Visualising VLM attention

Visualising the visualiser

Computer Vision
LLMs
Where does a VLM actually look at when looking at an image?
Author

Salman Naqvi

Published

Monday, 17 November 2025

In this notebook, I attempt to visualize how a VLM places attention on any input image, and on any input instruction.

This notebook follows the fastai style guide.

Important

Some of the the cell outputs have been deleted to keep the PDF concise. Rerun the notebook to see full outputs.

Setup

Traceback (most recent call last):
  File "/tmp/ipykernel_265136/256214625.py", line 5, in suplog
    try: yield
  File "/tmp/ipykernel_265136/256214625.py", line 7, in <module>
    with suplog(): 5/0
ZeroDivisionError: division by zero

This is a nifty snippet to keep errors in a notebook for future reference, but to allow myself to run all cells without halting execution.

Using a slow image processor as `use_fast` is unset and a slow processor was saved with this model. `use_fast=True` will be the default behavior in v4.50, even if the model was saved with a slow processor. This will result in minor differences in outputs. You'll still be able to use a slow processor with `use_fast=False`.
('eager', 'eager')
Qwen2_5_VLConfig {
  "architectures": [
    "Qwen2_5_VLForConditionalGeneration"
  ],
  "attention_dropout": 0.0,
  "bos_token_id": 151643,
  "eos_token_id": 151645,
  "hidden_act": "silu",
  "hidden_size": 2048,
  "image_token_id": 151655,
  "initializer_range": 0.02,
  "intermediate_size": 11008,
  "max_position_embeddings": 128000,
  "max_window_layers": 70,
  "model_type": "qwen2_5_vl",
  "num_attention_heads": 16,
  "num_hidden_layers": 36,
  "num_key_value_heads": 2,
  "rms_norm_eps": 1e-06,
  "rope_scaling": {
    "mrope_section": [
      16,
      24,
      24
    ],
    "rope_type": "default",
    "type": "default"
  },
  "rope_theta": 1000000.0,
  "sliding_window": 32768,
  "tie_word_embeddings": true,
  "torch_dtype": "bfloat16",
  "transformers_version": "4.50.0",
  "use_cache": true,
  "use_sliding_window": false,
  "video_token_id": 151656,
  "vision_config": {
    "depth": 32,
    "fullatt_block_indexes": [
      7,
      15,
      23,
      31
    ],
    "hidden_act": "silu",
    "hidden_size": 1280,
    "in_channels": 3,
    "in_chans": 3,
    "intermediate_size": 3420,
    "model_type": "qwen2_5_vl",
    "num_heads": 16,
    "out_hidden_size": 2048,
    "patch_size": 14,
    "spatial_merge_size": 2,
    "spatial_patch_size": 14,
    "temporal_patch_size": 2,
    "tokens_per_second": 2,
    "torch_dtype": "bfloat16",
    "window_size": 112
  },
  "vision_end_token_id": 151653,
  "vision_start_token_id": 151652,
  "vision_token_id": 151654,
  "vocab_size": 151936
}

Getting Started

In this notebook, I’ll be using the recent RoboBrain 2.0 model, which blows other VLMs out of the water. It’s a refreshing model, since the reason it performed so well was because the team worked directly with the data.

Future note to self: if too much attention is placed on the system prompt, remove it.

'<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>Describe the cat on the left.<think><|im_end|>\n<|im_start|>assistant\n'

I’ll remove the system prompt as it seems the VLM places a lot of attention there. If I leave it in, the visualizations become skewed.

'<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>Describe the cat on the left.<think><|im_end|>\n<|im_start|>assistant\n'

Over here, the image has been padded so that each dimension is a multiple of 14, which matches the kernel size.

([<PIL.Image.Image image mode=RGB size=644x476>], 14)
(46.0, 34.0)
{'input_ids': tensor([[151644,    872,    198, 151652, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151653,
          74785,    279,   8251,    389,    279,   2115,  15757,  26865,     29,
         151645,    198, 151644,  77091,    198]], device='cuda:0'), 'attention_mask': tensor([[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
         1, 1]], device='cuda:0'), 'pixel_values': tensor([[ 0.2515,  0.3099,  0.3391,  ..., -0.6270, -0.3995, -0.4990],
        [ 0.5143,  0.2807,  0.5581,  ..., -0.2857, -0.4137, -0.2573],
        [-0.0113,  0.0909, -0.0842,  ..., -1.0963, -1.0252, -0.9683],
        ...,
        [ 1.7114,  1.6238,  1.6238,  ...,  1.1505,  1.0652,  1.0225],
        [ 1.4486,  1.5800,  1.5216,  ..., -0.3426, -0.2146,  0.2688],
        [ 1.6530,  1.6676,  1.5508,  ..., -1.0678, -0.8545, -0.8830]],
       device='cuda:0'), 'image_grid_thw': tensor([[ 1, 34, 46]], device='cuda:0')}
dict_keys(['input_ids', 'attention_mask', 'pixel_values', 'image_grid_thw'])
torch.Size([1, 410])

Here I can see the tokens I might need to be aware off when visualizing the attention. If I keep them, it may skew the visualization as a lot of attention may be placed on those tokens.

Most likely, I need to be wary of the very first <|im_start|>, and set its attention to zero when visualizing.

['<|im_start|>',
 'user',
 'Ċ',
 '<|vision_start|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|vision_end|>',
 'Describe',
 'Ġthe',
 'Ġcat',
 'Ġon',
 'Ġthe',
 'Ġleft',
 '.<',
 'think',
 '>',
 '<|im_end|>',
 'Ċ',
 '<|im_start|>',
 'assistant',
 'Ċ']
GenerateDecoderOnlyOutput(sequences=tensor([[151644,    872,    198, 151652, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655,
         151655, 151655, 151655, 151655, 151655, 151655, 151655, 151655, 151653,
          74785,    279,   8251,    389,    279,   2115,  15757,  26865,     29,
         151645,    198, 151644,  77091,    198,    785,   8251,    389,    279,
           2115,    374,    264,  51735,  32775,   8251,  12233,    264,   6176,
           4772,     13,   1084,    374,  34644,   1495,  44669,    323,   7952,
            311,    387,  30367,     13, 151645]], device='cuda:0'), scores=None, logits=None, attentions=((tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4316e-01, 2.5684e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6530e-02, 5.0439e-01, 4.5923e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7796e-03, 1.6203e-03, 6.6109e-03,  ..., 1.9934e-01,
           0.0000e+00, 0.0000e+00],
          [1.4343e-03, 3.3360e-03, 1.1284e-02,  ..., 3.9771e-01,
           1.0059e-01, 0.0000e+00],
          [9.5129e-05, 2.2335e-03, 7.0953e-03,  ..., 4.6265e-02,
           1.7200e-01, 5.8154e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9141e-04, 2.5940e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8413e-06, 2.3186e-05, 8.0943e-05,  ..., 9.5459e-01,
           0.0000e+00, 0.0000e+00],
          [7.1168e-05, 2.8157e-04, 8.2245e-03,  ..., 5.3650e-02,
           5.0879e-01, 0.0000e+00],
          [1.0729e-06, 1.6928e-05, 6.3515e-04,  ..., 6.7616e-04,
           2.8763e-02, 9.5215e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1124e-04, 9.9902e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5359e-05, 4.3058e-04, 9.9951e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0000e+00,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0986e-02,
           9.8877e-01, 0.0000e+00],
          [6.4969e-06, 2.2650e-05, 4.0985e-02,  ..., 2.2650e-05,
           2.7609e-04, 4.9927e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9312e-02, 9.7070e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0651e-01, 1.9897e-01, 6.9434e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7684e-07, 6.0797e-06, 6.0797e-06,  ..., 9.9268e-01,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 8.3447e-07, 4.5121e-05,  ..., 4.0588e-03,
           9.9316e-01, 0.0000e+00],
          [8.1024e-03, 9.1858e-03, 1.9440e-02,  ..., 1.1795e-02,
           1.7151e-02, 7.6904e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5915e-02, 5.9143e-02, 9.2480e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0719e-02, 9.4604e-03, 1.0071e-02,  ..., 2.7374e-02,
           0.0000e+00, 0.0000e+00],
          [1.1459e-02, 1.0765e-02, 1.0109e-02,  ..., 7.8735e-03,
           1.2985e-02, 0.0000e+00],
          [2.3193e-03, 6.3057e-03, 4.6600e-02,  ..., 2.6283e-03,
           1.9424e-02, 1.5283e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3018e-01, 5.6982e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.5571e-01, 5.8643e-01, 5.8044e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0340e-02, 1.4427e-02, 6.8130e-03,  ..., 2.6947e-02,
           0.0000e+00, 0.0000e+00],
          [2.4475e-02, 1.3947e-02, 4.3221e-03,  ..., 3.9734e-02,
           3.6194e-02, 0.0000e+00],
          [8.5526e-03, 1.2444e-02, 1.3962e-03,  ..., 2.0508e-02,
           4.6234e-02, 3.3493e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 1.9934e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7510e-01, 2.6245e-01, 2.6245e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9380e-04, 1.0414e-03, 7.8125e-03,  ..., 5.6543e-01,
           0.0000e+00, 0.0000e+00],
          [7.8392e-04, 2.8667e-03, 1.6495e-02,  ..., 2.5806e-01,
           4.2554e-01, 0.0000e+00],
          [1.2255e-03, 3.5458e-03, 9.3384e-03,  ..., 4.4556e-02,
           1.1743e-01, 1.4380e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6387e-01, 3.6224e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0004e-01, 6.5234e-01, 2.4756e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0564e-05, 1.6093e-06, 5.9605e-08,  ..., 4.1504e-02,
           0.0000e+00, 0.0000e+00],
          [3.3140e-05, 7.4685e-05, 1.6928e-05,  ..., 7.0752e-01,
           1.8457e-01, 0.0000e+00],
          [2.0862e-06, 7.0572e-05, 1.6165e-04,  ..., 5.5756e-02,
           7.6953e-01, 1.6650e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4180e-01, 1.5820e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.5615e-01, 3.3740e-01, 1.0620e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0942e-03, 1.4763e-02, 3.6530e-02,  ..., 3.1250e-02,
           0.0000e+00, 0.0000e+00],
          [7.3433e-03, 7.5760e-03, 4.1595e-02,  ..., 1.4978e-01,
           2.3206e-01, 0.0000e+00],
          [1.1871e-02, 3.0792e-02, 3.6011e-02,  ..., 3.2776e-02,
           1.4917e-01, 6.9397e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 2.5830e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4263e-01, 2.6025e-01, 2.9712e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1372e-02, 2.9022e-02, 4.3579e-02,  ..., 9.5154e-02,
           0.0000e+00, 0.0000e+00],
          [8.6288e-03, 2.0523e-02, 4.0192e-02,  ..., 1.2769e-01,
           2.3315e-01, 0.0000e+00],
          [1.6680e-03, 7.0229e-03, 1.2039e-02,  ..., 5.1483e-02,
           1.5137e-01, 8.4229e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4697e-01, 3.5303e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9893e-01, 1.9434e-01, 4.0674e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.4289e-03, 1.4206e-02, 1.0095e-01,  ..., 6.1493e-03,
           0.0000e+00, 0.0000e+00],
          [1.7691e-03, 6.5956e-03, 5.2917e-02,  ..., 2.7451e-02,
           1.0944e-01, 0.0000e+00],
          [7.9880e-03, 1.3489e-02, 3.9795e-02,  ..., 2.8549e-02,
           7.0679e-02, 1.5076e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9160e-01, 1.0822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6890e-01, 4.6997e-01, 1.6113e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1887e-02, 3.3600e-02, 1.8692e-02,  ..., 3.4204e-01,
           0.0000e+00, 0.0000e+00],
          [4.3449e-03, 2.4033e-02, 2.5391e-02,  ..., 6.9580e-02,
           4.0649e-01, 0.0000e+00],
          [8.3590e-04, 7.3929e-03, 1.7977e-03,  ..., 1.6431e-01,
           2.7515e-01, 2.1606e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1826e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2510e-01, 2.0776e-01, 6.6956e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0966e-02, 1.4944e-03, 4.0627e-03,  ..., 7.6660e-02,
           0.0000e+00, 0.0000e+00],
          [2.2415e-02, 4.0817e-03, 1.0422e-02,  ..., 4.7455e-02,
           1.1298e-01, 0.0000e+00],
          [1.7786e-04, 7.8321e-05, 1.6844e-04,  ..., 1.3831e-01,
           6.2012e-01, 1.2207e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5146e-01, 2.4866e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5581e-01, 2.3828e-01, 3.0591e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9973e-03, 5.5733e-03, 8.0795e-03,  ..., 2.6321e-03,
           0.0000e+00, 0.0000e+00],
          [1.1795e-02, 5.0926e-03, 1.1040e-02,  ..., 2.2392e-03,
           1.3770e-01, 0.0000e+00],
          [7.9575e-03, 5.2261e-03, 7.1182e-03,  ..., 3.3321e-03,
           1.5173e-01, 4.3030e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6809e-01, 8.3203e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.2864e-01, 2.3962e-01, 5.3174e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9897e-02, 1.0612e-02, 1.5198e-02,  ..., 1.4935e-03,
           0.0000e+00, 0.0000e+00],
          [1.3489e-02, 7.1106e-03, 7.7782e-03,  ..., 3.8643e-03,
           8.2474e-03, 0.0000e+00],
          [3.9291e-03, 1.2903e-03, 1.0471e-03,  ..., 3.5706e-02,
           3.1891e-02, 1.0413e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1143e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0248e-02, 9.2383e-01, 5.5908e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6413e-02, 7.4339e-04, 2.4700e-04,  ..., 4.6460e-01,
           0.0000e+00, 0.0000e+00],
          [4.2763e-03, 1.2445e-03, 1.4544e-03,  ..., 5.0195e-01,
           1.7480e-01, 0.0000e+00],
          [4.2200e-05, 5.6505e-04, 2.3232e-03,  ..., 3.0609e-02,
           7.8955e-01, 1.3928e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1650e-01, 8.3618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2314e-01, 1.6394e-01, 1.1267e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6040e-02, 1.0773e-02, 4.5380e-02,  ..., 3.8818e-02,
           0.0000e+00, 0.0000e+00],
          [3.5645e-02, 7.0724e-03, 1.8051e-02,  ..., 2.4292e-02,
           1.2146e-01, 0.0000e+00],
          [1.6937e-02, 2.5177e-03, 1.3718e-02,  ..., 2.4078e-02,
           3.9062e-02, 1.7932e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7990e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3267e-01, 1.0986e-01, 6.5771e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6642e-02, 1.5879e-03, 3.3360e-03,  ..., 1.8213e-01,
           0.0000e+00, 0.0000e+00],
          [1.7288e-02, 4.7607e-03, 1.5732e-02,  ..., 2.2241e-01,
           2.8101e-01, 0.0000e+00],
          [1.3023e-02, 1.8110e-03, 1.0712e-02,  ..., 4.9530e-02,
           1.0162e-01, 4.6631e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6182e-01, 1.3831e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6216e-01, 1.4137e-02, 5.2393e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1128e-01, 2.6108e-02, 1.1072e-03,  ..., 3.9154e-02,
           0.0000e+00, 0.0000e+00],
          [2.5928e-01, 1.8272e-03, 1.2445e-01,  ..., 3.6001e-05,
           7.5745e-02, 0.0000e+00],
          [7.4072e-01, 4.9448e-04, 2.7351e-03,  ..., 3.5667e-03,
           3.2684e-02, 1.5419e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 2.3327e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 7.9956e-03, 1.2802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 1.0800e-04, 1.5259e-05,  ..., 6.1941e-04,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.0908e-05, 4.1056e-04,  ..., 1.1955e-02,
           2.4429e-02, 0.0000e+00],
          [8.3789e-01, 4.9233e-05, 1.1092e-04,  ..., 1.0651e-02,
           3.4058e-02, 9.6970e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.7275e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.5106e-02, 1.5381e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6846e-01, 1.4937e-04, 7.3969e-05,  ..., 5.6946e-02,
           0.0000e+00, 0.0000e+00],
          [7.7881e-01, 1.3530e-04, 2.4116e-04,  ..., 6.4564e-04,
           1.2840e-02, 0.0000e+00],
          [8.0615e-01, 8.6963e-05, 9.3079e-04,  ..., 1.5926e-04,
           6.0120e-03, 2.1149e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 1.0138e-03, 6.6071e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1885e-01, 4.9591e-03, 1.0330e-02,  ..., 7.1167e-02,
           0.0000e+00, 0.0000e+00],
          [6.8213e-01, 6.4316e-03, 6.5575e-03,  ..., 1.2457e-05,
           4.9408e-02, 0.0000e+00],
          [8.4082e-01, 1.0557e-03, 4.4098e-03,  ..., 4.2140e-05,
           6.8426e-04, 5.7983e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8291e-01, 1.7242e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 1.9501e-02, 2.6443e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0479e-01, 2.0683e-04, 9.8610e-04,  ..., 9.0456e-04,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.7376e-03, 2.8324e-03,  ..., 1.9646e-03,
           2.3438e-02, 0.0000e+00],
          [5.4688e-01, 3.4404e-04, 6.7139e-04,  ..., 2.3544e-04,
           1.5030e-02, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9586e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2334e-01, 2.8275e-02, 4.8370e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3701e-01, 9.6679e-05, 3.9649e-04,  ..., 6.2752e-04,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 1.2379e-03, 3.8052e-03,  ..., 6.2323e-04,
           3.2104e-02, 0.0000e+00],
          [7.5684e-01, 3.4761e-04, 1.7176e-03,  ..., 2.4796e-04,
           6.6757e-03, 2.1469e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1235e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 3.9482e-03, 4.6921e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6187e-01, 1.0431e-05, 2.2113e-05,  ..., 6.0081e-03,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 6.3539e-05, 5.8460e-04,  ..., 5.2452e-06,
           2.0981e-02, 0.0000e+00],
          [6.6162e-01, 1.7941e-05, 1.0166e-03,  ..., 4.7922e-04,
           1.4820e-03, 1.6098e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.0712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3374e-02, 8.1635e-03, 3.5715e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9541e-01, 5.9605e-06, 5.9605e-08,  ..., 2.6875e-03,
           0.0000e+00, 0.0000e+00],
          [9.2822e-01, 2.3007e-04, 1.2040e-05,  ..., 1.7654e-02,
           1.9470e-02, 0.0000e+00],
          [8.7891e-01, 2.4915e-05, 9.0420e-05,  ..., 3.6907e-03,
           1.0187e-01, 1.4641e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 4.0833e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2920e-01, 4.8859e-02, 2.1820e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3018e-01, 2.4247e-04, 5.3704e-05,  ..., 7.7972e-03,
           0.0000e+00, 0.0000e+00],
          [9.0625e-01, 1.1749e-03, 1.2217e-03,  ..., 3.5248e-02,
           1.9638e-02, 0.0000e+00],
          [9.0479e-01, 4.1747e-04, 2.6131e-04,  ..., 5.4359e-03,
           9.6512e-03, 1.0956e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 5.4993e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.1450e-01, 3.4241e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7510e-01, 3.3474e-04, 9.2220e-04,  ..., 3.5706e-03,
           0.0000e+00, 0.0000e+00],
          [7.4512e-01, 4.6825e-04, 6.5136e-04,  ..., 9.2621e-03,
           3.0869e-02, 0.0000e+00],
          [6.6162e-01, 2.3687e-04, 2.0623e-04,  ..., 4.4823e-03,
           3.7903e-02, 8.8806e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7028e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 3.2837e-02, 3.6957e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3994e-01, 5.5046e-03, 1.2650e-02,  ..., 5.8708e-03,
           0.0000e+00, 0.0000e+00],
          [6.9141e-01, 1.0643e-02, 1.0078e-02,  ..., 7.0496e-03,
           3.6865e-02, 0.0000e+00],
          [4.1699e-01, 5.6419e-03, 5.1384e-03,  ..., 5.6915e-03,
           5.7648e-02, 2.2400e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4824e-01, 5.1849e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0127e-01, 1.3257e-01, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.2445e-04, 1.0324e-04,  ..., 4.9927e-02,
           0.0000e+00, 0.0000e+00],
          [6.1768e-01, 2.5797e-04, 2.5535e-04,  ..., 9.6985e-02,
           9.6191e-02, 0.0000e+00],
          [5.4834e-01, 8.7798e-05, 1.6689e-04,  ..., 4.7272e-02,
           1.2732e-01, 1.2976e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3588e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 2.1469e-02, 3.3905e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.4255e-01, 9.2840e-04, 7.3147e-04,  ..., 2.2919e-02,
           0.0000e+00, 0.0000e+00],
          [3.0200e-01, 9.4652e-04, 1.0891e-03,  ..., 1.3756e-02,
           7.5684e-02, 0.0000e+00],
          [4.0796e-01, 1.8358e-03, 7.1335e-04,  ..., 1.2772e-02,
           7.2998e-02, 4.8004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 2.3392e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2568e-01, 1.6895e-01, 5.1270e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1123e-01, 5.3692e-04, 4.4227e-05,  ..., 4.5929e-02,
           0.0000e+00, 0.0000e+00],
          [3.7915e-01, 7.6771e-04, 4.3631e-05,  ..., 4.4525e-02,
           2.7710e-01, 0.0000e+00],
          [2.9761e-01, 1.4448e-04, 4.3511e-06,  ..., 6.6414e-03,
           4.4897e-01, 5.8441e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 3.1860e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 2.5665e-02, 1.9028e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8564e-01, 3.8195e-04, 1.2207e-04,  ..., 2.9812e-03,
           0.0000e+00, 0.0000e+00],
          [7.7637e-01, 1.3866e-03, 3.2234e-03,  ..., 6.1989e-03,
           1.0529e-02, 0.0000e+00],
          [7.8125e-01, 2.8181e-04, 6.3896e-05,  ..., 4.9210e-03,
           2.2171e-02, 3.0945e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2604e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 1.1498e-04, 3.6865e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9111e-01, 7.3552e-05, 1.1325e-06,  ..., 1.0364e-01,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 4.2260e-05, 1.4305e-06,  ..., 5.8949e-05,
           2.2446e-02, 0.0000e+00],
          [9.4385e-01, 1.8537e-05, 3.2544e-05,  ..., 1.0258e-04,
           3.6073e-04, 5.2582e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4014e-01, 1.1917e-02, 0.0000e+00,  ..., 6.0997e-03,
           6.0997e-03, 6.0997e-03],
          [2.8540e-01, 1.5497e-06, 7.4158e-03,  ..., 7.4425e-03,
           7.4425e-03, 0.0000e+00],
          ...,
          [9.3018e-01, 1.3933e-03, 3.7253e-05,  ..., 7.2021e-03,
           0.0000e+00, 0.0000e+00],
          [9.8779e-01, 2.1183e-04, 9.8348e-06,  ..., 4.1723e-07,
           6.9733e-03, 0.0000e+00],
          [9.8096e-01, 5.4538e-05, 4.5514e-04,  ..., 2.3842e-07,
           1.1921e-05, 1.3779e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-02, 4.1628e-04, 0.0000e+00,  ..., 4.4823e-03,
           0.0000e+00, 4.4823e-03],
          [1.7053e-01, 2.3842e-07, 8.7404e-04,  ..., 5.7526e-03,
           5.7526e-03, 0.0000e+00],
          ...,
          [9.9365e-01, 7.3791e-05, 5.3644e-07,  ..., 2.0618e-03,
           0.0000e+00, 0.0000e+00],
          [9.9512e-01, 2.8074e-05, 6.5565e-07,  ..., 4.1723e-07,
           3.9139e-03, 0.0000e+00],
          [9.9805e-01, 4.8876e-06, 7.9274e-06,  ..., 5.9605e-08,
           7.7486e-07, 1.5945e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7914e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0615e-01, 1.9080e-01, 3.0651e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5425e-01, 2.3782e-05, 1.3769e-05,  ..., 4.8859e-02,
           0.0000e+00, 0.0000e+00],
          [5.7080e-01, 1.1081e-04, 9.5487e-05,  ..., 8.8989e-02,
           9.0332e-02, 0.0000e+00],
          [7.5342e-01, 3.3808e-04, 2.2292e-05,  ..., 2.7664e-02,
           1.3538e-01, 1.3496e-02]],

         [[5.9277e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5662e-01, 3.0994e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.4082e-01, 6.5422e-04, 6.8045e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6719e-01, 1.0473e-04, 8.1539e-05,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [7.0752e-01, 3.7422e-03, 7.0286e-04,  ..., 1.1086e-05,
           3.4149e-02, 0.0000e+00],
          [9.2480e-01, 1.4150e-04, 4.0627e-04,  ..., 1.2755e-05,
           9.4843e-04, 1.4572e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 9.0027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.7802e-02, 4.3907e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7612e-01, 4.2038e-03, 8.9455e-04,  ..., 1.2268e-01,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 1.8911e-03, 6.9571e-04,  ..., 6.1531e-03,
           1.8356e-02, 0.0000e+00],
          [8.2227e-01, 1.4400e-03, 2.4021e-04,  ..., 1.5097e-03,
           1.7242e-02, 1.7166e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5106e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0869e-01, 8.5022e-02, 6.3057e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1162e-01, 7.6151e-04, 2.4056e-04,  ..., 6.1951e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 4.3793e-03, 5.4407e-04,  ..., 4.9324e-03,
           2.0966e-02, 0.0000e+00],
          [8.6230e-01, 1.1921e-03, 3.2592e-04,  ..., 4.8943e-03,
           1.5625e-02, 2.9266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.7042e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1309e-01, 3.2806e-02, 5.4230e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6865e-01, 2.5034e-05, 5.9307e-05,  ..., 6.3599e-02,
           0.0000e+00, 0.0000e+00],
          [8.2520e-01, 4.2796e-04, 8.6975e-04,  ..., 1.3519e-02,
           9.7107e-02, 0.0000e+00],
          [8.7354e-01, 5.1320e-05, 1.1891e-04,  ..., 3.2444e-03,
           1.7273e-02, 4.6082e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9471e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2021e-01, 2.6611e-01, 1.3817e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8770e-01, 1.6451e-04, 4.8816e-05,  ..., 7.7698e-02,
           0.0000e+00, 0.0000e+00],
          [8.7500e-01, 3.3140e-04, 1.4877e-04,  ..., 3.1910e-03,
           7.4463e-02, 0.0000e+00],
          [8.7830e-02, 3.7050e-04, 4.4584e-05,  ..., 3.6030e-03,
           8.9355e-01, 3.5515e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9805e-01, 1.9121e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 8.2703e-03, 7.5302e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4893e-01, 5.5313e-05, 1.3638e-04,  ..., 4.8523e-02,
           0.0000e+00, 0.0000e+00],
          [5.1904e-01, 2.7776e-04, 1.0710e-03,  ..., 2.8934e-03,
           1.1650e-02, 0.0000e+00],
          [6.9629e-01, 1.5771e-04, 2.1303e-04,  ..., 1.9398e-03,
           7.5378e-03, 2.8076e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5125e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6914e-01, 7.2327e-02, 5.8502e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0566e-01, 5.9414e-04, 1.3161e-03,  ..., 6.1005e-02,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 7.0047e-04, 2.0945e-04,  ..., 1.6546e-03,
           2.1942e-02, 0.0000e+00],
          [5.6152e-01, 7.2432e-04, 1.9264e-04,  ..., 4.8180e-03,
           3.0151e-02, 6.9458e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 1.0078e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 2.3755e-01, 2.3361e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 2.7001e-05, 5.4359e-05,  ..., 3.2330e-03,
           0.0000e+00, 0.0000e+00],
          [9.5361e-01, 7.1645e-05, 4.4823e-05,  ..., 2.8782e-03,
           9.6970e-03, 0.0000e+00],
          [9.0820e-01, 1.0967e-04, 1.7285e-05,  ..., 3.7174e-03,
           4.5105e-02, 1.1147e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.3030e-02, 6.8016e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5293e-01, 1.0973e-04, 2.9743e-05,  ..., 2.5511e-04,
           0.0000e+00, 0.0000e+00],
          [6.3574e-01, 1.1864e-03, 6.1083e-04,  ..., 1.9722e-03,
           2.6718e-02, 0.0000e+00],
          [7.1436e-01, 3.7956e-04, 1.0496e-04,  ..., 1.6937e-03,
           1.3695e-02, 1.4694e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9932e-01, 1.9336e-01, 7.4959e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1406e-01, 6.3467e-04, 2.1601e-04,  ..., 4.5013e-04,
           0.0000e+00, 0.0000e+00],
          [5.8789e-01, 3.4122e-03, 4.3225e-04,  ..., 7.0457e-03,
           2.9953e-02, 0.0000e+00],
          [4.8145e-01, 5.6763e-03, 2.9516e-04,  ..., 6.5231e-03,
           4.9438e-02, 2.0691e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-01, 1.0638e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9126e-01, 6.4746e-01, 6.1310e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8909e-01, 5.7077e-04, 1.4858e-03,  ..., 2.2385e-02,
           0.0000e+00, 0.0000e+00],
          [3.4790e-01, 3.4618e-03, 1.7891e-03,  ..., 9.7809e-03,
           7.8735e-02, 0.0000e+00],
          [6.1328e-01, 1.4267e-03, 8.6164e-04,  ..., 1.4519e-02,
           7.0190e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6172e-01, 2.3853e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 1.8054e-01, 8.2153e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9775e-01, 8.5211e-04, 6.9284e-04,  ..., 2.2202e-03,
           0.0000e+00, 0.0000e+00],
          [3.3130e-01, 6.7482e-03, 4.0779e-03,  ..., 5.2023e-04,
           1.5572e-02, 0.0000e+00],
          [5.9082e-01, 1.8854e-03, 1.0052e-03,  ..., 5.8413e-04,
           7.5760e-03, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.7449e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1602e-01, 7.2327e-02, 1.1589e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7646e-01, 2.4140e-05, 1.0002e-04,  ..., 1.0023e-03,
           0.0000e+00, 0.0000e+00],
          [1.2634e-02, 5.4240e-06, 1.3173e-05,  ..., 1.8167e-04,
           1.1021e-04, 0.0000e+00],
          [1.3664e-02, 2.1577e-05, 1.5199e-05,  ..., 2.6774e-04,
           2.7370e-04, 5.4550e-04]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 1.7456e-02, 7.4501e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 5.0962e-05, 2.0564e-05,  ..., 1.6670e-03,
           0.0000e+00, 0.0000e+00],
          [8.4082e-01, 3.1257e-04, 1.5771e-04,  ..., 2.6684e-03,
           8.6670e-03, 0.0000e+00],
          [9.1406e-01, 6.3419e-05, 1.3173e-05,  ..., 1.8902e-03,
           2.7943e-03, 3.4904e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7852e-01, 2.1271e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 9.6054e-03, 1.4503e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0176e-01, 5.0783e-04, 9.0170e-04,  ..., 2.2461e-01,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 3.3069e-04, 5.3549e-04,  ..., 4.0321e-03,
           1.2891e-01, 0.0000e+00],
          [2.4939e-01, 2.5196e-03, 7.6828e-03,  ..., 6.1321e-04,
           6.0539e-03, 1.6101e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.0403e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7461e-01, 2.1423e-02, 3.8376e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3291e-01, 2.2054e-05, 2.7478e-05,  ..., 1.3718e-02,
           0.0000e+00, 0.0000e+00],
          [1.6650e-01, 9.7096e-05, 1.0788e-04,  ..., 2.1782e-03,
           1.8740e-03, 0.0000e+00],
          [9.0759e-02, 4.7922e-05, 6.3717e-05,  ..., 1.4668e-03,
           6.3896e-04, 4.7340e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3311e-01, 6.6895e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 6.8262e-01, 7.6065e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9258e-01, 2.5606e-04, 2.4533e-04,  ..., 1.0010e-02,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 6.8617e-04, 4.6253e-05,  ..., 5.0659e-03,
           6.7558e-03, 0.0000e+00],
          [8.6035e-01, 8.8549e-04, 1.8454e-04,  ..., 8.0795e-03,
           3.2898e-02, 3.6564e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1543e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 2.8534e-02, 3.5980e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7246e-01, 4.6086e-04, 2.2736e-03,  ..., 3.1952e-02,
           0.0000e+00, 0.0000e+00],
          [3.1982e-01, 3.9983e-04, 7.0429e-04,  ..., 2.1912e-02,
           1.4880e-01, 0.0000e+00],
          [6.5723e-01, 1.6463e-04, 4.4918e-04,  ..., 6.2370e-03,
           5.2948e-02, 4.5868e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6582e-01, 3.4332e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 3.0884e-01, 1.6479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.0830e-01, 2.9802e-05, 2.9087e-05,  ..., 5.5328e-02,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 4.9353e-05, 2.5809e-05,  ..., 6.5918e-03,
           1.8860e-02, 0.0000e+00],
          [4.5508e-01, 2.8431e-05, 6.5148e-05,  ..., 2.7649e-02,
           2.8003e-01, 2.5620e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 1.0901e-01, 7.2098e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.8672e-01, 1.3340e-04, 3.1638e-04,  ..., 1.2199e-02,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.3328e-04, 4.2701e-04,  ..., 4.9438e-03,
           1.2604e-02, 0.0000e+00],
          [7.1680e-01, 5.5122e-04, 9.0218e-04,  ..., 7.4043e-03,
           2.0370e-02, 1.4412e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 6.3965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1299e-01, 1.5271e-01, 3.4424e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4805e-01, 6.8617e-04, 5.6887e-04,  ..., 7.7400e-03,
           0.0000e+00, 0.0000e+00],
          [5.4590e-01, 7.1716e-04, 4.5061e-04,  ..., 2.5043e-03,
           3.8391e-02, 0.0000e+00],
          [4.1138e-01, 7.5769e-04, 5.5122e-04,  ..., 7.8888e-03,
           3.5950e-02, 2.4463e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8223e-01, 1.9971e-01, 1.7838e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.8535e-01, 3.3903e-04, 5.1141e-05,  ..., 4.3411e-03,
           0.0000e+00, 0.0000e+00],
          [9.0918e-01, 3.2973e-04, 2.6393e-04,  ..., 4.6654e-03,
           8.6975e-03, 0.0000e+00],
          [7.1631e-01, 3.0112e-04, 2.4796e-04,  ..., 1.7456e-02,
           1.3623e-01, 1.1559e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.4027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9609e-01, 2.8114e-03, 9.2840e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8555e-01, 8.1062e-06, 1.2696e-05,  ..., 1.6336e-03,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 3.6955e-06, 1.0550e-05,  ..., 8.4019e-04,
           2.7218e-03, 0.0000e+00],
          [9.8584e-01, 1.8477e-06, 4.7684e-06,  ..., 3.2544e-04,
           1.0405e-03, 9.0170e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9673e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1787e-02, 4.0829e-05, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.1765e-01, 2.1935e-05, 1.6966e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1689e-01, 1.3232e-04, 2.7156e-04,  ..., 6.6948e-04,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.3403e-03, 4.0054e-04,  ..., 3.9339e-05,
           8.3069e-02, 0.0000e+00],
          [5.6006e-01, 2.9874e-04, 6.7825e-03,  ..., 9.3222e-05,
           3.6740e-04, 8.0505e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1024e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4043e-01, 5.7312e-02, 2.0084e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2073e-01, 2.0087e-05, 6.2823e-05,  ..., 2.6672e-02,
           0.0000e+00, 0.0000e+00],
          [6.5771e-01, 3.2234e-04, 1.2970e-04,  ..., 2.0618e-03,
           1.0925e-02, 0.0000e+00],
          [6.3770e-01, 1.2201e-04, 4.3511e-05,  ..., 9.6512e-03,
           1.2482e-01, 1.4923e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.8763e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 3.1311e-02, 1.4671e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.3916e-01, 1.1921e-07, 4.7684e-07,  ..., 1.8148e-03,
           0.0000e+00, 0.0000e+00],
          [4.6484e-01, 1.4472e-04, 1.6689e-05,  ..., 3.3200e-05,
           8.6441e-03, 0.0000e+00],
          [2.9907e-01, 6.7949e-06, 5.8651e-05,  ..., 4.6134e-04,
           7.6675e-04, 2.6428e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.3400e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.0236e-01, 2.0950e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 1.7107e-04, 7.5042e-05,  ..., 5.9938e-04,
           0.0000e+00, 0.0000e+00],
          [4.9146e-01, 1.0958e-03, 3.8099e-04,  ..., 4.3106e-03,
           2.7420e-02, 0.0000e+00],
          [7.5195e-01, 8.4496e-04, 4.6468e-04,  ..., 4.1847e-03,
           1.5030e-02, 9.9106e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5736e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 9.3460e-03, 1.6449e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7041e-01, 3.4904e-04, 5.0783e-04,  ..., 6.4453e-02,
           0.0000e+00, 0.0000e+00],
          [4.9292e-01, 8.6308e-04, 1.3876e-03,  ..., 2.7893e-02,
           5.6244e-02, 0.0000e+00],
          [4.6387e-01, 2.5005e-03, 2.8687e-03,  ..., 2.8030e-02,
           4.1016e-02, 4.9316e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7617e-01, 4.2383e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2412e-01, 2.5879e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1604e-02, 2.2449e-03, 8.2397e-04,  ..., 8.5144e-03,
           0.0000e+00, 0.0000e+00],
          [1.9751e-01, 1.2169e-03, 8.7452e-04,  ..., 8.4448e-04,
           1.6495e-02, 0.0000e+00],
          [2.4634e-01, 1.4076e-03, 9.1314e-04,  ..., 4.4250e-03,
           2.7069e-02, 4.2358e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9473e-01, 4.0552e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7139e-01, 4.9591e-02, 2.7881e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5867e-02, 1.7405e-05, 1.7047e-05,  ..., 8.8867e-01,
           0.0000e+00, 0.0000e+00],
          [4.4189e-01, 6.7353e-06, 3.3379e-05,  ..., 1.9093e-03,
           4.9097e-01, 0.0000e+00],
          [4.2578e-01, 5.2452e-06, 3.7134e-05,  ..., 3.0537e-03,
           3.4607e-02, 4.6582e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7725e-01, 3.2251e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3657e-01, 4.0552e-01, 3.5791e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3516e-01, 1.9395e-04, 2.3019e-04,  ..., 2.6646e-03,
           0.0000e+00, 0.0000e+00],
          [5.4077e-02, 8.0681e-04, 3.1223e-03,  ..., 1.2665e-03,
           2.2812e-02, 0.0000e+00],
          [1.4929e-01, 1.8950e-03, 4.1046e-03,  ..., 2.4090e-03,
           2.1851e-02, 3.6896e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9166e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.1879e-02, 3.6087e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4902e-01, 3.9864e-04, 4.1676e-04,  ..., 5.4054e-03,
           0.0000e+00, 0.0000e+00],
          [7.6843e-02, 6.7520e-04, 4.6611e-04,  ..., 8.4534e-03,
           4.6265e-02, 0.0000e+00],
          [4.5288e-01, 1.0509e-03, 8.3017e-04,  ..., 3.6240e-03,
           2.4384e-02, 5.3680e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.5725e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.2573e-02, 1.6556e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7427e-01, 4.9591e-05, 2.1207e-04,  ..., 3.5645e-02,
           0.0000e+00, 0.0000e+00],
          [1.2091e-01, 4.7517e-04, 8.7214e-04,  ..., 6.4354e-03,
           2.5845e-03, 0.0000e+00],
          [3.2642e-01, 3.0828e-04, 7.9775e-04,  ..., 4.4174e-03,
           6.0310e-03, 1.9257e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4549e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6865e-01, 1.1395e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1310e-02, 2.2054e-06, 5.1856e-06,  ..., 4.1046e-03,
           0.0000e+00, 0.0000e+00],
          [2.9834e-01, 1.4734e-04, 2.0862e-04,  ..., 1.5808e-02,
           6.8169e-03, 0.0000e+00],
          [3.1152e-01, 2.2471e-05, 4.5598e-05,  ..., 2.9907e-02,
           1.5686e-02, 1.0612e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1934e-01, 1.8079e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 1.9791e-02, 3.3142e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1350e-02, 4.1366e-05, 1.7226e-04,  ..., 7.2656e-01,
           0.0000e+00, 0.0000e+00],
          [4.2700e-01, 1.0288e-04, 7.7248e-04,  ..., 2.0676e-03,
           4.8267e-01, 0.0000e+00],
          [3.5718e-01, 2.0742e-05, 2.3258e-04,  ..., 2.7523e-03,
           2.9877e-02, 5.6250e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8584e-01, 5.1416e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3628e-01, 1.7786e-01, 3.8574e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4259e-02, 3.5286e-05, 6.6936e-05,  ..., 8.8770e-01,
           0.0000e+00, 0.0000e+00],
          [1.4587e-01, 1.3125e-04, 3.7265e-04,  ..., 2.9016e-04,
           7.6611e-01, 0.0000e+00],
          [1.0394e-01, 1.3053e-04, 2.2340e-04,  ..., 1.4601e-03,
           1.3318e-01, 6.4941e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8926e-01, 1.0590e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 4.0131e-02, 1.2756e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7078e-01, 9.1314e-05, 1.1140e-04,  ..., 6.3610e-04,
           0.0000e+00, 0.0000e+00],
          [8.7891e-01, 3.1328e-04, 1.3189e-03,  ..., 5.3883e-04,
           8.9569e-03, 0.0000e+00],
          [8.1641e-01, 2.6131e-04, 9.4891e-04,  ..., 3.8567e-03,
           1.4427e-02, 8.2245e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1055e-01, 1.8921e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.7930e-01, 1.3416e-01, 8.6365e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2544e-01, 2.0683e-05, 1.8120e-05,  ..., 3.9429e-02,
           0.0000e+00, 0.0000e+00],
          [1.4320e-02, 9.8765e-05, 9.3699e-04,  ..., 4.5598e-05,
           7.5836e-03, 0.0000e+00],
          [1.9165e-01, 1.5199e-04, 5.6887e-04,  ..., 8.0729e-04,
           6.0577e-03, 1.3191e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 2.6465e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5459e-01, 5.9662e-02, 4.8584e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4290e-02, 1.0133e-06, 5.7220e-06,  ..., 9.6143e-01,
           0.0000e+00, 0.0000e+00],
          [1.4404e-01, 1.2696e-05, 8.8394e-05,  ..., 2.3918e-03,
           8.0811e-01, 0.0000e+00],
          [1.2042e-01, 7.8082e-06, 1.8895e-05,  ..., 2.3537e-03,
           3.3752e-02, 8.0566e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 4.4861e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2617e-01, 5.9906e-02, 1.1414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.5625e-01, 4.6611e-04, 1.7691e-04,  ..., 2.2766e-02,
           0.0000e+00, 0.0000e+00],
          [1.6797e-01, 2.1982e-04, 6.8378e-04,  ..., 9.1124e-04,
           4.9951e-01, 0.0000e+00],
          [1.6870e-01, 1.7023e-04, 6.9332e-04,  ..., 2.2087e-03,
           2.1277e-01, 2.6733e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 1.7792e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 9.8877e-03, 5.1994e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4473e-01, 6.8069e-05, 1.7166e-05,  ..., 1.3332e-03,
           0.0000e+00, 0.0000e+00],
          [8.3740e-01, 4.2510e-04, 1.2913e-03,  ..., 1.5039e-03,
           1.1263e-03, 0.0000e+00],
          [8.7988e-01, 1.8942e-04, 4.3869e-04,  ..., 2.4128e-03,
           7.1335e-04, 2.0199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4351e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 4.6768e-03, 2.6245e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1228e-01, 3.1924e-04, 3.3069e-04,  ..., 7.8735e-03,
           0.0000e+00, 0.0000e+00],
          [7.4707e-01, 4.4227e-04, 7.5531e-04,  ..., 1.3332e-03,
           1.5869e-02, 0.0000e+00],
          [4.3018e-01, 9.6893e-04, 7.5150e-04,  ..., 2.6226e-03,
           3.8025e-02, 2.9739e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3071e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4014e-01, 2.9126e-01, 6.8481e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0605e-01, 2.4617e-05, 1.2982e-04,  ..., 9.9564e-03,
           0.0000e+00, 0.0000e+00],
          [9.3945e-01, 9.3877e-05, 1.0353e-04,  ..., 1.6842e-03,
           1.0338e-02, 0.0000e+00],
          [7.6611e-01, 3.7372e-05, 6.1631e-05,  ..., 1.7910e-03,
           9.2834e-02, 1.8341e-02]],

         [[8.4814e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3756e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0801e-01, 1.7929e-03, 2.6283e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7031e-01, 2.7776e-04, 5.1117e-04,  ..., 1.7838e-02,
           0.0000e+00, 0.0000e+00],
          [6.3477e-01, 8.8072e-04, 2.7943e-03,  ..., 6.0499e-05,
           1.5671e-02, 0.0000e+00],
          [6.3330e-01, 4.7302e-04, 1.1730e-03,  ..., 5.7745e-04,
           2.5520e-03, 3.3173e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8340e-01, 1.6479e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4502e-01, 3.3179e-01, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8467e-01, 2.9385e-05, 1.5497e-06,  ..., 2.1713e-02,
           0.0000e+00, 0.0000e+00],
          [7.5244e-01, 1.4746e-04, 5.6863e-05,  ..., 1.2091e-01,
           5.4993e-02, 0.0000e+00],
          [4.6704e-01, 5.2452e-06, 5.4896e-05,  ..., 9.9411e-03,
           4.4922e-01, 6.9092e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.1569e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 1.1719e-02, 1.2405e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7437e-01, 1.4515e-03, 1.6584e-03,  ..., 5.3635e-03,
           0.0000e+00, 0.0000e+00],
          [6.8262e-01, 3.9024e-03, 5.1613e-03,  ..., 1.5554e-03,
           3.5801e-03, 0.0000e+00],
          [7.0068e-01, 1.6909e-03, 3.3092e-03,  ..., 3.2654e-03,
           7.6065e-03, 1.9516e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 2.0920e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9971e-01, 2.8149e-01, 1.8967e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1826e-01, 3.2663e-05, 3.0696e-05,  ..., 3.0029e-02,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 9.1314e-05, 5.6505e-05,  ..., 5.1453e-02,
           2.7679e-02, 0.0000e+00],
          [7.5781e-01, 3.3617e-05, 6.1095e-05,  ..., 8.0383e-02,
           7.8125e-02, 1.8143e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.3940e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 8.9722e-03, 8.7738e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 6.7890e-05, 1.7107e-05,  ..., 6.9656e-03,
           0.0000e+00, 0.0000e+00],
          [3.7109e-01, 1.6394e-03, 1.2894e-03,  ..., 6.5193e-03,
           1.8753e-02, 0.0000e+00],
          [7.4268e-01, 6.4278e-04, 2.2960e-04,  ..., 5.7411e-03,
           1.1101e-03, 5.6305e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 1.9958e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8145e-01, 9.2239e-03, 9.2010e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8684e-02, 2.3496e-04, 3.6311e-04,  ..., 5.5695e-03,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.6751e-04, 8.2731e-05,  ..., 1.9133e-05,
           1.7529e-03, 0.0000e+00],
          [9.1699e-01, 3.1805e-04, 1.5104e-04,  ..., 2.7847e-04,
           8.2321e-03, 1.6754e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 4.8157e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 3.2043e-02, 1.6205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8833e-01, 2.7061e-04, 1.5426e-04,  ..., 1.3733e-02,
           0.0000e+00, 0.0000e+00],
          [9.1992e-01, 2.3327e-03, 5.4264e-04,  ..., 3.8147e-05,
           4.8370e-03, 0.0000e+00],
          [5.8740e-01, 1.7452e-03, 6.4516e-04,  ..., 4.7016e-04,
           6.7078e-02, 4.1748e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 2.5131e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0322e-01, 1.5112e-01, 4.5593e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.0312e-01, 5.7936e-05, 5.5492e-05,  ..., 3.1143e-02,
           0.0000e+00, 0.0000e+00],
          [1.8848e-01, 2.6608e-04, 2.7466e-04,  ..., 7.1655e-02,
           8.5938e-02, 0.0000e+00],
          [3.4033e-01, 8.7857e-05, 1.3554e-04,  ..., 6.2683e-02,
           6.1646e-02, 6.0150e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9272e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 8.3252e-02, 2.6749e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6113e-01, 2.6166e-05, 1.4365e-05,  ..., 3.9276e-02,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 6.6948e-04, 9.4461e-04,  ..., 7.1594e-02,
           3.6346e-02, 0.0000e+00],
          [7.3877e-01, 3.4642e-04, 2.6774e-04,  ..., 3.2867e-02,
           7.9346e-03, 7.5684e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.2797e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9941e-01, 6.1523e-02, 3.8879e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0400e-01, 4.3631e-05, 1.0425e-04,  ..., 3.0319e-02,
           0.0000e+00, 0.0000e+00],
          [8.1494e-01, 1.2999e-03, 3.9978e-03,  ..., 3.5839e-03,
           1.9257e-02, 0.0000e+00],
          [7.3340e-01, 4.8923e-04, 9.4652e-04,  ..., 3.1338e-03,
           5.0446e-02, 5.6427e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.0303e-01, 3.9697e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4221e-01, 3.9337e-02, 8.1836e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7842e-02, 6.7353e-06, 2.9504e-05,  ..., 8.4814e-01,
           0.0000e+00, 0.0000e+00],
          [4.2786e-02, 1.9729e-05, 7.5281e-05,  ..., 1.4610e-03,
           9.3652e-01, 0.0000e+00],
          [6.3721e-02, 3.8743e-06, 3.0339e-05,  ..., 1.8616e-03,
           2.9404e-02, 8.7988e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 1.8591e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3384e-01, 5.2148e-01, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2583e-01, 7.0751e-05, 6.4909e-05,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [7.2070e-01, 3.2783e-04, 1.1355e-04,  ..., 2.7969e-02,
           5.9753e-02, 0.0000e+00],
          [4.5215e-01, 1.0467e-04, 6.5386e-05,  ..., 1.9348e-02,
           1.2854e-01, 2.0947e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6025e-01, 2.3975e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9600e-01, 3.7933e-02, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3481e-01, 3.3970e-03, 3.4695e-03,  ..., 1.1940e-02,
           0.0000e+00, 0.0000e+00],
          [5.0244e-01, 2.7161e-02, 3.9612e-02,  ..., 2.0828e-03,
           4.4098e-02, 0.0000e+00],
          [5.2881e-01, 8.4763e-03, 1.0414e-02,  ..., 4.0359e-03,
           2.7756e-02, 8.0017e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6826e-01, 3.1921e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6621e-01, 9.6008e-02, 3.7659e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9385e-01, 1.1683e-05, 8.5235e-06,  ..., 1.1147e-02,
           0.0000e+00, 0.0000e+00],
          [2.7637e-01, 7.2975e-03, 4.6005e-03,  ..., 2.8496e-03,
           6.8016e-03, 0.0000e+00],
          [2.9980e-01, 8.8263e-04, 7.1049e-04,  ..., 5.3368e-03,
           1.2199e-02, 3.0777e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8047e-01, 1.9516e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 4.6806e-03, 4.1695e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9590e-01, 2.6882e-05, 3.3140e-05,  ..., 7.3929e-03,
           0.0000e+00, 0.0000e+00],
          [3.5767e-01, 5.5084e-03, 3.3417e-03,  ..., 2.5868e-05,
           2.0798e-02, 0.0000e+00],
          [7.7979e-01, 5.6791e-04, 6.1655e-04,  ..., 5.0449e-04,
           3.3665e-03, 1.4374e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9185e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.5375e-02, 1.3771e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2189e-01, 2.9325e-05, 8.8811e-06,  ..., 2.8305e-03,
           0.0000e+00, 0.0000e+00],
          [2.9831e-02, 4.4918e-04, 4.0293e-04,  ..., 4.7326e-04,
           1.0452e-02, 0.0000e+00],
          [6.4148e-02, 1.5223e-04, 1.0306e-04,  ..., 4.7951e-03,
           4.9400e-04, 5.5199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0820e-01, 9.2041e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6938e-01, 4.2505e-01, 2.0557e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0015e-01, 1.0192e-05, 6.6757e-06,  ..., 5.6953e-03,
           0.0000e+00, 0.0000e+00],
          [1.7822e-01, 3.3021e-04, 3.8624e-04,  ..., 7.3853e-03,
           3.6346e-02, 0.0000e+00],
          [3.0176e-01, 3.0756e-04, 3.2878e-04,  ..., 5.8861e-03,
           5.2032e-02, 4.1504e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 2.0859e-02, 2.0172e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0654e-01, 1.1480e-04, 6.2346e-05,  ..., 6.5002e-02,
           0.0000e+00, 0.0000e+00],
          [4.4873e-01, 2.7637e-03, 4.4708e-03,  ..., 3.7785e-03,
           1.0254e-01, 0.0000e+00],
          [4.8560e-01, 1.6909e-03, 1.8425e-03,  ..., 7.4310e-03,
           2.7710e-02, 5.9357e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1982e-01, 1.8005e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4075e-01, 6.6895e-01, 1.9019e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.2571e-02, 3.1590e-05, 3.2783e-06,  ..., 2.4323e-02,
           0.0000e+00, 0.0000e+00],
          [2.5955e-02, 4.2772e-04, 3.8195e-04,  ..., 3.9787e-03,
           1.4000e-02, 0.0000e+00],
          [7.0534e-03, 3.0279e-04, 6.9737e-05,  ..., 7.6866e-03,
           6.2752e-03, 1.2383e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4883e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 2.5894e-02, 4.5700e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7876e-01, 8.5831e-06, 1.0669e-05,  ..., 1.1864e-03,
           0.0000e+00, 0.0000e+00],
          [3.5181e-01, 9.8610e-04, 8.5163e-04,  ..., 4.8714e-03,
           7.3280e-03, 0.0000e+00],
          [3.6182e-01, 4.0913e-04, 3.0494e-04,  ..., 4.5395e-03,
           6.0577e-03, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7089e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9512e-01, 3.7671e-01, 1.2817e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8197e-02, 7.2718e-06, 3.8147e-06,  ..., 5.8517e-03,
           0.0000e+00, 0.0000e+00],
          [5.8008e-01, 1.2808e-03, 3.0422e-03,  ..., 7.8821e-04,
           6.6833e-02, 0.0000e+00],
          [1.1591e-01, 8.4591e-04, 8.6594e-04,  ..., 1.4477e-03,
           4.3976e-02, 3.5889e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5205e-01, 1.4807e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 3.8544e-02, 2.7390e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0225e-01, 5.3704e-05, 6.3956e-05,  ..., 3.2654e-03,
           0.0000e+00, 0.0000e+00],
          [3.5474e-01, 2.9697e-03, 1.7807e-02,  ..., 2.8753e-04,
           1.0864e-01, 0.0000e+00],
          [5.7129e-01, 2.0599e-03, 5.7335e-03,  ..., 7.6914e-04,
           1.6785e-02, 4.0283e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1113e-01, 8.8989e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 3.6831e-03, 1.8196e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5854e-01, 1.5478e-03, 2.1458e-03,  ..., 2.2507e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 3.7785e-03, 1.7462e-03,  ..., 3.5524e-05,
           7.5134e-02, 0.0000e+00],
          [1.2891e-01, 2.1935e-03, 2.9202e-03,  ..., 5.0366e-05,
           5.3613e-01, 2.4506e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.2292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.3941e-02, 1.5205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0039e-02, 4.0829e-05, 3.5167e-05,  ..., 9.9659e-04,
           0.0000e+00, 0.0000e+00],
          [6.1914e-01, 4.3945e-03, 2.2850e-03,  ..., 3.5834e-04,
           2.0203e-02, 0.0000e+00],
          [2.9346e-01, 2.2659e-03, 2.2411e-03,  ..., 1.8721e-03,
           6.6650e-02, 5.1392e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0430e-01, 9.5520e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 2.3438e-01, 3.5248e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6450e-01, 1.9193e-05, 1.6570e-05,  ..., 1.6418e-02,
           0.0000e+00, 0.0000e+00],
          [9.5154e-02, 8.0729e-04, 7.1812e-04,  ..., 1.3336e-02,
           2.1072e-02, 0.0000e+00],
          [2.7637e-01, 5.4979e-04, 7.0572e-04,  ..., 1.3496e-02,
           3.4729e-02, 5.2765e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 5.6458e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2578e-01, 4.1870e-02, 3.2440e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8301e-01, 3.9482e-04, 2.3222e-04,  ..., 2.5421e-02,
           0.0000e+00, 0.0000e+00],
          [4.8047e-01, 8.4610e-03, 1.5152e-02,  ..., 2.6627e-03,
           2.4490e-02, 0.0000e+00],
          [6.0059e-01, 2.7161e-03, 3.7785e-03,  ..., 6.0387e-03,
           2.3560e-02, 5.4352e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 1.1539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 7.8142e-05, 2.1954e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7803e-01, 5.6446e-05, 8.7738e-05,  ..., 4.8399e-04,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 9.0599e-05, 3.1495e-04,  ..., 7.8678e-05,
           4.3091e-02, 0.0000e+00],
          [9.9023e-01, 1.3995e-04, 7.5483e-04,  ..., 3.1590e-05,
           5.1594e-04, 1.7796e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.1643e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 3.2711e-03, 2.4147e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9541e-01, 4.4727e-04, 5.5075e-04,  ..., 2.0466e-03,
           0.0000e+00, 0.0000e+00],
          [8.5596e-01, 8.0633e-04, 1.2493e-03,  ..., 1.3151e-03,
           1.7593e-02, 0.0000e+00],
          [8.8428e-01, 2.3603e-04, 4.8137e-04,  ..., 1.2245e-03,
           6.9504e-03, 1.3878e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9062e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4775e-01, 2.2293e-02, 2.9999e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0419e-01, 1.7099e-03, 8.1396e-04,  ..., 2.4353e-02,
           0.0000e+00, 0.0000e+00],
          [8.4814e-01, 5.5389e-03, 4.0512e-03,  ..., 4.6074e-05,
           4.0283e-02, 0.0000e+00],
          [3.5205e-01, 8.5449e-03, 3.7918e-03,  ..., 3.6669e-04,
           1.0248e-01, 9.0454e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0576e-01, 9.3994e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3936e-01, 1.2573e-01, 3.4668e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2227e-01, 2.1210e-03, 8.9788e-04,  ..., 1.0201e-02,
           0.0000e+00, 0.0000e+00],
          [1.3062e-01, 9.1019e-03, 1.8196e-03,  ..., 5.8055e-05,
           6.5674e-02, 0.0000e+00],
          [6.8787e-02, 5.7831e-03, 1.5202e-03,  ..., 4.4250e-04,
           1.1981e-01, 1.3062e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 4.1479e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1812e-01, 2.0068e-01, 4.8120e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1179e-02, 1.8477e-06, 1.4305e-06,  ..., 6.4355e-01,
           0.0000e+00, 0.0000e+00],
          [7.6538e-02, 1.3266e-03, 2.3289e-03,  ..., 3.0384e-03,
           6.7676e-01, 0.0000e+00],
          [3.7598e-02, 8.2910e-05, 1.4901e-04,  ..., 1.6556e-02,
           1.0712e-01, 5.8838e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6436e-01, 3.5553e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0186e-01, 5.6519e-02, 4.1840e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2539e-01, 2.4676e-05, 1.6332e-05,  ..., 2.0142e-02,
           0.0000e+00, 0.0000e+00],
          [2.8516e-01, 1.3571e-03, 2.3270e-03,  ..., 3.5896e-03,
           7.4341e-02, 0.0000e+00],
          [2.5488e-01, 2.4080e-04, 3.0136e-04,  ..., 9.6512e-03,
           5.3864e-02, 6.3477e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2041e-01, 7.9773e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9336e-01, 2.0984e-01, 9.6802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1758e-01, 6.0797e-06, 3.8147e-06,  ..., 1.2070e-02,
           0.0000e+00, 0.0000e+00],
          [2.3022e-01, 6.8009e-05, 8.2672e-05,  ..., 2.5436e-02,
           3.6713e-02, 0.0000e+00],
          [4.2017e-01, 5.3823e-05, 5.3585e-05,  ..., 5.6854e-02,
           1.5549e-02, 6.8604e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0898e-01, 2.9102e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1074e-01, 1.7444e-01, 3.1470e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2852e-01, 1.2159e-05, 6.9737e-06,  ..., 2.7783e-01,
           0.0000e+00, 0.0000e+00],
          [3.7280e-01, 1.2054e-03, 1.2236e-03,  ..., 2.8019e-03,
           3.7280e-01, 0.0000e+00],
          [2.8638e-01, 3.6240e-04, 3.4046e-04,  ..., 3.8147e-02,
           9.0759e-02, 3.8818e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2104e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 1.5945e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.5059e-01, 3.5703e-05, 1.3351e-05,  ..., 5.2986e-03,
           0.0000e+00, 0.0000e+00],
          [4.9390e-01, 6.7215e-03, 3.2074e-02,  ..., 4.0591e-05,
           1.7615e-01, 0.0000e+00],
          [7.5732e-01, 1.5802e-03, 4.0359e-03,  ..., 2.0730e-04,
           3.2257e-02, 3.1250e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6904e-01, 2.3096e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 2.9419e-01, 1.2073e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8384e-01, 3.8028e-05, 3.7432e-05,  ..., 1.3115e-02,
           0.0000e+00, 0.0000e+00],
          [3.2983e-01, 2.7657e-03, 3.3894e-03,  ..., 4.9591e-04,
           3.2983e-01, 0.0000e+00],
          [1.8567e-01, 5.5552e-04, 4.2582e-04,  ..., 1.7643e-03,
           1.5881e-01, 1.3379e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6982e-01, 4.3018e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9941e-01, 6.3171e-02, 5.3760e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6294e-02, 2.8551e-05, 1.1563e-05,  ..., 7.8271e-01,
           0.0000e+00, 0.0000e+00],
          [1.4319e-01, 1.1282e-03, 1.3189e-03,  ..., 2.8062e-04,
           7.9834e-01, 0.0000e+00],
          [3.3862e-01, 2.2244e-04, 3.6097e-04,  ..., 2.0771e-03,
           1.1340e-01, 4.9268e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4229e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 7.4196e-03, 5.3024e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4153e-02, 1.3828e-05, 2.9802e-06,  ..., 1.2760e-03,
           0.0000e+00, 0.0000e+00],
          [5.6152e-01, 3.8818e-02, 1.7090e-02,  ..., 2.4438e-05,
           1.1145e-01, 0.0000e+00],
          [3.9551e-01, 1.4748e-02, 8.8043e-03,  ..., 6.8998e-04,
           2.8870e-02, 1.3147e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.5596e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 6.2904e-03, 3.7556e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6853e-02, 2.1338e-05, 9.7752e-06,  ..., 7.0333e-04,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 2.2755e-03, 2.8534e-03,  ..., 6.1214e-05,
           3.0930e-02, 0.0000e+00],
          [6.6650e-01, 1.4524e-03, 1.6136e-03,  ..., 4.7874e-04,
           2.4078e-02, 2.7298e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 2.9205e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.2842e-02, 9.3765e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7896e-01, 3.7789e-04, 2.2304e-04,  ..., 5.7068e-03,
           0.0000e+00, 0.0000e+00],
          [4.0869e-01, 9.6054e-03, 9.7580e-03,  ..., 2.3222e-04,
           9.4788e-02, 0.0000e+00],
          [3.8306e-01, 5.2338e-03, 5.8403e-03,  ..., 7.3957e-04,
           6.0608e-02, 2.1606e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 4.3518e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 2.9694e-02, 2.4429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6895e-01, 1.2517e-05, 1.1802e-05,  ..., 6.3110e-02,
           0.0000e+00, 0.0000e+00],
          [4.5386e-01, 3.1972e-04, 3.5954e-04,  ..., 1.2383e-02,
           6.3843e-02, 0.0000e+00],
          [2.8516e-01, 3.5262e-04, 2.5296e-04,  ..., 1.8097e-02,
           9.4788e-02, 1.7163e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5195e-01, 2.4805e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7266e-01, 1.7960e-02, 9.2468e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.6533e-01, 3.9935e-05, 2.2769e-05,  ..., 2.9087e-03,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 2.7790e-03, 1.3542e-03,  ..., 4.8370e-03,
           1.8402e-02, 0.0000e+00],
          [9.3652e-01, 2.0485e-03, 5.7364e-04,  ..., 6.4125e-03,
           1.2848e-02, 6.9847e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5488e-01, 2.4512e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 1.5312e-02, 7.0686e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7188e-01, 3.9637e-05, 1.4961e-05,  ..., 1.7761e-02,
           0.0000e+00, 0.0000e+00],
          [8.4229e-01, 3.8395e-03, 3.4161e-03,  ..., 1.6037e-02,
           3.7018e-02, 0.0000e+00],
          [8.4668e-01, 9.9945e-04, 5.5170e-04,  ..., 1.1986e-02,
           1.6769e-02, 1.4458e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 5.2612e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 5.1788e-02, 4.4312e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7126e-01, 2.5005e-03, 3.7251e-03,  ..., 1.6174e-02,
           0.0000e+00, 0.0000e+00],
          [2.4719e-01, 4.8943e-03, 5.7220e-03,  ..., 1.2350e-04,
           2.2864e-01, 0.0000e+00],
          [1.9580e-01, 1.8167e-03, 2.6436e-03,  ..., 5.9462e-04,
           3.4888e-01, 1.5491e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2285e-01, 7.6965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 7.0801e-02, 3.9093e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1562e-01, 3.0518e-05, 4.6909e-05,  ..., 3.3508e-02,
           0.0000e+00, 0.0000e+00],
          [1.8628e-01, 1.5869e-03, 9.0408e-04,  ..., 1.2878e-02,
           4.1565e-02, 0.0000e+00],
          [3.2349e-01, 1.3506e-04, 1.0270e-04,  ..., 6.1737e-02,
           6.2561e-03, 8.1787e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8682e-01, 1.3222e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5850e-01, 1.9882e-02, 2.1835e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 1.5745e-03, 3.5477e-03,  ..., 3.1548e-03,
           0.0000e+00, 0.0000e+00],
          [4.9268e-01, 6.1035e-03, 1.0147e-02,  ..., 5.8126e-04,
           2.7573e-02, 0.0000e+00],
          [3.7793e-01, 3.9749e-03, 8.4152e-03,  ..., 9.2936e-04,
           7.0984e-02, 6.1676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.5778e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 9.3384e-03, 1.5396e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7810e-02, 1.3888e-05, 1.4484e-05,  ..., 1.5610e-02,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 7.9584e-04, 2.1973e-03,  ..., 2.3532e-04,
           1.1450e-01, 0.0000e+00],
          [4.2407e-01, 8.5115e-04, 2.0103e-03,  ..., 2.5215e-03,
           2.6489e-02, 3.8849e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3359e-01, 6.6589e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3457e-01, 3.5675e-02, 2.9800e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1536e-01, 4.5598e-05, 7.8797e-05,  ..., 7.1449e-03,
           0.0000e+00, 0.0000e+00],
          [4.4360e-01, 1.9302e-03, 3.6621e-03,  ..., 4.4346e-05,
           1.2512e-01, 0.0000e+00],
          [1.9507e-01, 7.9107e-04, 2.2888e-03,  ..., 5.5647e-04,
           4.8553e-02, 1.2018e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.8982e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 2.3163e-02, 2.9968e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9580e-02, 4.7684e-06, 1.0788e-05,  ..., 4.1771e-03,
           0.0000e+00, 0.0000e+00],
          [5.6396e-01, 8.4782e-04, 2.0504e-03,  ..., 4.6086e-04,
           1.2988e-01, 0.0000e+00],
          [4.4458e-01, 6.0415e-04, 2.2621e-03,  ..., 1.9951e-03,
           5.5664e-02, 2.8656e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9590e-01, 2.0435e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 1.4832e-01, 1.1023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.5969e-02, 5.1022e-05, 3.8505e-05,  ..., 1.6541e-02,
           0.0000e+00, 0.0000e+00],
          [9.4238e-02, 1.5717e-03, 2.9812e-03,  ..., 4.1628e-04,
           3.8666e-02, 0.0000e+00],
          [9.1629e-03, 1.8430e-04, 3.9029e-04,  ..., 2.7943e-03,
           8.6060e-03, 9.1125e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.3697e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1064e-01, 3.6438e-02, 5.3009e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9820e-03, 9.4771e-06, 2.1458e-06,  ..., 9.8133e-04,
           0.0000e+00, 0.0000e+00],
          [6.6064e-01, 1.0242e-03, 1.2951e-03,  ..., 1.7250e-04,
           3.0411e-02, 0.0000e+00],
          [5.1758e-01, 1.3876e-03, 1.4544e-03,  ..., 5.3177e-03,
           6.9389e-03, 1.0034e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3148e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 1.2321e-02, 1.6708e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4834e-01, 4.3333e-05, 4.2319e-05,  ..., 3.9940e-03,
           0.0000e+00, 0.0000e+00],
          [5.9619e-01, 1.0309e-03, 2.2526e-03,  ..., 3.0971e-04,
           6.7253e-03, 0.0000e+00],
          [6.5918e-01, 1.6994e-03, 2.9125e-03,  ..., 2.6679e-04,
           6.5651e-03, 1.5747e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5693e-01, 8.1604e-02, 6.1584e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1462e-02, 1.0133e-06, 7.7486e-07,  ..., 1.9150e-03,
           0.0000e+00, 0.0000e+00],
          [2.4402e-01, 7.3528e-04, 7.2432e-04,  ..., 1.7047e-05,
           2.3056e-02, 0.0000e+00],
          [2.7441e-01, 2.4629e-04, 2.7800e-04,  ..., 1.2407e-03,
           1.5358e-02, 1.2366e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 2.4612e-02, 3.7231e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.4126e-05, 2.5272e-05,  ..., 6.4621e-03,
           0.0000e+00, 0.0000e+00],
          [3.1274e-01, 8.4496e-04, 1.1816e-03,  ..., 6.9141e-06,
           1.0559e-01, 0.0000e+00],
          [1.7444e-01, 3.1638e-04, 6.9666e-04,  ..., 1.7071e-04,
           1.7310e-01, 5.1971e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3643e-01, 1.6345e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 4.1565e-02, 1.6821e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4749e-02, 2.3961e-05, 3.7551e-05,  ..., 3.5962e-01,
           0.0000e+00, 0.0000e+00],
          [5.1367e-01, 2.5272e-04, 6.5041e-04,  ..., 7.4148e-05,
           2.3157e-01, 0.0000e+00],
          [4.0527e-01, 9.2685e-05, 3.5524e-04,  ..., 1.0281e-03,
           5.4413e-02, 4.0527e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 3.0334e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2139e-01, 2.2186e-02, 5.6213e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1558e-01, 1.1325e-05, 9.8348e-06,  ..., 5.4398e-03,
           0.0000e+00, 0.0000e+00],
          [4.7632e-01, 4.8280e-04, 1.6270e-03,  ..., 1.4210e-04,
           5.8228e-02, 0.0000e+00],
          [7.3438e-01, 2.7370e-04, 6.4421e-04,  ..., 1.5631e-03,
           2.0035e-02, 4.8065e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5752e-01, 4.2389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.0071e-02, 2.4933e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7295e-01, 9.0599e-06, 3.4213e-05,  ..., 8.5068e-03,
           0.0000e+00, 0.0000e+00],
          [6.2012e-01, 5.7888e-04, 3.1281e-03,  ..., 7.9572e-05,
           4.2175e-02, 0.0000e+00],
          [7.6953e-01, 2.3699e-04, 2.1458e-03,  ..., 4.3583e-04,
           6.2065e-03, 1.7548e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 2.2293e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 1.7273e-02, 4.6600e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5332e-01, 4.1723e-06, 5.0664e-06,  ..., 2.9888e-03,
           0.0000e+00, 0.0000e+00],
          [3.0103e-01, 3.6163e-03, 1.8951e-02,  ..., 1.1998e-04,
           5.0720e-02, 0.0000e+00],
          [5.7227e-01, 3.3779e-03, 8.6899e-03,  ..., 9.3079e-04,
           1.0239e-02, 3.9276e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5459e-01, 4.5349e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1699e-01, 4.1901e-02, 4.0924e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9651e-02, 2.4974e-05, 1.5259e-05,  ..., 9.6664e-03,
           0.0000e+00, 0.0000e+00],
          [3.6499e-01, 1.8272e-03, 3.1204e-03,  ..., 1.0090e-03,
           7.5928e-02, 0.0000e+00],
          [4.6411e-01, 6.3515e-04, 5.6934e-04,  ..., 9.3689e-03,
           1.6388e-02, 4.4525e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 3.2837e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 1.8631e-02, 2.2385e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6365e-02, 1.5736e-05, 7.3910e-06,  ..., 4.3091e-02,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 4.6635e-04, 9.4604e-04,  ..., 3.8362e-04,
           1.4122e-02, 0.0000e+00],
          [9.8511e-02, 9.9421e-05, 9.1255e-05,  ..., 2.4139e-02,
           3.2158e-03, 2.1805e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 3.6896e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5605e-01, 1.5640e-02, 2.8427e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9241e-02, 2.9802e-07, 1.1921e-07,  ..., 8.9765e-05,
           0.0000e+00, 0.0000e+00],
          [5.4541e-01, 2.1172e-03, 4.0665e-03,  ..., 3.0637e-04,
           4.6082e-03, 0.0000e+00],
          [5.3662e-01, 2.1420e-03, 3.4637e-03,  ..., 2.3441e-03,
           6.9733e-03, 2.0981e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 7.6370e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8232e-01, 1.0956e-01, 8.1253e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0625e-01, 1.9312e-05, 2.3842e-06,  ..., 2.0340e-02,
           0.0000e+00, 0.0000e+00],
          [6.5332e-01, 6.7055e-05, 4.8161e-05,  ..., 9.4116e-02,
           2.2354e-02, 0.0000e+00],
          [6.1279e-01, 6.9737e-06, 1.1003e-04,  ..., 9.6924e-02,
           1.0986e-01, 3.1219e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5801e-01, 4.2084e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 6.5430e-02, 5.9113e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9766e-01, 3.5524e-05, 1.2350e-04,  ..., 6.7627e-02,
           0.0000e+00, 0.0000e+00],
          [5.2002e-01, 3.7503e-04, 1.2884e-03,  ..., 2.0157e-02,
           3.5370e-02, 0.0000e+00],
          [4.6948e-01, 8.4937e-05, 2.5558e-04,  ..., 7.3120e-02,
           6.7627e-02, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7095e-01, 4.7461e-01, 5.4504e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4551e-01, 1.3769e-04, 5.2869e-05,  ..., 8.4656e-02,
           0.0000e+00, 0.0000e+00],
          [5.7861e-01, 2.4533e-04, 2.4343e-04,  ..., 2.3132e-02,
           9.3689e-02, 0.0000e+00],
          [5.0049e-01, 5.1618e-05, 3.6693e-04,  ..., 3.5431e-02,
           2.8516e-01, 9.7046e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9443e-01, 2.0557e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9395e-01, 7.4158e-02, 1.3171e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8687e-01, 4.1127e-06, 7.3910e-06,  ..., 2.2995e-02,
           0.0000e+00, 0.0000e+00],
          [4.9683e-02, 4.5395e-04, 6.8426e-04,  ..., 2.4676e-04,
           3.7781e-02, 0.0000e+00],
          [2.8351e-02, 8.2493e-05, 3.2616e-04,  ..., 4.1466e-03,
           6.5369e-02, 8.8684e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 5.6396e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1798e-01, 3.7109e-02, 8.4473e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5732e-02, 2.0266e-05, 5.9903e-05,  ..., 5.3320e-01,
           0.0000e+00, 0.0000e+00],
          [7.4951e-02, 1.6975e-04, 4.6806e-03,  ..., 9.7504e-03,
           7.2803e-01, 0.0000e+00],
          [1.4099e-02, 9.6560e-06, 1.3709e-04,  ..., 4.1656e-03,
           1.9424e-02, 8.9990e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9746e-01, 5.1025e-02, 5.1422e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.3450e-01, 8.0645e-05, 3.5667e-04,  ..., 2.0523e-03,
           0.0000e+00, 0.0000e+00],
          [5.3271e-01, 1.9369e-03, 5.4741e-03,  ..., 1.0133e-06,
           1.1147e-02, 0.0000e+00],
          [1.1969e-01, 3.5357e-04, 1.4715e-03,  ..., 2.9802e-06,
           4.9286e-02, 1.5762e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 6.3721e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7451e-01, 8.0078e-02, 4.5349e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4150e-01, 4.5943e-04, 7.5245e-04,  ..., 3.0975e-03,
           0.0000e+00, 0.0000e+00],
          [5.3906e-01, 2.9011e-03, 4.7455e-03,  ..., 3.1986e-03,
           7.6721e-02, 0.0000e+00],
          [2.6514e-01, 3.3703e-03, 6.2599e-03,  ..., 2.6245e-03,
           2.7246e-01, 1.1182e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 1.9073e-02, 1.3664e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7051e-01, 3.7551e-06, 7.1526e-06,  ..., 3.5065e-02,
           0.0000e+00, 0.0000e+00],
          [3.8818e-01, 2.1946e-04, 3.0231e-04,  ..., 1.4992e-02,
           2.9465e-02, 0.0000e+00],
          [1.4966e-01, 8.6665e-05, 1.0997e-04,  ..., 1.2871e-02,
           1.1009e-02, 6.6406e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 5.9326e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 2.3877e-01, 8.6182e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8350e-01, 7.1645e-05, 5.6565e-05,  ..., 1.6455e-01,
           0.0000e+00, 0.0000e+00],
          [3.2397e-01, 4.9639e-04, 5.0449e-04,  ..., 7.2815e-02,
           1.6675e-01, 0.0000e+00],
          [1.5173e-01, 4.5180e-04, 7.4053e-04,  ..., 1.3260e-02,
           4.8779e-01, 1.7053e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 2.1744e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4385e-01, 2.0721e-02, 3.5614e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0918e-01, 1.2314e-04, 4.3511e-04,  ..., 2.9516e-04,
           0.0000e+00, 0.0000e+00],
          [8.8428e-01, 2.8133e-04, 7.0190e-04,  ..., 3.1567e-04,
           2.4796e-02, 0.0000e+00],
          [8.0322e-01, 1.5533e-04, 7.5722e-04,  ..., 6.3133e-04,
           5.1361e-02, 4.0009e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 9.3651e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 3.5620e-04, 1.9112e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.8877e-01, 6.7353e-06, 2.6643e-05,  ..., 9.9599e-05,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.5010e-04, 1.6756e-03,  ..., 2.5105e-04,
           2.6569e-03, 0.0000e+00],
          [9.7803e-01, 8.9347e-05, 5.0402e-04,  ..., 1.9288e-04,
           5.1308e-03, 3.3646e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2672e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 1.0696e-02, 1.5381e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2666e-01, 2.4533e-04, 1.6367e-04,  ..., 1.1768e-03,
           0.0000e+00, 0.0000e+00],
          [7.2168e-01, 1.3199e-03, 6.1989e-03,  ..., 3.9406e-03,
           2.2842e-02, 0.0000e+00],
          [3.6719e-01, 5.2691e-04, 2.2964e-03,  ..., 8.4381e-03,
           1.5396e-02, 1.4050e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3634e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8770e-01, 5.2887e-02, 5.9479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 3.0279e-05, 2.2173e-05,  ..., 4.9782e-03,
           0.0000e+00, 0.0000e+00],
          [5.3467e-01, 4.6158e-04, 8.5592e-04,  ..., 3.6316e-03,
           5.2948e-02, 0.0000e+00],
          [2.7100e-01, 1.8227e-04, 2.4724e-04,  ..., 8.9874e-03,
           3.0899e-02, 2.9251e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9119e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 8.5449e-03, 1.1497e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5498e-01, 1.2517e-06, 3.5763e-07,  ..., 5.7373e-03,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 6.1941e-04, 9.7466e-04,  ..., 1.0727e-02,
           3.8025e-02, 0.0000e+00],
          [3.4277e-01, 1.3447e-04, 6.9201e-05,  ..., 4.0314e-02,
           1.1368e-02, 3.5004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4756e-01, 2.5244e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.7249e-01, 7.8369e-02, 7.4902e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.3691e-01, 5.5492e-05, 2.1112e-04,  ..., 2.7930e-01,
           0.0000e+00, 0.0000e+00],
          [1.8079e-01, 1.1806e-03, 1.4381e-02,  ..., 2.0714e-03,
           7.0410e-01, 0.0000e+00],
          [1.0321e-01, 2.8086e-04, 1.3199e-03,  ..., 5.4283e-03,
           1.3049e-01, 7.0508e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 1.2463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5977e-01, 1.5930e-01, 8.0750e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1270e-01, 5.6028e-05, 6.8963e-05,  ..., 3.0548e-02,
           0.0000e+00, 0.0000e+00],
          [4.4849e-01, 9.7930e-05, 1.9217e-04,  ..., 5.1727e-03,
           5.0934e-02, 0.0000e+00],
          [1.6614e-01, 2.4796e-05, 7.1108e-05,  ..., 3.2837e-02,
           9.5032e-02, 1.1420e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5703e-01, 4.2816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0674e-01, 6.1218e-02, 3.1952e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4033e-01, 1.4997e-04, 6.3002e-05,  ..., 7.7858e-03,
           0.0000e+00, 0.0000e+00],
          [5.2393e-01, 8.6963e-05, 2.1589e-04,  ..., 1.8860e-02,
           3.4943e-02, 0.0000e+00],
          [5.3027e-01, 4.5538e-05, 4.5717e-05,  ..., 4.3854e-02,
           1.8936e-02, 2.3941e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4629e-01, 5.3802e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 3.6652e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5708e-01, 5.0831e-04, 2.7394e-04,  ..., 2.8748e-02,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.4211e-04, 6.2275e-04,  ..., 2.6722e-03,
           1.8082e-02, 0.0000e+00],
          [5.7715e-01, 3.7193e-04, 9.0504e-04,  ..., 7.2098e-03,
           2.7206e-02, 3.2074e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 7.5867e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8672e-01, 4.6631e-02, 6.6772e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3193e-01, 6.4850e-05, 2.0719e-04,  ..., 9.5444e-03,
           0.0000e+00, 0.0000e+00],
          [6.8945e-01, 1.8442e-04, 2.1954e-03,  ..., 9.2087e-03,
           1.7670e-02, 0.0000e+00],
          [2.2400e-01, 2.2697e-04, 1.3962e-03,  ..., 7.8506e-03,
           3.7140e-02, 6.5735e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 5.3986e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8721e-01, 2.6581e-02, 8.6487e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3711e-01, 4.2200e-05, 7.4446e-05,  ..., 1.3451e-02,
           0.0000e+00, 0.0000e+00],
          [7.9297e-01, 4.9686e-04, 4.7493e-03,  ..., 1.4048e-03,
           4.2999e-02, 0.0000e+00],
          [6.9482e-01, 4.0126e-04, 5.1613e-03,  ..., 5.8250e-03,
           4.3396e-02, 7.2083e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2471e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.4198e-02, 2.0981e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4639e-01, 5.7220e-06, 2.1517e-05,  ..., 2.6566e-02,
           0.0000e+00, 0.0000e+00],
          [7.1533e-01, 4.3440e-04, 2.2240e-03,  ..., 8.8654e-03,
           8.9539e-02, 0.0000e+00],
          [5.4541e-01, 1.9026e-04, 7.2384e-04,  ..., 9.1629e-03,
           3.9825e-02, 1.5259e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0527e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 5.4626e-03, 4.7493e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.6234e-02, 1.6093e-06, 5.3644e-07,  ..., 1.6617e-02,
           0.0000e+00, 0.0000e+00],
          [1.4441e-01, 6.4039e-04, 8.5831e-04,  ..., 2.6550e-03,
           6.5193e-03, 0.0000e+00],
          [4.7989e-03, 1.9372e-05, 1.5616e-05,  ..., 1.6098e-02,
           3.9077e-04, 2.2144e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7363e-01, 2.6154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.6129e-02, 1.3260e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1104e-01, 5.8413e-06, 1.7881e-06,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 5.2691e-04, 1.3542e-03,  ..., 2.5902e-03,
           4.5815e-03, 0.0000e+00],
          [2.5073e-01, 7.8559e-05, 1.1277e-04,  ..., 3.2898e-02,
           2.7752e-03, 1.4656e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 5.3192e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 1.7670e-02, 3.2623e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2681e-01, 2.6345e-05, 4.5359e-05,  ..., 1.5556e-02,
           0.0000e+00, 0.0000e+00],
          [3.1494e-01, 1.6088e-03, 7.0953e-03,  ..., 3.2997e-03,
           3.9124e-02, 0.0000e+00],
          [1.0626e-01, 4.4107e-04, 1.8139e-03,  ..., 1.4267e-02,
           2.1088e-02, 8.0200e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.4283e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 3.8147e-03, 1.7914e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4053e-01, 5.4002e-05, 4.2820e-04,  ..., 1.5907e-03,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 8.9455e-04, 1.7786e-03,  ..., 4.8637e-04,
           3.4088e-02, 0.0000e+00],
          [4.0918e-01, 1.9348e-04, 1.2426e-03,  ..., 6.6519e-04,
           1.8152e-01, 1.1359e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0178e-01, 7.9834e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1455e-01, 3.5461e-02, 5.0018e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9814e-01, 1.0085e-04, 2.5749e-04,  ..., 1.0948e-02,
           0.0000e+00, 0.0000e+00],
          [4.1309e-01, 3.9253e-03, 3.8643e-03,  ..., 5.4359e-04,
           1.0944e-01, 0.0000e+00],
          [1.2842e-01, 7.5197e-04, 2.4261e-03,  ..., 6.4325e-04,
           1.4331e-01, 1.9897e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6680e-01, 3.3081e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 6.8245e-03, 1.1429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0029e-01, 1.9479e-04, 3.3140e-04,  ..., 4.5052e-03,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 8.2254e-04, 1.0080e-03,  ..., 2.0199e-03,
           1.8295e-02, 0.0000e+00],
          [5.8594e-01, 1.0633e-03, 1.4753e-03,  ..., 3.1738e-03,
           5.1208e-02, 4.0192e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 9.6191e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5049e-01, 2.7878e-02, 2.2144e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1133e-01, 2.6643e-05, 9.9838e-05,  ..., 2.5482e-02,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 8.0407e-05, 1.0598e-04,  ..., 8.1940e-03,
           8.4351e-02, 0.0000e+00],
          [7.1582e-01, 5.2452e-05, 7.2598e-05,  ..., 7.6752e-03,
           7.2815e-02, 9.5703e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5259e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 2.7115e-02, 3.2013e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5386e-01, 6.5327e-05, 1.9968e-04,  ..., 4.7546e-02,
           0.0000e+00, 0.0000e+00],
          [6.0596e-01, 1.8382e-04, 2.3282e-04,  ..., 2.2736e-02,
           1.1841e-01, 0.0000e+00],
          [3.6572e-01, 9.9838e-05, 2.0587e-04,  ..., 2.1545e-02,
           1.5918e-01, 1.8896e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0619e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 1.5454e-01, 1.6382e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9565e-01, 2.8396e-04, 3.0339e-05,  ..., 9.0837e-04,
           0.0000e+00, 0.0000e+00],
          [6.0254e-01, 2.1160e-04, 1.4949e-04,  ..., 1.0925e-02,
           2.1484e-01, 0.0000e+00],
          [1.5747e-01, 3.8922e-05, 1.6475e-04,  ..., 1.1314e-02,
           6.0840e-01, 1.8701e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 2.5375e-02, 4.4525e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6221e-01, 2.5809e-05, 6.7949e-05,  ..., 9.8495e-03,
           0.0000e+00, 0.0000e+00],
          [7.9736e-01, 8.5688e-04, 1.6909e-03,  ..., 1.2913e-03,
           2.2263e-02, 0.0000e+00],
          [4.0845e-01, 9.1076e-04, 2.4853e-03,  ..., 1.4534e-02,
           2.6520e-02, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0723e-01, 9.2712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5264e-01, 2.9221e-02, 1.8066e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7842e-01, 1.2875e-05, 1.0252e-05,  ..., 6.8665e-03,
           0.0000e+00, 0.0000e+00],
          [6.1719e-01, 7.2122e-05, 1.3208e-04,  ..., 2.2621e-03,
           1.2611e-02, 0.0000e+00],
          [2.1655e-01, 8.5592e-05, 1.8477e-04,  ..., 1.5686e-02,
           2.8412e-02, 3.5339e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2935e-01, 6.7041e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3538e-01, 5.4688e-02, 8.1006e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8356e-02, 4.1187e-05, 1.0747e-04,  ..., 1.8835e-01,
           0.0000e+00, 0.0000e+00],
          [8.8684e-02, 1.6756e-03, 4.2953e-03,  ..., 7.8735e-03,
           5.0635e-01, 0.0000e+00],
          [2.4338e-02, 4.1223e-04, 8.3590e-04,  ..., 1.8120e-03,
           2.6108e-02, 8.0566e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1070e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 3.5572e-03, 1.2054e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2080e-01, 7.5519e-05, 2.0230e-04,  ..., 7.9575e-03,
           0.0000e+00, 0.0000e+00],
          [3.7134e-01, 4.2152e-04, 1.0185e-03,  ..., 6.4125e-03,
           2.6538e-01, 0.0000e+00],
          [2.1777e-01, 5.1641e-04, 1.3180e-03,  ..., 1.0391e-02,
           2.9761e-01, 1.6565e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 6.4636e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 9.7046e-03, 1.9760e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7295e-01, 2.4891e-04, 3.6287e-04,  ..., 1.7786e-03,
           0.0000e+00, 0.0000e+00],
          [1.2457e-01, 7.4501e-03, 8.4763e-03,  ..., 1.0262e-02,
           4.9164e-02, 0.0000e+00],
          [8.9600e-02, 5.9128e-03, 7.1297e-03,  ..., 1.5640e-02,
           3.5919e-02, 6.0638e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 6.6101e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3164e-01, 2.3682e-02, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1790e-01, 4.9412e-05, 1.5533e-04,  ..., 1.0071e-02,
           0.0000e+00, 0.0000e+00],
          [2.7563e-01, 1.5154e-03, 2.8095e-03,  ..., 2.0920e-02,
           1.2016e-02, 0.0000e+00],
          [1.7615e-01, 6.8426e-04, 1.2054e-03,  ..., 2.2659e-02,
           7.6180e-03, 3.1342e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 5.7259e-03, 8.5983e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5244e-01, 1.5459e-03, 1.4296e-03,  ..., 5.2299e-03,
           0.0000e+00, 0.0000e+00],
          [7.2266e-01, 2.4109e-03, 1.1568e-03,  ..., 1.1749e-03,
           3.5431e-02, 0.0000e+00],
          [6.1523e-01, 2.9802e-04, 3.5930e-04,  ..., 2.2011e-03,
           2.0569e-02, 1.6266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.0000e-01, 5.0000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7783e-01, 2.0325e-01, 5.1904e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8931e-01, 5.6744e-04, 1.5421e-03,  ..., 2.7173e-01,
           0.0000e+00, 0.0000e+00],
          [2.0471e-01, 1.7710e-03, 3.8071e-03,  ..., 3.1891e-02,
           4.8340e-01, 0.0000e+00],
          [1.4502e-01, 1.0405e-03, 2.6569e-03,  ..., 1.9623e-02,
           4.4220e-02, 6.4014e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 1.2646e-03, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6318e-01, 2.1219e-04, 6.0940e-04,  ..., 6.3972e-03,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1711e-03, 1.8721e-03,  ..., 2.6398e-03,
           2.5848e-02, 0.0000e+00],
          [5.6934e-01, 4.9162e-04, 1.6499e-03,  ..., 9.2010e-03,
           4.5319e-02, 2.4246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1948e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 4.9095e-03, 1.4539e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.5020e-01, 6.7353e-06, 5.0247e-05,  ..., 2.2709e-04,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1849e-04, 6.1274e-04,  ..., 4.6196e-03,
           1.4488e-02, 0.0000e+00],
          [6.5820e-01, 3.7789e-05, 6.6280e-04,  ..., 4.9362e-03,
           1.2032e-02, 1.0803e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4404e-01, 3.5571e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9131e-01, 5.2490e-02, 3.5596e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7656e-01, 6.4969e-06, 5.3048e-05,  ..., 1.5879e-04,
           0.0000e+00, 0.0000e+00],
          [6.2500e-01, 2.0123e-04, 9.0075e-04,  ..., 4.9362e-03,
           4.8370e-02, 0.0000e+00],
          [6.8909e-02, 9.5665e-05, 9.2685e-05,  ..., 1.5344e-01,
           1.3206e-02, 3.6041e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2650e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 1.1169e-02, 1.5039e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 2.4140e-05, 1.2863e-04,  ..., 7.5877e-05,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 7.9298e-04, 2.2221e-03,  ..., 1.3018e-03,
           1.5222e-01, 0.0000e+00],
          [2.4585e-01, 1.4963e-03, 1.7529e-03,  ..., 1.8845e-02,
           1.4062e-01, 9.5520e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 2.1008e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 7.6752e-03, 1.8021e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1904e-01, 1.3628e-03, 6.5956e-03,  ..., 1.3062e-02,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 2.8276e-04, 4.9286e-03,  ..., 3.7117e-03,
           8.3313e-02, 0.0000e+00],
          [7.6221e-01, 2.5392e-04, 8.5449e-04,  ..., 4.5967e-03,
           2.2232e-02, 1.2665e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8613e-01, 2.1399e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6270e-01, 4.6967e-02, 1.9055e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8018e-01, 2.0623e-05, 2.5196e-03,  ..., 4.9362e-03,
           0.0000e+00, 0.0000e+00],
          [6.9824e-01, 5.7793e-04, 2.3766e-03,  ..., 1.3794e-02,
           1.6260e-01, 0.0000e+00],
          [8.5742e-01, 3.8981e-05, 8.1253e-04,  ..., 5.4283e-03,
           1.7593e-02, 9.6207e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.9866e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 7.7744e-03, 1.5717e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8262e-01, 4.0913e-04, 3.6359e-04,  ..., 1.0757e-02,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 1.3466e-03, 9.7227e-04,  ..., 6.9122e-03,
           4.9164e-02, 0.0000e+00],
          [5.0635e-01, 1.9236e-03, 1.8082e-03,  ..., 8.9111e-03,
           5.0049e-02, 2.6520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-02, 9.1064e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3428e-02, 1.3351e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1062e-06, 1.1265e-05, 3.2397e-01,  ..., 9.3174e-04,
           0.0000e+00, 0.0000e+00],
          [8.9836e-04, 3.8414e-03, 1.4026e-01,  ..., 1.5793e-02,
           8.2458e-02, 0.0000e+00],
          [5.6839e-04, 1.0252e-03, 1.7236e-01,  ..., 2.5314e-02,
           3.1769e-02, 3.9978e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8296e-01, 7.1680e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9276e-02, 3.0838e-02, 9.2969e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0121e-04, 2.0444e-04, 2.9248e-01,  ..., 8.4229e-03,
           0.0000e+00, 0.0000e+00],
          [4.5538e-04, 1.4811e-03, 2.6465e-01,  ..., 4.6806e-03,
           6.3843e-02, 0.0000e+00],
          [5.2738e-04, 2.9683e-04, 1.6028e-01,  ..., 9.5139e-03,
           1.1604e-02, 1.2947e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7424e-02, 9.5264e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.0131e-02, 4.1473e-02, 9.1846e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7115e-05, 4.3809e-05, 7.9041e-03,  ..., 1.5087e-03,
           0.0000e+00, 0.0000e+00],
          [1.3237e-03, 7.2670e-03, 8.8867e-02,  ..., 1.1566e-02,
           2.4927e-01, 0.0000e+00],
          [1.0719e-03, 2.0542e-03, 1.6162e-01,  ..., 1.2657e-02,
           4.2145e-02, 3.6041e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7178e-01, 4.2822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5845e-01, 8.5510e-02, 7.5586e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2214e-04, 6.2990e-04, 1.2335e-01,  ..., 5.9631e-02,
           0.0000e+00, 0.0000e+00],
          [2.5978e-03, 2.4986e-03, 2.4036e-01,  ..., 2.1164e-02,
           2.0557e-01, 0.0000e+00],
          [3.3307e-04, 1.8907e-04, 9.3811e-02,  ..., 1.1584e-01,
           2.5708e-01, 2.3035e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1567e-01, 6.8408e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1864e-02, 4.6005e-03, 9.8340e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1501e-03, 5.1260e-04, 3.1519e-01,  ..., 9.6130e-04,
           0.0000e+00, 0.0000e+00],
          [2.1517e-04, 7.9155e-04, 3.2300e-01,  ..., 5.7030e-04,
           6.8237e-02, 0.0000e+00],
          [6.0797e-04, 8.7070e-04, 2.7783e-01,  ..., 7.6866e-04,
           4.1321e-02, 5.1819e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9600e-01, 6.0400e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.2267e-03, 1.5068e-03, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.4392e-05, 6.9737e-05, 3.6182e-01,  ..., 2.8564e-02,
           0.0000e+00, 0.0000e+00],
          [2.2042e-04, 1.8132e-04, 2.3608e-01,  ..., 1.7242e-02,
           4.4037e-02, 0.0000e+00],
          [1.7631e-04, 2.9147e-05, 1.1548e-01,  ..., 1.9455e-02,
           7.5569e-03, 3.1830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7598e-01, 1.2421e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0109e-03, 2.2697e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6716e-05, 8.6427e-06, 4.1968e-01,  ..., 1.0258e-04,
           0.0000e+00, 0.0000e+00],
          [7.7188e-05, 7.0572e-05, 3.6914e-01,  ..., 1.9836e-03,
           8.6823e-03, 0.0000e+00],
          [3.5357e-04, 1.2058e-04, 2.2266e-01,  ..., 8.6670e-03,
           4.5166e-03, 2.2049e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 3.0493e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.7308e-03, 1.6651e-03, 9.9463e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7578e-05, 1.5676e-05, 2.8076e-01,  ..., 7.0953e-04,
           0.0000e+00, 0.0000e+00],
          [1.5128e-04, 8.7261e-05, 2.8662e-01,  ..., 1.7899e-02,
           4.7546e-02, 0.0000e+00],
          [1.8680e-04, 1.0645e-04, 1.6724e-01,  ..., 1.8188e-02,
           1.2596e-02, 1.1925e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2373e-01, 1.7615e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2567e-03, 1.0767e-03, 9.9365e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9339e-05, 5.7220e-06, 3.9648e-01,  ..., 1.5335e-03,
           0.0000e+00, 0.0000e+00],
          [6.9559e-05, 1.3769e-05, 3.2861e-01,  ..., 1.5137e-02,
           4.6997e-02, 0.0000e+00],
          [9.7513e-05, 2.0146e-05, 3.1543e-01,  ..., 1.1139e-02,
           9.2697e-03, 2.0416e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6975e-03, 1.5202e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0320e-04, 3.0398e-05, 1.4526e-01,  ..., 5.1785e-04,
           0.0000e+00, 0.0000e+00],
          [2.1303e-04, 9.0718e-05, 3.5400e-01,  ..., 1.4582e-03,
           1.4221e-02, 0.0000e+00],
          [7.7152e-04, 1.9884e-04, 1.8652e-01,  ..., 5.6992e-03,
           2.2678e-03, 1.1421e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-01, 1.0895e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7023e-04, 9.5034e-04, 9.9805e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2994e-05, 8.5235e-06, 3.7646e-01,  ..., 3.2258e-04,
           0.0000e+00, 0.0000e+00],
          [2.4855e-05, 2.6584e-05, 3.2129e-01,  ..., 1.5266e-02,
           6.0959e-03, 0.0000e+00],
          [1.9252e-05, 2.9683e-05, 2.4817e-01,  ..., 7.5531e-03,
           1.2617e-03, 5.5237e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8086e-01, 1.1920e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3733e-04, 5.1260e-04, 9.9854e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7893e-02, 1.0166e-03, 1.9067e-01,  ..., 1.3390e-02,
           0.0000e+00, 0.0000e+00],
          [4.9055e-05, 2.8789e-05, 3.1885e-01,  ..., 4.9210e-03,
           1.9800e-01, 0.0000e+00],
          [6.2764e-05, 2.2829e-05, 2.4365e-01,  ..., 5.0774e-03,
           5.3925e-02, 2.0203e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.8843e-01, 6.1182e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6564e-03, 1.2646e-03, 9.9512e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3195e-05, 4.5478e-05, 3.8770e-01,  ..., 8.4925e-04,
           0.0000e+00, 0.0000e+00],
          [4.3035e-05, 3.6001e-05, 3.4351e-01,  ..., 3.8086e-02,
           3.1395e-03, 0.0000e+00],
          [6.4552e-05, 4.0233e-05, 1.2573e-01,  ..., 1.9089e-02,
           3.9315e-04, 1.2337e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8281e-01, 1.1737e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7068e-03, 1.6725e-04, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1683e-04, 4.5300e-06, 3.9160e-01,  ..., 8.9073e-04,
           0.0000e+00, 0.0000e+00],
          [8.7202e-05, 5.1260e-06, 2.9126e-01,  ..., 1.6953e-02,
           1.7624e-02, 0.0000e+00],
          [1.5831e-04, 3.9935e-06, 1.7432e-01,  ..., 1.7532e-02,
           6.3248e-03, 1.0384e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5024e-01, 7.5000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1498e-03, 3.7670e-03, 9.9121e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0711e-04, 1.0902e-04, 4.2725e-01,  ..., 1.9703e-03,
           0.0000e+00, 0.0000e+00],
          [1.5020e-04, 1.5020e-04, 3.1665e-01,  ..., 3.9642e-02,
           4.3449e-03, 0.0000e+00],
          [1.6618e-04, 1.5664e-04, 2.2070e-01,  ..., 3.2684e-02,
           9.4175e-04, 7.0419e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9707e-01, 5.0293e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7298e-02, 2.9907e-03, 9.6973e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5024e-04, 3.7861e-04, 2.2754e-01,  ..., 3.0861e-03,
           0.0000e+00, 0.0000e+00],
          [1.0262e-03, 1.1187e-03, 2.7417e-01,  ..., 1.2871e-02,
           4.0894e-02, 0.0000e+00],
          [1.8728e-04, 1.8537e-04, 1.7029e-01,  ..., 2.0248e-02,
           1.4534e-02, 1.8738e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5537e-01, 7.4463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5106e-03, 1.4343e-03, 9.9707e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8716e-05, 3.5584e-05, 2.7148e-01,  ..., 1.2331e-03,
           0.0000e+00, 0.0000e+00],
          [4.7505e-05, 4.1664e-05, 2.3096e-01,  ..., 1.7807e-02,
           1.3817e-02, 0.0000e+00],
          [6.5148e-05, 5.7399e-05, 7.5745e-02,  ..., 1.6312e-02,
           6.2675e-03, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9375e-01, 4.0649e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6136e-03, 1.5936e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9935e-05, 2.2054e-05, 3.7598e-01,  ..., 7.6389e-04,
           0.0000e+00, 0.0000e+00],
          [1.0240e-04, 3.3379e-05, 2.6514e-01,  ..., 2.4658e-02,
           1.7212e-02, 0.0000e+00],
          [2.1017e-04, 3.4332e-05, 1.2720e-01,  ..., 2.9297e-02,
           4.1885e-03, 1.4618e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5684e-01, 2.4292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.5002e-02, 4.4342e-02, 8.9062e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1024e-03, 3.2926e-04, 3.1250e-01,  ..., 7.9651e-02,
           0.0000e+00, 0.0000e+00],
          [1.7290e-03, 2.0897e-04, 2.4536e-02,  ..., 3.2446e-01,
           2.6880e-01, 0.0000e+00],
          [8.2397e-04, 7.2718e-06, 4.3335e-03,  ..., 5.9521e-01,
           4.6692e-03, 9.3445e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.1230e-01, 3.8770e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2986e-03, 1.5945e-02, 9.7852e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1875e-04, 1.3947e-04, 2.6978e-01,  ..., 6.3038e-04,
           0.0000e+00, 0.0000e+00],
          [1.2836e-03, 3.6168e-04, 3.4570e-01,  ..., 1.5545e-04,
           3.9558e-03, 0.0000e+00],
          [9.1648e-04, 3.2330e-04, 2.7930e-01,  ..., 1.9062e-04,
           7.9041e-03, 1.7792e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5435e-01, 5.4590e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6959e-03, 4.8561e-03, 9.9023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7132e-04, 9.8324e-04, 1.0468e-01,  ..., 2.0134e-04,
           0.0000e+00, 0.0000e+00],
          [8.6725e-05, 1.0185e-03, 3.0884e-01,  ..., 2.6941e-05,
           1.0345e-01, 0.0000e+00],
          [1.4329e-04, 3.3188e-04, 2.4072e-01,  ..., 1.3065e-04,
           1.1731e-01, 5.0049e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3727e-02, 9.7607e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4189e-02, 3.4363e-02, 9.2139e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6335e-03, 9.5892e-04, 2.2986e-01,  ..., 2.3441e-03,
           0.0000e+00, 0.0000e+00],
          [2.5487e-04, 2.5010e-04, 9.4238e-02,  ..., 4.4327e-03,
           1.7615e-01, 0.0000e+00],
          [6.7472e-04, 1.3363e-04, 2.5177e-02,  ..., 4.1840e-02,
           1.6223e-01, 6.7078e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0248e-01, 8.9746e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.2769e-01, 6.8604e-02, 8.0371e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0185e-03, 3.2806e-03, 2.5879e-01,  ..., 8.4351e-02,
           0.0000e+00, 0.0000e+00],
          [8.9264e-04, 1.9007e-03, 7.1594e-02,  ..., 3.6285e-02,
           5.8350e-01, 0.0000e+00],
          [4.6206e-04, 8.5950e-05, 1.1154e-02,  ..., 2.4219e-01,
           4.4897e-01, 1.9312e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8052e-01, 7.1924e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3696e-01, 3.3630e-02, 8.2959e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1114e-03, 5.3120e-04, 1.9543e-01,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [1.4889e-04, 2.3186e-05, 1.9119e-02,  ..., 5.0598e-02,
           6.6895e-01, 0.0000e+00],
          [7.9334e-05, 9.7156e-06, 1.0818e-02,  ..., 5.9387e-02,
           5.1208e-02, 5.4504e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1475e-01, 8.8525e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6072e-02, 6.0425e-02, 9.0332e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2528e-03, 1.1911e-03, 1.3135e-01,  ..., 1.1932e-02,
           0.0000e+00, 0.0000e+00],
          [6.0539e-03, 5.4054e-03, 2.0850e-01,  ..., 1.7586e-03,
           9.0393e-02, 0.0000e+00],
          [6.7711e-03, 1.8911e-03, 1.4539e-01,  ..., 1.3924e-03,
           2.5253e-02, 7.4524e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8676e-02, 9.5117e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.3864e-02, 2.4368e-02, 9.2188e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9188e-03, 3.9825e-03, 1.6772e-01,  ..., 3.8483e-02,
           0.0000e+00, 0.0000e+00],
          [2.9087e-03, 3.9291e-03, 1.7236e-01,  ..., 1.2100e-02,
           1.9995e-01, 0.0000e+00],
          [1.2283e-03, 4.0207e-03, 2.7222e-01,  ..., 2.1267e-03,
           1.2703e-02, 4.5654e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2373e-01, 6.7627e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9703e-02, 5.1483e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.3923e-03, 3.3703e-03, 1.7322e-01,  ..., 3.3447e-02,
           0.0000e+00, 0.0000e+00],
          [3.0594e-03, 3.2177e-03, 1.1591e-01,  ..., 1.0071e-01,
           1.8884e-01, 0.0000e+00],
          [2.7275e-03, 1.0405e-03, 1.5308e-01,  ..., 1.9302e-02,
           2.8198e-02, 1.2891e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.6699e-01, 3.3276e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9006e-01, 6.9946e-02, 7.4023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4135e-04, 1.6797e-04, 2.1960e-01,  ..., 5.9113e-02,
           0.0000e+00, 0.0000e+00],
          [2.3746e-03, 1.8895e-04, 1.2659e-01,  ..., 3.6011e-02,
           3.1769e-02, 0.0000e+00],
          [1.0061e-03, 1.4222e-04, 4.8859e-02,  ..., 1.6663e-01,
           1.8152e-01, 2.5000e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2092e-02, 9.1797e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6251e-02, 7.5195e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5483e-04, 3.7594e-03, 2.4182e-01,  ..., 6.6101e-02,
           0.0000e+00, 0.0000e+00],
          [3.5119e-04, 2.6298e-04, 2.9541e-01,  ..., 1.5900e-02,
           8.2031e-02, 0.0000e+00],
          [4.9448e-04, 8.6117e-04, 2.0422e-01,  ..., 6.8245e-03,
           1.0086e-02, 3.1128e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9458e-01, 3.6353e-01, 4.4189e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.3742e-03, 1.2512e-03, 8.4351e-02,  ..., 2.5391e-01,
           0.0000e+00, 0.0000e+00],
          [3.9101e-03, 8.9645e-04, 5.3558e-02,  ..., 6.2103e-02,
           2.4756e-01, 0.0000e+00],
          [1.8728e-04, 4.7326e-05, 1.7059e-02,  ..., 9.8145e-02,
           3.0957e-01, 2.1948e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[5.9605e-07, 4.2319e-06, 6.4611e-04,  ..., 2.5803e-02,
           8.0273e-01, 1.3525e-01]],

         [[1.9014e-05, 1.9014e-05, 2.6531e-03,  ..., 2.3651e-02,
           2.8809e-01, 4.1919e-01]],

         [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 2.9802e-07,
           1.0986e-02, 9.8877e-01]],

         ...,

         [[2.5177e-03, 3.2310e-03, 4.1504e-03,  ..., 3.0670e-02,
           2.2656e-01, 5.0568e-02]],

         [[4.7569e-03, 1.2146e-02, 5.1147e-02,  ..., 3.9825e-02,
           1.3904e-01, 1.0175e-01]],

         [[2.0248e-02, 1.6785e-02, 2.5749e-03,  ..., 4.2877e-02,
           6.5765e-03, 6.2378e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1215e-03, 1.3323e-03, 1.0483e-02,  ..., 5.0781e-02,
           1.9775e-01, 2.6611e-01]],

         [[5.9605e-08, 2.0266e-06, 4.6670e-05,  ..., 9.4116e-02,
           7.7588e-01, 1.2671e-01]],

         [[3.3493e-03, 5.5237e-03, 1.7273e-02,  ..., 1.3586e-01,
           1.9775e-01, 1.9165e-01]],

         ...,

         [[4.5319e-03, 1.5450e-02, 1.5450e-02,  ..., 1.2250e-01,
           1.8494e-02, 9.6863e-02]],

         [[1.4267e-03, 4.0588e-03, 4.0192e-02,  ..., 6.7566e-02,
           3.6230e-01, 4.1443e-02]],

         [[1.0452e-03, 3.9444e-03, 1.8372e-02,  ..., 3.2324e-01,
           1.0577e-01, 1.8274e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1133e-02, 4.7874e-03, 1.4076e-02,  ..., 9.2529e-02,
           8.0383e-02, 1.0162e-01]],

         [[3.3569e-03, 1.0214e-03, 2.3251e-03,  ..., 2.9980e-01,
           3.9490e-02, 1.2817e-02]],

         [[1.7456e-02, 7.6904e-03, 7.7782e-03,  ..., 1.5114e-02,
           5.4413e-02, 2.8671e-02]],

         ...,

         [[4.0936e-04, 3.4332e-04, 1.5507e-03,  ..., 2.8540e-01,
           3.8379e-01, 1.4917e-01]],

         [[1.3947e-02, 2.5902e-03, 4.1885e-03,  ..., 6.7627e-02,
           1.6345e-01, 5.3070e-02]],

         [[9.6359e-03, 2.0123e-03, 2.0325e-02,  ..., 5.6549e-02,
           4.2114e-01, 9.5459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8955e-01, 2.2829e-05, 5.0485e-05,  ..., 1.4982e-03,
           3.7003e-03, 1.1658e-02]],

         [[9.6143e-01, 3.6359e-06, 3.6001e-05,  ..., 4.9477e-03,
           9.5749e-03, 4.0779e-03]],

         [[8.5010e-01, 3.7551e-05, 2.2292e-04,  ..., 7.1716e-03,
           3.6335e-03, 2.0660e-02]],

         ...,

         [[6.3428e-01, 4.7040e-04, 1.8635e-03,  ..., 1.5821e-03,
           4.9400e-03, 3.3112e-02]],

         [[2.6343e-01, 4.1771e-04, 8.1348e-04,  ..., 4.1870e-02,
           1.2917e-02, 3.8574e-02]],

         [[5.7227e-01, 3.1590e-04, 7.3385e-04,  ..., 2.0355e-02,
           3.3813e-02, 4.4128e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 6.8545e-06, 1.6212e-05,  ..., 1.7285e-04,
           3.8338e-04, 3.4790e-02]],

         [[8.7207e-01, 1.1921e-06, 1.5259e-05,  ..., 9.9945e-03,
           1.0199e-01, 1.2650e-02]],

         [[9.1650e-01, 3.8385e-05, 1.3304e-04,  ..., 6.9427e-04,
           2.2110e-02, 1.7593e-02]],

         ...,

         [[2.1680e-01, 1.5008e-04, 7.3016e-05,  ..., 1.0278e-01,
           2.5244e-01, 3.2715e-02]],

         [[9.7839e-02, 2.9011e-03, 2.0599e-03,  ..., 1.5100e-01,
           4.3335e-01, 2.5024e-02]],

         [[4.9292e-01, 8.5294e-05, 1.0532e-04,  ..., 7.5317e-02,
           1.1896e-01, 9.3506e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1807e-01, 7.0715e-04, 2.7585e-04,  ..., 5.5145e-02,
           6.5002e-02, 3.3905e-02]],

         [[3.4277e-01, 3.1829e-05, 3.4571e-05,  ..., 9.7473e-02,
           3.9355e-01, 4.2053e-02]],

         [[7.5586e-01, 9.7990e-05, 1.8477e-05,  ..., 8.3389e-03,
           4.4189e-02, 6.0242e-02]],

         ...,

         [[9.2383e-01, 4.1604e-05, 1.9073e-06,  ..., 1.8835e-05,
           3.4428e-04, 7.1960e-02]],

         [[9.6826e-01, 1.5640e-04, 3.6001e-05,  ..., 2.6822e-06,
           1.0061e-04, 1.0414e-02]],

         [[9.9609e-01, 2.3901e-05, 7.1526e-07,  ..., 5.9605e-08,
           8.7619e-06, 2.3708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1270e-01, 1.6809e-05, 8.7619e-06,  ..., 8.7830e-02,
           2.1753e-01, 2.4368e-02]],

         [[9.6045e-01, 1.4126e-05, 7.3910e-06,  ..., 7.3338e-04,
           1.0042e-03, 1.6953e-02]],

         [[8.2617e-01, 3.5930e-04, 2.9862e-05,  ..., 1.5083e-02,
           5.7335e-03, 5.0507e-02]],

         ...,

         [[9.5605e-01, 1.3530e-04, 4.3571e-05,  ..., 2.7504e-03,
           1.9112e-03, 9.3384e-03]],

         [[9.6533e-01, 7.5996e-05, 1.5736e-05,  ..., 5.0583e-03,
           1.7986e-03, 1.2878e-02]],

         [[6.2012e-01, 2.9850e-04, 1.1092e-04,  ..., 2.3486e-01,
           5.9929e-03, 1.0544e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0264e-01, 3.0339e-05, 3.5048e-05,  ..., 4.9629e-03,
           5.9052e-03, 1.4191e-02]],

         [[7.6318e-01, 7.7486e-05, 6.7592e-05,  ..., 8.9722e-03,
           7.9803e-03, 4.7882e-02]],

         [[8.7695e-01, 1.1253e-04, 3.6240e-05,  ..., 1.0391e-02,
           3.6469e-03, 8.0414e-03]],

         ...,

         [[6.9141e-01, 2.7156e-04, 4.2498e-05,  ..., 1.6449e-02,
           8.6365e-03, 6.1279e-02]],

         [[6.9775e-01, 6.0272e-04, 9.4295e-05,  ..., 2.7939e-02,
           3.9703e-02, 1.7212e-02]],

         [[7.1045e-01, 2.6655e-04, 1.6093e-04,  ..., 3.3722e-02,
           8.2214e-02, 2.9587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2212e-01, 2.8057e-03, 1.1206e-03,  ..., 1.3977e-02,
           3.4119e-02, 1.1032e-02]],

         [[8.1738e-01, 1.5521e-04, 3.4547e-04,  ..., 2.4452e-03,
           1.0376e-02, 5.0774e-03]],

         [[8.7305e-01, 5.6088e-05, 3.4809e-05,  ..., 2.0008e-03,
           3.9005e-03, 6.4049e-03]],

         ...,

         [[7.2168e-01, 4.3488e-04, 5.3930e-04,  ..., 3.4008e-03,
           1.4114e-02, 1.0901e-01]],

         [[8.6328e-01, 1.6832e-04, 6.0976e-05,  ..., 2.5539e-03,
           5.8632e-03, 2.8198e-02]],

         [[9.0771e-01, 2.8706e-04, 1.5128e-04,  ..., 2.2873e-02,
           4.1313e-03, 1.2688e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5610e-01, 2.9087e-04, 7.6056e-04,  ..., 7.0312e-02,
           1.8982e-01, 1.3977e-02]],

         [[8.1689e-01, 2.5988e-05, 1.7273e-04,  ..., 3.4851e-02,
           3.3264e-02, 2.6764e-02]],

         [[8.1006e-01, 7.9095e-05, 2.4843e-04,  ..., 6.9962e-03,
           8.7051e-03, 6.3171e-02]],

         ...,

         [[9.8511e-02, 2.1958e-04, 4.1199e-04,  ..., 3.0518e-02,
           7.4805e-01, 4.4403e-03]],

         [[7.4902e-01, 4.5776e-04, 1.7202e-04,  ..., 7.1167e-02,
           5.9204e-03, 1.0147e-02]],

         [[8.5596e-01, 1.1563e-05, 1.1444e-05,  ..., 4.6158e-03,
           5.9128e-03, 6.5002e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 3.1173e-05, 1.1539e-04,  ..., 9.3102e-05,
           3.3264e-03, 2.9358e-02]],

         [[4.8975e-01, 8.9169e-05, 7.7188e-05,  ..., 6.6162e-02,
           3.3508e-02, 1.0797e-01]],

         [[3.4814e-01, 5.8413e-06, 6.9737e-06,  ..., 9.9468e-04,
           3.7384e-03, 3.9368e-02]],

         ...,

         [[9.1748e-01, 1.8966e-04, 8.4162e-05,  ..., 5.2261e-03,
           5.4817e-03, 9.8724e-03]],

         [[4.1772e-01, 4.2820e-04, 2.6226e-04,  ..., 4.6692e-02,
           2.9633e-02, 2.6196e-01]],

         [[1.8042e-01, 5.5552e-04, 3.6287e-04,  ..., 2.5024e-02,
           4.9316e-02, 3.9520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6367e-01, 4.7684e-06, 7.8082e-06,  ..., 4.7379e-03,
           3.2288e-02, 6.7041e-01]],

         [[7.5537e-01, 6.5231e-04, 8.3590e-04,  ..., 1.6632e-02,
           1.1604e-02, 2.5043e-03]],

         [[8.7988e-01, 4.5228e-04, 5.7936e-04,  ..., 1.0300e-02,
           1.3832e-02, 2.6566e-02]],

         ...,

         [[4.0430e-01, 1.7905e-04, 4.5037e-04,  ..., 1.1139e-02,
           1.9394e-02, 3.8300e-02]],

         [[6.2939e-01, 4.6968e-05, 5.5134e-05,  ..., 9.7580e-03,
           3.8509e-03, 1.9577e-02]],

         [[2.9810e-01, 2.4974e-05, 8.3506e-05,  ..., 5.6610e-03,
           7.0862e-02, 5.5322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.5184e-04, 8.8930e-04,  ..., 1.2741e-02,
           8.3252e-02, 3.8208e-01]],

         [[5.4150e-01, 2.5821e-04, 6.7472e-04,  ..., 3.4851e-02,
           4.5227e-02, 6.7871e-02]],

         [[4.8608e-01, 5.8603e-04, 1.0443e-03,  ..., 4.8790e-03,
           1.3260e-02, 3.3539e-02]],

         ...,

         [[8.2947e-02, 2.1458e-06, 3.8147e-06,  ..., 5.6744e-04,
           6.9847e-03, 8.9844e-01]],

         [[4.8633e-01, 2.9421e-04, 4.8590e-04,  ..., 2.7054e-02,
           5.2246e-02, 1.3940e-01]],

         [[8.7842e-01, 1.3733e-04, 1.0735e-04,  ..., 1.7948e-03,
           3.3016e-03, 3.2940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 3.0899e-04, 3.9220e-04,  ..., 4.8447e-03,
           8.0719e-03, 1.4458e-02]],

         [[9.0088e-01, 2.5570e-05, 5.0724e-05,  ..., 7.8125e-03,
           1.6266e-02, 2.4902e-02]],

         [[5.2734e-01, 1.1134e-04, 2.8324e-04,  ..., 2.5291e-03,
           3.0136e-02, 3.1555e-02]],

         ...,

         [[3.9575e-01, 0.0000e+00, 3.9935e-06,  ..., 2.9099e-02,
           5.2344e-01, 4.5013e-02]],

         [[6.2744e-01, 1.0300e-03, 1.2751e-03,  ..., 1.7365e-02,
           2.9160e-02, 2.2629e-02]],

         [[9.1748e-01, 1.9670e-06, 7.9870e-06,  ..., 9.9945e-03,
           1.9119e-02, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8281e-01, 3.4809e-04, 8.9586e-05,  ..., 7.8011e-04,
           2.2030e-03, 6.1340e-03]],

         [[2.4976e-01, 7.1859e-04, 4.8018e-04,  ..., 1.5417e-01,
           2.8076e-01, 2.7130e-02]],

         [[4.8022e-01, 7.6246e-04, 4.2534e-04,  ..., 5.2643e-02,
           9.0576e-02, 3.1189e-02]],

         ...,

         [[3.5913e-01, 1.0192e-05, 4.1008e-05,  ..., 6.3232e-02,
           2.2168e-01, 8.7769e-02]],

         [[7.6318e-01, 5.7220e-06, 1.8120e-05,  ..., 3.1769e-02,
           8.1543e-02, 6.1340e-02]],

         [[1.6931e-01, 4.3452e-05, 1.0747e-04,  ..., 2.5903e-01,
           3.9795e-01, 2.2171e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7373e-02, 1.2517e-06, 2.5034e-06,  ..., 2.3365e-03,
           4.6112e-02, 8.5645e-01]],

         [[2.1960e-01, 1.0848e-05, 2.0385e-05,  ..., 1.2036e-01,
           4.8730e-01, 6.2195e-02]],

         [[5.0781e-01, 8.9550e-04, 1.1225e-03,  ..., 2.6245e-02,
           6.1493e-02, 1.1353e-02]],

         ...,

         [[1.9678e-01, 2.2161e-04, 2.4390e-04,  ..., 1.5198e-02,
           3.7170e-02, 1.0994e-02]],

         [[4.7925e-01, 5.7518e-05, 4.3273e-05,  ..., 2.8858e-03,
           1.4221e-02, 2.2858e-02]],

         [[9.9365e-02, 3.9959e-04, 2.0254e-04,  ..., 1.1683e-03,
           3.5477e-03, 6.1607e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 2.5487e-04, 3.8195e-04,  ..., 2.2308e-02,
           2.6901e-02, 2.1698e-02]],

         [[2.4841e-01, 4.2319e-04, 3.9911e-04,  ..., 2.3376e-02,
           6.0852e-02, 6.1829e-02]],

         [[1.2061e-01, 2.9659e-04, 1.5235e-04,  ..., 5.6648e-03,
           1.1620e-02, 1.0620e-02]],

         ...,

         [[7.2510e-01, 4.1246e-05, 2.8074e-05,  ..., 1.3485e-03,
           1.0712e-02, 2.8442e-02]],

         [[5.4883e-01, 7.1096e-04, 4.9305e-04,  ..., 6.9504e-03,
           2.4414e-02, 3.5919e-02]],

         [[8.9844e-01, 3.9554e-04, 1.8942e-04,  ..., 2.0199e-03,
           1.0712e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8726e-02, 1.6499e-03, 1.4124e-03,  ..., 5.4053e-01,
           1.1243e-01, 3.4218e-03]],

         [[3.9526e-01, 3.1967e-03, 1.6804e-03,  ..., 2.5818e-02,
           1.9150e-02, 3.1372e-02]],

         [[4.9341e-01, 1.4496e-04, 1.8835e-04,  ..., 1.4755e-02,
           6.8115e-02, 5.9174e-02]],

         ...,

         [[4.9219e-01, 2.8634e-04, 3.5477e-04,  ..., 1.0117e-02,
           2.2141e-02, 1.0400e-01]],

         [[9.8584e-01, 3.3975e-05, 1.0395e-04,  ..., 6.8545e-06,
           2.1577e-04, 2.1572e-03]],

         [[8.6963e-01, 1.3053e-04, 1.2374e-04,  ..., 8.7051e-03,
           1.0895e-02, 2.1179e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1689e-01, 3.5763e-03, 2.9640e-03,  ..., 1.8628e-01,
           1.8054e-01, 7.8857e-02]],

         [[7.9834e-02, 6.8169e-03, 3.1948e-03,  ..., 1.3477e-01,
           1.0095e-01, 2.5131e-02]],

         [[1.0645e-01, 6.4135e-05, 8.6308e-05,  ..., 1.8204e-02,
           1.1688e-01, 4.7314e-01]],

         ...,

         [[1.4807e-01, 5.7995e-05, 8.9586e-05,  ..., 5.6610e-02,
           1.6382e-01, 1.2199e-02]],

         [[7.6953e-01, 1.5020e-05, 2.3246e-05,  ..., 1.0391e-02,
           4.2419e-02, 1.1963e-02]],

         [[2.9590e-01, 1.7631e-04, 2.5058e-04,  ..., 5.1422e-02,
           3.3789e-01, 1.6467e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7061e-01, 2.7180e-05, 2.8014e-05,  ..., 1.1921e-03,
           5.5122e-03, 8.2275e-02]],

         [[2.5269e-01, 1.6081e-04, 7.2539e-05,  ..., 8.8684e-02,
           2.2656e-01, 3.7567e-02]],

         [[3.9526e-01, 6.2644e-05, 6.9857e-05,  ..., 9.0103e-03,
           7.3120e-02, 4.9194e-01]],

         ...,

         [[3.3423e-01, 3.9215e-03, 2.4929e-03,  ..., 6.9885e-03,
           1.9455e-02, 2.8305e-02]],

         [[5.2393e-01, 1.1330e-03, 1.0519e-03,  ..., 2.3743e-02,
           6.9824e-02, 2.4887e-02]],

         [[2.8296e-01, 2.0695e-03, 2.4872e-03,  ..., 4.6570e-02,
           4.9561e-02, 6.8787e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9248e-01, 3.3069e-04, 1.8263e-04,  ..., 1.2006e-01,
           1.2195e-01, 5.8502e-02]],

         [[8.0273e-01, 3.1972e-04, 4.7147e-05,  ..., 1.0674e-02,
           2.5349e-03, 3.6957e-02]],

         [[8.6719e-01, 7.4983e-05, 2.4855e-05,  ..., 1.0338e-02,
           3.7727e-03, 1.0666e-02]],

         ...,

         [[5.8716e-02, 1.7462e-03, 3.2616e-03,  ..., 4.6191e-01,
           1.8372e-01, 1.6571e-02]],

         [[3.8916e-01, 1.1164e-04, 1.0651e-04,  ..., 3.4733e-03,
           2.7740e-02, 2.9541e-02]],

         [[3.0933e-01, 1.5736e-03, 5.0392e-03,  ..., 1.0693e-01,
           1.6052e-01, 4.0253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9907e-01, 4.1914e-04, 9.2983e-04,  ..., 1.7960e-02,
           7.3303e-02, 8.3069e-02]],

         [[1.6553e-01, 5.6982e-04, 1.8387e-03,  ..., 2.8763e-02,
           1.0858e-01, 6.7932e-02]],

         [[2.7661e-01, 1.2004e-04, 5.7745e-04,  ..., 4.2755e-02,
           7.6782e-02, 5.5328e-02]],

         ...,

         [[4.4785e-03, 2.7061e-05, 6.0022e-05,  ..., 2.7599e-03,
           4.4525e-02, 3.8319e-03]],

         [[4.9585e-01, 4.4870e-04, 6.8951e-04,  ..., 4.2229e-03,
           4.5410e-02, 6.6040e-02]],

         [[5.9521e-01, 4.5013e-04, 6.9189e-04,  ..., 6.1188e-03,
           6.7871e-02, 3.5767e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3928e-01, 1.1158e-04, 1.0610e-04,  ..., 2.6932e-03,
           6.9458e-02, 1.4819e-01]],

         [[2.6367e-01, 4.3559e-04, 6.5899e-04,  ..., 6.3171e-02,
           9.4055e-02, 3.5706e-02]],

         [[3.4131e-01, 9.0957e-05, 3.2377e-04,  ..., 1.9257e-02,
           1.2268e-01, 4.0527e-01]],

         ...,

         [[8.3398e-01, 9.0480e-05, 2.4509e-04,  ..., 5.7983e-03,
           1.8005e-02, 1.8723e-02]],

         [[6.3818e-01, 2.4652e-04, 1.9703e-03,  ..., 3.9482e-03,
           1.7548e-02, 5.7098e-02]],

         [[4.1846e-01, 3.3665e-04, 3.5548e-04,  ..., 4.2305e-03,
           1.3344e-02, 2.8473e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9250e-01, 2.5129e-04, 3.4499e-04,  ..., 1.0773e-02,
           8.6731e-02, 1.4526e-01]],

         [[2.4826e-02, 2.6762e-05, 4.7743e-05,  ..., 1.1253e-03,
           1.4366e-02, 1.0109e-02]],

         [[5.1514e-01, 4.6802e-04, 1.1358e-03,  ..., 1.0735e-02,
           4.4312e-02, 4.2969e-02]],

         ...,

         [[6.3525e-01, 1.1921e-07, 1.6153e-05,  ..., 4.0619e-02,
           2.6904e-01, 2.0416e-02]],

         [[3.1763e-01, 1.0371e-05, 7.9811e-05,  ..., 1.3245e-01,
           1.1871e-01, 4.4006e-02]],

         [[5.3320e-01, 7.1526e-07, 8.1241e-05,  ..., 1.0834e-01,
           3.2349e-01, 1.4664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1067e-02, 3.6120e-05, 1.0407e-04,  ..., 2.6367e-02,
           1.9946e-01, 2.8076e-02]],

         [[7.7133e-03, 1.3113e-06, 8.1062e-06,  ..., 4.5967e-04,
           7.9575e-03, 9.6387e-01]],

         [[1.6504e-01, 3.9816e-04, 1.3733e-03,  ..., 6.4636e-02,
           3.9978e-02, 2.6733e-02]],

         ...,

         [[2.7783e-01, 6.9761e-04, 1.2894e-03,  ..., 1.4136e-01,
           2.3035e-01, 2.4857e-02]],

         [[5.9961e-01, 1.3471e-05, 2.3305e-05,  ..., 1.0689e-02,
           7.9895e-02, 7.0496e-03]],

         [[2.1973e-01, 5.6684e-05, 4.4751e-04,  ..., 2.4707e-01,
           4.2358e-01, 6.9946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1885e-01, 1.5497e-04, 5.8937e-04,  ..., 5.2765e-02,
           3.4332e-02, 1.8234e-02]],

         [[9.8340e-01, 2.5094e-05, 1.6844e-04,  ..., 2.4471e-03,
           8.2922e-04, 2.7990e-04]],

         [[2.2925e-01, 5.7042e-05, 2.9254e-04,  ..., 4.4518e-03,
           6.9771e-03, 2.3556e-03]],

         ...,

         [[6.6162e-01, 4.4405e-05, 7.0989e-05,  ..., 5.6801e-03,
           1.5442e-02, 3.2196e-02]],

         [[1.0175e-01, 1.2875e-05, 1.0431e-05,  ..., 2.3384e-03,
           1.0727e-02, 2.4551e-02]],

         [[1.4062e-01, 3.1173e-05, 5.9605e-05,  ..., 1.6785e-02,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5759e-01, 5.1498e-05, 1.0884e-04,  ..., 6.9092e-02,
           1.8567e-01, 7.6782e-02]],

         [[6.9873e-01, 7.7665e-05, 1.1289e-04,  ..., 8.6212e-03,
           1.5587e-02, 3.3722e-02]],

         [[2.4121e-01, 1.0526e-04, 4.7612e-04,  ..., 6.2927e-02,
           1.3745e-01, 3.3740e-01]],

         ...,

         [[5.8252e-01, 1.0622e-04, 4.8256e-04,  ..., 3.6957e-02,
           6.2347e-02, 5.1697e-02]],

         [[3.8867e-01, 7.2718e-05, 6.4945e-04,  ..., 2.1255e-02,
           1.0223e-01, 1.7798e-01]],

         [[5.7129e-01, 1.6034e-05, 9.0897e-05,  ..., 4.4327e-03,
           2.2507e-02, 6.0211e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4465e-02, 9.8348e-06, 1.0252e-05,  ..., 3.3760e-04,
           3.8338e-03, 1.3275e-02]],

         [[2.6642e-02, 3.7849e-05, 5.5194e-05,  ..., 1.1940e-03,
           1.5793e-02, 4.5319e-02]],

         [[1.0657e-01, 1.5163e-04, 6.7711e-04,  ..., 2.6321e-02,
           1.0992e-01, 2.6782e-01]],

         ...,

         [[1.1658e-01, 3.7313e-05, 3.3021e-04,  ..., 2.7979e-01,
           2.7979e-01, 9.3689e-02]],

         [[2.2241e-01, 1.7345e-04, 1.1673e-03,  ..., 1.0504e-01,
           2.0251e-01, 1.6272e-01]],

         [[9.4775e-01, 4.6909e-05, 1.2851e-04,  ..., 5.5504e-03,
           7.1259e-03, 1.1566e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.4617e-05, 3.8505e-05,  ..., 2.0401e-02,
           2.9739e-02, 2.5732e-01]],

         [[1.9263e-01, 5.2333e-05, 1.1748e-04,  ..., 1.1548e-01,
           4.5142e-01, 1.0229e-01]],

         [[1.0815e-01, 2.5034e-06, 2.3603e-05,  ..., 9.5764e-02,
           6.8359e-01, 8.4229e-02]],

         ...,

         [[3.1421e-01, 2.6655e-04, 1.6613e-03,  ..., 1.4465e-02,
           3.0869e-02, 7.2876e-02]],

         [[3.3569e-01, 2.7537e-04, 5.0068e-04,  ..., 4.5074e-02,
           9.3201e-02, 1.9531e-02]],

         [[3.7018e-02, 1.0234e-04, 1.0520e-04,  ..., 3.1338e-03,
           2.9739e-02, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4297e-01, 6.4254e-05, 4.7040e-04,  ..., 8.4229e-02,
           8.8623e-02, 1.0565e-01]],

         [[1.6418e-01, 3.1147e-03, 4.4098e-03,  ..., 1.3420e-02,
           3.7506e-02, 4.7211e-02]],

         [[2.8702e-02, 3.5524e-05, 1.2290e-04,  ..., 2.1973e-03,
           1.5610e-02, 9.0332e-03]],

         ...,

         [[8.9795e-01, 1.2648e-04, 1.7846e-04,  ..., 5.0926e-03,
           4.8599e-03, 1.3008e-02]],

         [[1.8799e-01, 1.1420e-04, 2.3067e-04,  ..., 8.1329e-03,
           4.8279e-02, 7.0947e-01]],

         [[6.0303e-01, 4.1068e-05, 1.9598e-04,  ..., 3.7933e-02,
           9.1003e-02, 1.6748e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 9.4414e-05, 1.9455e-03,  ..., 1.9379e-02,
           1.5602e-02, 4.4952e-02]],

         [[6.6772e-02, 6.5863e-05, 2.9874e-04,  ..., 7.7591e-03,
           1.5434e-02, 3.8055e-02]],

         [[2.1533e-01, 2.0103e-03, 4.0092e-03,  ..., 1.9763e-01,
           1.0956e-01, 8.3374e-02]],

         ...,

         [[7.7490e-01, 2.6464e-04, 7.2718e-04,  ..., 1.3756e-02,
           1.3786e-02, 3.1799e-02]],

         [[7.5879e-01, 1.0073e-05, 1.9302e-03,  ..., 1.6891e-02,
           1.4732e-02, 1.6464e-02]],

         [[6.7041e-01, 6.9261e-05, 1.6987e-04,  ..., 1.4671e-02,
           1.4503e-02, 5.2826e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9888e-03, 2.8400e-03, 6.5918e-02,  ..., 4.1260e-02,
           2.8122e-02, 4.6021e-02]],

         [[2.5344e-04, 2.6870e-04, 1.3086e-01,  ..., 8.4991e-03,
           1.4069e-02, 5.0873e-02]],

         [[5.5075e-04, 5.0402e-04, 7.9651e-02,  ..., 1.6769e-02,
           2.7740e-02, 1.0803e-01]],

         ...,

         [[1.6069e-04, 9.0122e-05, 1.5186e-01,  ..., 5.1270e-02,
           1.6174e-01, 3.0688e-01]],

         [[6.6757e-04, 6.5470e-04, 2.4475e-01,  ..., 4.3365e-02,
           5.0110e-02, 6.6895e-02]],

         [[1.4663e-05, 2.5630e-06, 1.0864e-02,  ..., 4.3130e-04,
           1.1816e-03, 7.4654e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3188e-04, 1.4496e-04, 2.1484e-01,  ..., 3.5973e-03,
           1.5564e-02, 1.0309e-01]],

         [[2.3603e-04, 1.3721e-04, 1.0962e-01,  ..., 1.4717e-02,
           9.2850e-03, 4.7150e-02]],

         [[1.4353e-04, 2.9266e-05, 2.9468e-01,  ..., 4.8370e-03,
           1.3405e-02, 7.5073e-02]],

         ...,

         [[1.0796e-03, 4.0269e-04, 1.4172e-01,  ..., 1.4839e-03,
           9.7961e-03, 5.5695e-02]],

         [[8.3447e-05, 9.7752e-05, 2.5317e-01,  ..., 2.7485e-03,
           9.1553e-03, 7.1411e-02]],

         [[4.3690e-05, 4.1425e-05, 2.5781e-01,  ..., 3.5736e-02,
           1.4026e-01, 1.0339e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-05, 9.8944e-05, 1.6516e-01,  ..., 5.2214e-04,
           1.5945e-02, 6.2073e-02]],

         [[1.6975e-04, 7.3910e-06, 7.6721e-02,  ..., 4.3983e-03,
           6.9695e-03, 4.4067e-02]],

         [[3.8743e-04, 4.8423e-04, 1.7200e-01,  ..., 1.1520e-03,
           9.2621e-03, 6.1829e-02]],

         ...,

         [[5.0664e-06, 1.5676e-05, 1.7871e-01,  ..., 4.5090e-03,
           1.4328e-02, 1.5527e-01]],

         [[4.2796e-05, 1.1128e-04, 3.9398e-02,  ..., 3.4409e-03,
           8.9951e-03, 3.1281e-02]],

         [[6.0797e-04, 1.4114e-04, 1.0101e-01,  ..., 2.3479e-03,
           8.5220e-03, 5.1605e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3358e-04, 1.0800e-04, 1.9638e-02,  ..., 9.9564e-03,
           1.3525e-01, 5.3516e-01]],

         [[5.8830e-05, 2.2531e-04, 2.3608e-01,  ..., 6.7383e-02,
           1.3196e-01, 6.2561e-02]],

         [[5.5850e-05, 3.8290e-04, 2.9468e-01,  ..., 8.1848e-02,
           4.2938e-02, 5.5046e-03]],

         ...,

         [[7.4053e-04, 5.3835e-04, 8.2458e-02,  ..., 1.1316e-01,
           2.7084e-02, 8.7769e-02]],

         [[1.6088e-03, 3.5715e-04, 4.0619e-02,  ..., 2.0703e-01,
           1.8701e-01, 3.6060e-01]],

         [[1.6534e-04, 3.7074e-05, 5.3558e-02,  ..., 2.2507e-02,
           7.1228e-02, 1.6174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8553e-03, 9.3699e-04, 1.3599e-01,  ..., 1.5015e-02,
           3.7323e-02, 7.0801e-02]],

         [[2.8276e-04, 9.8038e-04, 2.4902e-01,  ..., 5.4779e-03,
           2.7496e-02, 7.7759e-02]],

         [[1.3762e-03, 5.2786e-04, 1.8665e-01,  ..., 1.8478e-02,
           4.2725e-02, 4.0039e-02]],

         ...,

         [[2.0313e-03, 2.7919e-04, 1.2372e-01,  ..., 1.3458e-02,
           2.0203e-02, 3.7524e-01]],

         [[6.6805e-04, 1.4706e-03, 2.5732e-01,  ..., 8.2493e-04,
           4.6005e-03, 1.4771e-01]],

         [[9.2125e-04, 7.6962e-04, 1.5979e-01,  ..., 4.4037e-02,
           4.7607e-02, 1.6235e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.7180e-05, 6.6757e-06, 2.6054e-03,  ..., 2.2729e-01,
           4.5215e-01, 6.7200e-02]],

         [[5.5122e-04, 2.9516e-04, 4.3373e-03,  ..., 1.2671e-01,
           2.2241e-01, 3.9038e-01]],

         [[0.0000e+00, 0.0000e+00, 4.7684e-07,  ..., 2.4395e-03,
           1.0933e-02, 9.8438e-01]],

         ...,

         [[1.6510e-05, 3.6955e-06, 4.4882e-05,  ..., 2.4509e-03,
           2.4509e-03, 9.8877e-01]],

         [[6.0043e-03, 8.2016e-03, 5.2986e-03,  ..., 2.3743e-02,
           1.6479e-01, 1.0536e-02]],

         [[9.8495e-03, 3.5114e-03, 8.8787e-04,  ..., 5.9738e-03,
           2.9388e-02, 7.7454e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7980e-04, 6.5613e-04, 6.0349e-03,  ..., 7.2388e-02,
           1.2311e-01, 5.5176e-01]],

         [[0.0000e+00, 0.0000e+00, 9.5367e-07,  ..., 1.3989e-01,
           7.8027e-01, 7.2571e-02]],

         [[7.8440e-04, 1.3552e-03, 1.2077e-02,  ..., 1.4954e-01,
           1.4038e-01, 9.3567e-02]],

         ...,

         [[2.1458e-03, 3.8261e-03, 7.0381e-03,  ..., 2.7405e-02,
           1.5161e-01, 5.1709e-01]],

         [[2.3270e-03, 9.8572e-03, 3.6194e-02,  ..., 1.6040e-01,
           1.5002e-01, 2.2919e-02]],

         [[1.3390e-03, 2.7905e-03, 1.1841e-02,  ..., 7.4280e-02,
           3.2764e-01, 2.1155e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8441e-02, 7.9727e-03, 2.9617e-02,  ..., 4.4464e-02,
           8.0505e-02, 4.6234e-02]],

         [[8.7976e-04, 4.2868e-04, 1.7023e-03,  ..., 1.9943e-02,
           5.8517e-03, 2.2907e-03]],

         [[1.5053e-02, 5.0011e-03, 6.7825e-03,  ..., 1.3098e-01,
           5.9998e-02, 3.3661e-02]],

         ...,

         [[2.4490e-03, 1.9407e-04, 1.0328e-03,  ..., 1.6003e-01,
           2.4585e-01, 3.2080e-01]],

         [[1.9760e-02, 3.8166e-03, 4.8637e-03,  ..., 1.4270e-01,
           1.7896e-01, 7.8186e-02]],

         [[7.3776e-03, 7.3051e-04, 1.2070e-02,  ..., 1.5173e-01,
           2.8564e-01, 1.7188e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.8014e-06, 9.3877e-05,  ..., 4.7035e-03,
           3.1033e-03, 1.6052e-02]],

         [[9.2188e-01, 4.7684e-07, 2.8551e-05,  ..., 2.7115e-02,
           2.1561e-02, 1.1696e-02]],

         [[9.7314e-01, 1.2338e-05, 4.2915e-05,  ..., 1.2150e-03,
           3.4161e-03, 1.8320e-03]],

         ...,

         [[2.9114e-02, 1.9670e-05, 1.2422e-04,  ..., 1.6713e-04,
           1.6129e-04, 2.3708e-03]],

         [[6.5869e-01, 4.5204e-04, 1.1892e-03,  ..., 1.1505e-02,
           1.3557e-02, 7.7095e-03]],

         [[3.0103e-01, 3.4165e-04, 3.7813e-04,  ..., 1.4084e-02,
           2.0721e-02, 2.6718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9575e-01, 1.5497e-06, 4.8280e-06,  ..., 2.8706e-03,
           5.1928e-04, 9.1782e-03]],

         [[8.4668e-01, 0.0000e+00, 5.9605e-08,  ..., 6.9275e-03,
           9.5093e-02, 1.6403e-02]],

         [[9.2285e-01, 3.2485e-05, 1.3232e-05,  ..., 1.8244e-03,
           3.4668e-02, 3.6697e-03]],

         ...,

         [[2.8174e-01, 6.0129e-04, 1.8942e-04,  ..., 2.4573e-01,
           9.2529e-02, 6.6490e-03]],

         [[8.8745e-02, 9.9087e-04, 4.1199e-04,  ..., 3.1226e-01,
           8.8074e-02, 5.6343e-03]],

         [[6.3086e-01, 2.8551e-05, 8.0347e-05,  ..., 3.4576e-02,
           1.9543e-01, 5.5450e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7456e-01, 2.7227e-04, 1.5688e-04,  ..., 1.2793e-01,
           1.0730e-01, 7.1228e-02]],

         [[7.1680e-01, 1.7166e-05, 2.2590e-05,  ..., 7.8613e-02,
           1.3110e-01, 1.5747e-02]],

         [[8.8135e-01, 2.1160e-05, 1.2159e-05,  ..., 7.8583e-03,
           6.3232e-02, 1.5236e-02]],

         ...,

         [[4.9756e-01, 1.0133e-06, 2.9802e-07,  ..., 3.5524e-05,
           5.1117e-04, 4.8145e-01]],

         [[6.7773e-01, 8.7023e-06, 8.7023e-06,  ..., 8.1658e-06,
           2.0742e-05, 7.0953e-03]],

         [[9.7559e-01, 8.3447e-07, 5.9605e-07,  ..., 2.3842e-07,
           7.1526e-07, 1.2865e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7578e-01, 1.4842e-05, 5.0068e-06,  ..., 4.9683e-02,
           1.0162e-01, 1.1285e-01]],

         [[8.7891e-01, 1.5736e-05, 3.4571e-06,  ..., 2.4986e-04,
           5.3072e-04, 1.1620e-02]],

         [[7.2998e-01, 2.4629e-04, 5.7578e-05,  ..., 2.6531e-03,
           1.4641e-02, 9.8877e-02]],

         ...,

         [[8.9600e-01, 3.7122e-04, 8.9884e-05,  ..., 1.6317e-03,
           6.7444e-03, 1.5335e-02]],

         [[9.4385e-01, 6.4075e-05, 1.6570e-05,  ..., 1.7319e-03,
           1.5617e-02, 2.3590e-02]],

         [[9.1406e-01, 1.4281e-04, 1.8275e-04,  ..., 1.1349e-03,
           7.4196e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3838e-01, 6.2227e-05, 8.5771e-05,  ..., 2.3003e-03,
           2.0180e-03, 1.6739e-02]],

         [[7.6416e-01, 2.6441e-04, 8.2552e-05,  ..., 4.5433e-03,
           5.7220e-03, 2.9053e-02]],

         [[9.8486e-01, 1.0729e-06, 1.0133e-06,  ..., 4.5252e-04,
           1.5545e-03, 6.7673e-03]],

         ...,

         [[6.1963e-01, 1.0067e-04, 3.0339e-05,  ..., 1.0246e-02,
           1.1011e-01, 3.1403e-02]],

         [[9.1553e-01, 2.5439e-04, 2.8074e-05,  ..., 8.0032e-03,
           7.4348e-03, 2.8439e-03]],

         [[8.8232e-01, 1.6034e-05, 4.2558e-05,  ..., 5.9814e-03,
           2.6443e-02, 5.1758e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2837e-01, 9.7132e-04, 2.5535e-04,  ..., 1.2352e-02,
           6.6223e-03, 3.1357e-03]],

         [[9.8145e-01, 7.9274e-06, 1.3828e-05,  ..., 1.1263e-03,
           3.5496e-03, 2.0027e-04]],

         [[9.3018e-01, 8.3447e-06, 1.1265e-05,  ..., 3.2663e-04,
           3.5596e-04, 8.4734e-04]],

         ...,

         [[8.0420e-01, 2.9027e-05, 2.4438e-05,  ..., 1.2474e-03,
           1.6068e-02, 9.5581e-02]],

         [[9.6484e-01, 1.6093e-06, 2.7418e-06,  ..., 1.5621e-03,
           1.3039e-02, 6.5460e-03]],

         [[9.2334e-01, 1.1444e-05, 1.0967e-05,  ..., 3.9864e-04,
           1.2112e-03, 1.6281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9712e-02, 3.3200e-05, 2.7776e-04,  ..., 2.6880e-01,
           1.4612e-01, 1.7319e-02]],

         [[9.5410e-01, 4.3511e-06, 1.4246e-05,  ..., 6.6147e-03,
           6.6414e-03, 4.3030e-03]],

         [[7.9980e-01, 1.3232e-05, 5.0843e-05,  ..., 2.2202e-03,
           2.3132e-02, 2.6321e-02]],

         ...,

         [[9.0088e-01, 2.5034e-05, 4.3392e-05,  ..., 3.8757e-02,
           2.3117e-03, 1.0548e-03]],

         [[8.0859e-01, 9.6381e-05, 4.3750e-05,  ..., 1.0500e-03,
           1.6127e-03, 1.8921e-02]],

         [[8.5645e-01, 5.6028e-06, 7.3314e-06,  ..., 2.6264e-03,
           3.3478e-02, 2.3289e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6992e-01, 8.3447e-07, 5.3644e-06,  ..., 7.6294e-04,
           1.4853e-04, 9.4986e-03]],

         [[7.8125e-01, 4.4048e-05, 5.0724e-05,  ..., 2.4155e-02,
           5.7617e-02, 7.3242e-03]],

         [[7.2998e-01, 0.0000e+00, 1.7881e-07,  ..., 2.7161e-03,
           1.4200e-03, 1.5480e-02]],

         ...,

         [[7.8760e-01, 1.9073e-04, 8.5950e-05,  ..., 1.7471e-02,
           1.7242e-02, 6.6299e-03]],

         [[6.2939e-01, 7.6771e-05, 9.4414e-05,  ..., 2.6073e-03,
           5.6580e-02, 4.6844e-02]],

         [[3.2715e-01, 5.6839e-04, 1.0914e-04,  ..., 3.5339e-02,
           1.3832e-02, 1.0971e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1667e-01, 7.7486e-06, 5.7220e-06,  ..., 6.3944e-04,
           3.8280e-03, 7.6807e-01]],

         [[4.8730e-01, 3.8123e-04, 4.8780e-04,  ..., 1.5793e-02,
           1.8492e-03, 1.6098e-02]],

         [[7.8076e-01, 2.1708e-04, 5.3883e-04,  ..., 2.4475e-02,
           1.6266e-02, 5.2605e-03]],

         ...,

         [[6.6260e-01, 2.0468e-04, 2.2054e-04,  ..., 3.3493e-03,
           7.5188e-03, 2.8934e-03]],

         [[8.5107e-01, 1.6749e-05, 2.8253e-05,  ..., 2.0771e-03,
           9.2850e-03, 5.4138e-02]],

         [[4.7046e-01, 8.9407e-06, 3.8803e-05,  ..., 6.8893e-03,
           5.1849e-02, 3.6035e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9038e-01, 6.0797e-05, 8.5413e-05,  ..., 2.2812e-03,
           9.2010e-03, 3.3887e-01]],

         [[3.8428e-01, 1.8537e-04, 1.5795e-04,  ..., 1.6037e-02,
           1.2170e-01, 3.8109e-03]],

         [[4.4287e-01, 9.9778e-05, 5.0962e-05,  ..., 1.1377e-03,
           4.0016e-03, 7.9422e-03]],

         ...,

         [[1.6113e-02, 2.5630e-06, 2.1458e-06,  ..., 3.6061e-05,
           9.1219e-04, 8.0713e-01]],

         [[4.7095e-01, 5.9366e-05, 9.9421e-05,  ..., 1.8692e-03,
           2.1103e-02, 5.7983e-02]],

         [[5.0879e-01, 9.1195e-05, 1.1802e-05,  ..., 5.9223e-04,
           1.1482e-03, 7.0496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3564e-01, 1.0605e-03, 5.5695e-04,  ..., 6.6910e-03,
           1.0475e-02, 3.4103e-03]],

         [[6.5967e-01, 1.0806e-04, 3.9387e-04,  ..., 5.1636e-02,
           1.7249e-01, 5.6648e-03]],

         [[4.5825e-01, 1.4007e-05, 6.0081e-05,  ..., 1.2360e-02,
           1.1948e-02, 6.9466e-03]],

         ...,

         [[3.6035e-01, 0.0000e+00, 5.9605e-08,  ..., 5.8197e-02,
           5.0342e-01, 4.3701e-02]],

         [[4.0845e-01, 7.9441e-04, 5.9414e-04,  ..., 2.4597e-02,
           1.6693e-02, 8.3542e-03]],

         [[8.9014e-01, 2.9802e-07, 6.5565e-07,  ..., 2.5043e-03,
           1.4832e-02, 5.3955e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4180e-01, 6.5267e-05, 1.2159e-05,  ..., 7.4673e-04,
           3.5686e-03, 9.0942e-03]],

         [[6.9031e-02, 1.2791e-04, 1.0622e-04,  ..., 2.9590e-01,
           3.1250e-01, 1.1663e-03]],

         [[1.5662e-01, 1.3053e-04, 6.0618e-05,  ..., 1.9684e-02,
           1.8530e-01, 2.0103e-03]],

         ...,

         [[7.0996e-01, 6.5565e-07, 2.8014e-06,  ..., 4.4922e-02,
           1.0400e-01, 6.3843e-02]],

         [[5.9180e-01, 1.7881e-07, 4.1723e-07,  ..., 1.6464e-02,
           1.8066e-01, 1.7554e-01]],

         [[2.9761e-01, 3.0398e-06, 2.8014e-06,  ..., 2.6025e-01,
           2.7734e-01, 1.4610e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7415e-02, 5.3644e-07, 8.3447e-07,  ..., 2.3594e-03,
           4.6272e-03, 9.4287e-01]],

         [[1.0089e-01, 1.1981e-05, 3.6478e-05,  ..., 2.7759e-01,
           4.5581e-01, 3.5980e-02]],

         [[4.3066e-01, 3.6931e-04, 4.5919e-04,  ..., 9.7229e-02,
           2.3102e-02, 8.4457e-03]],

         ...,

         [[2.4072e-01, 8.1062e-05, 4.0591e-05,  ..., 4.2534e-03,
           6.6757e-03, 9.6436e-03]],

         [[5.9180e-01, 1.2994e-05, 1.2994e-05,  ..., 1.0086e-02,
           1.1627e-02, 1.0040e-02]],

         [[4.3213e-01, 1.0967e-04, 6.2227e-05,  ..., 9.2554e-04,
           4.5276e-04, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1025e-01, 5.8353e-05, 9.2328e-05,  ..., 8.8120e-03,
           4.5624e-02, 4.4739e-02]],

         [[3.5400e-01, 9.4318e-04, 4.1389e-04,  ..., 6.8207e-03,
           1.6495e-02, 1.7151e-02]],

         [[2.5488e-01, 4.0650e-05, 1.6510e-05,  ..., 3.0746e-03,
           1.0483e-02, 1.9653e-02]],

         ...,

         [[6.9775e-01, 1.8895e-05, 1.4246e-05,  ..., 1.6966e-03,
           1.2489e-02, 9.8801e-03]],

         [[5.0977e-01, 3.7909e-05, 3.4928e-05,  ..., 4.8599e-03,
           5.2063e-02, 8.8348e-03]],

         [[7.3584e-01, 5.2452e-05, 3.2842e-05,  ..., 5.7831e-03,
           6.3667e-03, 1.7105e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8174e-01, 6.5708e-04, 4.8828e-04,  ..., 9.0027e-02,
           1.3939e-02, 2.7847e-03]],

         [[3.3618e-01, 1.2314e-04, 6.6519e-05,  ..., 7.1669e-04,
           5.6553e-04, 6.6681e-03]],

         [[6.1230e-01, 1.3947e-05, 1.7226e-05,  ..., 7.7362e-03,
           2.4872e-02, 1.5656e-02]],

         ...,

         [[1.6382e-01, 2.7990e-04, 6.9797e-05,  ..., 1.8959e-03,
           7.4272e-03, 6.2927e-02]],

         [[9.5398e-02, 3.6538e-05, 3.0458e-05,  ..., 3.2187e-05,
           3.7372e-05, 6.7520e-03]],

         [[8.5059e-01, 1.7929e-04, 8.3387e-05,  ..., 4.0665e-03,
           7.0496e-03, 7.6818e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2441e-01, 2.6393e-04, 3.2592e-04,  ..., 5.6122e-02,
           2.5562e-01, 7.1907e-03]],

         [[2.5317e-01, 7.7820e-04, 3.5620e-04,  ..., 1.3741e-02,
           1.0658e-02, 1.6647e-02]],

         [[2.2253e-01, 1.2279e-05, 1.1265e-05,  ..., 4.7646e-03,
           3.6041e-02, 2.4060e-01]],

         ...,

         [[5.0195e-01, 1.6868e-05, 1.7285e-05,  ..., 4.1687e-02,
           3.7079e-02, 1.4580e-02]],

         [[8.2031e-01, 9.4771e-06, 1.1563e-05,  ..., 4.4403e-03,
           7.6141e-03, 3.4119e-02]],

         [[2.0007e-01, 3.3438e-05, 2.7061e-05,  ..., 8.7891e-03,
           1.3298e-02, 3.0518e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 8.3089e-05, 4.5180e-05,  ..., 2.0771e-03,
           1.5106e-02, 1.5587e-02]],

         [[2.0837e-01, 3.3021e-05, 4.8041e-05,  ..., 1.3245e-01,
           6.8726e-02, 1.3321e-02]],

         [[1.2262e-01, 7.2718e-06, 3.4571e-06,  ..., 2.5864e-03,
           3.0231e-03, 7.9980e-01]],

         ...,

         [[1.6931e-01, 2.3723e-04, 1.3208e-04,  ..., 1.1148e-03,
           3.8958e-04, 2.3453e-02]],

         [[4.9487e-01, 3.7575e-04, 2.0945e-04,  ..., 5.0468e-03,
           6.1569e-03, 9.1705e-03]],

         [[3.5010e-01, 6.6280e-04, 5.2023e-04,  ..., 4.3907e-03,
           8.3313e-03, 4.5471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7920e-01, 5.1260e-05, 2.3842e-05,  ..., 1.9653e-02,
           1.7883e-02, 5.1758e-02]],

         [[2.4976e-01, 2.1338e-05, 4.5300e-06,  ..., 6.2523e-03,
           9.0075e-04, 7.5417e-03]],

         [[8.0615e-01, 5.3644e-05, 2.4676e-05,  ..., 8.7509e-03,
           1.8196e-03, 2.9297e-03]],

         ...,

         [[1.3147e-01, 1.3943e-03, 1.9665e-03,  ..., 1.6882e-01,
           1.7120e-02, 7.8354e-03]],

         [[8.1201e-01, 2.8014e-05, 5.7101e-05,  ..., 1.6680e-03,
           1.0443e-03, 2.6505e-02]],

         [[4.9976e-01, 9.0647e-04, 1.6289e-03,  ..., 8.5510e-02,
           3.1464e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9321e-01, 3.0303e-04, 5.8842e-04,  ..., 4.8615e-02,
           4.2908e-02, 1.3924e-02]],

         [[1.9165e-01, 3.5286e-04, 4.4966e-04,  ..., 5.3192e-02,
           2.5925e-02, 3.3295e-02]],

         [[2.3865e-01, 5.0664e-05, 1.4043e-04,  ..., 2.2537e-02,
           1.9745e-02, 2.6978e-02]],

         ...,

         [[5.1208e-02, 8.5890e-05, 1.5068e-04,  ..., 2.4185e-02,
           9.9258e-03, 1.2352e-02]],

         [[6.4087e-02, 4.7684e-05, 4.8459e-05,  ..., 7.1192e-04,
           1.3514e-03, 5.7434e-02]],

         [[3.8501e-01, 2.5702e-04, 5.0688e-04,  ..., 1.7731e-02,
           1.1269e-02, 9.9487e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1924e-01, 2.1756e-05, 2.0921e-05,  ..., 3.8319e-03,
           4.5166e-03, 3.7811e-02]],

         [[3.2227e-01, 1.3459e-04, 2.7084e-04,  ..., 2.0599e-02,
           3.1647e-02, 3.5248e-03]],

         [[2.5122e-01, 2.5439e-04, 5.8031e-04,  ..., 6.9702e-02,
           9.0942e-02, 3.1006e-01]],

         ...,

         [[4.9463e-01, 1.6403e-04, 2.8014e-04,  ..., 3.7262e-02,
           1.7319e-02, 2.7039e-02]],

         [[1.1444e-01, 4.7088e-06, 5.0843e-05,  ..., 6.5422e-04,
           8.3351e-04, 7.0496e-02]],

         [[1.9141e-01, 2.8563e-04, 5.3358e-04,  ..., 4.9057e-03,
           4.0665e-03, 1.0777e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6636e-01, 2.3234e-04, 1.7917e-04,  ..., 1.3741e-02,
           9.5139e-03, 9.4238e-02]],

         [[1.0718e-01, 8.6427e-05, 1.2851e-04,  ..., 1.4503e-02,
           8.7280e-03, 7.2510e-02]],

         [[3.5571e-01, 2.5153e-04, 5.8508e-04,  ..., 2.6688e-02,
           1.9516e-02, 3.3356e-02]],

         ...,

         [[7.1680e-01, 0.0000e+00, 1.1921e-07,  ..., 6.2164e-02,
           1.3257e-01, 7.3792e-02]],

         [[4.2236e-01, 9.5367e-06, 4.2677e-05,  ..., 1.7883e-01,
           1.7200e-01, 3.1067e-02]],

         [[2.0715e-01, 1.7881e-07, 3.5763e-06,  ..., 5.4016e-02,
           7.0068e-01, 1.8677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0742e-02, 2.3842e-06, 9.5367e-06,  ..., 5.1239e-02,
           1.7853e-02, 4.0527e-02]],

         [[9.8114e-03, 3.5763e-07, 2.3842e-07,  ..., 8.5533e-05,
           1.9779e-03, 9.7754e-01]],

         [[5.3864e-02, 1.6809e-04, 7.6246e-04,  ..., 1.5222e-01,
           6.3965e-02, 2.5578e-03]],

         ...,

         [[1.6907e-01, 3.0899e-04, 7.9298e-04,  ..., 3.4277e-01,
           5.4688e-02, 7.5455e-03]],

         [[4.7681e-01, 1.0848e-05, 3.5822e-05,  ..., 7.6660e-02,
           1.2537e-01, 1.3928e-01]],

         [[1.0913e-01, 8.9407e-07, 2.1279e-05,  ..., 1.7993e-01,
           4.4873e-01, 1.8555e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 7.6294e-05, 3.8791e-04,  ..., 4.9042e-02,
           2.1515e-02, 1.2253e-02]],

         [[9.5264e-01, 2.7895e-05, 1.1194e-04,  ..., 2.8248e-03,
           3.9005e-04, 5.7411e-04]],

         [[7.7271e-02, 8.8811e-06, 4.3094e-05,  ..., 2.9964e-03,
           9.0647e-04, 1.2989e-03]],

         ...,

         [[7.6709e-01, 2.2471e-05, 2.6941e-05,  ..., 9.8114e-03,
           6.7017e-02, 4.3976e-02]],

         [[2.3840e-01, 3.8147e-06, 2.3246e-06,  ..., 5.1880e-03,
           4.3335e-03, 7.3700e-03]],

         [[7.8308e-02, 1.5557e-05, 3.9756e-05,  ..., 6.8970e-03,
           9.2087e-03, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9873e-01, 1.4663e-05, 2.6286e-05,  ..., 2.1826e-01,
           2.0178e-01, 3.8086e-02]],

         [[7.8467e-01, 2.1338e-05, 1.6749e-05,  ..., 4.7417e-03,
           1.9196e-02, 6.4148e-02]],

         [[2.4585e-01, 5.3823e-05, 9.3400e-05,  ..., 8.3984e-02,
           1.5503e-01, 2.7100e-01]],

         ...,

         [[5.6836e-01, 1.0788e-05, 6.8665e-05,  ..., 1.8997e-02,
           1.5784e-01, 5.5847e-02]],

         [[5.9717e-01, 2.8586e-04, 1.0414e-03,  ..., 6.3904e-02,
           6.3416e-02, 1.3504e-02]],

         [[6.3867e-01, 2.4617e-05, 1.5676e-04,  ..., 3.4393e-02,
           1.5417e-01, 1.1345e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7942e-02, 6.2227e-05, 3.7074e-05,  ..., 2.6684e-03,
           8.5449e-03, 5.6610e-02]],

         [[1.1206e-01, 4.6313e-05, 3.8505e-05,  ..., 7.1335e-03,
           8.9874e-03, 1.6144e-02]],

         [[2.4146e-01, 1.9741e-04, 6.1750e-04,  ..., 4.0985e-02,
           1.1407e-01, 1.1139e-01]],

         ...,

         [[6.8298e-02, 2.2948e-05, 1.9038e-04,  ..., 3.6377e-01,
           8.5022e-02, 1.8349e-03]],

         [[1.5430e-01, 7.8917e-05, 3.4285e-04,  ..., 2.2095e-01,
           3.7012e-01, 3.8033e-03]],

         [[5.8936e-01, 1.4019e-04, 2.8110e-04,  ..., 4.1382e-02,
           1.7554e-01, 7.5500e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2090e-01, 1.7583e-05, 5.3525e-05,  ..., 1.0727e-02,
           1.0400e-01, 3.6157e-01]],

         [[3.3594e-01, 6.7294e-05, 7.1347e-05,  ..., 1.3416e-01,
           7.9468e-02, 2.4866e-01]],

         [[5.4779e-02, 8.9407e-07, 1.1921e-06,  ..., 2.3889e-01,
           6.3428e-01, 2.9205e-02]],

         ...,

         [[5.9424e-01, 1.1760e-04, 2.6703e-04,  ..., 6.7291e-03,
           1.3916e-02, 1.0674e-02]],

         [[4.2188e-01, 2.9683e-05, 7.0632e-05,  ..., 3.7750e-02,
           6.5186e-02, 7.3738e-03]],

         [[2.1667e-02, 2.9802e-05, 1.9729e-05,  ..., 3.0022e-03,
           3.4882e-02, 8.0664e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2051e-01, 3.0041e-05, 3.3259e-04,  ..., 6.1676e-02,
           9.1858e-02, 1.5625e-01]],

         [[6.1401e-02, 3.2163e-04, 1.0405e-03,  ..., 1.1009e-02,
           1.5526e-02, 1.1673e-02]],

         [[2.5562e-01, 1.2159e-04, 4.8184e-04,  ..., 3.9185e-02,
           2.1301e-02, 3.9490e-02]],

         ...,

         [[8.3203e-01, 3.6407e-04, 2.5225e-04,  ..., 1.2245e-02,
           3.2257e-02, 2.2522e-02]],

         [[1.2756e-01, 1.2195e-04, 1.2577e-04,  ..., 8.8196e-03,
           4.3427e-02, 7.4609e-01]],

         [[6.8848e-01, 1.9073e-05, 1.6499e-04,  ..., 1.9836e-02,
           2.2351e-01, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5996e-01, 6.1154e-05, 1.1148e-03,  ..., 5.3864e-02,
           3.4180e-01, 3.1738e-02]],

         [[9.9304e-02, 1.9789e-05, 7.0274e-05,  ..., 7.5226e-03,
           1.2924e-02, 3.2471e-02]],

         [[3.4302e-01, 3.1877e-04, 2.2469e-03,  ..., 1.2280e-01,
           6.9946e-02, 3.3966e-02]],

         ...,

         [[5.8838e-01, 1.4472e-04, 1.6508e-03,  ..., 4.6814e-02,
           4.4312e-02, 6.9695e-03]],

         [[6.9727e-01, 7.8678e-06, 1.0366e-03,  ..., 3.3188e-03,
           5.9395e-03, 9.9640e-03]],

         [[6.9092e-01, 5.0485e-05, 1.0335e-04,  ..., 6.0364e-02,
           6.2042e-02, 4.5563e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1478e-04, 5.4789e-04, 1.1786e-01,  ..., 1.9699e-02,
           8.6243e-02, 6.6490e-03]],

         [[1.8024e-04, 1.1700e-04, 1.2421e-01,  ..., 2.5940e-02,
           6.9702e-02, 1.1761e-01]],

         [[6.0272e-04, 3.4213e-04, 8.1848e-02,  ..., 3.9429e-02,
           1.4880e-01, 3.9276e-02]],

         ...,

         [[1.8609e-04, 4.5240e-05, 5.5786e-02,  ..., 1.0101e-01,
           5.9961e-01, 7.3303e-02]],

         [[2.1672e-04, 8.6737e-04, 3.0396e-01,  ..., 3.6316e-02,
           2.7191e-02, 1.8654e-03]],

         [[4.9829e-05, 7.4804e-05, 4.3579e-02,  ..., 4.4518e-03,
           7.1144e-03, 1.9913e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2541e-04, 2.2972e-04, 3.0029e-01,  ..., 1.3504e-02,
           3.2928e-02, 2.9335e-03]],

         [[2.1017e-04, 3.9577e-04, 1.7126e-01,  ..., 1.0696e-02,
           2.3361e-02, 1.1475e-02]],

         [[3.6299e-05, 4.4227e-05, 2.6636e-01,  ..., 1.7166e-02,
           2.7634e-02, 1.2767e-04]],

         ...,

         [[1.4925e-03, 1.0691e-03, 2.2412e-01,  ..., 2.9526e-02,
           3.4241e-02, 6.1646e-03]],

         [[3.1888e-05, 8.3983e-05, 2.4939e-01,  ..., 4.0169e-03,
           3.0731e-02, 5.8746e-04]],

         [[6.5446e-05, 1.7107e-04, 1.1652e-01,  ..., 3.7915e-01,
           5.3772e-02, 8.0681e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6904e-04, 2.5082e-04, 1.5552e-01,  ..., 1.8356e-02,
           2.4414e-02, 9.0837e-05]],

         [[5.9938e-04, 5.7876e-05, 2.6147e-01,  ..., 1.4412e-02,
           2.0477e-02, 9.1248e-03]],

         [[4.3416e-04, 1.4772e-03, 2.4646e-01,  ..., 1.2039e-02,
           1.9608e-02, 1.5783e-03]],

         ...,

         [[1.4377e-04, 2.4331e-04, 1.4807e-01,  ..., 1.3771e-02,
           7.8918e-02, 5.7526e-02]],

         [[1.2469e-04, 7.6294e-04, 2.4207e-01,  ..., 2.0996e-02,
           1.4145e-02, 1.0513e-02]],

         [[3.6788e-04, 3.4571e-04, 1.3916e-01,  ..., 7.8201e-03,
           1.2207e-02, 6.3591e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8777e-04, 4.5359e-05, 2.9419e-02,  ..., 1.3501e-01,
           4.0283e-01, 1.9946e-01]],

         [[8.3864e-05, 2.3949e-04, 1.6138e-01,  ..., 1.7517e-01,
           5.4474e-02, 5.0964e-02]],

         [[7.0763e-04, 3.0746e-03, 1.6626e-01,  ..., 4.0100e-02,
           2.6913e-03, 2.0008e-03]],

         ...,

         [[7.0715e-04, 5.0306e-05, 1.0193e-02,  ..., 1.5535e-03,
           1.2732e-03, 8.1253e-03]],

         [[4.2076e-03, 1.4496e-04, 6.9824e-02,  ..., 1.3562e-01,
           1.5308e-01, 6.4575e-02]],

         [[1.0669e-05, 3.8147e-06, 3.3245e-03,  ..., 3.9581e-02,
           2.2552e-02, 1.2878e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3618e-03, 2.9869e-03, 1.9702e-01,  ..., 9.6970e-03,
           5.6992e-03, 4.9835e-02]],

         [[1.2531e-03, 2.1744e-03, 2.7539e-01,  ..., 1.4885e-02,
           2.1912e-02, 2.2171e-02]],

         [[2.7199e-03, 1.8635e-03, 1.8103e-01,  ..., 5.8655e-02,
           6.8176e-02, 2.5665e-02]],

         ...,

         [[4.6501e-03, 6.3658e-04, 1.7310e-01,  ..., 2.0523e-02,
           3.5187e-02, 9.4849e-02]],

         [[3.6182e-03, 1.2169e-03, 1.5503e-01,  ..., 1.8524e-02,
           1.1615e-01, 2.8735e-01]],

         [[5.0187e-05, 1.5640e-04, 5.3345e-02,  ..., 2.8549e-02,
           6.8652e-01, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.2994e-05, 4.7684e-07, 2.1636e-04,  ..., 1.2695e-01,
           7.7783e-01, 4.9713e-02]],

         [[7.5340e-05, 2.4438e-05, 4.9114e-04,  ..., 9.2697e-03,
           3.4103e-03, 9.4531e-01]],

         [[5.9605e-08, 0.0000e+00, 6.1393e-06,  ..., 2.0325e-04,
           2.0325e-04, 9.9854e-01]],

         ...,

         [[3.5591e-03, 5.7335e-03, 8.1482e-03,  ..., 4.9530e-02,
           1.1856e-02, 3.6890e-01]],

         [[9.0256e-03, 1.2329e-02, 1.6861e-02,  ..., 9.1125e-02,
           4.2610e-03, 8.5571e-02]],

         [[8.6136e-03, 3.5915e-03, 1.0948e-03,  ..., 1.0828e-01,
           7.2144e-02, 2.6535e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1052e-04, 1.2302e-03, 5.3444e-03,  ..., 1.4673e-01,
           1.5869e-01, 2.7856e-01]],

         [[0.0000e+00, 0.0000e+00, 1.7881e-07,  ..., 1.5881e-01,
           7.5732e-01, 7.1533e-02]],

         [[1.0567e-03, 1.5135e-03, 7.6866e-03,  ..., 1.7224e-01,
           6.9580e-02, 4.1534e-02]],

         ...,

         [[2.0921e-05, 6.8784e-05, 1.1015e-03,  ..., 6.5674e-01,
           2.2705e-01, 2.1957e-02]],

         [[1.3304e-03, 5.8479e-03, 2.2156e-02,  ..., 1.8542e-01,
           3.6652e-02, 4.7089e-02]],

         [[1.3371e-03, 5.2071e-03, 8.3160e-03,  ..., 4.3677e-01,
           7.0190e-02, 3.2654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6711e-02, 7.9041e-03, 3.3783e-02,  ..., 8.5632e-02,
           5.7495e-02, 6.8787e-02]],

         [[3.2482e-03, 9.1600e-04, 2.8763e-03,  ..., 1.1604e-02,
           6.7177e-03, 8.0414e-03]],

         [[1.4313e-02, 4.1885e-03, 4.2458e-03,  ..., 7.0984e-02,
           4.3060e-02, 5.9326e-02]],

         ...,

         [[2.4605e-03, 1.9252e-05, 3.1590e-05,  ..., 6.0577e-02,
           7.4951e-01, 1.4197e-01]],

         [[5.4016e-03, 1.2636e-03, 1.0471e-03,  ..., 2.4451e-01,
           1.0516e-01, 1.0681e-01]],

         [[1.6571e-02, 1.3027e-03, 1.3954e-02,  ..., 9.3872e-02,
           1.2244e-01, 2.5317e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1914e-01, 1.7285e-06, 5.9009e-06,  ..., 1.3485e-03,
           9.6054e-03, 4.9706e-03]],

         [[8.6963e-01, 5.4836e-06, 8.6427e-06,  ..., 4.7150e-02,
           5.4260e-02, 4.4861e-03]],

         [[8.7988e-01, 8.0466e-06, 3.3796e-05,  ..., 1.1330e-02,
           2.1881e-02, 3.0289e-02]],

         ...,

         [[9.0454e-02, 7.4089e-05, 2.2960e-04,  ..., 8.6308e-04,
           3.1376e-04, 8.0261e-03]],

         [[6.2305e-01, 1.3232e-04, 1.1463e-03,  ..., 1.1726e-02,
           4.3396e-02, 1.4061e-02]],

         [[4.1943e-01, 2.0730e-04, 5.1260e-04,  ..., 7.6660e-02,
           2.7557e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1748e-01, 7.6890e-06, 1.2159e-05,  ..., 9.1124e-04,
           1.5259e-05, 1.6052e-02]],

         [[8.5889e-01, 0.0000e+00, 0.0000e+00,  ..., 1.7715e-02,
           1.0193e-01, 1.7578e-02]],

         [[9.4580e-01, 7.2896e-05, 7.2718e-06,  ..., 3.4580e-03,
           6.6681e-03, 1.1612e-02]],

         ...,

         [[4.1626e-01, 3.2854e-04, 2.7442e-04,  ..., 7.5073e-02,
           5.1994e-03, 5.5656e-03]],

         [[1.8970e-01, 1.0004e-03, 5.1689e-04,  ..., 7.5928e-02,
           2.6436e-03, 2.7733e-03]],

         [[7.3633e-01, 1.5438e-05, 8.7857e-05,  ..., 9.4116e-02,
           9.3933e-02, 4.0741e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 4.1056e-04, 2.0003e-04,  ..., 8.1360e-02,
           5.1636e-02, 3.0991e-02]],

         [[6.8262e-01, 5.3644e-06, 9.3579e-06,  ..., 9.3140e-02,
           1.0120e-01, 1.5480e-02]],

         [[7.4268e-01, 2.5153e-05, 4.0531e-06,  ..., 9.9426e-02,
           1.2103e-01, 2.5387e-03]],

         ...,

         [[2.3694e-01, 2.3842e-07, 4.7684e-07,  ..., 9.1851e-05,
           1.6570e-04, 7.3828e-01]],

         [[1.5894e-01, 1.7881e-07, 5.1260e-06,  ..., 8.3447e-07,
           1.1921e-07, 1.2886e-02]],

         [[8.3691e-01, 1.1921e-07, 8.3447e-07,  ..., 7.7486e-07,
           5.9605e-08, 2.9266e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1948e-01, 1.5616e-05, 1.6153e-05,  ..., 6.7688e-02,
           6.4502e-01, 1.3397e-02]],

         [[9.1113e-01, 2.6822e-06, 1.7881e-06,  ..., 8.8787e-04,
           2.9922e-04, 3.6983e-03]],

         [[7.4951e-01, 3.2806e-04, 9.6977e-05,  ..., 9.6893e-03,
           8.4305e-03, 5.4413e-02]],

         ...,

         [[8.3447e-01, 7.2718e-05, 1.2934e-05,  ..., 2.6001e-02,
           1.0651e-01, 5.7755e-03]],

         [[9.2969e-01, 6.8784e-05, 2.7120e-05,  ..., 1.2390e-02,
           1.5884e-02, 2.7679e-02]],

         [[8.9551e-01, 1.2279e-04, 7.3075e-05,  ..., 1.0460e-02,
           6.9122e-03, 2.2659e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0078e-01, 2.8014e-05, 3.7193e-05,  ..., 5.4512e-03,
           2.3148e-02, 3.2440e-02]],

         [[5.3174e-01, 5.3024e-04, 1.4842e-04,  ..., 1.8311e-02,
           3.6896e-02, 1.0724e-01]],

         [[5.3467e-02, 1.4305e-06, 7.7486e-07,  ..., 1.8021e-02,
           8.7500e-01, 4.1412e-02]],

         ...,

         [[6.0449e-01, 1.0878e-04, 6.2704e-05,  ..., 9.1248e-02,
           2.5925e-02, 1.0925e-02]],

         [[9.3701e-01, 2.0814e-04, 3.1650e-05,  ..., 4.5013e-03,
           3.7327e-03, 1.3733e-03]],

         [[7.1094e-01, 3.8743e-06, 1.8179e-05,  ..., 4.5349e-02,
           1.8970e-01, 1.6235e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6997e-01, 1.7958e-03, 8.5831e-04,  ..., 6.7215e-03,
           1.4372e-03, 8.3494e-04]],

         [[9.6436e-01, 1.1623e-05, 3.2067e-05,  ..., 3.5095e-03,
           8.6355e-04, 4.2915e-04]],

         [[8.9502e-01, 8.1658e-06, 7.5102e-06,  ..., 1.9989e-03,
           3.1204e-03, 7.3576e-04]],

         ...,

         [[6.5186e-01, 9.9361e-05, 9.4056e-05,  ..., 3.1235e-02,
           5.2834e-03, 2.1252e-01]],

         [[8.9502e-01, 2.9802e-07, 1.7285e-06,  ..., 5.2216e-02,
           1.8967e-02, 1.7258e-02]],

         [[9.4043e-01, 3.5763e-06, 7.7486e-06,  ..., 7.0877e-03,
           2.2446e-02, 8.2245e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4194e-01, 2.6846e-04, 1.1940e-03,  ..., 5.6458e-02,
           8.5144e-03, 3.9368e-03]],

         [[6.7432e-01, 1.0848e-05, 4.2260e-05,  ..., 3.0319e-02,
           2.2424e-01, 1.1444e-02]],

         [[8.3496e-01, 6.2585e-06, 6.1870e-05,  ..., 6.6101e-02,
           2.8351e-02, 2.6169e-02]],

         ...,

         [[7.3438e-01, 6.8665e-05, 2.0790e-04,  ..., 1.2337e-02,
           1.2074e-03, 3.8075e-04]],

         [[7.0898e-01, 1.8096e-04, 4.8041e-05,  ..., 2.1790e-02,
           9.7656e-02, 5.3291e-03]],

         [[8.4961e-01, 5.3644e-06, 7.4506e-06,  ..., 1.1591e-01,
           7.2861e-04, 2.9907e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8867e-02, 3.0398e-06, 2.0385e-05,  ..., 3.6407e-04,
           6.0618e-05, 4.5662e-03]],

         [[8.3643e-01, 6.3181e-06, 9.1195e-06,  ..., 6.7444e-02,
           6.1302e-03, 6.3744e-03]],

         [[8.0957e-01, 4.7684e-07, 4.0531e-06,  ..., 5.8098e-03,
           7.1049e-04, 2.4986e-03]],

         ...,

         [[9.3066e-01, 7.5936e-05, 3.8028e-05,  ..., 9.1400e-03,
           1.2350e-03, 1.4019e-03]],

         [[8.6816e-01, 1.2165e-04, 6.0439e-05,  ..., 2.6520e-02,
           4.5776e-03, 6.9466e-03]],

         [[7.3389e-01, 5.6362e-04, 2.0409e-04,  ..., 1.0605e-02,
           3.2234e-04, 4.8018e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 1.0192e-05, 4.9472e-06,  ..., 2.3331e-02,
           3.2074e-02, 5.5664e-01]],

         [[8.0713e-01, 4.2748e-04, 3.3426e-04,  ..., 4.5242e-03,
           6.1493e-03, 3.3379e-03]],

         [[8.0762e-01, 6.1035e-04, 5.6934e-04,  ..., 5.1025e-02,
           2.7771e-03, 2.6798e-03]],

         ...,

         [[5.6787e-01, 4.7565e-05, 1.4102e-04,  ..., 1.2993e-02,
           2.9312e-02, 5.0201e-03]],

         [[8.7128e-03, 1.0133e-06, 1.0729e-06,  ..., 6.5756e-04,
           8.7256e-01, 5.6152e-03]],

         [[6.9763e-02, 9.1791e-06, 2.2709e-05,  ..., 5.3131e-02,
           2.0065e-02, 8.1787e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0381e-01, 2.0373e-04, 3.3069e-04,  ..., 3.3905e-02,
           3.0014e-02, 4.2310e-01]],

         [[6.1621e-01, 8.0943e-05, 1.3494e-04,  ..., 6.2988e-02,
           2.0325e-02, 5.3520e-03]],

         [[3.7109e-01, 1.2815e-04, 1.0496e-04,  ..., 5.1880e-03,
           7.5867e-02, 8.3847e-03]],

         ...,

         [[1.0681e-02, 1.0729e-06, 2.7418e-06,  ..., 8.5163e-04,
           7.9811e-05, 9.8242e-01]],

         [[7.8125e-01, 1.7083e-04, 1.4675e-04,  ..., 1.1475e-02,
           1.3268e-02, 9.8267e-03]],

         [[8.4180e-01, 7.9811e-05, 2.8789e-05,  ..., 1.9951e-03,
           7.1259e-03, 1.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5986e-01, 6.2609e-04, 1.2922e-03,  ..., 1.3962e-02,
           7.2441e-03, 5.9891e-03]],

         [[7.5488e-01, 1.5974e-05, 6.1691e-05,  ..., 1.5417e-01,
           6.1569e-03, 9.9945e-03]],

         [[3.1152e-01, 2.6584e-05, 6.0380e-05,  ..., 2.6367e-02,
           6.0844e-04, 2.5654e-03]],

         ...,

         [[6.6211e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1346e-01,
           1.9836e-01, 2.0416e-02]],

         [[5.2051e-01, 4.9543e-04, 5.5504e-04,  ..., 1.9897e-02,
           1.6418e-02, 5.8517e-03]],

         [[1.5210e-01, 1.1921e-07, 1.1921e-07,  ..., 1.7960e-02,
           8.0518e-01, 1.6983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9111e-01, 4.2081e-05, 1.3471e-05,  ..., 5.5084e-03,
           5.5122e-03, 3.9482e-03]],

         [[7.0496e-02, 5.8603e-04, 5.3835e-04,  ..., 2.0679e-01,
           3.0231e-04, 6.6137e-04]],

         [[5.0049e-01, 2.2113e-04, 1.6177e-04,  ..., 1.3867e-01,
           1.0777e-04, 2.9397e-04]],

         ...,

         [[4.1577e-01, 5.9605e-07, 2.3842e-06,  ..., 3.5767e-01,
           6.2073e-02, 5.9967e-03]],

         [[1.9751e-01, 5.9605e-08, 5.9605e-08,  ..., 3.2788e-01,
           4.4141e-01, 8.3389e-03]],

         [[8.9294e-02, 2.5630e-06, 6.1393e-06,  ..., 2.2949e-01,
           1.3351e-02, 2.6417e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6294e-02, 6.5565e-07, 1.2517e-06,  ..., 5.2643e-03,
           2.4765e-02, 8.0127e-01]],

         [[1.2238e-01, 9.4771e-06, 2.3007e-05,  ..., 4.0918e-01,
           5.4535e-02, 1.7395e-02]],

         [[2.5781e-01, 2.8563e-04, 3.9768e-04,  ..., 7.3303e-02,
           1.2810e-02, 8.7738e-03]],

         ...,

         [[3.3887e-01, 5.4538e-05, 8.3506e-05,  ..., 4.1840e-02,
           1.4626e-02, 4.8180e-03]],

         [[5.6152e-01, 3.0398e-06, 1.0490e-05,  ..., 2.0905e-02,
           1.7662e-03, 3.6488e-03]],

         [[1.8799e-01, 1.3876e-04, 1.6773e-04,  ..., 1.2379e-03,
           1.1467e-02, 8.0185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4902e-01, 2.1935e-05, 5.3167e-05,  ..., 3.5858e-02,
           1.0277e-02, 1.5251e-02]],

         [[2.2607e-01, 2.2376e-04, 3.6240e-04,  ..., 7.0007e-02,
           2.6169e-02, 1.8478e-02]],

         [[2.2430e-02, 2.6822e-06, 2.0862e-06,  ..., 8.2855e-03,
           6.0156e-01, 1.1818e-02]],

         ...,

         [[7.0898e-01, 4.9233e-05, 3.8207e-05,  ..., 1.9257e-02,
           3.7193e-03, 1.7746e-02]],

         [[5.5225e-01, 3.2425e-05, 5.4955e-05,  ..., 1.0254e-01,
           7.3128e-03, 1.4153e-02]],

         [[7.4854e-01, 9.1314e-05, 6.3121e-05,  ..., 1.1383e-02,
           1.1276e-02, 6.8130e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8809e-01, 1.9608e-03, 8.5688e-04,  ..., 1.2123e-02,
           2.7823e-04, 5.0688e-04]],

         [[1.7908e-01, 2.8658e-04, 2.5630e-04,  ..., 1.2817e-02,
           1.4641e-02, 1.5373e-02]],

         [[5.1807e-01, 2.3663e-05, 4.4167e-05,  ..., 7.8186e-02,
           4.5441e-02, 1.2306e-02]],

         ...,

         [[2.1021e-01, 1.1116e-04, 4.1544e-05,  ..., 6.1646e-02,
           2.7145e-02, 8.1635e-03]],

         [[8.9893e-01, 1.0598e-04, 1.4210e-04,  ..., 2.7728e-04,
           5.5432e-06, 8.7128e-03]],

         [[9.1553e-01, 7.7784e-05, 6.8009e-05,  ..., 1.8341e-02,
           4.4518e-03, 1.0881e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8799e-01, 5.7125e-04, 1.3075e-03,  ..., 8.4778e-02,
           1.8435e-03, 3.8818e-02]],

         [[2.8931e-01, 4.2915e-03, 3.1166e-03,  ..., 2.0477e-02,
           1.7338e-03, 2.5635e-03]],

         [[1.9812e-01, 1.6809e-05, 2.3901e-05,  ..., 1.3000e-01,
           1.1200e-01, 1.6040e-01]],

         ...,

         [[1.7334e-01, 1.4484e-05, 6.0618e-05,  ..., 6.6833e-02,
           1.0872e-02, 2.2869e-03]],

         [[3.6035e-01, 1.2100e-05, 3.1173e-05,  ..., 8.2947e-02,
           2.2559e-01, 3.1738e-02]],

         [[2.0984e-01, 9.2387e-06, 1.9848e-05,  ..., 1.7676e-01,
           2.3962e-01, 1.5710e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3252e-01, 1.1462e-04, 1.8609e-04,  ..., 2.4750e-02,
           8.9645e-03, 2.5543e-02]],

         [[1.9507e-01, 5.2571e-05, 1.2422e-04,  ..., 2.2107e-01,
           9.5032e-02, 6.8855e-03]],

         [[5.3284e-02, 4.1127e-06, 1.8477e-06,  ..., 7.4816e-04,
           8.3008e-03, 9.3018e-01]],

         ...,

         [[4.0680e-02, 1.2898e-04, 3.1304e-04,  ..., 1.7452e-03,
           2.9926e-03, 1.3306e-02]],

         [[3.0933e-01, 2.7657e-04, 4.4560e-04,  ..., 1.2085e-02,
           2.5391e-02, 5.5908e-02]],

         [[2.0410e-01, 1.7729e-03, 2.5806e-03,  ..., 3.2806e-02,
           4.7073e-03, 6.7177e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7715e-01, 1.1790e-04, 1.0085e-04,  ..., 5.3680e-02,
           5.0049e-02, 1.8265e-02]],

         [[3.7594e-03, 9.5367e-07, 1.3709e-06,  ..., 1.6868e-05,
           2.0027e-05, 1.5891e-04]],

         [[8.9404e-01, 2.5988e-05, 2.1815e-05,  ..., 8.0919e-04,
           4.6825e-04, 2.7537e-04]],

         ...,

         [[1.1536e-01, 1.1663e-03, 2.3746e-03,  ..., 2.9617e-02,
           1.6876e-02, 3.8635e-02]],

         [[3.3960e-01, 3.7551e-05, 8.1420e-05,  ..., 8.1100e-03,
           2.1582e-01, 4.4525e-02]],

         [[5.5762e-01, 1.2579e-03, 3.8452e-03,  ..., 2.0950e-02,
           1.0368e-02, 2.8641e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7437e-01, 1.4722e-04, 6.9666e-04,  ..., 5.7556e-02,
           4.2786e-02, 3.4912e-02]],

         [[6.7444e-02, 1.3757e-04, 6.0225e-04,  ..., 2.2949e-02,
           2.3823e-03, 4.7089e-02]],

         [[2.4500e-01, 3.7909e-05, 1.9717e-04,  ..., 1.5182e-02,
           4.9667e-03, 7.1289e-02]],

         ...,

         [[8.5449e-02, 2.5558e-04, 3.3855e-04,  ..., 1.1625e-03,
           6.3858e-03, 3.8513e-02]],

         [[2.5482e-02, 3.5405e-05, 6.3181e-05,  ..., 3.1590e-04,
           3.4485e-03, 1.5701e-02]],

         [[2.3108e-01, 1.0514e-04, 2.2781e-04,  ..., 2.2171e-02,
           2.0828e-02, 1.5236e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8396e-01, 1.5318e-05, 2.5034e-05,  ..., 2.0325e-02,
           7.0374e-02, 1.4783e-01]],

         [[4.8242e-01, 1.6439e-04, 4.9257e-04,  ..., 6.4270e-02,
           5.1956e-03, 1.7578e-02]],

         [[2.6611e-01, 1.7607e-04, 4.7851e-04,  ..., 7.2144e-02,
           1.0559e-02, 4.7778e-01]],

         ...,

         [[6.9971e-01, 1.4687e-04, 3.7503e-04,  ..., 2.0966e-02,
           9.1629e-03, 4.5074e-02]],

         [[2.9688e-01, 1.1462e-04, 2.2447e-04,  ..., 2.1133e-03,
           3.2978e-03, 3.6285e-02]],

         [[1.0602e-01, 7.3552e-05, 2.1279e-04,  ..., 1.5974e-03,
           1.0719e-03, 1.5396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3391e-01, 9.8884e-05, 1.4329e-04,  ..., 5.1239e-02,
           1.2482e-01, 1.2683e-01]],

         [[1.0815e-01, 4.3511e-05, 5.2392e-05,  ..., 5.9586e-03,
           3.1952e-02, 4.6143e-02]],

         [[4.5020e-01, 1.7798e-04, 3.1853e-04,  ..., 1.1360e-02,
           5.3772e-02, 8.0078e-02]],

         ...,

         [[5.1270e-01, 0.0000e+00, 0.0000e+00,  ..., 5.7526e-02,
           4.0552e-01, 1.4206e-02]],

         [[2.0154e-01, 7.6890e-06, 4.3452e-05,  ..., 2.8857e-01,
           1.3428e-01, 7.2823e-03]],

         [[2.5024e-01, 1.1325e-06, 1.3113e-06,  ..., 6.4270e-02,
           6.5918e-01, 1.4229e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1331e-02, 2.2411e-05, 5.7340e-05,  ..., 2.5818e-02,
           7.2998e-02, 9.7198e-03]],

         [[2.3590e-02, 5.4240e-06, 3.7551e-06,  ..., 3.3989e-03,
           4.8401e-02, 8.8525e-01]],

         [[3.7378e-01, 1.3371e-03, 3.6507e-03,  ..., 2.8824e-02,
           1.5821e-03, 1.2688e-02]],

         ...,

         [[2.7417e-01, 3.9744e-04, 7.8535e-04,  ..., 4.5471e-02,
           1.4198e-02, 3.6240e-03]],

         [[1.2952e-01, 1.0729e-06, 6.1393e-06,  ..., 1.0132e-01,
           6.5283e-01, 2.9831e-02]],

         [[7.3059e-02, 5.9605e-07, 3.5167e-06,  ..., 2.9810e-01,
           4.8779e-01, 2.8732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 1.5974e-04, 4.9686e-04,  ..., 8.1940e-03,
           5.1270e-03, 6.7368e-03]],

         [[9.6289e-01, 2.2471e-05, 8.3268e-05,  ..., 7.2896e-05,
           6.3241e-05, 1.2022e-04]],

         [[3.5669e-01, 6.4790e-05, 1.2589e-04,  ..., 5.0354e-04,
           6.4945e-04, 1.1091e-03]],

         ...,

         [[8.4619e-01, 1.5736e-05, 1.4126e-05,  ..., 1.2260e-02,
           3.8361e-02, 9.5444e-03]],

         [[4.6826e-01, 2.1458e-05, 1.7464e-05,  ..., 3.3321e-03,
           4.7379e-03, 1.0841e-02]],

         [[1.1554e-01, 1.6809e-05, 5.6386e-05,  ..., 9.3384e-03,
           6.3782e-02, 7.1875e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2559e-01, 1.4067e-05, 1.8716e-05,  ..., 1.3257e-01,
           1.2262e-01, 3.8147e-02]],

         [[7.4268e-01, 2.4498e-05, 1.8537e-05,  ..., 1.4305e-02,
           4.7089e-02, 1.3817e-02]],

         [[6.4026e-02, 1.5795e-05, 3.0458e-05,  ..., 5.7831e-02,
           2.8687e-01, 2.3636e-02]],

         ...,

         [[6.3428e-01, 1.6570e-05, 7.5459e-05,  ..., 8.5876e-02,
           8.4534e-02, 3.9001e-02]],

         [[6.5967e-01, 7.9215e-05, 6.0606e-04,  ..., 7.0068e-02,
           1.7303e-02, 8.1100e-03]],

         [[6.5674e-01, 1.5676e-05, 9.7394e-05,  ..., 2.6276e-02,
           5.5962e-03, 2.1118e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9438e-02, 1.1444e-05, 6.7353e-06,  ..., 4.2534e-03,
           3.8818e-02, 3.4981e-03]],

         [[2.5171e-01, 7.7486e-05, 6.8903e-05,  ..., 1.0674e-02,
           7.5111e-03, 5.9433e-03]],

         [[1.7200e-01, 1.1790e-04, 3.1424e-04,  ..., 6.8420e-02,
           7.3975e-02, 7.5745e-02]],

         ...,

         [[5.6213e-02, 1.6391e-05, 1.6940e-04,  ..., 8.5022e-02,
           3.8853e-03, 7.6065e-03]],

         [[1.2390e-01, 2.3127e-05, 1.6963e-04,  ..., 2.4646e-01,
           9.7046e-03, 3.7201e-02]],

         [[9.1650e-01, 6.7949e-06, 1.7345e-05,  ..., 2.7679e-02,
           1.9318e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0789e-01, 1.7285e-06, 5.3644e-06,  ..., 1.3525e-01,
           5.9424e-01, 2.8351e-02]],

         [[5.2344e-01, 1.2398e-05, 1.0550e-05,  ..., 1.8738e-02,
           1.7944e-01, 1.4819e-01]],

         [[8.8440e-02, 2.9802e-07, 4.1723e-07,  ..., 2.4915e-01,
           5.9277e-01, 2.9068e-02]],

         ...,

         [[3.7671e-01, 1.4460e-04, 4.3249e-04,  ..., 1.0811e-02,
           1.0239e-02, 2.9968e-02]],

         [[3.5742e-01, 1.0604e-04, 1.1396e-04,  ..., 1.0815e-01,
           6.9824e-02, 3.0029e-02]],

         [[4.7211e-02, 8.7976e-05, 7.4267e-05,  ..., 4.5410e-02,
           1.4099e-01, 5.2393e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9756e-01, 1.0729e-05, 1.1909e-04,  ..., 6.7322e-02,
           1.4038e-01, 1.5906e-01]],

         [[3.5400e-01, 6.7282e-04, 2.0084e-03,  ..., 1.9135e-02,
           5.5656e-03, 5.1689e-03]],

         [[1.2024e-01, 4.0054e-05, 2.0015e-04,  ..., 9.2010e-03,
           3.9902e-03, 1.5900e-02]],

         ...,

         [[6.5283e-01, 1.0109e-04, 1.2577e-04,  ..., 1.5839e-02,
           1.4575e-01, 5.6183e-02]],

         [[1.7175e-01, 5.9009e-05, 1.4377e-04,  ..., 7.7858e-03,
           4.9225e-02, 7.3438e-01]],

         [[7.4268e-01, 9.7573e-05, 3.8576e-04,  ..., 1.2978e-02,
           1.5656e-02, 2.0618e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1211e-01, 1.8609e-04, 9.5224e-04,  ..., 4.1199e-02,
           1.6077e-01, 7.8674e-02]],

         [[1.9800e-01, 5.2631e-05, 1.0973e-04,  ..., 1.1986e-02,
           5.8075e-02, 4.9683e-02]],

         [[1.7236e-01, 1.3638e-04, 2.5730e-03,  ..., 1.4465e-01,
           1.2817e-01, 2.7710e-02]],

         ...,

         [[8.1445e-01, 6.0141e-05, 1.8752e-04,  ..., 1.4664e-02,
           3.5896e-03, 3.6652e-02]],

         [[7.0801e-01, 1.7285e-06, 1.6093e-04,  ..., 1.1654e-03,
           1.0109e-03, 1.4524e-03]],

         [[6.1963e-01, 4.4167e-05, 4.0591e-05,  ..., 5.1453e-02,
           6.9214e-02, 1.3184e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6955e-04, 6.3610e-04, 1.6125e-01,  ..., 1.1792e-01,
           1.5078e-03, 1.4534e-02]],

         [[2.9087e-04, 2.7323e-04, 1.7761e-01,  ..., 1.1377e-01,
           9.0027e-02, 7.9269e-03]],

         [[2.0826e-04, 2.8610e-04, 6.7322e-02,  ..., 1.3330e-01,
           8.3801e-02, 1.3867e-01]],

         ...,

         [[2.7013e-04, 7.4446e-05, 1.0730e-01,  ..., 1.7419e-01,
           2.1167e-01, 1.3989e-01]],

         [[2.4652e-04, 5.4264e-04, 2.3938e-01,  ..., 4.9042e-02,
           3.0022e-03, 1.8753e-02]],

         [[4.8459e-05, 5.9009e-06, 2.7786e-02,  ..., 1.8616e-03,
           1.2183e-04, 2.8839e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4894e-04, 6.4909e-05, 2.7515e-01,  ..., 1.0498e-02,
           9.2125e-04, 1.7281e-03]],

         [[3.1090e-04, 1.3959e-04, 2.7197e-01,  ..., 9.4452e-03,
           2.4948e-03, 7.7581e-04]],

         [[4.3988e-05, 6.5565e-06, 2.9175e-01,  ..., 9.2316e-03,
           1.6034e-05, 8.8596e-04]],

         ...,

         [[2.9049e-03, 4.6778e-04, 2.7295e-01,  ..., 2.7130e-02,
           2.4757e-03, 8.0967e-04]],

         [[8.8930e-05, 6.3777e-05, 3.0640e-01,  ..., 1.7151e-02,
           2.5725e-04, 9.4509e-04]],

         [[2.2411e-05, 8.8215e-06, 2.0251e-01,  ..., 2.9175e-02,
           1.2493e-04, 2.5845e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2149e-04, 5.9187e-05, 2.1240e-01,  ..., 1.0933e-02,
           1.6630e-05, 8.4591e-04]],

         [[5.5313e-04, 1.0490e-05, 2.8320e-01,  ..., 1.1780e-02,
           2.7447e-03, 7.3290e-04]],

         [[4.4942e-04, 2.2376e-04, 2.5977e-01,  ..., 8.6517e-03,
           3.8815e-04, 1.1501e-03]],

         ...,

         [[2.6226e-05, 5.0664e-05, 2.1765e-01,  ..., 8.1970e-02,
           3.7689e-02, 3.3386e-02]],

         [[1.5008e-04, 9.3639e-05, 2.3108e-01,  ..., 9.0637e-03,
           5.9433e-03, 1.3418e-03]],

         [[6.8474e-04, 5.6446e-05, 2.1899e-01,  ..., 7.0648e-03,
           1.5106e-03, 7.8201e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9855e-03, 5.9128e-04, 4.9225e-02,  ..., 2.6416e-01,
           1.8433e-01, 4.0741e-03]],

         [[1.2577e-04, 2.6846e-04, 1.3220e-01,  ..., 4.0649e-02,
           6.9702e-02, 3.7292e-02]],

         [[4.9353e-04, 8.7976e-04, 7.7026e-02,  ..., 1.4715e-03,
           7.8869e-04, 2.0075e-04]],

         ...,

         [[2.4967e-03, 5.3263e-04, 3.9215e-02,  ..., 7.9041e-03,
           4.4365e-03, 5.5084e-03]],

         [[1.1711e-03, 2.1994e-04, 8.6548e-02,  ..., 5.0110e-02,
           6.5369e-02, 1.8723e-02]],

         [[1.1528e-04, 3.8266e-05, 2.4338e-02,  ..., 4.6021e-02,
           1.0699e-01, 7.8857e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7656e-03, 1.7033e-03, 2.1130e-01,  ..., 8.6517e-03,
           7.2144e-02, 2.2446e-02]],

         [[1.0262e-03, 1.2312e-03, 2.9590e-01,  ..., 1.7426e-02,
           1.3573e-02, 3.8361e-02]],

         [[3.2711e-03, 8.4972e-04, 1.9812e-01,  ..., 2.5894e-02,
           6.3843e-02, 2.9739e-02]],

         ...,

         [[1.8752e-04, 1.1921e-04, 3.6865e-02,  ..., 7.3364e-02,
           6.5869e-01, 8.8440e-02]],

         [[1.1320e-03, 4.5919e-04, 2.4646e-01,  ..., 3.8086e-02,
           3.1586e-02, 1.1914e-01]],

         [[1.3173e-04, 4.0746e-04, 2.0618e-01,  ..., 1.7358e-01,
           2.9480e-02, 1.2093e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.2817e-03, 7.7009e-05, 5.8699e-04,  ..., 4.3793e-02,
           2.9468e-01, 4.7095e-01]],

         [[2.2488e-03, 1.2808e-03, 1.2154e-02,  ..., 2.7122e-03,
           2.2705e-02, 4.5605e-01]],

         [[3.1829e-05, 2.6226e-06, 1.1104e-04,  ..., 1.1104e-04,
           8.9941e-01, 9.4788e-02]],

         ...,

         [[2.6054e-03, 3.3455e-03, 5.5161e-03,  ..., 2.8015e-02,
           3.0127e-01, 4.2953e-03]],

         [[3.1338e-03, 3.5515e-03, 1.1642e-02,  ..., 1.3905e-03,
           7.5928e-02, 1.8213e-01]],

         [[8.3084e-03, 5.8899e-03, 1.1969e-03,  ..., 7.1777e-02,
           2.2583e-02, 2.1881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8065e-04, 2.2926e-03, 6.5308e-03,  ..., 1.0059e-01,
           7.3608e-02, 1.5332e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 2.1997e-01,
           6.9873e-01, 7.4829e-02]],

         [[1.8339e-03, 2.1439e-03, 6.6032e-03,  ..., 2.9587e-02,
           2.6520e-02, 1.6083e-02]],

         ...,

         [[6.2895e-04, 1.6823e-03, 6.5002e-03,  ..., 6.6711e-02,
           6.4011e-03, 2.0981e-02]],

         [[4.4136e-03, 9.5673e-03, 4.2694e-02,  ..., 2.5696e-02,
           2.3666e-02, 2.5208e-02]],

         [[4.2200e-04, 9.1457e-04, 2.8172e-03,  ..., 2.3462e-01,
           5.8984e-01, 5.4840e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7618e-02, 4.0092e-03, 1.7548e-02,  ..., 8.5083e-02,
           1.0022e-01, 7.8674e-02]],

         [[2.4605e-03, 7.0477e-04, 2.5673e-03,  ..., 2.7122e-03,
           3.9444e-03, 8.1110e-04]],

         [[8.9798e-03, 2.4261e-03, 1.9875e-03,  ..., 2.7878e-02,
           3.5645e-02, 3.6926e-02]],

         ...,

         [[4.9286e-03, 3.0422e-04, 4.3917e-04,  ..., 3.3765e-01,
           3.1226e-01, 2.3755e-01]],

         [[3.0708e-03, 6.1893e-04, 1.3256e-03,  ..., 1.8555e-01,
           4.1412e-02, 3.0777e-02]],

         [[1.9333e-02, 1.8196e-03, 9.2773e-03,  ..., 5.5084e-02,
           4.0698e-01, 8.4656e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4551e-01, 2.8014e-06, 9.4771e-06,  ..., 9.0576e-02,
           1.4076e-02, 1.8982e-02]],

         [[9.0283e-01, 8.9407e-06, 1.4842e-05,  ..., 4.7607e-02,
           8.3771e-03, 9.3689e-03]],

         [[4.3164e-01, 1.1325e-06, 1.2100e-05,  ..., 1.1438e-01,
           3.7659e-02, 1.1002e-02]],

         ...,

         [[4.5288e-01, 3.6144e-04, 2.0924e-03,  ..., 5.6458e-04,
           2.9678e-03, 8.0719e-03]],

         [[5.1465e-01, 2.4915e-04, 1.3351e-03,  ..., 2.7267e-02,
           5.6854e-02, 1.5244e-02]],

         [[4.3481e-01, 1.2934e-04, 3.6740e-04,  ..., 2.9358e-02,
           4.1901e-02, 1.9104e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 5.1856e-06, 9.2983e-06,  ..., 2.9325e-04,
           5.5885e-04, 7.3738e-03]],

         [[1.3110e-01, 1.1921e-07, 0.0000e+00,  ..., 9.8648e-03,
           8.4521e-01, 1.3084e-02]],

         [[9.5117e-01, 1.0216e-04, 2.7835e-05,  ..., 1.1425e-03,
           7.6675e-03, 3.0689e-03]],

         ...,

         [[3.6108e-01, 1.3781e-04, 3.4904e-04,  ..., 1.1131e-02,
           7.2365e-03, 3.5954e-03]],

         [[1.5125e-01, 1.3914e-03, 6.4230e-04,  ..., 3.8986e-03,
           3.9711e-03, 1.6613e-03]],

         [[6.5381e-01, 1.2398e-05, 8.6248e-05,  ..., 6.7322e-02,
           4.9713e-02, 1.3000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3105e-01, 3.4046e-04, 1.8656e-04,  ..., 8.2397e-02,
           3.3630e-02, 2.3087e-02]],

         [[6.2305e-01, 8.3447e-06, 1.2755e-05,  ..., 7.7332e-02,
           1.4771e-01, 2.3834e-02]],

         [[8.2031e-01, 1.1742e-05, 3.1590e-06,  ..., 1.0278e-01,
           4.5471e-03, 8.6288e-03]],

         ...,

         [[2.8882e-01, 4.7684e-07, 2.9802e-06,  ..., 2.9278e-04,
           1.3031e-02, 6.8359e-01]],

         [[3.4814e-01, 4.7684e-07, 3.3402e-04,  ..., 2.9802e-07,
           1.1367e-04, 7.0984e-02]],

         [[8.9844e-01, 8.3447e-07, 2.4319e-05,  ..., 5.3644e-07,
           2.8312e-05, 4.6539e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2876e-01, 1.1683e-05, 4.1127e-06,  ..., 4.3750e-01,
           2.5879e-01, 1.4908e-02]],

         [[9.1406e-01, 7.6890e-06, 3.8743e-06,  ..., 3.9062e-03,
           1.2512e-03, 6.1951e-03]],

         [[6.9922e-01, 4.1795e-04, 2.5058e-04,  ..., 1.1650e-02,
           2.5970e-02, 2.6901e-02]],

         ...,

         [[8.5840e-01, 3.0100e-05, 1.2040e-05,  ..., 7.4707e-02,
           1.4839e-02, 8.0338e-03]],

         [[6.9971e-01, 3.5882e-05, 7.3314e-06,  ..., 2.2324e-02,
           1.9312e-01, 6.7383e-02]],

         [[8.9502e-01, 1.2231e-04, 2.8908e-05,  ..., 4.0512e-03,
           3.1372e-02, 2.6764e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0312e-01, 1.5914e-05, 1.0192e-05,  ..., 7.4707e-02,
           8.4778e-02, 5.0903e-02]],

         [[5.3369e-01, 9.8050e-05, 2.1672e-04,  ..., 4.4586e-02,
           6.3904e-02, 1.3245e-01]],

         [[1.7297e-01, 2.9802e-06, 4.2319e-06,  ..., 6.7578e-01,
           1.1322e-01, 7.5760e-03]],

         ...,

         [[6.6943e-01, 8.3685e-05, 4.6730e-05,  ..., 3.1555e-02,
           1.2543e-02, 1.0178e-02]],

         [[8.6621e-01, 1.7929e-04, 3.8624e-05,  ..., 6.8855e-03,
           4.2076e-03, 9.2459e-04]],

         [[8.0371e-01, 6.9737e-06, 8.4043e-06,  ..., 4.1565e-02,
           7.0557e-02, 1.3153e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.5936e-03, 1.3008e-03,  ..., 2.1954e-03,
           6.0701e-04, 5.5504e-04]],

         [[9.6094e-01, 1.1981e-05, 2.8968e-05,  ..., 9.3651e-04,
           8.8310e-04, 5.4407e-04]],

         [[8.7646e-01, 1.8656e-05, 1.2040e-05,  ..., 2.4223e-03,
           1.5631e-03, 1.9760e-03]],

         ...,

         [[6.0840e-01, 6.5744e-05, 1.0067e-04,  ..., 1.7319e-02,
           1.4600e-01, 8.2520e-02]],

         [[9.2969e-01, 6.5565e-07, 1.7285e-06,  ..., 8.3160e-03,
           2.4292e-02, 9.5749e-03]],

         [[9.5801e-01, 2.0862e-06, 2.1458e-06,  ..., 3.0499e-03,
           1.2657e-02, 1.1452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 4.9353e-04, 1.3733e-03,  ..., 5.1727e-03,
           4.8866e-03, 1.9951e-03]],

         [[5.9717e-01, 2.6345e-05, 3.2544e-05,  ..., 8.2458e-02,
           2.3315e-01, 2.9648e-02]],

         [[9.4775e-01, 8.4639e-06, 2.1815e-05,  ..., 9.5062e-03,
           5.3444e-03, 3.5458e-03]],

         ...,

         [[6.1035e-01, 1.3340e-04, 2.0754e-04,  ..., 1.9360e-03,
           5.2357e-04, 2.0826e-04]],

         [[7.7783e-01, 5.6505e-04, 7.6830e-05,  ..., 2.5955e-02,
           1.1581e-02, 4.9210e-03]],

         [[3.0469e-01, 1.1086e-05, 1.2815e-05,  ..., 7.8659e-03,
           5.2246e-01, 1.0083e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7981e-01, 4.8876e-06, 2.3723e-05,  ..., 2.1410e-04,
           6.1703e-04, 3.1395e-03]],

         [[7.5146e-01, 8.2850e-06, 1.4603e-05,  ..., 4.1473e-02,
           3.2166e-02, 3.7022e-03]],

         [[6.6309e-01, 5.9605e-07, 6.6161e-06,  ..., 3.7136e-03,
           2.0275e-03, 1.9045e-03]],

         ...,

         [[9.5703e-01, 9.3997e-05, 5.4181e-05,  ..., 5.0402e-04,
           6.7282e-04, 7.5817e-05]],

         [[5.3223e-01, 9.4831e-05, 3.7551e-05,  ..., 5.6992e-03,
           4.1565e-02, 3.0151e-01]],

         [[6.4990e-01, 4.0007e-04, 1.3912e-04,  ..., 3.4428e-04,
           1.5640e-03, 6.9809e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5283e-01, 2.3067e-05, 8.9407e-06,  ..., 2.1000e-03,
           1.6266e-02, 8.1348e-01]],

         [[8.5254e-01, 3.5262e-04, 2.6274e-04,  ..., 3.5744e-03,
           1.7557e-03, 7.6008e-04]],

         [[8.2129e-01, 3.8266e-04, 2.7466e-04,  ..., 5.8784e-03,
           7.0229e-03, 1.6394e-03]],

         ...,

         [[5.1904e-01, 5.9903e-05, 2.0742e-04,  ..., 3.3966e-02,
           3.9825e-02, 3.5419e-03]],

         [[1.8726e-01, 4.1723e-06, 5.1260e-06,  ..., 4.7729e-01,
           1.0242e-01, 6.4507e-03]],

         [[1.4392e-01, 2.3842e-05, 3.2842e-05,  ..., 1.1536e-02,
           1.1163e-01, 6.6992e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6709e-01, 2.2972e-04, 2.2793e-04,  ..., 2.6810e-02,
           1.1792e-01, 4.6973e-01]],

         [[6.4355e-01, 5.5611e-05, 5.3465e-05,  ..., 2.2064e-02,
           4.7638e-02, 3.7060e-03]],

         [[4.6460e-01, 1.8823e-04, 3.7718e-04,  ..., 4.1290e-02,
           2.6489e-02, 1.4801e-03]],

         ...,

         [[1.3786e-02, 8.3447e-07, 2.2650e-06,  ..., 7.3910e-06,
           5.2528e-03, 9.7803e-01]],

         [[7.7295e-01, 1.5545e-04, 1.9121e-04,  ..., 5.5618e-03,
           4.3144e-03, 3.4912e-02]],

         [[7.8711e-01, 1.4138e-04, 4.8637e-05,  ..., 1.9867e-02,
           2.0798e-02, 7.2098e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9624e-01, 4.5586e-04, 1.0376e-03,  ..., 6.6261e-03,
           3.0899e-03, 1.7338e-03]],

         [[8.6035e-01, 1.7226e-05, 5.0068e-05,  ..., 4.0169e-03,
           3.2776e-02, 7.9575e-03]],

         [[3.6646e-01, 3.0756e-05, 9.1195e-05,  ..., 7.6628e-04,
           2.4490e-03, 5.2681e-03]],

         ...,

         [[3.8989e-01, 3.5763e-07, 0.0000e+00,  ..., 6.8542e-02,
           5.2393e-01, 1.4458e-02]],

         [[5.4492e-01, 8.4209e-04, 9.8038e-04,  ..., 1.3222e-02,
           4.8370e-03, 6.3801e-04]],

         [[6.3770e-01, 2.9206e-06, 1.3709e-06,  ..., 2.9028e-01,
           3.6926e-02, 4.2267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9648e-01, 8.1241e-05, 2.4498e-05,  ..., 5.1193e-03,
           2.2373e-03, 4.7588e-04]],

         [[1.3904e-01, 3.0684e-04, 2.6536e-04,  ..., 1.1177e-03,
           1.4839e-03, 4.8780e-04]],

         [[7.7832e-01, 9.7573e-05, 4.3988e-05,  ..., 3.2067e-04,
           5.9891e-04, 3.2878e-04]],

         ...,

         [[6.5527e-01, 2.3842e-07, 1.7881e-07,  ..., 7.6172e-02,
           9.0332e-02, 1.2856e-02]],

         [[8.1836e-01, 5.9605e-08, 5.9605e-08,  ..., 7.3059e-02,
           5.3467e-02, 1.4977e-02]],

         [[5.0293e-01, 3.2783e-06, 6.8545e-06,  ..., 4.0771e-02,
           3.2745e-02, 8.5144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8167e-02, 1.0133e-06, 1.7285e-06,  ..., 1.0004e-03,
           1.0887e-02, 8.9941e-01]],

         [[1.4319e-01, 3.8743e-06, 9.1195e-06,  ..., 1.7200e-01,
           2.2791e-01, 3.7628e-02]],

         [[4.2773e-01, 1.8287e-04, 2.0754e-04,  ..., 1.2413e-02,
           7.1068e-03, 2.0294e-03]],

         ...,

         [[4.3359e-01, 1.0377e-04, 1.6844e-04,  ..., 2.8229e-02,
           1.0490e-02, 5.6686e-03]],

         [[4.9854e-01, 3.7551e-06, 2.1815e-05,  ..., 1.2039e-02,
           1.2688e-02, 1.1185e-02]],

         [[1.8750e-01, 2.0826e-04, 2.4915e-04,  ..., 2.3098e-03,
           3.8834e-03, 1.6222e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2539e-01, 2.7120e-05, 5.5492e-05,  ..., 1.7838e-02,
           1.0323e-02, 6.0921e-03]],

         [[3.5059e-01, 3.1900e-04, 2.3162e-04,  ..., 8.1360e-02,
           3.2745e-02, 1.3329e-02]],

         [[1.6553e-01, 3.1471e-05, 3.8981e-05,  ..., 2.5732e-01,
           3.0029e-02, 8.6899e-03]],

         ...,

         [[7.7637e-01, 5.4836e-06, 7.9870e-06,  ..., 1.4275e-02,
           7.1487e-03, 4.8828e-04]],

         [[6.4600e-01, 5.2273e-05, 7.6115e-05,  ..., 1.1925e-02,
           2.2903e-02, 1.5764e-03]],

         [[8.8477e-01, 4.3154e-05, 5.1439e-05,  ..., 8.4381e-03,
           1.0424e-03, 6.3705e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8979e-01, 2.7294e-03, 1.2264e-03,  ..., 7.8058e-04,
           1.2732e-03, 5.8603e-04]],

         [[2.9956e-01, 1.0729e-03, 7.4434e-04,  ..., 6.5575e-03,
           2.8381e-03, 4.1924e-03]],

         [[3.7695e-01, 2.0146e-05, 2.0504e-05,  ..., 2.0984e-01,
           1.3806e-01, 1.5022e-02]],

         ...,

         [[3.5156e-01, 2.1899e-04, 1.3340e-04,  ..., 4.6577e-03,
           1.4896e-03, 2.2171e-02]],

         [[9.5020e-01, 6.2585e-05, 2.7823e-04,  ..., 1.7881e-07,
           2.7657e-05, 2.3937e-03]],

         [[8.2422e-01, 5.6744e-05, 6.6936e-05,  ..., 1.5427e-02,
           4.4785e-03, 1.5640e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9990e-01, 4.3988e-04, 8.0919e-04,  ..., 1.1978e-02,
           2.9419e-02, 7.6752e-03]],

         [[4.5386e-01, 1.9894e-03, 1.4496e-03,  ..., 1.2474e-02,
           5.1994e-03, 2.0695e-03]],

         [[1.5442e-01, 2.5451e-05, 2.1219e-05,  ..., 1.0262e-02,
           3.5620e-01, 2.4475e-01]],

         ...,

         [[1.9727e-01, 1.0550e-05, 3.1114e-05,  ..., 2.3560e-02,
           1.1528e-02, 5.4979e-04]],

         [[4.2676e-01, 3.7968e-05, 4.3094e-05,  ..., 1.1395e-01,
           7.2998e-02, 2.3788e-02]],

         [[4.2920e-01, 1.1480e-04, 1.2082e-04,  ..., 4.5959e-02,
           1.0480e-01, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4131e-01, 1.5831e-04, 2.5702e-04,  ..., 6.7711e-04,
           2.5558e-03, 1.5900e-02]],

         [[1.1365e-01, 9.0599e-06, 1.0729e-05,  ..., 2.4426e-01,
           7.4524e-02, 7.8506e-03]],

         [[2.6465e-01, 5.8174e-05, 1.9789e-05,  ..., 2.3975e-03,
           1.0522e-01, 5.8691e-01]],

         ...,

         [[7.9041e-02, 3.1066e-04, 7.8726e-04,  ..., 6.7329e-04,
           3.4199e-03, 1.0048e-02]],

         [[3.3984e-01, 4.1699e-04, 5.1880e-04,  ..., 4.2610e-03,
           2.7039e-02, 2.7679e-02]],

         [[2.1509e-01, 1.9817e-03, 2.0046e-03,  ..., 3.8033e-03,
           2.8267e-03, 3.8490e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6655e-01, 4.4823e-05, 5.1498e-05,  ..., 6.7749e-02,
           3.6530e-02, 8.8120e-03]],

         [[2.1152e-03, 2.9802e-07, 5.9605e-08,  ..., 1.5199e-04,
           1.0610e-05, 3.7708e-03]],

         [[8.1543e-01, 1.5855e-05, 3.1590e-06,  ..., 2.8496e-03,
           1.3971e-04, 4.2033e-04]],

         ...,

         [[9.1003e-02, 8.4496e-04, 1.6031e-03,  ..., 1.5327e-02,
           1.6830e-02, 2.7275e-03]],

         [[5.2588e-01, 2.5809e-05, 4.9770e-05,  ..., 1.8164e-01,
           2.0065e-02, 8.9035e-03]],

         [[4.8169e-01, 6.0034e-04, 1.3638e-03,  ..., 9.8419e-03,
           1.6876e-02, 1.4973e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6265e-01, 5.0008e-05, 1.8728e-04,  ..., 3.4576e-02,
           1.8799e-02, 6.6650e-02]],

         [[8.6365e-02, 4.6670e-05, 1.3828e-04,  ..., 7.4310e-03,
           3.6011e-02, 2.5543e-02]],

         [[2.2742e-01, 3.7193e-05, 1.1367e-04,  ..., 3.0003e-03,
           1.9882e-02, 2.8687e-02]],

         ...,

         [[1.2988e-01, 7.0751e-05, 7.5281e-05,  ..., 6.6719e-03,
           3.3875e-02, 3.9612e-02]],

         [[7.8430e-02, 7.8559e-05, 6.5148e-05,  ..., 4.0932e-03,
           1.1841e-02, 2.2476e-02]],

         [[4.4678e-01, 4.5359e-05, 8.2731e-05,  ..., 2.0889e-02,
           2.3666e-02, 1.3916e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6150e-01, 1.2398e-05, 1.1802e-05,  ..., 2.4368e-02,
           7.2205e-02, 6.4697e-02]],

         [[4.7363e-01, 2.3031e-04, 4.3201e-04,  ..., 7.2479e-03,
           1.4992e-02, 2.1469e-02]],

         [[1.9617e-01, 2.7061e-04, 6.8045e-04,  ..., 4.9095e-03,
           2.5977e-01, 3.0151e-01]],

         ...,

         [[6.0352e-01, 6.1750e-05, 2.0885e-04,  ..., 7.9575e-03,
           7.5562e-02, 5.3131e-02]],

         [[1.9617e-01, 5.0247e-05, 2.5415e-04,  ..., 1.3962e-03,
           8.4839e-03, 2.9846e-02]],

         [[1.7834e-01, 3.6418e-05, 1.4019e-04,  ..., 3.9625e-04,
           6.5956e-03, 1.7654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6675e-01, 7.0035e-05, 1.2934e-04,  ..., 5.4565e-02,
           7.1167e-02, 7.9407e-02]],

         [[1.5686e-01, 2.2113e-05, 2.0862e-05,  ..., 1.2192e-02,
           6.7329e-03, 1.3077e-02]],

         [[6.2061e-01, 1.1545e-04, 1.9491e-04,  ..., 2.0111e-02,
           1.3290e-02, 8.6823e-03]],

         ...,

         [[6.4893e-01, 5.3644e-07, 1.1921e-07,  ..., 1.4490e-01,
           1.9495e-01, 3.4065e-03]],

         [[3.9575e-01, 6.4373e-06, 2.8610e-05,  ..., 1.7285e-01,
           9.1064e-02, 2.9736e-03]],

         [[9.6191e-02, 1.7881e-06, 8.3447e-07,  ..., 4.8370e-02,
           8.4424e-01, 5.1384e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4209e-01, 7.4089e-05, 1.8847e-04,  ..., 7.3120e-02,
           1.9836e-02, 3.7270e-03]],

         [[7.9834e-02, 9.9540e-06, 8.4043e-06,  ..., 1.1414e-02,
           9.2590e-02, 6.8408e-01]],

         [[3.0249e-01, 9.9945e-04, 2.6073e-03,  ..., 1.2655e-03,
           7.6027e-03, 1.0576e-03]],

         ...,

         [[2.7197e-01, 2.6298e-04, 5.4169e-04,  ..., 2.5192e-02,
           9.8267e-03, 2.4509e-03]],

         [[8.4839e-02, 4.1723e-07, 1.3709e-06,  ..., 6.5723e-01,
           2.1667e-01, 1.7303e-02]],

         [[1.5112e-01, 2.9802e-07, 1.0133e-06,  ..., 4.0112e-01,
           3.5669e-01, 1.7349e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 1.4782e-04, 4.5228e-04,  ..., 3.7193e-03,
           4.2496e-03, 5.7936e-04]],

         [[9.7998e-01, 1.5795e-05, 6.3956e-05,  ..., 3.3855e-05,
           3.3557e-05, 1.0067e-04]],

         [[7.8271e-01, 5.3525e-05, 1.4734e-04,  ..., 6.1989e-04,
           3.2163e-04, 5.8270e-04]],

         ...,

         [[8.4863e-01, 2.0325e-05, 2.5272e-05,  ..., 2.4460e-02,
           1.9043e-02, 1.4603e-02]],

         [[5.7471e-01, 8.2850e-06, 8.5235e-06,  ..., 3.6106e-03,
           5.7678e-03, 1.6434e-02]],

         [[1.2408e-01, 3.0220e-05, 9.0182e-05,  ..., 5.3673e-03,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6111e-02, 9.5367e-06, 9.0003e-06,  ..., 1.1432e-01,
           9.5520e-02, 3.0460e-03]],

         [[7.6660e-01, 2.3305e-05, 2.2233e-05,  ..., 3.1647e-02,
           1.3138e-02, 1.4046e-02]],

         [[1.3977e-01, 4.6790e-05, 7.8797e-05,  ..., 1.4026e-01,
           4.3304e-02, 4.2114e-02]],

         ...,

         [[6.7578e-01, 1.1563e-05, 3.9995e-05,  ..., 4.7089e-02,
           9.1492e-02, 2.5208e-02]],

         [[6.2744e-01, 1.3888e-04, 8.9979e-04,  ..., 1.6205e-02,
           8.2397e-03, 2.5894e-02]],

         [[6.0254e-01, 1.1444e-05, 6.1989e-05,  ..., 6.4354e-03,
           2.9297e-02, 5.9631e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6724e-01, 5.6982e-05, 6.8486e-05,  ..., 3.4790e-02,
           5.2528e-03, 1.7365e-02]],

         [[2.0459e-01, 6.1870e-05, 5.7459e-05,  ..., 1.6922e-02,
           1.3542e-02, 1.3916e-02]],

         [[7.8857e-02, 1.0610e-04, 2.0134e-04,  ..., 5.7251e-02,
           7.0129e-02, 7.0129e-02]],

         ...,

         [[3.7048e-02, 6.9141e-06, 1.0169e-04,  ..., 5.6381e-03,
           1.0956e-02, 1.2102e-03]],

         [[1.0004e-01, 7.8678e-06, 5.6624e-05,  ..., 1.1223e-02,
           2.8671e-02, 1.2520e-02]],

         [[9.0088e-01, 1.6391e-05, 2.0087e-05,  ..., 1.3054e-02,
           4.9255e-02, 3.7403e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8257e-01, 2.4438e-06, 1.1027e-05,  ..., 2.1289e-01,
           1.7786e-01, 1.4868e-01]],

         [[3.9233e-01, 2.8014e-05, 3.2306e-05,  ..., 3.8391e-02,
           3.6377e-02, 2.9053e-01]],

         [[5.5122e-03, 1.7881e-07, 1.1921e-07,  ..., 4.7339e-01,
           4.1455e-01, 3.8552e-04]],

         ...,

         [[1.6272e-01, 1.2970e-04, 3.1042e-04,  ..., 1.6357e-02,
           3.5461e-02, 6.5247e-02]],

         [[3.4766e-01, 9.4235e-05, 4.9055e-05,  ..., 4.2175e-02,
           1.6553e-01, 1.8417e-02]],

         [[2.0691e-02, 2.6226e-05, 6.5207e-05,  ..., 3.2043e-02,
           2.3270e-02, 7.4707e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9370e-01, 3.2187e-06, 5.2333e-05,  ..., 9.9121e-02,
           1.6211e-01, 3.0054e-01]],

         [[5.8984e-01, 3.1757e-04, 8.6164e-04,  ..., 3.4943e-03,
           9.9564e-03, 2.6642e-02]],

         [[8.7952e-02, 7.5698e-06, 4.1187e-05,  ..., 4.7722e-03,
           2.8564e-02, 3.2349e-02]],

         ...,

         [[6.8555e-01, 7.1228e-05, 7.7009e-05,  ..., 8.8562e-02,
           7.2266e-02, 2.4200e-02]],

         [[9.6313e-02, 6.3241e-05, 1.6415e-04,  ..., 9.8419e-03,
           1.5732e-02, 8.4521e-01]],

         [[1.3477e-01, 2.4557e-05, 7.6354e-05,  ..., 6.6071e-03,
           7.8809e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8027e-01, 4.5240e-05, 4.3774e-04,  ..., 3.6499e-02,
           4.8920e-02, 6.1829e-02]],

         [[7.3059e-02, 1.7703e-05, 3.8922e-05,  ..., 2.9068e-02,
           7.9346e-02, 6.5491e-02]],

         [[5.3833e-02, 1.2481e-04, 5.8603e-04,  ..., 1.2085e-01,
           2.7344e-01, 3.0441e-02]],

         ...,

         [[8.3887e-01, 1.3196e-04, 1.3459e-04,  ..., 7.1182e-03,
           1.6006e-02, 4.0970e-03]],

         [[7.5391e-01, 1.3709e-06, 7.3814e-04,  ..., 4.2343e-03,
           1.5278e-03, 5.2917e-02]],

         [[5.7959e-01, 8.3685e-05, 8.0645e-05,  ..., 3.6743e-02,
           7.2510e-02, 1.0388e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1192e-04, 4.9019e-04, 1.4417e-01,  ..., 1.6651e-03,
           9.0714e-03, 3.3295e-02]],

         [[1.0842e-04, 9.5189e-05, 1.3232e-01,  ..., 9.9915e-02,
           3.6392e-03, 1.9669e-02]],

         [[2.2089e-04, 1.9765e-04, 4.7974e-02,  ..., 3.9917e-02,
           8.9600e-02, 1.0522e-01]],

         ...,

         [[3.8505e-05, 1.7285e-05, 2.9587e-02,  ..., 2.9199e-01,
           3.1079e-01, 6.9336e-02]],

         [[8.5020e-04, 9.6369e-04, 1.7456e-01,  ..., 1.5976e-02,
           3.0334e-02, 1.7548e-02]],

         [[3.5763e-06, 4.7684e-07, 1.0300e-02,  ..., 6.6757e-06,
           8.9109e-05, 8.0633e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7680e-04, 5.1618e-05, 2.9663e-01,  ..., 7.3814e-04,
           9.7752e-04, 3.4882e-02]],

         [[1.1408e-04, 6.8426e-05, 2.4597e-01,  ..., 4.6310e-03,
           2.7800e-04, 2.2522e-02]],

         [[2.0087e-05, 6.8545e-06, 2.5366e-01,  ..., 1.7643e-05,
           6.3658e-04, 2.2171e-02]],

         ...,

         [[9.4891e-04, 2.0885e-04, 2.7832e-01,  ..., 4.5700e-03,
           3.6359e-04, 6.1989e-03]],

         [[3.9637e-05, 4.5478e-05, 3.1079e-01,  ..., 2.0099e-04,
           3.3259e-04, 3.1494e-02]],

         [[6.4492e-05, 4.4942e-05, 2.2302e-01,  ..., 6.3133e-04,
           1.0948e-03, 8.9188e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7716e-05, 3.2783e-05, 2.1777e-01,  ..., 2.2829e-05,
           4.8327e-04, 3.0640e-02]],

         [[7.0989e-05, 3.8147e-06, 2.6538e-01,  ..., 1.1307e-02,
           3.6073e-04, 2.2491e-02]],

         [[6.3598e-05, 1.5819e-04, 2.6294e-01,  ..., 1.6451e-03,
           7.8344e-04, 1.8616e-02]],

         ...,

         [[1.8597e-05, 3.9697e-05, 1.6895e-01,  ..., 6.6650e-02,
           1.1627e-02, 1.3257e-01]],

         [[1.7285e-05, 4.5896e-05, 1.6028e-01,  ..., 2.1103e-02,
           9.1314e-04, 1.5991e-02]],

         [[1.1545e-04, 4.3094e-05, 1.6296e-01,  ..., 3.6411e-03,
           4.2486e-04, 9.1171e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3631e-05, 2.6107e-05, 4.6844e-03,  ..., 1.2372e-01,
           6.5332e-01, 4.6600e-02]],

         [[5.5850e-05, 2.7919e-04, 1.2585e-01,  ..., 9.1370e-02,
           5.2277e-02, 9.4681e-03]],

         [[8.8334e-05, 6.6614e-04, 9.3506e-02,  ..., 5.9175e-04,
           1.8060e-04, 3.0613e-04]],

         ...,

         [[3.0861e-03, 5.7411e-04, 2.7084e-02,  ..., 4.3602e-03,
           5.0087e-03, 6.6223e-03]],

         [[9.1124e-04, 3.0470e-04, 1.6541e-02,  ..., 6.6956e-02,
           3.0249e-01, 2.7481e-02]],

         [[2.3711e-04, 6.1154e-05, 4.1534e-02,  ..., 3.3508e-02,
           1.0056e-02, 3.4973e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0055, 0.0010, 0.1638,  ..., 0.1348, 0.0202, 0.0223]],

         [[0.0004, 0.0008, 0.2350,  ..., 0.0208, 0.0168, 0.1542]],

         [[0.0034, 0.0011, 0.1915,  ..., 0.0368, 0.0211, 0.0542]],

         ...,

         [[0.0019, 0.0007, 0.0858,  ..., 0.2012, 0.0757, 0.1299]],

         [[0.0008, 0.0014, 0.1866,  ..., 0.0077, 0.0101, 0.2759]],

         [[0.0003, 0.0008, 0.1636,  ..., 0.0365, 0.0151, 0.0424]]]],
       device='cuda:0', dtype=torch.float16)), (tensor([[[[5.7888e-04, 1.2910e-04, 7.9107e-04,  ..., 7.1924e-01,
           1.5552e-01, 6.6895e-02]],

         [[7.1335e-04, 2.9731e-04, 4.6539e-03,  ..., 3.0334e-02,
           3.9355e-01, 1.7288e-02]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.3709e-06,
           2.0337e-04, 9.9951e-01]],

         ...,

         [[1.6272e-05, 2.6822e-05, 5.3835e-04,  ..., 3.9787e-03,
           6.5575e-03, 9.7314e-01]],

         [[6.8817e-03, 6.4659e-03, 5.3596e-03,  ..., 2.7222e-02,
           4.7791e-02, 8.3008e-03]],

         [[1.5556e-02, 3.3646e-03, 9.3412e-04,  ..., 1.8188e-02,
           1.2115e-02, 4.0985e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5736e-03, 2.2202e-03, 9.9487e-03,  ..., 4.3884e-02,
           1.6821e-01, 2.6050e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 1.5369e-01,
           7.6855e-01, 7.3792e-02]],

         [[8.1587e-04, 1.8673e-03, 7.1602e-03,  ..., 2.4994e-02,
           2.4216e-02, 1.1436e-02]],

         ...,

         [[1.7233e-03, 2.1610e-03, 8.0948e-03,  ..., 1.9882e-02,
           7.0496e-02, 1.0663e-01]],

         [[4.7569e-03, 1.2535e-02, 3.8177e-02,  ..., 1.3290e-02,
           1.7410e-02, 1.2932e-02]],

         [[1.9894e-03, 3.4103e-03, 7.1106e-03,  ..., 2.1106e-01,
           2.5208e-02, 6.0944e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7148e-02, 7.7286e-03, 4.2938e-02,  ..., 5.2216e-02,
           5.0201e-02, 4.2938e-02]],

         [[2.8381e-03, 1.0900e-03, 4.2114e-03,  ..., 7.3624e-03,
           2.3708e-03, 1.0216e-02]],

         [[7.5951e-03, 2.0599e-03, 2.0676e-03,  ..., 3.2471e-02,
           3.6530e-02, 2.3300e-02]],

         ...,

         [[3.8452e-03, 5.0449e-04, 1.7881e-03,  ..., 1.7542e-01,
           2.3059e-01, 1.9263e-01]],

         [[3.3493e-03, 1.7548e-04, 5.8222e-04,  ..., 7.6843e-02,
           3.5187e-02, 1.0101e-01]],

         [[4.4975e-03, 1.0147e-03, 2.1744e-03,  ..., 4.8828e-01,
           1.6614e-01, 7.7881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8418e-01, 3.2187e-06, 4.0531e-06,  ..., 1.9054e-03,
           2.4185e-03, 1.2573e-02]],

         [[9.4336e-01, 4.3511e-06, 9.1195e-06,  ..., 2.8648e-03,
           2.3636e-02, 5.1155e-03]],

         [[6.9189e-01, 5.1856e-06, 2.0862e-05,  ..., 4.4403e-03,
           1.6460e-03, 2.1606e-02]],

         ...,

         [[8.4351e-02, 4.4441e-04, 6.4659e-04,  ..., 2.6011e-04,
           7.3862e-04, 8.9264e-03]],

         [[5.4199e-01, 1.5676e-04, 1.1539e-03,  ..., 2.0538e-02,
           1.7563e-02, 1.1841e-02]],

         [[3.4692e-01, 2.2840e-04, 4.8971e-04,  ..., 6.2439e-02,
           2.9251e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1226e-01, 2.9206e-06, 3.5167e-06,  ..., 4.4346e-05,
           3.9160e-05, 6.6986e-03]],

         [[3.5767e-01, 5.0664e-06, 5.9605e-08,  ..., 3.8965e-01,
           2.3083e-01, 1.8829e-02]],

         [[8.9453e-01, 2.7323e-04, 1.7524e-05,  ..., 3.5126e-02,
           1.8188e-02, 4.8447e-03]],

         ...,

         [[3.5596e-01, 1.1045e-04, 2.2840e-04,  ..., 8.4457e-03,
           7.8506e-03, 2.2110e-02]],

         [[1.4429e-01, 1.9522e-03, 1.6851e-03,  ..., 3.0327e-03,
           1.3723e-03, 6.9885e-03]],

         [[1.9751e-01, 2.5868e-05, 7.9453e-05,  ..., 2.6587e-01,
           8.4656e-02, 4.9164e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 2.8729e-04, 1.3578e-04,  ..., 4.1870e-02,
           2.1301e-02, 2.9984e-02]],

         [[6.3330e-01, 1.1504e-05, 1.1444e-05,  ..., 1.2286e-01,
           9.5520e-03, 1.5078e-03]],

         [[8.6963e-01, 2.4259e-05, 5.6028e-06,  ..., 6.2714e-03,
           3.7903e-02, 2.3804e-03]],

         ...,

         [[4.2822e-01, 4.7684e-07, 1.7881e-06,  ..., 6.5279e-04,
           8.2350e-04, 5.4443e-01]],

         [[4.6582e-01, 1.3113e-06, 5.2750e-05,  ..., 2.5034e-06,
           2.3961e-05, 1.8799e-02]],

         [[9.3311e-01, 1.0729e-06, 8.4043e-06,  ..., 2.9802e-07,
           3.1590e-06, 1.9073e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4595e-01, 8.8274e-05, 1.2994e-05,  ..., 1.6260e-01,
           1.2830e-01, 2.5116e-02]],

         [[8.9746e-01, 4.4107e-06, 1.8477e-06,  ..., 5.4061e-05,
           1.6749e-05, 5.7907e-03]],

         [[7.8125e-01, 1.7500e-04, 5.4836e-05,  ..., 5.7487e-03,
           3.8509e-03, 3.3752e-02]],

         ...,

         [[9.0234e-01, 2.7227e-04, 1.8597e-05,  ..., 9.6359e-03,
           7.9117e-03, 1.2451e-02]],

         [[7.6807e-01, 1.2118e-04, 3.9518e-05,  ..., 6.7627e-02,
           4.4098e-02, 6.1615e-02]],

         [[8.8818e-01, 6.1989e-05, 3.4094e-05,  ..., 4.1565e-02,
           2.5497e-02, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7588e-01, 5.4955e-05, 3.1590e-05,  ..., 2.3239e-02,
           1.4923e-02, 5.0537e-02]],

         [[5.5713e-01, 9.4354e-05, 1.3733e-04,  ..., 1.0559e-01,
           2.2095e-02, 8.3557e-02]],

         [[8.3435e-02, 2.1458e-06, 3.5167e-06,  ..., 3.5254e-01,
           6.7749e-03, 2.3174e-03]],

         ...,

         [[5.8740e-01, 4.8995e-05, 1.7524e-05,  ..., 2.2003e-02,
           1.7975e-02, 2.8137e-02]],

         [[8.7939e-01, 2.9278e-04, 7.4267e-05,  ..., 2.7885e-03,
           1.0662e-03, 7.1602e-03]],

         [[3.9331e-01, 1.0669e-05, 9.0599e-06,  ..., 1.5881e-01,
           1.6052e-01, 6.4148e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 3.4161e-03, 1.2865e-03,  ..., 8.2731e-04,
           5.9128e-04, 8.2779e-03]],

         [[9.6045e-01, 9.9540e-06, 3.5346e-05,  ..., 1.7424e-03,
           5.5695e-04, 4.7827e-04]],

         [[8.7012e-01, 5.7757e-05, 5.2989e-05,  ..., 2.1324e-03,
           1.3142e-03, 1.1482e-02]],

         ...,

         [[5.7422e-01, 1.4770e-04, 3.4070e-04,  ..., 1.8265e-02,
           3.2501e-02, 1.6870e-01]],

         [[8.4570e-01, 4.4107e-06, 1.8418e-05,  ..., 4.9713e-02,
           1.3885e-02, 6.1684e-03]],

         [[8.7793e-01, 8.7619e-06, 1.3828e-05,  ..., 1.3329e-02,
           1.5434e-02, 6.6101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1899e-01, 3.8242e-04, 8.1873e-04,  ..., 2.1152e-03,
           1.4257e-03, 1.2505e-02]],

         [[5.1465e-01, 2.8491e-05, 2.4974e-05,  ..., 2.3413e-01,
           7.1045e-02, 3.0945e-02]],

         [[8.8086e-01, 2.2590e-05, 2.4021e-05,  ..., 4.0802e-02,
           1.5373e-02, 9.5444e-03]],

         ...,

         [[7.5000e-01, 9.8288e-05, 1.3304e-04,  ..., 2.6560e-04,
           8.3923e-05, 4.2033e-04]],

         [[7.8027e-01, 1.6356e-04, 4.5478e-05,  ..., 3.3630e-02,
           1.7685e-02, 6.6101e-02]],

         [[5.5518e-01, 1.2934e-05, 1.4484e-05,  ..., 1.9116e-01,
           1.2390e-01, 1.9638e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3416e-02, 4.1723e-06, 1.6153e-05,  ..., 7.8678e-06,
           9.4175e-06, 2.5673e-03]],

         [[4.8779e-01, 1.1265e-05, 2.0266e-05,  ..., 6.1035e-02,
           1.1276e-02, 4.8584e-02]],

         [[8.3887e-01, 8.3447e-07, 2.2054e-06,  ..., 2.0754e-04,
           5.9664e-05, 1.8644e-03]],

         ...,

         [[9.4238e-01, 5.8711e-05, 5.6267e-05,  ..., 2.9526e-03,
           2.8110e-04, 1.7567e-03]],

         [[7.3340e-01, 1.6546e-04, 6.3539e-05,  ..., 3.3951e-03,
           9.0820e-02, 1.3596e-02]],

         [[6.2939e-01, 7.2289e-04, 1.3268e-04,  ..., 7.9679e-04,
           2.1112e-04, 7.9651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1431e-01, 1.5199e-05, 1.6689e-05,  ..., 3.0575e-03,
           5.7373e-03, 5.4883e-01]],

         [[5.0732e-01, 8.8549e-04, 8.9073e-04,  ..., 3.2043e-03,
           2.8849e-04, 1.7487e-02]],

         [[6.0840e-01, 3.2377e-04, 3.1090e-04,  ..., 3.6373e-03,
           7.8058e-04, 7.8354e-03]],

         ...,

         [[3.2642e-01, 9.2506e-05, 3.0684e-04,  ..., 2.1713e-02,
           1.9932e-03, 6.7291e-03]],

         [[4.0796e-01, 3.5703e-05, 4.3213e-05,  ..., 1.2262e-01,
           9.5825e-03, 3.2532e-02]],

         [[3.3960e-01, 4.5598e-05, 1.1283e-04,  ..., 1.0315e-02,
           3.1860e-02, 5.4395e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4717e-01, 1.6046e-04, 2.0921e-04,  ..., 1.6510e-02,
           5.7678e-02, 5.0000e-01]],

         [[6.9141e-01, 8.1420e-05, 1.1623e-04,  ..., 1.8753e-02,
           3.8185e-03, 1.5762e-02]],

         [[4.1602e-01, 3.4261e-04, 6.1798e-04,  ..., 6.1684e-03,
           9.7227e-04, 9.3002e-03]],

         ...,

         [[9.3323e-02, 3.6895e-05, 8.4937e-05,  ..., 4.1628e-04,
           8.7967e-03, 8.4473e-01]],

         [[7.4512e-01, 3.2282e-04, 2.7823e-04,  ..., 2.7294e-03,
           1.4511e-02, 2.1484e-02]],

         [[7.3633e-01, 1.0198e-04, 3.8981e-05,  ..., 2.7924e-02,
           7.8821e-04, 1.1475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.3324e-04, 1.1854e-03,  ..., 1.4137e-02,
           6.4964e-03, 8.0505e-02]],

         [[6.9385e-01, 1.9014e-05, 5.2571e-05,  ..., 3.4058e-02,
           1.3351e-02, 1.1432e-01]],

         [[1.8774e-01, 5.1260e-06, 1.6391e-05,  ..., 1.8013e-04,
           7.7963e-04, 1.0307e-02]],

         ...,

         [[8.1836e-01, 1.3113e-06, 6.5565e-07,  ..., 7.7454e-02,
           5.6366e-02, 2.3499e-02]],

         [[5.4785e-01, 7.2813e-04, 1.2035e-03,  ..., 3.8986e-03,
           7.8440e-04, 1.4259e-02]],

         [[6.3330e-01, 2.3842e-06, 2.5630e-06,  ..., 5.5847e-02,
           2.1255e-02, 9.4238e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5010e-01, 5.4598e-05, 1.9848e-05,  ..., 4.7607e-03,
           6.2275e-04, 3.6182e-03]],

         [[1.4880e-01, 3.7384e-04, 3.1281e-04,  ..., 6.0892e-04,
           5.1022e-04, 1.7471e-03]],

         [[6.4746e-01, 2.2197e-04, 1.3304e-04,  ..., 1.6365e-03,
           5.6458e-04, 1.2779e-02]],

         ...,

         [[5.2783e-01, 4.1723e-07, 4.1723e-07,  ..., 3.1158e-02,
           3.7460e-03, 1.5526e-02]],

         [[3.4595e-01, 5.9605e-08, 5.9605e-08,  ..., 1.6284e-01,
           6.6589e-02, 8.2153e-02]],

         [[3.0811e-01, 3.3975e-06, 8.5235e-06,  ..., 5.3253e-03,
           2.0695e-03, 1.5823e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6721e-02, 2.3842e-07, 6.5565e-07,  ..., 7.2908e-04,
           2.6393e-04, 8.9551e-01]],

         [[1.0455e-01, 8.4043e-06, 2.5332e-05,  ..., 1.6968e-01,
           6.3538e-02, 6.3782e-02]],

         [[2.9590e-01, 2.3162e-04, 4.9400e-04,  ..., 9.9182e-03,
           1.7424e-03, 3.2501e-02]],

         ...,

         [[4.3457e-01, 7.1347e-05, 1.3804e-04,  ..., 2.3911e-02,
           3.3150e-03, 2.5391e-02]],

         [[7.2461e-01, 1.0133e-06, 1.2875e-05,  ..., 7.5264e-03,
           2.2335e-03, 6.8169e-03]],

         [[3.9941e-01, 1.2612e-04, 1.1754e-04,  ..., 1.0696e-02,
           1.7891e-03, 1.0017e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8643e-01, 2.4557e-05, 4.0352e-05,  ..., 1.7914e-02,
           8.5449e-03, 1.5002e-01]],

         [[4.7803e-01, 3.9887e-04, 5.7888e-04,  ..., 1.0941e-02,
           4.5166e-03, 3.8483e-02]],

         [[5.2185e-02, 7.7486e-06, 1.1921e-05,  ..., 6.9153e-02,
           6.9275e-03, 4.0344e-02]],

         ...,

         [[6.8213e-01, 2.4557e-05, 2.0325e-05,  ..., 2.6978e-02,
           2.7561e-03, 8.3740e-02]],

         [[5.6006e-01, 1.2994e-05, 2.6584e-05,  ..., 1.0559e-01,
           3.4351e-03, 1.1688e-01]],

         [[8.2520e-01, 4.6015e-05, 4.9412e-05,  ..., 1.4467e-03,
           4.6134e-04, 9.8190e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8882e-01, 1.7090e-03, 7.8917e-04,  ..., 4.2963e-04,
           2.1935e-04, 4.8294e-03]],

         [[3.7598e-01, 6.1226e-04, 4.5180e-04,  ..., 1.6222e-03,
           2.1420e-03, 7.8735e-03]],

         [[3.5400e-01, 2.2471e-05, 3.2306e-05,  ..., 1.6333e-01,
           1.3374e-02, 4.1931e-02]],

         ...,

         [[3.7280e-01, 2.3663e-04, 9.1434e-05,  ..., 7.8430e-03,
           7.8583e-03, 4.4006e-02]],

         [[8.3203e-01, 6.4194e-05, 6.0987e-04,  ..., 3.2783e-06,
           3.5214e-04, 5.8861e-03]],

         [[8.4570e-01, 1.5640e-04, 1.3089e-04,  ..., 2.3880e-03,
           7.0143e-04, 2.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3789e-01, 2.4557e-04, 3.5739e-04,  ..., 8.5449e-02,
           1.1925e-02, 3.9734e-02]],

         [[2.6709e-01, 1.3475e-03, 7.4148e-04,  ..., 9.0637e-03,
           2.5177e-03, 1.1024e-02]],

         [[9.7778e-02, 1.2815e-05, 1.2279e-05,  ..., 4.3030e-02,
           3.0289e-02, 4.3823e-01]],

         ...,

         [[3.6548e-01, 2.4796e-05, 5.2035e-05,  ..., 1.0612e-02,
           8.6689e-04, 1.3214e-02]],

         [[5.3320e-01, 8.7976e-05, 8.1480e-05,  ..., 2.3621e-02,
           1.0857e-02, 5.7770e-02]],

         [[2.8101e-01, 3.9279e-05, 3.3855e-05,  ..., 6.4148e-02,
           3.8025e-02, 2.6611e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3340e-01, 3.1292e-05, 3.2783e-05,  ..., 7.1907e-03,
           1.8646e-02, 8.0185e-03]],

         [[1.2451e-01, 1.7405e-05, 1.1444e-05,  ..., 9.5459e-02,
           1.0880e-02, 3.8574e-02]],

         [[1.6406e-01, 3.9637e-05, 2.2233e-05,  ..., 9.2545e-03,
           9.1095e-03, 7.4658e-01]],

         ...,

         [[3.9520e-02, 5.2631e-05, 8.2791e-05,  ..., 6.0940e-04,
           1.4734e-03, 2.3788e-02]],

         [[1.4221e-01, 2.8110e-04, 2.5702e-04,  ..., 2.2339e-02,
           3.6255e-02, 3.6888e-03]],

         [[2.4426e-01, 1.2321e-03, 9.0885e-04,  ..., 4.2496e-03,
           7.7896e-03, 2.6398e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 3.9935e-05, 3.6538e-05,  ..., 2.7527e-02,
           6.4888e-03, 4.1016e-02]],

         [[3.2910e-01, 9.6679e-05, 1.9312e-05,  ..., 5.7411e-04,
           1.0866e-04, 9.4833e-03]],

         [[8.1885e-01, 3.2961e-05, 7.4506e-06,  ..., 1.4057e-03,
           7.9036e-05, 6.5002e-03]],

         ...,

         [[4.5685e-02, 5.4455e-04, 1.0252e-03,  ..., 3.8090e-03,
           9.4843e-04, 1.0857e-02]],

         [[5.3906e-01, 5.6922e-05, 1.2338e-04,  ..., 3.4122e-03,
           2.5368e-03, 1.7059e-02]],

         [[4.2676e-01, 6.7759e-04, 2.1362e-03,  ..., 3.6133e-02,
           3.1090e-03, 1.6418e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4438e-01, 3.8266e-05, 9.0718e-05,  ..., 5.1498e-03,
           9.7733e-03, 1.0490e-02]],

         [[7.9529e-02, 1.4532e-04, 4.3058e-04,  ..., 1.1276e-02,
           1.2680e-02, 2.5421e-02]],

         [[1.4587e-01, 9.8050e-05, 2.0444e-04,  ..., 9.1095e-03,
           1.1879e-02, 1.7014e-02]],

         ...,

         [[3.2104e-02, 5.9128e-05, 8.7380e-05,  ..., 4.6692e-02,
           3.1113e-02, 5.0476e-02]],

         [[1.0370e-01, 7.8380e-05, 7.5936e-05,  ..., 6.3248e-03,
           1.1101e-02, 3.8147e-02]],

         [[5.3369e-01, 8.5890e-05, 1.2493e-04,  ..., 1.4214e-02,
           7.7286e-03, 1.4000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3623e-01, 2.0325e-05, 1.8656e-05,  ..., 5.2094e-02,
           1.1177e-02, 3.8116e-02]],

         [[4.0454e-01, 1.9670e-04, 4.1461e-04,  ..., 1.2024e-02,
           1.3412e-02, 1.7639e-02]],

         [[1.2323e-01, 7.4863e-05, 2.0993e-04,  ..., 4.3243e-02,
           5.0964e-02, 6.3086e-01]],

         ...,

         [[5.6348e-01, 8.2254e-05, 1.4555e-04,  ..., 1.8539e-02,
           1.1787e-02, 3.8055e-02]],

         [[2.0740e-01, 2.1696e-05, 2.9993e-04,  ..., 2.9125e-03,
           1.9287e-02, 5.0446e-02]],

         [[4.0430e-01, 4.0340e-04, 5.8031e-04,  ..., 3.0403e-03,
           3.0155e-03, 7.3471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6760e-01, 2.0730e-04, 1.6665e-04,  ..., 2.7802e-02,
           2.4719e-02, 8.3008e-02]],

         [[5.9723e-02, 5.2452e-05, 7.9572e-05,  ..., 1.7242e-02,
           1.7929e-02, 3.1235e-02]],

         [[2.0996e-01, 4.9353e-05, 7.4565e-05,  ..., 4.0627e-03,
           1.4830e-03, 1.6870e-03]],

         ...,

         [[7.4121e-01, 1.0729e-05, 1.3292e-05,  ..., 4.7394e-02,
           8.5144e-02, 5.4108e-02]],

         [[3.1982e-01, 1.9372e-05, 8.9645e-05,  ..., 2.4673e-02,
           4.9744e-03, 2.8168e-02]],

         [[1.9409e-01, 2.0981e-05, 2.8551e-05,  ..., 4.3066e-01,
           1.4307e-01, 2.4872e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2253e-02, 6.6757e-06, 3.1948e-05,  ..., 2.7332e-03,
           5.2977e-04, 1.3145e-02]],

         [[1.5900e-02, 5.3644e-07, 1.1921e-06,  ..., 4.4751e-04,
           3.0117e-03, 9.7607e-01]],

         [[2.7145e-02, 8.6546e-05, 4.4918e-04,  ..., 2.7943e-03,
           7.6234e-05, 4.5991e-04]],

         ...,

         [[1.0669e-01, 3.0708e-04, 9.4128e-04,  ..., 8.9645e-03,
           1.1549e-03, 9.0332e-03]],

         [[8.7280e-02, 7.1526e-07, 1.8477e-06,  ..., 6.3904e-02,
           7.0267e-03, 2.3224e-02]],

         [[2.9114e-02, 4.1723e-07, 1.9670e-06,  ..., 4.2285e-01,
           2.0129e-01, 1.1932e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0664e-01, 2.0564e-04, 7.5150e-04,  ..., 7.0343e-03,
           8.7166e-04, 1.1238e-02]],

         [[9.7900e-01, 1.4865e-04, 2.9111e-04,  ..., 2.9862e-05,
           1.9550e-05, 4.4727e-04]],

         [[6.6260e-01, 2.6941e-05, 2.5797e-04,  ..., 9.4318e-04,
           1.1024e-03, 2.9114e-02]],

         ...,

         [[2.9492e-01, 1.8895e-05, 1.6153e-05,  ..., 3.5571e-01,
           1.1731e-01, 4.5929e-02]],

         [[4.4458e-01, 2.1458e-06, 5.9605e-07,  ..., 1.0357e-03,
           1.0519e-03, 5.9586e-03]],

         [[6.4636e-02, 2.3425e-05, 7.6890e-05,  ..., 1.2026e-03,
           4.2648e-03, 8.7842e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8054e-02, 7.6890e-06, 1.1921e-05,  ..., 1.3512e-02,
           1.8282e-03, 2.1667e-02]],

         [[5.7959e-01, 5.4538e-05, 2.5928e-05,  ..., 7.9346e-02,
           1.6449e-02, 6.2286e-02]],

         [[3.1006e-01, 3.3307e-04, 3.9458e-04,  ..., 7.8369e-02,
           6.8115e-02, 7.9041e-02]],

         ...,

         [[3.3813e-01, 1.7107e-05, 8.8334e-05,  ..., 8.8867e-02,
           2.1439e-02, 8.0261e-02]],

         [[6.8506e-01, 5.3763e-05, 1.3387e-04,  ..., 2.3766e-03,
           1.8806e-03, 1.0567e-02]],

         [[5.0586e-01, 4.8041e-05, 1.9228e-04,  ..., 1.1261e-02,
           7.5607e-03, 2.8992e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9662e-03, 3.1590e-06, 1.2517e-06,  ..., 5.2223e-03,
           3.4714e-04, 8.8644e-04]],

         [[4.8401e-02, 5.5432e-06, 1.7881e-06,  ..., 6.6423e-04,
           2.5105e-04, 1.0653e-03]],

         [[1.1908e-01, 7.3195e-05, 1.4961e-04,  ..., 8.7830e-02,
           1.6632e-02, 1.4819e-01]],

         ...,

         [[5.6000e-02, 5.2691e-05, 3.8028e-04,  ..., 7.8201e-03,
           1.4238e-03, 6.4812e-03]],

         [[1.2366e-01, 7.2837e-05, 3.5834e-04,  ..., 1.6998e-02,
           1.1044e-03, 1.1147e-02]],

         [[1.6980e-01, 5.2869e-05, 5.2273e-05,  ..., 2.6099e-01,
           2.5238e-02, 1.7075e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1123e-01, 1.0848e-05, 1.4544e-05,  ..., 1.4087e-01,
           4.2786e-02, 8.9539e-02]],

         [[3.2861e-01, 7.9691e-05, 1.4412e-04,  ..., 1.5282e-02,
           7.3166e-03, 2.6929e-01]],

         [[4.5679e-01, 5.4240e-06, 1.5438e-05,  ..., 1.0938e-01,
           1.1823e-01, 1.6736e-01]],

         ...,

         [[4.7192e-01, 1.4114e-04, 1.4138e-04,  ..., 1.6403e-02,
           3.7613e-03, 1.3283e-02]],

         [[1.7908e-01, 8.7023e-06, 7.8082e-06,  ..., 1.7357e-03,
           3.9792e-04, 1.4587e-02]],

         [[7.6904e-03, 2.5868e-05, 7.2241e-05,  ..., 1.0033e-02,
           2.7054e-02, 8.6816e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6025e-01, 5.5194e-05, 2.0671e-04,  ..., 5.9021e-02,
           2.9785e-02, 1.7212e-01]],

         [[1.1749e-01, 1.3733e-03, 1.4992e-03,  ..., 6.2027e-03,
           6.5231e-03, 1.8082e-02]],

         [[1.6223e-01, 2.0456e-04, 3.3593e-04,  ..., 1.3847e-02,
           2.0905e-03, 6.8283e-03]],

         ...,

         [[6.7578e-01, 6.1131e-04, 1.9693e-04,  ..., 7.1228e-02,
           1.9165e-02, 3.0151e-02]],

         [[6.4453e-02, 2.2531e-04, 6.3181e-04,  ..., 2.8763e-03,
           1.0193e-02, 8.3545e-01]],

         [[3.2642e-01, 3.3379e-05, 1.6177e-04,  ..., 2.5415e-01,
           3.3667e-01, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5469e-01, 2.7776e-04, 1.0977e-03,  ..., 9.8267e-02,
           2.7686e-01, 5.9662e-03]],

         [[1.2793e-01, 1.5914e-05, 3.3498e-05,  ..., 7.0763e-03,
           5.4779e-03, 1.0645e-01]],

         [[5.4102e-01, 3.9840e-04, 2.1820e-03,  ..., 8.2626e-03,
           6.4163e-03, 1.2802e-02]],

         ...,

         [[6.5918e-01, 2.7323e-04, 4.5586e-03,  ..., 2.3605e-02,
           2.9602e-03, 1.3496e-02]],

         [[8.9355e-01, 2.1374e-04, 2.0027e-03,  ..., 1.6413e-03,
           8.6927e-04, 1.1833e-02]],

         [[4.0015e-01, 1.6832e-04, 1.7810e-04,  ..., 2.8711e-01,
           3.1342e-02, 5.8319e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2343e-04, 5.9843e-04, 2.2729e-01,  ..., 1.2817e-02,
           1.0880e-02, 1.8330e-03]],

         [[8.0287e-05, 2.5058e-04, 1.9556e-01,  ..., 2.4815e-03,
           7.2365e-03, 2.9583e-03]],

         [[1.0395e-03, 1.0748e-03, 1.2030e-01,  ..., 2.7054e-02,
           1.2726e-02, 2.8793e-02]],

         ...,

         [[2.8849e-04, 4.6730e-05, 5.6519e-02,  ..., 7.7820e-02,
           7.0312e-02, 2.4170e-01]],

         [[2.8825e-04, 2.1629e-03, 1.9617e-01,  ..., 1.5915e-02,
           2.3193e-03, 5.5885e-03]],

         [[3.5644e-04, 4.1676e-04, 2.5635e-01,  ..., 9.5596e-03,
           5.4474e-03, 1.8005e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5125e-03, 2.0447e-03, 3.0225e-01,  ..., 3.5591e-03,
           3.0804e-03, 7.8201e-04]],

         [[1.7462e-03, 1.5774e-03, 2.3230e-01,  ..., 1.2426e-03,
           2.2945e-03, 1.2379e-03]],

         [[1.1902e-03, 2.4748e-04, 3.3789e-01,  ..., 9.8324e-04,
           1.3123e-03, 1.9321e-03]],

         ...,

         [[2.8076e-03, 5.9891e-04, 2.4878e-01,  ..., 5.3167e-04,
           9.0075e-04, 1.0509e-03]],

         [[1.6892e-04, 3.3069e-04, 3.4082e-01,  ..., 9.9945e-04,
           4.7836e-03, 2.8014e-04]],

         [[2.9981e-05, 9.8825e-05, 1.0590e-01,  ..., 1.2970e-03,
           7.9250e-04, 1.6165e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3390e-03, 1.1101e-03, 2.3682e-01,  ..., 5.5552e-04,
           1.9302e-03, 9.6512e-04]],

         [[5.2032e-03, 2.0409e-04, 2.7954e-01,  ..., 1.4277e-03,
           3.3188e-03, 1.1450e-04]],

         [[1.3933e-03, 1.8969e-03, 3.2275e-01,  ..., 1.2102e-03,
           1.8597e-03, 8.9788e-04]],

         ...,

         [[3.5763e-04, 7.7677e-04, 1.6968e-01,  ..., 3.5980e-02,
           2.2079e-02, 1.1963e-02]],

         [[7.0238e-04, 1.5078e-03, 2.5513e-01,  ..., 1.3180e-03,
           1.2922e-03, 5.0640e-04]],

         [[2.8439e-03, 1.4162e-03, 1.5906e-01,  ..., 1.3838e-03,
           1.2255e-03, 8.1682e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3355e-04, 3.8207e-05, 2.1042e-02,  ..., 2.6596e-02,
           3.4698e-02, 1.8604e-01]],

         [[1.0699e-04, 2.5272e-04, 1.7285e-01,  ..., 8.0490e-03,
           3.8033e-03, 3.2593e-02]],

         [[2.7704e-04, 1.0939e-03, 3.1421e-01,  ..., 1.5628e-04,
           1.8358e-04, 1.2636e-03]],

         ...,

         [[9.6321e-05, 1.1027e-05, 1.6510e-02,  ..., 1.1396e-04,
           1.9026e-04, 1.5747e-02]],

         [[5.7678e-03, 4.6015e-04, 1.5112e-01,  ..., 1.0948e-02,
           4.1473e-02, 9.3445e-02]],

         [[1.5378e-05, 4.5896e-06, 4.3983e-03,  ..., 5.0545e-04,
           7.7629e-04, 2.5977e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0117e-03, 1.3189e-03, 2.0801e-01,  ..., 6.7635e-03,
           1.2863e-02, 4.1779e-02]],

         [[1.0357e-03, 1.1921e-03, 2.9077e-01,  ..., 1.0748e-03,
           6.3858e-03, 1.9257e-02]],

         [[4.8676e-03, 2.1591e-03, 2.0190e-01,  ..., 1.9455e-02,
           4.5410e-02, 2.7481e-02]],

         ...,

         [[3.1319e-03, 3.3808e-04, 2.2839e-01,  ..., 1.0090e-03,
           4.9667e-03, 1.2779e-02]],

         [[1.5345e-03, 3.1605e-03, 1.1816e-01,  ..., 5.2856e-02,
           4.0833e-02, 3.7549e-01]],

         [[5.9748e-04, 8.9109e-05, 2.9694e-02,  ..., 8.5266e-02,
           1.2402e-01, 2.7954e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.1231e-04, 2.7275e-04, 6.9618e-04,  ..., 1.7041e-01,
           4.7778e-01, 2.7222e-01]],

         [[4.0436e-03, 7.9632e-04, 1.4122e-02,  ..., 2.0752e-01,
           2.9892e-02, 2.4780e-02]],

         [[5.9605e-08, 5.9605e-08, 7.4506e-05,  ..., 5.5027e-04,
           3.3379e-04, 9.9512e-01]],

         ...,

         [[2.8610e-06, 9.9540e-06, 2.0027e-04,  ..., 2.0027e-04,
           6.6338e-03, 9.8438e-01]],

         [[4.9629e-03, 6.3744e-03, 1.9638e-02,  ..., 2.0905e-02,
           4.3793e-03, 1.2024e-01]],

         [[9.9869e-03, 4.4327e-03, 5.9986e-04,  ..., 5.7465e-02,
           3.2745e-02, 1.6464e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3341e-04, 1.6203e-03, 4.7607e-03,  ..., 1.2280e-01,
           1.2091e-01, 1.4807e-01]],

         [[1.1921e-07, 2.3842e-07, 1.1921e-07,  ..., 1.5063e-01,
           7.4170e-01, 1.0193e-01]],

         [[2.3174e-03, 3.0231e-03, 1.1772e-02,  ..., 1.5114e-02,
           1.8799e-02, 1.9409e-02]],

         ...,

         [[5.4777e-05, 8.2850e-05, 1.3905e-03,  ..., 6.7017e-02,
           3.1226e-01, 1.3013e-01]],

         [[3.6449e-03, 6.8626e-03, 2.6306e-02,  ..., 1.2527e-02,
           1.3702e-02, 4.9744e-02]],

         [[1.4496e-04, 3.3975e-04, 1.5230e-03,  ..., 2.6367e-02,
           7.2205e-02, 1.0266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2077e-02, 1.4648e-03, 2.6321e-03,  ..., 2.3328e-01,
           1.4380e-01, 6.9519e-02]],

         [[3.6907e-03, 9.6655e-04, 3.5362e-03,  ..., 2.0237e-03,
           8.8577e-03, 2.0111e-02]],

         [[6.3362e-03, 1.5354e-03, 1.0910e-03,  ..., 3.3081e-02,
           2.5955e-02, 4.5044e-02]],

         ...,

         [[1.1808e-04, 1.3151e-03, 4.5166e-03,  ..., 9.5825e-02,
           5.3027e-01, 2.6660e-01]],

         [[2.3193e-03, 2.6011e-04, 4.2081e-04,  ..., 4.1443e-02,
           1.1896e-01, 1.3477e-01]],

         [[4.6295e-02, 7.8583e-03, 1.8707e-02,  ..., 3.9276e-02,
           2.8290e-02, 8.1848e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2920e-01, 1.6689e-06, 1.7881e-07,  ..., 5.6229e-03,
           3.6449e-03, 2.6360e-03]],

         [[8.8477e-01, 5.0068e-06, 1.1861e-05,  ..., 2.7390e-02,
           4.6448e-02, 2.2411e-03]],

         [[8.9062e-01, 7.7486e-07, 1.6689e-06,  ..., 4.5662e-03,
           1.2217e-03, 4.2023e-02]],

         ...,

         [[5.2979e-01, 2.1887e-04, 1.1368e-03,  ..., 2.6226e-03,
           1.0853e-03, 3.6072e-02]],

         [[5.5811e-01, 2.4819e-04, 5.1165e-04,  ..., 1.7593e-02,
           1.6663e-02, 1.2718e-02]],

         [[7.0654e-01, 3.8743e-04, 4.8733e-04,  ..., 5.7449e-03,
           5.2109e-03, 2.3438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0273e-01, 2.2113e-05, 8.7619e-06,  ..., 2.7027e-03,
           3.2091e-04, 2.7802e-02]],

         [[9.4922e-01, 7.4506e-06, 4.1723e-07,  ..., 5.5237e-03,
           3.0090e-02, 1.4740e-02]],

         [[9.5117e-01, 2.7251e-04, 2.9624e-05,  ..., 2.8610e-03,
           3.2291e-03, 8.4457e-03]],

         ...,

         [[4.5239e-01, 9.4593e-05, 1.2934e-04,  ..., 4.9248e-03,
           3.6438e-02, 2.4460e-02]],

         [[1.4270e-01, 4.0054e-03, 1.6546e-03,  ..., 1.6270e-03,
           4.7264e-03, 1.2581e-02]],

         [[6.5820e-01, 6.8784e-05, 2.1148e-04,  ..., 8.2092e-02,
           5.9937e-02, 1.6113e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8296e-01, 2.7776e-04, 1.5342e-04,  ..., 2.1118e-02,
           3.1219e-02, 2.8503e-02]],

         [[6.7090e-01, 1.1086e-05, 7.3910e-06,  ..., 7.8087e-03,
           2.4376e-03, 2.2110e-02]],

         [[2.8125e-01, 1.1146e-05, 5.6624e-06,  ..., 5.3894e-02,
           1.9226e-02, 1.7807e-02]],

         ...,

         [[6.2402e-01, 2.0325e-05, 1.9073e-06,  ..., 5.6803e-05,
           1.0443e-04, 3.7183e-01]],

         [[9.5166e-01, 1.3280e-04, 5.3227e-05,  ..., 2.9802e-07,
           1.1921e-07, 1.4023e-02]],

         [[9.9268e-01, 1.8477e-05, 2.3246e-06,  ..., 0.0000e+00,
           0.0000e+00, 3.9101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0063e-01, 1.2755e-05, 6.0201e-06,  ..., 5.2368e-02,
           2.2217e-01, 1.6006e-02]],

         [[9.5654e-01, 1.1742e-05, 5.4240e-06,  ..., 1.2696e-05,
           2.8658e-04, 1.0208e-02]],

         [[8.0420e-01, 3.3832e-04, 6.7413e-05,  ..., 2.2144e-03,
           3.6736e-03, 6.1707e-02]],

         ...,

         [[7.2217e-01, 1.9097e-04, 3.2663e-05,  ..., 7.9498e-03,
           1.2100e-02, 2.4292e-02]],

         [[4.2725e-01, 9.4891e-05, 1.6570e-05,  ..., 7.1831e-03,
           7.9651e-02, 2.4323e-02]],

         [[8.9844e-01, 1.1200e-04, 8.7202e-05,  ..., 4.0169e-03,
           9.3994e-03, 1.1902e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0625e-01, 5.2452e-06, 6.2585e-06,  ..., 6.1127e-02,
           2.9834e-01, 9.0515e-02]],

         [[7.3877e-01, 1.1396e-04, 5.1975e-05,  ..., 2.2449e-03,
           3.4981e-03, 2.9037e-02]],

         [[4.2627e-01, 1.2577e-05, 4.7088e-06,  ..., 1.2917e-02,
           7.5012e-02, 1.7532e-02]],

         ...,

         [[7.0361e-01, 1.4913e-04, 5.1558e-05,  ..., 1.7776e-02,
           2.6077e-02, 2.6779e-02]],

         [[8.6279e-01, 3.6860e-04, 1.0562e-04,  ..., 1.1854e-03,
           1.7593e-02, 3.9635e-03]],

         [[3.9307e-01, 1.3590e-05, 2.2411e-05,  ..., 5.8899e-02,
           1.0120e-01, 3.5706e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3232e-01, 2.2793e-03, 1.8320e-03,  ..., 1.7583e-04,
           3.6144e-03, 1.9817e-03]],

         [[9.5020e-01, 3.8803e-05, 6.7651e-05,  ..., 1.6003e-03,
           1.1530e-03, 3.8090e-03]],

         [[8.8232e-01, 7.0870e-05, 4.8339e-05,  ..., 1.8635e-03,
           1.9958e-02, 1.8415e-03]],

         ...,

         [[8.3447e-01, 2.8205e-04, 3.0375e-04,  ..., 1.6718e-03,
           1.0967e-03, 6.2988e-02]],

         [[9.3164e-01, 4.0531e-06, 7.5698e-06,  ..., 3.0060e-03,
           5.6076e-03, 3.0869e-02]],

         [[8.5742e-01, 4.0293e-05, 3.4213e-05,  ..., 4.9057e-03,
           6.5674e-02, 3.8185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4158e-01, 7.4387e-04, 8.2684e-04,  ..., 8.0442e-04,
           1.7471e-02, 7.7057e-03]],

         [[8.1299e-01, 8.2076e-05, 1.0455e-04,  ..., 2.8591e-03,
           2.6978e-02, 6.5956e-03]],

         [[3.0347e-01, 1.8299e-05, 4.2200e-05,  ..., 4.6844e-02,
           3.5010e-01, 8.0338e-03]],

         ...,

         [[1.9543e-01, 6.2323e-04, 1.3161e-03,  ..., 1.7428e-04,
           2.3327e-03, 7.1754e-03]],

         [[6.0791e-01, 8.8751e-05, 5.7876e-05,  ..., 1.1581e-02,
           2.7515e-01, 5.4893e-03]],

         [[8.9209e-01, 4.0531e-06, 8.6427e-06,  ..., 3.1929e-03,
           3.4199e-03, 4.1718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.9426e-05, 2.2161e-04,  ..., 2.9254e-04,
           1.0052e-03, 1.7365e-02]],

         [[4.4604e-01, 5.3227e-05, 1.0425e-04,  ..., 1.5167e-02,
           1.1066e-01, 2.1179e-02]],

         [[6.9824e-01, 1.0550e-05, 3.6538e-05,  ..., 9.0265e-04,
           2.7580e-03, 2.4979e-02]],

         ...,

         [[9.5166e-01, 4.7743e-05, 2.5332e-05,  ..., 6.4135e-04,
           2.5463e-03, 3.0861e-03]],

         [[7.4170e-01, 4.2176e-04, 1.3018e-04,  ..., 4.1504e-02,
           6.9351e-03, 2.4780e-02]],

         [[2.3315e-01, 1.5812e-03, 5.0116e-04,  ..., 1.9360e-04,
           3.3913e-03, 1.6815e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9453e-01, 3.8147e-06, 6.7353e-06,  ..., 4.3011e-04,
           9.4986e-03, 5.3955e-01]],

         [[6.1670e-01, 8.0633e-04, 1.4725e-03,  ..., 3.7050e-04,
           1.3412e-02, 1.0862e-03]],

         [[7.6074e-01, 4.6921e-04, 5.1641e-04,  ..., 1.1520e-03,
           1.5068e-02, 1.9608e-02]],

         ...,

         [[5.6885e-01, 1.2589e-04, 2.4748e-04,  ..., 2.4700e-03,
           1.2375e-02, 8.2932e-03]],

         [[8.0994e-02, 2.1040e-05, 1.2755e-05,  ..., 3.8719e-04,
           7.5226e-03, 4.0649e-02]],

         [[1.4270e-01, 2.6464e-05, 7.7844e-05,  ..., 8.5602e-03,
           9.0561e-03, 7.1436e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7810e-01, 1.1140e-04, 1.7059e-04,  ..., 7.6332e-03,
           5.4741e-03, 5.8984e-01]],

         [[5.3369e-01, 2.1195e-04, 3.6073e-04,  ..., 4.8327e-04,
           3.2806e-02, 8.9951e-03]],

         [[6.7529e-01, 1.3280e-04, 1.5402e-04,  ..., 2.0564e-04,
           2.6627e-03, 9.5901e-03]],

         ...,

         [[2.5162e-02, 2.9206e-06, 1.0967e-05,  ..., 7.3671e-04,
           3.7212e-03, 9.5898e-01]],

         [[7.6807e-01, 1.1253e-04, 1.4508e-04,  ..., 4.3907e-03,
           2.2163e-03, 9.4833e-03]],

         [[7.6855e-01, 1.4806e-04, 7.7069e-05,  ..., 7.9060e-04,
           1.9867e-02, 7.3586e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1919e-01, 5.2071e-04, 7.6914e-04,  ..., 3.9597e-03,
           2.6367e-01, 1.0849e-02]],

         [[7.2070e-01, 7.6234e-05, 1.5223e-04,  ..., 2.3346e-03,
           4.9377e-02, 2.9724e-02]],

         [[3.7671e-01, 7.2837e-05, 1.8597e-04,  ..., 2.8920e-04,
           9.2459e-04, 1.2733e-02]],

         ...,

         [[7.8271e-01, 8.9407e-07, 1.9670e-06,  ..., 5.7831e-03,
           1.6736e-01, 3.1342e-02]],

         [[5.0195e-01, 9.1887e-04, 9.4414e-04,  ..., 5.6601e-04,
           1.4656e-02, 1.3916e-02]],

         [[5.2490e-01, 4.7684e-07, 1.0729e-06,  ..., 1.8295e-02,
           3.4692e-01, 6.1951e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0180e-04, 2.3901e-05,  ..., 7.2050e-04,
           3.4218e-03, 5.2147e-03]],

         [[1.5161e-01, 6.0701e-04, 6.1512e-04,  ..., 3.3474e-04,
           5.0011e-03, 1.3130e-02]],

         [[4.5557e-01, 8.5735e-04, 8.5735e-04,  ..., 1.8466e-04,
           2.4063e-02, 1.4534e-02]],

         ...,

         [[5.0098e-01, 2.9802e-06, 3.8743e-06,  ..., 1.7490e-03,
           4.0955e-02, 5.0049e-02]],

         [[2.9858e-01, 1.4305e-06, 1.2517e-06,  ..., 4.7798e-03,
           5.7068e-02, 2.8564e-02]],

         [[1.7041e-01, 3.0041e-05, 5.2512e-05,  ..., 3.7766e-04,
           1.3268e-02, 1.1101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0171e-02, 8.3447e-07, 1.0729e-06,  ..., 1.3065e-04,
           6.6338e-03, 8.7207e-01]],

         [[3.0615e-01, 6.7353e-06, 1.9312e-05,  ..., 1.1978e-02,
           7.1777e-02, 6.5857e-02]],

         [[4.4312e-01, 1.0128e-03, 1.0872e-03,  ..., 4.3273e-04,
           9.3307e-03, 1.1795e-02]],

         ...,

         [[2.1509e-01, 4.3452e-05, 5.5313e-05,  ..., 3.9330e-03,
           8.4045e-02, 2.9602e-02]],

         [[5.8594e-01, 5.7220e-06, 1.5616e-05,  ..., 1.4782e-03,
           3.6907e-03, 1.4816e-02]],

         [[8.9722e-02, 7.8201e-05, 7.4625e-05,  ..., 7.9679e-04,
           1.2428e-02, 8.3008e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0928e-01, 4.6194e-05, 7.8380e-05,  ..., 6.5880e-03,
           1.7603e-01, 8.2321e-03]],

         [[2.2754e-01, 4.1771e-04, 4.1032e-04,  ..., 4.6692e-03,
           5.5115e-02, 4.3427e-02]],

         [[7.3181e-02, 7.7248e-05, 2.4319e-05,  ..., 5.8985e-04,
           3.4546e-02, 4.7607e-02]],

         ...,

         [[6.7627e-01, 7.8857e-05, 2.3901e-05,  ..., 1.8377e-03,
           5.8411e-02, 2.5513e-02]],

         [[3.1201e-01, 1.0449e-04, 1.3804e-04,  ..., 3.3450e-04,
           6.1920e-02, 4.4067e-02]],

         [[6.8506e-01, 2.4092e-04, 1.1784e-04,  ..., 6.5470e-04,
           2.8717e-02, 2.2812e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9854e-02, 1.4505e-03, 1.3819e-03,  ..., 6.9916e-05,
           3.7956e-03, 3.0975e-03]],

         [[1.7029e-01, 4.9925e-04, 3.0589e-04,  ..., 1.6460e-03,
           4.1901e-02, 1.2291e-02]],

         [[5.2051e-01, 2.8849e-05, 5.4836e-05,  ..., 1.0939e-03,
           1.0406e-01, 1.9775e-02]],

         ...,

         [[5.0000e-01, 3.1114e-04, 1.4615e-04,  ..., 1.9894e-03,
           9.8267e-03, 3.8422e-02]],

         [[9.6143e-01, 8.8871e-05, 1.9515e-04,  ..., 1.0252e-05,
           1.9372e-05, 3.8395e-03]],

         [[9.1357e-01, 1.1140e-04, 1.5104e-04,  ..., 5.0259e-04,
           1.6308e-03, 3.8414e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9048e-01, 5.7888e-04, 7.5483e-04,  ..., 3.5667e-04,
           1.0185e-02, 2.5192e-02]],

         [[2.0630e-01, 2.9888e-03, 1.3847e-03,  ..., 7.7677e-04,
           1.7334e-02, 1.6937e-02]],

         [[3.1204e-02, 7.2122e-06, 8.0466e-06,  ..., 6.4392e-03,
           2.0309e-02, 6.1719e-01]],

         ...,

         [[3.0566e-01, 2.5630e-05, 2.9325e-05,  ..., 5.2357e-04,
           3.3875e-02, 1.3062e-02]],

         [[5.1562e-01, 1.1504e-05, 8.4639e-06,  ..., 2.4834e-03,
           2.3242e-01, 5.7297e-03]],

         [[1.4270e-01, 2.1636e-05, 1.7762e-05,  ..., 3.2024e-03,
           7.4585e-02, 3.7012e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8135e-01, 4.6074e-05, 2.3901e-05,  ..., 1.4553e-03,
           1.0481e-03, 1.0925e-02]],

         [[6.4331e-02, 4.3631e-05, 2.6464e-05,  ..., 1.5612e-03,
           8.5266e-02, 2.2583e-02]],

         [[1.7957e-01, 4.0770e-05, 2.3246e-05,  ..., 4.0507e-04,
           1.0124e-02, 7.4414e-01]],

         ...,

         [[1.2927e-01, 4.8828e-04, 2.2447e-04,  ..., 6.5231e-04,
           3.4981e-03, 1.0941e-02]],

         [[3.5010e-01, 8.6784e-04, 5.3883e-04,  ..., 6.8817e-03,
           5.1537e-03, 7.3242e-03]],

         [[3.6841e-01, 1.1015e-03, 1.0843e-03,  ..., 1.1169e-02,
           9.8190e-03, 4.6921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4585e-01, 2.2507e-04, 9.9480e-05,  ..., 5.2643e-03,
           6.8848e-02, 3.9825e-02]],

         [[9.2578e-01, 7.8499e-05, 1.1444e-05,  ..., 2.1505e-04,
           2.2411e-03, 9.2163e-03]],

         [[8.9600e-01, 3.9876e-05, 9.6560e-06,  ..., 8.2970e-04,
           7.8125e-03, 2.6779e-03]],

         ...,

         [[8.0750e-02, 2.0218e-03, 3.4389e-03,  ..., 2.9802e-04,
           4.9248e-03, 4.8866e-03]],

         [[5.1465e-01, 3.0935e-05, 3.5048e-05,  ..., 1.8997e-03,
           5.1758e-02, 6.2790e-03]],

         [[2.0142e-01, 5.7030e-04, 1.7157e-03,  ..., 1.9217e-03,
           1.6403e-02, 3.0884e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0820e-01, 2.1040e-04, 2.2221e-04,  ..., 7.8354e-03,
           4.9400e-03, 1.2917e-02]],

         [[3.3228e-01, 2.8920e-04, 6.9904e-04,  ..., 3.1586e-03,
           1.3718e-02, 3.1891e-02]],

         [[2.9028e-01, 7.9215e-05, 1.9073e-04,  ..., 1.1986e-02,
           3.5797e-02, 1.1345e-02]],

         ...,

         [[1.3245e-02, 5.9485e-05, 9.5010e-05,  ..., 1.9388e-03,
           1.5244e-02, 2.2888e-02]],

         [[1.3245e-01, 1.5998e-04, 1.0008e-04,  ..., 6.2275e-04,
           1.9178e-03, 2.3880e-03]],

         [[6.4893e-01, 2.4104e-04, 3.9124e-04,  ..., 2.7580e-03,
           4.5815e-03, 2.6951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1542e-01, 1.7881e-05, 1.7107e-05,  ..., 2.0981e-03,
           1.6251e-02, 9.1858e-03]],

         [[2.6440e-01, 1.0127e-04, 1.8120e-04,  ..., 5.5084e-03,
           1.8127e-02, 6.3896e-03]],

         [[2.0337e-01, 4.7088e-05, 1.5676e-04,  ..., 6.9046e-03,
           1.8051e-02, 6.2158e-01]],

         ...,

         [[7.0361e-01, 3.5465e-05, 6.5267e-05,  ..., 4.1351e-03,
           3.9703e-02, 2.7924e-02]],

         [[3.9160e-01, 2.5868e-05, 1.3876e-04,  ..., 3.7193e-03,
           7.6332e-03, 2.5620e-02]],

         [[4.8657e-01, 8.4341e-05, 9.3699e-05,  ..., 5.3930e-04,
           1.9121e-03, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3684e-01, 1.6236e-04, 8.1360e-05,  ..., 3.2425e-03,
           2.3224e-02, 1.5228e-02]],

         [[6.1340e-02, 2.4915e-05, 2.6405e-05,  ..., 8.4991e-03,
           1.3391e-01, 1.6373e-02]],

         [[4.9756e-01, 1.5318e-04, 1.3280e-04,  ..., 1.5955e-03,
           5.4207e-03, 1.3893e-02]],

         ...,

         [[2.6636e-01, 4.4703e-06, 2.9266e-05,  ..., 1.7303e-02,
           6.3916e-01, 2.8290e-02]],

         [[4.0601e-01, 3.3498e-05, 7.9751e-05,  ..., 3.9787e-03,
           3.5187e-02, 4.3457e-02]],

         [[4.9951e-01, 2.2352e-05, 7.5817e-05,  ..., 3.5267e-03,
           3.3276e-01, 7.3059e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6865e-02, 1.1683e-05, 5.7042e-05,  ..., 2.0542e-03,
           3.8025e-02, 2.7817e-02]],

         [[1.1864e-02, 4.7684e-06, 9.8944e-06,  ..., 2.6631e-04,
           6.3477e-03, 9.6484e-01]],

         [[2.6520e-02, 2.2829e-05, 1.1903e-04,  ..., 1.6046e-04,
           1.2407e-03, 2.7618e-03]],

         ...,

         [[9.7595e-02, 2.5415e-04, 9.0599e-04,  ..., 6.9332e-04,
           1.5869e-02, 1.1566e-02]],

         [[4.4727e-01, 3.9935e-06, 9.7156e-06,  ..., 8.5020e-04,
           1.3123e-01, 4.0802e-02]],

         [[2.5488e-01, 7.7486e-07, 3.6955e-06,  ..., 2.0432e-02,
           3.4961e-01, 1.4294e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9395e-01, 2.3758e-04, 1.1396e-03,  ..., 5.2452e-04,
           8.0032e-03, 6.7635e-03]],

         [[9.8486e-01, 3.1948e-05, 1.9491e-04,  ..., 1.8716e-05,
           3.3689e-04, 1.0376e-03]],

         [[8.1934e-01, 1.8883e-04, 2.1994e-04,  ..., 3.1815e-03,
           1.1871e-02, 6.1569e-03]],

         ...,

         [[6.5332e-01, 6.1333e-05, 6.8426e-05,  ..., 1.8814e-02,
           5.7983e-02, 1.1063e-02]],

         [[4.8413e-01, 1.2517e-06, 2.9802e-07,  ..., 9.4175e-04,
           1.4820e-03, 4.8981e-03]],

         [[1.9873e-01, 2.5272e-05, 5.7459e-05,  ..., 2.3997e-04,
           4.5044e-02, 6.6162e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2074e-02, 9.0003e-06, 1.2636e-05,  ..., 3.9368e-03,
           1.7334e-01, 8.8501e-02]],

         [[6.1914e-01, 1.1981e-04, 6.1274e-05,  ..., 1.2848e-02,
           7.0557e-02, 2.8503e-02]],

         [[1.8530e-01, 5.5122e-04, 3.5477e-04,  ..., 2.3743e-02,
           6.0638e-02, 1.8280e-02]],

         ...,

         [[4.3701e-01, 2.5630e-05, 1.0747e-04,  ..., 3.0212e-02,
           2.0972e-01, 1.6953e-02]],

         [[6.4551e-01, 2.3007e-05, 7.4208e-05,  ..., 5.6915e-03,
           2.9251e-02, 6.2408e-02]],

         [[3.9014e-01, 1.8835e-05, 6.1929e-05,  ..., 5.9204e-03,
           2.6749e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7863e-03, 1.9670e-06, 1.6093e-06,  ..., 5.3167e-04,
           2.5768e-03, 4.4861e-03]],

         [[8.2520e-02, 8.4639e-06, 4.4107e-06,  ..., 6.6566e-04,
           5.2376e-03, 5.8861e-03]],

         [[1.8958e-01, 9.5308e-05, 3.0708e-04,  ..., 1.7227e-02,
           1.2537e-01, 3.4821e-02]],

         ...,

         [[8.3374e-02, 1.1683e-05, 1.4198e-04,  ..., 1.7996e-03,
           1.0277e-02, 6.4941e-02]],

         [[2.0288e-01, 3.9995e-05, 3.4571e-04,  ..., 1.5488e-03,
           1.7456e-02, 5.4626e-02]],

         [[7.9199e-01, 2.4915e-05, 7.7367e-05,  ..., 4.8599e-03,
           3.2196e-02, 3.2196e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4316e-01, 1.3769e-05, 2.9385e-05,  ..., 1.0681e-02,
           3.2074e-02, 4.1840e-02]],

         [[3.2617e-01, 3.7551e-05, 1.5211e-04,  ..., 1.8509e-02,
           1.7468e-01, 5.4321e-02]],

         [[1.5942e-01, 2.8610e-06, 1.1146e-05,  ..., 2.7370e-03,
           6.6357e-01, 1.0956e-01]],

         ...,

         [[4.5703e-01, 7.6950e-05, 1.4067e-04,  ..., 5.7755e-03,
           1.1436e-02, 1.0132e-02]],

         [[2.9590e-01, 2.5809e-05, 1.5795e-05,  ..., 1.1444e-03,
           5.2216e-02, 4.3304e-02]],

         [[5.9021e-02, 1.0747e-04, 4.1842e-04,  ..., 1.5884e-02,
           6.4819e-02, 5.9619e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4546e-01, 7.5340e-05, 1.7405e-04,  ..., 1.7136e-02,
           1.0870e-01, 1.4259e-02]],

         [[3.3496e-01, 3.9530e-04, 4.8256e-04,  ..., 4.1847e-03,
           6.1836e-03, 3.1109e-03]],

         [[2.0679e-01, 8.7023e-05, 1.8466e-04,  ..., 3.2654e-03,
           4.6616e-03, 9.4910e-03]],

         ...,

         [[7.0996e-01, 4.5896e-04, 4.6635e-04,  ..., 6.5880e-03,
           2.0782e-02, 3.9429e-02]],

         [[1.4392e-01, 1.6844e-04, 4.9496e-04,  ..., 3.0327e-03,
           2.2064e-02, 7.5391e-01]],

         [[8.8184e-01, 2.4259e-05, 4.9829e-05,  ..., 1.3390e-02,
           6.1302e-03, 2.4628e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4766e-01, 1.0097e-04, 1.1864e-03,  ..., 1.7792e-02,
           1.3863e-02, 7.3700e-03]],

         [[4.8645e-02, 4.6492e-06, 2.7835e-05,  ..., 9.5797e-04,
           5.5786e-02, 1.8463e-02]],

         [[1.5686e-01, 8.0884e-05, 3.6645e-04,  ..., 2.8122e-02,
           7.1838e-02, 1.3904e-01]],

         ...,

         [[6.9922e-01, 1.6582e-04, 1.0080e-03,  ..., 4.5166e-03,
           3.2959e-02, 4.0863e-02]],

         [[8.0518e-01, 1.7881e-06, 3.2544e-04,  ..., 2.5272e-03,
           6.6261e-03, 2.7409e-03]],

         [[2.7588e-01, 1.1235e-04, 1.0341e-04,  ..., 4.3121e-02,
           1.2189e-01, 4.4312e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8394e-05, 1.2207e-04, 1.5515e-01,  ..., 5.0850e-03,
           7.7200e-04, 4.1748e-02]],

         [[2.9862e-05, 2.2411e-05, 4.1595e-02,  ..., 6.4945e-04,
           1.4019e-03, 8.3542e-03]],

         [[5.8222e-04, 8.4305e-04, 8.8013e-02,  ..., 3.3417e-03,
           1.1147e-02, 1.8738e-02]],

         ...,

         [[4.6206e-04, 1.6224e-04, 2.1545e-01,  ..., 1.0315e-02,
           4.3762e-02, 1.4465e-01]],

         [[2.7132e-04, 2.9278e-04, 3.2861e-01,  ..., 9.1934e-04,
           5.3749e-03, 8.3923e-03]],

         [[1.4782e-05, 1.5616e-05, 1.4355e-01,  ..., 6.6996e-04,
           5.4359e-03, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0969e-04, 1.3959e-04, 2.8052e-01,  ..., 1.1873e-03,
           2.7618e-03, 1.5343e-02]],

         [[1.3328e-04, 1.3196e-04, 1.2646e-01,  ..., 1.8616e-03,
           1.2302e-03, 1.0139e-02]],

         [[9.5546e-05, 2.7180e-05, 2.8418e-01,  ..., 1.7920e-03,
           4.0245e-03, 2.2263e-02]],

         ...,

         [[4.1771e-04, 6.7532e-05, 3.7018e-02,  ..., 2.2447e-04,
           6.0797e-04, 1.6136e-03]],

         [[1.9312e-05, 2.2113e-05, 2.2534e-01,  ..., 3.0670e-03,
           2.0957e-04, 8.9798e-03]],

         [[1.0610e-04, 1.3387e-04, 2.1741e-01,  ..., 2.0256e-03,
           5.2643e-04, 1.6769e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0637e-05, 3.2425e-05, 8.7585e-02,  ..., 8.1825e-04,
           8.0776e-04, 3.0575e-03]],

         [[2.5201e-04, 1.1802e-05, 2.0142e-01,  ..., 1.8396e-03,
           8.5711e-05, 1.1360e-02]],

         [[1.0622e-04, 1.2219e-04, 1.9727e-01,  ..., 9.7656e-04,
           1.0729e-03, 7.4158e-03]],

         ...,

         [[1.5414e-04, 7.8619e-05, 1.3635e-01,  ..., 5.5618e-03,
           3.8528e-03, 2.3499e-02]],

         [[1.9073e-05, 4.5121e-05, 7.1045e-02,  ..., 3.8767e-04,
           5.0449e-04, 4.5242e-03]],

         [[6.9559e-05, 2.9445e-05, 1.1188e-01,  ..., 7.3910e-04,
           6.5994e-04, 4.8409e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9605e-05, 5.0068e-06, 1.4328e-02,  ..., 2.3479e-03,
           1.1011e-01, 2.0093e-01]],

         [[3.0518e-05, 8.3923e-05, 1.5942e-01,  ..., 5.7030e-03,
           7.5256e-02, 3.5828e-02]],

         [[1.7166e-04, 5.4932e-04, 2.8955e-01,  ..., 6.0177e-04,
           8.3733e-04, 5.8699e-04]],

         ...,

         [[3.4142e-04, 4.9293e-05, 6.7871e-02,  ..., 4.0352e-05,
           2.4338e-03, 9.0714e-03]],

         [[4.0665e-03, 2.0480e-04, 1.2030e-01,  ..., 2.6684e-03,
           4.6204e-02, 2.5171e-01]],

         [[7.2122e-06, 1.3113e-06, 1.5020e-03,  ..., 1.6463e-04,
           8.5327e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8321e-03, 1.4305e-03, 1.2891e-01,  ..., 1.4130e-02,
           1.7578e-02, 7.2876e-02]],

         [[2.3098e-03, 1.4591e-03, 2.4158e-01,  ..., 7.2098e-03,
           1.3161e-02, 2.7008e-02]],

         [[8.6288e-03, 1.1053e-03, 2.2351e-01,  ..., 5.2223e-03,
           7.4806e-03, 1.1208e-02]],

         ...,

         [[1.6441e-03, 9.9123e-05, 2.3840e-01,  ..., 1.5812e-03,
           6.3515e-03, 1.4453e-01]],

         [[2.3174e-03, 2.9526e-03, 1.9751e-01,  ..., 4.2389e-02,
           5.7373e-03, 5.5298e-02]],

         [[1.5950e-04, 7.7426e-05, 3.7231e-02,  ..., 7.5645e-03,
           2.5439e-01, 3.8501e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.7001e-05, 6.4790e-05, 1.2608e-03,  ..., 8.8379e-02,
           4.7803e-01, 3.2837e-01]],

         [[2.2125e-03, 1.1835e-03, 1.7395e-02,  ..., 3.8815e-03,
           6.8787e-02, 4.2139e-01]],

         [[9.5367e-07, 1.1921e-06, 2.2087e-03,  ..., 8.5592e-05,
           5.6946e-02, 8.9062e-01]],

         ...,

         [[1.7881e-07, 8.3447e-07, 7.4267e-05,  ..., 1.2243e-04,
           2.4586e-03, 9.9170e-01]],

         [[2.7199e-03, 2.3994e-03, 1.4702e-02,  ..., 6.1302e-03,
           2.4246e-02, 2.0300e-01]],

         [[1.1749e-02, 4.3221e-03, 9.6464e-04,  ..., 3.1952e-02,
           2.6489e-02, 2.8198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0118e-03, 4.6082e-03, 1.6327e-02,  ..., 7.3181e-02,
           1.0651e-01, 1.4551e-01]],

         [[1.7881e-07, 3.5763e-07, 4.1723e-07,  ..., 1.7346e-01,
           7.4219e-01, 7.8186e-02]],

         [[2.5635e-03, 4.7150e-03, 1.7517e-02,  ..., 2.5879e-02,
           4.2664e-02, 1.9852e-02]],

         ...,

         [[1.2808e-03, 6.8521e-04, 6.8130e-03,  ..., 7.7393e-02,
           1.2634e-02, 7.3242e-02]],

         [[4.8714e-03, 1.4091e-02, 4.4617e-02,  ..., 1.2581e-02,
           3.6560e-02, 1.0933e-02]],

         [[1.8797e-03, 1.2070e-02, 2.8732e-02,  ..., 5.4932e-02,
           6.7810e-02, 3.0106e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2490e-02, 6.6757e-03, 1.5396e-02,  ..., 6.6895e-02,
           4.5959e-02, 4.3518e-02]],

         [[3.8414e-03, 8.4352e-04, 3.3245e-03,  ..., 4.8370e-03,
           8.6899e-03, 3.0270e-03]],

         [[4.9629e-03, 1.3208e-03, 1.2264e-03,  ..., 2.0264e-02,
           3.7994e-02, 2.2949e-02]],

         ...,

         [[2.9812e-03, 1.4639e-03, 2.3575e-03,  ..., 3.5254e-01,
           1.0754e-01, 1.5405e-01]],

         [[3.0689e-03, 5.2929e-04, 3.3236e-04,  ..., 2.8711e-01,
           1.2952e-01, 5.2307e-02]],

         [[2.6566e-02, 6.7711e-03, 1.7975e-02,  ..., 2.6566e-02,
           3.8135e-01, 9.4910e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6035e-01, 1.7285e-06, 1.1921e-06,  ..., 2.1915e-03,
           2.1458e-03, 1.0811e-02]],

         [[9.3359e-01, 4.8876e-06, 1.1086e-05,  ..., 8.8882e-03,
           1.2314e-02, 4.8208e-04]],

         [[8.8232e-01, 1.1921e-06, 4.7684e-06,  ..., 6.9504e-03,
           1.8280e-02, 1.9165e-02]],

         ...,

         [[6.3379e-01, 8.7798e-05, 8.4114e-04,  ..., 5.1618e-05,
           2.6684e-03, 2.4200e-02]],

         [[5.0342e-01, 1.3387e-04, 3.1519e-04,  ..., 2.8061e-02,
           2.4002e-02, 1.0948e-02]],

         [[6.2158e-01, 3.1114e-04, 4.3416e-04,  ..., 2.0050e-02,
           4.9622e-02, 2.2278e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6182e-01, 4.4703e-06, 1.7881e-06,  ..., 7.7200e-04,
           2.3499e-03, 3.6652e-02]],

         [[1.0907e-01, 5.3644e-07, 1.7881e-07,  ..., 2.5597e-03,
           8.7842e-01, 9.5596e-03]],

         [[9.4824e-01, 1.0818e-04, 5.7936e-05,  ..., 1.2007e-03,
           1.5656e-02, 4.8294e-03]],

         ...,

         [[4.4360e-01, 2.5415e-04, 1.2094e-04,  ..., 5.8624e-02,
           5.4016e-02, 3.1311e-02]],

         [[1.3232e-01, 9.4175e-04, 6.2084e-04,  ..., 5.9929e-03,
           1.7319e-02, 7.3738e-03]],

         [[4.4824e-01, 8.7202e-05, 1.4317e-04,  ..., 1.1877e-01,
           6.4819e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1348e-01, 3.6001e-04, 1.5426e-04,  ..., 4.5898e-02,
           5.9296e-02, 3.0640e-02]],

         [[6.4502e-01, 1.9372e-05, 1.2517e-05,  ..., 1.2398e-02,
           1.2891e-01, 1.8326e-02]],

         [[5.4980e-01, 1.7405e-05, 4.3511e-06,  ..., 1.7868e-02,
           6.3721e-02, 1.2474e-02]],

         ...,

         [[6.1670e-01, 1.7881e-06, 4.7684e-07,  ..., 6.2585e-06,
           5.4779e-03, 3.7402e-01]],

         [[9.2383e-01, 1.9789e-05, 2.9981e-05,  ..., 0.0000e+00,
           1.3483e-04, 4.3243e-02]],

         [[9.8389e-01, 2.3842e-06, 7.1526e-07,  ..., 0.0000e+00,
           2.7180e-05, 9.4223e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 1.9431e-05, 2.4438e-06,  ..., 1.2018e-01,
           1.3232e-01, 1.0727e-02]],

         [[9.5947e-01, 1.0967e-05, 2.6822e-06,  ..., 3.1614e-04,
           1.5564e-03, 1.1902e-02]],

         [[7.7197e-01, 5.7459e-04, 6.1989e-05,  ..., 9.2888e-04,
           2.3941e-02, 5.9906e-02]],

         ...,

         [[8.0273e-01, 4.7505e-05, 1.1981e-05,  ..., 6.9962e-03,
           1.8890e-02, 1.8280e-02]],

         [[7.9932e-01, 1.5581e-04, 4.9233e-05,  ..., 3.6865e-02,
           2.7878e-02, 3.1647e-02]],

         [[9.2285e-01, 6.2644e-05, 4.0948e-05,  ..., 6.5727e-03,
           9.7809e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 6.1393e-06, 6.9737e-06,  ..., 2.6221e-01,
           3.3417e-02, 3.4821e-02]],

         [[7.6025e-01, 2.9564e-05, 9.3997e-05,  ..., 5.6725e-03,
           1.8433e-02, 5.0507e-02]],

         [[9.3652e-01, 3.0160e-05, 2.5034e-06,  ..., 1.1932e-02,
           6.6795e-03, 1.7529e-03]],

         ...,

         [[6.7920e-01, 1.9467e-04, 8.2731e-05,  ..., 1.1627e-02,
           2.7252e-02, 1.9485e-02]],

         [[8.7793e-01, 1.4353e-04, 3.6597e-05,  ..., 5.9624e-03,
           9.1629e-03, 4.3602e-03]],

         [[7.5635e-01, 3.4571e-05, 1.7524e-05,  ..., 2.0599e-02,
           6.6101e-02, 2.8351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6055e-01, 1.7023e-03, 8.2970e-04,  ..., 2.9411e-03,
           1.9445e-03, 3.4523e-03]],

         [[9.1064e-01, 8.7082e-05, 1.2863e-04,  ..., 9.7752e-04,
           7.9956e-03, 9.7961e-03]],

         [[8.7842e-01, 6.3598e-05, 2.9564e-05,  ..., 4.7874e-03,
           7.5436e-04, 2.4338e-03]],

         ...,

         [[5.2295e-01, 3.8505e-05, 1.3447e-04,  ..., 1.9188e-03,
           3.4607e-02, 3.0249e-01]],

         [[5.4102e-01, 4.7684e-06, 6.3181e-06,  ..., 3.3665e-03,
           2.9224e-01, 1.0883e-01]],

         [[9.0576e-01, 2.7418e-05, 2.1696e-05,  ..., 2.7893e-02,
           5.4398e-03, 1.0880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5405e-01, 6.1035e-04, 1.1740e-03,  ..., 1.6129e-02,
           1.6174e-02, 3.7251e-03]],

         [[8.8818e-01, 4.5359e-05, 8.0884e-05,  ..., 7.3662e-03,
           3.7994e-02, 1.5358e-02]],

         [[5.8740e-01, 4.2796e-05, 6.1810e-05,  ..., 1.5955e-01,
           1.9180e-02, 3.2257e-02]],

         ...,

         [[3.4180e-01, 6.5756e-04, 7.1764e-04,  ..., 2.9240e-03,
           1.0925e-02, 1.6832e-03]],

         [[7.6172e-01, 7.1943e-05, 4.4048e-05,  ..., 1.2457e-01,
           7.4539e-03, 6.5346e-03]],

         [[1.6016e-01, 1.7285e-06, 6.5565e-06,  ..., 2.1992e-03,
           7.3486e-01, 8.5754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9824e-01, 6.3598e-05, 1.1116e-04,  ..., 1.4663e-04,
           7.8964e-04, 1.5961e-02]],

         [[5.5908e-01, 9.8348e-05, 9.3460e-05,  ..., 7.6416e-02,
           7.7881e-02, 1.6861e-02]],

         [[7.5342e-01, 4.5300e-06, 9.1791e-06,  ..., 1.6556e-03,
           3.8357e-03, 1.6022e-02]],

         ...,

         [[9.4922e-01, 4.9889e-05, 2.0802e-05,  ..., 2.7180e-03,
           4.6005e-03, 1.6909e-03]],

         [[3.8745e-01, 2.1100e-04, 6.7949e-05,  ..., 5.8441e-03,
           5.5962e-03, 5.1709e-01]],

         [[2.4194e-01, 3.5524e-04, 1.8680e-04,  ..., 2.3773e-02,
           4.4617e-02, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6182e-01, 1.7285e-06, 5.4240e-06,  ..., 6.6710e-04,
           7.2365e-03, 6.0840e-01]],

         [[6.2744e-01, 5.0688e-04, 9.1457e-04,  ..., 7.1945e-03,
           8.8215e-04, 1.7529e-03]],

         [[6.4502e-01, 6.0272e-04, 6.9714e-04,  ..., 1.6602e-02,
           3.6957e-02, 1.6571e-02]],

         ...,

         [[6.0498e-01, 1.3793e-04, 2.3556e-04,  ..., 3.8834e-03,
           3.0365e-03, 5.3482e-03]],

         [[2.2571e-01, 9.7215e-05, 5.9187e-05,  ..., 4.3259e-03,
           7.6904e-02, 2.5192e-02]],

         [[2.1545e-01, 3.0696e-05, 6.8605e-05,  ..., 3.2616e-03,
           6.0516e-02, 6.5918e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 5.8413e-05, 9.1851e-05,  ..., 3.7136e-03,
           7.6904e-02, 3.7720e-01]],

         [[3.3252e-01, 4.8161e-04, 4.8542e-04,  ..., 2.9282e-02,
           1.8542e-01, 4.1107e-02]],

         [[4.2432e-01, 2.1553e-04, 1.5771e-04,  ..., 2.1191e-03,
           4.4403e-03, 3.6526e-03]],

         ...,

         [[2.8488e-02, 1.3113e-06, 1.7285e-06,  ..., 8.5175e-05,
           5.1308e-03, 9.6191e-01]],

         [[8.4961e-01, 5.5671e-05, 4.8757e-05,  ..., 1.2856e-03,
           4.1733e-03, 2.8763e-02]],

         [[6.9141e-01, 2.3329e-04, 7.6056e-05,  ..., 1.1208e-02,
           4.7836e-03, 6.4659e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3516e-01, 4.3631e-04, 5.4741e-04,  ..., 8.8684e-02,
           1.3275e-02, 9.2010e-03]],

         [[7.7979e-01, 4.3869e-05, 1.6499e-04,  ..., 2.2858e-02,
           5.0812e-02, 3.9154e-02]],

         [[4.1699e-01, 1.6260e-04, 1.3793e-04,  ..., 1.2636e-03,
           4.6501e-03, 1.6068e-02]],

         ...,

         [[2.8027e-01, 5.9605e-08, 8.9407e-07,  ..., 4.0985e-02,
           6.4453e-01, 2.7740e-02]],

         [[3.8745e-01, 5.5361e-04, 7.1335e-04,  ..., 1.7059e-02,
           1.1269e-02, 4.8561e-03]],

         [[7.6514e-01, 1.3709e-06, 2.1458e-06,  ..., 5.0507e-02,
           9.0942e-02, 2.3468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1484e-01, 6.7890e-05, 1.9014e-05,  ..., 3.7365e-03,
           3.0708e-03, 7.8735e-03]],

         [[1.1322e-01, 4.3678e-04, 4.7469e-04,  ..., 6.2370e-03,
           1.7380e-02, 4.1428e-03]],

         [[3.9697e-01, 2.5558e-04, 2.5606e-04,  ..., 6.4819e-02,
           1.0394e-01, 1.5762e-02]],

         ...,

         [[3.2544e-01, 1.6093e-06, 1.4901e-06,  ..., 5.0049e-02,
           4.4873e-01, 1.7365e-02]],

         [[7.7734e-01, 1.7881e-07, 1.1921e-07,  ..., 2.0676e-02,
           8.9539e-02, 1.6113e-02]],

         [[4.0771e-01, 1.6332e-05, 3.4690e-05,  ..., 1.8295e-02,
           6.2378e-02, 7.2708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9773e-02, 4.1723e-07, 5.9605e-07,  ..., 8.4448e-04,
           7.0915e-03, 8.8818e-01]],

         [[1.7615e-01, 8.4043e-06, 2.0444e-05,  ..., 8.6670e-02,
           2.6245e-01, 3.5919e-02]],

         [[3.4058e-01, 1.3103e-03, 1.1044e-03,  ..., 2.2278e-02,
           1.9119e-02, 8.5678e-03]],

         ...,

         [[2.8198e-01, 5.0306e-05, 2.7180e-05,  ..., 8.5144e-02,
           2.1469e-02, 1.3817e-02]],

         [[6.1230e-01, 5.5432e-06, 8.1062e-06,  ..., 4.2267e-03,
           6.6338e-03, 1.3954e-02]],

         [[9.5459e-02, 8.5056e-05, 4.2081e-05,  ..., 5.3358e-04,
           3.5739e-04, 2.0027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0352e-01, 1.9133e-04, 8.8990e-05,  ..., 3.3997e-02,
           7.9498e-03, 3.3722e-02]],

         [[3.3008e-01, 4.7565e-04, 1.4567e-04,  ..., 7.6904e-02,
           3.1555e-02, 2.9770e-02]],

         [[8.6441e-03, 2.2173e-05, 2.5034e-06,  ..., 1.2238e-02,
           2.7252e-02, 3.5877e-03]],

         ...,

         [[7.0117e-01, 5.8830e-05, 2.2709e-05,  ..., 5.6915e-02,
           1.9287e-02, 1.0239e-02]],

         [[2.6416e-01, 2.1183e-04, 1.0926e-04,  ..., 1.3184e-01,
           7.0312e-02, 4.6631e-02]],

         [[8.9893e-01, 2.4176e-04, 6.8486e-05,  ..., 7.3357e-03,
           3.0384e-03, 9.0179e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3342e-01, 1.0586e-03, 1.7157e-03,  ..., 5.5161e-03,
           2.9755e-03, 3.0875e-04]],

         [[2.6489e-01, 4.4131e-04, 2.3580e-04,  ..., 2.8702e-02,
           7.1220e-03, 8.4610e-03]],

         [[2.7002e-01, 4.1127e-05, 6.1393e-05,  ..., 1.0162e-01,
           1.4502e-01, 5.9998e-02]],

         ...,

         [[7.3730e-01, 2.1100e-04, 1.0806e-04,  ..., 3.0632e-03,
           1.6479e-02, 2.5681e-02]],

         [[9.7461e-01, 1.7107e-05, 2.4676e-05,  ..., 3.3975e-06,
           4.6372e-05, 2.7084e-03]],

         [[7.8223e-01, 2.6870e-04, 3.2043e-04,  ..., 5.5046e-03,
           1.0902e-02, 4.9362e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0358e-01, 1.1039e-04, 8.6010e-05,  ..., 7.7400e-03,
           6.1829e-02, 8.2397e-03]],

         [[1.7688e-01, 4.3602e-03, 2.1076e-03,  ..., 1.8646e-02,
           1.6586e-02, 1.1757e-02]],

         [[6.2622e-02, 5.4240e-06, 6.1393e-06,  ..., 2.1637e-02,
           2.3071e-01, 5.2393e-01]],

         ...,

         [[2.6636e-01, 2.5094e-05, 1.9431e-05,  ..., 6.7566e-02,
           4.9438e-02, 5.7030e-03]],

         [[7.1777e-01, 2.5332e-05, 1.8597e-05,  ..., 5.4504e-02,
           1.4328e-02, 1.3046e-02]],

         [[2.4475e-01, 1.5104e-04, 1.1086e-04,  ..., 1.8143e-02,
           1.8323e-01, 2.6465e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3545e-01, 4.6432e-05, 2.1935e-05,  ..., 1.0910e-03,
           4.1161e-03, 3.0426e-02]],

         [[5.6274e-02, 3.9339e-05, 2.4617e-05,  ..., 2.6025e-01,
           1.0681e-01, 2.6169e-02]],

         [[1.9421e-01, 1.6212e-05, 1.8060e-05,  ..., 1.0509e-03,
           4.6112e-02, 7.3291e-01]],

         ...,

         [[1.5320e-01, 7.3195e-04, 2.1219e-04,  ..., 6.2370e-04,
           1.8549e-03, 8.1863e-03]],

         [[3.7964e-01, 4.4298e-04, 1.8978e-04,  ..., 4.1046e-03,
           5.4359e-03, 1.2016e-02]],

         [[3.9819e-01, 1.1635e-03, 1.2531e-03,  ..., 1.0910e-02,
           7.1526e-03, 1.2360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8933e-01, 1.2189e-04, 6.7592e-05,  ..., 8.5327e-02,
           5.6396e-02, 1.9333e-02]],

         [[9.6533e-01, 5.4836e-06, 5.9605e-07,  ..., 2.8191e-03,
           9.6560e-06, 1.2226e-03]],

         [[8.4961e-01, 4.9353e-05, 6.6757e-06,  ..., 1.4618e-02,
           1.7023e-04, 1.7872e-03]],

         ...,

         [[1.0632e-01, 1.9627e-03, 2.3670e-03,  ..., 7.0114e-03,
           5.3749e-03, 2.0237e-03]],

         [[6.4258e-01, 5.8889e-05, 6.5207e-05,  ..., 3.7384e-02,
           7.7171e-03, 1.4870e-02]],

         [[2.9614e-01, 1.1644e-03, 2.4452e-03,  ..., 9.7504e-03,
           2.2308e-02, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6450e-01, 1.8060e-04, 2.3377e-04,  ..., 1.0178e-02,
           2.9678e-02, 4.8920e-02]],

         [[2.7222e-01, 2.6417e-04, 5.3358e-04,  ..., 1.4771e-02,
           2.0996e-02, 2.5314e-02]],

         [[2.7490e-01, 7.6711e-05, 1.0693e-04,  ..., 1.7563e-02,
           5.3177e-03, 7.3242e-03]],

         ...,

         [[3.5889e-02, 2.2030e-04, 2.9635e-04,  ..., 3.2196e-02,
           2.5452e-02, 1.8341e-02]],

         [[5.5725e-02, 1.9479e-04, 9.9480e-05,  ..., 7.5817e-04,
           3.8743e-04, 8.0261e-03]],

         [[5.5908e-01, 3.5048e-04, 4.3273e-04,  ..., 3.4294e-03,
           2.7771e-03, 9.4681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4661e-01, 1.3292e-05, 1.1444e-05,  ..., 7.8278e-03,
           3.3150e-03, 8.8043e-03]],

         [[3.5034e-01, 1.7512e-04, 2.8205e-04,  ..., 1.9928e-02,
           8.0185e-03, 8.1787e-03]],

         [[2.5073e-01, 5.1022e-05, 1.1951e-04,  ..., 1.1818e-02,
           1.6443e-01, 3.8525e-01]],

         ...,

         [[5.2588e-01, 7.4446e-05, 1.2517e-04,  ..., 6.5857e-02,
           5.1666e-02, 3.6072e-02]],

         [[7.1777e-01, 3.0458e-05, 2.6226e-04,  ..., 2.2144e-03,
           1.1330e-02, 1.9882e-02]],

         [[2.4927e-01, 3.9935e-05, 8.2970e-05,  ..., 1.6537e-03,
           8.6594e-03, 3.2959e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0813e-01, 2.1040e-04, 1.7929e-04,  ..., 1.3992e-02,
           1.2947e-02, 2.9053e-02]],

         [[1.3623e-01, 1.7047e-04, 2.4748e-04,  ..., 7.5867e-02,
           2.8122e-02, 1.4702e-02]],

         [[4.3408e-01, 1.7595e-04, 1.3316e-04,  ..., 5.4207e-03,
           1.7166e-02, 6.4514e-02]],

         ...,

         [[4.6533e-01, 8.3447e-07, 1.4067e-05,  ..., 1.5588e-01,
           3.2983e-01, 2.6245e-02]],

         [[3.7817e-01, 4.1842e-05, 6.9261e-05,  ..., 1.1444e-01,
           1.7310e-01, 3.8330e-02]],

         [[3.5059e-01, 7.7486e-07, 9.6560e-06,  ..., 9.2224e-02,
           4.8682e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9520e-02, 7.9870e-06, 3.1590e-05,  ..., 4.4434e-02,
           5.0751e-02, 1.0803e-02]],

         [[1.6281e-02, 1.0848e-05, 1.4544e-05,  ..., 6.2132e-04,
           2.1912e-02, 9.3164e-01]],

         [[3.5583e-02, 4.1962e-05, 1.4591e-04,  ..., 3.0251e-03,
           6.8169e-03, 3.6049e-03]],

         ...,

         [[1.7627e-01, 3.1638e-04, 1.2674e-03,  ..., 1.8295e-02,
           3.4058e-02, 8.9874e-03]],

         [[2.9932e-01, 6.4969e-06, 8.1062e-06,  ..., 1.1768e-01,
           2.4524e-01, 3.8361e-02]],

         [[2.9395e-01, 1.1921e-07, 8.9407e-07,  ..., 1.0565e-01,
           4.6240e-01, 5.7892e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1055e-01, 1.9360e-04, 8.8835e-04,  ..., 7.5531e-03,
           5.7678e-03, 1.0529e-02]],

         [[9.8438e-01, 2.1040e-05, 6.2168e-05,  ..., 2.7704e-04,
           3.2640e-04, 8.8024e-04]],

         [[9.0186e-01, 1.7035e-04, 8.7857e-05,  ..., 3.4084e-03,
           1.1778e-03, 1.3351e-03]],

         ...,

         [[7.7881e-01, 3.9160e-05, 3.7909e-05,  ..., 3.4760e-02,
           8.2550e-03, 1.1826e-02]],

         [[3.5498e-01, 1.4901e-06, 4.7684e-07,  ..., 1.6308e-03,
           2.8629e-03, 1.0887e-02]],

         [[1.4478e-01, 1.1981e-05, 1.7881e-05,  ..., 5.8365e-03,
           5.1605e-02, 7.2363e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9669e-02, 7.5102e-06, 1.0729e-05,  ..., 1.6333e-01,
           1.8359e-01, 8.2764e-02]],

         [[7.0703e-01, 6.2168e-05, 4.1962e-05,  ..., 3.7323e-02,
           2.9770e-02, 4.1016e-02]],

         [[1.8665e-01, 7.8106e-04, 5.4312e-04,  ..., 4.3488e-02,
           2.2293e-02, 5.7793e-03]],

         ...,

         [[5.9814e-01, 1.6749e-05, 7.1406e-05,  ..., 1.2250e-01,
           2.5085e-02, 4.6196e-03]],

         [[6.3379e-01, 1.6809e-05, 4.8757e-05,  ..., 1.9287e-02,
           2.9648e-02, 2.9877e-02]],

         [[2.8320e-01, 7.5698e-06, 3.4392e-05,  ..., 1.8524e-02,
           1.0828e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5053e-02, 3.8147e-06, 4.7684e-06,  ..., 2.2717e-03,
           7.0534e-03, 1.0506e-02]],

         [[1.5710e-01, 1.6630e-05, 8.9407e-06,  ..., 7.8201e-03,
           8.9340e-03, 5.3436e-02]],

         [[2.3853e-01, 1.3375e-04, 4.1103e-04,  ..., 6.5186e-02,
           2.8488e-02, 7.2144e-02]],

         ...,

         [[6.1066e-02, 3.4571e-06, 4.5419e-05,  ..., 1.5198e-02,
           9.3079e-02, 1.5833e-01]],

         [[8.9172e-02, 3.8743e-06, 5.2512e-05,  ..., 1.4786e-02,
           9.0576e-02, 9.4971e-02]],

         [[9.4189e-01, 6.3777e-06, 1.4544e-05,  ..., 8.6746e-03,
           1.5945e-02, 4.6425e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 2.0862e-06, 1.1086e-05,  ..., 2.1164e-02,
           1.6266e-02, 6.8176e-02]],

         [[3.2422e-01, 2.0683e-05, 6.3837e-05,  ..., 9.2163e-02,
           2.5439e-01, 4.8584e-02]],

         [[8.8501e-02, 4.1723e-07, 1.9073e-06,  ..., 7.0312e-02,
           7.7344e-01, 4.9072e-02]],

         ...,

         [[6.2354e-01, 7.4148e-05, 1.6642e-04,  ..., 5.6725e-03,
           1.0597e-02, 9.8038e-03]],

         [[3.8184e-01, 2.1875e-05, 1.2100e-05,  ..., 3.9307e-02,
           1.0358e-01, 2.5375e-02]],

         [[7.0374e-02, 5.1141e-05, 2.1040e-04,  ..., 2.2491e-02,
           6.5613e-02, 6.6797e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 8.7440e-05, 1.6320e-04,  ..., 4.2786e-02,
           2.4658e-02, 3.2654e-02]],

         [[4.3481e-01, 3.3188e-04, 2.0695e-04,  ..., 1.2999e-03,
           1.3313e-03, 1.2726e-02]],

         [[3.0713e-01, 8.6010e-05, 2.0015e-04,  ..., 2.5253e-03,
           7.2212e-03, 5.0140e-02]],

         ...,

         [[8.2861e-01, 1.6212e-04, 2.3055e-04,  ..., 1.2985e-02,
           1.2985e-02, 9.6512e-03]],

         [[1.7407e-01, 7.4983e-05, 2.8753e-04,  ..., 4.9400e-03,
           3.7079e-02, 7.3291e-01]],

         [[5.0488e-01, 2.4974e-05, 6.8963e-05,  ..., 1.4107e-02,
           2.7466e-01, 1.2378e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1309e-01, 3.8862e-05, 5.5647e-04,  ..., 9.2697e-03,
           1.0880e-02, 1.5350e-02]],

         [[7.6843e-02, 6.4373e-06, 5.2273e-05,  ..., 1.4549e-02,
           4.2450e-02, 4.4464e-02]],

         [[2.3804e-01, 7.7903e-05, 5.6076e-04,  ..., 8.1299e-02,
           1.2158e-01, 1.2549e-01]],

         ...,

         [[8.6475e-01, 7.1943e-05, 3.5644e-04,  ..., 7.9956e-03,
           1.0590e-02, 2.4780e-02]],

         [[9.0039e-01, 8.9407e-07, 3.1090e-04,  ..., 2.5043e-03,
           3.0661e-04, 7.4158e-03]],

         [[5.4395e-01, 5.8651e-05, 1.1730e-04,  ..., 1.0834e-01,
           6.2943e-03, 1.4099e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3663e-05, 2.7180e-05, 1.7798e-01,  ..., 4.1199e-04,
           2.3254e-02, 1.0339e-01]],

         [[5.5134e-05, 1.7822e-05, 1.0384e-02,  ..., 1.0948e-03,
           2.9068e-03, 7.9041e-03]],

         [[2.9325e-04, 4.6730e-04, 1.0724e-01,  ..., 4.9744e-03,
           2.8519e-02, 5.6519e-02]],

         ...,

         [[3.9172e-04, 5.1618e-05, 1.9751e-01,  ..., 3.0289e-02,
           1.2854e-01, 1.7297e-01]],

         [[1.1021e-04, 5.7638e-05, 2.9224e-01,  ..., 2.9221e-03,
           6.4583e-03, 1.3779e-02]],

         [[4.1127e-06, 2.9206e-06, 2.5879e-01,  ..., 3.2845e-03,
           4.1199e-03, 3.8483e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3956e-05, 5.6624e-05, 2.8125e-01,  ..., 1.6203e-03,
           6.2103e-03, 5.2826e-02]],

         [[5.3287e-05, 1.9968e-05, 1.7651e-01,  ..., 5.4216e-04,
           3.8223e-03, 3.8452e-02]],

         [[6.8307e-05, 7.2718e-06, 3.1519e-01,  ..., 9.3508e-04,
           6.4659e-03, 4.8340e-02]],

         ...,

         [[1.9836e-04, 2.4676e-05, 2.3941e-02,  ..., 5.4359e-04,
           9.5034e-04, 9.9411e-03]],

         [[1.1027e-05, 5.9605e-06, 2.4927e-01,  ..., 1.1677e-04,
           3.5553e-03, 3.7628e-02]],

         [[4.8816e-05, 3.6001e-05, 2.5391e-01,  ..., 3.1161e-04,
           8.5602e-03, 3.2562e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8716e-05, 6.3777e-06, 1.0327e-01,  ..., 3.3545e-04,
           1.6146e-03, 1.4565e-02]],

         [[6.3062e-05, 1.3113e-06, 2.2290e-01,  ..., 3.0994e-05,
           3.5744e-03, 2.9465e-02]],

         [[5.1200e-05, 1.5795e-05, 2.3291e-01,  ..., 2.6941e-04,
           3.1452e-03, 2.4170e-02]],

         ...,

         [[5.1141e-05, 1.6928e-05, 1.4893e-01,  ..., 1.2245e-03,
           1.3115e-02, 4.2664e-02]],

         [[1.0073e-05, 1.1086e-05, 9.4971e-02,  ..., 2.8896e-04,
           2.7370e-03, 1.4801e-02]],

         [[4.0472e-05, 5.1856e-06, 1.5356e-01,  ..., 3.1233e-04,
           2.4242e-03, 1.7624e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6134e-05, 2.6822e-06, 1.1559e-02,  ..., 1.1284e-02,
           3.9795e-01, 2.2144e-01]],

         [[9.6560e-06, 4.9472e-05, 9.8633e-02,  ..., 1.3269e-01,
           4.4617e-02, 3.8330e-02]],

         [[3.8862e-05, 2.7871e-04, 3.1860e-01,  ..., 2.0161e-03,
           7.9250e-04, 7.7057e-04]],

         ...,

         [[5.9891e-04, 5.6386e-05, 1.8567e-01,  ..., 1.3237e-03,
           4.4365e-03, 1.4572e-02]],

         [[1.1988e-03, 9.5844e-05, 4.3945e-02,  ..., 1.8753e-02,
           2.4805e-01, 3.1348e-01]],

         [[1.8501e-04, 1.7166e-05, 4.2450e-02,  ..., 1.1650e-02,
           1.1816e-01, 1.4758e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1324e-03, 4.3726e-04, 9.3689e-02,  ..., 1.0971e-02,
           5.9082e-02, 9.0088e-02]],

         [[7.8630e-04, 1.2426e-03, 2.4292e-01,  ..., 9.9106e-03,
           1.3977e-02, 1.4816e-02]],

         [[2.9640e-03, 3.4666e-04, 1.4819e-01,  ..., 4.9706e-03,
           5.3680e-02, 6.8665e-02]],

         ...,

         [[3.8052e-04, 8.2314e-05, 2.4292e-02,  ..., 6.1417e-03,
           7.3828e-01, 1.3550e-01]],

         [[1.0805e-03, 3.2520e-03, 2.9053e-01,  ..., 1.0157e-03,
           6.0768e-03, 3.1097e-02]],

         [[4.1103e-04, 3.3021e-04, 2.1973e-01,  ..., 2.8595e-02,
           2.9739e-02, 3.4760e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[3.8123e-04, 1.7452e-04, 6.9695e-03,  ..., 1.1249e-01,
           5.3662e-01, 7.9773e-02]],

         [[3.8147e-04, 5.5504e-04, 6.7635e-03,  ..., 1.4320e-02,
           1.0577e-01, 7.8174e-01]],

         [[1.0014e-05, 2.1279e-05, 7.4208e-05,  ..., 1.5712e-04,
           2.4567e-03, 9.9121e-01]],

         ...,

         [[2.2054e-06, 6.0797e-06, 7.3731e-05,  ..., 2.0039e-04,
           4.0245e-03, 9.8535e-01]],

         [[3.4237e-03, 2.5043e-03, 2.3518e-03,  ..., 3.6804e-02,
           7.7881e-02, 2.6917e-02]],

         [[1.9806e-02, 5.3291e-03, 1.6251e-03,  ..., 1.9196e-02,
           3.1647e-02, 5.7281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0041e-04, 8.8310e-04, 3.6602e-03,  ..., 3.3661e-02,
           7.5867e-02, 6.7627e-01]],

         [[2.3842e-07, 2.9802e-07, 1.1921e-06,  ..., 1.0779e-01,
           7.7197e-01, 1.1475e-01]],

         [[9.3174e-04, 2.6970e-03, 1.8417e-02,  ..., 6.7383e-02,
           4.2175e-02, 1.8323e-01]],

         ...,

         [[2.6684e-03, 6.7635e-03, 1.2245e-02,  ..., 6.3515e-03,
           2.8915e-02, 4.1846e-01]],

         [[2.6913e-03, 1.0918e-02, 3.9612e-02,  ..., 2.7649e-02,
           3.1586e-02, 2.6382e-02]],

         [[5.0087e-03, 1.2108e-02, 1.9043e-02,  ..., 2.0007e-01,
           4.9774e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0797, 0.0073, 0.0407,  ..., 0.0317, 0.0271, 0.0395]],

         [[0.0009, 0.0004, 0.0023,  ..., 0.0082, 0.0031, 0.0075]],

         [[0.0091, 0.0022, 0.0018,  ..., 0.0525, 0.0319, 0.0242]],

         ...,

         [[0.0175, 0.0009, 0.0030,  ..., 0.0721, 0.1962, 0.3416]],

         [[0.0207, 0.0023, 0.0026,  ..., 0.0334, 0.0279, 0.0581]],

         [[0.0115, 0.0041, 0.0156,  ..., 0.0721, 0.2273, 0.3181]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[6.9189e-01, 1.6093e-06, 3.2365e-05,  ..., 9.1858e-03,
           2.7786e-02, 6.3858e-03]],

         [[8.4766e-01, 2.9802e-06, 7.8499e-05,  ..., 5.4283e-03,
           5.3444e-03, 1.3863e-02]],

         [[9.6875e-01, 2.1458e-06, 9.8348e-06,  ..., 3.0842e-03,
           1.3208e-03, 4.2801e-03]],

         ...,

         [[3.5156e-01, 2.0087e-04, 1.2732e-03,  ..., 3.9935e-05,
           2.5392e-05, 9.3307e-03]],

         [[7.3633e-01, 1.7548e-04, 4.4608e-04,  ..., 6.4316e-03,
           2.6932e-03, 5.2299e-03]],

         [[2.7832e-01, 1.6439e-04, 3.0708e-04,  ..., 3.1616e-02,
           2.9922e-02, 1.1360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1895e-01, 1.6809e-05, 5.9605e-07,  ..., 1.9445e-03,
           4.1723e-04, 2.1225e-02]],

         [[8.6328e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8275e-02,
           5.8380e-02, 4.9011e-02]],

         [[9.1846e-01, 3.4308e-04, 1.0836e-04,  ..., 5.8136e-03,
           1.0818e-02, 4.9133e-03]],

         ...,

         [[1.7554e-01, 6.1703e-04, 2.3746e-04,  ..., 1.2396e-01,
           1.5015e-01, 1.9760e-02]],

         [[1.6113e-01, 1.4229e-03, 4.4441e-04,  ..., 2.5055e-02,
           7.5455e-03, 7.0190e-03]],

         [[4.9121e-01, 1.1736e-04, 3.5143e-04,  ..., 2.5833e-02,
           3.5370e-02, 1.9470e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7505e-01, 2.7704e-04, 9.3937e-05,  ..., 9.0027e-02,
           5.2521e-02, 1.1346e-01]],

         [[6.6406e-01, 1.2374e-04, 4.1783e-05,  ..., 6.8237e-02,
           8.1848e-02, 9.5825e-02]],

         [[8.0811e-01, 8.3447e-05, 2.1577e-05,  ..., 1.0361e-02,
           1.5808e-02, 4.0588e-02]],

         ...,

         [[9.0527e-01, 3.4988e-05, 5.9605e-07,  ..., 4.7684e-05,
           5.0068e-06, 8.9905e-02]],

         [[8.7109e-01, 4.3678e-04, 6.7949e-06,  ..., 2.3246e-06,
           1.7881e-07, 3.0640e-02]],

         [[9.4678e-01, 3.3951e-04, 1.0133e-06,  ..., 1.4901e-06,
           5.9605e-08, 1.2749e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9380e-01, 1.6212e-05, 7.0333e-06,  ..., 7.4036e-02,
           2.6709e-01, 1.8042e-01]],

         [[9.0088e-01, 2.8253e-05, 6.4373e-06,  ..., 1.8358e-04,
           2.0492e-04, 3.7659e-02]],

         [[7.1240e-01, 9.1195e-05, 2.2948e-05,  ..., 1.6804e-03,
           6.7377e-04, 5.3375e-02]],

         ...,

         [[6.8896e-01, 2.8777e-04, 2.6107e-05,  ..., 3.0594e-03,
           5.8327e-03, 2.1500e-02]],

         [[7.5293e-01, 1.3599e-03, 1.1909e-04,  ..., 7.3318e-03,
           1.0315e-02, 8.5815e-02]],

         [[9.3799e-01, 1.1265e-04, 4.9412e-05,  ..., 1.1263e-03,
           2.8210e-03, 1.3809e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1260e-01, 2.7239e-05, 4.2856e-05,  ..., 6.0425e-03,
           2.4796e-03, 8.0948e-03]],

         [[9.6094e-01, 1.8120e-05, 7.5102e-06,  ..., 7.0524e-04,
           6.6805e-04, 1.1612e-02]],

         [[9.8486e-01, 3.4571e-06, 1.6689e-06,  ..., 7.3576e-04,
           1.8902e-03, 6.5079e-03]],

         ...,

         [[8.1787e-01, 1.6010e-04, 1.1146e-05,  ..., 8.4229e-03,
           7.8201e-03, 2.1271e-02]],

         [[8.9258e-01, 2.2078e-04, 4.1306e-05,  ..., 2.0294e-03,
           7.6151e-04, 1.2581e-02]],

         [[9.6973e-01, 9.0003e-06, 6.3777e-06,  ..., 1.2074e-03,
           5.3368e-03, 6.6948e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4741e-01, 9.2888e-04, 4.5085e-04,  ..., 4.9162e-04,
           2.3441e-03, 2.5803e-02]],

         [[9.3652e-01, 2.5213e-05, 5.2154e-05,  ..., 1.0395e-03,
           2.1248e-03, 1.3943e-03]],

         [[9.3066e-01, 3.1650e-05, 1.0252e-05,  ..., 1.5092e-04,
           1.8682e-03, 3.0685e-02]],

         ...,

         [[9.6338e-01, 1.5855e-05, 1.6153e-05,  ..., 2.5272e-04,
           1.6375e-03, 1.2520e-02]],

         [[9.0869e-01, 2.3842e-06, 1.1086e-05,  ..., 1.3199e-02,
           4.3640e-02, 3.5191e-03]],

         [[8.4961e-01, 3.5763e-06, 6.1393e-06,  ..., 5.3978e-04,
           4.3678e-03, 1.0321e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0999e-01, 2.1839e-04, 7.7438e-04,  ..., 3.4515e-02,
           1.8204e-02, 1.3763e-02]],

         [[9.1406e-01, 2.8849e-05, 4.0710e-05,  ..., 3.8261e-03,
           1.1536e-02, 1.4526e-02]],

         [[7.1436e-01, 2.9981e-05, 5.3048e-05,  ..., 1.0185e-02,
           1.3879e-01, 4.9561e-02]],

         ...,

         [[7.0898e-01, 1.6093e-04, 1.7536e-04,  ..., 2.2583e-03,
           1.1148e-03, 9.9335e-03]],

         [[7.8906e-01, 1.0824e-04, 4.3392e-05,  ..., 4.7016e-04,
           2.0866e-03, 5.9082e-02]],

         [[7.5293e-01, 1.4663e-05, 2.3782e-05,  ..., 7.7942e-02,
           4.9164e-02, 1.6479e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 7.3612e-05, 8.1062e-06,  ..., 3.5167e-06,
           4.2915e-06, 1.3733e-02]],

         [[8.9160e-01, 7.0989e-05, 2.1029e-04,  ..., 5.0545e-03,
           5.6839e-03, 8.0643e-03]],

         [[8.9111e-01, 1.7285e-05, 5.5432e-06,  ..., 4.0460e-04,
           2.1327e-04, 1.9531e-02]],

         ...,

         [[8.1006e-01, 3.2973e-04, 1.5819e-04,  ..., 1.7071e-03,
           2.2354e-03, 2.7557e-02]],

         [[7.3389e-01, 8.8632e-05, 1.0002e-04,  ..., 4.8943e-03,
           3.5004e-02, 1.0437e-02]],

         [[3.6743e-01, 1.0185e-03, 1.8835e-04,  ..., 1.1765e-02,
           1.9394e-02, 4.4586e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 7.7486e-06, 6.6757e-06,  ..., 4.6110e-04,
           1.9045e-03, 5.8057e-01]],

         [[5.1123e-01, 5.1069e-04, 7.5674e-04,  ..., 9.0742e-04,
           1.4334e-03, 4.5197e-02]],

         [[7.9688e-01, 2.4772e-04, 4.8232e-04,  ..., 6.7635e-03,
           6.3286e-03, 1.7441e-02]],

         ...,

         [[7.3584e-01, 4.8041e-05, 9.0420e-05,  ..., 5.6505e-04,
           2.6741e-03, 1.5478e-03]],

         [[7.9297e-01, 5.2810e-05, 3.6001e-05,  ..., 1.2001e-02,
           5.7587e-02, 1.9913e-02]],

         [[2.7319e-01, 1.4901e-04, 2.5845e-04,  ..., 1.7838e-02,
           3.5370e-02, 5.5469e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1396e-01, 6.6042e-05, 4.8518e-05,  ..., 3.1624e-03,
           2.0203e-02, 5.6494e-01]],

         [[7.4072e-01, 8.7738e-05, 8.2076e-05,  ..., 1.0361e-02,
           8.3435e-02, 8.8654e-03]],

         [[2.2693e-01, 2.8038e-04, 1.3936e-04,  ..., 1.3485e-03,
           3.4904e-03, 5.8838e-02]],

         ...,

         [[4.1870e-02, 5.0664e-06, 3.3975e-06,  ..., 4.7112e-04,
           5.0621e-03, 9.4043e-01]],

         [[8.1885e-01, 1.9610e-05, 3.2842e-05,  ..., 2.4052e-03,
           9.6817e-03, 2.6001e-02]],

         [[8.4717e-01, 2.2352e-04, 4.5776e-05,  ..., 1.8272e-03,
           2.6520e-02, 1.0246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2129e-01, 7.5054e-04, 5.6648e-04,  ..., 3.0327e-03,
           3.0098e-03, 1.6460e-03]],

         [[7.8223e-01, 5.3346e-05, 1.8036e-04,  ..., 4.7546e-02,
           6.7932e-02, 1.4771e-02]],

         [[2.8101e-01, 8.5652e-05, 7.5877e-05,  ..., 3.3879e-04,
           8.9598e-04, 2.3300e-02]],

         ...,

         [[6.4502e-01, 0.0000e+00, 1.7881e-07,  ..., 3.2349e-02,
           2.7905e-01, 3.6072e-02]],

         [[3.4082e-01, 3.7146e-04, 7.9679e-04,  ..., 6.0272e-03,
           4.1847e-03, 4.9973e-04]],

         [[9.2969e-01, 5.9605e-08, 5.9605e-08,  ..., 9.2936e-04,
           8.3389e-03, 5.8167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 5.1618e-05, 1.2636e-05,  ..., 1.7643e-03,
           5.5618e-03, 1.7624e-02]],

         [[1.2408e-01, 3.7169e-04, 4.6253e-04,  ..., 4.9744e-02,
           3.6896e-02, 2.9984e-03]],

         [[4.0381e-01, 5.7161e-05, 8.0884e-05,  ..., 3.7476e-02,
           2.7539e-01, 3.5522e-02]],

         ...,

         [[6.8701e-01, 4.0531e-06, 3.0398e-06,  ..., 8.7708e-02,
           5.3528e-02, 7.0068e-02]],

         [[6.5820e-01, 2.9802e-07, 3.5763e-07,  ..., 2.8553e-03,
           2.6108e-02, 3.0591e-01]],

         [[6.8701e-01, 1.6093e-05, 2.3603e-05,  ..., 5.8868e-02,
           5.7068e-02, 6.5552e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0472e-02, 2.3842e-07, 2.3842e-07,  ..., 1.0723e-04,
           8.3160e-04, 9.6680e-01]],

         [[7.0740e-02, 8.2850e-06, 1.4961e-05,  ..., 2.2046e-01,
           3.0493e-01, 5.7983e-02]],

         [[2.3425e-01, 4.5848e-04, 9.4461e-04,  ..., 1.8890e-02,
           1.7105e-02, 6.0028e-02]],

         ...,

         [[5.2344e-01, 4.4286e-05, 4.6074e-05,  ..., 2.6398e-03,
           8.3694e-03, 1.2573e-02]],

         [[6.4258e-01, 1.2398e-05, 9.3579e-06,  ..., 3.4904e-03,
           5.2376e-03, 2.4429e-02]],

         [[9.2236e-01, 2.2650e-05, 1.3351e-05,  ..., 1.5378e-04,
           3.8815e-04, 5.0735e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4072e-01, 2.4199e-05, 2.0504e-05,  ..., 1.5507e-03,
           1.1581e-02, 1.5283e-01]],

         [[1.5283e-01, 3.7217e-04, 6.5207e-05,  ..., 2.1076e-03,
           2.2995e-02, 1.0028e-01]],

         [[2.5415e-01, 2.2650e-05, 1.6391e-05,  ..., 9.7733e-03,
           1.5442e-02, 8.9539e-02]],

         ...,

         [[8.7646e-01, 3.6657e-05, 2.7955e-05,  ..., 2.6512e-03,
           1.8225e-03, 1.0735e-02]],

         [[7.0996e-01, 8.0049e-05, 3.9995e-05,  ..., 1.1208e-02,
           7.0251e-02, 6.7566e-02]],

         [[6.2744e-01, 1.4722e-05, 7.6294e-06,  ..., 6.8331e-04,
           4.6043e-03, 1.8762e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9531e-01, 7.1716e-04, 7.9107e-04,  ..., 1.8494e-02,
           1.2413e-02, 1.8280e-02]],

         [[2.7100e-01, 6.2227e-05, 4.6611e-05,  ..., 8.1396e-04,
           9.8724e-03, 4.4586e-02]],

         [[6.8164e-01, 2.8789e-05, 8.1360e-05,  ..., 1.7426e-02,
           7.5256e-02, 9.0454e-02]],

         ...,

         [[4.9194e-01, 3.8815e-04, 6.1631e-05,  ..., 9.9087e-04,
           3.8776e-03, 8.0627e-02]],

         [[6.2061e-01, 1.0786e-03, 5.9748e-04,  ..., 2.5630e-06,
           1.1981e-05, 1.4999e-02]],

         [[3.9844e-01, 2.2602e-04, 1.7059e-04,  ..., 1.0735e-02,
           1.0025e-02, 1.2016e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.4971e-02, 1.4281e-04, 1.0204e-04,  ..., 9.2041e-02,
           1.5173e-01, 7.9803e-03]],

         [[3.7012e-01, 1.6904e-04, 1.1688e-04,  ..., 5.3864e-03,
           5.5023e-02, 3.6652e-02]],

         [[2.9370e-01, 3.0696e-05, 1.6212e-05,  ..., 8.2703e-03,
           1.2927e-01, 4.1431e-01]],

         ...,

         [[5.6201e-01, 1.2577e-05, 2.0206e-05,  ..., 2.8320e-02,
           8.3557e-02, 8.3557e-02]],

         [[8.2031e-01, 1.1861e-05, 1.7643e-05,  ..., 2.6321e-03,
           3.3531e-03, 9.4971e-02]],

         [[5.4980e-01, 1.1122e-04, 1.0407e-04,  ..., 8.3160e-03,
           4.3732e-02, 2.3462e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.1128e-04, 1.9968e-05,  ..., 3.0255e-04,
           6.8855e-03, 3.4973e-02]],

         [[1.7480e-01, 4.6372e-05, 3.3438e-05,  ..., 8.1299e-02,
           2.0764e-01, 1.1829e-01]],

         [[3.1836e-01, 2.2411e-05, 1.0252e-05,  ..., 2.4068e-04,
           9.0790e-04, 6.6357e-01]],

         ...,

         [[6.4148e-02, 1.0592e-04, 3.2306e-05,  ..., 1.3232e-05,
           3.7313e-05, 2.0020e-02]],

         [[2.7905e-01, 3.5191e-04, 1.6367e-04,  ..., 3.6964e-03,
           1.4114e-02, 8.5602e-03]],

         [[3.1738e-01, 2.9397e-04, 1.3638e-04,  ..., 5.2071e-03,
           2.6047e-02, 2.0126e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8696e-01, 1.1867e-04, 7.5102e-05,  ..., 7.1449e-03,
           8.7509e-03, 3.9520e-02]],

         [[9.7461e-01, 1.0532e-04, 4.4703e-06,  ..., 1.2517e-06,
           4.4703e-06, 7.2956e-05]],

         [[9.2822e-01, 6.0034e-04, 1.0931e-04,  ..., 1.8883e-04,
           5.8711e-05, 1.4172e-03]],

         ...,

         [[1.8787e-01, 5.1165e-04, 9.3365e-04,  ..., 9.3536e-03,
           6.8436e-03, 3.5309e-02]],

         [[8.3398e-01, 2.7061e-05, 2.9266e-05,  ..., 1.0891e-03,
           1.6613e-03, 7.7576e-02]],

         [[1.7285e-01, 1.0195e-03, 1.2989e-03,  ..., 7.9575e-03,
           8.7357e-03, 3.2227e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0010e-01, 1.7631e-04, 2.2995e-04,  ..., 5.5275e-03,
           1.6251e-02, 2.5955e-02]],

         [[3.7695e-01, 1.5736e-04, 2.9874e-04,  ..., 3.8433e-03,
           6.9046e-03, 2.0935e-02]],

         [[4.9048e-01, 2.3568e-04, 2.8658e-04,  ..., 6.5088e-04,
           1.7824e-03, 3.1830e-02]],

         ...,

         [[8.4686e-03, 4.2379e-05, 2.5690e-05,  ..., 5.2185e-03,
           3.9764e-02, 1.2909e-02]],

         [[3.4180e-03, 4.1127e-06, 1.0729e-06,  ..., 7.9870e-06,
           8.1658e-05, 1.0696e-02]],

         [[3.4058e-01, 1.4222e-04, 9.6202e-05,  ..., 5.9891e-04,
           2.6016e-03, 3.1677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4004e-01, 2.9027e-05, 4.9114e-05,  ..., 1.6527e-03,
           1.0445e-02, 6.2012e-02]],

         [[4.8560e-01, 1.6105e-04, 2.0599e-04,  ..., 2.2018e-02,
           6.2225e-02, 6.4583e-03]],

         [[2.6562e-01, 4.5002e-05, 1.2183e-04,  ..., 1.1609e-01,
           1.3049e-01, 2.8271e-01]],

         ...,

         [[5.7910e-01, 4.0483e-04, 3.9697e-04,  ..., 3.2166e-02,
           2.9282e-02, 5.9174e-02]],

         [[1.8274e-01, 4.5538e-05, 3.3855e-05,  ..., 1.5414e-04,
           3.9053e-04, 1.7712e-01]],

         [[4.6704e-01, 1.8752e-04, 1.9503e-04,  ..., 8.6212e-03,
           1.6235e-02, 3.6865e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5742e-01, 2.0790e-04, 2.0957e-04,  ..., 7.2746e-03,
           2.9800e-02, 8.8928e-02]],

         [[2.8003e-01, 9.6560e-05, 1.7822e-04,  ..., 5.5008e-03,
           3.1921e-02, 6.5979e-02]],

         [[5.0195e-01, 2.8205e-04, 3.0041e-04,  ..., 1.0338e-02,
           8.0078e-02, 3.0396e-02]],

         ...,

         [[7.2998e-01, 5.9605e-08, 5.3644e-07,  ..., 3.1586e-02,
           1.9043e-01, 2.8748e-02]],

         [[3.4277e-01, 3.9399e-05, 3.0935e-05,  ..., 9.2957e-02,
           2.2302e-01, 7.9529e-02]],

         [[5.9033e-01, 4.7684e-07, 3.1590e-06,  ..., 1.9730e-02,
           3.1104e-01, 6.1249e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 1.8656e-05, 1.5259e-05,  ..., 4.7089e-02,
           1.6431e-01, 1.6266e-02]],

         [[1.2964e-01, 1.6272e-05, 1.8835e-05,  ..., 3.9749e-03,
           1.5717e-02, 7.6953e-01]],

         [[1.8787e-01, 1.2267e-04, 2.2209e-04,  ..., 3.1525e-02,
           6.2439e-02, 4.8714e-03]],

         ...,

         [[1.6968e-01, 1.0812e-04, 3.0494e-04,  ..., 1.0492e-01,
           4.2389e-02, 3.1616e-02]],

         [[5.4736e-01, 1.4663e-05, 8.7023e-06,  ..., 1.0126e-01,
           2.1851e-01, 4.2358e-02]],

         [[3.3276e-01, 0.0000e+00, 1.7881e-07,  ..., 8.2520e-02,
           4.5483e-01, 7.3975e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4229e-01, 7.7069e-05, 2.1052e-04,  ..., 2.2488e-03,
           3.1586e-03, 3.1036e-02]],

         [[9.8926e-01, 1.3828e-05, 1.7166e-05,  ..., 4.3273e-05,
           2.5570e-05, 3.0947e-04]],

         [[9.5801e-01, 1.2106e-04, 2.1935e-05,  ..., 5.2166e-04,
           3.8457e-04, 2.6913e-03]],

         ...,

         [[8.4033e-01, 5.4240e-05, 5.7757e-05,  ..., 3.6564e-03,
           2.2400e-02, 1.0522e-01]],

         [[9.1406e-01, 8.5235e-06, 3.0994e-06,  ..., 7.4148e-04,
           1.6069e-03, 6.4545e-03]],

         [[4.5825e-01, 9.1791e-05, 1.1879e-04,  ..., 4.5624e-03,
           1.6434e-02, 4.5825e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4023e-01, 1.5259e-05, 2.7180e-05,  ..., 1.2366e-01,
           2.4207e-01, 7.6477e-02]],

         [[8.1055e-01, 1.8179e-05, 9.5367e-06,  ..., 7.8201e-03,
           2.1606e-02, 1.2482e-01]],

         [[3.1860e-01, 1.8954e-04, 2.3341e-04,  ..., 5.2032e-02,
           2.9068e-02, 2.2778e-01]],

         ...,

         [[8.2959e-01, 5.3644e-06, 1.7107e-05,  ..., 3.4515e-02,
           5.6458e-02, 3.0457e-02]],

         [[8.4033e-01, 2.1696e-05, 4.6492e-05,  ..., 5.4054e-03,
           6.8855e-03, 5.0079e-02]],

         [[4.8291e-01, 2.6703e-05, 3.9697e-05,  ..., 4.1229e-02,
           9.3628e-02, 1.0364e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6499e-01, 4.7028e-05, 2.7657e-05,  ..., 9.6512e-03,
           6.7566e-02, 5.1392e-02]],

         [[3.4692e-01, 3.9577e-05, 1.7822e-05,  ..., 1.7433e-03,
           8.6823e-03, 6.2683e-02]],

         [[5.1221e-01, 1.2493e-04, 2.2149e-04,  ..., 1.2238e-02,
           3.4576e-02, 9.0393e-02]],

         ...,

         [[3.9337e-02, 1.3113e-06, 2.1100e-05,  ..., 1.2903e-01,
           2.1106e-01, 1.8738e-02]],

         [[4.9780e-01, 1.1086e-05, 6.8009e-05,  ..., 5.0079e-02,
           9.5032e-02, 2.6531e-03]],

         [[5.8252e-01, 4.9829e-05, 4.9829e-05,  ..., 1.2793e-01,
           1.1835e-01, 5.8594e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0088e-01, 5.3644e-07, 5.3644e-07,  ..., 1.1070e-02,
           5.5542e-02, 5.0488e-01]],

         [[7.2070e-01, 7.8678e-06, 2.0981e-05,  ..., 2.6352e-02,
           7.0801e-02, 2.7893e-02]],

         [[4.2798e-01, 2.9802e-07, 7.7486e-07,  ..., 5.0720e-02,
           2.7856e-01, 2.2473e-01]],

         ...,

         [[8.9453e-01, 1.4341e-04, 1.9038e-04,  ..., 1.0977e-03,
           3.2139e-03, 2.8748e-02]],

         [[8.4814e-01, 1.5199e-05, 8.3447e-06,  ..., 9.2392e-03,
           3.6682e-02, 1.9485e-02]],

         [[1.7151e-01, 1.5736e-04, 2.2852e-04,  ..., 1.3756e-02,
           3.1982e-02, 6.1768e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 3.0518e-05, 8.1301e-05,  ..., 2.8214e-02,
           3.4302e-02, 1.6052e-01]],

         [[3.9014e-01, 4.0555e-04, 3.0541e-04,  ..., 7.5912e-04,
           4.8180e-03, 2.2705e-02]],

         [[2.3730e-01, 4.4942e-05, 1.4997e-04,  ..., 1.4534e-02,
           8.4595e-02, 8.3313e-02]],

         ...,

         [[8.7695e-01, 2.0218e-04, 5.3692e-04,  ..., 1.6571e-02,
           2.4490e-02, 1.6830e-02]],

         [[2.2180e-01, 1.9300e-04, 2.6798e-04,  ..., 5.5542e-03,
           1.7105e-02, 6.7236e-01]],

         [[6.1572e-01, 2.4498e-05, 6.9201e-05,  ..., 7.5867e-02,
           2.1606e-01, 6.1920e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0632e-01, 1.1623e-05, 1.4126e-04,  ..., 6.0364e-02,
           7.1240e-01, 3.8208e-02]],

         [[1.7969e-01, 1.6332e-05, 8.0228e-05,  ..., 1.1131e-02,
           8.9294e-02, 4.5074e-02]],

         [[4.9487e-01, 2.9850e-04, 1.8482e-03,  ..., 3.9581e-02,
           1.1255e-01, 6.9885e-02]],

         ...,

         [[8.4375e-01, 9.5427e-05, 2.4247e-04,  ..., 3.6926e-02,
           2.9327e-02, 8.0490e-03]],

         [[9.6094e-01, 2.2650e-06, 1.4830e-04,  ..., 3.4308e-04,
           1.0262e-03, 5.6343e-03]],

         [[7.7637e-01, 5.5373e-05, 3.8326e-05,  ..., 1.1818e-02,
           6.1417e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4809e-05, 5.1916e-05, 1.8958e-01,  ..., 3.3966e-02,
           1.7395e-01, 1.4671e-02]],

         [[9.8050e-05, 4.6730e-05, 2.9556e-02,  ..., 1.1353e-02,
           3.2227e-02, 1.2549e-01]],

         [[1.4770e-04, 1.9062e-04, 1.0907e-01,  ..., 4.6539e-02,
           7.0984e-02, 1.2659e-01]],

         ...,

         [[3.0065e-04, 1.5557e-05, 6.7505e-02,  ..., 3.7292e-02,
           7.0361e-01, 5.8655e-02]],

         [[5.0545e-04, 2.3592e-04, 3.0737e-01,  ..., 4.1504e-03,
           5.2452e-03, 9.2957e-02]],

         [[7.1585e-05, 2.2757e-04, 2.9639e-01,  ..., 1.3535e-02,
           2.2675e-02, 2.7130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6606e-04, 2.2435e-04, 2.6196e-01,  ..., 1.6235e-02,
           6.6223e-02, 8.1635e-03]],

         [[1.4269e-04, 1.8394e-04, 2.7563e-01,  ..., 1.0689e-02,
           3.0685e-02, 9.5825e-03]],

         [[3.3832e-04, 1.4782e-04, 3.3252e-01,  ..., 7.5798e-03,
           2.3163e-02, 1.4467e-03]],

         ...,

         [[6.5899e-04, 1.6057e-04, 1.4453e-01,  ..., 3.1815e-03,
           1.8234e-02, 2.5366e-01]],

         [[1.3685e-04, 1.0717e-04, 3.2764e-01,  ..., 8.3618e-03,
           2.8412e-02, 4.2229e-03]],

         [[3.5405e-05, 9.8705e-05, 3.0664e-01,  ..., 7.0152e-03,
           1.0490e-02, 9.6464e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6192e-04, 2.0516e-04, 2.8003e-01,  ..., 7.8812e-03,
           2.0004e-02, 8.5974e-04]],

         [[3.7813e-04, 1.4484e-05, 2.9736e-01,  ..., 6.0768e-03,
           1.3275e-02, 2.3483e-02]],

         [[8.0681e-04, 4.2844e-04, 3.2812e-01,  ..., 6.9427e-03,
           1.5465e-02, 7.5378e-03]],

         ...,

         [[4.8995e-05, 2.3484e-05, 1.7273e-01,  ..., 1.6327e-02,
           3.2715e-02, 3.2788e-01]],

         [[1.1295e-04, 2.0814e-04, 2.1606e-01,  ..., 1.0971e-02,
           1.5961e-02, 1.6861e-02]],

         [[4.1032e-04, 7.3135e-05, 2.9395e-01,  ..., 6.2447e-03,
           1.5213e-02, 1.8066e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6473e-05, 6.4373e-06, 2.0966e-02,  ..., 6.8176e-02,
           5.1562e-01, 2.6343e-01]],

         [[1.6451e-05, 5.2214e-05, 2.0813e-01,  ..., 3.1052e-02,
           4.5380e-02, 1.9821e-02]],

         [[5.5790e-05, 3.1662e-04, 2.3218e-01,  ..., 1.4839e-03,
           1.3628e-03, 9.7504e-03]],

         ...,

         [[2.2411e-04, 1.4067e-05, 2.8564e-01,  ..., 6.3241e-05,
           2.9397e-04, 1.1873e-03]],

         [[1.0118e-03, 9.0003e-05, 5.3772e-02,  ..., 6.2866e-02,
           1.8262e-01, 4.2969e-01]],

         [[1.6546e-04, 1.2398e-05, 5.7495e-02,  ..., 5.6824e-02,
           1.5259e-01, 3.9746e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2173e-04, 5.5885e-04, 2.0740e-01,  ..., 6.2866e-03,
           1.0612e-02, 1.5649e-01]],

         [[2.2829e-04, 3.0518e-04, 2.8589e-01,  ..., 6.7253e-03,
           9.9411e-03, 5.7861e-02]],

         [[1.2398e-03, 5.6791e-04, 1.7395e-01,  ..., 7.1106e-02,
           1.9702e-01, 8.3435e-02]],

         ...,

         [[1.0605e-03, 1.7309e-04, 1.4783e-01,  ..., 4.5441e-02,
           1.3354e-01, 2.7173e-01]],

         [[2.0409e-04, 6.2370e-04, 2.9883e-01,  ..., 4.9820e-03,
           8.1726e-02, 5.3589e-02]],

         [[2.3901e-04, 5.1677e-05, 1.3281e-01,  ..., 1.1908e-01,
           1.7322e-01, 6.4758e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.7351e-03, 1.1053e-03, 1.7288e-02,  ..., 1.6406e-01,
           2.7905e-01, 1.9788e-01]],

         [[2.5928e-05, 2.4629e-04, 1.8196e-03,  ..., 1.1864e-02,
           1.9562e-02, 9.4238e-01]],

         [[2.3842e-07, 3.5346e-05, 1.0133e-05,  ..., 1.0133e-05,
           4.3058e-04, 9.9951e-01]],

         ...,

         [[0.0000e+00, 1.7881e-07, 6.1393e-06,  ..., 2.7478e-05,
           9.1028e-04, 9.9854e-01]],

         [[3.2711e-03, 2.8858e-03, 3.9444e-03,  ..., 8.7036e-02,
           4.2419e-02, 5.1178e-02]],

         [[1.5305e-02, 4.1847e-03, 2.0065e-03,  ..., 1.4153e-02,
           5.9570e-02, 6.3416e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6253e-05, 2.2066e-04, 1.9665e-03,  ..., 2.9816e-02,
           2.3450e-01, 5.6250e-01]],

         [[0.0000e+00, 5.9605e-08, 4.1723e-07,  ..., 8.2642e-02,
           8.6133e-01, 5.3375e-02]],

         [[2.9278e-04, 8.2111e-04, 6.9847e-03,  ..., 6.7322e-02,
           3.5254e-01, 4.2114e-02]],

         ...,

         [[2.1801e-03, 6.3591e-03, 1.1879e-02,  ..., 3.0334e-02,
           3.0884e-01, 8.9905e-02]],

         [[1.6146e-03, 1.1032e-02, 3.1921e-02,  ..., 2.4292e-02,
           2.3727e-02, 1.9745e-02]],

         [[6.5804e-03, 1.8173e-02, 1.9196e-02,  ..., 1.7471e-02,
           1.0876e-01, 1.0620e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5378e-02, 7.2632e-03, 4.0344e-02,  ..., 3.3966e-02,
           6.1035e-02, 3.5339e-02]],

         [[2.6627e-03, 1.0386e-03, 5.8174e-03,  ..., 6.0005e-03,
           1.7090e-02, 1.2802e-02]],

         [[4.3945e-03, 8.2541e-04, 9.4652e-04,  ..., 3.1952e-02,
           2.7557e-02, 1.9852e-02]],

         ...,

         [[1.3618e-02, 4.7803e-05, 1.2159e-04,  ..., 1.3647e-01,
           5.8838e-01, 2.0178e-01]],

         [[7.4654e-03, 1.0881e-03, 1.7452e-03,  ..., 8.9539e-02,
           5.4718e-02, 1.3440e-01]],

         [[5.2032e-03, 8.3590e-04, 3.1052e-03,  ..., 1.1475e-01,
           5.7373e-01, 1.9519e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.0381e-01, 1.4341e-04, 8.3923e-05,  ..., 4.5547e-03,
           8.4457e-03, 2.9469e-03]],

         [[7.6123e-01, 9.1791e-06, 1.7703e-05,  ..., 1.2306e-02,
           5.8594e-02, 1.4209e-01]],

         [[7.9590e-01, 1.6093e-06, 1.6093e-06,  ..., 4.5319e-03,
           1.5610e-02, 1.6235e-01]],

         ...,

         [[5.2686e-01, 1.0264e-04, 6.6566e-04,  ..., 2.4092e-04,
           2.5809e-05, 3.4218e-03]],

         [[6.5967e-01, 3.1662e-04, 8.8549e-04,  ..., 1.7105e-02,
           2.0004e-02, 1.7441e-02]],

         [[6.6406e-01, 1.1134e-04, 3.5334e-04,  ..., 4.9072e-02,
           1.4557e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6084e-01, 6.1989e-06, 8.9407e-07,  ..., 3.4637e-03,
           8.4877e-04, 9.3994e-02]],

         [[5.5371e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8954e-03,
           3.7256e-01, 6.9824e-02]],

         [[7.9443e-01, 9.8407e-05, 2.4498e-05,  ..., 7.3586e-03,
           1.6431e-01, 1.7334e-02]],

         ...,

         [[5.9863e-01, 5.4932e-04, 2.8300e-04,  ..., 5.6549e-02,
           8.3389e-03, 1.1826e-02]],

         [[3.7109e-01, 1.4200e-03, 6.9284e-04,  ..., 1.1002e-02,
           1.6068e-02, 1.6846e-02]],

         [[3.5156e-01, 3.9876e-05, 4.4942e-05,  ..., 3.6835e-02,
           5.5957e-01, 1.4313e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6099e-01, 3.0088e-04, 1.1426e-04,  ..., 6.5369e-02,
           1.2402e-01, 8.9233e-02]],

         [[6.7920e-01, 8.1599e-05, 3.8981e-05,  ..., 7.3486e-02,
           1.6565e-01, 2.1332e-02]],

         [[7.7344e-01, 4.1366e-05, 2.5868e-05,  ..., 1.0757e-02,
           1.1316e-01, 7.7698e-02]],

         ...,

         [[7.9736e-01, 1.7881e-07, 5.9605e-08,  ..., 1.4246e-05,
           2.6524e-05, 1.9641e-01]],

         [[9.4092e-01, 2.6822e-06, 1.0729e-06,  ..., 0.0000e+00,
           0.0000e+00, 1.7578e-02]],

         [[9.7461e-01, 1.7881e-07, 5.9605e-08,  ..., 0.0000e+00,
           0.0000e+00, 2.1820e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 4.2379e-05, 5.1260e-06,  ..., 2.5406e-02,
           2.7686e-01, 4.3671e-02]],

         [[9.2578e-01, 1.0073e-05, 5.9605e-07,  ..., 1.4818e-04,
           2.8896e-03, 5.7800e-02]],

         [[8.7939e-01, 6.3181e-05, 2.4140e-05,  ..., 3.4771e-03,
           5.2185e-03, 8.6365e-03]],

         ...,

         [[1.4697e-01, 4.5002e-05, 7.3314e-06,  ..., 3.1757e-03,
           8.2324e-01, 5.2376e-03]],

         [[6.8506e-01, 7.4005e-04, 1.3292e-04,  ..., 2.3010e-02,
           5.9967e-02, 1.4636e-01]],

         [[4.4702e-01, 2.4962e-04, 5.0843e-05,  ..., 4.1237e-03,
           9.7229e-02, 4.4360e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7119e-01, 7.5102e-06, 9.3579e-06,  ..., 5.6839e-04,
           5.9586e-03, 6.8588e-03]],

         [[9.1260e-01, 1.0312e-05, 2.6822e-06,  ..., 2.3804e-03,
           6.5575e-03, 5.0751e-02]],

         [[9.7266e-01, 3.3975e-06, 5.3644e-07,  ..., 4.7565e-04,
           2.0966e-02, 3.7861e-03]],

         ...,

         [[8.2812e-01, 8.1003e-05, 1.6630e-05,  ..., 2.8324e-03,
           2.1576e-02, 5.8533e-02]],

         [[9.4873e-01, 8.1539e-05, 5.0664e-06,  ..., 6.4373e-04,
           9.1476e-03, 9.8343e-03]],

         [[9.0576e-01, 1.2577e-05, 3.0994e-06,  ..., 2.0866e-03,
           2.4979e-02, 5.6091e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7695e-01, 1.1835e-03, 2.8110e-04,  ..., 9.8267e-03,
           4.6112e-02, 3.9856e-02]],

         [[9.7461e-01, 3.9935e-06, 1.2279e-05,  ..., 1.0490e-03,
           1.7910e-03, 2.2106e-03]],

         [[7.4561e-01, 4.5002e-05, 2.8014e-06,  ..., 6.6614e-04,
           1.6382e-01, 5.5573e-02]],

         ...,

         [[5.2490e-01, 1.1384e-05, 1.7285e-05,  ..., 1.0376e-03,
           1.9409e-02, 4.3945e-01]],

         [[9.6680e-01, 1.3709e-06, 3.5167e-06,  ..., 6.8092e-03,
           1.2703e-03, 1.3596e-02]],

         [[4.5605e-01, 2.0266e-06, 4.2915e-06,  ..., 1.1845e-03,
           5.2002e-01, 5.4321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5415e-01, 1.2720e-04, 8.7595e-04,  ..., 5.1758e-02,
           2.5635e-02, 1.0559e-02]],

         [[7.1680e-01, 1.7345e-05, 5.4836e-06,  ..., 5.6267e-03,
           2.4524e-01, 1.6632e-02]],

         [[9.4092e-01, 1.0252e-05, 9.7752e-06,  ..., 9.6893e-03,
           1.6708e-02, 1.0712e-02]],

         ...,

         [[9.0967e-01, 3.1412e-05, 2.7955e-05,  ..., 8.2874e-04,
           1.3107e-02, 2.8563e-04]],

         [[5.4834e-01, 1.8477e-05, 5.7817e-06,  ..., 1.8463e-03,
           3.6963e-01, 4.0497e-02]],

         [[8.9502e-01, 2.0266e-05, 2.7061e-05,  ..., 2.0798e-02,
           7.2975e-03, 3.5370e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8301e-01, 1.5259e-04, 3.3379e-05,  ..., 7.7486e-07,
           6.5327e-05, 1.6079e-03]],

         [[9.3115e-01, 2.0325e-05, 2.8908e-05,  ..., 1.7319e-03,
           2.6932e-02, 1.2527e-02]],

         [[9.6729e-01, 1.7285e-05, 1.9908e-05,  ..., 4.7159e-04,
           5.3978e-03, 6.6376e-03]],

         ...,

         [[8.8574e-01, 3.0136e-04, 1.7846e-04,  ..., 1.2808e-03,
           2.6566e-02, 4.2648e-03]],

         [[8.7451e-01, 6.3419e-05, 6.0737e-05,  ..., 1.0849e-02,
           4.7569e-03, 6.1913e-03]],

         [[7.4121e-01, 2.4929e-03, 4.5562e-04,  ..., 1.8097e-02,
           8.1253e-03, 9.5749e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5576e-01, 1.9133e-05, 1.9312e-05,  ..., 5.8556e-04,
           1.1902e-02, 3.2739e-01]],

         [[5.0537e-01, 9.4509e-04, 1.7004e-03,  ..., 1.7099e-03,
           5.8044e-02, 3.9581e-02]],

         [[8.2471e-01, 2.6512e-04, 4.9829e-04,  ..., 3.9043e-03,
           6.8207e-03, 1.1177e-02]],

         ...,

         [[6.5137e-01, 2.0468e-04, 2.7871e-04,  ..., 5.8556e-03,
           1.1223e-02, 1.3504e-03]],

         [[7.4414e-01, 7.9811e-05, 7.1228e-05,  ..., 6.5857e-02,
           6.3354e-02, 8.2169e-03]],

         [[5.5127e-01, 9.3102e-05, 1.6475e-04,  ..., 1.1276e-02,
           2.7969e-02, 3.6694e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3843e-02, 3.3557e-05, 1.1683e-05,  ..., 1.6870e-03,
           5.8441e-02, 8.5449e-01]],

         [[6.3965e-01, 5.0664e-05, 2.3901e-05,  ..., 1.6687e-01,
           4.8279e-02, 1.0498e-02]],

         [[1.4270e-01, 5.9128e-04, 1.1694e-04,  ..., 1.2306e-02,
           2.1997e-01, 6.1249e-02]],

         ...,

         [[2.0874e-01, 1.1384e-05, 1.2994e-05,  ..., 3.1910e-03,
           4.8981e-02, 6.8311e-01]],

         [[8.7402e-01, 4.7326e-05, 2.7180e-05,  ..., 4.7302e-03,
           2.0889e-02, 1.5602e-02]],

         [[8.6816e-01, 1.3626e-04, 2.1577e-05,  ..., 1.7975e-02,
           4.2816e-02, 2.4277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9199e-01, 4.2439e-04, 9.4175e-04,  ..., 8.9722e-03,
           1.6144e-02, 3.6812e-03]],

         [[8.2959e-01, 1.0014e-05, 4.8637e-05,  ..., 4.4952e-02,
           3.5522e-02, 1.3252e-02]],

         [[3.5107e-01, 3.0696e-05, 3.1292e-05,  ..., 2.4071e-03,
           2.9526e-02, 3.6221e-03]],

         ...,

         [[2.4536e-01, 0.0000e+00, 0.0000e+00,  ..., 3.6743e-02,
           7.0166e-01, 1.4389e-02]],

         [[4.0137e-01, 5.4312e-04, 8.1062e-04,  ..., 5.9166e-03,
           2.7142e-03, 1.2457e-04]],

         [[5.8643e-01, 1.7881e-07, 0.0000e+00,  ..., 3.1738e-03,
           3.9453e-01, 1.2199e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3975e-01, 4.8637e-05, 7.3314e-06,  ..., 1.2077e-02,
           4.1107e-02, 2.4933e-02]],

         [[4.0436e-02, 9.8610e-04, 4.1842e-04,  ..., 5.1727e-02,
           1.8501e-03, 6.4754e-04]],

         [[2.9980e-01, 6.4194e-05, 9.7871e-05,  ..., 3.1543e-01,
           4.2877e-02, 7.7133e-03]],

         ...,

         [[6.1279e-01, 1.7285e-06, 7.1526e-07,  ..., 6.4087e-02,
           1.8860e-01, 2.2430e-02]],

         [[7.2632e-02, 5.9605e-08, 0.0000e+00,  ..., 1.1147e-02,
           9.0918e-01, 4.1962e-03]],

         [[5.1709e-01, 1.7464e-05, 3.4690e-05,  ..., 9.4543e-02,
           1.0236e-01, 1.1963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6987e-02, 5.3644e-07, 7.7486e-07,  ..., 1.8752e-04,
           2.0508e-02, 9.3896e-01]],

         [[5.0812e-02, 5.5432e-06, 1.3351e-05,  ..., 3.8892e-01,
           1.2927e-01, 3.6530e-02]],

         [[1.3562e-01, 5.6648e-04, 7.9679e-04,  ..., 2.7039e-02,
           3.4302e-02, 1.0445e-02]],

         ...,

         [[6.6895e-01, 7.9870e-05, 2.7061e-05,  ..., 1.0033e-02,
           2.5818e-02, 1.6289e-03]],

         [[6.3965e-01, 6.3181e-06, 1.3709e-06,  ..., 3.8872e-03,
           1.1951e-01, 5.9570e-02]],

         [[6.0059e-01, 3.3438e-05, 1.6212e-05,  ..., 2.3377e-04,
           1.0358e-01, 2.1149e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7974e-02, 2.1458e-06, 3.5763e-07,  ..., 2.3060e-03,
           9.1553e-01, 6.2180e-03]],

         [[2.0740e-01, 3.2973e-04, 6.3479e-05,  ..., 4.3152e-02,
           1.6992e-01, 9.7595e-02]],

         [[4.1919e-01, 3.5644e-05, 3.7432e-05,  ..., 2.8976e-02,
           1.0071e-01, 1.0638e-01]],

         ...,

         [[9.0332e-01, 3.1173e-05, 2.2948e-05,  ..., 3.7346e-03,
           2.2003e-02, 1.2733e-02]],

         [[5.2490e-01, 2.9743e-05, 1.7226e-05,  ..., 5.4474e-02,
           2.6392e-01, 2.6978e-02]],

         [[7.3584e-01, 1.3351e-05, 8.2850e-06,  ..., 3.3817e-03,
           1.5857e-01, 2.7451e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6609e-02, 1.5383e-03, 2.3708e-03,  ..., 6.9695e-03,
           5.3024e-03, 4.9133e-03]],

         [[5.6305e-02, 4.3809e-05, 2.0146e-05,  ..., 2.6398e-02,
           6.2317e-02, 2.9434e-02]],

         [[3.4229e-01, 1.5020e-05, 2.3723e-05,  ..., 1.5186e-01,
           3.2397e-01, 8.4229e-02]],

         ...,

         [[4.3359e-01, 4.1437e-04, 4.1485e-05,  ..., 9.9373e-04,
           6.9153e-02, 2.4353e-02]],

         [[8.3984e-01, 2.9349e-04, 1.9693e-04,  ..., 3.5763e-05,
           4.5252e-04, 1.0780e-02]],

         [[5.6934e-01, 1.0842e-04, 1.0228e-04,  ..., 1.7059e-02,
           2.2755e-03, 1.2026e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-02, 7.9095e-05, 9.1016e-05,  ..., 1.1853e-01,
           1.9007e-03, 1.1925e-02]],

         [[3.1323e-01, 4.3726e-04, 2.9683e-04,  ..., 6.5613e-02,
           2.9587e-02, 1.8951e-02]],

         [[6.1920e-02, 2.7418e-06, 1.4305e-06,  ..., 5.4230e-02,
           3.1689e-01, 4.7217e-01]],

         ...,

         [[3.1665e-01, 6.7353e-06, 8.4043e-06,  ..., 2.8833e-01,
           1.0858e-01, 1.7715e-02]],

         [[2.4036e-01, 4.9472e-06, 5.9009e-06,  ..., 2.5883e-03,
           6.6895e-01, 5.8899e-02]],

         [[3.6401e-01, 3.8981e-05, 2.2531e-05,  ..., 8.7280e-03,
           3.0884e-01, 2.4060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7119e-01, 6.5863e-05, 1.1265e-05,  ..., 3.4668e-02,
           1.4368e-01, 7.0007e-02]],

         [[4.0497e-02, 6.4969e-06, 2.5034e-06,  ..., 1.9019e-01,
           4.1528e-01, 7.4463e-02]],

         [[1.4001e-01, 4.2319e-06, 9.5367e-07,  ..., 2.8777e-04,
           2.2156e-02, 8.3105e-01]],

         ...,

         [[3.7872e-02, 2.8610e-05, 1.5914e-05,  ..., 1.8239e-04,
           3.9291e-03, 6.0387e-03]],

         [[2.6880e-01, 2.6298e-04, 1.0830e-04,  ..., 1.4809e-02,
           3.2104e-02, 3.1860e-02]],

         [[1.7468e-01, 2.2376e-04, 9.1255e-05,  ..., 4.8126e-02,
           2.7206e-02, 1.1253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7305e-01, 1.0419e-04, 4.1306e-05,  ..., 5.5351e-03,
           5.2917e-02, 7.5195e-02]],

         [[9.2334e-01, 2.9564e-04, 2.0266e-06,  ..., 5.3644e-07,
           4.8995e-05, 1.5249e-03]],

         [[9.1504e-01, 3.1424e-04, 2.0087e-05,  ..., 2.7359e-05,
           1.6994e-03, 5.5275e-03]],

         ...,

         [[1.6919e-01, 4.9829e-04, 1.0138e-03,  ..., 5.0316e-03,
           1.7014e-02, 3.1799e-02]],

         [[8.1104e-01, 2.3186e-05, 2.5094e-05,  ..., 9.0599e-04,
           8.2825e-02, 4.5044e-02]],

         [[1.8188e-01, 7.7629e-04, 1.3304e-03,  ..., 1.7319e-02,
           4.6722e-02, 5.6366e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8135e-01, 1.6296e-04, 2.1935e-04,  ..., 2.6779e-02,
           5.0812e-02, 4.4159e-02]],

         [[2.3132e-01, 1.1384e-04, 2.5249e-04,  ..., 1.1429e-02,
           2.0065e-02, 4.6265e-02]],

         [[3.4473e-01, 1.1927e-04, 1.6046e-04,  ..., 3.6812e-03,
           5.6274e-02, 3.7170e-02]],

         ...,

         [[1.9287e-02, 7.0632e-05, 5.0902e-05,  ..., 8.7769e-02,
           3.3844e-02, 3.9551e-02]],

         [[7.2823e-03, 7.6294e-06, 2.3842e-06,  ..., 2.6941e-04,
           1.4038e-02, 3.9978e-02]],

         [[5.0439e-01, 4.7684e-05, 4.7684e-05,  ..., 8.2779e-03,
           1.8262e-01, 5.8380e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9102e-01, 2.7537e-05, 3.0398e-05,  ..., 1.2993e-02,
           1.9238e-01, 1.4526e-01]],

         [[3.5425e-01, 1.3781e-04, 2.4581e-04,  ..., 9.6069e-02,
           1.8768e-02, 2.4292e-02]],

         [[1.9373e-01, 2.9504e-05, 9.3460e-05,  ..., 5.2948e-02,
           1.3110e-01, 3.7061e-01]],

         ...,

         [[5.2441e-01, 2.0480e-04, 2.2316e-04,  ..., 4.4403e-02,
           8.7585e-02, 7.9773e-02]],

         [[3.6224e-02, 7.5102e-06, 8.8811e-06,  ..., 8.3685e-05,
           4.8256e-03, 4.0405e-02]],

         [[2.1960e-01, 1.1319e-04, 1.1635e-04,  ..., 2.3880e-02,
           1.2985e-02, 1.9485e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7285e-01, 9.3043e-05, 6.8605e-05,  ..., 3.2501e-02,
           2.6562e-01, 1.0162e-01]],

         [[5.6396e-02, 1.0073e-05, 1.7107e-05,  ..., 4.9011e-02,
           3.8843e-01, 4.4281e-02]],

         [[3.3911e-01, 9.5427e-05, 5.2571e-05,  ..., 7.9285e-02,
           2.3071e-02, 1.4107e-02]],

         ...,

         [[6.1182e-01, 2.9802e-07, 5.9605e-08,  ..., 5.0995e-02,
           2.8442e-01, 4.2297e-02]],

         [[2.1509e-01, 1.5080e-05, 9.8348e-06,  ..., 1.7017e-01,
           2.2546e-01, 4.9896e-02]],

         [[1.5906e-01, 7.1526e-07, 5.9605e-07,  ..., 8.6487e-02,
           7.1289e-01, 3.5217e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1858e-03, 1.1921e-06, 1.8477e-06,  ..., 9.5703e-02,
           8.0414e-03, 1.4786e-02]],

         [[2.0599e-02, 1.1921e-06, 1.9073e-06,  ..., 2.6131e-04,
           1.1375e-02, 9.6191e-01]],

         [[6.0486e-02, 6.0499e-05, 9.8944e-05,  ..., 5.2155e-02,
           2.0695e-03, 5.5389e-03]],

         ...,

         [[1.2830e-01, 1.1438e-04, 3.9339e-04,  ..., 3.1921e-02,
           2.2293e-02, 7.2937e-03]],

         [[4.3335e-02, 1.7881e-07, 1.1921e-07,  ..., 6.3428e-01,
           2.7115e-02, 7.8278e-03]],

         [[1.7749e-01, 5.9605e-08, 5.9605e-08,  ..., 1.7615e-01,
           5.4883e-01, 6.5308e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 8.9526e-05, 4.1723e-04,  ..., 5.2147e-03,
           2.7969e-02, 4.9835e-02]],

         [[9.7949e-01, 8.1599e-05, 1.4365e-05,  ..., 1.0079e-04,
           6.5613e-04, 3.1776e-03]],

         [[8.7988e-01, 1.0693e-04, 9.3997e-05,  ..., 1.1635e-03,
           1.4221e-02, 2.2385e-02]],

         ...,

         [[4.2944e-01, 6.5565e-06, 6.5565e-06,  ..., 6.1859e-02,
           3.8477e-01, 7.9407e-02]],

         [[7.0508e-01, 6.2585e-06, 2.6226e-06,  ..., 6.1989e-03,
           6.4964e-03, 7.4768e-03]],

         [[2.5879e-01, 1.5676e-05, 4.0054e-05,  ..., 1.6632e-03,
           7.3395e-03, 6.9238e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4551e-01, 1.2577e-05, 1.8418e-05,  ..., 1.9128e-01,
           1.1292e-01, 7.3792e-02]],

         [[4.2407e-01, 9.6560e-06, 9.2983e-06,  ..., 4.2999e-02,
           3.2275e-01, 1.4539e-01]],

         [[2.3669e-01, 1.6999e-04, 2.7418e-04,  ..., 1.1559e-02,
           2.4707e-01, 2.3401e-01]],

         ...,

         [[6.0596e-01, 5.4836e-06, 1.7881e-05,  ..., 6.1890e-02,
           7.0129e-02, 1.0608e-01]],

         [[7.5293e-01, 2.7835e-05, 6.2764e-05,  ..., 8.3313e-03,
           1.5503e-02, 1.7029e-02]],

         [[3.0225e-01, 2.9445e-05, 7.1645e-05,  ..., 1.4844e-01,
           2.5208e-02, 3.0396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8665e-02, 7.6294e-06, 6.9141e-06,  ..., 7.9651e-02,
           3.1128e-03, 5.7716e-03]],

         [[4.2017e-01, 2.4378e-05, 1.3530e-05,  ..., 1.6296e-02,
           9.8419e-03, 7.4272e-03]],

         [[2.6392e-01, 1.1843e-04, 2.7490e-04,  ..., 1.7960e-02,
           1.0577e-01, 1.0828e-01]],

         ...,

         [[2.5940e-02, 7.7486e-06, 1.0604e-04,  ..., 6.7322e-02,
           5.6992e-03, 2.2888e-02]],

         [[1.3257e-01, 2.0027e-05, 1.1003e-04,  ..., 1.1511e-01,
           3.2158e-03, 8.6060e-03]],

         [[5.3613e-01, 2.4259e-05, 2.6345e-05,  ..., 1.0394e-01,
           3.9764e-02, 7.9712e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1118e-02, 1.1921e-07, 0.0000e+00,  ..., 6.6986e-03,
           9.5947e-01, 9.4452e-03]],

         [[5.1318e-01, 7.2718e-06, 2.4915e-05,  ..., 5.3436e-02,
           1.7593e-02, 2.2400e-02]],

         [[3.0493e-01, 1.0133e-06, 1.0729e-06,  ..., 7.8613e-02,
           5.0830e-01, 6.6162e-02]],

         ...,

         [[7.9150e-01, 2.4414e-04, 3.1352e-04,  ..., 3.1223e-03,
           5.9013e-03, 4.4556e-03]],

         [[4.9976e-01, 4.7684e-06, 2.3842e-06,  ..., 2.1997e-01,
           3.1204e-02, 1.0292e-02]],

         [[4.6387e-02, 3.4869e-05, 8.4102e-05,  ..., 3.2692e-03,
           1.1749e-01, 7.6025e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7100e-01, 2.0742e-05, 7.7128e-05,  ..., 2.6520e-02,
           1.8921e-01, 1.6309e-01]],

         [[2.8345e-01, 4.8876e-04, 3.8815e-04,  ..., 5.7945e-03,
           2.7634e-02, 1.9073e-02]],

         [[1.1023e-01, 3.4988e-05, 1.2171e-04,  ..., 4.3823e-02,
           2.6581e-02, 1.5625e-02]],

         ...,

         [[6.4014e-01, 2.2864e-04, 3.0756e-04,  ..., 1.4145e-02,
           3.6682e-02, 1.6968e-01]],

         [[1.2335e-01, 1.1247e-04, 1.4663e-04,  ..., 1.2405e-02,
           8.6121e-02, 7.2070e-01]],

         [[6.4758e-02, 7.0333e-06, 7.0930e-06,  ..., 4.7852e-01,
           6.4758e-02, 5.0446e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7463e-02, 3.8743e-06, 6.9559e-05,  ..., 6.6699e-01,
           7.5989e-02, 2.0294e-02]],

         [[1.1041e-01, 1.1563e-05, 7.8619e-05,  ..., 1.4807e-01,
           4.3427e-02, 4.7119e-02]],

         [[2.8882e-01, 2.6536e-04, 1.3523e-03,  ..., 8.1482e-02,
           4.7150e-02, 1.0461e-01]],

         ...,

         [[7.8418e-01, 2.4188e-04, 5.9652e-04,  ..., 3.5126e-02,
           1.0445e-02, 5.5504e-03]],

         [[9.3213e-01, 2.1458e-06, 5.6362e-04,  ..., 6.1655e-04,
           4.2496e-03, 1.0170e-02]],

         [[3.1470e-01, 5.1498e-05, 4.0889e-05,  ..., 6.9809e-03,
           4.6417e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5923e-05, 4.0829e-05, 1.2524e-01,  ..., 1.7664e-01,
           1.0902e-02, 1.7151e-02]],

         [[7.3791e-05, 1.5378e-05, 7.2098e-03,  ..., 9.1858e-03,
           3.3875e-02, 1.0574e-02]],

         [[7.3969e-05, 1.5533e-04, 5.7159e-02,  ..., 2.7115e-02,
           2.1484e-01, 7.3425e-02]],

         ...,

         [[2.9397e-04, 1.7107e-05, 6.8420e-02,  ..., 4.8218e-01,
           6.1310e-02, 5.8502e-02]],

         [[5.8985e-04, 2.4867e-04, 1.8970e-01,  ..., 1.3214e-02,
           8.6853e-02, 1.8066e-02]],

         [[2.3544e-05, 4.6492e-05, 1.5942e-01,  ..., 1.2589e-02,
           1.0193e-02, 2.7428e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4046e-04, 3.3784e-04, 2.8833e-01,  ..., 5.0903e-02,
           2.2373e-03, 6.0806e-03]],

         [[1.7166e-04, 1.4865e-04, 1.8396e-01,  ..., 3.1464e-02,
           4.6768e-03, 7.2289e-04]],

         [[9.4604e-04, 2.4772e-04, 3.5034e-01,  ..., 2.0386e-02,
           4.6301e-04, 1.1803e-02]],

         ...,

         [[4.5609e-04, 8.9645e-05, 5.9723e-02,  ..., 1.4465e-02,
           1.9739e-01, 1.9699e-02]],

         [[8.2731e-05, 5.6267e-05, 2.7661e-01,  ..., 3.2288e-02,
           9.7847e-04, 1.6136e-03]],

         [[4.8697e-05, 7.5281e-05, 3.0151e-01,  ..., 1.9882e-02,
           1.6451e-03, 5.3101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6367e-04, 6.9022e-05, 2.3047e-01,  ..., 1.3206e-02,
           2.6369e-04, 3.9711e-03]],

         [[3.8385e-04, 1.1563e-05, 2.4548e-01,  ..., 1.5823e-02,
           3.4027e-02, 5.9700e-04]],

         [[3.8671e-04, 1.6153e-04, 2.7808e-01,  ..., 1.8707e-02,
           5.4626e-03, 6.7616e-04]],

         ...,

         [[1.8895e-05, 9.5963e-06, 8.7402e-02,  ..., 2.6230e-02,
           4.0723e-01, 8.6731e-02]],

         [[4.9591e-05, 7.6473e-05, 1.2341e-01,  ..., 1.4236e-02,
           6.7253e-03, 4.9400e-03]],

         [[2.1505e-04, 3.4213e-05, 1.9373e-01,  ..., 1.2978e-02,
           1.1581e-02, 4.4327e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3910e-06, 3.5763e-07, 1.3256e-03,  ..., 7.2510e-01,
           3.2867e-02, 5.0934e-02]],

         [[2.2709e-05, 7.5281e-05, 1.3611e-01,  ..., 4.2328e-02,
           3.1647e-02, 3.0304e-02]],

         [[1.1551e-04, 5.9795e-04, 3.1030e-01,  ..., 8.3017e-04,
           4.0627e-03, 1.5697e-03]],

         ...,

         [[1.8275e-04, 5.7817e-06, 2.6782e-01,  ..., 2.0004e-02,
           4.2267e-03, 4.3869e-03]],

         [[6.9618e-04, 3.2842e-05, 1.6281e-02,  ..., 2.8760e-01,
           3.2568e-01, 1.9910e-01]],

         [[7.1943e-05, 5.1856e-06, 3.6316e-02,  ..., 1.5906e-01,
           4.6265e-02, 1.5771e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4277e-03, 2.8110e-04, 1.2286e-01,  ..., 1.1299e-02,
           5.4535e-02, 1.0931e-01]],

         [[4.4680e-04, 8.7738e-04, 2.9517e-01,  ..., 1.9932e-03,
           3.2520e-03, 1.5945e-02]],

         [[8.2779e-04, 2.3139e-04, 1.4697e-01,  ..., 6.9946e-02,
           1.1853e-01, 7.2754e-02]],

         ...,

         [[5.1451e-04, 1.0294e-04, 7.6965e-02,  ..., 1.8896e-01,
           1.1377e-01, 3.0200e-01]],

         [[3.9506e-04, 1.0824e-03, 2.6270e-01,  ..., 3.6682e-02,
           4.2908e-02, 3.7842e-02]],

         [[1.1569e-04, 1.0616e-04, 1.1023e-01,  ..., 1.5918e-01,
           1.1646e-01, 7.7576e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.0862e-03, 3.3140e-04, 1.4542e-02,  ..., 8.6304e-02,
           3.8696e-01, 8.1116e-02]],

         [[1.7500e-04, 1.8816e-03, 1.3908e-02,  ..., 1.0826e-02,
           9.6497e-02, 4.3262e-01]],

         [[0.0000e+00, 2.2650e-06, 7.4208e-05,  ..., 4.7684e-07,
           9.0456e-04, 9.9170e-01]],

         ...,

         [[5.9605e-08, 1.3709e-06, 4.4823e-05,  ..., 1.6510e-05,
           4.0321e-03, 9.8682e-01]],

         [[2.7122e-03, 4.2000e-03, 3.9482e-03,  ..., 1.0727e-02,
           2.7390e-02, 1.0727e-02]],

         [[1.7120e-02, 2.8839e-03, 5.1737e-04,  ..., 5.1117e-02,
           1.6083e-02, 5.7953e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2648e-04, 3.7169e-04, 4.1885e-03,  ..., 6.9763e-02,
           9.8389e-02, 5.6592e-01]],

         [[0.0000e+00, 0.0000e+00, 2.3842e-07,  ..., 4.2480e-02,
           9.3701e-01, 1.8265e-02]],

         [[7.1669e-04, 1.6155e-03, 8.3313e-03,  ..., 2.0825e-01,
           3.3997e-02, 2.8183e-02]],

         ...,

         [[2.0618e-03, 5.3062e-03, 6.2485e-03,  ..., 1.7163e-01,
           3.9490e-02, 2.6172e-01]],

         [[2.5940e-03, 1.3725e-02, 2.9510e-02,  ..., 2.2537e-02,
           2.7512e-02, 7.7286e-03]],

         [[1.7061e-03, 6.6414e-03, 6.3400e-03,  ..., 2.7808e-01,
           2.4927e-01, 5.6061e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7791e-02, 5.1575e-03, 2.9449e-02,  ..., 7.4585e-02,
           5.9479e-02, 4.3854e-02]],

         [[1.1368e-03, 5.9652e-04, 2.5711e-03,  ..., 1.1795e-02,
           7.7362e-03, 1.1816e-03]],

         [[3.8681e-03, 6.9618e-04, 8.5640e-04,  ..., 2.4353e-02,
           1.6220e-02, 1.2939e-02]],

         ...,

         [[1.4229e-02, 6.1560e-04, 3.4261e-04,  ..., 1.3708e-01,
           5.2588e-01, 2.6025e-01]],

         [[3.4103e-03, 4.3702e-04, 8.1015e-04,  ..., 5.0507e-02,
           5.4626e-02, 1.8585e-02]],

         [[9.8801e-03, 1.6909e-03, 1.0117e-02,  ..., 8.4045e-02,
           3.5376e-01, 1.0876e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0801e-01, 3.5763e-06, 6.5327e-05,  ..., 5.5771e-03,
           2.6657e-02, 1.6083e-02]],

         [[6.7090e-01, 2.2054e-06, 4.3035e-05,  ..., 7.7477e-03,
           2.1521e-01, 1.5022e-02]],

         [[8.1885e-01, 5.5432e-06, 1.1146e-05,  ..., 8.4778e-02,
           5.9235e-02, 9.8038e-03]],

         ...,

         [[3.2715e-02, 7.3314e-06, 3.3677e-05,  ..., 1.5032e-04,
           2.1398e-05, 6.0501e-03]],

         [[6.9141e-01, 1.1659e-04, 3.9959e-04,  ..., 8.5983e-03,
           1.2268e-02, 1.4015e-02]],

         [[3.4863e-01, 2.0278e-04, 2.3139e-04,  ..., 1.4786e-02,
           3.3722e-02, 3.8055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4570e-01, 1.7285e-06, 8.3447e-07,  ..., 7.0076e-03,
           7.1478e-04, 1.8570e-02]],

         [[7.2559e-01, 7.7486e-07, 5.9605e-08,  ..., 6.9519e-02,
           1.9324e-01, 7.9193e-03]],

         [[9.4385e-01, 1.7977e-04, 9.8884e-05,  ..., 5.3062e-03,
           7.3204e-03, 1.0712e-02]],

         ...,

         [[2.3926e-01, 6.6471e-04, 1.4734e-04,  ..., 3.1738e-02,
           4.0283e-02, 1.2238e-02]],

         [[1.2408e-01, 1.4324e-03, 6.9618e-04,  ..., 4.4060e-03,
           4.4518e-03, 2.9831e-03]],

         [[4.8145e-01, 4.8876e-04, 2.1648e-04,  ..., 1.8347e-01,
           2.2217e-01, 4.4434e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6321e-01, 2.0540e-04, 8.3983e-05,  ..., 9.4910e-02,
           1.0529e-01, 5.4901e-02]],

         [[4.6997e-01, 8.3745e-05, 1.5080e-05,  ..., 2.0667e-01,
           1.3538e-01, 3.8490e-03]],

         [[5.1465e-01, 1.1086e-05, 7.6294e-06,  ..., 2.3181e-01,
           1.2274e-01, 4.5593e-02]],

         ...,

         [[7.7637e-01, 2.2650e-06, 5.9605e-08,  ..., 7.6234e-05,
           1.6093e-06, 2.1045e-01]],

         [[9.0430e-01, 2.8312e-05, 1.1921e-07,  ..., 1.9073e-06,
           0.0000e+00, 1.1688e-02]],

         [[9.6924e-01, 1.0610e-05, 0.0000e+00,  ..., 1.1921e-07,
           0.0000e+00, 4.4785e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2878e-01, 1.9073e-05, 1.1921e-06,  ..., 2.1411e-01,
           5.2588e-01, 2.7039e-02]],

         [[8.3887e-01, 2.6226e-06, 1.2517e-06,  ..., 9.1248e-03,
           1.1480e-04, 2.2598e-02]],

         [[6.6943e-01, 2.2995e-04, 4.9531e-05,  ..., 1.8158e-02,
           2.8076e-03, 2.3575e-02]],

         ...,

         [[5.9375e-01, 4.3750e-04, 3.8028e-05,  ..., 2.8369e-01,
           3.6133e-02, 3.1738e-02]],

         [[8.9355e-01, 2.0051e-04, 4.0591e-05,  ..., 1.4595e-02,
           2.8290e-02, 2.5269e-02]],

         [[2.8320e-01, 2.2137e-04, 3.0100e-05,  ..., 1.7761e-01,
           5.1172e-01, 6.6681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 4.3511e-06, 2.8014e-06,  ..., 3.5614e-02,
           2.1655e-01, 2.1896e-02]],

         [[9.1016e-01, 6.9320e-05, 1.3053e-05,  ..., 9.9945e-03,
           1.2054e-02, 1.7090e-02]],

         [[9.5654e-01, 1.7881e-07, 1.1921e-07,  ..., 2.2278e-03,
           1.7639e-02, 2.0615e-02]],

         ...,

         [[6.3428e-01, 2.3901e-05, 6.5565e-06,  ..., 4.0283e-02,
           1.4746e-01, 3.3661e-02]],

         [[8.8330e-01, 1.7571e-04, 2.6524e-05,  ..., 8.8043e-03,
           5.5969e-02, 3.1261e-03]],

         [[6.8506e-01, 2.0742e-05, 4.6492e-06,  ..., 8.6365e-02,
           1.3196e-01, 3.0502e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 8.9216e-04, 1.6177e-04,  ..., 1.4885e-02,
           1.1147e-02, 7.3967e-03]],

         [[9.7949e-01, 5.0068e-06, 1.5378e-05,  ..., 8.0919e-04,
           2.9602e-03, 1.7428e-04]],

         [[9.1162e-01, 8.5235e-06, 5.5432e-06,  ..., 1.5656e-02,
           6.0242e-02, 9.1982e-04]],

         ...,

         [[7.6074e-01, 4.6730e-05, 1.3113e-05,  ..., 4.1351e-02,
           5.1880e-02, 9.5947e-02]],

         [[8.3203e-01, 4.1723e-07, 4.1127e-06,  ..., 1.0433e-03,
           5.1880e-03, 3.6163e-03]],

         [[7.4219e-01, 1.1325e-06, 1.5497e-06,  ..., 1.7322e-01,
           4.7424e-02, 1.5167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3862e-02, 8.5354e-05, 6.8426e-04,  ..., 1.2787e-02,
           8.4839e-03, 3.3234e-02]],

         [[5.5713e-01, 7.0930e-06, 1.7226e-05,  ..., 5.0903e-02,
           3.5059e-01, 2.3300e-02]],

         [[4.7925e-01, 1.3471e-05, 5.0962e-05,  ..., 6.1279e-02,
           2.7490e-01, 1.7654e-02]],

         ...,

         [[8.3936e-01, 4.0889e-05, 8.3923e-05,  ..., 7.5493e-03,
           4.4918e-04, 2.9430e-03]],

         [[4.0723e-01, 3.5644e-05, 2.8729e-05,  ..., 1.8066e-01,
           3.0737e-01, 4.2267e-02]],

         [[6.2695e-01, 2.3842e-06, 5.3048e-06,  ..., 6.5765e-03,
           6.8176e-02, 5.7869e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3008e-01, 6.7949e-06, 8.6427e-06,  ..., 1.7965e-04,
           3.4273e-05, 6.6910e-03]],

         [[9.0479e-01, 9.0599e-06, 1.4961e-05,  ..., 5.7373e-03,
           1.3443e-02, 1.7061e-03]],

         [[5.8105e-01, 2.3842e-07, 1.1921e-06,  ..., 1.0956e-02,
           1.0014e-03, 1.3855e-02]],

         ...,

         [[8.4668e-01, 1.3411e-04, 6.3360e-05,  ..., 2.6688e-02,
           1.6479e-02, 1.4740e-02]],

         [[5.4688e-01, 1.1736e-04, 9.5010e-05,  ..., 7.1045e-02,
           4.0283e-03, 4.1229e-02]],

         [[3.1763e-01, 5.4646e-04, 1.0848e-04,  ..., 1.9470e-02,
           6.2866e-03, 3.3173e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6270e-01, 5.4836e-06, 2.2352e-05,  ..., 1.6928e-03,
           2.4014e-03, 7.2559e-01]],

         [[3.9600e-01, 5.5075e-04, 1.3485e-03,  ..., 4.7668e-02,
           1.0742e-01, 6.7200e-02]],

         [[7.0068e-01, 3.3545e-04, 8.0109e-04,  ..., 2.0065e-02,
           3.1647e-02, 1.4008e-02]],

         ...,

         [[6.0596e-01, 2.1231e-04, 1.8084e-04,  ..., 1.8082e-02,
           7.9651e-02, 8.2855e-03]],

         [[7.5732e-01, 4.9055e-05, 4.6074e-05,  ..., 1.1505e-02,
           6.4758e-02, 6.5430e-02]],

         [[3.2178e-01, 2.8014e-05, 5.6624e-05,  ..., 5.0293e-02,
           2.4071e-03, 4.4043e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2139e-01, 8.2135e-05, 9.2328e-05,  ..., 8.2458e-02,
           4.9774e-02, 2.0654e-01]],

         [[4.9365e-01, 2.2018e-04, 1.4555e-04,  ..., 3.2043e-02,
           1.3293e-01, 1.0551e-02]],

         [[4.5410e-01, 1.3316e-04, 6.7472e-05,  ..., 6.2408e-02,
           3.7659e-02, 2.5360e-02]],

         ...,

         [[4.6600e-02, 2.0266e-06, 9.8348e-06,  ..., 8.3590e-04,
           1.9016e-03, 8.4570e-01]],

         [[5.5420e-01, 8.2493e-05, 9.3460e-05,  ..., 1.6068e-02,
           2.7542e-03, 1.6296e-02]],

         [[5.6348e-01, 6.3777e-05, 7.8678e-06,  ..., 9.3750e-02,
           2.1045e-01, 1.1932e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.3371e-03, 8.3685e-04,  ..., 3.3783e-02,
           3.4698e-02, 1.8816e-03]],

         [[4.0991e-01, 1.5736e-04, 3.8648e-04,  ..., 3.0106e-02,
           3.5492e-02, 6.9466e-03]],

         [[2.8369e-01, 2.0921e-05, 7.5698e-05,  ..., 1.0262e-02,
           8.2970e-04, 1.3634e-02]],

         ...,

         [[4.6094e-01, 3.3379e-06, 1.7881e-07,  ..., 5.0354e-02,
           4.6802e-01, 1.8448e-02]],

         [[3.4082e-01, 6.7759e-04, 8.2207e-04,  ..., 1.1795e-02,
           1.8482e-03, 7.6904e-03]],

         [[8.0078e-01, 2.5034e-06, 3.5763e-07,  ..., 3.3020e-02,
           1.3086e-01, 3.3264e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 5.2154e-05, 1.2338e-05,  ..., 2.0584e-02,
           1.3039e-02, 3.4241e-02]],

         [[9.7290e-02, 4.8566e-04, 4.1866e-04,  ..., 2.5463e-04,
           1.5569e-04, 3.4389e-03]],

         [[3.6743e-01, 9.0539e-05, 7.3314e-05,  ..., 5.7869e-03,
           8.8959e-03, 7.5912e-03]],

         ...,

         [[4.5093e-01, 2.0266e-06, 1.6689e-06,  ..., 1.2817e-01,
           3.8177e-02, 2.7863e-02]],

         [[3.1250e-01, 7.1526e-07, 1.1921e-07,  ..., 5.5225e-01,
           3.2349e-02, 6.5041e-03]],

         [[4.7314e-01, 1.6570e-05, 2.0742e-05,  ..., 1.0933e-02,
           7.3166e-03, 6.3210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9011e-02, 8.9407e-07, 4.5300e-06,  ..., 1.2007e-03,
           1.6193e-03, 9.3945e-01]],

         [[1.0059e-01, 6.3777e-06, 1.3471e-05,  ..., 6.2317e-02,
           6.5979e-02, 1.9913e-02]],

         [[2.0544e-01, 3.6550e-04, 7.3385e-04,  ..., 2.2125e-02,
           7.0457e-03, 1.1169e-02]],

         ...,

         [[2.9224e-01, 4.6790e-05, 1.5616e-05,  ..., 1.4026e-01,
           1.8079e-01, 2.7863e-02]],

         [[5.1709e-01, 1.1921e-06, 5.3644e-07,  ..., 1.1681e-02,
           2.0969e-04, 6.9656e-03]],

         [[5.7617e-01, 7.1645e-05, 2.6345e-05,  ..., 3.6865e-02,
           5.6877e-03, 1.1429e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0071e-01, 5.9009e-06, 1.9670e-06,  ..., 3.1128e-01,
           5.3809e-01, 1.4282e-02]],

         [[1.9360e-01, 2.2447e-04, 5.8293e-05,  ..., 4.5074e-02,
           5.9021e-02, 2.8214e-02]],

         [[1.5393e-01, 1.1504e-05, 3.3975e-06,  ..., 2.2705e-02,
           2.2446e-02, 8.2703e-02]],

         ...,

         [[6.9482e-01, 2.6226e-05, 1.4067e-05,  ..., 2.6016e-02,
           8.3008e-02, 1.8448e-02]],

         [[4.4922e-01, 1.3053e-05, 1.0669e-05,  ..., 8.2458e-02,
           8.5388e-02, 3.4912e-02]],

         [[6.2939e-01, 7.2896e-05, 2.5213e-05,  ..., 8.8928e-02,
           6.3538e-02, 2.7237e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9775e-01, 9.7227e-04, 1.3180e-03,  ..., 2.4915e-04,
           1.1265e-04, 7.8630e-04]],

         [[3.2275e-01, 6.2525e-05, 7.4267e-05,  ..., 1.6088e-03,
           1.3866e-03, 1.3908e-02]],

         [[3.8940e-01, 2.7418e-06, 4.8876e-06,  ..., 6.5308e-02,
           4.9500e-02, 1.5884e-02]],

         ...,

         [[2.9102e-01, 2.2030e-04, 4.0233e-05,  ..., 2.4078e-02,
           7.8583e-03, 2.3056e-02]],

         [[4.6069e-01, 7.1824e-05, 6.3777e-05,  ..., 5.8234e-05,
           4.8280e-06, 1.3382e-02]],

         [[7.5000e-01, 1.0866e-04, 1.2052e-04,  ..., 1.7500e-03,
           3.1776e-03, 8.5735e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1248e-02, 4.5955e-05, 4.0233e-05,  ..., 3.6144e-04,
           6.0034e-04, 4.0703e-03]],

         [[1.7676e-01, 3.5763e-04, 2.2125e-04,  ..., 3.8910e-03,
           1.9875e-03, 2.3926e-02]],

         [[6.5491e-02, 7.5698e-06, 4.6492e-06,  ..., 2.7954e-02,
           2.1271e-02, 2.4915e-01]],

         ...,

         [[2.0972e-01, 5.5432e-06, 5.7220e-06,  ..., 3.3051e-02,
           1.0675e-01, 1.4275e-02]],

         [[1.7749e-01, 1.7285e-06, 2.2054e-06,  ..., 3.8159e-01,
           3.1641e-01, 5.6274e-02]],

         [[1.1700e-01, 1.1444e-05, 8.1658e-06,  ..., 1.9409e-02,
           9.9304e-02, 3.3862e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9326e-01, 2.0623e-05, 4.2319e-06,  ..., 3.5614e-02,
           1.7365e-02, 4.3640e-02]],

         [[2.8488e-02, 5.4240e-06, 6.2585e-06,  ..., 3.6011e-02,
           1.0002e-02, 1.3046e-02]],

         [[1.4624e-01, 8.6427e-06, 2.1458e-06,  ..., 3.4409e-03,
           3.1143e-02, 7.5439e-01]],

         ...,

         [[1.3940e-01, 2.3842e-04, 8.8751e-05,  ..., 3.1776e-03,
           2.3422e-03, 1.4351e-02]],

         [[2.4829e-01, 1.7357e-04, 9.5129e-05,  ..., 1.7166e-02,
           1.5747e-02, 3.5217e-02]],

         [[3.9380e-01, 3.7050e-04, 3.0470e-04,  ..., 1.2484e-03,
           7.6008e-04, 1.8597e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4905e-01, 2.4736e-05, 1.5914e-05,  ..., 2.0660e-02,
           2.7359e-02, 4.5471e-02]],

         [[7.0654e-01, 8.8573e-05, 6.8545e-06,  ..., 3.8457e-04,
           4.5052e-03, 2.1782e-03]],

         [[9.2383e-01, 7.3910e-05, 2.7955e-05,  ..., 5.6553e-04,
           9.4748e-04, 1.6117e-03]],

         ...,

         [[1.6028e-01, 1.4086e-03, 2.4338e-03,  ..., 1.3153e-02,
           1.0406e-02, 1.6113e-02]],

         [[3.6206e-01, 7.0333e-06, 9.5367e-06,  ..., 1.7102e-01,
           3.0957e-01, 4.7485e-02]],

         [[3.5522e-01, 6.5947e-04, 1.7242e-03,  ..., 2.1011e-02,
           3.2776e-02, 4.5532e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4180e-01, 9.3937e-05, 1.6952e-04,  ..., 5.1498e-03,
           9.3231e-03, 1.7014e-02]],

         [[2.7222e-01, 1.2875e-04, 2.2602e-04,  ..., 3.3741e-03,
           5.9204e-03, 2.8915e-02]],

         [[2.1790e-01, 2.1338e-05, 4.1306e-05,  ..., 7.0038e-03,
           1.2009e-02, 3.1372e-02]],

         ...,

         [[4.5441e-02, 3.5465e-05, 6.3181e-05,  ..., 2.1708e-04,
           1.4460e-04, 2.2842e-02]],

         [[9.3140e-02, 2.7120e-05, 1.9848e-05,  ..., 2.9011e-03,
           1.3704e-03, 5.3070e-02]],

         [[5.4297e-01, 1.7512e-04, 2.3568e-04,  ..., 2.6199e-02,
           2.7023e-02, 3.0151e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3667e-01, 6.7949e-06, 7.9870e-06,  ..., 1.7029e-02,
           2.8961e-02, 5.0812e-02]],

         [[3.0493e-01, 4.6611e-05, 1.1957e-04,  ..., 1.1215e-03,
           1.8196e-03, 4.5586e-03]],

         [[1.3660e-01, 3.6538e-05, 1.0777e-04,  ..., 5.1308e-03,
           3.6469e-02, 3.8916e-01]],

         ...,

         [[3.6841e-01, 4.2379e-05, 5.4181e-05,  ..., 4.8706e-02,
           1.5002e-01, 1.1237e-01]],

         [[2.3621e-01, 2.6822e-06, 1.4067e-05,  ..., 9.0647e-04,
           3.0851e-04, 6.2561e-02]],

         [[2.9321e-01, 1.3185e-04, 2.2864e-04,  ..., 3.0182e-02,
           3.4729e-02, 4.3144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2058e-01, 1.0014e-04, 5.6982e-05,  ..., 3.9551e-02,
           4.8828e-02, 5.3192e-02]],

         [[5.5603e-02, 3.0398e-05, 2.0266e-05,  ..., 8.2825e-02,
           6.0577e-02, 4.6448e-02]],

         [[3.0591e-01, 1.1975e-04, 1.4782e-04,  ..., 6.3972e-03,
           1.3390e-02, 2.5513e-02]],

         ...,

         [[5.9570e-01, 7.2122e-06, 2.3842e-07,  ..., 2.2751e-02,
           2.9492e-01, 7.9407e-02]],

         [[1.9958e-01, 2.1040e-05, 3.6418e-05,  ..., 4.8157e-02,
           2.7008e-02, 7.5745e-02]],

         [[3.0737e-01, 2.0564e-05, 1.2577e-05,  ..., 5.7739e-02,
           5.4785e-01, 3.8452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2392e-03, 8.3447e-07, 2.7418e-06,  ..., 4.8399e-04,
           2.2469e-03, 1.9714e-02]],

         [[2.5040e-02, 1.5497e-06, 4.7088e-06,  ..., 7.9823e-04,
           3.8483e-02, 9.1064e-01]],

         [[2.6962e-02, 2.7239e-05, 1.0455e-04,  ..., 1.8239e-04,
           6.4659e-04, 2.0466e-03]],

         ...,

         [[5.1910e-02, 1.1736e-04, 3.4237e-04,  ..., 7.9956e-03,
           8.0566e-03, 1.1490e-02]],

         [[2.2278e-01, 5.9605e-07, 4.1723e-07,  ..., 1.9089e-02,
           5.6976e-02, 2.8992e-02]],

         [[1.3171e-01, 7.1526e-07, 7.1526e-07,  ..., 2.4695e-01,
           3.2983e-01, 8.0505e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8945e-01, 1.6201e-04, 9.1791e-04,  ..., 1.3977e-02,
           9.3842e-03, 1.3702e-02]],

         [[9.8291e-01, 6.7532e-05, 5.7757e-05,  ..., 3.6645e-04,
           1.8203e-04, 9.9182e-04]],

         [[8.7695e-01, 3.5703e-05, 1.2958e-04,  ..., 1.4290e-02,
           4.3221e-03, 1.1749e-02]],

         ...,

         [[1.4612e-01, 1.4305e-06, 3.4571e-06,  ..., 1.9971e-01,
           2.6465e-01, 5.8105e-02]],

         [[3.9697e-01, 1.9670e-06, 8.9407e-07,  ..., 1.1873e-03,
           2.0676e-03, 1.2836e-03]],

         [[1.3574e-01, 1.1325e-05, 3.5167e-05,  ..., 4.4227e-04,
           1.1497e-02, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1504e-02, 2.8014e-06, 1.2040e-05,  ..., 3.4149e-02,
           4.8157e-02, 4.3152e-02]],

         [[8.0750e-02, 1.0133e-06, 1.1325e-06,  ..., 2.2473e-01,
           6.0156e-01, 4.1077e-02]],

         [[2.4890e-01, 2.1136e-04, 1.9538e-04,  ..., 7.0435e-02,
           1.4062e-01, 7.1838e-02]],

         ...,

         [[3.2007e-01, 2.8014e-06, 1.2279e-05,  ..., 3.1677e-02,
           1.9409e-01, 7.7820e-02]],

         [[6.2402e-01, 1.0848e-05, 3.9697e-05,  ..., 1.6754e-02,
           8.3740e-02, 4.3793e-02]],

         [[4.2773e-01, 5.9605e-06, 2.7716e-05,  ..., 3.4302e-02,
           8.4839e-02, 2.5482e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5221e-02, 3.5763e-06, 1.4305e-06,  ..., 1.6975e-04,
           2.9445e-04, 2.5681e-02]],

         [[2.8491e-01, 6.9141e-06, 2.9206e-06,  ..., 1.0063e-02,
           3.4302e-02, 1.0063e-02]],

         [[2.4829e-01, 4.6372e-05, 1.0163e-04,  ..., 3.7476e-02,
           1.2683e-01, 4.8492e-02]],

         ...,

         [[1.5213e-02, 9.4771e-06, 1.2457e-04,  ..., 3.0112e-04,
           2.0580e-03, 9.4223e-04]],

         [[7.6965e-02, 1.8001e-05, 1.3530e-04,  ..., 1.6844e-04,
           6.1607e-04, 2.9392e-03]],

         [[1.7310e-01, 2.1040e-05, 1.9729e-05,  ..., 1.1597e-02,
           9.7107e-02, 6.3660e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6549e-02, 5.9605e-07, 9.5367e-07,  ..., 8.5742e-01,
           3.6224e-02, 4.2763e-03]],

         [[3.4595e-01, 4.6492e-06, 3.6001e-05,  ..., 4.8370e-03,
           4.6768e-03, 9.3811e-02]],

         [[1.6626e-01, 8.9407e-07, 1.0729e-06,  ..., 9.1064e-02,
           6.7285e-01, 5.0690e-02]],

         ...,

         [[5.7471e-01, 5.1081e-05, 6.2883e-05,  ..., 2.3766e-03,
           5.7487e-03, 1.3206e-02]],

         [[1.5039e-01, 1.7881e-06, 3.2783e-06,  ..., 5.3520e-03,
           5.0278e-03, 5.1460e-03]],

         [[6.0577e-03, 1.0788e-05, 2.4021e-05,  ..., 7.4806e-03,
           1.5381e-01, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0483e-01, 3.0518e-05, 1.3244e-04,  ..., 4.4647e-02,
           2.4512e-01, 5.3009e-02]],

         [[2.8735e-01, 1.4877e-04, 1.8907e-04,  ..., 1.3069e-02,
           8.6746e-03, 1.2375e-02]],

         [[1.3086e-01, 1.9181e-04, 2.9135e-04,  ..., 1.7605e-03,
           8.7967e-03, 3.5686e-03]],

         ...,

         [[5.8887e-01, 6.1321e-04, 3.7766e-04,  ..., 1.0132e-02,
           1.2341e-01, 1.0452e-02]],

         [[9.4299e-02, 1.3745e-04, 4.9496e-04,  ..., 1.8570e-02,
           2.1378e-02, 7.3047e-01]],

         [[6.6895e-01, 1.0252e-05, 2.5749e-05,  ..., 2.8931e-02,
           5.5756e-02, 7.4310e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9854e-01, 4.6432e-05, 4.2772e-04,  ..., 8.4961e-02,
           2.0142e-01, 3.2959e-02]],

         [[4.9774e-02, 3.7551e-06, 1.8299e-05,  ..., 1.3290e-02,
           7.9346e-03, 8.5999e-02]],

         [[4.2798e-01, 3.0684e-04, 7.2050e-04,  ..., 3.9429e-02,
           4.2877e-02, 3.6896e-02]],

         ...,

         [[4.1626e-01, 1.2112e-04, 7.0620e-04,  ..., 2.8038e-03,
           4.8637e-03, 5.8899e-03]],

         [[8.0225e-01, 2.2650e-05, 1.3828e-03,  ..., 4.3564e-03,
           5.8784e-03, 3.8853e-03]],

         [[2.9321e-01, 4.1962e-05, 1.3065e-04,  ..., 3.8376e-03,
           5.0354e-02, 1.6663e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2201e-04, 8.8274e-05, 8.8501e-02,  ..., 1.0023e-03,
           4.3716e-03, 4.7455e-03]],

         [[2.0444e-05, 2.3901e-05, 7.1167e-02,  ..., 1.0460e-02,
           6.8855e-03, 8.5205e-02]],

         [[2.3770e-04, 2.0850e-04, 4.9927e-02,  ..., 2.0981e-02,
           1.4502e-01, 2.7908e-02]],

         ...,

         [[4.7374e-04, 7.7009e-05, 7.7515e-02,  ..., 9.8755e-02,
           1.9800e-01, 1.8591e-01]],

         [[4.1890e-04, 1.7366e-03, 2.3462e-01,  ..., 8.5449e-03,
           1.5717e-02, 3.7632e-03]],

         [[1.9431e-05, 8.7082e-05, 1.6113e-01,  ..., 9.8896e-04,
           8.1301e-04, 1.1740e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5739e-04, 7.9632e-04, 3.0762e-01,  ..., 4.9829e-04,
           2.9240e-03, 2.6302e-03]],

         [[1.7941e-04, 2.7347e-04, 1.0242e-01,  ..., 2.3758e-04,
           1.1092e-04, 1.3723e-03]],

         [[1.6749e-04, 1.0711e-04, 2.8833e-01,  ..., 5.9605e-05,
           2.1172e-03, 7.2598e-05]],

         ...,

         [[4.6277e-04, 4.7374e-04, 2.6440e-01,  ..., 1.1124e-02,
           2.4071e-03, 8.9931e-04]],

         [[2.8670e-05, 1.0788e-04, 2.4902e-01,  ..., 7.4744e-05,
           4.9591e-04, 3.7730e-05]],

         [[8.3387e-05, 3.9220e-04, 1.0626e-01,  ..., 1.0520e-04,
           7.5006e-04, 4.6778e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8203e-04, 1.2445e-04, 1.1975e-01,  ..., 1.0312e-05,
           2.5806e-03, 8.9407e-06]],

         [[7.0333e-04, 3.6895e-05, 2.7393e-01,  ..., 3.2024e-03,
           1.7035e-04, 1.5545e-03]],

         [[8.5449e-04, 7.2956e-04, 2.7441e-01,  ..., 3.6621e-04,
           2.7537e-04, 1.2755e-04]],

         ...,

         [[1.1927e-04, 1.0705e-04, 1.3074e-01,  ..., 9.4681e-03,
           4.8462e-02, 1.1467e-02]],

         [[1.0026e-04, 3.2067e-04, 2.4915e-01,  ..., 5.3358e-04,
           9.8896e-04, 1.8730e-03]],

         [[2.7132e-04, 1.8501e-04, 1.3464e-01,  ..., 1.0767e-03,
           1.2064e-03, 1.2789e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2034e-04, 2.2709e-05, 1.8600e-02,  ..., 8.7402e-02,
           6.3904e-02, 8.4045e-02]],

         [[8.3387e-05, 1.7464e-04, 9.8816e-02,  ..., 7.3700e-03,
           2.2659e-02, 2.4399e-02]],

         [[7.8678e-04, 3.8433e-03, 1.9336e-01,  ..., 1.1158e-03,
           7.6103e-04, 1.7185e-03]],

         ...,

         [[1.7166e-04, 2.8014e-06, 3.1006e-02,  ..., 2.8553e-03,
           5.5218e-04, 2.6520e-02]],

         [[3.0537e-03, 2.9349e-04, 6.2622e-02,  ..., 1.8701e-01,
           2.3096e-01, 5.1331e-02]],

         [[5.7817e-06, 2.4438e-06, 1.5039e-03,  ..., 2.6722e-03,
           1.7109e-03, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2120e-03, 1.3618e-03, 2.2388e-01,  ..., 2.9202e-03,
           3.8185e-03, 4.5135e-02]],

         [[1.2531e-03, 1.3628e-03, 2.5317e-01,  ..., 6.5947e-04,
           1.3628e-03, 2.1698e-02]],

         [[2.8515e-03, 1.4219e-03, 2.7637e-01,  ..., 5.2071e-03,
           9.2239e-03, 7.0457e-03]],

         ...,

         [[1.4629e-03, 1.7965e-04, 2.2766e-01,  ..., 1.7548e-02,
           4.1046e-03, 9.6375e-02]],

         [[3.1972e-04, 8.2922e-04, 1.0864e-01,  ..., 5.3345e-02,
           1.0779e-01, 2.2644e-01]],

         [[3.1590e-05, 3.4273e-05, 1.5747e-02,  ..., 1.1011e-01,
           3.8135e-01, 4.3457e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[7.7546e-05, 5.2166e-04, 6.7635e-03,  ..., 7.7393e-02,
           5.5420e-01, 1.6394e-01]],

         [[1.0413e-04, 5.6314e-04, 7.7705e-03,  ..., 1.3641e-02,
           4.4739e-02, 2.4182e-01]],

         [[0.0000e+00, 1.7881e-07, 2.7061e-05,  ..., 3.4750e-05,
           8.5068e-03, 9.8340e-01]],

         ...,

         [[0.0000e+00, 2.3842e-07, 4.7088e-06,  ..., 2.0146e-04,
           2.4529e-03, 9.8975e-01]],

         [[4.3373e-03, 5.2299e-03, 8.6212e-03,  ..., 1.7151e-02,
           1.5137e-02, 2.2018e-02]],

         [[1.9958e-02, 8.8577e-03, 2.4605e-03,  ..., 2.6443e-02,
           3.5034e-02, 2.9053e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6652e-05, 1.4329e-04, 4.3182e-03,  ..., 1.6556e-02,
           6.5491e-02, 4.1382e-01]],

         [[0.0000e+00, 1.1921e-07, 1.3113e-06,  ..., 1.5173e-01,
           7.7051e-01, 6.8359e-02]],

         [[9.6083e-04, 2.9583e-03, 1.4114e-02,  ..., 7.7972e-03,
           1.0994e-02, 2.2552e-02]],

         ...,

         [[1.3113e-05, 3.0613e-04, 6.3801e-04,  ..., 2.1851e-01,
           4.4849e-01, 2.3999e-01]],

         [[2.7447e-03, 8.3542e-03, 2.5436e-02,  ..., 1.1734e-02,
           7.3433e-03, 9.2850e-03]],

         [[1.6861e-03, 5.7487e-03, 6.7711e-03,  ..., 8.2520e-02,
           2.5970e-02, 2.3636e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1768e-02, 4.8370e-03, 1.3466e-02,  ..., 6.7871e-02,
           4.6997e-02, 5.8502e-02]],

         [[1.3371e-03, 4.9019e-04, 1.9379e-03,  ..., 6.2332e-03,
           6.2418e-04, 3.9291e-03]],

         [[4.6539e-03, 9.3460e-04, 1.0843e-03,  ..., 1.6891e-02,
           1.2260e-02, 1.5259e-02]],

         ...,

         [[4.9257e-04, 2.2087e-03, 3.8452e-03,  ..., 6.9763e-02,
           7.2705e-01, 1.7273e-01]],

         [[3.4165e-04, 4.4823e-05, 4.4107e-05,  ..., 8.3618e-02,
           2.0966e-02, 2.7145e-02]],

         [[3.3951e-03, 9.5034e-04, 1.7380e-02,  ..., 1.8646e-02,
           2.4506e-02, 3.5449e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7354e-01, 1.8477e-06, 1.4305e-06,  ..., 2.2202e-03,
           7.9346e-03, 1.8740e-03]],

         [[8.6914e-01, 3.3379e-06, 3.6359e-06,  ..., 1.3893e-02,
           2.9297e-02, 7.0620e-04]],

         [[6.0352e-01, 1.9670e-06, 7.9870e-06,  ..., 6.5613e-02,
           2.4147e-03, 1.3293e-01]],

         ...,

         [[4.7803e-01, 2.3949e-04, 8.7357e-04,  ..., 1.0365e-04,
           1.3649e-05, 1.0834e-01]],

         [[4.2871e-01, 5.3024e-04, 6.4993e-04,  ..., 1.8051e-02,
           1.4389e-02, 5.4718e-02]],

         [[4.8560e-01, 4.0245e-04, 8.5354e-04,  ..., 1.2787e-02,
           7.7286e-03, 1.6678e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6377e-01, 3.7432e-04, 9.4175e-06,  ..., 2.7065e-03,
           1.5855e-05, 4.4647e-02]],

         [[9.2920e-01, 9.7871e-05, 1.6689e-06,  ..., 1.5198e-02,
           2.1362e-02, 3.2654e-02]],

         [[9.4434e-01, 2.8300e-04, 2.3818e-04,  ..., 4.0531e-04,
           9.7942e-04, 6.0349e-03]],

         ...,

         [[4.0112e-01, 1.9801e-04, 1.2034e-04,  ..., 1.4877e-02,
           4.2381e-03, 1.7212e-02]],

         [[1.7151e-01, 1.6737e-03, 5.4455e-04,  ..., 5.5199e-03,
           3.7403e-03, 4.8027e-03]],

         [[8.0273e-01, 1.2579e-03, 3.8671e-04,  ..., 6.8115e-02,
           3.4973e-02, 1.5015e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2695e-01, 1.9729e-04, 8.6546e-05,  ..., 1.0858e-01,
           7.5256e-02, 3.2990e-02]],

         [[6.5332e-01, 9.0897e-05, 1.0252e-05,  ..., 7.1594e-02,
           2.3300e-02, 3.9154e-02]],

         [[4.4580e-01, 1.7464e-05, 3.1590e-05,  ..., 2.6779e-02,
           4.7424e-02, 4.5288e-02]],

         ...,

         [[7.9590e-01, 1.1665e-04, 1.1325e-06,  ..., 4.7684e-07,
           1.6809e-05, 1.9702e-01]],

         [[8.6279e-01, 5.3453e-04, 5.6028e-06,  ..., 0.0000e+00,
           2.9802e-07, 3.8391e-02]],

         [[9.7168e-01, 3.8266e-04, 2.3842e-07,  ..., 0.0000e+00,
           5.9605e-08, 8.5754e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 8.9407e-06, 1.8477e-06,  ..., 1.9287e-01,
           4.2700e-01, 1.8784e-02]],

         [[9.4727e-01, 9.5963e-06, 1.8477e-06,  ..., 3.1590e-04,
           6.1750e-04, 1.5167e-02]],

         [[7.7539e-01, 1.3232e-04, 5.0843e-05,  ..., 1.8845e-03,
           4.7836e-03, 1.4397e-02]],

         ...,

         [[8.8672e-01, 2.8992e-04, 2.4199e-05,  ..., 2.3518e-03,
           2.9999e-02, 6.9122e-03]],

         [[6.5674e-01, 4.3392e-04, 6.8247e-05,  ..., 5.0201e-02,
           1.5312e-02, 1.3110e-01]],

         [[8.3984e-01, 8.5592e-05, 7.7665e-05,  ..., 2.2869e-03,
           1.5297e-03, 1.3318e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0732e-01, 8.8215e-06, 3.6955e-06,  ..., 2.2424e-01,
           4.0497e-02, 9.8999e-02]],

         [[6.7578e-01, 4.6194e-05, 1.3113e-05,  ..., 6.6490e-03,
           2.5635e-02, 1.0828e-01]],

         [[7.4658e-01, 6.8545e-06, 1.2517e-06,  ..., 7.7591e-03,
           1.2634e-01, 7.5195e-02]],

         ...,

         [[4.5386e-01, 5.1916e-05, 3.7372e-05,  ..., 1.4221e-01,
           7.3059e-02, 4.7729e-02]],

         [[8.3594e-01, 1.3900e-04, 9.2208e-05,  ..., 1.7502e-02,
           4.4441e-03, 9.5520e-03]],

         [[7.4609e-01, 3.6716e-05, 1.5557e-05,  ..., 6.8970e-02,
           2.1912e-02, 2.4536e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3760e-01, 1.4038e-03, 6.4993e-04,  ..., 4.6005e-03,
           4.3907e-03, 4.5128e-03]],

         [[9.1211e-01, 4.9591e-05, 8.4341e-05,  ..., 5.5771e-03,
           1.2989e-03, 4.1122e-03]],

         [[7.8711e-01, 4.4107e-05, 1.1683e-05,  ..., 5.6488e-02,
           3.0945e-02, 1.2405e-02]],

         ...,

         [[9.1162e-01, 4.0174e-05, 2.0444e-05,  ..., 3.0403e-03,
           2.8458e-03, 1.9852e-02]],

         [[6.4795e-01, 4.1723e-06, 3.3200e-05,  ..., 2.0046e-03,
           5.5389e-03, 7.8003e-02]],

         [[9.6289e-01, 5.6624e-06, 4.5896e-06,  ..., 3.4313e-03,
           2.7905e-03, 7.6981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3184e-01, 1.7512e-04, 7.2575e-04,  ..., 1.1301e-03,
           1.6975e-03, 1.5869e-02]],

         [[8.8477e-01, 1.3769e-05, 1.8537e-05,  ..., 1.8066e-02,
           1.9577e-02, 5.2368e-02]],

         [[4.7827e-01, 2.8610e-05, 3.5810e-04,  ..., 6.6833e-02,
           2.3911e-02, 1.5587e-02]],

         ...,

         [[4.0259e-01, 1.3697e-04, 2.2399e-04,  ..., 1.0204e-03,
           3.1929e-03, 1.4563e-03]],

         [[7.9443e-01, 8.5175e-05, 3.9577e-05,  ..., 1.2039e-02,
           1.1337e-02, 6.7139e-02]],

         [[6.2793e-01, 1.4961e-05, 1.7583e-05,  ..., 7.1678e-03,
           1.0109e-03, 1.8848e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0419e-04, 1.4985e-04,  ..., 3.8028e-05,
           4.5896e-06, 1.1719e-02]],

         [[8.6084e-01, 2.2888e-05, 3.6538e-05,  ..., 9.7275e-03,
           5.7983e-03, 1.5366e-02]],

         [[8.9160e-01, 1.5318e-05, 2.2471e-05,  ..., 6.0349e-03,
           7.9823e-04, 1.1253e-02]],

         ...,

         [[9.2969e-01, 7.7367e-05, 5.0545e-05,  ..., 3.0117e-03,
           3.2959e-03, 7.5722e-03]],

         [[8.0762e-01, 1.2183e-04, 1.2088e-04,  ..., 9.9277e-04,
           2.4529e-03, 2.6302e-03]],

         [[6.2891e-01, 9.8324e-04, 3.1304e-04,  ..., 2.9011e-03,
           3.3321e-03, 1.0254e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2290e-01, 6.5565e-07, 1.3828e-05,  ..., 7.4482e-04,
           1.5137e-02, 7.5391e-01]],

         [[8.8281e-01, 3.3212e-04, 5.7316e-04,  ..., 3.6793e-03,
           3.1643e-03, 7.0572e-04]],

         [[7.1191e-01, 5.1641e-04, 7.3957e-04,  ..., 1.0872e-02,
           1.7685e-02, 1.3580e-02]],

         ...,

         [[6.3037e-01, 9.6321e-05, 1.9836e-04,  ..., 6.8436e-03,
           7.3166e-03, 1.9951e-03]],

         [[7.7197e-01, 7.7605e-05, 3.5226e-05,  ..., 1.2138e-02,
           5.7678e-02, 5.1849e-02]],

         [[2.4426e-01, 1.2922e-04, 1.8871e-04,  ..., 2.2812e-03,
           1.9821e-02, 5.9521e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3711e-01, 3.7193e-05, 4.2498e-05,  ..., 6.8321e-03,
           1.3245e-02, 3.8647e-01]],

         [[7.3828e-01, 9.5189e-05, 1.1486e-04,  ..., 9.5978e-03,
           1.9852e-02, 7.1045e-02]],

         [[3.0469e-01, 7.9775e-04, 1.4472e-04,  ..., 3.6530e-02,
           4.7760e-02, 1.9699e-02]],

         ...,

         [[9.5215e-02, 3.4571e-06, 7.9870e-06,  ..., 1.3947e-05,
           2.1279e-04, 8.9795e-01]],

         [[8.1445e-01, 6.5804e-05, 5.6922e-05,  ..., 7.8726e-04,
           5.0545e-03, 3.3142e-02]],

         [[9.1016e-01, 1.4532e-04, 3.9577e-05,  ..., 1.7319e-02,
           1.9684e-03, 1.1055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2695e-01, 1.4067e-03, 1.4009e-03,  ..., 1.6510e-02,
           3.4885e-03, 2.7649e-02]],

         [[8.5840e-01, 7.4863e-05, 1.1772e-04,  ..., 6.0272e-03,
           5.1231e-03, 3.3905e-02]],

         [[6.4600e-01, 3.0971e-04, 1.1003e-04,  ..., 3.3512e-03,
           1.4696e-03, 1.0208e-02]],

         ...,

         [[6.1768e-01, 5.1498e-05, 8.6427e-06,  ..., 1.6953e-02,
           3.0103e-01, 6.1218e-02]],

         [[3.8940e-01, 5.2071e-04, 5.6982e-04,  ..., 5.5962e-03,
           1.4549e-02, 1.4629e-03]],

         [[8.1152e-01, 4.1068e-05, 1.4007e-05,  ..., 2.1667e-02,
           1.4062e-01, 2.3407e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5439e-01, 7.7784e-05, 1.5914e-05,  ..., 5.7755e-03,
           2.1820e-02, 1.8723e-02]],

         [[8.1360e-02, 1.7233e-03, 1.5736e-03,  ..., 1.6856e-04,
           1.7166e-03, 4.7455e-03]],

         [[6.2256e-01, 2.5249e-04, 2.1517e-04,  ..., 2.6779e-03,
           1.1047e-02, 2.6215e-02]],

         ...,

         [[4.6875e-01, 1.8775e-05, 5.3048e-06,  ..., 3.5583e-02,
           7.1350e-02, 1.4380e-01]],

         [[7.4414e-01, 7.1526e-06, 2.4438e-06,  ..., 2.5055e-02,
           5.1483e-02, 7.8674e-02]],

         [[3.4204e-01, 5.6803e-05, 5.9783e-05,  ..., 7.2060e-03,
           1.9516e-02, 3.5736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0444e-02, 1.1921e-07, 4.7684e-07,  ..., 1.6439e-04,
           2.2697e-03, 9.1016e-01]],

         [[3.5132e-01, 2.9325e-05, 4.9710e-05,  ..., 6.6650e-02,
           9.0210e-02, 8.2642e-02]],

         [[1.8896e-01, 4.5991e-04, 7.5245e-04,  ..., 6.6910e-03,
           1.8173e-02, 3.3020e-02]],

         ...,

         [[4.3311e-01, 1.7190e-04, 1.3340e-04,  ..., 8.5938e-02,
           9.1309e-02, 6.3553e-03]],

         [[4.3311e-01, 7.6294e-06, 1.4305e-06,  ..., 1.5710e-01,
           2.6260e-02, 3.5858e-02]],

         [[1.7578e-01, 1.2076e-04, 7.2122e-06,  ..., 4.1771e-03,
           2.9144e-03, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2646e-01, 2.5332e-05, 2.1636e-05,  ..., 7.7515e-02,
           1.4877e-02, 4.0039e-02]],

         [[2.3157e-01, 1.1045e-04, 4.2319e-05,  ..., 3.1543e-01,
           8.7585e-02, 2.6291e-02]],

         [[7.1436e-01, 7.4267e-05, 9.0659e-05,  ..., 1.0071e-02,
           1.5594e-02, 4.6753e-02]],

         ...,

         [[8.7256e-01, 3.8564e-05, 1.6272e-05,  ..., 2.2186e-02,
           2.0279e-02, 7.8201e-03]],

         [[8.1738e-01, 2.6643e-05, 5.4896e-05,  ..., 2.9716e-03,
           7.5722e-03, 4.4861e-02]],

         [[5.3271e-01, 1.0490e-04, 3.1650e-05,  ..., 1.0327e-01,
           1.1084e-01, 1.0391e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5747e-01, 8.1015e-04, 1.1702e-03,  ..., 4.0650e-04,
           4.7779e-04, 1.9970e-03]],

         [[1.5698e-01, 1.2231e-04, 1.0985e-04,  ..., 6.5088e-04,
           7.1831e-03, 3.9307e-02]],

         [[6.6260e-01, 1.4126e-05, 3.7968e-05,  ..., 2.5101e-02,
           7.7576e-02, 2.4796e-02]],

         ...,

         [[8.3789e-01, 6.9022e-05, 1.6749e-05,  ..., 4.0588e-03,
           8.5926e-04, 2.2720e-02]],

         [[9.2578e-01, 2.6643e-05, 1.7703e-05,  ..., 4.7684e-07,
           4.7684e-07, 1.5495e-02]],

         [[5.9473e-01, 2.2876e-04, 4.1580e-04,  ..., 2.8992e-02,
           1.3115e-02, 5.6915e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8196e-02, 1.7571e-04, 1.0574e-04,  ..., 4.3907e-03,
           1.3847e-02, 2.3376e-02]],

         [[1.5723e-01, 3.4380e-04, 1.5497e-04,  ..., 5.1308e-03,
           4.6814e-02, 4.7180e-02]],

         [[5.1562e-01, 4.2915e-05, 2.7597e-05,  ..., 1.1215e-02,
           1.5091e-02, 1.8530e-01]],

         ...,

         [[2.5732e-01, 5.1260e-06, 7.6294e-06,  ..., 2.2180e-01,
           7.1472e-02, 6.1722e-03]],

         [[7.0557e-01, 9.0599e-06, 6.3777e-06,  ..., 1.8799e-02,
           2.7145e-02, 1.2854e-01]],

         [[1.9397e-01, 1.2398e-05, 8.9407e-06,  ..., 2.4460e-02,
           5.3436e-02, 6.5088e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-01, 6.2168e-05, 1.4305e-05,  ..., 1.1780e-01,
           4.9103e-02, 2.6276e-02]],

         [[2.0850e-01, 3.5167e-05, 6.6757e-05,  ..., 4.5105e-02,
           1.0651e-01, 4.8004e-02]],

         [[2.5708e-01, 1.6987e-05, 3.5763e-06,  ..., 1.3494e-03,
           3.3913e-03, 7.2119e-01]],

         ...,

         [[7.8979e-02, 2.2447e-04, 2.1219e-05,  ..., 2.4498e-05,
           7.2002e-05, 5.2612e-02]],

         [[3.4521e-01, 8.4400e-05, 3.7491e-05,  ..., 1.2138e-02,
           2.5009e-02, 4.8294e-03]],

         [[1.4075e-01, 7.4744e-05, 4.2498e-05,  ..., 1.9501e-02,
           6.5979e-02, 1.4725e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5254e-01, 2.6166e-05, 1.4067e-05,  ..., 2.2369e-02,
           3.4637e-02, 3.8635e-02]],

         [[9.7803e-01, 4.9531e-05, 1.1027e-05,  ..., 5.1260e-06,
           5.0068e-06, 1.7560e-04]],

         [[8.3350e-01, 2.6679e-04, 1.6630e-04,  ..., 2.7943e-04,
           1.1864e-03, 3.2234e-03]],

         ...,

         [[9.8999e-02, 6.6710e-04, 9.6321e-04,  ..., 2.7710e-02,
           1.9501e-02, 3.6407e-02]],

         [[6.5820e-01, 3.2842e-05, 3.2604e-05,  ..., 3.0792e-02,
           3.8940e-02, 2.4750e-02]],

         [[2.6904e-01, 5.6362e-04, 1.0958e-03,  ..., 3.8452e-02,
           4.7485e-02, 3.9062e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 1.2279e-04, 1.0175e-04,  ..., 8.3557e-02,
           8.8989e-02, 2.1973e-02]],

         [[2.7563e-01, 7.7844e-05, 1.1969e-04,  ..., 8.2626e-03,
           1.3939e-02, 7.1899e-02]],

         [[2.1960e-01, 4.5002e-05, 1.0633e-04,  ..., 1.1253e-03,
           3.5496e-03, 1.1543e-02]],

         ...,

         [[3.2227e-02, 2.3842e-04, 2.6608e-04,  ..., 6.5851e-04,
           7.1907e-03, 2.2842e-02]],

         [[8.0200e-02, 2.4140e-05, 1.0371e-05,  ..., 5.1856e-05,
           4.7982e-05, 1.6980e-01]],

         [[4.9634e-01, 1.3280e-04, 1.1903e-04,  ..., 1.3863e-02,
           8.4076e-03, 2.2858e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0391e-01, 3.5286e-05, 2.6941e-05,  ..., 1.1574e-02,
           2.0950e-02, 3.3478e-02]],

         [[2.1912e-01, 4.4048e-05, 7.2062e-05,  ..., 4.8790e-03,
           2.2034e-02, 8.4305e-03]],

         [[4.0649e-01, 9.2268e-05, 1.7238e-04,  ..., 3.4084e-03,
           2.3483e-02, 2.8833e-01]],

         ...,

         [[4.0430e-01, 8.1122e-05, 6.2943e-05,  ..., 4.1962e-02,
           1.0201e-02, 1.9821e-02]],

         [[7.8320e-01, 8.6010e-05, 7.8321e-05,  ..., 7.0155e-05,
           1.0782e-04, 4.0558e-02]],

         [[7.1143e-01, 7.2777e-05, 2.1470e-04,  ..., 9.5291e-03,
           3.2673e-03, 1.4198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6953e-01, 1.6880e-04, 6.9976e-05,  ..., 2.8061e-02,
           4.0680e-02, 5.3894e-02]],

         [[2.8516e-01, 1.6117e-04, 1.5807e-04,  ..., 1.0490e-03,
           2.4338e-02, 8.3618e-02]],

         [[4.2334e-01, 1.8525e-04, 1.0681e-04,  ..., 3.8177e-02,
           3.4760e-02, 1.6937e-02]],

         ...,

         [[5.2637e-01, 8.6069e-05, 1.5438e-05,  ..., 5.4199e-02,
           3.5083e-01, 5.6366e-02]],

         [[2.9395e-01, 1.3053e-05, 4.6730e-05,  ..., 8.4229e-02,
           2.5732e-01, 1.0236e-01]],

         [[6.1377e-01, 1.4496e-04, 8.3566e-05,  ..., 5.0018e-02,
           1.8433e-01, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4399e-02, 4.6492e-06, 1.2398e-05,  ..., 8.8043e-03,
           9.8022e-02, 2.6382e-02]],

         [[3.4839e-01, 8.5175e-05, 1.3888e-04,  ..., 1.2253e-02,
           2.4094e-02, 3.0762e-01]],

         [[7.5989e-02, 2.7120e-05, 1.3745e-04,  ..., 3.1376e-03,
           1.0406e-02, 2.3727e-02]],

         ...,

         [[1.0077e-01, 1.6022e-04, 5.5075e-04,  ..., 8.7357e-03,
           1.9089e-02, 1.5396e-02]],

         [[4.8926e-01, 7.2718e-06, 2.2709e-05,  ..., 5.1361e-02,
           1.3013e-01, 1.7236e-01]],

         [[3.1299e-01, 2.9206e-06, 7.8678e-06,  ..., 8.4900e-02,
           3.5596e-01, 1.6101e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 7.9751e-05, 3.5524e-04,  ..., 4.8248e-02,
           3.4485e-02, 3.7262e-02]],

         [[9.6045e-01, 2.1219e-05, 1.1760e-04,  ..., 5.5027e-04,
           1.7557e-03, 1.2798e-03]],

         [[8.8525e-01, 1.6856e-04, 2.3866e-04,  ..., 6.8665e-03,
           1.3130e-02, 7.3662e-03]],

         ...,

         [[8.5400e-01, 7.9274e-06, 1.2279e-05,  ..., 3.9337e-02,
           1.9165e-02, 2.6611e-02]],

         [[9.1309e-01, 4.3511e-06, 4.4107e-06,  ..., 1.3094e-03,
           8.3256e-04, 6.2485e-03]],

         [[2.7930e-01, 1.0490e-05, 2.3305e-05,  ..., 4.2725e-03,
           2.5574e-02, 6.1963e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3196e-01, 5.4240e-06, 2.4319e-05,  ..., 4.1534e-02,
           7.8857e-02, 2.7295e-01]],

         [[7.2070e-01, 2.5868e-05, 1.6987e-05,  ..., 1.0065e-01,
           1.9440e-02, 1.6632e-02]],

         [[1.2988e-01, 1.3173e-04, 7.8261e-05,  ..., 9.2102e-02,
           6.2561e-02, 3.3234e-02]],

         ...,

         [[6.4307e-01, 1.3530e-05, 5.1439e-05,  ..., 2.6123e-02,
           4.7699e-02, 4.3762e-02]],

         [[4.9634e-01, 1.7166e-05, 6.6996e-05,  ..., 1.3159e-01,
           9.9304e-02, 3.1006e-02]],

         [[7.2998e-01, 8.7023e-06, 4.0472e-05,  ..., 4.1199e-02,
           1.9455e-02, 3.1082e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9785e-01, 1.2279e-05, 3.5167e-06,  ..., 5.9433e-03,
           2.8580e-02, 2.2656e-01]],

         [[6.0938e-01, 9.1195e-06, 2.5034e-06,  ..., 4.2725e-03,
           7.2765e-04, 1.1188e-01]],

         [[2.9443e-01, 5.8830e-05, 5.4955e-05,  ..., 3.4332e-02,
           3.3813e-02, 1.5039e-01]],

         ...,

         [[1.0437e-01, 2.5630e-06, 5.5492e-05,  ..., 8.0017e-02,
           2.0874e-02, 6.9519e-02]],

         [[1.2213e-01, 3.8147e-06, 3.2902e-05,  ..., 4.3549e-02,
           3.4454e-02, 4.0894e-02]],

         [[8.4473e-01, 1.7583e-05, 2.7001e-05,  ..., 1.1147e-02,
           1.3229e-02, 1.1864e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 2.8014e-06, 9.5367e-06,  ..., 1.4183e-02,
           2.0920e-02, 5.3223e-02]],

         [[5.3027e-01, 4.8935e-05, 2.2078e-04,  ..., 1.9394e-02,
           1.8909e-01, 9.0027e-02]],

         [[6.5332e-01, 1.6928e-05, 3.1531e-05,  ..., 1.2077e-02,
           1.0748e-01, 2.1875e-01]],

         ...,

         [[4.2505e-01, 2.5034e-05, 1.9252e-05,  ..., 2.8702e-02,
           6.2225e-02, 4.3678e-03]],

         [[5.6543e-01, 1.2755e-05, 6.8545e-06,  ..., 2.1408e-02,
           2.7481e-02, 1.1572e-01]],

         [[1.6833e-01, 9.6262e-05, 3.1567e-04,  ..., 9.5947e-02,
           5.7281e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3037e-01, 7.4506e-06, 1.2815e-05,  ..., 1.1955e-02,
           3.6888e-03, 2.9541e-01]],

         [[7.1582e-01, 9.1612e-05, 7.3314e-05,  ..., 1.8673e-03,
           1.1024e-03, 1.0980e-01]],

         [[4.8779e-01, 2.7239e-05, 6.2346e-05,  ..., 6.0692e-03,
           1.2817e-03, 1.7944e-01]],

         ...,

         [[5.8301e-01, 1.0383e-04, 3.2496e-04,  ..., 5.4230e-02,
           1.9028e-02, 1.0785e-01]],

         [[5.5908e-01, 3.6144e-04, 1.0796e-03,  ..., 1.5617e-02,
           3.4119e-02, 2.1228e-01]],

         [[6.0840e-01, 5.9903e-05, 1.2290e-04,  ..., 1.8372e-02,
           3.4332e-02, 2.4207e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8262e-01, 3.4392e-05, 2.5082e-04,  ..., 1.2527e-02,
           9.9121e-02, 7.5378e-02]],

         [[2.2961e-01, 1.9670e-06, 3.2127e-05,  ..., 5.0735e-03,
           5.1613e-03, 4.4604e-01]],

         [[2.5708e-01, 1.1420e-04, 7.8201e-04,  ..., 1.1060e-01,
           1.3818e-01, 7.2815e-02]],

         ...,

         [[8.3154e-01, 7.1645e-05, 4.1699e-04,  ..., 1.4847e-02,
           1.0925e-02, 4.4060e-03]],

         [[9.5068e-01, 1.7881e-07, 1.5771e-04,  ..., 5.5170e-04,
           1.2836e-03, 7.9489e-04]],

         [[7.3730e-01, 1.6868e-05, 6.4254e-05,  ..., 1.6632e-02,
           1.0010e-02, 8.0795e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1100e-05, 1.1921e-05, 2.1741e-01,  ..., 3.0785e-03,
           7.2098e-03, 5.3284e-02]],

         [[7.0333e-06, 5.0664e-06, 2.0737e-02,  ..., 3.8643e-03,
           1.2402e-01, 5.6229e-03]],

         [[1.5116e-04, 1.3387e-04, 6.7078e-02,  ..., 2.9190e-02,
           3.0838e-02, 1.4626e-02]],

         ...,

         [[5.1796e-05, 9.6560e-06, 1.4612e-01,  ..., 1.4816e-02,
           1.5320e-01, 3.1421e-01]],

         [[2.0134e-04, 4.1902e-05, 3.0518e-01,  ..., 6.0692e-03,
           7.5531e-03, 1.3733e-02]],

         [[1.1206e-05, 1.2517e-06, 2.4475e-01,  ..., 7.4911e-04,
           7.8535e-04, 6.4880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8413e-05, 1.1504e-05, 2.5977e-01,  ..., 1.1806e-03,
           2.8877e-03, 2.9617e-02]],

         [[3.0518e-05, 8.6427e-06, 2.7222e-01,  ..., 1.2636e-04,
           5.1975e-04, 3.2257e-02]],

         [[2.1875e-05, 3.1590e-06, 3.2251e-01,  ..., 1.3866e-03,
           8.3745e-05, 1.4305e-03]],

         ...,

         [[6.8367e-05, 1.4424e-05, 9.4421e-02,  ..., 2.3365e-03,
           1.2503e-03, 5.4131e-03]],

         [[3.6955e-06, 2.0266e-06, 2.8516e-01,  ..., 5.4836e-04,
           2.1815e-04, 7.3357e-03]],

         [[1.2815e-05, 1.0431e-05, 2.9712e-01,  ..., 1.0931e-04,
           7.1406e-05, 5.2643e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0192e-05, 1.7881e-06, 2.0081e-01,  ..., 1.9045e-03,
           1.2219e-05, 3.0914e-02]],

         [[1.7643e-05, 2.9802e-07, 2.3828e-01,  ..., 5.4169e-04,
           1.0738e-03, 1.3514e-03]],

         [[2.0385e-05, 4.0531e-06, 2.2229e-01,  ..., 4.7231e-04,
           1.6069e-04, 3.3112e-03]],

         ...,

         [[2.5868e-04, 3.0518e-05, 2.1802e-01,  ..., 2.3697e-02,
           1.0765e-02, 1.3428e-01]],

         [[8.3447e-06, 2.9206e-06, 1.0846e-01,  ..., 1.0490e-03,
           6.8521e-04, 1.7227e-02]],

         [[3.6836e-05, 2.9206e-06, 2.0776e-01,  ..., 1.5678e-03,
           6.4611e-04, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8651e-05, 4.2319e-06, 2.8549e-02,  ..., 5.9814e-03,
           1.2466e-02, 6.4941e-01]],

         [[8.1658e-06, 3.3498e-05, 1.3940e-01,  ..., 5.3131e-02,
           7.1228e-02, 1.1200e-02]],

         [[9.1612e-05, 2.7108e-04, 3.2861e-01,  ..., 2.4438e-04,
           4.7898e-04, 1.0252e-04]],

         ...,

         [[8.0061e-04, 1.3232e-05, 1.8530e-01,  ..., 8.1158e-04,
           4.5586e-04, 5.0926e-03]],

         [[4.5848e-04, 1.5616e-05, 9.2773e-02,  ..., 1.1040e-02,
           1.4915e-02, 5.5518e-01]],

         [[3.0351e-04, 1.2279e-05, 7.6172e-02,  ..., 1.4481e-02,
           1.0269e-02, 4.5215e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6011e-04, 1.4663e-04, 1.1298e-01,  ..., 5.1544e-02,
           1.4999e-02, 6.4392e-02]],

         [[7.7629e-04, 4.7827e-04, 2.4707e-01,  ..., 5.7869e-03,
           2.0844e-02, 6.9153e-02]],

         [[3.4523e-04, 9.9242e-05, 1.8079e-01,  ..., 1.3695e-02,
           1.2421e-02, 5.5786e-02]],

         ...,

         [[1.1581e-04, 4.4286e-05, 1.6821e-01,  ..., 3.6957e-02,
           8.5938e-02, 1.5808e-01]],

         [[3.6407e-04, 3.9196e-04, 2.7759e-01,  ..., 1.1276e-02,
           1.0996e-03, 7.8308e-02]],

         [[5.2166e-04, 1.4377e-04, 2.2412e-01,  ..., 1.9882e-02,
           3.4912e-02, 1.6785e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[5.6028e-06, 8.2493e-05, 6.6948e-04,  ..., 7.7393e-02,
           4.0552e-01, 4.0552e-01]],

         [[3.4475e-04, 2.2256e-04, 3.7060e-03,  ..., 4.4708e-03,
           2.9144e-02, 4.0234e-01]],

         [[5.9605e-08, 5.9605e-08, 8.9347e-05,  ..., 3.2842e-05,
           2.9583e-03, 9.2969e-01]],

         ...,

         [[0.0000e+00, 4.7684e-07, 6.0201e-06,  ..., 2.7120e-05,
           2.0027e-04, 9.8438e-01]],

         [[3.1185e-03, 2.0123e-03, 7.4806e-03,  ..., 1.5678e-03,
           4.8294e-03, 1.8115e-01]],

         [[1.0628e-02, 3.9101e-03, 5.6314e-04,  ..., 1.1426e-01,
           7.8491e-02, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3167e-04, 1.4906e-03, 1.2283e-02,  ..., 5.5939e-02,
           1.2408e-01, 1.0126e-01]],

         [[0.0000e+00, 0.0000e+00, 3.5763e-07,  ..., 8.5022e-02,
           8.4570e-01, 6.2225e-02]],

         [[3.8319e-03, 5.9357e-03, 1.6388e-02,  ..., 1.0414e-02,
           2.1042e-02, 1.4694e-02]],

         ...,

         [[4.2439e-04, 3.0632e-03, 4.4556e-03,  ..., 3.7988e-01,
           4.7180e-02, 5.7800e-02]],

         [[4.5204e-03, 1.3763e-02, 3.3661e-02,  ..., 1.2100e-02,
           1.1459e-02, 8.7509e-03]],

         [[2.0866e-03, 1.0353e-02, 9.4223e-03,  ..., 1.1481e-01,
           7.3547e-02, 5.3406e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7292e-02, 3.9597e-03, 1.6541e-02,  ..., 3.9093e-02,
           7.6538e-02, 5.0964e-02]],

         [[2.1992e-03, 6.2513e-04, 2.7905e-03,  ..., 8.0919e-04,
           3.9520e-03, 3.8891e-03]],

         [[3.4599e-03, 6.0606e-04, 6.7902e-04,  ..., 1.2367e-02,
           1.4572e-02, 1.3901e-02]],

         ...,

         [[6.3300e-05, 1.5526e-03, 1.8463e-02,  ..., 2.1301e-01,
           4.6167e-01, 2.1301e-01]],

         [[1.1063e-03, 2.5964e-04, 4.3321e-04,  ..., 6.6345e-02,
           3.8391e-02, 1.9608e-02]],

         [[2.7752e-03, 8.6641e-04, 8.8882e-03,  ..., 3.5156e-02,
           3.8696e-01, 1.5881e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3203e-01, 1.1325e-06, 1.6689e-06,  ..., 5.0354e-03,
           3.4389e-03, 6.1073e-03]],

         [[9.4727e-01, 4.8876e-06, 5.6028e-06,  ..., 1.9638e-02,
           1.0666e-02, 4.7226e-03]],

         [[4.3311e-01, 1.0729e-06, 1.1921e-06,  ..., 4.9324e-03,
           7.7324e-03, 5.5580e-03]],

         ...,

         [[6.4990e-01, 1.9169e-04, 1.3828e-03,  ..., 5.2929e-04,
           8.0538e-04, 2.7252e-02]],

         [[5.9766e-01, 1.0711e-04, 2.3782e-04,  ..., 1.9363e-02,
           7.3914e-02, 9.9258e-03]],

         [[5.6787e-01, 2.2542e-04, 4.7064e-04,  ..., 1.1162e-02,
           1.0144e-01, 2.1255e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2163e-01, 6.6161e-06, 6.1989e-06,  ..., 2.6155e-04,
           1.3208e-03, 1.9806e-02]],

         [[1.0132e-01, 3.6955e-06, 5.3048e-06,  ..., 2.3346e-03,
           8.8770e-01, 8.0795e-03]],

         [[9.2773e-01, 7.5519e-05, 4.9174e-05,  ..., 1.5173e-03,
           4.6272e-03, 2.9354e-03]],

         ...,

         [[3.0200e-01, 1.2779e-04, 1.1212e-04,  ..., 1.1284e-02,
           9.7656e-02, 1.4717e-02]],

         [[1.6919e-01, 5.2071e-04, 5.0449e-04,  ..., 4.9210e-03,
           4.3259e-03, 2.2354e-03]],

         [[7.8320e-01, 2.5153e-04, 5.1069e-04,  ..., 1.3084e-02,
           1.1084e-01, 7.4921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0569e-01, 2.3878e-04, 9.8765e-05,  ..., 7.6355e-02,
           7.7271e-02, 1.9211e-02]],

         [[6.8555e-01, 7.3135e-05, 4.0233e-05,  ..., 3.1525e-02,
           1.8518e-01, 1.4923e-02]],

         [[7.9785e-01, 3.6120e-05, 1.6689e-05,  ..., 1.2009e-02,
           1.5182e-02, 2.5215e-03]],

         ...,

         [[8.3643e-01, 1.1921e-05, 3.4571e-06,  ..., 2.0504e-05,
           5.6982e-05, 1.5857e-01]],

         [[9.5947e-01, 9.1672e-05, 1.3554e-04,  ..., 5.9605e-08,
           4.1723e-07, 1.9989e-02]],

         [[9.8242e-01, 3.8505e-05, 4.4703e-06,  ..., 1.7881e-07,
           1.1921e-07, 1.1383e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9568e-01, 2.9206e-06, 6.5565e-07,  ..., 6.1157e-02,
           6.7383e-01, 2.0203e-02]],

         [[8.7988e-01, 4.7088e-06, 1.4305e-06,  ..., 4.3106e-03,
           3.4237e-03, 5.0011e-03]],

         [[7.0557e-01, 5.1403e-04, 3.3450e-04,  ..., 8.5907e-03,
           9.3460e-03, 1.7960e-02]],

         ...,

         [[9.0527e-01, 5.1737e-05, 1.4365e-05,  ..., 2.4673e-02,
           3.2234e-03, 1.7380e-02]],

         [[5.3809e-01, 2.7657e-05, 2.1696e-05,  ..., 1.0948e-02,
           3.6621e-01, 2.3941e-02]],

         [[4.9268e-01, 5.8830e-05, 6.8784e-05,  ..., 6.7902e-03,
           4.5117e-01, 8.9951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1045e-01, 8.1658e-06, 9.1791e-06,  ..., 4.7180e-02,
           4.5837e-02, 4.7684e-03]],

         [[9.0771e-01, 4.6074e-05, 2.4915e-05,  ..., 1.6144e-02,
           1.5327e-02, 6.5613e-03]],

         [[9.5898e-01, 1.6689e-05, 4.6492e-06,  ..., 3.7632e-03,
           2.7985e-02, 1.5936e-03]],

         ...,

         [[7.1240e-01, 2.1446e-04, 7.3850e-05,  ..., 1.5244e-02,
           3.5034e-02, 5.3101e-03]],

         [[8.8232e-01, 1.6189e-04, 2.8133e-05,  ..., 3.0308e-03,
           9.6664e-03, 1.1368e-03]],

         [[9.0137e-01, 1.5616e-05, 1.1861e-05,  ..., 7.4158e-03,
           4.1962e-02, 3.9139e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6069e-01, 9.6369e-04, 6.7282e-04,  ..., 7.4043e-03,
           4.5815e-03, 2.5005e-03]],

         [[9.3799e-01, 7.1406e-05, 8.4162e-05,  ..., 9.1505e-04,
           9.9754e-04, 9.4032e-04]],

         [[8.5352e-01, 4.5896e-05, 1.6510e-05,  ..., 1.0323e-02,
           9.7198e-03, 3.2692e-03]],

         ...,

         [[6.8066e-01, 1.8120e-05, 2.3961e-05,  ..., 6.1989e-03,
           1.6800e-02, 2.2632e-01]],

         [[6.7139e-01, 1.6689e-06, 7.0333e-06,  ..., 1.0056e-02,
           2.7393e-01, 1.8906e-02]],

         [[9.5312e-01, 1.2100e-05, 1.4901e-05,  ..., 3.8013e-03,
           1.0956e-02, 8.8501e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8127e-01, 3.9005e-04, 1.0624e-03,  ..., 1.2306e-02,
           4.2755e-02, 3.8719e-03]],

         [[7.2070e-01, 1.3113e-06, 3.5167e-06,  ..., 9.3231e-03,
           2.4719e-01, 1.0162e-02]],

         [[8.3154e-01, 1.2279e-05, 2.6941e-05,  ..., 1.0445e-02,
           1.5282e-02, 5.8708e-03]],

         ...,

         [[6.9580e-01, 2.4939e-04, 2.6298e-04,  ..., 4.7646e-03,
           2.6798e-03, 5.3120e-04]],

         [[7.7930e-01, 1.3828e-04, 7.6354e-05,  ..., 1.0803e-02,
           4.7058e-02, 1.6165e-03]],

         [[1.7426e-02, 3.5763e-07, 2.9802e-07,  ..., 4.4489e-04,
           9.7363e-01, 6.6147e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3301e-01, 5.4061e-05, 5.5313e-05,  ..., 4.0472e-05,
           3.9935e-04, 5.3024e-03]],

         [[8.5449e-01, 3.0339e-05, 3.6895e-05,  ..., 7.9575e-03,
           4.5166e-02, 1.0786e-03]],

         [[9.3213e-01, 2.0862e-06, 9.1195e-06,  ..., 3.5248e-03,
           1.9236e-03, 4.6883e-03]],

         ...,

         [[9.5801e-01, 9.9719e-05, 8.1062e-05,  ..., 2.5578e-03,
           3.9978e-03, 3.5501e-04]],

         [[3.2300e-01, 1.3781e-04, 1.0484e-04,  ..., 2.0714e-03,
           1.6693e-02, 5.8057e-01]],

         [[7.0947e-01, 2.6870e-04, 1.0985e-04,  ..., 4.9248e-03,
           2.8473e-02, 8.7433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6907e-01, 2.9802e-07, 2.3842e-06,  ..., 9.3746e-04,
           4.7569e-03, 8.1885e-01]],

         [[7.5781e-01, 2.2495e-04, 3.2377e-04,  ..., 4.5509e-03,
           7.4196e-04, 5.0354e-04]],

         [[6.9189e-01, 7.5340e-04, 1.3399e-03,  ..., 1.4023e-02,
           1.2016e-02, 1.7757e-03]],

         ...,

         [[4.0674e-01, 1.9670e-04, 2.9063e-04,  ..., 2.0187e-02,
           4.0512e-03, 2.5654e-03]],

         [[9.0674e-01, 4.5359e-05, 4.0650e-05,  ..., 4.4365e-03,
           1.6647e-02, 1.6022e-02]],

         [[1.2445e-01, 3.3557e-05, 3.3021e-05,  ..., 2.8057e-03,
           4.8706e-02, 7.8809e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1448e-01, 8.5235e-06, 7.0930e-06,  ..., 9.2545e-03,
           3.7720e-02, 6.9775e-01]],

         [[5.1855e-01, 5.5790e-05, 6.2943e-05,  ..., 8.2626e-03,
           2.7979e-01, 1.9699e-02]],

         [[3.8550e-01, 2.2256e-04, 4.5359e-05,  ..., 5.3223e-02,
           2.2583e-02, 1.0872e-02]],

         ...,

         [[9.7839e-02, 1.8477e-06, 2.7418e-06,  ..., 3.8087e-05,
           8.3923e-03, 8.7891e-01]],

         [[8.8135e-01, 2.9206e-05, 4.8757e-05,  ..., 2.2182e-03,
           6.0349e-03, 1.6541e-02]],

         [[8.1104e-01, 1.3769e-04, 3.7313e-05,  ..., 3.0365e-03,
           2.2644e-02, 3.0537e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 1.2159e-03, 9.8419e-04,  ..., 2.8553e-03,
           2.1957e-02, 6.2218e-03]],

         [[8.6914e-01, 5.3287e-05, 9.8825e-05,  ..., 6.2332e-03,
           3.3752e-02, 1.0895e-02]],

         [[5.4395e-01, 3.0375e-04, 1.3483e-04,  ..., 4.1695e-03,
           1.1810e-02, 6.4087e-03]],

         ...,

         [[3.2495e-01, 1.1027e-05, 2.0087e-05,  ..., 3.7689e-02,
           6.1621e-01, 1.8127e-02]],

         [[4.3042e-01, 3.3760e-04, 6.6614e-04,  ..., 1.7746e-02,
           1.9550e-03, 4.3511e-04]],

         [[8.9844e-01, 7.8082e-06, 1.1444e-05,  ..., 1.1841e-02,
           6.3293e-02, 2.3285e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4609e-01, 1.9729e-05, 1.1921e-05,  ..., 1.8539e-02,
           1.0117e-02, 3.0994e-03]],

         [[9.7107e-02, 1.2846e-03, 9.5940e-04,  ..., 5.2452e-03,
           1.3466e-02, 1.2741e-03]],

         [[4.3701e-01, 2.7061e-04, 2.2054e-04,  ..., 3.3966e-02,
           1.1786e-01, 6.1188e-03]],

         ...,

         [[3.3789e-01, 3.8087e-05, 1.8477e-05,  ..., 4.1840e-02,
           5.2490e-01, 3.3508e-02]],

         [[8.7939e-01, 9.9540e-06, 3.2783e-06,  ..., 1.4343e-02,
           7.8003e-02, 1.9150e-02]],

         [[6.7529e-01, 3.0935e-05, 2.3186e-05,  ..., 2.1896e-02,
           3.9001e-02, 3.5210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2642e-02, 1.1921e-07, 2.3842e-07,  ..., 1.2314e-04,
           9.6817e-03, 9.0234e-01]],

         [[1.7065e-01, 6.6161e-06, 1.1206e-05,  ..., 8.2214e-02,
           3.7720e-01, 3.7781e-02]],

         [[1.8262e-01, 3.1662e-04, 5.8079e-04,  ..., 2.6718e-02,
           4.6539e-02, 4.0321e-03]],

         ...,

         [[5.1270e-01, 3.0017e-04, 1.8001e-04,  ..., 4.2175e-02,
           8.8348e-03, 1.5421e-03]],

         [[5.8008e-01, 1.4246e-05, 5.3644e-06,  ..., 7.3547e-02,
           4.6417e-02, 1.4000e-02]],

         [[9.3933e-02, 5.6684e-05, 1.5676e-05,  ..., 1.5812e-03,
           6.4964e-03, 1.6632e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6445e-01, 3.2485e-05, 3.0696e-05,  ..., 1.4771e-02,
           2.8473e-02, 6.4575e-02]],

         [[2.5903e-01, 1.1766e-04, 5.0843e-05,  ..., 1.5283e-01,
           5.6427e-02, 4.9622e-02]],

         [[5.7129e-01, 2.1863e-04, 1.2982e-04,  ..., 4.9133e-03,
           7.6965e-02, 3.8849e-02]],

         ...,

         [[7.8223e-01, 1.8418e-05, 1.5199e-05,  ..., 2.3987e-02,
           3.6865e-02, 8.1940e-03]],

         [[5.8203e-01, 2.4855e-05, 4.4703e-05,  ..., 1.4519e-02,
           2.0837e-01, 8.0185e-03]],

         [[6.8896e-01, 3.4153e-05, 1.2755e-05,  ..., 6.8726e-02,
           5.3024e-03, 2.4738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8662e-01, 1.3447e-03, 1.5888e-03,  ..., 1.0138e-03,
           3.7651e-03, 2.6512e-04]],

         [[2.3267e-01, 1.5700e-04, 1.2994e-04,  ..., 7.5912e-03,
           2.3636e-02, 1.1337e-02]],

         [[2.4829e-01, 5.0664e-06, 1.7166e-05,  ..., 9.8755e-02,
           3.8745e-01, 2.8625e-02]],

         ...,

         [[7.8564e-01, 5.2691e-05, 2.8431e-05,  ..., 1.1778e-03,
           2.7725e-02, 2.5955e-02]],

         [[9.1895e-01, 4.0352e-05, 5.6565e-05,  ..., 2.1458e-06,
           4.5943e-04, 1.0193e-02]],

         [[5.9229e-01, 1.7333e-04, 3.7026e-04,  ..., 6.9351e-03,
           6.5155e-03, 3.3283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5552e-02, 2.4772e-04, 1.2660e-04,  ..., 4.4342e-02,
           3.2715e-02, 2.5616e-03]],

         [[5.3558e-02, 5.4598e-04, 2.0361e-04,  ..., 2.1652e-02,
           3.0289e-02, 3.6736e-03]],

         [[4.5923e-01, 2.7955e-05, 1.7047e-05,  ..., 3.8815e-03,
           2.4207e-01, 1.6638e-01]],

         ...,

         [[2.9932e-01, 4.5896e-06, 5.9009e-06,  ..., 4.8096e-02,
           1.8036e-02, 2.1381e-03]],

         [[5.9912e-01, 6.4373e-06, 2.9206e-06,  ..., 4.3640e-03,
           2.2217e-01, 9.1187e-02]],

         [[1.2622e-01, 5.9605e-06, 2.9802e-06,  ..., 9.8877e-03,
           5.1904e-01, 3.0054e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 1.7524e-05, 4.1723e-06,  ..., 4.8309e-02,
           3.1189e-02, 2.3926e-02]],

         [[1.5002e-01, 5.1916e-05, 8.9705e-05,  ..., 8.4106e-02,
           3.8306e-01, 5.6915e-02]],

         [[2.5977e-01, 7.7486e-06, 2.2054e-06,  ..., 7.1812e-04,
           1.0492e-01, 6.2305e-01]],

         ...,

         [[4.4098e-02, 8.2791e-05, 1.1683e-05,  ..., 1.1683e-05,
           2.9774e-03, 1.0887e-02]],

         [[4.0112e-01, 8.1599e-05, 3.7611e-05,  ..., 3.7323e-02,
           9.9258e-03, 3.4447e-03]],

         [[1.7944e-01, 7.3314e-05, 4.8757e-05,  ..., 8.3435e-02,
           4.2686e-03, 6.0921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5337e-01, 1.7881e-05, 1.4961e-05,  ..., 1.7990e-02,
           3.6621e-02, 3.7811e-02]],

         [[9.3262e-01, 1.2636e-05, 2.5034e-06,  ..., 4.9472e-06,
           9.0361e-05, 1.3512e-02]],

         [[6.8994e-01, 9.4235e-05, 1.1802e-05,  ..., 2.4300e-03,
           4.2992e-03, 2.0996e-02]],

         ...,

         [[2.4829e-01, 7.6008e-04, 1.0147e-03,  ..., 1.1986e-02,
           1.8707e-02, 5.8403e-03]],

         [[7.9395e-01, 2.5570e-05, 2.3663e-05,  ..., 1.1688e-02,
           1.4099e-02, 1.5732e-02]],

         [[2.1973e-01, 3.7575e-04, 5.0354e-04,  ..., 5.0568e-02,
           4.6051e-02, 1.0277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4644e-01, 1.0961e-04, 1.4293e-04,  ..., 8.0994e-02,
           1.0880e-02, 1.1398e-02]],

         [[2.8198e-01, 6.1333e-05, 1.2827e-04,  ..., 1.2985e-02,
           4.6387e-02, 4.0955e-02]],

         [[2.0984e-01, 5.7220e-05, 7.6413e-05,  ..., 1.5774e-03,
           7.1220e-03, 6.3362e-03]],

         ...,

         [[1.0107e-01, 6.7043e-04, 6.7043e-04,  ..., 1.0002e-02,
           5.4932e-02, 2.0859e-02]],

         [[6.1829e-02, 7.2420e-05, 2.4259e-05,  ..., 1.8775e-04,
           5.1270e-02, 8.0627e-02]],

         [[5.1123e-01, 1.5736e-04, 1.2946e-04,  ..., 2.1423e-02,
           9.9640e-03, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9395e-01, 4.3094e-05, 2.5034e-05,  ..., 1.3855e-02,
           9.8450e-02, 6.9275e-02]],

         [[2.7612e-01, 9.3758e-05, 1.3220e-04,  ..., 5.1086e-02,
           3.6224e-02, 1.6846e-02]],

         [[3.8403e-01, 1.9121e-04, 2.8706e-04,  ..., 2.1835e-02,
           1.4026e-01, 1.9165e-01]],

         ...,

         [[4.7095e-01, 9.5069e-05, 9.0182e-05,  ..., 1.2070e-02,
           4.0192e-02, 6.7871e-02]],

         [[6.4307e-01, 3.8981e-05, 1.4949e-04,  ..., 6.5947e-04,
           2.9617e-02, 1.0419e-01]],

         [[4.6826e-01, 5.1022e-05, 1.0842e-04,  ..., 7.3967e-03,
           1.4824e-02, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9958e-01, 1.3340e-04, 5.3644e-05,  ..., 1.8784e-02,
           5.7617e-02, 4.9286e-02]],

         [[4.2480e-01, 2.4772e-04, 2.0099e-04,  ..., 1.1772e-02,
           1.1084e-01, 4.5502e-02]],

         [[4.2261e-01, 3.5930e-04, 2.6655e-04,  ..., 2.8320e-02,
           2.4887e-02, 1.3428e-02]],

         ...,

         [[5.9668e-01, 4.8280e-05, 3.6240e-04,  ..., 3.8422e-02,
           3.1934e-01, 3.3661e-02]],

         [[2.6367e-01, 9.1195e-06, 6.5744e-05,  ..., 8.1665e-02,
           4.1162e-01, 5.6580e-02]],

         [[2.0813e-01, 7.0930e-06, 9.0361e-05,  ..., 1.1925e-02,
           7.3779e-01, 3.5065e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9007e-02, 5.8413e-06, 9.0599e-06,  ..., 9.8145e-02,
           2.2461e-01, 1.1035e-01]],

         [[1.7432e-01, 2.8729e-05, 2.3007e-05,  ..., 3.1185e-03,
           3.4332e-02, 6.6797e-01]],

         [[5.6641e-02, 2.5928e-05, 1.2183e-04,  ..., 1.0078e-02,
           2.9266e-02, 6.1111e-03]],

         ...,

         [[1.5076e-01, 1.7929e-04, 7.3147e-04,  ..., 4.1870e-02,
           3.8574e-02, 1.5343e-02]],

         [[1.7029e-01, 1.7881e-06, 4.1127e-06,  ..., 8.9569e-03,
           7.3096e-01, 5.8167e-02]],

         [[4.7192e-01, 1.0729e-06, 3.3379e-06,  ..., 1.0529e-01,
           3.3203e-01, 4.3732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5381e-01, 6.8605e-05, 2.6512e-04,  ..., 2.3438e-02,
           3.6163e-02, 6.9313e-03]],

         [[9.5947e-01, 3.5763e-05, 1.3006e-04,  ..., 1.0805e-03,
           3.6192e-04, 3.9434e-04]],

         [[8.4863e-01, 2.3520e-04, 1.8895e-04,  ..., 1.3405e-02,
           4.3869e-03, 2.1038e-03]],

         ...,

         [[9.1797e-01, 1.0192e-05, 1.5676e-05,  ..., 9.1400e-03,
           2.1591e-02, 5.0468e-03]],

         [[8.9551e-01, 3.1590e-06, 4.6492e-06,  ..., 6.1131e-04,
           2.6760e-03, 4.0817e-03]],

         [[2.7295e-01, 5.9605e-06, 1.6034e-05,  ..., 3.2520e-03,
           4.0558e-02, 6.2451e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1957e-01, 2.6226e-06, 1.1921e-05,  ..., 6.0120e-02,
           3.6816e-01, 1.5112e-01]],

         [[8.0420e-01, 2.5094e-05, 1.9073e-05,  ..., 1.2794e-02,
           2.0935e-02, 2.2552e-02]],

         [[1.8079e-01, 2.8944e-04, 1.3626e-04,  ..., 5.6427e-02,
           3.5034e-02, 1.1307e-02]],

         ...,

         [[6.5869e-01, 1.2040e-05, 4.5061e-05,  ..., 1.7548e-02,
           6.4209e-02, 3.1036e-02]],

         [[3.8037e-01, 1.7583e-05, 8.7917e-05,  ..., 5.8350e-02,
           1.8143e-02, 3.5126e-02]],

         [[6.9238e-01, 8.1658e-06, 5.1856e-05,  ..., 1.3397e-02,
           3.0670e-02, 5.4230e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9185e-01, 2.8253e-05, 1.5855e-05,  ..., 1.0124e-02,
           1.3440e-01, 1.9861e-01]],

         [[5.4150e-01, 8.8811e-06, 2.1458e-06,  ..., 2.4819e-04,
           2.6535e-02, 1.9763e-01]],

         [[2.1570e-01, 6.3598e-05, 5.2333e-05,  ..., 3.5767e-02,
           6.2286e-02, 1.6931e-01]],

         ...,

         [[8.8196e-02, 1.2517e-06, 2.6345e-05,  ..., 2.1286e-02,
           1.2250e-01, 4.5776e-02]],

         [[9.6619e-02, 4.1127e-06, 3.7014e-05,  ..., 2.8122e-02,
           4.6356e-02, 1.4587e-02]],

         [[9.5312e-01, 6.0201e-06, 1.5676e-05,  ..., 4.6272e-03,
           8.7814e-03, 9.1124e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3594e-01, 1.0133e-06, 8.8811e-06,  ..., 7.3318e-03,
           1.5587e-02, 8.2458e-02]],

         [[4.4019e-01, 1.6689e-05, 1.6630e-04,  ..., 1.0492e-01,
           1.7920e-01, 1.0455e-01]],

         [[1.5759e-01, 2.1458e-06, 6.5923e-05,  ..., 4.0924e-02,
           7.5732e-01, 3.4332e-02]],

         ...,

         [[4.2651e-01, 5.3465e-05, 4.0352e-05,  ..., 6.0486e-02,
           4.4670e-03, 2.1408e-02]],

         [[3.5913e-01, 1.0550e-05, 7.5698e-06,  ..., 1.6037e-02,
           3.2959e-01, 7.9529e-02]],

         [[1.1731e-01, 6.8247e-05, 2.4629e-04,  ..., 1.8066e-02,
           3.3081e-02, 3.5010e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4160e-01, 1.1683e-05, 1.8537e-05,  ..., 5.5733e-03,
           1.1505e-01, 1.2830e-01]],

         [[6.8604e-01, 4.8220e-05, 4.8816e-05,  ..., 3.8776e-03,
           3.3508e-02, 9.0332e-02]],

         [[3.0469e-01, 2.6405e-05, 5.4240e-05,  ..., 3.7460e-03,
           4.5654e-02, 1.9360e-01]],

         ...,

         [[6.7432e-01, 4.3869e-05, 1.2398e-04,  ..., 5.6534e-03,
           8.5754e-02, 3.1052e-02]],

         [[3.8892e-01, 1.6749e-04, 3.9554e-04,  ..., 9.5825e-03,
           1.4389e-02, 4.3384e-01]],

         [[2.9028e-01, 5.0128e-05, 7.7605e-05,  ..., 1.7700e-02,
           4.8608e-01, 1.6028e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2285e-01, 1.1206e-05, 1.1319e-04,  ..., 8.9951e-03,
           8.1863e-03, 3.0487e-02]],

         [[1.2756e-01, 1.3709e-06, 2.6584e-05,  ..., 3.5820e-03,
           3.1567e-01, 1.5564e-01]],

         [[2.7710e-01, 2.3544e-04, 9.5463e-04,  ..., 8.5815e-02,
           6.1798e-02, 8.3191e-02]],

         ...,

         [[7.9639e-01, 5.3525e-05, 3.4785e-04,  ..., 1.8112e-02,
           5.3711e-03, 1.0239e-02]],

         [[8.8184e-01, 2.3842e-07, 2.3127e-04,  ..., 2.9087e-03,
           1.3733e-03, 1.3977e-02]],

         [[5.6152e-01, 2.8670e-05, 1.1957e-04,  ..., 1.3580e-02,
           3.3932e-03, 9.9335e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6093e-05, 7.9870e-06, 1.5479e-01,  ..., 8.5678e-03,
           5.2490e-02, 3.0542e-01]],

         [[6.5565e-06, 2.3246e-06, 1.3466e-02,  ..., 1.2189e-01,
           3.3131e-03, 9.1782e-03]],

         [[1.4305e-04, 1.0079e-04, 5.8319e-02,  ..., 1.3527e-02,
           1.2131e-02, 1.3452e-01]],

         ...,

         [[3.7789e-05, 1.0014e-05, 1.3318e-01,  ..., 9.3689e-02,
           2.4683e-01, 1.7371e-01]],

         [[1.1557e-04, 1.7822e-05, 2.7783e-01,  ..., 6.6109e-03,
           7.4615e-03, 1.6556e-02]],

         [[3.2187e-06, 1.7881e-07, 3.2812e-01,  ..., 5.7697e-04,
           2.6932e-02, 7.6111e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3896e-05, 1.0252e-05, 2.6953e-01,  ..., 1.2436e-03,
           1.4977e-02, 1.3660e-01]],

         [[3.7491e-05, 3.8147e-06, 1.6736e-01,  ..., 2.1992e-03,
           4.0070e-02, 2.9370e-01]],

         [[1.7762e-05, 6.5565e-07, 2.9785e-01,  ..., 1.3649e-04,
           1.1168e-03, 7.2937e-02]],

         ...,

         [[2.9862e-05, 2.5630e-06, 6.4209e-02,  ..., 2.3956e-03,
           2.3308e-03, 5.1208e-02]],

         [[2.3246e-06, 7.1526e-07, 2.5708e-01,  ..., 4.6992e-04,
           5.8174e-03, 1.3550e-01]],

         [[2.7895e-05, 1.1802e-05, 3.0078e-01,  ..., 2.3544e-04,
           8.3828e-04, 4.6844e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9009e-06, 8.3447e-07, 1.5015e-01,  ..., 5.2035e-05,
           2.3026e-02, 1.2744e-01]],

         [[7.7486e-06, 5.9605e-08, 1.6296e-01,  ..., 5.7106e-03,
           6.4325e-04, 9.1431e-02]],

         [[8.7619e-06, 2.0862e-06, 1.5698e-01,  ..., 5.5742e-04,
           1.9760e-03, 9.9792e-02]],

         ...,

         [[9.3222e-05, 5.2452e-06, 1.8616e-01,  ..., 7.7744e-03,
           9.5825e-02, 2.0605e-01]],

         [[3.8147e-06, 1.2517e-06, 7.6416e-02,  ..., 2.5654e-03,
           5.7564e-03, 1.1298e-01]],

         [[2.2709e-05, 1.3113e-06, 1.6699e-01,  ..., 2.9526e-03,
           3.1250e-02, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2338e-05, 1.2517e-06, 1.8723e-02,  ..., 4.5166e-03,
           7.1387e-01, 1.4734e-01]],

         [[7.3314e-06, 7.1049e-05, 8.3740e-02,  ..., 9.4116e-02,
           2.2186e-02, 1.9775e-02]],

         [[3.6716e-05, 4.0388e-04, 3.2886e-01,  ..., 1.1263e-03,
           4.3297e-04, 5.3596e-04]],

         ...,

         [[1.2903e-03, 4.3988e-05, 2.5684e-01,  ..., 1.8525e-04,
           1.0004e-03, 2.7599e-03]],

         [[9.3222e-04, 5.3704e-05, 6.5613e-02,  ..., 6.3820e-03,
           2.3633e-01, 4.5728e-01]],

         [[2.1017e-04, 1.7405e-05, 6.6223e-02,  ..., 2.3022e-03,
           2.0569e-01, 3.6523e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4211e-04, 7.9453e-05, 8.9294e-02,  ..., 1.6129e-02,
           1.9989e-02, 4.6844e-02]],

         [[6.9666e-04, 3.6168e-04, 2.0789e-01,  ..., 2.7054e-02,
           3.2135e-02, 6.8787e-02]],

         [[3.5334e-04, 1.5354e-04, 1.4697e-01,  ..., 9.0179e-03,
           3.5797e-02, 6.4331e-02]],

         ...,

         [[1.0347e-04, 2.1040e-05, 3.7109e-02,  ..., 5.5618e-03,
           5.8545e-01, 2.4402e-01]],

         [[3.1710e-04, 5.6076e-04, 3.3960e-01,  ..., 3.5930e-04,
           4.8065e-03, 3.9001e-02]],

         [[4.2248e-04, 1.3614e-04, 2.8540e-01,  ..., 2.9793e-03,
           1.8539e-02, 3.1525e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[7.8082e-06, 3.2961e-05, 2.0390e-03,  ..., 6.1462e-02,
           5.1465e-01, 1.1487e-01]],

         [[1.1110e-04, 3.3855e-05, 7.7105e-04,  ..., 2.8931e-02,
           1.5637e-01, 5.1270e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 8.3447e-07,
           2.4700e-03, 9.9658e-01]],

         ...,

         [[0.0000e+00, 5.9605e-08, 1.7881e-07,  ..., 2.0015e-04,
           6.6261e-03, 9.8340e-01]],

         [[9.1705e-03, 7.6065e-03, 3.8242e-03,  ..., 2.0676e-02,
           2.6550e-02, 4.3335e-03]],

         [[8.9722e-03, 3.6259e-03, 1.1768e-03,  ..., 1.6769e-02,
           1.6769e-02, 5.0049e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5048e-04, 3.5048e-04, 3.0270e-03,  ..., 2.6154e-02,
           9.4177e-02, 5.4199e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 1.4624e-01,
           7.9053e-01, 5.3802e-02]],

         [[5.2929e-04, 1.8759e-03, 1.2238e-02,  ..., 3.8879e-02,
           3.6530e-02, 7.2632e-02]],

         ...,

         [[5.6725e-03, 7.8735e-03, 1.6281e-02,  ..., 8.6517e-03,
           1.8448e-02, 8.9417e-02]],

         [[2.1458e-03, 6.3667e-03, 2.9907e-02,  ..., 1.0788e-02,
           2.1790e-02, 1.1002e-02]],

         [[2.9049e-03, 7.7744e-03, 1.4076e-02,  ..., 5.7434e-02,
           6.7139e-02, 1.1786e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6519e-02, 5.5084e-03, 2.9541e-02,  ..., 2.6901e-02,
           1.9836e-02, 2.6901e-02]],

         [[8.1158e-04, 3.4904e-04, 1.8291e-03,  ..., 3.8719e-03,
           4.7836e-03, 3.5820e-03]],

         [[3.1414e-03, 5.6791e-04, 6.5327e-04,  ..., 1.9547e-02,
           1.8723e-02, 1.6586e-02]],

         ...,

         [[1.0204e-04, 1.9670e-04, 1.0170e-02,  ..., 1.3708e-01,
           3.7866e-01, 1.5784e-01]],

         [[4.5776e-03, 6.2180e-04, 2.1877e-03,  ..., 6.7810e-02,
           1.3458e-02, 2.0996e-02]],

         [[2.7542e-02, 5.5962e-03, 1.8494e-02,  ..., 8.8928e-02,
           9.2468e-02, 1.3245e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5791e-01, 5.9605e-08, 4.1723e-07,  ..., 3.7766e-03,
           5.1918e-03, 2.8625e-02]],

         [[8.8818e-01, 1.3113e-06, 5.3048e-06,  ..., 1.3306e-02,
           5.1788e-02, 1.4694e-02]],

         [[9.4629e-01, 1.6689e-06, 9.7752e-06,  ..., 6.9351e-03,
           1.2875e-03, 1.2100e-02]],

         ...,

         [[7.2754e-01, 4.9114e-04, 1.0986e-03,  ..., 2.1946e-04,
           1.1164e-04, 1.9623e-02]],

         [[6.5039e-01, 1.3351e-04, 6.0225e-04,  ..., 5.0354e-02,
           5.3062e-03, 1.5472e-02]],

         [[2.5439e-01, 1.4508e-04, 5.4884e-04,  ..., 4.2261e-01,
           2.9663e-02, 1.8280e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9453e-01, 5.6505e-05, 2.8193e-05,  ..., 7.7391e-04,
           3.0577e-05, 4.7699e-02]],

         [[8.0957e-01, 1.1921e-06, 9.6560e-06,  ..., 6.1554e-02,
           9.1064e-02, 3.5767e-02]],

         [[8.5986e-01, 5.1796e-05, 1.7583e-05,  ..., 8.2016e-04,
           1.3580e-02, 4.0512e-03]],

         ...,

         [[2.8613e-01, 2.2459e-04, 3.4809e-04,  ..., 1.2549e-01,
           4.2053e-02, 2.3682e-02]],

         [[2.0691e-01, 1.9855e-03, 9.0981e-04,  ..., 4.9286e-03,
           2.1172e-03, 3.9139e-03]],

         [[4.8145e-01, 8.8334e-05, 4.9400e-04,  ..., 1.1328e-01,
           3.9764e-02, 1.2512e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0337e-01, 1.9586e-04, 7.2598e-05,  ..., 7.1472e-02,
           3.1708e-02, 3.5553e-02]],

         [[6.9189e-01, 4.9591e-05, 3.2246e-05,  ..., 9.6924e-02,
           5.3162e-02, 8.8562e-02]],

         [[9.1602e-01, 1.4603e-04, 5.3287e-05,  ..., 3.1719e-03,
           7.0496e-03, 1.4214e-02]],

         ...,

         [[9.3311e-01, 6.7949e-06, 2.1458e-06,  ..., 1.1402e-04,
           1.7583e-05, 6.4392e-02]],

         [[9.3359e-01, 3.4571e-05, 3.0160e-05,  ..., 4.2319e-06,
           9.5367e-07, 2.2903e-02]],

         [[9.5508e-01, 9.0003e-06, 1.7881e-06,  ..., 3.6955e-06,
           7.7486e-07, 2.2537e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 4.7088e-06, 3.1590e-06,  ..., 1.9165e-01,
           2.5000e-01, 8.0383e-02]],

         [[8.2275e-01, 2.3246e-06, 1.3709e-06,  ..., 9.7322e-04,
           5.1880e-04, 2.4338e-02]],

         [[8.1006e-01, 1.0234e-04, 4.3988e-05,  ..., 1.1444e-03,
           1.4925e-03, 1.5160e-02]],

         ...,

         [[6.9531e-01, 2.0337e-04, 3.2306e-05,  ..., 1.3466e-02,
           5.5275e-03, 1.9763e-01]],

         [[5.1953e-01, 1.7798e-04, 5.4300e-05,  ..., 6.7322e-02,
           3.6865e-02, 1.7358e-01]],

         [[8.5010e-01, 3.7313e-05, 5.9187e-05,  ..., 6.1707e-02,
           1.0445e-02, 4.5685e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4424e-01, 1.4126e-05, 1.0431e-05,  ..., 3.0533e-02,
           3.9940e-03, 1.5060e-02]],

         [[9.0820e-01, 6.4969e-05, 1.5557e-05,  ..., 1.0170e-02,
           2.8191e-03, 2.0264e-02]],

         [[9.7656e-01, 1.3113e-06, 5.9605e-07,  ..., 9.3002e-03,
           4.8599e-03, 4.4708e-03]],

         ...,

         [[6.4111e-01, 2.6798e-04, 5.2154e-05,  ..., 3.4454e-02,
           9.9411e-03, 2.5864e-02]],

         [[8.6523e-01, 1.3316e-04, 2.0087e-05,  ..., 5.4131e-03,
           2.1696e-04, 6.1226e-03]],

         [[9.5557e-01, 1.5378e-05, 4.2915e-06,  ..., 6.6299e-03,
           4.8904e-03, 9.1858e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2446e-01, 2.4605e-03, 1.2093e-03,  ..., 6.9427e-03,
           1.5459e-03, 1.2283e-02]],

         [[9.2920e-01, 2.0862e-05, 6.0916e-05,  ..., 1.9321e-03,
           2.5196e-03, 4.9133e-03]],

         [[8.8574e-01, 3.4392e-05, 1.1086e-05,  ..., 1.7500e-03,
           1.5440e-03, 1.4061e-02]],

         ...,

         [[8.0566e-01, 3.4571e-05, 3.1233e-05,  ..., 4.3564e-03,
           2.5650e-02, 1.0498e-01]],

         [[7.9688e-01, 4.5300e-06, 9.7156e-06,  ..., 9.7290e-02,
           6.8787e-02, 6.7444e-03]],

         [[8.5791e-01, 5.6028e-06, 1.8001e-05,  ..., 7.1945e-03,
           9.9411e-03, 8.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1792e-01, 1.9073e-04, 7.5865e-04,  ..., 1.0767e-01,
           1.0033e-02, 1.0674e-02]],

         [[8.0322e-01, 1.4901e-06, 7.2718e-06,  ..., 6.2988e-02,
           7.2449e-02, 1.7456e-02]],

         [[8.5840e-01, 3.2187e-06, 8.2254e-06,  ..., 1.4458e-02,
           2.3148e-02, 2.8915e-02]],

         ...,

         [[7.5879e-01, 1.2314e-04, 1.2410e-04,  ..., 2.4090e-03,
           5.2118e-04, 1.1120e-03]],

         [[6.6504e-01, 4.8220e-05, 1.9908e-05,  ..., 4.9896e-02,
           4.2572e-03, 9.2712e-02]],

         [[1.3245e-01, 2.6822e-06, 1.6093e-06,  ..., 7.7441e-01,
           6.2561e-02, 1.4130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6216e-01, 1.2505e-04, 7.3195e-05,  ..., 1.7345e-05,
           4.9472e-06, 3.7117e-03]],

         [[9.0186e-01, 1.4126e-05, 3.1114e-05,  ..., 2.4246e-02,
           1.2722e-03, 8.3113e-04]],

         [[9.2871e-01, 1.0729e-05, 5.9485e-05,  ..., 1.7529e-03,
           1.3411e-04, 1.7868e-02]],

         ...,

         [[8.2520e-01, 2.2554e-04, 1.5080e-04,  ..., 7.9193e-03,
           2.5215e-03, 3.8357e-03]],

         [[7.8271e-01, 6.7472e-05, 5.9962e-05,  ..., 6.8903e-04,
           9.4482e-02, 9.5596e-03]],

         [[6.2402e-01, 7.9632e-04, 2.6989e-04,  ..., 4.9988e-02,
           1.4465e-02, 2.0081e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1863e-01, 1.7881e-07, 1.4901e-06,  ..., 6.6662e-04,
           6.7444e-03, 7.6611e-01]],

         [[5.3320e-01, 6.9475e-04, 1.2875e-03,  ..., 2.3918e-03,
           8.2731e-04, 5.9853e-03]],

         [[6.9043e-01, 5.8889e-04, 7.3671e-04,  ..., 9.5825e-03,
           2.7008e-03, 1.2001e-02]],

         ...,

         [[6.6309e-01, 1.5581e-04, 1.9395e-04,  ..., 2.8286e-03,
           2.4090e-03, 7.8201e-04]],

         [[4.0771e-01, 8.4043e-06, 9.6560e-06,  ..., 1.2537e-01,
           4.0405e-01, 1.8997e-02]],

         [[4.1382e-01, 1.2195e-04, 1.8167e-04,  ..., 2.3254e-02,
           6.6101e-02, 4.1968e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9810e-01, 1.0139e-04, 3.4511e-05,  ..., 5.2299e-03,
           8.7891e-03, 5.1514e-01]],

         [[4.4434e-01, 5.4240e-06, 7.5698e-06,  ..., 1.9177e-01,
           2.8394e-01, 9.8038e-03]],

         [[3.0908e-01, 1.4782e-04, 8.1956e-05,  ..., 4.4830e-02,
           2.2003e-02, 2.0416e-02]],

         ...,

         [[7.8064e-02, 3.9935e-06, 1.9670e-06,  ..., 2.0278e-04,
           2.1782e-03, 9.1113e-01]],

         [[8.7842e-01, 3.0577e-05, 2.4259e-05,  ..., 2.2907e-03,
           2.0409e-03, 8.2245e-03]],

         [[7.8809e-01, 5.4955e-05, 2.6882e-05,  ..., 3.3905e-02,
           3.0457e-02, 3.2288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 1.2293e-03, 8.4782e-04,  ..., 9.2239e-03,
           2.4967e-03, 2.6379e-03]],

         [[8.9355e-01, 5.4479e-05, 1.0794e-04,  ..., 1.7014e-02,
           1.3954e-02, 1.2283e-02]],

         [[5.0928e-01, 1.4162e-04, 6.5565e-05,  ..., 6.4182e-04,
           8.8453e-04, 8.3923e-03]],

         ...,

         [[5.5273e-01, 5.9605e-08, 3.1590e-06,  ..., 8.3618e-02,
           3.2300e-01, 3.5767e-02]],

         [[4.7168e-01, 6.3944e-04, 8.3733e-04,  ..., 4.9133e-03,
           1.5821e-03, 5.2404e-04]],

         [[8.9453e-01, 2.3842e-07, 4.2319e-06,  ..., 1.7044e-02,
           6.2408e-02, 2.2339e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7095e-01, 2.0361e-04, 6.7174e-05,  ..., 4.9927e-02,
           2.3819e-02, 8.2153e-02]],

         [[1.2103e-01, 2.5578e-03, 1.1511e-03,  ..., 5.2910e-03,
           3.2806e-04, 1.7481e-03]],

         [[7.6758e-01, 9.1136e-05, 7.5221e-05,  ..., 3.2013e-02,
           7.0152e-03, 4.7150e-02]],

         ...,

         [[7.1777e-01, 3.1352e-05, 1.5199e-05,  ..., 8.3984e-02,
           9.3323e-02, 6.6528e-02]],

         [[8.1543e-01, 4.5300e-06, 3.6955e-06,  ..., 2.5604e-02,
           5.7983e-02, 9.1309e-02]],

         [[8.1445e-01, 4.7386e-05, 2.5988e-05,  ..., 5.2155e-02,
           1.9943e-02, 1.6754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1500e-02, 0.0000e+00, 0.0000e+00,  ..., 7.4291e-04,
           1.8177e-03, 9.7314e-01]],

         [[1.2024e-01, 1.5318e-05, 1.8895e-05,  ..., 3.1934e-01,
           1.8616e-01, 8.5571e-02]],

         [[2.0825e-01, 2.6417e-04, 4.9591e-04,  ..., 4.8309e-02,
           4.8828e-03, 1.4824e-02]],

         ...,

         [[4.4263e-01, 2.7680e-04, 2.7466e-04,  ..., 2.1484e-02,
           1.4084e-02, 1.7128e-03]],

         [[5.9180e-01, 1.2636e-05, 6.3777e-06,  ..., 2.5803e-02,
           5.4436e-03, 1.0773e-02]],

         [[3.2324e-01, 4.9472e-05, 2.1279e-05,  ..., 2.5368e-03,
           5.9090e-03, 1.0475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0303e-01, 9.8348e-06, 6.3777e-06,  ..., 3.1342e-02,
           2.4078e-02, 2.0996e-01]],

         [[3.7036e-01, 2.7323e-04, 1.4567e-04,  ..., 1.4160e-02,
           3.6530e-02, 5.4840e-02]],

         [[1.1218e-01, 1.4842e-05, 6.9737e-06,  ..., 5.8411e-02,
           6.6064e-01, 4.5502e-02]],

         ...,

         [[8.3252e-01, 1.5557e-05, 3.1829e-05,  ..., 3.4241e-02,
           1.4275e-02, 2.2903e-02]],

         [[6.9775e-01, 1.4603e-05, 2.3067e-05,  ..., 1.1615e-01,
           2.9251e-02, 5.1941e-02]],

         [[8.0176e-01, 1.9848e-05, 8.7023e-06,  ..., 8.7051e-03,
           7.2975e-03, 4.4006e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5491e-01, 2.2068e-03, 2.0370e-03,  ..., 3.7518e-03,
           5.4646e-04, 2.0618e-03]],

         [[3.3130e-01, 1.2374e-04, 6.5625e-05,  ..., 4.3549e-02,
           5.0629e-02, 1.6663e-02]],

         [[4.8315e-01, 2.8610e-06, 5.1856e-06,  ..., 1.6565e-01,
           1.5137e-01, 9.9304e-02]],

         ...,

         [[3.1519e-01, 4.5300e-05, 1.5378e-05,  ..., 1.1492e-03,
           3.3226e-03, 3.6716e-03]],

         [[1.3374e-02, 2.9802e-06, 4.1723e-06,  ..., 2.9802e-07,
           1.7881e-06, 1.3342e-03]],

         [[4.4092e-01, 1.5318e-04, 3.6693e-04,  ..., 3.4428e-03,
           4.4823e-03, 4.3917e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5328e-02, 1.2386e-04, 7.9989e-05,  ..., 5.6641e-02,
           8.3542e-03, 5.9814e-02]],

         [[1.7822e-01, 3.5620e-04, 8.9705e-05,  ..., 6.6589e-02,
           1.4061e-02, 1.3527e-02]],

         [[4.4971e-01, 3.4988e-05, 1.4007e-05,  ..., 3.6072e-02,
           9.7290e-02, 3.3179e-01]],

         ...,

         [[5.3809e-01, 1.6689e-06, 4.9472e-06,  ..., 4.5593e-02,
           3.8391e-02, 1.4572e-02]],

         [[7.8711e-01, 2.3842e-06, 3.2187e-06,  ..., 8.4900e-02,
           4.8004e-02, 2.1133e-02]],

         [[1.5784e-01, 4.5300e-06, 2.2054e-06,  ..., 4.1504e-02,
           9.1370e-02, 6.2891e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 3.9995e-05, 9.9540e-06,  ..., 5.5771e-03,
           9.7885e-03, 4.3884e-02]],

         [[8.9539e-02, 2.7418e-06, 6.0797e-06,  ..., 4.4775e-01,
           3.1738e-01, 1.8768e-02]],

         [[1.1060e-01, 6.7353e-06, 1.7881e-06,  ..., 3.6678e-03,
           7.5264e-03, 8.6963e-01]],

         ...,

         [[4.3701e-02, 1.0294e-04, 5.3644e-06,  ..., 1.8034e-03,
           1.0767e-03, 4.8767e-02]],

         [[2.1667e-01, 1.9872e-04, 1.1867e-04,  ..., 3.2959e-02,
           2.4887e-02, 3.0243e-02]],

         [[2.0276e-01, 1.0997e-04, 4.7982e-05,  ..., 7.5195e-02,
           3.7506e-02, 4.2847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0771e-01, 1.8418e-05, 2.7299e-05,  ..., 3.0716e-02,
           3.0716e-02, 1.4429e-01]],

         [[1.6089e-01, 2.6822e-06, 5.3644e-07,  ..., 2.2542e-04,
           2.4962e-04, 4.9362e-03]],

         [[5.1514e-01, 8.5235e-05, 1.0133e-05,  ..., 2.9697e-03,
           6.0797e-04, 1.2306e-02]],

         ...,

         [[1.8738e-01, 9.6846e-04, 1.0967e-03,  ..., 3.5767e-02,
           1.0406e-02, 1.2169e-02]],

         [[6.4258e-01, 1.4901e-05, 1.3351e-05,  ..., 7.3242e-02,
           6.8787e-02, 6.8787e-02]],

         [[2.1289e-01, 4.3750e-04, 5.1165e-04,  ..., 2.4078e-02,
           1.0513e-02, 3.2379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8457e-01, 9.2804e-05, 2.0111e-04,  ..., 1.1780e-02,
           1.7548e-02, 5.5328e-02]],

         [[8.9355e-02, 1.7881e-05, 3.0458e-05,  ..., 1.3702e-02,
           1.7044e-02, 1.3069e-02]],

         [[3.4106e-01, 2.5153e-05, 2.4915e-05,  ..., 2.9736e-03,
           4.4289e-03, 5.0293e-02]],

         ...,

         [[3.1036e-02, 8.1897e-05, 2.6584e-05,  ..., 1.1713e-01,
           9.4116e-02, 1.1780e-02]],

         [[1.3159e-01, 1.5974e-05, 3.8743e-06,  ..., 1.3657e-02,
           1.9257e-02, 9.4788e-02]],

         [[8.2764e-01, 1.0139e-04, 7.9572e-05,  ..., 1.1810e-02,
           8.7738e-03, 2.4231e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 5.4240e-06, 2.3842e-06,  ..., 7.1533e-02,
           1.1255e-01, 1.4001e-01]],

         [[1.4050e-01, 2.9862e-05, 3.3438e-05,  ..., 1.4380e-01,
           2.0923e-01, 3.3112e-02]],

         [[2.4463e-01, 1.1712e-04, 1.7238e-04,  ..., 5.5450e-02,
           7.0068e-02, 5.1367e-01]],

         ...,

         [[4.5215e-01, 1.8656e-04, 1.2875e-04,  ..., 2.8458e-02,
           3.8605e-02, 6.5125e-02]],

         [[2.7344e-01, 1.6987e-05, 1.3137e-04,  ..., 7.9250e-04,
           4.0550e-03, 1.4417e-01]],

         [[3.8794e-01, 5.1200e-05, 5.5552e-05,  ..., 1.1444e-02,
           8.2642e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0996e-01, 4.4525e-05, 2.2531e-05,  ..., 6.8726e-02,
           1.3245e-01, 1.4893e-01]],

         [[4.6948e-01, 4.1068e-05, 9.2983e-06,  ..., 1.0236e-01,
           8.8196e-02, 1.0724e-01]],

         [[4.9487e-01, 3.5906e-04, 2.8801e-04,  ..., 1.4320e-02,
           8.2245e-03, 8.0948e-03]],

         ...,

         [[6.3428e-01, 1.1921e-07, 3.7551e-05,  ..., 4.7028e-02,
           2.6855e-01, 4.0863e-02]],

         [[3.2593e-01, 1.1921e-06, 2.4676e-05,  ..., 1.9922e-01,
           2.5781e-01, 1.8539e-02]],

         [[5.2783e-01, 1.9670e-06, 8.8990e-05,  ..., 7.2510e-02,
           2.8687e-01, 8.4778e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1765e-02, 1.2517e-06, 8.9407e-07,  ..., 2.8296e-01,
           5.8057e-01, 2.6108e-02]],

         [[2.6709e-01, 7.3314e-06, 7.5102e-06,  ..., 2.4395e-03,
           1.7883e-02, 6.6113e-01]],

         [[1.2317e-01, 6.1393e-05, 1.4925e-04,  ..., 1.2317e-01,
           4.3762e-02, 2.5223e-02]],

         ...,

         [[9.9365e-02, 1.2982e-04, 4.3750e-04,  ..., 4.5135e-02,
           3.9520e-02, 3.1525e-02]],

         [[8.8501e-02, 1.7881e-06, 2.6226e-06,  ..., 4.2896e-01,
           3.9062e-01, 7.9346e-02]],

         [[3.0078e-01, 5.9605e-07, 7.6294e-06,  ..., 1.5662e-01,
           4.4775e-01, 7.5134e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2988e-01, 5.5194e-05, 1.8680e-04,  ..., 3.0624e-02,
           9.3765e-03, 6.5857e-02]],

         [[9.6436e-01, 2.8372e-05, 4.7922e-05,  ..., 1.8513e-04,
           5.1916e-05, 1.9140e-03]],

         [[9.5215e-01, 7.6175e-05, 3.4869e-05,  ..., 2.0027e-03,
           4.9067e-04, 2.3975e-03]],

         ...,

         [[9.1943e-01, 4.3511e-06, 4.8876e-06,  ..., 7.9575e-03,
           7.3586e-03, 5.5206e-02]],

         [[9.6387e-01, 1.5497e-06, 1.0133e-06,  ..., 3.4976e-04,
           1.4725e-03, 3.1166e-03]],

         [[2.3108e-01, 3.2187e-06, 6.4373e-06,  ..., 6.2561e-03,
           1.3664e-02, 7.3438e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6309e-01, 6.5565e-07, 2.6822e-06,  ..., 1.0992e-01,
           1.8628e-01, 4.6460e-01]],

         [[9.4727e-01, 7.2122e-06, 5.0068e-06,  ..., 5.3520e-03,
           7.7896e-03, 1.8188e-02]],

         [[3.6182e-01, 1.7297e-04, 2.0552e-04,  ..., 8.1055e-02,
           3.6377e-02, 4.7272e-02]],

         ...,

         [[7.0312e-01, 9.4175e-06, 2.1935e-05,  ..., 7.8918e-02,
           6.9092e-02, 3.1891e-02]],

         [[8.5645e-01, 9.9540e-06, 3.9816e-05,  ..., 8.2703e-03,
           6.0005e-03, 1.9226e-02]],

         [[7.9492e-01, 6.2585e-06, 2.1994e-05,  ..., 2.7847e-02,
           8.4473e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 3.5167e-06, 1.4305e-06,  ..., 2.2864e-01,
           4.4067e-01, 7.4829e-02]],

         [[6.6064e-01, 3.1590e-06, 7.1526e-07,  ..., 7.5722e-03,
           1.2805e-01, 1.4062e-01]],

         [[5.4541e-01, 5.8472e-05, 3.4571e-05,  ..., 1.4534e-02,
           6.0242e-02, 4.7302e-02]],

         ...,

         [[8.1726e-02, 5.3644e-07, 1.3769e-05,  ..., 6.8787e-02,
           8.5632e-02, 1.8701e-01]],

         [[1.5881e-01, 2.3246e-06, 2.3842e-05,  ..., 9.7839e-02,
           6.7200e-02, 1.9257e-02]],

         [[9.3311e-01, 2.5094e-05, 3.2246e-05,  ..., 1.6830e-02,
           4.6005e-03, 3.2368e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6543e-01, 1.4305e-06, 2.3365e-05,  ..., 9.3384e-03,
           4.2419e-02, 3.4692e-01]],

         [[6.5234e-01, 1.9073e-06, 6.0141e-05,  ..., 4.9133e-02,
           1.1029e-01, 5.5237e-02]],

         [[5.3662e-01, 8.9407e-07, 3.5346e-05,  ..., 4.3060e-02,
           1.3733e-01, 2.6172e-01]],

         ...,

         [[4.9097e-01, 1.4591e-04, 8.0585e-05,  ..., 1.4145e-02,
           4.1260e-02, 2.4246e-02]],

         [[5.4535e-02, 4.1723e-07, 4.1723e-07,  ..., 4.3555e-01,
           4.2212e-01, 5.1208e-02]],

         [[6.7993e-02, 5.1498e-05, 1.7321e-04,  ..., 6.6528e-03,
           2.6413e-02, 6.7578e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7354e-01, 1.4067e-05, 1.2755e-05,  ..., 1.1566e-01,
           1.9067e-01, 1.8054e-01]],

         [[7.9004e-01, 1.4305e-05, 1.3471e-05,  ..., 1.5343e-02,
           5.7678e-02, 4.4556e-02]],

         [[1.6101e-01, 1.2577e-05, 3.0935e-05,  ..., 5.1056e-02,
           2.5732e-01, 1.3879e-01]],

         ...,

         [[6.5283e-01, 3.7789e-05, 1.0103e-04,  ..., 1.3684e-01,
           4.1748e-02, 4.1077e-02]],

         [[2.5293e-01, 1.2350e-04, 2.8706e-04,  ..., 1.1721e-03,
           1.2794e-02, 6.6650e-01]],

         [[1.4343e-01, 5.4240e-06, 9.8348e-06,  ..., 4.6289e-01,
           3.1323e-01, 5.3589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1655e-01, 1.4722e-05, 9.1672e-05,  ..., 4.1351e-02,
           5.9570e-01, 1.0852e-01]],

         [[5.5450e-02, 5.9605e-07, 4.8280e-06,  ..., 1.1829e-01,
           1.7480e-01, 2.6562e-01]],

         [[4.7217e-01, 1.9479e-04, 6.1321e-04,  ..., 1.4374e-02,
           4.9103e-02, 8.8013e-02]],

         ...,

         [[9.1357e-01, 4.3929e-05, 1.1456e-04,  ..., 4.1656e-03,
           5.2452e-03, 1.6510e-02]],

         [[9.6436e-01, 1.7881e-07, 9.4175e-05,  ..., 1.8823e-04,
           6.0558e-04, 8.7814e-03]],

         [[6.6992e-01, 2.2531e-05, 3.8624e-05,  ..., 3.0556e-03,
           1.1131e-02, 6.2805e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0788e-05, 6.6161e-06, 1.8677e-01,  ..., 3.1219e-02,
           2.6343e-01, 1.9302e-02]],

         [[2.9802e-06, 8.9407e-07, 7.5912e-03,  ..., 2.8362e-03,
           1.1940e-02, 2.7485e-03]],

         [[9.1195e-05, 8.9169e-05, 6.0394e-02,  ..., 1.0872e-02,
           6.4819e-02, 4.4458e-01]],

         ...,

         [[1.0967e-05, 1.4901e-06, 7.5623e-02,  ..., 7.1594e-02,
           4.8169e-01, 1.6785e-01]],

         [[1.1927e-04, 3.3796e-05, 3.3594e-01,  ..., 6.4507e-03,
           8.0566e-03, 9.4910e-03]],

         [[7.2718e-06, 5.3644e-07, 3.0103e-01,  ..., 1.3123e-02,
           5.5664e-02, 5.5664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3273e-05, 2.5511e-05, 2.6465e-01,  ..., 5.4283e-03,
           5.4169e-02, 1.7868e-02]],

         [[3.8207e-05, 9.8944e-06, 2.3279e-01,  ..., 3.1250e-02,
           1.4575e-01, 8.6136e-03]],

         [[4.9829e-05, 7.0930e-06, 2.8076e-01,  ..., 1.6365e-03,
           4.3365e-02, 1.0345e-02]],

         ...,

         [[1.2875e-05, 4.0531e-06, 2.4841e-01,  ..., 1.7004e-03,
           2.7863e-02, 1.3794e-02]],

         [[2.3842e-06, 1.4901e-06, 2.8149e-01,  ..., 6.5956e-03,
           7.0618e-02, 4.3602e-03]],

         [[1.2457e-05, 1.7047e-05, 3.1714e-01,  ..., 6.5422e-04,
           2.8366e-02, 5.3520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-06, 2.7418e-06, 1.6943e-01,  ..., 1.9318e-02,
           1.1292e-01, 9.0027e-03]],

         [[1.5736e-05, 4.1723e-07, 2.4414e-01,  ..., 3.1877e-04,
           4.2755e-02, 1.4656e-02]],

         [[2.2113e-05, 7.6890e-06, 2.5122e-01,  ..., 1.9407e-03,
           6.5552e-02, 1.0864e-02]],

         ...,

         [[3.9101e-05, 1.8477e-06, 1.6418e-01,  ..., 1.1548e-01,
           2.1411e-01, 1.6162e-01]],

         [[8.5235e-06, 4.2319e-06, 2.6514e-01,  ..., 4.3716e-03,
           4.6997e-02, 2.8503e-02]],

         [[4.3511e-05, 3.7551e-06, 2.2449e-01,  ..., 2.2308e-02,
           7.5195e-02, 2.4033e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-06, 2.3842e-07, 8.5220e-03,  ..., 2.4329e-01,
           5.5273e-01, 1.5466e-01]],

         [[3.1769e-05, 1.2100e-04, 7.6965e-02,  ..., 1.6830e-02,
           1.5747e-02, 1.4771e-02]],

         [[2.0921e-05, 2.3735e-04, 3.1396e-01,  ..., 5.4264e-04,
           4.6468e-04, 6.1369e-04]],

         ...,

         [[2.8586e-04, 9.0003e-06, 2.8540e-01,  ..., 5.2261e-04,
           1.7233e-03, 3.8967e-03]],

         [[2.9755e-04, 9.8348e-06, 2.9099e-02,  ..., 2.4124e-02,
           1.9958e-01, 6.6211e-01]],

         [[6.6936e-05, 7.3314e-06, 3.3722e-02,  ..., 8.6792e-02,
           3.9673e-01, 2.8906e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1505e-04, 8.4519e-05, 1.5125e-01,  ..., 1.3847e-02,
           2.6794e-02, 5.7861e-02]],

         [[1.3423e-04, 1.7035e-04, 1.6260e-01,  ..., 2.4368e-02,
           4.3762e-02, 1.9617e-01]],

         [[5.5742e-04, 1.2922e-04, 1.5149e-01,  ..., 2.4918e-02,
           7.5256e-02, 4.9927e-02]],

         ...,

         [[8.7142e-05, 2.0206e-05, 5.1147e-02,  ..., 6.5674e-02,
           4.0234e-01, 2.9443e-01]],

         [[2.0337e-04, 5.2357e-04, 3.7061e-01,  ..., 7.6332e-03,
           4.0619e-02, 4.7852e-02]],

         [[1.4532e-04, 6.8903e-05, 1.8188e-01,  ..., 6.7444e-02,
           1.5088e-01, 1.6956e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.2696e-05, 3.3975e-06, 4.7588e-04,  ..., 1.6943e-01,
           4.4653e-01, 8.2581e-02]],

         [[9.0551e-04, 2.2900e-04, 2.4624e-03,  ..., 1.0468e-01,
           4.9438e-02, 5.3174e-01]],

         [[2.9802e-07, 0.0000e+00, 8.3447e-07,  ..., 9.0456e-04,
           4.0550e-03, 9.9219e-01]],

         ...,

         [[1.7881e-07, 5.9605e-08, 2.9802e-07,  ..., 7.4446e-05,
           9.0694e-04, 9.9463e-01]],

         [[6.6071e-03, 8.7509e-03, 4.2648e-03,  ..., 7.1045e-02,
           1.5358e-02, 2.3788e-02]],

         [[4.8370e-03, 1.6203e-03, 5.7745e-04,  ..., 1.1971e-02,
           2.0370e-02, 1.6876e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2171e-04, 3.9911e-04, 3.7270e-03,  ..., 4.6143e-02,
           1.6101e-01, 4.9585e-01]],

         [[2.3842e-07, 0.0000e+00, 0.0000e+00,  ..., 1.0016e-01,
           8.1299e-01, 7.8003e-02]],

         [[1.7393e-04, 7.6723e-04, 1.0933e-02,  ..., 5.2155e-02,
           1.1749e-01, 6.9092e-02]],

         ...,

         [[2.1915e-03, 2.3499e-03, 5.5084e-03,  ..., 1.5479e-01,
           1.1414e-01, 2.7588e-01]],

         [[1.4372e-03, 7.4158e-03, 3.0975e-02,  ..., 4.1992e-02,
           1.0658e-02, 1.8417e-02]],

         [[7.6294e-04, 3.7861e-03, 4.3564e-03,  ..., 6.6040e-02,
           1.0229e-01, 5.1453e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 5.6686e-03, 3.1616e-02,  ..., 3.2867e-02,
           4.4922e-02, 3.9978e-02]],

         [[1.7757e-03, 8.0681e-04, 3.9101e-03,  ..., 5.2185e-03,
           9.5978e-03, 1.0544e-02]],

         [[7.3853e-03, 1.2197e-03, 1.4772e-03,  ..., 2.2568e-02,
           2.3376e-02, 2.1454e-02]],

         ...,

         [[2.8074e-05, 3.0994e-06, 2.0123e-04,  ..., 3.0334e-02,
           4.0576e-01, 4.1553e-01]],

         [[2.8954e-03, 9.6941e-04, 2.5940e-03,  ..., 4.2877e-02,
           2.8778e-02, 4.3549e-02]],

         [[2.3670e-03, 5.1355e-04, 2.4986e-03,  ..., 1.4185e-01,
           1.1310e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6904e-01, 1.7822e-05, 1.7405e-05,  ..., 2.2202e-03,
           4.9438e-03, 8.0872e-03]],

         [[9.2041e-01, 6.3181e-06, 5.2452e-06,  ..., 1.4114e-02,
           2.1988e-02, 1.6815e-02]],

         [[9.3750e-01, 1.7881e-07, 5.5432e-06,  ..., 5.4779e-03,
           1.2634e-02, 4.6387e-03]],

         ...,

         [[5.0586e-01, 2.2960e-04, 1.3962e-03,  ..., 2.5606e-04,
           3.5596e-04, 1.8494e-02]],

         [[5.3076e-01, 8.3804e-05, 1.2522e-03,  ..., 5.9319e-03,
           4.4250e-02, 1.0780e-02]],

         [[3.8818e-01, 2.3401e-04, 4.6921e-04,  ..., 4.1412e-02,
           8.2474e-03, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3350e-01, 4.7684e-06, 1.2100e-05,  ..., 8.2970e-04,
           8.1837e-05, 1.0269e-02]],

         [[9.2188e-01, 0.0000e+00, 0.0000e+00,  ..., 2.9392e-03,
           2.6077e-02, 4.5288e-02]],

         [[9.1504e-01, 1.5974e-05, 1.6093e-06,  ..., 2.9068e-03,
           1.8677e-02, 1.3496e-02]],

         ...,

         [[3.5742e-01, 1.8167e-04, 2.0897e-04,  ..., 8.0688e-02,
           7.2876e-02, 6.6566e-03]],

         [[2.4536e-01, 5.3358e-04, 2.6298e-04,  ..., 8.7128e-03,
           4.6577e-03, 8.7051e-03]],

         [[6.9629e-01, 1.6451e-05, 1.2136e-04,  ..., 1.1665e-02,
           1.6992e-01, 5.8838e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7383e-01, 1.2094e-04, 5.8234e-05,  ..., 2.8320e-02,
           5.9631e-02, 4.9744e-02]],

         [[6.5967e-01, 1.0252e-05, 9.0003e-06,  ..., 1.6739e-02,
           7.0435e-02, 1.5884e-02]],

         [[6.6797e-01, 2.8789e-05, 4.7028e-05,  ..., 4.8584e-02,
           1.5076e-01, 1.0474e-01]],

         ...,

         [[8.3691e-01, 2.3842e-07, 4.1723e-07,  ..., 2.2769e-05,
           6.0797e-06, 1.5442e-01]],

         [[8.8135e-01, 2.9802e-07, 3.8147e-06,  ..., 5.9605e-07,
           2.3842e-07, 3.5614e-02]],

         [[9.8438e-01, 5.9605e-08, 2.9802e-07,  ..., 2.9802e-07,
           5.9605e-08, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7563e-01, 4.0531e-06, 2.5034e-06,  ..., 1.7371e-01,
           3.5742e-01, 2.5681e-02]],

         [[9.4385e-01, 1.4305e-06, 1.9670e-06,  ..., 1.9097e-04,
           2.9802e-04, 3.9406e-03]],

         [[7.7783e-01, 1.1218e-04, 2.0087e-05,  ..., 1.4324e-03,
           7.7133e-03, 1.5778e-02]],

         ...,

         [[5.8789e-01, 3.1638e-04, 9.3162e-05,  ..., 2.0660e-02,
           1.9128e-01, 5.0293e-02]],

         [[3.9868e-01, 4.1187e-05, 1.1086e-05,  ..., 8.6288e-03,
           3.2422e-01, 6.8176e-02]],

         [[8.3545e-01, 3.5644e-05, 6.7353e-05,  ..., 8.6136e-03,
           2.3438e-02, 2.0325e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6338e-01, 2.9802e-06, 3.0994e-06,  ..., 7.2670e-04,
           4.0855e-03, 3.6335e-03]],

         [[9.6289e-01, 7.7486e-06, 1.6093e-06,  ..., 9.6798e-04,
           2.1515e-03, 1.5274e-02]],

         [[9.5508e-01, 2.0266e-06, 4.7684e-07,  ..., 7.0534e-03,
           9.6436e-03, 1.1055e-02]],

         ...,

         [[7.9395e-01, 1.8239e-04, 5.2631e-05,  ..., 5.4016e-03,
           2.6215e-02, 1.5022e-02]],

         [[8.9355e-01, 1.1694e-04, 1.2636e-05,  ..., 2.2721e-04,
           8.0566e-03, 1.3733e-02]],

         [[9.4824e-01, 6.1989e-06, 2.2054e-06,  ..., 1.4915e-02,
           1.2932e-02, 3.9520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7031e-01, 1.6203e-03, 1.2817e-03,  ..., 2.2316e-03,
           1.2329e-02, 8.2626e-03]],

         [[9.3945e-01, 1.1683e-05, 2.0504e-05,  ..., 5.4893e-03,
           2.3060e-03, 1.5974e-03]],

         [[8.8232e-01, 1.5855e-05, 8.1658e-06,  ..., 2.5578e-03,
           2.1851e-02, 7.7896e-03]],

         ...,

         [[7.2607e-01, 6.2048e-05, 2.3842e-05,  ..., 8.9569e-03,
           1.3336e-02, 1.9312e-01]],

         [[7.4463e-01, 2.7418e-06, 5.9605e-06,  ..., 5.9998e-02,
           2.8057e-03, 3.0289e-03]],

         [[7.0264e-01, 5.9009e-06, 6.9141e-06,  ..., 1.1787e-02,
           2.1692e-01, 2.4048e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9226e-01, 1.6105e-04, 7.7724e-04,  ..., 1.9302e-02,
           1.8097e-02, 4.7340e-03]],

         [[7.7441e-01, 4.7684e-07, 1.9670e-06,  ..., 2.3453e-02,
           7.9712e-02, 2.3453e-02]],

         [[7.6953e-01, 1.0014e-05, 2.1517e-05,  ..., 5.1178e-02,
           9.2468e-02, 2.5955e-02]],

         ...,

         [[8.7158e-01, 6.2764e-05, 4.9651e-05,  ..., 8.6975e-04,
           3.1872e-03, 1.5364e-03]],

         [[6.0059e-01, 2.2948e-05, 1.3292e-05,  ..., 4.2305e-03,
           1.7078e-01, 9.6924e-02]],

         [[4.7046e-01, 5.4836e-06, 8.1658e-06,  ..., 1.0999e-01,
           6.2195e-02, 2.2705e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6504e-01, 1.0842e-04, 2.7180e-05,  ..., 2.1875e-05,
           2.1064e-04, 1.7715e-02]],

         [[9.5361e-01, 6.9737e-06, 4.8280e-06,  ..., 1.5421e-03,
           3.5305e-03, 5.2986e-03]],

         [[7.9297e-01, 3.3975e-06, 1.6212e-05,  ..., 1.2112e-03,
           1.4275e-02, 1.5617e-02]],

         ...,

         [[8.3887e-01, 1.4365e-04, 8.8871e-05,  ..., 1.1215e-03,
           9.0485e-03, 5.7297e-03]],

         [[8.5938e-01, 7.1764e-05, 7.5221e-05,  ..., 2.1271e-02,
           1.3206e-02, 5.5962e-03]],

         [[7.5439e-01, 1.1101e-03, 4.8494e-04,  ..., 9.8114e-03,
           1.2016e-02, 9.5596e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 1.2517e-06, 9.5367e-07,  ..., 1.8990e-04,
           2.3499e-03, 4.6191e-01]],

         [[3.8940e-01, 4.6515e-04, 9.2936e-04,  ..., 1.4391e-03,
           1.3123e-02, 5.6549e-02]],

         [[7.3730e-01, 3.4523e-04, 5.2595e-04,  ..., 2.3861e-03,
           7.7095e-03, 2.7542e-02]],

         ...,

         [[4.4849e-01, 1.6379e-04, 1.9991e-04,  ..., 1.5228e-02,
           5.0545e-03, 8.6164e-04]],

         [[3.8403e-01, 4.1246e-05, 2.8491e-05,  ..., 7.5806e-02,
           2.3071e-01, 3.9185e-02]],

         [[2.9907e-01, 4.2260e-05, 3.8922e-05,  ..., 1.0231e-02,
           1.5030e-02, 6.2598e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9473e-01, 3.0279e-05, 4.2558e-05,  ..., 9.1171e-03,
           1.0826e-02, 3.4912e-01]],

         [[4.5776e-01, 4.0412e-05, 7.2300e-05,  ..., 1.0992e-01,
           1.2341e-01, 5.1697e-02]],

         [[9.8145e-02, 9.3281e-05, 1.1927e-04,  ..., 3.6407e-02,
           3.9917e-02, 6.9275e-02]],

         ...,

         [[2.0508e-01, 3.5763e-06, 8.6427e-06,  ..., 5.1498e-04,
           2.6646e-03, 7.8125e-01]],

         [[7.3438e-01, 2.5928e-05, 5.9545e-05,  ..., 2.6535e-02,
           6.9466e-03, 2.0950e-02]],

         [[6.9336e-01, 7.9751e-05, 1.6868e-05,  ..., 4.4250e-02,
           7.5134e-02, 3.6499e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0146e-01, 3.9554e-04, 1.2722e-03,  ..., 1.7929e-02,
           2.2583e-02, 1.4868e-01]],

         [[6.9043e-01, 3.8683e-05, 1.4722e-04,  ..., 4.3884e-02,
           1.4709e-01, 2.6459e-02]],

         [[6.6553e-01, 1.2445e-04, 1.2994e-04,  ..., 1.3176e-02,
           3.3259e-05, 9.3231e-03]],

         ...,

         [[3.3862e-01, 0.0000e+00, 0.0000e+00,  ..., 1.4519e-02,
           6.0986e-01, 2.9083e-02]],

         [[5.7373e-01, 4.9496e-04, 8.4209e-04,  ..., 1.7214e-03,
           1.6651e-03, 2.3854e-04]],

         [[7.3193e-01, 0.0000e+00, 5.9605e-08,  ..., 8.5068e-03,
           6.1951e-02, 1.9019e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2969e-01, 5.4717e-05, 2.4557e-05,  ..., 2.9968e-02,
           2.4261e-02, 5.5389e-02]],

         [[9.1125e-02, 1.2445e-03, 1.3647e-03,  ..., 1.6327e-03,
           2.8687e-03, 2.2488e-03]],

         [[1.9873e-01, 2.2411e-04, 8.1658e-05,  ..., 3.6743e-02,
           2.2607e-01, 2.2888e-02]],

         ...,

         [[7.5732e-01, 2.6941e-05, 2.0802e-05,  ..., 4.2145e-02,
           6.5796e-02, 1.2169e-02]],

         [[5.7178e-01, 1.7881e-06, 2.9206e-06,  ..., 1.0815e-01,
           1.8286e-01, 6.2927e-02]],

         [[7.0117e-01, 4.3213e-05, 2.7895e-05,  ..., 2.2797e-02,
           4.1321e-02, 7.0007e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6821e-01, 9.5367e-07, 2.3842e-07,  ..., 5.1022e-04,
           6.6261e-03, 8.0566e-01]],

         [[1.3025e-01, 1.4305e-06, 2.5630e-06,  ..., 1.6724e-01,
           2.3218e-01, 2.0676e-02]],

         [[1.1017e-01, 3.6740e-04, 5.3120e-04,  ..., 1.1406e-02,
           1.8265e-02, 2.6260e-02]],

         ...,

         [[5.9717e-01, 8.4817e-05, 8.1599e-05,  ..., 3.9864e-03,
           1.3281e-01, 1.9419e-04]],

         [[4.9561e-01, 5.3048e-06, 2.3246e-06,  ..., 1.0266e-01,
           3.0880e-03, 3.6133e-02]],

         [[6.9922e-01, 8.6963e-05, 1.6248e-04,  ..., 9.1505e-04,
           5.6458e-04, 3.9093e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5808e-01, 1.0729e-06, 2.1458e-06,  ..., 8.2169e-03,
           7.1680e-01, 4.8401e-02]],

         [[2.1057e-01, 5.4479e-05, 2.4915e-05,  ..., 1.5527e-01,
           8.9172e-02, 9.7168e-02]],

         [[7.9834e-01, 3.3975e-06, 1.3411e-05,  ..., 5.8861e-03,
           1.2006e-01, 1.1749e-02]],

         ...,

         [[7.1289e-01, 2.2650e-06, 1.0312e-05,  ..., 2.0233e-02,
           7.4829e-02, 1.5091e-02]],

         [[6.6650e-01, 1.0133e-06, 3.8147e-06,  ..., 2.1801e-03,
           2.5098e-01, 2.2629e-02]],

         [[6.2598e-01, 3.7074e-05, 3.3617e-05,  ..., 2.8038e-03,
           2.5940e-02, 4.4983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2825e-02, 4.2176e-04, 9.8228e-04,  ..., 1.0357e-03,
           1.0815e-03, 1.0612e-02]],

         [[1.8115e-01, 5.6207e-05, 1.1462e-04,  ..., 2.2141e-02,
           2.2316e-03, 1.6296e-01]],

         [[7.8674e-02, 5.9605e-07, 7.7486e-07,  ..., 1.0339e-01,
           5.5908e-01, 1.0626e-01]],

         ...,

         [[7.7051e-01, 1.2422e-04, 4.7147e-05,  ..., 5.6419e-03,
           2.6035e-03, 5.6366e-02]],

         [[7.9883e-01, 1.4663e-05, 9.0003e-06,  ..., 1.3053e-05,
           5.0664e-06, 9.0103e-03]],

         [[4.8267e-01, 1.9038e-04, 3.4475e-04,  ..., 7.0038e-03,
           1.6422e-03, 1.8559e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9713e-02, 3.6788e-04, 1.6975e-04,  ..., 6.8359e-03,
           2.0096e-02, 4.5891e-03]],

         [[2.5757e-01, 1.9121e-04, 6.3956e-05,  ..., 6.1760e-03,
           4.1275e-03, 4.3106e-03]],

         [[2.4048e-01, 4.2915e-06, 2.5630e-06,  ..., 2.0898e-01,
           1.8433e-01, 2.1558e-01]],

         ...,

         [[3.8379e-01, 2.3842e-07, 2.9802e-07,  ..., 1.4877e-02,
           3.4607e-02, 3.2177e-03]],

         [[4.4800e-01, 4.1723e-07, 3.5763e-07,  ..., 2.3926e-02,
           3.2764e-01, 6.5063e-02]],

         [[2.9907e-01, 1.4901e-06, 1.6689e-06,  ..., 6.5674e-02,
           1.0663e-01, 4.7046e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6826e-01, 5.3465e-05, 2.3723e-05,  ..., 5.3375e-02,
           1.2482e-02, 2.5208e-02]],

         [[2.2491e-02, 1.0133e-06, 2.3246e-06,  ..., 1.8530e-01,
           2.6978e-01, 6.0181e-02]],

         [[2.3413e-01, 7.3910e-06, 2.3246e-06,  ..., 4.0894e-03,
           5.8594e-03, 7.4365e-01]],

         ...,

         [[1.3757e-01, 3.7026e-04, 9.0361e-05,  ..., 3.7599e-04,
           1.7834e-04, 3.8433e-03]],

         [[4.7974e-01, 2.6679e-04, 1.5783e-04,  ..., 6.4392e-02,
           1.0269e-02, 2.1744e-02]],

         [[1.2408e-01, 7.3910e-05, 9.3460e-05,  ..., 2.4246e-02,
           3.1929e-03, 1.3603e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4790e-01, 6.6757e-06, 8.4639e-06,  ..., 2.2949e-02,
           4.9347e-02, 2.0142e-01]],

         [[6.9385e-01, 4.7088e-06, 7.1526e-07,  ..., 9.6464e-04,
           7.6294e-04, 1.3180e-03]],

         [[4.9731e-01, 1.2887e-04, 2.8968e-05,  ..., 2.9633e-02,
           2.1347e-02, 3.3760e-03]],

         ...,

         [[8.4595e-02, 2.6917e-04, 3.3498e-04,  ..., 7.7438e-03,
           4.6959e-03, 7.3891e-03]],

         [[6.8799e-01, 1.6093e-05, 1.4067e-05,  ..., 1.4961e-02,
           3.4790e-02, 9.7534e-02]],

         [[1.8054e-01, 4.2224e-04, 9.7370e-04,  ..., 7.9346e-03,
           2.0584e-02, 5.3802e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4424e-01, 4.4703e-05, 5.6744e-05,  ..., 3.2013e-02,
           6.4209e-02, 1.4575e-01]],

         [[2.1509e-01, 1.4007e-05, 4.8459e-05,  ..., 4.7241e-02,
           2.2675e-02, 7.5500e-02]],

         [[3.9990e-01, 2.4438e-05, 3.5822e-05,  ..., 7.6752e-03,
           1.6113e-02, 9.6436e-02]],

         ...,

         [[3.1067e-02, 5.0485e-05, 4.6730e-05,  ..., 7.1106e-02,
           5.2338e-03, 4.8126e-02]],

         [[3.0566e-01, 1.0252e-04, 7.0453e-05,  ..., 2.3575e-02,
           1.5221e-02, 1.5613e-01]],

         [[6.4160e-01, 7.1526e-05, 1.0091e-04,  ..., 1.9379e-02,
           1.4854e-02, 6.6589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9789e-05, 1.4544e-05,  ..., 4.0344e-02,
           5.1392e-02, 3.1470e-01]],

         [[6.6846e-01, 1.8811e-04, 3.9673e-04,  ..., 6.8665e-03,
           5.6038e-03, 1.5480e-02]],

         [[3.8477e-01, 1.3065e-04, 2.0790e-04,  ..., 6.7383e-02,
           6.1371e-02, 1.0437e-01]],

         ...,

         [[5.8008e-01, 1.3328e-04, 8.9109e-05,  ..., 3.9795e-02,
           5.9265e-02, 1.0730e-01]],

         [[4.1431e-01, 2.8133e-05, 1.2553e-04,  ..., 3.6469e-02,
           1.4038e-03, 1.8970e-01]],

         [[4.7144e-01, 2.0385e-04, 2.1279e-04,  ..., 2.3102e-02,
           4.3518e-02, 8.9966e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 8.0526e-05, 6.2525e-05,  ..., 5.5481e-02,
           1.2598e-01, 1.1121e-01]],

         [[1.7786e-01, 4.0054e-05, 2.0742e-05,  ..., 1.7932e-01,
           2.2668e-01, 1.5210e-01]],

         [[4.2480e-01, 2.0897e-04, 1.7095e-04,  ..., 3.7415e-02,
           5.4550e-03, 1.8234e-02]],

         ...,

         [[4.0698e-01, 0.0000e+00, 8.9407e-07,  ..., 2.9251e-02,
           4.8315e-01, 6.9092e-02]],

         [[1.9678e-01, 5.9605e-07, 3.9935e-06,  ..., 6.1920e-02,
           2.3880e-02, 1.8738e-02]],

         [[1.7822e-01, 0.0000e+00, 5.9605e-07,  ..., 6.3538e-02,
           7.0459e-01, 3.7354e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2619e-02, 1.4901e-06, 1.6093e-06,  ..., 5.6122e-02,
           1.7834e-03, 1.7517e-02]],

         [[1.8262e-01, 4.8280e-06, 2.0862e-06,  ..., 7.3624e-03,
           1.3863e-02, 7.3975e-01]],

         [[4.9835e-02, 3.8028e-05, 2.1851e-04,  ..., 3.0994e-04,
           2.6512e-04, 1.0757e-03]],

         ...,

         [[9.4910e-03, 3.1233e-05, 1.2046e-04,  ..., 1.7490e-03,
           2.9755e-03, 3.4523e-03]],

         [[1.8768e-02, 0.0000e+00, 0.0000e+00,  ..., 7.5439e-02,
           2.2125e-02, 1.9073e-02]],

         [[5.4077e-02, 0.0000e+00, 5.9605e-08,  ..., 1.7871e-01,
           5.7227e-01, 7.6233e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 1.2767e-04, 5.2977e-04,  ..., 7.0038e-03,
           1.9882e-02, 2.4536e-02]],

         [[9.6924e-01, 1.8096e-04, 1.3137e-04,  ..., 5.5969e-05,
           2.4021e-04, 6.0987e-04]],

         [[9.3652e-01, 6.1989e-06, 1.5438e-05,  ..., 3.1490e-03,
           3.4313e-03, 4.6883e-03]],

         ...,

         [[3.2324e-01, 4.1723e-07, 4.7684e-07,  ..., 1.3062e-01,
           1.7041e-01, 8.4351e-02]],

         [[8.4668e-01, 6.3181e-06, 9.1791e-06,  ..., 5.3883e-04,
           8.4782e-04, 4.3411e-03]],

         [[1.3220e-01, 4.6492e-06, 1.0371e-05,  ..., 5.4026e-04,
           1.4687e-03, 8.4863e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8958e-01, 2.3842e-07, 8.9407e-07,  ..., 3.6896e-02,
           1.4307e-01, 3.0298e-01]],

         [[4.2822e-01, 6.5565e-07, 7.1526e-07,  ..., 6.6162e-02,
           1.5747e-01, 1.2366e-01]],

         [[3.3252e-01, 3.1531e-05, 4.3750e-05,  ..., 2.6886e-02,
           7.7148e-02, 5.7129e-02]],

         ...,

         [[1.6113e-01, 6.5565e-07, 2.5034e-06,  ..., 1.0321e-01,
           5.5664e-02, 4.2694e-02]],

         [[4.2993e-01, 4.5896e-06, 2.3365e-05,  ..., 4.6051e-02,
           3.5309e-02, 3.1403e-02]],

         [[7.0410e-01, 3.2783e-06, 1.7762e-05,  ..., 4.0985e-02,
           5.1270e-03, 1.0361e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4053e-01, 1.9789e-05, 1.2755e-05,  ..., 1.0565e-01,
           8.6670e-03, 1.9128e-01]],

         [[5.1465e-01, 6.7949e-06, 5.1856e-06,  ..., 4.4983e-02,
           1.0513e-02, 1.7517e-01]],

         [[4.2896e-01, 5.4538e-05, 1.4532e-04,  ..., 3.0838e-02,
           2.0065e-02, 6.4758e-02]],

         ...,

         [[1.0645e-01, 3.4571e-06, 4.4823e-05,  ..., 1.4183e-02,
           1.8219e-02, 6.0539e-03]],

         [[1.8750e-01, 3.3379e-05, 2.1613e-04,  ..., 1.7538e-03,
           1.2150e-03, 6.9189e-04]],

         [[2.7161e-02, 2.7418e-06, 6.3181e-06,  ..., 9.0942e-03,
           6.1569e-03, 1.3031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0828e-01, 5.9605e-08, 3.5763e-07,  ..., 2.1484e-02,
           8.0615e-01, 4.1901e-02]],

         [[5.0232e-02, 1.7881e-07, 1.6093e-06,  ..., 3.1605e-03,
           2.3670e-03, 1.5198e-02]],

         [[3.6084e-01, 3.5763e-07, 2.6822e-06,  ..., 1.6846e-02,
           4.2847e-01, 1.7651e-01]],

         ...,

         [[6.5918e-01, 5.6446e-05, 2.6345e-05,  ..., 1.5625e-02,
           1.5869e-02, 4.3152e-02]],

         [[2.6392e-01, 4.4703e-06, 5.8413e-06,  ..., 1.5759e-01,
           7.2144e-02, 8.4351e-02]],

         [[1.0876e-01, 2.3067e-05, 7.6652e-05,  ..., 5.6854e-02,
           1.1945e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3887e-01, 3.8147e-06, 1.6749e-05,  ..., 4.9591e-02,
           1.0834e-01, 3.5522e-01]],

         [[7.8174e-01, 8.7023e-06, 3.0756e-05,  ..., 3.5034e-02,
           2.5131e-02, 4.5166e-02]],

         [[4.8779e-01, 6.3956e-05, 1.2898e-04,  ..., 1.0468e-01,
           1.7288e-02, 9.9854e-02]],

         ...,

         [[1.0498e-01, 1.1921e-05, 6.8545e-06,  ..., 3.4088e-02,
           1.0239e-02, 1.8539e-02]],

         [[4.9072e-02, 1.8954e-05, 1.1975e-04,  ..., 1.4503e-02,
           4.8309e-02, 8.1689e-01]],

         [[2.3120e-01, 7.4506e-06, 1.7464e-05,  ..., 1.2189e-01,
           1.5894e-01, 6.1279e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6875e-01, 5.7518e-05, 2.1303e-04,  ..., 9.3750e-02,
           2.1289e-01, 7.5317e-02]],

         [[2.9761e-01, 1.9610e-05, 5.5969e-05,  ..., 3.3783e-02,
           1.8909e-01, 1.7224e-01]],

         [[6.5479e-01, 1.5008e-04, 5.3596e-04,  ..., 1.6449e-02,
           1.9852e-02, 1.7624e-02]],

         ...,

         [[5.1123e-01, 4.8220e-05, 3.0303e-04,  ..., 3.0106e-02,
           1.7029e-02, 9.7733e-03]],

         [[8.8965e-01, 6.0797e-06, 4.3654e-04,  ..., 2.5768e-03,
           1.4023e-02, 8.4076e-03]],

         [[5.7080e-01, 5.4240e-06, 4.7266e-05,  ..., 1.8387e-02,
           1.8173e-02, 2.6077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9743e-05, 2.6286e-05, 1.7261e-01,  ..., 1.0974e-01,
           4.6444e-04, 5.3177e-03]],

         [[1.3292e-05, 7.0333e-06, 5.0598e-02,  ..., 4.1290e-02,
           5.7220e-03, 3.2864e-03]],

         [[9.1672e-05, 4.7445e-05, 6.9092e-02,  ..., 1.0699e-01,
           2.3364e-01, 1.1212e-01]],

         ...,

         [[8.9109e-05, 7.7486e-06, 6.7688e-02,  ..., 7.7271e-02,
           1.3892e-01, 1.6235e-01]],

         [[4.4560e-04, 1.3046e-03, 2.9517e-01,  ..., 1.0460e-02,
           4.3297e-03, 6.6490e-03]],

         [[1.6272e-04, 8.7440e-05, 3.9258e-01,  ..., 4.8294e-03,
           8.3923e-04, 1.4496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6921e-04, 4.7493e-04, 3.1104e-01,  ..., 7.4844e-03,
           2.5320e-04, 2.2926e-03]],

         [[2.2364e-04, 2.8944e-04, 2.7246e-01,  ..., 2.4567e-02,
           4.3273e-04, 4.0741e-03]],

         [[1.9777e-04, 9.9242e-05, 3.2617e-01,  ..., 6.2599e-03,
           5.6219e-04, 2.7514e-04]],

         ...,

         [[1.5509e-04, 3.4404e-04, 3.0371e-01,  ..., 3.8548e-03,
           1.5574e-03, 5.9366e-04]],

         [[9.9540e-06, 5.8830e-05, 3.0688e-01,  ..., 7.8659e-03,
           8.7500e-05, 4.0007e-04]],

         [[1.9073e-05, 1.1712e-04, 1.9128e-01,  ..., 5.2605e-03,
           1.4186e-04, 3.6502e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1127e-04, 2.1541e-04, 2.3218e-01,  ..., 1.1269e-02,
           1.0663e-04, 2.3098e-03]],

         [[4.0865e-04, 1.7524e-05, 3.1421e-01,  ..., 5.3825e-03,
           6.8712e-04, 2.0385e-05]],

         [[5.1451e-04, 4.2415e-04, 3.1104e-01,  ..., 9.5444e-03,
           4.1509e-04, 4.7922e-05]],

         ...,

         [[1.2267e-04, 2.7061e-05, 1.2695e-01,  ..., 1.0199e-01,
           6.6895e-02, 7.4890e-02]],

         [[7.7903e-05, 1.4806e-04, 3.1348e-01,  ..., 5.6076e-03,
           2.0046e-03, 4.9067e-04]],

         [[5.1260e-04, 2.0289e-04, 2.2046e-01,  ..., 5.9433e-03,
           9.4795e-04, 6.5279e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6061e-05, 1.4305e-06, 2.5902e-03,  ..., 7.1106e-02,
           2.1228e-01, 3.7750e-02]],

         [[4.1056e-04, 7.3814e-04, 1.2805e-01,  ..., 3.5706e-02,
           3.7354e-02, 3.6133e-02]],

         [[6.1226e-04, 3.0079e-03, 1.2201e-01,  ..., 6.5923e-05,
           7.1824e-05, 5.2452e-04]],

         ...,

         [[4.1847e-03, 5.6148e-05, 1.9153e-01,  ..., 1.2169e-02,
           1.8600e-02, 5.2765e-02]],

         [[1.1444e-03, 3.4332e-05, 1.1353e-02,  ..., 8.1299e-02,
           6.1523e-01, 1.1920e-01]],

         [[4.1127e-06, 8.3447e-07, 6.1083e-04,  ..., 9.9411e-03,
           1.1261e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6670e-03, 9.6035e-04, 2.3938e-01,  ..., 1.1482e-03,
           3.2845e-03, 4.7882e-02]],

         [[7.7057e-04, 1.0767e-03, 2.6318e-01,  ..., 3.3989e-03,
           1.5621e-03, 1.4168e-02]],

         [[3.7766e-03, 8.3637e-04, 2.1362e-01,  ..., 1.7746e-02,
           5.7281e-02, 2.8915e-02]],

         ...,

         [[2.9302e-04, 3.7372e-05, 8.6487e-02,  ..., 4.9934e-03,
           1.2169e-02, 2.4765e-02]],

         [[1.8120e-04, 8.3780e-04, 1.0309e-01,  ..., 6.0608e-02,
           7.6599e-02, 4.4775e-01]],

         [[3.3081e-05, 3.9935e-06, 4.0009e-02,  ..., 1.2915e-01,
           3.6206e-01, 1.8494e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[5.7280e-05, 2.4438e-06, 1.5569e-04,  ..., 2.9678e-02,
           1.8750e-01, 6.1475e-01]],

         [[1.3527e-02, 2.5826e-03, 7.7095e-03,  ..., 9.0103e-03,
           3.6774e-02, 2.0837e-01]],

         [[3.8147e-05, 3.8147e-05, 1.5396e-02,  ..., 1.7107e-04,
           1.3316e-04, 5.0977e-01]],

         ...,

         [[1.7853e-03, 1.3905e-03, 1.3065e-03,  ..., 9.0637e-03,
           3.4760e-02, 7.0610e-03]],

         [[6.9466e-03, 5.4092e-03, 3.7537e-02,  ..., 3.0823e-03,
           1.5656e-02, 2.0300e-01]],

         [[4.2686e-03, 4.0131e-03, 1.1379e-04,  ..., 2.4567e-02,
           4.4495e-02, 8.2321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4717e-05, 4.8018e-04, 2.2202e-03,  ..., 3.5828e-02,
           7.7026e-02, 8.5938e-02]],

         [[8.3208e-05, 7.3910e-06, 2.0862e-06,  ..., 1.2561e-01,
           5.2100e-01, 3.3105e-01]],

         [[3.6201e-03, 5.2681e-03, 2.1164e-02,  ..., 4.2343e-03,
           3.9787e-03, 9.3918e-03]],

         ...,

         [[6.6423e-04, 4.5300e-04, 2.6894e-03,  ..., 2.6535e-02,
           1.3657e-02, 5.8868e-02]],

         [[3.8528e-03, 5.6953e-03, 5.0964e-02,  ..., 2.8973e-03,
           6.6566e-03, 1.7471e-02]],

         [[5.9223e-04, 4.6043e-03, 3.2387e-03,  ..., 4.3335e-02,
           6.5063e-02, 2.0349e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1244e-03, 9.2268e-05, 1.7107e-04,  ..., 1.6943e-01,
           1.8909e-01, 1.2695e-01]],

         [[4.9973e-03, 1.9188e-03, 3.8013e-03,  ..., 1.5511e-02,
           1.4343e-02, 1.2032e-02]],

         [[1.3475e-03, 1.2100e-04, 1.0151e-04,  ..., 9.0332e-03,
           7.9727e-03, 4.6204e-02]],

         ...,

         [[1.2112e-04, 2.9802e-07, 3.5763e-06,  ..., 2.2202e-02,
           2.8125e-01, 6.4355e-01]],

         [[4.1618e-03, 6.4325e-04, 1.5678e-03,  ..., 1.3123e-02,
           6.2981e-03, 5.6122e-02]],

         [[1.7914e-02, 1.6794e-03, 1.0132e-02,  ..., 1.7227e-02,
           3.2196e-02, 1.5601e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8574e-01, 5.9605e-07, 4.2915e-06,  ..., 2.5730e-03,
           2.3556e-03, 1.7120e-02]],

         [[9.3799e-01, 1.6689e-05, 3.0398e-06,  ..., 6.5079e-03,
           9.3002e-03, 1.7075e-02]],

         [[8.5156e-01, 3.5763e-07, 1.6093e-06,  ..., 8.0414e-03,
           1.4000e-03, 2.5986e-02]],

         ...,

         [[8.1641e-01, 2.7609e-04, 1.2865e-03,  ..., 2.6608e-04,
           2.0862e-04, 3.2215e-03]],

         [[1.3391e-01, 5.9187e-05, 1.5533e-04,  ..., 3.1052e-02,
           2.0691e-02, 5.5786e-02]],

         [[8.2764e-01, 1.8513e-04, 6.0415e-04,  ..., 3.3360e-03,
           8.2932e-03, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4082e-01, 1.2815e-05, 4.9472e-06,  ..., 9.9421e-05,
           2.2471e-05, 9.0408e-03]],

         [[9.6875e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1683e-03,
           1.8860e-02, 9.5520e-03]],

         [[8.7158e-01, 7.8392e-04, 7.6115e-05,  ..., 4.6997e-03,
           1.8311e-02, 8.7814e-03]],

         ...,

         [[3.6353e-01, 1.9562e-04, 8.5950e-05,  ..., 6.6986e-03,
           7.6408e-03, 1.4946e-02]],

         [[7.4890e-02, 2.3842e-03, 1.1606e-03,  ..., 3.6550e-04,
           8.7166e-04, 3.4447e-03]],

         [[7.0264e-01, 7.9274e-06, 1.0967e-05,  ..., 8.3740e-02,
           8.8623e-02, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3315e-01, 3.6645e-04, 1.4246e-04,  ..., 4.4434e-02,
           3.3478e-02, 4.4617e-02]],

         [[6.6602e-01, 1.9121e-04, 1.7107e-05,  ..., 5.7755e-03,
           2.4891e-03, 2.5085e-02]],

         [[4.2334e-01, 5.2691e-05, 2.7478e-05,  ..., 1.9196e-02,
           4.5380e-02, 1.4610e-02]],

         ...,

         [[8.7305e-01, 1.2100e-05, 6.5565e-07,  ..., 4.5598e-05,
           3.8862e-05, 1.1383e-01]],

         [[8.6523e-01, 2.8193e-05, 3.9935e-06,  ..., 6.5565e-07,
           2.9802e-07, 1.7044e-02]],

         [[9.6240e-01, 9.3579e-06, 4.7684e-07,  ..., 1.1921e-07,
           5.9605e-08, 1.4771e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7812e-01, 6.4969e-06, 7.1526e-07,  ..., 7.7271e-02,
           9.5703e-02, 1.1299e-02]],

         [[9.1650e-01, 1.2040e-05, 2.0266e-06,  ..., 2.2674e-04,
           1.6081e-04, 8.0490e-03]],

         [[7.5879e-01, 7.8630e-04, 9.1732e-05,  ..., 7.1573e-04,
           6.4898e-04, 3.1403e-02]],

         ...,

         [[8.0469e-01, 8.6963e-05, 2.8670e-05,  ..., 1.8072e-03,
           3.9434e-04, 1.5358e-02]],

         [[8.3740e-01, 3.0398e-05, 1.4424e-05,  ..., 1.1307e-02,
           1.6975e-03, 3.2318e-02]],

         [[9.2773e-01, 9.7334e-05, 6.3837e-05,  ..., 1.0424e-03,
           7.3910e-04, 1.2993e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4443e-01, 1.6391e-05, 1.1265e-05,  ..., 2.1469e-02,
           1.5762e-02, 5.6305e-02]],

         [[7.0654e-01, 9.2864e-05, 4.8876e-06,  ..., 2.9984e-03,
           3.9711e-03, 1.4355e-01]],

         [[4.2651e-01, 2.7418e-06, 7.1526e-07,  ..., 5.5771e-03,
           1.3336e-02, 4.0527e-02]],

         ...,

         [[2.9297e-01, 8.6188e-05, 3.4273e-05,  ..., 8.6609e-02,
           8.1970e-02, 2.0233e-02]],

         [[5.6543e-01, 7.2384e-04, 6.8963e-05,  ..., 1.4122e-02,
           3.4882e-02, 2.1454e-02]],

         [[1.4136e-01, 9.6560e-06, 1.8299e-05,  ..., 2.2308e-02,
           1.3725e-02, 8.2642e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9536e-01, 2.8591e-03, 2.5635e-03,  ..., 1.0271e-03,
           1.7681e-03, 4.7874e-03]],

         [[8.0664e-01, 6.0141e-05, 3.2759e-04,  ..., 6.3286e-03,
           3.4008e-03, 1.0979e-02]],

         [[4.4458e-01, 2.7323e-04, 6.2883e-05,  ..., 1.5918e-01,
           3.9490e-02, 2.3514e-02]],

         ...,

         [[8.4961e-01, 3.2234e-04, 7.1347e-05,  ..., 1.4639e-04,
           6.4731e-05, 4.7333e-02]],

         [[2.6172e-01, 2.5868e-05, 7.2479e-05,  ..., 1.2083e-03,
           5.4598e-04, 6.4354e-03]],

         [[9.5801e-01, 4.0770e-05, 3.9220e-05,  ..., 8.4000e-03,
           6.1560e-04, 5.9700e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8853e-01, 9.3555e-04, 2.1496e-03,  ..., 1.5211e-04,
           1.0455e-04, 1.2993e-02]],

         [[9.1357e-01, 7.6294e-06, 2.8610e-05,  ..., 5.3864e-03,
           5.1804e-03, 5.4512e-03]],

         [[3.0151e-01, 1.9312e-05, 2.1601e-04,  ..., 2.2888e-02,
           1.3130e-02, 1.9470e-02]],

         ...,

         [[1.5793e-02, 8.1718e-05, 3.1018e-04,  ..., 3.4094e-05,
           3.3021e-05, 5.3825e-03]],

         [[8.9990e-01, 7.9811e-05, 7.9334e-05,  ..., 1.4259e-02,
           1.4442e-02, 2.6951e-03]],

         [[5.7129e-01, 2.6047e-05, 3.0696e-05,  ..., 1.5116e-03,
           4.1628e-04, 1.4488e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7720e-01, 9.1672e-05, 9.0075e-04,  ..., 1.0967e-04,
           8.6427e-06, 6.0028e-02]],

         [[3.0298e-01, 6.5207e-05, 6.9678e-05,  ..., 4.1199e-04,
           8.4877e-04, 3.2715e-02]],

         [[7.0605e-01, 1.5497e-06, 2.4915e-05,  ..., 3.8195e-04,
           4.1068e-05, 4.6631e-02]],

         ...,

         [[8.5889e-01, 1.9276e-04, 7.3433e-05,  ..., 2.1629e-03,
           4.2496e-03, 7.8278e-03]],

         [[4.6973e-01, 8.4066e-04, 2.6608e-04,  ..., 1.0147e-02,
           5.7945e-03, 2.1820e-02]],

         [[3.1128e-01, 8.1301e-04, 2.5797e-04,  ..., 2.4414e-04,
           2.5082e-04, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 3.9339e-06, 6.5565e-06,  ..., 1.2434e-04,
           7.9060e-04, 7.6611e-01]],

         [[6.3818e-01, 1.8282e-03, 2.4261e-03,  ..., 3.2902e-03,
           8.8196e-03, 2.1000e-03]],

         [[6.1670e-01, 5.5790e-04, 5.7840e-04,  ..., 6.8398e-03,
           1.7365e-02, 5.8746e-02]],

         ...,

         [[5.0830e-01, 1.7643e-04, 2.5988e-04,  ..., 1.0155e-02,
           5.8556e-03, 9.6497e-02]],

         [[5.0000e-01, 2.0683e-05, 2.4140e-05,  ..., 2.7679e-02,
           6.4270e-02, 5.3650e-02]],

         [[4.3237e-01, 1.1754e-04, 1.7786e-04,  ..., 3.0017e-04,
           1.4486e-03, 4.4678e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 3.7122e-04, 5.2166e-04,  ..., 1.6127e-03,
           2.5272e-03, 3.5107e-01]],

         [[8.3252e-01, 2.0719e-04, 5.4359e-04,  ..., 4.6692e-03,
           7.7286e-03, 1.3199e-02]],

         [[8.3984e-01, 3.4285e-04, 3.6788e-04,  ..., 1.1511e-03,
           3.3360e-03, 2.1912e-02]],

         ...,

         [[1.4355e-01, 3.6716e-05, 3.5703e-05,  ..., 4.1604e-05,
           4.0650e-04, 7.7783e-01]],

         [[4.8242e-01, 2.0981e-04, 1.9789e-04,  ..., 4.9257e-04,
           8.2636e-04, 3.7598e-02]],

         [[7.2754e-01, 6.4552e-05, 2.9325e-05,  ..., 2.1942e-02,
           1.0010e-01, 8.9645e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1953e-01, 2.2781e-04, 2.0337e-04,  ..., 3.0792e-02,
           1.0431e-01, 2.5482e-02]],

         [[7.8223e-01, 1.9908e-05, 2.1040e-05,  ..., 3.9642e-02,
           7.9193e-03, 2.4765e-02]],

         [[4.1602e-01, 1.0413e-04, 2.2662e-04,  ..., 2.2769e-05,
           9.7156e-06, 2.8061e-02]],

         ...,

         [[9.5801e-01, 0.0000e+00, 0.0000e+00,  ..., 2.2926e-03,
           2.3743e-02, 1.3809e-02]],

         [[5.8936e-01, 1.4191e-03, 1.9321e-03,  ..., 8.5926e-04,
           1.9131e-03, 1.3786e-02]],

         [[5.0635e-01, 2.3842e-07, 0.0000e+00,  ..., 1.7944e-02,
           3.6377e-01, 5.6244e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8809e-01, 1.0598e-04, 3.6776e-05,  ..., 2.9510e-02,
           8.1329e-03, 1.7822e-02]],

         [[3.1763e-01, 1.5974e-03, 1.0738e-03,  ..., 5.0640e-04,
           5.1641e-04, 1.2878e-01]],

         [[7.7588e-01, 3.0327e-04, 2.0754e-04,  ..., 1.9140e-03,
           4.5252e-04, 2.1988e-02]],

         ...,

         [[4.7754e-01, 2.5034e-06, 2.0266e-06,  ..., 5.0507e-03,
           1.7662e-03, 2.6337e-02]],

         [[5.5469e-01, 7.6294e-06, 3.9339e-06,  ..., 2.0325e-02,
           5.1498e-03, 3.0716e-02]],

         [[2.7930e-01, 8.8811e-05, 1.5771e-04,  ..., 4.7278e-04,
           4.8018e-04, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8877e-02, 6.3777e-06, 5.9009e-06,  ..., 1.0204e-04,
           2.9254e-04, 8.7109e-01]],

         [[6.2842e-01, 1.8477e-05, 3.2187e-05,  ..., 1.1650e-02,
           6.2866e-03, 5.1880e-02]],

         [[2.3035e-01, 1.1425e-03, 2.0084e-03,  ..., 7.0238e-04,
           1.5736e-03, 3.1052e-02]],

         ...,

         [[3.4741e-01, 9.5308e-05, 1.0884e-04,  ..., 6.3667e-03,
           4.8180e-03, 5.5023e-02]],

         [[4.9194e-01, 4.2915e-06, 4.7088e-06,  ..., 8.1873e-04,
           1.7917e-04, 5.8197e-02]],

         [[9.8450e-02, 4.2558e-05, 1.8895e-05,  ..., 2.4979e-02,
           1.1151e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9792e-02, 5.4836e-06, 1.4901e-06,  ..., 2.5879e-01,
           3.2471e-01, 3.1158e-02]],

         [[3.2910e-01, 4.3917e-04, 5.2595e-04,  ..., 2.9488e-03,
           3.1757e-03, 8.9294e-02]],

         [[2.7148e-01, 1.9789e-05, 8.5235e-06,  ..., 1.1696e-02,
           6.0120e-02, 6.9763e-02]],

         ...,

         [[4.3481e-01, 1.3828e-05, 9.4175e-06,  ..., 4.4952e-02,
           3.0411e-02, 1.0248e-01]],

         [[6.1475e-01, 1.2994e-05, 2.1338e-05,  ..., 2.6917e-02,
           3.0853e-02, 2.2049e-02]],

         [[6.6895e-01, 1.7452e-04, 1.2839e-04,  ..., 3.7003e-03,
           2.9160e-02, 5.7770e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1261e-01, 1.0166e-03, 1.3332e-03,  ..., 2.7061e-05,
           1.0765e-04, 6.2561e-03]],

         [[3.0322e-01, 1.0195e-03, 8.7166e-04,  ..., 1.6546e-04,
           1.3266e-03, 2.8931e-02]],

         [[4.0015e-01, 8.9407e-07, 1.5497e-06,  ..., 8.3923e-03,
           1.0582e-02, 1.5221e-02]],

         ...,

         [[6.2939e-01, 1.8084e-04, 1.9193e-04,  ..., 3.6430e-04,
           1.1501e-03, 2.6550e-02]],

         [[9.7900e-01, 3.3438e-05, 6.6698e-05,  ..., 9.5367e-07,
           1.2517e-06, 3.7403e-03]],

         [[6.7236e-01, 2.1315e-04, 3.2210e-04,  ..., 1.6289e-03,
           2.1229e-03, 1.0941e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2424e-01, 4.6802e-04, 7.2479e-04,  ..., 8.4066e-04,
           2.4283e-04, 6.2256e-02]],

         [[3.5913e-01, 4.2458e-03, 2.2564e-03,  ..., 6.6423e-04,
           6.8808e-04, 6.7993e-02]],

         [[3.9856e-02, 2.8074e-05, 2.9325e-05,  ..., 1.0872e-03,
           6.2943e-04, 6.1377e-01]],

         ...,

         [[1.2152e-01, 9.9540e-06, 1.8299e-05,  ..., 1.4362e-03,
           1.9836e-03, 3.2959e-02]],

         [[4.5776e-02, 6.5565e-07, 1.7881e-07,  ..., 2.2192e-01,
           3.2544e-01, 3.5950e-02]],

         [[1.9116e-01, 1.0014e-05, 6.6757e-06,  ..., 1.1040e-02,
           2.7328e-02, 5.3613e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.1098e-04, 5.0008e-05,  ..., 6.1569e-03,
           3.2444e-03, 4.8431e-02]],

         [[4.4952e-02, 1.2875e-05, 2.3365e-05,  ..., 3.4122e-03,
           4.1809e-03, 2.9480e-02]],

         [[1.9238e-01, 3.2425e-05, 1.0073e-05,  ..., 1.0252e-03,
           1.2688e-02, 7.4902e-01]],

         ...,

         [[1.3367e-01, 1.4858e-03, 3.5977e-04,  ..., 1.8454e-04,
           4.9973e-04, 1.4091e-02]],

         [[3.4399e-01, 1.0786e-03, 7.0143e-04,  ..., 1.7233e-03,
           7.5293e-04, 1.4709e-02]],

         [[1.9922e-01, 7.3814e-04, 9.3699e-04,  ..., 7.9823e-04,
           2.2125e-03, 2.1317e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 3.1650e-05, 2.9385e-05,  ..., 1.4702e-02,
           2.0416e-02, 1.3953e-01]],

         [[9.7363e-01, 1.0848e-05, 3.5167e-06,  ..., 1.5795e-04,
           3.5226e-05, 2.9335e-03]],

         [[6.6992e-01, 1.4913e-04, 7.9811e-05,  ..., 7.0457e-03,
           1.8024e-03, 2.1713e-02]],

         ...,

         [[1.2366e-01, 1.3113e-03, 2.3727e-03,  ..., 1.0529e-03,
           1.8482e-03, 3.7140e-02]],

         [[1.4880e-01, 2.2173e-05, 2.0504e-05,  ..., 4.3976e-02,
           1.8225e-01, 3.4790e-02]],

         [[3.2642e-01, 1.9703e-03, 6.1188e-03,  ..., 1.7262e-03,
           6.3629e-03, 7.7515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7908e-01, 1.1140e-04, 2.3401e-04,  ..., 1.3466e-03,
           3.4122e-03, 1.7319e-02]],

         [[3.1421e-01, 3.9995e-05, 2.2662e-04,  ..., 7.7248e-04,
           3.5458e-03, 3.3112e-02]],

         [[3.1567e-01, 4.0174e-05, 1.1998e-04,  ..., 2.7523e-03,
           1.0887e-02, 8.0261e-03]],

         ...,

         [[1.3132e-03, 3.9339e-06, 1.0192e-05,  ..., 1.1921e-06,
           4.6492e-06, 1.0223e-03]],

         [[5.1758e-01, 1.5318e-04, 1.5080e-04,  ..., 2.8019e-03,
           2.0386e-02, 1.7065e-01]],

         [[7.2998e-01, 1.1843e-04, 1.6308e-04,  ..., 5.8403e-03,
           6.4163e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5625e-01, 6.2764e-05, 8.3447e-05,  ..., 4.7660e-04,
           9.6817e-03, 4.7668e-02]],

         [[5.8740e-01, 2.9945e-04, 5.9986e-04,  ..., 5.4836e-04,
           2.8744e-03, 1.5656e-02]],

         [[8.8928e-02, 1.1504e-05, 4.2379e-05,  ..., 2.0428e-03,
           1.0139e-02, 7.9883e-01]],

         ...,

         [[2.2510e-01, 3.7670e-05, 2.5749e-05,  ..., 4.7913e-02,
           1.5710e-01, 1.9556e-01]],

         [[6.3770e-01, 8.1837e-05, 3.0184e-04,  ..., 7.8869e-04,
           7.8869e-04, 2.2171e-02]],

         [[2.6807e-01, 7.5912e-04, 6.1464e-04,  ..., 6.0272e-04,
           4.1199e-03, 1.3351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 7.1466e-05, 3.6597e-05,  ..., 5.9738e-03,
           2.2995e-02, 1.0065e-01]],

         [[2.6047e-02, 1.5020e-05, 1.3828e-05,  ..., 5.7220e-03,
           3.7231e-03, 5.2216e-02]],

         [[2.1338e-01, 1.0663e-04, 1.3578e-04,  ..., 3.0258e-02,
           2.7557e-02, 4.9896e-02]],

         ...,

         [[3.2056e-01, 0.0000e+00, 5.9605e-08,  ..., 2.1301e-02,
           4.3799e-01, 1.5625e-01]],

         [[3.7085e-01, 1.8537e-05, 8.5533e-05,  ..., 6.4659e-04,
           2.5787e-03, 8.4045e-02]],

         [[8.7451e-01, 4.1723e-07, 1.4901e-06,  ..., 3.5172e-03,
           3.0151e-02, 5.0903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0933e-02, 4.5896e-06, 1.8537e-05,  ..., 1.2338e-05,
           1.4677e-03, 9.4421e-02]],

         [[2.9572e-02, 2.0385e-05, 5.0366e-05,  ..., 4.4894e-04,
           3.9711e-03, 9.1992e-01]],

         [[1.7383e-01, 1.4019e-04, 9.6750e-04,  ..., 1.6093e-05,
           1.6201e-04, 1.8906e-02]],

         ...,

         [[1.5955e-01, 5.6219e-04, 1.7376e-03,  ..., 8.4686e-04,
           2.9621e-03, 5.2002e-02]],

         [[3.8403e-01, 1.5497e-06, 5.3644e-06,  ..., 7.8487e-04,
           7.1869e-03, 6.9153e-02]],

         [[2.4927e-01, 1.4901e-06, 1.6689e-06,  ..., 3.9581e-02,
           2.3145e-01, 1.6345e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8506e-01, 1.7536e-04, 8.5068e-04,  ..., 6.0692e-03,
           6.8245e-03, 2.8503e-02]],

         [[9.8389e-01, 1.2982e-04, 2.1899e-04,  ..., 3.7551e-05,
           4.4584e-05, 6.8283e-04]],

         [[7.9688e-01, 2.3234e-04, 7.8583e-04,  ..., 3.0594e-03,
           3.4008e-03, 4.3335e-03]],

         ...,

         [[3.0487e-02, 5.9605e-07, 1.7285e-06,  ..., 5.2147e-03,
           4.7211e-02, 2.1622e-02]],

         [[3.1152e-01, 5.6624e-06, 5.2452e-06,  ..., 9.7454e-05,
           2.7537e-04, 1.7138e-03]],

         [[1.7444e-01, 3.3081e-05, 1.0431e-04,  ..., 3.1638e-04,
           5.4779e-03, 7.2266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1151e-01, 6.1989e-06, 2.0683e-05,  ..., 3.5839e-03,
           1.6388e-02, 1.3550e-01]],

         [[6.2103e-02, 1.5497e-06, 1.5497e-06,  ..., 3.8574e-02,
           2.4561e-01, 5.5237e-02]],

         [[1.7932e-01, 7.2479e-05, 1.0580e-04,  ..., 2.5345e-02,
           9.4543e-02, 2.4368e-02]],

         ...,

         [[1.4075e-01, 1.0908e-05, 6.7472e-05,  ..., 3.3760e-03,
           1.6296e-02, 8.5388e-02]],

         [[3.7280e-01, 9.0003e-06, 1.0014e-04,  ..., 2.0905e-03,
           7.7400e-03, 7.7515e-02]],

         [[5.3809e-01, 5.7220e-06, 2.9087e-05,  ..., 1.0033e-03,
           7.2670e-03, 1.3501e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2953e-03, 2.3842e-07, 2.3842e-07,  ..., 3.0577e-05,
           1.5342e-04, 4.1962e-03]],

         [[1.8384e-01, 3.5167e-06, 2.1458e-06,  ..., 2.4643e-03,
           2.1622e-02, 9.8190e-03]],

         [[6.1676e-02, 1.5914e-05, 5.3048e-05,  ..., 5.6038e-03,
           2.1820e-02, 5.9326e-02]],

         ...,

         [[1.4087e-01, 8.2970e-05, 5.3692e-04,  ..., 1.7185e-03,
           1.5049e-03, 1.3867e-01]],

         [[1.6028e-01, 5.8270e-04, 2.0504e-03,  ..., 5.9652e-04,
           4.1008e-04, 1.2292e-01]],

         [[1.2085e-01, 3.2902e-04, 9.0122e-04,  ..., 2.2469e-03,
           1.4320e-02, 8.1116e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 4.3511e-06, 1.5795e-05,  ..., 1.7078e-01,
           2.8870e-02, 7.9880e-03]],

         [[1.0748e-01, 5.8413e-06, 3.3557e-05,  ..., 1.7614e-03,
           8.0200e-02, 3.1592e-01]],

         [[6.9141e-01, 3.5763e-07, 6.5565e-07,  ..., 9.5215e-03,
           9.8450e-02, 1.7139e-01]],

         ...,

         [[5.4736e-01, 1.8799e-04, 5.2738e-04,  ..., 1.9064e-03,
           4.2305e-03, 2.2415e-02]],

         [[4.3921e-01, 8.5235e-06, 4.5180e-05,  ..., 2.4338e-03,
           4.8027e-03, 3.3875e-02]],

         [[7.0923e-02, 3.2210e-04, 4.0555e-04,  ..., 2.7344e-02,
           1.5343e-02, 5.2832e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6711e-01, 4.8697e-05, 2.1243e-04,  ..., 1.5915e-02,
           1.4046e-02, 5.9143e-02]],

         [[1.5857e-01, 5.3177e-03, 5.3177e-03,  ..., 2.9030e-03,
           3.1147e-03, 3.8391e-02]],

         [[3.3911e-01, 9.5272e-04, 1.1339e-03,  ..., 1.3000e-02,
           1.8616e-02, 6.8909e-02]],

         ...,

         [[6.8799e-01, 3.8052e-04, 4.8494e-04,  ..., 2.4433e-03,
           4.8218e-03, 1.4389e-02]],

         [[2.3108e-01, 7.0190e-04, 1.2131e-03,  ..., 3.3493e-03,
           4.3427e-02, 5.6299e-01]],

         [[7.7295e-01, 3.7718e-04, 1.4687e-03,  ..., 2.3289e-03,
           3.8109e-03, 4.1595e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0354e-02, 6.1393e-06, 5.9128e-05,  ..., 6.0368e-04,
           7.8297e-04, 1.3904e-01]],

         [[5.7617e-01, 2.1732e-04, 2.6679e-04,  ..., 2.2430e-03,
           8.1558e-03, 2.2125e-02]],

         [[5.2393e-01, 7.0858e-04, 1.7052e-03,  ..., 1.4099e-02,
           5.3314e-02, 4.6417e-02]],

         ...,

         [[9.4482e-02, 1.0193e-02, 8.6212e-03,  ..., 5.7364e-04,
           3.5405e-04, 9.9182e-03]],

         [[8.0566e-01, 4.3213e-05, 4.1771e-04,  ..., 1.0902e-02,
           2.4815e-03, 5.8289e-03]],

         [[3.4058e-01, 2.4140e-04, 3.7026e-04,  ..., 7.9346e-04,
           1.3056e-03, 2.6459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9397e-04, 2.3060e-03, 1.8689e-01,  ..., 2.5535e-04,
           8.9121e-04, 8.8623e-02]],

         [[4.3678e-04, 9.9182e-04, 1.7249e-01,  ..., 4.7798e-03,
           1.9159e-03, 3.8177e-02]],

         [[5.0783e-04, 6.8855e-04, 1.2402e-01,  ..., 2.6001e-02,
           4.1466e-03, 5.3986e-02]],

         ...,

         [[3.3832e-04, 2.4366e-04, 1.2878e-01,  ..., 1.9245e-03,
           1.4305e-03, 2.4817e-01]],

         [[3.3398e-03, 1.5442e-02, 4.1321e-02,  ..., 1.0568e-04,
           8.9347e-05, 8.8167e-04]],

         [[1.0319e-03, 6.9857e-04, 4.1870e-02,  ..., 1.0080e-03,
           5.9509e-04, 6.7825e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8501e-03, 2.8157e-04, 1.9543e-01,  ..., 7.3576e-04,
           1.1902e-03, 1.8188e-02]],

         [[2.3766e-03, 2.5196e-03, 1.9336e-01,  ..., 5.0211e-04,
           4.0131e-03, 2.8732e-02]],

         [[2.5787e-03, 4.0865e-04, 2.9102e-01,  ..., 1.3959e-04,
           3.0756e-05, 3.9703e-02]],

         ...,

         [[1.6449e-02, 1.9875e-03, 1.6266e-02,  ..., 1.3294e-03,
           5.5313e-05, 2.5425e-03]],

         [[2.9588e-04, 6.6948e-04, 2.7173e-01,  ..., 4.7088e-04,
           3.9577e-04, 1.1932e-02]],

         [[2.1954e-03, 5.0278e-03, 1.3451e-02,  ..., 3.2187e-06,
           9.1195e-06, 6.6071e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5191e-03, 1.0977e-03, 1.3599e-01,  ..., 1.8358e-03,
           2.1011e-02, 4.7150e-02]],

         [[3.2768e-03, 4.3511e-05, 1.2732e-01,  ..., 8.7051e-03,
           3.1769e-05, 2.6077e-02]],

         [[8.6670e-03, 4.3068e-03, 2.0911e-01,  ..., 4.1084e-03,
           6.0260e-05, 5.3711e-02]],

         ...,

         [[1.5907e-03, 8.6164e-04, 1.8396e-01,  ..., 8.8730e-03,
           1.7071e-03, 4.1504e-02]],

         [[8.2111e-04, 8.5258e-04, 1.4880e-01,  ..., 4.5746e-02,
           1.0529e-03, 2.1271e-02]],

         [[7.8964e-03, 1.0672e-03, 5.7251e-02,  ..., 4.6997e-03,
           3.6573e-04, 6.1035e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6594e-04, 9.5069e-05, 2.5909e-02,  ..., 4.5204e-03,
           7.3242e-02, 3.6035e-01]],

         [[1.1196e-03, 5.0974e-04, 2.8784e-01,  ..., 1.1078e-02,
           3.9101e-03, 6.7711e-03]],

         [[5.1918e-03, 1.6251e-02, 4.3221e-03,  ..., 2.2948e-05,
           9.0241e-05, 7.6294e-02]],

         ...,

         [[2.1935e-03, 1.2970e-03, 2.7298e-02,  ..., 3.7527e-04,
           4.5090e-03, 4.8022e-01]],

         [[1.7710e-03, 2.6131e-04, 1.3344e-02,  ..., 1.2178e-03,
           1.2733e-02, 8.7207e-01]],

         [[6.8626e-03, 1.2598e-03, 2.2156e-02,  ..., 2.4471e-03,
           1.5900e-02, 6.3770e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5907e-03, 1.4849e-03, 1.7029e-01,  ..., 1.8097e-02,
           4.2236e-02, 5.3406e-02]],

         [[2.3117e-03, 6.4754e-04, 2.6733e-01,  ..., 8.4066e-04,
           3.0899e-03, 4.9286e-02]],

         [[4.5204e-03, 1.2493e-03, 2.6099e-01,  ..., 3.4294e-03,
           7.0381e-03, 5.9601e-02]],

         ...,

         [[6.8188e-04, 2.6083e-04, 1.3940e-01,  ..., 1.1778e-03,
           1.3069e-02, 3.2153e-01]],

         [[3.9601e-04, 2.8477e-03, 2.4268e-01,  ..., 4.7302e-03,
           1.1501e-03, 1.3501e-01]],

         [[2.8229e-04, 3.7611e-05, 4.3549e-02,  ..., 4.4136e-03,
           8.9874e-03, 6.1084e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.2812e-03, 1.2088e-04, 9.8133e-04,  ..., 1.0010e-01,
           6.3232e-01, 1.1340e-01]],

         [[7.6246e-04, 1.1688e-04, 6.7282e-04,  ..., 1.3244e-04,
           2.8610e-02, 9.4727e-01]],

         [[8.3447e-07, 1.3709e-06, 2.2650e-06,  ..., 2.2650e-06,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[3.1686e-04, 6.7091e-04, 4.0698e-04,  ..., 3.8605e-03,
           4.7028e-02, 7.3584e-01]],

         [[5.2719e-03, 4.6539e-03, 9.2545e-03,  ..., 2.6779e-02,
           7.7454e-02, 7.7454e-02]],

         [[6.7291e-03, 3.2806e-03, 2.0969e-04,  ..., 1.4702e-02,
           5.5809e-03, 2.2064e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5310e-04, 1.9388e-03, 5.8784e-03,  ..., 2.9861e-02,
           2.4231e-01, 1.2183e-01]],

         [[2.7955e-05, 4.9829e-05, 1.0610e-05,  ..., 1.0376e-01,
           8.2861e-01, 6.0974e-02]],

         [[1.3132e-03, 3.1013e-03, 1.0651e-02,  ..., 9.6970e-03,
           5.0842e-02, 2.6367e-02]],

         ...,

         [[1.3857e-03, 1.7653e-03, 6.7673e-03,  ..., 6.8359e-02,
           1.0260e-01, 2.8717e-02]],

         [[3.3550e-03, 9.0790e-03, 3.3081e-02,  ..., 2.0706e-02,
           6.2317e-02, 2.9541e-02]],

         [[2.3210e-04, 1.7290e-03, 4.8757e-04,  ..., 5.3772e-02,
           8.3447e-01, 3.3661e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2163e-03, 8.3780e-04, 2.6207e-03,  ..., 1.0468e-01,
           1.5234e-01, 8.6121e-02]],

         [[1.3533e-03, 3.4213e-04, 1.8787e-03,  ..., 3.3245e-03,
           3.7956e-03, 3.2082e-03]],

         [[2.8172e-03, 4.1389e-04, 4.6515e-04,  ..., 1.2627e-02,
           9.3994e-02, 1.0468e-02]],

         ...,

         [[3.9253e-03, 7.9930e-05, 1.2875e-04,  ..., 7.5867e-02,
           6.0107e-01, 1.9373e-01]],

         [[5.8889e-04, 1.1420e-04, 2.7609e-04,  ..., 5.3329e-03,
           7.7591e-03, 6.4316e-03]],

         [[3.1342e-02, 4.9973e-03, 1.1002e-02,  ..., 2.4033e-02,
           2.9980e-01, 4.8553e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1836e-01, 3.2783e-06, 1.1921e-06,  ..., 4.2677e-04,
           5.3864e-03, 1.8682e-03]],

         [[9.5703e-01, 4.2319e-06, 5.3644e-07,  ..., 1.5926e-03,
           2.8259e-02, 4.2381e-03]],

         [[7.4023e-01, 5.3644e-07, 2.3842e-07,  ..., 1.1909e-02,
           1.1253e-02, 2.4765e-02]],

         ...,

         [[1.7749e-01, 1.7941e-05, 1.7500e-04,  ..., 1.5998e-04,
           1.0826e-02, 1.1253e-02]],

         [[1.1304e-01, 2.8014e-06, 3.9101e-05,  ..., 4.5929e-03,
           2.3605e-02, 6.9237e-03]],

         [[7.3291e-01, 3.3903e-04, 2.4939e-04,  ..., 2.4071e-03,
           1.4519e-02, 5.8317e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0469e-01, 2.5034e-06, 1.5497e-06,  ..., 4.7469e-04,
           8.2855e-03, 5.2460e-02]],

         [[2.9590e-01, 5.3644e-07, 0.0000e+00,  ..., 1.4969e-02,
           6.7236e-01, 1.6357e-02]],

         [[9.6094e-01, 1.2034e-04, 5.5432e-06,  ..., 4.3182e-03,
           1.2863e-02, 4.5319e-03]],

         ...,

         [[5.7764e-01, 5.3883e-05, 2.5630e-05,  ..., 3.4580e-03,
           5.2368e-02, 1.5366e-02]],

         [[1.2732e-01, 4.9448e-04, 4.5753e-04,  ..., 9.5272e-04,
           1.4511e-02, 4.2343e-03]],

         [[2.3584e-01, 1.4305e-05, 7.6890e-06,  ..., 5.1422e-02,
           5.5469e-01, 3.7170e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2690e-01, 2.1076e-04, 8.1897e-05,  ..., 3.6316e-02,
           6.1981e-02, 2.3117e-02]],

         [[6.4600e-01, 4.5061e-05, 9.3579e-06,  ..., 6.8092e-03,
           1.2610e-01, 2.7267e-02]],

         [[5.2344e-01, 6.3181e-06, 7.5102e-06,  ..., 1.2062e-02,
           4.9225e-02, 1.2917e-02]],

         ...,

         [[8.4863e-01, 1.2159e-05, 7.1526e-06,  ..., 1.2636e-05,
           1.7748e-03, 1.4331e-01]],

         [[9.4727e-01, 2.0444e-05, 2.2006e-04,  ..., 1.7881e-07,
           3.2365e-05, 2.4902e-02]],

         [[9.8877e-01, 2.0266e-06, 2.3961e-05,  ..., 0.0000e+00,
           5.3048e-06, 8.7738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8379e-01, 1.4246e-05, 2.6822e-06,  ..., 7.4829e-02,
           3.3154e-01, 2.0752e-02]],

         [[6.2549e-01, 2.7418e-06, 7.7486e-07,  ..., 2.5129e-04,
           4.7073e-03, 5.6458e-03]],

         [[9.0430e-01, 2.5940e-04, 3.1829e-05,  ..., 1.6880e-04,
           1.1078e-02, 4.2534e-03]],

         ...,

         [[6.3135e-01, 6.9082e-05, 9.7156e-06,  ..., 7.0858e-04,
           6.4011e-03, 2.6642e-02]],

         [[2.3303e-01, 7.3314e-05, 6.9141e-06,  ..., 3.9635e-03,
           4.7112e-03, 4.6173e-02]],

         [[9.0479e-01, 9.0122e-05, 6.9320e-05,  ..., 2.7542e-03,
           1.6159e-02, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6685e-01, 8.3447e-07, 9.5367e-07,  ..., 7.3181e-02,
           9.3201e-02, 1.3901e-02]],

         [[4.1064e-01, 4.1366e-05, 2.3305e-05,  ..., 1.3687e-02,
           9.1370e-02, 1.0242e-01]],

         [[7.7783e-01, 1.5378e-05, 2.5034e-06,  ..., 4.3869e-03,
           2.5101e-02, 5.9280e-03]],

         ...,

         [[6.9629e-01, 4.3535e-04, 9.6381e-05,  ..., 1.1658e-02,
           9.3842e-03, 2.5482e-02]],

         [[8.4131e-01, 1.9455e-04, 5.1141e-05,  ..., 3.8166e-03,
           1.1665e-02, 4.9210e-03]],

         [[7.3193e-01, 3.8147e-06, 6.3777e-06,  ..., 8.7967e-03,
           5.7495e-02, 3.2990e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5811e-01, 1.1663e-03, 9.9754e-04,  ..., 1.7786e-03,
           5.6992e-03, 9.8991e-04]],

         [[9.1650e-01, 6.6221e-05, 2.0158e-04,  ..., 1.7967e-03,
           5.8022e-03, 5.0812e-03]],

         [[7.0801e-01, 4.1664e-05, 1.7941e-05,  ..., 2.8931e-02,
           4.2191e-03, 4.3869e-03]],

         ...,

         [[6.3721e-01, 5.2392e-05, 1.1265e-05,  ..., 1.8275e-04,
           9.3994e-03, 2.4414e-01]],

         [[8.7207e-01, 1.4722e-05, 4.5180e-05,  ..., 1.1683e-03,
           3.3207e-03, 3.6407e-02]],

         [[9.0820e-01, 1.3769e-05, 1.9491e-05,  ..., 4.2076e-03,
           1.8740e-03, 7.2556e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0764e-01, 4.5228e-04, 1.3475e-03,  ..., 2.0638e-03,
           1.2219e-01, 6.3324e-03]],

         [[8.7891e-01, 3.2783e-06, 6.7949e-06,  ..., 2.6150e-03,
           5.7129e-02, 2.7039e-02]],

         [[3.5498e-01, 1.7285e-06, 1.9073e-05,  ..., 7.5951e-03,
           7.3586e-03, 1.9623e-02]],

         ...,

         [[5.2979e-01, 8.4281e-05, 1.4448e-04,  ..., 1.1683e-03,
           6.1096e-02, 1.5545e-03]],

         [[8.6279e-01, 6.8605e-05, 3.9458e-05,  ..., 7.8087e-03,
           6.0806e-03, 7.3013e-03]],

         [[8.7354e-01, 5.2452e-06, 1.0192e-05,  ..., 1.9112e-03,
           4.8103e-03, 1.0963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2891e-01, 1.9431e-05, 3.3975e-05,  ..., 1.7786e-04,
           1.8204e-02, 6.8398e-03]],

         [[7.5537e-01, 1.1444e-05, 1.0848e-05,  ..., 1.8097e-02,
           4.8248e-02, 1.9272e-02]],

         [[1.1560e-01, 0.0000e+00, 1.7881e-07,  ..., 2.0046e-03,
           6.7993e-02, 5.6496e-03]],

         ...,

         [[9.2480e-01, 4.9174e-05, 2.2054e-05,  ..., 3.1452e-03,
           4.4708e-03, 4.3068e-03]],

         [[4.7559e-01, 8.1837e-05, 2.1303e-04,  ..., 2.2736e-03,
           1.5594e-02, 2.7710e-01]],

         [[5.4932e-01, 2.9731e-04, 7.4625e-05,  ..., 2.6512e-03,
           6.6948e-03, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2329e-01, 8.9407e-07, 1.7285e-06,  ..., 2.4629e-04,
           5.5237e-03, 8.5938e-01]],

         [[7.9004e-01, 8.5545e-04, 7.7820e-04,  ..., 1.7868e-02,
           2.0561e-03, 5.5742e-04]],

         [[6.1328e-01, 4.6372e-04, 6.3705e-04,  ..., 2.4353e-02,
           5.6824e-02, 1.4824e-02]],

         ...,

         [[5.0098e-01, 7.5817e-05, 1.5867e-04,  ..., 1.3222e-02,
           2.7527e-02, 1.3077e-02]],

         [[1.6199e-01, 4.0293e-05, 3.6657e-05,  ..., 6.8092e-03,
           1.0139e-02, 4.4128e-02]],

         [[1.8250e-01, 3.3677e-05, 1.1617e-04,  ..., 3.7789e-04,
           7.5378e-02, 6.5332e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9983e-01, 2.4259e-05, 4.6253e-05,  ..., 2.7733e-03,
           4.0741e-02, 6.1328e-01]],

         [[6.5869e-01, 3.1495e-04, 3.0541e-04,  ..., 2.9053e-02,
           6.1340e-02, 5.0659e-03]],

         [[6.1572e-01, 1.3328e-04, 6.9141e-05,  ..., 7.9117e-03,
           4.4136e-03, 1.2039e-02]],

         ...,

         [[9.5444e-03, 5.9605e-07, 4.2915e-06,  ..., 1.9789e-05,
           7.6294e-04, 9.8291e-01]],

         [[6.2158e-01, 6.4850e-05, 5.1320e-05,  ..., 3.3875e-03,
           6.6147e-03, 1.2866e-01]],

         [[7.1094e-01, 5.1200e-05, 1.5497e-05,  ..., 6.7444e-02,
           2.4319e-03, 2.7027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1069e-01, 3.5000e-04, 4.9543e-04,  ..., 8.6914e-02,
           3.6865e-02, 8.6288e-03]],

         [[8.5254e-01, 6.2585e-05, 1.0520e-04,  ..., 4.4556e-03,
           5.7007e-02, 1.2695e-02]],

         [[2.8931e-01, 7.0333e-06, 2.7478e-05,  ..., 1.3895e-03,
           2.5345e-02, 1.1719e-02]],

         ...,

         [[2.1570e-01, 5.9605e-08, 0.0000e+00,  ..., 9.9487e-03,
           7.5293e-01, 2.0203e-02]],

         [[4.7241e-01, 5.7745e-04, 6.9618e-04,  ..., 1.0422e-02,
           2.1652e-02, 8.3847e-03]],

         [[8.9502e-01, 1.7285e-06, 2.9802e-07,  ..., 1.9791e-02,
           3.1219e-02, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1445e-01, 4.6909e-05, 1.6451e-05,  ..., 5.1575e-03,
           1.9512e-03, 7.4501e-03]],

         [[1.2988e-01, 3.9005e-04, 3.7003e-04,  ..., 3.0780e-04,
           2.6123e-01, 4.5471e-03]],

         [[4.2432e-01, 2.6393e-04, 2.4164e-04,  ..., 6.9427e-03,
           1.2744e-01, 1.4198e-02]],

         ...,

         [[4.1724e-01, 7.9274e-06, 2.9206e-06,  ..., 9.4452e-03,
           4.1260e-01, 1.0382e-01]],

         [[7.6367e-01, 2.1458e-06, 1.0133e-06,  ..., 6.4611e-04,
           1.4636e-01, 6.5735e-02]],

         [[4.6704e-01, 2.1398e-05, 1.2279e-05,  ..., 1.2112e-03,
           3.4229e-01, 5.9052e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5054e-02, 2.5034e-06, 5.1856e-06,  ..., 5.9605e-05,
           1.5465e-02, 8.9551e-01]],

         [[4.2554e-01, 4.5300e-06, 5.8413e-06,  ..., 9.7275e-03,
           3.4204e-01, 1.3440e-01]],

         [[2.5952e-01, 1.6391e-04, 2.0516e-04,  ..., 2.2602e-03,
           4.8096e-02, 9.1248e-03]],

         ...,

         [[1.9678e-01, 1.6093e-05, 6.7949e-06,  ..., 3.8300e-02,
           1.0889e-01, 1.3206e-02]],

         [[2.6904e-01, 1.4305e-06, 5.9605e-07,  ..., 2.0993e-04,
           2.2925e-01, 1.6495e-02]],

         [[3.1470e-01, 1.3423e-04, 4.8816e-05,  ..., 9.3155e-03,
           9.8419e-03, 2.8946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1519e-01, 3.4332e-05, 2.3425e-05,  ..., 4.7760e-02,
           9.4604e-02, 8.4000e-03]],

         [[9.7778e-02, 6.3539e-05, 1.6809e-05,  ..., 9.0942e-03,
           1.6504e-01, 1.2924e-02]],

         [[9.9426e-02, 6.0201e-05, 5.1260e-06,  ..., 5.2071e-03,
           2.6962e-02, 2.7176e-02]],

         ...,

         [[4.8560e-01, 1.0610e-05, 7.6294e-06,  ..., 1.7822e-02,
           1.3074e-01, 1.9196e-02]],

         [[5.4639e-01, 1.5199e-05, 1.7285e-05,  ..., 5.6488e-02,
           6.2744e-02, 3.1067e-02]],

         [[2.7222e-01, 2.7180e-05, 1.5557e-05,  ..., 1.2659e-01,
           2.2388e-01, 5.9433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.6648e-04, 7.8249e-04,  ..., 6.2108e-05,
           1.1971e-02, 1.4162e-03]],

         [[1.9238e-01, 7.9870e-05, 1.4853e-04,  ..., 1.1635e-03,
           3.1647e-02, 8.2703e-03]],

         [[4.6680e-01, 4.2319e-06, 4.4703e-06,  ..., 4.2992e-03,
           1.4624e-01, 3.4607e-02]],

         ...,

         [[6.4062e-01, 4.9472e-05, 2.3067e-05,  ..., 1.5078e-03,
           1.9272e-02, 2.8763e-03]],

         [[9.8535e-01, 1.8299e-05, 2.4736e-05,  ..., 1.9073e-06,
           2.0313e-04, 2.8057e-03]],

         [[6.9629e-01, 1.0759e-04, 1.2779e-04,  ..., 9.3613e-03,
           2.8183e-02, 3.4237e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8408e-01, 9.5606e-05, 1.1444e-04,  ..., 6.8474e-04,
           1.2067e-01, 1.1406e-02]],

         [[1.5649e-01, 8.0872e-04, 5.0020e-04,  ..., 1.3786e-02,
           1.0510e-01, 1.8402e-02]],

         [[9.5703e-02, 4.3511e-06, 3.9339e-06,  ..., 1.9703e-03,
           7.3364e-02, 5.1318e-01]],

         ...,

         [[2.4561e-01, 3.9935e-06, 3.5167e-06,  ..., 8.3252e-02,
           2.4756e-01, 1.6586e-02]],

         [[4.1943e-01, 7.3910e-06, 6.1393e-06,  ..., 1.7517e-02,
           5.3314e-02, 2.4597e-02]],

         [[1.6187e-01, 2.0266e-06, 2.8014e-06,  ..., 5.7602e-03,
           1.9080e-01, 2.4683e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4873e-01, 9.5963e-06, 6.0201e-06,  ..., 1.2146e-02,
           6.4636e-02, 7.7209e-03]],

         [[2.1765e-01, 1.9073e-05, 2.9504e-05,  ..., 1.7414e-03,
           4.9347e-02, 3.6102e-02]],

         [[1.0931e-01, 4.5896e-06, 2.0862e-06,  ..., 4.5657e-05,
           1.8997e-02, 8.2031e-01]],

         ...,

         [[8.6670e-02, 7.9632e-05, 2.8849e-05,  ..., 4.9055e-05,
           2.8515e-03, 6.4278e-03]],

         [[3.5498e-01, 3.3665e-04, 1.7321e-04,  ..., 4.7379e-03,
           1.2581e-02, 3.6755e-03]],

         [[2.5513e-01, 2.0456e-04, 1.3053e-04,  ..., 1.4397e-02,
           7.6050e-02, 1.3847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9365e-02, 2.4319e-05, 1.1086e-05,  ..., 6.5674e-02,
           1.0333e-01, 2.2690e-02]],

         [[8.9893e-01, 1.1718e-04, 2.2292e-05,  ..., 4.5300e-06,
           7.7009e-04, 3.9406e-03]],

         [[9.1016e-01, 7.4804e-05, 1.7822e-05,  ..., 2.9862e-05,
           7.0429e-04, 2.4204e-03]],

         ...,

         [[7.7332e-02, 5.8556e-04, 1.2016e-03,  ..., 5.5580e-03,
           5.1514e-02, 3.8490e-03]],

         [[4.2896e-01, 1.0610e-05, 2.0564e-05,  ..., 4.8126e-02,
           3.3081e-02, 2.2736e-02]],

         [[3.9526e-01, 5.1212e-04, 1.5650e-03,  ..., 1.9669e-02,
           5.7831e-02, 7.9498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9004e-01, 3.2246e-05, 7.1228e-05,  ..., 5.7030e-03,
           1.2848e-02, 1.7151e-02]],

         [[1.4648e-01, 3.1710e-05, 1.0008e-04,  ..., 2.0580e-03,
           2.1774e-02, 3.8208e-02]],

         [[1.4307e-01, 1.2755e-05, 3.4034e-05,  ..., 6.2370e-03,
           6.9008e-03, 4.9324e-03]],

         ...,

         [[1.5747e-02, 1.3053e-05, 3.0875e-05,  ..., 1.7583e-05,
           3.6259e-03, 5.5313e-03]],

         [[1.9165e-01, 4.7028e-05, 4.6313e-05,  ..., 2.5034e-04,
           1.2451e-02, 9.2468e-03]],

         [[5.5859e-01, 7.2300e-05, 1.2010e-04,  ..., 9.4681e-03,
           7.8506e-03, 9.7656e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2632e-02, 5.4240e-06, 7.5698e-06,  ..., 1.9817e-03,
           1.0971e-02, 2.0493e-02]],

         [[2.0801e-01, 8.6486e-05, 1.4102e-04,  ..., 1.7042e-03,
           3.9635e-03, 7.5760e-03]],

         [[1.6565e-01, 2.9385e-05, 1.5521e-04,  ..., 1.4219e-03,
           9.6619e-02, 4.9463e-01]],

         ...,

         [[2.7295e-01, 1.8477e-05, 1.2696e-05,  ..., 2.1713e-02,
           1.0120e-01, 5.4893e-03]],

         [[1.6248e-01, 1.7107e-05, 7.6592e-05,  ..., 2.8348e-04,
           7.2517e-03, 4.4434e-02]],

         [[2.9004e-01, 2.4796e-05, 3.2842e-05,  ..., 4.4990e-04,
           6.9275e-03, 1.8072e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5063e-02, 3.5703e-05, 2.2948e-05,  ..., 2.8362e-03,
           2.3560e-02, 3.6774e-02]],

         [[1.2863e-02, 4.1723e-06, 7.7486e-06,  ..., 1.9681e-04,
           9.7046e-03, 2.1530e-02]],

         [[2.3523e-01, 1.4722e-05, 2.1040e-05,  ..., 1.4830e-03,
           8.6365e-03, 3.2520e-03]],

         ...,

         [[6.3232e-01, 1.1921e-07, 0.0000e+00,  ..., 1.0321e-01,
           1.8408e-01, 4.2023e-02]],

         [[5.3516e-01, 7.9870e-06, 1.9610e-05,  ..., 2.9953e-02,
           1.6321e-01, 2.6230e-02]],

         [[4.3066e-01, 1.1325e-06, 1.0729e-06,  ..., 1.3313e-02,
           4.9561e-01, 4.9438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2232e-02, 2.3842e-06, 1.7464e-05,  ..., 1.7548e-03,
           5.1270e-02, 2.4033e-02]],

         [[9.9106e-03, 4.7684e-06, 2.8849e-05,  ..., 8.1182e-05,
           3.2940e-03, 9.7998e-01]],

         [[1.0052e-01, 2.9087e-05, 2.0778e-04,  ..., 4.2558e-04,
           2.8137e-02, 6.0463e-04]],

         ...,

         [[1.9568e-01, 8.0884e-05, 3.4952e-04,  ..., 3.6888e-03,
           1.0760e-01, 8.1711e-03]],

         [[5.4541e-01, 5.9605e-07, 1.3113e-06,  ..., 8.3771e-03,
           9.9670e-02, 4.1565e-02]],

         [[2.0288e-01, 4.2915e-06, 2.7418e-06,  ..., 2.9572e-02,
           5.6250e-01, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 1.0943e-04, 6.6423e-04,  ..., 1.5472e-02,
           4.1565e-02, 6.5269e-03]],

         [[9.7266e-01, 4.9293e-05, 3.4595e-04,  ..., 1.5354e-04,
           6.4373e-04, 3.1519e-04]],

         [[8.1494e-01, 4.2260e-05, 1.0258e-04,  ..., 8.1787e-03,
           1.4580e-02, 1.3990e-03]],

         ...,

         [[3.2690e-01, 1.7285e-06, 3.0994e-06,  ..., 2.0889e-02,
           2.7252e-02, 1.1009e-02]],

         [[4.0112e-01, 4.7684e-07, 2.3842e-07,  ..., 3.8314e-04,
           1.8864e-03, 2.0714e-03]],

         [[1.2561e-01, 1.0192e-05, 3.2842e-05,  ..., 2.0266e-04,
           2.1835e-02, 7.8174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3025e-01, 1.8477e-06, 6.9737e-06,  ..., 1.8326e-02,
           3.3789e-01, 6.3232e-02]],

         [[2.8174e-01, 5.0664e-06, 3.8147e-06,  ..., 1.4966e-01,
           4.3701e-02, 4.2542e-02]],

         [[2.5195e-01, 1.4687e-04, 2.1279e-04,  ..., 2.2369e-02,
           1.2650e-02, 8.3923e-03]],

         ...,

         [[2.4707e-01, 1.1563e-05, 5.1856e-05,  ..., 3.8177e-02,
           9.0881e-02, 1.6678e-02]],

         [[5.3516e-01, 1.4782e-05, 7.3731e-05,  ..., 1.1734e-02,
           7.4158e-02, 9.0714e-03]],

         [[4.9780e-01, 2.0266e-06, 1.0014e-05,  ..., 7.3280e-03,
           1.6040e-01, 1.2077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6591e-02, 4.3511e-06, 3.9339e-06,  ..., 7.7903e-05,
           1.4191e-03, 1.1162e-02]],

         [[3.1396e-01, 3.9697e-05, 1.7881e-05,  ..., 3.0609e-02,
           9.9030e-03, 2.2745e-04]],

         [[2.7637e-01, 3.5167e-05, 7.2777e-05,  ..., 1.8372e-02,
           3.6804e-02, 4.2114e-03]],

         ...,

         [[3.0518e-02, 1.1802e-05, 1.1265e-04,  ..., 9.9659e-04,
           1.3782e-01, 6.7024e-03]],

         [[8.9355e-02, 5.4240e-05, 2.9802e-04,  ..., 1.8942e-04,
           2.3169e-01, 1.0674e-02]],

         [[6.7090e-01, 4.1127e-05, 4.9055e-05,  ..., 7.8125e-03,
           8.0139e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4248e-01, 1.7881e-05, 3.3140e-05,  ..., 2.0706e-02,
           1.7609e-02, 1.1237e-01]],

         [[2.4939e-01, 6.3777e-06, 3.7432e-05,  ..., 1.0236e-01,
           2.8809e-01, 1.3135e-01]],

         [[2.3352e-01, 1.8477e-06, 1.8477e-06,  ..., 1.4984e-02,
           6.4746e-01, 8.9355e-02]],

         ...,

         [[5.2783e-01, 8.0585e-05, 1.3816e-04,  ..., 4.1428e-03,
           1.9989e-02, 1.1299e-02]],

         [[1.5808e-01, 3.5167e-06, 7.6890e-06,  ..., 6.1798e-03,
           1.0291e-01, 8.8013e-02]],

         [[4.0436e-02, 9.1612e-05, 3.1471e-04,  ..., 2.8000e-02,
           5.3558e-02, 4.2139e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1567e-01, 4.2915e-05, 1.8322e-04,  ..., 3.9337e-02,
           3.5126e-02, 8.3771e-03]],

         [[6.6406e-01, 4.7064e-04, 1.0376e-03,  ..., 1.7118e-03,
           4.3869e-03, 1.6985e-03]],

         [[7.2266e-02, 8.4817e-05, 1.4651e-04,  ..., 1.9569e-03,
           4.7874e-03, 1.0767e-03]],

         ...,

         [[4.2603e-01, 4.9114e-04, 3.3212e-04,  ..., 1.4572e-02,
           1.7853e-02, 2.1881e-02]],

         [[7.5684e-02, 6.1870e-05, 2.0933e-04,  ..., 1.6113e-02,
           3.4119e-02, 7.8906e-01]],

         [[8.9795e-01, 3.5167e-05, 8.2374e-05,  ..., 5.7755e-03,
           1.4519e-02, 5.3406e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9746e-01, 8.9347e-05, 3.8981e-04,  ..., 1.8711e-03,
           2.2858e-02, 1.1909e-02]],

         [[1.3623e-01, 2.2531e-05, 6.2823e-05,  ..., 7.2899e-03,
           3.4851e-02, 3.4851e-02]],

         [[2.5635e-01, 2.7728e-04, 8.0156e-04,  ..., 1.5356e-01,
           5.8777e-02, 2.5818e-02]],

         ...,

         [[4.2871e-01, 1.4198e-04, 1.6222e-03,  ..., 2.8324e-03,
           2.4292e-02, 2.0447e-02]],

         [[8.6963e-01, 2.3484e-05, 6.9904e-04,  ..., 1.0386e-03,
           3.3321e-03, 1.8082e-03]],

         [[3.7573e-01, 3.8147e-05, 1.1253e-04,  ..., 2.1801e-03,
           1.1452e-02, 4.7760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0902e-05, 1.1158e-04, 1.7419e-01,  ..., 1.6813e-03,
           5.2734e-02, 1.4137e-02]],

         [[2.5213e-05, 9.3877e-05, 1.7822e-01,  ..., 5.0688e-04,
           1.7181e-02, 1.2230e-02]],

         [[1.4997e-04, 1.7262e-04, 1.0297e-01,  ..., 1.3039e-02,
           2.8580e-02, 1.1192e-02]],

         ...,

         [[2.0671e-04, 1.5295e-04, 1.6846e-01,  ..., 4.9286e-03,
           7.2449e-02, 2.2314e-01]],

         [[4.4751e-04, 9.3603e-04, 1.8921e-01,  ..., 2.4471e-03,
           1.4137e-02, 2.2598e-02]],

         [[2.3961e-05, 6.9559e-05, 1.6589e-01,  ..., 4.0710e-05,
           4.3182e-03, 1.8148e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1195e-05, 1.8930e-04, 2.8369e-01,  ..., 1.3530e-04,
           1.3802e-02, 2.5330e-03]],

         [[1.6189e-04, 1.9526e-04, 1.7554e-01,  ..., 1.1511e-03,
           1.0704e-02, 7.4863e-04]],

         [[7.0274e-05, 3.2544e-05, 3.1641e-01,  ..., 2.7776e-05,
           1.8265e-02, 5.8031e-04]],

         ...,

         [[4.0627e-04, 2.4438e-04, 2.8052e-01,  ..., 2.8133e-04,
           2.5284e-02, 1.2589e-03]],

         [[2.8670e-05, 1.3173e-04, 2.9102e-01,  ..., 1.2153e-04,
           8.8959e-03, 3.9244e-04]],

         [[1.4961e-05, 7.1764e-05, 6.4758e-02,  ..., 1.0127e-04,
           1.2976e-01, 2.3956e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0341e-04, 1.1975e-04, 1.8665e-01,  ..., 1.0738e-03,
           9.4376e-03, 4.4799e-04]],

         [[4.8923e-04, 3.4392e-05, 2.8149e-01,  ..., 1.7881e-05,
           1.0963e-02, 4.9496e-04]],

         [[1.5295e-04, 2.4796e-04, 3.1470e-01,  ..., 5.4836e-06,
           1.1826e-02, 1.0376e-03]],

         ...,

         [[1.6189e-04, 1.8084e-04, 1.4673e-01,  ..., 4.1466e-03,
           1.5656e-02, 8.6441e-03]],

         [[7.4744e-05, 2.1899e-04, 2.6489e-01,  ..., 2.5678e-04,
           1.1421e-02, 1.3189e-03]],

         [[4.5490e-04, 3.3855e-04, 1.7004e-01,  ..., 1.2720e-04,
           6.3171e-03, 1.2531e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7214e-04, 1.5378e-05, 3.3722e-02,  ..., 1.0864e-02,
           6.3362e-03, 3.3521e-01]],

         [[5.1856e-05, 9.5904e-05, 1.0199e-01,  ..., 4.5990e-02,
           1.2665e-02, 7.0374e-02]],

         [[2.2662e-04, 1.0223e-03, 3.5840e-01,  ..., 2.1267e-03,
           6.6795e-03, 1.5802e-03]],

         ...,

         [[2.4939e-04, 1.8656e-05, 3.3630e-02,  ..., 1.5610e-02,
           7.9870e-04, 3.6804e-02]],

         [[6.1455e-03, 8.9836e-04, 2.0667e-01,  ..., 4.0619e-02,
           3.8544e-02, 6.1829e-02]],

         [[2.6584e-05, 1.5259e-05, 1.3115e-02,  ..., 2.2232e-02,
           2.2449e-03, 3.0322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8515e-04, 8.1348e-04, 2.0020e-01,  ..., 1.4526e-02,
           4.5380e-02, 7.0801e-02]],

         [[3.7193e-04, 4.9496e-04, 2.4146e-01,  ..., 3.1185e-03,
           1.9745e-02, 3.3844e-02]],

         [[1.5402e-03, 1.3371e-03, 2.3975e-01,  ..., 6.9313e-03,
           4.4281e-02, 1.1971e-02]],

         ...,

         [[5.8651e-04, 9.4235e-05, 2.5586e-01,  ..., 2.2316e-03,
           2.2089e-04, 9.0332e-03]],

         [[1.7223e-03, 2.4109e-03, 1.2317e-01,  ..., 2.8954e-03,
           1.7609e-02, 2.8638e-01]],

         [[1.9121e-04, 6.4552e-05, 5.2612e-02,  ..., 2.3842e-03,
           2.7084e-02, 2.2498e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.3571e-03, 1.6880e-03, 2.2373e-03,  ..., 2.7515e-01,
           1.0126e-01, 4.5386e-01]],

         [[2.3537e-03, 8.1348e-04, 4.9820e-03,  ..., 1.5503e-01,
           7.3242e-02, 1.9714e-02]],

         [[2.9802e-07, 2.2650e-06, 1.2255e-04,  ..., 2.4605e-03,
           9.0504e-04, 9.9268e-01]],

         ...,

         [[6.0201e-06, 2.0981e-05, 3.4630e-05,  ..., 6.9523e-04,
           6.5956e-03, 9.7900e-01]],

         [[3.2806e-03, 3.2806e-03, 1.6663e-02,  ..., 2.9251e-02,
           2.2781e-02, 9.0088e-02]],

         [[4.3221e-03, 2.9716e-03, 2.4390e-04,  ..., 7.5874e-03,
           4.9469e-02, 1.6068e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-04, 2.5005e-03, 4.8180e-03,  ..., 1.7249e-01,
           1.2427e-01, 1.4307e-01]],

         [[3.1590e-06, 1.5748e-04, 1.7846e-04,  ..., 1.5247e-01,
           7.1582e-01, 1.1871e-01]],

         [[3.2578e-03, 3.3607e-03, 8.3237e-03,  ..., 7.7698e-02,
           3.8483e-02, 8.7204e-03]],

         ...,

         [[3.2663e-04, 5.4693e-04, 4.3964e-04,  ..., 1.4221e-02,
           7.2363e-01, 2.1387e-01]],

         [[4.0092e-03, 1.0078e-02, 3.0579e-02,  ..., 6.3477e-02,
           3.4119e-02, 1.8250e-02]],

         [[1.5473e-04, 1.0490e-03, 1.3554e-04,  ..., 7.6074e-01,
           1.2512e-01, 1.0597e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8295e-02, 1.7357e-03, 5.4092e-03,  ..., 1.8335e-01,
           7.7637e-02, 5.3772e-02]],

         [[2.1400e-03, 4.2486e-04, 1.8673e-03,  ..., 1.7748e-03,
           2.1400e-03, 1.9722e-03]],

         [[2.8286e-03, 4.8208e-04, 5.3120e-04,  ..., 8.3984e-02,
           1.0887e-02, 1.2672e-02]],

         ...,

         [[2.7809e-03, 4.1652e-04, 2.7752e-04,  ..., 2.6855e-01,
           4.3579e-01, 1.7334e-01]],

         [[8.0872e-04, 2.0611e-04, 5.3024e-04,  ..., 1.5869e-02,
           3.2318e-02, 1.3573e-02]],

         [[1.7685e-02, 3.4561e-03, 7.7286e-03,  ..., 2.2766e-01,
           9.1248e-02, 4.8462e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 3.5763e-07, 2.9802e-07,  ..., 8.8654e-03,
           1.8091e-03, 5.3482e-03]],

         [[9.7559e-01, 1.7405e-05, 2.3246e-06,  ..., 5.3444e-03,
           1.9274e-03, 6.0081e-04]],

         [[6.4258e-01, 6.5565e-07, 5.3644e-07,  ..., 6.2370e-03,
           5.0201e-03, 1.7029e-02]],

         ...,

         [[2.0813e-01, 1.3053e-05, 1.3220e-04,  ..., 9.3613e-03,
           2.2945e-03, 1.0567e-02]],

         [[2.9224e-01, 2.9206e-05, 1.6415e-04,  ..., 4.7333e-02,
           7.5493e-03, 4.4899e-03]],

         [[6.9189e-01, 1.5616e-04, 3.1257e-04,  ..., 2.6932e-02,
           1.2941e-03, 3.0746e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0576e-01, 1.4305e-06, 1.5497e-06,  ..., 9.2010e-03,
           3.0499e-03, 2.9968e-02]],

         [[2.4146e-01, 4.5359e-05, 1.7881e-07,  ..., 4.7668e-02,
           6.9434e-01, 1.5503e-02]],

         [[9.7168e-01, 4.0507e-04, 2.0862e-05,  ..., 8.0109e-03,
           5.6992e-03, 1.2283e-03]],

         ...,

         [[5.5908e-01, 5.4777e-05, 3.2842e-05,  ..., 1.6138e-01,
           2.4216e-02, 1.4381e-02]],

         [[7.3486e-02, 1.1444e-03, 6.1131e-04,  ..., 1.1742e-02,
           2.1858e-03, 2.6836e-03]],

         [[6.5137e-01, 3.8385e-05, 9.9540e-06,  ..., 1.7590e-01,
           3.8788e-02, 2.1301e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1204e-01, 1.2910e-04, 6.9618e-05,  ..., 9.3323e-02,
           3.2166e-02, 1.2115e-02]],

         [[5.3906e-01, 2.1458e-05, 5.6028e-06,  ..., 2.7856e-01,
           1.1554e-01, 2.0828e-02]],

         [[7.9932e-01, 1.1742e-05, 6.1989e-06,  ..., 2.0584e-02,
           5.1819e-02, 2.5421e-02]],

         ...,

         [[7.0117e-01, 8.2254e-06, 8.9407e-07,  ..., 2.0647e-04,
           1.2302e-03, 2.9346e-01]],

         [[9.5410e-01, 4.8161e-05, 3.8803e-05,  ..., 8.3447e-06,
           2.7275e-04, 2.9510e-02]],

         [[9.9072e-01, 2.4617e-05, 2.5630e-06,  ..., 7.7486e-07,
           3.3796e-05, 7.2937e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3613e-01, 2.1100e-05, 9.5367e-07,  ..., 8.3557e-02,
           2.8271e-01, 1.3374e-02]],

         [[4.2065e-01, 1.7881e-06, 1.3113e-06,  ..., 3.1776e-03,
           2.0337e-04, 4.1695e-03]],

         [[8.4766e-01, 9.1934e-04, 4.8161e-05,  ..., 1.7014e-02,
           5.6877e-03, 4.7607e-03]],

         ...,

         [[8.3105e-01, 3.4356e-04, 2.6882e-05,  ..., 1.0025e-02,
           2.0111e-02, 1.5686e-02]],

         [[5.9375e-01, 1.6069e-04, 1.8597e-05,  ..., 2.1301e-02,
           1.1115e-01, 7.9895e-02]],

         [[9.3701e-01, 7.1764e-05, 6.6400e-05,  ..., 1.5350e-02,
           3.0670e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9722e-01, 6.7353e-06, 1.0967e-05,  ..., 1.5381e-01,
           1.0139e-02, 1.3763e-02]],

         [[3.2275e-01, 5.5671e-05, 9.8348e-06,  ..., 9.3445e-02,
           4.7241e-02, 7.8308e-02]],

         [[9.0967e-01, 3.6538e-05, 1.0729e-05,  ..., 9.0408e-03,
           2.2232e-02, 5.6152e-03]],

         ...,

         [[5.7373e-01, 5.8651e-04, 2.5821e-04,  ..., 1.2543e-02,
           3.3051e-02, 8.0185e-03]],

         [[8.1250e-01, 3.9887e-04, 4.9114e-05,  ..., 6.1417e-03,
           5.9776e-03, 1.2627e-02]],

         [[3.9697e-01, 5.3048e-06, 5.2452e-06,  ..., 1.0669e-01,
           2.9150e-01, 3.6072e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4883e-01, 1.1597e-03, 1.0767e-03,  ..., 9.1934e-03,
           1.2388e-03, 4.3154e-04]],

         [[8.6328e-01, 4.1068e-05, 9.9301e-05,  ..., 6.9618e-03,
           1.8021e-02, 1.1436e-02]],

         [[5.8838e-01, 8.4281e-05, 3.3021e-05,  ..., 5.1537e-03,
           1.0887e-02, 8.4991e-03]],

         ...,

         [[7.8662e-01, 3.5346e-05, 2.0564e-05,  ..., 3.0251e-03,
           4.3396e-02, 3.0655e-02]],

         [[7.3291e-01, 5.3644e-06, 6.7949e-06,  ..., 2.9411e-03,
           1.9958e-01, 4.3030e-02]],

         [[9.3359e-01, 1.4842e-05, 4.8280e-06,  ..., 1.0824e-03,
           2.9221e-02, 5.4550e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0796e-01, 8.0252e-04, 1.2722e-03,  ..., 8.3618e-02,
           2.8057e-03, 1.8291e-03]],

         [[8.2812e-01, 1.6689e-06, 1.4305e-06,  ..., 2.9007e-02,
           1.2067e-01, 1.3832e-02]],

         [[6.7236e-01, 3.2783e-06, 8.5831e-06,  ..., 1.1345e-02,
           2.6276e-02, 1.2238e-02]],

         ...,

         [[3.0591e-01, 2.8777e-04, 3.8147e-04,  ..., 1.0907e-01,
           1.0862e-03, 1.7300e-03]],

         [[9.4629e-01, 2.4676e-05, 1.7285e-05,  ..., 3.3875e-03,
           1.2817e-02, 4.7646e-03]],

         [[9.4043e-01, 6.4969e-06, 7.0333e-06,  ..., 2.9697e-03,
           4.7226e-03, 2.2736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9639e-01, 5.1677e-05, 9.6560e-05,  ..., 6.6956e-02,
           1.5526e-03, 7.6103e-03]],

         [[7.8418e-01, 1.6332e-05, 1.1861e-05,  ..., 3.0487e-02,
           2.0752e-02, 1.2756e-02]],

         [[2.3975e-01, 7.7486e-07, 2.0266e-06,  ..., 1.1108e-01,
           7.9155e-04, 3.3836e-03]],

         ...,

         [[8.4570e-01, 1.2541e-04, 9.1016e-05,  ..., 8.2474e-03,
           8.9111e-03, 2.5574e-02]],

         [[4.1992e-01, 2.6083e-04, 3.2091e-04,  ..., 1.5411e-02,
           1.3672e-01, 1.9751e-01]],

         [[4.1113e-01, 3.8981e-04, 2.7061e-04,  ..., 7.7324e-03,
           2.2125e-03, 2.8553e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0017e-02, 3.5763e-07, 1.4901e-06,  ..., 4.7946e-04,
           1.2909e-02, 9.0137e-01]],

         [[6.8311e-01, 1.8139e-03, 2.0866e-03,  ..., 2.4414e-03,
           8.4686e-04, 2.2829e-04]],

         [[5.4541e-01, 5.9891e-04, 4.3130e-04,  ..., 9.8572e-02,
           1.9638e-02, 3.8147e-02]],

         ...,

         [[6.1670e-01, 1.2362e-04, 1.8334e-04,  ..., 3.0533e-02,
           1.5671e-02, 3.6049e-03]],

         [[2.0129e-01, 4.4227e-05, 7.2062e-05,  ..., 3.7048e-02,
           2.5537e-01, 4.2114e-02]],

         [[1.2421e-01, 4.4167e-05, 2.7180e-04,  ..., 1.1559e-02,
           5.9082e-02, 7.4170e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2471e-01, 1.9038e-04, 2.0182e-04,  ..., 2.6260e-02,
           1.1462e-01, 4.0161e-01]],

         [[5.7959e-01, 1.6820e-04, 2.1601e-04,  ..., 9.9548e-02,
           4.0039e-02, 2.9251e-02]],

         [[4.9756e-01, 1.8835e-04, 9.3579e-05,  ..., 1.0307e-02,
           4.2786e-02, 4.3640e-03]],

         ...,

         [[1.7441e-02, 1.7881e-07, 2.6226e-06,  ..., 7.0214e-05,
           4.5509e-03, 9.7510e-01]],

         [[8.8330e-01, 5.1677e-05, 4.6313e-05,  ..., 2.2812e-03,
           1.3924e-02, 1.4694e-02]],

         [[6.7578e-01, 8.6546e-05, 4.9293e-05,  ..., 2.9964e-03,
           1.0391e-02, 2.5330e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 6.1178e-04, 6.1178e-04,  ..., 2.7740e-02,
           6.2904e-03, 7.3166e-03]],

         [[8.9551e-01, 4.2439e-05, 9.1970e-05,  ..., 2.4811e-02,
           2.0447e-02, 3.0212e-02]],

         [[4.2334e-01, 1.8060e-05, 2.9445e-05,  ..., 4.7760e-02,
           4.2534e-03, 2.1229e-03]],

         ...,

         [[4.4507e-01, 3.6955e-06, 1.7881e-07,  ..., 1.2164e-01,
           3.9893e-01, 3.1494e-02]],

         [[4.7217e-01, 5.9891e-04, 7.1383e-04,  ..., 1.4183e-02,
           9.3765e-03, 3.7556e-03]],

         [[6.6553e-01, 6.9737e-06, 1.2517e-06,  ..., 2.3376e-02,
           2.6367e-01, 3.4515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7041e-01, 9.1195e-05, 2.8372e-05,  ..., 2.6741e-03,
           1.5762e-02, 5.4474e-03]],

         [[1.8958e-01, 4.8566e-04, 5.2977e-04,  ..., 1.6724e-01,
           2.9430e-03, 2.5120e-03]],

         [[5.8789e-01, 7.4291e-04, 8.7881e-04,  ..., 6.5063e-02,
           5.1613e-03, 8.1177e-03]],

         ...,

         [[3.8892e-01, 9.5367e-06, 5.5432e-06,  ..., 4.0186e-01,
           1.3855e-01, 2.0538e-02]],

         [[4.4360e-01, 6.4373e-06, 1.4305e-06,  ..., 1.3525e-01,
           3.8428e-01, 2.8336e-02]],

         [[4.9365e-01, 4.3988e-05, 2.8133e-05,  ..., 2.7832e-01,
           5.2002e-02, 1.5640e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0660e-02, 6.5565e-07, 2.5034e-06,  ..., 1.0223e-03,
           1.5297e-02, 9.5361e-01]],

         [[3.3276e-01, 1.4305e-06, 2.0266e-06,  ..., 3.2642e-01,
           2.2437e-01, 4.9866e-02]],

         [[2.1619e-01, 2.7609e-04, 3.2210e-04,  ..., 4.1107e-02,
           6.4316e-03, 2.8229e-03]],

         ...,

         [[2.0923e-01, 3.3319e-05, 1.1683e-05,  ..., 1.4209e-01,
           2.5085e-02, 7.2250e-03]],

         [[2.1912e-01, 2.8014e-06, 1.7881e-06,  ..., 2.6489e-01,
           5.9967e-03, 6.7215e-03]],

         [[3.0469e-01, 1.5736e-04, 5.2691e-05,  ..., 1.2711e-02,
           2.4719e-02, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 5.9843e-05, 3.9399e-05,  ..., 7.4341e-02,
           6.7291e-03, 5.5008e-03]],

         [[7.9956e-02, 7.7009e-05, 1.7822e-05,  ..., 2.2253e-01,
           8.3313e-03, 3.5133e-03]],

         [[1.6064e-01, 9.7394e-05, 1.0133e-05,  ..., 7.1838e-02,
           6.0516e-02, 1.3611e-02]],

         ...,

         [[3.9941e-01, 1.8179e-05, 9.1195e-06,  ..., 1.9934e-01,
           1.6235e-02, 1.3832e-02]],

         [[7.0801e-01, 3.3319e-05, 5.3406e-05,  ..., 3.2990e-02,
           3.9642e-02, 4.6692e-02]],

         [[3.2397e-01, 4.7684e-05, 3.4213e-05,  ..., 1.9043e-01,
           7.3547e-03, 1.0132e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5564e-01, 1.6403e-03, 2.0981e-03,  ..., 2.0050e-02,
           9.1887e-04, 9.2649e-04]],

         [[6.7200e-02, 7.6652e-05, 1.2016e-04,  ..., 6.1401e-02,
           1.4359e-02, 6.5613e-03]],

         [[4.5874e-01, 6.9737e-06, 9.1791e-06,  ..., 1.4893e-01,
           1.0480e-01, 2.7328e-02]],

         ...,

         [[6.2109e-01, 7.8261e-05, 4.9293e-05,  ..., 1.5900e-02,
           6.4087e-03, 9.4376e-03]],

         [[9.7412e-01, 1.0073e-05, 2.6464e-05,  ..., 2.0981e-04,
           4.0984e-04, 6.3248e-03]],

         [[6.1377e-01, 8.2195e-05, 1.9908e-04,  ..., 4.2175e-02,
           7.0877e-03, 1.0056e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5928e-01, 1.7858e-04, 2.8086e-04,  ..., 1.2445e-01,
           7.1869e-03, 1.0216e-02]],

         [[2.3181e-01, 1.3571e-03, 7.2050e-04,  ..., 1.2598e-01,
           1.0841e-02, 6.0806e-03]],

         [[1.6064e-01, 6.6161e-06, 6.4373e-06,  ..., 2.6428e-02,
           2.1277e-01, 4.8706e-01]],

         ...,

         [[2.4915e-01, 4.5896e-06, 4.6492e-06,  ..., 2.7588e-01,
           1.9211e-02, 1.1208e-02]],

         [[5.3662e-01, 6.9141e-06, 4.7088e-06,  ..., 9.1797e-02,
           1.1334e-01, 2.2324e-02]],

         [[1.9690e-01, 3.9339e-06, 4.4703e-06,  ..., 1.1218e-01,
           3.7354e-01, 1.2915e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4131e-01, 7.8082e-06, 4.1723e-06,  ..., 1.4233e-01,
           3.6182e-03, 8.4152e-03]],

         [[1.6895e-01, 1.6749e-05, 1.9908e-05,  ..., 7.8552e-02,
           7.8552e-02, 1.9547e-02]],

         [[1.4099e-01, 4.8280e-06, 1.3709e-06,  ..., 5.5542e-03,
           4.4373e-02, 7.8662e-01]],

         ...,

         [[1.4014e-01, 4.9353e-04, 1.8156e-04,  ..., 1.2634e-02,
           7.0877e-03, 4.2000e-03]],

         [[4.1968e-01, 4.1699e-04, 2.7514e-04,  ..., 1.1765e-02,
           3.3436e-03, 4.0817e-03]],

         [[2.3889e-01, 3.2687e-04, 1.9991e-04,  ..., 8.2520e-02,
           1.7166e-02, 8.5678e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4319e-01, 6.7472e-05, 2.4557e-05,  ..., 6.6589e-02,
           2.8198e-02, 9.0866e-03]],

         [[8.8037e-01, 1.8489e-04, 1.0490e-05,  ..., 3.2246e-05,
           2.3782e-05, 4.5753e-04]],

         [[8.6816e-01, 6.7592e-05, 7.0333e-06,  ..., 2.9826e-04,
           1.2243e-04, 1.2655e-03]],

         ...,

         [[6.2988e-02, 9.9468e-04, 2.0256e-03,  ..., 4.6814e-02,
           2.6207e-03, 4.5538e-04]],

         [[4.4263e-01, 1.5497e-05, 2.2054e-05,  ..., 3.5217e-02,
           5.5389e-02, 4.0741e-03]],

         [[3.4497e-01, 6.3038e-04, 1.7958e-03,  ..., 7.2876e-02,
           9.6359e-03, 2.1545e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8062e-01, 1.3268e-04, 2.6393e-04,  ..., 7.4196e-03,
           1.7792e-02, 2.1469e-02]],

         [[3.0981e-01, 1.6224e-04, 3.9554e-04,  ..., 1.1200e-02,
           1.2794e-02, 4.1542e-03]],

         [[2.4011e-01, 4.4644e-05, 1.0341e-04,  ..., 1.1765e-02,
           6.3477e-03, 2.3727e-03]],

         ...,

         [[1.5839e-02, 2.6584e-05, 6.1750e-05,  ..., 1.0281e-03,
           3.0231e-03, 2.7523e-03]],

         [[1.1975e-01, 7.7426e-05, 8.1122e-05,  ..., 1.8356e-02,
           4.3602e-03, 3.2921e-03]],

         [[5.6885e-01, 9.3758e-05, 1.1945e-04,  ..., 8.7738e-03,
           7.1640e-03, 1.9283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-02, 1.2517e-05, 1.1742e-05,  ..., 1.5381e-02,
           1.9897e-02, 3.5686e-03]],

         [[1.9897e-01, 9.1910e-05, 1.2910e-04,  ..., 9.2316e-03,
           1.1398e-02, 4.7874e-03]],

         [[2.6025e-01, 9.1851e-05, 3.9744e-04,  ..., 5.2887e-02,
           3.4668e-02, 4.8242e-01]],

         ...,

         [[4.5239e-01, 1.9848e-05, 2.6524e-05,  ..., 7.0984e-02,
           6.7596e-03, 4.5395e-03]],

         [[4.8779e-01, 1.0359e-04, 2.9397e-04,  ..., 6.6910e-03,
           4.1962e-02, 1.8326e-02]],

         [[3.6035e-01, 4.0174e-05, 5.8651e-05,  ..., 2.5043e-03,
           9.8038e-04, 2.8820e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9182e-02, 5.5790e-05, 3.0696e-05,  ..., 2.7985e-02,
           2.7756e-02, 1.0826e-02]],

         [[2.0645e-02, 5.6624e-06, 9.0599e-06,  ..., 2.7130e-02,
           7.3181e-02, 1.6232e-03]],

         [[3.5132e-01, 4.8280e-05, 5.8651e-05,  ..., 1.7014e-02,
           7.2632e-03, 8.4534e-03]],

         ...,

         [[4.7656e-01, 2.0266e-06, 3.5763e-07,  ..., 8.2825e-02,
           3.4863e-01, 2.0935e-02]],

         [[4.8462e-01, 1.1802e-05, 2.4259e-05,  ..., 1.8250e-01,
           7.0923e-02, 2.0309e-02]],

         [[4.0112e-01, 3.6955e-06, 3.5763e-06,  ..., 2.1143e-01,
           3.3789e-01, 3.7018e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6495e-02, 1.6689e-06, 1.2517e-05,  ..., 3.6896e-02,
           3.3081e-02, 1.0818e-02]],

         [[1.1261e-02, 2.3246e-06, 2.8014e-05,  ..., 7.9679e-04,
           1.7853e-02, 9.5947e-01]],

         [[9.7473e-02, 2.3127e-05, 1.3876e-04,  ..., 3.3417e-02,
           1.1482e-03, 1.4620e-03]],

         ...,

         [[2.9932e-01, 1.0496e-04, 3.4237e-04,  ..., 1.1450e-01,
           1.0246e-02, 6.1035e-03]],

         [[4.6362e-01, 6.5565e-07, 1.6093e-06,  ..., 7.2510e-02,
           5.3467e-02, 2.9877e-02]],

         [[2.3547e-01, 1.4246e-05, 8.5831e-06,  ..., 1.9226e-01,
           4.3311e-01, 1.0614e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2451e-01, 1.5986e-04, 9.2983e-04,  ..., 3.3630e-02,
           4.4975e-03, 5.4893e-03]],

         [[9.8389e-01, 2.1160e-05, 8.7440e-05,  ..., 4.6897e-04,
           6.9976e-05, 3.2735e-04]],

         [[8.0273e-01, 3.0994e-05, 2.8014e-05,  ..., 9.7961e-03,
           7.0095e-04, 4.8184e-04]],

         ...,

         [[3.6597e-01, 2.9802e-06, 3.3379e-06,  ..., 2.9114e-02,
           1.5106e-02, 6.1035e-03]],

         [[3.3569e-01, 9.5367e-07, 3.5763e-07,  ..., 1.0767e-03,
           1.2207e-03, 2.1935e-03]],

         [[1.2207e-01, 7.6890e-06, 3.0220e-05,  ..., 5.0392e-03,
           3.4973e-02, 7.5928e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7627e-02, 1.5497e-06, 4.9472e-06,  ..., 2.6538e-01,
           1.6345e-01, 6.3049e-02]],

         [[3.2031e-01, 1.0073e-05, 6.9737e-06,  ..., 3.6346e-02,
           3.2990e-02, 2.2400e-02]],

         [[1.5039e-01, 1.8501e-04, 1.7929e-04,  ..., 9.8038e-03,
           9.7275e-03, 2.6455e-03]],

         ...,

         [[2.5635e-01, 8.1062e-06, 4.0352e-05,  ..., 1.0767e-01,
           1.6769e-02, 6.0959e-03]],

         [[5.2441e-01, 9.1791e-06, 4.8280e-05,  ..., 7.4402e-02,
           1.3130e-02, 2.3041e-02]],

         [[5.1270e-01, 4.4703e-06, 1.4782e-05,  ..., 8.5022e-02,
           1.0315e-02, 2.5528e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8654e-03, 8.3447e-07, 5.3644e-07,  ..., 2.3136e-03,
           3.5065e-02, 3.6392e-03]],

         [[2.6758e-01, 2.5749e-05, 1.0669e-05,  ..., 3.7727e-03,
           2.5463e-04, 4.1676e-04]],

         [[2.1655e-01, 5.2929e-05, 9.0361e-05,  ..., 2.3361e-02,
           2.8896e-03, 2.9125e-03]],

         ...,

         [[6.1554e-02, 8.0466e-06, 1.2648e-04,  ..., 1.0803e-01,
           1.2901e-02, 2.5269e-02]],

         [[2.7686e-01, 2.0742e-05, 1.4615e-04,  ..., 1.0028e-01,
           1.0910e-02, 1.9135e-02]],

         [[8.8574e-01, 2.4021e-05, 3.4153e-05,  ..., 2.5131e-02,
           1.8387e-02, 1.7273e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8613e-01, 1.0014e-05, 1.4603e-05,  ..., 1.5266e-02,
           4.3854e-02, 7.0862e-02]],

         [[3.4619e-01, 8.8215e-06, 5.0843e-05,  ..., 1.4539e-01,
           1.3037e-01, 6.1340e-02]],

         [[1.1584e-01, 8.2254e-06, 7.3910e-06,  ..., 2.9590e-01,
           5.1318e-01, 5.5603e-02]],

         ...,

         [[4.4824e-01, 4.0650e-05, 7.7248e-05,  ..., 9.1248e-03,
           1.1223e-02, 4.5891e-03]],

         [[7.7271e-02, 1.9670e-06, 2.6226e-06,  ..., 5.7404e-02,
           2.1497e-01, 1.6449e-02]],

         [[4.7089e-02, 9.4295e-05, 7.1764e-04,  ..., 6.7558e-03,
           1.1810e-02, 5.7373e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2983e-01, 3.0339e-05, 1.2636e-04,  ..., 3.0685e-02,
           6.7139e-03, 5.6534e-03]],

         [[5.3760e-01, 1.0180e-04, 2.9778e-04,  ..., 3.9940e-03,
           1.6012e-03, 1.4277e-03]],

         [[9.3323e-02, 7.0512e-05, 1.4448e-04,  ..., 3.1300e-03,
           7.9489e-04, 1.7090e-03]],

         ...,

         [[5.6689e-01, 2.9683e-04, 3.2353e-04,  ..., 9.4528e-03,
           1.7380e-02, 6.0669e-02]],

         [[7.4463e-02, 3.2604e-05, 1.9038e-04,  ..., 9.6130e-03,
           3.1036e-02, 8.1299e-01]],

         [[8.6035e-01, 2.2411e-05, 4.7445e-05,  ..., 1.0826e-02,
           3.4599e-03, 4.0222e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2139e-01, 8.5473e-05, 5.6171e-04,  ..., 7.2021e-03,
           5.2986e-03, 4.6654e-03]],

         [[5.7831e-02, 6.5565e-06, 3.9279e-05,  ..., 1.8997e-02,
           6.0120e-02, 2.0538e-02]],

         [[1.2805e-01, 1.2791e-04, 3.2163e-04,  ..., 6.0974e-02,
           6.8054e-02, 5.8655e-02]],

         ...,

         [[6.8945e-01, 4.5121e-05, 3.5238e-04,  ..., 2.1942e-02,
           4.7455e-03, 2.4719e-02]],

         [[8.4277e-01, 1.1921e-06, 2.1207e-04,  ..., 3.3245e-03,
           8.3351e-04, 9.6941e-04]],

         [[3.2275e-01, 2.6941e-05, 3.7074e-05,  ..., 2.2545e-03,
           6.7520e-03, 1.7929e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4254e-05, 4.7445e-05, 9.9609e-02,  ..., 3.0746e-02,
           1.1444e-02, 2.3758e-02]],

         [[1.1683e-05, 1.2219e-05, 6.4209e-02,  ..., 5.0507e-03,
           4.8370e-03, 5.7907e-03]],

         [[1.8394e-04, 2.3675e-04, 8.7097e-02,  ..., 1.2604e-02,
           2.8400e-03, 9.1858e-03]],

         ...,

         [[7.2300e-05, 5.2691e-05, 2.0813e-01,  ..., 5.8228e-02,
           7.1350e-02, 1.4307e-01]],

         [[1.8036e-04, 2.1255e-04, 2.6099e-01,  ..., 1.2848e-02,
           3.4580e-03, 7.5226e-03]],

         [[1.1384e-05, 1.2815e-05, 1.6504e-01,  ..., 2.2984e-03,
           3.4308e-04, 1.6174e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1301e-04, 1.1301e-04, 2.9395e-01,  ..., 7.6485e-03,
           1.2293e-03, 3.1033e-03]],

         [[6.2168e-05, 7.6771e-05, 1.0962e-01,  ..., 5.4131e-03,
           1.1683e-04, 2.1877e-03]],

         [[2.7597e-05, 1.1444e-05, 2.7710e-01,  ..., 1.3062e-02,
           1.6510e-04, 3.9673e-03]],

         ...,

         [[1.6975e-04, 6.9559e-05, 2.2290e-01,  ..., 6.7825e-03,
           2.6655e-04, 1.7624e-03]],

         [[4.8280e-06, 1.3530e-05, 2.1387e-01,  ..., 4.7989e-03,
           6.3062e-05, 1.7042e-03]],

         [[3.3140e-05, 5.0843e-05, 2.2449e-01,  ..., 8.0017e-02,
           6.8426e-04, 1.8570e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1325e-05, 1.5020e-05, 1.1218e-01,  ..., 4.7417e-03,
           7.6234e-05, 8.4305e-04]],

         [[9.2328e-05, 6.1393e-06, 2.3730e-01,  ..., 8.9493e-03,
           1.0985e-04, 2.2907e-03]],

         [[4.4227e-05, 6.2764e-05, 2.4084e-01,  ..., 7.7438e-03,
           2.0039e-04, 2.4948e-03]],

         ...,

         [[5.4121e-05, 2.7478e-05, 1.1621e-01,  ..., 1.1780e-02,
           1.4801e-03, 7.0343e-03]],

         [[1.5855e-05, 4.0948e-05, 1.8579e-01,  ..., 6.6910e-03,
           2.8443e-04, 2.9736e-03]],

         [[5.4181e-05, 3.5942e-05, 1.2561e-01,  ..., 3.7498e-03,
           1.9372e-04, 2.5978e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3241e-05, 6.6757e-06, 1.9714e-02,  ..., 1.3466e-03,
           2.4011e-01, 8.6975e-02]],

         [[4.6074e-05, 5.3227e-05, 1.2512e-01,  ..., 1.0406e-02,
           6.3416e-02, 3.1189e-02]],

         [[1.9932e-04, 4.9400e-04, 3.0664e-01,  ..., 5.8517e-03,
           8.9025e-04, 7.5102e-04]],

         ...,

         [[1.7178e-04, 6.0201e-06, 3.9032e-02,  ..., 1.7452e-04,
           3.0098e-03, 5.3635e-03]],

         [[2.2202e-03, 2.2960e-04, 1.1884e-01,  ..., 2.3178e-02,
           7.6721e-02, 8.2947e-02]],

         [[6.3181e-06, 2.3842e-06, 1.9255e-03,  ..., 5.9652e-04,
           2.0190e-01, 1.0553e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1625e-03, 6.1893e-04, 1.7712e-01,  ..., 3.0304e-02,
           2.0416e-02, 6.0486e-02]],

         [[3.6645e-04, 3.4571e-04, 2.2559e-01,  ..., 1.4366e-02,
           1.7883e-02, 5.2124e-02]],

         [[1.6642e-03, 4.9400e-04, 2.6978e-01,  ..., 2.5497e-02,
           7.8354e-03, 7.9041e-03]],

         ...,

         [[4.5037e-04, 3.9995e-05, 2.7710e-01,  ..., 2.8634e-04,
           3.7422e-03, 2.8091e-02]],

         [[1.5802e-03, 1.7767e-03, 1.8408e-01,  ..., 9.4528e-03,
           1.4648e-02, 4.8767e-02]],

         [[1.1355e-04, 5.9366e-05, 4.2023e-02,  ..., 6.9695e-03,
           1.5381e-01, 2.8271e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[3.9458e-04, 1.1063e-03, 1.4801e-02,  ..., 8.7891e-02,
           3.5864e-01, 8.7891e-02]],

         [[1.2577e-04, 2.3496e-04, 1.6317e-03,  ..., 1.2833e-02,
           1.3794e-01, 1.2964e-01]],

         [[0.0000e+00, 5.9605e-08, 1.3709e-06,  ..., 2.9206e-06,
           5.5218e-04, 9.9854e-01]],

         ...,

         [[5.9605e-08, 1.1921e-07, 0.0000e+00,  ..., 1.2279e-04,
           9.0742e-04, 9.9512e-01]],

         [[4.0436e-03, 5.8823e-03, 3.7994e-03,  ..., 2.9877e-02,
           2.4765e-02, 5.8823e-03]],

         [[4.6272e-03, 1.1702e-03, 2.0981e-04,  ..., 2.1393e-02,
           1.0757e-02, 8.2031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5373e-05, 5.0926e-04, 1.6699e-03,  ..., 1.1955e-02,
           8.8379e-02, 5.4102e-01]],

         [[5.9605e-08, 1.2755e-05, 1.9014e-04,  ..., 9.8511e-02,
           7.9932e-01, 8.6914e-02]],

         [[4.4584e-04, 2.4872e-03, 6.5498e-03,  ..., 4.6906e-02,
           2.9358e-02, 4.6906e-02]],

         ...,

         [[1.3423e-04, 7.9679e-04, 6.0606e-04,  ..., 2.0923e-01,
           2.0923e-01, 4.0308e-01]],

         [[1.3075e-03, 6.0921e-03, 2.7206e-02,  ..., 3.1311e-02,
           2.1103e-02, 1.9821e-02]],

         [[7.3385e-04, 2.1400e-03, 1.1911e-03,  ..., 1.0229e-01,
           5.7959e-01, 1.1060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0356, 0.0031, 0.0159,  ..., 0.0487, 0.0373, 0.0433]],

         [[0.0009, 0.0004, 0.0019,  ..., 0.0032, 0.0038, 0.0158]],

         [[0.0034, 0.0007, 0.0008,  ..., 0.0130, 0.0167, 0.0170]],

         ...,

         [[0.0029, 0.0036, 0.0070,  ..., 0.2454, 0.2454, 0.2306]],

         [[0.0028, 0.0010, 0.0012,  ..., 0.0289, 0.0291, 0.1469]],

         [[0.0192, 0.0083, 0.0321,  ..., 0.0375, 0.0350, 0.2372]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[8.9551e-01, 1.4305e-06, 1.1921e-06,  ..., 7.1383e-04,
           1.3599e-03, 1.2428e-02]],

         [[9.3604e-01, 2.1577e-05, 1.6689e-06,  ..., 1.3016e-02,
           2.5635e-02, 1.0498e-02]],

         [[8.3887e-01, 7.9870e-06, 5.3048e-06,  ..., 2.9736e-03,
           4.6844e-03, 4.4830e-02]],

         ...,

         [[1.8213e-01, 5.1260e-05, 3.9625e-04,  ..., 4.1127e-06,
           3.3975e-06, 2.6550e-02]],

         [[5.0586e-01, 2.1231e-04, 4.2605e-04,  ..., 1.7166e-02,
           5.7411e-03, 2.3285e-02]],

         [[4.6021e-01, 3.2091e-04, 2.2554e-04,  ..., 2.6276e-02,
           7.2266e-02, 1.6830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8857e-01, 1.4186e-05, 5.3048e-06,  ..., 7.4506e-05,
           7.2181e-05, 3.8666e-02]],

         [[9.5020e-01, 1.2934e-04, 7.2241e-05,  ..., 4.7302e-03,
           3.1113e-02, 1.2772e-02]],

         [[9.4824e-01, 2.0456e-04, 1.2898e-04,  ..., 3.3531e-03,
           1.7105e-02, 6.8550e-03]],

         ...,

         [[3.8623e-01, 2.8539e-04, 1.2589e-04,  ..., 8.1116e-02,
           1.0559e-01, 3.2196e-02]],

         [[2.5439e-01, 7.5674e-04, 3.5095e-04,  ..., 8.9417e-03,
           4.6692e-03, 4.7646e-03]],

         [[7.8857e-01, 4.7755e-04, 2.0647e-04,  ..., 2.0798e-02,
           2.3422e-02, 5.7587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9592e-01, 2.9516e-04, 7.0393e-05,  ..., 6.8848e-02,
           3.4302e-02, 4.2084e-02]],

         [[7.4707e-01, 4.0054e-05, 1.1206e-05,  ..., 4.6539e-02,
           3.9673e-02, 4.3488e-02]],

         [[6.0596e-01, 1.2636e-05, 3.8624e-05,  ..., 2.5803e-02,
           8.8867e-02, 4.2572e-02]],

         ...,

         [[8.3936e-01, 6.9141e-06, 2.9802e-07,  ..., 1.2517e-06,
           4.4107e-05, 1.5222e-01]],

         [[9.2969e-01, 6.6400e-05, 1.9073e-06,  ..., 0.0000e+00,
           6.5565e-07, 1.9791e-02]],

         [[9.8828e-01, 1.7941e-05, 5.9605e-08,  ..., 0.0000e+00,
           5.9605e-08, 4.9210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 3.1292e-05, 4.7684e-06,  ..., 2.3816e-01,
           1.4685e-01, 4.4739e-02]],

         [[6.6455e-01, 7.8082e-06, 1.8477e-06,  ..., 1.1516e-04,
           2.1875e-05, 7.4402e-02]],

         [[7.2070e-01, 5.7161e-05, 9.8944e-06,  ..., 1.9684e-03,
           3.2776e-02, 1.1642e-02]],

         ...,

         [[6.6797e-01, 5.3835e-04, 2.0742e-05,  ..., 3.6755e-03,
           2.5177e-03, 2.3766e-03]],

         [[5.9717e-01, 2.3687e-04, 4.3631e-05,  ..., 9.3613e-03,
           2.0233e-02, 1.8066e-01]],

         [[8.4570e-01, 1.4663e-04, 1.7345e-04,  ..., 1.0605e-03,
           5.5542e-03, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8271e-01, 1.0550e-05, 1.7107e-05,  ..., 6.7291e-03,
           4.5738e-03, 2.6505e-02]],

         [[6.6260e-01, 4.3511e-05, 3.1352e-05,  ..., 1.0750e-02,
           6.8207e-03, 3.0334e-02]],

         [[9.5508e-01, 1.3590e-05, 5.5432e-06,  ..., 5.6686e-03,
           4.6997e-03, 9.1629e-03]],

         ...,

         [[5.7275e-01, 1.5616e-04, 5.7697e-05,  ..., 4.0192e-02,
           1.8661e-02, 4.1046e-02]],

         [[8.7354e-01, 1.0759e-04, 2.5153e-05,  ..., 5.1918e-03,
           4.2953e-03, 1.3733e-02]],

         [[8.5547e-01, 1.6153e-05, 1.5140e-05,  ..., 1.9745e-02,
           4.1443e-02, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 5.0068e-04, 3.3641e-04,  ..., 8.7929e-04,
           6.6376e-04, 2.7618e-03]],

         [[9.3408e-01, 1.6868e-05, 6.1333e-05,  ..., 8.4152e-03,
           2.7962e-03, 3.4389e-03]],

         [[9.0234e-01, 1.0788e-05, 7.9870e-06,  ..., 8.4829e-04,
           6.4564e-04, 1.9045e-03]],

         ...,

         [[8.5059e-01, 1.6451e-05, 2.4199e-05,  ..., 2.4796e-02,
           1.0529e-02, 1.8463e-02]],

         [[8.1104e-01, 3.2485e-05, 2.3246e-05,  ..., 3.7018e-02,
           2.3422e-02, 9.0820e-02]],

         [[9.3652e-01, 2.0742e-05, 1.7405e-05,  ..., 3.1853e-03,
           1.5516e-03, 1.6037e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3816e-01, 1.7226e-04, 1.0471e-03,  ..., 2.1866e-02,
           1.1009e-02, 1.3794e-02]],

         [[8.6377e-01, 1.7166e-05, 1.6510e-05,  ..., 1.9547e-02,
           3.2104e-02, 2.0889e-02]],

         [[6.3428e-01, 3.2187e-05, 7.8380e-05,  ..., 2.8458e-02,
           1.6602e-02, 2.2247e-02]],

         ...,

         [[7.4951e-01, 3.0756e-05, 1.3173e-04,  ..., 2.7237e-03,
           6.0234e-03, 1.7560e-04]],

         [[7.2510e-01, 4.8637e-05, 3.7909e-05,  ..., 1.1932e-02,
           4.5776e-03, 2.8442e-02]],

         [[8.7109e-01, 3.6001e-05, 4.0352e-05,  ..., 5.5771e-03,
           4.9225e-02, 1.8478e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9375e-01, 4.5002e-05, 3.9697e-05,  ..., 4.1127e-06,
           7.1526e-07, 2.5208e-02]],

         [[8.1934e-01, 1.4126e-05, 3.6955e-06,  ..., 3.3905e-02,
           2.2247e-02, 5.3680e-02]],

         [[7.3096e-01, 1.6689e-06, 2.0862e-06,  ..., 1.3981e-03,
           7.9453e-05, 2.9251e-02]],

         ...,

         [[9.2627e-01, 5.2452e-05, 4.2617e-05,  ..., 1.5993e-03,
           4.5586e-03, 8.0414e-03]],

         [[7.9395e-01, 6.7890e-05, 3.8981e-05,  ..., 2.6154e-02,
           9.0942e-03, 9.8801e-04]],

         [[4.5386e-01, 4.3964e-04, 8.6904e-05,  ..., 4.8294e-03,
           1.5554e-03, 1.3397e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0640e-01, 2.6822e-06, 4.8280e-06,  ..., 2.3117e-03,
           2.1667e-03, 6.7725e-01]],

         [[8.0127e-01, 4.8542e-04, 6.4468e-04,  ..., 5.2452e-04,
           1.3673e-04, 1.7920e-03]],

         [[6.6650e-01, 6.3848e-04, 8.1205e-04,  ..., 4.6997e-03,
           5.7106e-03, 2.1942e-02]],

         ...,

         [[5.8936e-01, 1.5485e-04, 1.5485e-04,  ..., 5.1193e-03,
           6.5947e-04, 7.5626e-04]],

         [[7.7930e-01, 1.1116e-04, 9.0003e-05,  ..., 1.1604e-02,
           1.8873e-03, 1.4816e-02]],

         [[5.3125e-01, 5.5611e-05, 1.6403e-04,  ..., 1.3687e-02,
           1.8433e-02, 3.4473e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3854e-02, 2.2054e-06, 2.0862e-06,  ..., 9.4128e-04,
           1.0910e-03, 9.4336e-01]],

         [[7.2607e-01, 8.1778e-05, 1.1313e-04,  ..., 2.5879e-02,
           5.2734e-02, 7.0801e-03]],

         [[4.5679e-01, 1.4389e-04, 3.4213e-05,  ..., 6.5269e-03,
           5.3072e-04, 2.6352e-02]],

         ...,

         [[4.0359e-03, 3.5763e-07, 1.3709e-06,  ..., 2.9111e-04,
           5.6696e-04, 9.8975e-01]],

         [[4.4019e-01, 2.9027e-05, 4.1544e-05,  ..., 1.1131e-02,
           5.6877e-03, 3.1763e-01]],

         [[8.8574e-01, 7.3254e-05, 1.1206e-05,  ..., 7.1144e-04,
           4.7588e-04, 3.4542e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9336e-01, 3.0279e-04, 3.7241e-04,  ..., 9.8343e-03,
           3.2864e-03, 2.0325e-02]],

         [[6.9482e-01, 4.3809e-05, 1.0717e-04,  ..., 6.6040e-02,
           1.6589e-01, 1.1726e-02]],

         [[5.7520e-01, 1.8239e-05, 8.5592e-05,  ..., 4.1580e-03,
           2.2995e-04, 1.5266e-02]],

         ...,

         [[4.8584e-01, 3.7014e-05, 2.0683e-05,  ..., 7.6843e-02,
           4.1138e-01, 1.9226e-02]],

         [[4.7192e-01, 4.9734e-04, 7.6723e-04,  ..., 1.0597e-02,
           3.0518e-03, 2.7776e-04]],

         [[8.6230e-01, 1.5557e-05, 1.1086e-05,  ..., 2.7725e-02,
           6.5186e-02, 3.9825e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0703e-01, 1.4603e-05, 1.3411e-05,  ..., 2.4948e-02,
           4.0359e-03, 4.5441e-02]],

         [[1.0175e-01, 6.0272e-04, 3.2902e-04,  ..., 1.2108e-02,
           1.4191e-02, 5.5027e-04]],

         [[3.5620e-01, 1.1187e-03, 9.0122e-04,  ..., 2.6840e-02,
           5.1025e-02, 5.8022e-03]],

         ...,

         [[4.8682e-01, 1.0461e-04, 5.6207e-05,  ..., 1.4539e-01,
           1.6541e-01, 8.1863e-03]],

         [[7.5244e-01, 7.5698e-05, 6.0141e-05,  ..., 6.8542e-02,
           7.2937e-02, 6.2622e-02]],

         [[5.6689e-01, 2.4796e-04, 9.8646e-05,  ..., 6.3477e-02,
           4.8004e-02, 8.1940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7811e-02, 1.6093e-06, 1.6093e-06,  ..., 4.3392e-04,
           1.1129e-03, 9.5654e-01]],

         [[1.8103e-01, 8.0466e-06, 7.3314e-06,  ..., 2.1582e-01,
           2.8052e-01, 5.9814e-02]],

         [[1.7297e-01, 4.2176e-04, 4.7541e-04,  ..., 1.9913e-02,
           4.4670e-03, 9.0103e-03]],

         ...,

         [[3.7573e-01, 1.9407e-04, 5.6803e-05,  ..., 2.5452e-02,
           6.2714e-03, 3.4637e-03]],

         [[6.3721e-01, 1.1742e-05, 4.5896e-06,  ..., 1.2428e-02,
           1.6479e-03, 1.1032e-02]],

         [[7.2754e-01, 4.4942e-05, 6.8545e-05,  ..., 3.0251e-03,
           2.1541e-04, 2.5272e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8145e-01, 5.2571e-05, 5.4300e-05,  ..., 3.8147e-02,
           1.5602e-02, 4.8035e-02]],

         [[2.2327e-01, 6.9618e-05, 3.1114e-05,  ..., 1.6754e-02,
           3.1605e-03, 6.7017e-02]],

         [[5.3223e-01, 9.9778e-05, 9.6440e-05,  ..., 1.6983e-02,
           7.9269e-03, 1.9318e-02]],

         ...,

         [[6.8652e-01, 2.3782e-05, 1.4842e-05,  ..., 2.5299e-02,
           1.6144e-02, 1.7807e-02]],

         [[7.4512e-01, 5.7936e-05, 5.9426e-05,  ..., 3.5553e-02,
           7.2632e-02, 1.7670e-02]],

         [[3.9233e-01, 4.9472e-06, 3.8147e-06,  ..., 1.2810e-02,
           6.6719e-03, 1.3016e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 1.0319e-03, 7.7677e-04,  ..., 1.9569e-03,
           8.2970e-04, 3.4294e-03]],

         [[1.7273e-01, 4.6194e-05, 4.1604e-05,  ..., 5.5962e-03,
           9.6560e-04, 2.4078e-02]],

         [[3.0591e-01, 1.3173e-05, 1.2696e-05,  ..., 1.6309e-01,
           2.1619e-01, 4.8218e-02]],

         ...,

         [[5.0830e-01, 1.5891e-04, 4.5598e-05,  ..., 1.8110e-03,
           1.4458e-03, 4.6539e-02]],

         [[5.3418e-01, 2.5094e-05, 3.1352e-05,  ..., 2.0266e-06,
           8.2254e-06, 2.3880e-03]],

         [[6.0010e-01, 8.4758e-05, 1.2398e-04,  ..., 5.3368e-03,
           5.0621e-03, 5.3644e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2781e-01, 4.8578e-05, 2.8551e-05,  ..., 1.9913e-02,
           4.4891e-02, 6.8817e-03]],

         [[1.6992e-01, 2.8825e-04, 1.1247e-04,  ..., 1.3832e-02,
           2.1801e-03, 1.5930e-02]],

         [[2.9199e-01, 1.6630e-05, 7.7486e-06,  ..., 1.1798e-01,
           9.5520e-02, 1.7432e-01]],

         ...,

         [[3.7012e-01, 6.5565e-06, 3.8743e-06,  ..., 2.5787e-02,
           2.1210e-02, 2.0630e-02]],

         [[6.8896e-01, 1.1802e-05, 5.9009e-06,  ..., 4.1687e-02,
           8.8806e-03, 1.1157e-01]],

         [[2.7344e-01, 2.0623e-05, 1.6928e-05,  ..., 1.3123e-01,
           7.1899e-02, 3.9478e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 1.7643e-05, 6.6757e-06,  ..., 3.9291e-03,
           4.0512e-03, 4.0283e-02]],

         [[1.5918e-01, 6.6757e-06, 1.6510e-05,  ..., 2.3157e-01,
           8.5205e-02, 2.9434e-02]],

         [[1.4441e-01, 8.1658e-06, 1.1325e-06,  ..., 4.3602e-03,
           7.4158e-03, 8.0518e-01]],

         ...,

         [[1.7188e-01, 2.0444e-04, 1.4102e-04,  ..., 1.4102e-04,
           2.2805e-04, 2.0859e-02]],

         [[3.7476e-01, 1.5461e-04, 7.5102e-05,  ..., 2.6169e-03,
           2.2545e-03, 2.7802e-02]],

         [[2.0825e-01, 1.3900e-04, 8.1718e-05,  ..., 1.8921e-02,
           3.8147e-03, 6.0959e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2324e-01, 1.9431e-05, 1.1206e-05,  ..., 2.0676e-02,
           6.1569e-03, 6.6223e-02]],

         [[7.7930e-01, 2.1505e-04, 3.3617e-05,  ..., 4.3511e-06,
           1.0729e-05, 7.5459e-05]],

         [[8.7402e-01, 1.9145e-04, 5.6803e-05,  ..., 8.1718e-05,
           1.9598e-04, 8.3494e-04]],

         ...,

         [[1.4673e-01, 5.8603e-04, 1.1206e-03,  ..., 4.1656e-03,
           5.4169e-04, 5.1003e-03]],

         [[6.5332e-01, 1.0312e-05, 1.5974e-05,  ..., 3.5736e-02,
           2.1133e-03, 2.1011e-02]],

         [[3.6401e-01, 5.3453e-04, 1.1768e-03,  ..., 5.0697e-03,
           6.8245e-03, 1.3680e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8428e-01, 7.5161e-05, 8.7917e-05,  ..., 1.0239e-02,
           1.2062e-02, 1.1787e-02]],

         [[2.6270e-01, 2.2328e-04, 2.9802e-04,  ..., 3.1300e-03,
           9.4700e-04, 3.5004e-02]],

         [[3.5034e-01, 4.4584e-05, 6.3896e-05,  ..., 6.2037e-04,
           2.8849e-04, 5.2757e-03]],

         ...,

         [[5.4871e-02, 4.9293e-05, 7.6294e-05,  ..., 3.0956e-03,
           4.1656e-03, 5.4871e-02]],

         [[2.4475e-02, 2.3007e-05, 2.7776e-05,  ..., 4.1461e-04,
           3.5453e-04, 6.5491e-02]],

         [[4.9243e-01, 7.6830e-05, 8.5711e-05,  ..., 3.9406e-03,
           1.2016e-03, 1.0063e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0386e-01, 3.1531e-05, 2.8253e-05,  ..., 2.9297e-03,
           7.7629e-04, 4.2725e-02]],

         [[3.9746e-01, 3.1233e-04, 3.6526e-04,  ..., 4.7722e-03,
           2.6779e-03, 9.0561e-03]],

         [[1.4233e-01, 6.8963e-05, 1.6737e-04,  ..., 1.1963e-02,
           3.9520e-02, 5.9912e-01]],

         ...,

         [[5.4883e-01, 9.6262e-05, 1.8704e-04,  ..., 1.6968e-02,
           8.5297e-03, 7.3730e-02]],

         [[1.3635e-01, 7.6294e-06, 4.7565e-05,  ..., 3.8314e-04,
           2.3234e-04, 6.5430e-02]],

         [[3.6108e-01, 1.9205e-04, 3.4642e-04,  ..., 5.7364e-04,
           4.6086e-04, 9.3307e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6626e-01, 1.1969e-04, 6.9976e-05,  ..., 2.1301e-02,
           1.0468e-02, 1.2164e-01]],

         [[9.9426e-02, 6.3062e-05, 1.3196e-04,  ..., 1.7725e-01,
           1.6113e-02, 1.2352e-02]],

         [[5.2441e-01, 4.0591e-05, 1.1533e-04,  ..., 6.4964e-03,
           8.1177e-03, 1.1230e-02]],

         ...,

         [[6.5576e-01, 1.8775e-05, 2.8729e-05,  ..., 4.8248e-02,
           2.3010e-01, 5.1361e-02]],

         [[5.5225e-01, 1.1683e-05, 2.9325e-05,  ..., 3.5858e-02,
           2.9266e-02, 2.0432e-02]],

         [[2.9907e-01, 8.1658e-06, 1.2815e-05,  ..., 1.1896e-01,
           4.7778e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4250e-02, 1.8477e-06, 1.2398e-05,  ..., 1.0132e-01,
           3.2135e-02, 4.9377e-02]],

         [[4.2480e-02, 1.6689e-06, 5.8413e-06,  ..., 1.6336e-03,
           4.3373e-03, 9.2236e-01]],

         [[4.8523e-02, 2.1100e-05, 6.7234e-05,  ..., 1.2756e-02,
           1.1261e-02, 1.8555e-02]],

         ...,

         [[1.6711e-01, 5.3883e-05, 1.4007e-04,  ..., 3.5583e-02,
           2.3514e-02, 1.8234e-02]],

         [[5.6250e-01, 4.0531e-06, 8.1062e-06,  ..., 5.4626e-02,
           3.3783e-02, 8.3618e-02]],

         [[1.4429e-01, 1.4067e-05, 1.2994e-05,  ..., 1.0315e-01,
           6.2939e-01, 7.6965e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6016e-01, 1.0377e-04, 5.1689e-04,  ..., 3.7746e-03,
           4.5013e-03, 1.4473e-02]],

         [[9.6436e-01, 3.6597e-05, 7.0393e-05,  ..., 2.5570e-05,
           5.3942e-05, 6.5899e-04]],

         [[8.8867e-01, 2.6822e-05, 6.7949e-06,  ..., 6.1274e-05,
           2.2829e-05, 3.8719e-04]],

         ...,

         [[7.7832e-01, 6.9737e-06, 9.8944e-06,  ..., 1.3809e-02,
           9.0561e-03, 3.5828e-02]],

         [[6.0352e-01, 1.8477e-06, 5.9605e-07,  ..., 4.8566e-04,
           2.7251e-04, 6.1989e-03]],

         [[1.0931e-01, 6.4969e-06, 9.2983e-06,  ..., 2.8896e-03,
           1.0490e-02, 7.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3497e-02, 3.2783e-06, 1.0133e-05,  ..., 2.1313e-01,
           1.2830e-01, 9.6069e-02]],

         [[5.6494e-01, 1.5318e-05, 1.0371e-05,  ..., 1.9867e-02,
           1.9958e-02, 2.2217e-01]],

         [[1.2195e-01, 1.5354e-04, 1.3220e-04,  ..., 4.0863e-02,
           7.9803e-03, 7.1716e-02]],

         ...,

         [[5.0146e-01, 1.1981e-05, 5.6684e-05,  ..., 4.2816e-02,
           2.2202e-02, 6.7871e-02]],

         [[5.6787e-01, 1.6868e-05, 3.5167e-05,  ..., 2.8503e-02,
           1.3046e-02, 2.4185e-02]],

         [[6.3135e-01, 6.2585e-06, 1.6034e-05,  ..., 3.7823e-03,
           5.2528e-03, 4.8676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9440e-02, 3.1590e-06, 1.8477e-06,  ..., 4.1168e-02,
           5.8403e-03, 2.0218e-02]],

         [[1.5381e-01, 5.9187e-05, 2.5690e-05,  ..., 1.0166e-03,
           7.2384e-04, 7.3090e-03]],

         [[2.8198e-01, 6.8009e-05, 1.6022e-04,  ..., 6.5804e-03,
           3.4809e-03, 5.4657e-02]],

         ...,

         [[5.6305e-02, 5.4240e-06, 7.0691e-05,  ..., 1.1894e-02,
           3.1586e-02, 3.4668e-02]],

         [[1.1975e-01, 7.9274e-06, 5.5790e-05,  ..., 9.8267e-03,
           1.7807e-02, 9.8267e-03]],

         [[7.7246e-01, 2.5272e-05, 4.8339e-05,  ..., 5.1758e-02,
           8.5327e-02, 4.0314e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0889e-01, 2.1458e-06, 7.2718e-06,  ..., 2.8366e-02,
           8.4045e-02, 2.1619e-01]],

         [[3.7158e-01, 2.7835e-05, 8.5831e-05,  ..., 9.3933e-02,
           1.2396e-01, 1.5369e-01]],

         [[3.4253e-01, 6.6757e-06, 4.8757e-05,  ..., 6.7139e-02,
           4.8291e-01, 7.9468e-02]],

         ...,

         [[4.2358e-01, 3.2961e-05, 3.1471e-05,  ..., 1.3191e-02,
           3.6507e-03, 1.1284e-02]],

         [[1.8604e-01, 5.6028e-06, 1.0252e-05,  ..., 1.8176e-01,
           9.2834e-02, 4.7791e-02]],

         [[6.5979e-02, 2.7359e-05, 8.1599e-05,  ..., 3.3112e-03,
           1.3298e-02, 6.2109e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9121e-01, 1.6868e-05, 1.0830e-04,  ..., 8.0338e-03,
           4.7798e-03, 1.3855e-01]],

         [[2.0337e-01, 2.6047e-05, 8.2672e-05,  ..., 3.4857e-04,
           3.1114e-04, 4.0741e-03]],

         [[2.5195e-01, 6.4254e-05, 2.7275e-04,  ..., 7.6199e-04,
           1.1568e-03, 5.1178e-02]],

         ...,

         [[4.7217e-01, 3.4857e-04, 5.3596e-04,  ..., 1.9180e-02,
           4.1901e-02, 2.8198e-01]],

         [[5.0879e-01, 1.5306e-04, 5.9605e-04,  ..., 3.1052e-02,
           7.3303e-02, 2.4426e-01]],

         [[5.7861e-01, 3.2723e-05, 4.5061e-05,  ..., 2.9888e-03,
           1.5671e-02, 3.6206e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5488e-01, 3.7730e-05, 2.5058e-04,  ..., 5.9357e-02,
           5.4230e-02, 3.3539e-02]],

         [[1.4087e-01, 4.1723e-06, 5.5909e-05,  ..., 2.7618e-02,
           8.4106e-02, 1.0968e-01]],

         [[3.3984e-01, 1.3804e-04, 6.3848e-04,  ..., 3.6682e-02,
           5.2948e-02, 9.2590e-02]],

         ...,

         [[5.3467e-01, 4.1604e-05, 1.5628e-04,  ..., 2.5665e-02,
           6.1676e-02, 9.7733e-03]],

         [[9.0967e-01, 1.0133e-06, 4.6182e-04,  ..., 2.7275e-04,
           1.6356e-04, 3.0804e-03]],

         [[6.4258e-01, 3.2783e-06, 9.7752e-06,  ..., 2.1561e-02,
           1.7120e-02, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7193e-05, 1.7762e-05, 1.7249e-01,  ..., 5.4779e-03,
           1.3405e-02, 2.2186e-02]],

         [[2.3782e-05, 1.6510e-05, 1.4539e-01,  ..., 1.0162e-02,
           1.3466e-02, 1.2070e-02]],

         [[4.0174e-05, 4.9591e-05, 9.8633e-02,  ..., 6.0387e-03,
           1.6739e-02, 3.6560e-02]],

         ...,

         [[6.5506e-05, 7.8082e-05, 2.8174e-01,  ..., 9.2773e-03,
           9.2224e-02, 1.3843e-01]],

         [[2.2209e-04, 1.1826e-04, 2.7710e-01,  ..., 2.6016e-03,
           4.7112e-03, 2.9770e-02]],

         [[4.4465e-05, 1.6093e-05, 3.1299e-01,  ..., 5.9938e-04,
           1.1110e-03, 5.1003e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2493e-04, 6.2108e-05, 3.2227e-01,  ..., 4.8661e-04,
           7.5626e-04, 4.2343e-03]],

         [[5.3465e-05, 4.3213e-05, 2.0825e-01,  ..., 1.8704e-04,
           9.3126e-04, 2.4529e-03]],

         [[1.2398e-05, 4.2319e-06, 2.7002e-01,  ..., 1.0931e-04,
           8.9741e-04, 5.0163e-04]],

         ...,

         [[1.1438e-04, 1.0496e-04, 2.7075e-01,  ..., 5.3835e-04,
           1.8101e-03, 2.9259e-03]],

         [[1.1921e-06, 1.7881e-06, 2.3254e-01,  ..., 1.4353e-04,
           1.0118e-03, 4.8256e-03]],

         [[1.0276e-04, 1.4544e-04, 1.8872e-01,  ..., 1.5163e-03,
           1.2642e-02, 2.4891e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0014e-05, 6.8545e-06, 1.7029e-01,  ..., 9.8288e-05,
           4.3678e-04, 2.1896e-02]],

         [[3.0220e-05, 1.6093e-06, 2.4927e-01,  ..., 1.2803e-04,
           9.4223e-04, 3.7909e-04]],

         [[1.7643e-05, 1.9372e-05, 2.4829e-01,  ..., 1.5986e-04,
           9.4223e-04, 8.2827e-04]],

         ...,

         [[1.5497e-05, 1.6212e-05, 1.2939e-01,  ..., 1.9646e-03,
           7.1869e-03, 3.1787e-01]],

         [[1.0014e-05, 2.1398e-05, 2.2314e-01,  ..., 2.2221e-04,
           1.1349e-03, 2.7657e-03]],

         [[6.3062e-05, 2.6524e-05, 1.6211e-01,  ..., 2.8229e-04,
           1.1272e-03, 1.3306e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1842e-05, 8.7619e-06, 1.2787e-02,  ..., 1.4636e-01,
           3.8867e-01, 1.8652e-01]],

         [[2.0564e-05, 2.3007e-05, 8.1421e-02,  ..., 4.9194e-02,
           3.0914e-02, 1.5099e-02]],

         [[1.4102e-04, 3.8314e-04, 3.0127e-01,  ..., 2.7771e-03,
           1.0662e-03, 1.4420e-03]],

         ...,

         [[5.1308e-04, 2.2292e-05, 1.3660e-01,  ..., 2.4166e-03,
           3.0041e-03, 2.0294e-02]],

         [[7.6532e-04, 1.3793e-04, 4.5837e-02,  ..., 9.9731e-02,
           2.8979e-01, 2.7856e-01]],

         [[9.4950e-05, 2.0802e-05, 1.5839e-02,  ..., 3.2745e-02,
           3.9948e-02, 4.0674e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9083e-03, 1.2045e-03, 2.3621e-01,  ..., 2.5730e-03,
           5.7220e-03, 6.3599e-02]],

         [[1.5986e-04, 2.4462e-04, 2.3853e-01,  ..., 1.8829e-02,
           2.9388e-02, 4.3091e-02]],

         [[1.5335e-03, 4.1366e-04, 2.4353e-01,  ..., 1.5327e-02,
           5.7831e-02, 6.2042e-02]],

         ...,

         [[1.6689e-04, 4.6670e-05, 2.3132e-01,  ..., 1.5495e-02,
           4.2450e-02, 5.5359e-02]],

         [[8.1921e-04, 7.5150e-04, 1.2549e-01,  ..., 9.7504e-03,
           3.8269e-02, 3.2275e-01]],

         [[1.2934e-05, 2.9266e-05, 8.3252e-02,  ..., 7.1777e-02,
           1.9055e-01, 1.4722e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.3828e-05, 1.9670e-04, 1.9255e-03,  ..., 3.0121e-02,
           6.8555e-01, 1.0510e-01]],

         [[4.6074e-05, 3.1650e-05, 1.8227e-04,  ..., 3.4389e-03,
           1.9791e-02, 9.5361e-01]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.6689e-05,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[5.9605e-06, 3.6359e-06, 4.7684e-07,  ..., 5.3835e-04,
           6.5613e-03, 9.7363e-01]],

         [[3.9635e-03, 2.2583e-03, 3.2845e-03,  ..., 5.8228e-02,
           1.1467e-02, 2.2812e-02]],

         [[3.3455e-03, 2.0294e-03, 2.6631e-04,  ..., 4.3396e-02,
           1.1792e-01, 6.5125e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8036e-05, 8.3542e-04, 2.0676e-03,  ..., 8.2581e-02,
           2.9736e-01, 2.0447e-01]],

         [[0.0000e+00, 1.7881e-07, 8.9407e-06,  ..., 1.2164e-01,
           7.8125e-01, 7.7332e-02]],

         [[1.2903e-03, 4.2305e-03, 1.1871e-02,  ..., 2.1500e-02,
           3.8910e-02, 1.2634e-02]],

         ...,

         [[1.7996e-03, 5.0850e-03, 4.0245e-03,  ..., 4.2017e-01,
           1.1487e-01, 1.3397e-02]],

         [[2.5368e-03, 1.1505e-02, 1.9806e-02,  ..., 1.9485e-02,
           1.7334e-02, 7.7820e-03]],

         [[3.4313e-03, 7.2632e-03, 6.9847e-03,  ..., 2.1069e-01,
           7.2205e-02, 5.8960e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3732e-02, 4.5547e-03, 2.3773e-02,  ..., 1.3763e-02,
           1.9867e-02, 1.3924e-02]],

         [[1.2712e-03, 4.5323e-04, 2.2850e-03,  ..., 6.0654e-03,
           3.0090e-02, 1.1421e-02]],

         [[8.6823e-03, 1.5812e-03, 2.4776e-03,  ..., 1.7197e-02,
           2.0981e-02, 1.1772e-02]],

         ...,

         [[2.0638e-03, 1.6203e-03, 8.1024e-03,  ..., 3.7750e-02,
           1.1627e-01, 5.5469e-01]],

         [[1.6632e-03, 4.2367e-04, 1.0824e-03,  ..., 3.0884e-02,
           1.0535e-01, 4.1901e-02]],

         [[8.9121e-04, 4.2748e-04, 5.3329e-03,  ..., 5.3864e-02,
           5.7910e-01, 1.5466e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4863e-01, 2.3842e-06, 2.5630e-06,  ..., 1.5545e-03,
           8.8043e-03, 2.3682e-02]],

         [[9.5215e-01, 1.0788e-05, 3.0994e-06,  ..., 4.7760e-03,
           1.7395e-02, 1.5732e-02]],

         [[9.1357e-01, 4.7684e-06, 1.6689e-06,  ..., 2.2736e-03,
           1.9012e-02, 1.8524e-02]],

         ...,

         [[6.4355e-01, 1.1837e-04, 9.6560e-04,  ..., 1.1384e-05,
           9.4299e-03, 2.2278e-02]],

         [[5.5127e-01, 1.1843e-04, 3.6526e-04,  ..., 1.6510e-02,
           5.0262e-02, 1.3741e-02]],

         [[4.8267e-01, 7.4983e-05, 2.4331e-04,  ..., 2.7130e-02,
           8.0322e-02, 9.8267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1797e-01, 2.1994e-05, 1.5497e-05,  ..., 1.8358e-04,
           2.5043e-03, 1.6205e-02]],

         [[3.0396e-01, 6.5565e-07, 5.3704e-05,  ..., 6.5327e-04,
           6.8604e-01, 7.7515e-03]],

         [[9.6045e-01, 2.4676e-05, 4.3452e-05,  ..., 3.0098e-03,
           7.5684e-03, 1.7319e-03]],

         ...,

         [[3.6060e-01, 2.2662e-04, 2.1482e-04,  ..., 1.3904e-01,
           5.1758e-02, 2.9480e-02]],

         [[2.4817e-01, 8.8120e-04, 4.2653e-04,  ..., 5.2299e-03,
           8.7128e-03, 6.1150e-03]],

         [[6.3965e-01, 6.6280e-04, 8.9741e-04,  ..., 8.4000e-03,
           2.4951e-01, 3.7903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8628e-01, 1.3542e-04, 4.2915e-05,  ..., 4.4861e-02,
           6.2622e-02, 3.9948e-02]],

         [[7.1240e-01, 4.6134e-05, 1.8358e-05,  ..., 2.1255e-02,
           1.9531e-01, 1.4816e-02]],

         [[7.9443e-01, 3.5703e-05, 2.3067e-05,  ..., 3.8635e-02,
           4.8462e-02, 1.1490e-02]],

         ...,

         [[9.5264e-01, 6.1572e-05, 6.5565e-07,  ..., 4.1723e-07,
           2.0111e-04, 3.8055e-02]],

         [[9.5312e-01, 2.1458e-04, 2.5630e-06,  ..., 0.0000e+00,
           1.1325e-06, 9.2773e-03]],

         [[9.8682e-01, 4.7028e-05, 1.1921e-07,  ..., 0.0000e+00,
           5.9605e-08, 7.8964e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3743e-01, 2.3246e-05, 4.5896e-06,  ..., 7.0068e-02,
           4.3994e-01, 2.9617e-02]],

         [[8.0664e-01, 2.3842e-06, 3.5763e-07,  ..., 1.2267e-04,
           2.3788e-02, 4.7211e-02]],

         [[8.4424e-01, 3.4511e-05, 6.7949e-06,  ..., 6.6109e-03,
           5.9509e-03, 8.2474e-03]],

         ...,

         [[7.8174e-01, 3.0780e-04, 1.7345e-05,  ..., 4.2152e-03,
           1.4999e-02, 1.2566e-02]],

         [[5.3906e-01, 3.3438e-05, 1.2159e-05,  ..., 1.3702e-02,
           2.2766e-02, 3.3081e-01]],

         [[3.0762e-01, 9.0420e-05, 6.5625e-05,  ..., 4.3335e-03,
           6.1670e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1152e-01, 3.3081e-05, 1.4484e-05,  ..., 3.0327e-03,
           1.4313e-02, 1.6113e-02]],

         [[7.8857e-01, 3.7253e-05, 1.8239e-05,  ..., 3.3379e-03,
           9.2392e-03, 6.3721e-02]],

         [[7.4805e-01, 4.0531e-06, 3.9935e-06,  ..., 1.2032e-02,
           1.5137e-01, 2.6031e-02]],

         ...,

         [[5.8447e-01, 9.3758e-05, 1.8954e-05,  ..., 8.0261e-03,
           5.8868e-02, 1.6998e-02]],

         [[8.5010e-01, 9.3162e-05, 3.1948e-05,  ..., 2.2583e-03,
           1.1856e-02, 1.3695e-02]],

         [[8.2666e-01, 1.8179e-05, 1.7762e-05,  ..., 2.6260e-02,
           7.6111e-02, 7.2021e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9971e-01, 8.9073e-04, 4.8423e-04,  ..., 5.0163e-04,
           2.7428e-03, 1.9226e-03]],

         [[9.3652e-01, 8.4639e-06, 4.9472e-05,  ..., 1.2178e-03,
           2.7657e-03, 1.1578e-03]],

         [[7.4805e-01, 3.9876e-05, 1.1504e-05,  ..., 3.4943e-03,
           5.5199e-03, 9.3002e-03]],

         ...,

         [[8.1738e-01, 1.4901e-05, 2.6822e-05,  ..., 2.9507e-03,
           1.1589e-02, 9.7656e-02]],

         [[3.1812e-01, 1.4067e-05, 7.8082e-06,  ..., 3.7903e-02,
           5.2637e-01, 4.8859e-02]],

         [[9.0625e-01, 5.0247e-05, 1.9193e-05,  ..., 9.9468e-04,
           8.1406e-03, 2.5986e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0642e-01, 4.6039e-04, 1.9045e-03,  ..., 8.6746e-03,
           1.5686e-02, 1.3145e-02]],

         [[7.0801e-01, 1.6868e-05, 4.3452e-05,  ..., 3.9612e-02,
           1.1304e-01, 3.0899e-02]],

         [[7.3242e-01, 1.5795e-05, 2.9743e-05,  ..., 9.7961e-03,
           1.9821e-02, 1.3573e-02]],

         ...,

         [[8.5400e-01, 3.5465e-05, 8.8394e-05,  ..., 1.4410e-03,
           1.6618e-04, 5.4061e-05]],

         [[6.3232e-01, 1.1241e-04, 7.0095e-05,  ..., 1.1589e-02,
           7.5500e-02, 2.3163e-02]],

         [[9.5703e-01, 8.4043e-06, 1.9133e-05,  ..., 2.6627e-03,
           6.2943e-03, 9.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1172e-01, 2.0599e-04, 1.7345e-04,  ..., 1.1921e-06,
           2.3575e-03, 1.1337e-02]],

         [[7.3779e-01, 4.1127e-06, 1.7285e-06,  ..., 5.2216e-02,
           7.1838e-02, 7.3624e-03]],

         [[7.3975e-01, 5.9009e-06, 5.7817e-06,  ..., 3.6621e-04,
           5.4047e-02, 9.6893e-03]],

         ...,

         [[9.1260e-01, 1.0157e-04, 7.3731e-05,  ..., 3.0155e-03,
           8.3313e-03, 2.5539e-03]],

         [[8.5400e-01, 6.9797e-05, 3.9160e-05,  ..., 9.0332e-03,
           1.0443e-03, 8.2254e-04]],

         [[5.6787e-01, 9.8801e-04, 3.6430e-04,  ..., 1.8034e-03,
           1.5518e-02, 6.5498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5295e-01, 4.7684e-07, 1.3113e-06,  ..., 1.7226e-04,
           3.3875e-02, 8.0469e-01]],

         [[7.9932e-01, 3.0565e-04, 5.6601e-04,  ..., 9.6262e-05,
           1.8215e-03, 1.3895e-03]],

         [[6.0693e-01, 5.9223e-04, 1.0653e-03,  ..., 4.8943e-03,
           1.7319e-02, 8.2092e-03]],

         ...,

         [[6.6260e-01, 2.3592e-04, 1.9407e-04,  ..., 1.6003e-03,
           2.3651e-03, 5.0163e-04]],

         [[8.0762e-01, 7.8619e-05, 6.5446e-05,  ..., 3.5172e-03,
           1.8799e-02, 5.6305e-03]],

         [[3.0518e-01, 2.3603e-05, 1.1224e-04,  ..., 3.2234e-03,
           1.0388e-01, 5.1709e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5098e-01, 9.8348e-06, 9.5963e-06,  ..., 1.5144e-03,
           3.4399e-01, 3.7207e-01]],

         [[7.5732e-01, 5.1320e-05, 6.3598e-05,  ..., 3.9032e-02,
           2.9861e-02, 1.3794e-02]],

         [[6.1182e-01, 2.4319e-04, 4.9233e-05,  ..., 7.1192e-04,
           4.0741e-02, 2.1347e-02]],

         ...,

         [[9.9106e-03, 7.1526e-07, 1.9073e-06,  ..., 2.5630e-05,
           3.8605e-02, 9.4629e-01]],

         [[5.2588e-01, 3.2365e-05, 4.5300e-05,  ..., 1.0786e-03,
           6.5002e-02, 1.6174e-01]],

         [[9.1748e-01, 5.3704e-05, 1.1027e-05,  ..., 5.4121e-04,
           3.5992e-03, 2.3651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-01, 3.1281e-04, 3.0804e-04,  ..., 4.0321e-03,
           1.9547e-02, 1.7914e-02]],

         [[6.8213e-01, 7.6354e-05, 8.1897e-05,  ..., 1.7847e-01,
           1.3954e-02, 4.8256e-03]],

         [[6.2158e-01, 4.2379e-05, 1.3161e-04,  ..., 3.3998e-04,
           6.0234e-03, 6.3629e-03]],

         ...,

         [[7.6416e-01, 1.0550e-05, 3.7968e-05,  ..., 1.1011e-01,
           1.0815e-01, 6.7558e-03]],

         [[5.8447e-01, 1.4627e-04, 2.3019e-04,  ..., 2.5826e-03,
           3.3498e-04, 2.5487e-04]],

         [[5.0537e-01, 8.4639e-06, 4.2975e-05,  ..., 3.4851e-02,
           3.9600e-01, 5.7220e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2520e-01, 1.1086e-05, 7.7486e-06,  ..., 1.5345e-03,
           2.7908e-02, 1.9943e-02]],

         [[6.3477e-02, 7.1907e-04, 3.4237e-04,  ..., 1.3947e-02,
           7.3433e-04, 2.6727e-04]],

         [[4.2432e-01, 6.1083e-04, 4.5133e-04,  ..., 5.7312e-02,
           1.0025e-02, 3.9043e-03]],

         ...,

         [[6.7285e-01, 1.0216e-04, 6.8009e-05,  ..., 1.1926e-01,
           2.1500e-02, 5.1765e-03]],

         [[4.2822e-01, 8.4162e-05, 1.2201e-04,  ..., 2.2980e-02,
           5.0879e-01, 1.2581e-02]],

         [[5.7178e-01, 1.6212e-04, 1.3185e-04,  ..., 6.1554e-02,
           1.7624e-02, 1.6891e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-02, 1.6689e-06, 2.3842e-06,  ..., 7.7963e-05,
           1.3718e-02, 9.3213e-01]],

         [[1.3586e-01, 1.4842e-05, 1.0133e-05,  ..., 2.7002e-01,
           2.6587e-01, 1.9714e-02]],

         [[2.3730e-01, 5.5838e-04, 5.8317e-04,  ..., 7.8583e-03,
           2.0538e-02, 9.4604e-03]],

         ...,

         [[3.8257e-01, 1.3769e-04, 5.2869e-05,  ..., 8.6365e-03,
           9.6512e-03, 2.7275e-03]],

         [[7.5000e-01, 3.0577e-05, 1.5080e-05,  ..., 3.0060e-03,
           1.8997e-02, 4.7302e-03]],

         [[7.4023e-01, 7.4804e-05, 6.7890e-05,  ..., 3.4475e-04,
           1.4061e-02, 4.4098e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9341e-01, 2.3961e-05, 2.4021e-05,  ..., 9.5825e-03,
           1.9702e-01, 8.4412e-02]],

         [[2.5269e-01, 1.0628e-04, 4.6074e-05,  ..., 4.2381e-03,
           9.3994e-02, 3.7262e-02]],

         [[6.0156e-01, 3.5524e-05, 4.9472e-05,  ..., 8.6823e-03,
           9.3323e-02, 2.3407e-02]],

         ...,

         [[7.5635e-01, 1.6093e-05, 1.2755e-05,  ..., 1.1711e-02,
           3.8391e-02, 1.7578e-02]],

         [[8.1055e-01, 2.7597e-05, 2.2411e-05,  ..., 4.8859e-02,
           2.3911e-02, 1.5434e-02]],

         [[4.4116e-01, 2.8610e-06, 2.9206e-06,  ..., 6.8054e-03,
           1.3641e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4341e-01, 7.2289e-04, 9.4128e-04,  ..., 1.7414e-03,
           8.0261e-03, 4.0550e-03]],

         [[1.1395e-01, 5.4002e-05, 4.3571e-05,  ..., 2.4338e-03,
           4.3945e-02, 1.2054e-02]],

         [[3.4717e-01, 1.2577e-05, 9.9540e-06,  ..., 1.5161e-01,
           2.7686e-01, 2.6978e-02]],

         ...,

         [[5.8447e-01, 2.2376e-04, 9.9242e-05,  ..., 7.2956e-04,
           2.5375e-02, 1.1551e-02]],

         [[9.2725e-01, 4.2617e-05, 4.6074e-05,  ..., 2.3842e-07,
           3.2425e-05, 3.4103e-03]],

         [[6.2793e-01, 1.3101e-04, 1.5771e-04,  ..., 4.2610e-03,
           4.0746e-04, 1.8585e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6924e-02, 4.6551e-05, 4.4107e-05,  ..., 8.1604e-02,
           5.5084e-03, 2.5425e-03]],

         [[1.8408e-01, 1.2398e-04, 5.9366e-05,  ..., 5.1193e-03,
           2.1469e-02, 7.3051e-03]],

         [[1.8787e-01, 2.3842e-06, 1.0729e-06,  ..., 3.2135e-02,
           4.3018e-01, 1.8494e-01]],

         ...,

         [[5.1562e-01, 7.3910e-06, 4.3511e-06,  ..., 2.3361e-02,
           5.8289e-02, 1.4061e-02]],

         [[2.7344e-01, 1.5497e-06, 1.6093e-06,  ..., 3.5248e-03,
           6.2598e-01, 5.0598e-02]],

         [[1.5588e-01, 2.5034e-06, 3.2783e-06,  ..., 2.3712e-02,
           6.1621e-01, 1.2622e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1963e-01, 1.5318e-05, 5.7817e-06,  ..., 5.1155e-03,
           4.1534e-02, 1.9928e-02]],

         [[1.3086e-01, 3.3975e-06, 4.3511e-06,  ..., 1.2683e-01,
           2.2266e-01, 1.4015e-02]],

         [[1.1737e-01, 7.3910e-06, 7.1526e-07,  ..., 2.3746e-04,
           2.5787e-02, 8.4033e-01]],

         ...,

         [[8.7952e-02, 5.3644e-05, 3.6836e-05,  ..., 1.0705e-04,
           2.8778e-02, 1.8005e-02]],

         [[3.7720e-01, 1.1748e-04, 4.8459e-05,  ..., 1.4420e-03,
           9.2239e-03, 2.1820e-03]],

         [[2.0581e-01, 1.1975e-04, 6.7472e-05,  ..., 6.6147e-03,
           1.0910e-02, 1.7605e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9372e-05, 1.0848e-05,  ..., 3.9597e-03,
           4.6051e-02, 4.5685e-02]],

         [[9.6582e-01, 1.5914e-05, 1.4305e-06,  ..., 1.1921e-07,
           1.9670e-06, 2.9335e-03]],

         [[8.2666e-01, 5.5909e-05, 6.7353e-06,  ..., 5.4598e-05,
           4.2105e-04, 8.1635e-03]],

         ...,

         [[2.1545e-01, 6.4421e-04, 1.1129e-03,  ..., 1.5574e-03,
           5.9242e-03, 3.1700e-03]],

         [[7.1094e-01, 1.0967e-05, 2.1517e-05,  ..., 1.6556e-03,
           2.5497e-02, 1.7517e-02]],

         [[2.8931e-01, 5.9700e-04, 1.6356e-03,  ..., 7.8964e-03,
           1.1391e-02, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1299e-01, 1.3280e-04, 1.2577e-04,  ..., 1.0468e-02,
           9.7580e-03, 1.4839e-03]],

         [[3.8062e-01, 1.8287e-04, 3.2854e-04,  ..., 3.0937e-03,
           1.9394e-02, 1.4420e-02]],

         [[4.3628e-01, 1.0872e-04, 9.6738e-05,  ..., 2.7776e-04,
           4.2114e-03, 4.2419e-03]],

         ...,

         [[4.3060e-02, 4.3154e-05, 6.7830e-05,  ..., 9.0256e-03,
           1.4575e-01, 3.5706e-02]],

         [[5.5267e-02, 5.4479e-05, 5.3644e-05,  ..., 1.3208e-03,
           1.5991e-01, 4.1718e-02]],

         [[4.8047e-01, 5.8830e-05, 6.8784e-05,  ..., 1.9627e-03,
           2.7969e-02, 1.1124e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0103e-01, 2.0981e-05, 2.3246e-05,  ..., 1.4839e-03,
           5.6580e-02, 2.0172e-02]],

         [[3.2324e-01, 2.2948e-04, 2.9707e-04,  ..., 3.6182e-03,
           1.8524e-02, 8.2169e-03]],

         [[3.0518e-01, 5.3525e-05, 1.1557e-04,  ..., 2.5253e-02,
           1.6467e-01, 2.8223e-01]],

         ...,

         [[5.6104e-01, 1.0937e-04, 1.6415e-04,  ..., 8.6517e-03,
           8.7402e-02, 5.8228e-02]],

         [[1.8640e-01, 3.9935e-06, 3.8862e-05,  ..., 1.9872e-04,
           4.8645e-02, 8.8074e-02]],

         [[2.5830e-01, 1.5450e-04, 2.0790e-04,  ..., 6.2084e-04,
           1.0414e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7004e-01, 7.6771e-05, 3.6061e-05,  ..., 1.2711e-02,
           1.7822e-01, 6.9275e-02]],

         [[1.0144e-01, 1.9073e-05, 3.3259e-05,  ..., 2.2293e-02,
           1.0632e-01, 1.5556e-02]],

         [[4.3652e-01, 5.3346e-05, 1.4877e-04,  ..., 1.5656e-02,
           1.0971e-02, 7.0000e-03]],

         ...,

         [[4.2017e-01, 2.3842e-07, 1.0431e-05,  ..., 3.8483e-02,
           5.1465e-01, 1.7349e-02]],

         [[3.1714e-01, 1.0133e-06, 8.6427e-06,  ..., 7.6538e-02,
           3.2715e-01, 1.6159e-02]],

         [[1.1072e-01, 1.7881e-07, 3.5167e-06,  ..., 3.3752e-02,
           8.3105e-01, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0117e-02, 9.5367e-07, 5.4836e-06,  ..., 9.3750e-02,
           9.8206e-02, 1.7075e-02]],

         [[2.0279e-02, 5.9605e-07, 1.0729e-06,  ..., 1.3185e-04,
           1.0681e-02, 9.5410e-01]],

         [[2.5558e-02, 2.1040e-05, 7.7009e-05,  ..., 1.8402e-02,
           9.5520e-03, 4.3030e-03]],

         ...,

         [[6.4575e-02, 6.4492e-05, 1.4126e-04,  ..., 4.4556e-02,
           3.1097e-02, 7.6523e-03]],

         [[1.0632e-01, 1.8477e-06, 2.8610e-06,  ..., 1.2476e-01,
           6.1182e-01, 4.5349e-02]],

         [[9.7351e-02, 4.8876e-06, 1.4603e-05,  ..., 1.6956e-01,
           6.0596e-01, 8.9722e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 1.1361e-04, 6.2561e-04,  ..., 2.5635e-03,
           1.0216e-02, 1.5457e-02]],

         [[9.7266e-01, 6.3717e-05, 2.7299e-05,  ..., 5.9664e-05,
           4.8208e-04, 1.6117e-03]],

         [[9.0332e-01, 1.2636e-05, 4.4703e-06,  ..., 4.2975e-05,
           3.7861e-04, 8.5220e-03]],

         ...,

         [[5.4834e-01, 5.0664e-06, 6.2585e-06,  ..., 1.7090e-02,
           1.6211e-01, 1.8958e-01]],

         [[4.3311e-01, 1.7881e-06, 7.1526e-07,  ..., 3.8576e-04,
           8.5754e-03, 4.8103e-03]],

         [[1.1932e-01, 2.8014e-06, 3.8743e-06,  ..., 1.3895e-03,
           7.0129e-02, 7.4268e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6627e-02, 4.1723e-07, 1.3709e-06,  ..., 1.5686e-01,
           3.2446e-01, 5.6366e-02]],

         [[2.3242e-01, 1.7881e-06, 1.2517e-06,  ..., 2.4979e-02,
           4.8462e-01, 2.0044e-01]],

         [[1.0431e-01, 9.2030e-05, 9.2089e-05,  ..., 2.0615e-02,
           1.5967e-01, 3.7628e-02]],

         ...,

         [[1.8652e-01, 1.4901e-06, 9.9540e-06,  ..., 7.5928e-02,
           3.3765e-01, 2.0959e-01]],

         [[5.7471e-01, 1.4484e-05, 4.3690e-05,  ..., 1.9196e-02,
           1.0773e-02, 2.9510e-02]],

         [[6.7236e-01, 5.0068e-06, 1.5736e-05,  ..., 9.5901e-03,
           2.4689e-02, 3.3997e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3804e-02, 1.9670e-06, 1.3709e-06,  ..., 7.3738e-03,
           5.7526e-02, 3.0090e-02]],

         [[1.6809e-01, 2.0206e-05, 1.0133e-05,  ..., 2.7013e-04,
           4.8828e-03, 4.3411e-03]],

         [[2.6196e-01, 2.3484e-05, 7.7248e-05,  ..., 3.4828e-03,
           1.0669e-01, 6.6772e-02]],

         ...,

         [[3.9307e-02, 4.2915e-06, 6.4433e-05,  ..., 2.7451e-02,
           2.2049e-02, 1.0017e-02]],

         [[1.1084e-01, 1.2279e-05, 6.9439e-05,  ..., 2.3224e-02,
           9.5367e-03, 6.2523e-03]],

         [[2.7783e-01, 2.2292e-05, 3.5107e-05,  ..., 2.5708e-01,
           1.6077e-01, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0654e-01, 2.3842e-07, 1.0133e-06,  ..., 4.2145e-02,
           6.1426e-01, 1.1774e-01]],

         [[2.8125e-01, 4.0531e-06, 2.8253e-05,  ..., 9.3506e-02,
           2.8027e-01, 1.0675e-01]],

         [[1.9336e-01, 5.3644e-07, 1.0014e-05,  ..., 8.2520e-02,
           6.5918e-01, 4.7211e-02]],

         ...,

         [[4.7607e-01, 3.8505e-05, 4.7386e-05,  ..., 3.9291e-03,
           4.7569e-03, 6.9504e-03]],

         [[7.1594e-02, 2.6226e-06, 3.0398e-06,  ..., 5.7983e-02,
           3.5254e-01, 9.4849e-02]],

         [[4.5715e-02, 2.1756e-05, 6.3419e-05,  ..., 3.0746e-03,
           1.5222e-01, 5.9277e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4751e-01, 3.0220e-05, 9.2506e-05,  ..., 4.3030e-03,
           1.4868e-01, 9.6802e-02]],

         [[9.8083e-02, 9.1195e-06, 3.1531e-05,  ..., 1.0854e-04,
           2.0847e-03, 1.7281e-03]],

         [[2.1118e-01, 8.2970e-05, 1.4770e-04,  ..., 1.2217e-03,
           6.2927e-02, 1.9562e-02]],

         ...,

         [[3.3478e-02, 5.5730e-05, 5.3585e-05,  ..., 2.2659e-02,
           7.5000e-01, 1.5967e-01]],

         [[3.1250e-01, 1.1075e-04, 7.6246e-04,  ..., 2.9984e-02,
           1.0468e-01, 4.2041e-01]],

         [[4.1968e-01, 2.0742e-05, 4.1306e-05,  ..., 2.1225e-02,
           2.6685e-01, 2.5464e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 3.2723e-05, 1.6534e-04,  ..., 3.4943e-02,
           6.7627e-02, 7.1678e-03]],

         [[6.6223e-02, 1.8477e-06, 1.6391e-05,  ..., 1.3684e-01,
           1.5454e-01, 3.2013e-02]],

         [[4.0894e-01, 1.5008e-04, 7.2241e-04,  ..., 3.7964e-02,
           6.0059e-02, 1.5854e-02]],

         ...,

         [[3.0029e-01, 6.7949e-05, 1.5736e-04,  ..., 1.1945e-01,
           6.2675e-03, 6.9389e-03]],

         [[9.0771e-01, 5.9605e-06, 4.9257e-04,  ..., 1.1718e-04,
           1.3447e-03, 1.3260e-02]],

         [[5.1270e-01, 7.3314e-06, 2.5749e-05,  ..., 3.3417e-02,
           2.6031e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5809e-05, 1.2994e-05, 1.0345e-01,  ..., 1.3412e-02,
           1.6891e-02, 1.3840e-02]],

         [[3.0220e-05, 3.8564e-05, 1.7725e-01,  ..., 2.0920e-02,
           1.0040e-02, 9.0332e-03]],

         [[1.4687e-04, 1.0639e-04, 9.6802e-02,  ..., 1.6373e-02,
           8.4412e-02, 1.2000e-01]],

         ...,

         [[7.0572e-05, 1.2100e-05, 1.9080e-01,  ..., 9.0149e-02,
           2.6294e-01, 1.2225e-01]],

         [[7.9536e-04, 1.0157e-03, 2.0764e-01,  ..., 1.2085e-02,
           3.7384e-02, 2.7451e-02]],

         [[2.5558e-04, 6.8247e-05, 2.7588e-01,  ..., 1.1997e-03,
           7.9498e-03, 2.8572e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7241e-04, 1.4079e-04, 3.2812e-01,  ..., 7.5245e-04,
           1.8187e-03, 2.1763e-03]],

         [[2.9707e-04, 1.5628e-04, 2.1533e-01,  ..., 7.6771e-04,
           1.1129e-03, 1.3371e-03]],

         [[6.8545e-05, 1.6689e-05, 2.9272e-01,  ..., 8.5831e-04,
           1.6761e-04, 2.5616e-03]],

         ...,

         [[6.2752e-04, 4.8137e-04, 2.8687e-01,  ..., 1.5240e-03,
           1.7328e-03, 3.1447e-04]],

         [[6.0201e-06, 1.0133e-05, 2.3059e-01,  ..., 9.0981e-04,
           2.0103e-03, 1.8816e-03]],

         [[6.0320e-05, 2.1601e-04, 1.5149e-01,  ..., 1.3130e-02,
           2.6875e-03, 6.2609e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0562e-04, 4.8518e-05, 1.6785e-01,  ..., 4.8399e-04,
           1.5976e-02, 1.3237e-03]],

         [[2.4986e-04, 8.2254e-06, 2.7295e-01,  ..., 8.4829e-04,
           1.1712e-04, 5.0259e-04]],

         [[1.9932e-04, 1.4389e-04, 2.8394e-01,  ..., 1.0166e-03,
           4.0126e-04, 8.7976e-04]],

         ...,

         [[3.6657e-05, 2.2650e-05, 5.8105e-02,  ..., 6.3438e-03,
           4.7534e-01, 8.8623e-02]],

         [[6.9737e-05, 1.1301e-04, 2.4500e-01,  ..., 9.6083e-04,
           2.1877e-03, 2.1496e-03]],

         [[4.4370e-04, 1.1724e-04, 1.7383e-01,  ..., 1.4019e-03,
           1.0239e-02, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3411e-05, 7.7486e-07, 1.0891e-03,  ..., 1.7480e-01,
           7.1338e-01, 1.9608e-02]],

         [[8.5711e-05, 8.2552e-05, 9.7717e-02,  ..., 5.2521e-02,
           1.3863e-02, 6.4850e-03]],

         [[5.3835e-04, 1.5316e-03, 2.3499e-01,  ..., 1.1387e-03,
           1.8806e-03, 1.0691e-03]],

         ...,

         [[1.1826e-03, 3.0756e-05, 7.2205e-02,  ..., 2.3438e-02,
           1.1353e-01, 2.7069e-02]],

         [[4.6659e-04, 6.1154e-05, 2.4948e-02,  ..., 2.4524e-01,
           4.4751e-01, 7.5378e-02]],

         [[1.0252e-05, 3.6359e-06, 1.9779e-03,  ..., 1.1279e-01,
           3.6987e-01, 2.2253e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1656e-03, 1.1091e-03, 2.2510e-01,  ..., 4.6577e-03,
           5.0018e-02, 6.8909e-02]],

         [[3.7622e-04, 6.9475e-04, 2.5879e-01,  ..., 9.1324e-03,
           1.3390e-02, 2.2430e-02]],

         [[1.7729e-03, 1.1063e-03, 2.5342e-01,  ..., 2.7328e-02,
           4.1046e-02, 1.7731e-02]],

         ...,

         [[2.1803e-04, 6.3419e-05, 1.9666e-01,  ..., 7.1777e-02,
           1.4160e-01, 5.9052e-02]],

         [[2.1100e-04, 1.6952e-04, 4.6265e-02,  ..., 2.0218e-02,
           1.8591e-01, 4.8975e-01]],

         [[5.9009e-06, 5.3644e-06, 1.8494e-02,  ..., 1.8103e-01,
           4.1772e-01, 1.3147e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.0383e-04, 5.3883e-05, 2.8515e-03,  ..., 5.9082e-02,
           5.1025e-01, 1.1749e-01]],

         [[1.2064e-03, 2.3758e-04, 1.5488e-03,  ..., 1.0643e-03,
           2.1378e-02, 8.0225e-01]],

         [[5.9605e-08, 0.0000e+00, 1.7881e-07,  ..., 2.7359e-05,
           5.2185e-03, 9.9463e-01]],

         ...,

         [[1.3709e-06, 2.9802e-07, 0.0000e+00,  ..., 1.6689e-05,
           9.0981e-04, 9.9805e-01]],

         [[1.9608e-03, 5.1651e-03, 1.2655e-03,  ..., 7.0572e-03,
           2.2430e-02, 9.9564e-03]],

         [[7.3509e-03, 2.5406e-03, 5.3263e-04,  ..., 7.9041e-02,
           3.2959e-02, 3.3997e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7752e-06, 7.4744e-05, 9.5963e-05,  ..., 8.3694e-03,
           2.7451e-02, 8.8086e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 1.3721e-01,
           7.5342e-01, 8.0688e-02]],

         [[1.2817e-03, 3.5954e-03, 9.1782e-03,  ..., 1.8829e-02,
           6.7177e-03, 1.3359e-02]],

         ...,

         [[1.1606e-03, 1.7977e-03, 3.6888e-03,  ..., 2.6270e-01,
           2.6270e-01, 4.6722e-02]],

         [[2.2469e-03, 7.5684e-03, 4.4586e-02,  ..., 9.4604e-03,
           5.3902e-03, 8.6136e-03]],

         [[3.3092e-03, 6.7368e-03, 1.6037e-02,  ..., 6.1005e-02,
           5.7281e-02, 3.6713e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0680e-02, 3.4466e-03, 1.4565e-02,  ..., 3.9429e-02,
           3.4546e-02, 4.2297e-02]],

         [[1.8368e-03, 5.2404e-04, 3.9177e-03,  ..., 1.5625e-02,
           6.4621e-03, 8.9722e-03]],

         [[5.3520e-03, 1.1044e-03, 1.3485e-03,  ..., 2.1164e-02,
           1.0727e-02, 1.1597e-02]],

         ...,

         [[2.1160e-04, 6.5804e-05, 9.3746e-04,  ..., 1.1719e-01,
           4.4922e-01, 2.7246e-01]],

         [[9.5940e-04, 2.7800e-04, 3.8457e-04,  ..., 2.4512e-01,
           1.2421e-01, 6.2927e-02]],

         [[1.4732e-02, 1.4820e-03, 6.4354e-03,  ..., 3.1433e-02,
           3.5919e-02, 5.6592e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.3604e-01, 7.0930e-06, 5.3048e-06,  ..., 3.6354e-03,
           2.0790e-03, 1.9302e-02]],

         [[9.1602e-01, 6.6757e-06, 2.8014e-06,  ..., 2.6077e-02,
           3.3356e-02, 1.2444e-02]],

         [[9.3018e-01, 5.1856e-06, 1.4305e-05,  ..., 2.8900e-02,
           4.7607e-03, 1.9119e-02]],

         ...,

         [[4.7754e-01, 1.2934e-04, 6.6948e-04,  ..., 5.8889e-05,
           1.8084e-04, 2.4780e-01]],

         [[2.1924e-01, 6.0499e-05, 2.6631e-04,  ..., 4.2261e-01,
           1.6553e-01, 3.4149e-02]],

         [[5.6787e-01, 1.7309e-04, 6.5422e-04,  ..., 9.2346e-02,
           2.5742e-02, 7.2205e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.0723e-01, 3.0041e-05, 2.6345e-05,  ..., 1.8203e-04,
           1.0496e-04, 4.9561e-02]],

         [[8.7646e-01, 5.9605e-08, 2.2054e-06,  ..., 1.0109e-02,
           7.9529e-02, 2.5253e-02]],

         [[9.4531e-01, 1.3423e-04, 3.6538e-05,  ..., 6.4802e-04,
           4.8637e-03, 1.8244e-03]],

         ...,

         [[5.5908e-01, 1.2791e-04, 2.3115e-04,  ..., 3.1174e-02,
           4.4403e-02, 5.9845e-02]],

         [[3.7549e-01, 1.0777e-03, 9.5558e-04,  ..., 1.2451e-02,
           7.1640e-03, 1.1047e-02]],

         [[6.8896e-01, 3.4809e-04, 5.3024e-04,  ..., 1.3196e-01,
           7.4585e-02, 4.2938e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0752e-01, 8.2195e-05, 2.9624e-05,  ..., 7.8430e-02,
           4.6661e-02, 4.4403e-02]],

         [[7.5830e-01, 6.6221e-05, 1.1861e-05,  ..., 9.5703e-02,
           1.9058e-02, 4.6539e-02]],

         [[8.0127e-01, 6.8069e-05, 2.4080e-05,  ..., 9.7107e-02,
           1.5656e-02, 3.2745e-02]],

         ...,

         [[9.1553e-01, 2.2054e-06, 2.9802e-07,  ..., 1.7881e-07,
           2.1458e-06, 8.3191e-02]],

         [[9.6484e-01, 1.0192e-05, 3.6955e-06,  ..., 0.0000e+00,
           5.9605e-08, 2.7328e-02]],

         [[9.8828e-01, 9.5367e-07, 1.1921e-07,  ..., 0.0000e+00,
           0.0000e+00, 1.0780e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2708e-01, 1.4067e-05, 2.0266e-06,  ..., 4.9097e-01,
           2.1692e-01, 4.6143e-02]],

         [[8.9795e-01, 5.9009e-06, 1.7881e-06,  ..., 5.9652e-04,
           3.5763e-04, 1.2619e-02]],

         [[8.7061e-01, 1.5450e-04, 1.6630e-05,  ..., 8.1100e-03,
           3.5591e-03, 2.2232e-02]],

         ...,

         [[9.1357e-01, 1.1587e-04, 7.1526e-06,  ..., 4.6921e-03,
           1.3771e-02, 1.4038e-02]],

         [[5.7715e-01, 6.8128e-05, 1.0908e-05,  ..., 5.1086e-02,
           1.5271e-01, 7.7209e-02]],

         [[4.8560e-01, 1.4067e-05, 5.2452e-06,  ..., 8.4229e-02,
           3.7891e-01, 1.4282e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3525e-01, 1.5199e-05, 3.3736e-05,  ..., 6.0059e-02,
           1.3953e-01, 5.3040e-02]],

         [[8.6279e-01, 1.5378e-05, 1.3173e-05,  ..., 3.3207e-03,
           3.5057e-03, 2.8152e-02]],

         [[5.9229e-01, 7.6294e-06, 7.0930e-06,  ..., 9.8022e-02,
           1.8091e-01, 5.4382e-02]],

         ...,

         [[5.3320e-01, 3.3379e-05, 1.3173e-05,  ..., 8.7158e-02,
           4.7394e-02, 9.6008e-02]],

         [[8.6035e-01, 6.9439e-05, 1.5140e-05,  ..., 1.1559e-02,
           9.1248e-03, 1.7532e-02]],

         [[8.5449e-01, 1.5020e-05, 9.4175e-06,  ..., 5.3253e-02,
           1.4374e-02, 8.6823e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0879e-01, 7.3767e-04, 4.5276e-04,  ..., 7.7019e-03,
           4.5204e-03, 1.0384e-02]],

         [[9.0869e-01, 2.5034e-06, 1.8179e-05,  ..., 5.8327e-03,
           3.4962e-03, 5.0278e-03]],

         [[8.4326e-01, 3.4571e-06, 3.6955e-06,  ..., 4.9591e-03,
           7.6485e-03, 1.0872e-02]],

         ...,

         [[7.5781e-01, 2.4378e-05, 5.2869e-05,  ..., 7.4883e-03,
           2.4734e-02, 1.0602e-01]],

         [[1.8359e-01, 1.5557e-05, 7.2122e-06,  ..., 5.1270e-01,
           1.8860e-01, 2.9099e-02]],

         [[4.3311e-01, 1.2040e-05, 1.2994e-05,  ..., 1.4095e-03,
           3.2715e-02, 5.2051e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6328e-01, 2.9731e-04, 8.5020e-04,  ..., 1.4320e-02,
           9.6359e-03, 5.7983e-03]],

         [[6.1816e-01, 1.2577e-05, 4.6790e-05,  ..., 5.4626e-02,
           1.3733e-01, 1.3049e-01]],

         [[6.3135e-01, 1.2398e-05, 1.6510e-05,  ..., 5.1880e-02,
           1.1359e-01, 5.6915e-02]],

         ...,

         [[8.5254e-01, 2.5392e-05, 5.1081e-05,  ..., 1.9569e-03,
           4.2582e-04, 1.3552e-03]],

         [[5.0635e-01, 1.5259e-05, 5.3644e-06,  ..., 8.4961e-02,
           1.7236e-01, 1.7102e-01]],

         [[9.6680e-01, 3.0994e-06, 5.1260e-06,  ..., 2.1992e-03,
           1.5297e-02, 7.8583e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3398e-01, 8.1182e-05, 4.6611e-05,  ..., 1.4663e-05,
           4.7386e-05, 1.4282e-02]],

         [[8.2031e-01, 8.1658e-06, 1.0669e-05,  ..., 3.6102e-02,
           2.0020e-02, 2.8793e-02]],

         [[8.9258e-01, 1.6093e-06, 1.2517e-06,  ..., 5.1346e-03,
           1.2579e-03, 9.3307e-03]],

         ...,

         [[8.1006e-01, 1.5855e-04, 9.9182e-05,  ..., 2.6627e-02,
           1.2207e-02, 1.4709e-02]],

         [[7.7881e-01, 9.5010e-05, 4.8161e-05,  ..., 1.8167e-03,
           1.7681e-03, 4.6425e-03]],

         [[7.4609e-01, 2.6846e-04, 1.9801e-04,  ..., 2.6505e-02,
           1.3786e-02, 6.9656e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6143e-01, 1.7881e-06, 2.3246e-06,  ..., 1.0815e-03,
           8.6670e-03, 5.1465e-01]],

         [[7.2314e-01, 6.0844e-04, 6.5470e-04,  ..., 2.7924e-03,
           2.4261e-03, 8.5678e-03]],

         [[5.3271e-01, 4.9686e-04, 9.2697e-04,  ..., 2.0996e-02,
           1.2390e-02, 6.1707e-02]],

         ...,

         [[5.2783e-01, 1.6105e-04, 3.6860e-04,  ..., 9.8267e-03,
           8.7891e-03, 3.0499e-03]],

         [[4.5532e-01, 3.6955e-05, 4.4942e-05,  ..., 2.0776e-01,
           1.3745e-01, 3.4363e-02]],

         [[4.2896e-01, 4.2915e-05, 1.1808e-04,  ..., 4.7974e-02,
           6.8176e-02, 3.4375e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0630e-01, 1.3471e-05, 1.4722e-05,  ..., 9.6252e-02,
           5.7831e-02, 5.8447e-01]],

         [[6.1914e-01, 4.4167e-05, 5.7399e-05,  ..., 2.6093e-02,
           8.2275e-02, 9.0515e-02]],

         [[3.0811e-01, 2.2352e-05, 1.0908e-05,  ..., 2.5977e-01,
           2.0300e-01, 6.6223e-02]],

         ...,

         [[1.3382e-02, 1.3113e-06, 1.0729e-06,  ..., 6.2287e-05,
           2.2602e-03, 9.8096e-01]],

         [[5.7959e-01, 6.3777e-05, 4.5061e-05,  ..., 4.1382e-02,
           4.4159e-02, 1.0156e-01]],

         [[8.0664e-01, 4.3988e-05, 1.3411e-05,  ..., 2.4246e-02,
           2.2949e-02, 5.0446e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9658e-01, 2.6369e-04, 6.0558e-04,  ..., 1.8845e-02,
           4.4708e-02, 2.2009e-01]],

         [[6.4697e-01, 2.1458e-05, 3.4451e-05,  ..., 1.4847e-02,
           2.7069e-02, 3.0838e-02]],

         [[6.9531e-01, 2.8729e-05, 8.9943e-05,  ..., 1.7996e-03,
           1.1940e-03, 1.9272e-02]],

         ...,

         [[2.8784e-01, 5.9605e-08, 1.4901e-06,  ..., 1.7029e-02,
           6.7871e-01, 1.4336e-02]],

         [[5.1904e-01, 3.8123e-04, 5.5027e-04,  ..., 2.7027e-03,
           3.0384e-03, 2.1343e-03]],

         [[2.9224e-01, 1.7881e-07, 1.5497e-06,  ..., 4.6600e-02,
           5.8105e-01, 7.5928e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0605e-01, 2.2888e-05, 1.5140e-05,  ..., 4.2419e-02,
           4.5776e-02, 4.9255e-02]],

         [[1.1859e-01, 7.2861e-04, 3.2854e-04,  ..., 4.9400e-04,
           3.2282e-04, 2.0828e-03]],

         [[6.5479e-01, 1.4603e-04, 1.9956e-04,  ..., 4.0665e-03,
           3.8509e-03, 1.7700e-02]],

         ...,

         [[6.1914e-01, 2.0683e-05, 3.2783e-05,  ..., 5.0903e-02,
           1.0645e-01, 4.3060e-02]],

         [[4.7021e-01, 3.3796e-05, 6.1393e-05,  ..., 7.3486e-02,
           4.0894e-02, 3.3447e-01]],

         [[5.3076e-01, 4.4584e-05, 5.6803e-05,  ..., 4.4464e-02,
           3.0441e-02, 2.9282e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7708e-02, 1.1325e-06, 1.6093e-06,  ..., 4.5061e-04,
           4.3221e-03, 8.9600e-01]],

         [[2.2766e-01, 1.2934e-05, 2.5749e-05,  ..., 2.2229e-01,
           8.0688e-02, 5.1178e-02]],

         [[1.7847e-01, 8.2445e-04, 1.0443e-03,  ..., 1.3107e-02,
           3.9558e-03, 1.2955e-02]],

         ...,

         [[3.6499e-01, 1.0633e-04, 9.3818e-05,  ..., 1.2439e-01,
           1.4917e-01, 4.4342e-02]],

         [[6.4551e-01, 1.6809e-05, 6.0618e-05,  ..., 4.8309e-02,
           1.2459e-02, 4.2175e-02]],

         [[3.8208e-01, 5.6446e-05, 4.3273e-05,  ..., 1.2622e-01,
           7.4524e-02, 2.1423e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0059e-01, 5.4359e-05, 3.7849e-05,  ..., 4.1351e-02,
           1.2152e-01, 6.1829e-02]],

         [[3.4033e-01, 1.3947e-04, 1.2279e-04,  ..., 7.9590e-02,
           5.6854e-02, 2.8046e-02]],

         [[3.2593e-01, 2.4498e-05, 2.3663e-05,  ..., 2.6392e-01,
           2.8320e-01, 4.9011e-02]],

         ...,

         [[5.8594e-01, 9.7156e-06, 1.7703e-05,  ..., 1.1536e-01,
           1.4355e-01, 2.8503e-02]],

         [[7.6709e-01, 7.6890e-06, 1.0073e-05,  ..., 9.8877e-03,
           4.2114e-02, 1.0754e-01]],

         [[6.2109e-01, 7.5102e-06, 7.5698e-06,  ..., 1.0399e-02,
           7.6103e-03, 1.5312e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4326e-01, 9.7036e-04, 1.4830e-03,  ..., 2.1667e-03,
           1.0967e-03, 2.8458e-03]],

         [[1.4209e-01, 9.1910e-05, 7.7963e-05,  ..., 7.7576e-02,
           1.8570e-02, 2.6031e-02]],

         [[4.5288e-01, 1.1802e-05, 1.7941e-05,  ..., 1.4197e-01,
           1.4819e-01, 9.5276e-02]],

         ...,

         [[8.0908e-01, 5.9128e-05, 4.6015e-05,  ..., 7.0000e-03,
           5.8708e-03, 1.7090e-02]],

         [[9.9316e-01, 6.4373e-06, 3.1173e-05,  ..., 1.1921e-07,
           1.0729e-06, 1.5373e-03]],

         [[7.7441e-01, 1.2684e-04, 1.1373e-04,  ..., 1.1511e-03,
           1.2474e-03, 4.9734e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6528e-01, 4.6313e-05, 4.9710e-05,  ..., 3.4294e-03,
           1.2417e-03, 1.0162e-02]],

         [[2.7124e-01, 1.8704e-04, 9.6083e-05,  ..., 1.9180e-02,
           4.7760e-03, 1.8738e-02]],

         [[2.8711e-01, 2.5630e-06, 7.1526e-07,  ..., 2.1667e-01,
           9.6191e-02, 2.0520e-01]],

         ...,

         [[3.4839e-01, 4.7088e-06, 3.8147e-06,  ..., 1.2329e-01,
           9.0881e-02, 1.9821e-02]],

         [[3.7793e-01, 2.1458e-06, 2.1458e-06,  ..., 2.8296e-01,
           1.4453e-01, 7.8003e-02]],

         [[2.0435e-01, 8.9407e-07, 1.9670e-06,  ..., 2.5830e-01,
           1.8323e-01, 2.1582e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6689e-01, 5.3644e-05, 2.6166e-05,  ..., 3.5126e-02,
           1.2329e-02, 7.0923e-02]],

         [[1.2177e-01, 2.5034e-06, 2.1458e-06,  ..., 3.7500e-01,
           1.3794e-01, 4.0802e-02]],

         [[1.7065e-01, 6.4373e-06, 1.0133e-06,  ..., 1.1864e-03,
           4.9934e-03, 8.1396e-01]],

         ...,

         [[1.8030e-01, 1.1653e-04, 7.1526e-05,  ..., 2.9354e-03,
           7.3671e-04, 8.5602e-03]],

         [[5.0781e-01, 1.3506e-04, 6.4671e-05,  ..., 5.3024e-03,
           1.2836e-03, 2.1591e-03]],

         [[2.3987e-01, 1.3471e-04, 1.0234e-04,  ..., 5.8319e-02,
           9.8267e-03, 1.2230e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4351e-01, 2.8312e-05, 1.2636e-05,  ..., 1.0156e-01,
           8.8928e-02, 8.2886e-02]],

         [[9.8535e-01, 1.0550e-05, 2.9802e-07,  ..., 7.7486e-06,
           4.8757e-05, 1.8990e-04]],

         [[8.9893e-01, 6.1750e-05, 1.0073e-05,  ..., 5.8126e-04,
           1.1787e-03, 4.2610e-03]],

         ...,

         [[1.3342e-01, 7.2193e-04, 1.3075e-03,  ..., 6.1417e-03,
           2.2602e-03, 7.4081e-03]],

         [[5.3125e-01, 3.1471e-05, 4.5419e-05,  ..., 4.1626e-02,
           4.4983e-02, 2.9510e-02]],

         [[4.5288e-01, 1.0386e-03, 2.9125e-03,  ..., 1.8692e-02,
           1.6113e-02, 4.8096e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6606e-01, 1.2660e-04, 1.4126e-04,  ..., 1.0872e-02,
           2.2621e-03, 7.4730e-03]],

         [[3.9648e-01, 3.9077e-04, 4.9400e-04,  ..., 1.2543e-02,
           4.4022e-03, 3.9856e-02]],

         [[5.3906e-01, 5.3048e-05, 1.1951e-04,  ..., 5.7144e-03,
           3.1796e-03, 1.2581e-02]],

         ...,

         [[1.3049e-01, 2.2948e-04, 2.1219e-04,  ..., 9.2590e-02,
           2.2339e-02, 7.2083e-02]],

         [[1.8262e-01, 1.1623e-04, 1.5640e-04,  ..., 7.2632e-02,
           2.7573e-02, 6.6162e-02]],

         [[5.2832e-01, 5.3644e-05, 8.8453e-05,  ..., 1.9852e-02,
           1.5228e-02, 3.8269e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8613e-01, 4.2617e-05, 6.0320e-05,  ..., 1.9067e-01,
           6.8481e-02, 2.3865e-02]],

         [[3.4644e-01, 4.6515e-04, 5.2500e-04,  ..., 3.7659e-02,
           1.6724e-02, 2.7344e-02]],

         [[3.6060e-01, 7.0572e-05, 1.6987e-04,  ..., 5.0354e-02,
           3.1250e-02, 3.2837e-01]],

         ...,

         [[7.1289e-01, 6.9082e-05, 1.2076e-04,  ..., 3.3875e-02,
           3.3081e-02, 3.9917e-02]],

         [[7.8516e-01, 1.6570e-05, 9.1732e-05,  ..., 2.4366e-04,
           5.8889e-04, 7.7759e-02]],

         [[4.3066e-01, 6.3598e-05, 1.8263e-04,  ..., 1.2901e-02,
           1.3107e-02, 4.3907e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3572e-01, 9.5189e-05, 7.7546e-05,  ..., 1.4294e-01,
           9.0881e-02, 2.9037e-02]],

         [[1.7236e-01, 4.0352e-05, 3.0935e-05,  ..., 9.5215e-02,
           1.8600e-02, 2.6855e-02]],

         [[6.0742e-01, 7.8082e-05, 1.4138e-04,  ..., 2.0065e-02,
           1.5266e-02, 1.2077e-02]],

         ...,

         [[2.0850e-01, 0.0000e+00, 2.3842e-07,  ..., 9.3994e-02,
           6.5234e-01, 3.6804e-02]],

         [[2.8320e-01, 1.5497e-06, 1.4365e-05,  ..., 1.4929e-01,
           8.0505e-02, 5.0385e-02]],

         [[5.3223e-01, 2.9802e-07, 1.1802e-05,  ..., 6.7688e-02,
           3.1787e-01, 5.9723e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0918e-02, 5.3644e-07, 5.5432e-06,  ..., 1.4954e-01,
           2.4414e-02, 1.3588e-02]],

         [[6.5308e-02, 1.8477e-06, 2.2650e-06,  ..., 1.3142e-03,
           1.5038e-02, 8.8770e-01]],

         [[1.0864e-01, 3.6716e-05, 2.3305e-04,  ..., 1.9638e-02,
           5.6915e-03, 3.1616e-02]],

         ...,

         [[4.4067e-02, 9.6023e-05, 2.5010e-04,  ..., 2.3773e-02,
           7.9651e-03, 6.3477e-03]],

         [[3.7903e-02, 3.5763e-07, 4.1723e-07,  ..., 5.6592e-01,
           2.8442e-01, 3.0228e-02]],

         [[2.8833e-01, 1.0133e-06, 5.4240e-06,  ..., 9.3201e-02,
           3.7305e-01, 2.1252e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1621e-01, 1.2898e-04, 6.1464e-04,  ..., 1.1108e-02,
           1.9043e-02, 6.4148e-02]],

         [[9.7559e-01, 4.2856e-05, 6.6578e-05,  ..., 3.6502e-04,
           3.5930e-04, 2.6550e-03]],

         [[9.2773e-01, 5.3048e-06, 6.7949e-06,  ..., 7.2622e-04,
           3.7327e-03, 2.0020e-02]],

         ...,

         [[5.8350e-01, 4.5896e-06, 7.0930e-06,  ..., 2.3331e-02,
           1.4294e-01, 1.9543e-01]],

         [[5.4834e-01, 2.0862e-06, 1.0729e-06,  ..., 7.8201e-03,
           3.9330e-03, 4.5967e-03]],

         [[1.7529e-01, 6.6161e-06, 1.3530e-05,  ..., 2.2995e-02,
           3.8513e-02, 7.1533e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6804e-02, 1.7881e-07, 4.1723e-07,  ..., 2.1008e-01,
           1.6235e-01, 2.2534e-01]],

         [[3.6035e-01, 2.9802e-06, 1.6689e-06,  ..., 1.7981e-01,
           2.0532e-01, 1.3684e-01]],

         [[7.8552e-02, 2.3961e-05, 3.3379e-05,  ..., 8.5632e-02,
           2.3499e-02, 2.5696e-02]],

         ...,

         [[2.0178e-01, 2.3246e-06, 1.1921e-05,  ..., 2.1484e-01,
           2.9590e-01, 8.6121e-02]],

         [[5.4932e-01, 1.6332e-05, 5.6088e-05,  ..., 1.4519e-02,
           5.2704e-02, 1.4748e-02]],

         [[8.2275e-01, 3.3975e-06, 8.8811e-06,  ..., 6.7902e-03,
           1.8906e-02, 1.3191e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3549e-02, 4.3511e-06, 4.5300e-06,  ..., 5.5481e-02,
           2.1896e-02, 1.2772e-02]],

         [[2.3694e-01, 1.7345e-05, 1.2279e-05,  ..., 1.1612e-02,
           8.5602e-03, 3.8891e-03]],

         [[2.7002e-01, 2.4974e-05, 7.3075e-05,  ..., 9.2590e-02,
           7.6721e-02, 4.9164e-02]],

         ...,

         [[3.5645e-02, 6.4969e-06, 9.1851e-05,  ..., 1.2711e-02,
           6.9618e-03, 2.8854e-02]],

         [[2.2363e-01, 3.5465e-05, 1.9157e-04,  ..., 6.0539e-03,
           3.3970e-03, 8.6746e-03]],

         [[1.6370e-01, 2.5749e-05, 4.9591e-05,  ..., 1.4001e-01,
           1.2164e-01, 1.0406e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4971e-01, 7.1526e-07, 2.9206e-06,  ..., 7.0618e-02,
           1.1963e-01, 2.9272e-01]],

         [[2.2229e-01, 7.1526e-06, 3.4332e-05,  ..., 1.9922e-01,
           2.9688e-01, 5.2399e-02]],

         [[2.3877e-01, 0.0000e+00, 5.3644e-07,  ..., 4.1992e-02,
           5.7080e-01, 1.3708e-01]],

         ...,

         [[5.6641e-01, 2.1935e-05, 3.8505e-05,  ..., 3.4752e-03,
           3.9520e-03, 1.2293e-03]],

         [[1.3159e-01, 5.0664e-06, 3.5167e-06,  ..., 2.8516e-01,
           2.1863e-01, 6.4148e-02]],

         [[4.0863e-02, 1.6332e-05, 3.4988e-05,  ..., 4.8889e-02,
           1.4368e-01, 5.2979e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6074e-01, 1.8477e-05, 7.5698e-05,  ..., 1.8494e-01,
           1.2708e-01, 6.6956e-02]],

         [[9.9243e-02, 7.3910e-06, 4.2975e-05,  ..., 2.7390e-03,
           1.6356e-03, 6.6328e-04]],

         [[2.6318e-01, 1.2457e-04, 2.6202e-04,  ..., 1.3232e-01,
           3.4393e-02, 9.7427e-03]],

         ...,

         [[2.2385e-02, 1.7047e-05, 3.2127e-05,  ..., 5.0146e-01,
           4.0942e-01, 3.2562e-02]],

         [[1.0706e-01, 2.9087e-05, 3.2520e-04,  ..., 6.9504e-03,
           4.4617e-02, 7.7881e-01]],

         [[4.3701e-01, 3.8564e-05, 3.3736e-05,  ..., 6.0089e-02,
           4.1699e-01, 5.3864e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4609e-01, 1.3232e-05, 8.7559e-05,  ..., 6.8054e-02,
           4.8828e-02, 2.0950e-02]],

         [[1.6113e-01, 1.6689e-06, 1.7762e-05,  ..., 1.2500e-01,
           1.6626e-01, 6.1646e-02]],

         [[5.0195e-01, 1.2910e-04, 5.2166e-04,  ..., 3.6285e-02,
           2.1164e-02, 1.3908e-02]],

         ...,

         [[5.0146e-01, 6.2525e-05, 1.4424e-04,  ..., 4.8790e-03,
           9.1553e-03, 8.0948e-03]],

         [[9.2383e-01, 8.4043e-06, 7.1955e-04,  ..., 4.3845e-04,
           3.4447e-03, 3.2864e-03]],

         [[4.8438e-01, 1.1146e-05, 4.1783e-05,  ..., 3.3600e-02,
           7.0862e-02, 2.5574e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7285e-05, 1.3530e-05, 1.1401e-01,  ..., 7.0381e-03,
           1.5076e-02, 2.1725e-03]],

         [[3.6418e-05, 5.7101e-05, 1.5991e-01,  ..., 1.6266e-02,
           2.4429e-02, 4.4327e-03]],

         [[1.8513e-04, 1.7405e-04, 7.2327e-02,  ..., 6.8481e-02,
           1.0522e-01, 1.9012e-02]],

         ...,

         [[2.3365e-05, 2.3842e-06, 7.0801e-02,  ..., 9.6802e-02,
           5.9277e-01, 9.9060e-02]],

         [[8.6737e-04, 1.4553e-03, 1.7285e-01,  ..., 4.2206e-02,
           2.6611e-02, 1.0262e-02]],

         [[7.5865e-04, 4.0150e-04, 2.5879e-01,  ..., 1.5541e-02,
           3.7498e-03, 2.6379e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3757e-04, 7.0035e-05, 2.5879e-01,  ..., 4.1504e-03,
           3.1681e-03, 5.1928e-04]],

         [[2.1672e-04, 2.2757e-04, 1.8262e-01,  ..., 2.3994e-03,
           1.7900e-03, 1.5604e-04]],

         [[9.8228e-05, 5.4777e-05, 2.6904e-01,  ..., 2.7370e-04,
           1.9159e-03, 1.3475e-03]],

         ...,

         [[5.8556e-04, 5.4598e-04, 2.3401e-01,  ..., 2.8095e-03,
           3.2783e-04, 6.0892e-04]],

         [[9.3579e-06, 1.8001e-05, 2.5195e-01,  ..., 1.9226e-03,
           1.6642e-03, 3.5024e-04]],

         [[3.4153e-05, 1.3530e-04, 1.3428e-01,  ..., 2.3937e-03,
           1.0624e-03, 3.0065e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4746e-04, 7.5161e-05, 1.8604e-01,  ..., 2.6810e-02,
           1.8501e-03, 2.0790e-04]],

         [[3.5191e-04, 1.4722e-05, 2.5903e-01,  ..., 2.9182e-04,
           7.3910e-04, 2.5940e-04]],

         [[2.8062e-04, 2.4819e-04, 2.8076e-01,  ..., 7.9298e-04,
           1.0185e-03, 5.0926e-04]],

         ...,

         [[6.0141e-05, 6.3539e-05, 6.1493e-02,  ..., 3.6377e-01,
           5.4260e-02, 1.0101e-01]],

         [[7.6234e-05, 1.3196e-04, 2.4902e-01,  ..., 2.6817e-03,
           1.4095e-03, 6.9237e-04]],

         [[4.2343e-04, 1.6940e-04, 1.7480e-01,  ..., 1.1620e-02,
           5.8212e-03, 3.9101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3840e-04, 5.5432e-06, 4.4403e-03,  ..., 6.8018e-01,
           1.8445e-01, 3.6621e-02]],

         [[1.7142e-04, 1.9014e-04, 1.3892e-01,  ..., 5.7220e-03,
           3.4504e-03, 9.4147e-03]],

         [[5.9891e-04, 2.0981e-03, 1.7542e-01,  ..., 2.4300e-03,
           1.4877e-03, 1.0132e-02]],

         ...,

         [[1.0538e-03, 3.0756e-05, 7.7881e-02,  ..., 9.2529e-02,
           4.9896e-02, 3.6652e-02]],

         [[1.7700e-03, 2.0599e-04, 9.0942e-02,  ..., 1.8164e-01,
           2.7271e-01, 1.1230e-01]],

         [[3.2008e-05, 5.3048e-06, 4.7798e-03,  ..., 1.4355e-01,
           3.4985e-01, 2.0898e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2578e-03, 7.6771e-04, 2.2668e-01,  ..., 1.3771e-02,
           1.7303e-02, 6.7017e-02]],

         [[3.6240e-04, 5.7459e-04, 2.5415e-01,  ..., 4.3831e-03,
           6.9466e-03, 1.5022e-02]],

         [[1.1454e-03, 1.0452e-03, 2.7441e-01,  ..., 9.4604e-03,
           1.4313e-02, 2.5162e-02]],

         ...,

         [[5.0259e-04, 9.9778e-05, 1.7480e-01,  ..., 1.3098e-01,
           1.0944e-01, 1.2207e-01]],

         [[1.1301e-04, 5.9545e-05, 3.3356e-02,  ..., 2.7664e-02,
           1.2793e-01, 6.3916e-01]],

         [[2.3246e-06, 7.1526e-07, 1.0406e-02,  ..., 3.7256e-01,
           3.7256e-01, 7.6843e-02]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.3544e-05, 1.7285e-06, 2.2340e-04,  ..., 1.2317e-01,
           6.4551e-01, 1.3965e-01]],

         [[1.8775e-05, 6.4969e-06, 2.9087e-05,  ..., 2.7585e-04,
           2.7847e-03, 9.9170e-01]],

         [[1.3981e-03, 1.7948e-03, 2.1866e-02,  ..., 8.0490e-03,
           1.7029e-02, 1.7029e-02]],

         ...,

         [[1.5631e-03, 1.0414e-03, 4.5490e-04,  ..., 5.6854e-02,
           5.1788e-02, 9.2285e-02]],

         [[5.6877e-03, 2.8610e-03, 1.1673e-02,  ..., 1.3229e-02,
           9.6741e-03, 4.9133e-02]],

         [[4.6883e-03, 2.6703e-03, 1.8179e-04,  ..., 7.1045e-02,
           4.4464e-02, 2.4551e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6260e-04, 1.2989e-03, 2.9736e-03,  ..., 2.1301e-02,
           1.7017e-01, 3.0322e-01]],

         [[1.7881e-07, 0.0000e+00, 0.0000e+00,  ..., 7.8125e-02,
           8.3984e-01, 6.6833e-02]],

         [[3.6392e-03, 3.8128e-03, 9.5901e-03,  ..., 5.1308e-03,
           8.7280e-03, 7.9498e-03]],

         ...,

         [[3.0689e-03, 7.0801e-03, 1.4534e-02,  ..., 3.4851e-02,
           5.1117e-02, 4.6906e-02]],

         [[3.6850e-03, 9.7427e-03, 2.7985e-02,  ..., 4.9171e-03,
           6.7253e-03, 1.4801e-02]],

         [[3.0079e-03, 7.4158e-03, 1.6205e-02,  ..., 3.8879e-02,
           6.9275e-02, 5.3131e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2392e-03, 3.7265e-04, 3.6049e-03,  ..., 1.4404e-01,
           3.6499e-01, 6.2927e-02]],

         [[1.9932e-03, 6.8092e-04, 3.6678e-03,  ..., 6.0005e-03,
           9.0790e-03, 4.4937e-03]],

         [[2.5673e-03, 4.2915e-04, 3.8171e-04,  ..., 1.3039e-02,
           1.3992e-02, 1.5854e-02]],

         ...,

         [[1.2445e-04, 1.7285e-06, 1.7345e-05,  ..., 4.1931e-02,
           4.3018e-01, 4.2358e-01]],

         [[1.1997e-03, 2.1255e-04, 8.3733e-04,  ..., 1.3977e-01,
           1.2238e-01, 3.1677e-02]],

         [[2.2240e-03, 4.0817e-04, 1.3075e-03,  ..., 3.2440e-02,
           8.2959e-01, 4.9835e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4326e-01, 3.4571e-06, 1.0133e-06,  ..., 7.4234e-03,
           1.9974e-02, 5.5847e-03]],

         [[7.0801e-01, 1.2636e-05, 6.7353e-06,  ..., 2.1698e-02,
           1.1487e-01, 1.3269e-01]],

         [[9.4629e-01, 7.0333e-06, 9.5367e-07,  ..., 6.6185e-03,
           5.1727e-03, 1.8600e-02]],

         ...,

         [[7.3096e-01, 3.3915e-05, 4.1389e-04,  ..., 5.0211e-04,
           2.4719e-03, 2.6413e-02]],

         [[1.8140e-01, 1.4067e-05, 4.6968e-05,  ..., 1.7859e-01,
           7.5012e-02, 6.1951e-02]],

         [[7.0557e-01, 3.1948e-05, 1.8108e-04,  ..., 1.6403e-02,
           3.6804e-02, 3.8422e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7490e-01, 8.1658e-06, 2.2054e-06,  ..., 4.2953e-03,
           1.9169e-03, 3.3264e-02]],

         [[9.2822e-01, 0.0000e+00, 0.0000e+00,  ..., 7.0992e-03,
           5.4321e-02, 6.9771e-03]],

         [[8.7207e-01, 2.9492e-04, 2.5272e-05,  ..., 1.9836e-03,
           2.0508e-02, 4.7028e-02]],

         ...,

         [[4.7974e-01, 7.6175e-05, 1.0842e-04,  ..., 4.0222e-02,
           9.5581e-02, 2.7924e-02]],

         [[2.6001e-01, 1.0920e-03, 6.8521e-04,  ..., 1.8988e-03,
           4.1313e-03, 2.2354e-03]],

         [[7.5488e-01, 8.4996e-05, 1.4448e-04,  ..., 8.7219e-02,
           9.0149e-02, 1.6312e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9614e-01, 2.7370e-04, 6.7353e-05,  ..., 5.4779e-02,
           5.1331e-02, 1.9684e-02]],

         [[7.0557e-01, 1.3947e-05, 2.4438e-06,  ..., 2.3376e-02,
           9.4299e-02, 4.6196e-03]],

         [[4.4312e-01, 1.2934e-05, 1.1265e-05,  ..., 3.1494e-02,
           4.3701e-01, 5.1727e-03]],

         ...,

         [[9.0332e-01, 6.7353e-06, 6.5565e-07,  ..., 1.6689e-06,
           5.2154e-05, 9.4910e-02]],

         [[9.6436e-01, 1.8656e-05, 1.3113e-06,  ..., 0.0000e+00,
           1.3113e-06, 3.2288e-02]],

         [[9.7314e-01, 2.9802e-06, 0.0000e+00,  ..., 0.0000e+00,
           5.9605e-08, 2.6596e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5518e-01, 1.0878e-04, 8.8215e-06,  ..., 1.0645e-01,
           2.0239e-01, 1.0117e-02]],

         [[8.5156e-01, 5.4240e-06, 7.1526e-07,  ..., 1.6413e-03,
           2.9945e-03, 4.1580e-03]],

         [[8.6963e-01, 2.2471e-05, 3.6955e-06,  ..., 4.0588e-03,
           2.2568e-02, 1.4168e-02]],

         ...,

         [[7.8564e-01, 3.9458e-05, 2.6822e-06,  ..., 1.2032e-02,
           2.4734e-02, 2.8336e-02]],

         [[3.2227e-01, 2.2829e-05, 1.9073e-06,  ..., 2.0398e-01,
           2.5293e-01, 2.2964e-02]],

         [[8.7842e-01, 8.9407e-06, 7.6294e-06,  ..., 5.5237e-02,
           9.8038e-03, 7.0648e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6929e-01, 6.5565e-07, 7.1526e-07,  ..., 3.6230e-01,
           1.5625e-01, 4.5471e-02]],

         [[8.2178e-01, 2.1160e-05, 4.0531e-06,  ..., 8.5907e-03,
           1.3390e-02, 6.6795e-03]],

         [[1.6040e-01, 8.9407e-07, 2.9802e-07,  ..., 2.6660e-01,
           2.3291e-01, 4.4861e-03]],

         ...,

         [[3.1909e-01, 6.0618e-05, 3.7491e-05,  ..., 4.1077e-02,
           2.4756e-01, 9.5215e-03]],

         [[8.5254e-01, 1.7262e-04, 2.7955e-05,  ..., 7.6485e-03,
           8.6365e-03, 8.0156e-04]],

         [[4.4385e-01, 9.9540e-06, 1.0908e-05,  ..., 7.2083e-02,
           5.5115e-02, 1.0864e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1191e-01, 7.5769e-04, 7.5483e-04,  ..., 1.3771e-03,
           5.3024e-03, 8.4877e-04]],

         [[9.2480e-01, 5.1260e-06, 5.3883e-05,  ..., 1.3466e-03,
           3.4237e-03, 1.6953e-02]],

         [[6.3330e-01, 7.2718e-06, 1.0610e-05,  ..., 3.6133e-02,
           3.3661e-02, 3.3813e-02]],

         ...,

         [[6.9971e-01, 2.8491e-05, 6.4194e-05,  ..., 9.9754e-04,
           5.5733e-03, 2.0361e-01]],

         [[3.6963e-01, 1.3530e-05, 7.5102e-06,  ..., 9.9304e-02,
           3.1464e-02, 5.7335e-03]],

         [[8.8916e-01, 1.2338e-05, 1.2159e-05,  ..., 1.9012e-02,
           6.8420e-02, 1.8854e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9565e-01, 2.7633e-04, 6.5756e-04,  ..., 5.9891e-03,
           5.2032e-03, 5.7030e-03]],

         [[8.1592e-01, 4.1723e-06, 5.6028e-06,  ..., 2.7878e-02,
           9.4299e-02, 3.9940e-03]],

         [[6.9629e-01, 6.6161e-06, 1.2279e-05,  ..., 6.3904e-02,
           6.4026e-02, 1.0178e-02]],

         ...,

         [[5.3174e-01, 6.8247e-05, 2.1553e-04,  ..., 1.2207e-04,
           5.0211e-04, 7.6103e-03]],

         [[6.2012e-01, 1.8120e-05, 7.5102e-06,  ..., 1.2500e-01,
           1.6748e-01, 1.3382e-02]],

         [[9.5020e-01, 2.9206e-06, 4.4703e-06,  ..., 1.3229e-02,
           4.6883e-03, 5.2490e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4307e-01, 3.2246e-05, 9.0003e-05,  ..., 7.5960e-04,
           1.2627e-03, 3.7292e-02]],

         [[6.2744e-01, 1.6689e-05, 6.6161e-06,  ..., 1.7502e-02,
           6.1859e-02, 2.9938e-02]],

         [[6.9824e-01, 2.9802e-07, 8.9407e-07,  ..., 6.4774e-03,
           5.2452e-03, 5.0049e-02]],

         ...,

         [[8.3057e-01, 9.4950e-05, 5.2214e-05,  ..., 5.9395e-03,
           1.1711e-02, 3.5419e-03]],

         [[6.0498e-01, 2.7251e-04, 1.1230e-04,  ..., 1.4334e-03,
           1.3208e-03, 1.1823e-01]],

         [[7.0801e-01, 3.7646e-04, 4.2844e-04,  ..., 4.3221e-03,
           3.5820e-03, 5.5695e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6833e-01, 6.8545e-06, 3.4571e-06,  ..., 2.3770e-04,
           1.0353e-02, 8.0029e-01]],

         [[8.6377e-01, 9.2363e-04, 8.8453e-04,  ..., 3.7646e-04,
           1.5812e-03, 2.0134e-04]],

         [[3.8306e-01, 4.0722e-04, 5.4646e-04,  ..., 1.5442e-02,
           5.7159e-02, 4.6997e-02]],

         ...,

         [[7.6270e-01, 6.4671e-05, 1.0216e-04,  ..., 6.6490e-03,
           6.7024e-03, 1.1398e-02]],

         [[2.8101e-01, 1.6093e-05, 2.4736e-05,  ..., 2.1960e-01,
           1.2878e-01, 2.9495e-02]],

         [[3.4448e-01, 5.0545e-05, 1.4234e-04,  ..., 3.9482e-03,
           1.3496e-02, 5.0879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2275e-01, 6.5327e-05, 7.5459e-05,  ..., 2.5787e-02,
           7.6416e-02, 2.9907e-01]],

         [[6.2012e-01, 2.4498e-05, 3.1710e-05,  ..., 2.2629e-02,
           1.4648e-01, 6.0669e-02]],

         [[6.5430e-01, 7.1943e-05, 2.4974e-05,  ..., 5.7953e-02,
           3.9642e-02, 1.9073e-02]],

         ...,

         [[9.2957e-02, 7.0333e-06, 2.5034e-05,  ..., 3.4094e-05,
           6.6223e-03, 8.8232e-01]],

         [[7.8027e-01, 7.9989e-05, 5.1677e-05,  ..., 1.0406e-02,
           8.7814e-03, 1.5442e-02]],

         [[5.8838e-01, 2.2173e-05, 9.5367e-06,  ..., 3.5858e-02,
           2.4390e-01, 1.8784e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6147e-01, 1.6069e-04, 7.3850e-05,  ..., 6.4209e-02,
           4.6875e-01, 4.9103e-02]],

         [[6.9873e-01, 1.2577e-05, 1.6749e-05,  ..., 4.8218e-03,
           1.3802e-02, 6.0150e-02]],

         [[5.6250e-01, 3.0696e-05, 2.7180e-05,  ..., 2.2411e-03,
           1.6298e-03, 6.2408e-03]],

         ...,

         [[7.6025e-01, 0.0000e+00, 5.9605e-08,  ..., 7.8011e-03,
           1.9104e-01, 3.4210e-02]],

         [[4.7534e-01, 2.7394e-04, 3.0804e-04,  ..., 3.4485e-03,
           5.0087e-03, 5.7411e-03]],

         [[6.2286e-02, 5.9605e-08, 0.0000e+00,  ..., 3.7506e-02,
           8.5156e-01, 4.5227e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8760e-01, 6.5267e-05, 3.7491e-05,  ..., 1.4061e-02,
           4.0833e-02, 1.2230e-02]],

         [[1.4551e-01, 7.5912e-04, 3.5477e-04,  ..., 2.4080e-04,
           5.4789e-04, 1.5060e-02]],

         [[7.7490e-01, 2.9874e-04, 2.4045e-04,  ..., 6.9571e-04,
           3.2768e-03, 1.7502e-02]],

         ...,

         [[5.8740e-01, 5.7280e-05, 2.7835e-05,  ..., 2.6016e-02,
           5.4810e-02, 5.6000e-02]],

         [[4.0503e-01, 4.0710e-05, 3.0994e-05,  ..., 3.3356e-02,
           2.3376e-01, 1.6953e-02]],

         [[4.0283e-01, 3.8564e-05, 4.8578e-05,  ..., 8.3389e-03,
           1.1055e-02, 2.5742e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0577e-01, 1.1325e-06, 1.0729e-06,  ..., 7.9572e-05,
           7.7209e-03, 8.6523e-01]],

         [[5.3076e-01, 1.3769e-05, 2.4676e-05,  ..., 1.4351e-02,
           3.2654e-02, 4.0771e-02]],

         [[1.4294e-01, 2.5344e-04, 2.7514e-04,  ..., 2.2449e-03,
           7.8278e-03, 9.2468e-03]],

         ...,

         [[2.9712e-01, 4.2856e-05, 4.1366e-05,  ..., 4.2084e-02,
           1.0486e-01, 6.8115e-02]],

         [[5.0635e-01, 7.1526e-06, 9.4771e-06,  ..., 2.3224e-02,
           3.4454e-02, 6.8787e-02]],

         [[3.4204e-01, 3.9816e-05, 2.2769e-05,  ..., 1.3342e-01,
           9.3750e-02, 2.3239e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 6.4552e-05, 1.9252e-05,  ..., 2.0544e-01,
           1.8640e-01, 5.3192e-02]],

         [[2.1838e-01, 6.2943e-05, 3.1114e-05,  ..., 4.4373e-02,
           2.7557e-02, 3.1830e-02]],

         [[1.5637e-01, 3.7789e-05, 1.0431e-05,  ..., 2.6392e-01,
           1.4026e-01, 6.3660e-02]],

         ...,

         [[1.7480e-01, 1.6451e-05, 1.0729e-05,  ..., 2.2620e-01,
           1.4722e-01, 8.0627e-02]],

         [[5.7812e-01, 8.3447e-06, 9.5963e-06,  ..., 3.1494e-02,
           1.3623e-01, 9.7778e-02]],

         [[2.4561e-01, 1.0669e-05, 1.3173e-05,  ..., 2.2583e-02,
           8.2275e-02, 9.4666e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6907e-01, 7.6485e-04, 1.0109e-03,  ..., 4.2558e-04,
           1.3628e-03, 1.9817e-03]],

         [[9.3445e-02, 9.1851e-05, 9.0361e-05,  ..., 8.1177e-03,
           1.1650e-02, 1.5404e-02]],

         [[3.5815e-01, 1.7285e-06, 1.6093e-06,  ..., 1.1676e-01,
           2.3767e-01, 2.9755e-02]],

         ...,

         [[6.8994e-01, 2.6762e-05, 3.4750e-05,  ..., 8.5526e-03,
           2.3148e-02, 6.3904e-02]],

         [[9.8193e-01, 1.7881e-05, 4.8935e-05,  ..., 1.4305e-06,
           1.1027e-05, 6.9189e-04]],

         [[6.4111e-01, 6.3598e-05, 1.0079e-04,  ..., 1.2465e-03,
           1.5879e-03, 6.8817e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2988e-01, 7.8917e-05, 6.1929e-05,  ..., 4.8709e-04,
           2.1172e-03, 1.2665e-02]],

         [[1.3293e-01, 4.5943e-04, 2.5558e-04,  ..., 4.7684e-03,
           2.0386e-02, 2.4979e-02]],

         [[1.1755e-01, 2.5630e-06, 9.5367e-07,  ..., 1.6937e-02,
           1.9501e-02, 6.3086e-01]],

         ...,

         [[3.2544e-01, 6.7949e-06, 4.7088e-06,  ..., 2.6825e-02,
           3.4424e-02, 1.3023e-02]],

         [[2.1436e-01, 1.6689e-06, 7.1526e-07,  ..., 2.2290e-01,
           2.0776e-01, 2.1561e-02]],

         [[1.2964e-01, 1.0133e-06, 5.9605e-07,  ..., 1.3586e-01,
           1.7871e-01, 1.3696e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3730e-01, 1.6928e-05, 6.1393e-06,  ..., 1.0205e-01,
           1.6052e-01, 7.4646e-02]],

         [[1.4966e-01, 1.0550e-05, 1.0848e-05,  ..., 7.1838e-02,
           3.5553e-02, 2.4048e-02]],

         [[1.3171e-01, 4.0531e-06, 1.1921e-06,  ..., 5.9128e-04,
           3.0319e-02, 8.0664e-01]],

         ...,

         [[1.8408e-01, 2.8229e-04, 1.4758e-04,  ..., 4.7569e-03,
           6.0577e-03, 4.1412e-02]],

         [[3.6865e-01, 2.2745e-04, 1.4043e-04,  ..., 1.5125e-03,
           1.4944e-03, 8.6670e-03]],

         [[1.6113e-01, 2.4605e-04, 2.6822e-04,  ..., 5.0583e-03,
           3.0460e-03, 9.6054e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8665e-01, 1.4186e-05, 1.1861e-05,  ..., 9.0942e-02,
           5.1422e-02, 1.2140e-01]],

         [[9.4531e-01, 1.2934e-05, 7.1526e-07,  ..., 3.0935e-05,
           7.7724e-05, 7.5483e-04]],

         [[8.6963e-01, 2.8908e-05, 6.4969e-06,  ..., 2.1434e-04,
           1.0424e-03, 4.8599e-03]],

         ...,

         [[1.2433e-01, 8.1825e-04, 1.3180e-03,  ..., 1.9178e-03,
           2.3689e-03, 7.3509e-03]],

         [[3.7036e-01, 1.3411e-05, 1.9372e-05,  ..., 5.2521e-02,
           3.1342e-02, 1.3916e-02]],

         [[4.1748e-01, 6.7902e-04, 1.9493e-03,  ..., 1.6449e-02,
           3.1708e-02, 1.6968e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5498e-01, 9.6440e-05, 1.9944e-04,  ..., 1.5564e-03,
           7.3090e-03, 1.8814e-02]],

         [[2.8589e-01, 8.7976e-05, 1.7357e-04,  ..., 9.6283e-03,
           5.5420e-02, 4.5929e-02]],

         [[3.8477e-01, 5.7280e-05, 1.4508e-04,  ..., 1.4200e-03,
           4.4785e-03, 7.0457e-03]],

         ...,

         [[9.3628e-02, 1.2815e-04, 1.8942e-04,  ..., 2.6550e-03,
           8.4381e-03, 5.5023e-02]],

         [[1.1237e-01, 7.4983e-05, 8.1062e-05,  ..., 9.5215e-03,
           8.8043e-03, 4.2664e-02]],

         [[4.8071e-01, 4.6194e-05, 7.1526e-05,  ..., 1.8066e-02,
           5.4779e-02, 2.1118e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8787e-02, 1.2636e-05, 1.6689e-05,  ..., 2.1637e-02,
           1.7944e-02, 4.6906e-02]],

         [[2.8564e-01, 2.6560e-04, 4.0174e-04,  ..., 2.5902e-03,
           1.0239e-02, 1.9440e-02]],

         [[2.2644e-01, 5.3048e-05, 1.2434e-04,  ..., 2.1225e-02,
           4.7485e-02, 3.9453e-01]],

         ...,

         [[1.2439e-01, 1.7583e-05, 1.3649e-05,  ..., 2.1484e-01,
           2.2351e-01, 2.4689e-02]],

         [[5.2002e-01, 3.7134e-05, 9.1195e-05,  ..., 4.6768e-03,
           1.1047e-02, 5.1910e-02]],

         [[1.9385e-01, 5.7638e-05, 6.7651e-05,  ..., 2.7428e-03,
           4.4174e-03, 7.1182e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2219e-01, 4.5121e-05, 2.3246e-05,  ..., 6.1920e-02,
           4.1901e-02, 4.6387e-02]],

         [[1.4001e-01, 2.1398e-05, 4.1246e-05,  ..., 5.5809e-03,
           6.0120e-03, 1.1147e-02]],

         [[3.5254e-01, 4.1246e-05, 6.5565e-05,  ..., 5.6213e-02,
           5.8441e-02, 4.1779e-02]],

         ...,

         [[8.9600e-02, 0.0000e+00, 0.0000e+00,  ..., 6.6589e-02,
           8.2373e-01, 1.5091e-02]],

         [[3.3740e-01, 2.0266e-06, 1.5974e-05,  ..., 2.3682e-02,
           1.3416e-01, 1.0449e-01]],

         [[5.1855e-01, 6.5565e-07, 9.5367e-07,  ..., 7.3547e-02,
           3.1445e-01, 7.8918e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6387e-02, 1.9073e-06, 2.3782e-05,  ..., 1.3184e-02,
           2.4826e-02, 1.1127e-01]],

         [[4.9438e-02, 3.0398e-05, 4.2498e-05,  ..., 7.7724e-04,
           1.3832e-02, 8.6914e-01]],

         [[2.2595e-01, 1.0014e-04, 4.6992e-04,  ..., 2.7580e-03,
           1.5747e-02, 2.6169e-02]],

         ...,

         [[1.6248e-01, 1.0097e-04, 2.9778e-04,  ..., 4.6463e-03,
           9.1324e-03, 8.3771e-03]],

         [[2.5293e-01, 2.9802e-06, 3.0994e-06,  ..., 1.1267e-01,
           9.4849e-02, 2.8931e-02]],

         [[1.3086e-01, 1.1325e-06, 1.3709e-06,  ..., 1.3293e-01,
           6.5918e-01, 3.3600e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5283e-01, 2.2793e-04, 1.1950e-03,  ..., 1.5839e-02,
           4.7089e-02, 1.4534e-02]],

         [[9.8047e-01, 2.8789e-05, 9.7990e-05,  ..., 3.1376e-04,
           1.1082e-03, 6.2895e-04]],

         [[7.6123e-01, 2.2590e-05, 3.6240e-05,  ..., 2.6657e-02,
           1.0962e-01, 2.7939e-02]],

         ...,

         [[1.6833e-01, 2.2650e-06, 3.7551e-06,  ..., 3.5083e-01,
           3.7329e-01, 3.3142e-02]],

         [[4.9854e-01, 1.0848e-05, 6.0797e-06,  ..., 5.3673e-03,
           4.4861e-03, 2.7618e-03]],

         [[1.1389e-01, 6.6161e-06, 1.7226e-05,  ..., 2.2736e-03,
           3.9368e-02, 7.1973e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3650e-02, 1.1325e-06, 2.7418e-06,  ..., 5.8929e-02,
           2.0410e-01, 1.2976e-01]],

         [[1.4429e-01, 2.8610e-06, 1.4305e-06,  ..., 2.4927e-01,
           3.2007e-01, 5.9204e-02]],

         [[1.1145e-01, 5.6744e-05, 4.1544e-05,  ..., 4.4312e-02,
           6.0333e-02, 1.8768e-02]],

         ...,

         [[1.3586e-01, 5.4240e-06, 2.6464e-05,  ..., 1.6138e-01,
           1.6516e-01, 8.6365e-02]],

         [[5.7861e-01, 8.1658e-06, 6.5267e-05,  ..., 3.6133e-02,
           6.5430e-02, 8.4656e-02]],

         [[1.2158e-01, 1.4901e-06, 1.2159e-05,  ..., 1.1249e-01,
           6.3232e-01, 7.3181e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8147e-02, 2.9206e-06, 2.8610e-06,  ..., 1.5167e-02,
           1.0262e-02, 7.3357e-03]],

         [[3.0396e-01, 1.9312e-05, 1.1563e-05,  ..., 1.6909e-03,
           2.6493e-03, 2.3117e-03]],

         [[1.7639e-01, 2.0623e-05, 3.1829e-05,  ..., 7.9468e-02,
           6.1432e-02, 2.5604e-02]],

         ...,

         [[7.3975e-02, 3.9339e-05, 4.3297e-04,  ..., 5.9776e-03,
           3.9581e-02, 2.5955e-02]],

         [[2.3560e-01, 6.7055e-05, 3.8290e-04,  ..., 2.3460e-03,
           1.7059e-02, 1.8173e-02]],

         [[3.8672e-01, 3.1531e-05, 6.9976e-05,  ..., 4.4769e-02,
           4.6906e-02, 5.5695e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3770e-01, 2.3842e-07, 2.9802e-07,  ..., 8.9600e-02,
           7.3584e-01, 1.1528e-02]],

         [[1.0358e-01, 3.0398e-06, 6.7353e-06,  ..., 2.6343e-01,
           4.3115e-01, 3.2867e-02]],

         [[5.1074e-01, 0.0000e+00, 3.5763e-07,  ..., 4.8859e-02,
           2.8882e-01, 1.2573e-01]],

         ...,

         [[6.2891e-01, 3.8624e-05, 1.2374e-04,  ..., 5.7678e-03,
           5.6114e-03, 6.8779e-03]],

         [[3.6865e-02, 2.5630e-06, 4.4107e-06,  ..., 1.4807e-01,
           1.1090e-01, 6.2195e-02]],

         [[4.9286e-02, 5.8055e-05, 1.4091e-04,  ..., 8.7219e-02,
           2.6440e-01, 2.5830e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5840e-01, 6.4671e-05, 2.0957e-04,  ..., 5.8044e-02,
           1.0107e-01, 1.3107e-02]],

         [[6.9824e-01, 3.9506e-04, 7.9870e-04,  ..., 8.3876e-04,
           5.8556e-04, 2.2526e-03]],

         [[1.4526e-01, 1.0914e-04, 1.6940e-04,  ..., 7.3738e-03,
           2.9968e-02, 1.5190e-02]],

         ...,

         [[6.5857e-02, 3.7611e-05, 6.8605e-05,  ..., 1.3940e-01,
           8.5876e-02, 8.3084e-03]],

         [[1.6431e-01, 9.2328e-05, 2.4700e-04,  ..., 8.2642e-02,
           1.3843e-01, 4.2627e-01]],

         [[6.2939e-01, 5.3823e-05, 2.1958e-04,  ..., 8.0017e-02,
           4.1504e-02, 1.6769e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5576e-01, 1.8609e-04, 3.3331e-04,  ..., 1.4343e-02,
           1.9073e-02, 1.8814e-02]],

         [[2.8976e-02, 1.8954e-05, 2.3961e-05,  ..., 4.1199e-02,
           2.6184e-02, 4.0558e-02]],

         [[1.4258e-01, 1.9813e-04, 4.8351e-04,  ..., 7.7820e-02,
           7.7271e-02, 1.8997e-02]],

         ...,

         [[3.4595e-01, 8.9526e-05, 2.2399e-04,  ..., 1.8921e-02,
           2.7100e-02, 3.4943e-02]],

         [[6.0205e-01, 1.7405e-05, 4.0436e-04,  ..., 5.5786e-02,
           1.6077e-01, 1.4076e-02]],

         [[2.8174e-01, 6.7651e-05, 8.6308e-05,  ..., 9.6207e-03,
           1.3863e-02, 2.4094e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7280e-05, 2.6345e-05, 1.3477e-01,  ..., 4.4022e-03,
           4.6921e-04, 2.4353e-02]],

         [[9.2983e-06, 1.7047e-05, 8.8806e-02,  ..., 7.4310e-03,
           1.7242e-03, 1.0315e-02]],

         [[2.3580e-04, 1.0014e-04, 5.0629e-02,  ..., 1.0553e-01,
           7.3059e-02, 1.6251e-02]],

         ...,

         [[8.7619e-05, 2.4319e-05, 1.1456e-01,  ..., 4.4159e-02,
           7.5745e-02, 1.1279e-01]],

         [[1.9872e-04, 3.5501e-04, 2.5439e-01,  ..., 1.6510e-02,
           1.7441e-02, 1.2962e-02]],

         [[7.7784e-05, 4.8637e-05, 2.0618e-01,  ..., 3.9649e-04,
           1.1425e-03, 8.9874e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0340e-04, 2.7609e-04, 2.7148e-01,  ..., 3.4237e-04,
           5.2404e-04, 2.1423e-02]],

         [[1.1361e-04, 8.3089e-05, 1.7224e-01,  ..., 7.2062e-05,
           8.9347e-05, 1.4145e-02]],

         [[7.1824e-05, 1.9670e-05, 2.8711e-01,  ..., 1.3900e-04,
           1.3294e-03, 2.4887e-02]],

         ...,

         [[6.6698e-05, 4.3690e-05, 2.8735e-01,  ..., 2.0289e-04,
           3.4189e-04, 9.0561e-03]],

         [[1.1384e-05, 1.2934e-05, 2.2852e-01,  ..., 6.1572e-05,
           8.5652e-05, 1.1024e-02]],

         [[1.7107e-05, 4.8101e-05, 2.2351e-01,  ..., 2.8515e-04,
           2.8729e-04, 2.3926e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9353e-05, 3.3975e-05, 1.6077e-01,  ..., 9.5367e-05,
           4.4942e-05, 8.0643e-03]],

         [[1.6010e-04, 6.3777e-06, 2.3999e-01,  ..., 6.2108e-05,
           1.3566e-04, 1.2138e-02]],

         [[9.8586e-05, 8.5235e-05, 2.6294e-01,  ..., 4.3452e-05,
           2.3508e-04, 1.3138e-02]],

         ...,

         [[9.7513e-05, 2.5094e-05, 4.4403e-02,  ..., 7.2212e-03,
           1.8701e-01, 1.0841e-02]],

         [[3.6180e-05, 6.3896e-05, 2.4854e-01,  ..., 2.6751e-04,
           4.5252e-04, 8.5373e-03]],

         [[1.7178e-04, 6.4611e-05, 1.7456e-01,  ..., 6.6376e-04,
           2.7351e-03, 1.1559e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5199e-05, 9.5367e-07, 3.8280e-03,  ..., 1.3184e-01,
           2.3819e-02, 1.0925e-01]],

         [[5.7042e-05, 4.5657e-05, 1.2732e-01,  ..., 8.0795e-03,
           1.1818e-02, 2.8305e-02]],

         [[2.0766e-04, 2.5654e-04, 3.4180e-01,  ..., 5.1832e-04,
           2.3289e-03, 1.2436e-03]],

         ...,

         [[1.2422e-04, 2.0862e-06, 2.4033e-02,  ..., 1.2619e-02,
           5.2681e-03, 1.6418e-02]],

         [[5.3120e-04, 7.0512e-05, 6.8909e-02,  ..., 1.3550e-01,
           9.6436e-02, 9.3140e-02]],

         [[1.1325e-06, 1.7881e-07, 5.1880e-04,  ..., 8.8989e-02,
           1.9699e-02, 2.3071e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1635e-04, 2.4092e-04, 9.8083e-02,  ..., 1.2354e-01,
           1.9360e-01, 8.2275e-02]],

         [[5.3024e-04, 2.4986e-04, 2.2412e-01,  ..., 9.5291e-03,
           7.6561e-03, 3.9398e-02]],

         [[1.2245e-03, 3.6645e-04, 2.7148e-01,  ..., 3.8795e-03,
           1.1101e-02, 1.6449e-02]],

         ...,

         [[6.8307e-05, 1.4901e-05, 2.4768e-01,  ..., 2.8820e-03,
           7.7133e-03, 2.7359e-02]],

         [[1.7774e-04, 1.9741e-04, 6.6528e-02,  ..., 1.8478e-02,
           1.6693e-02, 2.0825e-01]],

         [[7.9572e-05, 3.5703e-05, 5.9052e-02,  ..., 3.1372e-02,
           2.9694e-02, 1.7761e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.9255e-03, 4.5729e-04, 2.1820e-03,  ..., 5.6274e-02,
           6.2402e-01, 7.0007e-02]],

         [[2.2259e-03, 5.3368e-03, 1.8631e-02,  ..., 1.2802e-02,
           8.3496e-02, 7.3669e-02]],

         [[0.0000e+00, 4.7684e-07, 2.2650e-06,  ..., 2.2650e-06,
           9.0981e-04, 9.9805e-01]],

         ...,

         [[6.0201e-06, 7.3612e-05, 8.9693e-04,  ..., 2.0015e-04,
           6.6299e-03, 9.8389e-01]],

         [[3.1261e-03, 3.3264e-03, 4.8409e-03,  ..., 1.5869e-02,
           1.5869e-02, 1.4008e-02]],

         [[1.1749e-02, 4.0588e-03, 2.1517e-04,  ..., 7.6599e-02,
           1.0696e-02, 3.5065e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5153e-05, 1.9777e-04, 8.8596e-04,  ..., 3.3264e-02,
           9.9243e-02, 4.8853e-01]],

         [[7.5102e-06, 1.7881e-07, 0.0000e+00,  ..., 1.5735e-01,
           7.7441e-01, 5.1056e-02]],

         [[2.6932e-03, 3.9177e-03, 6.2599e-03,  ..., 1.2070e-02,
           1.9897e-02, 2.5543e-02]],

         ...,

         [[3.7849e-05, 1.4269e-04, 1.3840e-04,  ..., 1.1078e-02,
           9.3652e-01, 4.3121e-02]],

         [[1.6079e-03, 6.6109e-03, 2.8381e-02,  ..., 8.4229e-03,
           2.5238e-02, 3.2166e-02]],

         [[2.1756e-04, 1.5459e-03, 9.8324e-04,  ..., 5.2429e-02,
           3.9355e-01, 1.0675e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0350e-02, 3.7556e-03, 2.2202e-02,  ..., 7.8125e-02,
           5.8044e-02, 6.4270e-02]],

         [[1.5469e-03, 4.6086e-04, 2.5501e-03,  ..., 4.5471e-03,
           1.0262e-03, 1.0574e-02]],

         [[6.4812e-03, 1.4181e-03, 1.4629e-03,  ..., 1.9348e-02,
           2.0752e-02, 1.6037e-02]],

         ...,

         [[4.3373e-03, 1.7905e-04, 2.3532e-04,  ..., 7.8064e-02,
           6.8506e-01, 1.3806e-01]],

         [[3.4547e-04, 5.9545e-05, 7.4089e-05,  ..., 9.9548e-02,
           1.0910e-02, 4.5227e-02]],

         [[1.3954e-02, 3.0899e-03, 1.6312e-02,  ..., 2.9770e-02,
           8.8867e-02, 1.8237e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7783e-01, 2.2054e-06, 2.5034e-06,  ..., 4.0131e-02,
           6.9008e-03, 1.2932e-02]],

         [[5.3174e-01, 3.0696e-05, 3.2902e-05,  ..., 7.2693e-02,
           7.8003e-02, 8.6441e-03]],

         [[7.2900e-01, 2.3246e-05, 3.3975e-06,  ..., 2.0081e-02,
           2.6588e-03, 1.0315e-02]],

         ...,

         [[4.4897e-01, 7.9811e-05, 8.0633e-04,  ..., 2.6379e-03,
           7.1716e-03, 9.1248e-02]],

         [[5.3613e-01, 5.5408e-04, 1.1253e-03,  ..., 1.2222e-02,
           1.7990e-02, 2.5970e-02]],

         [[7.0020e-01, 4.9531e-05, 3.1161e-04,  ..., 3.2898e-02,
           2.1561e-02, 7.3090e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3008e-01, 5.1439e-05, 5.1260e-06,  ..., 6.9809e-03,
           1.8227e-04, 6.7993e-02]],

         [[5.5713e-01, 2.3842e-07, 0.0000e+00,  ..., 2.6871e-02,
           3.8208e-01, 3.0487e-02]],

         [[9.4873e-01, 2.8920e-04, 2.4021e-05,  ..., 9.0170e-04,
           1.3565e-02, 1.6909e-03]],

         ...,

         [[5.8643e-01, 2.0945e-04, 1.4591e-04,  ..., 4.2389e-02,
           5.7739e-02, 5.1056e-02]],

         [[2.6074e-01, 1.5774e-03, 9.8133e-04,  ..., 5.1880e-03,
           1.5116e-03, 5.6763e-03]],

         [[6.5381e-01, 1.0502e-04, 9.4712e-05,  ..., 5.4840e-02,
           8.1055e-02, 6.4392e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8127e-01, 1.0371e-04, 2.6107e-05,  ..., 7.3059e-02,
           3.1952e-02, 2.2873e-02]],

         [[5.1318e-01, 4.6670e-05, 4.9472e-06,  ..., 8.5693e-02,
           1.6394e-01, 4.2236e-02]],

         [[4.2432e-01, 1.5318e-05, 5.9009e-06,  ..., 2.8290e-02,
           2.6054e-03, 3.2776e-02]],

         ...,

         [[8.6572e-01, 1.1325e-05, 1.7881e-07,  ..., 1.4424e-05,
           1.3709e-06, 1.2976e-01]],

         [[9.4775e-01, 3.2365e-05, 7.1526e-07,  ..., 2.5034e-06,
           5.9605e-08, 2.3438e-02]],

         [[9.8486e-01, 6.8545e-06, 0.0000e+00,  ..., 6.5565e-07,
           0.0000e+00, 1.2627e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5264e-01, 3.9160e-05, 2.9802e-06,  ..., 9.8145e-02,
           1.5808e-01, 2.7145e-02]],

         [[1.5857e-01, 2.6822e-06, 8.9407e-07,  ..., 1.3618e-02,
           4.7779e-04, 7.0877e-03]],

         [[7.4756e-01, 6.4254e-05, 1.2696e-05,  ..., 1.2222e-02,
           1.9083e-03, 1.5068e-02]],

         ...,

         [[9.4531e-01, 2.7013e-04, 1.9431e-05,  ..., 1.5888e-03,
           3.0346e-03, 4.5280e-03]],

         [[5.2881e-01, 7.5281e-05, 2.9266e-05,  ..., 3.8910e-02,
           2.1866e-02, 2.6953e-01]],

         [[8.3936e-01, 5.4598e-05, 1.8358e-05,  ..., 9.3222e-04,
           2.5463e-03, 1.3562e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2393e-01, 1.0490e-05, 5.3644e-06,  ..., 4.6875e-02,
           3.5339e-02, 2.7145e-02]],

         [[4.7925e-01, 2.2650e-05, 8.1658e-06,  ..., 1.5915e-02,
           1.2711e-02, 1.0077e-01]],

         [[9.1748e-01, 9.6560e-06, 3.8147e-06,  ..., 3.3226e-03,
           2.4261e-03, 1.1642e-02]],

         ...,

         [[4.2212e-01, 2.8634e-04, 4.2558e-05,  ..., 1.4050e-01,
           2.2644e-02, 4.5959e-02]],

         [[7.0801e-01, 1.2457e-04, 3.3796e-05,  ..., 3.3875e-02,
           1.1368e-03, 3.9185e-02]],

         [[6.5723e-01, 7.8022e-05, 5.0962e-05,  ..., 1.2543e-02,
           1.0907e-01, 3.7445e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9521e-01, 1.3933e-03, 7.8773e-04,  ..., 3.4885e-03,
           6.9523e-04, 9.2459e-04]],

         [[8.8135e-01, 1.2875e-05, 6.6340e-05,  ..., 2.6817e-03,
           3.0041e-03, 1.3306e-02]],

         [[7.3584e-01, 2.6464e-05, 1.1742e-05,  ..., 2.8412e-02,
           1.0536e-02, 6.3019e-03]],

         ...,

         [[7.7051e-01, 3.6657e-05, 2.6166e-05,  ..., 9.8705e-04,
           2.4216e-02, 3.3508e-02]],

         [[8.3789e-01, 6.9678e-05, 4.9591e-05,  ..., 7.1907e-03,
           1.5533e-02, 9.6680e-02]],

         [[9.2188e-01, 1.8120e-05, 1.7405e-05,  ..., 1.0757e-02,
           4.7188e-03, 1.9348e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0737e-01, 1.8525e-04, 3.7551e-04,  ..., 2.7275e-03,
           4.2419e-03, 3.6049e-03]],

         [[7.8369e-01, 8.7023e-06, 8.5235e-06,  ..., 3.1342e-02,
           8.6487e-02, 3.8696e-02]],

         [[8.2959e-01, 7.2718e-06, 3.0160e-05,  ..., 3.4149e-02,
           5.2185e-03, 1.7487e-02]],

         ...,

         [[7.6074e-01, 1.0270e-04, 8.4519e-05,  ..., 3.9148e-04,
           6.9733e-03, 4.3416e-04]],

         [[9.0771e-01, 1.9670e-05, 1.0490e-05,  ..., 1.6296e-02,
           6.1264e-03, 8.0109e-03]],

         [[9.6729e-01, 6.4373e-06, 1.1563e-05,  ..., 1.1387e-03,
           2.1572e-03, 9.5749e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5254e-01, 2.5153e-05, 2.3663e-05,  ..., 7.4148e-04,
           1.0812e-04, 2.0325e-02]],

         [[7.5342e-01, 6.4313e-05, 2.7895e-05,  ..., 1.9272e-02,
           1.2195e-01, 7.5493e-03]],

         [[7.7734e-01, 2.3246e-06, 1.1325e-06,  ..., 6.0394e-02,
           6.2180e-03, 2.4155e-02]],

         ...,

         [[8.9648e-01, 3.8862e-05, 4.4346e-05,  ..., 8.6060e-03,
           1.3828e-03, 4.4212e-03]],

         [[7.1094e-01, 1.2040e-04, 6.5982e-05,  ..., 1.6365e-03,
           4.3121e-02, 3.4847e-03]],

         [[5.6152e-01, 6.4993e-04, 3.6812e-04,  ..., 4.4556e-03,
           8.2550e-03, 1.3252e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8530e-01, 1.3709e-06, 1.9073e-06,  ..., 1.7376e-03,
           5.5847e-03, 7.9883e-01]],

         [[5.6494e-01, 1.6899e-03, 1.1435e-03,  ..., 1.0307e-02,
           2.8744e-03, 1.3371e-03]],

         [[4.7192e-01, 7.4100e-04, 3.9172e-04,  ..., 4.3030e-02,
           4.7272e-02, 3.8147e-02]],

         ...,

         [[6.2793e-01, 9.4295e-05, 1.8179e-04,  ..., 1.1742e-02,
           1.5701e-02, 1.7939e-03]],

         [[2.9077e-01, 1.0830e-04, 1.0413e-04,  ..., 9.4681e-03,
           1.1162e-02, 1.3893e-02]],

         [[3.9746e-01, 2.4414e-04, 2.4331e-04,  ..., 2.0847e-03,
           4.4586e-02, 4.3213e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2461e-01, 1.3626e-04, 1.5557e-04,  ..., 6.4964e-03,
           5.9021e-02, 4.9878e-01]],

         [[6.7188e-01, 6.9380e-05, 1.1307e-04,  ..., 6.2988e-02,
           1.4587e-01, 2.2018e-02]],

         [[4.6338e-01, 1.5545e-04, 5.6088e-05,  ..., 2.9922e-02,
           2.9175e-02, 9.9030e-03]],

         ...,

         [[6.4758e-02, 1.6689e-06, 9.2983e-06,  ..., 8.1301e-04,
           5.5885e-03, 9.2236e-01]],

         [[7.9541e-01, 8.9705e-05, 8.2016e-05,  ..., 7.0114e-03,
           5.6152e-03, 3.7048e-02]],

         [[7.0215e-01, 1.1963e-04, 5.1618e-05,  ..., 5.8228e-02,
           9.1782e-03, 3.6411e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4604e-01, 7.6294e-04, 7.0810e-04,  ..., 1.2115e-01,
           6.8054e-02, 4.1473e-02]],

         [[8.8232e-01, 4.4167e-05, 1.4651e-04,  ..., 4.3182e-03,
           4.0466e-02, 2.5116e-02]],

         [[3.6255e-01, 1.0550e-04, 5.7101e-05,  ..., 1.5076e-02,
           1.5594e-02, 7.9712e-02]],

         ...,

         [[4.7119e-01, 5.9605e-07, 5.9605e-08,  ..., 3.1128e-02,
           4.7119e-01, 2.2675e-02]],

         [[4.4214e-01, 6.1607e-04, 7.3290e-04,  ..., 1.3779e-02,
           1.0147e-02, 1.3199e-03]],

         [[7.9443e-01, 6.4969e-06, 7.1526e-07,  ..., 5.0751e-02,
           1.2421e-01, 1.7471e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7480e-01, 7.9155e-05, 4.4405e-05,  ..., 3.3875e-02,
           1.3290e-02, 4.5624e-03]],

         [[1.4050e-01, 8.6355e-04, 4.6730e-04,  ..., 5.1165e-04,
           2.1088e-02, 4.7150e-03]],

         [[5.0732e-01, 6.3562e-04, 7.0810e-04,  ..., 6.1378e-03,
           2.9541e-02, 3.4027e-02]],

         ...,

         [[3.5547e-01, 3.1710e-05, 2.0862e-05,  ..., 9.0515e-02,
           4.3774e-01, 4.6600e-02]],

         [[6.7041e-01, 6.8247e-05, 5.4419e-05,  ..., 6.3843e-02,
           8.4229e-02, 7.1594e-02]],

         [[5.9326e-01, 1.1081e-04, 2.0456e-04,  ..., 1.1322e-02,
           5.9570e-02, 8.1100e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0568e-02, 4.7684e-07, 4.1723e-07,  ..., 3.2425e-04,
           8.0032e-03, 9.3604e-01]],

         [[3.9795e-01, 3.4273e-05, 6.5684e-05,  ..., 6.6650e-02,
           2.8442e-01, 8.6060e-02]],

         [[1.8762e-01, 3.9458e-04, 6.8617e-04,  ..., 1.0895e-02,
           2.3987e-02, 6.2103e-03]],

         ...,

         [[3.5205e-01, 7.4506e-05, 5.6207e-05,  ..., 5.3162e-02,
           1.4905e-01, 3.3360e-03]],

         [[9.0698e-02, 2.1458e-06, 1.0729e-06,  ..., 1.4722e-01,
           1.1969e-01, 1.3504e-02]],

         [[3.1567e-01, 5.2154e-05, 6.4850e-05,  ..., 4.5837e-02,
           4.8859e-02, 6.7482e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6074e-01, 1.0031e-04, 3.2485e-05,  ..., 1.8201e-01,
           1.9458e-01, 2.5818e-02]],

         [[1.4392e-01, 6.9439e-05, 2.6584e-05,  ..., 7.2937e-02,
           8.8684e-02, 1.2283e-02]],

         [[2.2046e-01, 1.9073e-05, 6.3777e-06,  ..., 4.2908e-02,
           1.1987e-01, 6.8298e-02]],

         ...,

         [[3.2422e-01, 1.2398e-05, 8.7023e-06,  ..., 9.8145e-02,
           1.7627e-01, 1.7120e-02]],

         [[6.0059e-01, 9.8944e-06, 1.1981e-05,  ..., 7.5806e-02,
           1.2067e-01, 4.6875e-02]],

         [[3.0933e-01, 2.1517e-05, 3.2425e-05,  ..., 8.9294e-02,
           1.4441e-01, 1.4015e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7271e-01, 9.1600e-04, 1.6794e-03,  ..., 1.3990e-03,
           2.1839e-03, 1.9684e-03]],

         [[9.1858e-02, 9.8526e-05, 1.4019e-04,  ..., 2.0462e-02,
           4.2725e-02, 1.8158e-02]],

         [[4.3091e-01, 7.2122e-06, 1.2696e-05,  ..., 5.1666e-02,
           3.0420e-01, 1.9836e-02]],

         ...,

         [[7.4219e-01, 1.2434e-04, 5.2154e-05,  ..., 2.4048e-02,
           1.5106e-02, 1.3952e-03]],

         [[9.9365e-01, 1.3530e-05, 5.8591e-05,  ..., 4.7684e-07,
           3.0339e-05, 1.9684e-03]],

         [[7.5928e-01, 7.9632e-05, 1.7059e-04,  ..., 6.7024e-03,
           1.3664e-02, 1.2789e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7188e-01, 1.0359e-04, 1.3411e-04,  ..., 1.0323e-02,
           4.3457e-02, 1.7288e-02]],

         [[2.0825e-01, 5.4979e-04, 3.2425e-04,  ..., 1.7502e-02,
           3.2684e-02, 9.5901e-03]],

         [[1.8591e-01, 2.0146e-05, 7.3910e-06,  ..., 1.2550e-02,
           9.0576e-02, 4.1553e-01]],

         ...,

         [[2.2205e-01, 1.7583e-05, 1.8179e-05,  ..., 1.0822e-01,
           2.9663e-01, 2.6321e-02]],

         [[4.1162e-01, 2.8670e-05, 1.3769e-05,  ..., 8.5571e-02,
           1.5027e-01, 5.9753e-02]],

         [[2.0227e-01, 9.6560e-06, 3.7551e-06,  ..., 1.3367e-01,
           2.3645e-01, 1.3794e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.4421e-02, 6.8545e-06, 1.9073e-06,  ..., 4.1016e-01,
           2.0947e-01, 1.9180e-02]],

         [[2.0190e-01, 3.8624e-05, 2.4557e-05,  ..., 7.5439e-02,
           1.6748e-01, 2.2308e-02]],

         [[1.0797e-01, 6.9141e-06, 4.3511e-06,  ..., 4.1199e-03,
           2.3346e-02, 8.4912e-01]],

         ...,

         [[2.4841e-01, 2.4199e-04, 1.0997e-04,  ..., 2.3651e-02,
           2.6382e-02, 1.8997e-02]],

         [[4.7192e-01, 2.4056e-04, 1.3626e-04,  ..., 3.7262e-02,
           4.0924e-02, 3.0884e-02]],

         [[2.3181e-01, 2.1887e-04, 2.1136e-04,  ..., 3.4454e-02,
           4.0924e-02, 1.1101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4587e-01, 2.8908e-05, 1.6212e-05,  ..., 6.3721e-02,
           2.8784e-01, 1.2848e-02]],

         [[9.6533e-01, 3.4213e-04, 1.1635e-04,  ..., 5.6267e-05,
           4.3225e-04, 1.9038e-04]],

         [[8.8965e-01, 3.0208e-04, 1.2589e-04,  ..., 1.4353e-03,
           5.8556e-03, 3.0384e-03]],

         ...,

         [[1.2207e-01, 9.2506e-04, 1.7557e-03,  ..., 8.0566e-03,
           1.7593e-02, 3.5191e-03]],

         [[4.7705e-01, 2.6345e-05, 2.7180e-05,  ..., 4.4373e-02,
           2.7328e-02, 4.6043e-03]],

         [[3.4985e-01, 9.7513e-04, 3.2463e-03,  ..., 6.1737e-02,
           2.1851e-02, 2.4185e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9951e-01, 1.9431e-04, 2.8062e-04,  ..., 1.3947e-02,
           1.5320e-02, 1.6174e-02]],

         [[3.4619e-01, 1.8132e-04, 3.0351e-04,  ..., 1.3342e-01,
           6.4026e-02, 2.2125e-02]],

         [[4.5557e-01, 8.9109e-05, 1.7583e-04,  ..., 2.8900e-02,
           3.4058e-02, 1.0223e-02]],

         ...,

         [[4.3579e-02, 5.9664e-05, 8.1539e-05,  ..., 7.6904e-03,
           2.7695e-02, 1.7609e-02]],

         [[1.6357e-01, 1.0091e-04, 8.7678e-05,  ..., 3.0258e-02,
           4.0070e-02, 5.2277e-02]],

         [[4.6753e-01, 3.6359e-05, 6.1870e-05,  ..., 8.9233e-02,
           3.4393e-02, 5.8823e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5735e-02, 1.4961e-05, 1.5497e-05,  ..., 3.6011e-02,
           1.0181e-01, 1.9730e-02]],

         [[3.4131e-01, 1.5712e-04, 2.5702e-04,  ..., 3.8574e-02,
           4.7638e-02, 1.5717e-02]],

         [[4.4067e-01, 9.1076e-05, 2.7180e-04,  ..., 2.3178e-02,
           2.3727e-02, 3.5693e-01]],

         ...,

         [[3.4912e-01, 6.5923e-05, 6.2585e-05,  ..., 1.5613e-01,
           8.8257e-02, 8.3389e-03]],

         [[7.7344e-01, 4.9531e-05, 1.8120e-04,  ..., 1.6928e-03,
           2.6062e-02, 3.2959e-02]],

         [[3.1494e-01, 6.2585e-05, 8.2612e-05,  ..., 3.0945e-02,
           9.1476e-03, 2.3117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2756e-01, 9.2804e-05, 4.6492e-05,  ..., 1.1090e-01,
           6.4697e-02, 2.8030e-02]],

         [[5.0323e-02, 1.2517e-05, 1.2159e-05,  ..., 1.3443e-02,
           2.6733e-02, 5.2223e-03]],

         [[4.2603e-01, 1.5545e-04, 1.5032e-04,  ..., 4.7424e-02,
           5.3314e-02, 1.9623e-02]],

         ...,

         [[3.6401e-01, 1.1921e-07, 5.9605e-08,  ..., 3.6401e-01,
           2.2791e-01, 7.9803e-03]],

         [[3.3325e-01, 2.2650e-06, 1.7345e-05,  ..., 9.4727e-02,
           2.2900e-01, 3.7109e-02]],

         [[6.5625e-01, 8.9407e-07, 1.2517e-06,  ..., 2.3529e-02,
           2.4524e-01, 6.0547e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1422e-03, 1.2517e-06, 1.1861e-05,  ..., 1.6220e-02,
           1.4124e-01, 9.1019e-03]],

         [[3.8574e-02, 3.5882e-05, 7.6234e-05,  ..., 5.1260e-04,
           8.2092e-03, 9.1260e-01]],

         [[2.5293e-01, 1.4567e-04, 6.2180e-04,  ..., 1.8387e-02,
           1.6541e-02, 1.1024e-02]],

         ...,

         [[1.1194e-01, 1.4722e-04, 5.1022e-04,  ..., 8.1100e-03,
           9.2621e-03, 5.1346e-03]],

         [[3.4668e-01, 1.2934e-05, 1.8239e-05,  ..., 3.0045e-02,
           6.2134e-02, 9.7778e-02]],

         [[3.4546e-01, 2.9802e-06, 2.7418e-06,  ..., 1.1755e-01,
           3.8379e-01, 7.5623e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7578e-01, 2.5249e-04, 1.2379e-03,  ..., 6.5918e-02,
           1.5411e-02, 9.2392e-03]],

         [[9.8145e-01, 5.1796e-05, 2.9278e-04,  ..., 9.1267e-04,
           3.7885e-04, 3.9554e-04]],

         [[9.2676e-01, 1.0413e-04, 2.0945e-04,  ..., 8.9417e-03,
           5.1346e-03, 1.5850e-03]],

         ...,

         [[5.6738e-01, 7.3910e-06, 1.4246e-05,  ..., 1.0992e-01,
           6.6711e-02, 1.8524e-02]],

         [[7.1875e-01, 2.1398e-05, 1.7583e-05,  ..., 6.5155e-03,
           3.0308e-03, 2.8458e-03]],

         [[1.5869e-01, 1.2815e-05, 4.5180e-05,  ..., 8.8120e-03,
           2.3590e-02, 7.1094e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7688e-02, 3.3975e-06, 5.7220e-06,  ..., 1.0315e-01,
           2.5146e-01, 1.2445e-01]],

         [[4.0747e-01, 9.7156e-06, 6.9141e-06,  ..., 1.5283e-01,
           8.9844e-02, 4.2938e-02]],

         [[1.4404e-01, 1.1522e-04, 1.8823e-04,  ..., 2.6749e-02,
           2.5116e-02, 8.9188e-03]],

         ...,

         [[4.2358e-01, 1.3173e-05, 5.9128e-05,  ..., 1.3965e-01,
           1.4185e-01, 4.2938e-02]],

         [[6.0156e-01, 5.6028e-06, 3.3855e-05,  ..., 4.2572e-02,
           7.1289e-02, 2.8809e-02]],

         [[6.1084e-01, 8.5235e-06, 3.3319e-05,  ..., 6.7505e-02,
           1.2415e-01, 2.2079e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3193e-02, 3.8743e-06, 3.0398e-06,  ..., 7.8354e-03,
           7.1869e-03, 3.2139e-03]],

         [[3.7305e-01, 7.6175e-05, 5.4479e-05,  ..., 2.5787e-02,
           1.0101e-02, 4.7874e-03]],

         [[2.7466e-01, 8.0824e-05, 1.3649e-04,  ..., 9.5703e-02,
           3.1067e-02, 1.0406e-02]],

         ...,

         [[4.8218e-02, 3.6180e-05, 4.0126e-04,  ..., 2.5604e-02,
           3.1372e-02, 2.1912e-02]],

         [[3.4375e-01, 6.8843e-05, 6.9523e-04,  ..., 2.0004e-02,
           1.6586e-02, 2.6093e-02]],

         [[7.8271e-01, 7.2718e-06, 1.4484e-05,  ..., 3.3844e-02,
           3.1799e-02, 2.9877e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4473e-01, 2.0266e-06, 6.9737e-06,  ..., 3.5431e-02,
           6.8512e-03, 1.7532e-02]],

         [[3.5156e-01, 5.9605e-06, 1.5438e-05,  ..., 1.6479e-01,
           1.5723e-01, 1.0394e-01]],

         [[2.2937e-01, 5.9605e-08, 2.3842e-07,  ..., 3.0807e-02,
           5.2490e-01, 2.0007e-01]],

         ...,

         [[6.7773e-01, 3.5584e-05, 1.4532e-04,  ..., 6.3896e-03,
           5.2338e-03, 5.3177e-03]],

         [[7.9834e-02, 2.9206e-06, 4.1127e-06,  ..., 6.2683e-02,
           2.5757e-01, 1.6479e-02]],

         [[1.2683e-01, 1.0085e-04, 2.0981e-04,  ..., 3.0664e-01,
           2.2034e-02, 2.7930e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2432e-01, 2.5988e-05, 8.1837e-05,  ..., 2.4561e-01,
           2.1210e-02, 1.6708e-02]],

         [[7.0654e-01, 3.2425e-04, 3.6120e-04,  ..., 1.2941e-03,
           1.7376e-03, 2.0752e-03]],

         [[4.5972e-01, 3.7730e-05, 9.6560e-05,  ..., 8.9172e-02,
           2.7405e-02, 1.0078e-02]],

         ...,

         [[4.0405e-01, 1.2255e-04, 2.8920e-04,  ..., 5.7312e-02,
           2.6443e-02, 2.6099e-01]],

         [[2.7026e-01, 1.4949e-04, 4.8256e-04,  ..., 4.0161e-02,
           2.5391e-01, 3.0151e-01]],

         [[8.3447e-01, 4.5717e-05, 2.3758e-04,  ..., 1.8723e-02,
           1.6785e-02, 1.7593e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1455e-01, 4.4703e-05, 1.9407e-04,  ..., 6.9427e-03,
           7.3738e-03, 9.4681e-03]],

         [[4.8401e-02, 4.3690e-05, 5.4061e-05,  ..., 7.1945e-03,
           4.3060e-02, 5.5481e-02]],

         [[1.8384e-01, 9.5785e-05, 3.5882e-04,  ..., 8.8928e-02,
           3.7628e-02, 1.0895e-01]],

         ...,

         [[7.1826e-01, 2.4498e-05, 1.3649e-04,  ..., 1.3519e-02,
           3.4119e-02, 2.8183e-02]],

         [[9.3066e-01, 2.4438e-06, 3.0828e-04,  ..., 3.8071e-03,
           4.0970e-03, 5.1355e-04]],

         [[6.5820e-01, 3.3915e-05, 7.9989e-05,  ..., 2.7710e-02,
           2.5879e-02, 3.2196e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0146e-05, 3.6001e-05, 1.9885e-01,  ..., 1.6251e-03,
           6.9214e-02, 1.5144e-03]],

         [[2.1875e-05, 3.5226e-05, 9.3445e-02,  ..., 1.3332e-03,
           2.5162e-02, 1.2770e-03]],

         [[3.3355e-04, 2.4748e-04, 1.7395e-01,  ..., 1.0474e-01,
           4.9652e-02, 1.2703e-02]],

         ...,

         [[1.6665e-04, 1.3280e-04, 2.4402e-01,  ..., 1.7258e-02,
           2.1484e-02, 1.0657e-01]],

         [[3.3760e-04, 2.9850e-04, 2.3633e-01,  ..., 2.7786e-02,
           1.8219e-02, 5.6458e-03]],

         [[3.5226e-05, 3.3081e-05, 2.9663e-01,  ..., 5.5199e-03,
           1.1597e-02, 1.4172e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2183e-04, 1.9932e-04, 2.8174e-01,  ..., 2.1858e-03,
           1.5900e-02, 2.4853e-03]],

         [[7.0512e-05, 7.6532e-05, 2.0447e-01,  ..., 3.3498e-04,
           1.1269e-02, 2.7037e-04]],

         [[2.2888e-05, 1.3828e-05, 2.9980e-01,  ..., 3.5591e-03,
           1.5762e-02, 8.8596e-04]],

         ...,

         [[2.6035e-04, 1.4770e-04, 1.6882e-01,  ..., 9.5034e-04,
           6.1951e-03, 4.4799e-04]],

         [[1.0848e-05, 1.8597e-05, 2.6440e-01,  ..., 5.2643e-04,
           1.6006e-02, 1.0431e-04]],

         [[1.5676e-05, 4.8280e-05, 2.3608e-01,  ..., 5.3930e-04,
           1.9913e-02, 4.3154e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3232e-05, 2.5451e-05, 2.2461e-01,  ..., 2.3055e-04,
           7.4921e-03, 5.5361e-04]],

         [[5.2273e-05, 3.7551e-06, 2.6270e-01,  ..., 6.1607e-04,
           1.1269e-02, 1.4663e-05]],

         [[3.6001e-05, 7.2420e-05, 2.9443e-01,  ..., 1.0376e-03,
           1.0727e-02, 9.8050e-05]],

         ...,

         [[2.9504e-05, 1.3947e-05, 7.1167e-02,  ..., 3.9404e-01,
           1.2131e-02, 7.0496e-03]],

         [[1.2338e-05, 4.2498e-05, 2.3376e-01,  ..., 1.8129e-03,
           8.0948e-03, 4.2558e-04]],

         [[8.1778e-05, 5.4479e-05, 2.1899e-01,  ..., 7.7019e-03,
           1.1971e-02, 1.5402e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5199e-05, 2.3246e-06, 1.7685e-02,  ..., 5.4901e-02,
           1.5393e-01, 1.7041e-01]],

         [[3.0935e-05, 4.5896e-05, 8.9844e-02,  ..., 1.6113e-02,
           2.2888e-02, 4.0497e-02]],

         [[1.1760e-04, 3.2139e-04, 3.0859e-01,  ..., 2.2812e-03,
           7.7820e-04, 8.3637e-04]],

         ...,

         [[3.8695e-04, 3.0756e-05, 8.3923e-02,  ..., 1.4334e-03,
           3.4256e-03, 1.5503e-02]],

         [[4.6587e-04, 5.2273e-05, 8.5388e-02,  ..., 3.5858e-02,
           6.2927e-02, 2.5293e-01]],

         [[2.9027e-05, 2.5630e-06, 2.2629e-02,  ..., 2.7618e-02,
           5.3650e-02, 2.4426e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8578e-03, 2.5201e-04, 1.6907e-01,  ..., 3.9307e-02,
           1.5327e-02, 5.3192e-02]],

         [[3.2973e-04, 2.6178e-04, 2.5098e-01,  ..., 9.1171e-03,
           1.5129e-02, 2.1744e-02]],

         [[4.1885e-03, 6.3562e-04, 2.5391e-01,  ..., 2.0561e-03,
           4.3373e-03, 4.2801e-03]],

         ...,

         [[2.7752e-04, 5.1379e-05, 2.5439e-01,  ..., 4.3106e-03,
           1.7715e-02, 7.6965e-02]],

         [[3.7241e-04, 4.4560e-04, 2.0532e-01,  ..., 4.1618e-03,
           6.5002e-03, 2.2559e-01]],

         [[1.4985e-04, 1.5521e-04, 1.0944e-01,  ..., 1.6144e-02,
           6.6406e-02, 3.4790e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.0361e-04, 8.4877e-05, 2.2364e-04,  ..., 7.0251e-02,
           8.5596e-01, 2.9297e-02]],

         [[4.1723e-07, 1.3709e-06, 1.2994e-05,  ..., 9.5963e-05,
           2.0313e-04, 9.9854e-01]],

         [[2.5129e-04, 6.8331e-04, 2.9053e-02,  ..., 6.4812e-03,
           8.3237e-03, 3.7292e-02]],

         ...,

         [[5.9605e-08, 7.1526e-07, 5.4836e-05,  ..., 6.7482e-03,
           9.4092e-01, 4.6844e-02]],

         [[4.3106e-03, 5.8899e-03, 5.1994e-03,  ..., 2.9922e-02,
           1.0338e-02, 1.1115e-01]],

         [[6.7863e-03, 3.5210e-03, 4.9162e-04,  ..., 3.1372e-02,
           3.9062e-02, 5.1727e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1283e-04, 1.1940e-03, 4.0398e-03,  ..., 6.6223e-02,
           2.5000e-01, 1.0919e-01]],

         [[5.7817e-06, 6.3181e-06, 5.3644e-07,  ..., 1.1731e-01,
           7.8955e-01, 8.0627e-02]],

         [[5.9586e-03, 5.9586e-03, 1.3855e-02,  ..., 1.0452e-02,
           1.8356e-02, 8.1406e-03]],

         ...,

         [[1.1377e-03, 1.6556e-03, 2.3537e-03,  ..., 5.9424e-01,
           2.4011e-01, 4.4769e-02]],

         [[1.1730e-03, 3.7556e-03, 2.3102e-02,  ..., 2.6688e-02,
           5.0262e-02, 2.4872e-02]],

         [[1.5390e-04, 2.5940e-03, 3.5725e-03,  ..., 1.8762e-01,
           1.5076e-01, 2.8763e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8214e-02, 2.6665e-03, 1.9699e-02,  ..., 4.7272e-02,
           6.4087e-02, 2.7359e-02]],

         [[1.1444e-03, 3.5453e-04, 1.7586e-03,  ..., 1.4467e-03,
           1.0521e-02, 2.9053e-02]],

         [[6.4354e-03, 1.3332e-03, 1.3380e-03,  ..., 2.8168e-02,
           1.9363e-02, 1.0605e-02]],

         ...,

         [[6.2332e-03, 1.0338e-03, 2.1210e-03,  ..., 2.6709e-01,
           1.9385e-01, 3.6499e-01]],

         [[4.1795e-04, 1.6618e-04, 1.9586e-04,  ..., 1.7487e-02,
           2.4857e-02, 2.6260e-02]],

         [[1.8644e-03, 2.4462e-04, 1.3647e-03,  ..., 1.8692e-02,
           8.1348e-01, 1.0669e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.0186e-01, 3.1590e-06, 1.6689e-06,  ..., 2.7180e-03,
           6.1226e-03, 4.2839e-03]],

         [[9.1943e-01, 1.9252e-05, 9.5367e-06,  ..., 3.4882e-02,
           1.2367e-02, 1.0895e-02]],

         [[8.5498e-01, 3.6836e-05, 1.1921e-06,  ..., 3.6194e-02,
           1.0078e-02, 1.8646e-02]],

         ...,

         [[5.8447e-01, 1.2457e-04, 5.8889e-04,  ..., 5.4665e-03,
           8.1787e-03, 5.8319e-02]],

         [[4.9365e-01, 4.3082e-04, 9.4366e-04,  ..., 1.4908e-02,
           1.8265e-02, 8.6975e-03]],

         [[7.2363e-01, 5.6028e-05, 3.3903e-04,  ..., 2.5253e-02,
           2.1774e-02, 2.8275e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6670e-01, 5.3883e-05, 2.0206e-05,  ..., 1.8997e-03,
           2.2068e-03, 1.7639e-02]],

         [[6.2103e-02, 8.9407e-06, 0.0000e+00,  ..., 1.5278e-03,
           9.1943e-01, 1.6068e-02]],

         [[9.6094e-01, 5.4932e-04, 1.2338e-05,  ..., 6.0959e-03,
           6.9351e-03, 1.4534e-03]],

         ...,

         [[6.7334e-01, 2.6393e-04, 2.5988e-04,  ..., 2.7084e-02,
           5.2521e-02, 2.4048e-02]],

         [[2.4182e-01, 1.3056e-03, 9.6273e-04,  ..., 4.9553e-03,
           7.4120e-03, 6.5422e-03]],

         [[4.9146e-01, 4.0603e-04, 4.3344e-04,  ..., 2.6611e-02,
           4.0430e-01, 3.8330e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1104e-01, 1.9610e-04, 5.3823e-05,  ..., 5.0171e-02,
           2.5574e-02, 1.1795e-02]],

         [[7.6221e-01, 8.9586e-05, 9.2387e-06,  ..., 6.5247e-02,
           1.1755e-01, 8.6517e-03]],

         [[8.2959e-01, 2.3603e-05, 1.8239e-05,  ..., 4.1580e-03,
           3.6713e-02, 7.4997e-03]],

         ...,

         [[9.0039e-01, 8.9407e-07, 1.1921e-06,  ..., 1.9073e-06,
           6.5565e-04, 9.6130e-02]],

         [[9.7119e-01, 4.7684e-07, 1.6272e-05,  ..., 0.0000e+00,
           5.0664e-06, 2.1805e-02]],

         [[9.9414e-01, 5.9605e-08, 7.1526e-07,  ..., 0.0000e+00,
           1.4901e-06, 5.2261e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8022e-01, 8.3745e-05, 2.1458e-06,  ..., 8.1482e-02,
           3.0054e-01, 1.0696e-02]],

         [[4.4556e-01, 1.0490e-05, 3.3975e-06,  ..., 2.2972e-04,
           4.2572e-03, 3.8548e-03]],

         [[8.8672e-01, 1.4722e-04, 2.5272e-05,  ..., 4.0817e-04,
           4.1809e-03, 4.7264e-03]],

         ...,

         [[9.5801e-01, 1.3661e-04, 3.8147e-06,  ..., 8.5831e-03,
           6.8321e-03, 2.5444e-03]],

         [[6.4355e-01, 1.1778e-04, 1.1683e-05,  ..., 4.3915e-02,
           7.0557e-02, 2.5787e-02]],

         [[5.0098e-01, 8.4817e-05, 1.6272e-05,  ..., 6.3705e-03,
           4.1943e-01, 5.0934e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2998e-01, 8.7619e-06, 2.8014e-06,  ..., 4.9347e-02,
           4.0283e-02, 3.5992e-03]],

         [[7.1436e-01, 9.2983e-06, 3.8147e-06,  ..., 9.4223e-03,
           2.1545e-02, 1.3268e-02]],

         [[7.3096e-01, 6.0201e-06, 2.1458e-06,  ..., 5.6038e-03,
           1.6772e-01, 2.8488e-02]],

         ...,

         [[5.6201e-01, 2.9516e-04, 3.7551e-05,  ..., 8.4152e-03,
           1.6098e-02, 8.8348e-03]],

         [[7.3633e-01, 1.2034e-04, 2.0087e-05,  ..., 1.2741e-03,
           2.0035e-02, 1.7899e-02]],

         [[6.4746e-01, 1.0043e-04, 1.7345e-05,  ..., 4.3396e-02,
           1.7517e-01, 1.8158e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1221e-01, 7.6675e-04, 7.9393e-04,  ..., 8.5878e-04,
           1.7080e-03, 3.0079e-03]],

         [[8.6328e-01, 2.7299e-05, 1.6475e-04,  ..., 4.9667e-03,
           1.5182e-02, 2.0618e-03]],

         [[5.7080e-01, 1.1343e-04, 6.0976e-05,  ..., 9.7046e-03,
           1.3527e-02, 5.7068e-03]],

         ...,

         [[4.7705e-01, 6.7770e-05, 3.1888e-05,  ..., 1.0818e-02,
           1.4435e-02, 4.3018e-01]],

         [[3.7280e-01, 3.1650e-05, 3.0279e-05,  ..., 4.2191e-03,
           4.6582e-01, 1.0352e-01]],

         [[8.6035e-01, 2.5153e-05, 1.2577e-05,  ..., 7.6942e-03,
           2.3727e-02, 7.5989e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7051e-01, 2.1720e-04, 4.3035e-04,  ..., 3.8128e-03,
           6.3972e-03, 2.9640e-03]],

         [[5.3369e-01, 9.3579e-06, 4.0531e-06,  ..., 1.6443e-01,
           1.5881e-01, 5.9235e-02]],

         [[8.6768e-01, 6.6161e-06, 1.1206e-05,  ..., 4.5624e-03,
           1.0880e-02, 7.1182e-03]],

         ...,

         [[5.1514e-01, 2.1374e-04, 3.1710e-04,  ..., 1.1322e-02,
           6.4230e-04, 1.0948e-03]],

         [[8.1592e-01, 5.7042e-05, 3.9399e-05,  ..., 2.0752e-03,
           1.0931e-01, 9.8953e-03]],

         [[9.5898e-01, 4.9472e-06, 1.0610e-05,  ..., 4.1962e-03,
           4.8485e-03, 8.7051e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0322e-01, 2.5630e-05, 2.6882e-05,  ..., 1.2326e-04,
           7.3318e-03, 1.4641e-02]],

         [[7.4121e-01, 6.1572e-05, 3.2425e-05,  ..., 8.7280e-02,
           1.7075e-02, 6.5422e-03]],

         [[7.6514e-01, 1.3113e-06, 1.9073e-06,  ..., 4.3221e-03,
           4.5868e-02, 1.3443e-02]],

         ...,

         [[7.7539e-01, 2.1398e-04, 1.1015e-04,  ..., 2.6588e-03,
           7.4272e-03, 7.1220e-03]],

         [[7.1729e-01, 1.5593e-04, 1.2386e-04,  ..., 3.7140e-02,
           2.6474e-03, 9.4528e-03]],

         [[8.0615e-01, 4.4394e-04, 4.8566e-04,  ..., 2.0771e-03,
           1.1206e-03, 1.4935e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4990e-01, 1.0133e-06, 3.7551e-06,  ..., 1.1883e-03,
           1.8768e-02, 8.1689e-01]],

         [[6.0156e-01, 1.8625e-03, 1.4057e-03,  ..., 1.6537e-03,
           1.2531e-03, 1.3132e-03]],

         [[5.8545e-01, 5.4836e-04, 4.0102e-04,  ..., 2.2964e-02,
           2.8137e-02, 2.3148e-02]],

         ...,

         [[6.5869e-01, 7.7128e-05, 1.2130e-04,  ..., 2.4368e-02,
           2.8591e-03, 2.0847e-03]],

         [[6.2695e-01, 6.9737e-05, 3.4690e-05,  ..., 3.3020e-02,
           3.5797e-02, 2.1454e-02]],

         [[4.7461e-01, 9.9897e-05, 2.0027e-04,  ..., 2.6932e-02,
           6.9031e-02, 2.9785e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4580e-01, 7.3731e-05, 1.2743e-04,  ..., 4.0802e-02,
           6.7688e-02, 2.9224e-01]],

         [[6.2842e-01, 5.6744e-05, 9.1374e-05,  ..., 1.3831e-01,
           5.8746e-02, 4.4403e-02]],

         [[4.6582e-01, 1.7178e-04, 7.6532e-05,  ..., 5.7465e-02,
           1.0460e-02, 9.9792e-03]],

         ...,

         [[6.1737e-02, 7.1526e-07, 1.1623e-05,  ..., 6.4707e-04,
           1.2985e-02, 9.1650e-01]],

         [[8.4473e-01, 7.2360e-05, 7.0393e-05,  ..., 3.7212e-03,
           1.1086e-02, 1.1253e-02]],

         [[6.9385e-01, 6.7174e-05, 4.8757e-05,  ..., 1.3023e-02,
           4.5853e-03, 3.8986e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8096e-01, 6.6757e-04, 7.7724e-04,  ..., 4.3732e-02,
           2.5726e-02, 2.6138e-02]],

         [[6.8115e-01, 2.2829e-05, 7.2896e-05,  ..., 8.6121e-02,
           8.1909e-02, 5.8380e-02]],

         [[4.9463e-01, 8.2314e-05, 6.2346e-05,  ..., 1.5587e-02,
           7.4501e-03, 1.4832e-02]],

         ...,

         [[4.0894e-01, 8.9407e-06, 1.0729e-06,  ..., 2.6660e-01,
           3.0225e-01, 9.3536e-03]],

         [[4.7974e-01, 6.2418e-04, 5.8413e-04,  ..., 7.7629e-03,
           2.2259e-03, 1.7109e-03]],

         [[6.0254e-01, 1.5736e-05, 1.7881e-06,  ..., 1.6345e-01,
           1.7834e-01, 1.7914e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2266e-01, 6.7353e-05, 3.8981e-05,  ..., 2.3254e-02,
           6.7825e-03, 3.3245e-03]],

         [[1.0718e-01, 8.6308e-04, 5.9938e-04,  ..., 1.0345e-02,
           1.6823e-03, 1.4257e-03]],

         [[7.2217e-01, 6.3181e-04, 6.0558e-04,  ..., 1.0078e-02,
           1.5526e-02, 1.7853e-02]],

         ...,

         [[3.6304e-01, 2.5630e-05, 1.5914e-05,  ..., 4.6558e-01,
           5.9753e-02, 6.0806e-03]],

         [[4.6362e-01, 3.1710e-05, 1.5080e-05,  ..., 8.1970e-02,
           3.9648e-01, 1.2161e-02]],

         [[4.7021e-01, 1.3447e-04, 1.3864e-04,  ..., 8.4351e-02,
           9.1324e-03, 9.8515e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9083e-02, 2.9802e-07, 3.5763e-07,  ..., 1.5318e-04,
           8.9035e-03, 9.5947e-01]],

         [[3.7866e-01, 1.2636e-05, 2.7180e-05,  ..., 2.5635e-01,
           1.8103e-01, 1.2199e-02]],

         [[1.5247e-01, 1.7178e-04, 2.4986e-04,  ..., 3.0243e-02,
           6.8970e-03, 2.6455e-03]],

         ...,

         [[4.5215e-01, 3.4451e-05, 4.2558e-05,  ..., 1.6406e-01,
           6.7291e-03, 2.4204e-03]],

         [[2.1912e-01, 2.8014e-06, 1.3709e-06,  ..., 1.3257e-01,
           2.1103e-02, 4.1542e-03]],

         [[4.3237e-01, 9.1672e-05, 5.4300e-05,  ..., 7.7759e-02,
           5.7678e-03, 3.3436e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9780e-01, 2.3699e-04, 4.4346e-05,  ..., 1.3391e-01,
           4.6112e-02, 1.2611e-02]],

         [[1.8359e-01, 5.0604e-05, 2.1636e-05,  ..., 1.0059e-01,
           2.1927e-02, 5.1880e-03]],

         [[3.8721e-01, 1.7762e-05, 9.8348e-06,  ..., 1.6016e-01,
           7.5073e-02, 2.0203e-02]],

         ...,

         [[4.8486e-01, 2.4617e-05, 2.3663e-05,  ..., 1.4001e-01,
           2.1057e-02, 4.1466e-03]],

         [[6.7383e-01, 1.5795e-05, 1.7822e-05,  ..., 8.9783e-02,
           6.8542e-02, 3.4882e-02]],

         [[3.7451e-01, 2.8074e-05, 5.6207e-05,  ..., 1.0645e-01,
           1.3802e-02, 6.3210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 4.6802e-04, 7.6199e-04,  ..., 1.6499e-03,
           1.8129e-03, 6.5517e-04]],

         [[8.4106e-02, 1.1373e-04, 8.1897e-05,  ..., 7.3669e-02,
           3.7933e-02, 6.3362e-03]],

         [[3.9160e-01, 1.3709e-05, 1.6749e-05,  ..., 3.5376e-01,
           6.1951e-02, 7.6180e-03]],

         ...,

         [[8.7061e-01, 6.4433e-05, 4.4823e-05,  ..., 1.1353e-02,
           2.8591e-03, 1.0796e-03]],

         [[9.9414e-01, 5.0902e-05, 8.7917e-05,  ..., 7.3314e-06,
           5.2691e-05, 1.9569e-03]],

         [[8.1543e-01, 5.4300e-05, 1.0198e-04,  ..., 1.9058e-02,
           4.4746e-03, 1.9646e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1216e-01, 8.0645e-05, 7.1168e-05,  ..., 8.4351e-02,
           1.4778e-02, 7.2021e-03]],

         [[2.1655e-01, 4.3464e-04, 2.5153e-04,  ..., 3.9764e-02,
           1.6068e-02, 2.6836e-03]],

         [[2.5195e-01, 2.7418e-05, 1.4305e-05,  ..., 3.6011e-02,
           3.9014e-01, 1.9312e-01]],

         ...,

         [[2.2058e-01, 1.7107e-05, 1.5259e-05,  ..., 3.4180e-01,
           5.8685e-02, 7.4921e-03]],

         [[4.6509e-01, 1.6928e-05, 7.6890e-06,  ..., 1.7114e-01,
           1.2720e-01, 3.7292e-02]],

         [[2.1167e-01, 1.5974e-05, 5.9605e-06,  ..., 1.4893e-01,
           2.8931e-01, 1.6748e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1993e-01, 1.1683e-05, 4.3511e-06,  ..., 2.1045e-01,
           4.1443e-02, 8.2932e-03]],

         [[2.8198e-01, 4.8339e-05, 3.5346e-05,  ..., 1.4185e-01,
           6.9153e-02, 2.4033e-03]],

         [[9.4116e-02, 2.9206e-06, 3.9339e-06,  ..., 2.4300e-03,
           8.0505e-02, 8.1299e-01]],

         ...,

         [[2.0105e-01, 1.0043e-04, 3.4571e-05,  ..., 1.4336e-02,
           3.6831e-03, 5.7945e-03]],

         [[6.4209e-01, 2.9087e-04, 1.3137e-04,  ..., 1.8646e-02,
           1.2520e-02, 9.7885e-03]],

         [[1.3159e-01, 1.6475e-04, 1.3816e-04,  ..., 7.8003e-02,
           3.4332e-02, 2.8515e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3145e-01, 3.4869e-05, 2.3663e-05,  ..., 2.1570e-01,
           1.3260e-02, 4.3716e-03]],

         [[9.7070e-01, 5.4502e-04, 6.4611e-05,  ..., 7.5519e-05,
           9.8526e-05, 6.7978e-03]],

         [[8.5840e-01, 2.4629e-04, 7.1704e-05,  ..., 2.8858e-03,
           5.8746e-03, 6.0158e-03]],

         ...,

         [[1.6882e-01, 3.9005e-04, 7.6342e-04,  ..., 2.4704e-02,
           4.1618e-03, 1.0357e-03]],

         [[6.1768e-01, 2.4915e-05, 2.2531e-05,  ..., 2.7557e-02,
           8.5373e-03, 1.9054e-03]],

         [[2.9810e-01, 5.2404e-04, 1.6775e-03,  ..., 2.8152e-02,
           2.7084e-02, 1.0773e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3613e-01, 1.1432e-04, 1.9443e-04,  ..., 1.8204e-02,
           1.6449e-02, 3.4733e-03]],

         [[4.3091e-01, 1.5879e-04, 3.0851e-04,  ..., 3.2715e-02,
           1.2611e-02, 1.3222e-02]],

         [[4.6777e-01, 9.2983e-05, 2.2125e-04,  ..., 2.3468e-02,
           1.2657e-02, 4.4441e-03]],

         ...,

         [[6.0822e-02, 2.7001e-05, 5.2094e-05,  ..., 5.8929e-02,
           9.6207e-03, 6.2141e-03]],

         [[1.5735e-01, 3.1233e-05, 4.3392e-05,  ..., 2.1957e-02,
           2.8362e-03, 7.9575e-03]],

         [[5.5957e-01, 2.9445e-05, 5.9962e-05,  ..., 2.7847e-02,
           7.7362e-03, 2.5501e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9083e-02, 6.1393e-06, 7.8082e-06,  ..., 1.1066e-01,
           1.3420e-02, 3.0422e-03]],

         [[2.8296e-01, 2.3949e-04, 3.5405e-04,  ..., 4.3732e-02,
           1.6083e-02, 6.0120e-03]],

         [[4.7168e-01, 1.1945e-04, 3.8862e-04,  ..., 1.5167e-02,
           8.3923e-02, 2.7100e-01]],

         ...,

         [[2.8882e-01, 2.4021e-05, 2.8491e-05,  ..., 7.1350e-02,
           7.4615e-03, 2.6913e-03]],

         [[7.8857e-01, 3.6478e-05, 9.3579e-05,  ..., 2.6352e-02,
           1.0567e-02, 1.2070e-02]],

         [[1.6748e-01, 8.2254e-06, 2.6524e-05,  ..., 1.1391e-02,
           2.5826e-03, 7.5150e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1798e-01, 4.5002e-05, 3.0637e-05,  ..., 1.1615e-01,
           3.1494e-02, 6.7329e-03]],

         [[1.3428e-01, 2.5153e-05, 3.6418e-05,  ..., 7.8308e-02,
           1.1917e-02, 1.6003e-03]],

         [[4.7534e-01, 1.2445e-04, 1.5461e-04,  ..., 5.0110e-02,
           2.2400e-02, 4.4479e-03]],

         ...,

         [[5.4199e-01, 1.0729e-06, 2.3842e-07,  ..., 2.5610e-01,
           8.1848e-02, 2.9564e-03]],

         [[2.9761e-01, 2.0266e-06, 9.4175e-06,  ..., 2.0947e-01,
           1.4392e-01, 1.1726e-02]],

         [[2.0923e-01, 6.3777e-06, 3.0398e-06,  ..., 6.1859e-02,
           7.1924e-01, 7.2174e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4397e-02, 7.1526e-07, 6.3181e-06,  ..., 2.5122e-01,
           1.5442e-02, 2.9011e-03]],

         [[2.2324e-02, 4.7684e-06, 2.5630e-05,  ..., 2.6107e-04,
           2.3041e-02, 9.3457e-01]],

         [[3.5913e-01, 1.1706e-04, 4.3201e-04,  ..., 1.9867e-02,
           1.0223e-02, 3.0937e-03]],

         ...,

         [[1.6760e-01, 1.3387e-04, 3.7003e-04,  ..., 1.9791e-02,
           1.2482e-02, 2.8515e-03]],

         [[3.4009e-01, 4.0531e-06, 6.9141e-06,  ..., 6.3904e-02,
           1.2122e-01, 9.4223e-03]],

         [[4.1260e-01, 5.1856e-06, 3.9935e-06,  ..., 1.8164e-01,
           3.3691e-01, 2.7115e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7236e-01, 1.8847e-04, 1.1826e-03,  ..., 1.2032e-02,
           8.0185e-03, 2.6741e-03]],

         [[9.8340e-01, 1.3888e-05, 1.1533e-04,  ..., 1.2672e-04,
           2.6083e-04, 2.4128e-04]],

         [[9.4580e-01, 2.0444e-05, 8.5711e-05,  ..., 1.1835e-03,
           6.0940e-04, 4.2534e-04]],

         ...,

         [[5.6396e-01, 5.1856e-06, 7.5698e-06,  ..., 5.8533e-02,
           2.5177e-02, 3.8605e-03]],

         [[7.3340e-01, 1.5974e-05, 1.3173e-05,  ..., 1.9970e-03,
           1.4048e-03, 8.0681e-04]],

         [[1.0114e-01, 3.5763e-06, 1.8716e-05,  ..., 1.5116e-03,
           4.2816e-02, 7.8271e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3538e-02, 1.0729e-06, 3.0994e-06,  ..., 2.6343e-01,
           2.9858e-01, 2.5177e-02]],

         [[3.8208e-01, 1.2457e-05, 8.1658e-06,  ..., 1.0242e-01,
           4.9927e-02, 3.1300e-03]],

         [[1.9580e-01, 7.9513e-05, 9.7871e-05,  ..., 5.5237e-02,
           2.2751e-02, 2.1324e-03]],

         ...,

         [[3.8379e-01, 1.0133e-05, 4.4048e-05,  ..., 1.7566e-01,
           9.2590e-02, 8.4763e-03]],

         [[5.4639e-01, 3.9339e-06, 2.5511e-05,  ..., 9.4177e-02,
           4.9255e-02, 4.1473e-02]],

         [[5.5225e-01, 2.2054e-06, 8.4639e-06,  ..., 1.4514e-01,
           4.2908e-02, 2.4261e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9791e-02, 1.0729e-06, 7.1526e-07,  ..., 7.2250e-03,
           3.0346e-03, 4.8971e-04]],

         [[5.0537e-01, 3.1829e-05, 1.4067e-05,  ..., 2.5921e-03,
           1.2150e-03, 7.0000e-04]],

         [[2.9004e-01, 2.7895e-05, 3.1650e-05,  ..., 3.3325e-02,
           8.2932e-03, 2.2755e-03]],

         ...,

         [[4.9408e-02, 3.8564e-05, 4.4799e-04,  ..., 2.3346e-02,
           2.1255e-02, 1.0361e-02]],

         [[2.7173e-01, 8.2970e-05, 9.9468e-04,  ..., 2.7756e-02,
           3.0960e-02, 8.8730e-03]],

         [[8.7598e-01, 7.8082e-06, 1.2100e-05,  ..., 1.9958e-02,
           2.5635e-02, 4.0894e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2080e-01, 1.1325e-06, 3.1590e-06,  ..., 1.3962e-02,
           7.0862e-02, 2.2522e-02]],

         [[2.1069e-01, 1.3709e-06, 6.4969e-06,  ..., 1.4429e-01,
           2.0508e-01, 4.1992e-02]],

         [[6.5918e-02, 0.0000e+00, 1.1921e-07,  ..., 1.8994e-01,
           7.2266e-01, 9.2392e-03]],

         ...,

         [[5.3320e-01, 2.1815e-05, 1.0788e-04,  ..., 1.3252e-02,
           1.0002e-02, 4.5090e-03]],

         [[6.9885e-02, 1.9670e-06, 3.8743e-06,  ..., 3.3081e-01,
           3.2745e-02, 1.2627e-02]],

         [[6.4575e-02, 6.3181e-05, 9.0301e-05,  ..., 4.1428e-03,
           1.0986e-01, 2.3621e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2910e-01, 4.5657e-05, 1.2589e-04,  ..., 2.4216e-02,
           2.2751e-02, 4.9400e-03]],

         [[7.3486e-01, 4.8161e-04, 6.5041e-04,  ..., 1.9007e-03,
           1.7862e-03, 9.8991e-04]],

         [[2.1729e-01, 6.2108e-05, 7.2300e-05,  ..., 1.7090e-02,
           5.8365e-03, 2.3403e-03]],

         ...,

         [[1.2091e-01, 4.3511e-05, 9.8825e-05,  ..., 2.1347e-02,
           4.3530e-01, 1.4900e-02]],

         [[1.3000e-01, 3.0220e-05, 1.3018e-04,  ..., 6.4331e-02,
           7.7637e-02, 5.7373e-01]],

         [[8.5254e-01, 2.4796e-05, 2.1756e-04,  ..., 1.3351e-02,
           2.6550e-02, 1.3351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9346e-01, 5.5790e-05, 2.4354e-04,  ..., 1.4877e-02,
           1.6342e-02, 2.3994e-03]],

         [[1.4368e-01, 7.4983e-05, 1.3244e-04,  ..., 4.7729e-02,
           5.3680e-02, 1.2878e-02]],

         [[1.0541e-01, 2.0087e-04, 6.4898e-04,  ..., 4.7302e-02,
           1.4465e-01, 6.7101e-03]],

         ...,

         [[7.1289e-01, 2.1696e-05, 8.5592e-05,  ..., 1.4397e-02,
           2.5757e-02, 1.0490e-02]],

         [[8.7842e-01, 1.7226e-05, 2.9087e-04,  ..., 1.4343e-03,
           9.5139e-03, 2.1458e-03]],

         [[5.2637e-01, 3.0279e-05, 2.6166e-05,  ..., 1.5762e-02,
           1.2695e-01, 1.1292e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3021e-05, 4.8757e-05, 1.8469e-01,  ..., 4.8157e-02,
           2.4700e-03, 7.7400e-03]],

         [[1.6689e-05, 6.0737e-05, 1.9434e-01,  ..., 2.1378e-02,
           2.4834e-03, 5.2605e-03]],

         [[1.1700e-04, 1.6987e-04, 1.0065e-01,  ..., 3.3722e-02,
           6.1768e-02, 1.8829e-02]],

         ...,

         [[3.2139e-04, 9.1732e-05, 1.1353e-01,  ..., 7.0496e-02,
           3.3105e-01, 1.0748e-01]],

         [[2.0838e-04, 2.0278e-04, 1.1603e-01,  ..., 3.1235e-02,
           2.8656e-02, 1.0460e-02]],

         [[2.2173e-05, 2.7359e-05, 2.3425e-01,  ..., 9.0866e-03,
           1.1826e-03, 2.7695e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4199e-05, 2.9743e-05, 2.7856e-01,  ..., 1.2054e-02,
           1.4963e-03, 2.4300e-03]],

         [[4.1902e-05, 3.4451e-05, 2.4072e-01,  ..., 8.2397e-03,
           6.3276e-04, 1.0891e-03]],

         [[1.0252e-05, 4.7088e-06, 3.1812e-01,  ..., 5.7144e-03,
           9.3269e-04, 6.9284e-04]],

         ...,

         [[7.5638e-05, 4.3333e-05, 2.9053e-01,  ..., 6.0043e-03,
           2.2519e-04, 6.3276e-04]],

         [[7.4506e-06, 2.2113e-05, 2.7856e-01,  ..., 7.8735e-03,
           4.1413e-04, 2.9016e-04]],

         [[3.5763e-06, 1.7643e-05, 2.0264e-01,  ..., 1.4915e-02,
           2.8181e-04, 1.0309e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2517e-05, 1.1861e-05, 2.2864e-01,  ..., 3.8128e-03,
           6.9237e-04, 2.4021e-04]],

         [[6.3062e-05, 2.8014e-06, 2.8271e-01,  ..., 5.4474e-03,
           2.1577e-05, 1.7548e-04]],

         [[1.3769e-05, 2.0266e-05, 3.0054e-01,  ..., 5.9509e-03,
           1.0753e-04, 4.0746e-04]],

         ...,

         [[5.5850e-05, 2.3782e-05, 1.2164e-01,  ..., 1.2329e-02,
           3.3112e-02, 5.8022e-03]],

         [[1.4484e-05, 3.6001e-05, 2.7881e-01,  ..., 4.9706e-03,
           5.9557e-04, 5.4455e-04]],

         [[9.9719e-05, 4.9055e-05, 1.9946e-01,  ..., 5.8632e-03,
           3.3522e-04, 4.8208e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9407e-05, 1.2994e-05, 3.6713e-02,  ..., 1.6199e-01,
           1.4868e-01, 7.4768e-02]],

         [[4.4584e-05, 5.2571e-05, 9.6741e-02,  ..., 2.5726e-02,
           2.6443e-02, 6.7253e-03]],

         [[3.5667e-04, 6.3372e-04, 2.9468e-01,  ..., 1.2741e-03,
           1.1806e-03, 4.8828e-04]],

         ...,

         [[2.6488e-04, 3.1829e-05, 3.6652e-02,  ..., 4.1687e-02,
           1.0117e-02, 2.1118e-02]],

         [[1.1597e-03, 1.9085e-04, 1.6663e-01,  ..., 5.6702e-02,
           8.2214e-02, 2.1561e-02]],

         [[4.7088e-06, 1.0133e-06, 2.9526e-03,  ..., 1.6956e-01,
           6.4880e-02, 1.9824e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0395e-03, 2.7204e-04, 2.0435e-01,  ..., 1.2199e-02,
           6.0272e-02, 1.2810e-02]],

         [[4.8923e-04, 4.4870e-04, 2.5708e-01,  ..., 9.3079e-03,
           1.2138e-02, 7.8812e-03]],

         [[1.3409e-03, 9.1791e-04, 2.2913e-01,  ..., 4.1626e-02,
           8.3923e-02, 1.5053e-02]],

         ...,

         [[3.6836e-04, 5.3465e-05, 2.8857e-01,  ..., 4.2152e-03,
           4.9011e-02, 1.6159e-02]],

         [[1.5116e-03, 1.0719e-03, 2.4158e-01,  ..., 1.0948e-02,
           5.2643e-02, 1.9263e-01]],

         [[2.0814e-04, 1.1277e-04, 3.7842e-02,  ..., 1.6833e-01,
           1.9373e-01, 3.3740e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[2.0039e-04, 5.6219e-04, 3.4428e-03,  ..., 1.1401e-01,
           5.1123e-01, 2.4146e-01]],

         [[9.0599e-05, 6.6280e-05, 2.3365e-03,  ..., 5.9662e-03,
           1.1255e-01, 3.4668e-01]],

         [[0.0000e+00, 2.3842e-07, 9.2506e-05,  ..., 3.4034e-05,
           1.3725e-02, 7.4951e-01]],

         ...,

         [[5.9605e-08, 1.3709e-06, 2.7418e-05,  ..., 4.5180e-05,
           2.4681e-03, 9.9561e-01]],

         [[1.7767e-03, 2.7523e-03, 3.7613e-03,  ..., 2.6108e-02,
           7.5562e-02, 5.8838e-02]],

         [[7.0343e-03, 1.3847e-03, 3.5024e-04,  ..., 9.4116e-02,
           2.2354e-02, 1.1597e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5175e-05, 6.2943e-04, 2.1954e-03,  ..., 1.1993e-01,
           4.9988e-02, 6.5186e-02]],

         [[1.1921e-07, 9.9540e-06, 8.7619e-06,  ..., 1.3892e-01,
           7.7490e-01, 7.5562e-02]],

         [[3.4447e-03, 3.7842e-03, 9.0790e-03,  ..., 3.5889e-02,
           8.2626e-03, 7.5264e-03]],

         ...,

         [[8.5950e-05, 2.8419e-04, 5.5599e-04,  ..., 1.4038e-01,
           7.4756e-01, 3.0853e-02]],

         [[2.3880e-03, 5.7297e-03, 2.3743e-02,  ..., 4.6875e-02,
           1.8784e-02, 2.1118e-02]],

         [[3.5429e-04, 1.3580e-03, 2.1362e-03,  ..., 1.9080e-01,
           2.8137e-02, 4.3488e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3270e-02, 2.3403e-03, 1.8997e-02,  ..., 8.8501e-02,
           3.8361e-02, 2.7847e-02]],

         [[1.2150e-03, 3.4404e-04, 2.9144e-03,  ..., 6.1722e-03,
           2.8763e-02, 2.7599e-03]],

         [[6.4774e-03, 1.4744e-03, 1.5211e-03,  ..., 1.9714e-02,
           1.1597e-02, 1.8967e-02]],

         ...,

         [[5.5313e-04, 1.2350e-04, 1.2255e-04,  ..., 1.2329e-01,
           3.3173e-02, 3.4241e-02]],

         [[8.2159e-04, 1.2255e-04, 2.2900e-04,  ..., 5.3284e-02,
           4.3488e-02, 3.3600e-02]],

         [[6.2714e-03, 7.3195e-04, 5.8022e-03,  ..., 5.1025e-01,
           1.8054e-01, 7.4646e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9785e-01, 1.1921e-07, 4.7684e-07,  ..., 1.2489e-02,
           1.1635e-02, 5.1498e-03]],

         [[8.4863e-01, 8.7619e-06, 4.2379e-05,  ..., 1.3222e-02,
           8.6243e-02, 1.7853e-03]],

         [[8.3887e-01, 9.4771e-06, 8.3447e-07,  ..., 4.0131e-02,
           1.1711e-03, 1.1719e-02]],

         ...,

         [[5.3955e-01, 6.0499e-05, 4.1699e-04,  ..., 3.2864e-03,
           4.1542e-03, 1.6296e-02]],

         [[4.6924e-01, 1.3721e-04, 2.9325e-04,  ..., 6.4850e-03,
           1.1539e-03, 1.0843e-03]],

         [[4.4116e-01, 1.2720e-04, 2.0027e-04,  ..., 1.5601e-01,
           1.8530e-01, 8.0185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6123e-01, 2.3782e-05, 1.0371e-05,  ..., 6.5460e-03,
           3.0918e-03, 2.3849e-02]],

         [[1.0010e-01, 3.1471e-05, 3.5763e-07,  ..., 2.1863e-01,
           6.6943e-01, 8.7433e-03]],

         [[9.3945e-01, 2.2042e-04, 2.3246e-05,  ..., 2.5578e-03,
           1.7614e-03, 7.4577e-04]],

         ...,

         [[4.2920e-01, 6.3777e-05, 7.9334e-05,  ..., 1.5417e-01,
           9.5764e-02, 1.4343e-02]],

         [[3.4033e-01, 1.0948e-03, 9.5224e-04,  ..., 3.0155e-03,
           3.9330e-03, 1.3628e-03]],

         [[7.3926e-01, 2.2709e-04, 3.6001e-04,  ..., 1.0944e-01,
           3.4302e-02, 1.5205e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7969e-01, 1.3626e-04, 4.5061e-05,  ..., 2.2400e-02,
           1.5182e-02, 1.1856e-02]],

         [[3.0347e-01, 6.8367e-05, 4.1127e-06,  ..., 2.2205e-01,
           1.0498e-02, 1.3571e-03]],

         [[6.0986e-01, 4.0710e-05, 1.7226e-05,  ..., 7.3608e-02,
           5.7251e-02, 4.6616e-03]],

         ...,

         [[6.4453e-01, 3.2187e-06, 5.3644e-07,  ..., 1.4572e-03,
           2.8877e-03, 3.4912e-01]],

         [[9.3945e-01, 5.5432e-06, 1.2636e-05,  ..., 8.7619e-06,
           2.3878e-04, 5.2490e-02]],

         [[9.9121e-01, 8.9407e-07, 2.3842e-07,  ..., 8.9407e-07,
           1.0490e-05, 7.8278e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4424e-01, 1.5140e-05, 1.0729e-06,  ..., 2.6831e-01,
           1.6284e-01, 1.9958e-02]],

         [[4.5947e-01, 6.9737e-06, 1.8477e-06,  ..., 3.8128e-03,
           4.2892e-04, 5.0507e-03]],

         [[8.8477e-01, 3.2592e-04, 6.6459e-05,  ..., 7.5645e-03,
           2.9163e-03, 1.3781e-03]],

         ...,

         [[9.3018e-01, 2.8467e-04, 1.5855e-05,  ..., 7.5455e-03,
           3.2845e-03, 5.0240e-03]],

         [[3.1128e-01, 9.5069e-05, 7.8678e-06,  ..., 1.6882e-01,
           8.8806e-02, 9.4727e-02]],

         [[6.3574e-01, 1.4257e-04, 2.6703e-05,  ..., 1.9690e-01,
           1.1023e-01, 2.2568e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7715e-01, 7.2718e-06, 4.8280e-06,  ..., 3.6469e-02,
           3.5267e-03, 1.3527e-02]],

         [[3.1982e-01, 3.2067e-05, 3.8862e-05,  ..., 1.6434e-02,
           7.6866e-03, 1.5945e-02]],

         [[7.4756e-01, 1.5378e-05, 3.9935e-06,  ..., 1.0968e-01,
           1.7120e-02, 1.2650e-02]],

         ...,

         [[5.8447e-01, 1.0955e-04, 4.3750e-05,  ..., 2.3315e-02,
           1.3885e-02, 1.2787e-02]],

         [[8.0469e-01, 1.6189e-04, 2.8670e-05,  ..., 1.2489e-02,
           8.8730e-03, 5.8746e-03]],

         [[5.2148e-01, 4.0829e-05, 6.1989e-05,  ..., 1.1749e-01,
           5.7251e-02, 1.1009e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0654e-01, 5.9319e-04, 3.8743e-04,  ..., 8.4972e-04,
           1.1482e-03, 6.1464e-04]],

         [[8.6670e-01, 2.6226e-05, 3.9935e-05,  ..., 1.9714e-02,
           2.3689e-03, 3.9520e-03]],

         [[8.5303e-01, 2.5868e-05, 1.5318e-05,  ..., 1.4572e-03,
           1.2808e-03, 6.6471e-04]],

         ...,

         [[7.0361e-01, 2.2411e-05, 4.0293e-05,  ..., 8.8120e-03,
           1.2894e-02, 6.0730e-02]],

         [[2.5439e-01, 2.1458e-06, 3.9339e-06,  ..., 3.4106e-01,
           1.9214e-01, 7.7454e-02]],

         [[6.2158e-01, 5.1856e-06, 2.0862e-06,  ..., 1.5173e-01,
           1.2854e-01, 9.6588e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7759e-01, 3.2973e-04, 6.4802e-04,  ..., 9.2316e-03,
           3.7746e-03, 1.3351e-03]],

         [[3.8867e-01, 8.8215e-06, 4.5300e-06,  ..., 1.4062e-01,
           2.8589e-01, 3.6743e-02]],

         [[8.3398e-01, 1.5497e-06, 6.2585e-06,  ..., 9.0103e-03,
           4.7302e-03, 3.1757e-03]],

         ...,

         [[7.3584e-01, 1.5879e-04, 1.4687e-04,  ..., 3.1400e-04,
           3.9530e-04, 1.9038e-04]],

         [[7.7637e-01, 2.6941e-05, 1.5557e-05,  ..., 6.5613e-02,
           3.8940e-02, 7.6942e-03]],

         [[8.8818e-01, 2.9802e-06, 5.3644e-06,  ..., 6.5384e-03,
           1.1345e-02, 4.0894e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4062e-01, 9.1195e-06, 6.4373e-06,  ..., 5.0163e-04,
           3.5429e-04, 5.6725e-03]],

         [[6.8213e-01, 9.0599e-06, 2.4974e-05,  ..., 3.0701e-02,
           2.4811e-02, 3.6964e-03]],

         [[4.6704e-01, 8.9407e-07, 4.7684e-07,  ..., 1.3580e-03,
           2.7800e-04, 1.1845e-03]],

         ...,

         [[8.8037e-01, 4.6432e-05, 2.7120e-05,  ..., 7.2975e-03,
           8.0414e-03, 3.9444e-03]],

         [[7.2559e-01, 1.0699e-04, 6.4135e-05,  ..., 2.4872e-03,
           4.5776e-03, 7.7095e-03]],

         [[5.7520e-01, 3.5262e-04, 1.9407e-04,  ..., 8.5602e-03,
           5.6419e-03, 4.3221e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4319e-01, 5.3644e-07, 2.5630e-06,  ..., 9.6798e-04,
           4.3488e-02, 7.9199e-01]],

         [[6.5674e-01, 6.7282e-04, 7.4387e-04,  ..., 4.4632e-04,
           6.8903e-04, 2.9159e-04]],

         [[5.6445e-01, 4.8876e-04, 3.8576e-04,  ..., 2.6108e-02,
           2.3224e-02, 9.9106e-03]],

         ...,

         [[7.7051e-01, 2.4080e-05, 5.2035e-05,  ..., 7.8869e-04,
           9.0742e-04, 1.9989e-03]],

         [[4.2310e-01, 2.7776e-05, 3.8743e-05,  ..., 1.0109e-02,
           4.8294e-03, 1.2787e-02]],

         [[2.5659e-01, 1.8716e-05, 6.3121e-05,  ..., 2.0050e-02,
           4.1138e-02, 4.9585e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4451e-01, 2.9683e-05, 3.7193e-05,  ..., 1.3481e-02,
           3.0853e-02, 2.7563e-01]],

         [[3.4229e-01, 7.3969e-05, 8.8513e-05,  ..., 6.4209e-02,
           1.0632e-01, 2.7100e-02]],

         [[5.0244e-01, 3.7313e-05, 1.0848e-05,  ..., 8.4925e-04,
           5.3978e-04, 1.4391e-03]],

         ...,

         [[1.3390e-02, 5.9605e-08, 6.5565e-07,  ..., 1.2302e-04,
           7.0419e-03, 9.7217e-01]],

         [[6.7920e-01, 2.6524e-05, 3.9041e-05,  ..., 8.7738e-03,
           1.0719e-02, 4.1901e-02]],

         [[3.0420e-01, 5.1618e-05, 1.8716e-05,  ..., 4.6043e-03,
           3.1223e-03, 9.4604e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2383e-01, 2.8157e-04, 4.3797e-04,  ..., 2.2491e-02,
           1.6205e-02, 4.4937e-03]],

         [[1.6528e-01, 8.1658e-06, 3.6359e-05,  ..., 1.8042e-01,
           5.6006e-01, 3.1494e-02]],

         [[5.5566e-01, 1.2338e-05, 2.1040e-05,  ..., 1.8892e-03,
           2.7542e-03, 3.9062e-03]],

         ...,

         [[5.4199e-01, 1.3947e-05, 1.7881e-05,  ..., 1.9177e-01,
           2.1143e-01, 1.5045e-02]],

         [[3.9502e-01, 2.6679e-04, 2.4557e-04,  ..., 2.7199e-03,
           2.4185e-03, 3.3779e-03]],

         [[7.5195e-01, 3.2723e-05, 1.6451e-05,  ..., 6.5430e-02,
           2.6245e-02, 2.7267e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6211e-01, 3.3379e-05, 1.6689e-05,  ..., 1.0891e-03,
           6.1941e-04, 5.9433e-03]],

         [[1.1200e-01, 4.8018e-04, 3.9148e-04,  ..., 5.3329e-03,
           4.1389e-03, 2.5272e-03]],

         [[3.5449e-01, 2.0540e-04, 2.1565e-04,  ..., 1.4551e-01,
           1.6736e-01, 1.7853e-02]],

         ...,

         [[4.7681e-01, 1.1683e-05, 2.4498e-05,  ..., 7.3608e-02,
           1.3023e-02, 6.8207e-03]],

         [[6.2451e-01, 1.1325e-05, 3.2067e-05,  ..., 2.1924e-01,
           2.7832e-02, 2.1988e-02]],

         [[5.4297e-01, 5.6922e-05, 4.5538e-05,  ..., 2.4841e-02,
           1.1360e-02, 9.6970e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5024e-02, 1.1921e-07, 1.1921e-07,  ..., 3.4189e-04,
           6.2065e-03, 9.6533e-01]],

         [[1.7480e-01, 4.0531e-06, 1.9789e-05,  ..., 3.5571e-01,
           8.4045e-02, 1.9760e-02]],

         [[2.6001e-01, 3.8934e-04, 6.3896e-04,  ..., 4.9744e-03,
           1.6136e-03, 1.4858e-03]],

         ...,

         [[3.4937e-01, 2.9802e-05, 1.5557e-05,  ..., 1.8280e-02,
           3.8090e-03, 5.8327e-03]],

         [[2.2363e-01, 1.6689e-06, 1.2517e-06,  ..., 8.1024e-03,
           7.0286e-04, 2.3060e-03]],

         [[3.4619e-01, 4.2319e-05, 2.3842e-05,  ..., 2.2602e-04,
           8.7798e-05, 2.3556e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9478e-01, 1.6272e-04, 8.1062e-05,  ..., 2.7817e-02,
           1.1307e-02, 8.5144e-03]],

         [[1.3818e-01, 3.2246e-05, 1.2696e-05,  ..., 7.0686e-03,
           1.2550e-03, 8.1635e-03]],

         [[2.5928e-01, 1.2159e-04, 3.4750e-05,  ..., 3.6469e-02,
           5.5923e-03, 2.0218e-02]],

         ...,

         [[3.1348e-01, 6.6757e-06, 3.8743e-06,  ..., 1.2634e-02,
           1.3113e-03, 4.0245e-03]],

         [[2.7637e-01, 8.4043e-06, 6.1989e-06,  ..., 1.9604e-01,
           2.5757e-01, 6.3110e-02]],

         [[2.1509e-01, 4.7088e-06, 6.6161e-06,  ..., 1.5640e-02,
           2.6875e-03, 5.5542e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6021e-01, 4.6730e-04, 5.5552e-04,  ..., 7.0229e-03,
           2.1935e-03, 5.5552e-04]],

         [[1.3550e-01, 3.5584e-05, 2.8193e-05,  ..., 2.4586e-03,
           2.2244e-04, 3.9635e-03]],

         [[3.6694e-01, 8.8811e-06, 1.4961e-05,  ..., 4.4861e-02,
           2.2736e-02, 1.4595e-02]],

         ...,

         [[7.4805e-01, 4.4644e-05, 3.4034e-05,  ..., 4.2953e-03,
           1.4000e-03, 5.2910e-03]],

         [[9.8291e-01, 2.5511e-05, 4.8637e-05,  ..., 4.4763e-05,
           1.6356e-04, 1.0557e-03]],

         [[7.4121e-01, 5.9366e-05, 1.5616e-04,  ..., 5.1155e-03,
           4.2152e-03, 2.4605e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5674e-01, 6.8009e-05, 6.5446e-05,  ..., 5.6763e-02,
           6.8481e-02, 6.1226e-03]],

         [[6.5735e-02, 2.5821e-04, 1.3041e-04,  ..., 7.7286e-03,
           1.3638e-03, 3.2463e-03]],

         [[8.1238e-02, 7.2718e-06, 2.5034e-06,  ..., 7.1106e-02,
           3.7170e-02, 5.8154e-01]],

         ...,

         [[3.6108e-01, 3.7432e-05, 2.1577e-05,  ..., 2.5452e-02,
           5.0507e-03, 2.6207e-03]],

         [[6.0645e-01, 5.2750e-05, 2.3425e-05,  ..., 1.8906e-02,
           3.0499e-03, 2.0126e-02]],

         [[2.0081e-01, 5.9605e-06, 4.8280e-06,  ..., 8.1177e-02,
           3.0090e-02, 1.8286e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4668e-01, 1.1504e-05, 3.9935e-06,  ..., 1.0147e-02,
           2.6054e-03, 7.6599e-03]],

         [[1.2683e-01, 1.4484e-05, 1.3173e-05,  ..., 7.0068e-02,
           4.0779e-03, 9.6283e-03]],

         [[8.6304e-02, 1.1325e-06, 1.0133e-06,  ..., 8.6823e-03,
           4.2725e-02, 8.3203e-01]],

         ...,

         [[1.6187e-01, 4.8101e-05, 1.7643e-05,  ..., 2.0370e-03,
           9.3317e-04, 5.9433e-03]],

         [[3.7573e-01, 6.4254e-05, 3.1769e-05,  ..., 1.3901e-02,
           2.8248e-03, 5.2757e-03]],

         [[2.1924e-01, 5.0247e-05, 3.8326e-05,  ..., 1.0017e-02,
           1.1368e-03, 3.7880e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.5764e-02, 5.3048e-06, 2.8014e-06,  ..., 1.0834e-02,
           2.1667e-03, 6.2218e-03]],

         [[9.4434e-01, 9.3639e-05, 3.1590e-06,  ..., 2.2113e-04,
           1.5593e-03, 1.9875e-03]],

         [[8.9893e-01, 2.5630e-05, 2.6822e-06,  ..., 3.6182e-03,
           1.5202e-03, 3.5610e-03]],

         ...,

         [[1.5967e-01, 7.9346e-04, 1.8444e-03,  ..., 7.0572e-04,
           2.0695e-04, 9.4223e-04]],

         [[5.6738e-01, 1.1981e-05, 1.4126e-05,  ..., 8.4019e-04,
           1.6809e-04, 4.6844e-03]],

         [[4.3066e-01, 3.2830e-04, 1.0109e-03,  ..., 2.1118e-02,
           1.1841e-02, 3.6983e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8389e-01, 9.9242e-05, 1.8108e-04,  ..., 5.4169e-03,
           1.1902e-03, 2.4094e-02]],

         [[3.3032e-01, 9.7036e-05, 1.4007e-04,  ..., 1.1307e-02,
           9.8953e-03, 4.4250e-03]],

         [[4.7974e-01, 4.5240e-05, 1.2434e-04,  ..., 3.4962e-03,
           1.9913e-03, 3.7804e-03]],

         ...,

         [[1.6626e-01, 5.1618e-05, 7.5042e-05,  ..., 4.8676e-03,
           2.0294e-03, 7.8964e-03]],

         [[2.1594e-01, 7.0214e-05, 6.6459e-05,  ..., 1.5736e-03,
           7.9107e-04, 5.0774e-03]],

         [[3.3740e-01, 1.2517e-05, 2.3723e-05,  ..., 1.3161e-03,
           3.0065e-04, 1.6899e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5604e-02, 3.3975e-06, 3.0994e-06,  ..., 3.0575e-03,
           8.1015e-04, 5.5809e-03]],

         [[2.2815e-01, 1.1313e-04, 1.7726e-04,  ..., 4.9248e-03,
           3.2291e-03, 4.8866e-03]],

         [[2.1985e-01, 2.9445e-05, 1.1647e-04,  ..., 1.8051e-02,
           1.6953e-02, 5.8350e-01]],

         ...,

         [[2.0837e-01, 1.4842e-05, 1.9014e-05,  ..., 5.8632e-03,
           2.0657e-03, 1.0208e-02]],

         [[5.7568e-01, 6.1274e-05, 1.2279e-04,  ..., 6.7558e-03,
           2.9984e-03, 2.1133e-02]],

         [[1.2122e-01, 1.2338e-05, 3.1948e-05,  ..., 1.5593e-04,
           1.3435e-04, 3.2063e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6821e-01, 4.7207e-05, 2.4080e-05,  ..., 9.6741e-03,
           1.6813e-03, 1.4923e-02]],

         [[9.5581e-02, 1.8299e-05, 3.8147e-05,  ..., 2.1706e-03,
           2.4724e-04, 2.4967e-03]],

         [[4.3652e-01, 5.5492e-05, 8.2493e-05,  ..., 1.0071e-02,
           2.4776e-03, 2.0737e-02]],

         ...,

         [[6.8115e-01, 6.1393e-06, 7.9870e-06,  ..., 6.0455e-02,
           1.6266e-02, 1.3176e-02]],

         [[4.7461e-01, 3.8743e-06, 2.8670e-05,  ..., 7.8735e-02,
           2.4200e-02, 4.9255e-02]],

         [[1.4795e-01, 9.0599e-06, 2.1279e-05,  ..., 3.8989e-01,
           4.2822e-01, 6.1073e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8448e-02, 1.0729e-06, 1.2159e-05,  ..., 6.5765e-03,
           1.2169e-03, 3.0823e-03]],

         [[1.2405e-02, 1.1921e-06, 6.9141e-06,  ..., 6.4707e-04,
           1.7185e-03, 9.7021e-01]],

         [[1.1188e-01, 5.8174e-05, 1.6332e-04,  ..., 2.6779e-02,
           6.4850e-03, 3.6240e-04]],

         ...,

         [[2.1301e-01, 1.8477e-04, 4.5466e-04,  ..., 1.9424e-02,
           5.9242e-03, 3.1166e-03]],

         [[5.0879e-01, 1.4484e-05, 2.5809e-05,  ..., 9.8267e-02,
           1.4656e-02, 4.3091e-02]],

         [[3.9600e-01, 7.4506e-06, 2.8193e-05,  ..., 2.3645e-01,
           1.0870e-01, 5.9326e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5234e-01, 2.1183e-04, 1.3533e-03,  ..., 8.6365e-03,
           2.1420e-03, 1.9646e-03]],

         [[9.8242e-01, 4.2617e-05, 1.5819e-04,  ..., 1.7571e-04,
           1.1325e-04, 9.4593e-05]],

         [[9.0430e-01, 2.7657e-05, 2.8312e-05,  ..., 9.9468e-04,
           2.8729e-04, 7.2765e-04]],

         ...,

         [[6.7139e-01, 7.2718e-06, 1.0312e-05,  ..., 1.5305e-02,
           3.9940e-03, 6.0883e-03]],

         [[7.3291e-01, 6.8545e-06, 5.0664e-06,  ..., 8.0633e-04,
           4.3845e-04, 1.9798e-03]],

         [[8.3557e-02, 3.2187e-06, 6.9141e-06,  ..., 1.8177e-03,
           5.2185e-03, 8.4424e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4943e-02, 5.3644e-07, 2.1458e-06,  ..., 2.4634e-01,
           6.6284e-02, 1.6251e-02]],

         [[4.5630e-01, 2.5988e-05, 1.5318e-05,  ..., 4.9255e-02,
           7.4921e-03, 8.9188e-03]],

         [[1.4905e-01, 1.9324e-04, 1.9562e-04,  ..., 9.9030e-03,
           5.7268e-04, 1.1177e-03]],

         ...,

         [[3.1836e-01, 1.4067e-05, 5.3465e-05,  ..., 4.6570e-02,
           6.5536e-03, 4.1504e-03]],

         [[5.5615e-01, 6.8545e-06, 2.6524e-05,  ..., 3.5828e-02,
           2.6627e-02, 2.1225e-02]],

         [[4.6631e-01, 5.6624e-06, 1.7583e-05,  ..., 2.6520e-02,
           1.5587e-02, 3.2501e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2227e-02, 1.0431e-05, 7.4506e-06,  ..., 3.4771e-03,
           4.3845e-04, 4.3945e-03]],

         [[2.9028e-01, 4.9353e-05, 2.5988e-05,  ..., 1.1358e-03,
           5.4502e-04, 3.4313e-03]],

         [[2.0068e-01, 5.8591e-05, 7.2479e-05,  ..., 3.6602e-03,
           5.7697e-04, 2.1362e-03]],

         ...,

         [[7.0557e-02, 2.0862e-05, 3.2163e-04,  ..., 2.4002e-02,
           1.2451e-02, 7.3776e-03]],

         [[3.4448e-01, 3.1590e-05, 3.6740e-04,  ..., 2.0050e-02,
           8.4915e-03, 3.4313e-03]],

         [[9.0625e-01, 1.0550e-05, 2.0206e-05,  ..., 2.8687e-02,
           5.5580e-03, 1.6184e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0078e-01, 3.5763e-06, 8.5831e-06,  ..., 1.5991e-02,
           6.0081e-03, 6.2744e-02]],

         [[3.4106e-01, 3.2187e-06, 2.6882e-05,  ..., 7.8491e-02,
           2.7176e-02, 3.3234e-02]],

         [[4.3488e-02, 0.0000e+00, 2.9802e-07,  ..., 5.9570e-01,
           2.8589e-01, 5.8632e-03]],

         ...,

         [[5.1953e-01, 3.0756e-05, 1.1098e-04,  ..., 8.0719e-03,
           4.5280e-03, 6.0692e-03]],

         [[8.6304e-02, 8.0466e-06, 1.4067e-05,  ..., 4.9957e-02,
           4.7668e-02, 3.8300e-02]],

         [[5.2673e-02, 2.5511e-05, 6.3539e-05,  ..., 2.1118e-02,
           9.9792e-03, 2.9150e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5181e-01, 2.6822e-05, 9.1195e-05,  ..., 5.6648e-03,
           2.5730e-03, 5.5771e-03]],

         [[4.8584e-01, 5.6505e-05, 1.0425e-04,  ..., 1.0405e-03,
           4.8089e-04, 1.6232e-03]],

         [[2.5269e-01, 6.4373e-05, 8.7142e-05,  ..., 1.8196e-03,
           1.0948e-03, 1.1505e-02]],

         ...,

         [[4.0210e-01, 1.2863e-04, 2.1219e-04,  ..., 8.9722e-02,
           9.2392e-03, 3.9795e-02]],

         [[1.1438e-01, 3.9935e-05, 1.5903e-04,  ..., 4.0161e-02,
           2.3239e-02, 6.5820e-01]],

         [[8.2422e-01, 2.6345e-05, 1.2958e-04,  ..., 1.6327e-02,
           6.2904e-03, 5.5206e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6768e-01, 8.3864e-05, 2.7156e-04,  ..., 1.0605e-02,
           1.4801e-03, 3.7842e-03]],

         [[7.4463e-02, 3.4571e-05, 9.9838e-05,  ..., 1.0376e-01,
           5.1575e-02, 3.2013e-02]],

         [[6.5125e-02, 6.9678e-05, 1.8549e-04,  ..., 1.5991e-01,
           1.4023e-02, 1.6022e-02]],

         ...,

         [[7.6318e-01, 2.1160e-05, 9.3699e-05,  ..., 1.1276e-02,
           6.3744e-03, 1.4275e-02]],

         [[8.4912e-01, 8.9407e-07, 9.8586e-05,  ..., 2.9278e-04,
           3.6979e-04, 2.3384e-03]],

         [[4.6729e-01, 4.6492e-05, 4.5598e-05,  ..., 2.1347e-02,
           2.6199e-02, 4.3610e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1636e-05, 1.0788e-05, 1.0657e-01,  ..., 6.4468e-04,
           3.5057e-03, 1.6342e-02]],

         [[9.9540e-06, 1.2100e-05, 1.2646e-01,  ..., 5.1689e-04,
           2.3708e-03, 1.1765e-02]],

         [[1.3220e-04, 1.0931e-04, 1.0748e-01,  ..., 6.9542e-03,
           4.2839e-03, 1.7410e-02]],

         ...,

         [[3.9518e-05, 6.4671e-05, 1.9812e-01,  ..., 5.6335e-02,
           9.8877e-02, 1.7346e-01]],

         [[1.3709e-04, 1.0848e-04, 2.2620e-01,  ..., 3.5572e-03,
           1.6804e-03, 5.7983e-03]],

         [[2.0087e-05, 2.1160e-05, 2.2266e-01,  ..., 5.1451e-04,
           5.9891e-04, 2.8019e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8930e-05, 7.6056e-05, 2.8271e-01,  ..., 1.0281e-03,
           1.1196e-03, 4.6959e-03]],

         [[3.2663e-05, 3.1352e-05, 1.6736e-01,  ..., 1.7154e-04,
           4.5371e-04, 5.4436e-03]],

         [[1.2219e-05, 7.1526e-06, 2.7856e-01,  ..., 6.0225e-04,
           7.3528e-04, 6.2790e-03]],

         ...,

         [[7.4804e-05, 4.4763e-05, 1.7761e-01,  ..., 1.3947e-04,
           4.2605e-04, 3.2673e-03]],

         [[4.5300e-06, 6.4373e-06, 2.1167e-01,  ..., 5.6207e-05,
           1.5509e-04, 2.5425e-03]],

         [[9.3579e-06, 2.8670e-05, 2.6489e-01,  ..., 3.4785e-04,
           8.5592e-04, 6.7596e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6161e-06, 9.1791e-06, 1.6406e-01,  ..., 4.5776e-04,
           3.9339e-04, 1.4019e-03]],

         [[4.0293e-05, 2.5630e-06, 2.3120e-01,  ..., 8.4043e-06,
           2.1923e-04, 4.8180e-03]],

         [[2.0683e-05, 3.0935e-05, 2.5220e-01,  ..., 8.7082e-05,
           5.7220e-04, 3.7251e-03]],

         ...,

         [[1.7524e-05, 6.3181e-06, 4.4647e-02,  ..., 7.4883e-03,
           3.3627e-03, 6.3324e-03]],

         [[7.9870e-06, 2.2650e-05, 2.1167e-01,  ..., 3.0494e-04,
           6.0749e-04, 4.0970e-03]],

         [[3.5703e-05, 2.0981e-05, 1.4880e-01,  ..., 1.3697e-04,
           5.3549e-04, 4.4594e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8743e-05, 3.9339e-06, 1.6220e-02,  ..., 1.3696e-01,
           7.8003e-02, 1.4807e-01]],

         [[2.1935e-05, 3.2246e-05, 7.4829e-02,  ..., 4.9652e-02,
           2.5269e-02, 1.4938e-02]],

         [[2.0432e-04, 2.4915e-04, 3.1445e-01,  ..., 1.0471e-03,
           3.5858e-04, 5.5408e-04]],

         ...,

         [[2.0504e-04, 5.1260e-06, 6.7810e-02,  ..., 5.3291e-03,
           1.5678e-03, 8.5526e-03]],

         [[5.3596e-04, 8.0287e-05, 7.8613e-02,  ..., 1.6907e-01,
           3.9551e-02, 7.4158e-02]],

         [[8.5831e-06, 7.1526e-07, 2.0332e-03,  ..., 3.2166e-02,
           1.1053e-01, 1.0968e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0065e-03, 2.5463e-04, 1.4502e-01,  ..., 3.3508e-02,
           8.9340e-03, 4.6509e-02]],

         [[4.0293e-04, 3.7050e-04, 2.0874e-01,  ..., 1.2779e-02,
           3.9368e-03, 1.7960e-02]],

         [[1.5316e-03, 5.3024e-04, 2.6880e-01,  ..., 4.5624e-03,
           3.5591e-03, 2.3880e-03]],

         ...,

         [[2.9850e-04, 3.8087e-05, 2.7344e-01,  ..., 2.7939e-02,
           3.5248e-03, 4.1626e-02]],

         [[6.4611e-04, 4.9162e-04, 1.8347e-01,  ..., 9.1553e-02,
           1.2009e-02, 8.6670e-02]],

         [[1.0777e-04, 7.6711e-05, 4.4495e-02,  ..., 2.2424e-01,
           3.8971e-02, 3.2642e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[1.2875e-05, 2.3568e-04, 4.0474e-03,  ..., 8.1299e-02,
           3.7598e-01, 5.9479e-02]],

         [[1.3053e-05, 2.2888e-05, 3.5763e-04,  ..., 1.3420e-02,
           8.7524e-02, 7.3291e-01]],

         [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 3.9215e-03,
           1.7578e-02, 9.5996e-01]],

         ...,

         [[5.9605e-08, 5.9605e-08, 1.3113e-06,  ..., 1.4381e-03,
           2.8885e-02, 9.5654e-01]],

         [[9.6970e-03, 4.5815e-03, 2.6093e-03,  ..., 1.5991e-02,
           9.1095e-03, 5.5275e-03]],

         [[1.0971e-02, 6.8665e-03, 2.0730e-04,  ..., 7.3090e-03,
           4.0359e-03, 1.2047e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8280e-06, 1.5795e-05, 1.3232e-04,  ..., 1.6281e-02,
           6.4392e-02, 6.7090e-01]],

         [[0.0000e+00, 2.9802e-07, 7.0930e-06,  ..., 1.1816e-01,
           8.0713e-01, 6.4209e-02]],

         [[2.5487e-04, 1.6108e-03, 4.9591e-03,  ..., 1.6266e-02,
           1.5282e-02, 2.6810e-02]],

         ...,

         [[1.0866e-04, 2.9755e-04, 1.5841e-03,  ..., 1.7334e-01,
           3.3667e-01, 1.5417e-01]],

         [[2.1515e-03, 9.0942e-03, 2.9587e-02,  ..., 2.2507e-02,
           1.1497e-02, 9.4147e-03]],

         [[2.1439e-03, 3.1452e-03, 9.9182e-03,  ..., 9.3842e-03,
           4.7188e-03, 3.7994e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0328, 0.0029, 0.0199,  ..., 0.0210, 0.0173, 0.0349]],

         [[0.0014, 0.0006, 0.0036,  ..., 0.0620, 0.0104, 0.0025]],

         [[0.0067, 0.0016, 0.0020,  ..., 0.0090, 0.0149, 0.0094]],

         ...,

         [[0.0065, 0.0011, 0.0030,  ..., 0.0884, 0.0589, 0.2189]],

         [[0.0012, 0.0002, 0.0002,  ..., 0.0555, 0.0419, 0.0260]],

         [[0.0105, 0.0007, 0.0059,  ..., 0.0444, 0.1401, 0.1945]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[8.7354e-01, 2.9802e-07, 5.9605e-07,  ..., 4.7188e-03,
           2.1954e-03, 5.3673e-03]],

         [[8.3203e-01, 3.1590e-06, 1.6809e-05,  ..., 1.5091e-02,
           1.4771e-02, 1.7960e-02]],

         [[8.6914e-01, 8.5235e-06, 1.3769e-05,  ..., 3.3436e-03,
           5.0783e-04, 1.7441e-02]],

         ...,

         [[4.4556e-01, 2.9385e-05, 4.5419e-04,  ..., 1.1164e-04,
           1.6868e-05, 2.3438e-02]],

         [[4.0161e-01, 9.0718e-05, 2.1446e-04,  ..., 5.7716e-03,
           2.3422e-03, 4.3488e-02]],

         [[2.7539e-01, 1.4293e-04, 1.6212e-04,  ..., 1.2122e-01,
           6.6956e-02, 1.3184e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2129e-01, 1.5378e-05, 7.3314e-06,  ..., 4.3702e-04,
           4.5538e-05, 4.8279e-02]],

         [[8.2812e-01, 7.6294e-06, 1.6987e-05,  ..., 7.6233e-02,
           6.2195e-02, 2.9663e-02]],

         [[9.4971e-01, 4.1783e-05, 3.0696e-05,  ..., 8.1635e-04,
           5.6114e-03, 4.6883e-03]],

         ...,

         [[1.8872e-01, 3.4118e-04, 2.6798e-04,  ..., 1.5466e-01,
           2.3895e-02, 3.2654e-02]],

         [[4.0527e-01, 1.0653e-03, 6.5565e-04,  ..., 4.0207e-03,
           2.2430e-03, 7.1106e-03]],

         [[5.4199e-01, 1.0794e-04, 1.2672e-04,  ..., 2.8564e-02,
           2.2156e-02, 7.6172e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1877e-01, 1.0216e-04, 2.8610e-05,  ..., 2.8641e-02,
           2.9587e-02, 3.3997e-02]],

         [[7.5391e-01, 1.1581e-04, 1.1623e-05,  ..., 2.6520e-02,
           1.7807e-02, 2.0248e-02]],

         [[7.8125e-01, 2.3544e-05, 5.2452e-06,  ..., 4.7882e-02,
           9.9258e-03, 2.8885e-02]],

         ...,

         [[9.5752e-01, 4.7684e-07, 2.3842e-07,  ..., 7.1526e-06,
           1.7941e-05, 3.6926e-02]],

         [[9.6191e-01, 1.1325e-06, 4.2319e-06,  ..., 1.9073e-06,
           2.2650e-06, 1.1513e-02]],

         [[9.8633e-01, 5.9605e-08, 1.7881e-07,  ..., 1.8477e-06,
           5.3644e-06, 9.1553e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3110e-01, 1.3053e-05, 1.5497e-06,  ..., 2.0715e-01,
           1.9617e-01, 2.3544e-02]],

         [[8.7939e-01, 2.3305e-05, 4.7088e-06,  ..., 6.6221e-05,
           1.6749e-05, 6.9275e-03]],

         [[7.4854e-01, 3.2663e-05, 4.2319e-06,  ..., 3.0346e-03,
           2.8172e-03, 1.9073e-02]],

         ...,

         [[8.3008e-01, 3.2496e-04, 2.9564e-05,  ..., 1.7290e-03,
           5.6152e-03, 6.4926e-03]],

         [[2.4011e-01, 6.1393e-05, 1.4544e-05,  ..., 2.1500e-02,
           3.4698e-02, 3.2886e-01]],

         [[7.7588e-01, 8.7380e-05, 4.0174e-05,  ..., 1.2199e-02,
           3.6713e-02, 9.5398e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4863e-01, 2.2888e-05, 2.4855e-05,  ..., 3.4008e-03,
           5.8060e-03, 1.5404e-02]],

         [[8.7158e-01, 1.2219e-05, 7.6294e-06,  ..., 2.3270e-03,
           3.9978e-03, 3.5797e-02]],

         [[9.4189e-01, 1.3113e-06, 1.1921e-06,  ..., 3.9787e-03,
           3.4637e-03, 1.1398e-02]],

         ...,

         [[5.6494e-01, 6.4671e-05, 2.2590e-05,  ..., 1.9867e-02,
           1.1047e-02, 4.5135e-02]],

         [[8.1592e-01, 3.1412e-05, 9.4771e-06,  ..., 2.3834e-02,
           5.8632e-03, 8.4152e-03]],

         [[9.2188e-01, 2.8849e-05, 2.9266e-05,  ..., 6.7978e-03,
           6.1417e-03, 3.1548e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 5.6171e-04, 2.4533e-04,  ..., 3.5362e-03,
           2.4586e-03, 2.2919e-02]],

         [[9.2285e-01, 4.2319e-06, 6.4969e-06,  ..., 1.2970e-03,
           1.8082e-03, 4.4746e-03]],

         [[8.9697e-01, 2.6882e-05, 2.6882e-05,  ..., 3.8171e-04,
           2.7704e-04, 4.1351e-03]],

         ...,

         [[8.0713e-01, 6.0201e-06, 1.6809e-05,  ..., 8.0566e-03,
           1.0765e-02, 7.1960e-02]],

         [[2.8418e-01, 2.6822e-06, 5.7817e-06,  ..., 1.3794e-01,
           4.1675e-01, 2.3956e-02]],

         [[8.6719e-01, 8.8811e-06, 2.1935e-05,  ..., 2.3834e-02,
           7.8125e-03, 3.0914e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0764e-01, 1.2070e-04, 3.7026e-04,  ..., 1.8250e-02,
           7.7934e-03, 5.0316e-03]],

         [[8.0127e-01, 2.8670e-05, 1.2219e-05,  ..., 2.8320e-02,
           2.8992e-02, 1.8967e-02]],

         [[8.3887e-01, 5.9605e-06, 7.3910e-06,  ..., 6.5155e-03,
           8.5831e-03, 2.3285e-02]],

         ...,

         [[9.0430e-01, 2.4259e-05, 2.5094e-05,  ..., 7.5674e-04,
           5.0402e-04, 1.1454e-03]],

         [[7.1240e-01, 1.7285e-05, 2.3365e-05,  ..., 2.3460e-03,
           2.9335e-03, 1.4990e-01]],

         [[7.8271e-01, 6.2585e-06, 9.2983e-06,  ..., 1.7548e-02,
           1.3159e-01, 1.9943e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2656e-01, 6.0558e-05, 1.1384e-05,  ..., 3.9339e-06,
           5.9605e-06, 3.4882e-02]],

         [[8.5254e-01, 6.1393e-05, 6.2823e-05,  ..., 1.3565e-02,
           6.2218e-03, 9.3994e-03]],

         [[7.6123e-01, 1.4246e-05, 5.5432e-06,  ..., 5.5790e-04,
           8.5890e-05, 1.8478e-02]],

         ...,

         [[8.0566e-01, 1.0180e-04, 5.1796e-05,  ..., 1.1482e-02,
           9.3994e-03, 6.7902e-03]],

         [[8.4814e-01, 2.8431e-05, 1.7405e-05,  ..., 8.1396e-04,
           1.3790e-03, 3.8681e-03]],

         [[6.0205e-01, 4.8447e-04, 1.5855e-04,  ..., 8.9722e-03,
           6.0883e-03, 3.4119e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3335e-01, 2.2650e-06, 5.7220e-06,  ..., 1.7042e-03,
           5.9090e-03, 5.3906e-01]],

         [[4.1992e-01, 5.7220e-04, 6.6137e-04,  ..., 5.4121e-04,
           2.3115e-04, 1.4709e-02]],

         [[4.3823e-01, 7.1716e-04, 7.1573e-04,  ..., 2.2507e-02,
           1.1497e-02, 3.9185e-02]],

         ...,

         [[4.2261e-01, 3.3593e-04, 2.3460e-04,  ..., 2.0103e-03,
           3.1643e-03, 1.0681e-03]],

         [[6.8311e-01, 6.0499e-05, 6.9380e-05,  ..., 6.6833e-03,
           1.3985e-02, 2.2644e-02]],

         [[4.6362e-01, 3.0994e-06, 2.9087e-05,  ..., 4.8370e-03,
           8.3618e-03, 4.1309e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0078e-01, 4.4703e-06, 3.8743e-06,  ..., 1.3294e-03,
           3.1586e-03, 5.4395e-01]],

         [[4.0918e-01, 1.5306e-04, 1.7214e-04,  ..., 4.5532e-02,
           1.4490e-01, 3.1616e-02]],

         [[4.0308e-01, 2.4271e-04, 4.7982e-05,  ..., 9.9468e-04,
           1.5163e-03, 2.8412e-02]],

         ...,

         [[2.4597e-02, 5.3644e-07, 6.5565e-07,  ..., 1.3745e-04,
           1.7376e-03, 9.6338e-01]],

         [[5.8447e-01, 3.7849e-05, 2.3961e-05,  ..., 4.7112e-03,
           8.6441e-03, 1.2891e-01]],

         [[5.9570e-01, 5.1737e-05, 1.9610e-05,  ..., 3.0270e-03,
           1.2751e-03, 4.0619e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9487e-01, 2.0826e-04, 3.1376e-04,  ..., 7.7820e-03,
           7.6180e-03, 6.2408e-02]],

         [[5.0049e-01, 1.9670e-05, 2.6584e-05,  ..., 2.4646e-01,
           9.5947e-02, 1.8921e-02]],

         [[2.9004e-01, 9.7156e-06, 1.3411e-05,  ..., 1.2565e-04,
           1.0455e-04, 5.0774e-03]],

         ...,

         [[5.3760e-01, 3.2187e-06, 2.3186e-05,  ..., 7.8247e-02,
           3.2935e-01, 4.7760e-02]],

         [[4.7656e-01, 2.9635e-04, 5.5790e-04,  ..., 2.4567e-03,
           2.7046e-03, 2.9526e-03]],

         [[8.2080e-01, 4.9472e-06, 8.2850e-06,  ..., 1.7181e-02,
           6.2805e-02, 8.5999e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8955e-01, 1.8358e-05, 1.5259e-05,  ..., 8.8692e-04,
           4.2191e-03, 9.2316e-03]],

         [[1.2646e-01, 4.5514e-04, 2.0349e-04,  ..., 5.9357e-03,
           5.4665e-03, 1.0204e-03]],

         [[4.1870e-01, 1.3351e-04, 1.3065e-04,  ..., 5.7648e-02,
           5.6885e-02, 1.7487e-02]],

         ...,

         [[5.9326e-01, 1.8477e-06, 1.0848e-05,  ..., 6.9641e-02,
           1.0181e-01, 2.5742e-02]],

         [[5.9082e-01, 7.3314e-06, 3.5584e-05,  ..., 1.1749e-01,
           8.0627e-02, 8.5327e-02]],

         [[7.1436e-01, 9.5963e-06, 2.4915e-05,  ..., 1.6861e-02,
           2.4338e-02, 2.6367e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6008e-02, 2.6226e-06, 1.2517e-06,  ..., 2.1040e-04,
           2.5444e-03, 8.9014e-01]],

         [[1.0651e-01, 3.1590e-06, 8.2254e-06,  ..., 1.9971e-01,
           1.8250e-01, 1.3092e-02]],

         [[1.7029e-01, 6.9714e-04, 1.3590e-03,  ..., 2.4796e-03,
           2.0390e-03, 3.3550e-03]],

         ...,

         [[3.8354e-01, 9.7334e-05, 4.2498e-05,  ..., 1.0597e-02,
           1.0963e-02, 9.2697e-03]],

         [[1.3281e-01, 5.3644e-07, 1.7285e-06,  ..., 1.7452e-04,
           6.2752e-04, 4.8981e-03]],

         [[7.6123e-01, 1.1742e-05, 1.8597e-05,  ..., 3.2282e-04,
           1.7519e-03, 2.1271e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7178e-01, 4.2140e-05, 4.1366e-05,  ..., 1.4717e-02,
           1.0315e-02, 4.3427e-02]],

         [[2.9980e-01, 6.8188e-05, 3.3200e-05,  ..., 1.9503e-03,
           6.7253e-03, 2.5085e-02]],

         [[3.9624e-01, 2.8908e-05, 1.3471e-05,  ..., 2.6657e-02,
           9.9426e-02, 6.8054e-02]],

         ...,

         [[5.2148e-01, 1.0729e-05, 9.0599e-06,  ..., 2.1820e-03,
           1.1658e-02, 3.7628e-02]],

         [[4.9487e-01, 1.2338e-05, 1.0371e-05,  ..., 1.3013e-01,
           1.6711e-01, 3.0899e-02]],

         [[3.5034e-01, 8.9407e-06, 9.8944e-06,  ..., 1.3113e-03,
           4.8637e-03, 1.2817e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9536e-01, 9.3746e-04, 1.0815e-03,  ..., 1.4887e-03,
           5.7650e-04, 2.1591e-03]],

         [[2.2827e-01, 3.6120e-05, 6.2525e-05,  ..., 1.7989e-04,
           3.0365e-03, 3.0182e-02]],

         [[3.9355e-01, 2.9206e-06, 3.8743e-06,  ..., 2.4963e-02,
           1.6089e-01, 5.8289e-02]],

         ...,

         [[5.9375e-01, 1.0711e-04, 5.7995e-05,  ..., 5.1641e-04,
           2.6588e-03, 1.0002e-02]],

         [[9.8828e-01, 4.7684e-06, 2.8908e-05,  ..., 8.3447e-07,
           4.4703e-06, 5.5656e-03]],

         [[7.6953e-01, 6.8009e-05, 1.0473e-04,  ..., 8.6737e-04,
           8.1015e-04, 3.2043e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5537e-01, 6.9380e-05, 5.6624e-05,  ..., 5.9692e-02,
           8.5983e-03, 3.8757e-03]],

         [[1.7969e-01, 1.9610e-04, 8.4996e-05,  ..., 1.3247e-03,
           5.1346e-03, 9.8267e-03]],

         [[2.1985e-01, 3.2187e-06, 1.6093e-06,  ..., 1.9669e-02,
           1.8225e-01, 2.1814e-01]],

         ...,

         [[4.6216e-01, 1.9491e-05, 1.3053e-05,  ..., 3.4271e-02,
           2.7634e-02, 4.5563e-02]],

         [[5.9082e-01, 9.4175e-06, 8.7023e-06,  ..., 9.1858e-03,
           3.6896e-02, 1.9641e-01]],

         [[1.7371e-01, 3.5763e-06, 1.7285e-06,  ..., 9.4986e-03,
           3.5858e-02, 2.9785e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8711e-01, 1.0490e-05, 2.9802e-06,  ..., 1.4601e-03,
           4.9400e-03, 1.3641e-02]],

         [[1.4526e-01, 3.5167e-06, 6.6161e-06,  ..., 6.1493e-02,
           1.1133e-01, 2.9510e-02]],

         [[8.5754e-02, 1.8477e-06, 5.9605e-07,  ..., 8.6737e-04,
           3.0270e-03, 8.6621e-01]],

         ...,

         [[7.5073e-02, 1.1742e-05, 1.0490e-05,  ..., 5.1856e-05,
           5.2595e-04, 1.6495e-02]],

         [[4.8535e-01, 8.9347e-05, 4.0352e-05,  ..., 2.1114e-03,
           1.2146e-02, 6.9904e-04]],

         [[2.0386e-01, 8.3148e-05, 6.4135e-05,  ..., 9.4032e-04,
           3.1185e-03, 1.0635e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7271e-01, 2.1458e-05, 1.0431e-05,  ..., 2.0485e-03,
           4.7264e-03, 2.9877e-02]],

         [[9.6777e-01, 4.6432e-05, 1.9670e-06,  ..., 5.4240e-06,
           5.8413e-06, 3.0935e-05]],

         [[8.0859e-01, 6.2704e-05, 6.6757e-06,  ..., 3.1090e-04,
           3.2711e-04, 7.9727e-04]],

         ...,

         [[2.3193e-01, 5.1928e-04, 1.0414e-03,  ..., 8.7643e-04,
           1.6632e-03, 6.5804e-03]],

         [[6.4551e-01, 1.3828e-05, 1.8358e-05,  ..., 6.3610e-04,
           7.7515e-03, 3.9978e-02]],

         [[5.0732e-01, 1.0757e-03, 2.9469e-03,  ..., 9.9716e-03,
           2.4242e-03, 2.2293e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4482e-01, 1.3173e-04, 1.5521e-04,  ..., 2.1420e-03,
           3.4851e-02, 2.1420e-03]],

         [[4.3506e-01, 1.5295e-04, 2.2793e-04,  ..., 1.0155e-02,
           1.1238e-02, 2.6550e-02]],

         [[6.2549e-01, 5.0604e-05, 9.3818e-05,  ..., 2.4014e-03,
           6.1302e-03, 9.4223e-03]],

         ...,

         [[1.0614e-01, 1.0145e-04, 1.0628e-04,  ..., 5.1231e-03,
           3.6102e-02, 2.4048e-02]],

         [[3.6865e-01, 5.6624e-05, 5.8413e-05,  ..., 3.7289e-03,
           3.4271e-02, 5.9235e-02]],

         [[5.8496e-01, 3.0577e-05, 6.8903e-05,  ..., 6.0463e-04,
           4.3983e-03, 1.7120e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2754e-01, 1.4782e-05, 1.7762e-05,  ..., 2.9087e-03,
           2.6337e-02, 2.1164e-02]],

         [[2.2131e-01, 1.5235e-04, 1.6725e-04,  ..., 5.8975e-03,
           2.4826e-02, 1.7334e-02]],

         [[4.2212e-01, 3.1829e-05, 1.3459e-04,  ..., 4.5090e-03,
           1.9440e-02, 3.5547e-01]],

         ...,

         [[5.9570e-01, 1.1522e-04, 1.4281e-04,  ..., 3.5152e-03,
           1.4801e-02, 3.2318e-02]],

         [[5.4297e-01, 3.4571e-06, 1.8001e-05,  ..., 3.8242e-04,
           4.5166e-03, 9.4299e-02]],

         [[5.3418e-01, 1.4341e-04, 2.4211e-04,  ..., 1.0891e-03,
           1.1620e-02, 1.3588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7319e-01, 5.3763e-05, 3.5465e-05,  ..., 3.1071e-03,
           2.9480e-02, 4.3915e-02]],

         [[1.9434e-01, 2.1696e-05, 1.9908e-05,  ..., 5.4703e-03,
           4.0741e-02, 5.8350e-02]],

         [[6.1084e-01, 6.4313e-05, 7.5221e-05,  ..., 5.3253e-03,
           2.0737e-02, 9.6054e-03]],

         ...,

         [[6.6162e-01, 2.8610e-06, 3.4034e-05,  ..., 5.8716e-02,
           1.6992e-01, 7.9651e-02]],

         [[2.5586e-01, 8.3447e-07, 1.1146e-05,  ..., 4.1443e-02,
           1.1444e-01, 1.7822e-02]],

         [[4.4458e-01, 5.0664e-06, 7.9215e-05,  ..., 6.4026e-02,
           3.4082e-01, 5.1453e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4719e-03, 4.7684e-07, 3.3379e-06,  ..., 8.6288e-03,
           8.6487e-02, 6.1188e-03]],

         [[5.0659e-02, 7.7486e-07, 1.6093e-06,  ..., 1.6439e-04,
           1.4977e-02, 8.9111e-01]],

         [[1.7871e-01, 1.4925e-04, 4.5538e-04,  ..., 5.6343e-03,
           1.5764e-03, 7.9803e-03]],

         ...,

         [[1.9730e-02, 1.4937e-04, 2.9421e-04,  ..., 6.5660e-04,
           3.8981e-04, 4.4174e-03]],

         [[1.2744e-01, 3.0994e-06, 5.0068e-06,  ..., 4.5959e-02,
           4.2944e-01, 4.3854e-02]],

         [[1.1499e-01, 4.1723e-07, 2.1458e-06,  ..., 1.4417e-01,
           4.8413e-01, 4.6265e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7861e-01, 2.1493e-04, 9.0456e-04,  ..., 6.6032e-03,
           3.1013e-03, 4.1901e-02]],

         [[9.8242e-01, 9.1672e-05, 7.7546e-05,  ..., 1.2803e-04,
           1.1414e-04, 8.5163e-04]],

         [[9.2773e-01, 5.7995e-05, 2.5511e-05,  ..., 6.1655e-04,
           1.1120e-03, 8.6746e-03]],

         ...,

         [[5.6445e-01, 4.8280e-06, 8.0466e-06,  ..., 1.9608e-02,
           6.9519e-02, 2.2095e-01]],

         [[6.1572e-01, 2.5451e-05, 2.0921e-05,  ..., 4.5109e-04,
           2.3079e-03, 3.8967e-03]],

         [[1.5686e-01, 9.4175e-06, 1.2279e-05,  ..., 7.1526e-04,
           7.4539e-03, 7.8467e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8013e-02, 4.1723e-07, 1.0729e-06,  ..., 8.9417e-02,
           5.4657e-02, 1.3110e-01]],

         [[4.4580e-01, 2.0266e-06, 1.9073e-06,  ..., 6.3721e-02,
           6.2225e-02, 1.8152e-01]],

         [[1.9214e-01, 3.8505e-05, 6.0976e-05,  ..., 4.6616e-03,
           6.7329e-03, 2.2247e-02]],

         ...,

         [[1.3135e-01, 3.1590e-06, 1.2994e-05,  ..., 8.5449e-02,
           4.5380e-02, 1.9556e-01]],

         [[4.5850e-01, 3.3319e-05, 1.1122e-04,  ..., 1.0132e-02,
           1.4282e-02, 2.0950e-02]],

         [[8.0420e-01, 7.0333e-06, 2.1160e-05,  ..., 4.7798e-03,
           1.5076e-02, 1.4839e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1997e-01, 2.2769e-05, 1.7941e-05,  ..., 1.2531e-03,
           1.3527e-02, 2.3926e-02]],

         [[2.6978e-01, 7.5459e-05, 3.6836e-05,  ..., 4.2343e-04,
           3.6583e-03, 1.0300e-02]],

         [[3.4888e-01, 5.5730e-05, 9.4771e-05,  ..., 6.6280e-04,
           2.3956e-03, 3.8513e-02]],

         ...,

         [[5.0171e-02, 2.6345e-05, 2.4533e-04,  ..., 2.0111e-02,
           4.0550e-03, 7.5150e-03]],

         [[1.9421e-01, 1.1802e-04, 4.8518e-04,  ..., 3.7575e-03,
           8.3208e-04, 3.2139e-03]],

         [[4.8981e-02, 1.3590e-05, 2.0921e-05,  ..., 3.6407e-02,
           1.1452e-02, 2.8336e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 7.1526e-07, 3.3975e-06,  ..., 1.0406e-02,
           9.5154e-02, 2.6270e-01]],

         [[1.9617e-01, 6.5565e-07, 5.6028e-06,  ..., 3.3966e-02,
           8.8806e-02, 4.2755e-02]],

         [[5.0391e-01, 0.0000e+00, 3.5763e-07,  ..., 4.9133e-02,
           2.2278e-01, 1.9055e-01]],

         ...,

         [[8.0127e-01, 4.1366e-05, 7.5519e-05,  ..., 1.4868e-03,
           2.2507e-03, 2.1477e-03]],

         [[3.3447e-01, 1.0014e-05, 1.5974e-05,  ..., 1.5900e-02,
           1.7908e-01, 4.4922e-02]],

         [[6.5430e-02, 1.8954e-05, 2.7418e-05,  ..., 3.7193e-03,
           9.5367e-03, 5.0244e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3667e-01, 3.1233e-05, 8.3208e-05,  ..., 4.0436e-03,
           6.9313e-03, 3.7781e-02]],

         [[3.8354e-01, 1.7333e-04, 4.7159e-04,  ..., 2.1398e-04,
           7.7581e-04, 2.7828e-03]],

         [[4.1504e-01, 1.7667e-04, 2.0456e-04,  ..., 1.3313e-03,
           9.4604e-03, 1.3031e-02]],

         ...,

         [[2.6978e-01, 4.2081e-05, 1.3280e-04,  ..., 1.4755e-02,
           4.4067e-02, 3.5205e-01]],

         [[2.2510e-01, 6.0678e-05, 4.4489e-04,  ..., 1.1383e-02,
           3.9124e-02, 4.7656e-01]],

         [[1.8738e-01, 4.9472e-06, 1.5795e-05,  ..., 3.3081e-02,
           5.9521e-01, 3.9886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1494e-01, 1.5616e-05, 6.2287e-05,  ..., 1.3130e-02,
           3.8745e-01, 4.9835e-02]],

         [[3.9624e-01, 2.6047e-05, 1.3328e-04,  ..., 2.6230e-02,
           1.1176e-01, 1.2610e-01]],

         [[5.0195e-01, 3.3855e-04, 1.8110e-03,  ..., 3.9253e-03,
           1.2444e-02, 2.5238e-02]],

         ...,

         [[3.1470e-01, 9.5427e-05, 5.1260e-04,  ..., 5.3833e-02,
           7.2754e-02, 3.0308e-03]],

         [[8.7012e-01, 2.9922e-05, 2.4748e-04,  ..., 9.1851e-05,
           3.2187e-04, 3.3569e-03]],

         [[3.0420e-01, 1.2577e-05, 5.4538e-05,  ..., 3.7933e-02,
           2.7756e-02, 2.6989e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8670e-05, 3.3200e-05, 1.3391e-01,  ..., 2.0523e-02,
           6.5735e-02, 3.4046e-03]],

         [[7.7069e-05, 1.2469e-04, 1.2323e-01,  ..., 1.8906e-02,
           6.6467e-02, 6.0883e-03]],

         [[9.1732e-05, 1.8501e-04, 8.8989e-02,  ..., 9.5596e-03,
           3.5248e-02, 3.0624e-02]],

         ...,

         [[2.9147e-05, 4.1783e-05, 6.6406e-02,  ..., 4.4922e-02,
           5.4736e-01, 5.8136e-02]],

         [[6.9094e-04, 1.9073e-03, 1.7371e-01,  ..., 2.4986e-03,
           5.2261e-03, 4.5052e-03]],

         [[3.2473e-04, 8.8978e-04, 2.9517e-01,  ..., 4.5609e-04,
           1.9436e-03, 3.7174e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2972e-04, 1.4317e-04, 2.9126e-01,  ..., 1.5516e-03,
           4.5624e-03, 7.2718e-04]],

         [[3.0899e-04, 2.8706e-04, 2.2668e-01,  ..., 9.7084e-04,
           5.8098e-03, 2.8586e-04]],

         [[2.2507e-04, 9.1434e-05, 3.0591e-01,  ..., 8.4591e-04,
           4.1466e-03, 2.7156e-04]],

         ...,

         [[3.5501e-04, 6.8808e-04, 1.1261e-01,  ..., 7.3373e-05,
           6.1798e-04, 2.4819e-04]],

         [[1.9670e-05, 5.0426e-05, 2.6367e-01,  ..., 3.5834e-04,
           2.8515e-03, 5.2738e-04]],

         [[2.3782e-05, 7.9036e-05, 1.0681e-01,  ..., 2.0134e-04,
           9.6083e-04, 1.4877e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5320e-04, 1.3912e-04, 2.4133e-01,  ..., 4.5967e-04,
           9.0122e-04, 6.1750e-04]],

         [[6.5517e-04, 2.8491e-05, 2.9028e-01,  ..., 2.4676e-04,
           3.0537e-03, 7.1406e-05]],

         [[5.4789e-04, 3.7599e-04, 3.2617e-01,  ..., 4.0722e-04,
           2.1381e-03, 2.2233e-04]],

         ...,

         [[7.8857e-05, 1.0502e-04, 7.5500e-02,  ..., 7.5951e-03,
           1.1360e-02, 6.9580e-02]],

         [[8.1003e-05, 1.9407e-04, 2.7710e-01,  ..., 8.9359e-04,
           4.7112e-03, 4.2391e-04]],

         [[5.2500e-04, 3.2592e-04, 1.9507e-01,  ..., 7.2575e-04,
           4.3755e-03, 1.3561e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8387e-03, 7.5758e-05, 1.8860e-02,  ..., 4.1840e-02,
           1.2396e-01, 1.6553e-01]],

         [[2.1482e-04, 1.8466e-04, 1.4148e-01,  ..., 7.8430e-03,
           1.6251e-02, 1.1528e-02]],

         [[1.1225e-03, 2.2049e-03, 2.0361e-01,  ..., 5.5647e-04,
           3.7074e-04, 1.3481e-02]],

         ...,

         [[6.1703e-04, 6.0499e-05, 1.4819e-01,  ..., 2.6112e-03,
           1.6052e-02, 8.4473e-02]],

         [[1.1110e-03, 9.0063e-05, 7.3730e-02,  ..., 5.3596e-03,
           1.8896e-01, 2.4268e-01]],

         [[2.6989e-04, 1.5795e-05, 1.3031e-02,  ..., 1.1368e-02,
           8.7708e-02, 1.8640e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7700e-03, 5.0402e-04, 2.1838e-01,  ..., 2.1057e-03,
           1.7586e-03, 4.6967e-02]],

         [[2.3615e-04, 6.8474e-04, 2.5806e-01,  ..., 1.8978e-03,
           2.9049e-03, 1.3618e-02]],

         [[1.3494e-03, 8.2874e-04, 2.0374e-01,  ..., 2.2858e-02,
           1.5552e-01, 5.0110e-02]],

         ...,

         [[3.8910e-04, 9.6083e-05, 1.9080e-01,  ..., 4.3831e-03,
           1.4259e-02, 8.7341e-02]],

         [[3.7265e-04, 5.7268e-04, 1.2659e-01,  ..., 7.1983e-03,
           6.3660e-02, 3.0371e-01]],

         [[1.1444e-05, 5.6028e-06, 2.8885e-02,  ..., 4.1687e-02,
           5.7007e-02, 2.0691e-01]]]], device='cuda:0', dtype=torch.float16)), (tensor([[[[8.3447e-07, 3.2783e-06, 3.7742e-04,  ..., 1.1139e-01,
           1.4758e-01, 6.0205e-01]],

         [[2.8801e-03, 4.1509e-04, 2.7924e-03,  ..., 1.1871e-01,
           2.6077e-02, 1.9580e-01]],

         [[1.5497e-06, 4.1723e-06, 5.8365e-03,  ..., 7.1106e-02,
           3.0637e-05, 4.0894e-01]],

         ...,

         [[2.0390e-03, 1.2360e-03, 1.2360e-03,  ..., 4.0955e-02,
           1.1017e-02, 5.2071e-03]],

         [[4.7913e-03, 3.7308e-03, 1.8951e-02,  ..., 4.2267e-03,
           5.1003e-03, 1.7969e-01]],

         [[4.9248e-03, 3.9558e-03, 9.0182e-05,  ..., 2.7466e-02,
           2.4246e-02, 6.9466e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8532e-05, 2.5439e-04, 8.8787e-04,  ..., 4.2786e-02,
           6.5247e-02, 8.3740e-02]],

         [[0.0000e+00, 1.7881e-07, 9.4771e-06,  ..., 1.1157e-01,
           5.3223e-01, 3.1299e-01]],

         [[4.7798e-03, 6.7406e-03, 2.5040e-02,  ..., 2.4414e-03,
           2.8992e-03, 6.7406e-03]],

         ...,

         [[3.6931e-04, 7.3433e-04, 6.6328e-04,  ..., 2.7065e-03,
           6.8588e-03, 3.7659e-02]],

         [[3.3264e-03, 4.6921e-03, 4.9255e-02,  ..., 5.0507e-03,
           1.5469e-03, 1.5083e-02]],

         [[5.7554e-04, 4.3221e-03, 3.8433e-03,  ..., 1.9073e-02,
           4.9835e-02, 1.5845e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1648e-04, 9.4950e-05, 1.0306e-04,  ..., 1.6003e-01,
           2.4976e-01, 1.2177e-01]],

         [[2.5272e-03, 1.5879e-03, 3.7346e-03,  ..., 1.4153e-02,
           4.6844e-03, 5.0659e-03]],

         [[1.7672e-03, 2.7084e-04, 2.3627e-04,  ..., 8.0719e-03,
           5.5923e-03, 3.8818e-02]],

         ...,

         [[3.0994e-06, 1.3113e-06, 5.6839e-04,  ..., 1.6876e-02,
           1.7444e-01, 6.5332e-01]],

         [[5.2757e-03, 7.4816e-04, 2.4929e-03,  ..., 8.8348e-03,
           6.5651e-03, 4.2816e-02]],

         [[9.0256e-03, 8.9025e-04, 7.7209e-03,  ..., 9.6130e-03,
           2.8473e-02, 7.3242e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8320e-01, 9.5367e-07, 8.5235e-06,  ..., 4.2081e-04,
           6.4926e-03, 1.1177e-02]],

         [[8.9551e-01, 3.5763e-06, 5.4836e-06,  ..., 1.1482e-02,
           1.1375e-02, 1.5305e-02]],

         [[6.1475e-01, 2.0862e-06, 3.9935e-06,  ..., 3.7289e-04,
           4.3068e-03, 1.5640e-02]],

         ...,

         [[8.2422e-01, 2.3508e-04, 1.1349e-03,  ..., 8.4281e-05,
           9.4461e-04, 2.0676e-03]],

         [[1.3074e-01, 5.1022e-05, 1.4484e-04,  ..., 3.3051e-02,
           6.2500e-02, 3.9886e-02]],

         [[8.5010e-01, 1.6212e-04, 5.9795e-04,  ..., 1.2808e-03,
           5.7869e-03, 6.7253e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7417e-01, 6.1989e-06, 1.7285e-06,  ..., 2.2411e-04,
           1.0103e-04, 3.5095e-03]],

         [[9.6045e-01, 0.0000e+00, 4.7684e-07,  ..., 1.3332e-03,
           2.2278e-02, 1.3611e-02]],

         [[7.9932e-01, 1.5140e-04, 7.6413e-05,  ..., 3.2501e-03,
           2.1744e-02, 1.0490e-02]],

         ...,

         [[3.7524e-01, 1.9765e-04, 1.7536e-04,  ..., 1.1765e-02,
           2.3758e-02, 2.1210e-02]],

         [[1.0999e-01, 3.0632e-03, 2.8610e-03,  ..., 8.6784e-05,
           6.9761e-04, 2.8038e-03]],

         [[7.1338e-01, 5.8413e-05, 1.1170e-04,  ..., 1.9531e-02,
           8.9417e-02, 8.6670e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1558e-01, 1.8740e-04, 5.5194e-05,  ..., 1.5015e-02,
           2.7832e-02, 2.8168e-02]],

         [[8.1396e-01, 1.0496e-04, 8.1658e-06,  ..., 2.9392e-03,
           7.6370e-03, 2.2842e-02]],

         [[5.5518e-01, 1.6904e-04, 2.7835e-05,  ..., 6.0883e-03,
           3.0350e-02, 8.0032e-03]],

         ...,

         [[7.2314e-01, 2.4915e-05, 4.7684e-07,  ..., 3.9244e-04,
           6.1572e-05, 2.6855e-01]],

         [[9.1064e-01, 1.2171e-04, 6.8545e-06,  ..., 1.6034e-05,
           1.6689e-06, 5.1117e-02]],

         [[9.5654e-01, 2.0742e-05, 6.5565e-07,  ..., 8.3447e-06,
           2.3842e-07, 3.7628e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0747e-01, 5.5432e-06, 2.9802e-07,  ..., 5.1697e-02,
           2.0752e-01, 1.8875e-02]],

         [[6.4355e-01, 7.8678e-06, 1.2517e-06,  ..., 2.9564e-04,
           1.2617e-03, 2.4548e-03]],

         [[7.2754e-01, 4.4680e-04, 6.7949e-05,  ..., 4.0283e-03,
           2.9469e-03, 3.1830e-02]],

         ...,

         [[7.7588e-01, 1.2743e-04, 1.9252e-05,  ..., 5.9166e-03,
           3.4094e-04, 7.4005e-03]],

         [[7.7197e-01, 4.4823e-05, 3.4213e-05,  ..., 1.5373e-02,
           6.5422e-03, 2.4567e-02]],

         [[8.8037e-01, 8.1182e-05, 1.2338e-04,  ..., 1.1299e-02,
           8.5220e-03, 1.0521e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2944e-01, 6.3777e-06, 6.0201e-06,  ..., 2.6245e-02,
           5.7800e-02, 7.6050e-02]],

         [[7.2607e-01, 2.9504e-05, 2.9206e-06,  ..., 6.0349e-03,
           8.7280e-03, 4.4067e-02]],

         [[4.9487e-01, 7.1526e-07, 7.1526e-07,  ..., 2.9755e-02,
           8.4106e-02, 3.7506e-02]],

         ...,

         [[1.1859e-01, 2.0504e-05, 1.0312e-05,  ..., 3.9917e-02,
           1.7188e-01, 5.8365e-03]],

         [[5.4297e-01, 5.6171e-04, 8.8871e-05,  ..., 9.7046e-03,
           1.5121e-02, 1.0742e-02]],

         [[9.1980e-02, 6.4373e-06, 6.4373e-06,  ..., 1.6388e-02,
           7.8583e-03, 3.7842e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8838e-01, 2.1133e-03, 2.5482e-03,  ..., 3.4618e-04,
           1.6651e-03, 1.7242e-03]],

         [[6.0400e-01, 2.3782e-05, 3.2485e-05,  ..., 7.2823e-03,
           2.0332e-03, 1.1101e-02]],

         [[4.5898e-01, 1.1134e-04, 4.2439e-05,  ..., 6.6757e-03,
           1.3329e-02, 1.3351e-02]],

         ...,

         [[7.7783e-01, 9.3579e-05, 6.5863e-05,  ..., 3.1590e-04,
           1.0771e-04, 9.2468e-02]],

         [[5.6055e-01, 5.9843e-05, 7.6890e-05,  ..., 4.4891e-02,
           8.0338e-03, 5.9052e-03]],

         [[9.7656e-01, 5.4836e-05, 2.7716e-05,  ..., 2.8181e-04,
           1.0490e-03, 2.8300e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6309e-01, 5.7793e-04, 1.3332e-03,  ..., 1.3149e-04,
           2.1434e-04, 4.2953e-03]],

         [[9.2725e-01, 3.9935e-06, 1.5736e-05,  ..., 5.8670e-03,
           1.3641e-02, 5.3635e-03]],

         [[3.4766e-01, 9.8944e-06, 7.5936e-05,  ..., 5.2887e-02,
           2.8915e-02, 2.0050e-02]],

         ...,

         [[2.6871e-02, 1.0169e-04, 3.0971e-04,  ..., 9.0003e-06,
           2.2888e-05, 2.4414e-03]],

         [[8.9941e-01, 5.6922e-05, 5.9545e-05,  ..., 2.2087e-03,
           2.0950e-02, 4.9171e-03]],

         [[8.4375e-01, 2.1815e-05, 3.8147e-05,  ..., 3.0853e-02,
           5.9433e-03, 1.3596e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7444e-01, 9.7811e-05, 2.6178e-04,  ..., 8.0168e-05,
           2.7478e-05, 1.8875e-02]],

         [[3.1860e-01, 6.2227e-05, 6.0320e-05,  ..., 3.9177e-03,
           3.7308e-03, 3.9093e-02]],

         [[4.6729e-01, 2.8014e-06, 1.1802e-05,  ..., 3.4976e-04,
           2.9564e-04, 2.4765e-02]],

         ...,

         [[6.8945e-01, 1.6272e-04, 5.0604e-05,  ..., 1.1833e-02,
           1.3588e-02, 6.5842e-03]],

         [[3.4253e-01, 3.7670e-04, 1.7118e-04,  ..., 4.9324e-03,
           3.7899e-03, 1.8219e-02]],

         [[2.3206e-01, 1.4029e-03, 5.8746e-04,  ..., 9.0122e-04,
           1.1206e-03, 1.7639e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1191e-01, 5.8413e-06, 1.2577e-05,  ..., 9.1505e-04,
           9.4318e-04, 7.3975e-01]],

         [[6.3818e-01, 1.9913e-03, 3.4065e-03,  ..., 3.1531e-05,
           3.4943e-03, 1.1015e-03]],

         [[4.5093e-01, 5.7888e-04, 6.3372e-04,  ..., 9.4910e-03,
           3.3325e-02, 5.0812e-02]],

         ...,

         [[4.5288e-01, 1.1259e-04, 1.9228e-04,  ..., 1.8280e-02,
           1.0742e-02, 6.2378e-02]],

         [[5.2588e-01, 1.5616e-05, 1.8835e-05,  ..., 1.6449e-02,
           1.2756e-01, 2.1027e-02]],

         [[4.5557e-01, 1.6034e-05, 4.3929e-05,  ..., 6.4278e-03,
           5.9471e-03, 4.0234e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2388e-01, 1.1861e-04, 1.6403e-04,  ..., 5.3902e-03,
           7.9117e-03, 5.3271e-01]],

         [[8.5059e-01, 6.1095e-05, 1.3149e-04,  ..., 3.5114e-03,
           1.7181e-02, 6.8359e-03]],

         [[8.0273e-01, 2.3782e-04, 1.2732e-04,  ..., 4.5319e-03,
           1.7929e-02, 2.1057e-02]],

         ...,

         [[2.2620e-01, 1.5318e-05, 1.2517e-05,  ..., 4.6372e-04,
           6.3896e-04, 7.1582e-01]],

         [[4.4824e-01, 1.7679e-04, 1.5247e-04,  ..., 1.0157e-03,
           2.0962e-03, 2.5101e-02]],

         [[7.4121e-01, 5.3883e-05, 2.7418e-05,  ..., 5.1270e-03,
           8.2092e-02, 5.4703e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6538e-01, 7.7367e-05, 8.1420e-05,  ..., 2.7588e-02,
           3.2153e-01, 4.1412e-02]],

         [[6.8408e-01, 1.5080e-05, 1.8239e-05,  ..., 1.7410e-02,
           7.2365e-03, 3.0884e-02]],

         [[4.1333e-01, 3.5405e-05, 5.1320e-05,  ..., 1.3733e-04,
           3.1304e-04, 1.3824e-02]],

         ...,

         [[9.3066e-01, 0.0000e+00, 7.7486e-07,  ..., 3.4904e-03,
           4.3640e-02, 2.0325e-02]],

         [[5.7959e-01, 9.3317e-04, 1.2732e-03,  ..., 1.6766e-03,
           3.0651e-03, 6.9008e-03]],

         [[6.4062e-01, 3.5763e-07, 2.6822e-06,  ..., 2.0599e-02,
           2.4500e-01, 7.0557e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6416e-01, 6.7890e-05, 3.8981e-05,  ..., 6.6948e-03,
           2.6825e-02, 9.9945e-03]],

         [[3.1445e-01, 1.2941e-03, 5.7459e-04,  ..., 1.6963e-04,
           1.9562e-04, 6.4514e-02]],

         [[8.5791e-01, 2.6298e-04, 2.3448e-04,  ..., 5.1022e-04,
           3.1996e-04, 1.2985e-02]],

         ...,

         [[6.8115e-01, 3.2783e-06, 1.2517e-05,  ..., 7.3776e-03,
           4.1565e-02, 3.8574e-02]],

         [[7.3096e-01, 1.0490e-05, 2.9862e-05,  ..., 4.7531e-03,
           3.1128e-02, 3.3783e-02]],

         [[3.1836e-01, 1.2732e-04, 3.3808e-04,  ..., 6.3896e-04,
           5.7945e-03, 4.9347e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4575e-02, 5.6624e-06, 3.0994e-06,  ..., 1.9646e-04,
           5.8937e-04, 9.0869e-01]],

         [[7.6562e-01, 1.3649e-05, 2.3007e-05,  ..., 1.0818e-02,
           9.9182e-03, 5.5054e-02]],

         [[1.6052e-01, 3.0160e-04, 2.6631e-04,  ..., 1.4412e-04,
           3.6907e-04, 1.0773e-02]],

         ...,

         [[4.2773e-01, 2.2590e-05, 3.8266e-05,  ..., 2.6684e-03,
           6.8474e-03, 5.1666e-02]],

         [[4.6606e-01, 8.9407e-07, 1.7881e-06,  ..., 6.1226e-04,
           2.9850e-03, 2.2293e-02]],

         [[3.6084e-01, 2.1935e-05, 1.4722e-05,  ..., 2.7039e-02,
           1.0541e-01, 2.7512e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.5154e-02, 1.1384e-05, 5.4836e-06,  ..., 6.0616e-03,
           2.6685e-01, 2.3697e-02]],

         [[3.4131e-01, 1.6916e-04, 1.1569e-04,  ..., 9.8515e-04,
           4.3297e-03, 4.1718e-02]],

         [[8.7952e-02, 2.3842e-05, 1.0788e-05,  ..., 2.4429e-02,
           9.9670e-02, 5.0507e-02]],

         ...,

         [[3.7109e-01, 5.6028e-06, 3.3975e-06,  ..., 2.4200e-02,
           5.9418e-02, 5.8960e-02]],

         [[5.1318e-01, 1.4126e-05, 2.1100e-05,  ..., 1.1559e-02,
           2.4857e-02, 2.2278e-02]],

         [[5.3467e-01, 5.4419e-05, 6.0380e-05,  ..., 2.4090e-03,
           2.9812e-03, 3.5828e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1096e-01, 1.6346e-03, 1.6031e-03,  ..., 9.6560e-06,
           1.1748e-04, 6.1874e-03]],

         [[1.7773e-01, 8.7309e-04, 6.7806e-04,  ..., 2.6202e-04,
           2.4166e-03, 2.6520e-02]],

         [[2.3865e-01, 7.1526e-07, 1.0133e-06,  ..., 8.1024e-03,
           4.6265e-02, 2.7725e-02]],

         ...,

         [[6.2012e-01, 4.3511e-05, 9.5189e-05,  ..., 9.9850e-04,
           1.2760e-03, 4.9896e-02]],

         [[9.8682e-01, 1.7524e-05, 7.2896e-05,  ..., 1.3113e-05,
           7.0333e-06, 2.1744e-03]],

         [[6.1914e-01, 1.6952e-04, 2.2221e-04,  ..., 9.1648e-04,
           5.8413e-04, 1.7151e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9639e-01, 4.5609e-04, 1.0042e-03,  ..., 7.6234e-05,
           3.6597e-05, 6.6162e-02]],

         [[3.2837e-01, 3.2196e-03, 2.1610e-03,  ..., 1.7071e-04,
           5.5075e-04, 4.6936e-02]],

         [[3.5858e-02, 1.4484e-05, 8.7023e-06,  ..., 6.1340e-03,
           4.0150e-04, 7.5488e-01]],

         ...,

         [[1.3843e-01, 1.3888e-05, 1.3888e-05,  ..., 4.8780e-04,
           2.8629e-03, 1.7471e-02]],

         [[3.9795e-02, 5.9605e-07, 1.7881e-07,  ..., 7.4196e-03,
           4.0210e-01, 1.2924e-02]],

         [[1.2537e-01, 1.3411e-05, 6.6161e-06,  ..., 5.5542e-03,
           2.6291e-02, 3.1274e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9761e-01, 5.9605e-05, 3.9697e-05,  ..., 1.6365e-03,
           5.0430e-03, 6.1432e-02]],

         [[1.1493e-01, 9.4771e-06, 1.5080e-05,  ..., 5.3749e-03,
           5.3749e-03, 3.2928e-02]],

         [[1.0785e-01, 1.0669e-05, 2.6822e-06,  ..., 7.3814e-04,
           7.3395e-03, 8.2227e-01]],

         ...,

         [[2.6025e-01, 9.2363e-04, 6.5517e-04,  ..., 1.6727e-03,
           3.1013e-03, 2.9648e-02]],

         [[5.4541e-01, 5.2929e-04, 4.1223e-04,  ..., 9.1791e-04,
           3.5810e-04, 2.1687e-03]],

         [[3.0347e-01, 1.0443e-03, 1.5373e-03,  ..., 4.0746e-04,
           8.1015e-04, 4.4670e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7954e-01, 5.2989e-05, 5.2571e-05,  ..., 1.6613e-03,
           4.8065e-03, 4.2847e-02]],

         [[9.5898e-01, 2.1040e-05, 1.9670e-06,  ..., 2.3842e-06,
           6.5565e-06, 1.3132e-03]],

         [[7.4023e-01, 5.5611e-05, 1.6034e-05,  ..., 9.5367e-05,
           1.6034e-04, 5.1460e-03]],

         ...,

         [[6.4270e-02, 1.1768e-03, 1.7662e-03,  ..., 2.4438e-05,
           1.5438e-04, 3.1799e-02]],

         [[1.1377e-01, 9.3579e-06, 1.3292e-05,  ..., 6.6223e-03,
           6.7932e-02, 2.4612e-02]],

         [[2.5073e-01, 1.6756e-03, 6.0806e-03,  ..., 1.0986e-03,
           1.3077e-02, 7.8857e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6123e-01, 1.2457e-04, 2.5558e-04,  ..., 2.7485e-03,
           2.2774e-03, 3.1952e-02]],

         [[4.3311e-01, 4.3035e-04, 2.0695e-03,  ..., 1.2846e-03,
           1.4334e-03, 3.2898e-02]],

         [[4.8926e-01, 9.5010e-05, 3.2401e-04,  ..., 9.8228e-04,
           9.9754e-04, 1.1597e-02]],

         ...,

         [[4.3640e-03, 1.9252e-05, 5.4955e-05,  ..., 1.2767e-04,
           1.2970e-04, 9.3842e-03]],

         [[3.8647e-01, 2.5392e-04, 3.1590e-04,  ..., 5.9624e-03,
           5.9624e-03, 4.7638e-02]],

         [[6.6064e-01, 6.0558e-05, 1.1492e-04,  ..., 1.3361e-03,
           6.3744e-03, 9.7198e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8408e-01, 3.5584e-05, 6.3241e-05,  ..., 3.8815e-03,
           2.0020e-02, 8.4290e-02]],

         [[4.4678e-01, 2.6298e-04, 6.2370e-04,  ..., 2.7828e-03,
           1.1185e-02, 3.6652e-02]],

         [[2.3389e-01, 6.1095e-05, 1.6487e-04,  ..., 2.7027e-03,
           3.1403e-02, 4.1382e-01]],

         ...,

         [[3.8086e-01, 8.3447e-05, 7.6294e-05,  ..., 1.0475e-02,
           2.7588e-02, 5.4016e-02]],

         [[5.6104e-01, 6.4790e-05, 2.0027e-04,  ..., 6.0844e-03,
           4.1199e-03, 1.9653e-02]],

         [[5.0244e-01, 6.5374e-04, 1.2751e-03,  ..., 9.1324e-03,
           4.9248e-03, 1.0109e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4575e-01, 9.2983e-05, 6.4790e-05,  ..., 8.5831e-03,
           8.9111e-02, 6.2683e-02]],

         [[4.1138e-02, 1.4305e-05, 1.4842e-05,  ..., 1.3752e-03,
           1.4191e-03, 1.1162e-02]],

         [[2.8247e-01, 8.8513e-05, 1.5330e-04,  ..., 6.2180e-03,
           7.1869e-03, 1.3527e-02]],

         ...,

         [[3.1665e-01, 0.0000e+00, 4.6492e-06,  ..., 2.7023e-02,
           5.1416e-01, 9.8083e-02]],

         [[4.4458e-01, 8.4639e-06, 6.3181e-05,  ..., 3.1147e-03,
           4.1924e-03, 5.3101e-02]],

         [[8.5791e-01, 2.3842e-07, 9.4771e-06,  ..., 8.2169e-03,
           5.7465e-02, 5.3131e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2527e-02, 3.5763e-06, 2.6762e-05,  ..., 1.2751e-03,
           3.9406e-03, 1.6248e-01]],

         [[4.0039e-02, 1.1563e-05, 1.1516e-04,  ..., 1.8587e-03,
           3.7556e-03, 9.0430e-01]],

         [[3.0933e-01, 2.1422e-04, 1.6432e-03,  ..., 2.2292e-05,
           7.0524e-04, 7.0679e-02]],

         ...,

         [[1.4429e-01, 7.0286e-04, 1.6317e-03,  ..., 1.4937e-04,
           2.8191e-03, 7.9712e-02]],

         [[2.3511e-01, 2.5034e-06, 9.2983e-06,  ..., 5.7487e-03,
           7.8888e-03, 5.1849e-02]],

         [[1.6040e-01, 1.1921e-06, 3.2783e-06,  ..., 6.5796e-02,
           3.2275e-01, 1.2390e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8213e-01, 2.0528e-04, 5.9557e-04,  ..., 2.9030e-03,
           2.4460e-02, 1.8677e-02]],

         [[9.8047e-01, 3.5024e-04, 6.1941e-04,  ..., 4.2260e-05,
           2.2960e-04, 3.8171e-04]],

         [[7.6904e-01, 3.8123e-04, 1.3514e-03,  ..., 1.2693e-03,
           7.6561e-03, 4.5891e-03]],

         ...,

         [[2.8824e-02, 6.5565e-07, 2.4438e-06,  ..., 8.2550e-03,
           1.0876e-01, 2.9739e-02]],

         [[4.5459e-01, 2.5690e-05, 4.4405e-05,  ..., 1.7052e-03,
           2.5997e-03, 2.0733e-03]],

         [[7.5500e-02, 3.0816e-05, 6.6698e-05,  ..., 4.7779e-04,
           2.2793e-03, 8.6377e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3364e-02, 1.4901e-06, 8.9407e-06,  ..., 6.1417e-03,
           3.7323e-02, 1.7883e-01]],

         [[2.9236e-02, 2.3842e-07, 4.1723e-07,  ..., 3.3142e-02,
           2.2119e-01, 4.4586e-02]],

         [[1.9397e-01, 2.8908e-05, 6.2048e-05,  ..., 6.2828e-03,
           3.4790e-02, 2.5345e-02]],

         ...,

         [[1.6748e-01, 1.1563e-05, 1.2612e-04,  ..., 1.3428e-02,
           9.7656e-02, 7.6050e-02]],

         [[3.1641e-01, 5.5432e-06, 1.4055e-04,  ..., 1.9592e-02,
           5.3177e-03, 2.8076e-02]],

         [[5.1611e-01, 1.3053e-05, 8.7798e-05,  ..., 2.1988e-02,
           1.4305e-02, 7.8552e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9346e-03, 2.3842e-06, 3.5763e-06,  ..., 7.5006e-04,
           3.6693e-04, 2.2564e-03]],

         [[8.4534e-02, 5.4836e-06, 4.2319e-06,  ..., 3.2711e-04,
           1.4019e-04, 2.8419e-04]],

         [[8.9539e-02, 4.7803e-05, 2.2888e-04,  ..., 2.6608e-03,
           1.9669e-02, 4.0680e-02]],

         ...,

         [[1.9763e-01, 2.1076e-04, 7.4720e-04,  ..., 3.2711e-03,
           7.0343e-03, 1.4233e-01]],

         [[1.3855e-01, 1.4925e-03, 4.6692e-03,  ..., 5.7983e-04,
           1.7443e-03, 6.1493e-02]],

         [[1.2207e-01, 2.7776e-04, 7.4911e-04,  ..., 1.8682e-03,
           4.4220e-02, 1.9666e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1562e-01, 5.6028e-06, 1.0252e-05,  ..., 1.9608e-02,
           1.7822e-01, 2.3941e-02]],

         [[1.2286e-01, 1.1683e-05, 5.2452e-05,  ..., 1.5320e-02,
           4.3823e-02, 3.3008e-01]],

         [[5.6689e-01, 0.0000e+00, 1.1921e-07,  ..., 3.3905e-02,
           1.8542e-01, 1.9897e-01]],

         ...,

         [[4.7925e-01, 2.7347e-04, 7.8201e-04,  ..., 8.7929e-04,
           4.3035e-04, 9.1629e-03]],

         [[4.3213e-01, 4.0114e-05, 1.1694e-04,  ..., 3.2349e-03,
           1.5617e-02, 4.4128e-02]],

         [[5.4169e-02, 2.3663e-04, 5.0497e-04,  ..., 2.7878e-02,
           6.7932e-02, 3.8184e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9043e-01, 3.4451e-05, 2.4152e-04,  ..., 1.1299e-02,
           2.8961e-02, 4.0863e-02]],

         [[2.4268e-01, 1.0460e-02, 1.2665e-02,  ..., 1.8492e-03,
           7.7677e-04, 1.3374e-02]],

         [[1.9861e-01, 9.4175e-04, 2.0809e-03,  ..., 3.1357e-03,
           2.9335e-03, 2.3819e-02]],

         ...,

         [[5.1318e-01, 2.0444e-04, 4.5371e-04,  ..., 2.5558e-02,
           3.4912e-02, 3.9581e-02]],

         [[1.8408e-01, 3.6383e-04, 7.0667e-04,  ..., 2.0340e-02,
           4.5837e-02, 4.9268e-01]],

         [[7.1729e-01, 5.7268e-04, 5.4779e-03,  ..., 7.6027e-03,
           4.2000e-03, 5.1971e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3110e-01, 9.3579e-06, 9.4473e-05,  ..., 1.2146e-02,
           1.3283e-02, 8.4961e-02]],

         [[4.1162e-01, 9.3043e-05, 2.2435e-04,  ..., 2.5749e-03,
           3.8071e-03, 5.4199e-02]],

         [[4.7803e-01, 1.0052e-03, 1.9312e-03,  ..., 3.4485e-03,
           2.6291e-02, 2.6962e-02]],

         ...,

         [[3.1891e-02, 8.9264e-03, 1.4153e-02,  ..., 2.9135e-04,
           7.7605e-05, 3.9940e-03]],

         [[8.1396e-01, 1.2338e-04, 9.0790e-04,  ..., 4.2582e-04,
           1.5068e-03, 5.4550e-03]],

         [[1.5686e-01, 2.3770e-04, 2.4235e-04,  ..., 6.8398e-03,
           9.2649e-04, 2.5208e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9809e-04, 1.3809e-03, 1.4026e-01,  ..., 8.3542e-03,
           3.5262e-04, 4.6814e-02]],

         [[7.6246e-04, 1.5221e-03, 1.6113e-01,  ..., 1.1139e-02,
           1.5860e-03, 2.9922e-02]],

         [[3.9768e-04, 8.2254e-04, 9.7717e-02,  ..., 1.3855e-02,
           2.2400e-02, 2.7237e-02]],

         ...,

         [[1.3590e-04, 1.6975e-04, 1.4612e-01,  ..., 6.1264e-03,
           5.5351e-03, 2.0935e-01]],

         [[2.8496e-03, 1.2474e-02, 2.3300e-02,  ..., 2.4891e-04,
           1.1313e-04, 5.4407e-04]],

         [[2.0065e-03, 5.4646e-04, 1.6998e-02,  ..., 1.2195e-04,
           5.2035e-05, 2.8419e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5992e-03, 1.8275e-04, 1.9800e-01,  ..., 3.3817e-03,
           1.8704e-04, 1.0826e-02]],

         [[4.9820e-03, 2.0142e-03, 1.9141e-01,  ..., 3.3989e-03,
           2.2590e-04, 1.6861e-02]],

         [[5.1956e-03, 3.3474e-04, 2.7588e-01,  ..., 2.1572e-03,
           2.0456e-04, 3.1464e-02]],

         ...,

         [[2.7374e-02, 1.5144e-03, 1.1726e-02,  ..., 1.6963e-04,
           1.0991e-04, 3.0479e-03]],

         [[3.0589e-04, 2.6846e-04, 2.6636e-01,  ..., 1.4544e-03,
           1.6155e-03, 1.4015e-02]],

         [[2.4624e-03, 3.2864e-03, 1.2596e-02,  ..., 1.9908e-05,
           8.4043e-06, 1.2115e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1553e-03, 1.0176e-03, 1.1578e-01,  ..., 7.8201e-04,
           6.3972e-03, 2.1851e-02]],

         [[7.1793e-03, 2.8908e-05, 1.0107e-01,  ..., 1.3762e-03,
           1.2350e-04, 1.5747e-02]],

         [[2.7084e-02, 4.4212e-03, 1.7737e-01,  ..., 1.9054e-03,
           2.1591e-03, 2.8061e-02]],

         ...,

         [[2.0332e-03, 4.8280e-04, 1.8018e-01,  ..., 8.5602e-03,
           1.1032e-02, 3.7170e-02]],

         [[2.3060e-03, 9.1934e-04, 1.2396e-01,  ..., 2.7065e-03,
           5.4455e-04, 1.0788e-02]],

         [[2.1622e-02, 9.5940e-04, 4.8340e-02,  ..., 9.1553e-04,
           1.0376e-03, 3.0422e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2598e-03, 7.2598e-05, 2.2156e-02,  ..., 8.3084e-03,
           3.1738e-02, 2.2717e-01]],

         [[1.0538e-03, 4.8590e-04, 2.9663e-01,  ..., 2.5806e-03,
           2.1725e-03, 3.1128e-03]],

         [[6.2828e-03, 1.6296e-02, 4.8943e-03,  ..., 3.8028e-05,
           6.3181e-04, 1.5576e-01]],

         ...,

         [[9.7609e-04, 2.7347e-04, 2.2705e-02,  ..., 2.1286e-03,
           4.2801e-03, 3.2446e-01]],

         [[1.0738e-03, 1.7846e-04, 1.4542e-02,  ..., 3.2177e-03,
           1.6739e-02, 7.4561e-01]],

         [[6.8817e-03, 4.4179e-04, 1.2711e-02,  ..., 5.7411e-04,
           6.6986e-03, 6.3672e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5354e-04, 1.4324e-03, 1.5833e-01,  ..., 3.6144e-03,
           3.4088e-02, 4.3762e-02]],

         [[1.3628e-03, 6.6090e-04, 2.5122e-01,  ..., 3.7632e-03,
           6.4774e-03, 5.1453e-02]],

         [[1.3189e-03, 5.8460e-04, 2.7393e-01,  ..., 3.8986e-03,
           7.8583e-03, 3.6926e-02]],

         ...,

         [[5.9605e-04, 2.0611e-04, 1.1816e-01,  ..., 2.6512e-03,
           2.3087e-02, 3.4180e-01]],

         [[2.1517e-04, 2.4433e-03, 1.6211e-01,  ..., 8.8654e-03,
           6.2370e-03, 1.7126e-01]],

         [[8.9169e-05, 1.1802e-05, 3.6682e-02,  ..., 6.5765e-03,
           2.4628e-02, 4.3994e-01]]]], device='cuda:0', dtype=torch.float16))), hidden_states=None, past_key_values=<transformers.cache_utils.DynamicCache object at 0x7db0a5b77850>)
['<|im_start|>user\n<|vision_start|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|vision_end|>Describe the cat on the left.<think><|im_end|>\n<|im_start|>assistant\nThe cat on the left is a tiger-striped cat wearing a green tag. It is laying down comfortably and appears to be relaxed.<|im_end|>']

Both the input prompt and the generated answer in total have this many tokens (the second dimension in the shape below.)

torch.Size([1, 437])
['The cat on the left is a tiger-striped cat wearing a green tag. It is laying down comfortably and appears to be relaxed.<|im_end|>']

The generated answer itself has this many tokens.

torch.Size([27])

What I’m going to do next is generate the attention matrix of the model before it generated the first token, by averaging the attention of every layer.

There were so many tokens generated in the answer. Therefore, we have so many attention states.

(#27) [(tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4316e-01, 2.5684e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6530e-02, 5.0439e-01, 4.5923e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7796e-03, 1.6203e-03, 6.6109e-03,  ..., 1.9934e-01,
           0.0000e+00, 0.0000e+00],
          [1.4343e-03, 3.3360e-03, 1.1284e-02,  ..., 3.9771e-01,
           1.0059e-01, 0.0000e+00],
          [9.5129e-05, 2.2335e-03, 7.0953e-03,  ..., 4.6265e-02,
           1.7200e-01, 5.8154e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9141e-04, 2.5940e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8413e-06, 2.3186e-05, 8.0943e-05,  ..., 9.5459e-01,
           0.0000e+00, 0.0000e+00],
          [7.1168e-05, 2.8157e-04, 8.2245e-03,  ..., 5.3650e-02,
           5.0879e-01, 0.0000e+00],
          [1.0729e-06, 1.6928e-05, 6.3515e-04,  ..., 6.7616e-04,
           2.8763e-02, 9.5215e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1124e-04, 9.9902e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5359e-05, 4.3058e-04, 9.9951e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0000e+00,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0986e-02,
           9.8877e-01, 0.0000e+00],
          [6.4969e-06, 2.2650e-05, 4.0985e-02,  ..., 2.2650e-05,
           2.7609e-04, 4.9927e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9312e-02, 9.7070e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0651e-01, 1.9897e-01, 6.9434e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7684e-07, 6.0797e-06, 6.0797e-06,  ..., 9.9268e-01,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 8.3447e-07, 4.5121e-05,  ..., 4.0588e-03,
           9.9316e-01, 0.0000e+00],
          [8.1024e-03, 9.1858e-03, 1.9440e-02,  ..., 1.1795e-02,
           1.7151e-02, 7.6904e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5915e-02, 5.9143e-02, 9.2480e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0719e-02, 9.4604e-03, 1.0071e-02,  ..., 2.7374e-02,
           0.0000e+00, 0.0000e+00],
          [1.1459e-02, 1.0765e-02, 1.0109e-02,  ..., 7.8735e-03,
           1.2985e-02, 0.0000e+00],
          [2.3193e-03, 6.3057e-03, 4.6600e-02,  ..., 2.6283e-03,
           1.9424e-02, 1.5283e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3018e-01, 5.6982e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.5571e-01, 5.8643e-01, 5.8044e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0340e-02, 1.4427e-02, 6.8130e-03,  ..., 2.6947e-02,
           0.0000e+00, 0.0000e+00],
          [2.4475e-02, 1.3947e-02, 4.3221e-03,  ..., 3.9734e-02,
           3.6194e-02, 0.0000e+00],
          [8.5526e-03, 1.2444e-02, 1.3962e-03,  ..., 2.0508e-02,
           4.6234e-02, 3.3493e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 1.9934e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7510e-01, 2.6245e-01, 2.6245e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9380e-04, 1.0414e-03, 7.8125e-03,  ..., 5.6543e-01,
           0.0000e+00, 0.0000e+00],
          [7.8392e-04, 2.8667e-03, 1.6495e-02,  ..., 2.5806e-01,
           4.2554e-01, 0.0000e+00],
          [1.2255e-03, 3.5458e-03, 9.3384e-03,  ..., 4.4556e-02,
           1.1743e-01, 1.4380e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6387e-01, 3.6224e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0004e-01, 6.5234e-01, 2.4756e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0564e-05, 1.6093e-06, 5.9605e-08,  ..., 4.1504e-02,
           0.0000e+00, 0.0000e+00],
          [3.3140e-05, 7.4685e-05, 1.6928e-05,  ..., 7.0752e-01,
           1.8457e-01, 0.0000e+00],
          [2.0862e-06, 7.0572e-05, 1.6165e-04,  ..., 5.5756e-02,
           7.6953e-01, 1.6650e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4180e-01, 1.5820e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.5615e-01, 3.3740e-01, 1.0620e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0942e-03, 1.4763e-02, 3.6530e-02,  ..., 3.1250e-02,
           0.0000e+00, 0.0000e+00],
          [7.3433e-03, 7.5760e-03, 4.1595e-02,  ..., 1.4978e-01,
           2.3206e-01, 0.0000e+00],
          [1.1871e-02, 3.0792e-02, 3.6011e-02,  ..., 3.2776e-02,
           1.4917e-01, 6.9397e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 2.5830e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4263e-01, 2.6025e-01, 2.9712e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1372e-02, 2.9022e-02, 4.3579e-02,  ..., 9.5154e-02,
           0.0000e+00, 0.0000e+00],
          [8.6288e-03, 2.0523e-02, 4.0192e-02,  ..., 1.2769e-01,
           2.3315e-01, 0.0000e+00],
          [1.6680e-03, 7.0229e-03, 1.2039e-02,  ..., 5.1483e-02,
           1.5137e-01, 8.4229e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4697e-01, 3.5303e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9893e-01, 1.9434e-01, 4.0674e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.4289e-03, 1.4206e-02, 1.0095e-01,  ..., 6.1493e-03,
           0.0000e+00, 0.0000e+00],
          [1.7691e-03, 6.5956e-03, 5.2917e-02,  ..., 2.7451e-02,
           1.0944e-01, 0.0000e+00],
          [7.9880e-03, 1.3489e-02, 3.9795e-02,  ..., 2.8549e-02,
           7.0679e-02, 1.5076e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9160e-01, 1.0822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6890e-01, 4.6997e-01, 1.6113e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1887e-02, 3.3600e-02, 1.8692e-02,  ..., 3.4204e-01,
           0.0000e+00, 0.0000e+00],
          [4.3449e-03, 2.4033e-02, 2.5391e-02,  ..., 6.9580e-02,
           4.0649e-01, 0.0000e+00],
          [8.3590e-04, 7.3929e-03, 1.7977e-03,  ..., 1.6431e-01,
           2.7515e-01, 2.1606e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1826e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2510e-01, 2.0776e-01, 6.6956e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0966e-02, 1.4944e-03, 4.0627e-03,  ..., 7.6660e-02,
           0.0000e+00, 0.0000e+00],
          [2.2415e-02, 4.0817e-03, 1.0422e-02,  ..., 4.7455e-02,
           1.1298e-01, 0.0000e+00],
          [1.7786e-04, 7.8321e-05, 1.6844e-04,  ..., 1.3831e-01,
           6.2012e-01, 1.2207e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5146e-01, 2.4866e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5581e-01, 2.3828e-01, 3.0591e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9973e-03, 5.5733e-03, 8.0795e-03,  ..., 2.6321e-03,
           0.0000e+00, 0.0000e+00],
          [1.1795e-02, 5.0926e-03, 1.1040e-02,  ..., 2.2392e-03,
           1.3770e-01, 0.0000e+00],
          [7.9575e-03, 5.2261e-03, 7.1182e-03,  ..., 3.3321e-03,
           1.5173e-01, 4.3030e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6809e-01, 8.3203e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.2864e-01, 2.3962e-01, 5.3174e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9897e-02, 1.0612e-02, 1.5198e-02,  ..., 1.4935e-03,
           0.0000e+00, 0.0000e+00],
          [1.3489e-02, 7.1106e-03, 7.7782e-03,  ..., 3.8643e-03,
           8.2474e-03, 0.0000e+00],
          [3.9291e-03, 1.2903e-03, 1.0471e-03,  ..., 3.5706e-02,
           3.1891e-02, 1.0413e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1143e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0248e-02, 9.2383e-01, 5.5908e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6413e-02, 7.4339e-04, 2.4700e-04,  ..., 4.6460e-01,
           0.0000e+00, 0.0000e+00],
          [4.2763e-03, 1.2445e-03, 1.4544e-03,  ..., 5.0195e-01,
           1.7480e-01, 0.0000e+00],
          [4.2200e-05, 5.6505e-04, 2.3232e-03,  ..., 3.0609e-02,
           7.8955e-01, 1.3928e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1650e-01, 8.3618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2314e-01, 1.6394e-01, 1.1267e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6040e-02, 1.0773e-02, 4.5380e-02,  ..., 3.8818e-02,
           0.0000e+00, 0.0000e+00],
          [3.5645e-02, 7.0724e-03, 1.8051e-02,  ..., 2.4292e-02,
           1.2146e-01, 0.0000e+00],
          [1.6937e-02, 2.5177e-03, 1.3718e-02,  ..., 2.4078e-02,
           3.9062e-02, 1.7932e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7990e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3267e-01, 1.0986e-01, 6.5771e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6642e-02, 1.5879e-03, 3.3360e-03,  ..., 1.8213e-01,
           0.0000e+00, 0.0000e+00],
          [1.7288e-02, 4.7607e-03, 1.5732e-02,  ..., 2.2241e-01,
           2.8101e-01, 0.0000e+00],
          [1.3023e-02, 1.8110e-03, 1.0712e-02,  ..., 4.9530e-02,
           1.0162e-01, 4.6631e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6182e-01, 1.3831e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6216e-01, 1.4137e-02, 5.2393e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1128e-01, 2.6108e-02, 1.1072e-03,  ..., 3.9154e-02,
           0.0000e+00, 0.0000e+00],
          [2.5928e-01, 1.8272e-03, 1.2445e-01,  ..., 3.6001e-05,
           7.5745e-02, 0.0000e+00],
          [7.4072e-01, 4.9448e-04, 2.7351e-03,  ..., 3.5667e-03,
           3.2684e-02, 1.5419e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 2.3327e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 7.9956e-03, 1.2802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 1.0800e-04, 1.5259e-05,  ..., 6.1941e-04,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.0908e-05, 4.1056e-04,  ..., 1.1955e-02,
           2.4429e-02, 0.0000e+00],
          [8.3789e-01, 4.9233e-05, 1.1092e-04,  ..., 1.0651e-02,
           3.4058e-02, 9.6970e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.7275e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.5106e-02, 1.5381e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6846e-01, 1.4937e-04, 7.3969e-05,  ..., 5.6946e-02,
           0.0000e+00, 0.0000e+00],
          [7.7881e-01, 1.3530e-04, 2.4116e-04,  ..., 6.4564e-04,
           1.2840e-02, 0.0000e+00],
          [8.0615e-01, 8.6963e-05, 9.3079e-04,  ..., 1.5926e-04,
           6.0120e-03, 2.1149e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 1.0138e-03, 6.6071e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1885e-01, 4.9591e-03, 1.0330e-02,  ..., 7.1167e-02,
           0.0000e+00, 0.0000e+00],
          [6.8213e-01, 6.4316e-03, 6.5575e-03,  ..., 1.2457e-05,
           4.9408e-02, 0.0000e+00],
          [8.4082e-01, 1.0557e-03, 4.4098e-03,  ..., 4.2140e-05,
           6.8426e-04, 5.7983e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8291e-01, 1.7242e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 1.9501e-02, 2.6443e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0479e-01, 2.0683e-04, 9.8610e-04,  ..., 9.0456e-04,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.7376e-03, 2.8324e-03,  ..., 1.9646e-03,
           2.3438e-02, 0.0000e+00],
          [5.4688e-01, 3.4404e-04, 6.7139e-04,  ..., 2.3544e-04,
           1.5030e-02, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9586e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2334e-01, 2.8275e-02, 4.8370e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3701e-01, 9.6679e-05, 3.9649e-04,  ..., 6.2752e-04,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 1.2379e-03, 3.8052e-03,  ..., 6.2323e-04,
           3.2104e-02, 0.0000e+00],
          [7.5684e-01, 3.4761e-04, 1.7176e-03,  ..., 2.4796e-04,
           6.6757e-03, 2.1469e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1235e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 3.9482e-03, 4.6921e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6187e-01, 1.0431e-05, 2.2113e-05,  ..., 6.0081e-03,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 6.3539e-05, 5.8460e-04,  ..., 5.2452e-06,
           2.0981e-02, 0.0000e+00],
          [6.6162e-01, 1.7941e-05, 1.0166e-03,  ..., 4.7922e-04,
           1.4820e-03, 1.6098e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.0712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3374e-02, 8.1635e-03, 3.5715e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9541e-01, 5.9605e-06, 5.9605e-08,  ..., 2.6875e-03,
           0.0000e+00, 0.0000e+00],
          [9.2822e-01, 2.3007e-04, 1.2040e-05,  ..., 1.7654e-02,
           1.9470e-02, 0.0000e+00],
          [8.7891e-01, 2.4915e-05, 9.0420e-05,  ..., 3.6907e-03,
           1.0187e-01, 1.4641e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 4.0833e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2920e-01, 4.8859e-02, 2.1820e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3018e-01, 2.4247e-04, 5.3704e-05,  ..., 7.7972e-03,
           0.0000e+00, 0.0000e+00],
          [9.0625e-01, 1.1749e-03, 1.2217e-03,  ..., 3.5248e-02,
           1.9638e-02, 0.0000e+00],
          [9.0479e-01, 4.1747e-04, 2.6131e-04,  ..., 5.4359e-03,
           9.6512e-03, 1.0956e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 5.4993e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.1450e-01, 3.4241e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7510e-01, 3.3474e-04, 9.2220e-04,  ..., 3.5706e-03,
           0.0000e+00, 0.0000e+00],
          [7.4512e-01, 4.6825e-04, 6.5136e-04,  ..., 9.2621e-03,
           3.0869e-02, 0.0000e+00],
          [6.6162e-01, 2.3687e-04, 2.0623e-04,  ..., 4.4823e-03,
           3.7903e-02, 8.8806e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7028e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 3.2837e-02, 3.6957e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3994e-01, 5.5046e-03, 1.2650e-02,  ..., 5.8708e-03,
           0.0000e+00, 0.0000e+00],
          [6.9141e-01, 1.0643e-02, 1.0078e-02,  ..., 7.0496e-03,
           3.6865e-02, 0.0000e+00],
          [4.1699e-01, 5.6419e-03, 5.1384e-03,  ..., 5.6915e-03,
           5.7648e-02, 2.2400e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4824e-01, 5.1849e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0127e-01, 1.3257e-01, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.2445e-04, 1.0324e-04,  ..., 4.9927e-02,
           0.0000e+00, 0.0000e+00],
          [6.1768e-01, 2.5797e-04, 2.5535e-04,  ..., 9.6985e-02,
           9.6191e-02, 0.0000e+00],
          [5.4834e-01, 8.7798e-05, 1.6689e-04,  ..., 4.7272e-02,
           1.2732e-01, 1.2976e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3588e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 2.1469e-02, 3.3905e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.4255e-01, 9.2840e-04, 7.3147e-04,  ..., 2.2919e-02,
           0.0000e+00, 0.0000e+00],
          [3.0200e-01, 9.4652e-04, 1.0891e-03,  ..., 1.3756e-02,
           7.5684e-02, 0.0000e+00],
          [4.0796e-01, 1.8358e-03, 7.1335e-04,  ..., 1.2772e-02,
           7.2998e-02, 4.8004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 2.3392e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2568e-01, 1.6895e-01, 5.1270e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1123e-01, 5.3692e-04, 4.4227e-05,  ..., 4.5929e-02,
           0.0000e+00, 0.0000e+00],
          [3.7915e-01, 7.6771e-04, 4.3631e-05,  ..., 4.4525e-02,
           2.7710e-01, 0.0000e+00],
          [2.9761e-01, 1.4448e-04, 4.3511e-06,  ..., 6.6414e-03,
           4.4897e-01, 5.8441e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 3.1860e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 2.5665e-02, 1.9028e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8564e-01, 3.8195e-04, 1.2207e-04,  ..., 2.9812e-03,
           0.0000e+00, 0.0000e+00],
          [7.7637e-01, 1.3866e-03, 3.2234e-03,  ..., 6.1989e-03,
           1.0529e-02, 0.0000e+00],
          [7.8125e-01, 2.8181e-04, 6.3896e-05,  ..., 4.9210e-03,
           2.2171e-02, 3.0945e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2604e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 1.1498e-04, 3.6865e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9111e-01, 7.3552e-05, 1.1325e-06,  ..., 1.0364e-01,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 4.2260e-05, 1.4305e-06,  ..., 5.8949e-05,
           2.2446e-02, 0.0000e+00],
          [9.4385e-01, 1.8537e-05, 3.2544e-05,  ..., 1.0258e-04,
           3.6073e-04, 5.2582e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4014e-01, 1.1917e-02, 0.0000e+00,  ..., 6.0997e-03,
           6.0997e-03, 6.0997e-03],
          [2.8540e-01, 1.5497e-06, 7.4158e-03,  ..., 7.4425e-03,
           7.4425e-03, 0.0000e+00],
          ...,
          [9.3018e-01, 1.3933e-03, 3.7253e-05,  ..., 7.2021e-03,
           0.0000e+00, 0.0000e+00],
          [9.8779e-01, 2.1183e-04, 9.8348e-06,  ..., 4.1723e-07,
           6.9733e-03, 0.0000e+00],
          [9.8096e-01, 5.4538e-05, 4.5514e-04,  ..., 2.3842e-07,
           1.1921e-05, 1.3779e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-02, 4.1628e-04, 0.0000e+00,  ..., 4.4823e-03,
           0.0000e+00, 4.4823e-03],
          [1.7053e-01, 2.3842e-07, 8.7404e-04,  ..., 5.7526e-03,
           5.7526e-03, 0.0000e+00],
          ...,
          [9.9365e-01, 7.3791e-05, 5.3644e-07,  ..., 2.0618e-03,
           0.0000e+00, 0.0000e+00],
          [9.9512e-01, 2.8074e-05, 6.5565e-07,  ..., 4.1723e-07,
           3.9139e-03, 0.0000e+00],
          [9.9805e-01, 4.8876e-06, 7.9274e-06,  ..., 5.9605e-08,
           7.7486e-07, 1.5945e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7914e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0615e-01, 1.9080e-01, 3.0651e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5425e-01, 2.3782e-05, 1.3769e-05,  ..., 4.8859e-02,
           0.0000e+00, 0.0000e+00],
          [5.7080e-01, 1.1081e-04, 9.5487e-05,  ..., 8.8989e-02,
           9.0332e-02, 0.0000e+00],
          [7.5342e-01, 3.3808e-04, 2.2292e-05,  ..., 2.7664e-02,
           1.3538e-01, 1.3496e-02]],

         [[5.9277e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5662e-01, 3.0994e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.4082e-01, 6.5422e-04, 6.8045e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6719e-01, 1.0473e-04, 8.1539e-05,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [7.0752e-01, 3.7422e-03, 7.0286e-04,  ..., 1.1086e-05,
           3.4149e-02, 0.0000e+00],
          [9.2480e-01, 1.4150e-04, 4.0627e-04,  ..., 1.2755e-05,
           9.4843e-04, 1.4572e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 9.0027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.7802e-02, 4.3907e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7612e-01, 4.2038e-03, 8.9455e-04,  ..., 1.2268e-01,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 1.8911e-03, 6.9571e-04,  ..., 6.1531e-03,
           1.8356e-02, 0.0000e+00],
          [8.2227e-01, 1.4400e-03, 2.4021e-04,  ..., 1.5097e-03,
           1.7242e-02, 1.7166e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5106e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0869e-01, 8.5022e-02, 6.3057e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1162e-01, 7.6151e-04, 2.4056e-04,  ..., 6.1951e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 4.3793e-03, 5.4407e-04,  ..., 4.9324e-03,
           2.0966e-02, 0.0000e+00],
          [8.6230e-01, 1.1921e-03, 3.2592e-04,  ..., 4.8943e-03,
           1.5625e-02, 2.9266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.7042e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1309e-01, 3.2806e-02, 5.4230e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6865e-01, 2.5034e-05, 5.9307e-05,  ..., 6.3599e-02,
           0.0000e+00, 0.0000e+00],
          [8.2520e-01, 4.2796e-04, 8.6975e-04,  ..., 1.3519e-02,
           9.7107e-02, 0.0000e+00],
          [8.7354e-01, 5.1320e-05, 1.1891e-04,  ..., 3.2444e-03,
           1.7273e-02, 4.6082e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9471e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2021e-01, 2.6611e-01, 1.3817e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8770e-01, 1.6451e-04, 4.8816e-05,  ..., 7.7698e-02,
           0.0000e+00, 0.0000e+00],
          [8.7500e-01, 3.3140e-04, 1.4877e-04,  ..., 3.1910e-03,
           7.4463e-02, 0.0000e+00],
          [8.7830e-02, 3.7050e-04, 4.4584e-05,  ..., 3.6030e-03,
           8.9355e-01, 3.5515e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9805e-01, 1.9121e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 8.2703e-03, 7.5302e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4893e-01, 5.5313e-05, 1.3638e-04,  ..., 4.8523e-02,
           0.0000e+00, 0.0000e+00],
          [5.1904e-01, 2.7776e-04, 1.0710e-03,  ..., 2.8934e-03,
           1.1650e-02, 0.0000e+00],
          [6.9629e-01, 1.5771e-04, 2.1303e-04,  ..., 1.9398e-03,
           7.5378e-03, 2.8076e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5125e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6914e-01, 7.2327e-02, 5.8502e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0566e-01, 5.9414e-04, 1.3161e-03,  ..., 6.1005e-02,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 7.0047e-04, 2.0945e-04,  ..., 1.6546e-03,
           2.1942e-02, 0.0000e+00],
          [5.6152e-01, 7.2432e-04, 1.9264e-04,  ..., 4.8180e-03,
           3.0151e-02, 6.9458e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 1.0078e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 2.3755e-01, 2.3361e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 2.7001e-05, 5.4359e-05,  ..., 3.2330e-03,
           0.0000e+00, 0.0000e+00],
          [9.5361e-01, 7.1645e-05, 4.4823e-05,  ..., 2.8782e-03,
           9.6970e-03, 0.0000e+00],
          [9.0820e-01, 1.0967e-04, 1.7285e-05,  ..., 3.7174e-03,
           4.5105e-02, 1.1147e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.3030e-02, 6.8016e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5293e-01, 1.0973e-04, 2.9743e-05,  ..., 2.5511e-04,
           0.0000e+00, 0.0000e+00],
          [6.3574e-01, 1.1864e-03, 6.1083e-04,  ..., 1.9722e-03,
           2.6718e-02, 0.0000e+00],
          [7.1436e-01, 3.7956e-04, 1.0496e-04,  ..., 1.6937e-03,
           1.3695e-02, 1.4694e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9932e-01, 1.9336e-01, 7.4959e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1406e-01, 6.3467e-04, 2.1601e-04,  ..., 4.5013e-04,
           0.0000e+00, 0.0000e+00],
          [5.8789e-01, 3.4122e-03, 4.3225e-04,  ..., 7.0457e-03,
           2.9953e-02, 0.0000e+00],
          [4.8145e-01, 5.6763e-03, 2.9516e-04,  ..., 6.5231e-03,
           4.9438e-02, 2.0691e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-01, 1.0638e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9126e-01, 6.4746e-01, 6.1310e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8909e-01, 5.7077e-04, 1.4858e-03,  ..., 2.2385e-02,
           0.0000e+00, 0.0000e+00],
          [3.4790e-01, 3.4618e-03, 1.7891e-03,  ..., 9.7809e-03,
           7.8735e-02, 0.0000e+00],
          [6.1328e-01, 1.4267e-03, 8.6164e-04,  ..., 1.4519e-02,
           7.0190e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6172e-01, 2.3853e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 1.8054e-01, 8.2153e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9775e-01, 8.5211e-04, 6.9284e-04,  ..., 2.2202e-03,
           0.0000e+00, 0.0000e+00],
          [3.3130e-01, 6.7482e-03, 4.0779e-03,  ..., 5.2023e-04,
           1.5572e-02, 0.0000e+00],
          [5.9082e-01, 1.8854e-03, 1.0052e-03,  ..., 5.8413e-04,
           7.5760e-03, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.7449e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1602e-01, 7.2327e-02, 1.1589e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7646e-01, 2.4140e-05, 1.0002e-04,  ..., 1.0023e-03,
           0.0000e+00, 0.0000e+00],
          [1.2634e-02, 5.4240e-06, 1.3173e-05,  ..., 1.8167e-04,
           1.1021e-04, 0.0000e+00],
          [1.3664e-02, 2.1577e-05, 1.5199e-05,  ..., 2.6774e-04,
           2.7370e-04, 5.4550e-04]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 1.7456e-02, 7.4501e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 5.0962e-05, 2.0564e-05,  ..., 1.6670e-03,
           0.0000e+00, 0.0000e+00],
          [8.4082e-01, 3.1257e-04, 1.5771e-04,  ..., 2.6684e-03,
           8.6670e-03, 0.0000e+00],
          [9.1406e-01, 6.3419e-05, 1.3173e-05,  ..., 1.8902e-03,
           2.7943e-03, 3.4904e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7852e-01, 2.1271e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 9.6054e-03, 1.4503e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0176e-01, 5.0783e-04, 9.0170e-04,  ..., 2.2461e-01,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 3.3069e-04, 5.3549e-04,  ..., 4.0321e-03,
           1.2891e-01, 0.0000e+00],
          [2.4939e-01, 2.5196e-03, 7.6828e-03,  ..., 6.1321e-04,
           6.0539e-03, 1.6101e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.0403e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7461e-01, 2.1423e-02, 3.8376e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3291e-01, 2.2054e-05, 2.7478e-05,  ..., 1.3718e-02,
           0.0000e+00, 0.0000e+00],
          [1.6650e-01, 9.7096e-05, 1.0788e-04,  ..., 2.1782e-03,
           1.8740e-03, 0.0000e+00],
          [9.0759e-02, 4.7922e-05, 6.3717e-05,  ..., 1.4668e-03,
           6.3896e-04, 4.7340e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3311e-01, 6.6895e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 6.8262e-01, 7.6065e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9258e-01, 2.5606e-04, 2.4533e-04,  ..., 1.0010e-02,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 6.8617e-04, 4.6253e-05,  ..., 5.0659e-03,
           6.7558e-03, 0.0000e+00],
          [8.6035e-01, 8.8549e-04, 1.8454e-04,  ..., 8.0795e-03,
           3.2898e-02, 3.6564e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1543e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 2.8534e-02, 3.5980e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7246e-01, 4.6086e-04, 2.2736e-03,  ..., 3.1952e-02,
           0.0000e+00, 0.0000e+00],
          [3.1982e-01, 3.9983e-04, 7.0429e-04,  ..., 2.1912e-02,
           1.4880e-01, 0.0000e+00],
          [6.5723e-01, 1.6463e-04, 4.4918e-04,  ..., 6.2370e-03,
           5.2948e-02, 4.5868e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6582e-01, 3.4332e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 3.0884e-01, 1.6479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.0830e-01, 2.9802e-05, 2.9087e-05,  ..., 5.5328e-02,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 4.9353e-05, 2.5809e-05,  ..., 6.5918e-03,
           1.8860e-02, 0.0000e+00],
          [4.5508e-01, 2.8431e-05, 6.5148e-05,  ..., 2.7649e-02,
           2.8003e-01, 2.5620e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 1.0901e-01, 7.2098e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.8672e-01, 1.3340e-04, 3.1638e-04,  ..., 1.2199e-02,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.3328e-04, 4.2701e-04,  ..., 4.9438e-03,
           1.2604e-02, 0.0000e+00],
          [7.1680e-01, 5.5122e-04, 9.0218e-04,  ..., 7.4043e-03,
           2.0370e-02, 1.4412e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 6.3965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1299e-01, 1.5271e-01, 3.4424e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4805e-01, 6.8617e-04, 5.6887e-04,  ..., 7.7400e-03,
           0.0000e+00, 0.0000e+00],
          [5.4590e-01, 7.1716e-04, 4.5061e-04,  ..., 2.5043e-03,
           3.8391e-02, 0.0000e+00],
          [4.1138e-01, 7.5769e-04, 5.5122e-04,  ..., 7.8888e-03,
           3.5950e-02, 2.4463e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8223e-01, 1.9971e-01, 1.7838e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.8535e-01, 3.3903e-04, 5.1141e-05,  ..., 4.3411e-03,
           0.0000e+00, 0.0000e+00],
          [9.0918e-01, 3.2973e-04, 2.6393e-04,  ..., 4.6654e-03,
           8.6975e-03, 0.0000e+00],
          [7.1631e-01, 3.0112e-04, 2.4796e-04,  ..., 1.7456e-02,
           1.3623e-01, 1.1559e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.4027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9609e-01, 2.8114e-03, 9.2840e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8555e-01, 8.1062e-06, 1.2696e-05,  ..., 1.6336e-03,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 3.6955e-06, 1.0550e-05,  ..., 8.4019e-04,
           2.7218e-03, 0.0000e+00],
          [9.8584e-01, 1.8477e-06, 4.7684e-06,  ..., 3.2544e-04,
           1.0405e-03, 9.0170e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9673e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1787e-02, 4.0829e-05, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.1765e-01, 2.1935e-05, 1.6966e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1689e-01, 1.3232e-04, 2.7156e-04,  ..., 6.6948e-04,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.3403e-03, 4.0054e-04,  ..., 3.9339e-05,
           8.3069e-02, 0.0000e+00],
          [5.6006e-01, 2.9874e-04, 6.7825e-03,  ..., 9.3222e-05,
           3.6740e-04, 8.0505e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1024e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4043e-01, 5.7312e-02, 2.0084e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2073e-01, 2.0087e-05, 6.2823e-05,  ..., 2.6672e-02,
           0.0000e+00, 0.0000e+00],
          [6.5771e-01, 3.2234e-04, 1.2970e-04,  ..., 2.0618e-03,
           1.0925e-02, 0.0000e+00],
          [6.3770e-01, 1.2201e-04, 4.3511e-05,  ..., 9.6512e-03,
           1.2482e-01, 1.4923e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.8763e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 3.1311e-02, 1.4671e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.3916e-01, 1.1921e-07, 4.7684e-07,  ..., 1.8148e-03,
           0.0000e+00, 0.0000e+00],
          [4.6484e-01, 1.4472e-04, 1.6689e-05,  ..., 3.3200e-05,
           8.6441e-03, 0.0000e+00],
          [2.9907e-01, 6.7949e-06, 5.8651e-05,  ..., 4.6134e-04,
           7.6675e-04, 2.6428e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.3400e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.0236e-01, 2.0950e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 1.7107e-04, 7.5042e-05,  ..., 5.9938e-04,
           0.0000e+00, 0.0000e+00],
          [4.9146e-01, 1.0958e-03, 3.8099e-04,  ..., 4.3106e-03,
           2.7420e-02, 0.0000e+00],
          [7.5195e-01, 8.4496e-04, 4.6468e-04,  ..., 4.1847e-03,
           1.5030e-02, 9.9106e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5736e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 9.3460e-03, 1.6449e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7041e-01, 3.4904e-04, 5.0783e-04,  ..., 6.4453e-02,
           0.0000e+00, 0.0000e+00],
          [4.9292e-01, 8.6308e-04, 1.3876e-03,  ..., 2.7893e-02,
           5.6244e-02, 0.0000e+00],
          [4.6387e-01, 2.5005e-03, 2.8687e-03,  ..., 2.8030e-02,
           4.1016e-02, 4.9316e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7617e-01, 4.2383e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2412e-01, 2.5879e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1604e-02, 2.2449e-03, 8.2397e-04,  ..., 8.5144e-03,
           0.0000e+00, 0.0000e+00],
          [1.9751e-01, 1.2169e-03, 8.7452e-04,  ..., 8.4448e-04,
           1.6495e-02, 0.0000e+00],
          [2.4634e-01, 1.4076e-03, 9.1314e-04,  ..., 4.4250e-03,
           2.7069e-02, 4.2358e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9473e-01, 4.0552e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7139e-01, 4.9591e-02, 2.7881e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5867e-02, 1.7405e-05, 1.7047e-05,  ..., 8.8867e-01,
           0.0000e+00, 0.0000e+00],
          [4.4189e-01, 6.7353e-06, 3.3379e-05,  ..., 1.9093e-03,
           4.9097e-01, 0.0000e+00],
          [4.2578e-01, 5.2452e-06, 3.7134e-05,  ..., 3.0537e-03,
           3.4607e-02, 4.6582e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7725e-01, 3.2251e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3657e-01, 4.0552e-01, 3.5791e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3516e-01, 1.9395e-04, 2.3019e-04,  ..., 2.6646e-03,
           0.0000e+00, 0.0000e+00],
          [5.4077e-02, 8.0681e-04, 3.1223e-03,  ..., 1.2665e-03,
           2.2812e-02, 0.0000e+00],
          [1.4929e-01, 1.8950e-03, 4.1046e-03,  ..., 2.4090e-03,
           2.1851e-02, 3.6896e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9166e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.1879e-02, 3.6087e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4902e-01, 3.9864e-04, 4.1676e-04,  ..., 5.4054e-03,
           0.0000e+00, 0.0000e+00],
          [7.6843e-02, 6.7520e-04, 4.6611e-04,  ..., 8.4534e-03,
           4.6265e-02, 0.0000e+00],
          [4.5288e-01, 1.0509e-03, 8.3017e-04,  ..., 3.6240e-03,
           2.4384e-02, 5.3680e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.5725e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.2573e-02, 1.6556e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7427e-01, 4.9591e-05, 2.1207e-04,  ..., 3.5645e-02,
           0.0000e+00, 0.0000e+00],
          [1.2091e-01, 4.7517e-04, 8.7214e-04,  ..., 6.4354e-03,
           2.5845e-03, 0.0000e+00],
          [3.2642e-01, 3.0828e-04, 7.9775e-04,  ..., 4.4174e-03,
           6.0310e-03, 1.9257e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4549e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6865e-01, 1.1395e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1310e-02, 2.2054e-06, 5.1856e-06,  ..., 4.1046e-03,
           0.0000e+00, 0.0000e+00],
          [2.9834e-01, 1.4734e-04, 2.0862e-04,  ..., 1.5808e-02,
           6.8169e-03, 0.0000e+00],
          [3.1152e-01, 2.2471e-05, 4.5598e-05,  ..., 2.9907e-02,
           1.5686e-02, 1.0612e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1934e-01, 1.8079e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 1.9791e-02, 3.3142e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1350e-02, 4.1366e-05, 1.7226e-04,  ..., 7.2656e-01,
           0.0000e+00, 0.0000e+00],
          [4.2700e-01, 1.0288e-04, 7.7248e-04,  ..., 2.0676e-03,
           4.8267e-01, 0.0000e+00],
          [3.5718e-01, 2.0742e-05, 2.3258e-04,  ..., 2.7523e-03,
           2.9877e-02, 5.6250e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8584e-01, 5.1416e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3628e-01, 1.7786e-01, 3.8574e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4259e-02, 3.5286e-05, 6.6936e-05,  ..., 8.8770e-01,
           0.0000e+00, 0.0000e+00],
          [1.4587e-01, 1.3125e-04, 3.7265e-04,  ..., 2.9016e-04,
           7.6611e-01, 0.0000e+00],
          [1.0394e-01, 1.3053e-04, 2.2340e-04,  ..., 1.4601e-03,
           1.3318e-01, 6.4941e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8926e-01, 1.0590e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 4.0131e-02, 1.2756e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7078e-01, 9.1314e-05, 1.1140e-04,  ..., 6.3610e-04,
           0.0000e+00, 0.0000e+00],
          [8.7891e-01, 3.1328e-04, 1.3189e-03,  ..., 5.3883e-04,
           8.9569e-03, 0.0000e+00],
          [8.1641e-01, 2.6131e-04, 9.4891e-04,  ..., 3.8567e-03,
           1.4427e-02, 8.2245e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1055e-01, 1.8921e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.7930e-01, 1.3416e-01, 8.6365e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2544e-01, 2.0683e-05, 1.8120e-05,  ..., 3.9429e-02,
           0.0000e+00, 0.0000e+00],
          [1.4320e-02, 9.8765e-05, 9.3699e-04,  ..., 4.5598e-05,
           7.5836e-03, 0.0000e+00],
          [1.9165e-01, 1.5199e-04, 5.6887e-04,  ..., 8.0729e-04,
           6.0577e-03, 1.3191e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 2.6465e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5459e-01, 5.9662e-02, 4.8584e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4290e-02, 1.0133e-06, 5.7220e-06,  ..., 9.6143e-01,
           0.0000e+00, 0.0000e+00],
          [1.4404e-01, 1.2696e-05, 8.8394e-05,  ..., 2.3918e-03,
           8.0811e-01, 0.0000e+00],
          [1.2042e-01, 7.8082e-06, 1.8895e-05,  ..., 2.3537e-03,
           3.3752e-02, 8.0566e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 4.4861e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2617e-01, 5.9906e-02, 1.1414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.5625e-01, 4.6611e-04, 1.7691e-04,  ..., 2.2766e-02,
           0.0000e+00, 0.0000e+00],
          [1.6797e-01, 2.1982e-04, 6.8378e-04,  ..., 9.1124e-04,
           4.9951e-01, 0.0000e+00],
          [1.6870e-01, 1.7023e-04, 6.9332e-04,  ..., 2.2087e-03,
           2.1277e-01, 2.6733e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 1.7792e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 9.8877e-03, 5.1994e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4473e-01, 6.8069e-05, 1.7166e-05,  ..., 1.3332e-03,
           0.0000e+00, 0.0000e+00],
          [8.3740e-01, 4.2510e-04, 1.2913e-03,  ..., 1.5039e-03,
           1.1263e-03, 0.0000e+00],
          [8.7988e-01, 1.8942e-04, 4.3869e-04,  ..., 2.4128e-03,
           7.1335e-04, 2.0199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4351e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 4.6768e-03, 2.6245e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1228e-01, 3.1924e-04, 3.3069e-04,  ..., 7.8735e-03,
           0.0000e+00, 0.0000e+00],
          [7.4707e-01, 4.4227e-04, 7.5531e-04,  ..., 1.3332e-03,
           1.5869e-02, 0.0000e+00],
          [4.3018e-01, 9.6893e-04, 7.5150e-04,  ..., 2.6226e-03,
           3.8025e-02, 2.9739e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3071e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4014e-01, 2.9126e-01, 6.8481e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0605e-01, 2.4617e-05, 1.2982e-04,  ..., 9.9564e-03,
           0.0000e+00, 0.0000e+00],
          [9.3945e-01, 9.3877e-05, 1.0353e-04,  ..., 1.6842e-03,
           1.0338e-02, 0.0000e+00],
          [7.6611e-01, 3.7372e-05, 6.1631e-05,  ..., 1.7910e-03,
           9.2834e-02, 1.8341e-02]],

         [[8.4814e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3756e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0801e-01, 1.7929e-03, 2.6283e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7031e-01, 2.7776e-04, 5.1117e-04,  ..., 1.7838e-02,
           0.0000e+00, 0.0000e+00],
          [6.3477e-01, 8.8072e-04, 2.7943e-03,  ..., 6.0499e-05,
           1.5671e-02, 0.0000e+00],
          [6.3330e-01, 4.7302e-04, 1.1730e-03,  ..., 5.7745e-04,
           2.5520e-03, 3.3173e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8340e-01, 1.6479e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4502e-01, 3.3179e-01, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8467e-01, 2.9385e-05, 1.5497e-06,  ..., 2.1713e-02,
           0.0000e+00, 0.0000e+00],
          [7.5244e-01, 1.4746e-04, 5.6863e-05,  ..., 1.2091e-01,
           5.4993e-02, 0.0000e+00],
          [4.6704e-01, 5.2452e-06, 5.4896e-05,  ..., 9.9411e-03,
           4.4922e-01, 6.9092e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.1569e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 1.1719e-02, 1.2405e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7437e-01, 1.4515e-03, 1.6584e-03,  ..., 5.3635e-03,
           0.0000e+00, 0.0000e+00],
          [6.8262e-01, 3.9024e-03, 5.1613e-03,  ..., 1.5554e-03,
           3.5801e-03, 0.0000e+00],
          [7.0068e-01, 1.6909e-03, 3.3092e-03,  ..., 3.2654e-03,
           7.6065e-03, 1.9516e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 2.0920e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9971e-01, 2.8149e-01, 1.8967e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1826e-01, 3.2663e-05, 3.0696e-05,  ..., 3.0029e-02,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 9.1314e-05, 5.6505e-05,  ..., 5.1453e-02,
           2.7679e-02, 0.0000e+00],
          [7.5781e-01, 3.3617e-05, 6.1095e-05,  ..., 8.0383e-02,
           7.8125e-02, 1.8143e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.3940e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 8.9722e-03, 8.7738e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 6.7890e-05, 1.7107e-05,  ..., 6.9656e-03,
           0.0000e+00, 0.0000e+00],
          [3.7109e-01, 1.6394e-03, 1.2894e-03,  ..., 6.5193e-03,
           1.8753e-02, 0.0000e+00],
          [7.4268e-01, 6.4278e-04, 2.2960e-04,  ..., 5.7411e-03,
           1.1101e-03, 5.6305e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 1.9958e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8145e-01, 9.2239e-03, 9.2010e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8684e-02, 2.3496e-04, 3.6311e-04,  ..., 5.5695e-03,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.6751e-04, 8.2731e-05,  ..., 1.9133e-05,
           1.7529e-03, 0.0000e+00],
          [9.1699e-01, 3.1805e-04, 1.5104e-04,  ..., 2.7847e-04,
           8.2321e-03, 1.6754e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 4.8157e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 3.2043e-02, 1.6205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8833e-01, 2.7061e-04, 1.5426e-04,  ..., 1.3733e-02,
           0.0000e+00, 0.0000e+00],
          [9.1992e-01, 2.3327e-03, 5.4264e-04,  ..., 3.8147e-05,
           4.8370e-03, 0.0000e+00],
          [5.8740e-01, 1.7452e-03, 6.4516e-04,  ..., 4.7016e-04,
           6.7078e-02, 4.1748e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 2.5131e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0322e-01, 1.5112e-01, 4.5593e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.0312e-01, 5.7936e-05, 5.5492e-05,  ..., 3.1143e-02,
           0.0000e+00, 0.0000e+00],
          [1.8848e-01, 2.6608e-04, 2.7466e-04,  ..., 7.1655e-02,
           8.5938e-02, 0.0000e+00],
          [3.4033e-01, 8.7857e-05, 1.3554e-04,  ..., 6.2683e-02,
           6.1646e-02, 6.0150e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9272e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 8.3252e-02, 2.6749e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6113e-01, 2.6166e-05, 1.4365e-05,  ..., 3.9276e-02,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 6.6948e-04, 9.4461e-04,  ..., 7.1594e-02,
           3.6346e-02, 0.0000e+00],
          [7.3877e-01, 3.4642e-04, 2.6774e-04,  ..., 3.2867e-02,
           7.9346e-03, 7.5684e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.2797e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9941e-01, 6.1523e-02, 3.8879e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0400e-01, 4.3631e-05, 1.0425e-04,  ..., 3.0319e-02,
           0.0000e+00, 0.0000e+00],
          [8.1494e-01, 1.2999e-03, 3.9978e-03,  ..., 3.5839e-03,
           1.9257e-02, 0.0000e+00],
          [7.3340e-01, 4.8923e-04, 9.4652e-04,  ..., 3.1338e-03,
           5.0446e-02, 5.6427e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.0303e-01, 3.9697e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4221e-01, 3.9337e-02, 8.1836e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7842e-02, 6.7353e-06, 2.9504e-05,  ..., 8.4814e-01,
           0.0000e+00, 0.0000e+00],
          [4.2786e-02, 1.9729e-05, 7.5281e-05,  ..., 1.4610e-03,
           9.3652e-01, 0.0000e+00],
          [6.3721e-02, 3.8743e-06, 3.0339e-05,  ..., 1.8616e-03,
           2.9404e-02, 8.7988e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 1.8591e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3384e-01, 5.2148e-01, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2583e-01, 7.0751e-05, 6.4909e-05,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [7.2070e-01, 3.2783e-04, 1.1355e-04,  ..., 2.7969e-02,
           5.9753e-02, 0.0000e+00],
          [4.5215e-01, 1.0467e-04, 6.5386e-05,  ..., 1.9348e-02,
           1.2854e-01, 2.0947e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6025e-01, 2.3975e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9600e-01, 3.7933e-02, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3481e-01, 3.3970e-03, 3.4695e-03,  ..., 1.1940e-02,
           0.0000e+00, 0.0000e+00],
          [5.0244e-01, 2.7161e-02, 3.9612e-02,  ..., 2.0828e-03,
           4.4098e-02, 0.0000e+00],
          [5.2881e-01, 8.4763e-03, 1.0414e-02,  ..., 4.0359e-03,
           2.7756e-02, 8.0017e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6826e-01, 3.1921e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6621e-01, 9.6008e-02, 3.7659e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9385e-01, 1.1683e-05, 8.5235e-06,  ..., 1.1147e-02,
           0.0000e+00, 0.0000e+00],
          [2.7637e-01, 7.2975e-03, 4.6005e-03,  ..., 2.8496e-03,
           6.8016e-03, 0.0000e+00],
          [2.9980e-01, 8.8263e-04, 7.1049e-04,  ..., 5.3368e-03,
           1.2199e-02, 3.0777e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8047e-01, 1.9516e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 4.6806e-03, 4.1695e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9590e-01, 2.6882e-05, 3.3140e-05,  ..., 7.3929e-03,
           0.0000e+00, 0.0000e+00],
          [3.5767e-01, 5.5084e-03, 3.3417e-03,  ..., 2.5868e-05,
           2.0798e-02, 0.0000e+00],
          [7.7979e-01, 5.6791e-04, 6.1655e-04,  ..., 5.0449e-04,
           3.3665e-03, 1.4374e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9185e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.5375e-02, 1.3771e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2189e-01, 2.9325e-05, 8.8811e-06,  ..., 2.8305e-03,
           0.0000e+00, 0.0000e+00],
          [2.9831e-02, 4.4918e-04, 4.0293e-04,  ..., 4.7326e-04,
           1.0452e-02, 0.0000e+00],
          [6.4148e-02, 1.5223e-04, 1.0306e-04,  ..., 4.7951e-03,
           4.9400e-04, 5.5199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0820e-01, 9.2041e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6938e-01, 4.2505e-01, 2.0557e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0015e-01, 1.0192e-05, 6.6757e-06,  ..., 5.6953e-03,
           0.0000e+00, 0.0000e+00],
          [1.7822e-01, 3.3021e-04, 3.8624e-04,  ..., 7.3853e-03,
           3.6346e-02, 0.0000e+00],
          [3.0176e-01, 3.0756e-04, 3.2878e-04,  ..., 5.8861e-03,
           5.2032e-02, 4.1504e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 2.0859e-02, 2.0172e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0654e-01, 1.1480e-04, 6.2346e-05,  ..., 6.5002e-02,
           0.0000e+00, 0.0000e+00],
          [4.4873e-01, 2.7637e-03, 4.4708e-03,  ..., 3.7785e-03,
           1.0254e-01, 0.0000e+00],
          [4.8560e-01, 1.6909e-03, 1.8425e-03,  ..., 7.4310e-03,
           2.7710e-02, 5.9357e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1982e-01, 1.8005e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4075e-01, 6.6895e-01, 1.9019e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.2571e-02, 3.1590e-05, 3.2783e-06,  ..., 2.4323e-02,
           0.0000e+00, 0.0000e+00],
          [2.5955e-02, 4.2772e-04, 3.8195e-04,  ..., 3.9787e-03,
           1.4000e-02, 0.0000e+00],
          [7.0534e-03, 3.0279e-04, 6.9737e-05,  ..., 7.6866e-03,
           6.2752e-03, 1.2383e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4883e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 2.5894e-02, 4.5700e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7876e-01, 8.5831e-06, 1.0669e-05,  ..., 1.1864e-03,
           0.0000e+00, 0.0000e+00],
          [3.5181e-01, 9.8610e-04, 8.5163e-04,  ..., 4.8714e-03,
           7.3280e-03, 0.0000e+00],
          [3.6182e-01, 4.0913e-04, 3.0494e-04,  ..., 4.5395e-03,
           6.0577e-03, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7089e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9512e-01, 3.7671e-01, 1.2817e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8197e-02, 7.2718e-06, 3.8147e-06,  ..., 5.8517e-03,
           0.0000e+00, 0.0000e+00],
          [5.8008e-01, 1.2808e-03, 3.0422e-03,  ..., 7.8821e-04,
           6.6833e-02, 0.0000e+00],
          [1.1591e-01, 8.4591e-04, 8.6594e-04,  ..., 1.4477e-03,
           4.3976e-02, 3.5889e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5205e-01, 1.4807e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 3.8544e-02, 2.7390e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0225e-01, 5.3704e-05, 6.3956e-05,  ..., 3.2654e-03,
           0.0000e+00, 0.0000e+00],
          [3.5474e-01, 2.9697e-03, 1.7807e-02,  ..., 2.8753e-04,
           1.0864e-01, 0.0000e+00],
          [5.7129e-01, 2.0599e-03, 5.7335e-03,  ..., 7.6914e-04,
           1.6785e-02, 4.0283e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1113e-01, 8.8989e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 3.6831e-03, 1.8196e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5854e-01, 1.5478e-03, 2.1458e-03,  ..., 2.2507e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 3.7785e-03, 1.7462e-03,  ..., 3.5524e-05,
           7.5134e-02, 0.0000e+00],
          [1.2891e-01, 2.1935e-03, 2.9202e-03,  ..., 5.0366e-05,
           5.3613e-01, 2.4506e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.2292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.3941e-02, 1.5205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0039e-02, 4.0829e-05, 3.5167e-05,  ..., 9.9659e-04,
           0.0000e+00, 0.0000e+00],
          [6.1914e-01, 4.3945e-03, 2.2850e-03,  ..., 3.5834e-04,
           2.0203e-02, 0.0000e+00],
          [2.9346e-01, 2.2659e-03, 2.2411e-03,  ..., 1.8721e-03,
           6.6650e-02, 5.1392e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0430e-01, 9.5520e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 2.3438e-01, 3.5248e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6450e-01, 1.9193e-05, 1.6570e-05,  ..., 1.6418e-02,
           0.0000e+00, 0.0000e+00],
          [9.5154e-02, 8.0729e-04, 7.1812e-04,  ..., 1.3336e-02,
           2.1072e-02, 0.0000e+00],
          [2.7637e-01, 5.4979e-04, 7.0572e-04,  ..., 1.3496e-02,
           3.4729e-02, 5.2765e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 5.6458e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2578e-01, 4.1870e-02, 3.2440e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8301e-01, 3.9482e-04, 2.3222e-04,  ..., 2.5421e-02,
           0.0000e+00, 0.0000e+00],
          [4.8047e-01, 8.4610e-03, 1.5152e-02,  ..., 2.6627e-03,
           2.4490e-02, 0.0000e+00],
          [6.0059e-01, 2.7161e-03, 3.7785e-03,  ..., 6.0387e-03,
           2.3560e-02, 5.4352e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 1.1539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 7.8142e-05, 2.1954e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7803e-01, 5.6446e-05, 8.7738e-05,  ..., 4.8399e-04,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 9.0599e-05, 3.1495e-04,  ..., 7.8678e-05,
           4.3091e-02, 0.0000e+00],
          [9.9023e-01, 1.3995e-04, 7.5483e-04,  ..., 3.1590e-05,
           5.1594e-04, 1.7796e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.1643e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 3.2711e-03, 2.4147e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9541e-01, 4.4727e-04, 5.5075e-04,  ..., 2.0466e-03,
           0.0000e+00, 0.0000e+00],
          [8.5596e-01, 8.0633e-04, 1.2493e-03,  ..., 1.3151e-03,
           1.7593e-02, 0.0000e+00],
          [8.8428e-01, 2.3603e-04, 4.8137e-04,  ..., 1.2245e-03,
           6.9504e-03, 1.3878e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9062e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4775e-01, 2.2293e-02, 2.9999e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0419e-01, 1.7099e-03, 8.1396e-04,  ..., 2.4353e-02,
           0.0000e+00, 0.0000e+00],
          [8.4814e-01, 5.5389e-03, 4.0512e-03,  ..., 4.6074e-05,
           4.0283e-02, 0.0000e+00],
          [3.5205e-01, 8.5449e-03, 3.7918e-03,  ..., 3.6669e-04,
           1.0248e-01, 9.0454e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0576e-01, 9.3994e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3936e-01, 1.2573e-01, 3.4668e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2227e-01, 2.1210e-03, 8.9788e-04,  ..., 1.0201e-02,
           0.0000e+00, 0.0000e+00],
          [1.3062e-01, 9.1019e-03, 1.8196e-03,  ..., 5.8055e-05,
           6.5674e-02, 0.0000e+00],
          [6.8787e-02, 5.7831e-03, 1.5202e-03,  ..., 4.4250e-04,
           1.1981e-01, 1.3062e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 4.1479e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1812e-01, 2.0068e-01, 4.8120e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1179e-02, 1.8477e-06, 1.4305e-06,  ..., 6.4355e-01,
           0.0000e+00, 0.0000e+00],
          [7.6538e-02, 1.3266e-03, 2.3289e-03,  ..., 3.0384e-03,
           6.7676e-01, 0.0000e+00],
          [3.7598e-02, 8.2910e-05, 1.4901e-04,  ..., 1.6556e-02,
           1.0712e-01, 5.8838e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6436e-01, 3.5553e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0186e-01, 5.6519e-02, 4.1840e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2539e-01, 2.4676e-05, 1.6332e-05,  ..., 2.0142e-02,
           0.0000e+00, 0.0000e+00],
          [2.8516e-01, 1.3571e-03, 2.3270e-03,  ..., 3.5896e-03,
           7.4341e-02, 0.0000e+00],
          [2.5488e-01, 2.4080e-04, 3.0136e-04,  ..., 9.6512e-03,
           5.3864e-02, 6.3477e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2041e-01, 7.9773e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9336e-01, 2.0984e-01, 9.6802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1758e-01, 6.0797e-06, 3.8147e-06,  ..., 1.2070e-02,
           0.0000e+00, 0.0000e+00],
          [2.3022e-01, 6.8009e-05, 8.2672e-05,  ..., 2.5436e-02,
           3.6713e-02, 0.0000e+00],
          [4.2017e-01, 5.3823e-05, 5.3585e-05,  ..., 5.6854e-02,
           1.5549e-02, 6.8604e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0898e-01, 2.9102e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1074e-01, 1.7444e-01, 3.1470e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2852e-01, 1.2159e-05, 6.9737e-06,  ..., 2.7783e-01,
           0.0000e+00, 0.0000e+00],
          [3.7280e-01, 1.2054e-03, 1.2236e-03,  ..., 2.8019e-03,
           3.7280e-01, 0.0000e+00],
          [2.8638e-01, 3.6240e-04, 3.4046e-04,  ..., 3.8147e-02,
           9.0759e-02, 3.8818e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2104e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 1.5945e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.5059e-01, 3.5703e-05, 1.3351e-05,  ..., 5.2986e-03,
           0.0000e+00, 0.0000e+00],
          [4.9390e-01, 6.7215e-03, 3.2074e-02,  ..., 4.0591e-05,
           1.7615e-01, 0.0000e+00],
          [7.5732e-01, 1.5802e-03, 4.0359e-03,  ..., 2.0730e-04,
           3.2257e-02, 3.1250e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6904e-01, 2.3096e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 2.9419e-01, 1.2073e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8384e-01, 3.8028e-05, 3.7432e-05,  ..., 1.3115e-02,
           0.0000e+00, 0.0000e+00],
          [3.2983e-01, 2.7657e-03, 3.3894e-03,  ..., 4.9591e-04,
           3.2983e-01, 0.0000e+00],
          [1.8567e-01, 5.5552e-04, 4.2582e-04,  ..., 1.7643e-03,
           1.5881e-01, 1.3379e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6982e-01, 4.3018e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9941e-01, 6.3171e-02, 5.3760e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6294e-02, 2.8551e-05, 1.1563e-05,  ..., 7.8271e-01,
           0.0000e+00, 0.0000e+00],
          [1.4319e-01, 1.1282e-03, 1.3189e-03,  ..., 2.8062e-04,
           7.9834e-01, 0.0000e+00],
          [3.3862e-01, 2.2244e-04, 3.6097e-04,  ..., 2.0771e-03,
           1.1340e-01, 4.9268e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4229e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 7.4196e-03, 5.3024e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4153e-02, 1.3828e-05, 2.9802e-06,  ..., 1.2760e-03,
           0.0000e+00, 0.0000e+00],
          [5.6152e-01, 3.8818e-02, 1.7090e-02,  ..., 2.4438e-05,
           1.1145e-01, 0.0000e+00],
          [3.9551e-01, 1.4748e-02, 8.8043e-03,  ..., 6.8998e-04,
           2.8870e-02, 1.3147e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.5596e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 6.2904e-03, 3.7556e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6853e-02, 2.1338e-05, 9.7752e-06,  ..., 7.0333e-04,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 2.2755e-03, 2.8534e-03,  ..., 6.1214e-05,
           3.0930e-02, 0.0000e+00],
          [6.6650e-01, 1.4524e-03, 1.6136e-03,  ..., 4.7874e-04,
           2.4078e-02, 2.7298e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 2.9205e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.2842e-02, 9.3765e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7896e-01, 3.7789e-04, 2.2304e-04,  ..., 5.7068e-03,
           0.0000e+00, 0.0000e+00],
          [4.0869e-01, 9.6054e-03, 9.7580e-03,  ..., 2.3222e-04,
           9.4788e-02, 0.0000e+00],
          [3.8306e-01, 5.2338e-03, 5.8403e-03,  ..., 7.3957e-04,
           6.0608e-02, 2.1606e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 4.3518e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 2.9694e-02, 2.4429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6895e-01, 1.2517e-05, 1.1802e-05,  ..., 6.3110e-02,
           0.0000e+00, 0.0000e+00],
          [4.5386e-01, 3.1972e-04, 3.5954e-04,  ..., 1.2383e-02,
           6.3843e-02, 0.0000e+00],
          [2.8516e-01, 3.5262e-04, 2.5296e-04,  ..., 1.8097e-02,
           9.4788e-02, 1.7163e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5195e-01, 2.4805e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7266e-01, 1.7960e-02, 9.2468e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.6533e-01, 3.9935e-05, 2.2769e-05,  ..., 2.9087e-03,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 2.7790e-03, 1.3542e-03,  ..., 4.8370e-03,
           1.8402e-02, 0.0000e+00],
          [9.3652e-01, 2.0485e-03, 5.7364e-04,  ..., 6.4125e-03,
           1.2848e-02, 6.9847e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5488e-01, 2.4512e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 1.5312e-02, 7.0686e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7188e-01, 3.9637e-05, 1.4961e-05,  ..., 1.7761e-02,
           0.0000e+00, 0.0000e+00],
          [8.4229e-01, 3.8395e-03, 3.4161e-03,  ..., 1.6037e-02,
           3.7018e-02, 0.0000e+00],
          [8.4668e-01, 9.9945e-04, 5.5170e-04,  ..., 1.1986e-02,
           1.6769e-02, 1.4458e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 5.2612e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 5.1788e-02, 4.4312e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7126e-01, 2.5005e-03, 3.7251e-03,  ..., 1.6174e-02,
           0.0000e+00, 0.0000e+00],
          [2.4719e-01, 4.8943e-03, 5.7220e-03,  ..., 1.2350e-04,
           2.2864e-01, 0.0000e+00],
          [1.9580e-01, 1.8167e-03, 2.6436e-03,  ..., 5.9462e-04,
           3.4888e-01, 1.5491e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2285e-01, 7.6965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 7.0801e-02, 3.9093e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1562e-01, 3.0518e-05, 4.6909e-05,  ..., 3.3508e-02,
           0.0000e+00, 0.0000e+00],
          [1.8628e-01, 1.5869e-03, 9.0408e-04,  ..., 1.2878e-02,
           4.1565e-02, 0.0000e+00],
          [3.2349e-01, 1.3506e-04, 1.0270e-04,  ..., 6.1737e-02,
           6.2561e-03, 8.1787e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8682e-01, 1.3222e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5850e-01, 1.9882e-02, 2.1835e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 1.5745e-03, 3.5477e-03,  ..., 3.1548e-03,
           0.0000e+00, 0.0000e+00],
          [4.9268e-01, 6.1035e-03, 1.0147e-02,  ..., 5.8126e-04,
           2.7573e-02, 0.0000e+00],
          [3.7793e-01, 3.9749e-03, 8.4152e-03,  ..., 9.2936e-04,
           7.0984e-02, 6.1676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.5778e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 9.3384e-03, 1.5396e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7810e-02, 1.3888e-05, 1.4484e-05,  ..., 1.5610e-02,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 7.9584e-04, 2.1973e-03,  ..., 2.3532e-04,
           1.1450e-01, 0.0000e+00],
          [4.2407e-01, 8.5115e-04, 2.0103e-03,  ..., 2.5215e-03,
           2.6489e-02, 3.8849e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3359e-01, 6.6589e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3457e-01, 3.5675e-02, 2.9800e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1536e-01, 4.5598e-05, 7.8797e-05,  ..., 7.1449e-03,
           0.0000e+00, 0.0000e+00],
          [4.4360e-01, 1.9302e-03, 3.6621e-03,  ..., 4.4346e-05,
           1.2512e-01, 0.0000e+00],
          [1.9507e-01, 7.9107e-04, 2.2888e-03,  ..., 5.5647e-04,
           4.8553e-02, 1.2018e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.8982e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 2.3163e-02, 2.9968e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9580e-02, 4.7684e-06, 1.0788e-05,  ..., 4.1771e-03,
           0.0000e+00, 0.0000e+00],
          [5.6396e-01, 8.4782e-04, 2.0504e-03,  ..., 4.6086e-04,
           1.2988e-01, 0.0000e+00],
          [4.4458e-01, 6.0415e-04, 2.2621e-03,  ..., 1.9951e-03,
           5.5664e-02, 2.8656e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9590e-01, 2.0435e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 1.4832e-01, 1.1023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.5969e-02, 5.1022e-05, 3.8505e-05,  ..., 1.6541e-02,
           0.0000e+00, 0.0000e+00],
          [9.4238e-02, 1.5717e-03, 2.9812e-03,  ..., 4.1628e-04,
           3.8666e-02, 0.0000e+00],
          [9.1629e-03, 1.8430e-04, 3.9029e-04,  ..., 2.7943e-03,
           8.6060e-03, 9.1125e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.3697e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1064e-01, 3.6438e-02, 5.3009e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9820e-03, 9.4771e-06, 2.1458e-06,  ..., 9.8133e-04,
           0.0000e+00, 0.0000e+00],
          [6.6064e-01, 1.0242e-03, 1.2951e-03,  ..., 1.7250e-04,
           3.0411e-02, 0.0000e+00],
          [5.1758e-01, 1.3876e-03, 1.4544e-03,  ..., 5.3177e-03,
           6.9389e-03, 1.0034e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3148e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 1.2321e-02, 1.6708e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4834e-01, 4.3333e-05, 4.2319e-05,  ..., 3.9940e-03,
           0.0000e+00, 0.0000e+00],
          [5.9619e-01, 1.0309e-03, 2.2526e-03,  ..., 3.0971e-04,
           6.7253e-03, 0.0000e+00],
          [6.5918e-01, 1.6994e-03, 2.9125e-03,  ..., 2.6679e-04,
           6.5651e-03, 1.5747e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5693e-01, 8.1604e-02, 6.1584e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1462e-02, 1.0133e-06, 7.7486e-07,  ..., 1.9150e-03,
           0.0000e+00, 0.0000e+00],
          [2.4402e-01, 7.3528e-04, 7.2432e-04,  ..., 1.7047e-05,
           2.3056e-02, 0.0000e+00],
          [2.7441e-01, 2.4629e-04, 2.7800e-04,  ..., 1.2407e-03,
           1.5358e-02, 1.2366e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 2.4612e-02, 3.7231e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.4126e-05, 2.5272e-05,  ..., 6.4621e-03,
           0.0000e+00, 0.0000e+00],
          [3.1274e-01, 8.4496e-04, 1.1816e-03,  ..., 6.9141e-06,
           1.0559e-01, 0.0000e+00],
          [1.7444e-01, 3.1638e-04, 6.9666e-04,  ..., 1.7071e-04,
           1.7310e-01, 5.1971e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3643e-01, 1.6345e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 4.1565e-02, 1.6821e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4749e-02, 2.3961e-05, 3.7551e-05,  ..., 3.5962e-01,
           0.0000e+00, 0.0000e+00],
          [5.1367e-01, 2.5272e-04, 6.5041e-04,  ..., 7.4148e-05,
           2.3157e-01, 0.0000e+00],
          [4.0527e-01, 9.2685e-05, 3.5524e-04,  ..., 1.0281e-03,
           5.4413e-02, 4.0527e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 3.0334e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2139e-01, 2.2186e-02, 5.6213e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1558e-01, 1.1325e-05, 9.8348e-06,  ..., 5.4398e-03,
           0.0000e+00, 0.0000e+00],
          [4.7632e-01, 4.8280e-04, 1.6270e-03,  ..., 1.4210e-04,
           5.8228e-02, 0.0000e+00],
          [7.3438e-01, 2.7370e-04, 6.4421e-04,  ..., 1.5631e-03,
           2.0035e-02, 4.8065e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5752e-01, 4.2389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.0071e-02, 2.4933e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7295e-01, 9.0599e-06, 3.4213e-05,  ..., 8.5068e-03,
           0.0000e+00, 0.0000e+00],
          [6.2012e-01, 5.7888e-04, 3.1281e-03,  ..., 7.9572e-05,
           4.2175e-02, 0.0000e+00],
          [7.6953e-01, 2.3699e-04, 2.1458e-03,  ..., 4.3583e-04,
           6.2065e-03, 1.7548e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 2.2293e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 1.7273e-02, 4.6600e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5332e-01, 4.1723e-06, 5.0664e-06,  ..., 2.9888e-03,
           0.0000e+00, 0.0000e+00],
          [3.0103e-01, 3.6163e-03, 1.8951e-02,  ..., 1.1998e-04,
           5.0720e-02, 0.0000e+00],
          [5.7227e-01, 3.3779e-03, 8.6899e-03,  ..., 9.3079e-04,
           1.0239e-02, 3.9276e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5459e-01, 4.5349e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1699e-01, 4.1901e-02, 4.0924e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9651e-02, 2.4974e-05, 1.5259e-05,  ..., 9.6664e-03,
           0.0000e+00, 0.0000e+00],
          [3.6499e-01, 1.8272e-03, 3.1204e-03,  ..., 1.0090e-03,
           7.5928e-02, 0.0000e+00],
          [4.6411e-01, 6.3515e-04, 5.6934e-04,  ..., 9.3689e-03,
           1.6388e-02, 4.4525e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 3.2837e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 1.8631e-02, 2.2385e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6365e-02, 1.5736e-05, 7.3910e-06,  ..., 4.3091e-02,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 4.6635e-04, 9.4604e-04,  ..., 3.8362e-04,
           1.4122e-02, 0.0000e+00],
          [9.8511e-02, 9.9421e-05, 9.1255e-05,  ..., 2.4139e-02,
           3.2158e-03, 2.1805e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 3.6896e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5605e-01, 1.5640e-02, 2.8427e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9241e-02, 2.9802e-07, 1.1921e-07,  ..., 8.9765e-05,
           0.0000e+00, 0.0000e+00],
          [5.4541e-01, 2.1172e-03, 4.0665e-03,  ..., 3.0637e-04,
           4.6082e-03, 0.0000e+00],
          [5.3662e-01, 2.1420e-03, 3.4637e-03,  ..., 2.3441e-03,
           6.9733e-03, 2.0981e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 7.6370e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8232e-01, 1.0956e-01, 8.1253e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0625e-01, 1.9312e-05, 2.3842e-06,  ..., 2.0340e-02,
           0.0000e+00, 0.0000e+00],
          [6.5332e-01, 6.7055e-05, 4.8161e-05,  ..., 9.4116e-02,
           2.2354e-02, 0.0000e+00],
          [6.1279e-01, 6.9737e-06, 1.1003e-04,  ..., 9.6924e-02,
           1.0986e-01, 3.1219e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5801e-01, 4.2084e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 6.5430e-02, 5.9113e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9766e-01, 3.5524e-05, 1.2350e-04,  ..., 6.7627e-02,
           0.0000e+00, 0.0000e+00],
          [5.2002e-01, 3.7503e-04, 1.2884e-03,  ..., 2.0157e-02,
           3.5370e-02, 0.0000e+00],
          [4.6948e-01, 8.4937e-05, 2.5558e-04,  ..., 7.3120e-02,
           6.7627e-02, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7095e-01, 4.7461e-01, 5.4504e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4551e-01, 1.3769e-04, 5.2869e-05,  ..., 8.4656e-02,
           0.0000e+00, 0.0000e+00],
          [5.7861e-01, 2.4533e-04, 2.4343e-04,  ..., 2.3132e-02,
           9.3689e-02, 0.0000e+00],
          [5.0049e-01, 5.1618e-05, 3.6693e-04,  ..., 3.5431e-02,
           2.8516e-01, 9.7046e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9443e-01, 2.0557e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9395e-01, 7.4158e-02, 1.3171e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8687e-01, 4.1127e-06, 7.3910e-06,  ..., 2.2995e-02,
           0.0000e+00, 0.0000e+00],
          [4.9683e-02, 4.5395e-04, 6.8426e-04,  ..., 2.4676e-04,
           3.7781e-02, 0.0000e+00],
          [2.8351e-02, 8.2493e-05, 3.2616e-04,  ..., 4.1466e-03,
           6.5369e-02, 8.8684e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 5.6396e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1798e-01, 3.7109e-02, 8.4473e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5732e-02, 2.0266e-05, 5.9903e-05,  ..., 5.3320e-01,
           0.0000e+00, 0.0000e+00],
          [7.4951e-02, 1.6975e-04, 4.6806e-03,  ..., 9.7504e-03,
           7.2803e-01, 0.0000e+00],
          [1.4099e-02, 9.6560e-06, 1.3709e-04,  ..., 4.1656e-03,
           1.9424e-02, 8.9990e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9746e-01, 5.1025e-02, 5.1422e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.3450e-01, 8.0645e-05, 3.5667e-04,  ..., 2.0523e-03,
           0.0000e+00, 0.0000e+00],
          [5.3271e-01, 1.9369e-03, 5.4741e-03,  ..., 1.0133e-06,
           1.1147e-02, 0.0000e+00],
          [1.1969e-01, 3.5357e-04, 1.4715e-03,  ..., 2.9802e-06,
           4.9286e-02, 1.5762e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 6.3721e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7451e-01, 8.0078e-02, 4.5349e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4150e-01, 4.5943e-04, 7.5245e-04,  ..., 3.0975e-03,
           0.0000e+00, 0.0000e+00],
          [5.3906e-01, 2.9011e-03, 4.7455e-03,  ..., 3.1986e-03,
           7.6721e-02, 0.0000e+00],
          [2.6514e-01, 3.3703e-03, 6.2599e-03,  ..., 2.6245e-03,
           2.7246e-01, 1.1182e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 1.9073e-02, 1.3664e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7051e-01, 3.7551e-06, 7.1526e-06,  ..., 3.5065e-02,
           0.0000e+00, 0.0000e+00],
          [3.8818e-01, 2.1946e-04, 3.0231e-04,  ..., 1.4992e-02,
           2.9465e-02, 0.0000e+00],
          [1.4966e-01, 8.6665e-05, 1.0997e-04,  ..., 1.2871e-02,
           1.1009e-02, 6.6406e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 5.9326e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 2.3877e-01, 8.6182e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8350e-01, 7.1645e-05, 5.6565e-05,  ..., 1.6455e-01,
           0.0000e+00, 0.0000e+00],
          [3.2397e-01, 4.9639e-04, 5.0449e-04,  ..., 7.2815e-02,
           1.6675e-01, 0.0000e+00],
          [1.5173e-01, 4.5180e-04, 7.4053e-04,  ..., 1.3260e-02,
           4.8779e-01, 1.7053e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 2.1744e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4385e-01, 2.0721e-02, 3.5614e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0918e-01, 1.2314e-04, 4.3511e-04,  ..., 2.9516e-04,
           0.0000e+00, 0.0000e+00],
          [8.8428e-01, 2.8133e-04, 7.0190e-04,  ..., 3.1567e-04,
           2.4796e-02, 0.0000e+00],
          [8.0322e-01, 1.5533e-04, 7.5722e-04,  ..., 6.3133e-04,
           5.1361e-02, 4.0009e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 9.3651e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 3.5620e-04, 1.9112e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.8877e-01, 6.7353e-06, 2.6643e-05,  ..., 9.9599e-05,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.5010e-04, 1.6756e-03,  ..., 2.5105e-04,
           2.6569e-03, 0.0000e+00],
          [9.7803e-01, 8.9347e-05, 5.0402e-04,  ..., 1.9288e-04,
           5.1308e-03, 3.3646e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2672e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 1.0696e-02, 1.5381e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2666e-01, 2.4533e-04, 1.6367e-04,  ..., 1.1768e-03,
           0.0000e+00, 0.0000e+00],
          [7.2168e-01, 1.3199e-03, 6.1989e-03,  ..., 3.9406e-03,
           2.2842e-02, 0.0000e+00],
          [3.6719e-01, 5.2691e-04, 2.2964e-03,  ..., 8.4381e-03,
           1.5396e-02, 1.4050e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3634e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8770e-01, 5.2887e-02, 5.9479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 3.0279e-05, 2.2173e-05,  ..., 4.9782e-03,
           0.0000e+00, 0.0000e+00],
          [5.3467e-01, 4.6158e-04, 8.5592e-04,  ..., 3.6316e-03,
           5.2948e-02, 0.0000e+00],
          [2.7100e-01, 1.8227e-04, 2.4724e-04,  ..., 8.9874e-03,
           3.0899e-02, 2.9251e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9119e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 8.5449e-03, 1.1497e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5498e-01, 1.2517e-06, 3.5763e-07,  ..., 5.7373e-03,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 6.1941e-04, 9.7466e-04,  ..., 1.0727e-02,
           3.8025e-02, 0.0000e+00],
          [3.4277e-01, 1.3447e-04, 6.9201e-05,  ..., 4.0314e-02,
           1.1368e-02, 3.5004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4756e-01, 2.5244e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.7249e-01, 7.8369e-02, 7.4902e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.3691e-01, 5.5492e-05, 2.1112e-04,  ..., 2.7930e-01,
           0.0000e+00, 0.0000e+00],
          [1.8079e-01, 1.1806e-03, 1.4381e-02,  ..., 2.0714e-03,
           7.0410e-01, 0.0000e+00],
          [1.0321e-01, 2.8086e-04, 1.3199e-03,  ..., 5.4283e-03,
           1.3049e-01, 7.0508e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 1.2463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5977e-01, 1.5930e-01, 8.0750e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1270e-01, 5.6028e-05, 6.8963e-05,  ..., 3.0548e-02,
           0.0000e+00, 0.0000e+00],
          [4.4849e-01, 9.7930e-05, 1.9217e-04,  ..., 5.1727e-03,
           5.0934e-02, 0.0000e+00],
          [1.6614e-01, 2.4796e-05, 7.1108e-05,  ..., 3.2837e-02,
           9.5032e-02, 1.1420e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5703e-01, 4.2816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0674e-01, 6.1218e-02, 3.1952e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4033e-01, 1.4997e-04, 6.3002e-05,  ..., 7.7858e-03,
           0.0000e+00, 0.0000e+00],
          [5.2393e-01, 8.6963e-05, 2.1589e-04,  ..., 1.8860e-02,
           3.4943e-02, 0.0000e+00],
          [5.3027e-01, 4.5538e-05, 4.5717e-05,  ..., 4.3854e-02,
           1.8936e-02, 2.3941e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4629e-01, 5.3802e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 3.6652e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5708e-01, 5.0831e-04, 2.7394e-04,  ..., 2.8748e-02,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.4211e-04, 6.2275e-04,  ..., 2.6722e-03,
           1.8082e-02, 0.0000e+00],
          [5.7715e-01, 3.7193e-04, 9.0504e-04,  ..., 7.2098e-03,
           2.7206e-02, 3.2074e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 7.5867e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8672e-01, 4.6631e-02, 6.6772e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3193e-01, 6.4850e-05, 2.0719e-04,  ..., 9.5444e-03,
           0.0000e+00, 0.0000e+00],
          [6.8945e-01, 1.8442e-04, 2.1954e-03,  ..., 9.2087e-03,
           1.7670e-02, 0.0000e+00],
          [2.2400e-01, 2.2697e-04, 1.3962e-03,  ..., 7.8506e-03,
           3.7140e-02, 6.5735e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 5.3986e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8721e-01, 2.6581e-02, 8.6487e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3711e-01, 4.2200e-05, 7.4446e-05,  ..., 1.3451e-02,
           0.0000e+00, 0.0000e+00],
          [7.9297e-01, 4.9686e-04, 4.7493e-03,  ..., 1.4048e-03,
           4.2999e-02, 0.0000e+00],
          [6.9482e-01, 4.0126e-04, 5.1613e-03,  ..., 5.8250e-03,
           4.3396e-02, 7.2083e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2471e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.4198e-02, 2.0981e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4639e-01, 5.7220e-06, 2.1517e-05,  ..., 2.6566e-02,
           0.0000e+00, 0.0000e+00],
          [7.1533e-01, 4.3440e-04, 2.2240e-03,  ..., 8.8654e-03,
           8.9539e-02, 0.0000e+00],
          [5.4541e-01, 1.9026e-04, 7.2384e-04,  ..., 9.1629e-03,
           3.9825e-02, 1.5259e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0527e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 5.4626e-03, 4.7493e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.6234e-02, 1.6093e-06, 5.3644e-07,  ..., 1.6617e-02,
           0.0000e+00, 0.0000e+00],
          [1.4441e-01, 6.4039e-04, 8.5831e-04,  ..., 2.6550e-03,
           6.5193e-03, 0.0000e+00],
          [4.7989e-03, 1.9372e-05, 1.5616e-05,  ..., 1.6098e-02,
           3.9077e-04, 2.2144e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7363e-01, 2.6154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.6129e-02, 1.3260e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1104e-01, 5.8413e-06, 1.7881e-06,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 5.2691e-04, 1.3542e-03,  ..., 2.5902e-03,
           4.5815e-03, 0.0000e+00],
          [2.5073e-01, 7.8559e-05, 1.1277e-04,  ..., 3.2898e-02,
           2.7752e-03, 1.4656e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 5.3192e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 1.7670e-02, 3.2623e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2681e-01, 2.6345e-05, 4.5359e-05,  ..., 1.5556e-02,
           0.0000e+00, 0.0000e+00],
          [3.1494e-01, 1.6088e-03, 7.0953e-03,  ..., 3.2997e-03,
           3.9124e-02, 0.0000e+00],
          [1.0626e-01, 4.4107e-04, 1.8139e-03,  ..., 1.4267e-02,
           2.1088e-02, 8.0200e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.4283e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 3.8147e-03, 1.7914e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4053e-01, 5.4002e-05, 4.2820e-04,  ..., 1.5907e-03,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 8.9455e-04, 1.7786e-03,  ..., 4.8637e-04,
           3.4088e-02, 0.0000e+00],
          [4.0918e-01, 1.9348e-04, 1.2426e-03,  ..., 6.6519e-04,
           1.8152e-01, 1.1359e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0178e-01, 7.9834e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1455e-01, 3.5461e-02, 5.0018e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9814e-01, 1.0085e-04, 2.5749e-04,  ..., 1.0948e-02,
           0.0000e+00, 0.0000e+00],
          [4.1309e-01, 3.9253e-03, 3.8643e-03,  ..., 5.4359e-04,
           1.0944e-01, 0.0000e+00],
          [1.2842e-01, 7.5197e-04, 2.4261e-03,  ..., 6.4325e-04,
           1.4331e-01, 1.9897e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6680e-01, 3.3081e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 6.8245e-03, 1.1429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0029e-01, 1.9479e-04, 3.3140e-04,  ..., 4.5052e-03,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 8.2254e-04, 1.0080e-03,  ..., 2.0199e-03,
           1.8295e-02, 0.0000e+00],
          [5.8594e-01, 1.0633e-03, 1.4753e-03,  ..., 3.1738e-03,
           5.1208e-02, 4.0192e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 9.6191e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5049e-01, 2.7878e-02, 2.2144e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1133e-01, 2.6643e-05, 9.9838e-05,  ..., 2.5482e-02,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 8.0407e-05, 1.0598e-04,  ..., 8.1940e-03,
           8.4351e-02, 0.0000e+00],
          [7.1582e-01, 5.2452e-05, 7.2598e-05,  ..., 7.6752e-03,
           7.2815e-02, 9.5703e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5259e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 2.7115e-02, 3.2013e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5386e-01, 6.5327e-05, 1.9968e-04,  ..., 4.7546e-02,
           0.0000e+00, 0.0000e+00],
          [6.0596e-01, 1.8382e-04, 2.3282e-04,  ..., 2.2736e-02,
           1.1841e-01, 0.0000e+00],
          [3.6572e-01, 9.9838e-05, 2.0587e-04,  ..., 2.1545e-02,
           1.5918e-01, 1.8896e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0619e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 1.5454e-01, 1.6382e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9565e-01, 2.8396e-04, 3.0339e-05,  ..., 9.0837e-04,
           0.0000e+00, 0.0000e+00],
          [6.0254e-01, 2.1160e-04, 1.4949e-04,  ..., 1.0925e-02,
           2.1484e-01, 0.0000e+00],
          [1.5747e-01, 3.8922e-05, 1.6475e-04,  ..., 1.1314e-02,
           6.0840e-01, 1.8701e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 2.5375e-02, 4.4525e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6221e-01, 2.5809e-05, 6.7949e-05,  ..., 9.8495e-03,
           0.0000e+00, 0.0000e+00],
          [7.9736e-01, 8.5688e-04, 1.6909e-03,  ..., 1.2913e-03,
           2.2263e-02, 0.0000e+00],
          [4.0845e-01, 9.1076e-04, 2.4853e-03,  ..., 1.4534e-02,
           2.6520e-02, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0723e-01, 9.2712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5264e-01, 2.9221e-02, 1.8066e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7842e-01, 1.2875e-05, 1.0252e-05,  ..., 6.8665e-03,
           0.0000e+00, 0.0000e+00],
          [6.1719e-01, 7.2122e-05, 1.3208e-04,  ..., 2.2621e-03,
           1.2611e-02, 0.0000e+00],
          [2.1655e-01, 8.5592e-05, 1.8477e-04,  ..., 1.5686e-02,
           2.8412e-02, 3.5339e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2935e-01, 6.7041e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3538e-01, 5.4688e-02, 8.1006e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8356e-02, 4.1187e-05, 1.0747e-04,  ..., 1.8835e-01,
           0.0000e+00, 0.0000e+00],
          [8.8684e-02, 1.6756e-03, 4.2953e-03,  ..., 7.8735e-03,
           5.0635e-01, 0.0000e+00],
          [2.4338e-02, 4.1223e-04, 8.3590e-04,  ..., 1.8120e-03,
           2.6108e-02, 8.0566e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1070e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 3.5572e-03, 1.2054e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2080e-01, 7.5519e-05, 2.0230e-04,  ..., 7.9575e-03,
           0.0000e+00, 0.0000e+00],
          [3.7134e-01, 4.2152e-04, 1.0185e-03,  ..., 6.4125e-03,
           2.6538e-01, 0.0000e+00],
          [2.1777e-01, 5.1641e-04, 1.3180e-03,  ..., 1.0391e-02,
           2.9761e-01, 1.6565e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 6.4636e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 9.7046e-03, 1.9760e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7295e-01, 2.4891e-04, 3.6287e-04,  ..., 1.7786e-03,
           0.0000e+00, 0.0000e+00],
          [1.2457e-01, 7.4501e-03, 8.4763e-03,  ..., 1.0262e-02,
           4.9164e-02, 0.0000e+00],
          [8.9600e-02, 5.9128e-03, 7.1297e-03,  ..., 1.5640e-02,
           3.5919e-02, 6.0638e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 6.6101e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3164e-01, 2.3682e-02, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1790e-01, 4.9412e-05, 1.5533e-04,  ..., 1.0071e-02,
           0.0000e+00, 0.0000e+00],
          [2.7563e-01, 1.5154e-03, 2.8095e-03,  ..., 2.0920e-02,
           1.2016e-02, 0.0000e+00],
          [1.7615e-01, 6.8426e-04, 1.2054e-03,  ..., 2.2659e-02,
           7.6180e-03, 3.1342e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 5.7259e-03, 8.5983e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5244e-01, 1.5459e-03, 1.4296e-03,  ..., 5.2299e-03,
           0.0000e+00, 0.0000e+00],
          [7.2266e-01, 2.4109e-03, 1.1568e-03,  ..., 1.1749e-03,
           3.5431e-02, 0.0000e+00],
          [6.1523e-01, 2.9802e-04, 3.5930e-04,  ..., 2.2011e-03,
           2.0569e-02, 1.6266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.0000e-01, 5.0000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7783e-01, 2.0325e-01, 5.1904e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8931e-01, 5.6744e-04, 1.5421e-03,  ..., 2.7173e-01,
           0.0000e+00, 0.0000e+00],
          [2.0471e-01, 1.7710e-03, 3.8071e-03,  ..., 3.1891e-02,
           4.8340e-01, 0.0000e+00],
          [1.4502e-01, 1.0405e-03, 2.6569e-03,  ..., 1.9623e-02,
           4.4220e-02, 6.4014e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 1.2646e-03, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6318e-01, 2.1219e-04, 6.0940e-04,  ..., 6.3972e-03,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1711e-03, 1.8721e-03,  ..., 2.6398e-03,
           2.5848e-02, 0.0000e+00],
          [5.6934e-01, 4.9162e-04, 1.6499e-03,  ..., 9.2010e-03,
           4.5319e-02, 2.4246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1948e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 4.9095e-03, 1.4539e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.5020e-01, 6.7353e-06, 5.0247e-05,  ..., 2.2709e-04,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1849e-04, 6.1274e-04,  ..., 4.6196e-03,
           1.4488e-02, 0.0000e+00],
          [6.5820e-01, 3.7789e-05, 6.6280e-04,  ..., 4.9362e-03,
           1.2032e-02, 1.0803e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4404e-01, 3.5571e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9131e-01, 5.2490e-02, 3.5596e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7656e-01, 6.4969e-06, 5.3048e-05,  ..., 1.5879e-04,
           0.0000e+00, 0.0000e+00],
          [6.2500e-01, 2.0123e-04, 9.0075e-04,  ..., 4.9362e-03,
           4.8370e-02, 0.0000e+00],
          [6.8909e-02, 9.5665e-05, 9.2685e-05,  ..., 1.5344e-01,
           1.3206e-02, 3.6041e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2650e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 1.1169e-02, 1.5039e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 2.4140e-05, 1.2863e-04,  ..., 7.5877e-05,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 7.9298e-04, 2.2221e-03,  ..., 1.3018e-03,
           1.5222e-01, 0.0000e+00],
          [2.4585e-01, 1.4963e-03, 1.7529e-03,  ..., 1.8845e-02,
           1.4062e-01, 9.5520e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 2.1008e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 7.6752e-03, 1.8021e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1904e-01, 1.3628e-03, 6.5956e-03,  ..., 1.3062e-02,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 2.8276e-04, 4.9286e-03,  ..., 3.7117e-03,
           8.3313e-02, 0.0000e+00],
          [7.6221e-01, 2.5392e-04, 8.5449e-04,  ..., 4.5967e-03,
           2.2232e-02, 1.2665e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8613e-01, 2.1399e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6270e-01, 4.6967e-02, 1.9055e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8018e-01, 2.0623e-05, 2.5196e-03,  ..., 4.9362e-03,
           0.0000e+00, 0.0000e+00],
          [6.9824e-01, 5.7793e-04, 2.3766e-03,  ..., 1.3794e-02,
           1.6260e-01, 0.0000e+00],
          [8.5742e-01, 3.8981e-05, 8.1253e-04,  ..., 5.4283e-03,
           1.7593e-02, 9.6207e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.9866e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 7.7744e-03, 1.5717e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8262e-01, 4.0913e-04, 3.6359e-04,  ..., 1.0757e-02,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 1.3466e-03, 9.7227e-04,  ..., 6.9122e-03,
           4.9164e-02, 0.0000e+00],
          [5.0635e-01, 1.9236e-03, 1.8082e-03,  ..., 8.9111e-03,
           5.0049e-02, 2.6520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-02, 9.1064e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3428e-02, 1.3351e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1062e-06, 1.1265e-05, 3.2397e-01,  ..., 9.3174e-04,
           0.0000e+00, 0.0000e+00],
          [8.9836e-04, 3.8414e-03, 1.4026e-01,  ..., 1.5793e-02,
           8.2458e-02, 0.0000e+00],
          [5.6839e-04, 1.0252e-03, 1.7236e-01,  ..., 2.5314e-02,
           3.1769e-02, 3.9978e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8296e-01, 7.1680e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9276e-02, 3.0838e-02, 9.2969e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0121e-04, 2.0444e-04, 2.9248e-01,  ..., 8.4229e-03,
           0.0000e+00, 0.0000e+00],
          [4.5538e-04, 1.4811e-03, 2.6465e-01,  ..., 4.6806e-03,
           6.3843e-02, 0.0000e+00],
          [5.2738e-04, 2.9683e-04, 1.6028e-01,  ..., 9.5139e-03,
           1.1604e-02, 1.2947e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7424e-02, 9.5264e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.0131e-02, 4.1473e-02, 9.1846e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7115e-05, 4.3809e-05, 7.9041e-03,  ..., 1.5087e-03,
           0.0000e+00, 0.0000e+00],
          [1.3237e-03, 7.2670e-03, 8.8867e-02,  ..., 1.1566e-02,
           2.4927e-01, 0.0000e+00],
          [1.0719e-03, 2.0542e-03, 1.6162e-01,  ..., 1.2657e-02,
           4.2145e-02, 3.6041e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7178e-01, 4.2822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5845e-01, 8.5510e-02, 7.5586e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2214e-04, 6.2990e-04, 1.2335e-01,  ..., 5.9631e-02,
           0.0000e+00, 0.0000e+00],
          [2.5978e-03, 2.4986e-03, 2.4036e-01,  ..., 2.1164e-02,
           2.0557e-01, 0.0000e+00],
          [3.3307e-04, 1.8907e-04, 9.3811e-02,  ..., 1.1584e-01,
           2.5708e-01, 2.3035e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1567e-01, 6.8408e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1864e-02, 4.6005e-03, 9.8340e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1501e-03, 5.1260e-04, 3.1519e-01,  ..., 9.6130e-04,
           0.0000e+00, 0.0000e+00],
          [2.1517e-04, 7.9155e-04, 3.2300e-01,  ..., 5.7030e-04,
           6.8237e-02, 0.0000e+00],
          [6.0797e-04, 8.7070e-04, 2.7783e-01,  ..., 7.6866e-04,
           4.1321e-02, 5.1819e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9600e-01, 6.0400e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.2267e-03, 1.5068e-03, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.4392e-05, 6.9737e-05, 3.6182e-01,  ..., 2.8564e-02,
           0.0000e+00, 0.0000e+00],
          [2.2042e-04, 1.8132e-04, 2.3608e-01,  ..., 1.7242e-02,
           4.4037e-02, 0.0000e+00],
          [1.7631e-04, 2.9147e-05, 1.1548e-01,  ..., 1.9455e-02,
           7.5569e-03, 3.1830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7598e-01, 1.2421e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0109e-03, 2.2697e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6716e-05, 8.6427e-06, 4.1968e-01,  ..., 1.0258e-04,
           0.0000e+00, 0.0000e+00],
          [7.7188e-05, 7.0572e-05, 3.6914e-01,  ..., 1.9836e-03,
           8.6823e-03, 0.0000e+00],
          [3.5357e-04, 1.2058e-04, 2.2266e-01,  ..., 8.6670e-03,
           4.5166e-03, 2.2049e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 3.0493e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.7308e-03, 1.6651e-03, 9.9463e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7578e-05, 1.5676e-05, 2.8076e-01,  ..., 7.0953e-04,
           0.0000e+00, 0.0000e+00],
          [1.5128e-04, 8.7261e-05, 2.8662e-01,  ..., 1.7899e-02,
           4.7546e-02, 0.0000e+00],
          [1.8680e-04, 1.0645e-04, 1.6724e-01,  ..., 1.8188e-02,
           1.2596e-02, 1.1925e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2373e-01, 1.7615e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2567e-03, 1.0767e-03, 9.9365e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9339e-05, 5.7220e-06, 3.9648e-01,  ..., 1.5335e-03,
           0.0000e+00, 0.0000e+00],
          [6.9559e-05, 1.3769e-05, 3.2861e-01,  ..., 1.5137e-02,
           4.6997e-02, 0.0000e+00],
          [9.7513e-05, 2.0146e-05, 3.1543e-01,  ..., 1.1139e-02,
           9.2697e-03, 2.0416e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6975e-03, 1.5202e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0320e-04, 3.0398e-05, 1.4526e-01,  ..., 5.1785e-04,
           0.0000e+00, 0.0000e+00],
          [2.1303e-04, 9.0718e-05, 3.5400e-01,  ..., 1.4582e-03,
           1.4221e-02, 0.0000e+00],
          [7.7152e-04, 1.9884e-04, 1.8652e-01,  ..., 5.6992e-03,
           2.2678e-03, 1.1421e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-01, 1.0895e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7023e-04, 9.5034e-04, 9.9805e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2994e-05, 8.5235e-06, 3.7646e-01,  ..., 3.2258e-04,
           0.0000e+00, 0.0000e+00],
          [2.4855e-05, 2.6584e-05, 3.2129e-01,  ..., 1.5266e-02,
           6.0959e-03, 0.0000e+00],
          [1.9252e-05, 2.9683e-05, 2.4817e-01,  ..., 7.5531e-03,
           1.2617e-03, 5.5237e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8086e-01, 1.1920e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3733e-04, 5.1260e-04, 9.9854e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7893e-02, 1.0166e-03, 1.9067e-01,  ..., 1.3390e-02,
           0.0000e+00, 0.0000e+00],
          [4.9055e-05, 2.8789e-05, 3.1885e-01,  ..., 4.9210e-03,
           1.9800e-01, 0.0000e+00],
          [6.2764e-05, 2.2829e-05, 2.4365e-01,  ..., 5.0774e-03,
           5.3925e-02, 2.0203e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.8843e-01, 6.1182e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6564e-03, 1.2646e-03, 9.9512e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3195e-05, 4.5478e-05, 3.8770e-01,  ..., 8.4925e-04,
           0.0000e+00, 0.0000e+00],
          [4.3035e-05, 3.6001e-05, 3.4351e-01,  ..., 3.8086e-02,
           3.1395e-03, 0.0000e+00],
          [6.4552e-05, 4.0233e-05, 1.2573e-01,  ..., 1.9089e-02,
           3.9315e-04, 1.2337e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8281e-01, 1.1737e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7068e-03, 1.6725e-04, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1683e-04, 4.5300e-06, 3.9160e-01,  ..., 8.9073e-04,
           0.0000e+00, 0.0000e+00],
          [8.7202e-05, 5.1260e-06, 2.9126e-01,  ..., 1.6953e-02,
           1.7624e-02, 0.0000e+00],
          [1.5831e-04, 3.9935e-06, 1.7432e-01,  ..., 1.7532e-02,
           6.3248e-03, 1.0384e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5024e-01, 7.5000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1498e-03, 3.7670e-03, 9.9121e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0711e-04, 1.0902e-04, 4.2725e-01,  ..., 1.9703e-03,
           0.0000e+00, 0.0000e+00],
          [1.5020e-04, 1.5020e-04, 3.1665e-01,  ..., 3.9642e-02,
           4.3449e-03, 0.0000e+00],
          [1.6618e-04, 1.5664e-04, 2.2070e-01,  ..., 3.2684e-02,
           9.4175e-04, 7.0419e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9707e-01, 5.0293e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7298e-02, 2.9907e-03, 9.6973e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5024e-04, 3.7861e-04, 2.2754e-01,  ..., 3.0861e-03,
           0.0000e+00, 0.0000e+00],
          [1.0262e-03, 1.1187e-03, 2.7417e-01,  ..., 1.2871e-02,
           4.0894e-02, 0.0000e+00],
          [1.8728e-04, 1.8537e-04, 1.7029e-01,  ..., 2.0248e-02,
           1.4534e-02, 1.8738e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5537e-01, 7.4463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5106e-03, 1.4343e-03, 9.9707e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8716e-05, 3.5584e-05, 2.7148e-01,  ..., 1.2331e-03,
           0.0000e+00, 0.0000e+00],
          [4.7505e-05, 4.1664e-05, 2.3096e-01,  ..., 1.7807e-02,
           1.3817e-02, 0.0000e+00],
          [6.5148e-05, 5.7399e-05, 7.5745e-02,  ..., 1.6312e-02,
           6.2675e-03, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9375e-01, 4.0649e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6136e-03, 1.5936e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9935e-05, 2.2054e-05, 3.7598e-01,  ..., 7.6389e-04,
           0.0000e+00, 0.0000e+00],
          [1.0240e-04, 3.3379e-05, 2.6514e-01,  ..., 2.4658e-02,
           1.7212e-02, 0.0000e+00],
          [2.1017e-04, 3.4332e-05, 1.2720e-01,  ..., 2.9297e-02,
           4.1885e-03, 1.4618e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5684e-01, 2.4292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.5002e-02, 4.4342e-02, 8.9062e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1024e-03, 3.2926e-04, 3.1250e-01,  ..., 7.9651e-02,
           0.0000e+00, 0.0000e+00],
          [1.7290e-03, 2.0897e-04, 2.4536e-02,  ..., 3.2446e-01,
           2.6880e-01, 0.0000e+00],
          [8.2397e-04, 7.2718e-06, 4.3335e-03,  ..., 5.9521e-01,
           4.6692e-03, 9.3445e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.1230e-01, 3.8770e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2986e-03, 1.5945e-02, 9.7852e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1875e-04, 1.3947e-04, 2.6978e-01,  ..., 6.3038e-04,
           0.0000e+00, 0.0000e+00],
          [1.2836e-03, 3.6168e-04, 3.4570e-01,  ..., 1.5545e-04,
           3.9558e-03, 0.0000e+00],
          [9.1648e-04, 3.2330e-04, 2.7930e-01,  ..., 1.9062e-04,
           7.9041e-03, 1.7792e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5435e-01, 5.4590e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6959e-03, 4.8561e-03, 9.9023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7132e-04, 9.8324e-04, 1.0468e-01,  ..., 2.0134e-04,
           0.0000e+00, 0.0000e+00],
          [8.6725e-05, 1.0185e-03, 3.0884e-01,  ..., 2.6941e-05,
           1.0345e-01, 0.0000e+00],
          [1.4329e-04, 3.3188e-04, 2.4072e-01,  ..., 1.3065e-04,
           1.1731e-01, 5.0049e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3727e-02, 9.7607e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4189e-02, 3.4363e-02, 9.2139e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6335e-03, 9.5892e-04, 2.2986e-01,  ..., 2.3441e-03,
           0.0000e+00, 0.0000e+00],
          [2.5487e-04, 2.5010e-04, 9.4238e-02,  ..., 4.4327e-03,
           1.7615e-01, 0.0000e+00],
          [6.7472e-04, 1.3363e-04, 2.5177e-02,  ..., 4.1840e-02,
           1.6223e-01, 6.7078e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0248e-01, 8.9746e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.2769e-01, 6.8604e-02, 8.0371e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0185e-03, 3.2806e-03, 2.5879e-01,  ..., 8.4351e-02,
           0.0000e+00, 0.0000e+00],
          [8.9264e-04, 1.9007e-03, 7.1594e-02,  ..., 3.6285e-02,
           5.8350e-01, 0.0000e+00],
          [4.6206e-04, 8.5950e-05, 1.1154e-02,  ..., 2.4219e-01,
           4.4897e-01, 1.9312e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8052e-01, 7.1924e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3696e-01, 3.3630e-02, 8.2959e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1114e-03, 5.3120e-04, 1.9543e-01,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [1.4889e-04, 2.3186e-05, 1.9119e-02,  ..., 5.0598e-02,
           6.6895e-01, 0.0000e+00],
          [7.9334e-05, 9.7156e-06, 1.0818e-02,  ..., 5.9387e-02,
           5.1208e-02, 5.4504e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1475e-01, 8.8525e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6072e-02, 6.0425e-02, 9.0332e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2528e-03, 1.1911e-03, 1.3135e-01,  ..., 1.1932e-02,
           0.0000e+00, 0.0000e+00],
          [6.0539e-03, 5.4054e-03, 2.0850e-01,  ..., 1.7586e-03,
           9.0393e-02, 0.0000e+00],
          [6.7711e-03, 1.8911e-03, 1.4539e-01,  ..., 1.3924e-03,
           2.5253e-02, 7.4524e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8676e-02, 9.5117e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.3864e-02, 2.4368e-02, 9.2188e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9188e-03, 3.9825e-03, 1.6772e-01,  ..., 3.8483e-02,
           0.0000e+00, 0.0000e+00],
          [2.9087e-03, 3.9291e-03, 1.7236e-01,  ..., 1.2100e-02,
           1.9995e-01, 0.0000e+00],
          [1.2283e-03, 4.0207e-03, 2.7222e-01,  ..., 2.1267e-03,
           1.2703e-02, 4.5654e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2373e-01, 6.7627e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9703e-02, 5.1483e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.3923e-03, 3.3703e-03, 1.7322e-01,  ..., 3.3447e-02,
           0.0000e+00, 0.0000e+00],
          [3.0594e-03, 3.2177e-03, 1.1591e-01,  ..., 1.0071e-01,
           1.8884e-01, 0.0000e+00],
          [2.7275e-03, 1.0405e-03, 1.5308e-01,  ..., 1.9302e-02,
           2.8198e-02, 1.2891e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.6699e-01, 3.3276e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9006e-01, 6.9946e-02, 7.4023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4135e-04, 1.6797e-04, 2.1960e-01,  ..., 5.9113e-02,
           0.0000e+00, 0.0000e+00],
          [2.3746e-03, 1.8895e-04, 1.2659e-01,  ..., 3.6011e-02,
           3.1769e-02, 0.0000e+00],
          [1.0061e-03, 1.4222e-04, 4.8859e-02,  ..., 1.6663e-01,
           1.8152e-01, 2.5000e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2092e-02, 9.1797e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6251e-02, 7.5195e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5483e-04, 3.7594e-03, 2.4182e-01,  ..., 6.6101e-02,
           0.0000e+00, 0.0000e+00],
          [3.5119e-04, 2.6298e-04, 2.9541e-01,  ..., 1.5900e-02,
           8.2031e-02, 0.0000e+00],
          [4.9448e-04, 8.6117e-04, 2.0422e-01,  ..., 6.8245e-03,
           1.0086e-02, 3.1128e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9458e-01, 3.6353e-01, 4.4189e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.3742e-03, 1.2512e-03, 8.4351e-02,  ..., 2.5391e-01,
           0.0000e+00, 0.0000e+00],
          [3.9101e-03, 8.9645e-04, 5.3558e-02,  ..., 6.2103e-02,
           2.4756e-01, 0.0000e+00],
          [1.8728e-04, 4.7326e-05, 1.7059e-02,  ..., 9.8145e-02,
           3.0957e-01, 2.1948e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.9605e-07, 4.2319e-06, 6.4611e-04,  ..., 2.5803e-02,
           8.0273e-01, 1.3525e-01]],

         [[1.9014e-05, 1.9014e-05, 2.6531e-03,  ..., 2.3651e-02,
           2.8809e-01, 4.1919e-01]],

         [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 2.9802e-07,
           1.0986e-02, 9.8877e-01]],

         ...,

         [[2.5177e-03, 3.2310e-03, 4.1504e-03,  ..., 3.0670e-02,
           2.2656e-01, 5.0568e-02]],

         [[4.7569e-03, 1.2146e-02, 5.1147e-02,  ..., 3.9825e-02,
           1.3904e-01, 1.0175e-01]],

         [[2.0248e-02, 1.6785e-02, 2.5749e-03,  ..., 4.2877e-02,
           6.5765e-03, 6.2378e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1215e-03, 1.3323e-03, 1.0483e-02,  ..., 5.0781e-02,
           1.9775e-01, 2.6611e-01]],

         [[5.9605e-08, 2.0266e-06, 4.6670e-05,  ..., 9.4116e-02,
           7.7588e-01, 1.2671e-01]],

         [[3.3493e-03, 5.5237e-03, 1.7273e-02,  ..., 1.3586e-01,
           1.9775e-01, 1.9165e-01]],

         ...,

         [[4.5319e-03, 1.5450e-02, 1.5450e-02,  ..., 1.2250e-01,
           1.8494e-02, 9.6863e-02]],

         [[1.4267e-03, 4.0588e-03, 4.0192e-02,  ..., 6.7566e-02,
           3.6230e-01, 4.1443e-02]],

         [[1.0452e-03, 3.9444e-03, 1.8372e-02,  ..., 3.2324e-01,
           1.0577e-01, 1.8274e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1133e-02, 4.7874e-03, 1.4076e-02,  ..., 9.2529e-02,
           8.0383e-02, 1.0162e-01]],

         [[3.3569e-03, 1.0214e-03, 2.3251e-03,  ..., 2.9980e-01,
           3.9490e-02, 1.2817e-02]],

         [[1.7456e-02, 7.6904e-03, 7.7782e-03,  ..., 1.5114e-02,
           5.4413e-02, 2.8671e-02]],

         ...,

         [[4.0936e-04, 3.4332e-04, 1.5507e-03,  ..., 2.8540e-01,
           3.8379e-01, 1.4917e-01]],

         [[1.3947e-02, 2.5902e-03, 4.1885e-03,  ..., 6.7627e-02,
           1.6345e-01, 5.3070e-02]],

         [[9.6359e-03, 2.0123e-03, 2.0325e-02,  ..., 5.6549e-02,
           4.2114e-01, 9.5459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8955e-01, 2.2829e-05, 5.0485e-05,  ..., 1.4982e-03,
           3.7003e-03, 1.1658e-02]],

         [[9.6143e-01, 3.6359e-06, 3.6001e-05,  ..., 4.9477e-03,
           9.5749e-03, 4.0779e-03]],

         [[8.5010e-01, 3.7551e-05, 2.2292e-04,  ..., 7.1716e-03,
           3.6335e-03, 2.0660e-02]],

         ...,

         [[6.3428e-01, 4.7040e-04, 1.8635e-03,  ..., 1.5821e-03,
           4.9400e-03, 3.3112e-02]],

         [[2.6343e-01, 4.1771e-04, 8.1348e-04,  ..., 4.1870e-02,
           1.2917e-02, 3.8574e-02]],

         [[5.7227e-01, 3.1590e-04, 7.3385e-04,  ..., 2.0355e-02,
           3.3813e-02, 4.4128e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 6.8545e-06, 1.6212e-05,  ..., 1.7285e-04,
           3.8338e-04, 3.4790e-02]],

         [[8.7207e-01, 1.1921e-06, 1.5259e-05,  ..., 9.9945e-03,
           1.0199e-01, 1.2650e-02]],

         [[9.1650e-01, 3.8385e-05, 1.3304e-04,  ..., 6.9427e-04,
           2.2110e-02, 1.7593e-02]],

         ...,

         [[2.1680e-01, 1.5008e-04, 7.3016e-05,  ..., 1.0278e-01,
           2.5244e-01, 3.2715e-02]],

         [[9.7839e-02, 2.9011e-03, 2.0599e-03,  ..., 1.5100e-01,
           4.3335e-01, 2.5024e-02]],

         [[4.9292e-01, 8.5294e-05, 1.0532e-04,  ..., 7.5317e-02,
           1.1896e-01, 9.3506e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1807e-01, 7.0715e-04, 2.7585e-04,  ..., 5.5145e-02,
           6.5002e-02, 3.3905e-02]],

         [[3.4277e-01, 3.1829e-05, 3.4571e-05,  ..., 9.7473e-02,
           3.9355e-01, 4.2053e-02]],

         [[7.5586e-01, 9.7990e-05, 1.8477e-05,  ..., 8.3389e-03,
           4.4189e-02, 6.0242e-02]],

         ...,

         [[9.2383e-01, 4.1604e-05, 1.9073e-06,  ..., 1.8835e-05,
           3.4428e-04, 7.1960e-02]],

         [[9.6826e-01, 1.5640e-04, 3.6001e-05,  ..., 2.6822e-06,
           1.0061e-04, 1.0414e-02]],

         [[9.9609e-01, 2.3901e-05, 7.1526e-07,  ..., 5.9605e-08,
           8.7619e-06, 2.3708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1270e-01, 1.6809e-05, 8.7619e-06,  ..., 8.7830e-02,
           2.1753e-01, 2.4368e-02]],

         [[9.6045e-01, 1.4126e-05, 7.3910e-06,  ..., 7.3338e-04,
           1.0042e-03, 1.6953e-02]],

         [[8.2617e-01, 3.5930e-04, 2.9862e-05,  ..., 1.5083e-02,
           5.7335e-03, 5.0507e-02]],

         ...,

         [[9.5605e-01, 1.3530e-04, 4.3571e-05,  ..., 2.7504e-03,
           1.9112e-03, 9.3384e-03]],

         [[9.6533e-01, 7.5996e-05, 1.5736e-05,  ..., 5.0583e-03,
           1.7986e-03, 1.2878e-02]],

         [[6.2012e-01, 2.9850e-04, 1.1092e-04,  ..., 2.3486e-01,
           5.9929e-03, 1.0544e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0264e-01, 3.0339e-05, 3.5048e-05,  ..., 4.9629e-03,
           5.9052e-03, 1.4191e-02]],

         [[7.6318e-01, 7.7486e-05, 6.7592e-05,  ..., 8.9722e-03,
           7.9803e-03, 4.7882e-02]],

         [[8.7695e-01, 1.1253e-04, 3.6240e-05,  ..., 1.0391e-02,
           3.6469e-03, 8.0414e-03]],

         ...,

         [[6.9141e-01, 2.7156e-04, 4.2498e-05,  ..., 1.6449e-02,
           8.6365e-03, 6.1279e-02]],

         [[6.9775e-01, 6.0272e-04, 9.4295e-05,  ..., 2.7939e-02,
           3.9703e-02, 1.7212e-02]],

         [[7.1045e-01, 2.6655e-04, 1.6093e-04,  ..., 3.3722e-02,
           8.2214e-02, 2.9587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2212e-01, 2.8057e-03, 1.1206e-03,  ..., 1.3977e-02,
           3.4119e-02, 1.1032e-02]],

         [[8.1738e-01, 1.5521e-04, 3.4547e-04,  ..., 2.4452e-03,
           1.0376e-02, 5.0774e-03]],

         [[8.7305e-01, 5.6088e-05, 3.4809e-05,  ..., 2.0008e-03,
           3.9005e-03, 6.4049e-03]],

         ...,

         [[7.2168e-01, 4.3488e-04, 5.3930e-04,  ..., 3.4008e-03,
           1.4114e-02, 1.0901e-01]],

         [[8.6328e-01, 1.6832e-04, 6.0976e-05,  ..., 2.5539e-03,
           5.8632e-03, 2.8198e-02]],

         [[9.0771e-01, 2.8706e-04, 1.5128e-04,  ..., 2.2873e-02,
           4.1313e-03, 1.2688e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5610e-01, 2.9087e-04, 7.6056e-04,  ..., 7.0312e-02,
           1.8982e-01, 1.3977e-02]],

         [[8.1689e-01, 2.5988e-05, 1.7273e-04,  ..., 3.4851e-02,
           3.3264e-02, 2.6764e-02]],

         [[8.1006e-01, 7.9095e-05, 2.4843e-04,  ..., 6.9962e-03,
           8.7051e-03, 6.3171e-02]],

         ...,

         [[9.8511e-02, 2.1958e-04, 4.1199e-04,  ..., 3.0518e-02,
           7.4805e-01, 4.4403e-03]],

         [[7.4902e-01, 4.5776e-04, 1.7202e-04,  ..., 7.1167e-02,
           5.9204e-03, 1.0147e-02]],

         [[8.5596e-01, 1.1563e-05, 1.1444e-05,  ..., 4.6158e-03,
           5.9128e-03, 6.5002e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 3.1173e-05, 1.1539e-04,  ..., 9.3102e-05,
           3.3264e-03, 2.9358e-02]],

         [[4.8975e-01, 8.9169e-05, 7.7188e-05,  ..., 6.6162e-02,
           3.3508e-02, 1.0797e-01]],

         [[3.4814e-01, 5.8413e-06, 6.9737e-06,  ..., 9.9468e-04,
           3.7384e-03, 3.9368e-02]],

         ...,

         [[9.1748e-01, 1.8966e-04, 8.4162e-05,  ..., 5.2261e-03,
           5.4817e-03, 9.8724e-03]],

         [[4.1772e-01, 4.2820e-04, 2.6226e-04,  ..., 4.6692e-02,
           2.9633e-02, 2.6196e-01]],

         [[1.8042e-01, 5.5552e-04, 3.6287e-04,  ..., 2.5024e-02,
           4.9316e-02, 3.9520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6367e-01, 4.7684e-06, 7.8082e-06,  ..., 4.7379e-03,
           3.2288e-02, 6.7041e-01]],

         [[7.5537e-01, 6.5231e-04, 8.3590e-04,  ..., 1.6632e-02,
           1.1604e-02, 2.5043e-03]],

         [[8.7988e-01, 4.5228e-04, 5.7936e-04,  ..., 1.0300e-02,
           1.3832e-02, 2.6566e-02]],

         ...,

         [[4.0430e-01, 1.7905e-04, 4.5037e-04,  ..., 1.1139e-02,
           1.9394e-02, 3.8300e-02]],

         [[6.2939e-01, 4.6968e-05, 5.5134e-05,  ..., 9.7580e-03,
           3.8509e-03, 1.9577e-02]],

         [[2.9810e-01, 2.4974e-05, 8.3506e-05,  ..., 5.6610e-03,
           7.0862e-02, 5.5322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.5184e-04, 8.8930e-04,  ..., 1.2741e-02,
           8.3252e-02, 3.8208e-01]],

         [[5.4150e-01, 2.5821e-04, 6.7472e-04,  ..., 3.4851e-02,
           4.5227e-02, 6.7871e-02]],

         [[4.8608e-01, 5.8603e-04, 1.0443e-03,  ..., 4.8790e-03,
           1.3260e-02, 3.3539e-02]],

         ...,

         [[8.2947e-02, 2.1458e-06, 3.8147e-06,  ..., 5.6744e-04,
           6.9847e-03, 8.9844e-01]],

         [[4.8633e-01, 2.9421e-04, 4.8590e-04,  ..., 2.7054e-02,
           5.2246e-02, 1.3940e-01]],

         [[8.7842e-01, 1.3733e-04, 1.0735e-04,  ..., 1.7948e-03,
           3.3016e-03, 3.2940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 3.0899e-04, 3.9220e-04,  ..., 4.8447e-03,
           8.0719e-03, 1.4458e-02]],

         [[9.0088e-01, 2.5570e-05, 5.0724e-05,  ..., 7.8125e-03,
           1.6266e-02, 2.4902e-02]],

         [[5.2734e-01, 1.1134e-04, 2.8324e-04,  ..., 2.5291e-03,
           3.0136e-02, 3.1555e-02]],

         ...,

         [[3.9575e-01, 0.0000e+00, 3.9935e-06,  ..., 2.9099e-02,
           5.2344e-01, 4.5013e-02]],

         [[6.2744e-01, 1.0300e-03, 1.2751e-03,  ..., 1.7365e-02,
           2.9160e-02, 2.2629e-02]],

         [[9.1748e-01, 1.9670e-06, 7.9870e-06,  ..., 9.9945e-03,
           1.9119e-02, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8281e-01, 3.4809e-04, 8.9586e-05,  ..., 7.8011e-04,
           2.2030e-03, 6.1340e-03]],

         [[2.4976e-01, 7.1859e-04, 4.8018e-04,  ..., 1.5417e-01,
           2.8076e-01, 2.7130e-02]],

         [[4.8022e-01, 7.6246e-04, 4.2534e-04,  ..., 5.2643e-02,
           9.0576e-02, 3.1189e-02]],

         ...,

         [[3.5913e-01, 1.0192e-05, 4.1008e-05,  ..., 6.3232e-02,
           2.2168e-01, 8.7769e-02]],

         [[7.6318e-01, 5.7220e-06, 1.8120e-05,  ..., 3.1769e-02,
           8.1543e-02, 6.1340e-02]],

         [[1.6931e-01, 4.3452e-05, 1.0747e-04,  ..., 2.5903e-01,
           3.9795e-01, 2.2171e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7373e-02, 1.2517e-06, 2.5034e-06,  ..., 2.3365e-03,
           4.6112e-02, 8.5645e-01]],

         [[2.1960e-01, 1.0848e-05, 2.0385e-05,  ..., 1.2036e-01,
           4.8730e-01, 6.2195e-02]],

         [[5.0781e-01, 8.9550e-04, 1.1225e-03,  ..., 2.6245e-02,
           6.1493e-02, 1.1353e-02]],

         ...,

         [[1.9678e-01, 2.2161e-04, 2.4390e-04,  ..., 1.5198e-02,
           3.7170e-02, 1.0994e-02]],

         [[4.7925e-01, 5.7518e-05, 4.3273e-05,  ..., 2.8858e-03,
           1.4221e-02, 2.2858e-02]],

         [[9.9365e-02, 3.9959e-04, 2.0254e-04,  ..., 1.1683e-03,
           3.5477e-03, 6.1607e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 2.5487e-04, 3.8195e-04,  ..., 2.2308e-02,
           2.6901e-02, 2.1698e-02]],

         [[2.4841e-01, 4.2319e-04, 3.9911e-04,  ..., 2.3376e-02,
           6.0852e-02, 6.1829e-02]],

         [[1.2061e-01, 2.9659e-04, 1.5235e-04,  ..., 5.6648e-03,
           1.1620e-02, 1.0620e-02]],

         ...,

         [[7.2510e-01, 4.1246e-05, 2.8074e-05,  ..., 1.3485e-03,
           1.0712e-02, 2.8442e-02]],

         [[5.4883e-01, 7.1096e-04, 4.9305e-04,  ..., 6.9504e-03,
           2.4414e-02, 3.5919e-02]],

         [[8.9844e-01, 3.9554e-04, 1.8942e-04,  ..., 2.0199e-03,
           1.0712e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8726e-02, 1.6499e-03, 1.4124e-03,  ..., 5.4053e-01,
           1.1243e-01, 3.4218e-03]],

         [[3.9526e-01, 3.1967e-03, 1.6804e-03,  ..., 2.5818e-02,
           1.9150e-02, 3.1372e-02]],

         [[4.9341e-01, 1.4496e-04, 1.8835e-04,  ..., 1.4755e-02,
           6.8115e-02, 5.9174e-02]],

         ...,

         [[4.9219e-01, 2.8634e-04, 3.5477e-04,  ..., 1.0117e-02,
           2.2141e-02, 1.0400e-01]],

         [[9.8584e-01, 3.3975e-05, 1.0395e-04,  ..., 6.8545e-06,
           2.1577e-04, 2.1572e-03]],

         [[8.6963e-01, 1.3053e-04, 1.2374e-04,  ..., 8.7051e-03,
           1.0895e-02, 2.1179e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1689e-01, 3.5763e-03, 2.9640e-03,  ..., 1.8628e-01,
           1.8054e-01, 7.8857e-02]],

         [[7.9834e-02, 6.8169e-03, 3.1948e-03,  ..., 1.3477e-01,
           1.0095e-01, 2.5131e-02]],

         [[1.0645e-01, 6.4135e-05, 8.6308e-05,  ..., 1.8204e-02,
           1.1688e-01, 4.7314e-01]],

         ...,

         [[1.4807e-01, 5.7995e-05, 8.9586e-05,  ..., 5.6610e-02,
           1.6382e-01, 1.2199e-02]],

         [[7.6953e-01, 1.5020e-05, 2.3246e-05,  ..., 1.0391e-02,
           4.2419e-02, 1.1963e-02]],

         [[2.9590e-01, 1.7631e-04, 2.5058e-04,  ..., 5.1422e-02,
           3.3789e-01, 1.6467e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7061e-01, 2.7180e-05, 2.8014e-05,  ..., 1.1921e-03,
           5.5122e-03, 8.2275e-02]],

         [[2.5269e-01, 1.6081e-04, 7.2539e-05,  ..., 8.8684e-02,
           2.2656e-01, 3.7567e-02]],

         [[3.9526e-01, 6.2644e-05, 6.9857e-05,  ..., 9.0103e-03,
           7.3120e-02, 4.9194e-01]],

         ...,

         [[3.3423e-01, 3.9215e-03, 2.4929e-03,  ..., 6.9885e-03,
           1.9455e-02, 2.8305e-02]],

         [[5.2393e-01, 1.1330e-03, 1.0519e-03,  ..., 2.3743e-02,
           6.9824e-02, 2.4887e-02]],

         [[2.8296e-01, 2.0695e-03, 2.4872e-03,  ..., 4.6570e-02,
           4.9561e-02, 6.8787e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9248e-01, 3.3069e-04, 1.8263e-04,  ..., 1.2006e-01,
           1.2195e-01, 5.8502e-02]],

         [[8.0273e-01, 3.1972e-04, 4.7147e-05,  ..., 1.0674e-02,
           2.5349e-03, 3.6957e-02]],

         [[8.6719e-01, 7.4983e-05, 2.4855e-05,  ..., 1.0338e-02,
           3.7727e-03, 1.0666e-02]],

         ...,

         [[5.8716e-02, 1.7462e-03, 3.2616e-03,  ..., 4.6191e-01,
           1.8372e-01, 1.6571e-02]],

         [[3.8916e-01, 1.1164e-04, 1.0651e-04,  ..., 3.4733e-03,
           2.7740e-02, 2.9541e-02]],

         [[3.0933e-01, 1.5736e-03, 5.0392e-03,  ..., 1.0693e-01,
           1.6052e-01, 4.0253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9907e-01, 4.1914e-04, 9.2983e-04,  ..., 1.7960e-02,
           7.3303e-02, 8.3069e-02]],

         [[1.6553e-01, 5.6982e-04, 1.8387e-03,  ..., 2.8763e-02,
           1.0858e-01, 6.7932e-02]],

         [[2.7661e-01, 1.2004e-04, 5.7745e-04,  ..., 4.2755e-02,
           7.6782e-02, 5.5328e-02]],

         ...,

         [[4.4785e-03, 2.7061e-05, 6.0022e-05,  ..., 2.7599e-03,
           4.4525e-02, 3.8319e-03]],

         [[4.9585e-01, 4.4870e-04, 6.8951e-04,  ..., 4.2229e-03,
           4.5410e-02, 6.6040e-02]],

         [[5.9521e-01, 4.5013e-04, 6.9189e-04,  ..., 6.1188e-03,
           6.7871e-02, 3.5767e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3928e-01, 1.1158e-04, 1.0610e-04,  ..., 2.6932e-03,
           6.9458e-02, 1.4819e-01]],

         [[2.6367e-01, 4.3559e-04, 6.5899e-04,  ..., 6.3171e-02,
           9.4055e-02, 3.5706e-02]],

         [[3.4131e-01, 9.0957e-05, 3.2377e-04,  ..., 1.9257e-02,
           1.2268e-01, 4.0527e-01]],

         ...,

         [[8.3398e-01, 9.0480e-05, 2.4509e-04,  ..., 5.7983e-03,
           1.8005e-02, 1.8723e-02]],

         [[6.3818e-01, 2.4652e-04, 1.9703e-03,  ..., 3.9482e-03,
           1.7548e-02, 5.7098e-02]],

         [[4.1846e-01, 3.3665e-04, 3.5548e-04,  ..., 4.2305e-03,
           1.3344e-02, 2.8473e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9250e-01, 2.5129e-04, 3.4499e-04,  ..., 1.0773e-02,
           8.6731e-02, 1.4526e-01]],

         [[2.4826e-02, 2.6762e-05, 4.7743e-05,  ..., 1.1253e-03,
           1.4366e-02, 1.0109e-02]],

         [[5.1514e-01, 4.6802e-04, 1.1358e-03,  ..., 1.0735e-02,
           4.4312e-02, 4.2969e-02]],

         ...,

         [[6.3525e-01, 1.1921e-07, 1.6153e-05,  ..., 4.0619e-02,
           2.6904e-01, 2.0416e-02]],

         [[3.1763e-01, 1.0371e-05, 7.9811e-05,  ..., 1.3245e-01,
           1.1871e-01, 4.4006e-02]],

         [[5.3320e-01, 7.1526e-07, 8.1241e-05,  ..., 1.0834e-01,
           3.2349e-01, 1.4664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1067e-02, 3.6120e-05, 1.0407e-04,  ..., 2.6367e-02,
           1.9946e-01, 2.8076e-02]],

         [[7.7133e-03, 1.3113e-06, 8.1062e-06,  ..., 4.5967e-04,
           7.9575e-03, 9.6387e-01]],

         [[1.6504e-01, 3.9816e-04, 1.3733e-03,  ..., 6.4636e-02,
           3.9978e-02, 2.6733e-02]],

         ...,

         [[2.7783e-01, 6.9761e-04, 1.2894e-03,  ..., 1.4136e-01,
           2.3035e-01, 2.4857e-02]],

         [[5.9961e-01, 1.3471e-05, 2.3305e-05,  ..., 1.0689e-02,
           7.9895e-02, 7.0496e-03]],

         [[2.1973e-01, 5.6684e-05, 4.4751e-04,  ..., 2.4707e-01,
           4.2358e-01, 6.9946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1885e-01, 1.5497e-04, 5.8937e-04,  ..., 5.2765e-02,
           3.4332e-02, 1.8234e-02]],

         [[9.8340e-01, 2.5094e-05, 1.6844e-04,  ..., 2.4471e-03,
           8.2922e-04, 2.7990e-04]],

         [[2.2925e-01, 5.7042e-05, 2.9254e-04,  ..., 4.4518e-03,
           6.9771e-03, 2.3556e-03]],

         ...,

         [[6.6162e-01, 4.4405e-05, 7.0989e-05,  ..., 5.6801e-03,
           1.5442e-02, 3.2196e-02]],

         [[1.0175e-01, 1.2875e-05, 1.0431e-05,  ..., 2.3384e-03,
           1.0727e-02, 2.4551e-02]],

         [[1.4062e-01, 3.1173e-05, 5.9605e-05,  ..., 1.6785e-02,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5759e-01, 5.1498e-05, 1.0884e-04,  ..., 6.9092e-02,
           1.8567e-01, 7.6782e-02]],

         [[6.9873e-01, 7.7665e-05, 1.1289e-04,  ..., 8.6212e-03,
           1.5587e-02, 3.3722e-02]],

         [[2.4121e-01, 1.0526e-04, 4.7612e-04,  ..., 6.2927e-02,
           1.3745e-01, 3.3740e-01]],

         ...,

         [[5.8252e-01, 1.0622e-04, 4.8256e-04,  ..., 3.6957e-02,
           6.2347e-02, 5.1697e-02]],

         [[3.8867e-01, 7.2718e-05, 6.4945e-04,  ..., 2.1255e-02,
           1.0223e-01, 1.7798e-01]],

         [[5.7129e-01, 1.6034e-05, 9.0897e-05,  ..., 4.4327e-03,
           2.2507e-02, 6.0211e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4465e-02, 9.8348e-06, 1.0252e-05,  ..., 3.3760e-04,
           3.8338e-03, 1.3275e-02]],

         [[2.6642e-02, 3.7849e-05, 5.5194e-05,  ..., 1.1940e-03,
           1.5793e-02, 4.5319e-02]],

         [[1.0657e-01, 1.5163e-04, 6.7711e-04,  ..., 2.6321e-02,
           1.0992e-01, 2.6782e-01]],

         ...,

         [[1.1658e-01, 3.7313e-05, 3.3021e-04,  ..., 2.7979e-01,
           2.7979e-01, 9.3689e-02]],

         [[2.2241e-01, 1.7345e-04, 1.1673e-03,  ..., 1.0504e-01,
           2.0251e-01, 1.6272e-01]],

         [[9.4775e-01, 4.6909e-05, 1.2851e-04,  ..., 5.5504e-03,
           7.1259e-03, 1.1566e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.4617e-05, 3.8505e-05,  ..., 2.0401e-02,
           2.9739e-02, 2.5732e-01]],

         [[1.9263e-01, 5.2333e-05, 1.1748e-04,  ..., 1.1548e-01,
           4.5142e-01, 1.0229e-01]],

         [[1.0815e-01, 2.5034e-06, 2.3603e-05,  ..., 9.5764e-02,
           6.8359e-01, 8.4229e-02]],

         ...,

         [[3.1421e-01, 2.6655e-04, 1.6613e-03,  ..., 1.4465e-02,
           3.0869e-02, 7.2876e-02]],

         [[3.3569e-01, 2.7537e-04, 5.0068e-04,  ..., 4.5074e-02,
           9.3201e-02, 1.9531e-02]],

         [[3.7018e-02, 1.0234e-04, 1.0520e-04,  ..., 3.1338e-03,
           2.9739e-02, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4297e-01, 6.4254e-05, 4.7040e-04,  ..., 8.4229e-02,
           8.8623e-02, 1.0565e-01]],

         [[1.6418e-01, 3.1147e-03, 4.4098e-03,  ..., 1.3420e-02,
           3.7506e-02, 4.7211e-02]],

         [[2.8702e-02, 3.5524e-05, 1.2290e-04,  ..., 2.1973e-03,
           1.5610e-02, 9.0332e-03]],

         ...,

         [[8.9795e-01, 1.2648e-04, 1.7846e-04,  ..., 5.0926e-03,
           4.8599e-03, 1.3008e-02]],

         [[1.8799e-01, 1.1420e-04, 2.3067e-04,  ..., 8.1329e-03,
           4.8279e-02, 7.0947e-01]],

         [[6.0303e-01, 4.1068e-05, 1.9598e-04,  ..., 3.7933e-02,
           9.1003e-02, 1.6748e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 9.4414e-05, 1.9455e-03,  ..., 1.9379e-02,
           1.5602e-02, 4.4952e-02]],

         [[6.6772e-02, 6.5863e-05, 2.9874e-04,  ..., 7.7591e-03,
           1.5434e-02, 3.8055e-02]],

         [[2.1533e-01, 2.0103e-03, 4.0092e-03,  ..., 1.9763e-01,
           1.0956e-01, 8.3374e-02]],

         ...,

         [[7.7490e-01, 2.6464e-04, 7.2718e-04,  ..., 1.3756e-02,
           1.3786e-02, 3.1799e-02]],

         [[7.5879e-01, 1.0073e-05, 1.9302e-03,  ..., 1.6891e-02,
           1.4732e-02, 1.6464e-02]],

         [[6.7041e-01, 6.9261e-05, 1.6987e-04,  ..., 1.4671e-02,
           1.4503e-02, 5.2826e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9888e-03, 2.8400e-03, 6.5918e-02,  ..., 4.1260e-02,
           2.8122e-02, 4.6021e-02]],

         [[2.5344e-04, 2.6870e-04, 1.3086e-01,  ..., 8.4991e-03,
           1.4069e-02, 5.0873e-02]],

         [[5.5075e-04, 5.0402e-04, 7.9651e-02,  ..., 1.6769e-02,
           2.7740e-02, 1.0803e-01]],

         ...,

         [[1.6069e-04, 9.0122e-05, 1.5186e-01,  ..., 5.1270e-02,
           1.6174e-01, 3.0688e-01]],

         [[6.6757e-04, 6.5470e-04, 2.4475e-01,  ..., 4.3365e-02,
           5.0110e-02, 6.6895e-02]],

         [[1.4663e-05, 2.5630e-06, 1.0864e-02,  ..., 4.3130e-04,
           1.1816e-03, 7.4654e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3188e-04, 1.4496e-04, 2.1484e-01,  ..., 3.5973e-03,
           1.5564e-02, 1.0309e-01]],

         [[2.3603e-04, 1.3721e-04, 1.0962e-01,  ..., 1.4717e-02,
           9.2850e-03, 4.7150e-02]],

         [[1.4353e-04, 2.9266e-05, 2.9468e-01,  ..., 4.8370e-03,
           1.3405e-02, 7.5073e-02]],

         ...,

         [[1.0796e-03, 4.0269e-04, 1.4172e-01,  ..., 1.4839e-03,
           9.7961e-03, 5.5695e-02]],

         [[8.3447e-05, 9.7752e-05, 2.5317e-01,  ..., 2.7485e-03,
           9.1553e-03, 7.1411e-02]],

         [[4.3690e-05, 4.1425e-05, 2.5781e-01,  ..., 3.5736e-02,
           1.4026e-01, 1.0339e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-05, 9.8944e-05, 1.6516e-01,  ..., 5.2214e-04,
           1.5945e-02, 6.2073e-02]],

         [[1.6975e-04, 7.3910e-06, 7.6721e-02,  ..., 4.3983e-03,
           6.9695e-03, 4.4067e-02]],

         [[3.8743e-04, 4.8423e-04, 1.7200e-01,  ..., 1.1520e-03,
           9.2621e-03, 6.1829e-02]],

         ...,

         [[5.0664e-06, 1.5676e-05, 1.7871e-01,  ..., 4.5090e-03,
           1.4328e-02, 1.5527e-01]],

         [[4.2796e-05, 1.1128e-04, 3.9398e-02,  ..., 3.4409e-03,
           8.9951e-03, 3.1281e-02]],

         [[6.0797e-04, 1.4114e-04, 1.0101e-01,  ..., 2.3479e-03,
           8.5220e-03, 5.1605e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3358e-04, 1.0800e-04, 1.9638e-02,  ..., 9.9564e-03,
           1.3525e-01, 5.3516e-01]],

         [[5.8830e-05, 2.2531e-04, 2.3608e-01,  ..., 6.7383e-02,
           1.3196e-01, 6.2561e-02]],

         [[5.5850e-05, 3.8290e-04, 2.9468e-01,  ..., 8.1848e-02,
           4.2938e-02, 5.5046e-03]],

         ...,

         [[7.4053e-04, 5.3835e-04, 8.2458e-02,  ..., 1.1316e-01,
           2.7084e-02, 8.7769e-02]],

         [[1.6088e-03, 3.5715e-04, 4.0619e-02,  ..., 2.0703e-01,
           1.8701e-01, 3.6060e-01]],

         [[1.6534e-04, 3.7074e-05, 5.3558e-02,  ..., 2.2507e-02,
           7.1228e-02, 1.6174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8553e-03, 9.3699e-04, 1.3599e-01,  ..., 1.5015e-02,
           3.7323e-02, 7.0801e-02]],

         [[2.8276e-04, 9.8038e-04, 2.4902e-01,  ..., 5.4779e-03,
           2.7496e-02, 7.7759e-02]],

         [[1.3762e-03, 5.2786e-04, 1.8665e-01,  ..., 1.8478e-02,
           4.2725e-02, 4.0039e-02]],

         ...,

         [[2.0313e-03, 2.7919e-04, 1.2372e-01,  ..., 1.3458e-02,
           2.0203e-02, 3.7524e-01]],

         [[6.6805e-04, 1.4706e-03, 2.5732e-01,  ..., 8.2493e-04,
           4.6005e-03, 1.4771e-01]],

         [[9.2125e-04, 7.6962e-04, 1.5979e-01,  ..., 4.4037e-02,
           4.7607e-02, 1.6235e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7180e-05, 6.6757e-06, 2.6054e-03,  ..., 2.2729e-01,
           4.5215e-01, 6.7200e-02]],

         [[5.5122e-04, 2.9516e-04, 4.3373e-03,  ..., 1.2671e-01,
           2.2241e-01, 3.9038e-01]],

         [[0.0000e+00, 0.0000e+00, 4.7684e-07,  ..., 2.4395e-03,
           1.0933e-02, 9.8438e-01]],

         ...,

         [[1.6510e-05, 3.6955e-06, 4.4882e-05,  ..., 2.4509e-03,
           2.4509e-03, 9.8877e-01]],

         [[6.0043e-03, 8.2016e-03, 5.2986e-03,  ..., 2.3743e-02,
           1.6479e-01, 1.0536e-02]],

         [[9.8495e-03, 3.5114e-03, 8.8787e-04,  ..., 5.9738e-03,
           2.9388e-02, 7.7454e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7980e-04, 6.5613e-04, 6.0349e-03,  ..., 7.2388e-02,
           1.2311e-01, 5.5176e-01]],

         [[0.0000e+00, 0.0000e+00, 9.5367e-07,  ..., 1.3989e-01,
           7.8027e-01, 7.2571e-02]],

         [[7.8440e-04, 1.3552e-03, 1.2077e-02,  ..., 1.4954e-01,
           1.4038e-01, 9.3567e-02]],

         ...,

         [[2.1458e-03, 3.8261e-03, 7.0381e-03,  ..., 2.7405e-02,
           1.5161e-01, 5.1709e-01]],

         [[2.3270e-03, 9.8572e-03, 3.6194e-02,  ..., 1.6040e-01,
           1.5002e-01, 2.2919e-02]],

         [[1.3390e-03, 2.7905e-03, 1.1841e-02,  ..., 7.4280e-02,
           3.2764e-01, 2.1155e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8441e-02, 7.9727e-03, 2.9617e-02,  ..., 4.4464e-02,
           8.0505e-02, 4.6234e-02]],

         [[8.7976e-04, 4.2868e-04, 1.7023e-03,  ..., 1.9943e-02,
           5.8517e-03, 2.2907e-03]],

         [[1.5053e-02, 5.0011e-03, 6.7825e-03,  ..., 1.3098e-01,
           5.9998e-02, 3.3661e-02]],

         ...,

         [[2.4490e-03, 1.9407e-04, 1.0328e-03,  ..., 1.6003e-01,
           2.4585e-01, 3.2080e-01]],

         [[1.9760e-02, 3.8166e-03, 4.8637e-03,  ..., 1.4270e-01,
           1.7896e-01, 7.8186e-02]],

         [[7.3776e-03, 7.3051e-04, 1.2070e-02,  ..., 1.5173e-01,
           2.8564e-01, 1.7188e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.8014e-06, 9.3877e-05,  ..., 4.7035e-03,
           3.1033e-03, 1.6052e-02]],

         [[9.2188e-01, 4.7684e-07, 2.8551e-05,  ..., 2.7115e-02,
           2.1561e-02, 1.1696e-02]],

         [[9.7314e-01, 1.2338e-05, 4.2915e-05,  ..., 1.2150e-03,
           3.4161e-03, 1.8320e-03]],

         ...,

         [[2.9114e-02, 1.9670e-05, 1.2422e-04,  ..., 1.6713e-04,
           1.6129e-04, 2.3708e-03]],

         [[6.5869e-01, 4.5204e-04, 1.1892e-03,  ..., 1.1505e-02,
           1.3557e-02, 7.7095e-03]],

         [[3.0103e-01, 3.4165e-04, 3.7813e-04,  ..., 1.4084e-02,
           2.0721e-02, 2.6718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9575e-01, 1.5497e-06, 4.8280e-06,  ..., 2.8706e-03,
           5.1928e-04, 9.1782e-03]],

         [[8.4668e-01, 0.0000e+00, 5.9605e-08,  ..., 6.9275e-03,
           9.5093e-02, 1.6403e-02]],

         [[9.2285e-01, 3.2485e-05, 1.3232e-05,  ..., 1.8244e-03,
           3.4668e-02, 3.6697e-03]],

         ...,

         [[2.8174e-01, 6.0129e-04, 1.8942e-04,  ..., 2.4573e-01,
           9.2529e-02, 6.6490e-03]],

         [[8.8745e-02, 9.9087e-04, 4.1199e-04,  ..., 3.1226e-01,
           8.8074e-02, 5.6343e-03]],

         [[6.3086e-01, 2.8551e-05, 8.0347e-05,  ..., 3.4576e-02,
           1.9543e-01, 5.5450e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7456e-01, 2.7227e-04, 1.5688e-04,  ..., 1.2793e-01,
           1.0730e-01, 7.1228e-02]],

         [[7.1680e-01, 1.7166e-05, 2.2590e-05,  ..., 7.8613e-02,
           1.3110e-01, 1.5747e-02]],

         [[8.8135e-01, 2.1160e-05, 1.2159e-05,  ..., 7.8583e-03,
           6.3232e-02, 1.5236e-02]],

         ...,

         [[4.9756e-01, 1.0133e-06, 2.9802e-07,  ..., 3.5524e-05,
           5.1117e-04, 4.8145e-01]],

         [[6.7773e-01, 8.7023e-06, 8.7023e-06,  ..., 8.1658e-06,
           2.0742e-05, 7.0953e-03]],

         [[9.7559e-01, 8.3447e-07, 5.9605e-07,  ..., 2.3842e-07,
           7.1526e-07, 1.2865e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7578e-01, 1.4842e-05, 5.0068e-06,  ..., 4.9683e-02,
           1.0162e-01, 1.1285e-01]],

         [[8.7891e-01, 1.5736e-05, 3.4571e-06,  ..., 2.4986e-04,
           5.3072e-04, 1.1620e-02]],

         [[7.2998e-01, 2.4629e-04, 5.7578e-05,  ..., 2.6531e-03,
           1.4641e-02, 9.8877e-02]],

         ...,

         [[8.9600e-01, 3.7122e-04, 8.9884e-05,  ..., 1.6317e-03,
           6.7444e-03, 1.5335e-02]],

         [[9.4385e-01, 6.4075e-05, 1.6570e-05,  ..., 1.7319e-03,
           1.5617e-02, 2.3590e-02]],

         [[9.1406e-01, 1.4281e-04, 1.8275e-04,  ..., 1.1349e-03,
           7.4196e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3838e-01, 6.2227e-05, 8.5771e-05,  ..., 2.3003e-03,
           2.0180e-03, 1.6739e-02]],

         [[7.6416e-01, 2.6441e-04, 8.2552e-05,  ..., 4.5433e-03,
           5.7220e-03, 2.9053e-02]],

         [[9.8486e-01, 1.0729e-06, 1.0133e-06,  ..., 4.5252e-04,
           1.5545e-03, 6.7673e-03]],

         ...,

         [[6.1963e-01, 1.0067e-04, 3.0339e-05,  ..., 1.0246e-02,
           1.1011e-01, 3.1403e-02]],

         [[9.1553e-01, 2.5439e-04, 2.8074e-05,  ..., 8.0032e-03,
           7.4348e-03, 2.8439e-03]],

         [[8.8232e-01, 1.6034e-05, 4.2558e-05,  ..., 5.9814e-03,
           2.6443e-02, 5.1758e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2837e-01, 9.7132e-04, 2.5535e-04,  ..., 1.2352e-02,
           6.6223e-03, 3.1357e-03]],

         [[9.8145e-01, 7.9274e-06, 1.3828e-05,  ..., 1.1263e-03,
           3.5496e-03, 2.0027e-04]],

         [[9.3018e-01, 8.3447e-06, 1.1265e-05,  ..., 3.2663e-04,
           3.5596e-04, 8.4734e-04]],

         ...,

         [[8.0420e-01, 2.9027e-05, 2.4438e-05,  ..., 1.2474e-03,
           1.6068e-02, 9.5581e-02]],

         [[9.6484e-01, 1.6093e-06, 2.7418e-06,  ..., 1.5621e-03,
           1.3039e-02, 6.5460e-03]],

         [[9.2334e-01, 1.1444e-05, 1.0967e-05,  ..., 3.9864e-04,
           1.2112e-03, 1.6281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9712e-02, 3.3200e-05, 2.7776e-04,  ..., 2.6880e-01,
           1.4612e-01, 1.7319e-02]],

         [[9.5410e-01, 4.3511e-06, 1.4246e-05,  ..., 6.6147e-03,
           6.6414e-03, 4.3030e-03]],

         [[7.9980e-01, 1.3232e-05, 5.0843e-05,  ..., 2.2202e-03,
           2.3132e-02, 2.6321e-02]],

         ...,

         [[9.0088e-01, 2.5034e-05, 4.3392e-05,  ..., 3.8757e-02,
           2.3117e-03, 1.0548e-03]],

         [[8.0859e-01, 9.6381e-05, 4.3750e-05,  ..., 1.0500e-03,
           1.6127e-03, 1.8921e-02]],

         [[8.5645e-01, 5.6028e-06, 7.3314e-06,  ..., 2.6264e-03,
           3.3478e-02, 2.3289e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6992e-01, 8.3447e-07, 5.3644e-06,  ..., 7.6294e-04,
           1.4853e-04, 9.4986e-03]],

         [[7.8125e-01, 4.4048e-05, 5.0724e-05,  ..., 2.4155e-02,
           5.7617e-02, 7.3242e-03]],

         [[7.2998e-01, 0.0000e+00, 1.7881e-07,  ..., 2.7161e-03,
           1.4200e-03, 1.5480e-02]],

         ...,

         [[7.8760e-01, 1.9073e-04, 8.5950e-05,  ..., 1.7471e-02,
           1.7242e-02, 6.6299e-03]],

         [[6.2939e-01, 7.6771e-05, 9.4414e-05,  ..., 2.6073e-03,
           5.6580e-02, 4.6844e-02]],

         [[3.2715e-01, 5.6839e-04, 1.0914e-04,  ..., 3.5339e-02,
           1.3832e-02, 1.0971e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1667e-01, 7.7486e-06, 5.7220e-06,  ..., 6.3944e-04,
           3.8280e-03, 7.6807e-01]],

         [[4.8730e-01, 3.8123e-04, 4.8780e-04,  ..., 1.5793e-02,
           1.8492e-03, 1.6098e-02]],

         [[7.8076e-01, 2.1708e-04, 5.3883e-04,  ..., 2.4475e-02,
           1.6266e-02, 5.2605e-03]],

         ...,

         [[6.6260e-01, 2.0468e-04, 2.2054e-04,  ..., 3.3493e-03,
           7.5188e-03, 2.8934e-03]],

         [[8.5107e-01, 1.6749e-05, 2.8253e-05,  ..., 2.0771e-03,
           9.2850e-03, 5.4138e-02]],

         [[4.7046e-01, 8.9407e-06, 3.8803e-05,  ..., 6.8893e-03,
           5.1849e-02, 3.6035e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9038e-01, 6.0797e-05, 8.5413e-05,  ..., 2.2812e-03,
           9.2010e-03, 3.3887e-01]],

         [[3.8428e-01, 1.8537e-04, 1.5795e-04,  ..., 1.6037e-02,
           1.2170e-01, 3.8109e-03]],

         [[4.4287e-01, 9.9778e-05, 5.0962e-05,  ..., 1.1377e-03,
           4.0016e-03, 7.9422e-03]],

         ...,

         [[1.6113e-02, 2.5630e-06, 2.1458e-06,  ..., 3.6061e-05,
           9.1219e-04, 8.0713e-01]],

         [[4.7095e-01, 5.9366e-05, 9.9421e-05,  ..., 1.8692e-03,
           2.1103e-02, 5.7983e-02]],

         [[5.0879e-01, 9.1195e-05, 1.1802e-05,  ..., 5.9223e-04,
           1.1482e-03, 7.0496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3564e-01, 1.0605e-03, 5.5695e-04,  ..., 6.6910e-03,
           1.0475e-02, 3.4103e-03]],

         [[6.5967e-01, 1.0806e-04, 3.9387e-04,  ..., 5.1636e-02,
           1.7249e-01, 5.6648e-03]],

         [[4.5825e-01, 1.4007e-05, 6.0081e-05,  ..., 1.2360e-02,
           1.1948e-02, 6.9466e-03]],

         ...,

         [[3.6035e-01, 0.0000e+00, 5.9605e-08,  ..., 5.8197e-02,
           5.0342e-01, 4.3701e-02]],

         [[4.0845e-01, 7.9441e-04, 5.9414e-04,  ..., 2.4597e-02,
           1.6693e-02, 8.3542e-03]],

         [[8.9014e-01, 2.9802e-07, 6.5565e-07,  ..., 2.5043e-03,
           1.4832e-02, 5.3955e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4180e-01, 6.5267e-05, 1.2159e-05,  ..., 7.4673e-04,
           3.5686e-03, 9.0942e-03]],

         [[6.9031e-02, 1.2791e-04, 1.0622e-04,  ..., 2.9590e-01,
           3.1250e-01, 1.1663e-03]],

         [[1.5662e-01, 1.3053e-04, 6.0618e-05,  ..., 1.9684e-02,
           1.8530e-01, 2.0103e-03]],

         ...,

         [[7.0996e-01, 6.5565e-07, 2.8014e-06,  ..., 4.4922e-02,
           1.0400e-01, 6.3843e-02]],

         [[5.9180e-01, 1.7881e-07, 4.1723e-07,  ..., 1.6464e-02,
           1.8066e-01, 1.7554e-01]],

         [[2.9761e-01, 3.0398e-06, 2.8014e-06,  ..., 2.6025e-01,
           2.7734e-01, 1.4610e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7415e-02, 5.3644e-07, 8.3447e-07,  ..., 2.3594e-03,
           4.6272e-03, 9.4287e-01]],

         [[1.0089e-01, 1.1981e-05, 3.6478e-05,  ..., 2.7759e-01,
           4.5581e-01, 3.5980e-02]],

         [[4.3066e-01, 3.6931e-04, 4.5919e-04,  ..., 9.7229e-02,
           2.3102e-02, 8.4457e-03]],

         ...,

         [[2.4072e-01, 8.1062e-05, 4.0591e-05,  ..., 4.2534e-03,
           6.6757e-03, 9.6436e-03]],

         [[5.9180e-01, 1.2994e-05, 1.2994e-05,  ..., 1.0086e-02,
           1.1627e-02, 1.0040e-02]],

         [[4.3213e-01, 1.0967e-04, 6.2227e-05,  ..., 9.2554e-04,
           4.5276e-04, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1025e-01, 5.8353e-05, 9.2328e-05,  ..., 8.8120e-03,
           4.5624e-02, 4.4739e-02]],

         [[3.5400e-01, 9.4318e-04, 4.1389e-04,  ..., 6.8207e-03,
           1.6495e-02, 1.7151e-02]],

         [[2.5488e-01, 4.0650e-05, 1.6510e-05,  ..., 3.0746e-03,
           1.0483e-02, 1.9653e-02]],

         ...,

         [[6.9775e-01, 1.8895e-05, 1.4246e-05,  ..., 1.6966e-03,
           1.2489e-02, 9.8801e-03]],

         [[5.0977e-01, 3.7909e-05, 3.4928e-05,  ..., 4.8599e-03,
           5.2063e-02, 8.8348e-03]],

         [[7.3584e-01, 5.2452e-05, 3.2842e-05,  ..., 5.7831e-03,
           6.3667e-03, 1.7105e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8174e-01, 6.5708e-04, 4.8828e-04,  ..., 9.0027e-02,
           1.3939e-02, 2.7847e-03]],

         [[3.3618e-01, 1.2314e-04, 6.6519e-05,  ..., 7.1669e-04,
           5.6553e-04, 6.6681e-03]],

         [[6.1230e-01, 1.3947e-05, 1.7226e-05,  ..., 7.7362e-03,
           2.4872e-02, 1.5656e-02]],

         ...,

         [[1.6382e-01, 2.7990e-04, 6.9797e-05,  ..., 1.8959e-03,
           7.4272e-03, 6.2927e-02]],

         [[9.5398e-02, 3.6538e-05, 3.0458e-05,  ..., 3.2187e-05,
           3.7372e-05, 6.7520e-03]],

         [[8.5059e-01, 1.7929e-04, 8.3387e-05,  ..., 4.0665e-03,
           7.0496e-03, 7.6818e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2441e-01, 2.6393e-04, 3.2592e-04,  ..., 5.6122e-02,
           2.5562e-01, 7.1907e-03]],

         [[2.5317e-01, 7.7820e-04, 3.5620e-04,  ..., 1.3741e-02,
           1.0658e-02, 1.6647e-02]],

         [[2.2253e-01, 1.2279e-05, 1.1265e-05,  ..., 4.7646e-03,
           3.6041e-02, 2.4060e-01]],

         ...,

         [[5.0195e-01, 1.6868e-05, 1.7285e-05,  ..., 4.1687e-02,
           3.7079e-02, 1.4580e-02]],

         [[8.2031e-01, 9.4771e-06, 1.1563e-05,  ..., 4.4403e-03,
           7.6141e-03, 3.4119e-02]],

         [[2.0007e-01, 3.3438e-05, 2.7061e-05,  ..., 8.7891e-03,
           1.3298e-02, 3.0518e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 8.3089e-05, 4.5180e-05,  ..., 2.0771e-03,
           1.5106e-02, 1.5587e-02]],

         [[2.0837e-01, 3.3021e-05, 4.8041e-05,  ..., 1.3245e-01,
           6.8726e-02, 1.3321e-02]],

         [[1.2262e-01, 7.2718e-06, 3.4571e-06,  ..., 2.5864e-03,
           3.0231e-03, 7.9980e-01]],

         ...,

         [[1.6931e-01, 2.3723e-04, 1.3208e-04,  ..., 1.1148e-03,
           3.8958e-04, 2.3453e-02]],

         [[4.9487e-01, 3.7575e-04, 2.0945e-04,  ..., 5.0468e-03,
           6.1569e-03, 9.1705e-03]],

         [[3.5010e-01, 6.6280e-04, 5.2023e-04,  ..., 4.3907e-03,
           8.3313e-03, 4.5471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7920e-01, 5.1260e-05, 2.3842e-05,  ..., 1.9653e-02,
           1.7883e-02, 5.1758e-02]],

         [[2.4976e-01, 2.1338e-05, 4.5300e-06,  ..., 6.2523e-03,
           9.0075e-04, 7.5417e-03]],

         [[8.0615e-01, 5.3644e-05, 2.4676e-05,  ..., 8.7509e-03,
           1.8196e-03, 2.9297e-03]],

         ...,

         [[1.3147e-01, 1.3943e-03, 1.9665e-03,  ..., 1.6882e-01,
           1.7120e-02, 7.8354e-03]],

         [[8.1201e-01, 2.8014e-05, 5.7101e-05,  ..., 1.6680e-03,
           1.0443e-03, 2.6505e-02]],

         [[4.9976e-01, 9.0647e-04, 1.6289e-03,  ..., 8.5510e-02,
           3.1464e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9321e-01, 3.0303e-04, 5.8842e-04,  ..., 4.8615e-02,
           4.2908e-02, 1.3924e-02]],

         [[1.9165e-01, 3.5286e-04, 4.4966e-04,  ..., 5.3192e-02,
           2.5925e-02, 3.3295e-02]],

         [[2.3865e-01, 5.0664e-05, 1.4043e-04,  ..., 2.2537e-02,
           1.9745e-02, 2.6978e-02]],

         ...,

         [[5.1208e-02, 8.5890e-05, 1.5068e-04,  ..., 2.4185e-02,
           9.9258e-03, 1.2352e-02]],

         [[6.4087e-02, 4.7684e-05, 4.8459e-05,  ..., 7.1192e-04,
           1.3514e-03, 5.7434e-02]],

         [[3.8501e-01, 2.5702e-04, 5.0688e-04,  ..., 1.7731e-02,
           1.1269e-02, 9.9487e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1924e-01, 2.1756e-05, 2.0921e-05,  ..., 3.8319e-03,
           4.5166e-03, 3.7811e-02]],

         [[3.2227e-01, 1.3459e-04, 2.7084e-04,  ..., 2.0599e-02,
           3.1647e-02, 3.5248e-03]],

         [[2.5122e-01, 2.5439e-04, 5.8031e-04,  ..., 6.9702e-02,
           9.0942e-02, 3.1006e-01]],

         ...,

         [[4.9463e-01, 1.6403e-04, 2.8014e-04,  ..., 3.7262e-02,
           1.7319e-02, 2.7039e-02]],

         [[1.1444e-01, 4.7088e-06, 5.0843e-05,  ..., 6.5422e-04,
           8.3351e-04, 7.0496e-02]],

         [[1.9141e-01, 2.8563e-04, 5.3358e-04,  ..., 4.9057e-03,
           4.0665e-03, 1.0777e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6636e-01, 2.3234e-04, 1.7917e-04,  ..., 1.3741e-02,
           9.5139e-03, 9.4238e-02]],

         [[1.0718e-01, 8.6427e-05, 1.2851e-04,  ..., 1.4503e-02,
           8.7280e-03, 7.2510e-02]],

         [[3.5571e-01, 2.5153e-04, 5.8508e-04,  ..., 2.6688e-02,
           1.9516e-02, 3.3356e-02]],

         ...,

         [[7.1680e-01, 0.0000e+00, 1.1921e-07,  ..., 6.2164e-02,
           1.3257e-01, 7.3792e-02]],

         [[4.2236e-01, 9.5367e-06, 4.2677e-05,  ..., 1.7883e-01,
           1.7200e-01, 3.1067e-02]],

         [[2.0715e-01, 1.7881e-07, 3.5763e-06,  ..., 5.4016e-02,
           7.0068e-01, 1.8677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0742e-02, 2.3842e-06, 9.5367e-06,  ..., 5.1239e-02,
           1.7853e-02, 4.0527e-02]],

         [[9.8114e-03, 3.5763e-07, 2.3842e-07,  ..., 8.5533e-05,
           1.9779e-03, 9.7754e-01]],

         [[5.3864e-02, 1.6809e-04, 7.6246e-04,  ..., 1.5222e-01,
           6.3965e-02, 2.5578e-03]],

         ...,

         [[1.6907e-01, 3.0899e-04, 7.9298e-04,  ..., 3.4277e-01,
           5.4688e-02, 7.5455e-03]],

         [[4.7681e-01, 1.0848e-05, 3.5822e-05,  ..., 7.6660e-02,
           1.2537e-01, 1.3928e-01]],

         [[1.0913e-01, 8.9407e-07, 2.1279e-05,  ..., 1.7993e-01,
           4.4873e-01, 1.8555e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 7.6294e-05, 3.8791e-04,  ..., 4.9042e-02,
           2.1515e-02, 1.2253e-02]],

         [[9.5264e-01, 2.7895e-05, 1.1194e-04,  ..., 2.8248e-03,
           3.9005e-04, 5.7411e-04]],

         [[7.7271e-02, 8.8811e-06, 4.3094e-05,  ..., 2.9964e-03,
           9.0647e-04, 1.2989e-03]],

         ...,

         [[7.6709e-01, 2.2471e-05, 2.6941e-05,  ..., 9.8114e-03,
           6.7017e-02, 4.3976e-02]],

         [[2.3840e-01, 3.8147e-06, 2.3246e-06,  ..., 5.1880e-03,
           4.3335e-03, 7.3700e-03]],

         [[7.8308e-02, 1.5557e-05, 3.9756e-05,  ..., 6.8970e-03,
           9.2087e-03, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9873e-01, 1.4663e-05, 2.6286e-05,  ..., 2.1826e-01,
           2.0178e-01, 3.8086e-02]],

         [[7.8467e-01, 2.1338e-05, 1.6749e-05,  ..., 4.7417e-03,
           1.9196e-02, 6.4148e-02]],

         [[2.4585e-01, 5.3823e-05, 9.3400e-05,  ..., 8.3984e-02,
           1.5503e-01, 2.7100e-01]],

         ...,

         [[5.6836e-01, 1.0788e-05, 6.8665e-05,  ..., 1.8997e-02,
           1.5784e-01, 5.5847e-02]],

         [[5.9717e-01, 2.8586e-04, 1.0414e-03,  ..., 6.3904e-02,
           6.3416e-02, 1.3504e-02]],

         [[6.3867e-01, 2.4617e-05, 1.5676e-04,  ..., 3.4393e-02,
           1.5417e-01, 1.1345e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7942e-02, 6.2227e-05, 3.7074e-05,  ..., 2.6684e-03,
           8.5449e-03, 5.6610e-02]],

         [[1.1206e-01, 4.6313e-05, 3.8505e-05,  ..., 7.1335e-03,
           8.9874e-03, 1.6144e-02]],

         [[2.4146e-01, 1.9741e-04, 6.1750e-04,  ..., 4.0985e-02,
           1.1407e-01, 1.1139e-01]],

         ...,

         [[6.8298e-02, 2.2948e-05, 1.9038e-04,  ..., 3.6377e-01,
           8.5022e-02, 1.8349e-03]],

         [[1.5430e-01, 7.8917e-05, 3.4285e-04,  ..., 2.2095e-01,
           3.7012e-01, 3.8033e-03]],

         [[5.8936e-01, 1.4019e-04, 2.8110e-04,  ..., 4.1382e-02,
           1.7554e-01, 7.5500e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2090e-01, 1.7583e-05, 5.3525e-05,  ..., 1.0727e-02,
           1.0400e-01, 3.6157e-01]],

         [[3.3594e-01, 6.7294e-05, 7.1347e-05,  ..., 1.3416e-01,
           7.9468e-02, 2.4866e-01]],

         [[5.4779e-02, 8.9407e-07, 1.1921e-06,  ..., 2.3889e-01,
           6.3428e-01, 2.9205e-02]],

         ...,

         [[5.9424e-01, 1.1760e-04, 2.6703e-04,  ..., 6.7291e-03,
           1.3916e-02, 1.0674e-02]],

         [[4.2188e-01, 2.9683e-05, 7.0632e-05,  ..., 3.7750e-02,
           6.5186e-02, 7.3738e-03]],

         [[2.1667e-02, 2.9802e-05, 1.9729e-05,  ..., 3.0022e-03,
           3.4882e-02, 8.0664e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2051e-01, 3.0041e-05, 3.3259e-04,  ..., 6.1676e-02,
           9.1858e-02, 1.5625e-01]],

         [[6.1401e-02, 3.2163e-04, 1.0405e-03,  ..., 1.1009e-02,
           1.5526e-02, 1.1673e-02]],

         [[2.5562e-01, 1.2159e-04, 4.8184e-04,  ..., 3.9185e-02,
           2.1301e-02, 3.9490e-02]],

         ...,

         [[8.3203e-01, 3.6407e-04, 2.5225e-04,  ..., 1.2245e-02,
           3.2257e-02, 2.2522e-02]],

         [[1.2756e-01, 1.2195e-04, 1.2577e-04,  ..., 8.8196e-03,
           4.3427e-02, 7.4609e-01]],

         [[6.8848e-01, 1.9073e-05, 1.6499e-04,  ..., 1.9836e-02,
           2.2351e-01, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5996e-01, 6.1154e-05, 1.1148e-03,  ..., 5.3864e-02,
           3.4180e-01, 3.1738e-02]],

         [[9.9304e-02, 1.9789e-05, 7.0274e-05,  ..., 7.5226e-03,
           1.2924e-02, 3.2471e-02]],

         [[3.4302e-01, 3.1877e-04, 2.2469e-03,  ..., 1.2280e-01,
           6.9946e-02, 3.3966e-02]],

         ...,

         [[5.8838e-01, 1.4472e-04, 1.6508e-03,  ..., 4.6814e-02,
           4.4312e-02, 6.9695e-03]],

         [[6.9727e-01, 7.8678e-06, 1.0366e-03,  ..., 3.3188e-03,
           5.9395e-03, 9.9640e-03]],

         [[6.9092e-01, 5.0485e-05, 1.0335e-04,  ..., 6.0364e-02,
           6.2042e-02, 4.5563e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1478e-04, 5.4789e-04, 1.1786e-01,  ..., 1.9699e-02,
           8.6243e-02, 6.6490e-03]],

         [[1.8024e-04, 1.1700e-04, 1.2421e-01,  ..., 2.5940e-02,
           6.9702e-02, 1.1761e-01]],

         [[6.0272e-04, 3.4213e-04, 8.1848e-02,  ..., 3.9429e-02,
           1.4880e-01, 3.9276e-02]],

         ...,

         [[1.8609e-04, 4.5240e-05, 5.5786e-02,  ..., 1.0101e-01,
           5.9961e-01, 7.3303e-02]],

         [[2.1672e-04, 8.6737e-04, 3.0396e-01,  ..., 3.6316e-02,
           2.7191e-02, 1.8654e-03]],

         [[4.9829e-05, 7.4804e-05, 4.3579e-02,  ..., 4.4518e-03,
           7.1144e-03, 1.9913e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2541e-04, 2.2972e-04, 3.0029e-01,  ..., 1.3504e-02,
           3.2928e-02, 2.9335e-03]],

         [[2.1017e-04, 3.9577e-04, 1.7126e-01,  ..., 1.0696e-02,
           2.3361e-02, 1.1475e-02]],

         [[3.6299e-05, 4.4227e-05, 2.6636e-01,  ..., 1.7166e-02,
           2.7634e-02, 1.2767e-04]],

         ...,

         [[1.4925e-03, 1.0691e-03, 2.2412e-01,  ..., 2.9526e-02,
           3.4241e-02, 6.1646e-03]],

         [[3.1888e-05, 8.3983e-05, 2.4939e-01,  ..., 4.0169e-03,
           3.0731e-02, 5.8746e-04]],

         [[6.5446e-05, 1.7107e-04, 1.1652e-01,  ..., 3.7915e-01,
           5.3772e-02, 8.0681e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6904e-04, 2.5082e-04, 1.5552e-01,  ..., 1.8356e-02,
           2.4414e-02, 9.0837e-05]],

         [[5.9938e-04, 5.7876e-05, 2.6147e-01,  ..., 1.4412e-02,
           2.0477e-02, 9.1248e-03]],

         [[4.3416e-04, 1.4772e-03, 2.4646e-01,  ..., 1.2039e-02,
           1.9608e-02, 1.5783e-03]],

         ...,

         [[1.4377e-04, 2.4331e-04, 1.4807e-01,  ..., 1.3771e-02,
           7.8918e-02, 5.7526e-02]],

         [[1.2469e-04, 7.6294e-04, 2.4207e-01,  ..., 2.0996e-02,
           1.4145e-02, 1.0513e-02]],

         [[3.6788e-04, 3.4571e-04, 1.3916e-01,  ..., 7.8201e-03,
           1.2207e-02, 6.3591e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8777e-04, 4.5359e-05, 2.9419e-02,  ..., 1.3501e-01,
           4.0283e-01, 1.9946e-01]],

         [[8.3864e-05, 2.3949e-04, 1.6138e-01,  ..., 1.7517e-01,
           5.4474e-02, 5.0964e-02]],

         [[7.0763e-04, 3.0746e-03, 1.6626e-01,  ..., 4.0100e-02,
           2.6913e-03, 2.0008e-03]],

         ...,

         [[7.0715e-04, 5.0306e-05, 1.0193e-02,  ..., 1.5535e-03,
           1.2732e-03, 8.1253e-03]],

         [[4.2076e-03, 1.4496e-04, 6.9824e-02,  ..., 1.3562e-01,
           1.5308e-01, 6.4575e-02]],

         [[1.0669e-05, 3.8147e-06, 3.3245e-03,  ..., 3.9581e-02,
           2.2552e-02, 1.2878e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3618e-03, 2.9869e-03, 1.9702e-01,  ..., 9.6970e-03,
           5.6992e-03, 4.9835e-02]],

         [[1.2531e-03, 2.1744e-03, 2.7539e-01,  ..., 1.4885e-02,
           2.1912e-02, 2.2171e-02]],

         [[2.7199e-03, 1.8635e-03, 1.8103e-01,  ..., 5.8655e-02,
           6.8176e-02, 2.5665e-02]],

         ...,

         [[4.6501e-03, 6.3658e-04, 1.7310e-01,  ..., 2.0523e-02,
           3.5187e-02, 9.4849e-02]],

         [[3.6182e-03, 1.2169e-03, 1.5503e-01,  ..., 1.8524e-02,
           1.1615e-01, 2.8735e-01]],

         [[5.0187e-05, 1.5640e-04, 5.3345e-02,  ..., 2.8549e-02,
           6.8652e-01, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2994e-05, 4.7684e-07, 2.1636e-04,  ..., 1.2695e-01,
           7.7783e-01, 4.9713e-02]],

         [[7.5340e-05, 2.4438e-05, 4.9114e-04,  ..., 9.2697e-03,
           3.4103e-03, 9.4531e-01]],

         [[5.9605e-08, 0.0000e+00, 6.1393e-06,  ..., 2.0325e-04,
           2.0325e-04, 9.9854e-01]],

         ...,

         [[3.5591e-03, 5.7335e-03, 8.1482e-03,  ..., 4.9530e-02,
           1.1856e-02, 3.6890e-01]],

         [[9.0256e-03, 1.2329e-02, 1.6861e-02,  ..., 9.1125e-02,
           4.2610e-03, 8.5571e-02]],

         [[8.6136e-03, 3.5915e-03, 1.0948e-03,  ..., 1.0828e-01,
           7.2144e-02, 2.6535e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1052e-04, 1.2302e-03, 5.3444e-03,  ..., 1.4673e-01,
           1.5869e-01, 2.7856e-01]],

         [[0.0000e+00, 0.0000e+00, 1.7881e-07,  ..., 1.5881e-01,
           7.5732e-01, 7.1533e-02]],

         [[1.0567e-03, 1.5135e-03, 7.6866e-03,  ..., 1.7224e-01,
           6.9580e-02, 4.1534e-02]],

         ...,

         [[2.0921e-05, 6.8784e-05, 1.1015e-03,  ..., 6.5674e-01,
           2.2705e-01, 2.1957e-02]],

         [[1.3304e-03, 5.8479e-03, 2.2156e-02,  ..., 1.8542e-01,
           3.6652e-02, 4.7089e-02]],

         [[1.3371e-03, 5.2071e-03, 8.3160e-03,  ..., 4.3677e-01,
           7.0190e-02, 3.2654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6711e-02, 7.9041e-03, 3.3783e-02,  ..., 8.5632e-02,
           5.7495e-02, 6.8787e-02]],

         [[3.2482e-03, 9.1600e-04, 2.8763e-03,  ..., 1.1604e-02,
           6.7177e-03, 8.0414e-03]],

         [[1.4313e-02, 4.1885e-03, 4.2458e-03,  ..., 7.0984e-02,
           4.3060e-02, 5.9326e-02]],

         ...,

         [[2.4605e-03, 1.9252e-05, 3.1590e-05,  ..., 6.0577e-02,
           7.4951e-01, 1.4197e-01]],

         [[5.4016e-03, 1.2636e-03, 1.0471e-03,  ..., 2.4451e-01,
           1.0516e-01, 1.0681e-01]],

         [[1.6571e-02, 1.3027e-03, 1.3954e-02,  ..., 9.3872e-02,
           1.2244e-01, 2.5317e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1914e-01, 1.7285e-06, 5.9009e-06,  ..., 1.3485e-03,
           9.6054e-03, 4.9706e-03]],

         [[8.6963e-01, 5.4836e-06, 8.6427e-06,  ..., 4.7150e-02,
           5.4260e-02, 4.4861e-03]],

         [[8.7988e-01, 8.0466e-06, 3.3796e-05,  ..., 1.1330e-02,
           2.1881e-02, 3.0289e-02]],

         ...,

         [[9.0454e-02, 7.4089e-05, 2.2960e-04,  ..., 8.6308e-04,
           3.1376e-04, 8.0261e-03]],

         [[6.2305e-01, 1.3232e-04, 1.1463e-03,  ..., 1.1726e-02,
           4.3396e-02, 1.4061e-02]],

         [[4.1943e-01, 2.0730e-04, 5.1260e-04,  ..., 7.6660e-02,
           2.7557e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1748e-01, 7.6890e-06, 1.2159e-05,  ..., 9.1124e-04,
           1.5259e-05, 1.6052e-02]],

         [[8.5889e-01, 0.0000e+00, 0.0000e+00,  ..., 1.7715e-02,
           1.0193e-01, 1.7578e-02]],

         [[9.4580e-01, 7.2896e-05, 7.2718e-06,  ..., 3.4580e-03,
           6.6681e-03, 1.1612e-02]],

         ...,

         [[4.1626e-01, 3.2854e-04, 2.7442e-04,  ..., 7.5073e-02,
           5.1994e-03, 5.5656e-03]],

         [[1.8970e-01, 1.0004e-03, 5.1689e-04,  ..., 7.5928e-02,
           2.6436e-03, 2.7733e-03]],

         [[7.3633e-01, 1.5438e-05, 8.7857e-05,  ..., 9.4116e-02,
           9.3933e-02, 4.0741e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 4.1056e-04, 2.0003e-04,  ..., 8.1360e-02,
           5.1636e-02, 3.0991e-02]],

         [[6.8262e-01, 5.3644e-06, 9.3579e-06,  ..., 9.3140e-02,
           1.0120e-01, 1.5480e-02]],

         [[7.4268e-01, 2.5153e-05, 4.0531e-06,  ..., 9.9426e-02,
           1.2103e-01, 2.5387e-03]],

         ...,

         [[2.3694e-01, 2.3842e-07, 4.7684e-07,  ..., 9.1851e-05,
           1.6570e-04, 7.3828e-01]],

         [[1.5894e-01, 1.7881e-07, 5.1260e-06,  ..., 8.3447e-07,
           1.1921e-07, 1.2886e-02]],

         [[8.3691e-01, 1.1921e-07, 8.3447e-07,  ..., 7.7486e-07,
           5.9605e-08, 2.9266e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1948e-01, 1.5616e-05, 1.6153e-05,  ..., 6.7688e-02,
           6.4502e-01, 1.3397e-02]],

         [[9.1113e-01, 2.6822e-06, 1.7881e-06,  ..., 8.8787e-04,
           2.9922e-04, 3.6983e-03]],

         [[7.4951e-01, 3.2806e-04, 9.6977e-05,  ..., 9.6893e-03,
           8.4305e-03, 5.4413e-02]],

         ...,

         [[8.3447e-01, 7.2718e-05, 1.2934e-05,  ..., 2.6001e-02,
           1.0651e-01, 5.7755e-03]],

         [[9.2969e-01, 6.8784e-05, 2.7120e-05,  ..., 1.2390e-02,
           1.5884e-02, 2.7679e-02]],

         [[8.9551e-01, 1.2279e-04, 7.3075e-05,  ..., 1.0460e-02,
           6.9122e-03, 2.2659e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0078e-01, 2.8014e-05, 3.7193e-05,  ..., 5.4512e-03,
           2.3148e-02, 3.2440e-02]],

         [[5.3174e-01, 5.3024e-04, 1.4842e-04,  ..., 1.8311e-02,
           3.6896e-02, 1.0724e-01]],

         [[5.3467e-02, 1.4305e-06, 7.7486e-07,  ..., 1.8021e-02,
           8.7500e-01, 4.1412e-02]],

         ...,

         [[6.0449e-01, 1.0878e-04, 6.2704e-05,  ..., 9.1248e-02,
           2.5925e-02, 1.0925e-02]],

         [[9.3701e-01, 2.0814e-04, 3.1650e-05,  ..., 4.5013e-03,
           3.7327e-03, 1.3733e-03]],

         [[7.1094e-01, 3.8743e-06, 1.8179e-05,  ..., 4.5349e-02,
           1.8970e-01, 1.6235e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6997e-01, 1.7958e-03, 8.5831e-04,  ..., 6.7215e-03,
           1.4372e-03, 8.3494e-04]],

         [[9.6436e-01, 1.1623e-05, 3.2067e-05,  ..., 3.5095e-03,
           8.6355e-04, 4.2915e-04]],

         [[8.9502e-01, 8.1658e-06, 7.5102e-06,  ..., 1.9989e-03,
           3.1204e-03, 7.3576e-04]],

         ...,

         [[6.5186e-01, 9.9361e-05, 9.4056e-05,  ..., 3.1235e-02,
           5.2834e-03, 2.1252e-01]],

         [[8.9502e-01, 2.9802e-07, 1.7285e-06,  ..., 5.2216e-02,
           1.8967e-02, 1.7258e-02]],

         [[9.4043e-01, 3.5763e-06, 7.7486e-06,  ..., 7.0877e-03,
           2.2446e-02, 8.2245e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4194e-01, 2.6846e-04, 1.1940e-03,  ..., 5.6458e-02,
           8.5144e-03, 3.9368e-03]],

         [[6.7432e-01, 1.0848e-05, 4.2260e-05,  ..., 3.0319e-02,
           2.2424e-01, 1.1444e-02]],

         [[8.3496e-01, 6.2585e-06, 6.1870e-05,  ..., 6.6101e-02,
           2.8351e-02, 2.6169e-02]],

         ...,

         [[7.3438e-01, 6.8665e-05, 2.0790e-04,  ..., 1.2337e-02,
           1.2074e-03, 3.8075e-04]],

         [[7.0898e-01, 1.8096e-04, 4.8041e-05,  ..., 2.1790e-02,
           9.7656e-02, 5.3291e-03]],

         [[8.4961e-01, 5.3644e-06, 7.4506e-06,  ..., 1.1591e-01,
           7.2861e-04, 2.9907e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8867e-02, 3.0398e-06, 2.0385e-05,  ..., 3.6407e-04,
           6.0618e-05, 4.5662e-03]],

         [[8.3643e-01, 6.3181e-06, 9.1195e-06,  ..., 6.7444e-02,
           6.1302e-03, 6.3744e-03]],

         [[8.0957e-01, 4.7684e-07, 4.0531e-06,  ..., 5.8098e-03,
           7.1049e-04, 2.4986e-03]],

         ...,

         [[9.3066e-01, 7.5936e-05, 3.8028e-05,  ..., 9.1400e-03,
           1.2350e-03, 1.4019e-03]],

         [[8.6816e-01, 1.2165e-04, 6.0439e-05,  ..., 2.6520e-02,
           4.5776e-03, 6.9466e-03]],

         [[7.3389e-01, 5.6362e-04, 2.0409e-04,  ..., 1.0605e-02,
           3.2234e-04, 4.8018e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 1.0192e-05, 4.9472e-06,  ..., 2.3331e-02,
           3.2074e-02, 5.5664e-01]],

         [[8.0713e-01, 4.2748e-04, 3.3426e-04,  ..., 4.5242e-03,
           6.1493e-03, 3.3379e-03]],

         [[8.0762e-01, 6.1035e-04, 5.6934e-04,  ..., 5.1025e-02,
           2.7771e-03, 2.6798e-03]],

         ...,

         [[5.6787e-01, 4.7565e-05, 1.4102e-04,  ..., 1.2993e-02,
           2.9312e-02, 5.0201e-03]],

         [[8.7128e-03, 1.0133e-06, 1.0729e-06,  ..., 6.5756e-04,
           8.7256e-01, 5.6152e-03]],

         [[6.9763e-02, 9.1791e-06, 2.2709e-05,  ..., 5.3131e-02,
           2.0065e-02, 8.1787e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0381e-01, 2.0373e-04, 3.3069e-04,  ..., 3.3905e-02,
           3.0014e-02, 4.2310e-01]],

         [[6.1621e-01, 8.0943e-05, 1.3494e-04,  ..., 6.2988e-02,
           2.0325e-02, 5.3520e-03]],

         [[3.7109e-01, 1.2815e-04, 1.0496e-04,  ..., 5.1880e-03,
           7.5867e-02, 8.3847e-03]],

         ...,

         [[1.0681e-02, 1.0729e-06, 2.7418e-06,  ..., 8.5163e-04,
           7.9811e-05, 9.8242e-01]],

         [[7.8125e-01, 1.7083e-04, 1.4675e-04,  ..., 1.1475e-02,
           1.3268e-02, 9.8267e-03]],

         [[8.4180e-01, 7.9811e-05, 2.8789e-05,  ..., 1.9951e-03,
           7.1259e-03, 1.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5986e-01, 6.2609e-04, 1.2922e-03,  ..., 1.3962e-02,
           7.2441e-03, 5.9891e-03]],

         [[7.5488e-01, 1.5974e-05, 6.1691e-05,  ..., 1.5417e-01,
           6.1569e-03, 9.9945e-03]],

         [[3.1152e-01, 2.6584e-05, 6.0380e-05,  ..., 2.6367e-02,
           6.0844e-04, 2.5654e-03]],

         ...,

         [[6.6211e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1346e-01,
           1.9836e-01, 2.0416e-02]],

         [[5.2051e-01, 4.9543e-04, 5.5504e-04,  ..., 1.9897e-02,
           1.6418e-02, 5.8517e-03]],

         [[1.5210e-01, 1.1921e-07, 1.1921e-07,  ..., 1.7960e-02,
           8.0518e-01, 1.6983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9111e-01, 4.2081e-05, 1.3471e-05,  ..., 5.5084e-03,
           5.5122e-03, 3.9482e-03]],

         [[7.0496e-02, 5.8603e-04, 5.3835e-04,  ..., 2.0679e-01,
           3.0231e-04, 6.6137e-04]],

         [[5.0049e-01, 2.2113e-04, 1.6177e-04,  ..., 1.3867e-01,
           1.0777e-04, 2.9397e-04]],

         ...,

         [[4.1577e-01, 5.9605e-07, 2.3842e-06,  ..., 3.5767e-01,
           6.2073e-02, 5.9967e-03]],

         [[1.9751e-01, 5.9605e-08, 5.9605e-08,  ..., 3.2788e-01,
           4.4141e-01, 8.3389e-03]],

         [[8.9294e-02, 2.5630e-06, 6.1393e-06,  ..., 2.2949e-01,
           1.3351e-02, 2.6417e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6294e-02, 6.5565e-07, 1.2517e-06,  ..., 5.2643e-03,
           2.4765e-02, 8.0127e-01]],

         [[1.2238e-01, 9.4771e-06, 2.3007e-05,  ..., 4.0918e-01,
           5.4535e-02, 1.7395e-02]],

         [[2.5781e-01, 2.8563e-04, 3.9768e-04,  ..., 7.3303e-02,
           1.2810e-02, 8.7738e-03]],

         ...,

         [[3.3887e-01, 5.4538e-05, 8.3506e-05,  ..., 4.1840e-02,
           1.4626e-02, 4.8180e-03]],

         [[5.6152e-01, 3.0398e-06, 1.0490e-05,  ..., 2.0905e-02,
           1.7662e-03, 3.6488e-03]],

         [[1.8799e-01, 1.3876e-04, 1.6773e-04,  ..., 1.2379e-03,
           1.1467e-02, 8.0185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4902e-01, 2.1935e-05, 5.3167e-05,  ..., 3.5858e-02,
           1.0277e-02, 1.5251e-02]],

         [[2.2607e-01, 2.2376e-04, 3.6240e-04,  ..., 7.0007e-02,
           2.6169e-02, 1.8478e-02]],

         [[2.2430e-02, 2.6822e-06, 2.0862e-06,  ..., 8.2855e-03,
           6.0156e-01, 1.1818e-02]],

         ...,

         [[7.0898e-01, 4.9233e-05, 3.8207e-05,  ..., 1.9257e-02,
           3.7193e-03, 1.7746e-02]],

         [[5.5225e-01, 3.2425e-05, 5.4955e-05,  ..., 1.0254e-01,
           7.3128e-03, 1.4153e-02]],

         [[7.4854e-01, 9.1314e-05, 6.3121e-05,  ..., 1.1383e-02,
           1.1276e-02, 6.8130e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8809e-01, 1.9608e-03, 8.5688e-04,  ..., 1.2123e-02,
           2.7823e-04, 5.0688e-04]],

         [[1.7908e-01, 2.8658e-04, 2.5630e-04,  ..., 1.2817e-02,
           1.4641e-02, 1.5373e-02]],

         [[5.1807e-01, 2.3663e-05, 4.4167e-05,  ..., 7.8186e-02,
           4.5441e-02, 1.2306e-02]],

         ...,

         [[2.1021e-01, 1.1116e-04, 4.1544e-05,  ..., 6.1646e-02,
           2.7145e-02, 8.1635e-03]],

         [[8.9893e-01, 1.0598e-04, 1.4210e-04,  ..., 2.7728e-04,
           5.5432e-06, 8.7128e-03]],

         [[9.1553e-01, 7.7784e-05, 6.8009e-05,  ..., 1.8341e-02,
           4.4518e-03, 1.0881e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8799e-01, 5.7125e-04, 1.3075e-03,  ..., 8.4778e-02,
           1.8435e-03, 3.8818e-02]],

         [[2.8931e-01, 4.2915e-03, 3.1166e-03,  ..., 2.0477e-02,
           1.7338e-03, 2.5635e-03]],

         [[1.9812e-01, 1.6809e-05, 2.3901e-05,  ..., 1.3000e-01,
           1.1200e-01, 1.6040e-01]],

         ...,

         [[1.7334e-01, 1.4484e-05, 6.0618e-05,  ..., 6.6833e-02,
           1.0872e-02, 2.2869e-03]],

         [[3.6035e-01, 1.2100e-05, 3.1173e-05,  ..., 8.2947e-02,
           2.2559e-01, 3.1738e-02]],

         [[2.0984e-01, 9.2387e-06, 1.9848e-05,  ..., 1.7676e-01,
           2.3962e-01, 1.5710e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3252e-01, 1.1462e-04, 1.8609e-04,  ..., 2.4750e-02,
           8.9645e-03, 2.5543e-02]],

         [[1.9507e-01, 5.2571e-05, 1.2422e-04,  ..., 2.2107e-01,
           9.5032e-02, 6.8855e-03]],

         [[5.3284e-02, 4.1127e-06, 1.8477e-06,  ..., 7.4816e-04,
           8.3008e-03, 9.3018e-01]],

         ...,

         [[4.0680e-02, 1.2898e-04, 3.1304e-04,  ..., 1.7452e-03,
           2.9926e-03, 1.3306e-02]],

         [[3.0933e-01, 2.7657e-04, 4.4560e-04,  ..., 1.2085e-02,
           2.5391e-02, 5.5908e-02]],

         [[2.0410e-01, 1.7729e-03, 2.5806e-03,  ..., 3.2806e-02,
           4.7073e-03, 6.7177e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7715e-01, 1.1790e-04, 1.0085e-04,  ..., 5.3680e-02,
           5.0049e-02, 1.8265e-02]],

         [[3.7594e-03, 9.5367e-07, 1.3709e-06,  ..., 1.6868e-05,
           2.0027e-05, 1.5891e-04]],

         [[8.9404e-01, 2.5988e-05, 2.1815e-05,  ..., 8.0919e-04,
           4.6825e-04, 2.7537e-04]],

         ...,

         [[1.1536e-01, 1.1663e-03, 2.3746e-03,  ..., 2.9617e-02,
           1.6876e-02, 3.8635e-02]],

         [[3.3960e-01, 3.7551e-05, 8.1420e-05,  ..., 8.1100e-03,
           2.1582e-01, 4.4525e-02]],

         [[5.5762e-01, 1.2579e-03, 3.8452e-03,  ..., 2.0950e-02,
           1.0368e-02, 2.8641e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7437e-01, 1.4722e-04, 6.9666e-04,  ..., 5.7556e-02,
           4.2786e-02, 3.4912e-02]],

         [[6.7444e-02, 1.3757e-04, 6.0225e-04,  ..., 2.2949e-02,
           2.3823e-03, 4.7089e-02]],

         [[2.4500e-01, 3.7909e-05, 1.9717e-04,  ..., 1.5182e-02,
           4.9667e-03, 7.1289e-02]],

         ...,

         [[8.5449e-02, 2.5558e-04, 3.3855e-04,  ..., 1.1625e-03,
           6.3858e-03, 3.8513e-02]],

         [[2.5482e-02, 3.5405e-05, 6.3181e-05,  ..., 3.1590e-04,
           3.4485e-03, 1.5701e-02]],

         [[2.3108e-01, 1.0514e-04, 2.2781e-04,  ..., 2.2171e-02,
           2.0828e-02, 1.5236e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8396e-01, 1.5318e-05, 2.5034e-05,  ..., 2.0325e-02,
           7.0374e-02, 1.4783e-01]],

         [[4.8242e-01, 1.6439e-04, 4.9257e-04,  ..., 6.4270e-02,
           5.1956e-03, 1.7578e-02]],

         [[2.6611e-01, 1.7607e-04, 4.7851e-04,  ..., 7.2144e-02,
           1.0559e-02, 4.7778e-01]],

         ...,

         [[6.9971e-01, 1.4687e-04, 3.7503e-04,  ..., 2.0966e-02,
           9.1629e-03, 4.5074e-02]],

         [[2.9688e-01, 1.1462e-04, 2.2447e-04,  ..., 2.1133e-03,
           3.2978e-03, 3.6285e-02]],

         [[1.0602e-01, 7.3552e-05, 2.1279e-04,  ..., 1.5974e-03,
           1.0719e-03, 1.5396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3391e-01, 9.8884e-05, 1.4329e-04,  ..., 5.1239e-02,
           1.2482e-01, 1.2683e-01]],

         [[1.0815e-01, 4.3511e-05, 5.2392e-05,  ..., 5.9586e-03,
           3.1952e-02, 4.6143e-02]],

         [[4.5020e-01, 1.7798e-04, 3.1853e-04,  ..., 1.1360e-02,
           5.3772e-02, 8.0078e-02]],

         ...,

         [[5.1270e-01, 0.0000e+00, 0.0000e+00,  ..., 5.7526e-02,
           4.0552e-01, 1.4206e-02]],

         [[2.0154e-01, 7.6890e-06, 4.3452e-05,  ..., 2.8857e-01,
           1.3428e-01, 7.2823e-03]],

         [[2.5024e-01, 1.1325e-06, 1.3113e-06,  ..., 6.4270e-02,
           6.5918e-01, 1.4229e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1331e-02, 2.2411e-05, 5.7340e-05,  ..., 2.5818e-02,
           7.2998e-02, 9.7198e-03]],

         [[2.3590e-02, 5.4240e-06, 3.7551e-06,  ..., 3.3989e-03,
           4.8401e-02, 8.8525e-01]],

         [[3.7378e-01, 1.3371e-03, 3.6507e-03,  ..., 2.8824e-02,
           1.5821e-03, 1.2688e-02]],

         ...,

         [[2.7417e-01, 3.9744e-04, 7.8535e-04,  ..., 4.5471e-02,
           1.4198e-02, 3.6240e-03]],

         [[1.2952e-01, 1.0729e-06, 6.1393e-06,  ..., 1.0132e-01,
           6.5283e-01, 2.9831e-02]],

         [[7.3059e-02, 5.9605e-07, 3.5167e-06,  ..., 2.9810e-01,
           4.8779e-01, 2.8732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 1.5974e-04, 4.9686e-04,  ..., 8.1940e-03,
           5.1270e-03, 6.7368e-03]],

         [[9.6289e-01, 2.2471e-05, 8.3268e-05,  ..., 7.2896e-05,
           6.3241e-05, 1.2022e-04]],

         [[3.5669e-01, 6.4790e-05, 1.2589e-04,  ..., 5.0354e-04,
           6.4945e-04, 1.1091e-03]],

         ...,

         [[8.4619e-01, 1.5736e-05, 1.4126e-05,  ..., 1.2260e-02,
           3.8361e-02, 9.5444e-03]],

         [[4.6826e-01, 2.1458e-05, 1.7464e-05,  ..., 3.3321e-03,
           4.7379e-03, 1.0841e-02]],

         [[1.1554e-01, 1.6809e-05, 5.6386e-05,  ..., 9.3384e-03,
           6.3782e-02, 7.1875e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2559e-01, 1.4067e-05, 1.8716e-05,  ..., 1.3257e-01,
           1.2262e-01, 3.8147e-02]],

         [[7.4268e-01, 2.4498e-05, 1.8537e-05,  ..., 1.4305e-02,
           4.7089e-02, 1.3817e-02]],

         [[6.4026e-02, 1.5795e-05, 3.0458e-05,  ..., 5.7831e-02,
           2.8687e-01, 2.3636e-02]],

         ...,

         [[6.3428e-01, 1.6570e-05, 7.5459e-05,  ..., 8.5876e-02,
           8.4534e-02, 3.9001e-02]],

         [[6.5967e-01, 7.9215e-05, 6.0606e-04,  ..., 7.0068e-02,
           1.7303e-02, 8.1100e-03]],

         [[6.5674e-01, 1.5676e-05, 9.7394e-05,  ..., 2.6276e-02,
           5.5962e-03, 2.1118e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9438e-02, 1.1444e-05, 6.7353e-06,  ..., 4.2534e-03,
           3.8818e-02, 3.4981e-03]],

         [[2.5171e-01, 7.7486e-05, 6.8903e-05,  ..., 1.0674e-02,
           7.5111e-03, 5.9433e-03]],

         [[1.7200e-01, 1.1790e-04, 3.1424e-04,  ..., 6.8420e-02,
           7.3975e-02, 7.5745e-02]],

         ...,

         [[5.6213e-02, 1.6391e-05, 1.6940e-04,  ..., 8.5022e-02,
           3.8853e-03, 7.6065e-03]],

         [[1.2390e-01, 2.3127e-05, 1.6963e-04,  ..., 2.4646e-01,
           9.7046e-03, 3.7201e-02]],

         [[9.1650e-01, 6.7949e-06, 1.7345e-05,  ..., 2.7679e-02,
           1.9318e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0789e-01, 1.7285e-06, 5.3644e-06,  ..., 1.3525e-01,
           5.9424e-01, 2.8351e-02]],

         [[5.2344e-01, 1.2398e-05, 1.0550e-05,  ..., 1.8738e-02,
           1.7944e-01, 1.4819e-01]],

         [[8.8440e-02, 2.9802e-07, 4.1723e-07,  ..., 2.4915e-01,
           5.9277e-01, 2.9068e-02]],

         ...,

         [[3.7671e-01, 1.4460e-04, 4.3249e-04,  ..., 1.0811e-02,
           1.0239e-02, 2.9968e-02]],

         [[3.5742e-01, 1.0604e-04, 1.1396e-04,  ..., 1.0815e-01,
           6.9824e-02, 3.0029e-02]],

         [[4.7211e-02, 8.7976e-05, 7.4267e-05,  ..., 4.5410e-02,
           1.4099e-01, 5.2393e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9756e-01, 1.0729e-05, 1.1909e-04,  ..., 6.7322e-02,
           1.4038e-01, 1.5906e-01]],

         [[3.5400e-01, 6.7282e-04, 2.0084e-03,  ..., 1.9135e-02,
           5.5656e-03, 5.1689e-03]],

         [[1.2024e-01, 4.0054e-05, 2.0015e-04,  ..., 9.2010e-03,
           3.9902e-03, 1.5900e-02]],

         ...,

         [[6.5283e-01, 1.0109e-04, 1.2577e-04,  ..., 1.5839e-02,
           1.4575e-01, 5.6183e-02]],

         [[1.7175e-01, 5.9009e-05, 1.4377e-04,  ..., 7.7858e-03,
           4.9225e-02, 7.3438e-01]],

         [[7.4268e-01, 9.7573e-05, 3.8576e-04,  ..., 1.2978e-02,
           1.5656e-02, 2.0618e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1211e-01, 1.8609e-04, 9.5224e-04,  ..., 4.1199e-02,
           1.6077e-01, 7.8674e-02]],

         [[1.9800e-01, 5.2631e-05, 1.0973e-04,  ..., 1.1986e-02,
           5.8075e-02, 4.9683e-02]],

         [[1.7236e-01, 1.3638e-04, 2.5730e-03,  ..., 1.4465e-01,
           1.2817e-01, 2.7710e-02]],

         ...,

         [[8.1445e-01, 6.0141e-05, 1.8752e-04,  ..., 1.4664e-02,
           3.5896e-03, 3.6652e-02]],

         [[7.0801e-01, 1.7285e-06, 1.6093e-04,  ..., 1.1654e-03,
           1.0109e-03, 1.4524e-03]],

         [[6.1963e-01, 4.4167e-05, 4.0591e-05,  ..., 5.1453e-02,
           6.9214e-02, 1.3184e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6955e-04, 6.3610e-04, 1.6125e-01,  ..., 1.1792e-01,
           1.5078e-03, 1.4534e-02]],

         [[2.9087e-04, 2.7323e-04, 1.7761e-01,  ..., 1.1377e-01,
           9.0027e-02, 7.9269e-03]],

         [[2.0826e-04, 2.8610e-04, 6.7322e-02,  ..., 1.3330e-01,
           8.3801e-02, 1.3867e-01]],

         ...,

         [[2.7013e-04, 7.4446e-05, 1.0730e-01,  ..., 1.7419e-01,
           2.1167e-01, 1.3989e-01]],

         [[2.4652e-04, 5.4264e-04, 2.3938e-01,  ..., 4.9042e-02,
           3.0022e-03, 1.8753e-02]],

         [[4.8459e-05, 5.9009e-06, 2.7786e-02,  ..., 1.8616e-03,
           1.2183e-04, 2.8839e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4894e-04, 6.4909e-05, 2.7515e-01,  ..., 1.0498e-02,
           9.2125e-04, 1.7281e-03]],

         [[3.1090e-04, 1.3959e-04, 2.7197e-01,  ..., 9.4452e-03,
           2.4948e-03, 7.7581e-04]],

         [[4.3988e-05, 6.5565e-06, 2.9175e-01,  ..., 9.2316e-03,
           1.6034e-05, 8.8596e-04]],

         ...,

         [[2.9049e-03, 4.6778e-04, 2.7295e-01,  ..., 2.7130e-02,
           2.4757e-03, 8.0967e-04]],

         [[8.8930e-05, 6.3777e-05, 3.0640e-01,  ..., 1.7151e-02,
           2.5725e-04, 9.4509e-04]],

         [[2.2411e-05, 8.8215e-06, 2.0251e-01,  ..., 2.9175e-02,
           1.2493e-04, 2.5845e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2149e-04, 5.9187e-05, 2.1240e-01,  ..., 1.0933e-02,
           1.6630e-05, 8.4591e-04]],

         [[5.5313e-04, 1.0490e-05, 2.8320e-01,  ..., 1.1780e-02,
           2.7447e-03, 7.3290e-04]],

         [[4.4942e-04, 2.2376e-04, 2.5977e-01,  ..., 8.6517e-03,
           3.8815e-04, 1.1501e-03]],

         ...,

         [[2.6226e-05, 5.0664e-05, 2.1765e-01,  ..., 8.1970e-02,
           3.7689e-02, 3.3386e-02]],

         [[1.5008e-04, 9.3639e-05, 2.3108e-01,  ..., 9.0637e-03,
           5.9433e-03, 1.3418e-03]],

         [[6.8474e-04, 5.6446e-05, 2.1899e-01,  ..., 7.0648e-03,
           1.5106e-03, 7.8201e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9855e-03, 5.9128e-04, 4.9225e-02,  ..., 2.6416e-01,
           1.8433e-01, 4.0741e-03]],

         [[1.2577e-04, 2.6846e-04, 1.3220e-01,  ..., 4.0649e-02,
           6.9702e-02, 3.7292e-02]],

         [[4.9353e-04, 8.7976e-04, 7.7026e-02,  ..., 1.4715e-03,
           7.8869e-04, 2.0075e-04]],

         ...,

         [[2.4967e-03, 5.3263e-04, 3.9215e-02,  ..., 7.9041e-03,
           4.4365e-03, 5.5084e-03]],

         [[1.1711e-03, 2.1994e-04, 8.6548e-02,  ..., 5.0110e-02,
           6.5369e-02, 1.8723e-02]],

         [[1.1528e-04, 3.8266e-05, 2.4338e-02,  ..., 4.6021e-02,
           1.0699e-01, 7.8857e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7656e-03, 1.7033e-03, 2.1130e-01,  ..., 8.6517e-03,
           7.2144e-02, 2.2446e-02]],

         [[1.0262e-03, 1.2312e-03, 2.9590e-01,  ..., 1.7426e-02,
           1.3573e-02, 3.8361e-02]],

         [[3.2711e-03, 8.4972e-04, 1.9812e-01,  ..., 2.5894e-02,
           6.3843e-02, 2.9739e-02]],

         ...,

         [[1.8752e-04, 1.1921e-04, 3.6865e-02,  ..., 7.3364e-02,
           6.5869e-01, 8.8440e-02]],

         [[1.1320e-03, 4.5919e-04, 2.4646e-01,  ..., 3.8086e-02,
           3.1586e-02, 1.1914e-01]],

         [[1.3173e-04, 4.0746e-04, 2.0618e-01,  ..., 1.7358e-01,
           2.9480e-02, 1.2093e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2817e-03, 7.7009e-05, 5.8699e-04,  ..., 4.3793e-02,
           2.9468e-01, 4.7095e-01]],

         [[2.2488e-03, 1.2808e-03, 1.2154e-02,  ..., 2.7122e-03,
           2.2705e-02, 4.5605e-01]],

         [[3.1829e-05, 2.6226e-06, 1.1104e-04,  ..., 1.1104e-04,
           8.9941e-01, 9.4788e-02]],

         ...,

         [[2.6054e-03, 3.3455e-03, 5.5161e-03,  ..., 2.8015e-02,
           3.0127e-01, 4.2953e-03]],

         [[3.1338e-03, 3.5515e-03, 1.1642e-02,  ..., 1.3905e-03,
           7.5928e-02, 1.8213e-01]],

         [[8.3084e-03, 5.8899e-03, 1.1969e-03,  ..., 7.1777e-02,
           2.2583e-02, 2.1881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8065e-04, 2.2926e-03, 6.5308e-03,  ..., 1.0059e-01,
           7.3608e-02, 1.5332e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 2.1997e-01,
           6.9873e-01, 7.4829e-02]],

         [[1.8339e-03, 2.1439e-03, 6.6032e-03,  ..., 2.9587e-02,
           2.6520e-02, 1.6083e-02]],

         ...,

         [[6.2895e-04, 1.6823e-03, 6.5002e-03,  ..., 6.6711e-02,
           6.4011e-03, 2.0981e-02]],

         [[4.4136e-03, 9.5673e-03, 4.2694e-02,  ..., 2.5696e-02,
           2.3666e-02, 2.5208e-02]],

         [[4.2200e-04, 9.1457e-04, 2.8172e-03,  ..., 2.3462e-01,
           5.8984e-01, 5.4840e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7618e-02, 4.0092e-03, 1.7548e-02,  ..., 8.5083e-02,
           1.0022e-01, 7.8674e-02]],

         [[2.4605e-03, 7.0477e-04, 2.5673e-03,  ..., 2.7122e-03,
           3.9444e-03, 8.1110e-04]],

         [[8.9798e-03, 2.4261e-03, 1.9875e-03,  ..., 2.7878e-02,
           3.5645e-02, 3.6926e-02]],

         ...,

         [[4.9286e-03, 3.0422e-04, 4.3917e-04,  ..., 3.3765e-01,
           3.1226e-01, 2.3755e-01]],

         [[3.0708e-03, 6.1893e-04, 1.3256e-03,  ..., 1.8555e-01,
           4.1412e-02, 3.0777e-02]],

         [[1.9333e-02, 1.8196e-03, 9.2773e-03,  ..., 5.5084e-02,
           4.0698e-01, 8.4656e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4551e-01, 2.8014e-06, 9.4771e-06,  ..., 9.0576e-02,
           1.4076e-02, 1.8982e-02]],

         [[9.0283e-01, 8.9407e-06, 1.4842e-05,  ..., 4.7607e-02,
           8.3771e-03, 9.3689e-03]],

         [[4.3164e-01, 1.1325e-06, 1.2100e-05,  ..., 1.1438e-01,
           3.7659e-02, 1.1002e-02]],

         ...,

         [[4.5288e-01, 3.6144e-04, 2.0924e-03,  ..., 5.6458e-04,
           2.9678e-03, 8.0719e-03]],

         [[5.1465e-01, 2.4915e-04, 1.3351e-03,  ..., 2.7267e-02,
           5.6854e-02, 1.5244e-02]],

         [[4.3481e-01, 1.2934e-04, 3.6740e-04,  ..., 2.9358e-02,
           4.1901e-02, 1.9104e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 5.1856e-06, 9.2983e-06,  ..., 2.9325e-04,
           5.5885e-04, 7.3738e-03]],

         [[1.3110e-01, 1.1921e-07, 0.0000e+00,  ..., 9.8648e-03,
           8.4521e-01, 1.3084e-02]],

         [[9.5117e-01, 1.0216e-04, 2.7835e-05,  ..., 1.1425e-03,
           7.6675e-03, 3.0689e-03]],

         ...,

         [[3.6108e-01, 1.3781e-04, 3.4904e-04,  ..., 1.1131e-02,
           7.2365e-03, 3.5954e-03]],

         [[1.5125e-01, 1.3914e-03, 6.4230e-04,  ..., 3.8986e-03,
           3.9711e-03, 1.6613e-03]],

         [[6.5381e-01, 1.2398e-05, 8.6248e-05,  ..., 6.7322e-02,
           4.9713e-02, 1.3000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3105e-01, 3.4046e-04, 1.8656e-04,  ..., 8.2397e-02,
           3.3630e-02, 2.3087e-02]],

         [[6.2305e-01, 8.3447e-06, 1.2755e-05,  ..., 7.7332e-02,
           1.4771e-01, 2.3834e-02]],

         [[8.2031e-01, 1.1742e-05, 3.1590e-06,  ..., 1.0278e-01,
           4.5471e-03, 8.6288e-03]],

         ...,

         [[2.8882e-01, 4.7684e-07, 2.9802e-06,  ..., 2.9278e-04,
           1.3031e-02, 6.8359e-01]],

         [[3.4814e-01, 4.7684e-07, 3.3402e-04,  ..., 2.9802e-07,
           1.1367e-04, 7.0984e-02]],

         [[8.9844e-01, 8.3447e-07, 2.4319e-05,  ..., 5.3644e-07,
           2.8312e-05, 4.6539e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2876e-01, 1.1683e-05, 4.1127e-06,  ..., 4.3750e-01,
           2.5879e-01, 1.4908e-02]],

         [[9.1406e-01, 7.6890e-06, 3.8743e-06,  ..., 3.9062e-03,
           1.2512e-03, 6.1951e-03]],

         [[6.9922e-01, 4.1795e-04, 2.5058e-04,  ..., 1.1650e-02,
           2.5970e-02, 2.6901e-02]],

         ...,

         [[8.5840e-01, 3.0100e-05, 1.2040e-05,  ..., 7.4707e-02,
           1.4839e-02, 8.0338e-03]],

         [[6.9971e-01, 3.5882e-05, 7.3314e-06,  ..., 2.2324e-02,
           1.9312e-01, 6.7383e-02]],

         [[8.9502e-01, 1.2231e-04, 2.8908e-05,  ..., 4.0512e-03,
           3.1372e-02, 2.6764e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0312e-01, 1.5914e-05, 1.0192e-05,  ..., 7.4707e-02,
           8.4778e-02, 5.0903e-02]],

         [[5.3369e-01, 9.8050e-05, 2.1672e-04,  ..., 4.4586e-02,
           6.3904e-02, 1.3245e-01]],

         [[1.7297e-01, 2.9802e-06, 4.2319e-06,  ..., 6.7578e-01,
           1.1322e-01, 7.5760e-03]],

         ...,

         [[6.6943e-01, 8.3685e-05, 4.6730e-05,  ..., 3.1555e-02,
           1.2543e-02, 1.0178e-02]],

         [[8.6621e-01, 1.7929e-04, 3.8624e-05,  ..., 6.8855e-03,
           4.2076e-03, 9.2459e-04]],

         [[8.0371e-01, 6.9737e-06, 8.4043e-06,  ..., 4.1565e-02,
           7.0557e-02, 1.3153e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.5936e-03, 1.3008e-03,  ..., 2.1954e-03,
           6.0701e-04, 5.5504e-04]],

         [[9.6094e-01, 1.1981e-05, 2.8968e-05,  ..., 9.3651e-04,
           8.8310e-04, 5.4407e-04]],

         [[8.7646e-01, 1.8656e-05, 1.2040e-05,  ..., 2.4223e-03,
           1.5631e-03, 1.9760e-03]],

         ...,

         [[6.0840e-01, 6.5744e-05, 1.0067e-04,  ..., 1.7319e-02,
           1.4600e-01, 8.2520e-02]],

         [[9.2969e-01, 6.5565e-07, 1.7285e-06,  ..., 8.3160e-03,
           2.4292e-02, 9.5749e-03]],

         [[9.5801e-01, 2.0862e-06, 2.1458e-06,  ..., 3.0499e-03,
           1.2657e-02, 1.1452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 4.9353e-04, 1.3733e-03,  ..., 5.1727e-03,
           4.8866e-03, 1.9951e-03]],

         [[5.9717e-01, 2.6345e-05, 3.2544e-05,  ..., 8.2458e-02,
           2.3315e-01, 2.9648e-02]],

         [[9.4775e-01, 8.4639e-06, 2.1815e-05,  ..., 9.5062e-03,
           5.3444e-03, 3.5458e-03]],

         ...,

         [[6.1035e-01, 1.3340e-04, 2.0754e-04,  ..., 1.9360e-03,
           5.2357e-04, 2.0826e-04]],

         [[7.7783e-01, 5.6505e-04, 7.6830e-05,  ..., 2.5955e-02,
           1.1581e-02, 4.9210e-03]],

         [[3.0469e-01, 1.1086e-05, 1.2815e-05,  ..., 7.8659e-03,
           5.2246e-01, 1.0083e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7981e-01, 4.8876e-06, 2.3723e-05,  ..., 2.1410e-04,
           6.1703e-04, 3.1395e-03]],

         [[7.5146e-01, 8.2850e-06, 1.4603e-05,  ..., 4.1473e-02,
           3.2166e-02, 3.7022e-03]],

         [[6.6309e-01, 5.9605e-07, 6.6161e-06,  ..., 3.7136e-03,
           2.0275e-03, 1.9045e-03]],

         ...,

         [[9.5703e-01, 9.3997e-05, 5.4181e-05,  ..., 5.0402e-04,
           6.7282e-04, 7.5817e-05]],

         [[5.3223e-01, 9.4831e-05, 3.7551e-05,  ..., 5.6992e-03,
           4.1565e-02, 3.0151e-01]],

         [[6.4990e-01, 4.0007e-04, 1.3912e-04,  ..., 3.4428e-04,
           1.5640e-03, 6.9809e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5283e-01, 2.3067e-05, 8.9407e-06,  ..., 2.1000e-03,
           1.6266e-02, 8.1348e-01]],

         [[8.5254e-01, 3.5262e-04, 2.6274e-04,  ..., 3.5744e-03,
           1.7557e-03, 7.6008e-04]],

         [[8.2129e-01, 3.8266e-04, 2.7466e-04,  ..., 5.8784e-03,
           7.0229e-03, 1.6394e-03]],

         ...,

         [[5.1904e-01, 5.9903e-05, 2.0742e-04,  ..., 3.3966e-02,
           3.9825e-02, 3.5419e-03]],

         [[1.8726e-01, 4.1723e-06, 5.1260e-06,  ..., 4.7729e-01,
           1.0242e-01, 6.4507e-03]],

         [[1.4392e-01, 2.3842e-05, 3.2842e-05,  ..., 1.1536e-02,
           1.1163e-01, 6.6992e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6709e-01, 2.2972e-04, 2.2793e-04,  ..., 2.6810e-02,
           1.1792e-01, 4.6973e-01]],

         [[6.4355e-01, 5.5611e-05, 5.3465e-05,  ..., 2.2064e-02,
           4.7638e-02, 3.7060e-03]],

         [[4.6460e-01, 1.8823e-04, 3.7718e-04,  ..., 4.1290e-02,
           2.6489e-02, 1.4801e-03]],

         ...,

         [[1.3786e-02, 8.3447e-07, 2.2650e-06,  ..., 7.3910e-06,
           5.2528e-03, 9.7803e-01]],

         [[7.7295e-01, 1.5545e-04, 1.9121e-04,  ..., 5.5618e-03,
           4.3144e-03, 3.4912e-02]],

         [[7.8711e-01, 1.4138e-04, 4.8637e-05,  ..., 1.9867e-02,
           2.0798e-02, 7.2098e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9624e-01, 4.5586e-04, 1.0376e-03,  ..., 6.6261e-03,
           3.0899e-03, 1.7338e-03]],

         [[8.6035e-01, 1.7226e-05, 5.0068e-05,  ..., 4.0169e-03,
           3.2776e-02, 7.9575e-03]],

         [[3.6646e-01, 3.0756e-05, 9.1195e-05,  ..., 7.6628e-04,
           2.4490e-03, 5.2681e-03]],

         ...,

         [[3.8989e-01, 3.5763e-07, 0.0000e+00,  ..., 6.8542e-02,
           5.2393e-01, 1.4458e-02]],

         [[5.4492e-01, 8.4209e-04, 9.8038e-04,  ..., 1.3222e-02,
           4.8370e-03, 6.3801e-04]],

         [[6.3770e-01, 2.9206e-06, 1.3709e-06,  ..., 2.9028e-01,
           3.6926e-02, 4.2267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9648e-01, 8.1241e-05, 2.4498e-05,  ..., 5.1193e-03,
           2.2373e-03, 4.7588e-04]],

         [[1.3904e-01, 3.0684e-04, 2.6536e-04,  ..., 1.1177e-03,
           1.4839e-03, 4.8780e-04]],

         [[7.7832e-01, 9.7573e-05, 4.3988e-05,  ..., 3.2067e-04,
           5.9891e-04, 3.2878e-04]],

         ...,

         [[6.5527e-01, 2.3842e-07, 1.7881e-07,  ..., 7.6172e-02,
           9.0332e-02, 1.2856e-02]],

         [[8.1836e-01, 5.9605e-08, 5.9605e-08,  ..., 7.3059e-02,
           5.3467e-02, 1.4977e-02]],

         [[5.0293e-01, 3.2783e-06, 6.8545e-06,  ..., 4.0771e-02,
           3.2745e-02, 8.5144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8167e-02, 1.0133e-06, 1.7285e-06,  ..., 1.0004e-03,
           1.0887e-02, 8.9941e-01]],

         [[1.4319e-01, 3.8743e-06, 9.1195e-06,  ..., 1.7200e-01,
           2.2791e-01, 3.7628e-02]],

         [[4.2773e-01, 1.8287e-04, 2.0754e-04,  ..., 1.2413e-02,
           7.1068e-03, 2.0294e-03]],

         ...,

         [[4.3359e-01, 1.0377e-04, 1.6844e-04,  ..., 2.8229e-02,
           1.0490e-02, 5.6686e-03]],

         [[4.9854e-01, 3.7551e-06, 2.1815e-05,  ..., 1.2039e-02,
           1.2688e-02, 1.1185e-02]],

         [[1.8750e-01, 2.0826e-04, 2.4915e-04,  ..., 2.3098e-03,
           3.8834e-03, 1.6222e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2539e-01, 2.7120e-05, 5.5492e-05,  ..., 1.7838e-02,
           1.0323e-02, 6.0921e-03]],

         [[3.5059e-01, 3.1900e-04, 2.3162e-04,  ..., 8.1360e-02,
           3.2745e-02, 1.3329e-02]],

         [[1.6553e-01, 3.1471e-05, 3.8981e-05,  ..., 2.5732e-01,
           3.0029e-02, 8.6899e-03]],

         ...,

         [[7.7637e-01, 5.4836e-06, 7.9870e-06,  ..., 1.4275e-02,
           7.1487e-03, 4.8828e-04]],

         [[6.4600e-01, 5.2273e-05, 7.6115e-05,  ..., 1.1925e-02,
           2.2903e-02, 1.5764e-03]],

         [[8.8477e-01, 4.3154e-05, 5.1439e-05,  ..., 8.4381e-03,
           1.0424e-03, 6.3705e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8979e-01, 2.7294e-03, 1.2264e-03,  ..., 7.8058e-04,
           1.2732e-03, 5.8603e-04]],

         [[2.9956e-01, 1.0729e-03, 7.4434e-04,  ..., 6.5575e-03,
           2.8381e-03, 4.1924e-03]],

         [[3.7695e-01, 2.0146e-05, 2.0504e-05,  ..., 2.0984e-01,
           1.3806e-01, 1.5022e-02]],

         ...,

         [[3.5156e-01, 2.1899e-04, 1.3340e-04,  ..., 4.6577e-03,
           1.4896e-03, 2.2171e-02]],

         [[9.5020e-01, 6.2585e-05, 2.7823e-04,  ..., 1.7881e-07,
           2.7657e-05, 2.3937e-03]],

         [[8.2422e-01, 5.6744e-05, 6.6936e-05,  ..., 1.5427e-02,
           4.4785e-03, 1.5640e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9990e-01, 4.3988e-04, 8.0919e-04,  ..., 1.1978e-02,
           2.9419e-02, 7.6752e-03]],

         [[4.5386e-01, 1.9894e-03, 1.4496e-03,  ..., 1.2474e-02,
           5.1994e-03, 2.0695e-03]],

         [[1.5442e-01, 2.5451e-05, 2.1219e-05,  ..., 1.0262e-02,
           3.5620e-01, 2.4475e-01]],

         ...,

         [[1.9727e-01, 1.0550e-05, 3.1114e-05,  ..., 2.3560e-02,
           1.1528e-02, 5.4979e-04]],

         [[4.2676e-01, 3.7968e-05, 4.3094e-05,  ..., 1.1395e-01,
           7.2998e-02, 2.3788e-02]],

         [[4.2920e-01, 1.1480e-04, 1.2082e-04,  ..., 4.5959e-02,
           1.0480e-01, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4131e-01, 1.5831e-04, 2.5702e-04,  ..., 6.7711e-04,
           2.5558e-03, 1.5900e-02]],

         [[1.1365e-01, 9.0599e-06, 1.0729e-05,  ..., 2.4426e-01,
           7.4524e-02, 7.8506e-03]],

         [[2.6465e-01, 5.8174e-05, 1.9789e-05,  ..., 2.3975e-03,
           1.0522e-01, 5.8691e-01]],

         ...,

         [[7.9041e-02, 3.1066e-04, 7.8726e-04,  ..., 6.7329e-04,
           3.4199e-03, 1.0048e-02]],

         [[3.3984e-01, 4.1699e-04, 5.1880e-04,  ..., 4.2610e-03,
           2.7039e-02, 2.7679e-02]],

         [[2.1509e-01, 1.9817e-03, 2.0046e-03,  ..., 3.8033e-03,
           2.8267e-03, 3.8490e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6655e-01, 4.4823e-05, 5.1498e-05,  ..., 6.7749e-02,
           3.6530e-02, 8.8120e-03]],

         [[2.1152e-03, 2.9802e-07, 5.9605e-08,  ..., 1.5199e-04,
           1.0610e-05, 3.7708e-03]],

         [[8.1543e-01, 1.5855e-05, 3.1590e-06,  ..., 2.8496e-03,
           1.3971e-04, 4.2033e-04]],

         ...,

         [[9.1003e-02, 8.4496e-04, 1.6031e-03,  ..., 1.5327e-02,
           1.6830e-02, 2.7275e-03]],

         [[5.2588e-01, 2.5809e-05, 4.9770e-05,  ..., 1.8164e-01,
           2.0065e-02, 8.9035e-03]],

         [[4.8169e-01, 6.0034e-04, 1.3638e-03,  ..., 9.8419e-03,
           1.6876e-02, 1.4973e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6265e-01, 5.0008e-05, 1.8728e-04,  ..., 3.4576e-02,
           1.8799e-02, 6.6650e-02]],

         [[8.6365e-02, 4.6670e-05, 1.3828e-04,  ..., 7.4310e-03,
           3.6011e-02, 2.5543e-02]],

         [[2.2742e-01, 3.7193e-05, 1.1367e-04,  ..., 3.0003e-03,
           1.9882e-02, 2.8687e-02]],

         ...,

         [[1.2988e-01, 7.0751e-05, 7.5281e-05,  ..., 6.6719e-03,
           3.3875e-02, 3.9612e-02]],

         [[7.8430e-02, 7.8559e-05, 6.5148e-05,  ..., 4.0932e-03,
           1.1841e-02, 2.2476e-02]],

         [[4.4678e-01, 4.5359e-05, 8.2731e-05,  ..., 2.0889e-02,
           2.3666e-02, 1.3916e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6150e-01, 1.2398e-05, 1.1802e-05,  ..., 2.4368e-02,
           7.2205e-02, 6.4697e-02]],

         [[4.7363e-01, 2.3031e-04, 4.3201e-04,  ..., 7.2479e-03,
           1.4992e-02, 2.1469e-02]],

         [[1.9617e-01, 2.7061e-04, 6.8045e-04,  ..., 4.9095e-03,
           2.5977e-01, 3.0151e-01]],

         ...,

         [[6.0352e-01, 6.1750e-05, 2.0885e-04,  ..., 7.9575e-03,
           7.5562e-02, 5.3131e-02]],

         [[1.9617e-01, 5.0247e-05, 2.5415e-04,  ..., 1.3962e-03,
           8.4839e-03, 2.9846e-02]],

         [[1.7834e-01, 3.6418e-05, 1.4019e-04,  ..., 3.9625e-04,
           6.5956e-03, 1.7654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6675e-01, 7.0035e-05, 1.2934e-04,  ..., 5.4565e-02,
           7.1167e-02, 7.9407e-02]],

         [[1.5686e-01, 2.2113e-05, 2.0862e-05,  ..., 1.2192e-02,
           6.7329e-03, 1.3077e-02]],

         [[6.2061e-01, 1.1545e-04, 1.9491e-04,  ..., 2.0111e-02,
           1.3290e-02, 8.6823e-03]],

         ...,

         [[6.4893e-01, 5.3644e-07, 1.1921e-07,  ..., 1.4490e-01,
           1.9495e-01, 3.4065e-03]],

         [[3.9575e-01, 6.4373e-06, 2.8610e-05,  ..., 1.7285e-01,
           9.1064e-02, 2.9736e-03]],

         [[9.6191e-02, 1.7881e-06, 8.3447e-07,  ..., 4.8370e-02,
           8.4424e-01, 5.1384e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4209e-01, 7.4089e-05, 1.8847e-04,  ..., 7.3120e-02,
           1.9836e-02, 3.7270e-03]],

         [[7.9834e-02, 9.9540e-06, 8.4043e-06,  ..., 1.1414e-02,
           9.2590e-02, 6.8408e-01]],

         [[3.0249e-01, 9.9945e-04, 2.6073e-03,  ..., 1.2655e-03,
           7.6027e-03, 1.0576e-03]],

         ...,

         [[2.7197e-01, 2.6298e-04, 5.4169e-04,  ..., 2.5192e-02,
           9.8267e-03, 2.4509e-03]],

         [[8.4839e-02, 4.1723e-07, 1.3709e-06,  ..., 6.5723e-01,
           2.1667e-01, 1.7303e-02]],

         [[1.5112e-01, 2.9802e-07, 1.0133e-06,  ..., 4.0112e-01,
           3.5669e-01, 1.7349e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 1.4782e-04, 4.5228e-04,  ..., 3.7193e-03,
           4.2496e-03, 5.7936e-04]],

         [[9.7998e-01, 1.5795e-05, 6.3956e-05,  ..., 3.3855e-05,
           3.3557e-05, 1.0067e-04]],

         [[7.8271e-01, 5.3525e-05, 1.4734e-04,  ..., 6.1989e-04,
           3.2163e-04, 5.8270e-04]],

         ...,

         [[8.4863e-01, 2.0325e-05, 2.5272e-05,  ..., 2.4460e-02,
           1.9043e-02, 1.4603e-02]],

         [[5.7471e-01, 8.2850e-06, 8.5235e-06,  ..., 3.6106e-03,
           5.7678e-03, 1.6434e-02]],

         [[1.2408e-01, 3.0220e-05, 9.0182e-05,  ..., 5.3673e-03,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6111e-02, 9.5367e-06, 9.0003e-06,  ..., 1.1432e-01,
           9.5520e-02, 3.0460e-03]],

         [[7.6660e-01, 2.3305e-05, 2.2233e-05,  ..., 3.1647e-02,
           1.3138e-02, 1.4046e-02]],

         [[1.3977e-01, 4.6790e-05, 7.8797e-05,  ..., 1.4026e-01,
           4.3304e-02, 4.2114e-02]],

         ...,

         [[6.7578e-01, 1.1563e-05, 3.9995e-05,  ..., 4.7089e-02,
           9.1492e-02, 2.5208e-02]],

         [[6.2744e-01, 1.3888e-04, 8.9979e-04,  ..., 1.6205e-02,
           8.2397e-03, 2.5894e-02]],

         [[6.0254e-01, 1.1444e-05, 6.1989e-05,  ..., 6.4354e-03,
           2.9297e-02, 5.9631e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6724e-01, 5.6982e-05, 6.8486e-05,  ..., 3.4790e-02,
           5.2528e-03, 1.7365e-02]],

         [[2.0459e-01, 6.1870e-05, 5.7459e-05,  ..., 1.6922e-02,
           1.3542e-02, 1.3916e-02]],

         [[7.8857e-02, 1.0610e-04, 2.0134e-04,  ..., 5.7251e-02,
           7.0129e-02, 7.0129e-02]],

         ...,

         [[3.7048e-02, 6.9141e-06, 1.0169e-04,  ..., 5.6381e-03,
           1.0956e-02, 1.2102e-03]],

         [[1.0004e-01, 7.8678e-06, 5.6624e-05,  ..., 1.1223e-02,
           2.8671e-02, 1.2520e-02]],

         [[9.0088e-01, 1.6391e-05, 2.0087e-05,  ..., 1.3054e-02,
           4.9255e-02, 3.7403e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8257e-01, 2.4438e-06, 1.1027e-05,  ..., 2.1289e-01,
           1.7786e-01, 1.4868e-01]],

         [[3.9233e-01, 2.8014e-05, 3.2306e-05,  ..., 3.8391e-02,
           3.6377e-02, 2.9053e-01]],

         [[5.5122e-03, 1.7881e-07, 1.1921e-07,  ..., 4.7339e-01,
           4.1455e-01, 3.8552e-04]],

         ...,

         [[1.6272e-01, 1.2970e-04, 3.1042e-04,  ..., 1.6357e-02,
           3.5461e-02, 6.5247e-02]],

         [[3.4766e-01, 9.4235e-05, 4.9055e-05,  ..., 4.2175e-02,
           1.6553e-01, 1.8417e-02]],

         [[2.0691e-02, 2.6226e-05, 6.5207e-05,  ..., 3.2043e-02,
           2.3270e-02, 7.4707e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9370e-01, 3.2187e-06, 5.2333e-05,  ..., 9.9121e-02,
           1.6211e-01, 3.0054e-01]],

         [[5.8984e-01, 3.1757e-04, 8.6164e-04,  ..., 3.4943e-03,
           9.9564e-03, 2.6642e-02]],

         [[8.7952e-02, 7.5698e-06, 4.1187e-05,  ..., 4.7722e-03,
           2.8564e-02, 3.2349e-02]],

         ...,

         [[6.8555e-01, 7.1228e-05, 7.7009e-05,  ..., 8.8562e-02,
           7.2266e-02, 2.4200e-02]],

         [[9.6313e-02, 6.3241e-05, 1.6415e-04,  ..., 9.8419e-03,
           1.5732e-02, 8.4521e-01]],

         [[1.3477e-01, 2.4557e-05, 7.6354e-05,  ..., 6.6071e-03,
           7.8809e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8027e-01, 4.5240e-05, 4.3774e-04,  ..., 3.6499e-02,
           4.8920e-02, 6.1829e-02]],

         [[7.3059e-02, 1.7703e-05, 3.8922e-05,  ..., 2.9068e-02,
           7.9346e-02, 6.5491e-02]],

         [[5.3833e-02, 1.2481e-04, 5.8603e-04,  ..., 1.2085e-01,
           2.7344e-01, 3.0441e-02]],

         ...,

         [[8.3887e-01, 1.3196e-04, 1.3459e-04,  ..., 7.1182e-03,
           1.6006e-02, 4.0970e-03]],

         [[7.5391e-01, 1.3709e-06, 7.3814e-04,  ..., 4.2343e-03,
           1.5278e-03, 5.2917e-02]],

         [[5.7959e-01, 8.3685e-05, 8.0645e-05,  ..., 3.6743e-02,
           7.2510e-02, 1.0388e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1192e-04, 4.9019e-04, 1.4417e-01,  ..., 1.6651e-03,
           9.0714e-03, 3.3295e-02]],

         [[1.0842e-04, 9.5189e-05, 1.3232e-01,  ..., 9.9915e-02,
           3.6392e-03, 1.9669e-02]],

         [[2.2089e-04, 1.9765e-04, 4.7974e-02,  ..., 3.9917e-02,
           8.9600e-02, 1.0522e-01]],

         ...,

         [[3.8505e-05, 1.7285e-05, 2.9587e-02,  ..., 2.9199e-01,
           3.1079e-01, 6.9336e-02]],

         [[8.5020e-04, 9.6369e-04, 1.7456e-01,  ..., 1.5976e-02,
           3.0334e-02, 1.7548e-02]],

         [[3.5763e-06, 4.7684e-07, 1.0300e-02,  ..., 6.6757e-06,
           8.9109e-05, 8.0633e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7680e-04, 5.1618e-05, 2.9663e-01,  ..., 7.3814e-04,
           9.7752e-04, 3.4882e-02]],

         [[1.1408e-04, 6.8426e-05, 2.4597e-01,  ..., 4.6310e-03,
           2.7800e-04, 2.2522e-02]],

         [[2.0087e-05, 6.8545e-06, 2.5366e-01,  ..., 1.7643e-05,
           6.3658e-04, 2.2171e-02]],

         ...,

         [[9.4891e-04, 2.0885e-04, 2.7832e-01,  ..., 4.5700e-03,
           3.6359e-04, 6.1989e-03]],

         [[3.9637e-05, 4.5478e-05, 3.1079e-01,  ..., 2.0099e-04,
           3.3259e-04, 3.1494e-02]],

         [[6.4492e-05, 4.4942e-05, 2.2302e-01,  ..., 6.3133e-04,
           1.0948e-03, 8.9188e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7716e-05, 3.2783e-05, 2.1777e-01,  ..., 2.2829e-05,
           4.8327e-04, 3.0640e-02]],

         [[7.0989e-05, 3.8147e-06, 2.6538e-01,  ..., 1.1307e-02,
           3.6073e-04, 2.2491e-02]],

         [[6.3598e-05, 1.5819e-04, 2.6294e-01,  ..., 1.6451e-03,
           7.8344e-04, 1.8616e-02]],

         ...,

         [[1.8597e-05, 3.9697e-05, 1.6895e-01,  ..., 6.6650e-02,
           1.1627e-02, 1.3257e-01]],

         [[1.7285e-05, 4.5896e-05, 1.6028e-01,  ..., 2.1103e-02,
           9.1314e-04, 1.5991e-02]],

         [[1.1545e-04, 4.3094e-05, 1.6296e-01,  ..., 3.6411e-03,
           4.2486e-04, 9.1171e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3631e-05, 2.6107e-05, 4.6844e-03,  ..., 1.2372e-01,
           6.5332e-01, 4.6600e-02]],

         [[5.5850e-05, 2.7919e-04, 1.2585e-01,  ..., 9.1370e-02,
           5.2277e-02, 9.4681e-03]],

         [[8.8334e-05, 6.6614e-04, 9.3506e-02,  ..., 5.9175e-04,
           1.8060e-04, 3.0613e-04]],

         ...,

         [[3.0861e-03, 5.7411e-04, 2.7084e-02,  ..., 4.3602e-03,
           5.0087e-03, 6.6223e-03]],

         [[9.1124e-04, 3.0470e-04, 1.6541e-02,  ..., 6.6956e-02,
           3.0249e-01, 2.7481e-02]],

         [[2.3711e-04, 6.1154e-05, 4.1534e-02,  ..., 3.3508e-02,
           1.0056e-02, 3.4973e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0055, 0.0010, 0.1638,  ..., 0.1348, 0.0202, 0.0223]],

         [[0.0004, 0.0008, 0.2350,  ..., 0.0208, 0.0168, 0.1542]],

         [[0.0034, 0.0011, 0.1915,  ..., 0.0368, 0.0211, 0.0542]],

         ...,

         [[0.0019, 0.0007, 0.0858,  ..., 0.2012, 0.0757, 0.1299]],

         [[0.0008, 0.0014, 0.1866,  ..., 0.0077, 0.0101, 0.2759]],

         [[0.0003, 0.0008, 0.1636,  ..., 0.0365, 0.0151, 0.0424]]]],
       device='cuda:0', dtype=torch.float16)),(tensor([[[[5.7888e-04, 1.2910e-04, 7.9107e-04,  ..., 7.1924e-01,
           1.5552e-01, 6.6895e-02]],

         [[7.1335e-04, 2.9731e-04, 4.6539e-03,  ..., 3.0334e-02,
           3.9355e-01, 1.7288e-02]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.3709e-06,
           2.0337e-04, 9.9951e-01]],

         ...,

         [[1.6272e-05, 2.6822e-05, 5.3835e-04,  ..., 3.9787e-03,
           6.5575e-03, 9.7314e-01]],

         [[6.8817e-03, 6.4659e-03, 5.3596e-03,  ..., 2.7222e-02,
           4.7791e-02, 8.3008e-03]],

         [[1.5556e-02, 3.3646e-03, 9.3412e-04,  ..., 1.8188e-02,
           1.2115e-02, 4.0985e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5736e-03, 2.2202e-03, 9.9487e-03,  ..., 4.3884e-02,
           1.6821e-01, 2.6050e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 1.5369e-01,
           7.6855e-01, 7.3792e-02]],

         [[8.1587e-04, 1.8673e-03, 7.1602e-03,  ..., 2.4994e-02,
           2.4216e-02, 1.1436e-02]],

         ...,

         [[1.7233e-03, 2.1610e-03, 8.0948e-03,  ..., 1.9882e-02,
           7.0496e-02, 1.0663e-01]],

         [[4.7569e-03, 1.2535e-02, 3.8177e-02,  ..., 1.3290e-02,
           1.7410e-02, 1.2932e-02]],

         [[1.9894e-03, 3.4103e-03, 7.1106e-03,  ..., 2.1106e-01,
           2.5208e-02, 6.0944e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7148e-02, 7.7286e-03, 4.2938e-02,  ..., 5.2216e-02,
           5.0201e-02, 4.2938e-02]],

         [[2.8381e-03, 1.0900e-03, 4.2114e-03,  ..., 7.3624e-03,
           2.3708e-03, 1.0216e-02]],

         [[7.5951e-03, 2.0599e-03, 2.0676e-03,  ..., 3.2471e-02,
           3.6530e-02, 2.3300e-02]],

         ...,

         [[3.8452e-03, 5.0449e-04, 1.7881e-03,  ..., 1.7542e-01,
           2.3059e-01, 1.9263e-01]],

         [[3.3493e-03, 1.7548e-04, 5.8222e-04,  ..., 7.6843e-02,
           3.5187e-02, 1.0101e-01]],

         [[4.4975e-03, 1.0147e-03, 2.1744e-03,  ..., 4.8828e-01,
           1.6614e-01, 7.7881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8418e-01, 3.2187e-06, 4.0531e-06,  ..., 1.9054e-03,
           2.4185e-03, 1.2573e-02]],

         [[9.4336e-01, 4.3511e-06, 9.1195e-06,  ..., 2.8648e-03,
           2.3636e-02, 5.1155e-03]],

         [[6.9189e-01, 5.1856e-06, 2.0862e-05,  ..., 4.4403e-03,
           1.6460e-03, 2.1606e-02]],

         ...,

         [[8.4351e-02, 4.4441e-04, 6.4659e-04,  ..., 2.6011e-04,
           7.3862e-04, 8.9264e-03]],

         [[5.4199e-01, 1.5676e-04, 1.1539e-03,  ..., 2.0538e-02,
           1.7563e-02, 1.1841e-02]],

         [[3.4692e-01, 2.2840e-04, 4.8971e-04,  ..., 6.2439e-02,
           2.9251e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1226e-01, 2.9206e-06, 3.5167e-06,  ..., 4.4346e-05,
           3.9160e-05, 6.6986e-03]],

         [[3.5767e-01, 5.0664e-06, 5.9605e-08,  ..., 3.8965e-01,
           2.3083e-01, 1.8829e-02]],

         [[8.9453e-01, 2.7323e-04, 1.7524e-05,  ..., 3.5126e-02,
           1.8188e-02, 4.8447e-03]],

         ...,

         [[3.5596e-01, 1.1045e-04, 2.2840e-04,  ..., 8.4457e-03,
           7.8506e-03, 2.2110e-02]],

         [[1.4429e-01, 1.9522e-03, 1.6851e-03,  ..., 3.0327e-03,
           1.3723e-03, 6.9885e-03]],

         [[1.9751e-01, 2.5868e-05, 7.9453e-05,  ..., 2.6587e-01,
           8.4656e-02, 4.9164e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 2.8729e-04, 1.3578e-04,  ..., 4.1870e-02,
           2.1301e-02, 2.9984e-02]],

         [[6.3330e-01, 1.1504e-05, 1.1444e-05,  ..., 1.2286e-01,
           9.5520e-03, 1.5078e-03]],

         [[8.6963e-01, 2.4259e-05, 5.6028e-06,  ..., 6.2714e-03,
           3.7903e-02, 2.3804e-03]],

         ...,

         [[4.2822e-01, 4.7684e-07, 1.7881e-06,  ..., 6.5279e-04,
           8.2350e-04, 5.4443e-01]],

         [[4.6582e-01, 1.3113e-06, 5.2750e-05,  ..., 2.5034e-06,
           2.3961e-05, 1.8799e-02]],

         [[9.3311e-01, 1.0729e-06, 8.4043e-06,  ..., 2.9802e-07,
           3.1590e-06, 1.9073e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4595e-01, 8.8274e-05, 1.2994e-05,  ..., 1.6260e-01,
           1.2830e-01, 2.5116e-02]],

         [[8.9746e-01, 4.4107e-06, 1.8477e-06,  ..., 5.4061e-05,
           1.6749e-05, 5.7907e-03]],

         [[7.8125e-01, 1.7500e-04, 5.4836e-05,  ..., 5.7487e-03,
           3.8509e-03, 3.3752e-02]],

         ...,

         [[9.0234e-01, 2.7227e-04, 1.8597e-05,  ..., 9.6359e-03,
           7.9117e-03, 1.2451e-02]],

         [[7.6807e-01, 1.2118e-04, 3.9518e-05,  ..., 6.7627e-02,
           4.4098e-02, 6.1615e-02]],

         [[8.8818e-01, 6.1989e-05, 3.4094e-05,  ..., 4.1565e-02,
           2.5497e-02, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7588e-01, 5.4955e-05, 3.1590e-05,  ..., 2.3239e-02,
           1.4923e-02, 5.0537e-02]],

         [[5.5713e-01, 9.4354e-05, 1.3733e-04,  ..., 1.0559e-01,
           2.2095e-02, 8.3557e-02]],

         [[8.3435e-02, 2.1458e-06, 3.5167e-06,  ..., 3.5254e-01,
           6.7749e-03, 2.3174e-03]],

         ...,

         [[5.8740e-01, 4.8995e-05, 1.7524e-05,  ..., 2.2003e-02,
           1.7975e-02, 2.8137e-02]],

         [[8.7939e-01, 2.9278e-04, 7.4267e-05,  ..., 2.7885e-03,
           1.0662e-03, 7.1602e-03]],

         [[3.9331e-01, 1.0669e-05, 9.0599e-06,  ..., 1.5881e-01,
           1.6052e-01, 6.4148e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 3.4161e-03, 1.2865e-03,  ..., 8.2731e-04,
           5.9128e-04, 8.2779e-03]],

         [[9.6045e-01, 9.9540e-06, 3.5346e-05,  ..., 1.7424e-03,
           5.5695e-04, 4.7827e-04]],

         [[8.7012e-01, 5.7757e-05, 5.2989e-05,  ..., 2.1324e-03,
           1.3142e-03, 1.1482e-02]],

         ...,

         [[5.7422e-01, 1.4770e-04, 3.4070e-04,  ..., 1.8265e-02,
           3.2501e-02, 1.6870e-01]],

         [[8.4570e-01, 4.4107e-06, 1.8418e-05,  ..., 4.9713e-02,
           1.3885e-02, 6.1684e-03]],

         [[8.7793e-01, 8.7619e-06, 1.3828e-05,  ..., 1.3329e-02,
           1.5434e-02, 6.6101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1899e-01, 3.8242e-04, 8.1873e-04,  ..., 2.1152e-03,
           1.4257e-03, 1.2505e-02]],

         [[5.1465e-01, 2.8491e-05, 2.4974e-05,  ..., 2.3413e-01,
           7.1045e-02, 3.0945e-02]],

         [[8.8086e-01, 2.2590e-05, 2.4021e-05,  ..., 4.0802e-02,
           1.5373e-02, 9.5444e-03]],

         ...,

         [[7.5000e-01, 9.8288e-05, 1.3304e-04,  ..., 2.6560e-04,
           8.3923e-05, 4.2033e-04]],

         [[7.8027e-01, 1.6356e-04, 4.5478e-05,  ..., 3.3630e-02,
           1.7685e-02, 6.6101e-02]],

         [[5.5518e-01, 1.2934e-05, 1.4484e-05,  ..., 1.9116e-01,
           1.2390e-01, 1.9638e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3416e-02, 4.1723e-06, 1.6153e-05,  ..., 7.8678e-06,
           9.4175e-06, 2.5673e-03]],

         [[4.8779e-01, 1.1265e-05, 2.0266e-05,  ..., 6.1035e-02,
           1.1276e-02, 4.8584e-02]],

         [[8.3887e-01, 8.3447e-07, 2.2054e-06,  ..., 2.0754e-04,
           5.9664e-05, 1.8644e-03]],

         ...,

         [[9.4238e-01, 5.8711e-05, 5.6267e-05,  ..., 2.9526e-03,
           2.8110e-04, 1.7567e-03]],

         [[7.3340e-01, 1.6546e-04, 6.3539e-05,  ..., 3.3951e-03,
           9.0820e-02, 1.3596e-02]],

         [[6.2939e-01, 7.2289e-04, 1.3268e-04,  ..., 7.9679e-04,
           2.1112e-04, 7.9651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1431e-01, 1.5199e-05, 1.6689e-05,  ..., 3.0575e-03,
           5.7373e-03, 5.4883e-01]],

         [[5.0732e-01, 8.8549e-04, 8.9073e-04,  ..., 3.2043e-03,
           2.8849e-04, 1.7487e-02]],

         [[6.0840e-01, 3.2377e-04, 3.1090e-04,  ..., 3.6373e-03,
           7.8058e-04, 7.8354e-03]],

         ...,

         [[3.2642e-01, 9.2506e-05, 3.0684e-04,  ..., 2.1713e-02,
           1.9932e-03, 6.7291e-03]],

         [[4.0796e-01, 3.5703e-05, 4.3213e-05,  ..., 1.2262e-01,
           9.5825e-03, 3.2532e-02]],

         [[3.3960e-01, 4.5598e-05, 1.1283e-04,  ..., 1.0315e-02,
           3.1860e-02, 5.4395e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4717e-01, 1.6046e-04, 2.0921e-04,  ..., 1.6510e-02,
           5.7678e-02, 5.0000e-01]],

         [[6.9141e-01, 8.1420e-05, 1.1623e-04,  ..., 1.8753e-02,
           3.8185e-03, 1.5762e-02]],

         [[4.1602e-01, 3.4261e-04, 6.1798e-04,  ..., 6.1684e-03,
           9.7227e-04, 9.3002e-03]],

         ...,

         [[9.3323e-02, 3.6895e-05, 8.4937e-05,  ..., 4.1628e-04,
           8.7967e-03, 8.4473e-01]],

         [[7.4512e-01, 3.2282e-04, 2.7823e-04,  ..., 2.7294e-03,
           1.4511e-02, 2.1484e-02]],

         [[7.3633e-01, 1.0198e-04, 3.8981e-05,  ..., 2.7924e-02,
           7.8821e-04, 1.1475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.3324e-04, 1.1854e-03,  ..., 1.4137e-02,
           6.4964e-03, 8.0505e-02]],

         [[6.9385e-01, 1.9014e-05, 5.2571e-05,  ..., 3.4058e-02,
           1.3351e-02, 1.1432e-01]],

         [[1.8774e-01, 5.1260e-06, 1.6391e-05,  ..., 1.8013e-04,
           7.7963e-04, 1.0307e-02]],

         ...,

         [[8.1836e-01, 1.3113e-06, 6.5565e-07,  ..., 7.7454e-02,
           5.6366e-02, 2.3499e-02]],

         [[5.4785e-01, 7.2813e-04, 1.2035e-03,  ..., 3.8986e-03,
           7.8440e-04, 1.4259e-02]],

         [[6.3330e-01, 2.3842e-06, 2.5630e-06,  ..., 5.5847e-02,
           2.1255e-02, 9.4238e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5010e-01, 5.4598e-05, 1.9848e-05,  ..., 4.7607e-03,
           6.2275e-04, 3.6182e-03]],

         [[1.4880e-01, 3.7384e-04, 3.1281e-04,  ..., 6.0892e-04,
           5.1022e-04, 1.7471e-03]],

         [[6.4746e-01, 2.2197e-04, 1.3304e-04,  ..., 1.6365e-03,
           5.6458e-04, 1.2779e-02]],

         ...,

         [[5.2783e-01, 4.1723e-07, 4.1723e-07,  ..., 3.1158e-02,
           3.7460e-03, 1.5526e-02]],

         [[3.4595e-01, 5.9605e-08, 5.9605e-08,  ..., 1.6284e-01,
           6.6589e-02, 8.2153e-02]],

         [[3.0811e-01, 3.3975e-06, 8.5235e-06,  ..., 5.3253e-03,
           2.0695e-03, 1.5823e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6721e-02, 2.3842e-07, 6.5565e-07,  ..., 7.2908e-04,
           2.6393e-04, 8.9551e-01]],

         [[1.0455e-01, 8.4043e-06, 2.5332e-05,  ..., 1.6968e-01,
           6.3538e-02, 6.3782e-02]],

         [[2.9590e-01, 2.3162e-04, 4.9400e-04,  ..., 9.9182e-03,
           1.7424e-03, 3.2501e-02]],

         ...,

         [[4.3457e-01, 7.1347e-05, 1.3804e-04,  ..., 2.3911e-02,
           3.3150e-03, 2.5391e-02]],

         [[7.2461e-01, 1.0133e-06, 1.2875e-05,  ..., 7.5264e-03,
           2.2335e-03, 6.8169e-03]],

         [[3.9941e-01, 1.2612e-04, 1.1754e-04,  ..., 1.0696e-02,
           1.7891e-03, 1.0017e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8643e-01, 2.4557e-05, 4.0352e-05,  ..., 1.7914e-02,
           8.5449e-03, 1.5002e-01]],

         [[4.7803e-01, 3.9887e-04, 5.7888e-04,  ..., 1.0941e-02,
           4.5166e-03, 3.8483e-02]],

         [[5.2185e-02, 7.7486e-06, 1.1921e-05,  ..., 6.9153e-02,
           6.9275e-03, 4.0344e-02]],

         ...,

         [[6.8213e-01, 2.4557e-05, 2.0325e-05,  ..., 2.6978e-02,
           2.7561e-03, 8.3740e-02]],

         [[5.6006e-01, 1.2994e-05, 2.6584e-05,  ..., 1.0559e-01,
           3.4351e-03, 1.1688e-01]],

         [[8.2520e-01, 4.6015e-05, 4.9412e-05,  ..., 1.4467e-03,
           4.6134e-04, 9.8190e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8882e-01, 1.7090e-03, 7.8917e-04,  ..., 4.2963e-04,
           2.1935e-04, 4.8294e-03]],

         [[3.7598e-01, 6.1226e-04, 4.5180e-04,  ..., 1.6222e-03,
           2.1420e-03, 7.8735e-03]],

         [[3.5400e-01, 2.2471e-05, 3.2306e-05,  ..., 1.6333e-01,
           1.3374e-02, 4.1931e-02]],

         ...,

         [[3.7280e-01, 2.3663e-04, 9.1434e-05,  ..., 7.8430e-03,
           7.8583e-03, 4.4006e-02]],

         [[8.3203e-01, 6.4194e-05, 6.0987e-04,  ..., 3.2783e-06,
           3.5214e-04, 5.8861e-03]],

         [[8.4570e-01, 1.5640e-04, 1.3089e-04,  ..., 2.3880e-03,
           7.0143e-04, 2.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3789e-01, 2.4557e-04, 3.5739e-04,  ..., 8.5449e-02,
           1.1925e-02, 3.9734e-02]],

         [[2.6709e-01, 1.3475e-03, 7.4148e-04,  ..., 9.0637e-03,
           2.5177e-03, 1.1024e-02]],

         [[9.7778e-02, 1.2815e-05, 1.2279e-05,  ..., 4.3030e-02,
           3.0289e-02, 4.3823e-01]],

         ...,

         [[3.6548e-01, 2.4796e-05, 5.2035e-05,  ..., 1.0612e-02,
           8.6689e-04, 1.3214e-02]],

         [[5.3320e-01, 8.7976e-05, 8.1480e-05,  ..., 2.3621e-02,
           1.0857e-02, 5.7770e-02]],

         [[2.8101e-01, 3.9279e-05, 3.3855e-05,  ..., 6.4148e-02,
           3.8025e-02, 2.6611e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3340e-01, 3.1292e-05, 3.2783e-05,  ..., 7.1907e-03,
           1.8646e-02, 8.0185e-03]],

         [[1.2451e-01, 1.7405e-05, 1.1444e-05,  ..., 9.5459e-02,
           1.0880e-02, 3.8574e-02]],

         [[1.6406e-01, 3.9637e-05, 2.2233e-05,  ..., 9.2545e-03,
           9.1095e-03, 7.4658e-01]],

         ...,

         [[3.9520e-02, 5.2631e-05, 8.2791e-05,  ..., 6.0940e-04,
           1.4734e-03, 2.3788e-02]],

         [[1.4221e-01, 2.8110e-04, 2.5702e-04,  ..., 2.2339e-02,
           3.6255e-02, 3.6888e-03]],

         [[2.4426e-01, 1.2321e-03, 9.0885e-04,  ..., 4.2496e-03,
           7.7896e-03, 2.6398e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 3.9935e-05, 3.6538e-05,  ..., 2.7527e-02,
           6.4888e-03, 4.1016e-02]],

         [[3.2910e-01, 9.6679e-05, 1.9312e-05,  ..., 5.7411e-04,
           1.0866e-04, 9.4833e-03]],

         [[8.1885e-01, 3.2961e-05, 7.4506e-06,  ..., 1.4057e-03,
           7.9036e-05, 6.5002e-03]],

         ...,

         [[4.5685e-02, 5.4455e-04, 1.0252e-03,  ..., 3.8090e-03,
           9.4843e-04, 1.0857e-02]],

         [[5.3906e-01, 5.6922e-05, 1.2338e-04,  ..., 3.4122e-03,
           2.5368e-03, 1.7059e-02]],

         [[4.2676e-01, 6.7759e-04, 2.1362e-03,  ..., 3.6133e-02,
           3.1090e-03, 1.6418e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4438e-01, 3.8266e-05, 9.0718e-05,  ..., 5.1498e-03,
           9.7733e-03, 1.0490e-02]],

         [[7.9529e-02, 1.4532e-04, 4.3058e-04,  ..., 1.1276e-02,
           1.2680e-02, 2.5421e-02]],

         [[1.4587e-01, 9.8050e-05, 2.0444e-04,  ..., 9.1095e-03,
           1.1879e-02, 1.7014e-02]],

         ...,

         [[3.2104e-02, 5.9128e-05, 8.7380e-05,  ..., 4.6692e-02,
           3.1113e-02, 5.0476e-02]],

         [[1.0370e-01, 7.8380e-05, 7.5936e-05,  ..., 6.3248e-03,
           1.1101e-02, 3.8147e-02]],

         [[5.3369e-01, 8.5890e-05, 1.2493e-04,  ..., 1.4214e-02,
           7.7286e-03, 1.4000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3623e-01, 2.0325e-05, 1.8656e-05,  ..., 5.2094e-02,
           1.1177e-02, 3.8116e-02]],

         [[4.0454e-01, 1.9670e-04, 4.1461e-04,  ..., 1.2024e-02,
           1.3412e-02, 1.7639e-02]],

         [[1.2323e-01, 7.4863e-05, 2.0993e-04,  ..., 4.3243e-02,
           5.0964e-02, 6.3086e-01]],

         ...,

         [[5.6348e-01, 8.2254e-05, 1.4555e-04,  ..., 1.8539e-02,
           1.1787e-02, 3.8055e-02]],

         [[2.0740e-01, 2.1696e-05, 2.9993e-04,  ..., 2.9125e-03,
           1.9287e-02, 5.0446e-02]],

         [[4.0430e-01, 4.0340e-04, 5.8031e-04,  ..., 3.0403e-03,
           3.0155e-03, 7.3471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6760e-01, 2.0730e-04, 1.6665e-04,  ..., 2.7802e-02,
           2.4719e-02, 8.3008e-02]],

         [[5.9723e-02, 5.2452e-05, 7.9572e-05,  ..., 1.7242e-02,
           1.7929e-02, 3.1235e-02]],

         [[2.0996e-01, 4.9353e-05, 7.4565e-05,  ..., 4.0627e-03,
           1.4830e-03, 1.6870e-03]],

         ...,

         [[7.4121e-01, 1.0729e-05, 1.3292e-05,  ..., 4.7394e-02,
           8.5144e-02, 5.4108e-02]],

         [[3.1982e-01, 1.9372e-05, 8.9645e-05,  ..., 2.4673e-02,
           4.9744e-03, 2.8168e-02]],

         [[1.9409e-01, 2.0981e-05, 2.8551e-05,  ..., 4.3066e-01,
           1.4307e-01, 2.4872e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2253e-02, 6.6757e-06, 3.1948e-05,  ..., 2.7332e-03,
           5.2977e-04, 1.3145e-02]],

         [[1.5900e-02, 5.3644e-07, 1.1921e-06,  ..., 4.4751e-04,
           3.0117e-03, 9.7607e-01]],

         [[2.7145e-02, 8.6546e-05, 4.4918e-04,  ..., 2.7943e-03,
           7.6234e-05, 4.5991e-04]],

         ...,

         [[1.0669e-01, 3.0708e-04, 9.4128e-04,  ..., 8.9645e-03,
           1.1549e-03, 9.0332e-03]],

         [[8.7280e-02, 7.1526e-07, 1.8477e-06,  ..., 6.3904e-02,
           7.0267e-03, 2.3224e-02]],

         [[2.9114e-02, 4.1723e-07, 1.9670e-06,  ..., 4.2285e-01,
           2.0129e-01, 1.1932e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0664e-01, 2.0564e-04, 7.5150e-04,  ..., 7.0343e-03,
           8.7166e-04, 1.1238e-02]],

         [[9.7900e-01, 1.4865e-04, 2.9111e-04,  ..., 2.9862e-05,
           1.9550e-05, 4.4727e-04]],

         [[6.6260e-01, 2.6941e-05, 2.5797e-04,  ..., 9.4318e-04,
           1.1024e-03, 2.9114e-02]],

         ...,

         [[2.9492e-01, 1.8895e-05, 1.6153e-05,  ..., 3.5571e-01,
           1.1731e-01, 4.5929e-02]],

         [[4.4458e-01, 2.1458e-06, 5.9605e-07,  ..., 1.0357e-03,
           1.0519e-03, 5.9586e-03]],

         [[6.4636e-02, 2.3425e-05, 7.6890e-05,  ..., 1.2026e-03,
           4.2648e-03, 8.7842e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8054e-02, 7.6890e-06, 1.1921e-05,  ..., 1.3512e-02,
           1.8282e-03, 2.1667e-02]],

         [[5.7959e-01, 5.4538e-05, 2.5928e-05,  ..., 7.9346e-02,
           1.6449e-02, 6.2286e-02]],

         [[3.1006e-01, 3.3307e-04, 3.9458e-04,  ..., 7.8369e-02,
           6.8115e-02, 7.9041e-02]],

         ...,

         [[3.3813e-01, 1.7107e-05, 8.8334e-05,  ..., 8.8867e-02,
           2.1439e-02, 8.0261e-02]],

         [[6.8506e-01, 5.3763e-05, 1.3387e-04,  ..., 2.3766e-03,
           1.8806e-03, 1.0567e-02]],

         [[5.0586e-01, 4.8041e-05, 1.9228e-04,  ..., 1.1261e-02,
           7.5607e-03, 2.8992e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9662e-03, 3.1590e-06, 1.2517e-06,  ..., 5.2223e-03,
           3.4714e-04, 8.8644e-04]],

         [[4.8401e-02, 5.5432e-06, 1.7881e-06,  ..., 6.6423e-04,
           2.5105e-04, 1.0653e-03]],

         [[1.1908e-01, 7.3195e-05, 1.4961e-04,  ..., 8.7830e-02,
           1.6632e-02, 1.4819e-01]],

         ...,

         [[5.6000e-02, 5.2691e-05, 3.8028e-04,  ..., 7.8201e-03,
           1.4238e-03, 6.4812e-03]],

         [[1.2366e-01, 7.2837e-05, 3.5834e-04,  ..., 1.6998e-02,
           1.1044e-03, 1.1147e-02]],

         [[1.6980e-01, 5.2869e-05, 5.2273e-05,  ..., 2.6099e-01,
           2.5238e-02, 1.7075e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1123e-01, 1.0848e-05, 1.4544e-05,  ..., 1.4087e-01,
           4.2786e-02, 8.9539e-02]],

         [[3.2861e-01, 7.9691e-05, 1.4412e-04,  ..., 1.5282e-02,
           7.3166e-03, 2.6929e-01]],

         [[4.5679e-01, 5.4240e-06, 1.5438e-05,  ..., 1.0938e-01,
           1.1823e-01, 1.6736e-01]],

         ...,

         [[4.7192e-01, 1.4114e-04, 1.4138e-04,  ..., 1.6403e-02,
           3.7613e-03, 1.3283e-02]],

         [[1.7908e-01, 8.7023e-06, 7.8082e-06,  ..., 1.7357e-03,
           3.9792e-04, 1.4587e-02]],

         [[7.6904e-03, 2.5868e-05, 7.2241e-05,  ..., 1.0033e-02,
           2.7054e-02, 8.6816e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6025e-01, 5.5194e-05, 2.0671e-04,  ..., 5.9021e-02,
           2.9785e-02, 1.7212e-01]],

         [[1.1749e-01, 1.3733e-03, 1.4992e-03,  ..., 6.2027e-03,
           6.5231e-03, 1.8082e-02]],

         [[1.6223e-01, 2.0456e-04, 3.3593e-04,  ..., 1.3847e-02,
           2.0905e-03, 6.8283e-03]],

         ...,

         [[6.7578e-01, 6.1131e-04, 1.9693e-04,  ..., 7.1228e-02,
           1.9165e-02, 3.0151e-02]],

         [[6.4453e-02, 2.2531e-04, 6.3181e-04,  ..., 2.8763e-03,
           1.0193e-02, 8.3545e-01]],

         [[3.2642e-01, 3.3379e-05, 1.6177e-04,  ..., 2.5415e-01,
           3.3667e-01, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5469e-01, 2.7776e-04, 1.0977e-03,  ..., 9.8267e-02,
           2.7686e-01, 5.9662e-03]],

         [[1.2793e-01, 1.5914e-05, 3.3498e-05,  ..., 7.0763e-03,
           5.4779e-03, 1.0645e-01]],

         [[5.4102e-01, 3.9840e-04, 2.1820e-03,  ..., 8.2626e-03,
           6.4163e-03, 1.2802e-02]],

         ...,

         [[6.5918e-01, 2.7323e-04, 4.5586e-03,  ..., 2.3605e-02,
           2.9602e-03, 1.3496e-02]],

         [[8.9355e-01, 2.1374e-04, 2.0027e-03,  ..., 1.6413e-03,
           8.6927e-04, 1.1833e-02]],

         [[4.0015e-01, 1.6832e-04, 1.7810e-04,  ..., 2.8711e-01,
           3.1342e-02, 5.8319e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2343e-04, 5.9843e-04, 2.2729e-01,  ..., 1.2817e-02,
           1.0880e-02, 1.8330e-03]],

         [[8.0287e-05, 2.5058e-04, 1.9556e-01,  ..., 2.4815e-03,
           7.2365e-03, 2.9583e-03]],

         [[1.0395e-03, 1.0748e-03, 1.2030e-01,  ..., 2.7054e-02,
           1.2726e-02, 2.8793e-02]],

         ...,

         [[2.8849e-04, 4.6730e-05, 5.6519e-02,  ..., 7.7820e-02,
           7.0312e-02, 2.4170e-01]],

         [[2.8825e-04, 2.1629e-03, 1.9617e-01,  ..., 1.5915e-02,
           2.3193e-03, 5.5885e-03]],

         [[3.5644e-04, 4.1676e-04, 2.5635e-01,  ..., 9.5596e-03,
           5.4474e-03, 1.8005e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5125e-03, 2.0447e-03, 3.0225e-01,  ..., 3.5591e-03,
           3.0804e-03, 7.8201e-04]],

         [[1.7462e-03, 1.5774e-03, 2.3230e-01,  ..., 1.2426e-03,
           2.2945e-03, 1.2379e-03]],

         [[1.1902e-03, 2.4748e-04, 3.3789e-01,  ..., 9.8324e-04,
           1.3123e-03, 1.9321e-03]],

         ...,

         [[2.8076e-03, 5.9891e-04, 2.4878e-01,  ..., 5.3167e-04,
           9.0075e-04, 1.0509e-03]],

         [[1.6892e-04, 3.3069e-04, 3.4082e-01,  ..., 9.9945e-04,
           4.7836e-03, 2.8014e-04]],

         [[2.9981e-05, 9.8825e-05, 1.0590e-01,  ..., 1.2970e-03,
           7.9250e-04, 1.6165e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3390e-03, 1.1101e-03, 2.3682e-01,  ..., 5.5552e-04,
           1.9302e-03, 9.6512e-04]],

         [[5.2032e-03, 2.0409e-04, 2.7954e-01,  ..., 1.4277e-03,
           3.3188e-03, 1.1450e-04]],

         [[1.3933e-03, 1.8969e-03, 3.2275e-01,  ..., 1.2102e-03,
           1.8597e-03, 8.9788e-04]],

         ...,

         [[3.5763e-04, 7.7677e-04, 1.6968e-01,  ..., 3.5980e-02,
           2.2079e-02, 1.1963e-02]],

         [[7.0238e-04, 1.5078e-03, 2.5513e-01,  ..., 1.3180e-03,
           1.2922e-03, 5.0640e-04]],

         [[2.8439e-03, 1.4162e-03, 1.5906e-01,  ..., 1.3838e-03,
           1.2255e-03, 8.1682e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3355e-04, 3.8207e-05, 2.1042e-02,  ..., 2.6596e-02,
           3.4698e-02, 1.8604e-01]],

         [[1.0699e-04, 2.5272e-04, 1.7285e-01,  ..., 8.0490e-03,
           3.8033e-03, 3.2593e-02]],

         [[2.7704e-04, 1.0939e-03, 3.1421e-01,  ..., 1.5628e-04,
           1.8358e-04, 1.2636e-03]],

         ...,

         [[9.6321e-05, 1.1027e-05, 1.6510e-02,  ..., 1.1396e-04,
           1.9026e-04, 1.5747e-02]],

         [[5.7678e-03, 4.6015e-04, 1.5112e-01,  ..., 1.0948e-02,
           4.1473e-02, 9.3445e-02]],

         [[1.5378e-05, 4.5896e-06, 4.3983e-03,  ..., 5.0545e-04,
           7.7629e-04, 2.5977e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0117e-03, 1.3189e-03, 2.0801e-01,  ..., 6.7635e-03,
           1.2863e-02, 4.1779e-02]],

         [[1.0357e-03, 1.1921e-03, 2.9077e-01,  ..., 1.0748e-03,
           6.3858e-03, 1.9257e-02]],

         [[4.8676e-03, 2.1591e-03, 2.0190e-01,  ..., 1.9455e-02,
           4.5410e-02, 2.7481e-02]],

         ...,

         [[3.1319e-03, 3.3808e-04, 2.2839e-01,  ..., 1.0090e-03,
           4.9667e-03, 1.2779e-02]],

         [[1.5345e-03, 3.1605e-03, 1.1816e-01,  ..., 5.2856e-02,
           4.0833e-02, 3.7549e-01]],

         [[5.9748e-04, 8.9109e-05, 2.9694e-02,  ..., 8.5266e-02,
           1.2402e-01, 2.7954e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.1231e-04, 2.7275e-04, 6.9618e-04,  ..., 1.7041e-01,
           4.7778e-01, 2.7222e-01]],

         [[4.0436e-03, 7.9632e-04, 1.4122e-02,  ..., 2.0752e-01,
           2.9892e-02, 2.4780e-02]],

         [[5.9605e-08, 5.9605e-08, 7.4506e-05,  ..., 5.5027e-04,
           3.3379e-04, 9.9512e-01]],

         ...,

         [[2.8610e-06, 9.9540e-06, 2.0027e-04,  ..., 2.0027e-04,
           6.6338e-03, 9.8438e-01]],

         [[4.9629e-03, 6.3744e-03, 1.9638e-02,  ..., 2.0905e-02,
           4.3793e-03, 1.2024e-01]],

         [[9.9869e-03, 4.4327e-03, 5.9986e-04,  ..., 5.7465e-02,
           3.2745e-02, 1.6464e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3341e-04, 1.6203e-03, 4.7607e-03,  ..., 1.2280e-01,
           1.2091e-01, 1.4807e-01]],

         [[1.1921e-07, 2.3842e-07, 1.1921e-07,  ..., 1.5063e-01,
           7.4170e-01, 1.0193e-01]],

         [[2.3174e-03, 3.0231e-03, 1.1772e-02,  ..., 1.5114e-02,
           1.8799e-02, 1.9409e-02]],

         ...,

         [[5.4777e-05, 8.2850e-05, 1.3905e-03,  ..., 6.7017e-02,
           3.1226e-01, 1.3013e-01]],

         [[3.6449e-03, 6.8626e-03, 2.6306e-02,  ..., 1.2527e-02,
           1.3702e-02, 4.9744e-02]],

         [[1.4496e-04, 3.3975e-04, 1.5230e-03,  ..., 2.6367e-02,
           7.2205e-02, 1.0266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2077e-02, 1.4648e-03, 2.6321e-03,  ..., 2.3328e-01,
           1.4380e-01, 6.9519e-02]],

         [[3.6907e-03, 9.6655e-04, 3.5362e-03,  ..., 2.0237e-03,
           8.8577e-03, 2.0111e-02]],

         [[6.3362e-03, 1.5354e-03, 1.0910e-03,  ..., 3.3081e-02,
           2.5955e-02, 4.5044e-02]],

         ...,

         [[1.1808e-04, 1.3151e-03, 4.5166e-03,  ..., 9.5825e-02,
           5.3027e-01, 2.6660e-01]],

         [[2.3193e-03, 2.6011e-04, 4.2081e-04,  ..., 4.1443e-02,
           1.1896e-01, 1.3477e-01]],

         [[4.6295e-02, 7.8583e-03, 1.8707e-02,  ..., 3.9276e-02,
           2.8290e-02, 8.1848e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2920e-01, 1.6689e-06, 1.7881e-07,  ..., 5.6229e-03,
           3.6449e-03, 2.6360e-03]],

         [[8.8477e-01, 5.0068e-06, 1.1861e-05,  ..., 2.7390e-02,
           4.6448e-02, 2.2411e-03]],

         [[8.9062e-01, 7.7486e-07, 1.6689e-06,  ..., 4.5662e-03,
           1.2217e-03, 4.2023e-02]],

         ...,

         [[5.2979e-01, 2.1887e-04, 1.1368e-03,  ..., 2.6226e-03,
           1.0853e-03, 3.6072e-02]],

         [[5.5811e-01, 2.4819e-04, 5.1165e-04,  ..., 1.7593e-02,
           1.6663e-02, 1.2718e-02]],

         [[7.0654e-01, 3.8743e-04, 4.8733e-04,  ..., 5.7449e-03,
           5.2109e-03, 2.3438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0273e-01, 2.2113e-05, 8.7619e-06,  ..., 2.7027e-03,
           3.2091e-04, 2.7802e-02]],

         [[9.4922e-01, 7.4506e-06, 4.1723e-07,  ..., 5.5237e-03,
           3.0090e-02, 1.4740e-02]],

         [[9.5117e-01, 2.7251e-04, 2.9624e-05,  ..., 2.8610e-03,
           3.2291e-03, 8.4457e-03]],

         ...,

         [[4.5239e-01, 9.4593e-05, 1.2934e-04,  ..., 4.9248e-03,
           3.6438e-02, 2.4460e-02]],

         [[1.4270e-01, 4.0054e-03, 1.6546e-03,  ..., 1.6270e-03,
           4.7264e-03, 1.2581e-02]],

         [[6.5820e-01, 6.8784e-05, 2.1148e-04,  ..., 8.2092e-02,
           5.9937e-02, 1.6113e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8296e-01, 2.7776e-04, 1.5342e-04,  ..., 2.1118e-02,
           3.1219e-02, 2.8503e-02]],

         [[6.7090e-01, 1.1086e-05, 7.3910e-06,  ..., 7.8087e-03,
           2.4376e-03, 2.2110e-02]],

         [[2.8125e-01, 1.1146e-05, 5.6624e-06,  ..., 5.3894e-02,
           1.9226e-02, 1.7807e-02]],

         ...,

         [[6.2402e-01, 2.0325e-05, 1.9073e-06,  ..., 5.6803e-05,
           1.0443e-04, 3.7183e-01]],

         [[9.5166e-01, 1.3280e-04, 5.3227e-05,  ..., 2.9802e-07,
           1.1921e-07, 1.4023e-02]],

         [[9.9268e-01, 1.8477e-05, 2.3246e-06,  ..., 0.0000e+00,
           0.0000e+00, 3.9101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0063e-01, 1.2755e-05, 6.0201e-06,  ..., 5.2368e-02,
           2.2217e-01, 1.6006e-02]],

         [[9.5654e-01, 1.1742e-05, 5.4240e-06,  ..., 1.2696e-05,
           2.8658e-04, 1.0208e-02]],

         [[8.0420e-01, 3.3832e-04, 6.7413e-05,  ..., 2.2144e-03,
           3.6736e-03, 6.1707e-02]],

         ...,

         [[7.2217e-01, 1.9097e-04, 3.2663e-05,  ..., 7.9498e-03,
           1.2100e-02, 2.4292e-02]],

         [[4.2725e-01, 9.4891e-05, 1.6570e-05,  ..., 7.1831e-03,
           7.9651e-02, 2.4323e-02]],

         [[8.9844e-01, 1.1200e-04, 8.7202e-05,  ..., 4.0169e-03,
           9.3994e-03, 1.1902e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0625e-01, 5.2452e-06, 6.2585e-06,  ..., 6.1127e-02,
           2.9834e-01, 9.0515e-02]],

         [[7.3877e-01, 1.1396e-04, 5.1975e-05,  ..., 2.2449e-03,
           3.4981e-03, 2.9037e-02]],

         [[4.2627e-01, 1.2577e-05, 4.7088e-06,  ..., 1.2917e-02,
           7.5012e-02, 1.7532e-02]],

         ...,

         [[7.0361e-01, 1.4913e-04, 5.1558e-05,  ..., 1.7776e-02,
           2.6077e-02, 2.6779e-02]],

         [[8.6279e-01, 3.6860e-04, 1.0562e-04,  ..., 1.1854e-03,
           1.7593e-02, 3.9635e-03]],

         [[3.9307e-01, 1.3590e-05, 2.2411e-05,  ..., 5.8899e-02,
           1.0120e-01, 3.5706e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3232e-01, 2.2793e-03, 1.8320e-03,  ..., 1.7583e-04,
           3.6144e-03, 1.9817e-03]],

         [[9.5020e-01, 3.8803e-05, 6.7651e-05,  ..., 1.6003e-03,
           1.1530e-03, 3.8090e-03]],

         [[8.8232e-01, 7.0870e-05, 4.8339e-05,  ..., 1.8635e-03,
           1.9958e-02, 1.8415e-03]],

         ...,

         [[8.3447e-01, 2.8205e-04, 3.0375e-04,  ..., 1.6718e-03,
           1.0967e-03, 6.2988e-02]],

         [[9.3164e-01, 4.0531e-06, 7.5698e-06,  ..., 3.0060e-03,
           5.6076e-03, 3.0869e-02]],

         [[8.5742e-01, 4.0293e-05, 3.4213e-05,  ..., 4.9057e-03,
           6.5674e-02, 3.8185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4158e-01, 7.4387e-04, 8.2684e-04,  ..., 8.0442e-04,
           1.7471e-02, 7.7057e-03]],

         [[8.1299e-01, 8.2076e-05, 1.0455e-04,  ..., 2.8591e-03,
           2.6978e-02, 6.5956e-03]],

         [[3.0347e-01, 1.8299e-05, 4.2200e-05,  ..., 4.6844e-02,
           3.5010e-01, 8.0338e-03]],

         ...,

         [[1.9543e-01, 6.2323e-04, 1.3161e-03,  ..., 1.7428e-04,
           2.3327e-03, 7.1754e-03]],

         [[6.0791e-01, 8.8751e-05, 5.7876e-05,  ..., 1.1581e-02,
           2.7515e-01, 5.4893e-03]],

         [[8.9209e-01, 4.0531e-06, 8.6427e-06,  ..., 3.1929e-03,
           3.4199e-03, 4.1718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.9426e-05, 2.2161e-04,  ..., 2.9254e-04,
           1.0052e-03, 1.7365e-02]],

         [[4.4604e-01, 5.3227e-05, 1.0425e-04,  ..., 1.5167e-02,
           1.1066e-01, 2.1179e-02]],

         [[6.9824e-01, 1.0550e-05, 3.6538e-05,  ..., 9.0265e-04,
           2.7580e-03, 2.4979e-02]],

         ...,

         [[9.5166e-01, 4.7743e-05, 2.5332e-05,  ..., 6.4135e-04,
           2.5463e-03, 3.0861e-03]],

         [[7.4170e-01, 4.2176e-04, 1.3018e-04,  ..., 4.1504e-02,
           6.9351e-03, 2.4780e-02]],

         [[2.3315e-01, 1.5812e-03, 5.0116e-04,  ..., 1.9360e-04,
           3.3913e-03, 1.6815e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9453e-01, 3.8147e-06, 6.7353e-06,  ..., 4.3011e-04,
           9.4986e-03, 5.3955e-01]],

         [[6.1670e-01, 8.0633e-04, 1.4725e-03,  ..., 3.7050e-04,
           1.3412e-02, 1.0862e-03]],

         [[7.6074e-01, 4.6921e-04, 5.1641e-04,  ..., 1.1520e-03,
           1.5068e-02, 1.9608e-02]],

         ...,

         [[5.6885e-01, 1.2589e-04, 2.4748e-04,  ..., 2.4700e-03,
           1.2375e-02, 8.2932e-03]],

         [[8.0994e-02, 2.1040e-05, 1.2755e-05,  ..., 3.8719e-04,
           7.5226e-03, 4.0649e-02]],

         [[1.4270e-01, 2.6464e-05, 7.7844e-05,  ..., 8.5602e-03,
           9.0561e-03, 7.1436e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7810e-01, 1.1140e-04, 1.7059e-04,  ..., 7.6332e-03,
           5.4741e-03, 5.8984e-01]],

         [[5.3369e-01, 2.1195e-04, 3.6073e-04,  ..., 4.8327e-04,
           3.2806e-02, 8.9951e-03]],

         [[6.7529e-01, 1.3280e-04, 1.5402e-04,  ..., 2.0564e-04,
           2.6627e-03, 9.5901e-03]],

         ...,

         [[2.5162e-02, 2.9206e-06, 1.0967e-05,  ..., 7.3671e-04,
           3.7212e-03, 9.5898e-01]],

         [[7.6807e-01, 1.1253e-04, 1.4508e-04,  ..., 4.3907e-03,
           2.2163e-03, 9.4833e-03]],

         [[7.6855e-01, 1.4806e-04, 7.7069e-05,  ..., 7.9060e-04,
           1.9867e-02, 7.3586e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1919e-01, 5.2071e-04, 7.6914e-04,  ..., 3.9597e-03,
           2.6367e-01, 1.0849e-02]],

         [[7.2070e-01, 7.6234e-05, 1.5223e-04,  ..., 2.3346e-03,
           4.9377e-02, 2.9724e-02]],

         [[3.7671e-01, 7.2837e-05, 1.8597e-04,  ..., 2.8920e-04,
           9.2459e-04, 1.2733e-02]],

         ...,

         [[7.8271e-01, 8.9407e-07, 1.9670e-06,  ..., 5.7831e-03,
           1.6736e-01, 3.1342e-02]],

         [[5.0195e-01, 9.1887e-04, 9.4414e-04,  ..., 5.6601e-04,
           1.4656e-02, 1.3916e-02]],

         [[5.2490e-01, 4.7684e-07, 1.0729e-06,  ..., 1.8295e-02,
           3.4692e-01, 6.1951e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0180e-04, 2.3901e-05,  ..., 7.2050e-04,
           3.4218e-03, 5.2147e-03]],

         [[1.5161e-01, 6.0701e-04, 6.1512e-04,  ..., 3.3474e-04,
           5.0011e-03, 1.3130e-02]],

         [[4.5557e-01, 8.5735e-04, 8.5735e-04,  ..., 1.8466e-04,
           2.4063e-02, 1.4534e-02]],

         ...,

         [[5.0098e-01, 2.9802e-06, 3.8743e-06,  ..., 1.7490e-03,
           4.0955e-02, 5.0049e-02]],

         [[2.9858e-01, 1.4305e-06, 1.2517e-06,  ..., 4.7798e-03,
           5.7068e-02, 2.8564e-02]],

         [[1.7041e-01, 3.0041e-05, 5.2512e-05,  ..., 3.7766e-04,
           1.3268e-02, 1.1101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0171e-02, 8.3447e-07, 1.0729e-06,  ..., 1.3065e-04,
           6.6338e-03, 8.7207e-01]],

         [[3.0615e-01, 6.7353e-06, 1.9312e-05,  ..., 1.1978e-02,
           7.1777e-02, 6.5857e-02]],

         [[4.4312e-01, 1.0128e-03, 1.0872e-03,  ..., 4.3273e-04,
           9.3307e-03, 1.1795e-02]],

         ...,

         [[2.1509e-01, 4.3452e-05, 5.5313e-05,  ..., 3.9330e-03,
           8.4045e-02, 2.9602e-02]],

         [[5.8594e-01, 5.7220e-06, 1.5616e-05,  ..., 1.4782e-03,
           3.6907e-03, 1.4816e-02]],

         [[8.9722e-02, 7.8201e-05, 7.4625e-05,  ..., 7.9679e-04,
           1.2428e-02, 8.3008e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0928e-01, 4.6194e-05, 7.8380e-05,  ..., 6.5880e-03,
           1.7603e-01, 8.2321e-03]],

         [[2.2754e-01, 4.1771e-04, 4.1032e-04,  ..., 4.6692e-03,
           5.5115e-02, 4.3427e-02]],

         [[7.3181e-02, 7.7248e-05, 2.4319e-05,  ..., 5.8985e-04,
           3.4546e-02, 4.7607e-02]],

         ...,

         [[6.7627e-01, 7.8857e-05, 2.3901e-05,  ..., 1.8377e-03,
           5.8411e-02, 2.5513e-02]],

         [[3.1201e-01, 1.0449e-04, 1.3804e-04,  ..., 3.3450e-04,
           6.1920e-02, 4.4067e-02]],

         [[6.8506e-01, 2.4092e-04, 1.1784e-04,  ..., 6.5470e-04,
           2.8717e-02, 2.2812e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9854e-02, 1.4505e-03, 1.3819e-03,  ..., 6.9916e-05,
           3.7956e-03, 3.0975e-03]],

         [[1.7029e-01, 4.9925e-04, 3.0589e-04,  ..., 1.6460e-03,
           4.1901e-02, 1.2291e-02]],

         [[5.2051e-01, 2.8849e-05, 5.4836e-05,  ..., 1.0939e-03,
           1.0406e-01, 1.9775e-02]],

         ...,

         [[5.0000e-01, 3.1114e-04, 1.4615e-04,  ..., 1.9894e-03,
           9.8267e-03, 3.8422e-02]],

         [[9.6143e-01, 8.8871e-05, 1.9515e-04,  ..., 1.0252e-05,
           1.9372e-05, 3.8395e-03]],

         [[9.1357e-01, 1.1140e-04, 1.5104e-04,  ..., 5.0259e-04,
           1.6308e-03, 3.8414e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9048e-01, 5.7888e-04, 7.5483e-04,  ..., 3.5667e-04,
           1.0185e-02, 2.5192e-02]],

         [[2.0630e-01, 2.9888e-03, 1.3847e-03,  ..., 7.7677e-04,
           1.7334e-02, 1.6937e-02]],

         [[3.1204e-02, 7.2122e-06, 8.0466e-06,  ..., 6.4392e-03,
           2.0309e-02, 6.1719e-01]],

         ...,

         [[3.0566e-01, 2.5630e-05, 2.9325e-05,  ..., 5.2357e-04,
           3.3875e-02, 1.3062e-02]],

         [[5.1562e-01, 1.1504e-05, 8.4639e-06,  ..., 2.4834e-03,
           2.3242e-01, 5.7297e-03]],

         [[1.4270e-01, 2.1636e-05, 1.7762e-05,  ..., 3.2024e-03,
           7.4585e-02, 3.7012e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8135e-01, 4.6074e-05, 2.3901e-05,  ..., 1.4553e-03,
           1.0481e-03, 1.0925e-02]],

         [[6.4331e-02, 4.3631e-05, 2.6464e-05,  ..., 1.5612e-03,
           8.5266e-02, 2.2583e-02]],

         [[1.7957e-01, 4.0770e-05, 2.3246e-05,  ..., 4.0507e-04,
           1.0124e-02, 7.4414e-01]],

         ...,

         [[1.2927e-01, 4.8828e-04, 2.2447e-04,  ..., 6.5231e-04,
           3.4981e-03, 1.0941e-02]],

         [[3.5010e-01, 8.6784e-04, 5.3883e-04,  ..., 6.8817e-03,
           5.1537e-03, 7.3242e-03]],

         [[3.6841e-01, 1.1015e-03, 1.0843e-03,  ..., 1.1169e-02,
           9.8190e-03, 4.6921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4585e-01, 2.2507e-04, 9.9480e-05,  ..., 5.2643e-03,
           6.8848e-02, 3.9825e-02]],

         [[9.2578e-01, 7.8499e-05, 1.1444e-05,  ..., 2.1505e-04,
           2.2411e-03, 9.2163e-03]],

         [[8.9600e-01, 3.9876e-05, 9.6560e-06,  ..., 8.2970e-04,
           7.8125e-03, 2.6779e-03]],

         ...,

         [[8.0750e-02, 2.0218e-03, 3.4389e-03,  ..., 2.9802e-04,
           4.9248e-03, 4.8866e-03]],

         [[5.1465e-01, 3.0935e-05, 3.5048e-05,  ..., 1.8997e-03,
           5.1758e-02, 6.2790e-03]],

         [[2.0142e-01, 5.7030e-04, 1.7157e-03,  ..., 1.9217e-03,
           1.6403e-02, 3.0884e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0820e-01, 2.1040e-04, 2.2221e-04,  ..., 7.8354e-03,
           4.9400e-03, 1.2917e-02]],

         [[3.3228e-01, 2.8920e-04, 6.9904e-04,  ..., 3.1586e-03,
           1.3718e-02, 3.1891e-02]],

         [[2.9028e-01, 7.9215e-05, 1.9073e-04,  ..., 1.1986e-02,
           3.5797e-02, 1.1345e-02]],

         ...,

         [[1.3245e-02, 5.9485e-05, 9.5010e-05,  ..., 1.9388e-03,
           1.5244e-02, 2.2888e-02]],

         [[1.3245e-01, 1.5998e-04, 1.0008e-04,  ..., 6.2275e-04,
           1.9178e-03, 2.3880e-03]],

         [[6.4893e-01, 2.4104e-04, 3.9124e-04,  ..., 2.7580e-03,
           4.5815e-03, 2.6951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1542e-01, 1.7881e-05, 1.7107e-05,  ..., 2.0981e-03,
           1.6251e-02, 9.1858e-03]],

         [[2.6440e-01, 1.0127e-04, 1.8120e-04,  ..., 5.5084e-03,
           1.8127e-02, 6.3896e-03]],

         [[2.0337e-01, 4.7088e-05, 1.5676e-04,  ..., 6.9046e-03,
           1.8051e-02, 6.2158e-01]],

         ...,

         [[7.0361e-01, 3.5465e-05, 6.5267e-05,  ..., 4.1351e-03,
           3.9703e-02, 2.7924e-02]],

         [[3.9160e-01, 2.5868e-05, 1.3876e-04,  ..., 3.7193e-03,
           7.6332e-03, 2.5620e-02]],

         [[4.8657e-01, 8.4341e-05, 9.3699e-05,  ..., 5.3930e-04,
           1.9121e-03, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3684e-01, 1.6236e-04, 8.1360e-05,  ..., 3.2425e-03,
           2.3224e-02, 1.5228e-02]],

         [[6.1340e-02, 2.4915e-05, 2.6405e-05,  ..., 8.4991e-03,
           1.3391e-01, 1.6373e-02]],

         [[4.9756e-01, 1.5318e-04, 1.3280e-04,  ..., 1.5955e-03,
           5.4207e-03, 1.3893e-02]],

         ...,

         [[2.6636e-01, 4.4703e-06, 2.9266e-05,  ..., 1.7303e-02,
           6.3916e-01, 2.8290e-02]],

         [[4.0601e-01, 3.3498e-05, 7.9751e-05,  ..., 3.9787e-03,
           3.5187e-02, 4.3457e-02]],

         [[4.9951e-01, 2.2352e-05, 7.5817e-05,  ..., 3.5267e-03,
           3.3276e-01, 7.3059e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6865e-02, 1.1683e-05, 5.7042e-05,  ..., 2.0542e-03,
           3.8025e-02, 2.7817e-02]],

         [[1.1864e-02, 4.7684e-06, 9.8944e-06,  ..., 2.6631e-04,
           6.3477e-03, 9.6484e-01]],

         [[2.6520e-02, 2.2829e-05, 1.1903e-04,  ..., 1.6046e-04,
           1.2407e-03, 2.7618e-03]],

         ...,

         [[9.7595e-02, 2.5415e-04, 9.0599e-04,  ..., 6.9332e-04,
           1.5869e-02, 1.1566e-02]],

         [[4.4727e-01, 3.9935e-06, 9.7156e-06,  ..., 8.5020e-04,
           1.3123e-01, 4.0802e-02]],

         [[2.5488e-01, 7.7486e-07, 3.6955e-06,  ..., 2.0432e-02,
           3.4961e-01, 1.4294e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9395e-01, 2.3758e-04, 1.1396e-03,  ..., 5.2452e-04,
           8.0032e-03, 6.7635e-03]],

         [[9.8486e-01, 3.1948e-05, 1.9491e-04,  ..., 1.8716e-05,
           3.3689e-04, 1.0376e-03]],

         [[8.1934e-01, 1.8883e-04, 2.1994e-04,  ..., 3.1815e-03,
           1.1871e-02, 6.1569e-03]],

         ...,

         [[6.5332e-01, 6.1333e-05, 6.8426e-05,  ..., 1.8814e-02,
           5.7983e-02, 1.1063e-02]],

         [[4.8413e-01, 1.2517e-06, 2.9802e-07,  ..., 9.4175e-04,
           1.4820e-03, 4.8981e-03]],

         [[1.9873e-01, 2.5272e-05, 5.7459e-05,  ..., 2.3997e-04,
           4.5044e-02, 6.6162e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2074e-02, 9.0003e-06, 1.2636e-05,  ..., 3.9368e-03,
           1.7334e-01, 8.8501e-02]],

         [[6.1914e-01, 1.1981e-04, 6.1274e-05,  ..., 1.2848e-02,
           7.0557e-02, 2.8503e-02]],

         [[1.8530e-01, 5.5122e-04, 3.5477e-04,  ..., 2.3743e-02,
           6.0638e-02, 1.8280e-02]],

         ...,

         [[4.3701e-01, 2.5630e-05, 1.0747e-04,  ..., 3.0212e-02,
           2.0972e-01, 1.6953e-02]],

         [[6.4551e-01, 2.3007e-05, 7.4208e-05,  ..., 5.6915e-03,
           2.9251e-02, 6.2408e-02]],

         [[3.9014e-01, 1.8835e-05, 6.1929e-05,  ..., 5.9204e-03,
           2.6749e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7863e-03, 1.9670e-06, 1.6093e-06,  ..., 5.3167e-04,
           2.5768e-03, 4.4861e-03]],

         [[8.2520e-02, 8.4639e-06, 4.4107e-06,  ..., 6.6566e-04,
           5.2376e-03, 5.8861e-03]],

         [[1.8958e-01, 9.5308e-05, 3.0708e-04,  ..., 1.7227e-02,
           1.2537e-01, 3.4821e-02]],

         ...,

         [[8.3374e-02, 1.1683e-05, 1.4198e-04,  ..., 1.7996e-03,
           1.0277e-02, 6.4941e-02]],

         [[2.0288e-01, 3.9995e-05, 3.4571e-04,  ..., 1.5488e-03,
           1.7456e-02, 5.4626e-02]],

         [[7.9199e-01, 2.4915e-05, 7.7367e-05,  ..., 4.8599e-03,
           3.2196e-02, 3.2196e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4316e-01, 1.3769e-05, 2.9385e-05,  ..., 1.0681e-02,
           3.2074e-02, 4.1840e-02]],

         [[3.2617e-01, 3.7551e-05, 1.5211e-04,  ..., 1.8509e-02,
           1.7468e-01, 5.4321e-02]],

         [[1.5942e-01, 2.8610e-06, 1.1146e-05,  ..., 2.7370e-03,
           6.6357e-01, 1.0956e-01]],

         ...,

         [[4.5703e-01, 7.6950e-05, 1.4067e-04,  ..., 5.7755e-03,
           1.1436e-02, 1.0132e-02]],

         [[2.9590e-01, 2.5809e-05, 1.5795e-05,  ..., 1.1444e-03,
           5.2216e-02, 4.3304e-02]],

         [[5.9021e-02, 1.0747e-04, 4.1842e-04,  ..., 1.5884e-02,
           6.4819e-02, 5.9619e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4546e-01, 7.5340e-05, 1.7405e-04,  ..., 1.7136e-02,
           1.0870e-01, 1.4259e-02]],

         [[3.3496e-01, 3.9530e-04, 4.8256e-04,  ..., 4.1847e-03,
           6.1836e-03, 3.1109e-03]],

         [[2.0679e-01, 8.7023e-05, 1.8466e-04,  ..., 3.2654e-03,
           4.6616e-03, 9.4910e-03]],

         ...,

         [[7.0996e-01, 4.5896e-04, 4.6635e-04,  ..., 6.5880e-03,
           2.0782e-02, 3.9429e-02]],

         [[1.4392e-01, 1.6844e-04, 4.9496e-04,  ..., 3.0327e-03,
           2.2064e-02, 7.5391e-01]],

         [[8.8184e-01, 2.4259e-05, 4.9829e-05,  ..., 1.3390e-02,
           6.1302e-03, 2.4628e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4766e-01, 1.0097e-04, 1.1864e-03,  ..., 1.7792e-02,
           1.3863e-02, 7.3700e-03]],

         [[4.8645e-02, 4.6492e-06, 2.7835e-05,  ..., 9.5797e-04,
           5.5786e-02, 1.8463e-02]],

         [[1.5686e-01, 8.0884e-05, 3.6645e-04,  ..., 2.8122e-02,
           7.1838e-02, 1.3904e-01]],

         ...,

         [[6.9922e-01, 1.6582e-04, 1.0080e-03,  ..., 4.5166e-03,
           3.2959e-02, 4.0863e-02]],

         [[8.0518e-01, 1.7881e-06, 3.2544e-04,  ..., 2.5272e-03,
           6.6261e-03, 2.7409e-03]],

         [[2.7588e-01, 1.1235e-04, 1.0341e-04,  ..., 4.3121e-02,
           1.2189e-01, 4.4312e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8394e-05, 1.2207e-04, 1.5515e-01,  ..., 5.0850e-03,
           7.7200e-04, 4.1748e-02]],

         [[2.9862e-05, 2.2411e-05, 4.1595e-02,  ..., 6.4945e-04,
           1.4019e-03, 8.3542e-03]],

         [[5.8222e-04, 8.4305e-04, 8.8013e-02,  ..., 3.3417e-03,
           1.1147e-02, 1.8738e-02]],

         ...,

         [[4.6206e-04, 1.6224e-04, 2.1545e-01,  ..., 1.0315e-02,
           4.3762e-02, 1.4465e-01]],

         [[2.7132e-04, 2.9278e-04, 3.2861e-01,  ..., 9.1934e-04,
           5.3749e-03, 8.3923e-03]],

         [[1.4782e-05, 1.5616e-05, 1.4355e-01,  ..., 6.6996e-04,
           5.4359e-03, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0969e-04, 1.3959e-04, 2.8052e-01,  ..., 1.1873e-03,
           2.7618e-03, 1.5343e-02]],

         [[1.3328e-04, 1.3196e-04, 1.2646e-01,  ..., 1.8616e-03,
           1.2302e-03, 1.0139e-02]],

         [[9.5546e-05, 2.7180e-05, 2.8418e-01,  ..., 1.7920e-03,
           4.0245e-03, 2.2263e-02]],

         ...,

         [[4.1771e-04, 6.7532e-05, 3.7018e-02,  ..., 2.2447e-04,
           6.0797e-04, 1.6136e-03]],

         [[1.9312e-05, 2.2113e-05, 2.2534e-01,  ..., 3.0670e-03,
           2.0957e-04, 8.9798e-03]],

         [[1.0610e-04, 1.3387e-04, 2.1741e-01,  ..., 2.0256e-03,
           5.2643e-04, 1.6769e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0637e-05, 3.2425e-05, 8.7585e-02,  ..., 8.1825e-04,
           8.0776e-04, 3.0575e-03]],

         [[2.5201e-04, 1.1802e-05, 2.0142e-01,  ..., 1.8396e-03,
           8.5711e-05, 1.1360e-02]],

         [[1.0622e-04, 1.2219e-04, 1.9727e-01,  ..., 9.7656e-04,
           1.0729e-03, 7.4158e-03]],

         ...,

         [[1.5414e-04, 7.8619e-05, 1.3635e-01,  ..., 5.5618e-03,
           3.8528e-03, 2.3499e-02]],

         [[1.9073e-05, 4.5121e-05, 7.1045e-02,  ..., 3.8767e-04,
           5.0449e-04, 4.5242e-03]],

         [[6.9559e-05, 2.9445e-05, 1.1188e-01,  ..., 7.3910e-04,
           6.5994e-04, 4.8409e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9605e-05, 5.0068e-06, 1.4328e-02,  ..., 2.3479e-03,
           1.1011e-01, 2.0093e-01]],

         [[3.0518e-05, 8.3923e-05, 1.5942e-01,  ..., 5.7030e-03,
           7.5256e-02, 3.5828e-02]],

         [[1.7166e-04, 5.4932e-04, 2.8955e-01,  ..., 6.0177e-04,
           8.3733e-04, 5.8699e-04]],

         ...,

         [[3.4142e-04, 4.9293e-05, 6.7871e-02,  ..., 4.0352e-05,
           2.4338e-03, 9.0714e-03]],

         [[4.0665e-03, 2.0480e-04, 1.2030e-01,  ..., 2.6684e-03,
           4.6204e-02, 2.5171e-01]],

         [[7.2122e-06, 1.3113e-06, 1.5020e-03,  ..., 1.6463e-04,
           8.5327e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8321e-03, 1.4305e-03, 1.2891e-01,  ..., 1.4130e-02,
           1.7578e-02, 7.2876e-02]],

         [[2.3098e-03, 1.4591e-03, 2.4158e-01,  ..., 7.2098e-03,
           1.3161e-02, 2.7008e-02]],

         [[8.6288e-03, 1.1053e-03, 2.2351e-01,  ..., 5.2223e-03,
           7.4806e-03, 1.1208e-02]],

         ...,

         [[1.6441e-03, 9.9123e-05, 2.3840e-01,  ..., 1.5812e-03,
           6.3515e-03, 1.4453e-01]],

         [[2.3174e-03, 2.9526e-03, 1.9751e-01,  ..., 4.2389e-02,
           5.7373e-03, 5.5298e-02]],

         [[1.5950e-04, 7.7426e-05, 3.7231e-02,  ..., 7.5645e-03,
           2.5439e-01, 3.8501e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7001e-05, 6.4790e-05, 1.2608e-03,  ..., 8.8379e-02,
           4.7803e-01, 3.2837e-01]],

         [[2.2125e-03, 1.1835e-03, 1.7395e-02,  ..., 3.8815e-03,
           6.8787e-02, 4.2139e-01]],

         [[9.5367e-07, 1.1921e-06, 2.2087e-03,  ..., 8.5592e-05,
           5.6946e-02, 8.9062e-01]],

         ...,

         [[1.7881e-07, 8.3447e-07, 7.4267e-05,  ..., 1.2243e-04,
           2.4586e-03, 9.9170e-01]],

         [[2.7199e-03, 2.3994e-03, 1.4702e-02,  ..., 6.1302e-03,
           2.4246e-02, 2.0300e-01]],

         [[1.1749e-02, 4.3221e-03, 9.6464e-04,  ..., 3.1952e-02,
           2.6489e-02, 2.8198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0118e-03, 4.6082e-03, 1.6327e-02,  ..., 7.3181e-02,
           1.0651e-01, 1.4551e-01]],

         [[1.7881e-07, 3.5763e-07, 4.1723e-07,  ..., 1.7346e-01,
           7.4219e-01, 7.8186e-02]],

         [[2.5635e-03, 4.7150e-03, 1.7517e-02,  ..., 2.5879e-02,
           4.2664e-02, 1.9852e-02]],

         ...,

         [[1.2808e-03, 6.8521e-04, 6.8130e-03,  ..., 7.7393e-02,
           1.2634e-02, 7.3242e-02]],

         [[4.8714e-03, 1.4091e-02, 4.4617e-02,  ..., 1.2581e-02,
           3.6560e-02, 1.0933e-02]],

         [[1.8797e-03, 1.2070e-02, 2.8732e-02,  ..., 5.4932e-02,
           6.7810e-02, 3.0106e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2490e-02, 6.6757e-03, 1.5396e-02,  ..., 6.6895e-02,
           4.5959e-02, 4.3518e-02]],

         [[3.8414e-03, 8.4352e-04, 3.3245e-03,  ..., 4.8370e-03,
           8.6899e-03, 3.0270e-03]],

         [[4.9629e-03, 1.3208e-03, 1.2264e-03,  ..., 2.0264e-02,
           3.7994e-02, 2.2949e-02]],

         ...,

         [[2.9812e-03, 1.4639e-03, 2.3575e-03,  ..., 3.5254e-01,
           1.0754e-01, 1.5405e-01]],

         [[3.0689e-03, 5.2929e-04, 3.3236e-04,  ..., 2.8711e-01,
           1.2952e-01, 5.2307e-02]],

         [[2.6566e-02, 6.7711e-03, 1.7975e-02,  ..., 2.6566e-02,
           3.8135e-01, 9.4910e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6035e-01, 1.7285e-06, 1.1921e-06,  ..., 2.1915e-03,
           2.1458e-03, 1.0811e-02]],

         [[9.3359e-01, 4.8876e-06, 1.1086e-05,  ..., 8.8882e-03,
           1.2314e-02, 4.8208e-04]],

         [[8.8232e-01, 1.1921e-06, 4.7684e-06,  ..., 6.9504e-03,
           1.8280e-02, 1.9165e-02]],

         ...,

         [[6.3379e-01, 8.7798e-05, 8.4114e-04,  ..., 5.1618e-05,
           2.6684e-03, 2.4200e-02]],

         [[5.0342e-01, 1.3387e-04, 3.1519e-04,  ..., 2.8061e-02,
           2.4002e-02, 1.0948e-02]],

         [[6.2158e-01, 3.1114e-04, 4.3416e-04,  ..., 2.0050e-02,
           4.9622e-02, 2.2278e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6182e-01, 4.4703e-06, 1.7881e-06,  ..., 7.7200e-04,
           2.3499e-03, 3.6652e-02]],

         [[1.0907e-01, 5.3644e-07, 1.7881e-07,  ..., 2.5597e-03,
           8.7842e-01, 9.5596e-03]],

         [[9.4824e-01, 1.0818e-04, 5.7936e-05,  ..., 1.2007e-03,
           1.5656e-02, 4.8294e-03]],

         ...,

         [[4.4360e-01, 2.5415e-04, 1.2094e-04,  ..., 5.8624e-02,
           5.4016e-02, 3.1311e-02]],

         [[1.3232e-01, 9.4175e-04, 6.2084e-04,  ..., 5.9929e-03,
           1.7319e-02, 7.3738e-03]],

         [[4.4824e-01, 8.7202e-05, 1.4317e-04,  ..., 1.1877e-01,
           6.4819e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1348e-01, 3.6001e-04, 1.5426e-04,  ..., 4.5898e-02,
           5.9296e-02, 3.0640e-02]],

         [[6.4502e-01, 1.9372e-05, 1.2517e-05,  ..., 1.2398e-02,
           1.2891e-01, 1.8326e-02]],

         [[5.4980e-01, 1.7405e-05, 4.3511e-06,  ..., 1.7868e-02,
           6.3721e-02, 1.2474e-02]],

         ...,

         [[6.1670e-01, 1.7881e-06, 4.7684e-07,  ..., 6.2585e-06,
           5.4779e-03, 3.7402e-01]],

         [[9.2383e-01, 1.9789e-05, 2.9981e-05,  ..., 0.0000e+00,
           1.3483e-04, 4.3243e-02]],

         [[9.8389e-01, 2.3842e-06, 7.1526e-07,  ..., 0.0000e+00,
           2.7180e-05, 9.4223e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 1.9431e-05, 2.4438e-06,  ..., 1.2018e-01,
           1.3232e-01, 1.0727e-02]],

         [[9.5947e-01, 1.0967e-05, 2.6822e-06,  ..., 3.1614e-04,
           1.5564e-03, 1.1902e-02]],

         [[7.7197e-01, 5.7459e-04, 6.1989e-05,  ..., 9.2888e-04,
           2.3941e-02, 5.9906e-02]],

         ...,

         [[8.0273e-01, 4.7505e-05, 1.1981e-05,  ..., 6.9962e-03,
           1.8890e-02, 1.8280e-02]],

         [[7.9932e-01, 1.5581e-04, 4.9233e-05,  ..., 3.6865e-02,
           2.7878e-02, 3.1647e-02]],

         [[9.2285e-01, 6.2644e-05, 4.0948e-05,  ..., 6.5727e-03,
           9.7809e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 6.1393e-06, 6.9737e-06,  ..., 2.6221e-01,
           3.3417e-02, 3.4821e-02]],

         [[7.6025e-01, 2.9564e-05, 9.3997e-05,  ..., 5.6725e-03,
           1.8433e-02, 5.0507e-02]],

         [[9.3652e-01, 3.0160e-05, 2.5034e-06,  ..., 1.1932e-02,
           6.6795e-03, 1.7529e-03]],

         ...,

         [[6.7920e-01, 1.9467e-04, 8.2731e-05,  ..., 1.1627e-02,
           2.7252e-02, 1.9485e-02]],

         [[8.7793e-01, 1.4353e-04, 3.6597e-05,  ..., 5.9624e-03,
           9.1629e-03, 4.3602e-03]],

         [[7.5635e-01, 3.4571e-05, 1.7524e-05,  ..., 2.0599e-02,
           6.6101e-02, 2.8351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6055e-01, 1.7023e-03, 8.2970e-04,  ..., 2.9411e-03,
           1.9445e-03, 3.4523e-03]],

         [[9.1064e-01, 8.7082e-05, 1.2863e-04,  ..., 9.7752e-04,
           7.9956e-03, 9.7961e-03]],

         [[8.7842e-01, 6.3598e-05, 2.9564e-05,  ..., 4.7874e-03,
           7.5436e-04, 2.4338e-03]],

         ...,

         [[5.2295e-01, 3.8505e-05, 1.3447e-04,  ..., 1.9188e-03,
           3.4607e-02, 3.0249e-01]],

         [[5.4102e-01, 4.7684e-06, 6.3181e-06,  ..., 3.3665e-03,
           2.9224e-01, 1.0883e-01]],

         [[9.0576e-01, 2.7418e-05, 2.1696e-05,  ..., 2.7893e-02,
           5.4398e-03, 1.0880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5405e-01, 6.1035e-04, 1.1740e-03,  ..., 1.6129e-02,
           1.6174e-02, 3.7251e-03]],

         [[8.8818e-01, 4.5359e-05, 8.0884e-05,  ..., 7.3662e-03,
           3.7994e-02, 1.5358e-02]],

         [[5.8740e-01, 4.2796e-05, 6.1810e-05,  ..., 1.5955e-01,
           1.9180e-02, 3.2257e-02]],

         ...,

         [[3.4180e-01, 6.5756e-04, 7.1764e-04,  ..., 2.9240e-03,
           1.0925e-02, 1.6832e-03]],

         [[7.6172e-01, 7.1943e-05, 4.4048e-05,  ..., 1.2457e-01,
           7.4539e-03, 6.5346e-03]],

         [[1.6016e-01, 1.7285e-06, 6.5565e-06,  ..., 2.1992e-03,
           7.3486e-01, 8.5754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9824e-01, 6.3598e-05, 1.1116e-04,  ..., 1.4663e-04,
           7.8964e-04, 1.5961e-02]],

         [[5.5908e-01, 9.8348e-05, 9.3460e-05,  ..., 7.6416e-02,
           7.7881e-02, 1.6861e-02]],

         [[7.5342e-01, 4.5300e-06, 9.1791e-06,  ..., 1.6556e-03,
           3.8357e-03, 1.6022e-02]],

         ...,

         [[9.4922e-01, 4.9889e-05, 2.0802e-05,  ..., 2.7180e-03,
           4.6005e-03, 1.6909e-03]],

         [[3.8745e-01, 2.1100e-04, 6.7949e-05,  ..., 5.8441e-03,
           5.5962e-03, 5.1709e-01]],

         [[2.4194e-01, 3.5524e-04, 1.8680e-04,  ..., 2.3773e-02,
           4.4617e-02, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6182e-01, 1.7285e-06, 5.4240e-06,  ..., 6.6710e-04,
           7.2365e-03, 6.0840e-01]],

         [[6.2744e-01, 5.0688e-04, 9.1457e-04,  ..., 7.1945e-03,
           8.8215e-04, 1.7529e-03]],

         [[6.4502e-01, 6.0272e-04, 6.9714e-04,  ..., 1.6602e-02,
           3.6957e-02, 1.6571e-02]],

         ...,

         [[6.0498e-01, 1.3793e-04, 2.3556e-04,  ..., 3.8834e-03,
           3.0365e-03, 5.3482e-03]],

         [[2.2571e-01, 9.7215e-05, 5.9187e-05,  ..., 4.3259e-03,
           7.6904e-02, 2.5192e-02]],

         [[2.1545e-01, 3.0696e-05, 6.8605e-05,  ..., 3.2616e-03,
           6.0516e-02, 6.5918e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 5.8413e-05, 9.1851e-05,  ..., 3.7136e-03,
           7.6904e-02, 3.7720e-01]],

         [[3.3252e-01, 4.8161e-04, 4.8542e-04,  ..., 2.9282e-02,
           1.8542e-01, 4.1107e-02]],

         [[4.2432e-01, 2.1553e-04, 1.5771e-04,  ..., 2.1191e-03,
           4.4403e-03, 3.6526e-03]],

         ...,

         [[2.8488e-02, 1.3113e-06, 1.7285e-06,  ..., 8.5175e-05,
           5.1308e-03, 9.6191e-01]],

         [[8.4961e-01, 5.5671e-05, 4.8757e-05,  ..., 1.2856e-03,
           4.1733e-03, 2.8763e-02]],

         [[6.9141e-01, 2.3329e-04, 7.6056e-05,  ..., 1.1208e-02,
           4.7836e-03, 6.4659e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3516e-01, 4.3631e-04, 5.4741e-04,  ..., 8.8684e-02,
           1.3275e-02, 9.2010e-03]],

         [[7.7979e-01, 4.3869e-05, 1.6499e-04,  ..., 2.2858e-02,
           5.0812e-02, 3.9154e-02]],

         [[4.1699e-01, 1.6260e-04, 1.3793e-04,  ..., 1.2636e-03,
           4.6501e-03, 1.6068e-02]],

         ...,

         [[2.8027e-01, 5.9605e-08, 8.9407e-07,  ..., 4.0985e-02,
           6.4453e-01, 2.7740e-02]],

         [[3.8745e-01, 5.5361e-04, 7.1335e-04,  ..., 1.7059e-02,
           1.1269e-02, 4.8561e-03]],

         [[7.6514e-01, 1.3709e-06, 2.1458e-06,  ..., 5.0507e-02,
           9.0942e-02, 2.3468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1484e-01, 6.7890e-05, 1.9014e-05,  ..., 3.7365e-03,
           3.0708e-03, 7.8735e-03]],

         [[1.1322e-01, 4.3678e-04, 4.7469e-04,  ..., 6.2370e-03,
           1.7380e-02, 4.1428e-03]],

         [[3.9697e-01, 2.5558e-04, 2.5606e-04,  ..., 6.4819e-02,
           1.0394e-01, 1.5762e-02]],

         ...,

         [[3.2544e-01, 1.6093e-06, 1.4901e-06,  ..., 5.0049e-02,
           4.4873e-01, 1.7365e-02]],

         [[7.7734e-01, 1.7881e-07, 1.1921e-07,  ..., 2.0676e-02,
           8.9539e-02, 1.6113e-02]],

         [[4.0771e-01, 1.6332e-05, 3.4690e-05,  ..., 1.8295e-02,
           6.2378e-02, 7.2708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9773e-02, 4.1723e-07, 5.9605e-07,  ..., 8.4448e-04,
           7.0915e-03, 8.8818e-01]],

         [[1.7615e-01, 8.4043e-06, 2.0444e-05,  ..., 8.6670e-02,
           2.6245e-01, 3.5919e-02]],

         [[3.4058e-01, 1.3103e-03, 1.1044e-03,  ..., 2.2278e-02,
           1.9119e-02, 8.5678e-03]],

         ...,

         [[2.8198e-01, 5.0306e-05, 2.7180e-05,  ..., 8.5144e-02,
           2.1469e-02, 1.3817e-02]],

         [[6.1230e-01, 5.5432e-06, 8.1062e-06,  ..., 4.2267e-03,
           6.6338e-03, 1.3954e-02]],

         [[9.5459e-02, 8.5056e-05, 4.2081e-05,  ..., 5.3358e-04,
           3.5739e-04, 2.0027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0352e-01, 1.9133e-04, 8.8990e-05,  ..., 3.3997e-02,
           7.9498e-03, 3.3722e-02]],

         [[3.3008e-01, 4.7565e-04, 1.4567e-04,  ..., 7.6904e-02,
           3.1555e-02, 2.9770e-02]],

         [[8.6441e-03, 2.2173e-05, 2.5034e-06,  ..., 1.2238e-02,
           2.7252e-02, 3.5877e-03]],

         ...,

         [[7.0117e-01, 5.8830e-05, 2.2709e-05,  ..., 5.6915e-02,
           1.9287e-02, 1.0239e-02]],

         [[2.6416e-01, 2.1183e-04, 1.0926e-04,  ..., 1.3184e-01,
           7.0312e-02, 4.6631e-02]],

         [[8.9893e-01, 2.4176e-04, 6.8486e-05,  ..., 7.3357e-03,
           3.0384e-03, 9.0179e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3342e-01, 1.0586e-03, 1.7157e-03,  ..., 5.5161e-03,
           2.9755e-03, 3.0875e-04]],

         [[2.6489e-01, 4.4131e-04, 2.3580e-04,  ..., 2.8702e-02,
           7.1220e-03, 8.4610e-03]],

         [[2.7002e-01, 4.1127e-05, 6.1393e-05,  ..., 1.0162e-01,
           1.4502e-01, 5.9998e-02]],

         ...,

         [[7.3730e-01, 2.1100e-04, 1.0806e-04,  ..., 3.0632e-03,
           1.6479e-02, 2.5681e-02]],

         [[9.7461e-01, 1.7107e-05, 2.4676e-05,  ..., 3.3975e-06,
           4.6372e-05, 2.7084e-03]],

         [[7.8223e-01, 2.6870e-04, 3.2043e-04,  ..., 5.5046e-03,
           1.0902e-02, 4.9362e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0358e-01, 1.1039e-04, 8.6010e-05,  ..., 7.7400e-03,
           6.1829e-02, 8.2397e-03]],

         [[1.7688e-01, 4.3602e-03, 2.1076e-03,  ..., 1.8646e-02,
           1.6586e-02, 1.1757e-02]],

         [[6.2622e-02, 5.4240e-06, 6.1393e-06,  ..., 2.1637e-02,
           2.3071e-01, 5.2393e-01]],

         ...,

         [[2.6636e-01, 2.5094e-05, 1.9431e-05,  ..., 6.7566e-02,
           4.9438e-02, 5.7030e-03]],

         [[7.1777e-01, 2.5332e-05, 1.8597e-05,  ..., 5.4504e-02,
           1.4328e-02, 1.3046e-02]],

         [[2.4475e-01, 1.5104e-04, 1.1086e-04,  ..., 1.8143e-02,
           1.8323e-01, 2.6465e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3545e-01, 4.6432e-05, 2.1935e-05,  ..., 1.0910e-03,
           4.1161e-03, 3.0426e-02]],

         [[5.6274e-02, 3.9339e-05, 2.4617e-05,  ..., 2.6025e-01,
           1.0681e-01, 2.6169e-02]],

         [[1.9421e-01, 1.6212e-05, 1.8060e-05,  ..., 1.0509e-03,
           4.6112e-02, 7.3291e-01]],

         ...,

         [[1.5320e-01, 7.3195e-04, 2.1219e-04,  ..., 6.2370e-04,
           1.8549e-03, 8.1863e-03]],

         [[3.7964e-01, 4.4298e-04, 1.8978e-04,  ..., 4.1046e-03,
           5.4359e-03, 1.2016e-02]],

         [[3.9819e-01, 1.1635e-03, 1.2531e-03,  ..., 1.0910e-02,
           7.1526e-03, 1.2360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8933e-01, 1.2189e-04, 6.7592e-05,  ..., 8.5327e-02,
           5.6396e-02, 1.9333e-02]],

         [[9.6533e-01, 5.4836e-06, 5.9605e-07,  ..., 2.8191e-03,
           9.6560e-06, 1.2226e-03]],

         [[8.4961e-01, 4.9353e-05, 6.6757e-06,  ..., 1.4618e-02,
           1.7023e-04, 1.7872e-03]],

         ...,

         [[1.0632e-01, 1.9627e-03, 2.3670e-03,  ..., 7.0114e-03,
           5.3749e-03, 2.0237e-03]],

         [[6.4258e-01, 5.8889e-05, 6.5207e-05,  ..., 3.7384e-02,
           7.7171e-03, 1.4870e-02]],

         [[2.9614e-01, 1.1644e-03, 2.4452e-03,  ..., 9.7504e-03,
           2.2308e-02, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6450e-01, 1.8060e-04, 2.3377e-04,  ..., 1.0178e-02,
           2.9678e-02, 4.8920e-02]],

         [[2.7222e-01, 2.6417e-04, 5.3358e-04,  ..., 1.4771e-02,
           2.0996e-02, 2.5314e-02]],

         [[2.7490e-01, 7.6711e-05, 1.0693e-04,  ..., 1.7563e-02,
           5.3177e-03, 7.3242e-03]],

         ...,

         [[3.5889e-02, 2.2030e-04, 2.9635e-04,  ..., 3.2196e-02,
           2.5452e-02, 1.8341e-02]],

         [[5.5725e-02, 1.9479e-04, 9.9480e-05,  ..., 7.5817e-04,
           3.8743e-04, 8.0261e-03]],

         [[5.5908e-01, 3.5048e-04, 4.3273e-04,  ..., 3.4294e-03,
           2.7771e-03, 9.4681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4661e-01, 1.3292e-05, 1.1444e-05,  ..., 7.8278e-03,
           3.3150e-03, 8.8043e-03]],

         [[3.5034e-01, 1.7512e-04, 2.8205e-04,  ..., 1.9928e-02,
           8.0185e-03, 8.1787e-03]],

         [[2.5073e-01, 5.1022e-05, 1.1951e-04,  ..., 1.1818e-02,
           1.6443e-01, 3.8525e-01]],

         ...,

         [[5.2588e-01, 7.4446e-05, 1.2517e-04,  ..., 6.5857e-02,
           5.1666e-02, 3.6072e-02]],

         [[7.1777e-01, 3.0458e-05, 2.6226e-04,  ..., 2.2144e-03,
           1.1330e-02, 1.9882e-02]],

         [[2.4927e-01, 3.9935e-05, 8.2970e-05,  ..., 1.6537e-03,
           8.6594e-03, 3.2959e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0813e-01, 2.1040e-04, 1.7929e-04,  ..., 1.3992e-02,
           1.2947e-02, 2.9053e-02]],

         [[1.3623e-01, 1.7047e-04, 2.4748e-04,  ..., 7.5867e-02,
           2.8122e-02, 1.4702e-02]],

         [[4.3408e-01, 1.7595e-04, 1.3316e-04,  ..., 5.4207e-03,
           1.7166e-02, 6.4514e-02]],

         ...,

         [[4.6533e-01, 8.3447e-07, 1.4067e-05,  ..., 1.5588e-01,
           3.2983e-01, 2.6245e-02]],

         [[3.7817e-01, 4.1842e-05, 6.9261e-05,  ..., 1.1444e-01,
           1.7310e-01, 3.8330e-02]],

         [[3.5059e-01, 7.7486e-07, 9.6560e-06,  ..., 9.2224e-02,
           4.8682e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9520e-02, 7.9870e-06, 3.1590e-05,  ..., 4.4434e-02,
           5.0751e-02, 1.0803e-02]],

         [[1.6281e-02, 1.0848e-05, 1.4544e-05,  ..., 6.2132e-04,
           2.1912e-02, 9.3164e-01]],

         [[3.5583e-02, 4.1962e-05, 1.4591e-04,  ..., 3.0251e-03,
           6.8169e-03, 3.6049e-03]],

         ...,

         [[1.7627e-01, 3.1638e-04, 1.2674e-03,  ..., 1.8295e-02,
           3.4058e-02, 8.9874e-03]],

         [[2.9932e-01, 6.4969e-06, 8.1062e-06,  ..., 1.1768e-01,
           2.4524e-01, 3.8361e-02]],

         [[2.9395e-01, 1.1921e-07, 8.9407e-07,  ..., 1.0565e-01,
           4.6240e-01, 5.7892e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1055e-01, 1.9360e-04, 8.8835e-04,  ..., 7.5531e-03,
           5.7678e-03, 1.0529e-02]],

         [[9.8438e-01, 2.1040e-05, 6.2168e-05,  ..., 2.7704e-04,
           3.2640e-04, 8.8024e-04]],

         [[9.0186e-01, 1.7035e-04, 8.7857e-05,  ..., 3.4084e-03,
           1.1778e-03, 1.3351e-03]],

         ...,

         [[7.7881e-01, 3.9160e-05, 3.7909e-05,  ..., 3.4760e-02,
           8.2550e-03, 1.1826e-02]],

         [[3.5498e-01, 1.4901e-06, 4.7684e-07,  ..., 1.6308e-03,
           2.8629e-03, 1.0887e-02]],

         [[1.4478e-01, 1.1981e-05, 1.7881e-05,  ..., 5.8365e-03,
           5.1605e-02, 7.2363e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9669e-02, 7.5102e-06, 1.0729e-05,  ..., 1.6333e-01,
           1.8359e-01, 8.2764e-02]],

         [[7.0703e-01, 6.2168e-05, 4.1962e-05,  ..., 3.7323e-02,
           2.9770e-02, 4.1016e-02]],

         [[1.8665e-01, 7.8106e-04, 5.4312e-04,  ..., 4.3488e-02,
           2.2293e-02, 5.7793e-03]],

         ...,

         [[5.9814e-01, 1.6749e-05, 7.1406e-05,  ..., 1.2250e-01,
           2.5085e-02, 4.6196e-03]],

         [[6.3379e-01, 1.6809e-05, 4.8757e-05,  ..., 1.9287e-02,
           2.9648e-02, 2.9877e-02]],

         [[2.8320e-01, 7.5698e-06, 3.4392e-05,  ..., 1.8524e-02,
           1.0828e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5053e-02, 3.8147e-06, 4.7684e-06,  ..., 2.2717e-03,
           7.0534e-03, 1.0506e-02]],

         [[1.5710e-01, 1.6630e-05, 8.9407e-06,  ..., 7.8201e-03,
           8.9340e-03, 5.3436e-02]],

         [[2.3853e-01, 1.3375e-04, 4.1103e-04,  ..., 6.5186e-02,
           2.8488e-02, 7.2144e-02]],

         ...,

         [[6.1066e-02, 3.4571e-06, 4.5419e-05,  ..., 1.5198e-02,
           9.3079e-02, 1.5833e-01]],

         [[8.9172e-02, 3.8743e-06, 5.2512e-05,  ..., 1.4786e-02,
           9.0576e-02, 9.4971e-02]],

         [[9.4189e-01, 6.3777e-06, 1.4544e-05,  ..., 8.6746e-03,
           1.5945e-02, 4.6425e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 2.0862e-06, 1.1086e-05,  ..., 2.1164e-02,
           1.6266e-02, 6.8176e-02]],

         [[3.2422e-01, 2.0683e-05, 6.3837e-05,  ..., 9.2163e-02,
           2.5439e-01, 4.8584e-02]],

         [[8.8501e-02, 4.1723e-07, 1.9073e-06,  ..., 7.0312e-02,
           7.7344e-01, 4.9072e-02]],

         ...,

         [[6.2354e-01, 7.4148e-05, 1.6642e-04,  ..., 5.6725e-03,
           1.0597e-02, 9.8038e-03]],

         [[3.8184e-01, 2.1875e-05, 1.2100e-05,  ..., 3.9307e-02,
           1.0358e-01, 2.5375e-02]],

         [[7.0374e-02, 5.1141e-05, 2.1040e-04,  ..., 2.2491e-02,
           6.5613e-02, 6.6797e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 8.7440e-05, 1.6320e-04,  ..., 4.2786e-02,
           2.4658e-02, 3.2654e-02]],

         [[4.3481e-01, 3.3188e-04, 2.0695e-04,  ..., 1.2999e-03,
           1.3313e-03, 1.2726e-02]],

         [[3.0713e-01, 8.6010e-05, 2.0015e-04,  ..., 2.5253e-03,
           7.2212e-03, 5.0140e-02]],

         ...,

         [[8.2861e-01, 1.6212e-04, 2.3055e-04,  ..., 1.2985e-02,
           1.2985e-02, 9.6512e-03]],

         [[1.7407e-01, 7.4983e-05, 2.8753e-04,  ..., 4.9400e-03,
           3.7079e-02, 7.3291e-01]],

         [[5.0488e-01, 2.4974e-05, 6.8963e-05,  ..., 1.4107e-02,
           2.7466e-01, 1.2378e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1309e-01, 3.8862e-05, 5.5647e-04,  ..., 9.2697e-03,
           1.0880e-02, 1.5350e-02]],

         [[7.6843e-02, 6.4373e-06, 5.2273e-05,  ..., 1.4549e-02,
           4.2450e-02, 4.4464e-02]],

         [[2.3804e-01, 7.7903e-05, 5.6076e-04,  ..., 8.1299e-02,
           1.2158e-01, 1.2549e-01]],

         ...,

         [[8.6475e-01, 7.1943e-05, 3.5644e-04,  ..., 7.9956e-03,
           1.0590e-02, 2.4780e-02]],

         [[9.0039e-01, 8.9407e-07, 3.1090e-04,  ..., 2.5043e-03,
           3.0661e-04, 7.4158e-03]],

         [[5.4395e-01, 5.8651e-05, 1.1730e-04,  ..., 1.0834e-01,
           6.2943e-03, 1.4099e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3663e-05, 2.7180e-05, 1.7798e-01,  ..., 4.1199e-04,
           2.3254e-02, 1.0339e-01]],

         [[5.5134e-05, 1.7822e-05, 1.0384e-02,  ..., 1.0948e-03,
           2.9068e-03, 7.9041e-03]],

         [[2.9325e-04, 4.6730e-04, 1.0724e-01,  ..., 4.9744e-03,
           2.8519e-02, 5.6519e-02]],

         ...,

         [[3.9172e-04, 5.1618e-05, 1.9751e-01,  ..., 3.0289e-02,
           1.2854e-01, 1.7297e-01]],

         [[1.1021e-04, 5.7638e-05, 2.9224e-01,  ..., 2.9221e-03,
           6.4583e-03, 1.3779e-02]],

         [[4.1127e-06, 2.9206e-06, 2.5879e-01,  ..., 3.2845e-03,
           4.1199e-03, 3.8483e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3956e-05, 5.6624e-05, 2.8125e-01,  ..., 1.6203e-03,
           6.2103e-03, 5.2826e-02]],

         [[5.3287e-05, 1.9968e-05, 1.7651e-01,  ..., 5.4216e-04,
           3.8223e-03, 3.8452e-02]],

         [[6.8307e-05, 7.2718e-06, 3.1519e-01,  ..., 9.3508e-04,
           6.4659e-03, 4.8340e-02]],

         ...,

         [[1.9836e-04, 2.4676e-05, 2.3941e-02,  ..., 5.4359e-04,
           9.5034e-04, 9.9411e-03]],

         [[1.1027e-05, 5.9605e-06, 2.4927e-01,  ..., 1.1677e-04,
           3.5553e-03, 3.7628e-02]],

         [[4.8816e-05, 3.6001e-05, 2.5391e-01,  ..., 3.1161e-04,
           8.5602e-03, 3.2562e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8716e-05, 6.3777e-06, 1.0327e-01,  ..., 3.3545e-04,
           1.6146e-03, 1.4565e-02]],

         [[6.3062e-05, 1.3113e-06, 2.2290e-01,  ..., 3.0994e-05,
           3.5744e-03, 2.9465e-02]],

         [[5.1200e-05, 1.5795e-05, 2.3291e-01,  ..., 2.6941e-04,
           3.1452e-03, 2.4170e-02]],

         ...,

         [[5.1141e-05, 1.6928e-05, 1.4893e-01,  ..., 1.2245e-03,
           1.3115e-02, 4.2664e-02]],

         [[1.0073e-05, 1.1086e-05, 9.4971e-02,  ..., 2.8896e-04,
           2.7370e-03, 1.4801e-02]],

         [[4.0472e-05, 5.1856e-06, 1.5356e-01,  ..., 3.1233e-04,
           2.4242e-03, 1.7624e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6134e-05, 2.6822e-06, 1.1559e-02,  ..., 1.1284e-02,
           3.9795e-01, 2.2144e-01]],

         [[9.6560e-06, 4.9472e-05, 9.8633e-02,  ..., 1.3269e-01,
           4.4617e-02, 3.8330e-02]],

         [[3.8862e-05, 2.7871e-04, 3.1860e-01,  ..., 2.0161e-03,
           7.9250e-04, 7.7057e-04]],

         ...,

         [[5.9891e-04, 5.6386e-05, 1.8567e-01,  ..., 1.3237e-03,
           4.4365e-03, 1.4572e-02]],

         [[1.1988e-03, 9.5844e-05, 4.3945e-02,  ..., 1.8753e-02,
           2.4805e-01, 3.1348e-01]],

         [[1.8501e-04, 1.7166e-05, 4.2450e-02,  ..., 1.1650e-02,
           1.1816e-01, 1.4758e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1324e-03, 4.3726e-04, 9.3689e-02,  ..., 1.0971e-02,
           5.9082e-02, 9.0088e-02]],

         [[7.8630e-04, 1.2426e-03, 2.4292e-01,  ..., 9.9106e-03,
           1.3977e-02, 1.4816e-02]],

         [[2.9640e-03, 3.4666e-04, 1.4819e-01,  ..., 4.9706e-03,
           5.3680e-02, 6.8665e-02]],

         ...,

         [[3.8052e-04, 8.2314e-05, 2.4292e-02,  ..., 6.1417e-03,
           7.3828e-01, 1.3550e-01]],

         [[1.0805e-03, 3.2520e-03, 2.9053e-01,  ..., 1.0157e-03,
           6.0768e-03, 3.1097e-02]],

         [[4.1103e-04, 3.3021e-04, 2.1973e-01,  ..., 2.8595e-02,
           2.9739e-02, 3.4760e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[3.8123e-04, 1.7452e-04, 6.9695e-03,  ..., 1.1249e-01,
           5.3662e-01, 7.9773e-02]],

         [[3.8147e-04, 5.5504e-04, 6.7635e-03,  ..., 1.4320e-02,
           1.0577e-01, 7.8174e-01]],

         [[1.0014e-05, 2.1279e-05, 7.4208e-05,  ..., 1.5712e-04,
           2.4567e-03, 9.9121e-01]],

         ...,

         [[2.2054e-06, 6.0797e-06, 7.3731e-05,  ..., 2.0039e-04,
           4.0245e-03, 9.8535e-01]],

         [[3.4237e-03, 2.5043e-03, 2.3518e-03,  ..., 3.6804e-02,
           7.7881e-02, 2.6917e-02]],

         [[1.9806e-02, 5.3291e-03, 1.6251e-03,  ..., 1.9196e-02,
           3.1647e-02, 5.7281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0041e-04, 8.8310e-04, 3.6602e-03,  ..., 3.3661e-02,
           7.5867e-02, 6.7627e-01]],

         [[2.3842e-07, 2.9802e-07, 1.1921e-06,  ..., 1.0779e-01,
           7.7197e-01, 1.1475e-01]],

         [[9.3174e-04, 2.6970e-03, 1.8417e-02,  ..., 6.7383e-02,
           4.2175e-02, 1.8323e-01]],

         ...,

         [[2.6684e-03, 6.7635e-03, 1.2245e-02,  ..., 6.3515e-03,
           2.8915e-02, 4.1846e-01]],

         [[2.6913e-03, 1.0918e-02, 3.9612e-02,  ..., 2.7649e-02,
           3.1586e-02, 2.6382e-02]],

         [[5.0087e-03, 1.2108e-02, 1.9043e-02,  ..., 2.0007e-01,
           4.9774e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0797, 0.0073, 0.0407,  ..., 0.0317, 0.0271, 0.0395]],

         [[0.0009, 0.0004, 0.0023,  ..., 0.0082, 0.0031, 0.0075]],

         [[0.0091, 0.0022, 0.0018,  ..., 0.0525, 0.0319, 0.0242]],

         ...,

         [[0.0175, 0.0009, 0.0030,  ..., 0.0721, 0.1962, 0.3416]],

         [[0.0207, 0.0023, 0.0026,  ..., 0.0334, 0.0279, 0.0581]],

         [[0.0115, 0.0041, 0.0156,  ..., 0.0721, 0.2273, 0.3181]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[6.9189e-01, 1.6093e-06, 3.2365e-05,  ..., 9.1858e-03,
           2.7786e-02, 6.3858e-03]],

         [[8.4766e-01, 2.9802e-06, 7.8499e-05,  ..., 5.4283e-03,
           5.3444e-03, 1.3863e-02]],

         [[9.6875e-01, 2.1458e-06, 9.8348e-06,  ..., 3.0842e-03,
           1.3208e-03, 4.2801e-03]],

         ...,

         [[3.5156e-01, 2.0087e-04, 1.2732e-03,  ..., 3.9935e-05,
           2.5392e-05, 9.3307e-03]],

         [[7.3633e-01, 1.7548e-04, 4.4608e-04,  ..., 6.4316e-03,
           2.6932e-03, 5.2299e-03]],

         [[2.7832e-01, 1.6439e-04, 3.0708e-04,  ..., 3.1616e-02,
           2.9922e-02, 1.1360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1895e-01, 1.6809e-05, 5.9605e-07,  ..., 1.9445e-03,
           4.1723e-04, 2.1225e-02]],

         [[8.6328e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8275e-02,
           5.8380e-02, 4.9011e-02]],

         [[9.1846e-01, 3.4308e-04, 1.0836e-04,  ..., 5.8136e-03,
           1.0818e-02, 4.9133e-03]],

         ...,

         [[1.7554e-01, 6.1703e-04, 2.3746e-04,  ..., 1.2396e-01,
           1.5015e-01, 1.9760e-02]],

         [[1.6113e-01, 1.4229e-03, 4.4441e-04,  ..., 2.5055e-02,
           7.5455e-03, 7.0190e-03]],

         [[4.9121e-01, 1.1736e-04, 3.5143e-04,  ..., 2.5833e-02,
           3.5370e-02, 1.9470e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7505e-01, 2.7704e-04, 9.3937e-05,  ..., 9.0027e-02,
           5.2521e-02, 1.1346e-01]],

         [[6.6406e-01, 1.2374e-04, 4.1783e-05,  ..., 6.8237e-02,
           8.1848e-02, 9.5825e-02]],

         [[8.0811e-01, 8.3447e-05, 2.1577e-05,  ..., 1.0361e-02,
           1.5808e-02, 4.0588e-02]],

         ...,

         [[9.0527e-01, 3.4988e-05, 5.9605e-07,  ..., 4.7684e-05,
           5.0068e-06, 8.9905e-02]],

         [[8.7109e-01, 4.3678e-04, 6.7949e-06,  ..., 2.3246e-06,
           1.7881e-07, 3.0640e-02]],

         [[9.4678e-01, 3.3951e-04, 1.0133e-06,  ..., 1.4901e-06,
           5.9605e-08, 1.2749e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9380e-01, 1.6212e-05, 7.0333e-06,  ..., 7.4036e-02,
           2.6709e-01, 1.8042e-01]],

         [[9.0088e-01, 2.8253e-05, 6.4373e-06,  ..., 1.8358e-04,
           2.0492e-04, 3.7659e-02]],

         [[7.1240e-01, 9.1195e-05, 2.2948e-05,  ..., 1.6804e-03,
           6.7377e-04, 5.3375e-02]],

         ...,

         [[6.8896e-01, 2.8777e-04, 2.6107e-05,  ..., 3.0594e-03,
           5.8327e-03, 2.1500e-02]],

         [[7.5293e-01, 1.3599e-03, 1.1909e-04,  ..., 7.3318e-03,
           1.0315e-02, 8.5815e-02]],

         [[9.3799e-01, 1.1265e-04, 4.9412e-05,  ..., 1.1263e-03,
           2.8210e-03, 1.3809e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1260e-01, 2.7239e-05, 4.2856e-05,  ..., 6.0425e-03,
           2.4796e-03, 8.0948e-03]],

         [[9.6094e-01, 1.8120e-05, 7.5102e-06,  ..., 7.0524e-04,
           6.6805e-04, 1.1612e-02]],

         [[9.8486e-01, 3.4571e-06, 1.6689e-06,  ..., 7.3576e-04,
           1.8902e-03, 6.5079e-03]],

         ...,

         [[8.1787e-01, 1.6010e-04, 1.1146e-05,  ..., 8.4229e-03,
           7.8201e-03, 2.1271e-02]],

         [[8.9258e-01, 2.2078e-04, 4.1306e-05,  ..., 2.0294e-03,
           7.6151e-04, 1.2581e-02]],

         [[9.6973e-01, 9.0003e-06, 6.3777e-06,  ..., 1.2074e-03,
           5.3368e-03, 6.6948e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4741e-01, 9.2888e-04, 4.5085e-04,  ..., 4.9162e-04,
           2.3441e-03, 2.5803e-02]],

         [[9.3652e-01, 2.5213e-05, 5.2154e-05,  ..., 1.0395e-03,
           2.1248e-03, 1.3943e-03]],

         [[9.3066e-01, 3.1650e-05, 1.0252e-05,  ..., 1.5092e-04,
           1.8682e-03, 3.0685e-02]],

         ...,

         [[9.6338e-01, 1.5855e-05, 1.6153e-05,  ..., 2.5272e-04,
           1.6375e-03, 1.2520e-02]],

         [[9.0869e-01, 2.3842e-06, 1.1086e-05,  ..., 1.3199e-02,
           4.3640e-02, 3.5191e-03]],

         [[8.4961e-01, 3.5763e-06, 6.1393e-06,  ..., 5.3978e-04,
           4.3678e-03, 1.0321e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0999e-01, 2.1839e-04, 7.7438e-04,  ..., 3.4515e-02,
           1.8204e-02, 1.3763e-02]],

         [[9.1406e-01, 2.8849e-05, 4.0710e-05,  ..., 3.8261e-03,
           1.1536e-02, 1.4526e-02]],

         [[7.1436e-01, 2.9981e-05, 5.3048e-05,  ..., 1.0185e-02,
           1.3879e-01, 4.9561e-02]],

         ...,

         [[7.0898e-01, 1.6093e-04, 1.7536e-04,  ..., 2.2583e-03,
           1.1148e-03, 9.9335e-03]],

         [[7.8906e-01, 1.0824e-04, 4.3392e-05,  ..., 4.7016e-04,
           2.0866e-03, 5.9082e-02]],

         [[7.5293e-01, 1.4663e-05, 2.3782e-05,  ..., 7.7942e-02,
           4.9164e-02, 1.6479e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 7.3612e-05, 8.1062e-06,  ..., 3.5167e-06,
           4.2915e-06, 1.3733e-02]],

         [[8.9160e-01, 7.0989e-05, 2.1029e-04,  ..., 5.0545e-03,
           5.6839e-03, 8.0643e-03]],

         [[8.9111e-01, 1.7285e-05, 5.5432e-06,  ..., 4.0460e-04,
           2.1327e-04, 1.9531e-02]],

         ...,

         [[8.1006e-01, 3.2973e-04, 1.5819e-04,  ..., 1.7071e-03,
           2.2354e-03, 2.7557e-02]],

         [[7.3389e-01, 8.8632e-05, 1.0002e-04,  ..., 4.8943e-03,
           3.5004e-02, 1.0437e-02]],

         [[3.6743e-01, 1.0185e-03, 1.8835e-04,  ..., 1.1765e-02,
           1.9394e-02, 4.4586e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 7.7486e-06, 6.6757e-06,  ..., 4.6110e-04,
           1.9045e-03, 5.8057e-01]],

         [[5.1123e-01, 5.1069e-04, 7.5674e-04,  ..., 9.0742e-04,
           1.4334e-03, 4.5197e-02]],

         [[7.9688e-01, 2.4772e-04, 4.8232e-04,  ..., 6.7635e-03,
           6.3286e-03, 1.7441e-02]],

         ...,

         [[7.3584e-01, 4.8041e-05, 9.0420e-05,  ..., 5.6505e-04,
           2.6741e-03, 1.5478e-03]],

         [[7.9297e-01, 5.2810e-05, 3.6001e-05,  ..., 1.2001e-02,
           5.7587e-02, 1.9913e-02]],

         [[2.7319e-01, 1.4901e-04, 2.5845e-04,  ..., 1.7838e-02,
           3.5370e-02, 5.5469e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1396e-01, 6.6042e-05, 4.8518e-05,  ..., 3.1624e-03,
           2.0203e-02, 5.6494e-01]],

         [[7.4072e-01, 8.7738e-05, 8.2076e-05,  ..., 1.0361e-02,
           8.3435e-02, 8.8654e-03]],

         [[2.2693e-01, 2.8038e-04, 1.3936e-04,  ..., 1.3485e-03,
           3.4904e-03, 5.8838e-02]],

         ...,

         [[4.1870e-02, 5.0664e-06, 3.3975e-06,  ..., 4.7112e-04,
           5.0621e-03, 9.4043e-01]],

         [[8.1885e-01, 1.9610e-05, 3.2842e-05,  ..., 2.4052e-03,
           9.6817e-03, 2.6001e-02]],

         [[8.4717e-01, 2.2352e-04, 4.5776e-05,  ..., 1.8272e-03,
           2.6520e-02, 1.0246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2129e-01, 7.5054e-04, 5.6648e-04,  ..., 3.0327e-03,
           3.0098e-03, 1.6460e-03]],

         [[7.8223e-01, 5.3346e-05, 1.8036e-04,  ..., 4.7546e-02,
           6.7932e-02, 1.4771e-02]],

         [[2.8101e-01, 8.5652e-05, 7.5877e-05,  ..., 3.3879e-04,
           8.9598e-04, 2.3300e-02]],

         ...,

         [[6.4502e-01, 0.0000e+00, 1.7881e-07,  ..., 3.2349e-02,
           2.7905e-01, 3.6072e-02]],

         [[3.4082e-01, 3.7146e-04, 7.9679e-04,  ..., 6.0272e-03,
           4.1847e-03, 4.9973e-04]],

         [[9.2969e-01, 5.9605e-08, 5.9605e-08,  ..., 9.2936e-04,
           8.3389e-03, 5.8167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 5.1618e-05, 1.2636e-05,  ..., 1.7643e-03,
           5.5618e-03, 1.7624e-02]],

         [[1.2408e-01, 3.7169e-04, 4.6253e-04,  ..., 4.9744e-02,
           3.6896e-02, 2.9984e-03]],

         [[4.0381e-01, 5.7161e-05, 8.0884e-05,  ..., 3.7476e-02,
           2.7539e-01, 3.5522e-02]],

         ...,

         [[6.8701e-01, 4.0531e-06, 3.0398e-06,  ..., 8.7708e-02,
           5.3528e-02, 7.0068e-02]],

         [[6.5820e-01, 2.9802e-07, 3.5763e-07,  ..., 2.8553e-03,
           2.6108e-02, 3.0591e-01]],

         [[6.8701e-01, 1.6093e-05, 2.3603e-05,  ..., 5.8868e-02,
           5.7068e-02, 6.5552e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0472e-02, 2.3842e-07, 2.3842e-07,  ..., 1.0723e-04,
           8.3160e-04, 9.6680e-01]],

         [[7.0740e-02, 8.2850e-06, 1.4961e-05,  ..., 2.2046e-01,
           3.0493e-01, 5.7983e-02]],

         [[2.3425e-01, 4.5848e-04, 9.4461e-04,  ..., 1.8890e-02,
           1.7105e-02, 6.0028e-02]],

         ...,

         [[5.2344e-01, 4.4286e-05, 4.6074e-05,  ..., 2.6398e-03,
           8.3694e-03, 1.2573e-02]],

         [[6.4258e-01, 1.2398e-05, 9.3579e-06,  ..., 3.4904e-03,
           5.2376e-03, 2.4429e-02]],

         [[9.2236e-01, 2.2650e-05, 1.3351e-05,  ..., 1.5378e-04,
           3.8815e-04, 5.0735e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4072e-01, 2.4199e-05, 2.0504e-05,  ..., 1.5507e-03,
           1.1581e-02, 1.5283e-01]],

         [[1.5283e-01, 3.7217e-04, 6.5207e-05,  ..., 2.1076e-03,
           2.2995e-02, 1.0028e-01]],

         [[2.5415e-01, 2.2650e-05, 1.6391e-05,  ..., 9.7733e-03,
           1.5442e-02, 8.9539e-02]],

         ...,

         [[8.7646e-01, 3.6657e-05, 2.7955e-05,  ..., 2.6512e-03,
           1.8225e-03, 1.0735e-02]],

         [[7.0996e-01, 8.0049e-05, 3.9995e-05,  ..., 1.1208e-02,
           7.0251e-02, 6.7566e-02]],

         [[6.2744e-01, 1.4722e-05, 7.6294e-06,  ..., 6.8331e-04,
           4.6043e-03, 1.8762e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9531e-01, 7.1716e-04, 7.9107e-04,  ..., 1.8494e-02,
           1.2413e-02, 1.8280e-02]],

         [[2.7100e-01, 6.2227e-05, 4.6611e-05,  ..., 8.1396e-04,
           9.8724e-03, 4.4586e-02]],

         [[6.8164e-01, 2.8789e-05, 8.1360e-05,  ..., 1.7426e-02,
           7.5256e-02, 9.0454e-02]],

         ...,

         [[4.9194e-01, 3.8815e-04, 6.1631e-05,  ..., 9.9087e-04,
           3.8776e-03, 8.0627e-02]],

         [[6.2061e-01, 1.0786e-03, 5.9748e-04,  ..., 2.5630e-06,
           1.1981e-05, 1.4999e-02]],

         [[3.9844e-01, 2.2602e-04, 1.7059e-04,  ..., 1.0735e-02,
           1.0025e-02, 1.2016e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.4971e-02, 1.4281e-04, 1.0204e-04,  ..., 9.2041e-02,
           1.5173e-01, 7.9803e-03]],

         [[3.7012e-01, 1.6904e-04, 1.1688e-04,  ..., 5.3864e-03,
           5.5023e-02, 3.6652e-02]],

         [[2.9370e-01, 3.0696e-05, 1.6212e-05,  ..., 8.2703e-03,
           1.2927e-01, 4.1431e-01]],

         ...,

         [[5.6201e-01, 1.2577e-05, 2.0206e-05,  ..., 2.8320e-02,
           8.3557e-02, 8.3557e-02]],

         [[8.2031e-01, 1.1861e-05, 1.7643e-05,  ..., 2.6321e-03,
           3.3531e-03, 9.4971e-02]],

         [[5.4980e-01, 1.1122e-04, 1.0407e-04,  ..., 8.3160e-03,
           4.3732e-02, 2.3462e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.1128e-04, 1.9968e-05,  ..., 3.0255e-04,
           6.8855e-03, 3.4973e-02]],

         [[1.7480e-01, 4.6372e-05, 3.3438e-05,  ..., 8.1299e-02,
           2.0764e-01, 1.1829e-01]],

         [[3.1836e-01, 2.2411e-05, 1.0252e-05,  ..., 2.4068e-04,
           9.0790e-04, 6.6357e-01]],

         ...,

         [[6.4148e-02, 1.0592e-04, 3.2306e-05,  ..., 1.3232e-05,
           3.7313e-05, 2.0020e-02]],

         [[2.7905e-01, 3.5191e-04, 1.6367e-04,  ..., 3.6964e-03,
           1.4114e-02, 8.5602e-03]],

         [[3.1738e-01, 2.9397e-04, 1.3638e-04,  ..., 5.2071e-03,
           2.6047e-02, 2.0126e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8696e-01, 1.1867e-04, 7.5102e-05,  ..., 7.1449e-03,
           8.7509e-03, 3.9520e-02]],

         [[9.7461e-01, 1.0532e-04, 4.4703e-06,  ..., 1.2517e-06,
           4.4703e-06, 7.2956e-05]],

         [[9.2822e-01, 6.0034e-04, 1.0931e-04,  ..., 1.8883e-04,
           5.8711e-05, 1.4172e-03]],

         ...,

         [[1.8787e-01, 5.1165e-04, 9.3365e-04,  ..., 9.3536e-03,
           6.8436e-03, 3.5309e-02]],

         [[8.3398e-01, 2.7061e-05, 2.9266e-05,  ..., 1.0891e-03,
           1.6613e-03, 7.7576e-02]],

         [[1.7285e-01, 1.0195e-03, 1.2989e-03,  ..., 7.9575e-03,
           8.7357e-03, 3.2227e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0010e-01, 1.7631e-04, 2.2995e-04,  ..., 5.5275e-03,
           1.6251e-02, 2.5955e-02]],

         [[3.7695e-01, 1.5736e-04, 2.9874e-04,  ..., 3.8433e-03,
           6.9046e-03, 2.0935e-02]],

         [[4.9048e-01, 2.3568e-04, 2.8658e-04,  ..., 6.5088e-04,
           1.7824e-03, 3.1830e-02]],

         ...,

         [[8.4686e-03, 4.2379e-05, 2.5690e-05,  ..., 5.2185e-03,
           3.9764e-02, 1.2909e-02]],

         [[3.4180e-03, 4.1127e-06, 1.0729e-06,  ..., 7.9870e-06,
           8.1658e-05, 1.0696e-02]],

         [[3.4058e-01, 1.4222e-04, 9.6202e-05,  ..., 5.9891e-04,
           2.6016e-03, 3.1677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4004e-01, 2.9027e-05, 4.9114e-05,  ..., 1.6527e-03,
           1.0445e-02, 6.2012e-02]],

         [[4.8560e-01, 1.6105e-04, 2.0599e-04,  ..., 2.2018e-02,
           6.2225e-02, 6.4583e-03]],

         [[2.6562e-01, 4.5002e-05, 1.2183e-04,  ..., 1.1609e-01,
           1.3049e-01, 2.8271e-01]],

         ...,

         [[5.7910e-01, 4.0483e-04, 3.9697e-04,  ..., 3.2166e-02,
           2.9282e-02, 5.9174e-02]],

         [[1.8274e-01, 4.5538e-05, 3.3855e-05,  ..., 1.5414e-04,
           3.9053e-04, 1.7712e-01]],

         [[4.6704e-01, 1.8752e-04, 1.9503e-04,  ..., 8.6212e-03,
           1.6235e-02, 3.6865e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5742e-01, 2.0790e-04, 2.0957e-04,  ..., 7.2746e-03,
           2.9800e-02, 8.8928e-02]],

         [[2.8003e-01, 9.6560e-05, 1.7822e-04,  ..., 5.5008e-03,
           3.1921e-02, 6.5979e-02]],

         [[5.0195e-01, 2.8205e-04, 3.0041e-04,  ..., 1.0338e-02,
           8.0078e-02, 3.0396e-02]],

         ...,

         [[7.2998e-01, 5.9605e-08, 5.3644e-07,  ..., 3.1586e-02,
           1.9043e-01, 2.8748e-02]],

         [[3.4277e-01, 3.9399e-05, 3.0935e-05,  ..., 9.2957e-02,
           2.2302e-01, 7.9529e-02]],

         [[5.9033e-01, 4.7684e-07, 3.1590e-06,  ..., 1.9730e-02,
           3.1104e-01, 6.1249e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 1.8656e-05, 1.5259e-05,  ..., 4.7089e-02,
           1.6431e-01, 1.6266e-02]],

         [[1.2964e-01, 1.6272e-05, 1.8835e-05,  ..., 3.9749e-03,
           1.5717e-02, 7.6953e-01]],

         [[1.8787e-01, 1.2267e-04, 2.2209e-04,  ..., 3.1525e-02,
           6.2439e-02, 4.8714e-03]],

         ...,

         [[1.6968e-01, 1.0812e-04, 3.0494e-04,  ..., 1.0492e-01,
           4.2389e-02, 3.1616e-02]],

         [[5.4736e-01, 1.4663e-05, 8.7023e-06,  ..., 1.0126e-01,
           2.1851e-01, 4.2358e-02]],

         [[3.3276e-01, 0.0000e+00, 1.7881e-07,  ..., 8.2520e-02,
           4.5483e-01, 7.3975e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4229e-01, 7.7069e-05, 2.1052e-04,  ..., 2.2488e-03,
           3.1586e-03, 3.1036e-02]],

         [[9.8926e-01, 1.3828e-05, 1.7166e-05,  ..., 4.3273e-05,
           2.5570e-05, 3.0947e-04]],

         [[9.5801e-01, 1.2106e-04, 2.1935e-05,  ..., 5.2166e-04,
           3.8457e-04, 2.6913e-03]],

         ...,

         [[8.4033e-01, 5.4240e-05, 5.7757e-05,  ..., 3.6564e-03,
           2.2400e-02, 1.0522e-01]],

         [[9.1406e-01, 8.5235e-06, 3.0994e-06,  ..., 7.4148e-04,
           1.6069e-03, 6.4545e-03]],

         [[4.5825e-01, 9.1791e-05, 1.1879e-04,  ..., 4.5624e-03,
           1.6434e-02, 4.5825e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4023e-01, 1.5259e-05, 2.7180e-05,  ..., 1.2366e-01,
           2.4207e-01, 7.6477e-02]],

         [[8.1055e-01, 1.8179e-05, 9.5367e-06,  ..., 7.8201e-03,
           2.1606e-02, 1.2482e-01]],

         [[3.1860e-01, 1.8954e-04, 2.3341e-04,  ..., 5.2032e-02,
           2.9068e-02, 2.2778e-01]],

         ...,

         [[8.2959e-01, 5.3644e-06, 1.7107e-05,  ..., 3.4515e-02,
           5.6458e-02, 3.0457e-02]],

         [[8.4033e-01, 2.1696e-05, 4.6492e-05,  ..., 5.4054e-03,
           6.8855e-03, 5.0079e-02]],

         [[4.8291e-01, 2.6703e-05, 3.9697e-05,  ..., 4.1229e-02,
           9.3628e-02, 1.0364e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6499e-01, 4.7028e-05, 2.7657e-05,  ..., 9.6512e-03,
           6.7566e-02, 5.1392e-02]],

         [[3.4692e-01, 3.9577e-05, 1.7822e-05,  ..., 1.7433e-03,
           8.6823e-03, 6.2683e-02]],

         [[5.1221e-01, 1.2493e-04, 2.2149e-04,  ..., 1.2238e-02,
           3.4576e-02, 9.0393e-02]],

         ...,

         [[3.9337e-02, 1.3113e-06, 2.1100e-05,  ..., 1.2903e-01,
           2.1106e-01, 1.8738e-02]],

         [[4.9780e-01, 1.1086e-05, 6.8009e-05,  ..., 5.0079e-02,
           9.5032e-02, 2.6531e-03]],

         [[5.8252e-01, 4.9829e-05, 4.9829e-05,  ..., 1.2793e-01,
           1.1835e-01, 5.8594e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0088e-01, 5.3644e-07, 5.3644e-07,  ..., 1.1070e-02,
           5.5542e-02, 5.0488e-01]],

         [[7.2070e-01, 7.8678e-06, 2.0981e-05,  ..., 2.6352e-02,
           7.0801e-02, 2.7893e-02]],

         [[4.2798e-01, 2.9802e-07, 7.7486e-07,  ..., 5.0720e-02,
           2.7856e-01, 2.2473e-01]],

         ...,

         [[8.9453e-01, 1.4341e-04, 1.9038e-04,  ..., 1.0977e-03,
           3.2139e-03, 2.8748e-02]],

         [[8.4814e-01, 1.5199e-05, 8.3447e-06,  ..., 9.2392e-03,
           3.6682e-02, 1.9485e-02]],

         [[1.7151e-01, 1.5736e-04, 2.2852e-04,  ..., 1.3756e-02,
           3.1982e-02, 6.1768e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 3.0518e-05, 8.1301e-05,  ..., 2.8214e-02,
           3.4302e-02, 1.6052e-01]],

         [[3.9014e-01, 4.0555e-04, 3.0541e-04,  ..., 7.5912e-04,
           4.8180e-03, 2.2705e-02]],

         [[2.3730e-01, 4.4942e-05, 1.4997e-04,  ..., 1.4534e-02,
           8.4595e-02, 8.3313e-02]],

         ...,

         [[8.7695e-01, 2.0218e-04, 5.3692e-04,  ..., 1.6571e-02,
           2.4490e-02, 1.6830e-02]],

         [[2.2180e-01, 1.9300e-04, 2.6798e-04,  ..., 5.5542e-03,
           1.7105e-02, 6.7236e-01]],

         [[6.1572e-01, 2.4498e-05, 6.9201e-05,  ..., 7.5867e-02,
           2.1606e-01, 6.1920e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0632e-01, 1.1623e-05, 1.4126e-04,  ..., 6.0364e-02,
           7.1240e-01, 3.8208e-02]],

         [[1.7969e-01, 1.6332e-05, 8.0228e-05,  ..., 1.1131e-02,
           8.9294e-02, 4.5074e-02]],

         [[4.9487e-01, 2.9850e-04, 1.8482e-03,  ..., 3.9581e-02,
           1.1255e-01, 6.9885e-02]],

         ...,

         [[8.4375e-01, 9.5427e-05, 2.4247e-04,  ..., 3.6926e-02,
           2.9327e-02, 8.0490e-03]],

         [[9.6094e-01, 2.2650e-06, 1.4830e-04,  ..., 3.4308e-04,
           1.0262e-03, 5.6343e-03]],

         [[7.7637e-01, 5.5373e-05, 3.8326e-05,  ..., 1.1818e-02,
           6.1417e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4809e-05, 5.1916e-05, 1.8958e-01,  ..., 3.3966e-02,
           1.7395e-01, 1.4671e-02]],

         [[9.8050e-05, 4.6730e-05, 2.9556e-02,  ..., 1.1353e-02,
           3.2227e-02, 1.2549e-01]],

         [[1.4770e-04, 1.9062e-04, 1.0907e-01,  ..., 4.6539e-02,
           7.0984e-02, 1.2659e-01]],

         ...,

         [[3.0065e-04, 1.5557e-05, 6.7505e-02,  ..., 3.7292e-02,
           7.0361e-01, 5.8655e-02]],

         [[5.0545e-04, 2.3592e-04, 3.0737e-01,  ..., 4.1504e-03,
           5.2452e-03, 9.2957e-02]],

         [[7.1585e-05, 2.2757e-04, 2.9639e-01,  ..., 1.3535e-02,
           2.2675e-02, 2.7130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6606e-04, 2.2435e-04, 2.6196e-01,  ..., 1.6235e-02,
           6.6223e-02, 8.1635e-03]],

         [[1.4269e-04, 1.8394e-04, 2.7563e-01,  ..., 1.0689e-02,
           3.0685e-02, 9.5825e-03]],

         [[3.3832e-04, 1.4782e-04, 3.3252e-01,  ..., 7.5798e-03,
           2.3163e-02, 1.4467e-03]],

         ...,

         [[6.5899e-04, 1.6057e-04, 1.4453e-01,  ..., 3.1815e-03,
           1.8234e-02, 2.5366e-01]],

         [[1.3685e-04, 1.0717e-04, 3.2764e-01,  ..., 8.3618e-03,
           2.8412e-02, 4.2229e-03]],

         [[3.5405e-05, 9.8705e-05, 3.0664e-01,  ..., 7.0152e-03,
           1.0490e-02, 9.6464e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6192e-04, 2.0516e-04, 2.8003e-01,  ..., 7.8812e-03,
           2.0004e-02, 8.5974e-04]],

         [[3.7813e-04, 1.4484e-05, 2.9736e-01,  ..., 6.0768e-03,
           1.3275e-02, 2.3483e-02]],

         [[8.0681e-04, 4.2844e-04, 3.2812e-01,  ..., 6.9427e-03,
           1.5465e-02, 7.5378e-03]],

         ...,

         [[4.8995e-05, 2.3484e-05, 1.7273e-01,  ..., 1.6327e-02,
           3.2715e-02, 3.2788e-01]],

         [[1.1295e-04, 2.0814e-04, 2.1606e-01,  ..., 1.0971e-02,
           1.5961e-02, 1.6861e-02]],

         [[4.1032e-04, 7.3135e-05, 2.9395e-01,  ..., 6.2447e-03,
           1.5213e-02, 1.8066e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6473e-05, 6.4373e-06, 2.0966e-02,  ..., 6.8176e-02,
           5.1562e-01, 2.6343e-01]],

         [[1.6451e-05, 5.2214e-05, 2.0813e-01,  ..., 3.1052e-02,
           4.5380e-02, 1.9821e-02]],

         [[5.5790e-05, 3.1662e-04, 2.3218e-01,  ..., 1.4839e-03,
           1.3628e-03, 9.7504e-03]],

         ...,

         [[2.2411e-04, 1.4067e-05, 2.8564e-01,  ..., 6.3241e-05,
           2.9397e-04, 1.1873e-03]],

         [[1.0118e-03, 9.0003e-05, 5.3772e-02,  ..., 6.2866e-02,
           1.8262e-01, 4.2969e-01]],

         [[1.6546e-04, 1.2398e-05, 5.7495e-02,  ..., 5.6824e-02,
           1.5259e-01, 3.9746e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2173e-04, 5.5885e-04, 2.0740e-01,  ..., 6.2866e-03,
           1.0612e-02, 1.5649e-01]],

         [[2.2829e-04, 3.0518e-04, 2.8589e-01,  ..., 6.7253e-03,
           9.9411e-03, 5.7861e-02]],

         [[1.2398e-03, 5.6791e-04, 1.7395e-01,  ..., 7.1106e-02,
           1.9702e-01, 8.3435e-02]],

         ...,

         [[1.0605e-03, 1.7309e-04, 1.4783e-01,  ..., 4.5441e-02,
           1.3354e-01, 2.7173e-01]],

         [[2.0409e-04, 6.2370e-04, 2.9883e-01,  ..., 4.9820e-03,
           8.1726e-02, 5.3589e-02]],

         [[2.3901e-04, 5.1677e-05, 1.3281e-01,  ..., 1.1908e-01,
           1.7322e-01, 6.4758e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7351e-03, 1.1053e-03, 1.7288e-02,  ..., 1.6406e-01,
           2.7905e-01, 1.9788e-01]],

         [[2.5928e-05, 2.4629e-04, 1.8196e-03,  ..., 1.1864e-02,
           1.9562e-02, 9.4238e-01]],

         [[2.3842e-07, 3.5346e-05, 1.0133e-05,  ..., 1.0133e-05,
           4.3058e-04, 9.9951e-01]],

         ...,

         [[0.0000e+00, 1.7881e-07, 6.1393e-06,  ..., 2.7478e-05,
           9.1028e-04, 9.9854e-01]],

         [[3.2711e-03, 2.8858e-03, 3.9444e-03,  ..., 8.7036e-02,
           4.2419e-02, 5.1178e-02]],

         [[1.5305e-02, 4.1847e-03, 2.0065e-03,  ..., 1.4153e-02,
           5.9570e-02, 6.3416e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6253e-05, 2.2066e-04, 1.9665e-03,  ..., 2.9816e-02,
           2.3450e-01, 5.6250e-01]],

         [[0.0000e+00, 5.9605e-08, 4.1723e-07,  ..., 8.2642e-02,
           8.6133e-01, 5.3375e-02]],

         [[2.9278e-04, 8.2111e-04, 6.9847e-03,  ..., 6.7322e-02,
           3.5254e-01, 4.2114e-02]],

         ...,

         [[2.1801e-03, 6.3591e-03, 1.1879e-02,  ..., 3.0334e-02,
           3.0884e-01, 8.9905e-02]],

         [[1.6146e-03, 1.1032e-02, 3.1921e-02,  ..., 2.4292e-02,
           2.3727e-02, 1.9745e-02]],

         [[6.5804e-03, 1.8173e-02, 1.9196e-02,  ..., 1.7471e-02,
           1.0876e-01, 1.0620e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5378e-02, 7.2632e-03, 4.0344e-02,  ..., 3.3966e-02,
           6.1035e-02, 3.5339e-02]],

         [[2.6627e-03, 1.0386e-03, 5.8174e-03,  ..., 6.0005e-03,
           1.7090e-02, 1.2802e-02]],

         [[4.3945e-03, 8.2541e-04, 9.4652e-04,  ..., 3.1952e-02,
           2.7557e-02, 1.9852e-02]],

         ...,

         [[1.3618e-02, 4.7803e-05, 1.2159e-04,  ..., 1.3647e-01,
           5.8838e-01, 2.0178e-01]],

         [[7.4654e-03, 1.0881e-03, 1.7452e-03,  ..., 8.9539e-02,
           5.4718e-02, 1.3440e-01]],

         [[5.2032e-03, 8.3590e-04, 3.1052e-03,  ..., 1.1475e-01,
           5.7373e-01, 1.9519e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.0381e-01, 1.4341e-04, 8.3923e-05,  ..., 4.5547e-03,
           8.4457e-03, 2.9469e-03]],

         [[7.6123e-01, 9.1791e-06, 1.7703e-05,  ..., 1.2306e-02,
           5.8594e-02, 1.4209e-01]],

         [[7.9590e-01, 1.6093e-06, 1.6093e-06,  ..., 4.5319e-03,
           1.5610e-02, 1.6235e-01]],

         ...,

         [[5.2686e-01, 1.0264e-04, 6.6566e-04,  ..., 2.4092e-04,
           2.5809e-05, 3.4218e-03]],

         [[6.5967e-01, 3.1662e-04, 8.8549e-04,  ..., 1.7105e-02,
           2.0004e-02, 1.7441e-02]],

         [[6.6406e-01, 1.1134e-04, 3.5334e-04,  ..., 4.9072e-02,
           1.4557e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6084e-01, 6.1989e-06, 8.9407e-07,  ..., 3.4637e-03,
           8.4877e-04, 9.3994e-02]],

         [[5.5371e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8954e-03,
           3.7256e-01, 6.9824e-02]],

         [[7.9443e-01, 9.8407e-05, 2.4498e-05,  ..., 7.3586e-03,
           1.6431e-01, 1.7334e-02]],

         ...,

         [[5.9863e-01, 5.4932e-04, 2.8300e-04,  ..., 5.6549e-02,
           8.3389e-03, 1.1826e-02]],

         [[3.7109e-01, 1.4200e-03, 6.9284e-04,  ..., 1.1002e-02,
           1.6068e-02, 1.6846e-02]],

         [[3.5156e-01, 3.9876e-05, 4.4942e-05,  ..., 3.6835e-02,
           5.5957e-01, 1.4313e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6099e-01, 3.0088e-04, 1.1426e-04,  ..., 6.5369e-02,
           1.2402e-01, 8.9233e-02]],

         [[6.7920e-01, 8.1599e-05, 3.8981e-05,  ..., 7.3486e-02,
           1.6565e-01, 2.1332e-02]],

         [[7.7344e-01, 4.1366e-05, 2.5868e-05,  ..., 1.0757e-02,
           1.1316e-01, 7.7698e-02]],

         ...,

         [[7.9736e-01, 1.7881e-07, 5.9605e-08,  ..., 1.4246e-05,
           2.6524e-05, 1.9641e-01]],

         [[9.4092e-01, 2.6822e-06, 1.0729e-06,  ..., 0.0000e+00,
           0.0000e+00, 1.7578e-02]],

         [[9.7461e-01, 1.7881e-07, 5.9605e-08,  ..., 0.0000e+00,
           0.0000e+00, 2.1820e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 4.2379e-05, 5.1260e-06,  ..., 2.5406e-02,
           2.7686e-01, 4.3671e-02]],

         [[9.2578e-01, 1.0073e-05, 5.9605e-07,  ..., 1.4818e-04,
           2.8896e-03, 5.7800e-02]],

         [[8.7939e-01, 6.3181e-05, 2.4140e-05,  ..., 3.4771e-03,
           5.2185e-03, 8.6365e-03]],

         ...,

         [[1.4697e-01, 4.5002e-05, 7.3314e-06,  ..., 3.1757e-03,
           8.2324e-01, 5.2376e-03]],

         [[6.8506e-01, 7.4005e-04, 1.3292e-04,  ..., 2.3010e-02,
           5.9967e-02, 1.4636e-01]],

         [[4.4702e-01, 2.4962e-04, 5.0843e-05,  ..., 4.1237e-03,
           9.7229e-02, 4.4360e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7119e-01, 7.5102e-06, 9.3579e-06,  ..., 5.6839e-04,
           5.9586e-03, 6.8588e-03]],

         [[9.1260e-01, 1.0312e-05, 2.6822e-06,  ..., 2.3804e-03,
           6.5575e-03, 5.0751e-02]],

         [[9.7266e-01, 3.3975e-06, 5.3644e-07,  ..., 4.7565e-04,
           2.0966e-02, 3.7861e-03]],

         ...,

         [[8.2812e-01, 8.1003e-05, 1.6630e-05,  ..., 2.8324e-03,
           2.1576e-02, 5.8533e-02]],

         [[9.4873e-01, 8.1539e-05, 5.0664e-06,  ..., 6.4373e-04,
           9.1476e-03, 9.8343e-03]],

         [[9.0576e-01, 1.2577e-05, 3.0994e-06,  ..., 2.0866e-03,
           2.4979e-02, 5.6091e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7695e-01, 1.1835e-03, 2.8110e-04,  ..., 9.8267e-03,
           4.6112e-02, 3.9856e-02]],

         [[9.7461e-01, 3.9935e-06, 1.2279e-05,  ..., 1.0490e-03,
           1.7910e-03, 2.2106e-03]],

         [[7.4561e-01, 4.5002e-05, 2.8014e-06,  ..., 6.6614e-04,
           1.6382e-01, 5.5573e-02]],

         ...,

         [[5.2490e-01, 1.1384e-05, 1.7285e-05,  ..., 1.0376e-03,
           1.9409e-02, 4.3945e-01]],

         [[9.6680e-01, 1.3709e-06, 3.5167e-06,  ..., 6.8092e-03,
           1.2703e-03, 1.3596e-02]],

         [[4.5605e-01, 2.0266e-06, 4.2915e-06,  ..., 1.1845e-03,
           5.2002e-01, 5.4321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5415e-01, 1.2720e-04, 8.7595e-04,  ..., 5.1758e-02,
           2.5635e-02, 1.0559e-02]],

         [[7.1680e-01, 1.7345e-05, 5.4836e-06,  ..., 5.6267e-03,
           2.4524e-01, 1.6632e-02]],

         [[9.4092e-01, 1.0252e-05, 9.7752e-06,  ..., 9.6893e-03,
           1.6708e-02, 1.0712e-02]],

         ...,

         [[9.0967e-01, 3.1412e-05, 2.7955e-05,  ..., 8.2874e-04,
           1.3107e-02, 2.8563e-04]],

         [[5.4834e-01, 1.8477e-05, 5.7817e-06,  ..., 1.8463e-03,
           3.6963e-01, 4.0497e-02]],

         [[8.9502e-01, 2.0266e-05, 2.7061e-05,  ..., 2.0798e-02,
           7.2975e-03, 3.5370e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8301e-01, 1.5259e-04, 3.3379e-05,  ..., 7.7486e-07,
           6.5327e-05, 1.6079e-03]],

         [[9.3115e-01, 2.0325e-05, 2.8908e-05,  ..., 1.7319e-03,
           2.6932e-02, 1.2527e-02]],

         [[9.6729e-01, 1.7285e-05, 1.9908e-05,  ..., 4.7159e-04,
           5.3978e-03, 6.6376e-03]],

         ...,

         [[8.8574e-01, 3.0136e-04, 1.7846e-04,  ..., 1.2808e-03,
           2.6566e-02, 4.2648e-03]],

         [[8.7451e-01, 6.3419e-05, 6.0737e-05,  ..., 1.0849e-02,
           4.7569e-03, 6.1913e-03]],

         [[7.4121e-01, 2.4929e-03, 4.5562e-04,  ..., 1.8097e-02,
           8.1253e-03, 9.5749e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5576e-01, 1.9133e-05, 1.9312e-05,  ..., 5.8556e-04,
           1.1902e-02, 3.2739e-01]],

         [[5.0537e-01, 9.4509e-04, 1.7004e-03,  ..., 1.7099e-03,
           5.8044e-02, 3.9581e-02]],

         [[8.2471e-01, 2.6512e-04, 4.9829e-04,  ..., 3.9043e-03,
           6.8207e-03, 1.1177e-02]],

         ...,

         [[6.5137e-01, 2.0468e-04, 2.7871e-04,  ..., 5.8556e-03,
           1.1223e-02, 1.3504e-03]],

         [[7.4414e-01, 7.9811e-05, 7.1228e-05,  ..., 6.5857e-02,
           6.3354e-02, 8.2169e-03]],

         [[5.5127e-01, 9.3102e-05, 1.6475e-04,  ..., 1.1276e-02,
           2.7969e-02, 3.6694e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3843e-02, 3.3557e-05, 1.1683e-05,  ..., 1.6870e-03,
           5.8441e-02, 8.5449e-01]],

         [[6.3965e-01, 5.0664e-05, 2.3901e-05,  ..., 1.6687e-01,
           4.8279e-02, 1.0498e-02]],

         [[1.4270e-01, 5.9128e-04, 1.1694e-04,  ..., 1.2306e-02,
           2.1997e-01, 6.1249e-02]],

         ...,

         [[2.0874e-01, 1.1384e-05, 1.2994e-05,  ..., 3.1910e-03,
           4.8981e-02, 6.8311e-01]],

         [[8.7402e-01, 4.7326e-05, 2.7180e-05,  ..., 4.7302e-03,
           2.0889e-02, 1.5602e-02]],

         [[8.6816e-01, 1.3626e-04, 2.1577e-05,  ..., 1.7975e-02,
           4.2816e-02, 2.4277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9199e-01, 4.2439e-04, 9.4175e-04,  ..., 8.9722e-03,
           1.6144e-02, 3.6812e-03]],

         [[8.2959e-01, 1.0014e-05, 4.8637e-05,  ..., 4.4952e-02,
           3.5522e-02, 1.3252e-02]],

         [[3.5107e-01, 3.0696e-05, 3.1292e-05,  ..., 2.4071e-03,
           2.9526e-02, 3.6221e-03]],

         ...,

         [[2.4536e-01, 0.0000e+00, 0.0000e+00,  ..., 3.6743e-02,
           7.0166e-01, 1.4389e-02]],

         [[4.0137e-01, 5.4312e-04, 8.1062e-04,  ..., 5.9166e-03,
           2.7142e-03, 1.2457e-04]],

         [[5.8643e-01, 1.7881e-07, 0.0000e+00,  ..., 3.1738e-03,
           3.9453e-01, 1.2199e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3975e-01, 4.8637e-05, 7.3314e-06,  ..., 1.2077e-02,
           4.1107e-02, 2.4933e-02]],

         [[4.0436e-02, 9.8610e-04, 4.1842e-04,  ..., 5.1727e-02,
           1.8501e-03, 6.4754e-04]],

         [[2.9980e-01, 6.4194e-05, 9.7871e-05,  ..., 3.1543e-01,
           4.2877e-02, 7.7133e-03]],

         ...,

         [[6.1279e-01, 1.7285e-06, 7.1526e-07,  ..., 6.4087e-02,
           1.8860e-01, 2.2430e-02]],

         [[7.2632e-02, 5.9605e-08, 0.0000e+00,  ..., 1.1147e-02,
           9.0918e-01, 4.1962e-03]],

         [[5.1709e-01, 1.7464e-05, 3.4690e-05,  ..., 9.4543e-02,
           1.0236e-01, 1.1963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6987e-02, 5.3644e-07, 7.7486e-07,  ..., 1.8752e-04,
           2.0508e-02, 9.3896e-01]],

         [[5.0812e-02, 5.5432e-06, 1.3351e-05,  ..., 3.8892e-01,
           1.2927e-01, 3.6530e-02]],

         [[1.3562e-01, 5.6648e-04, 7.9679e-04,  ..., 2.7039e-02,
           3.4302e-02, 1.0445e-02]],

         ...,

         [[6.6895e-01, 7.9870e-05, 2.7061e-05,  ..., 1.0033e-02,
           2.5818e-02, 1.6289e-03]],

         [[6.3965e-01, 6.3181e-06, 1.3709e-06,  ..., 3.8872e-03,
           1.1951e-01, 5.9570e-02]],

         [[6.0059e-01, 3.3438e-05, 1.6212e-05,  ..., 2.3377e-04,
           1.0358e-01, 2.1149e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7974e-02, 2.1458e-06, 3.5763e-07,  ..., 2.3060e-03,
           9.1553e-01, 6.2180e-03]],

         [[2.0740e-01, 3.2973e-04, 6.3479e-05,  ..., 4.3152e-02,
           1.6992e-01, 9.7595e-02]],

         [[4.1919e-01, 3.5644e-05, 3.7432e-05,  ..., 2.8976e-02,
           1.0071e-01, 1.0638e-01]],

         ...,

         [[9.0332e-01, 3.1173e-05, 2.2948e-05,  ..., 3.7346e-03,
           2.2003e-02, 1.2733e-02]],

         [[5.2490e-01, 2.9743e-05, 1.7226e-05,  ..., 5.4474e-02,
           2.6392e-01, 2.6978e-02]],

         [[7.3584e-01, 1.3351e-05, 8.2850e-06,  ..., 3.3817e-03,
           1.5857e-01, 2.7451e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6609e-02, 1.5383e-03, 2.3708e-03,  ..., 6.9695e-03,
           5.3024e-03, 4.9133e-03]],

         [[5.6305e-02, 4.3809e-05, 2.0146e-05,  ..., 2.6398e-02,
           6.2317e-02, 2.9434e-02]],

         [[3.4229e-01, 1.5020e-05, 2.3723e-05,  ..., 1.5186e-01,
           3.2397e-01, 8.4229e-02]],

         ...,

         [[4.3359e-01, 4.1437e-04, 4.1485e-05,  ..., 9.9373e-04,
           6.9153e-02, 2.4353e-02]],

         [[8.3984e-01, 2.9349e-04, 1.9693e-04,  ..., 3.5763e-05,
           4.5252e-04, 1.0780e-02]],

         [[5.6934e-01, 1.0842e-04, 1.0228e-04,  ..., 1.7059e-02,
           2.2755e-03, 1.2026e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-02, 7.9095e-05, 9.1016e-05,  ..., 1.1853e-01,
           1.9007e-03, 1.1925e-02]],

         [[3.1323e-01, 4.3726e-04, 2.9683e-04,  ..., 6.5613e-02,
           2.9587e-02, 1.8951e-02]],

         [[6.1920e-02, 2.7418e-06, 1.4305e-06,  ..., 5.4230e-02,
           3.1689e-01, 4.7217e-01]],

         ...,

         [[3.1665e-01, 6.7353e-06, 8.4043e-06,  ..., 2.8833e-01,
           1.0858e-01, 1.7715e-02]],

         [[2.4036e-01, 4.9472e-06, 5.9009e-06,  ..., 2.5883e-03,
           6.6895e-01, 5.8899e-02]],

         [[3.6401e-01, 3.8981e-05, 2.2531e-05,  ..., 8.7280e-03,
           3.0884e-01, 2.4060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7119e-01, 6.5863e-05, 1.1265e-05,  ..., 3.4668e-02,
           1.4368e-01, 7.0007e-02]],

         [[4.0497e-02, 6.4969e-06, 2.5034e-06,  ..., 1.9019e-01,
           4.1528e-01, 7.4463e-02]],

         [[1.4001e-01, 4.2319e-06, 9.5367e-07,  ..., 2.8777e-04,
           2.2156e-02, 8.3105e-01]],

         ...,

         [[3.7872e-02, 2.8610e-05, 1.5914e-05,  ..., 1.8239e-04,
           3.9291e-03, 6.0387e-03]],

         [[2.6880e-01, 2.6298e-04, 1.0830e-04,  ..., 1.4809e-02,
           3.2104e-02, 3.1860e-02]],

         [[1.7468e-01, 2.2376e-04, 9.1255e-05,  ..., 4.8126e-02,
           2.7206e-02, 1.1253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7305e-01, 1.0419e-04, 4.1306e-05,  ..., 5.5351e-03,
           5.2917e-02, 7.5195e-02]],

         [[9.2334e-01, 2.9564e-04, 2.0266e-06,  ..., 5.3644e-07,
           4.8995e-05, 1.5249e-03]],

         [[9.1504e-01, 3.1424e-04, 2.0087e-05,  ..., 2.7359e-05,
           1.6994e-03, 5.5275e-03]],

         ...,

         [[1.6919e-01, 4.9829e-04, 1.0138e-03,  ..., 5.0316e-03,
           1.7014e-02, 3.1799e-02]],

         [[8.1104e-01, 2.3186e-05, 2.5094e-05,  ..., 9.0599e-04,
           8.2825e-02, 4.5044e-02]],

         [[1.8188e-01, 7.7629e-04, 1.3304e-03,  ..., 1.7319e-02,
           4.6722e-02, 5.6366e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8135e-01, 1.6296e-04, 2.1935e-04,  ..., 2.6779e-02,
           5.0812e-02, 4.4159e-02]],

         [[2.3132e-01, 1.1384e-04, 2.5249e-04,  ..., 1.1429e-02,
           2.0065e-02, 4.6265e-02]],

         [[3.4473e-01, 1.1927e-04, 1.6046e-04,  ..., 3.6812e-03,
           5.6274e-02, 3.7170e-02]],

         ...,

         [[1.9287e-02, 7.0632e-05, 5.0902e-05,  ..., 8.7769e-02,
           3.3844e-02, 3.9551e-02]],

         [[7.2823e-03, 7.6294e-06, 2.3842e-06,  ..., 2.6941e-04,
           1.4038e-02, 3.9978e-02]],

         [[5.0439e-01, 4.7684e-05, 4.7684e-05,  ..., 8.2779e-03,
           1.8262e-01, 5.8380e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9102e-01, 2.7537e-05, 3.0398e-05,  ..., 1.2993e-02,
           1.9238e-01, 1.4526e-01]],

         [[3.5425e-01, 1.3781e-04, 2.4581e-04,  ..., 9.6069e-02,
           1.8768e-02, 2.4292e-02]],

         [[1.9373e-01, 2.9504e-05, 9.3460e-05,  ..., 5.2948e-02,
           1.3110e-01, 3.7061e-01]],

         ...,

         [[5.2441e-01, 2.0480e-04, 2.2316e-04,  ..., 4.4403e-02,
           8.7585e-02, 7.9773e-02]],

         [[3.6224e-02, 7.5102e-06, 8.8811e-06,  ..., 8.3685e-05,
           4.8256e-03, 4.0405e-02]],

         [[2.1960e-01, 1.1319e-04, 1.1635e-04,  ..., 2.3880e-02,
           1.2985e-02, 1.9485e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7285e-01, 9.3043e-05, 6.8605e-05,  ..., 3.2501e-02,
           2.6562e-01, 1.0162e-01]],

         [[5.6396e-02, 1.0073e-05, 1.7107e-05,  ..., 4.9011e-02,
           3.8843e-01, 4.4281e-02]],

         [[3.3911e-01, 9.5427e-05, 5.2571e-05,  ..., 7.9285e-02,
           2.3071e-02, 1.4107e-02]],

         ...,

         [[6.1182e-01, 2.9802e-07, 5.9605e-08,  ..., 5.0995e-02,
           2.8442e-01, 4.2297e-02]],

         [[2.1509e-01, 1.5080e-05, 9.8348e-06,  ..., 1.7017e-01,
           2.2546e-01, 4.9896e-02]],

         [[1.5906e-01, 7.1526e-07, 5.9605e-07,  ..., 8.6487e-02,
           7.1289e-01, 3.5217e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1858e-03, 1.1921e-06, 1.8477e-06,  ..., 9.5703e-02,
           8.0414e-03, 1.4786e-02]],

         [[2.0599e-02, 1.1921e-06, 1.9073e-06,  ..., 2.6131e-04,
           1.1375e-02, 9.6191e-01]],

         [[6.0486e-02, 6.0499e-05, 9.8944e-05,  ..., 5.2155e-02,
           2.0695e-03, 5.5389e-03]],

         ...,

         [[1.2830e-01, 1.1438e-04, 3.9339e-04,  ..., 3.1921e-02,
           2.2293e-02, 7.2937e-03]],

         [[4.3335e-02, 1.7881e-07, 1.1921e-07,  ..., 6.3428e-01,
           2.7115e-02, 7.8278e-03]],

         [[1.7749e-01, 5.9605e-08, 5.9605e-08,  ..., 1.7615e-01,
           5.4883e-01, 6.5308e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 8.9526e-05, 4.1723e-04,  ..., 5.2147e-03,
           2.7969e-02, 4.9835e-02]],

         [[9.7949e-01, 8.1599e-05, 1.4365e-05,  ..., 1.0079e-04,
           6.5613e-04, 3.1776e-03]],

         [[8.7988e-01, 1.0693e-04, 9.3997e-05,  ..., 1.1635e-03,
           1.4221e-02, 2.2385e-02]],

         ...,

         [[4.2944e-01, 6.5565e-06, 6.5565e-06,  ..., 6.1859e-02,
           3.8477e-01, 7.9407e-02]],

         [[7.0508e-01, 6.2585e-06, 2.6226e-06,  ..., 6.1989e-03,
           6.4964e-03, 7.4768e-03]],

         [[2.5879e-01, 1.5676e-05, 4.0054e-05,  ..., 1.6632e-03,
           7.3395e-03, 6.9238e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4551e-01, 1.2577e-05, 1.8418e-05,  ..., 1.9128e-01,
           1.1292e-01, 7.3792e-02]],

         [[4.2407e-01, 9.6560e-06, 9.2983e-06,  ..., 4.2999e-02,
           3.2275e-01, 1.4539e-01]],

         [[2.3669e-01, 1.6999e-04, 2.7418e-04,  ..., 1.1559e-02,
           2.4707e-01, 2.3401e-01]],

         ...,

         [[6.0596e-01, 5.4836e-06, 1.7881e-05,  ..., 6.1890e-02,
           7.0129e-02, 1.0608e-01]],

         [[7.5293e-01, 2.7835e-05, 6.2764e-05,  ..., 8.3313e-03,
           1.5503e-02, 1.7029e-02]],

         [[3.0225e-01, 2.9445e-05, 7.1645e-05,  ..., 1.4844e-01,
           2.5208e-02, 3.0396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8665e-02, 7.6294e-06, 6.9141e-06,  ..., 7.9651e-02,
           3.1128e-03, 5.7716e-03]],

         [[4.2017e-01, 2.4378e-05, 1.3530e-05,  ..., 1.6296e-02,
           9.8419e-03, 7.4272e-03]],

         [[2.6392e-01, 1.1843e-04, 2.7490e-04,  ..., 1.7960e-02,
           1.0577e-01, 1.0828e-01]],

         ...,

         [[2.5940e-02, 7.7486e-06, 1.0604e-04,  ..., 6.7322e-02,
           5.6992e-03, 2.2888e-02]],

         [[1.3257e-01, 2.0027e-05, 1.1003e-04,  ..., 1.1511e-01,
           3.2158e-03, 8.6060e-03]],

         [[5.3613e-01, 2.4259e-05, 2.6345e-05,  ..., 1.0394e-01,
           3.9764e-02, 7.9712e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1118e-02, 1.1921e-07, 0.0000e+00,  ..., 6.6986e-03,
           9.5947e-01, 9.4452e-03]],

         [[5.1318e-01, 7.2718e-06, 2.4915e-05,  ..., 5.3436e-02,
           1.7593e-02, 2.2400e-02]],

         [[3.0493e-01, 1.0133e-06, 1.0729e-06,  ..., 7.8613e-02,
           5.0830e-01, 6.6162e-02]],

         ...,

         [[7.9150e-01, 2.4414e-04, 3.1352e-04,  ..., 3.1223e-03,
           5.9013e-03, 4.4556e-03]],

         [[4.9976e-01, 4.7684e-06, 2.3842e-06,  ..., 2.1997e-01,
           3.1204e-02, 1.0292e-02]],

         [[4.6387e-02, 3.4869e-05, 8.4102e-05,  ..., 3.2692e-03,
           1.1749e-01, 7.6025e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7100e-01, 2.0742e-05, 7.7128e-05,  ..., 2.6520e-02,
           1.8921e-01, 1.6309e-01]],

         [[2.8345e-01, 4.8876e-04, 3.8815e-04,  ..., 5.7945e-03,
           2.7634e-02, 1.9073e-02]],

         [[1.1023e-01, 3.4988e-05, 1.2171e-04,  ..., 4.3823e-02,
           2.6581e-02, 1.5625e-02]],

         ...,

         [[6.4014e-01, 2.2864e-04, 3.0756e-04,  ..., 1.4145e-02,
           3.6682e-02, 1.6968e-01]],

         [[1.2335e-01, 1.1247e-04, 1.4663e-04,  ..., 1.2405e-02,
           8.6121e-02, 7.2070e-01]],

         [[6.4758e-02, 7.0333e-06, 7.0930e-06,  ..., 4.7852e-01,
           6.4758e-02, 5.0446e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7463e-02, 3.8743e-06, 6.9559e-05,  ..., 6.6699e-01,
           7.5989e-02, 2.0294e-02]],

         [[1.1041e-01, 1.1563e-05, 7.8619e-05,  ..., 1.4807e-01,
           4.3427e-02, 4.7119e-02]],

         [[2.8882e-01, 2.6536e-04, 1.3523e-03,  ..., 8.1482e-02,
           4.7150e-02, 1.0461e-01]],

         ...,

         [[7.8418e-01, 2.4188e-04, 5.9652e-04,  ..., 3.5126e-02,
           1.0445e-02, 5.5504e-03]],

         [[9.3213e-01, 2.1458e-06, 5.6362e-04,  ..., 6.1655e-04,
           4.2496e-03, 1.0170e-02]],

         [[3.1470e-01, 5.1498e-05, 4.0889e-05,  ..., 6.9809e-03,
           4.6417e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5923e-05, 4.0829e-05, 1.2524e-01,  ..., 1.7664e-01,
           1.0902e-02, 1.7151e-02]],

         [[7.3791e-05, 1.5378e-05, 7.2098e-03,  ..., 9.1858e-03,
           3.3875e-02, 1.0574e-02]],

         [[7.3969e-05, 1.5533e-04, 5.7159e-02,  ..., 2.7115e-02,
           2.1484e-01, 7.3425e-02]],

         ...,

         [[2.9397e-04, 1.7107e-05, 6.8420e-02,  ..., 4.8218e-01,
           6.1310e-02, 5.8502e-02]],

         [[5.8985e-04, 2.4867e-04, 1.8970e-01,  ..., 1.3214e-02,
           8.6853e-02, 1.8066e-02]],

         [[2.3544e-05, 4.6492e-05, 1.5942e-01,  ..., 1.2589e-02,
           1.0193e-02, 2.7428e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4046e-04, 3.3784e-04, 2.8833e-01,  ..., 5.0903e-02,
           2.2373e-03, 6.0806e-03]],

         [[1.7166e-04, 1.4865e-04, 1.8396e-01,  ..., 3.1464e-02,
           4.6768e-03, 7.2289e-04]],

         [[9.4604e-04, 2.4772e-04, 3.5034e-01,  ..., 2.0386e-02,
           4.6301e-04, 1.1803e-02]],

         ...,

         [[4.5609e-04, 8.9645e-05, 5.9723e-02,  ..., 1.4465e-02,
           1.9739e-01, 1.9699e-02]],

         [[8.2731e-05, 5.6267e-05, 2.7661e-01,  ..., 3.2288e-02,
           9.7847e-04, 1.6136e-03]],

         [[4.8697e-05, 7.5281e-05, 3.0151e-01,  ..., 1.9882e-02,
           1.6451e-03, 5.3101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6367e-04, 6.9022e-05, 2.3047e-01,  ..., 1.3206e-02,
           2.6369e-04, 3.9711e-03]],

         [[3.8385e-04, 1.1563e-05, 2.4548e-01,  ..., 1.5823e-02,
           3.4027e-02, 5.9700e-04]],

         [[3.8671e-04, 1.6153e-04, 2.7808e-01,  ..., 1.8707e-02,
           5.4626e-03, 6.7616e-04]],

         ...,

         [[1.8895e-05, 9.5963e-06, 8.7402e-02,  ..., 2.6230e-02,
           4.0723e-01, 8.6731e-02]],

         [[4.9591e-05, 7.6473e-05, 1.2341e-01,  ..., 1.4236e-02,
           6.7253e-03, 4.9400e-03]],

         [[2.1505e-04, 3.4213e-05, 1.9373e-01,  ..., 1.2978e-02,
           1.1581e-02, 4.4327e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3910e-06, 3.5763e-07, 1.3256e-03,  ..., 7.2510e-01,
           3.2867e-02, 5.0934e-02]],

         [[2.2709e-05, 7.5281e-05, 1.3611e-01,  ..., 4.2328e-02,
           3.1647e-02, 3.0304e-02]],

         [[1.1551e-04, 5.9795e-04, 3.1030e-01,  ..., 8.3017e-04,
           4.0627e-03, 1.5697e-03]],

         ...,

         [[1.8275e-04, 5.7817e-06, 2.6782e-01,  ..., 2.0004e-02,
           4.2267e-03, 4.3869e-03]],

         [[6.9618e-04, 3.2842e-05, 1.6281e-02,  ..., 2.8760e-01,
           3.2568e-01, 1.9910e-01]],

         [[7.1943e-05, 5.1856e-06, 3.6316e-02,  ..., 1.5906e-01,
           4.6265e-02, 1.5771e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4277e-03, 2.8110e-04, 1.2286e-01,  ..., 1.1299e-02,
           5.4535e-02, 1.0931e-01]],

         [[4.4680e-04, 8.7738e-04, 2.9517e-01,  ..., 1.9932e-03,
           3.2520e-03, 1.5945e-02]],

         [[8.2779e-04, 2.3139e-04, 1.4697e-01,  ..., 6.9946e-02,
           1.1853e-01, 7.2754e-02]],

         ...,

         [[5.1451e-04, 1.0294e-04, 7.6965e-02,  ..., 1.8896e-01,
           1.1377e-01, 3.0200e-01]],

         [[3.9506e-04, 1.0824e-03, 2.6270e-01,  ..., 3.6682e-02,
           4.2908e-02, 3.7842e-02]],

         [[1.1569e-04, 1.0616e-04, 1.1023e-01,  ..., 1.5918e-01,
           1.1646e-01, 7.7576e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.0862e-03, 3.3140e-04, 1.4542e-02,  ..., 8.6304e-02,
           3.8696e-01, 8.1116e-02]],

         [[1.7500e-04, 1.8816e-03, 1.3908e-02,  ..., 1.0826e-02,
           9.6497e-02, 4.3262e-01]],

         [[0.0000e+00, 2.2650e-06, 7.4208e-05,  ..., 4.7684e-07,
           9.0456e-04, 9.9170e-01]],

         ...,

         [[5.9605e-08, 1.3709e-06, 4.4823e-05,  ..., 1.6510e-05,
           4.0321e-03, 9.8682e-01]],

         [[2.7122e-03, 4.2000e-03, 3.9482e-03,  ..., 1.0727e-02,
           2.7390e-02, 1.0727e-02]],

         [[1.7120e-02, 2.8839e-03, 5.1737e-04,  ..., 5.1117e-02,
           1.6083e-02, 5.7953e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2648e-04, 3.7169e-04, 4.1885e-03,  ..., 6.9763e-02,
           9.8389e-02, 5.6592e-01]],

         [[0.0000e+00, 0.0000e+00, 2.3842e-07,  ..., 4.2480e-02,
           9.3701e-01, 1.8265e-02]],

         [[7.1669e-04, 1.6155e-03, 8.3313e-03,  ..., 2.0825e-01,
           3.3997e-02, 2.8183e-02]],

         ...,

         [[2.0618e-03, 5.3062e-03, 6.2485e-03,  ..., 1.7163e-01,
           3.9490e-02, 2.6172e-01]],

         [[2.5940e-03, 1.3725e-02, 2.9510e-02,  ..., 2.2537e-02,
           2.7512e-02, 7.7286e-03]],

         [[1.7061e-03, 6.6414e-03, 6.3400e-03,  ..., 2.7808e-01,
           2.4927e-01, 5.6061e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7791e-02, 5.1575e-03, 2.9449e-02,  ..., 7.4585e-02,
           5.9479e-02, 4.3854e-02]],

         [[1.1368e-03, 5.9652e-04, 2.5711e-03,  ..., 1.1795e-02,
           7.7362e-03, 1.1816e-03]],

         [[3.8681e-03, 6.9618e-04, 8.5640e-04,  ..., 2.4353e-02,
           1.6220e-02, 1.2939e-02]],

         ...,

         [[1.4229e-02, 6.1560e-04, 3.4261e-04,  ..., 1.3708e-01,
           5.2588e-01, 2.6025e-01]],

         [[3.4103e-03, 4.3702e-04, 8.1015e-04,  ..., 5.0507e-02,
           5.4626e-02, 1.8585e-02]],

         [[9.8801e-03, 1.6909e-03, 1.0117e-02,  ..., 8.4045e-02,
           3.5376e-01, 1.0876e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0801e-01, 3.5763e-06, 6.5327e-05,  ..., 5.5771e-03,
           2.6657e-02, 1.6083e-02]],

         [[6.7090e-01, 2.2054e-06, 4.3035e-05,  ..., 7.7477e-03,
           2.1521e-01, 1.5022e-02]],

         [[8.1885e-01, 5.5432e-06, 1.1146e-05,  ..., 8.4778e-02,
           5.9235e-02, 9.8038e-03]],

         ...,

         [[3.2715e-02, 7.3314e-06, 3.3677e-05,  ..., 1.5032e-04,
           2.1398e-05, 6.0501e-03]],

         [[6.9141e-01, 1.1659e-04, 3.9959e-04,  ..., 8.5983e-03,
           1.2268e-02, 1.4015e-02]],

         [[3.4863e-01, 2.0278e-04, 2.3139e-04,  ..., 1.4786e-02,
           3.3722e-02, 3.8055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4570e-01, 1.7285e-06, 8.3447e-07,  ..., 7.0076e-03,
           7.1478e-04, 1.8570e-02]],

         [[7.2559e-01, 7.7486e-07, 5.9605e-08,  ..., 6.9519e-02,
           1.9324e-01, 7.9193e-03]],

         [[9.4385e-01, 1.7977e-04, 9.8884e-05,  ..., 5.3062e-03,
           7.3204e-03, 1.0712e-02]],

         ...,

         [[2.3926e-01, 6.6471e-04, 1.4734e-04,  ..., 3.1738e-02,
           4.0283e-02, 1.2238e-02]],

         [[1.2408e-01, 1.4324e-03, 6.9618e-04,  ..., 4.4060e-03,
           4.4518e-03, 2.9831e-03]],

         [[4.8145e-01, 4.8876e-04, 2.1648e-04,  ..., 1.8347e-01,
           2.2217e-01, 4.4434e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6321e-01, 2.0540e-04, 8.3983e-05,  ..., 9.4910e-02,
           1.0529e-01, 5.4901e-02]],

         [[4.6997e-01, 8.3745e-05, 1.5080e-05,  ..., 2.0667e-01,
           1.3538e-01, 3.8490e-03]],

         [[5.1465e-01, 1.1086e-05, 7.6294e-06,  ..., 2.3181e-01,
           1.2274e-01, 4.5593e-02]],

         ...,

         [[7.7637e-01, 2.2650e-06, 5.9605e-08,  ..., 7.6234e-05,
           1.6093e-06, 2.1045e-01]],

         [[9.0430e-01, 2.8312e-05, 1.1921e-07,  ..., 1.9073e-06,
           0.0000e+00, 1.1688e-02]],

         [[9.6924e-01, 1.0610e-05, 0.0000e+00,  ..., 1.1921e-07,
           0.0000e+00, 4.4785e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2878e-01, 1.9073e-05, 1.1921e-06,  ..., 2.1411e-01,
           5.2588e-01, 2.7039e-02]],

         [[8.3887e-01, 2.6226e-06, 1.2517e-06,  ..., 9.1248e-03,
           1.1480e-04, 2.2598e-02]],

         [[6.6943e-01, 2.2995e-04, 4.9531e-05,  ..., 1.8158e-02,
           2.8076e-03, 2.3575e-02]],

         ...,

         [[5.9375e-01, 4.3750e-04, 3.8028e-05,  ..., 2.8369e-01,
           3.6133e-02, 3.1738e-02]],

         [[8.9355e-01, 2.0051e-04, 4.0591e-05,  ..., 1.4595e-02,
           2.8290e-02, 2.5269e-02]],

         [[2.8320e-01, 2.2137e-04, 3.0100e-05,  ..., 1.7761e-01,
           5.1172e-01, 6.6681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 4.3511e-06, 2.8014e-06,  ..., 3.5614e-02,
           2.1655e-01, 2.1896e-02]],

         [[9.1016e-01, 6.9320e-05, 1.3053e-05,  ..., 9.9945e-03,
           1.2054e-02, 1.7090e-02]],

         [[9.5654e-01, 1.7881e-07, 1.1921e-07,  ..., 2.2278e-03,
           1.7639e-02, 2.0615e-02]],

         ...,

         [[6.3428e-01, 2.3901e-05, 6.5565e-06,  ..., 4.0283e-02,
           1.4746e-01, 3.3661e-02]],

         [[8.8330e-01, 1.7571e-04, 2.6524e-05,  ..., 8.8043e-03,
           5.5969e-02, 3.1261e-03]],

         [[6.8506e-01, 2.0742e-05, 4.6492e-06,  ..., 8.6365e-02,
           1.3196e-01, 3.0502e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 8.9216e-04, 1.6177e-04,  ..., 1.4885e-02,
           1.1147e-02, 7.3967e-03]],

         [[9.7949e-01, 5.0068e-06, 1.5378e-05,  ..., 8.0919e-04,
           2.9602e-03, 1.7428e-04]],

         [[9.1162e-01, 8.5235e-06, 5.5432e-06,  ..., 1.5656e-02,
           6.0242e-02, 9.1982e-04]],

         ...,

         [[7.6074e-01, 4.6730e-05, 1.3113e-05,  ..., 4.1351e-02,
           5.1880e-02, 9.5947e-02]],

         [[8.3203e-01, 4.1723e-07, 4.1127e-06,  ..., 1.0433e-03,
           5.1880e-03, 3.6163e-03]],

         [[7.4219e-01, 1.1325e-06, 1.5497e-06,  ..., 1.7322e-01,
           4.7424e-02, 1.5167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3862e-02, 8.5354e-05, 6.8426e-04,  ..., 1.2787e-02,
           8.4839e-03, 3.3234e-02]],

         [[5.5713e-01, 7.0930e-06, 1.7226e-05,  ..., 5.0903e-02,
           3.5059e-01, 2.3300e-02]],

         [[4.7925e-01, 1.3471e-05, 5.0962e-05,  ..., 6.1279e-02,
           2.7490e-01, 1.7654e-02]],

         ...,

         [[8.3936e-01, 4.0889e-05, 8.3923e-05,  ..., 7.5493e-03,
           4.4918e-04, 2.9430e-03]],

         [[4.0723e-01, 3.5644e-05, 2.8729e-05,  ..., 1.8066e-01,
           3.0737e-01, 4.2267e-02]],

         [[6.2695e-01, 2.3842e-06, 5.3048e-06,  ..., 6.5765e-03,
           6.8176e-02, 5.7869e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3008e-01, 6.7949e-06, 8.6427e-06,  ..., 1.7965e-04,
           3.4273e-05, 6.6910e-03]],

         [[9.0479e-01, 9.0599e-06, 1.4961e-05,  ..., 5.7373e-03,
           1.3443e-02, 1.7061e-03]],

         [[5.8105e-01, 2.3842e-07, 1.1921e-06,  ..., 1.0956e-02,
           1.0014e-03, 1.3855e-02]],

         ...,

         [[8.4668e-01, 1.3411e-04, 6.3360e-05,  ..., 2.6688e-02,
           1.6479e-02, 1.4740e-02]],

         [[5.4688e-01, 1.1736e-04, 9.5010e-05,  ..., 7.1045e-02,
           4.0283e-03, 4.1229e-02]],

         [[3.1763e-01, 5.4646e-04, 1.0848e-04,  ..., 1.9470e-02,
           6.2866e-03, 3.3173e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6270e-01, 5.4836e-06, 2.2352e-05,  ..., 1.6928e-03,
           2.4014e-03, 7.2559e-01]],

         [[3.9600e-01, 5.5075e-04, 1.3485e-03,  ..., 4.7668e-02,
           1.0742e-01, 6.7200e-02]],

         [[7.0068e-01, 3.3545e-04, 8.0109e-04,  ..., 2.0065e-02,
           3.1647e-02, 1.4008e-02]],

         ...,

         [[6.0596e-01, 2.1231e-04, 1.8084e-04,  ..., 1.8082e-02,
           7.9651e-02, 8.2855e-03]],

         [[7.5732e-01, 4.9055e-05, 4.6074e-05,  ..., 1.1505e-02,
           6.4758e-02, 6.5430e-02]],

         [[3.2178e-01, 2.8014e-05, 5.6624e-05,  ..., 5.0293e-02,
           2.4071e-03, 4.4043e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2139e-01, 8.2135e-05, 9.2328e-05,  ..., 8.2458e-02,
           4.9774e-02, 2.0654e-01]],

         [[4.9365e-01, 2.2018e-04, 1.4555e-04,  ..., 3.2043e-02,
           1.3293e-01, 1.0551e-02]],

         [[4.5410e-01, 1.3316e-04, 6.7472e-05,  ..., 6.2408e-02,
           3.7659e-02, 2.5360e-02]],

         ...,

         [[4.6600e-02, 2.0266e-06, 9.8348e-06,  ..., 8.3590e-04,
           1.9016e-03, 8.4570e-01]],

         [[5.5420e-01, 8.2493e-05, 9.3460e-05,  ..., 1.6068e-02,
           2.7542e-03, 1.6296e-02]],

         [[5.6348e-01, 6.3777e-05, 7.8678e-06,  ..., 9.3750e-02,
           2.1045e-01, 1.1932e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.3371e-03, 8.3685e-04,  ..., 3.3783e-02,
           3.4698e-02, 1.8816e-03]],

         [[4.0991e-01, 1.5736e-04, 3.8648e-04,  ..., 3.0106e-02,
           3.5492e-02, 6.9466e-03]],

         [[2.8369e-01, 2.0921e-05, 7.5698e-05,  ..., 1.0262e-02,
           8.2970e-04, 1.3634e-02]],

         ...,

         [[4.6094e-01, 3.3379e-06, 1.7881e-07,  ..., 5.0354e-02,
           4.6802e-01, 1.8448e-02]],

         [[3.4082e-01, 6.7759e-04, 8.2207e-04,  ..., 1.1795e-02,
           1.8482e-03, 7.6904e-03]],

         [[8.0078e-01, 2.5034e-06, 3.5763e-07,  ..., 3.3020e-02,
           1.3086e-01, 3.3264e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 5.2154e-05, 1.2338e-05,  ..., 2.0584e-02,
           1.3039e-02, 3.4241e-02]],

         [[9.7290e-02, 4.8566e-04, 4.1866e-04,  ..., 2.5463e-04,
           1.5569e-04, 3.4389e-03]],

         [[3.6743e-01, 9.0539e-05, 7.3314e-05,  ..., 5.7869e-03,
           8.8959e-03, 7.5912e-03]],

         ...,

         [[4.5093e-01, 2.0266e-06, 1.6689e-06,  ..., 1.2817e-01,
           3.8177e-02, 2.7863e-02]],

         [[3.1250e-01, 7.1526e-07, 1.1921e-07,  ..., 5.5225e-01,
           3.2349e-02, 6.5041e-03]],

         [[4.7314e-01, 1.6570e-05, 2.0742e-05,  ..., 1.0933e-02,
           7.3166e-03, 6.3210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9011e-02, 8.9407e-07, 4.5300e-06,  ..., 1.2007e-03,
           1.6193e-03, 9.3945e-01]],

         [[1.0059e-01, 6.3777e-06, 1.3471e-05,  ..., 6.2317e-02,
           6.5979e-02, 1.9913e-02]],

         [[2.0544e-01, 3.6550e-04, 7.3385e-04,  ..., 2.2125e-02,
           7.0457e-03, 1.1169e-02]],

         ...,

         [[2.9224e-01, 4.6790e-05, 1.5616e-05,  ..., 1.4026e-01,
           1.8079e-01, 2.7863e-02]],

         [[5.1709e-01, 1.1921e-06, 5.3644e-07,  ..., 1.1681e-02,
           2.0969e-04, 6.9656e-03]],

         [[5.7617e-01, 7.1645e-05, 2.6345e-05,  ..., 3.6865e-02,
           5.6877e-03, 1.1429e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0071e-01, 5.9009e-06, 1.9670e-06,  ..., 3.1128e-01,
           5.3809e-01, 1.4282e-02]],

         [[1.9360e-01, 2.2447e-04, 5.8293e-05,  ..., 4.5074e-02,
           5.9021e-02, 2.8214e-02]],

         [[1.5393e-01, 1.1504e-05, 3.3975e-06,  ..., 2.2705e-02,
           2.2446e-02, 8.2703e-02]],

         ...,

         [[6.9482e-01, 2.6226e-05, 1.4067e-05,  ..., 2.6016e-02,
           8.3008e-02, 1.8448e-02]],

         [[4.4922e-01, 1.3053e-05, 1.0669e-05,  ..., 8.2458e-02,
           8.5388e-02, 3.4912e-02]],

         [[6.2939e-01, 7.2896e-05, 2.5213e-05,  ..., 8.8928e-02,
           6.3538e-02, 2.7237e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9775e-01, 9.7227e-04, 1.3180e-03,  ..., 2.4915e-04,
           1.1265e-04, 7.8630e-04]],

         [[3.2275e-01, 6.2525e-05, 7.4267e-05,  ..., 1.6088e-03,
           1.3866e-03, 1.3908e-02]],

         [[3.8940e-01, 2.7418e-06, 4.8876e-06,  ..., 6.5308e-02,
           4.9500e-02, 1.5884e-02]],

         ...,

         [[2.9102e-01, 2.2030e-04, 4.0233e-05,  ..., 2.4078e-02,
           7.8583e-03, 2.3056e-02]],

         [[4.6069e-01, 7.1824e-05, 6.3777e-05,  ..., 5.8234e-05,
           4.8280e-06, 1.3382e-02]],

         [[7.5000e-01, 1.0866e-04, 1.2052e-04,  ..., 1.7500e-03,
           3.1776e-03, 8.5735e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1248e-02, 4.5955e-05, 4.0233e-05,  ..., 3.6144e-04,
           6.0034e-04, 4.0703e-03]],

         [[1.7676e-01, 3.5763e-04, 2.2125e-04,  ..., 3.8910e-03,
           1.9875e-03, 2.3926e-02]],

         [[6.5491e-02, 7.5698e-06, 4.6492e-06,  ..., 2.7954e-02,
           2.1271e-02, 2.4915e-01]],

         ...,

         [[2.0972e-01, 5.5432e-06, 5.7220e-06,  ..., 3.3051e-02,
           1.0675e-01, 1.4275e-02]],

         [[1.7749e-01, 1.7285e-06, 2.2054e-06,  ..., 3.8159e-01,
           3.1641e-01, 5.6274e-02]],

         [[1.1700e-01, 1.1444e-05, 8.1658e-06,  ..., 1.9409e-02,
           9.9304e-02, 3.3862e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9326e-01, 2.0623e-05, 4.2319e-06,  ..., 3.5614e-02,
           1.7365e-02, 4.3640e-02]],

         [[2.8488e-02, 5.4240e-06, 6.2585e-06,  ..., 3.6011e-02,
           1.0002e-02, 1.3046e-02]],

         [[1.4624e-01, 8.6427e-06, 2.1458e-06,  ..., 3.4409e-03,
           3.1143e-02, 7.5439e-01]],

         ...,

         [[1.3940e-01, 2.3842e-04, 8.8751e-05,  ..., 3.1776e-03,
           2.3422e-03, 1.4351e-02]],

         [[2.4829e-01, 1.7357e-04, 9.5129e-05,  ..., 1.7166e-02,
           1.5747e-02, 3.5217e-02]],

         [[3.9380e-01, 3.7050e-04, 3.0470e-04,  ..., 1.2484e-03,
           7.6008e-04, 1.8597e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4905e-01, 2.4736e-05, 1.5914e-05,  ..., 2.0660e-02,
           2.7359e-02, 4.5471e-02]],

         [[7.0654e-01, 8.8573e-05, 6.8545e-06,  ..., 3.8457e-04,
           4.5052e-03, 2.1782e-03]],

         [[9.2383e-01, 7.3910e-05, 2.7955e-05,  ..., 5.6553e-04,
           9.4748e-04, 1.6117e-03]],

         ...,

         [[1.6028e-01, 1.4086e-03, 2.4338e-03,  ..., 1.3153e-02,
           1.0406e-02, 1.6113e-02]],

         [[3.6206e-01, 7.0333e-06, 9.5367e-06,  ..., 1.7102e-01,
           3.0957e-01, 4.7485e-02]],

         [[3.5522e-01, 6.5947e-04, 1.7242e-03,  ..., 2.1011e-02,
           3.2776e-02, 4.5532e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4180e-01, 9.3937e-05, 1.6952e-04,  ..., 5.1498e-03,
           9.3231e-03, 1.7014e-02]],

         [[2.7222e-01, 1.2875e-04, 2.2602e-04,  ..., 3.3741e-03,
           5.9204e-03, 2.8915e-02]],

         [[2.1790e-01, 2.1338e-05, 4.1306e-05,  ..., 7.0038e-03,
           1.2009e-02, 3.1372e-02]],

         ...,

         [[4.5441e-02, 3.5465e-05, 6.3181e-05,  ..., 2.1708e-04,
           1.4460e-04, 2.2842e-02]],

         [[9.3140e-02, 2.7120e-05, 1.9848e-05,  ..., 2.9011e-03,
           1.3704e-03, 5.3070e-02]],

         [[5.4297e-01, 1.7512e-04, 2.3568e-04,  ..., 2.6199e-02,
           2.7023e-02, 3.0151e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3667e-01, 6.7949e-06, 7.9870e-06,  ..., 1.7029e-02,
           2.8961e-02, 5.0812e-02]],

         [[3.0493e-01, 4.6611e-05, 1.1957e-04,  ..., 1.1215e-03,
           1.8196e-03, 4.5586e-03]],

         [[1.3660e-01, 3.6538e-05, 1.0777e-04,  ..., 5.1308e-03,
           3.6469e-02, 3.8916e-01]],

         ...,

         [[3.6841e-01, 4.2379e-05, 5.4181e-05,  ..., 4.8706e-02,
           1.5002e-01, 1.1237e-01]],

         [[2.3621e-01, 2.6822e-06, 1.4067e-05,  ..., 9.0647e-04,
           3.0851e-04, 6.2561e-02]],

         [[2.9321e-01, 1.3185e-04, 2.2864e-04,  ..., 3.0182e-02,
           3.4729e-02, 4.3144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2058e-01, 1.0014e-04, 5.6982e-05,  ..., 3.9551e-02,
           4.8828e-02, 5.3192e-02]],

         [[5.5603e-02, 3.0398e-05, 2.0266e-05,  ..., 8.2825e-02,
           6.0577e-02, 4.6448e-02]],

         [[3.0591e-01, 1.1975e-04, 1.4782e-04,  ..., 6.3972e-03,
           1.3390e-02, 2.5513e-02]],

         ...,

         [[5.9570e-01, 7.2122e-06, 2.3842e-07,  ..., 2.2751e-02,
           2.9492e-01, 7.9407e-02]],

         [[1.9958e-01, 2.1040e-05, 3.6418e-05,  ..., 4.8157e-02,
           2.7008e-02, 7.5745e-02]],

         [[3.0737e-01, 2.0564e-05, 1.2577e-05,  ..., 5.7739e-02,
           5.4785e-01, 3.8452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2392e-03, 8.3447e-07, 2.7418e-06,  ..., 4.8399e-04,
           2.2469e-03, 1.9714e-02]],

         [[2.5040e-02, 1.5497e-06, 4.7088e-06,  ..., 7.9823e-04,
           3.8483e-02, 9.1064e-01]],

         [[2.6962e-02, 2.7239e-05, 1.0455e-04,  ..., 1.8239e-04,
           6.4659e-04, 2.0466e-03]],

         ...,

         [[5.1910e-02, 1.1736e-04, 3.4237e-04,  ..., 7.9956e-03,
           8.0566e-03, 1.1490e-02]],

         [[2.2278e-01, 5.9605e-07, 4.1723e-07,  ..., 1.9089e-02,
           5.6976e-02, 2.8992e-02]],

         [[1.3171e-01, 7.1526e-07, 7.1526e-07,  ..., 2.4695e-01,
           3.2983e-01, 8.0505e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8945e-01, 1.6201e-04, 9.1791e-04,  ..., 1.3977e-02,
           9.3842e-03, 1.3702e-02]],

         [[9.8291e-01, 6.7532e-05, 5.7757e-05,  ..., 3.6645e-04,
           1.8203e-04, 9.9182e-04]],

         [[8.7695e-01, 3.5703e-05, 1.2958e-04,  ..., 1.4290e-02,
           4.3221e-03, 1.1749e-02]],

         ...,

         [[1.4612e-01, 1.4305e-06, 3.4571e-06,  ..., 1.9971e-01,
           2.6465e-01, 5.8105e-02]],

         [[3.9697e-01, 1.9670e-06, 8.9407e-07,  ..., 1.1873e-03,
           2.0676e-03, 1.2836e-03]],

         [[1.3574e-01, 1.1325e-05, 3.5167e-05,  ..., 4.4227e-04,
           1.1497e-02, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1504e-02, 2.8014e-06, 1.2040e-05,  ..., 3.4149e-02,
           4.8157e-02, 4.3152e-02]],

         [[8.0750e-02, 1.0133e-06, 1.1325e-06,  ..., 2.2473e-01,
           6.0156e-01, 4.1077e-02]],

         [[2.4890e-01, 2.1136e-04, 1.9538e-04,  ..., 7.0435e-02,
           1.4062e-01, 7.1838e-02]],

         ...,

         [[3.2007e-01, 2.8014e-06, 1.2279e-05,  ..., 3.1677e-02,
           1.9409e-01, 7.7820e-02]],

         [[6.2402e-01, 1.0848e-05, 3.9697e-05,  ..., 1.6754e-02,
           8.3740e-02, 4.3793e-02]],

         [[4.2773e-01, 5.9605e-06, 2.7716e-05,  ..., 3.4302e-02,
           8.4839e-02, 2.5482e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5221e-02, 3.5763e-06, 1.4305e-06,  ..., 1.6975e-04,
           2.9445e-04, 2.5681e-02]],

         [[2.8491e-01, 6.9141e-06, 2.9206e-06,  ..., 1.0063e-02,
           3.4302e-02, 1.0063e-02]],

         [[2.4829e-01, 4.6372e-05, 1.0163e-04,  ..., 3.7476e-02,
           1.2683e-01, 4.8492e-02]],

         ...,

         [[1.5213e-02, 9.4771e-06, 1.2457e-04,  ..., 3.0112e-04,
           2.0580e-03, 9.4223e-04]],

         [[7.6965e-02, 1.8001e-05, 1.3530e-04,  ..., 1.6844e-04,
           6.1607e-04, 2.9392e-03]],

         [[1.7310e-01, 2.1040e-05, 1.9729e-05,  ..., 1.1597e-02,
           9.7107e-02, 6.3660e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6549e-02, 5.9605e-07, 9.5367e-07,  ..., 8.5742e-01,
           3.6224e-02, 4.2763e-03]],

         [[3.4595e-01, 4.6492e-06, 3.6001e-05,  ..., 4.8370e-03,
           4.6768e-03, 9.3811e-02]],

         [[1.6626e-01, 8.9407e-07, 1.0729e-06,  ..., 9.1064e-02,
           6.7285e-01, 5.0690e-02]],

         ...,

         [[5.7471e-01, 5.1081e-05, 6.2883e-05,  ..., 2.3766e-03,
           5.7487e-03, 1.3206e-02]],

         [[1.5039e-01, 1.7881e-06, 3.2783e-06,  ..., 5.3520e-03,
           5.0278e-03, 5.1460e-03]],

         [[6.0577e-03, 1.0788e-05, 2.4021e-05,  ..., 7.4806e-03,
           1.5381e-01, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0483e-01, 3.0518e-05, 1.3244e-04,  ..., 4.4647e-02,
           2.4512e-01, 5.3009e-02]],

         [[2.8735e-01, 1.4877e-04, 1.8907e-04,  ..., 1.3069e-02,
           8.6746e-03, 1.2375e-02]],

         [[1.3086e-01, 1.9181e-04, 2.9135e-04,  ..., 1.7605e-03,
           8.7967e-03, 3.5686e-03]],

         ...,

         [[5.8887e-01, 6.1321e-04, 3.7766e-04,  ..., 1.0132e-02,
           1.2341e-01, 1.0452e-02]],

         [[9.4299e-02, 1.3745e-04, 4.9496e-04,  ..., 1.8570e-02,
           2.1378e-02, 7.3047e-01]],

         [[6.6895e-01, 1.0252e-05, 2.5749e-05,  ..., 2.8931e-02,
           5.5756e-02, 7.4310e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9854e-01, 4.6432e-05, 4.2772e-04,  ..., 8.4961e-02,
           2.0142e-01, 3.2959e-02]],

         [[4.9774e-02, 3.7551e-06, 1.8299e-05,  ..., 1.3290e-02,
           7.9346e-03, 8.5999e-02]],

         [[4.2798e-01, 3.0684e-04, 7.2050e-04,  ..., 3.9429e-02,
           4.2877e-02, 3.6896e-02]],

         ...,

         [[4.1626e-01, 1.2112e-04, 7.0620e-04,  ..., 2.8038e-03,
           4.8637e-03, 5.8899e-03]],

         [[8.0225e-01, 2.2650e-05, 1.3828e-03,  ..., 4.3564e-03,
           5.8784e-03, 3.8853e-03]],

         [[2.9321e-01, 4.1962e-05, 1.3065e-04,  ..., 3.8376e-03,
           5.0354e-02, 1.6663e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2201e-04, 8.8274e-05, 8.8501e-02,  ..., 1.0023e-03,
           4.3716e-03, 4.7455e-03]],

         [[2.0444e-05, 2.3901e-05, 7.1167e-02,  ..., 1.0460e-02,
           6.8855e-03, 8.5205e-02]],

         [[2.3770e-04, 2.0850e-04, 4.9927e-02,  ..., 2.0981e-02,
           1.4502e-01, 2.7908e-02]],

         ...,

         [[4.7374e-04, 7.7009e-05, 7.7515e-02,  ..., 9.8755e-02,
           1.9800e-01, 1.8591e-01]],

         [[4.1890e-04, 1.7366e-03, 2.3462e-01,  ..., 8.5449e-03,
           1.5717e-02, 3.7632e-03]],

         [[1.9431e-05, 8.7082e-05, 1.6113e-01,  ..., 9.8896e-04,
           8.1301e-04, 1.1740e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5739e-04, 7.9632e-04, 3.0762e-01,  ..., 4.9829e-04,
           2.9240e-03, 2.6302e-03]],

         [[1.7941e-04, 2.7347e-04, 1.0242e-01,  ..., 2.3758e-04,
           1.1092e-04, 1.3723e-03]],

         [[1.6749e-04, 1.0711e-04, 2.8833e-01,  ..., 5.9605e-05,
           2.1172e-03, 7.2598e-05]],

         ...,

         [[4.6277e-04, 4.7374e-04, 2.6440e-01,  ..., 1.1124e-02,
           2.4071e-03, 8.9931e-04]],

         [[2.8670e-05, 1.0788e-04, 2.4902e-01,  ..., 7.4744e-05,
           4.9591e-04, 3.7730e-05]],

         [[8.3387e-05, 3.9220e-04, 1.0626e-01,  ..., 1.0520e-04,
           7.5006e-04, 4.6778e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8203e-04, 1.2445e-04, 1.1975e-01,  ..., 1.0312e-05,
           2.5806e-03, 8.9407e-06]],

         [[7.0333e-04, 3.6895e-05, 2.7393e-01,  ..., 3.2024e-03,
           1.7035e-04, 1.5545e-03]],

         [[8.5449e-04, 7.2956e-04, 2.7441e-01,  ..., 3.6621e-04,
           2.7537e-04, 1.2755e-04]],

         ...,

         [[1.1927e-04, 1.0705e-04, 1.3074e-01,  ..., 9.4681e-03,
           4.8462e-02, 1.1467e-02]],

         [[1.0026e-04, 3.2067e-04, 2.4915e-01,  ..., 5.3358e-04,
           9.8896e-04, 1.8730e-03]],

         [[2.7132e-04, 1.8501e-04, 1.3464e-01,  ..., 1.0767e-03,
           1.2064e-03, 1.2789e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2034e-04, 2.2709e-05, 1.8600e-02,  ..., 8.7402e-02,
           6.3904e-02, 8.4045e-02]],

         [[8.3387e-05, 1.7464e-04, 9.8816e-02,  ..., 7.3700e-03,
           2.2659e-02, 2.4399e-02]],

         [[7.8678e-04, 3.8433e-03, 1.9336e-01,  ..., 1.1158e-03,
           7.6103e-04, 1.7185e-03]],

         ...,

         [[1.7166e-04, 2.8014e-06, 3.1006e-02,  ..., 2.8553e-03,
           5.5218e-04, 2.6520e-02]],

         [[3.0537e-03, 2.9349e-04, 6.2622e-02,  ..., 1.8701e-01,
           2.3096e-01, 5.1331e-02]],

         [[5.7817e-06, 2.4438e-06, 1.5039e-03,  ..., 2.6722e-03,
           1.7109e-03, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2120e-03, 1.3618e-03, 2.2388e-01,  ..., 2.9202e-03,
           3.8185e-03, 4.5135e-02]],

         [[1.2531e-03, 1.3628e-03, 2.5317e-01,  ..., 6.5947e-04,
           1.3628e-03, 2.1698e-02]],

         [[2.8515e-03, 1.4219e-03, 2.7637e-01,  ..., 5.2071e-03,
           9.2239e-03, 7.0457e-03]],

         ...,

         [[1.4629e-03, 1.7965e-04, 2.2766e-01,  ..., 1.7548e-02,
           4.1046e-03, 9.6375e-02]],

         [[3.1972e-04, 8.2922e-04, 1.0864e-01,  ..., 5.3345e-02,
           1.0779e-01, 2.2644e-01]],

         [[3.1590e-05, 3.4273e-05, 1.5747e-02,  ..., 1.1011e-01,
           3.8135e-01, 4.3457e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[7.7546e-05, 5.2166e-04, 6.7635e-03,  ..., 7.7393e-02,
           5.5420e-01, 1.6394e-01]],

         [[1.0413e-04, 5.6314e-04, 7.7705e-03,  ..., 1.3641e-02,
           4.4739e-02, 2.4182e-01]],

         [[0.0000e+00, 1.7881e-07, 2.7061e-05,  ..., 3.4750e-05,
           8.5068e-03, 9.8340e-01]],

         ...,

         [[0.0000e+00, 2.3842e-07, 4.7088e-06,  ..., 2.0146e-04,
           2.4529e-03, 9.8975e-01]],

         [[4.3373e-03, 5.2299e-03, 8.6212e-03,  ..., 1.7151e-02,
           1.5137e-02, 2.2018e-02]],

         [[1.9958e-02, 8.8577e-03, 2.4605e-03,  ..., 2.6443e-02,
           3.5034e-02, 2.9053e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6652e-05, 1.4329e-04, 4.3182e-03,  ..., 1.6556e-02,
           6.5491e-02, 4.1382e-01]],

         [[0.0000e+00, 1.1921e-07, 1.3113e-06,  ..., 1.5173e-01,
           7.7051e-01, 6.8359e-02]],

         [[9.6083e-04, 2.9583e-03, 1.4114e-02,  ..., 7.7972e-03,
           1.0994e-02, 2.2552e-02]],

         ...,

         [[1.3113e-05, 3.0613e-04, 6.3801e-04,  ..., 2.1851e-01,
           4.4849e-01, 2.3999e-01]],

         [[2.7447e-03, 8.3542e-03, 2.5436e-02,  ..., 1.1734e-02,
           7.3433e-03, 9.2850e-03]],

         [[1.6861e-03, 5.7487e-03, 6.7711e-03,  ..., 8.2520e-02,
           2.5970e-02, 2.3636e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1768e-02, 4.8370e-03, 1.3466e-02,  ..., 6.7871e-02,
           4.6997e-02, 5.8502e-02]],

         [[1.3371e-03, 4.9019e-04, 1.9379e-03,  ..., 6.2332e-03,
           6.2418e-04, 3.9291e-03]],

         [[4.6539e-03, 9.3460e-04, 1.0843e-03,  ..., 1.6891e-02,
           1.2260e-02, 1.5259e-02]],

         ...,

         [[4.9257e-04, 2.2087e-03, 3.8452e-03,  ..., 6.9763e-02,
           7.2705e-01, 1.7273e-01]],

         [[3.4165e-04, 4.4823e-05, 4.4107e-05,  ..., 8.3618e-02,
           2.0966e-02, 2.7145e-02]],

         [[3.3951e-03, 9.5034e-04, 1.7380e-02,  ..., 1.8646e-02,
           2.4506e-02, 3.5449e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7354e-01, 1.8477e-06, 1.4305e-06,  ..., 2.2202e-03,
           7.9346e-03, 1.8740e-03]],

         [[8.6914e-01, 3.3379e-06, 3.6359e-06,  ..., 1.3893e-02,
           2.9297e-02, 7.0620e-04]],

         [[6.0352e-01, 1.9670e-06, 7.9870e-06,  ..., 6.5613e-02,
           2.4147e-03, 1.3293e-01]],

         ...,

         [[4.7803e-01, 2.3949e-04, 8.7357e-04,  ..., 1.0365e-04,
           1.3649e-05, 1.0834e-01]],

         [[4.2871e-01, 5.3024e-04, 6.4993e-04,  ..., 1.8051e-02,
           1.4389e-02, 5.4718e-02]],

         [[4.8560e-01, 4.0245e-04, 8.5354e-04,  ..., 1.2787e-02,
           7.7286e-03, 1.6678e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6377e-01, 3.7432e-04, 9.4175e-06,  ..., 2.7065e-03,
           1.5855e-05, 4.4647e-02]],

         [[9.2920e-01, 9.7871e-05, 1.6689e-06,  ..., 1.5198e-02,
           2.1362e-02, 3.2654e-02]],

         [[9.4434e-01, 2.8300e-04, 2.3818e-04,  ..., 4.0531e-04,
           9.7942e-04, 6.0349e-03]],

         ...,

         [[4.0112e-01, 1.9801e-04, 1.2034e-04,  ..., 1.4877e-02,
           4.2381e-03, 1.7212e-02]],

         [[1.7151e-01, 1.6737e-03, 5.4455e-04,  ..., 5.5199e-03,
           3.7403e-03, 4.8027e-03]],

         [[8.0273e-01, 1.2579e-03, 3.8671e-04,  ..., 6.8115e-02,
           3.4973e-02, 1.5015e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2695e-01, 1.9729e-04, 8.6546e-05,  ..., 1.0858e-01,
           7.5256e-02, 3.2990e-02]],

         [[6.5332e-01, 9.0897e-05, 1.0252e-05,  ..., 7.1594e-02,
           2.3300e-02, 3.9154e-02]],

         [[4.4580e-01, 1.7464e-05, 3.1590e-05,  ..., 2.6779e-02,
           4.7424e-02, 4.5288e-02]],

         ...,

         [[7.9590e-01, 1.1665e-04, 1.1325e-06,  ..., 4.7684e-07,
           1.6809e-05, 1.9702e-01]],

         [[8.6279e-01, 5.3453e-04, 5.6028e-06,  ..., 0.0000e+00,
           2.9802e-07, 3.8391e-02]],

         [[9.7168e-01, 3.8266e-04, 2.3842e-07,  ..., 0.0000e+00,
           5.9605e-08, 8.5754e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 8.9407e-06, 1.8477e-06,  ..., 1.9287e-01,
           4.2700e-01, 1.8784e-02]],

         [[9.4727e-01, 9.5963e-06, 1.8477e-06,  ..., 3.1590e-04,
           6.1750e-04, 1.5167e-02]],

         [[7.7539e-01, 1.3232e-04, 5.0843e-05,  ..., 1.8845e-03,
           4.7836e-03, 1.4397e-02]],

         ...,

         [[8.8672e-01, 2.8992e-04, 2.4199e-05,  ..., 2.3518e-03,
           2.9999e-02, 6.9122e-03]],

         [[6.5674e-01, 4.3392e-04, 6.8247e-05,  ..., 5.0201e-02,
           1.5312e-02, 1.3110e-01]],

         [[8.3984e-01, 8.5592e-05, 7.7665e-05,  ..., 2.2869e-03,
           1.5297e-03, 1.3318e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0732e-01, 8.8215e-06, 3.6955e-06,  ..., 2.2424e-01,
           4.0497e-02, 9.8999e-02]],

         [[6.7578e-01, 4.6194e-05, 1.3113e-05,  ..., 6.6490e-03,
           2.5635e-02, 1.0828e-01]],

         [[7.4658e-01, 6.8545e-06, 1.2517e-06,  ..., 7.7591e-03,
           1.2634e-01, 7.5195e-02]],

         ...,

         [[4.5386e-01, 5.1916e-05, 3.7372e-05,  ..., 1.4221e-01,
           7.3059e-02, 4.7729e-02]],

         [[8.3594e-01, 1.3900e-04, 9.2208e-05,  ..., 1.7502e-02,
           4.4441e-03, 9.5520e-03]],

         [[7.4609e-01, 3.6716e-05, 1.5557e-05,  ..., 6.8970e-02,
           2.1912e-02, 2.4536e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3760e-01, 1.4038e-03, 6.4993e-04,  ..., 4.6005e-03,
           4.3907e-03, 4.5128e-03]],

         [[9.1211e-01, 4.9591e-05, 8.4341e-05,  ..., 5.5771e-03,
           1.2989e-03, 4.1122e-03]],

         [[7.8711e-01, 4.4107e-05, 1.1683e-05,  ..., 5.6488e-02,
           3.0945e-02, 1.2405e-02]],

         ...,

         [[9.1162e-01, 4.0174e-05, 2.0444e-05,  ..., 3.0403e-03,
           2.8458e-03, 1.9852e-02]],

         [[6.4795e-01, 4.1723e-06, 3.3200e-05,  ..., 2.0046e-03,
           5.5389e-03, 7.8003e-02]],

         [[9.6289e-01, 5.6624e-06, 4.5896e-06,  ..., 3.4313e-03,
           2.7905e-03, 7.6981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3184e-01, 1.7512e-04, 7.2575e-04,  ..., 1.1301e-03,
           1.6975e-03, 1.5869e-02]],

         [[8.8477e-01, 1.3769e-05, 1.8537e-05,  ..., 1.8066e-02,
           1.9577e-02, 5.2368e-02]],

         [[4.7827e-01, 2.8610e-05, 3.5810e-04,  ..., 6.6833e-02,
           2.3911e-02, 1.5587e-02]],

         ...,

         [[4.0259e-01, 1.3697e-04, 2.2399e-04,  ..., 1.0204e-03,
           3.1929e-03, 1.4563e-03]],

         [[7.9443e-01, 8.5175e-05, 3.9577e-05,  ..., 1.2039e-02,
           1.1337e-02, 6.7139e-02]],

         [[6.2793e-01, 1.4961e-05, 1.7583e-05,  ..., 7.1678e-03,
           1.0109e-03, 1.8848e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0419e-04, 1.4985e-04,  ..., 3.8028e-05,
           4.5896e-06, 1.1719e-02]],

         [[8.6084e-01, 2.2888e-05, 3.6538e-05,  ..., 9.7275e-03,
           5.7983e-03, 1.5366e-02]],

         [[8.9160e-01, 1.5318e-05, 2.2471e-05,  ..., 6.0349e-03,
           7.9823e-04, 1.1253e-02]],

         ...,

         [[9.2969e-01, 7.7367e-05, 5.0545e-05,  ..., 3.0117e-03,
           3.2959e-03, 7.5722e-03]],

         [[8.0762e-01, 1.2183e-04, 1.2088e-04,  ..., 9.9277e-04,
           2.4529e-03, 2.6302e-03]],

         [[6.2891e-01, 9.8324e-04, 3.1304e-04,  ..., 2.9011e-03,
           3.3321e-03, 1.0254e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2290e-01, 6.5565e-07, 1.3828e-05,  ..., 7.4482e-04,
           1.5137e-02, 7.5391e-01]],

         [[8.8281e-01, 3.3212e-04, 5.7316e-04,  ..., 3.6793e-03,
           3.1643e-03, 7.0572e-04]],

         [[7.1191e-01, 5.1641e-04, 7.3957e-04,  ..., 1.0872e-02,
           1.7685e-02, 1.3580e-02]],

         ...,

         [[6.3037e-01, 9.6321e-05, 1.9836e-04,  ..., 6.8436e-03,
           7.3166e-03, 1.9951e-03]],

         [[7.7197e-01, 7.7605e-05, 3.5226e-05,  ..., 1.2138e-02,
           5.7678e-02, 5.1849e-02]],

         [[2.4426e-01, 1.2922e-04, 1.8871e-04,  ..., 2.2812e-03,
           1.9821e-02, 5.9521e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3711e-01, 3.7193e-05, 4.2498e-05,  ..., 6.8321e-03,
           1.3245e-02, 3.8647e-01]],

         [[7.3828e-01, 9.5189e-05, 1.1486e-04,  ..., 9.5978e-03,
           1.9852e-02, 7.1045e-02]],

         [[3.0469e-01, 7.9775e-04, 1.4472e-04,  ..., 3.6530e-02,
           4.7760e-02, 1.9699e-02]],

         ...,

         [[9.5215e-02, 3.4571e-06, 7.9870e-06,  ..., 1.3947e-05,
           2.1279e-04, 8.9795e-01]],

         [[8.1445e-01, 6.5804e-05, 5.6922e-05,  ..., 7.8726e-04,
           5.0545e-03, 3.3142e-02]],

         [[9.1016e-01, 1.4532e-04, 3.9577e-05,  ..., 1.7319e-02,
           1.9684e-03, 1.1055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2695e-01, 1.4067e-03, 1.4009e-03,  ..., 1.6510e-02,
           3.4885e-03, 2.7649e-02]],

         [[8.5840e-01, 7.4863e-05, 1.1772e-04,  ..., 6.0272e-03,
           5.1231e-03, 3.3905e-02]],

         [[6.4600e-01, 3.0971e-04, 1.1003e-04,  ..., 3.3512e-03,
           1.4696e-03, 1.0208e-02]],

         ...,

         [[6.1768e-01, 5.1498e-05, 8.6427e-06,  ..., 1.6953e-02,
           3.0103e-01, 6.1218e-02]],

         [[3.8940e-01, 5.2071e-04, 5.6982e-04,  ..., 5.5962e-03,
           1.4549e-02, 1.4629e-03]],

         [[8.1152e-01, 4.1068e-05, 1.4007e-05,  ..., 2.1667e-02,
           1.4062e-01, 2.3407e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5439e-01, 7.7784e-05, 1.5914e-05,  ..., 5.7755e-03,
           2.1820e-02, 1.8723e-02]],

         [[8.1360e-02, 1.7233e-03, 1.5736e-03,  ..., 1.6856e-04,
           1.7166e-03, 4.7455e-03]],

         [[6.2256e-01, 2.5249e-04, 2.1517e-04,  ..., 2.6779e-03,
           1.1047e-02, 2.6215e-02]],

         ...,

         [[4.6875e-01, 1.8775e-05, 5.3048e-06,  ..., 3.5583e-02,
           7.1350e-02, 1.4380e-01]],

         [[7.4414e-01, 7.1526e-06, 2.4438e-06,  ..., 2.5055e-02,
           5.1483e-02, 7.8674e-02]],

         [[3.4204e-01, 5.6803e-05, 5.9783e-05,  ..., 7.2060e-03,
           1.9516e-02, 3.5736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0444e-02, 1.1921e-07, 4.7684e-07,  ..., 1.6439e-04,
           2.2697e-03, 9.1016e-01]],

         [[3.5132e-01, 2.9325e-05, 4.9710e-05,  ..., 6.6650e-02,
           9.0210e-02, 8.2642e-02]],

         [[1.8896e-01, 4.5991e-04, 7.5245e-04,  ..., 6.6910e-03,
           1.8173e-02, 3.3020e-02]],

         ...,

         [[4.3311e-01, 1.7190e-04, 1.3340e-04,  ..., 8.5938e-02,
           9.1309e-02, 6.3553e-03]],

         [[4.3311e-01, 7.6294e-06, 1.4305e-06,  ..., 1.5710e-01,
           2.6260e-02, 3.5858e-02]],

         [[1.7578e-01, 1.2076e-04, 7.2122e-06,  ..., 4.1771e-03,
           2.9144e-03, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2646e-01, 2.5332e-05, 2.1636e-05,  ..., 7.7515e-02,
           1.4877e-02, 4.0039e-02]],

         [[2.3157e-01, 1.1045e-04, 4.2319e-05,  ..., 3.1543e-01,
           8.7585e-02, 2.6291e-02]],

         [[7.1436e-01, 7.4267e-05, 9.0659e-05,  ..., 1.0071e-02,
           1.5594e-02, 4.6753e-02]],

         ...,

         [[8.7256e-01, 3.8564e-05, 1.6272e-05,  ..., 2.2186e-02,
           2.0279e-02, 7.8201e-03]],

         [[8.1738e-01, 2.6643e-05, 5.4896e-05,  ..., 2.9716e-03,
           7.5722e-03, 4.4861e-02]],

         [[5.3271e-01, 1.0490e-04, 3.1650e-05,  ..., 1.0327e-01,
           1.1084e-01, 1.0391e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5747e-01, 8.1015e-04, 1.1702e-03,  ..., 4.0650e-04,
           4.7779e-04, 1.9970e-03]],

         [[1.5698e-01, 1.2231e-04, 1.0985e-04,  ..., 6.5088e-04,
           7.1831e-03, 3.9307e-02]],

         [[6.6260e-01, 1.4126e-05, 3.7968e-05,  ..., 2.5101e-02,
           7.7576e-02, 2.4796e-02]],

         ...,

         [[8.3789e-01, 6.9022e-05, 1.6749e-05,  ..., 4.0588e-03,
           8.5926e-04, 2.2720e-02]],

         [[9.2578e-01, 2.6643e-05, 1.7703e-05,  ..., 4.7684e-07,
           4.7684e-07, 1.5495e-02]],

         [[5.9473e-01, 2.2876e-04, 4.1580e-04,  ..., 2.8992e-02,
           1.3115e-02, 5.6915e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8196e-02, 1.7571e-04, 1.0574e-04,  ..., 4.3907e-03,
           1.3847e-02, 2.3376e-02]],

         [[1.5723e-01, 3.4380e-04, 1.5497e-04,  ..., 5.1308e-03,
           4.6814e-02, 4.7180e-02]],

         [[5.1562e-01, 4.2915e-05, 2.7597e-05,  ..., 1.1215e-02,
           1.5091e-02, 1.8530e-01]],

         ...,

         [[2.5732e-01, 5.1260e-06, 7.6294e-06,  ..., 2.2180e-01,
           7.1472e-02, 6.1722e-03]],

         [[7.0557e-01, 9.0599e-06, 6.3777e-06,  ..., 1.8799e-02,
           2.7145e-02, 1.2854e-01]],

         [[1.9397e-01, 1.2398e-05, 8.9407e-06,  ..., 2.4460e-02,
           5.3436e-02, 6.5088e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-01, 6.2168e-05, 1.4305e-05,  ..., 1.1780e-01,
           4.9103e-02, 2.6276e-02]],

         [[2.0850e-01, 3.5167e-05, 6.6757e-05,  ..., 4.5105e-02,
           1.0651e-01, 4.8004e-02]],

         [[2.5708e-01, 1.6987e-05, 3.5763e-06,  ..., 1.3494e-03,
           3.3913e-03, 7.2119e-01]],

         ...,

         [[7.8979e-02, 2.2447e-04, 2.1219e-05,  ..., 2.4498e-05,
           7.2002e-05, 5.2612e-02]],

         [[3.4521e-01, 8.4400e-05, 3.7491e-05,  ..., 1.2138e-02,
           2.5009e-02, 4.8294e-03]],

         [[1.4075e-01, 7.4744e-05, 4.2498e-05,  ..., 1.9501e-02,
           6.5979e-02, 1.4725e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5254e-01, 2.6166e-05, 1.4067e-05,  ..., 2.2369e-02,
           3.4637e-02, 3.8635e-02]],

         [[9.7803e-01, 4.9531e-05, 1.1027e-05,  ..., 5.1260e-06,
           5.0068e-06, 1.7560e-04]],

         [[8.3350e-01, 2.6679e-04, 1.6630e-04,  ..., 2.7943e-04,
           1.1864e-03, 3.2234e-03]],

         ...,

         [[9.8999e-02, 6.6710e-04, 9.6321e-04,  ..., 2.7710e-02,
           1.9501e-02, 3.6407e-02]],

         [[6.5820e-01, 3.2842e-05, 3.2604e-05,  ..., 3.0792e-02,
           3.8940e-02, 2.4750e-02]],

         [[2.6904e-01, 5.6362e-04, 1.0958e-03,  ..., 3.8452e-02,
           4.7485e-02, 3.9062e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 1.2279e-04, 1.0175e-04,  ..., 8.3557e-02,
           8.8989e-02, 2.1973e-02]],

         [[2.7563e-01, 7.7844e-05, 1.1969e-04,  ..., 8.2626e-03,
           1.3939e-02, 7.1899e-02]],

         [[2.1960e-01, 4.5002e-05, 1.0633e-04,  ..., 1.1253e-03,
           3.5496e-03, 1.1543e-02]],

         ...,

         [[3.2227e-02, 2.3842e-04, 2.6608e-04,  ..., 6.5851e-04,
           7.1907e-03, 2.2842e-02]],

         [[8.0200e-02, 2.4140e-05, 1.0371e-05,  ..., 5.1856e-05,
           4.7982e-05, 1.6980e-01]],

         [[4.9634e-01, 1.3280e-04, 1.1903e-04,  ..., 1.3863e-02,
           8.4076e-03, 2.2858e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0391e-01, 3.5286e-05, 2.6941e-05,  ..., 1.1574e-02,
           2.0950e-02, 3.3478e-02]],

         [[2.1912e-01, 4.4048e-05, 7.2062e-05,  ..., 4.8790e-03,
           2.2034e-02, 8.4305e-03]],

         [[4.0649e-01, 9.2268e-05, 1.7238e-04,  ..., 3.4084e-03,
           2.3483e-02, 2.8833e-01]],

         ...,

         [[4.0430e-01, 8.1122e-05, 6.2943e-05,  ..., 4.1962e-02,
           1.0201e-02, 1.9821e-02]],

         [[7.8320e-01, 8.6010e-05, 7.8321e-05,  ..., 7.0155e-05,
           1.0782e-04, 4.0558e-02]],

         [[7.1143e-01, 7.2777e-05, 2.1470e-04,  ..., 9.5291e-03,
           3.2673e-03, 1.4198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6953e-01, 1.6880e-04, 6.9976e-05,  ..., 2.8061e-02,
           4.0680e-02, 5.3894e-02]],

         [[2.8516e-01, 1.6117e-04, 1.5807e-04,  ..., 1.0490e-03,
           2.4338e-02, 8.3618e-02]],

         [[4.2334e-01, 1.8525e-04, 1.0681e-04,  ..., 3.8177e-02,
           3.4760e-02, 1.6937e-02]],

         ...,

         [[5.2637e-01, 8.6069e-05, 1.5438e-05,  ..., 5.4199e-02,
           3.5083e-01, 5.6366e-02]],

         [[2.9395e-01, 1.3053e-05, 4.6730e-05,  ..., 8.4229e-02,
           2.5732e-01, 1.0236e-01]],

         [[6.1377e-01, 1.4496e-04, 8.3566e-05,  ..., 5.0018e-02,
           1.8433e-01, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4399e-02, 4.6492e-06, 1.2398e-05,  ..., 8.8043e-03,
           9.8022e-02, 2.6382e-02]],

         [[3.4839e-01, 8.5175e-05, 1.3888e-04,  ..., 1.2253e-02,
           2.4094e-02, 3.0762e-01]],

         [[7.5989e-02, 2.7120e-05, 1.3745e-04,  ..., 3.1376e-03,
           1.0406e-02, 2.3727e-02]],

         ...,

         [[1.0077e-01, 1.6022e-04, 5.5075e-04,  ..., 8.7357e-03,
           1.9089e-02, 1.5396e-02]],

         [[4.8926e-01, 7.2718e-06, 2.2709e-05,  ..., 5.1361e-02,
           1.3013e-01, 1.7236e-01]],

         [[3.1299e-01, 2.9206e-06, 7.8678e-06,  ..., 8.4900e-02,
           3.5596e-01, 1.6101e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 7.9751e-05, 3.5524e-04,  ..., 4.8248e-02,
           3.4485e-02, 3.7262e-02]],

         [[9.6045e-01, 2.1219e-05, 1.1760e-04,  ..., 5.5027e-04,
           1.7557e-03, 1.2798e-03]],

         [[8.8525e-01, 1.6856e-04, 2.3866e-04,  ..., 6.8665e-03,
           1.3130e-02, 7.3662e-03]],

         ...,

         [[8.5400e-01, 7.9274e-06, 1.2279e-05,  ..., 3.9337e-02,
           1.9165e-02, 2.6611e-02]],

         [[9.1309e-01, 4.3511e-06, 4.4107e-06,  ..., 1.3094e-03,
           8.3256e-04, 6.2485e-03]],

         [[2.7930e-01, 1.0490e-05, 2.3305e-05,  ..., 4.2725e-03,
           2.5574e-02, 6.1963e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3196e-01, 5.4240e-06, 2.4319e-05,  ..., 4.1534e-02,
           7.8857e-02, 2.7295e-01]],

         [[7.2070e-01, 2.5868e-05, 1.6987e-05,  ..., 1.0065e-01,
           1.9440e-02, 1.6632e-02]],

         [[1.2988e-01, 1.3173e-04, 7.8261e-05,  ..., 9.2102e-02,
           6.2561e-02, 3.3234e-02]],

         ...,

         [[6.4307e-01, 1.3530e-05, 5.1439e-05,  ..., 2.6123e-02,
           4.7699e-02, 4.3762e-02]],

         [[4.9634e-01, 1.7166e-05, 6.6996e-05,  ..., 1.3159e-01,
           9.9304e-02, 3.1006e-02]],

         [[7.2998e-01, 8.7023e-06, 4.0472e-05,  ..., 4.1199e-02,
           1.9455e-02, 3.1082e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9785e-01, 1.2279e-05, 3.5167e-06,  ..., 5.9433e-03,
           2.8580e-02, 2.2656e-01]],

         [[6.0938e-01, 9.1195e-06, 2.5034e-06,  ..., 4.2725e-03,
           7.2765e-04, 1.1188e-01]],

         [[2.9443e-01, 5.8830e-05, 5.4955e-05,  ..., 3.4332e-02,
           3.3813e-02, 1.5039e-01]],

         ...,

         [[1.0437e-01, 2.5630e-06, 5.5492e-05,  ..., 8.0017e-02,
           2.0874e-02, 6.9519e-02]],

         [[1.2213e-01, 3.8147e-06, 3.2902e-05,  ..., 4.3549e-02,
           3.4454e-02, 4.0894e-02]],

         [[8.4473e-01, 1.7583e-05, 2.7001e-05,  ..., 1.1147e-02,
           1.3229e-02, 1.1864e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 2.8014e-06, 9.5367e-06,  ..., 1.4183e-02,
           2.0920e-02, 5.3223e-02]],

         [[5.3027e-01, 4.8935e-05, 2.2078e-04,  ..., 1.9394e-02,
           1.8909e-01, 9.0027e-02]],

         [[6.5332e-01, 1.6928e-05, 3.1531e-05,  ..., 1.2077e-02,
           1.0748e-01, 2.1875e-01]],

         ...,

         [[4.2505e-01, 2.5034e-05, 1.9252e-05,  ..., 2.8702e-02,
           6.2225e-02, 4.3678e-03]],

         [[5.6543e-01, 1.2755e-05, 6.8545e-06,  ..., 2.1408e-02,
           2.7481e-02, 1.1572e-01]],

         [[1.6833e-01, 9.6262e-05, 3.1567e-04,  ..., 9.5947e-02,
           5.7281e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3037e-01, 7.4506e-06, 1.2815e-05,  ..., 1.1955e-02,
           3.6888e-03, 2.9541e-01]],

         [[7.1582e-01, 9.1612e-05, 7.3314e-05,  ..., 1.8673e-03,
           1.1024e-03, 1.0980e-01]],

         [[4.8779e-01, 2.7239e-05, 6.2346e-05,  ..., 6.0692e-03,
           1.2817e-03, 1.7944e-01]],

         ...,

         [[5.8301e-01, 1.0383e-04, 3.2496e-04,  ..., 5.4230e-02,
           1.9028e-02, 1.0785e-01]],

         [[5.5908e-01, 3.6144e-04, 1.0796e-03,  ..., 1.5617e-02,
           3.4119e-02, 2.1228e-01]],

         [[6.0840e-01, 5.9903e-05, 1.2290e-04,  ..., 1.8372e-02,
           3.4332e-02, 2.4207e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8262e-01, 3.4392e-05, 2.5082e-04,  ..., 1.2527e-02,
           9.9121e-02, 7.5378e-02]],

         [[2.2961e-01, 1.9670e-06, 3.2127e-05,  ..., 5.0735e-03,
           5.1613e-03, 4.4604e-01]],

         [[2.5708e-01, 1.1420e-04, 7.8201e-04,  ..., 1.1060e-01,
           1.3818e-01, 7.2815e-02]],

         ...,

         [[8.3154e-01, 7.1645e-05, 4.1699e-04,  ..., 1.4847e-02,
           1.0925e-02, 4.4060e-03]],

         [[9.5068e-01, 1.7881e-07, 1.5771e-04,  ..., 5.5170e-04,
           1.2836e-03, 7.9489e-04]],

         [[7.3730e-01, 1.6868e-05, 6.4254e-05,  ..., 1.6632e-02,
           1.0010e-02, 8.0795e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1100e-05, 1.1921e-05, 2.1741e-01,  ..., 3.0785e-03,
           7.2098e-03, 5.3284e-02]],

         [[7.0333e-06, 5.0664e-06, 2.0737e-02,  ..., 3.8643e-03,
           1.2402e-01, 5.6229e-03]],

         [[1.5116e-04, 1.3387e-04, 6.7078e-02,  ..., 2.9190e-02,
           3.0838e-02, 1.4626e-02]],

         ...,

         [[5.1796e-05, 9.6560e-06, 1.4612e-01,  ..., 1.4816e-02,
           1.5320e-01, 3.1421e-01]],

         [[2.0134e-04, 4.1902e-05, 3.0518e-01,  ..., 6.0692e-03,
           7.5531e-03, 1.3733e-02]],

         [[1.1206e-05, 1.2517e-06, 2.4475e-01,  ..., 7.4911e-04,
           7.8535e-04, 6.4880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8413e-05, 1.1504e-05, 2.5977e-01,  ..., 1.1806e-03,
           2.8877e-03, 2.9617e-02]],

         [[3.0518e-05, 8.6427e-06, 2.7222e-01,  ..., 1.2636e-04,
           5.1975e-04, 3.2257e-02]],

         [[2.1875e-05, 3.1590e-06, 3.2251e-01,  ..., 1.3866e-03,
           8.3745e-05, 1.4305e-03]],

         ...,

         [[6.8367e-05, 1.4424e-05, 9.4421e-02,  ..., 2.3365e-03,
           1.2503e-03, 5.4131e-03]],

         [[3.6955e-06, 2.0266e-06, 2.8516e-01,  ..., 5.4836e-04,
           2.1815e-04, 7.3357e-03]],

         [[1.2815e-05, 1.0431e-05, 2.9712e-01,  ..., 1.0931e-04,
           7.1406e-05, 5.2643e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0192e-05, 1.7881e-06, 2.0081e-01,  ..., 1.9045e-03,
           1.2219e-05, 3.0914e-02]],

         [[1.7643e-05, 2.9802e-07, 2.3828e-01,  ..., 5.4169e-04,
           1.0738e-03, 1.3514e-03]],

         [[2.0385e-05, 4.0531e-06, 2.2229e-01,  ..., 4.7231e-04,
           1.6069e-04, 3.3112e-03]],

         ...,

         [[2.5868e-04, 3.0518e-05, 2.1802e-01,  ..., 2.3697e-02,
           1.0765e-02, 1.3428e-01]],

         [[8.3447e-06, 2.9206e-06, 1.0846e-01,  ..., 1.0490e-03,
           6.8521e-04, 1.7227e-02]],

         [[3.6836e-05, 2.9206e-06, 2.0776e-01,  ..., 1.5678e-03,
           6.4611e-04, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8651e-05, 4.2319e-06, 2.8549e-02,  ..., 5.9814e-03,
           1.2466e-02, 6.4941e-01]],

         [[8.1658e-06, 3.3498e-05, 1.3940e-01,  ..., 5.3131e-02,
           7.1228e-02, 1.1200e-02]],

         [[9.1612e-05, 2.7108e-04, 3.2861e-01,  ..., 2.4438e-04,
           4.7898e-04, 1.0252e-04]],

         ...,

         [[8.0061e-04, 1.3232e-05, 1.8530e-01,  ..., 8.1158e-04,
           4.5586e-04, 5.0926e-03]],

         [[4.5848e-04, 1.5616e-05, 9.2773e-02,  ..., 1.1040e-02,
           1.4915e-02, 5.5518e-01]],

         [[3.0351e-04, 1.2279e-05, 7.6172e-02,  ..., 1.4481e-02,
           1.0269e-02, 4.5215e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6011e-04, 1.4663e-04, 1.1298e-01,  ..., 5.1544e-02,
           1.4999e-02, 6.4392e-02]],

         [[7.7629e-04, 4.7827e-04, 2.4707e-01,  ..., 5.7869e-03,
           2.0844e-02, 6.9153e-02]],

         [[3.4523e-04, 9.9242e-05, 1.8079e-01,  ..., 1.3695e-02,
           1.2421e-02, 5.5786e-02]],

         ...,

         [[1.1581e-04, 4.4286e-05, 1.6821e-01,  ..., 3.6957e-02,
           8.5938e-02, 1.5808e-01]],

         [[3.6407e-04, 3.9196e-04, 2.7759e-01,  ..., 1.1276e-02,
           1.0996e-03, 7.8308e-02]],

         [[5.2166e-04, 1.4377e-04, 2.2412e-01,  ..., 1.9882e-02,
           3.4912e-02, 1.6785e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.6028e-06, 8.2493e-05, 6.6948e-04,  ..., 7.7393e-02,
           4.0552e-01, 4.0552e-01]],

         [[3.4475e-04, 2.2256e-04, 3.7060e-03,  ..., 4.4708e-03,
           2.9144e-02, 4.0234e-01]],

         [[5.9605e-08, 5.9605e-08, 8.9347e-05,  ..., 3.2842e-05,
           2.9583e-03, 9.2969e-01]],

         ...,

         [[0.0000e+00, 4.7684e-07, 6.0201e-06,  ..., 2.7120e-05,
           2.0027e-04, 9.8438e-01]],

         [[3.1185e-03, 2.0123e-03, 7.4806e-03,  ..., 1.5678e-03,
           4.8294e-03, 1.8115e-01]],

         [[1.0628e-02, 3.9101e-03, 5.6314e-04,  ..., 1.1426e-01,
           7.8491e-02, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3167e-04, 1.4906e-03, 1.2283e-02,  ..., 5.5939e-02,
           1.2408e-01, 1.0126e-01]],

         [[0.0000e+00, 0.0000e+00, 3.5763e-07,  ..., 8.5022e-02,
           8.4570e-01, 6.2225e-02]],

         [[3.8319e-03, 5.9357e-03, 1.6388e-02,  ..., 1.0414e-02,
           2.1042e-02, 1.4694e-02]],

         ...,

         [[4.2439e-04, 3.0632e-03, 4.4556e-03,  ..., 3.7988e-01,
           4.7180e-02, 5.7800e-02]],

         [[4.5204e-03, 1.3763e-02, 3.3661e-02,  ..., 1.2100e-02,
           1.1459e-02, 8.7509e-03]],

         [[2.0866e-03, 1.0353e-02, 9.4223e-03,  ..., 1.1481e-01,
           7.3547e-02, 5.3406e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7292e-02, 3.9597e-03, 1.6541e-02,  ..., 3.9093e-02,
           7.6538e-02, 5.0964e-02]],

         [[2.1992e-03, 6.2513e-04, 2.7905e-03,  ..., 8.0919e-04,
           3.9520e-03, 3.8891e-03]],

         [[3.4599e-03, 6.0606e-04, 6.7902e-04,  ..., 1.2367e-02,
           1.4572e-02, 1.3901e-02]],

         ...,

         [[6.3300e-05, 1.5526e-03, 1.8463e-02,  ..., 2.1301e-01,
           4.6167e-01, 2.1301e-01]],

         [[1.1063e-03, 2.5964e-04, 4.3321e-04,  ..., 6.6345e-02,
           3.8391e-02, 1.9608e-02]],

         [[2.7752e-03, 8.6641e-04, 8.8882e-03,  ..., 3.5156e-02,
           3.8696e-01, 1.5881e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3203e-01, 1.1325e-06, 1.6689e-06,  ..., 5.0354e-03,
           3.4389e-03, 6.1073e-03]],

         [[9.4727e-01, 4.8876e-06, 5.6028e-06,  ..., 1.9638e-02,
           1.0666e-02, 4.7226e-03]],

         [[4.3311e-01, 1.0729e-06, 1.1921e-06,  ..., 4.9324e-03,
           7.7324e-03, 5.5580e-03]],

         ...,

         [[6.4990e-01, 1.9169e-04, 1.3828e-03,  ..., 5.2929e-04,
           8.0538e-04, 2.7252e-02]],

         [[5.9766e-01, 1.0711e-04, 2.3782e-04,  ..., 1.9363e-02,
           7.3914e-02, 9.9258e-03]],

         [[5.6787e-01, 2.2542e-04, 4.7064e-04,  ..., 1.1162e-02,
           1.0144e-01, 2.1255e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2163e-01, 6.6161e-06, 6.1989e-06,  ..., 2.6155e-04,
           1.3208e-03, 1.9806e-02]],

         [[1.0132e-01, 3.6955e-06, 5.3048e-06,  ..., 2.3346e-03,
           8.8770e-01, 8.0795e-03]],

         [[9.2773e-01, 7.5519e-05, 4.9174e-05,  ..., 1.5173e-03,
           4.6272e-03, 2.9354e-03]],

         ...,

         [[3.0200e-01, 1.2779e-04, 1.1212e-04,  ..., 1.1284e-02,
           9.7656e-02, 1.4717e-02]],

         [[1.6919e-01, 5.2071e-04, 5.0449e-04,  ..., 4.9210e-03,
           4.3259e-03, 2.2354e-03]],

         [[7.8320e-01, 2.5153e-04, 5.1069e-04,  ..., 1.3084e-02,
           1.1084e-01, 7.4921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0569e-01, 2.3878e-04, 9.8765e-05,  ..., 7.6355e-02,
           7.7271e-02, 1.9211e-02]],

         [[6.8555e-01, 7.3135e-05, 4.0233e-05,  ..., 3.1525e-02,
           1.8518e-01, 1.4923e-02]],

         [[7.9785e-01, 3.6120e-05, 1.6689e-05,  ..., 1.2009e-02,
           1.5182e-02, 2.5215e-03]],

         ...,

         [[8.3643e-01, 1.1921e-05, 3.4571e-06,  ..., 2.0504e-05,
           5.6982e-05, 1.5857e-01]],

         [[9.5947e-01, 9.1672e-05, 1.3554e-04,  ..., 5.9605e-08,
           4.1723e-07, 1.9989e-02]],

         [[9.8242e-01, 3.8505e-05, 4.4703e-06,  ..., 1.7881e-07,
           1.1921e-07, 1.1383e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9568e-01, 2.9206e-06, 6.5565e-07,  ..., 6.1157e-02,
           6.7383e-01, 2.0203e-02]],

         [[8.7988e-01, 4.7088e-06, 1.4305e-06,  ..., 4.3106e-03,
           3.4237e-03, 5.0011e-03]],

         [[7.0557e-01, 5.1403e-04, 3.3450e-04,  ..., 8.5907e-03,
           9.3460e-03, 1.7960e-02]],

         ...,

         [[9.0527e-01, 5.1737e-05, 1.4365e-05,  ..., 2.4673e-02,
           3.2234e-03, 1.7380e-02]],

         [[5.3809e-01, 2.7657e-05, 2.1696e-05,  ..., 1.0948e-02,
           3.6621e-01, 2.3941e-02]],

         [[4.9268e-01, 5.8830e-05, 6.8784e-05,  ..., 6.7902e-03,
           4.5117e-01, 8.9951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1045e-01, 8.1658e-06, 9.1791e-06,  ..., 4.7180e-02,
           4.5837e-02, 4.7684e-03]],

         [[9.0771e-01, 4.6074e-05, 2.4915e-05,  ..., 1.6144e-02,
           1.5327e-02, 6.5613e-03]],

         [[9.5898e-01, 1.6689e-05, 4.6492e-06,  ..., 3.7632e-03,
           2.7985e-02, 1.5936e-03]],

         ...,

         [[7.1240e-01, 2.1446e-04, 7.3850e-05,  ..., 1.5244e-02,
           3.5034e-02, 5.3101e-03]],

         [[8.8232e-01, 1.6189e-04, 2.8133e-05,  ..., 3.0308e-03,
           9.6664e-03, 1.1368e-03]],

         [[9.0137e-01, 1.5616e-05, 1.1861e-05,  ..., 7.4158e-03,
           4.1962e-02, 3.9139e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6069e-01, 9.6369e-04, 6.7282e-04,  ..., 7.4043e-03,
           4.5815e-03, 2.5005e-03]],

         [[9.3799e-01, 7.1406e-05, 8.4162e-05,  ..., 9.1505e-04,
           9.9754e-04, 9.4032e-04]],

         [[8.5352e-01, 4.5896e-05, 1.6510e-05,  ..., 1.0323e-02,
           9.7198e-03, 3.2692e-03]],

         ...,

         [[6.8066e-01, 1.8120e-05, 2.3961e-05,  ..., 6.1989e-03,
           1.6800e-02, 2.2632e-01]],

         [[6.7139e-01, 1.6689e-06, 7.0333e-06,  ..., 1.0056e-02,
           2.7393e-01, 1.8906e-02]],

         [[9.5312e-01, 1.2100e-05, 1.4901e-05,  ..., 3.8013e-03,
           1.0956e-02, 8.8501e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8127e-01, 3.9005e-04, 1.0624e-03,  ..., 1.2306e-02,
           4.2755e-02, 3.8719e-03]],

         [[7.2070e-01, 1.3113e-06, 3.5167e-06,  ..., 9.3231e-03,
           2.4719e-01, 1.0162e-02]],

         [[8.3154e-01, 1.2279e-05, 2.6941e-05,  ..., 1.0445e-02,
           1.5282e-02, 5.8708e-03]],

         ...,

         [[6.9580e-01, 2.4939e-04, 2.6298e-04,  ..., 4.7646e-03,
           2.6798e-03, 5.3120e-04]],

         [[7.7930e-01, 1.3828e-04, 7.6354e-05,  ..., 1.0803e-02,
           4.7058e-02, 1.6165e-03]],

         [[1.7426e-02, 3.5763e-07, 2.9802e-07,  ..., 4.4489e-04,
           9.7363e-01, 6.6147e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3301e-01, 5.4061e-05, 5.5313e-05,  ..., 4.0472e-05,
           3.9935e-04, 5.3024e-03]],

         [[8.5449e-01, 3.0339e-05, 3.6895e-05,  ..., 7.9575e-03,
           4.5166e-02, 1.0786e-03]],

         [[9.3213e-01, 2.0862e-06, 9.1195e-06,  ..., 3.5248e-03,
           1.9236e-03, 4.6883e-03]],

         ...,

         [[9.5801e-01, 9.9719e-05, 8.1062e-05,  ..., 2.5578e-03,
           3.9978e-03, 3.5501e-04]],

         [[3.2300e-01, 1.3781e-04, 1.0484e-04,  ..., 2.0714e-03,
           1.6693e-02, 5.8057e-01]],

         [[7.0947e-01, 2.6870e-04, 1.0985e-04,  ..., 4.9248e-03,
           2.8473e-02, 8.7433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6907e-01, 2.9802e-07, 2.3842e-06,  ..., 9.3746e-04,
           4.7569e-03, 8.1885e-01]],

         [[7.5781e-01, 2.2495e-04, 3.2377e-04,  ..., 4.5509e-03,
           7.4196e-04, 5.0354e-04]],

         [[6.9189e-01, 7.5340e-04, 1.3399e-03,  ..., 1.4023e-02,
           1.2016e-02, 1.7757e-03]],

         ...,

         [[4.0674e-01, 1.9670e-04, 2.9063e-04,  ..., 2.0187e-02,
           4.0512e-03, 2.5654e-03]],

         [[9.0674e-01, 4.5359e-05, 4.0650e-05,  ..., 4.4365e-03,
           1.6647e-02, 1.6022e-02]],

         [[1.2445e-01, 3.3557e-05, 3.3021e-05,  ..., 2.8057e-03,
           4.8706e-02, 7.8809e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1448e-01, 8.5235e-06, 7.0930e-06,  ..., 9.2545e-03,
           3.7720e-02, 6.9775e-01]],

         [[5.1855e-01, 5.5790e-05, 6.2943e-05,  ..., 8.2626e-03,
           2.7979e-01, 1.9699e-02]],

         [[3.8550e-01, 2.2256e-04, 4.5359e-05,  ..., 5.3223e-02,
           2.2583e-02, 1.0872e-02]],

         ...,

         [[9.7839e-02, 1.8477e-06, 2.7418e-06,  ..., 3.8087e-05,
           8.3923e-03, 8.7891e-01]],

         [[8.8135e-01, 2.9206e-05, 4.8757e-05,  ..., 2.2182e-03,
           6.0349e-03, 1.6541e-02]],

         [[8.1104e-01, 1.3769e-04, 3.7313e-05,  ..., 3.0365e-03,
           2.2644e-02, 3.0537e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 1.2159e-03, 9.8419e-04,  ..., 2.8553e-03,
           2.1957e-02, 6.2218e-03]],

         [[8.6914e-01, 5.3287e-05, 9.8825e-05,  ..., 6.2332e-03,
           3.3752e-02, 1.0895e-02]],

         [[5.4395e-01, 3.0375e-04, 1.3483e-04,  ..., 4.1695e-03,
           1.1810e-02, 6.4087e-03]],

         ...,

         [[3.2495e-01, 1.1027e-05, 2.0087e-05,  ..., 3.7689e-02,
           6.1621e-01, 1.8127e-02]],

         [[4.3042e-01, 3.3760e-04, 6.6614e-04,  ..., 1.7746e-02,
           1.9550e-03, 4.3511e-04]],

         [[8.9844e-01, 7.8082e-06, 1.1444e-05,  ..., 1.1841e-02,
           6.3293e-02, 2.3285e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4609e-01, 1.9729e-05, 1.1921e-05,  ..., 1.8539e-02,
           1.0117e-02, 3.0994e-03]],

         [[9.7107e-02, 1.2846e-03, 9.5940e-04,  ..., 5.2452e-03,
           1.3466e-02, 1.2741e-03]],

         [[4.3701e-01, 2.7061e-04, 2.2054e-04,  ..., 3.3966e-02,
           1.1786e-01, 6.1188e-03]],

         ...,

         [[3.3789e-01, 3.8087e-05, 1.8477e-05,  ..., 4.1840e-02,
           5.2490e-01, 3.3508e-02]],

         [[8.7939e-01, 9.9540e-06, 3.2783e-06,  ..., 1.4343e-02,
           7.8003e-02, 1.9150e-02]],

         [[6.7529e-01, 3.0935e-05, 2.3186e-05,  ..., 2.1896e-02,
           3.9001e-02, 3.5210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2642e-02, 1.1921e-07, 2.3842e-07,  ..., 1.2314e-04,
           9.6817e-03, 9.0234e-01]],

         [[1.7065e-01, 6.6161e-06, 1.1206e-05,  ..., 8.2214e-02,
           3.7720e-01, 3.7781e-02]],

         [[1.8262e-01, 3.1662e-04, 5.8079e-04,  ..., 2.6718e-02,
           4.6539e-02, 4.0321e-03]],

         ...,

         [[5.1270e-01, 3.0017e-04, 1.8001e-04,  ..., 4.2175e-02,
           8.8348e-03, 1.5421e-03]],

         [[5.8008e-01, 1.4246e-05, 5.3644e-06,  ..., 7.3547e-02,
           4.6417e-02, 1.4000e-02]],

         [[9.3933e-02, 5.6684e-05, 1.5676e-05,  ..., 1.5812e-03,
           6.4964e-03, 1.6632e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6445e-01, 3.2485e-05, 3.0696e-05,  ..., 1.4771e-02,
           2.8473e-02, 6.4575e-02]],

         [[2.5903e-01, 1.1766e-04, 5.0843e-05,  ..., 1.5283e-01,
           5.6427e-02, 4.9622e-02]],

         [[5.7129e-01, 2.1863e-04, 1.2982e-04,  ..., 4.9133e-03,
           7.6965e-02, 3.8849e-02]],

         ...,

         [[7.8223e-01, 1.8418e-05, 1.5199e-05,  ..., 2.3987e-02,
           3.6865e-02, 8.1940e-03]],

         [[5.8203e-01, 2.4855e-05, 4.4703e-05,  ..., 1.4519e-02,
           2.0837e-01, 8.0185e-03]],

         [[6.8896e-01, 3.4153e-05, 1.2755e-05,  ..., 6.8726e-02,
           5.3024e-03, 2.4738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8662e-01, 1.3447e-03, 1.5888e-03,  ..., 1.0138e-03,
           3.7651e-03, 2.6512e-04]],

         [[2.3267e-01, 1.5700e-04, 1.2994e-04,  ..., 7.5912e-03,
           2.3636e-02, 1.1337e-02]],

         [[2.4829e-01, 5.0664e-06, 1.7166e-05,  ..., 9.8755e-02,
           3.8745e-01, 2.8625e-02]],

         ...,

         [[7.8564e-01, 5.2691e-05, 2.8431e-05,  ..., 1.1778e-03,
           2.7725e-02, 2.5955e-02]],

         [[9.1895e-01, 4.0352e-05, 5.6565e-05,  ..., 2.1458e-06,
           4.5943e-04, 1.0193e-02]],

         [[5.9229e-01, 1.7333e-04, 3.7026e-04,  ..., 6.9351e-03,
           6.5155e-03, 3.3283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5552e-02, 2.4772e-04, 1.2660e-04,  ..., 4.4342e-02,
           3.2715e-02, 2.5616e-03]],

         [[5.3558e-02, 5.4598e-04, 2.0361e-04,  ..., 2.1652e-02,
           3.0289e-02, 3.6736e-03]],

         [[4.5923e-01, 2.7955e-05, 1.7047e-05,  ..., 3.8815e-03,
           2.4207e-01, 1.6638e-01]],

         ...,

         [[2.9932e-01, 4.5896e-06, 5.9009e-06,  ..., 4.8096e-02,
           1.8036e-02, 2.1381e-03]],

         [[5.9912e-01, 6.4373e-06, 2.9206e-06,  ..., 4.3640e-03,
           2.2217e-01, 9.1187e-02]],

         [[1.2622e-01, 5.9605e-06, 2.9802e-06,  ..., 9.8877e-03,
           5.1904e-01, 3.0054e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 1.7524e-05, 4.1723e-06,  ..., 4.8309e-02,
           3.1189e-02, 2.3926e-02]],

         [[1.5002e-01, 5.1916e-05, 8.9705e-05,  ..., 8.4106e-02,
           3.8306e-01, 5.6915e-02]],

         [[2.5977e-01, 7.7486e-06, 2.2054e-06,  ..., 7.1812e-04,
           1.0492e-01, 6.2305e-01]],

         ...,

         [[4.4098e-02, 8.2791e-05, 1.1683e-05,  ..., 1.1683e-05,
           2.9774e-03, 1.0887e-02]],

         [[4.0112e-01, 8.1599e-05, 3.7611e-05,  ..., 3.7323e-02,
           9.9258e-03, 3.4447e-03]],

         [[1.7944e-01, 7.3314e-05, 4.8757e-05,  ..., 8.3435e-02,
           4.2686e-03, 6.0921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5337e-01, 1.7881e-05, 1.4961e-05,  ..., 1.7990e-02,
           3.6621e-02, 3.7811e-02]],

         [[9.3262e-01, 1.2636e-05, 2.5034e-06,  ..., 4.9472e-06,
           9.0361e-05, 1.3512e-02]],

         [[6.8994e-01, 9.4235e-05, 1.1802e-05,  ..., 2.4300e-03,
           4.2992e-03, 2.0996e-02]],

         ...,

         [[2.4829e-01, 7.6008e-04, 1.0147e-03,  ..., 1.1986e-02,
           1.8707e-02, 5.8403e-03]],

         [[7.9395e-01, 2.5570e-05, 2.3663e-05,  ..., 1.1688e-02,
           1.4099e-02, 1.5732e-02]],

         [[2.1973e-01, 3.7575e-04, 5.0354e-04,  ..., 5.0568e-02,
           4.6051e-02, 1.0277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4644e-01, 1.0961e-04, 1.4293e-04,  ..., 8.0994e-02,
           1.0880e-02, 1.1398e-02]],

         [[2.8198e-01, 6.1333e-05, 1.2827e-04,  ..., 1.2985e-02,
           4.6387e-02, 4.0955e-02]],

         [[2.0984e-01, 5.7220e-05, 7.6413e-05,  ..., 1.5774e-03,
           7.1220e-03, 6.3362e-03]],

         ...,

         [[1.0107e-01, 6.7043e-04, 6.7043e-04,  ..., 1.0002e-02,
           5.4932e-02, 2.0859e-02]],

         [[6.1829e-02, 7.2420e-05, 2.4259e-05,  ..., 1.8775e-04,
           5.1270e-02, 8.0627e-02]],

         [[5.1123e-01, 1.5736e-04, 1.2946e-04,  ..., 2.1423e-02,
           9.9640e-03, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9395e-01, 4.3094e-05, 2.5034e-05,  ..., 1.3855e-02,
           9.8450e-02, 6.9275e-02]],

         [[2.7612e-01, 9.3758e-05, 1.3220e-04,  ..., 5.1086e-02,
           3.6224e-02, 1.6846e-02]],

         [[3.8403e-01, 1.9121e-04, 2.8706e-04,  ..., 2.1835e-02,
           1.4026e-01, 1.9165e-01]],

         ...,

         [[4.7095e-01, 9.5069e-05, 9.0182e-05,  ..., 1.2070e-02,
           4.0192e-02, 6.7871e-02]],

         [[6.4307e-01, 3.8981e-05, 1.4949e-04,  ..., 6.5947e-04,
           2.9617e-02, 1.0419e-01]],

         [[4.6826e-01, 5.1022e-05, 1.0842e-04,  ..., 7.3967e-03,
           1.4824e-02, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9958e-01, 1.3340e-04, 5.3644e-05,  ..., 1.8784e-02,
           5.7617e-02, 4.9286e-02]],

         [[4.2480e-01, 2.4772e-04, 2.0099e-04,  ..., 1.1772e-02,
           1.1084e-01, 4.5502e-02]],

         [[4.2261e-01, 3.5930e-04, 2.6655e-04,  ..., 2.8320e-02,
           2.4887e-02, 1.3428e-02]],

         ...,

         [[5.9668e-01, 4.8280e-05, 3.6240e-04,  ..., 3.8422e-02,
           3.1934e-01, 3.3661e-02]],

         [[2.6367e-01, 9.1195e-06, 6.5744e-05,  ..., 8.1665e-02,
           4.1162e-01, 5.6580e-02]],

         [[2.0813e-01, 7.0930e-06, 9.0361e-05,  ..., 1.1925e-02,
           7.3779e-01, 3.5065e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9007e-02, 5.8413e-06, 9.0599e-06,  ..., 9.8145e-02,
           2.2461e-01, 1.1035e-01]],

         [[1.7432e-01, 2.8729e-05, 2.3007e-05,  ..., 3.1185e-03,
           3.4332e-02, 6.6797e-01]],

         [[5.6641e-02, 2.5928e-05, 1.2183e-04,  ..., 1.0078e-02,
           2.9266e-02, 6.1111e-03]],

         ...,

         [[1.5076e-01, 1.7929e-04, 7.3147e-04,  ..., 4.1870e-02,
           3.8574e-02, 1.5343e-02]],

         [[1.7029e-01, 1.7881e-06, 4.1127e-06,  ..., 8.9569e-03,
           7.3096e-01, 5.8167e-02]],

         [[4.7192e-01, 1.0729e-06, 3.3379e-06,  ..., 1.0529e-01,
           3.3203e-01, 4.3732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5381e-01, 6.8605e-05, 2.6512e-04,  ..., 2.3438e-02,
           3.6163e-02, 6.9313e-03]],

         [[9.5947e-01, 3.5763e-05, 1.3006e-04,  ..., 1.0805e-03,
           3.6192e-04, 3.9434e-04]],

         [[8.4863e-01, 2.3520e-04, 1.8895e-04,  ..., 1.3405e-02,
           4.3869e-03, 2.1038e-03]],

         ...,

         [[9.1797e-01, 1.0192e-05, 1.5676e-05,  ..., 9.1400e-03,
           2.1591e-02, 5.0468e-03]],

         [[8.9551e-01, 3.1590e-06, 4.6492e-06,  ..., 6.1131e-04,
           2.6760e-03, 4.0817e-03]],

         [[2.7295e-01, 5.9605e-06, 1.6034e-05,  ..., 3.2520e-03,
           4.0558e-02, 6.2451e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1957e-01, 2.6226e-06, 1.1921e-05,  ..., 6.0120e-02,
           3.6816e-01, 1.5112e-01]],

         [[8.0420e-01, 2.5094e-05, 1.9073e-05,  ..., 1.2794e-02,
           2.0935e-02, 2.2552e-02]],

         [[1.8079e-01, 2.8944e-04, 1.3626e-04,  ..., 5.6427e-02,
           3.5034e-02, 1.1307e-02]],

         ...,

         [[6.5869e-01, 1.2040e-05, 4.5061e-05,  ..., 1.7548e-02,
           6.4209e-02, 3.1036e-02]],

         [[3.8037e-01, 1.7583e-05, 8.7917e-05,  ..., 5.8350e-02,
           1.8143e-02, 3.5126e-02]],

         [[6.9238e-01, 8.1658e-06, 5.1856e-05,  ..., 1.3397e-02,
           3.0670e-02, 5.4230e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9185e-01, 2.8253e-05, 1.5855e-05,  ..., 1.0124e-02,
           1.3440e-01, 1.9861e-01]],

         [[5.4150e-01, 8.8811e-06, 2.1458e-06,  ..., 2.4819e-04,
           2.6535e-02, 1.9763e-01]],

         [[2.1570e-01, 6.3598e-05, 5.2333e-05,  ..., 3.5767e-02,
           6.2286e-02, 1.6931e-01]],

         ...,

         [[8.8196e-02, 1.2517e-06, 2.6345e-05,  ..., 2.1286e-02,
           1.2250e-01, 4.5776e-02]],

         [[9.6619e-02, 4.1127e-06, 3.7014e-05,  ..., 2.8122e-02,
           4.6356e-02, 1.4587e-02]],

         [[9.5312e-01, 6.0201e-06, 1.5676e-05,  ..., 4.6272e-03,
           8.7814e-03, 9.1124e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3594e-01, 1.0133e-06, 8.8811e-06,  ..., 7.3318e-03,
           1.5587e-02, 8.2458e-02]],

         [[4.4019e-01, 1.6689e-05, 1.6630e-04,  ..., 1.0492e-01,
           1.7920e-01, 1.0455e-01]],

         [[1.5759e-01, 2.1458e-06, 6.5923e-05,  ..., 4.0924e-02,
           7.5732e-01, 3.4332e-02]],

         ...,

         [[4.2651e-01, 5.3465e-05, 4.0352e-05,  ..., 6.0486e-02,
           4.4670e-03, 2.1408e-02]],

         [[3.5913e-01, 1.0550e-05, 7.5698e-06,  ..., 1.6037e-02,
           3.2959e-01, 7.9529e-02]],

         [[1.1731e-01, 6.8247e-05, 2.4629e-04,  ..., 1.8066e-02,
           3.3081e-02, 3.5010e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4160e-01, 1.1683e-05, 1.8537e-05,  ..., 5.5733e-03,
           1.1505e-01, 1.2830e-01]],

         [[6.8604e-01, 4.8220e-05, 4.8816e-05,  ..., 3.8776e-03,
           3.3508e-02, 9.0332e-02]],

         [[3.0469e-01, 2.6405e-05, 5.4240e-05,  ..., 3.7460e-03,
           4.5654e-02, 1.9360e-01]],

         ...,

         [[6.7432e-01, 4.3869e-05, 1.2398e-04,  ..., 5.6534e-03,
           8.5754e-02, 3.1052e-02]],

         [[3.8892e-01, 1.6749e-04, 3.9554e-04,  ..., 9.5825e-03,
           1.4389e-02, 4.3384e-01]],

         [[2.9028e-01, 5.0128e-05, 7.7605e-05,  ..., 1.7700e-02,
           4.8608e-01, 1.6028e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2285e-01, 1.1206e-05, 1.1319e-04,  ..., 8.9951e-03,
           8.1863e-03, 3.0487e-02]],

         [[1.2756e-01, 1.3709e-06, 2.6584e-05,  ..., 3.5820e-03,
           3.1567e-01, 1.5564e-01]],

         [[2.7710e-01, 2.3544e-04, 9.5463e-04,  ..., 8.5815e-02,
           6.1798e-02, 8.3191e-02]],

         ...,

         [[7.9639e-01, 5.3525e-05, 3.4785e-04,  ..., 1.8112e-02,
           5.3711e-03, 1.0239e-02]],

         [[8.8184e-01, 2.3842e-07, 2.3127e-04,  ..., 2.9087e-03,
           1.3733e-03, 1.3977e-02]],

         [[5.6152e-01, 2.8670e-05, 1.1957e-04,  ..., 1.3580e-02,
           3.3932e-03, 9.9335e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6093e-05, 7.9870e-06, 1.5479e-01,  ..., 8.5678e-03,
           5.2490e-02, 3.0542e-01]],

         [[6.5565e-06, 2.3246e-06, 1.3466e-02,  ..., 1.2189e-01,
           3.3131e-03, 9.1782e-03]],

         [[1.4305e-04, 1.0079e-04, 5.8319e-02,  ..., 1.3527e-02,
           1.2131e-02, 1.3452e-01]],

         ...,

         [[3.7789e-05, 1.0014e-05, 1.3318e-01,  ..., 9.3689e-02,
           2.4683e-01, 1.7371e-01]],

         [[1.1557e-04, 1.7822e-05, 2.7783e-01,  ..., 6.6109e-03,
           7.4615e-03, 1.6556e-02]],

         [[3.2187e-06, 1.7881e-07, 3.2812e-01,  ..., 5.7697e-04,
           2.6932e-02, 7.6111e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3896e-05, 1.0252e-05, 2.6953e-01,  ..., 1.2436e-03,
           1.4977e-02, 1.3660e-01]],

         [[3.7491e-05, 3.8147e-06, 1.6736e-01,  ..., 2.1992e-03,
           4.0070e-02, 2.9370e-01]],

         [[1.7762e-05, 6.5565e-07, 2.9785e-01,  ..., 1.3649e-04,
           1.1168e-03, 7.2937e-02]],

         ...,

         [[2.9862e-05, 2.5630e-06, 6.4209e-02,  ..., 2.3956e-03,
           2.3308e-03, 5.1208e-02]],

         [[2.3246e-06, 7.1526e-07, 2.5708e-01,  ..., 4.6992e-04,
           5.8174e-03, 1.3550e-01]],

         [[2.7895e-05, 1.1802e-05, 3.0078e-01,  ..., 2.3544e-04,
           8.3828e-04, 4.6844e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9009e-06, 8.3447e-07, 1.5015e-01,  ..., 5.2035e-05,
           2.3026e-02, 1.2744e-01]],

         [[7.7486e-06, 5.9605e-08, 1.6296e-01,  ..., 5.7106e-03,
           6.4325e-04, 9.1431e-02]],

         [[8.7619e-06, 2.0862e-06, 1.5698e-01,  ..., 5.5742e-04,
           1.9760e-03, 9.9792e-02]],

         ...,

         [[9.3222e-05, 5.2452e-06, 1.8616e-01,  ..., 7.7744e-03,
           9.5825e-02, 2.0605e-01]],

         [[3.8147e-06, 1.2517e-06, 7.6416e-02,  ..., 2.5654e-03,
           5.7564e-03, 1.1298e-01]],

         [[2.2709e-05, 1.3113e-06, 1.6699e-01,  ..., 2.9526e-03,
           3.1250e-02, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2338e-05, 1.2517e-06, 1.8723e-02,  ..., 4.5166e-03,
           7.1387e-01, 1.4734e-01]],

         [[7.3314e-06, 7.1049e-05, 8.3740e-02,  ..., 9.4116e-02,
           2.2186e-02, 1.9775e-02]],

         [[3.6716e-05, 4.0388e-04, 3.2886e-01,  ..., 1.1263e-03,
           4.3297e-04, 5.3596e-04]],

         ...,

         [[1.2903e-03, 4.3988e-05, 2.5684e-01,  ..., 1.8525e-04,
           1.0004e-03, 2.7599e-03]],

         [[9.3222e-04, 5.3704e-05, 6.5613e-02,  ..., 6.3820e-03,
           2.3633e-01, 4.5728e-01]],

         [[2.1017e-04, 1.7405e-05, 6.6223e-02,  ..., 2.3022e-03,
           2.0569e-01, 3.6523e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4211e-04, 7.9453e-05, 8.9294e-02,  ..., 1.6129e-02,
           1.9989e-02, 4.6844e-02]],

         [[6.9666e-04, 3.6168e-04, 2.0789e-01,  ..., 2.7054e-02,
           3.2135e-02, 6.8787e-02]],

         [[3.5334e-04, 1.5354e-04, 1.4697e-01,  ..., 9.0179e-03,
           3.5797e-02, 6.4331e-02]],

         ...,

         [[1.0347e-04, 2.1040e-05, 3.7109e-02,  ..., 5.5618e-03,
           5.8545e-01, 2.4402e-01]],

         [[3.1710e-04, 5.6076e-04, 3.3960e-01,  ..., 3.5930e-04,
           4.8065e-03, 3.9001e-02]],

         [[4.2248e-04, 1.3614e-04, 2.8540e-01,  ..., 2.9793e-03,
           1.8539e-02, 3.1525e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[7.8082e-06, 3.2961e-05, 2.0390e-03,  ..., 6.1462e-02,
           5.1465e-01, 1.1487e-01]],

         [[1.1110e-04, 3.3855e-05, 7.7105e-04,  ..., 2.8931e-02,
           1.5637e-01, 5.1270e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 8.3447e-07,
           2.4700e-03, 9.9658e-01]],

         ...,

         [[0.0000e+00, 5.9605e-08, 1.7881e-07,  ..., 2.0015e-04,
           6.6261e-03, 9.8340e-01]],

         [[9.1705e-03, 7.6065e-03, 3.8242e-03,  ..., 2.0676e-02,
           2.6550e-02, 4.3335e-03]],

         [[8.9722e-03, 3.6259e-03, 1.1768e-03,  ..., 1.6769e-02,
           1.6769e-02, 5.0049e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5048e-04, 3.5048e-04, 3.0270e-03,  ..., 2.6154e-02,
           9.4177e-02, 5.4199e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 1.4624e-01,
           7.9053e-01, 5.3802e-02]],

         [[5.2929e-04, 1.8759e-03, 1.2238e-02,  ..., 3.8879e-02,
           3.6530e-02, 7.2632e-02]],

         ...,

         [[5.6725e-03, 7.8735e-03, 1.6281e-02,  ..., 8.6517e-03,
           1.8448e-02, 8.9417e-02]],

         [[2.1458e-03, 6.3667e-03, 2.9907e-02,  ..., 1.0788e-02,
           2.1790e-02, 1.1002e-02]],

         [[2.9049e-03, 7.7744e-03, 1.4076e-02,  ..., 5.7434e-02,
           6.7139e-02, 1.1786e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6519e-02, 5.5084e-03, 2.9541e-02,  ..., 2.6901e-02,
           1.9836e-02, 2.6901e-02]],

         [[8.1158e-04, 3.4904e-04, 1.8291e-03,  ..., 3.8719e-03,
           4.7836e-03, 3.5820e-03]],

         [[3.1414e-03, 5.6791e-04, 6.5327e-04,  ..., 1.9547e-02,
           1.8723e-02, 1.6586e-02]],

         ...,

         [[1.0204e-04, 1.9670e-04, 1.0170e-02,  ..., 1.3708e-01,
           3.7866e-01, 1.5784e-01]],

         [[4.5776e-03, 6.2180e-04, 2.1877e-03,  ..., 6.7810e-02,
           1.3458e-02, 2.0996e-02]],

         [[2.7542e-02, 5.5962e-03, 1.8494e-02,  ..., 8.8928e-02,
           9.2468e-02, 1.3245e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5791e-01, 5.9605e-08, 4.1723e-07,  ..., 3.7766e-03,
           5.1918e-03, 2.8625e-02]],

         [[8.8818e-01, 1.3113e-06, 5.3048e-06,  ..., 1.3306e-02,
           5.1788e-02, 1.4694e-02]],

         [[9.4629e-01, 1.6689e-06, 9.7752e-06,  ..., 6.9351e-03,
           1.2875e-03, 1.2100e-02]],

         ...,

         [[7.2754e-01, 4.9114e-04, 1.0986e-03,  ..., 2.1946e-04,
           1.1164e-04, 1.9623e-02]],

         [[6.5039e-01, 1.3351e-04, 6.0225e-04,  ..., 5.0354e-02,
           5.3062e-03, 1.5472e-02]],

         [[2.5439e-01, 1.4508e-04, 5.4884e-04,  ..., 4.2261e-01,
           2.9663e-02, 1.8280e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9453e-01, 5.6505e-05, 2.8193e-05,  ..., 7.7391e-04,
           3.0577e-05, 4.7699e-02]],

         [[8.0957e-01, 1.1921e-06, 9.6560e-06,  ..., 6.1554e-02,
           9.1064e-02, 3.5767e-02]],

         [[8.5986e-01, 5.1796e-05, 1.7583e-05,  ..., 8.2016e-04,
           1.3580e-02, 4.0512e-03]],

         ...,

         [[2.8613e-01, 2.2459e-04, 3.4809e-04,  ..., 1.2549e-01,
           4.2053e-02, 2.3682e-02]],

         [[2.0691e-01, 1.9855e-03, 9.0981e-04,  ..., 4.9286e-03,
           2.1172e-03, 3.9139e-03]],

         [[4.8145e-01, 8.8334e-05, 4.9400e-04,  ..., 1.1328e-01,
           3.9764e-02, 1.2512e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0337e-01, 1.9586e-04, 7.2598e-05,  ..., 7.1472e-02,
           3.1708e-02, 3.5553e-02]],

         [[6.9189e-01, 4.9591e-05, 3.2246e-05,  ..., 9.6924e-02,
           5.3162e-02, 8.8562e-02]],

         [[9.1602e-01, 1.4603e-04, 5.3287e-05,  ..., 3.1719e-03,
           7.0496e-03, 1.4214e-02]],

         ...,

         [[9.3311e-01, 6.7949e-06, 2.1458e-06,  ..., 1.1402e-04,
           1.7583e-05, 6.4392e-02]],

         [[9.3359e-01, 3.4571e-05, 3.0160e-05,  ..., 4.2319e-06,
           9.5367e-07, 2.2903e-02]],

         [[9.5508e-01, 9.0003e-06, 1.7881e-06,  ..., 3.6955e-06,
           7.7486e-07, 2.2537e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 4.7088e-06, 3.1590e-06,  ..., 1.9165e-01,
           2.5000e-01, 8.0383e-02]],

         [[8.2275e-01, 2.3246e-06, 1.3709e-06,  ..., 9.7322e-04,
           5.1880e-04, 2.4338e-02]],

         [[8.1006e-01, 1.0234e-04, 4.3988e-05,  ..., 1.1444e-03,
           1.4925e-03, 1.5160e-02]],

         ...,

         [[6.9531e-01, 2.0337e-04, 3.2306e-05,  ..., 1.3466e-02,
           5.5275e-03, 1.9763e-01]],

         [[5.1953e-01, 1.7798e-04, 5.4300e-05,  ..., 6.7322e-02,
           3.6865e-02, 1.7358e-01]],

         [[8.5010e-01, 3.7313e-05, 5.9187e-05,  ..., 6.1707e-02,
           1.0445e-02, 4.5685e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4424e-01, 1.4126e-05, 1.0431e-05,  ..., 3.0533e-02,
           3.9940e-03, 1.5060e-02]],

         [[9.0820e-01, 6.4969e-05, 1.5557e-05,  ..., 1.0170e-02,
           2.8191e-03, 2.0264e-02]],

         [[9.7656e-01, 1.3113e-06, 5.9605e-07,  ..., 9.3002e-03,
           4.8599e-03, 4.4708e-03]],

         ...,

         [[6.4111e-01, 2.6798e-04, 5.2154e-05,  ..., 3.4454e-02,
           9.9411e-03, 2.5864e-02]],

         [[8.6523e-01, 1.3316e-04, 2.0087e-05,  ..., 5.4131e-03,
           2.1696e-04, 6.1226e-03]],

         [[9.5557e-01, 1.5378e-05, 4.2915e-06,  ..., 6.6299e-03,
           4.8904e-03, 9.1858e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2446e-01, 2.4605e-03, 1.2093e-03,  ..., 6.9427e-03,
           1.5459e-03, 1.2283e-02]],

         [[9.2920e-01, 2.0862e-05, 6.0916e-05,  ..., 1.9321e-03,
           2.5196e-03, 4.9133e-03]],

         [[8.8574e-01, 3.4392e-05, 1.1086e-05,  ..., 1.7500e-03,
           1.5440e-03, 1.4061e-02]],

         ...,

         [[8.0566e-01, 3.4571e-05, 3.1233e-05,  ..., 4.3564e-03,
           2.5650e-02, 1.0498e-01]],

         [[7.9688e-01, 4.5300e-06, 9.7156e-06,  ..., 9.7290e-02,
           6.8787e-02, 6.7444e-03]],

         [[8.5791e-01, 5.6028e-06, 1.8001e-05,  ..., 7.1945e-03,
           9.9411e-03, 8.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1792e-01, 1.9073e-04, 7.5865e-04,  ..., 1.0767e-01,
           1.0033e-02, 1.0674e-02]],

         [[8.0322e-01, 1.4901e-06, 7.2718e-06,  ..., 6.2988e-02,
           7.2449e-02, 1.7456e-02]],

         [[8.5840e-01, 3.2187e-06, 8.2254e-06,  ..., 1.4458e-02,
           2.3148e-02, 2.8915e-02]],

         ...,

         [[7.5879e-01, 1.2314e-04, 1.2410e-04,  ..., 2.4090e-03,
           5.2118e-04, 1.1120e-03]],

         [[6.6504e-01, 4.8220e-05, 1.9908e-05,  ..., 4.9896e-02,
           4.2572e-03, 9.2712e-02]],

         [[1.3245e-01, 2.6822e-06, 1.6093e-06,  ..., 7.7441e-01,
           6.2561e-02, 1.4130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6216e-01, 1.2505e-04, 7.3195e-05,  ..., 1.7345e-05,
           4.9472e-06, 3.7117e-03]],

         [[9.0186e-01, 1.4126e-05, 3.1114e-05,  ..., 2.4246e-02,
           1.2722e-03, 8.3113e-04]],

         [[9.2871e-01, 1.0729e-05, 5.9485e-05,  ..., 1.7529e-03,
           1.3411e-04, 1.7868e-02]],

         ...,

         [[8.2520e-01, 2.2554e-04, 1.5080e-04,  ..., 7.9193e-03,
           2.5215e-03, 3.8357e-03]],

         [[7.8271e-01, 6.7472e-05, 5.9962e-05,  ..., 6.8903e-04,
           9.4482e-02, 9.5596e-03]],

         [[6.2402e-01, 7.9632e-04, 2.6989e-04,  ..., 4.9988e-02,
           1.4465e-02, 2.0081e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1863e-01, 1.7881e-07, 1.4901e-06,  ..., 6.6662e-04,
           6.7444e-03, 7.6611e-01]],

         [[5.3320e-01, 6.9475e-04, 1.2875e-03,  ..., 2.3918e-03,
           8.2731e-04, 5.9853e-03]],

         [[6.9043e-01, 5.8889e-04, 7.3671e-04,  ..., 9.5825e-03,
           2.7008e-03, 1.2001e-02]],

         ...,

         [[6.6309e-01, 1.5581e-04, 1.9395e-04,  ..., 2.8286e-03,
           2.4090e-03, 7.8201e-04]],

         [[4.0771e-01, 8.4043e-06, 9.6560e-06,  ..., 1.2537e-01,
           4.0405e-01, 1.8997e-02]],

         [[4.1382e-01, 1.2195e-04, 1.8167e-04,  ..., 2.3254e-02,
           6.6101e-02, 4.1968e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9810e-01, 1.0139e-04, 3.4511e-05,  ..., 5.2299e-03,
           8.7891e-03, 5.1514e-01]],

         [[4.4434e-01, 5.4240e-06, 7.5698e-06,  ..., 1.9177e-01,
           2.8394e-01, 9.8038e-03]],

         [[3.0908e-01, 1.4782e-04, 8.1956e-05,  ..., 4.4830e-02,
           2.2003e-02, 2.0416e-02]],

         ...,

         [[7.8064e-02, 3.9935e-06, 1.9670e-06,  ..., 2.0278e-04,
           2.1782e-03, 9.1113e-01]],

         [[8.7842e-01, 3.0577e-05, 2.4259e-05,  ..., 2.2907e-03,
           2.0409e-03, 8.2245e-03]],

         [[7.8809e-01, 5.4955e-05, 2.6882e-05,  ..., 3.3905e-02,
           3.0457e-02, 3.2288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 1.2293e-03, 8.4782e-04,  ..., 9.2239e-03,
           2.4967e-03, 2.6379e-03]],

         [[8.9355e-01, 5.4479e-05, 1.0794e-04,  ..., 1.7014e-02,
           1.3954e-02, 1.2283e-02]],

         [[5.0928e-01, 1.4162e-04, 6.5565e-05,  ..., 6.4182e-04,
           8.8453e-04, 8.3923e-03]],

         ...,

         [[5.5273e-01, 5.9605e-08, 3.1590e-06,  ..., 8.3618e-02,
           3.2300e-01, 3.5767e-02]],

         [[4.7168e-01, 6.3944e-04, 8.3733e-04,  ..., 4.9133e-03,
           1.5821e-03, 5.2404e-04]],

         [[8.9453e-01, 2.3842e-07, 4.2319e-06,  ..., 1.7044e-02,
           6.2408e-02, 2.2339e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7095e-01, 2.0361e-04, 6.7174e-05,  ..., 4.9927e-02,
           2.3819e-02, 8.2153e-02]],

         [[1.2103e-01, 2.5578e-03, 1.1511e-03,  ..., 5.2910e-03,
           3.2806e-04, 1.7481e-03]],

         [[7.6758e-01, 9.1136e-05, 7.5221e-05,  ..., 3.2013e-02,
           7.0152e-03, 4.7150e-02]],

         ...,

         [[7.1777e-01, 3.1352e-05, 1.5199e-05,  ..., 8.3984e-02,
           9.3323e-02, 6.6528e-02]],

         [[8.1543e-01, 4.5300e-06, 3.6955e-06,  ..., 2.5604e-02,
           5.7983e-02, 9.1309e-02]],

         [[8.1445e-01, 4.7386e-05, 2.5988e-05,  ..., 5.2155e-02,
           1.9943e-02, 1.6754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1500e-02, 0.0000e+00, 0.0000e+00,  ..., 7.4291e-04,
           1.8177e-03, 9.7314e-01]],

         [[1.2024e-01, 1.5318e-05, 1.8895e-05,  ..., 3.1934e-01,
           1.8616e-01, 8.5571e-02]],

         [[2.0825e-01, 2.6417e-04, 4.9591e-04,  ..., 4.8309e-02,
           4.8828e-03, 1.4824e-02]],

         ...,

         [[4.4263e-01, 2.7680e-04, 2.7466e-04,  ..., 2.1484e-02,
           1.4084e-02, 1.7128e-03]],

         [[5.9180e-01, 1.2636e-05, 6.3777e-06,  ..., 2.5803e-02,
           5.4436e-03, 1.0773e-02]],

         [[3.2324e-01, 4.9472e-05, 2.1279e-05,  ..., 2.5368e-03,
           5.9090e-03, 1.0475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0303e-01, 9.8348e-06, 6.3777e-06,  ..., 3.1342e-02,
           2.4078e-02, 2.0996e-01]],

         [[3.7036e-01, 2.7323e-04, 1.4567e-04,  ..., 1.4160e-02,
           3.6530e-02, 5.4840e-02]],

         [[1.1218e-01, 1.4842e-05, 6.9737e-06,  ..., 5.8411e-02,
           6.6064e-01, 4.5502e-02]],

         ...,

         [[8.3252e-01, 1.5557e-05, 3.1829e-05,  ..., 3.4241e-02,
           1.4275e-02, 2.2903e-02]],

         [[6.9775e-01, 1.4603e-05, 2.3067e-05,  ..., 1.1615e-01,
           2.9251e-02, 5.1941e-02]],

         [[8.0176e-01, 1.9848e-05, 8.7023e-06,  ..., 8.7051e-03,
           7.2975e-03, 4.4006e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5491e-01, 2.2068e-03, 2.0370e-03,  ..., 3.7518e-03,
           5.4646e-04, 2.0618e-03]],

         [[3.3130e-01, 1.2374e-04, 6.5625e-05,  ..., 4.3549e-02,
           5.0629e-02, 1.6663e-02]],

         [[4.8315e-01, 2.8610e-06, 5.1856e-06,  ..., 1.6565e-01,
           1.5137e-01, 9.9304e-02]],

         ...,

         [[3.1519e-01, 4.5300e-05, 1.5378e-05,  ..., 1.1492e-03,
           3.3226e-03, 3.6716e-03]],

         [[1.3374e-02, 2.9802e-06, 4.1723e-06,  ..., 2.9802e-07,
           1.7881e-06, 1.3342e-03]],

         [[4.4092e-01, 1.5318e-04, 3.6693e-04,  ..., 3.4428e-03,
           4.4823e-03, 4.3917e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5328e-02, 1.2386e-04, 7.9989e-05,  ..., 5.6641e-02,
           8.3542e-03, 5.9814e-02]],

         [[1.7822e-01, 3.5620e-04, 8.9705e-05,  ..., 6.6589e-02,
           1.4061e-02, 1.3527e-02]],

         [[4.4971e-01, 3.4988e-05, 1.4007e-05,  ..., 3.6072e-02,
           9.7290e-02, 3.3179e-01]],

         ...,

         [[5.3809e-01, 1.6689e-06, 4.9472e-06,  ..., 4.5593e-02,
           3.8391e-02, 1.4572e-02]],

         [[7.8711e-01, 2.3842e-06, 3.2187e-06,  ..., 8.4900e-02,
           4.8004e-02, 2.1133e-02]],

         [[1.5784e-01, 4.5300e-06, 2.2054e-06,  ..., 4.1504e-02,
           9.1370e-02, 6.2891e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 3.9995e-05, 9.9540e-06,  ..., 5.5771e-03,
           9.7885e-03, 4.3884e-02]],

         [[8.9539e-02, 2.7418e-06, 6.0797e-06,  ..., 4.4775e-01,
           3.1738e-01, 1.8768e-02]],

         [[1.1060e-01, 6.7353e-06, 1.7881e-06,  ..., 3.6678e-03,
           7.5264e-03, 8.6963e-01]],

         ...,

         [[4.3701e-02, 1.0294e-04, 5.3644e-06,  ..., 1.8034e-03,
           1.0767e-03, 4.8767e-02]],

         [[2.1667e-01, 1.9872e-04, 1.1867e-04,  ..., 3.2959e-02,
           2.4887e-02, 3.0243e-02]],

         [[2.0276e-01, 1.0997e-04, 4.7982e-05,  ..., 7.5195e-02,
           3.7506e-02, 4.2847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0771e-01, 1.8418e-05, 2.7299e-05,  ..., 3.0716e-02,
           3.0716e-02, 1.4429e-01]],

         [[1.6089e-01, 2.6822e-06, 5.3644e-07,  ..., 2.2542e-04,
           2.4962e-04, 4.9362e-03]],

         [[5.1514e-01, 8.5235e-05, 1.0133e-05,  ..., 2.9697e-03,
           6.0797e-04, 1.2306e-02]],

         ...,

         [[1.8738e-01, 9.6846e-04, 1.0967e-03,  ..., 3.5767e-02,
           1.0406e-02, 1.2169e-02]],

         [[6.4258e-01, 1.4901e-05, 1.3351e-05,  ..., 7.3242e-02,
           6.8787e-02, 6.8787e-02]],

         [[2.1289e-01, 4.3750e-04, 5.1165e-04,  ..., 2.4078e-02,
           1.0513e-02, 3.2379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8457e-01, 9.2804e-05, 2.0111e-04,  ..., 1.1780e-02,
           1.7548e-02, 5.5328e-02]],

         [[8.9355e-02, 1.7881e-05, 3.0458e-05,  ..., 1.3702e-02,
           1.7044e-02, 1.3069e-02]],

         [[3.4106e-01, 2.5153e-05, 2.4915e-05,  ..., 2.9736e-03,
           4.4289e-03, 5.0293e-02]],

         ...,

         [[3.1036e-02, 8.1897e-05, 2.6584e-05,  ..., 1.1713e-01,
           9.4116e-02, 1.1780e-02]],

         [[1.3159e-01, 1.5974e-05, 3.8743e-06,  ..., 1.3657e-02,
           1.9257e-02, 9.4788e-02]],

         [[8.2764e-01, 1.0139e-04, 7.9572e-05,  ..., 1.1810e-02,
           8.7738e-03, 2.4231e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 5.4240e-06, 2.3842e-06,  ..., 7.1533e-02,
           1.1255e-01, 1.4001e-01]],

         [[1.4050e-01, 2.9862e-05, 3.3438e-05,  ..., 1.4380e-01,
           2.0923e-01, 3.3112e-02]],

         [[2.4463e-01, 1.1712e-04, 1.7238e-04,  ..., 5.5450e-02,
           7.0068e-02, 5.1367e-01]],

         ...,

         [[4.5215e-01, 1.8656e-04, 1.2875e-04,  ..., 2.8458e-02,
           3.8605e-02, 6.5125e-02]],

         [[2.7344e-01, 1.6987e-05, 1.3137e-04,  ..., 7.9250e-04,
           4.0550e-03, 1.4417e-01]],

         [[3.8794e-01, 5.1200e-05, 5.5552e-05,  ..., 1.1444e-02,
           8.2642e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0996e-01, 4.4525e-05, 2.2531e-05,  ..., 6.8726e-02,
           1.3245e-01, 1.4893e-01]],

         [[4.6948e-01, 4.1068e-05, 9.2983e-06,  ..., 1.0236e-01,
           8.8196e-02, 1.0724e-01]],

         [[4.9487e-01, 3.5906e-04, 2.8801e-04,  ..., 1.4320e-02,
           8.2245e-03, 8.0948e-03]],

         ...,

         [[6.3428e-01, 1.1921e-07, 3.7551e-05,  ..., 4.7028e-02,
           2.6855e-01, 4.0863e-02]],

         [[3.2593e-01, 1.1921e-06, 2.4676e-05,  ..., 1.9922e-01,
           2.5781e-01, 1.8539e-02]],

         [[5.2783e-01, 1.9670e-06, 8.8990e-05,  ..., 7.2510e-02,
           2.8687e-01, 8.4778e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1765e-02, 1.2517e-06, 8.9407e-07,  ..., 2.8296e-01,
           5.8057e-01, 2.6108e-02]],

         [[2.6709e-01, 7.3314e-06, 7.5102e-06,  ..., 2.4395e-03,
           1.7883e-02, 6.6113e-01]],

         [[1.2317e-01, 6.1393e-05, 1.4925e-04,  ..., 1.2317e-01,
           4.3762e-02, 2.5223e-02]],

         ...,

         [[9.9365e-02, 1.2982e-04, 4.3750e-04,  ..., 4.5135e-02,
           3.9520e-02, 3.1525e-02]],

         [[8.8501e-02, 1.7881e-06, 2.6226e-06,  ..., 4.2896e-01,
           3.9062e-01, 7.9346e-02]],

         [[3.0078e-01, 5.9605e-07, 7.6294e-06,  ..., 1.5662e-01,
           4.4775e-01, 7.5134e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2988e-01, 5.5194e-05, 1.8680e-04,  ..., 3.0624e-02,
           9.3765e-03, 6.5857e-02]],

         [[9.6436e-01, 2.8372e-05, 4.7922e-05,  ..., 1.8513e-04,
           5.1916e-05, 1.9140e-03]],

         [[9.5215e-01, 7.6175e-05, 3.4869e-05,  ..., 2.0027e-03,
           4.9067e-04, 2.3975e-03]],

         ...,

         [[9.1943e-01, 4.3511e-06, 4.8876e-06,  ..., 7.9575e-03,
           7.3586e-03, 5.5206e-02]],

         [[9.6387e-01, 1.5497e-06, 1.0133e-06,  ..., 3.4976e-04,
           1.4725e-03, 3.1166e-03]],

         [[2.3108e-01, 3.2187e-06, 6.4373e-06,  ..., 6.2561e-03,
           1.3664e-02, 7.3438e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6309e-01, 6.5565e-07, 2.6822e-06,  ..., 1.0992e-01,
           1.8628e-01, 4.6460e-01]],

         [[9.4727e-01, 7.2122e-06, 5.0068e-06,  ..., 5.3520e-03,
           7.7896e-03, 1.8188e-02]],

         [[3.6182e-01, 1.7297e-04, 2.0552e-04,  ..., 8.1055e-02,
           3.6377e-02, 4.7272e-02]],

         ...,

         [[7.0312e-01, 9.4175e-06, 2.1935e-05,  ..., 7.8918e-02,
           6.9092e-02, 3.1891e-02]],

         [[8.5645e-01, 9.9540e-06, 3.9816e-05,  ..., 8.2703e-03,
           6.0005e-03, 1.9226e-02]],

         [[7.9492e-01, 6.2585e-06, 2.1994e-05,  ..., 2.7847e-02,
           8.4473e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 3.5167e-06, 1.4305e-06,  ..., 2.2864e-01,
           4.4067e-01, 7.4829e-02]],

         [[6.6064e-01, 3.1590e-06, 7.1526e-07,  ..., 7.5722e-03,
           1.2805e-01, 1.4062e-01]],

         [[5.4541e-01, 5.8472e-05, 3.4571e-05,  ..., 1.4534e-02,
           6.0242e-02, 4.7302e-02]],

         ...,

         [[8.1726e-02, 5.3644e-07, 1.3769e-05,  ..., 6.8787e-02,
           8.5632e-02, 1.8701e-01]],

         [[1.5881e-01, 2.3246e-06, 2.3842e-05,  ..., 9.7839e-02,
           6.7200e-02, 1.9257e-02]],

         [[9.3311e-01, 2.5094e-05, 3.2246e-05,  ..., 1.6830e-02,
           4.6005e-03, 3.2368e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6543e-01, 1.4305e-06, 2.3365e-05,  ..., 9.3384e-03,
           4.2419e-02, 3.4692e-01]],

         [[6.5234e-01, 1.9073e-06, 6.0141e-05,  ..., 4.9133e-02,
           1.1029e-01, 5.5237e-02]],

         [[5.3662e-01, 8.9407e-07, 3.5346e-05,  ..., 4.3060e-02,
           1.3733e-01, 2.6172e-01]],

         ...,

         [[4.9097e-01, 1.4591e-04, 8.0585e-05,  ..., 1.4145e-02,
           4.1260e-02, 2.4246e-02]],

         [[5.4535e-02, 4.1723e-07, 4.1723e-07,  ..., 4.3555e-01,
           4.2212e-01, 5.1208e-02]],

         [[6.7993e-02, 5.1498e-05, 1.7321e-04,  ..., 6.6528e-03,
           2.6413e-02, 6.7578e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7354e-01, 1.4067e-05, 1.2755e-05,  ..., 1.1566e-01,
           1.9067e-01, 1.8054e-01]],

         [[7.9004e-01, 1.4305e-05, 1.3471e-05,  ..., 1.5343e-02,
           5.7678e-02, 4.4556e-02]],

         [[1.6101e-01, 1.2577e-05, 3.0935e-05,  ..., 5.1056e-02,
           2.5732e-01, 1.3879e-01]],

         ...,

         [[6.5283e-01, 3.7789e-05, 1.0103e-04,  ..., 1.3684e-01,
           4.1748e-02, 4.1077e-02]],

         [[2.5293e-01, 1.2350e-04, 2.8706e-04,  ..., 1.1721e-03,
           1.2794e-02, 6.6650e-01]],

         [[1.4343e-01, 5.4240e-06, 9.8348e-06,  ..., 4.6289e-01,
           3.1323e-01, 5.3589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1655e-01, 1.4722e-05, 9.1672e-05,  ..., 4.1351e-02,
           5.9570e-01, 1.0852e-01]],

         [[5.5450e-02, 5.9605e-07, 4.8280e-06,  ..., 1.1829e-01,
           1.7480e-01, 2.6562e-01]],

         [[4.7217e-01, 1.9479e-04, 6.1321e-04,  ..., 1.4374e-02,
           4.9103e-02, 8.8013e-02]],

         ...,

         [[9.1357e-01, 4.3929e-05, 1.1456e-04,  ..., 4.1656e-03,
           5.2452e-03, 1.6510e-02]],

         [[9.6436e-01, 1.7881e-07, 9.4175e-05,  ..., 1.8823e-04,
           6.0558e-04, 8.7814e-03]],

         [[6.6992e-01, 2.2531e-05, 3.8624e-05,  ..., 3.0556e-03,
           1.1131e-02, 6.2805e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0788e-05, 6.6161e-06, 1.8677e-01,  ..., 3.1219e-02,
           2.6343e-01, 1.9302e-02]],

         [[2.9802e-06, 8.9407e-07, 7.5912e-03,  ..., 2.8362e-03,
           1.1940e-02, 2.7485e-03]],

         [[9.1195e-05, 8.9169e-05, 6.0394e-02,  ..., 1.0872e-02,
           6.4819e-02, 4.4458e-01]],

         ...,

         [[1.0967e-05, 1.4901e-06, 7.5623e-02,  ..., 7.1594e-02,
           4.8169e-01, 1.6785e-01]],

         [[1.1927e-04, 3.3796e-05, 3.3594e-01,  ..., 6.4507e-03,
           8.0566e-03, 9.4910e-03]],

         [[7.2718e-06, 5.3644e-07, 3.0103e-01,  ..., 1.3123e-02,
           5.5664e-02, 5.5664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3273e-05, 2.5511e-05, 2.6465e-01,  ..., 5.4283e-03,
           5.4169e-02, 1.7868e-02]],

         [[3.8207e-05, 9.8944e-06, 2.3279e-01,  ..., 3.1250e-02,
           1.4575e-01, 8.6136e-03]],

         [[4.9829e-05, 7.0930e-06, 2.8076e-01,  ..., 1.6365e-03,
           4.3365e-02, 1.0345e-02]],

         ...,

         [[1.2875e-05, 4.0531e-06, 2.4841e-01,  ..., 1.7004e-03,
           2.7863e-02, 1.3794e-02]],

         [[2.3842e-06, 1.4901e-06, 2.8149e-01,  ..., 6.5956e-03,
           7.0618e-02, 4.3602e-03]],

         [[1.2457e-05, 1.7047e-05, 3.1714e-01,  ..., 6.5422e-04,
           2.8366e-02, 5.3520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-06, 2.7418e-06, 1.6943e-01,  ..., 1.9318e-02,
           1.1292e-01, 9.0027e-03]],

         [[1.5736e-05, 4.1723e-07, 2.4414e-01,  ..., 3.1877e-04,
           4.2755e-02, 1.4656e-02]],

         [[2.2113e-05, 7.6890e-06, 2.5122e-01,  ..., 1.9407e-03,
           6.5552e-02, 1.0864e-02]],

         ...,

         [[3.9101e-05, 1.8477e-06, 1.6418e-01,  ..., 1.1548e-01,
           2.1411e-01, 1.6162e-01]],

         [[8.5235e-06, 4.2319e-06, 2.6514e-01,  ..., 4.3716e-03,
           4.6997e-02, 2.8503e-02]],

         [[4.3511e-05, 3.7551e-06, 2.2449e-01,  ..., 2.2308e-02,
           7.5195e-02, 2.4033e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-06, 2.3842e-07, 8.5220e-03,  ..., 2.4329e-01,
           5.5273e-01, 1.5466e-01]],

         [[3.1769e-05, 1.2100e-04, 7.6965e-02,  ..., 1.6830e-02,
           1.5747e-02, 1.4771e-02]],

         [[2.0921e-05, 2.3735e-04, 3.1396e-01,  ..., 5.4264e-04,
           4.6468e-04, 6.1369e-04]],

         ...,

         [[2.8586e-04, 9.0003e-06, 2.8540e-01,  ..., 5.2261e-04,
           1.7233e-03, 3.8967e-03]],

         [[2.9755e-04, 9.8348e-06, 2.9099e-02,  ..., 2.4124e-02,
           1.9958e-01, 6.6211e-01]],

         [[6.6936e-05, 7.3314e-06, 3.3722e-02,  ..., 8.6792e-02,
           3.9673e-01, 2.8906e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1505e-04, 8.4519e-05, 1.5125e-01,  ..., 1.3847e-02,
           2.6794e-02, 5.7861e-02]],

         [[1.3423e-04, 1.7035e-04, 1.6260e-01,  ..., 2.4368e-02,
           4.3762e-02, 1.9617e-01]],

         [[5.5742e-04, 1.2922e-04, 1.5149e-01,  ..., 2.4918e-02,
           7.5256e-02, 4.9927e-02]],

         ...,

         [[8.7142e-05, 2.0206e-05, 5.1147e-02,  ..., 6.5674e-02,
           4.0234e-01, 2.9443e-01]],

         [[2.0337e-04, 5.2357e-04, 3.7061e-01,  ..., 7.6332e-03,
           4.0619e-02, 4.7852e-02]],

         [[1.4532e-04, 6.8903e-05, 1.8188e-01,  ..., 6.7444e-02,
           1.5088e-01, 1.6956e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2696e-05, 3.3975e-06, 4.7588e-04,  ..., 1.6943e-01,
           4.4653e-01, 8.2581e-02]],

         [[9.0551e-04, 2.2900e-04, 2.4624e-03,  ..., 1.0468e-01,
           4.9438e-02, 5.3174e-01]],

         [[2.9802e-07, 0.0000e+00, 8.3447e-07,  ..., 9.0456e-04,
           4.0550e-03, 9.9219e-01]],

         ...,

         [[1.7881e-07, 5.9605e-08, 2.9802e-07,  ..., 7.4446e-05,
           9.0694e-04, 9.9463e-01]],

         [[6.6071e-03, 8.7509e-03, 4.2648e-03,  ..., 7.1045e-02,
           1.5358e-02, 2.3788e-02]],

         [[4.8370e-03, 1.6203e-03, 5.7745e-04,  ..., 1.1971e-02,
           2.0370e-02, 1.6876e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2171e-04, 3.9911e-04, 3.7270e-03,  ..., 4.6143e-02,
           1.6101e-01, 4.9585e-01]],

         [[2.3842e-07, 0.0000e+00, 0.0000e+00,  ..., 1.0016e-01,
           8.1299e-01, 7.8003e-02]],

         [[1.7393e-04, 7.6723e-04, 1.0933e-02,  ..., 5.2155e-02,
           1.1749e-01, 6.9092e-02]],

         ...,

         [[2.1915e-03, 2.3499e-03, 5.5084e-03,  ..., 1.5479e-01,
           1.1414e-01, 2.7588e-01]],

         [[1.4372e-03, 7.4158e-03, 3.0975e-02,  ..., 4.1992e-02,
           1.0658e-02, 1.8417e-02]],

         [[7.6294e-04, 3.7861e-03, 4.3564e-03,  ..., 6.6040e-02,
           1.0229e-01, 5.1453e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 5.6686e-03, 3.1616e-02,  ..., 3.2867e-02,
           4.4922e-02, 3.9978e-02]],

         [[1.7757e-03, 8.0681e-04, 3.9101e-03,  ..., 5.2185e-03,
           9.5978e-03, 1.0544e-02]],

         [[7.3853e-03, 1.2197e-03, 1.4772e-03,  ..., 2.2568e-02,
           2.3376e-02, 2.1454e-02]],

         ...,

         [[2.8074e-05, 3.0994e-06, 2.0123e-04,  ..., 3.0334e-02,
           4.0576e-01, 4.1553e-01]],

         [[2.8954e-03, 9.6941e-04, 2.5940e-03,  ..., 4.2877e-02,
           2.8778e-02, 4.3549e-02]],

         [[2.3670e-03, 5.1355e-04, 2.4986e-03,  ..., 1.4185e-01,
           1.1310e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6904e-01, 1.7822e-05, 1.7405e-05,  ..., 2.2202e-03,
           4.9438e-03, 8.0872e-03]],

         [[9.2041e-01, 6.3181e-06, 5.2452e-06,  ..., 1.4114e-02,
           2.1988e-02, 1.6815e-02]],

         [[9.3750e-01, 1.7881e-07, 5.5432e-06,  ..., 5.4779e-03,
           1.2634e-02, 4.6387e-03]],

         ...,

         [[5.0586e-01, 2.2960e-04, 1.3962e-03,  ..., 2.5606e-04,
           3.5596e-04, 1.8494e-02]],

         [[5.3076e-01, 8.3804e-05, 1.2522e-03,  ..., 5.9319e-03,
           4.4250e-02, 1.0780e-02]],

         [[3.8818e-01, 2.3401e-04, 4.6921e-04,  ..., 4.1412e-02,
           8.2474e-03, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3350e-01, 4.7684e-06, 1.2100e-05,  ..., 8.2970e-04,
           8.1837e-05, 1.0269e-02]],

         [[9.2188e-01, 0.0000e+00, 0.0000e+00,  ..., 2.9392e-03,
           2.6077e-02, 4.5288e-02]],

         [[9.1504e-01, 1.5974e-05, 1.6093e-06,  ..., 2.9068e-03,
           1.8677e-02, 1.3496e-02]],

         ...,

         [[3.5742e-01, 1.8167e-04, 2.0897e-04,  ..., 8.0688e-02,
           7.2876e-02, 6.6566e-03]],

         [[2.4536e-01, 5.3358e-04, 2.6298e-04,  ..., 8.7128e-03,
           4.6577e-03, 8.7051e-03]],

         [[6.9629e-01, 1.6451e-05, 1.2136e-04,  ..., 1.1665e-02,
           1.6992e-01, 5.8838e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7383e-01, 1.2094e-04, 5.8234e-05,  ..., 2.8320e-02,
           5.9631e-02, 4.9744e-02]],

         [[6.5967e-01, 1.0252e-05, 9.0003e-06,  ..., 1.6739e-02,
           7.0435e-02, 1.5884e-02]],

         [[6.6797e-01, 2.8789e-05, 4.7028e-05,  ..., 4.8584e-02,
           1.5076e-01, 1.0474e-01]],

         ...,

         [[8.3691e-01, 2.3842e-07, 4.1723e-07,  ..., 2.2769e-05,
           6.0797e-06, 1.5442e-01]],

         [[8.8135e-01, 2.9802e-07, 3.8147e-06,  ..., 5.9605e-07,
           2.3842e-07, 3.5614e-02]],

         [[9.8438e-01, 5.9605e-08, 2.9802e-07,  ..., 2.9802e-07,
           5.9605e-08, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7563e-01, 4.0531e-06, 2.5034e-06,  ..., 1.7371e-01,
           3.5742e-01, 2.5681e-02]],

         [[9.4385e-01, 1.4305e-06, 1.9670e-06,  ..., 1.9097e-04,
           2.9802e-04, 3.9406e-03]],

         [[7.7783e-01, 1.1218e-04, 2.0087e-05,  ..., 1.4324e-03,
           7.7133e-03, 1.5778e-02]],

         ...,

         [[5.8789e-01, 3.1638e-04, 9.3162e-05,  ..., 2.0660e-02,
           1.9128e-01, 5.0293e-02]],

         [[3.9868e-01, 4.1187e-05, 1.1086e-05,  ..., 8.6288e-03,
           3.2422e-01, 6.8176e-02]],

         [[8.3545e-01, 3.5644e-05, 6.7353e-05,  ..., 8.6136e-03,
           2.3438e-02, 2.0325e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6338e-01, 2.9802e-06, 3.0994e-06,  ..., 7.2670e-04,
           4.0855e-03, 3.6335e-03]],

         [[9.6289e-01, 7.7486e-06, 1.6093e-06,  ..., 9.6798e-04,
           2.1515e-03, 1.5274e-02]],

         [[9.5508e-01, 2.0266e-06, 4.7684e-07,  ..., 7.0534e-03,
           9.6436e-03, 1.1055e-02]],

         ...,

         [[7.9395e-01, 1.8239e-04, 5.2631e-05,  ..., 5.4016e-03,
           2.6215e-02, 1.5022e-02]],

         [[8.9355e-01, 1.1694e-04, 1.2636e-05,  ..., 2.2721e-04,
           8.0566e-03, 1.3733e-02]],

         [[9.4824e-01, 6.1989e-06, 2.2054e-06,  ..., 1.4915e-02,
           1.2932e-02, 3.9520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7031e-01, 1.6203e-03, 1.2817e-03,  ..., 2.2316e-03,
           1.2329e-02, 8.2626e-03]],

         [[9.3945e-01, 1.1683e-05, 2.0504e-05,  ..., 5.4893e-03,
           2.3060e-03, 1.5974e-03]],

         [[8.8232e-01, 1.5855e-05, 8.1658e-06,  ..., 2.5578e-03,
           2.1851e-02, 7.7896e-03]],

         ...,

         [[7.2607e-01, 6.2048e-05, 2.3842e-05,  ..., 8.9569e-03,
           1.3336e-02, 1.9312e-01]],

         [[7.4463e-01, 2.7418e-06, 5.9605e-06,  ..., 5.9998e-02,
           2.8057e-03, 3.0289e-03]],

         [[7.0264e-01, 5.9009e-06, 6.9141e-06,  ..., 1.1787e-02,
           2.1692e-01, 2.4048e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9226e-01, 1.6105e-04, 7.7724e-04,  ..., 1.9302e-02,
           1.8097e-02, 4.7340e-03]],

         [[7.7441e-01, 4.7684e-07, 1.9670e-06,  ..., 2.3453e-02,
           7.9712e-02, 2.3453e-02]],

         [[7.6953e-01, 1.0014e-05, 2.1517e-05,  ..., 5.1178e-02,
           9.2468e-02, 2.5955e-02]],

         ...,

         [[8.7158e-01, 6.2764e-05, 4.9651e-05,  ..., 8.6975e-04,
           3.1872e-03, 1.5364e-03]],

         [[6.0059e-01, 2.2948e-05, 1.3292e-05,  ..., 4.2305e-03,
           1.7078e-01, 9.6924e-02]],

         [[4.7046e-01, 5.4836e-06, 8.1658e-06,  ..., 1.0999e-01,
           6.2195e-02, 2.2705e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6504e-01, 1.0842e-04, 2.7180e-05,  ..., 2.1875e-05,
           2.1064e-04, 1.7715e-02]],

         [[9.5361e-01, 6.9737e-06, 4.8280e-06,  ..., 1.5421e-03,
           3.5305e-03, 5.2986e-03]],

         [[7.9297e-01, 3.3975e-06, 1.6212e-05,  ..., 1.2112e-03,
           1.4275e-02, 1.5617e-02]],

         ...,

         [[8.3887e-01, 1.4365e-04, 8.8871e-05,  ..., 1.1215e-03,
           9.0485e-03, 5.7297e-03]],

         [[8.5938e-01, 7.1764e-05, 7.5221e-05,  ..., 2.1271e-02,
           1.3206e-02, 5.5962e-03]],

         [[7.5439e-01, 1.1101e-03, 4.8494e-04,  ..., 9.8114e-03,
           1.2016e-02, 9.5596e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 1.2517e-06, 9.5367e-07,  ..., 1.8990e-04,
           2.3499e-03, 4.6191e-01]],

         [[3.8940e-01, 4.6515e-04, 9.2936e-04,  ..., 1.4391e-03,
           1.3123e-02, 5.6549e-02]],

         [[7.3730e-01, 3.4523e-04, 5.2595e-04,  ..., 2.3861e-03,
           7.7095e-03, 2.7542e-02]],

         ...,

         [[4.4849e-01, 1.6379e-04, 1.9991e-04,  ..., 1.5228e-02,
           5.0545e-03, 8.6164e-04]],

         [[3.8403e-01, 4.1246e-05, 2.8491e-05,  ..., 7.5806e-02,
           2.3071e-01, 3.9185e-02]],

         [[2.9907e-01, 4.2260e-05, 3.8922e-05,  ..., 1.0231e-02,
           1.5030e-02, 6.2598e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9473e-01, 3.0279e-05, 4.2558e-05,  ..., 9.1171e-03,
           1.0826e-02, 3.4912e-01]],

         [[4.5776e-01, 4.0412e-05, 7.2300e-05,  ..., 1.0992e-01,
           1.2341e-01, 5.1697e-02]],

         [[9.8145e-02, 9.3281e-05, 1.1927e-04,  ..., 3.6407e-02,
           3.9917e-02, 6.9275e-02]],

         ...,

         [[2.0508e-01, 3.5763e-06, 8.6427e-06,  ..., 5.1498e-04,
           2.6646e-03, 7.8125e-01]],

         [[7.3438e-01, 2.5928e-05, 5.9545e-05,  ..., 2.6535e-02,
           6.9466e-03, 2.0950e-02]],

         [[6.9336e-01, 7.9751e-05, 1.6868e-05,  ..., 4.4250e-02,
           7.5134e-02, 3.6499e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0146e-01, 3.9554e-04, 1.2722e-03,  ..., 1.7929e-02,
           2.2583e-02, 1.4868e-01]],

         [[6.9043e-01, 3.8683e-05, 1.4722e-04,  ..., 4.3884e-02,
           1.4709e-01, 2.6459e-02]],

         [[6.6553e-01, 1.2445e-04, 1.2994e-04,  ..., 1.3176e-02,
           3.3259e-05, 9.3231e-03]],

         ...,

         [[3.3862e-01, 0.0000e+00, 0.0000e+00,  ..., 1.4519e-02,
           6.0986e-01, 2.9083e-02]],

         [[5.7373e-01, 4.9496e-04, 8.4209e-04,  ..., 1.7214e-03,
           1.6651e-03, 2.3854e-04]],

         [[7.3193e-01, 0.0000e+00, 5.9605e-08,  ..., 8.5068e-03,
           6.1951e-02, 1.9019e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2969e-01, 5.4717e-05, 2.4557e-05,  ..., 2.9968e-02,
           2.4261e-02, 5.5389e-02]],

         [[9.1125e-02, 1.2445e-03, 1.3647e-03,  ..., 1.6327e-03,
           2.8687e-03, 2.2488e-03]],

         [[1.9873e-01, 2.2411e-04, 8.1658e-05,  ..., 3.6743e-02,
           2.2607e-01, 2.2888e-02]],

         ...,

         [[7.5732e-01, 2.6941e-05, 2.0802e-05,  ..., 4.2145e-02,
           6.5796e-02, 1.2169e-02]],

         [[5.7178e-01, 1.7881e-06, 2.9206e-06,  ..., 1.0815e-01,
           1.8286e-01, 6.2927e-02]],

         [[7.0117e-01, 4.3213e-05, 2.7895e-05,  ..., 2.2797e-02,
           4.1321e-02, 7.0007e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6821e-01, 9.5367e-07, 2.3842e-07,  ..., 5.1022e-04,
           6.6261e-03, 8.0566e-01]],

         [[1.3025e-01, 1.4305e-06, 2.5630e-06,  ..., 1.6724e-01,
           2.3218e-01, 2.0676e-02]],

         [[1.1017e-01, 3.6740e-04, 5.3120e-04,  ..., 1.1406e-02,
           1.8265e-02, 2.6260e-02]],

         ...,

         [[5.9717e-01, 8.4817e-05, 8.1599e-05,  ..., 3.9864e-03,
           1.3281e-01, 1.9419e-04]],

         [[4.9561e-01, 5.3048e-06, 2.3246e-06,  ..., 1.0266e-01,
           3.0880e-03, 3.6133e-02]],

         [[6.9922e-01, 8.6963e-05, 1.6248e-04,  ..., 9.1505e-04,
           5.6458e-04, 3.9093e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5808e-01, 1.0729e-06, 2.1458e-06,  ..., 8.2169e-03,
           7.1680e-01, 4.8401e-02]],

         [[2.1057e-01, 5.4479e-05, 2.4915e-05,  ..., 1.5527e-01,
           8.9172e-02, 9.7168e-02]],

         [[7.9834e-01, 3.3975e-06, 1.3411e-05,  ..., 5.8861e-03,
           1.2006e-01, 1.1749e-02]],

         ...,

         [[7.1289e-01, 2.2650e-06, 1.0312e-05,  ..., 2.0233e-02,
           7.4829e-02, 1.5091e-02]],

         [[6.6650e-01, 1.0133e-06, 3.8147e-06,  ..., 2.1801e-03,
           2.5098e-01, 2.2629e-02]],

         [[6.2598e-01, 3.7074e-05, 3.3617e-05,  ..., 2.8038e-03,
           2.5940e-02, 4.4983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2825e-02, 4.2176e-04, 9.8228e-04,  ..., 1.0357e-03,
           1.0815e-03, 1.0612e-02]],

         [[1.8115e-01, 5.6207e-05, 1.1462e-04,  ..., 2.2141e-02,
           2.2316e-03, 1.6296e-01]],

         [[7.8674e-02, 5.9605e-07, 7.7486e-07,  ..., 1.0339e-01,
           5.5908e-01, 1.0626e-01]],

         ...,

         [[7.7051e-01, 1.2422e-04, 4.7147e-05,  ..., 5.6419e-03,
           2.6035e-03, 5.6366e-02]],

         [[7.9883e-01, 1.4663e-05, 9.0003e-06,  ..., 1.3053e-05,
           5.0664e-06, 9.0103e-03]],

         [[4.8267e-01, 1.9038e-04, 3.4475e-04,  ..., 7.0038e-03,
           1.6422e-03, 1.8559e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9713e-02, 3.6788e-04, 1.6975e-04,  ..., 6.8359e-03,
           2.0096e-02, 4.5891e-03]],

         [[2.5757e-01, 1.9121e-04, 6.3956e-05,  ..., 6.1760e-03,
           4.1275e-03, 4.3106e-03]],

         [[2.4048e-01, 4.2915e-06, 2.5630e-06,  ..., 2.0898e-01,
           1.8433e-01, 2.1558e-01]],

         ...,

         [[3.8379e-01, 2.3842e-07, 2.9802e-07,  ..., 1.4877e-02,
           3.4607e-02, 3.2177e-03]],

         [[4.4800e-01, 4.1723e-07, 3.5763e-07,  ..., 2.3926e-02,
           3.2764e-01, 6.5063e-02]],

         [[2.9907e-01, 1.4901e-06, 1.6689e-06,  ..., 6.5674e-02,
           1.0663e-01, 4.7046e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6826e-01, 5.3465e-05, 2.3723e-05,  ..., 5.3375e-02,
           1.2482e-02, 2.5208e-02]],

         [[2.2491e-02, 1.0133e-06, 2.3246e-06,  ..., 1.8530e-01,
           2.6978e-01, 6.0181e-02]],

         [[2.3413e-01, 7.3910e-06, 2.3246e-06,  ..., 4.0894e-03,
           5.8594e-03, 7.4365e-01]],

         ...,

         [[1.3757e-01, 3.7026e-04, 9.0361e-05,  ..., 3.7599e-04,
           1.7834e-04, 3.8433e-03]],

         [[4.7974e-01, 2.6679e-04, 1.5783e-04,  ..., 6.4392e-02,
           1.0269e-02, 2.1744e-02]],

         [[1.2408e-01, 7.3910e-05, 9.3460e-05,  ..., 2.4246e-02,
           3.1929e-03, 1.3603e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4790e-01, 6.6757e-06, 8.4639e-06,  ..., 2.2949e-02,
           4.9347e-02, 2.0142e-01]],

         [[6.9385e-01, 4.7088e-06, 7.1526e-07,  ..., 9.6464e-04,
           7.6294e-04, 1.3180e-03]],

         [[4.9731e-01, 1.2887e-04, 2.8968e-05,  ..., 2.9633e-02,
           2.1347e-02, 3.3760e-03]],

         ...,

         [[8.4595e-02, 2.6917e-04, 3.3498e-04,  ..., 7.7438e-03,
           4.6959e-03, 7.3891e-03]],

         [[6.8799e-01, 1.6093e-05, 1.4067e-05,  ..., 1.4961e-02,
           3.4790e-02, 9.7534e-02]],

         [[1.8054e-01, 4.2224e-04, 9.7370e-04,  ..., 7.9346e-03,
           2.0584e-02, 5.3802e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4424e-01, 4.4703e-05, 5.6744e-05,  ..., 3.2013e-02,
           6.4209e-02, 1.4575e-01]],

         [[2.1509e-01, 1.4007e-05, 4.8459e-05,  ..., 4.7241e-02,
           2.2675e-02, 7.5500e-02]],

         [[3.9990e-01, 2.4438e-05, 3.5822e-05,  ..., 7.6752e-03,
           1.6113e-02, 9.6436e-02]],

         ...,

         [[3.1067e-02, 5.0485e-05, 4.6730e-05,  ..., 7.1106e-02,
           5.2338e-03, 4.8126e-02]],

         [[3.0566e-01, 1.0252e-04, 7.0453e-05,  ..., 2.3575e-02,
           1.5221e-02, 1.5613e-01]],

         [[6.4160e-01, 7.1526e-05, 1.0091e-04,  ..., 1.9379e-02,
           1.4854e-02, 6.6589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9789e-05, 1.4544e-05,  ..., 4.0344e-02,
           5.1392e-02, 3.1470e-01]],

         [[6.6846e-01, 1.8811e-04, 3.9673e-04,  ..., 6.8665e-03,
           5.6038e-03, 1.5480e-02]],

         [[3.8477e-01, 1.3065e-04, 2.0790e-04,  ..., 6.7383e-02,
           6.1371e-02, 1.0437e-01]],

         ...,

         [[5.8008e-01, 1.3328e-04, 8.9109e-05,  ..., 3.9795e-02,
           5.9265e-02, 1.0730e-01]],

         [[4.1431e-01, 2.8133e-05, 1.2553e-04,  ..., 3.6469e-02,
           1.4038e-03, 1.8970e-01]],

         [[4.7144e-01, 2.0385e-04, 2.1279e-04,  ..., 2.3102e-02,
           4.3518e-02, 8.9966e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 8.0526e-05, 6.2525e-05,  ..., 5.5481e-02,
           1.2598e-01, 1.1121e-01]],

         [[1.7786e-01, 4.0054e-05, 2.0742e-05,  ..., 1.7932e-01,
           2.2668e-01, 1.5210e-01]],

         [[4.2480e-01, 2.0897e-04, 1.7095e-04,  ..., 3.7415e-02,
           5.4550e-03, 1.8234e-02]],

         ...,

         [[4.0698e-01, 0.0000e+00, 8.9407e-07,  ..., 2.9251e-02,
           4.8315e-01, 6.9092e-02]],

         [[1.9678e-01, 5.9605e-07, 3.9935e-06,  ..., 6.1920e-02,
           2.3880e-02, 1.8738e-02]],

         [[1.7822e-01, 0.0000e+00, 5.9605e-07,  ..., 6.3538e-02,
           7.0459e-01, 3.7354e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2619e-02, 1.4901e-06, 1.6093e-06,  ..., 5.6122e-02,
           1.7834e-03, 1.7517e-02]],

         [[1.8262e-01, 4.8280e-06, 2.0862e-06,  ..., 7.3624e-03,
           1.3863e-02, 7.3975e-01]],

         [[4.9835e-02, 3.8028e-05, 2.1851e-04,  ..., 3.0994e-04,
           2.6512e-04, 1.0757e-03]],

         ...,

         [[9.4910e-03, 3.1233e-05, 1.2046e-04,  ..., 1.7490e-03,
           2.9755e-03, 3.4523e-03]],

         [[1.8768e-02, 0.0000e+00, 0.0000e+00,  ..., 7.5439e-02,
           2.2125e-02, 1.9073e-02]],

         [[5.4077e-02, 0.0000e+00, 5.9605e-08,  ..., 1.7871e-01,
           5.7227e-01, 7.6233e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 1.2767e-04, 5.2977e-04,  ..., 7.0038e-03,
           1.9882e-02, 2.4536e-02]],

         [[9.6924e-01, 1.8096e-04, 1.3137e-04,  ..., 5.5969e-05,
           2.4021e-04, 6.0987e-04]],

         [[9.3652e-01, 6.1989e-06, 1.5438e-05,  ..., 3.1490e-03,
           3.4313e-03, 4.6883e-03]],

         ...,

         [[3.2324e-01, 4.1723e-07, 4.7684e-07,  ..., 1.3062e-01,
           1.7041e-01, 8.4351e-02]],

         [[8.4668e-01, 6.3181e-06, 9.1791e-06,  ..., 5.3883e-04,
           8.4782e-04, 4.3411e-03]],

         [[1.3220e-01, 4.6492e-06, 1.0371e-05,  ..., 5.4026e-04,
           1.4687e-03, 8.4863e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8958e-01, 2.3842e-07, 8.9407e-07,  ..., 3.6896e-02,
           1.4307e-01, 3.0298e-01]],

         [[4.2822e-01, 6.5565e-07, 7.1526e-07,  ..., 6.6162e-02,
           1.5747e-01, 1.2366e-01]],

         [[3.3252e-01, 3.1531e-05, 4.3750e-05,  ..., 2.6886e-02,
           7.7148e-02, 5.7129e-02]],

         ...,

         [[1.6113e-01, 6.5565e-07, 2.5034e-06,  ..., 1.0321e-01,
           5.5664e-02, 4.2694e-02]],

         [[4.2993e-01, 4.5896e-06, 2.3365e-05,  ..., 4.6051e-02,
           3.5309e-02, 3.1403e-02]],

         [[7.0410e-01, 3.2783e-06, 1.7762e-05,  ..., 4.0985e-02,
           5.1270e-03, 1.0361e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4053e-01, 1.9789e-05, 1.2755e-05,  ..., 1.0565e-01,
           8.6670e-03, 1.9128e-01]],

         [[5.1465e-01, 6.7949e-06, 5.1856e-06,  ..., 4.4983e-02,
           1.0513e-02, 1.7517e-01]],

         [[4.2896e-01, 5.4538e-05, 1.4532e-04,  ..., 3.0838e-02,
           2.0065e-02, 6.4758e-02]],

         ...,

         [[1.0645e-01, 3.4571e-06, 4.4823e-05,  ..., 1.4183e-02,
           1.8219e-02, 6.0539e-03]],

         [[1.8750e-01, 3.3379e-05, 2.1613e-04,  ..., 1.7538e-03,
           1.2150e-03, 6.9189e-04]],

         [[2.7161e-02, 2.7418e-06, 6.3181e-06,  ..., 9.0942e-03,
           6.1569e-03, 1.3031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0828e-01, 5.9605e-08, 3.5763e-07,  ..., 2.1484e-02,
           8.0615e-01, 4.1901e-02]],

         [[5.0232e-02, 1.7881e-07, 1.6093e-06,  ..., 3.1605e-03,
           2.3670e-03, 1.5198e-02]],

         [[3.6084e-01, 3.5763e-07, 2.6822e-06,  ..., 1.6846e-02,
           4.2847e-01, 1.7651e-01]],

         ...,

         [[6.5918e-01, 5.6446e-05, 2.6345e-05,  ..., 1.5625e-02,
           1.5869e-02, 4.3152e-02]],

         [[2.6392e-01, 4.4703e-06, 5.8413e-06,  ..., 1.5759e-01,
           7.2144e-02, 8.4351e-02]],

         [[1.0876e-01, 2.3067e-05, 7.6652e-05,  ..., 5.6854e-02,
           1.1945e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3887e-01, 3.8147e-06, 1.6749e-05,  ..., 4.9591e-02,
           1.0834e-01, 3.5522e-01]],

         [[7.8174e-01, 8.7023e-06, 3.0756e-05,  ..., 3.5034e-02,
           2.5131e-02, 4.5166e-02]],

         [[4.8779e-01, 6.3956e-05, 1.2898e-04,  ..., 1.0468e-01,
           1.7288e-02, 9.9854e-02]],

         ...,

         [[1.0498e-01, 1.1921e-05, 6.8545e-06,  ..., 3.4088e-02,
           1.0239e-02, 1.8539e-02]],

         [[4.9072e-02, 1.8954e-05, 1.1975e-04,  ..., 1.4503e-02,
           4.8309e-02, 8.1689e-01]],

         [[2.3120e-01, 7.4506e-06, 1.7464e-05,  ..., 1.2189e-01,
           1.5894e-01, 6.1279e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6875e-01, 5.7518e-05, 2.1303e-04,  ..., 9.3750e-02,
           2.1289e-01, 7.5317e-02]],

         [[2.9761e-01, 1.9610e-05, 5.5969e-05,  ..., 3.3783e-02,
           1.8909e-01, 1.7224e-01]],

         [[6.5479e-01, 1.5008e-04, 5.3596e-04,  ..., 1.6449e-02,
           1.9852e-02, 1.7624e-02]],

         ...,

         [[5.1123e-01, 4.8220e-05, 3.0303e-04,  ..., 3.0106e-02,
           1.7029e-02, 9.7733e-03]],

         [[8.8965e-01, 6.0797e-06, 4.3654e-04,  ..., 2.5768e-03,
           1.4023e-02, 8.4076e-03]],

         [[5.7080e-01, 5.4240e-06, 4.7266e-05,  ..., 1.8387e-02,
           1.8173e-02, 2.6077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9743e-05, 2.6286e-05, 1.7261e-01,  ..., 1.0974e-01,
           4.6444e-04, 5.3177e-03]],

         [[1.3292e-05, 7.0333e-06, 5.0598e-02,  ..., 4.1290e-02,
           5.7220e-03, 3.2864e-03]],

         [[9.1672e-05, 4.7445e-05, 6.9092e-02,  ..., 1.0699e-01,
           2.3364e-01, 1.1212e-01]],

         ...,

         [[8.9109e-05, 7.7486e-06, 6.7688e-02,  ..., 7.7271e-02,
           1.3892e-01, 1.6235e-01]],

         [[4.4560e-04, 1.3046e-03, 2.9517e-01,  ..., 1.0460e-02,
           4.3297e-03, 6.6490e-03]],

         [[1.6272e-04, 8.7440e-05, 3.9258e-01,  ..., 4.8294e-03,
           8.3923e-04, 1.4496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6921e-04, 4.7493e-04, 3.1104e-01,  ..., 7.4844e-03,
           2.5320e-04, 2.2926e-03]],

         [[2.2364e-04, 2.8944e-04, 2.7246e-01,  ..., 2.4567e-02,
           4.3273e-04, 4.0741e-03]],

         [[1.9777e-04, 9.9242e-05, 3.2617e-01,  ..., 6.2599e-03,
           5.6219e-04, 2.7514e-04]],

         ...,

         [[1.5509e-04, 3.4404e-04, 3.0371e-01,  ..., 3.8548e-03,
           1.5574e-03, 5.9366e-04]],

         [[9.9540e-06, 5.8830e-05, 3.0688e-01,  ..., 7.8659e-03,
           8.7500e-05, 4.0007e-04]],

         [[1.9073e-05, 1.1712e-04, 1.9128e-01,  ..., 5.2605e-03,
           1.4186e-04, 3.6502e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1127e-04, 2.1541e-04, 2.3218e-01,  ..., 1.1269e-02,
           1.0663e-04, 2.3098e-03]],

         [[4.0865e-04, 1.7524e-05, 3.1421e-01,  ..., 5.3825e-03,
           6.8712e-04, 2.0385e-05]],

         [[5.1451e-04, 4.2415e-04, 3.1104e-01,  ..., 9.5444e-03,
           4.1509e-04, 4.7922e-05]],

         ...,

         [[1.2267e-04, 2.7061e-05, 1.2695e-01,  ..., 1.0199e-01,
           6.6895e-02, 7.4890e-02]],

         [[7.7903e-05, 1.4806e-04, 3.1348e-01,  ..., 5.6076e-03,
           2.0046e-03, 4.9067e-04]],

         [[5.1260e-04, 2.0289e-04, 2.2046e-01,  ..., 5.9433e-03,
           9.4795e-04, 6.5279e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6061e-05, 1.4305e-06, 2.5902e-03,  ..., 7.1106e-02,
           2.1228e-01, 3.7750e-02]],

         [[4.1056e-04, 7.3814e-04, 1.2805e-01,  ..., 3.5706e-02,
           3.7354e-02, 3.6133e-02]],

         [[6.1226e-04, 3.0079e-03, 1.2201e-01,  ..., 6.5923e-05,
           7.1824e-05, 5.2452e-04]],

         ...,

         [[4.1847e-03, 5.6148e-05, 1.9153e-01,  ..., 1.2169e-02,
           1.8600e-02, 5.2765e-02]],

         [[1.1444e-03, 3.4332e-05, 1.1353e-02,  ..., 8.1299e-02,
           6.1523e-01, 1.1920e-01]],

         [[4.1127e-06, 8.3447e-07, 6.1083e-04,  ..., 9.9411e-03,
           1.1261e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6670e-03, 9.6035e-04, 2.3938e-01,  ..., 1.1482e-03,
           3.2845e-03, 4.7882e-02]],

         [[7.7057e-04, 1.0767e-03, 2.6318e-01,  ..., 3.3989e-03,
           1.5621e-03, 1.4168e-02]],

         [[3.7766e-03, 8.3637e-04, 2.1362e-01,  ..., 1.7746e-02,
           5.7281e-02, 2.8915e-02]],

         ...,

         [[2.9302e-04, 3.7372e-05, 8.6487e-02,  ..., 4.9934e-03,
           1.2169e-02, 2.4765e-02]],

         [[1.8120e-04, 8.3780e-04, 1.0309e-01,  ..., 6.0608e-02,
           7.6599e-02, 4.4775e-01]],

         [[3.3081e-05, 3.9935e-06, 4.0009e-02,  ..., 1.2915e-01,
           3.6206e-01, 1.8494e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.7280e-05, 2.4438e-06, 1.5569e-04,  ..., 2.9678e-02,
           1.8750e-01, 6.1475e-01]],

         [[1.3527e-02, 2.5826e-03, 7.7095e-03,  ..., 9.0103e-03,
           3.6774e-02, 2.0837e-01]],

         [[3.8147e-05, 3.8147e-05, 1.5396e-02,  ..., 1.7107e-04,
           1.3316e-04, 5.0977e-01]],

         ...,

         [[1.7853e-03, 1.3905e-03, 1.3065e-03,  ..., 9.0637e-03,
           3.4760e-02, 7.0610e-03]],

         [[6.9466e-03, 5.4092e-03, 3.7537e-02,  ..., 3.0823e-03,
           1.5656e-02, 2.0300e-01]],

         [[4.2686e-03, 4.0131e-03, 1.1379e-04,  ..., 2.4567e-02,
           4.4495e-02, 8.2321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4717e-05, 4.8018e-04, 2.2202e-03,  ..., 3.5828e-02,
           7.7026e-02, 8.5938e-02]],

         [[8.3208e-05, 7.3910e-06, 2.0862e-06,  ..., 1.2561e-01,
           5.2100e-01, 3.3105e-01]],

         [[3.6201e-03, 5.2681e-03, 2.1164e-02,  ..., 4.2343e-03,
           3.9787e-03, 9.3918e-03]],

         ...,

         [[6.6423e-04, 4.5300e-04, 2.6894e-03,  ..., 2.6535e-02,
           1.3657e-02, 5.8868e-02]],

         [[3.8528e-03, 5.6953e-03, 5.0964e-02,  ..., 2.8973e-03,
           6.6566e-03, 1.7471e-02]],

         [[5.9223e-04, 4.6043e-03, 3.2387e-03,  ..., 4.3335e-02,
           6.5063e-02, 2.0349e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1244e-03, 9.2268e-05, 1.7107e-04,  ..., 1.6943e-01,
           1.8909e-01, 1.2695e-01]],

         [[4.9973e-03, 1.9188e-03, 3.8013e-03,  ..., 1.5511e-02,
           1.4343e-02, 1.2032e-02]],

         [[1.3475e-03, 1.2100e-04, 1.0151e-04,  ..., 9.0332e-03,
           7.9727e-03, 4.6204e-02]],

         ...,

         [[1.2112e-04, 2.9802e-07, 3.5763e-06,  ..., 2.2202e-02,
           2.8125e-01, 6.4355e-01]],

         [[4.1618e-03, 6.4325e-04, 1.5678e-03,  ..., 1.3123e-02,
           6.2981e-03, 5.6122e-02]],

         [[1.7914e-02, 1.6794e-03, 1.0132e-02,  ..., 1.7227e-02,
           3.2196e-02, 1.5601e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8574e-01, 5.9605e-07, 4.2915e-06,  ..., 2.5730e-03,
           2.3556e-03, 1.7120e-02]],

         [[9.3799e-01, 1.6689e-05, 3.0398e-06,  ..., 6.5079e-03,
           9.3002e-03, 1.7075e-02]],

         [[8.5156e-01, 3.5763e-07, 1.6093e-06,  ..., 8.0414e-03,
           1.4000e-03, 2.5986e-02]],

         ...,

         [[8.1641e-01, 2.7609e-04, 1.2865e-03,  ..., 2.6608e-04,
           2.0862e-04, 3.2215e-03]],

         [[1.3391e-01, 5.9187e-05, 1.5533e-04,  ..., 3.1052e-02,
           2.0691e-02, 5.5786e-02]],

         [[8.2764e-01, 1.8513e-04, 6.0415e-04,  ..., 3.3360e-03,
           8.2932e-03, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4082e-01, 1.2815e-05, 4.9472e-06,  ..., 9.9421e-05,
           2.2471e-05, 9.0408e-03]],

         [[9.6875e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1683e-03,
           1.8860e-02, 9.5520e-03]],

         [[8.7158e-01, 7.8392e-04, 7.6115e-05,  ..., 4.6997e-03,
           1.8311e-02, 8.7814e-03]],

         ...,

         [[3.6353e-01, 1.9562e-04, 8.5950e-05,  ..., 6.6986e-03,
           7.6408e-03, 1.4946e-02]],

         [[7.4890e-02, 2.3842e-03, 1.1606e-03,  ..., 3.6550e-04,
           8.7166e-04, 3.4447e-03]],

         [[7.0264e-01, 7.9274e-06, 1.0967e-05,  ..., 8.3740e-02,
           8.8623e-02, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3315e-01, 3.6645e-04, 1.4246e-04,  ..., 4.4434e-02,
           3.3478e-02, 4.4617e-02]],

         [[6.6602e-01, 1.9121e-04, 1.7107e-05,  ..., 5.7755e-03,
           2.4891e-03, 2.5085e-02]],

         [[4.2334e-01, 5.2691e-05, 2.7478e-05,  ..., 1.9196e-02,
           4.5380e-02, 1.4610e-02]],

         ...,

         [[8.7305e-01, 1.2100e-05, 6.5565e-07,  ..., 4.5598e-05,
           3.8862e-05, 1.1383e-01]],

         [[8.6523e-01, 2.8193e-05, 3.9935e-06,  ..., 6.5565e-07,
           2.9802e-07, 1.7044e-02]],

         [[9.6240e-01, 9.3579e-06, 4.7684e-07,  ..., 1.1921e-07,
           5.9605e-08, 1.4771e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7812e-01, 6.4969e-06, 7.1526e-07,  ..., 7.7271e-02,
           9.5703e-02, 1.1299e-02]],

         [[9.1650e-01, 1.2040e-05, 2.0266e-06,  ..., 2.2674e-04,
           1.6081e-04, 8.0490e-03]],

         [[7.5879e-01, 7.8630e-04, 9.1732e-05,  ..., 7.1573e-04,
           6.4898e-04, 3.1403e-02]],

         ...,

         [[8.0469e-01, 8.6963e-05, 2.8670e-05,  ..., 1.8072e-03,
           3.9434e-04, 1.5358e-02]],

         [[8.3740e-01, 3.0398e-05, 1.4424e-05,  ..., 1.1307e-02,
           1.6975e-03, 3.2318e-02]],

         [[9.2773e-01, 9.7334e-05, 6.3837e-05,  ..., 1.0424e-03,
           7.3910e-04, 1.2993e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4443e-01, 1.6391e-05, 1.1265e-05,  ..., 2.1469e-02,
           1.5762e-02, 5.6305e-02]],

         [[7.0654e-01, 9.2864e-05, 4.8876e-06,  ..., 2.9984e-03,
           3.9711e-03, 1.4355e-01]],

         [[4.2651e-01, 2.7418e-06, 7.1526e-07,  ..., 5.5771e-03,
           1.3336e-02, 4.0527e-02]],

         ...,

         [[2.9297e-01, 8.6188e-05, 3.4273e-05,  ..., 8.6609e-02,
           8.1970e-02, 2.0233e-02]],

         [[5.6543e-01, 7.2384e-04, 6.8963e-05,  ..., 1.4122e-02,
           3.4882e-02, 2.1454e-02]],

         [[1.4136e-01, 9.6560e-06, 1.8299e-05,  ..., 2.2308e-02,
           1.3725e-02, 8.2642e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9536e-01, 2.8591e-03, 2.5635e-03,  ..., 1.0271e-03,
           1.7681e-03, 4.7874e-03]],

         [[8.0664e-01, 6.0141e-05, 3.2759e-04,  ..., 6.3286e-03,
           3.4008e-03, 1.0979e-02]],

         [[4.4458e-01, 2.7323e-04, 6.2883e-05,  ..., 1.5918e-01,
           3.9490e-02, 2.3514e-02]],

         ...,

         [[8.4961e-01, 3.2234e-04, 7.1347e-05,  ..., 1.4639e-04,
           6.4731e-05, 4.7333e-02]],

         [[2.6172e-01, 2.5868e-05, 7.2479e-05,  ..., 1.2083e-03,
           5.4598e-04, 6.4354e-03]],

         [[9.5801e-01, 4.0770e-05, 3.9220e-05,  ..., 8.4000e-03,
           6.1560e-04, 5.9700e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8853e-01, 9.3555e-04, 2.1496e-03,  ..., 1.5211e-04,
           1.0455e-04, 1.2993e-02]],

         [[9.1357e-01, 7.6294e-06, 2.8610e-05,  ..., 5.3864e-03,
           5.1804e-03, 5.4512e-03]],

         [[3.0151e-01, 1.9312e-05, 2.1601e-04,  ..., 2.2888e-02,
           1.3130e-02, 1.9470e-02]],

         ...,

         [[1.5793e-02, 8.1718e-05, 3.1018e-04,  ..., 3.4094e-05,
           3.3021e-05, 5.3825e-03]],

         [[8.9990e-01, 7.9811e-05, 7.9334e-05,  ..., 1.4259e-02,
           1.4442e-02, 2.6951e-03]],

         [[5.7129e-01, 2.6047e-05, 3.0696e-05,  ..., 1.5116e-03,
           4.1628e-04, 1.4488e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7720e-01, 9.1672e-05, 9.0075e-04,  ..., 1.0967e-04,
           8.6427e-06, 6.0028e-02]],

         [[3.0298e-01, 6.5207e-05, 6.9678e-05,  ..., 4.1199e-04,
           8.4877e-04, 3.2715e-02]],

         [[7.0605e-01, 1.5497e-06, 2.4915e-05,  ..., 3.8195e-04,
           4.1068e-05, 4.6631e-02]],

         ...,

         [[8.5889e-01, 1.9276e-04, 7.3433e-05,  ..., 2.1629e-03,
           4.2496e-03, 7.8278e-03]],

         [[4.6973e-01, 8.4066e-04, 2.6608e-04,  ..., 1.0147e-02,
           5.7945e-03, 2.1820e-02]],

         [[3.1128e-01, 8.1301e-04, 2.5797e-04,  ..., 2.4414e-04,
           2.5082e-04, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 3.9339e-06, 6.5565e-06,  ..., 1.2434e-04,
           7.9060e-04, 7.6611e-01]],

         [[6.3818e-01, 1.8282e-03, 2.4261e-03,  ..., 3.2902e-03,
           8.8196e-03, 2.1000e-03]],

         [[6.1670e-01, 5.5790e-04, 5.7840e-04,  ..., 6.8398e-03,
           1.7365e-02, 5.8746e-02]],

         ...,

         [[5.0830e-01, 1.7643e-04, 2.5988e-04,  ..., 1.0155e-02,
           5.8556e-03, 9.6497e-02]],

         [[5.0000e-01, 2.0683e-05, 2.4140e-05,  ..., 2.7679e-02,
           6.4270e-02, 5.3650e-02]],

         [[4.3237e-01, 1.1754e-04, 1.7786e-04,  ..., 3.0017e-04,
           1.4486e-03, 4.4678e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 3.7122e-04, 5.2166e-04,  ..., 1.6127e-03,
           2.5272e-03, 3.5107e-01]],

         [[8.3252e-01, 2.0719e-04, 5.4359e-04,  ..., 4.6692e-03,
           7.7286e-03, 1.3199e-02]],

         [[8.3984e-01, 3.4285e-04, 3.6788e-04,  ..., 1.1511e-03,
           3.3360e-03, 2.1912e-02]],

         ...,

         [[1.4355e-01, 3.6716e-05, 3.5703e-05,  ..., 4.1604e-05,
           4.0650e-04, 7.7783e-01]],

         [[4.8242e-01, 2.0981e-04, 1.9789e-04,  ..., 4.9257e-04,
           8.2636e-04, 3.7598e-02]],

         [[7.2754e-01, 6.4552e-05, 2.9325e-05,  ..., 2.1942e-02,
           1.0010e-01, 8.9645e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1953e-01, 2.2781e-04, 2.0337e-04,  ..., 3.0792e-02,
           1.0431e-01, 2.5482e-02]],

         [[7.8223e-01, 1.9908e-05, 2.1040e-05,  ..., 3.9642e-02,
           7.9193e-03, 2.4765e-02]],

         [[4.1602e-01, 1.0413e-04, 2.2662e-04,  ..., 2.2769e-05,
           9.7156e-06, 2.8061e-02]],

         ...,

         [[9.5801e-01, 0.0000e+00, 0.0000e+00,  ..., 2.2926e-03,
           2.3743e-02, 1.3809e-02]],

         [[5.8936e-01, 1.4191e-03, 1.9321e-03,  ..., 8.5926e-04,
           1.9131e-03, 1.3786e-02]],

         [[5.0635e-01, 2.3842e-07, 0.0000e+00,  ..., 1.7944e-02,
           3.6377e-01, 5.6244e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8809e-01, 1.0598e-04, 3.6776e-05,  ..., 2.9510e-02,
           8.1329e-03, 1.7822e-02]],

         [[3.1763e-01, 1.5974e-03, 1.0738e-03,  ..., 5.0640e-04,
           5.1641e-04, 1.2878e-01]],

         [[7.7588e-01, 3.0327e-04, 2.0754e-04,  ..., 1.9140e-03,
           4.5252e-04, 2.1988e-02]],

         ...,

         [[4.7754e-01, 2.5034e-06, 2.0266e-06,  ..., 5.0507e-03,
           1.7662e-03, 2.6337e-02]],

         [[5.5469e-01, 7.6294e-06, 3.9339e-06,  ..., 2.0325e-02,
           5.1498e-03, 3.0716e-02]],

         [[2.7930e-01, 8.8811e-05, 1.5771e-04,  ..., 4.7278e-04,
           4.8018e-04, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8877e-02, 6.3777e-06, 5.9009e-06,  ..., 1.0204e-04,
           2.9254e-04, 8.7109e-01]],

         [[6.2842e-01, 1.8477e-05, 3.2187e-05,  ..., 1.1650e-02,
           6.2866e-03, 5.1880e-02]],

         [[2.3035e-01, 1.1425e-03, 2.0084e-03,  ..., 7.0238e-04,
           1.5736e-03, 3.1052e-02]],

         ...,

         [[3.4741e-01, 9.5308e-05, 1.0884e-04,  ..., 6.3667e-03,
           4.8180e-03, 5.5023e-02]],

         [[4.9194e-01, 4.2915e-06, 4.7088e-06,  ..., 8.1873e-04,
           1.7917e-04, 5.8197e-02]],

         [[9.8450e-02, 4.2558e-05, 1.8895e-05,  ..., 2.4979e-02,
           1.1151e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9792e-02, 5.4836e-06, 1.4901e-06,  ..., 2.5879e-01,
           3.2471e-01, 3.1158e-02]],

         [[3.2910e-01, 4.3917e-04, 5.2595e-04,  ..., 2.9488e-03,
           3.1757e-03, 8.9294e-02]],

         [[2.7148e-01, 1.9789e-05, 8.5235e-06,  ..., 1.1696e-02,
           6.0120e-02, 6.9763e-02]],

         ...,

         [[4.3481e-01, 1.3828e-05, 9.4175e-06,  ..., 4.4952e-02,
           3.0411e-02, 1.0248e-01]],

         [[6.1475e-01, 1.2994e-05, 2.1338e-05,  ..., 2.6917e-02,
           3.0853e-02, 2.2049e-02]],

         [[6.6895e-01, 1.7452e-04, 1.2839e-04,  ..., 3.7003e-03,
           2.9160e-02, 5.7770e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1261e-01, 1.0166e-03, 1.3332e-03,  ..., 2.7061e-05,
           1.0765e-04, 6.2561e-03]],

         [[3.0322e-01, 1.0195e-03, 8.7166e-04,  ..., 1.6546e-04,
           1.3266e-03, 2.8931e-02]],

         [[4.0015e-01, 8.9407e-07, 1.5497e-06,  ..., 8.3923e-03,
           1.0582e-02, 1.5221e-02]],

         ...,

         [[6.2939e-01, 1.8084e-04, 1.9193e-04,  ..., 3.6430e-04,
           1.1501e-03, 2.6550e-02]],

         [[9.7900e-01, 3.3438e-05, 6.6698e-05,  ..., 9.5367e-07,
           1.2517e-06, 3.7403e-03]],

         [[6.7236e-01, 2.1315e-04, 3.2210e-04,  ..., 1.6289e-03,
           2.1229e-03, 1.0941e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2424e-01, 4.6802e-04, 7.2479e-04,  ..., 8.4066e-04,
           2.4283e-04, 6.2256e-02]],

         [[3.5913e-01, 4.2458e-03, 2.2564e-03,  ..., 6.6423e-04,
           6.8808e-04, 6.7993e-02]],

         [[3.9856e-02, 2.8074e-05, 2.9325e-05,  ..., 1.0872e-03,
           6.2943e-04, 6.1377e-01]],

         ...,

         [[1.2152e-01, 9.9540e-06, 1.8299e-05,  ..., 1.4362e-03,
           1.9836e-03, 3.2959e-02]],

         [[4.5776e-02, 6.5565e-07, 1.7881e-07,  ..., 2.2192e-01,
           3.2544e-01, 3.5950e-02]],

         [[1.9116e-01, 1.0014e-05, 6.6757e-06,  ..., 1.1040e-02,
           2.7328e-02, 5.3613e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.1098e-04, 5.0008e-05,  ..., 6.1569e-03,
           3.2444e-03, 4.8431e-02]],

         [[4.4952e-02, 1.2875e-05, 2.3365e-05,  ..., 3.4122e-03,
           4.1809e-03, 2.9480e-02]],

         [[1.9238e-01, 3.2425e-05, 1.0073e-05,  ..., 1.0252e-03,
           1.2688e-02, 7.4902e-01]],

         ...,

         [[1.3367e-01, 1.4858e-03, 3.5977e-04,  ..., 1.8454e-04,
           4.9973e-04, 1.4091e-02]],

         [[3.4399e-01, 1.0786e-03, 7.0143e-04,  ..., 1.7233e-03,
           7.5293e-04, 1.4709e-02]],

         [[1.9922e-01, 7.3814e-04, 9.3699e-04,  ..., 7.9823e-04,
           2.2125e-03, 2.1317e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 3.1650e-05, 2.9385e-05,  ..., 1.4702e-02,
           2.0416e-02, 1.3953e-01]],

         [[9.7363e-01, 1.0848e-05, 3.5167e-06,  ..., 1.5795e-04,
           3.5226e-05, 2.9335e-03]],

         [[6.6992e-01, 1.4913e-04, 7.9811e-05,  ..., 7.0457e-03,
           1.8024e-03, 2.1713e-02]],

         ...,

         [[1.2366e-01, 1.3113e-03, 2.3727e-03,  ..., 1.0529e-03,
           1.8482e-03, 3.7140e-02]],

         [[1.4880e-01, 2.2173e-05, 2.0504e-05,  ..., 4.3976e-02,
           1.8225e-01, 3.4790e-02]],

         [[3.2642e-01, 1.9703e-03, 6.1188e-03,  ..., 1.7262e-03,
           6.3629e-03, 7.7515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7908e-01, 1.1140e-04, 2.3401e-04,  ..., 1.3466e-03,
           3.4122e-03, 1.7319e-02]],

         [[3.1421e-01, 3.9995e-05, 2.2662e-04,  ..., 7.7248e-04,
           3.5458e-03, 3.3112e-02]],

         [[3.1567e-01, 4.0174e-05, 1.1998e-04,  ..., 2.7523e-03,
           1.0887e-02, 8.0261e-03]],

         ...,

         [[1.3132e-03, 3.9339e-06, 1.0192e-05,  ..., 1.1921e-06,
           4.6492e-06, 1.0223e-03]],

         [[5.1758e-01, 1.5318e-04, 1.5080e-04,  ..., 2.8019e-03,
           2.0386e-02, 1.7065e-01]],

         [[7.2998e-01, 1.1843e-04, 1.6308e-04,  ..., 5.8403e-03,
           6.4163e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5625e-01, 6.2764e-05, 8.3447e-05,  ..., 4.7660e-04,
           9.6817e-03, 4.7668e-02]],

         [[5.8740e-01, 2.9945e-04, 5.9986e-04,  ..., 5.4836e-04,
           2.8744e-03, 1.5656e-02]],

         [[8.8928e-02, 1.1504e-05, 4.2379e-05,  ..., 2.0428e-03,
           1.0139e-02, 7.9883e-01]],

         ...,

         [[2.2510e-01, 3.7670e-05, 2.5749e-05,  ..., 4.7913e-02,
           1.5710e-01, 1.9556e-01]],

         [[6.3770e-01, 8.1837e-05, 3.0184e-04,  ..., 7.8869e-04,
           7.8869e-04, 2.2171e-02]],

         [[2.6807e-01, 7.5912e-04, 6.1464e-04,  ..., 6.0272e-04,
           4.1199e-03, 1.3351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 7.1466e-05, 3.6597e-05,  ..., 5.9738e-03,
           2.2995e-02, 1.0065e-01]],

         [[2.6047e-02, 1.5020e-05, 1.3828e-05,  ..., 5.7220e-03,
           3.7231e-03, 5.2216e-02]],

         [[2.1338e-01, 1.0663e-04, 1.3578e-04,  ..., 3.0258e-02,
           2.7557e-02, 4.9896e-02]],

         ...,

         [[3.2056e-01, 0.0000e+00, 5.9605e-08,  ..., 2.1301e-02,
           4.3799e-01, 1.5625e-01]],

         [[3.7085e-01, 1.8537e-05, 8.5533e-05,  ..., 6.4659e-04,
           2.5787e-03, 8.4045e-02]],

         [[8.7451e-01, 4.1723e-07, 1.4901e-06,  ..., 3.5172e-03,
           3.0151e-02, 5.0903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0933e-02, 4.5896e-06, 1.8537e-05,  ..., 1.2338e-05,
           1.4677e-03, 9.4421e-02]],

         [[2.9572e-02, 2.0385e-05, 5.0366e-05,  ..., 4.4894e-04,
           3.9711e-03, 9.1992e-01]],

         [[1.7383e-01, 1.4019e-04, 9.6750e-04,  ..., 1.6093e-05,
           1.6201e-04, 1.8906e-02]],

         ...,

         [[1.5955e-01, 5.6219e-04, 1.7376e-03,  ..., 8.4686e-04,
           2.9621e-03, 5.2002e-02]],

         [[3.8403e-01, 1.5497e-06, 5.3644e-06,  ..., 7.8487e-04,
           7.1869e-03, 6.9153e-02]],

         [[2.4927e-01, 1.4901e-06, 1.6689e-06,  ..., 3.9581e-02,
           2.3145e-01, 1.6345e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8506e-01, 1.7536e-04, 8.5068e-04,  ..., 6.0692e-03,
           6.8245e-03, 2.8503e-02]],

         [[9.8389e-01, 1.2982e-04, 2.1899e-04,  ..., 3.7551e-05,
           4.4584e-05, 6.8283e-04]],

         [[7.9688e-01, 2.3234e-04, 7.8583e-04,  ..., 3.0594e-03,
           3.4008e-03, 4.3335e-03]],

         ...,

         [[3.0487e-02, 5.9605e-07, 1.7285e-06,  ..., 5.2147e-03,
           4.7211e-02, 2.1622e-02]],

         [[3.1152e-01, 5.6624e-06, 5.2452e-06,  ..., 9.7454e-05,
           2.7537e-04, 1.7138e-03]],

         [[1.7444e-01, 3.3081e-05, 1.0431e-04,  ..., 3.1638e-04,
           5.4779e-03, 7.2266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1151e-01, 6.1989e-06, 2.0683e-05,  ..., 3.5839e-03,
           1.6388e-02, 1.3550e-01]],

         [[6.2103e-02, 1.5497e-06, 1.5497e-06,  ..., 3.8574e-02,
           2.4561e-01, 5.5237e-02]],

         [[1.7932e-01, 7.2479e-05, 1.0580e-04,  ..., 2.5345e-02,
           9.4543e-02, 2.4368e-02]],

         ...,

         [[1.4075e-01, 1.0908e-05, 6.7472e-05,  ..., 3.3760e-03,
           1.6296e-02, 8.5388e-02]],

         [[3.7280e-01, 9.0003e-06, 1.0014e-04,  ..., 2.0905e-03,
           7.7400e-03, 7.7515e-02]],

         [[5.3809e-01, 5.7220e-06, 2.9087e-05,  ..., 1.0033e-03,
           7.2670e-03, 1.3501e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2953e-03, 2.3842e-07, 2.3842e-07,  ..., 3.0577e-05,
           1.5342e-04, 4.1962e-03]],

         [[1.8384e-01, 3.5167e-06, 2.1458e-06,  ..., 2.4643e-03,
           2.1622e-02, 9.8190e-03]],

         [[6.1676e-02, 1.5914e-05, 5.3048e-05,  ..., 5.6038e-03,
           2.1820e-02, 5.9326e-02]],

         ...,

         [[1.4087e-01, 8.2970e-05, 5.3692e-04,  ..., 1.7185e-03,
           1.5049e-03, 1.3867e-01]],

         [[1.6028e-01, 5.8270e-04, 2.0504e-03,  ..., 5.9652e-04,
           4.1008e-04, 1.2292e-01]],

         [[1.2085e-01, 3.2902e-04, 9.0122e-04,  ..., 2.2469e-03,
           1.4320e-02, 8.1116e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 4.3511e-06, 1.5795e-05,  ..., 1.7078e-01,
           2.8870e-02, 7.9880e-03]],

         [[1.0748e-01, 5.8413e-06, 3.3557e-05,  ..., 1.7614e-03,
           8.0200e-02, 3.1592e-01]],

         [[6.9141e-01, 3.5763e-07, 6.5565e-07,  ..., 9.5215e-03,
           9.8450e-02, 1.7139e-01]],

         ...,

         [[5.4736e-01, 1.8799e-04, 5.2738e-04,  ..., 1.9064e-03,
           4.2305e-03, 2.2415e-02]],

         [[4.3921e-01, 8.5235e-06, 4.5180e-05,  ..., 2.4338e-03,
           4.8027e-03, 3.3875e-02]],

         [[7.0923e-02, 3.2210e-04, 4.0555e-04,  ..., 2.7344e-02,
           1.5343e-02, 5.2832e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6711e-01, 4.8697e-05, 2.1243e-04,  ..., 1.5915e-02,
           1.4046e-02, 5.9143e-02]],

         [[1.5857e-01, 5.3177e-03, 5.3177e-03,  ..., 2.9030e-03,
           3.1147e-03, 3.8391e-02]],

         [[3.3911e-01, 9.5272e-04, 1.1339e-03,  ..., 1.3000e-02,
           1.8616e-02, 6.8909e-02]],

         ...,

         [[6.8799e-01, 3.8052e-04, 4.8494e-04,  ..., 2.4433e-03,
           4.8218e-03, 1.4389e-02]],

         [[2.3108e-01, 7.0190e-04, 1.2131e-03,  ..., 3.3493e-03,
           4.3427e-02, 5.6299e-01]],

         [[7.7295e-01, 3.7718e-04, 1.4687e-03,  ..., 2.3289e-03,
           3.8109e-03, 4.1595e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0354e-02, 6.1393e-06, 5.9128e-05,  ..., 6.0368e-04,
           7.8297e-04, 1.3904e-01]],

         [[5.7617e-01, 2.1732e-04, 2.6679e-04,  ..., 2.2430e-03,
           8.1558e-03, 2.2125e-02]],

         [[5.2393e-01, 7.0858e-04, 1.7052e-03,  ..., 1.4099e-02,
           5.3314e-02, 4.6417e-02]],

         ...,

         [[9.4482e-02, 1.0193e-02, 8.6212e-03,  ..., 5.7364e-04,
           3.5405e-04, 9.9182e-03]],

         [[8.0566e-01, 4.3213e-05, 4.1771e-04,  ..., 1.0902e-02,
           2.4815e-03, 5.8289e-03]],

         [[3.4058e-01, 2.4140e-04, 3.7026e-04,  ..., 7.9346e-04,
           1.3056e-03, 2.6459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9397e-04, 2.3060e-03, 1.8689e-01,  ..., 2.5535e-04,
           8.9121e-04, 8.8623e-02]],

         [[4.3678e-04, 9.9182e-04, 1.7249e-01,  ..., 4.7798e-03,
           1.9159e-03, 3.8177e-02]],

         [[5.0783e-04, 6.8855e-04, 1.2402e-01,  ..., 2.6001e-02,
           4.1466e-03, 5.3986e-02]],

         ...,

         [[3.3832e-04, 2.4366e-04, 1.2878e-01,  ..., 1.9245e-03,
           1.4305e-03, 2.4817e-01]],

         [[3.3398e-03, 1.5442e-02, 4.1321e-02,  ..., 1.0568e-04,
           8.9347e-05, 8.8167e-04]],

         [[1.0319e-03, 6.9857e-04, 4.1870e-02,  ..., 1.0080e-03,
           5.9509e-04, 6.7825e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8501e-03, 2.8157e-04, 1.9543e-01,  ..., 7.3576e-04,
           1.1902e-03, 1.8188e-02]],

         [[2.3766e-03, 2.5196e-03, 1.9336e-01,  ..., 5.0211e-04,
           4.0131e-03, 2.8732e-02]],

         [[2.5787e-03, 4.0865e-04, 2.9102e-01,  ..., 1.3959e-04,
           3.0756e-05, 3.9703e-02]],

         ...,

         [[1.6449e-02, 1.9875e-03, 1.6266e-02,  ..., 1.3294e-03,
           5.5313e-05, 2.5425e-03]],

         [[2.9588e-04, 6.6948e-04, 2.7173e-01,  ..., 4.7088e-04,
           3.9577e-04, 1.1932e-02]],

         [[2.1954e-03, 5.0278e-03, 1.3451e-02,  ..., 3.2187e-06,
           9.1195e-06, 6.6071e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5191e-03, 1.0977e-03, 1.3599e-01,  ..., 1.8358e-03,
           2.1011e-02, 4.7150e-02]],

         [[3.2768e-03, 4.3511e-05, 1.2732e-01,  ..., 8.7051e-03,
           3.1769e-05, 2.6077e-02]],

         [[8.6670e-03, 4.3068e-03, 2.0911e-01,  ..., 4.1084e-03,
           6.0260e-05, 5.3711e-02]],

         ...,

         [[1.5907e-03, 8.6164e-04, 1.8396e-01,  ..., 8.8730e-03,
           1.7071e-03, 4.1504e-02]],

         [[8.2111e-04, 8.5258e-04, 1.4880e-01,  ..., 4.5746e-02,
           1.0529e-03, 2.1271e-02]],

         [[7.8964e-03, 1.0672e-03, 5.7251e-02,  ..., 4.6997e-03,
           3.6573e-04, 6.1035e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6594e-04, 9.5069e-05, 2.5909e-02,  ..., 4.5204e-03,
           7.3242e-02, 3.6035e-01]],

         [[1.1196e-03, 5.0974e-04, 2.8784e-01,  ..., 1.1078e-02,
           3.9101e-03, 6.7711e-03]],

         [[5.1918e-03, 1.6251e-02, 4.3221e-03,  ..., 2.2948e-05,
           9.0241e-05, 7.6294e-02]],

         ...,

         [[2.1935e-03, 1.2970e-03, 2.7298e-02,  ..., 3.7527e-04,
           4.5090e-03, 4.8022e-01]],

         [[1.7710e-03, 2.6131e-04, 1.3344e-02,  ..., 1.2178e-03,
           1.2733e-02, 8.7207e-01]],

         [[6.8626e-03, 1.2598e-03, 2.2156e-02,  ..., 2.4471e-03,
           1.5900e-02, 6.3770e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5907e-03, 1.4849e-03, 1.7029e-01,  ..., 1.8097e-02,
           4.2236e-02, 5.3406e-02]],

         [[2.3117e-03, 6.4754e-04, 2.6733e-01,  ..., 8.4066e-04,
           3.0899e-03, 4.9286e-02]],

         [[4.5204e-03, 1.2493e-03, 2.6099e-01,  ..., 3.4294e-03,
           7.0381e-03, 5.9601e-02]],

         ...,

         [[6.8188e-04, 2.6083e-04, 1.3940e-01,  ..., 1.1778e-03,
           1.3069e-02, 3.2153e-01]],

         [[3.9601e-04, 2.8477e-03, 2.4268e-01,  ..., 4.7302e-03,
           1.1501e-03, 1.3501e-01]],

         [[2.8229e-04, 3.7611e-05, 4.3549e-02,  ..., 4.4136e-03,
           8.9874e-03, 6.1084e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.2812e-03, 1.2088e-04, 9.8133e-04,  ..., 1.0010e-01,
           6.3232e-01, 1.1340e-01]],

         [[7.6246e-04, 1.1688e-04, 6.7282e-04,  ..., 1.3244e-04,
           2.8610e-02, 9.4727e-01]],

         [[8.3447e-07, 1.3709e-06, 2.2650e-06,  ..., 2.2650e-06,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[3.1686e-04, 6.7091e-04, 4.0698e-04,  ..., 3.8605e-03,
           4.7028e-02, 7.3584e-01]],

         [[5.2719e-03, 4.6539e-03, 9.2545e-03,  ..., 2.6779e-02,
           7.7454e-02, 7.7454e-02]],

         [[6.7291e-03, 3.2806e-03, 2.0969e-04,  ..., 1.4702e-02,
           5.5809e-03, 2.2064e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5310e-04, 1.9388e-03, 5.8784e-03,  ..., 2.9861e-02,
           2.4231e-01, 1.2183e-01]],

         [[2.7955e-05, 4.9829e-05, 1.0610e-05,  ..., 1.0376e-01,
           8.2861e-01, 6.0974e-02]],

         [[1.3132e-03, 3.1013e-03, 1.0651e-02,  ..., 9.6970e-03,
           5.0842e-02, 2.6367e-02]],

         ...,

         [[1.3857e-03, 1.7653e-03, 6.7673e-03,  ..., 6.8359e-02,
           1.0260e-01, 2.8717e-02]],

         [[3.3550e-03, 9.0790e-03, 3.3081e-02,  ..., 2.0706e-02,
           6.2317e-02, 2.9541e-02]],

         [[2.3210e-04, 1.7290e-03, 4.8757e-04,  ..., 5.3772e-02,
           8.3447e-01, 3.3661e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2163e-03, 8.3780e-04, 2.6207e-03,  ..., 1.0468e-01,
           1.5234e-01, 8.6121e-02]],

         [[1.3533e-03, 3.4213e-04, 1.8787e-03,  ..., 3.3245e-03,
           3.7956e-03, 3.2082e-03]],

         [[2.8172e-03, 4.1389e-04, 4.6515e-04,  ..., 1.2627e-02,
           9.3994e-02, 1.0468e-02]],

         ...,

         [[3.9253e-03, 7.9930e-05, 1.2875e-04,  ..., 7.5867e-02,
           6.0107e-01, 1.9373e-01]],

         [[5.8889e-04, 1.1420e-04, 2.7609e-04,  ..., 5.3329e-03,
           7.7591e-03, 6.4316e-03]],

         [[3.1342e-02, 4.9973e-03, 1.1002e-02,  ..., 2.4033e-02,
           2.9980e-01, 4.8553e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1836e-01, 3.2783e-06, 1.1921e-06,  ..., 4.2677e-04,
           5.3864e-03, 1.8682e-03]],

         [[9.5703e-01, 4.2319e-06, 5.3644e-07,  ..., 1.5926e-03,
           2.8259e-02, 4.2381e-03]],

         [[7.4023e-01, 5.3644e-07, 2.3842e-07,  ..., 1.1909e-02,
           1.1253e-02, 2.4765e-02]],

         ...,

         [[1.7749e-01, 1.7941e-05, 1.7500e-04,  ..., 1.5998e-04,
           1.0826e-02, 1.1253e-02]],

         [[1.1304e-01, 2.8014e-06, 3.9101e-05,  ..., 4.5929e-03,
           2.3605e-02, 6.9237e-03]],

         [[7.3291e-01, 3.3903e-04, 2.4939e-04,  ..., 2.4071e-03,
           1.4519e-02, 5.8317e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0469e-01, 2.5034e-06, 1.5497e-06,  ..., 4.7469e-04,
           8.2855e-03, 5.2460e-02]],

         [[2.9590e-01, 5.3644e-07, 0.0000e+00,  ..., 1.4969e-02,
           6.7236e-01, 1.6357e-02]],

         [[9.6094e-01, 1.2034e-04, 5.5432e-06,  ..., 4.3182e-03,
           1.2863e-02, 4.5319e-03]],

         ...,

         [[5.7764e-01, 5.3883e-05, 2.5630e-05,  ..., 3.4580e-03,
           5.2368e-02, 1.5366e-02]],

         [[1.2732e-01, 4.9448e-04, 4.5753e-04,  ..., 9.5272e-04,
           1.4511e-02, 4.2343e-03]],

         [[2.3584e-01, 1.4305e-05, 7.6890e-06,  ..., 5.1422e-02,
           5.5469e-01, 3.7170e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2690e-01, 2.1076e-04, 8.1897e-05,  ..., 3.6316e-02,
           6.1981e-02, 2.3117e-02]],

         [[6.4600e-01, 4.5061e-05, 9.3579e-06,  ..., 6.8092e-03,
           1.2610e-01, 2.7267e-02]],

         [[5.2344e-01, 6.3181e-06, 7.5102e-06,  ..., 1.2062e-02,
           4.9225e-02, 1.2917e-02]],

         ...,

         [[8.4863e-01, 1.2159e-05, 7.1526e-06,  ..., 1.2636e-05,
           1.7748e-03, 1.4331e-01]],

         [[9.4727e-01, 2.0444e-05, 2.2006e-04,  ..., 1.7881e-07,
           3.2365e-05, 2.4902e-02]],

         [[9.8877e-01, 2.0266e-06, 2.3961e-05,  ..., 0.0000e+00,
           5.3048e-06, 8.7738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8379e-01, 1.4246e-05, 2.6822e-06,  ..., 7.4829e-02,
           3.3154e-01, 2.0752e-02]],

         [[6.2549e-01, 2.7418e-06, 7.7486e-07,  ..., 2.5129e-04,
           4.7073e-03, 5.6458e-03]],

         [[9.0430e-01, 2.5940e-04, 3.1829e-05,  ..., 1.6880e-04,
           1.1078e-02, 4.2534e-03]],

         ...,

         [[6.3135e-01, 6.9082e-05, 9.7156e-06,  ..., 7.0858e-04,
           6.4011e-03, 2.6642e-02]],

         [[2.3303e-01, 7.3314e-05, 6.9141e-06,  ..., 3.9635e-03,
           4.7112e-03, 4.6173e-02]],

         [[9.0479e-01, 9.0122e-05, 6.9320e-05,  ..., 2.7542e-03,
           1.6159e-02, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6685e-01, 8.3447e-07, 9.5367e-07,  ..., 7.3181e-02,
           9.3201e-02, 1.3901e-02]],

         [[4.1064e-01, 4.1366e-05, 2.3305e-05,  ..., 1.3687e-02,
           9.1370e-02, 1.0242e-01]],

         [[7.7783e-01, 1.5378e-05, 2.5034e-06,  ..., 4.3869e-03,
           2.5101e-02, 5.9280e-03]],

         ...,

         [[6.9629e-01, 4.3535e-04, 9.6381e-05,  ..., 1.1658e-02,
           9.3842e-03, 2.5482e-02]],

         [[8.4131e-01, 1.9455e-04, 5.1141e-05,  ..., 3.8166e-03,
           1.1665e-02, 4.9210e-03]],

         [[7.3193e-01, 3.8147e-06, 6.3777e-06,  ..., 8.7967e-03,
           5.7495e-02, 3.2990e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5811e-01, 1.1663e-03, 9.9754e-04,  ..., 1.7786e-03,
           5.6992e-03, 9.8991e-04]],

         [[9.1650e-01, 6.6221e-05, 2.0158e-04,  ..., 1.7967e-03,
           5.8022e-03, 5.0812e-03]],

         [[7.0801e-01, 4.1664e-05, 1.7941e-05,  ..., 2.8931e-02,
           4.2191e-03, 4.3869e-03]],

         ...,

         [[6.3721e-01, 5.2392e-05, 1.1265e-05,  ..., 1.8275e-04,
           9.3994e-03, 2.4414e-01]],

         [[8.7207e-01, 1.4722e-05, 4.5180e-05,  ..., 1.1683e-03,
           3.3207e-03, 3.6407e-02]],

         [[9.0820e-01, 1.3769e-05, 1.9491e-05,  ..., 4.2076e-03,
           1.8740e-03, 7.2556e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0764e-01, 4.5228e-04, 1.3475e-03,  ..., 2.0638e-03,
           1.2219e-01, 6.3324e-03]],

         [[8.7891e-01, 3.2783e-06, 6.7949e-06,  ..., 2.6150e-03,
           5.7129e-02, 2.7039e-02]],

         [[3.5498e-01, 1.7285e-06, 1.9073e-05,  ..., 7.5951e-03,
           7.3586e-03, 1.9623e-02]],

         ...,

         [[5.2979e-01, 8.4281e-05, 1.4448e-04,  ..., 1.1683e-03,
           6.1096e-02, 1.5545e-03]],

         [[8.6279e-01, 6.8605e-05, 3.9458e-05,  ..., 7.8087e-03,
           6.0806e-03, 7.3013e-03]],

         [[8.7354e-01, 5.2452e-06, 1.0192e-05,  ..., 1.9112e-03,
           4.8103e-03, 1.0963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2891e-01, 1.9431e-05, 3.3975e-05,  ..., 1.7786e-04,
           1.8204e-02, 6.8398e-03]],

         [[7.5537e-01, 1.1444e-05, 1.0848e-05,  ..., 1.8097e-02,
           4.8248e-02, 1.9272e-02]],

         [[1.1560e-01, 0.0000e+00, 1.7881e-07,  ..., 2.0046e-03,
           6.7993e-02, 5.6496e-03]],

         ...,

         [[9.2480e-01, 4.9174e-05, 2.2054e-05,  ..., 3.1452e-03,
           4.4708e-03, 4.3068e-03]],

         [[4.7559e-01, 8.1837e-05, 2.1303e-04,  ..., 2.2736e-03,
           1.5594e-02, 2.7710e-01]],

         [[5.4932e-01, 2.9731e-04, 7.4625e-05,  ..., 2.6512e-03,
           6.6948e-03, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2329e-01, 8.9407e-07, 1.7285e-06,  ..., 2.4629e-04,
           5.5237e-03, 8.5938e-01]],

         [[7.9004e-01, 8.5545e-04, 7.7820e-04,  ..., 1.7868e-02,
           2.0561e-03, 5.5742e-04]],

         [[6.1328e-01, 4.6372e-04, 6.3705e-04,  ..., 2.4353e-02,
           5.6824e-02, 1.4824e-02]],

         ...,

         [[5.0098e-01, 7.5817e-05, 1.5867e-04,  ..., 1.3222e-02,
           2.7527e-02, 1.3077e-02]],

         [[1.6199e-01, 4.0293e-05, 3.6657e-05,  ..., 6.8092e-03,
           1.0139e-02, 4.4128e-02]],

         [[1.8250e-01, 3.3677e-05, 1.1617e-04,  ..., 3.7789e-04,
           7.5378e-02, 6.5332e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9983e-01, 2.4259e-05, 4.6253e-05,  ..., 2.7733e-03,
           4.0741e-02, 6.1328e-01]],

         [[6.5869e-01, 3.1495e-04, 3.0541e-04,  ..., 2.9053e-02,
           6.1340e-02, 5.0659e-03]],

         [[6.1572e-01, 1.3328e-04, 6.9141e-05,  ..., 7.9117e-03,
           4.4136e-03, 1.2039e-02]],

         ...,

         [[9.5444e-03, 5.9605e-07, 4.2915e-06,  ..., 1.9789e-05,
           7.6294e-04, 9.8291e-01]],

         [[6.2158e-01, 6.4850e-05, 5.1320e-05,  ..., 3.3875e-03,
           6.6147e-03, 1.2866e-01]],

         [[7.1094e-01, 5.1200e-05, 1.5497e-05,  ..., 6.7444e-02,
           2.4319e-03, 2.7027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1069e-01, 3.5000e-04, 4.9543e-04,  ..., 8.6914e-02,
           3.6865e-02, 8.6288e-03]],

         [[8.5254e-01, 6.2585e-05, 1.0520e-04,  ..., 4.4556e-03,
           5.7007e-02, 1.2695e-02]],

         [[2.8931e-01, 7.0333e-06, 2.7478e-05,  ..., 1.3895e-03,
           2.5345e-02, 1.1719e-02]],

         ...,

         [[2.1570e-01, 5.9605e-08, 0.0000e+00,  ..., 9.9487e-03,
           7.5293e-01, 2.0203e-02]],

         [[4.7241e-01, 5.7745e-04, 6.9618e-04,  ..., 1.0422e-02,
           2.1652e-02, 8.3847e-03]],

         [[8.9502e-01, 1.7285e-06, 2.9802e-07,  ..., 1.9791e-02,
           3.1219e-02, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1445e-01, 4.6909e-05, 1.6451e-05,  ..., 5.1575e-03,
           1.9512e-03, 7.4501e-03]],

         [[1.2988e-01, 3.9005e-04, 3.7003e-04,  ..., 3.0780e-04,
           2.6123e-01, 4.5471e-03]],

         [[4.2432e-01, 2.6393e-04, 2.4164e-04,  ..., 6.9427e-03,
           1.2744e-01, 1.4198e-02]],

         ...,

         [[4.1724e-01, 7.9274e-06, 2.9206e-06,  ..., 9.4452e-03,
           4.1260e-01, 1.0382e-01]],

         [[7.6367e-01, 2.1458e-06, 1.0133e-06,  ..., 6.4611e-04,
           1.4636e-01, 6.5735e-02]],

         [[4.6704e-01, 2.1398e-05, 1.2279e-05,  ..., 1.2112e-03,
           3.4229e-01, 5.9052e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5054e-02, 2.5034e-06, 5.1856e-06,  ..., 5.9605e-05,
           1.5465e-02, 8.9551e-01]],

         [[4.2554e-01, 4.5300e-06, 5.8413e-06,  ..., 9.7275e-03,
           3.4204e-01, 1.3440e-01]],

         [[2.5952e-01, 1.6391e-04, 2.0516e-04,  ..., 2.2602e-03,
           4.8096e-02, 9.1248e-03]],

         ...,

         [[1.9678e-01, 1.6093e-05, 6.7949e-06,  ..., 3.8300e-02,
           1.0889e-01, 1.3206e-02]],

         [[2.6904e-01, 1.4305e-06, 5.9605e-07,  ..., 2.0993e-04,
           2.2925e-01, 1.6495e-02]],

         [[3.1470e-01, 1.3423e-04, 4.8816e-05,  ..., 9.3155e-03,
           9.8419e-03, 2.8946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1519e-01, 3.4332e-05, 2.3425e-05,  ..., 4.7760e-02,
           9.4604e-02, 8.4000e-03]],

         [[9.7778e-02, 6.3539e-05, 1.6809e-05,  ..., 9.0942e-03,
           1.6504e-01, 1.2924e-02]],

         [[9.9426e-02, 6.0201e-05, 5.1260e-06,  ..., 5.2071e-03,
           2.6962e-02, 2.7176e-02]],

         ...,

         [[4.8560e-01, 1.0610e-05, 7.6294e-06,  ..., 1.7822e-02,
           1.3074e-01, 1.9196e-02]],

         [[5.4639e-01, 1.5199e-05, 1.7285e-05,  ..., 5.6488e-02,
           6.2744e-02, 3.1067e-02]],

         [[2.7222e-01, 2.7180e-05, 1.5557e-05,  ..., 1.2659e-01,
           2.2388e-01, 5.9433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.6648e-04, 7.8249e-04,  ..., 6.2108e-05,
           1.1971e-02, 1.4162e-03]],

         [[1.9238e-01, 7.9870e-05, 1.4853e-04,  ..., 1.1635e-03,
           3.1647e-02, 8.2703e-03]],

         [[4.6680e-01, 4.2319e-06, 4.4703e-06,  ..., 4.2992e-03,
           1.4624e-01, 3.4607e-02]],

         ...,

         [[6.4062e-01, 4.9472e-05, 2.3067e-05,  ..., 1.5078e-03,
           1.9272e-02, 2.8763e-03]],

         [[9.8535e-01, 1.8299e-05, 2.4736e-05,  ..., 1.9073e-06,
           2.0313e-04, 2.8057e-03]],

         [[6.9629e-01, 1.0759e-04, 1.2779e-04,  ..., 9.3613e-03,
           2.8183e-02, 3.4237e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8408e-01, 9.5606e-05, 1.1444e-04,  ..., 6.8474e-04,
           1.2067e-01, 1.1406e-02]],

         [[1.5649e-01, 8.0872e-04, 5.0020e-04,  ..., 1.3786e-02,
           1.0510e-01, 1.8402e-02]],

         [[9.5703e-02, 4.3511e-06, 3.9339e-06,  ..., 1.9703e-03,
           7.3364e-02, 5.1318e-01]],

         ...,

         [[2.4561e-01, 3.9935e-06, 3.5167e-06,  ..., 8.3252e-02,
           2.4756e-01, 1.6586e-02]],

         [[4.1943e-01, 7.3910e-06, 6.1393e-06,  ..., 1.7517e-02,
           5.3314e-02, 2.4597e-02]],

         [[1.6187e-01, 2.0266e-06, 2.8014e-06,  ..., 5.7602e-03,
           1.9080e-01, 2.4683e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4873e-01, 9.5963e-06, 6.0201e-06,  ..., 1.2146e-02,
           6.4636e-02, 7.7209e-03]],

         [[2.1765e-01, 1.9073e-05, 2.9504e-05,  ..., 1.7414e-03,
           4.9347e-02, 3.6102e-02]],

         [[1.0931e-01, 4.5896e-06, 2.0862e-06,  ..., 4.5657e-05,
           1.8997e-02, 8.2031e-01]],

         ...,

         [[8.6670e-02, 7.9632e-05, 2.8849e-05,  ..., 4.9055e-05,
           2.8515e-03, 6.4278e-03]],

         [[3.5498e-01, 3.3665e-04, 1.7321e-04,  ..., 4.7379e-03,
           1.2581e-02, 3.6755e-03]],

         [[2.5513e-01, 2.0456e-04, 1.3053e-04,  ..., 1.4397e-02,
           7.6050e-02, 1.3847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9365e-02, 2.4319e-05, 1.1086e-05,  ..., 6.5674e-02,
           1.0333e-01, 2.2690e-02]],

         [[8.9893e-01, 1.1718e-04, 2.2292e-05,  ..., 4.5300e-06,
           7.7009e-04, 3.9406e-03]],

         [[9.1016e-01, 7.4804e-05, 1.7822e-05,  ..., 2.9862e-05,
           7.0429e-04, 2.4204e-03]],

         ...,

         [[7.7332e-02, 5.8556e-04, 1.2016e-03,  ..., 5.5580e-03,
           5.1514e-02, 3.8490e-03]],

         [[4.2896e-01, 1.0610e-05, 2.0564e-05,  ..., 4.8126e-02,
           3.3081e-02, 2.2736e-02]],

         [[3.9526e-01, 5.1212e-04, 1.5650e-03,  ..., 1.9669e-02,
           5.7831e-02, 7.9498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9004e-01, 3.2246e-05, 7.1228e-05,  ..., 5.7030e-03,
           1.2848e-02, 1.7151e-02]],

         [[1.4648e-01, 3.1710e-05, 1.0008e-04,  ..., 2.0580e-03,
           2.1774e-02, 3.8208e-02]],

         [[1.4307e-01, 1.2755e-05, 3.4034e-05,  ..., 6.2370e-03,
           6.9008e-03, 4.9324e-03]],

         ...,

         [[1.5747e-02, 1.3053e-05, 3.0875e-05,  ..., 1.7583e-05,
           3.6259e-03, 5.5313e-03]],

         [[1.9165e-01, 4.7028e-05, 4.6313e-05,  ..., 2.5034e-04,
           1.2451e-02, 9.2468e-03]],

         [[5.5859e-01, 7.2300e-05, 1.2010e-04,  ..., 9.4681e-03,
           7.8506e-03, 9.7656e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2632e-02, 5.4240e-06, 7.5698e-06,  ..., 1.9817e-03,
           1.0971e-02, 2.0493e-02]],

         [[2.0801e-01, 8.6486e-05, 1.4102e-04,  ..., 1.7042e-03,
           3.9635e-03, 7.5760e-03]],

         [[1.6565e-01, 2.9385e-05, 1.5521e-04,  ..., 1.4219e-03,
           9.6619e-02, 4.9463e-01]],

         ...,

         [[2.7295e-01, 1.8477e-05, 1.2696e-05,  ..., 2.1713e-02,
           1.0120e-01, 5.4893e-03]],

         [[1.6248e-01, 1.7107e-05, 7.6592e-05,  ..., 2.8348e-04,
           7.2517e-03, 4.4434e-02]],

         [[2.9004e-01, 2.4796e-05, 3.2842e-05,  ..., 4.4990e-04,
           6.9275e-03, 1.8072e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5063e-02, 3.5703e-05, 2.2948e-05,  ..., 2.8362e-03,
           2.3560e-02, 3.6774e-02]],

         [[1.2863e-02, 4.1723e-06, 7.7486e-06,  ..., 1.9681e-04,
           9.7046e-03, 2.1530e-02]],

         [[2.3523e-01, 1.4722e-05, 2.1040e-05,  ..., 1.4830e-03,
           8.6365e-03, 3.2520e-03]],

         ...,

         [[6.3232e-01, 1.1921e-07, 0.0000e+00,  ..., 1.0321e-01,
           1.8408e-01, 4.2023e-02]],

         [[5.3516e-01, 7.9870e-06, 1.9610e-05,  ..., 2.9953e-02,
           1.6321e-01, 2.6230e-02]],

         [[4.3066e-01, 1.1325e-06, 1.0729e-06,  ..., 1.3313e-02,
           4.9561e-01, 4.9438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2232e-02, 2.3842e-06, 1.7464e-05,  ..., 1.7548e-03,
           5.1270e-02, 2.4033e-02]],

         [[9.9106e-03, 4.7684e-06, 2.8849e-05,  ..., 8.1182e-05,
           3.2940e-03, 9.7998e-01]],

         [[1.0052e-01, 2.9087e-05, 2.0778e-04,  ..., 4.2558e-04,
           2.8137e-02, 6.0463e-04]],

         ...,

         [[1.9568e-01, 8.0884e-05, 3.4952e-04,  ..., 3.6888e-03,
           1.0760e-01, 8.1711e-03]],

         [[5.4541e-01, 5.9605e-07, 1.3113e-06,  ..., 8.3771e-03,
           9.9670e-02, 4.1565e-02]],

         [[2.0288e-01, 4.2915e-06, 2.7418e-06,  ..., 2.9572e-02,
           5.6250e-01, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 1.0943e-04, 6.6423e-04,  ..., 1.5472e-02,
           4.1565e-02, 6.5269e-03]],

         [[9.7266e-01, 4.9293e-05, 3.4595e-04,  ..., 1.5354e-04,
           6.4373e-04, 3.1519e-04]],

         [[8.1494e-01, 4.2260e-05, 1.0258e-04,  ..., 8.1787e-03,
           1.4580e-02, 1.3990e-03]],

         ...,

         [[3.2690e-01, 1.7285e-06, 3.0994e-06,  ..., 2.0889e-02,
           2.7252e-02, 1.1009e-02]],

         [[4.0112e-01, 4.7684e-07, 2.3842e-07,  ..., 3.8314e-04,
           1.8864e-03, 2.0714e-03]],

         [[1.2561e-01, 1.0192e-05, 3.2842e-05,  ..., 2.0266e-04,
           2.1835e-02, 7.8174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3025e-01, 1.8477e-06, 6.9737e-06,  ..., 1.8326e-02,
           3.3789e-01, 6.3232e-02]],

         [[2.8174e-01, 5.0664e-06, 3.8147e-06,  ..., 1.4966e-01,
           4.3701e-02, 4.2542e-02]],

         [[2.5195e-01, 1.4687e-04, 2.1279e-04,  ..., 2.2369e-02,
           1.2650e-02, 8.3923e-03]],

         ...,

         [[2.4707e-01, 1.1563e-05, 5.1856e-05,  ..., 3.8177e-02,
           9.0881e-02, 1.6678e-02]],

         [[5.3516e-01, 1.4782e-05, 7.3731e-05,  ..., 1.1734e-02,
           7.4158e-02, 9.0714e-03]],

         [[4.9780e-01, 2.0266e-06, 1.0014e-05,  ..., 7.3280e-03,
           1.6040e-01, 1.2077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6591e-02, 4.3511e-06, 3.9339e-06,  ..., 7.7903e-05,
           1.4191e-03, 1.1162e-02]],

         [[3.1396e-01, 3.9697e-05, 1.7881e-05,  ..., 3.0609e-02,
           9.9030e-03, 2.2745e-04]],

         [[2.7637e-01, 3.5167e-05, 7.2777e-05,  ..., 1.8372e-02,
           3.6804e-02, 4.2114e-03]],

         ...,

         [[3.0518e-02, 1.1802e-05, 1.1265e-04,  ..., 9.9659e-04,
           1.3782e-01, 6.7024e-03]],

         [[8.9355e-02, 5.4240e-05, 2.9802e-04,  ..., 1.8942e-04,
           2.3169e-01, 1.0674e-02]],

         [[6.7090e-01, 4.1127e-05, 4.9055e-05,  ..., 7.8125e-03,
           8.0139e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4248e-01, 1.7881e-05, 3.3140e-05,  ..., 2.0706e-02,
           1.7609e-02, 1.1237e-01]],

         [[2.4939e-01, 6.3777e-06, 3.7432e-05,  ..., 1.0236e-01,
           2.8809e-01, 1.3135e-01]],

         [[2.3352e-01, 1.8477e-06, 1.8477e-06,  ..., 1.4984e-02,
           6.4746e-01, 8.9355e-02]],

         ...,

         [[5.2783e-01, 8.0585e-05, 1.3816e-04,  ..., 4.1428e-03,
           1.9989e-02, 1.1299e-02]],

         [[1.5808e-01, 3.5167e-06, 7.6890e-06,  ..., 6.1798e-03,
           1.0291e-01, 8.8013e-02]],

         [[4.0436e-02, 9.1612e-05, 3.1471e-04,  ..., 2.8000e-02,
           5.3558e-02, 4.2139e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1567e-01, 4.2915e-05, 1.8322e-04,  ..., 3.9337e-02,
           3.5126e-02, 8.3771e-03]],

         [[6.6406e-01, 4.7064e-04, 1.0376e-03,  ..., 1.7118e-03,
           4.3869e-03, 1.6985e-03]],

         [[7.2266e-02, 8.4817e-05, 1.4651e-04,  ..., 1.9569e-03,
           4.7874e-03, 1.0767e-03]],

         ...,

         [[4.2603e-01, 4.9114e-04, 3.3212e-04,  ..., 1.4572e-02,
           1.7853e-02, 2.1881e-02]],

         [[7.5684e-02, 6.1870e-05, 2.0933e-04,  ..., 1.6113e-02,
           3.4119e-02, 7.8906e-01]],

         [[8.9795e-01, 3.5167e-05, 8.2374e-05,  ..., 5.7755e-03,
           1.4519e-02, 5.3406e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9746e-01, 8.9347e-05, 3.8981e-04,  ..., 1.8711e-03,
           2.2858e-02, 1.1909e-02]],

         [[1.3623e-01, 2.2531e-05, 6.2823e-05,  ..., 7.2899e-03,
           3.4851e-02, 3.4851e-02]],

         [[2.5635e-01, 2.7728e-04, 8.0156e-04,  ..., 1.5356e-01,
           5.8777e-02, 2.5818e-02]],

         ...,

         [[4.2871e-01, 1.4198e-04, 1.6222e-03,  ..., 2.8324e-03,
           2.4292e-02, 2.0447e-02]],

         [[8.6963e-01, 2.3484e-05, 6.9904e-04,  ..., 1.0386e-03,
           3.3321e-03, 1.8082e-03]],

         [[3.7573e-01, 3.8147e-05, 1.1253e-04,  ..., 2.1801e-03,
           1.1452e-02, 4.7760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0902e-05, 1.1158e-04, 1.7419e-01,  ..., 1.6813e-03,
           5.2734e-02, 1.4137e-02]],

         [[2.5213e-05, 9.3877e-05, 1.7822e-01,  ..., 5.0688e-04,
           1.7181e-02, 1.2230e-02]],

         [[1.4997e-04, 1.7262e-04, 1.0297e-01,  ..., 1.3039e-02,
           2.8580e-02, 1.1192e-02]],

         ...,

         [[2.0671e-04, 1.5295e-04, 1.6846e-01,  ..., 4.9286e-03,
           7.2449e-02, 2.2314e-01]],

         [[4.4751e-04, 9.3603e-04, 1.8921e-01,  ..., 2.4471e-03,
           1.4137e-02, 2.2598e-02]],

         [[2.3961e-05, 6.9559e-05, 1.6589e-01,  ..., 4.0710e-05,
           4.3182e-03, 1.8148e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1195e-05, 1.8930e-04, 2.8369e-01,  ..., 1.3530e-04,
           1.3802e-02, 2.5330e-03]],

         [[1.6189e-04, 1.9526e-04, 1.7554e-01,  ..., 1.1511e-03,
           1.0704e-02, 7.4863e-04]],

         [[7.0274e-05, 3.2544e-05, 3.1641e-01,  ..., 2.7776e-05,
           1.8265e-02, 5.8031e-04]],

         ...,

         [[4.0627e-04, 2.4438e-04, 2.8052e-01,  ..., 2.8133e-04,
           2.5284e-02, 1.2589e-03]],

         [[2.8670e-05, 1.3173e-04, 2.9102e-01,  ..., 1.2153e-04,
           8.8959e-03, 3.9244e-04]],

         [[1.4961e-05, 7.1764e-05, 6.4758e-02,  ..., 1.0127e-04,
           1.2976e-01, 2.3956e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0341e-04, 1.1975e-04, 1.8665e-01,  ..., 1.0738e-03,
           9.4376e-03, 4.4799e-04]],

         [[4.8923e-04, 3.4392e-05, 2.8149e-01,  ..., 1.7881e-05,
           1.0963e-02, 4.9496e-04]],

         [[1.5295e-04, 2.4796e-04, 3.1470e-01,  ..., 5.4836e-06,
           1.1826e-02, 1.0376e-03]],

         ...,

         [[1.6189e-04, 1.8084e-04, 1.4673e-01,  ..., 4.1466e-03,
           1.5656e-02, 8.6441e-03]],

         [[7.4744e-05, 2.1899e-04, 2.6489e-01,  ..., 2.5678e-04,
           1.1421e-02, 1.3189e-03]],

         [[4.5490e-04, 3.3855e-04, 1.7004e-01,  ..., 1.2720e-04,
           6.3171e-03, 1.2531e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7214e-04, 1.5378e-05, 3.3722e-02,  ..., 1.0864e-02,
           6.3362e-03, 3.3521e-01]],

         [[5.1856e-05, 9.5904e-05, 1.0199e-01,  ..., 4.5990e-02,
           1.2665e-02, 7.0374e-02]],

         [[2.2662e-04, 1.0223e-03, 3.5840e-01,  ..., 2.1267e-03,
           6.6795e-03, 1.5802e-03]],

         ...,

         [[2.4939e-04, 1.8656e-05, 3.3630e-02,  ..., 1.5610e-02,
           7.9870e-04, 3.6804e-02]],

         [[6.1455e-03, 8.9836e-04, 2.0667e-01,  ..., 4.0619e-02,
           3.8544e-02, 6.1829e-02]],

         [[2.6584e-05, 1.5259e-05, 1.3115e-02,  ..., 2.2232e-02,
           2.2449e-03, 3.0322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8515e-04, 8.1348e-04, 2.0020e-01,  ..., 1.4526e-02,
           4.5380e-02, 7.0801e-02]],

         [[3.7193e-04, 4.9496e-04, 2.4146e-01,  ..., 3.1185e-03,
           1.9745e-02, 3.3844e-02]],

         [[1.5402e-03, 1.3371e-03, 2.3975e-01,  ..., 6.9313e-03,
           4.4281e-02, 1.1971e-02]],

         ...,

         [[5.8651e-04, 9.4235e-05, 2.5586e-01,  ..., 2.2316e-03,
           2.2089e-04, 9.0332e-03]],

         [[1.7223e-03, 2.4109e-03, 1.2317e-01,  ..., 2.8954e-03,
           1.7609e-02, 2.8638e-01]],

         [[1.9121e-04, 6.4552e-05, 5.2612e-02,  ..., 2.3842e-03,
           2.7084e-02, 2.2498e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.3571e-03, 1.6880e-03, 2.2373e-03,  ..., 2.7515e-01,
           1.0126e-01, 4.5386e-01]],

         [[2.3537e-03, 8.1348e-04, 4.9820e-03,  ..., 1.5503e-01,
           7.3242e-02, 1.9714e-02]],

         [[2.9802e-07, 2.2650e-06, 1.2255e-04,  ..., 2.4605e-03,
           9.0504e-04, 9.9268e-01]],

         ...,

         [[6.0201e-06, 2.0981e-05, 3.4630e-05,  ..., 6.9523e-04,
           6.5956e-03, 9.7900e-01]],

         [[3.2806e-03, 3.2806e-03, 1.6663e-02,  ..., 2.9251e-02,
           2.2781e-02, 9.0088e-02]],

         [[4.3221e-03, 2.9716e-03, 2.4390e-04,  ..., 7.5874e-03,
           4.9469e-02, 1.6068e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-04, 2.5005e-03, 4.8180e-03,  ..., 1.7249e-01,
           1.2427e-01, 1.4307e-01]],

         [[3.1590e-06, 1.5748e-04, 1.7846e-04,  ..., 1.5247e-01,
           7.1582e-01, 1.1871e-01]],

         [[3.2578e-03, 3.3607e-03, 8.3237e-03,  ..., 7.7698e-02,
           3.8483e-02, 8.7204e-03]],

         ...,

         [[3.2663e-04, 5.4693e-04, 4.3964e-04,  ..., 1.4221e-02,
           7.2363e-01, 2.1387e-01]],

         [[4.0092e-03, 1.0078e-02, 3.0579e-02,  ..., 6.3477e-02,
           3.4119e-02, 1.8250e-02]],

         [[1.5473e-04, 1.0490e-03, 1.3554e-04,  ..., 7.6074e-01,
           1.2512e-01, 1.0597e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8295e-02, 1.7357e-03, 5.4092e-03,  ..., 1.8335e-01,
           7.7637e-02, 5.3772e-02]],

         [[2.1400e-03, 4.2486e-04, 1.8673e-03,  ..., 1.7748e-03,
           2.1400e-03, 1.9722e-03]],

         [[2.8286e-03, 4.8208e-04, 5.3120e-04,  ..., 8.3984e-02,
           1.0887e-02, 1.2672e-02]],

         ...,

         [[2.7809e-03, 4.1652e-04, 2.7752e-04,  ..., 2.6855e-01,
           4.3579e-01, 1.7334e-01]],

         [[8.0872e-04, 2.0611e-04, 5.3024e-04,  ..., 1.5869e-02,
           3.2318e-02, 1.3573e-02]],

         [[1.7685e-02, 3.4561e-03, 7.7286e-03,  ..., 2.2766e-01,
           9.1248e-02, 4.8462e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 3.5763e-07, 2.9802e-07,  ..., 8.8654e-03,
           1.8091e-03, 5.3482e-03]],

         [[9.7559e-01, 1.7405e-05, 2.3246e-06,  ..., 5.3444e-03,
           1.9274e-03, 6.0081e-04]],

         [[6.4258e-01, 6.5565e-07, 5.3644e-07,  ..., 6.2370e-03,
           5.0201e-03, 1.7029e-02]],

         ...,

         [[2.0813e-01, 1.3053e-05, 1.3220e-04,  ..., 9.3613e-03,
           2.2945e-03, 1.0567e-02]],

         [[2.9224e-01, 2.9206e-05, 1.6415e-04,  ..., 4.7333e-02,
           7.5493e-03, 4.4899e-03]],

         [[6.9189e-01, 1.5616e-04, 3.1257e-04,  ..., 2.6932e-02,
           1.2941e-03, 3.0746e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0576e-01, 1.4305e-06, 1.5497e-06,  ..., 9.2010e-03,
           3.0499e-03, 2.9968e-02]],

         [[2.4146e-01, 4.5359e-05, 1.7881e-07,  ..., 4.7668e-02,
           6.9434e-01, 1.5503e-02]],

         [[9.7168e-01, 4.0507e-04, 2.0862e-05,  ..., 8.0109e-03,
           5.6992e-03, 1.2283e-03]],

         ...,

         [[5.5908e-01, 5.4777e-05, 3.2842e-05,  ..., 1.6138e-01,
           2.4216e-02, 1.4381e-02]],

         [[7.3486e-02, 1.1444e-03, 6.1131e-04,  ..., 1.1742e-02,
           2.1858e-03, 2.6836e-03]],

         [[6.5137e-01, 3.8385e-05, 9.9540e-06,  ..., 1.7590e-01,
           3.8788e-02, 2.1301e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1204e-01, 1.2910e-04, 6.9618e-05,  ..., 9.3323e-02,
           3.2166e-02, 1.2115e-02]],

         [[5.3906e-01, 2.1458e-05, 5.6028e-06,  ..., 2.7856e-01,
           1.1554e-01, 2.0828e-02]],

         [[7.9932e-01, 1.1742e-05, 6.1989e-06,  ..., 2.0584e-02,
           5.1819e-02, 2.5421e-02]],

         ...,

         [[7.0117e-01, 8.2254e-06, 8.9407e-07,  ..., 2.0647e-04,
           1.2302e-03, 2.9346e-01]],

         [[9.5410e-01, 4.8161e-05, 3.8803e-05,  ..., 8.3447e-06,
           2.7275e-04, 2.9510e-02]],

         [[9.9072e-01, 2.4617e-05, 2.5630e-06,  ..., 7.7486e-07,
           3.3796e-05, 7.2937e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3613e-01, 2.1100e-05, 9.5367e-07,  ..., 8.3557e-02,
           2.8271e-01, 1.3374e-02]],

         [[4.2065e-01, 1.7881e-06, 1.3113e-06,  ..., 3.1776e-03,
           2.0337e-04, 4.1695e-03]],

         [[8.4766e-01, 9.1934e-04, 4.8161e-05,  ..., 1.7014e-02,
           5.6877e-03, 4.7607e-03]],

         ...,

         [[8.3105e-01, 3.4356e-04, 2.6882e-05,  ..., 1.0025e-02,
           2.0111e-02, 1.5686e-02]],

         [[5.9375e-01, 1.6069e-04, 1.8597e-05,  ..., 2.1301e-02,
           1.1115e-01, 7.9895e-02]],

         [[9.3701e-01, 7.1764e-05, 6.6400e-05,  ..., 1.5350e-02,
           3.0670e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9722e-01, 6.7353e-06, 1.0967e-05,  ..., 1.5381e-01,
           1.0139e-02, 1.3763e-02]],

         [[3.2275e-01, 5.5671e-05, 9.8348e-06,  ..., 9.3445e-02,
           4.7241e-02, 7.8308e-02]],

         [[9.0967e-01, 3.6538e-05, 1.0729e-05,  ..., 9.0408e-03,
           2.2232e-02, 5.6152e-03]],

         ...,

         [[5.7373e-01, 5.8651e-04, 2.5821e-04,  ..., 1.2543e-02,
           3.3051e-02, 8.0185e-03]],

         [[8.1250e-01, 3.9887e-04, 4.9114e-05,  ..., 6.1417e-03,
           5.9776e-03, 1.2627e-02]],

         [[3.9697e-01, 5.3048e-06, 5.2452e-06,  ..., 1.0669e-01,
           2.9150e-01, 3.6072e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4883e-01, 1.1597e-03, 1.0767e-03,  ..., 9.1934e-03,
           1.2388e-03, 4.3154e-04]],

         [[8.6328e-01, 4.1068e-05, 9.9301e-05,  ..., 6.9618e-03,
           1.8021e-02, 1.1436e-02]],

         [[5.8838e-01, 8.4281e-05, 3.3021e-05,  ..., 5.1537e-03,
           1.0887e-02, 8.4991e-03]],

         ...,

         [[7.8662e-01, 3.5346e-05, 2.0564e-05,  ..., 3.0251e-03,
           4.3396e-02, 3.0655e-02]],

         [[7.3291e-01, 5.3644e-06, 6.7949e-06,  ..., 2.9411e-03,
           1.9958e-01, 4.3030e-02]],

         [[9.3359e-01, 1.4842e-05, 4.8280e-06,  ..., 1.0824e-03,
           2.9221e-02, 5.4550e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0796e-01, 8.0252e-04, 1.2722e-03,  ..., 8.3618e-02,
           2.8057e-03, 1.8291e-03]],

         [[8.2812e-01, 1.6689e-06, 1.4305e-06,  ..., 2.9007e-02,
           1.2067e-01, 1.3832e-02]],

         [[6.7236e-01, 3.2783e-06, 8.5831e-06,  ..., 1.1345e-02,
           2.6276e-02, 1.2238e-02]],

         ...,

         [[3.0591e-01, 2.8777e-04, 3.8147e-04,  ..., 1.0907e-01,
           1.0862e-03, 1.7300e-03]],

         [[9.4629e-01, 2.4676e-05, 1.7285e-05,  ..., 3.3875e-03,
           1.2817e-02, 4.7646e-03]],

         [[9.4043e-01, 6.4969e-06, 7.0333e-06,  ..., 2.9697e-03,
           4.7226e-03, 2.2736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9639e-01, 5.1677e-05, 9.6560e-05,  ..., 6.6956e-02,
           1.5526e-03, 7.6103e-03]],

         [[7.8418e-01, 1.6332e-05, 1.1861e-05,  ..., 3.0487e-02,
           2.0752e-02, 1.2756e-02]],

         [[2.3975e-01, 7.7486e-07, 2.0266e-06,  ..., 1.1108e-01,
           7.9155e-04, 3.3836e-03]],

         ...,

         [[8.4570e-01, 1.2541e-04, 9.1016e-05,  ..., 8.2474e-03,
           8.9111e-03, 2.5574e-02]],

         [[4.1992e-01, 2.6083e-04, 3.2091e-04,  ..., 1.5411e-02,
           1.3672e-01, 1.9751e-01]],

         [[4.1113e-01, 3.8981e-04, 2.7061e-04,  ..., 7.7324e-03,
           2.2125e-03, 2.8553e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0017e-02, 3.5763e-07, 1.4901e-06,  ..., 4.7946e-04,
           1.2909e-02, 9.0137e-01]],

         [[6.8311e-01, 1.8139e-03, 2.0866e-03,  ..., 2.4414e-03,
           8.4686e-04, 2.2829e-04]],

         [[5.4541e-01, 5.9891e-04, 4.3130e-04,  ..., 9.8572e-02,
           1.9638e-02, 3.8147e-02]],

         ...,

         [[6.1670e-01, 1.2362e-04, 1.8334e-04,  ..., 3.0533e-02,
           1.5671e-02, 3.6049e-03]],

         [[2.0129e-01, 4.4227e-05, 7.2062e-05,  ..., 3.7048e-02,
           2.5537e-01, 4.2114e-02]],

         [[1.2421e-01, 4.4167e-05, 2.7180e-04,  ..., 1.1559e-02,
           5.9082e-02, 7.4170e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2471e-01, 1.9038e-04, 2.0182e-04,  ..., 2.6260e-02,
           1.1462e-01, 4.0161e-01]],

         [[5.7959e-01, 1.6820e-04, 2.1601e-04,  ..., 9.9548e-02,
           4.0039e-02, 2.9251e-02]],

         [[4.9756e-01, 1.8835e-04, 9.3579e-05,  ..., 1.0307e-02,
           4.2786e-02, 4.3640e-03]],

         ...,

         [[1.7441e-02, 1.7881e-07, 2.6226e-06,  ..., 7.0214e-05,
           4.5509e-03, 9.7510e-01]],

         [[8.8330e-01, 5.1677e-05, 4.6313e-05,  ..., 2.2812e-03,
           1.3924e-02, 1.4694e-02]],

         [[6.7578e-01, 8.6546e-05, 4.9293e-05,  ..., 2.9964e-03,
           1.0391e-02, 2.5330e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 6.1178e-04, 6.1178e-04,  ..., 2.7740e-02,
           6.2904e-03, 7.3166e-03]],

         [[8.9551e-01, 4.2439e-05, 9.1970e-05,  ..., 2.4811e-02,
           2.0447e-02, 3.0212e-02]],

         [[4.2334e-01, 1.8060e-05, 2.9445e-05,  ..., 4.7760e-02,
           4.2534e-03, 2.1229e-03]],

         ...,

         [[4.4507e-01, 3.6955e-06, 1.7881e-07,  ..., 1.2164e-01,
           3.9893e-01, 3.1494e-02]],

         [[4.7217e-01, 5.9891e-04, 7.1383e-04,  ..., 1.4183e-02,
           9.3765e-03, 3.7556e-03]],

         [[6.6553e-01, 6.9737e-06, 1.2517e-06,  ..., 2.3376e-02,
           2.6367e-01, 3.4515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7041e-01, 9.1195e-05, 2.8372e-05,  ..., 2.6741e-03,
           1.5762e-02, 5.4474e-03]],

         [[1.8958e-01, 4.8566e-04, 5.2977e-04,  ..., 1.6724e-01,
           2.9430e-03, 2.5120e-03]],

         [[5.8789e-01, 7.4291e-04, 8.7881e-04,  ..., 6.5063e-02,
           5.1613e-03, 8.1177e-03]],

         ...,

         [[3.8892e-01, 9.5367e-06, 5.5432e-06,  ..., 4.0186e-01,
           1.3855e-01, 2.0538e-02]],

         [[4.4360e-01, 6.4373e-06, 1.4305e-06,  ..., 1.3525e-01,
           3.8428e-01, 2.8336e-02]],

         [[4.9365e-01, 4.3988e-05, 2.8133e-05,  ..., 2.7832e-01,
           5.2002e-02, 1.5640e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0660e-02, 6.5565e-07, 2.5034e-06,  ..., 1.0223e-03,
           1.5297e-02, 9.5361e-01]],

         [[3.3276e-01, 1.4305e-06, 2.0266e-06,  ..., 3.2642e-01,
           2.2437e-01, 4.9866e-02]],

         [[2.1619e-01, 2.7609e-04, 3.2210e-04,  ..., 4.1107e-02,
           6.4316e-03, 2.8229e-03]],

         ...,

         [[2.0923e-01, 3.3319e-05, 1.1683e-05,  ..., 1.4209e-01,
           2.5085e-02, 7.2250e-03]],

         [[2.1912e-01, 2.8014e-06, 1.7881e-06,  ..., 2.6489e-01,
           5.9967e-03, 6.7215e-03]],

         [[3.0469e-01, 1.5736e-04, 5.2691e-05,  ..., 1.2711e-02,
           2.4719e-02, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 5.9843e-05, 3.9399e-05,  ..., 7.4341e-02,
           6.7291e-03, 5.5008e-03]],

         [[7.9956e-02, 7.7009e-05, 1.7822e-05,  ..., 2.2253e-01,
           8.3313e-03, 3.5133e-03]],

         [[1.6064e-01, 9.7394e-05, 1.0133e-05,  ..., 7.1838e-02,
           6.0516e-02, 1.3611e-02]],

         ...,

         [[3.9941e-01, 1.8179e-05, 9.1195e-06,  ..., 1.9934e-01,
           1.6235e-02, 1.3832e-02]],

         [[7.0801e-01, 3.3319e-05, 5.3406e-05,  ..., 3.2990e-02,
           3.9642e-02, 4.6692e-02]],

         [[3.2397e-01, 4.7684e-05, 3.4213e-05,  ..., 1.9043e-01,
           7.3547e-03, 1.0132e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5564e-01, 1.6403e-03, 2.0981e-03,  ..., 2.0050e-02,
           9.1887e-04, 9.2649e-04]],

         [[6.7200e-02, 7.6652e-05, 1.2016e-04,  ..., 6.1401e-02,
           1.4359e-02, 6.5613e-03]],

         [[4.5874e-01, 6.9737e-06, 9.1791e-06,  ..., 1.4893e-01,
           1.0480e-01, 2.7328e-02]],

         ...,

         [[6.2109e-01, 7.8261e-05, 4.9293e-05,  ..., 1.5900e-02,
           6.4087e-03, 9.4376e-03]],

         [[9.7412e-01, 1.0073e-05, 2.6464e-05,  ..., 2.0981e-04,
           4.0984e-04, 6.3248e-03]],

         [[6.1377e-01, 8.2195e-05, 1.9908e-04,  ..., 4.2175e-02,
           7.0877e-03, 1.0056e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5928e-01, 1.7858e-04, 2.8086e-04,  ..., 1.2445e-01,
           7.1869e-03, 1.0216e-02]],

         [[2.3181e-01, 1.3571e-03, 7.2050e-04,  ..., 1.2598e-01,
           1.0841e-02, 6.0806e-03]],

         [[1.6064e-01, 6.6161e-06, 6.4373e-06,  ..., 2.6428e-02,
           2.1277e-01, 4.8706e-01]],

         ...,

         [[2.4915e-01, 4.5896e-06, 4.6492e-06,  ..., 2.7588e-01,
           1.9211e-02, 1.1208e-02]],

         [[5.3662e-01, 6.9141e-06, 4.7088e-06,  ..., 9.1797e-02,
           1.1334e-01, 2.2324e-02]],

         [[1.9690e-01, 3.9339e-06, 4.4703e-06,  ..., 1.1218e-01,
           3.7354e-01, 1.2915e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4131e-01, 7.8082e-06, 4.1723e-06,  ..., 1.4233e-01,
           3.6182e-03, 8.4152e-03]],

         [[1.6895e-01, 1.6749e-05, 1.9908e-05,  ..., 7.8552e-02,
           7.8552e-02, 1.9547e-02]],

         [[1.4099e-01, 4.8280e-06, 1.3709e-06,  ..., 5.5542e-03,
           4.4373e-02, 7.8662e-01]],

         ...,

         [[1.4014e-01, 4.9353e-04, 1.8156e-04,  ..., 1.2634e-02,
           7.0877e-03, 4.2000e-03]],

         [[4.1968e-01, 4.1699e-04, 2.7514e-04,  ..., 1.1765e-02,
           3.3436e-03, 4.0817e-03]],

         [[2.3889e-01, 3.2687e-04, 1.9991e-04,  ..., 8.2520e-02,
           1.7166e-02, 8.5678e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4319e-01, 6.7472e-05, 2.4557e-05,  ..., 6.6589e-02,
           2.8198e-02, 9.0866e-03]],

         [[8.8037e-01, 1.8489e-04, 1.0490e-05,  ..., 3.2246e-05,
           2.3782e-05, 4.5753e-04]],

         [[8.6816e-01, 6.7592e-05, 7.0333e-06,  ..., 2.9826e-04,
           1.2243e-04, 1.2655e-03]],

         ...,

         [[6.2988e-02, 9.9468e-04, 2.0256e-03,  ..., 4.6814e-02,
           2.6207e-03, 4.5538e-04]],

         [[4.4263e-01, 1.5497e-05, 2.2054e-05,  ..., 3.5217e-02,
           5.5389e-02, 4.0741e-03]],

         [[3.4497e-01, 6.3038e-04, 1.7958e-03,  ..., 7.2876e-02,
           9.6359e-03, 2.1545e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8062e-01, 1.3268e-04, 2.6393e-04,  ..., 7.4196e-03,
           1.7792e-02, 2.1469e-02]],

         [[3.0981e-01, 1.6224e-04, 3.9554e-04,  ..., 1.1200e-02,
           1.2794e-02, 4.1542e-03]],

         [[2.4011e-01, 4.4644e-05, 1.0341e-04,  ..., 1.1765e-02,
           6.3477e-03, 2.3727e-03]],

         ...,

         [[1.5839e-02, 2.6584e-05, 6.1750e-05,  ..., 1.0281e-03,
           3.0231e-03, 2.7523e-03]],

         [[1.1975e-01, 7.7426e-05, 8.1122e-05,  ..., 1.8356e-02,
           4.3602e-03, 3.2921e-03]],

         [[5.6885e-01, 9.3758e-05, 1.1945e-04,  ..., 8.7738e-03,
           7.1640e-03, 1.9283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-02, 1.2517e-05, 1.1742e-05,  ..., 1.5381e-02,
           1.9897e-02, 3.5686e-03]],

         [[1.9897e-01, 9.1910e-05, 1.2910e-04,  ..., 9.2316e-03,
           1.1398e-02, 4.7874e-03]],

         [[2.6025e-01, 9.1851e-05, 3.9744e-04,  ..., 5.2887e-02,
           3.4668e-02, 4.8242e-01]],

         ...,

         [[4.5239e-01, 1.9848e-05, 2.6524e-05,  ..., 7.0984e-02,
           6.7596e-03, 4.5395e-03]],

         [[4.8779e-01, 1.0359e-04, 2.9397e-04,  ..., 6.6910e-03,
           4.1962e-02, 1.8326e-02]],

         [[3.6035e-01, 4.0174e-05, 5.8651e-05,  ..., 2.5043e-03,
           9.8038e-04, 2.8820e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9182e-02, 5.5790e-05, 3.0696e-05,  ..., 2.7985e-02,
           2.7756e-02, 1.0826e-02]],

         [[2.0645e-02, 5.6624e-06, 9.0599e-06,  ..., 2.7130e-02,
           7.3181e-02, 1.6232e-03]],

         [[3.5132e-01, 4.8280e-05, 5.8651e-05,  ..., 1.7014e-02,
           7.2632e-03, 8.4534e-03]],

         ...,

         [[4.7656e-01, 2.0266e-06, 3.5763e-07,  ..., 8.2825e-02,
           3.4863e-01, 2.0935e-02]],

         [[4.8462e-01, 1.1802e-05, 2.4259e-05,  ..., 1.8250e-01,
           7.0923e-02, 2.0309e-02]],

         [[4.0112e-01, 3.6955e-06, 3.5763e-06,  ..., 2.1143e-01,
           3.3789e-01, 3.7018e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6495e-02, 1.6689e-06, 1.2517e-05,  ..., 3.6896e-02,
           3.3081e-02, 1.0818e-02]],

         [[1.1261e-02, 2.3246e-06, 2.8014e-05,  ..., 7.9679e-04,
           1.7853e-02, 9.5947e-01]],

         [[9.7473e-02, 2.3127e-05, 1.3876e-04,  ..., 3.3417e-02,
           1.1482e-03, 1.4620e-03]],

         ...,

         [[2.9932e-01, 1.0496e-04, 3.4237e-04,  ..., 1.1450e-01,
           1.0246e-02, 6.1035e-03]],

         [[4.6362e-01, 6.5565e-07, 1.6093e-06,  ..., 7.2510e-02,
           5.3467e-02, 2.9877e-02]],

         [[2.3547e-01, 1.4246e-05, 8.5831e-06,  ..., 1.9226e-01,
           4.3311e-01, 1.0614e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2451e-01, 1.5986e-04, 9.2983e-04,  ..., 3.3630e-02,
           4.4975e-03, 5.4893e-03]],

         [[9.8389e-01, 2.1160e-05, 8.7440e-05,  ..., 4.6897e-04,
           6.9976e-05, 3.2735e-04]],

         [[8.0273e-01, 3.0994e-05, 2.8014e-05,  ..., 9.7961e-03,
           7.0095e-04, 4.8184e-04]],

         ...,

         [[3.6597e-01, 2.9802e-06, 3.3379e-06,  ..., 2.9114e-02,
           1.5106e-02, 6.1035e-03]],

         [[3.3569e-01, 9.5367e-07, 3.5763e-07,  ..., 1.0767e-03,
           1.2207e-03, 2.1935e-03]],

         [[1.2207e-01, 7.6890e-06, 3.0220e-05,  ..., 5.0392e-03,
           3.4973e-02, 7.5928e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7627e-02, 1.5497e-06, 4.9472e-06,  ..., 2.6538e-01,
           1.6345e-01, 6.3049e-02]],

         [[3.2031e-01, 1.0073e-05, 6.9737e-06,  ..., 3.6346e-02,
           3.2990e-02, 2.2400e-02]],

         [[1.5039e-01, 1.8501e-04, 1.7929e-04,  ..., 9.8038e-03,
           9.7275e-03, 2.6455e-03]],

         ...,

         [[2.5635e-01, 8.1062e-06, 4.0352e-05,  ..., 1.0767e-01,
           1.6769e-02, 6.0959e-03]],

         [[5.2441e-01, 9.1791e-06, 4.8280e-05,  ..., 7.4402e-02,
           1.3130e-02, 2.3041e-02]],

         [[5.1270e-01, 4.4703e-06, 1.4782e-05,  ..., 8.5022e-02,
           1.0315e-02, 2.5528e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8654e-03, 8.3447e-07, 5.3644e-07,  ..., 2.3136e-03,
           3.5065e-02, 3.6392e-03]],

         [[2.6758e-01, 2.5749e-05, 1.0669e-05,  ..., 3.7727e-03,
           2.5463e-04, 4.1676e-04]],

         [[2.1655e-01, 5.2929e-05, 9.0361e-05,  ..., 2.3361e-02,
           2.8896e-03, 2.9125e-03]],

         ...,

         [[6.1554e-02, 8.0466e-06, 1.2648e-04,  ..., 1.0803e-01,
           1.2901e-02, 2.5269e-02]],

         [[2.7686e-01, 2.0742e-05, 1.4615e-04,  ..., 1.0028e-01,
           1.0910e-02, 1.9135e-02]],

         [[8.8574e-01, 2.4021e-05, 3.4153e-05,  ..., 2.5131e-02,
           1.8387e-02, 1.7273e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8613e-01, 1.0014e-05, 1.4603e-05,  ..., 1.5266e-02,
           4.3854e-02, 7.0862e-02]],

         [[3.4619e-01, 8.8215e-06, 5.0843e-05,  ..., 1.4539e-01,
           1.3037e-01, 6.1340e-02]],

         [[1.1584e-01, 8.2254e-06, 7.3910e-06,  ..., 2.9590e-01,
           5.1318e-01, 5.5603e-02]],

         ...,

         [[4.4824e-01, 4.0650e-05, 7.7248e-05,  ..., 9.1248e-03,
           1.1223e-02, 4.5891e-03]],

         [[7.7271e-02, 1.9670e-06, 2.6226e-06,  ..., 5.7404e-02,
           2.1497e-01, 1.6449e-02]],

         [[4.7089e-02, 9.4295e-05, 7.1764e-04,  ..., 6.7558e-03,
           1.1810e-02, 5.7373e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2983e-01, 3.0339e-05, 1.2636e-04,  ..., 3.0685e-02,
           6.7139e-03, 5.6534e-03]],

         [[5.3760e-01, 1.0180e-04, 2.9778e-04,  ..., 3.9940e-03,
           1.6012e-03, 1.4277e-03]],

         [[9.3323e-02, 7.0512e-05, 1.4448e-04,  ..., 3.1300e-03,
           7.9489e-04, 1.7090e-03]],

         ...,

         [[5.6689e-01, 2.9683e-04, 3.2353e-04,  ..., 9.4528e-03,
           1.7380e-02, 6.0669e-02]],

         [[7.4463e-02, 3.2604e-05, 1.9038e-04,  ..., 9.6130e-03,
           3.1036e-02, 8.1299e-01]],

         [[8.6035e-01, 2.2411e-05, 4.7445e-05,  ..., 1.0826e-02,
           3.4599e-03, 4.0222e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2139e-01, 8.5473e-05, 5.6171e-04,  ..., 7.2021e-03,
           5.2986e-03, 4.6654e-03]],

         [[5.7831e-02, 6.5565e-06, 3.9279e-05,  ..., 1.8997e-02,
           6.0120e-02, 2.0538e-02]],

         [[1.2805e-01, 1.2791e-04, 3.2163e-04,  ..., 6.0974e-02,
           6.8054e-02, 5.8655e-02]],

         ...,

         [[6.8945e-01, 4.5121e-05, 3.5238e-04,  ..., 2.1942e-02,
           4.7455e-03, 2.4719e-02]],

         [[8.4277e-01, 1.1921e-06, 2.1207e-04,  ..., 3.3245e-03,
           8.3351e-04, 9.6941e-04]],

         [[3.2275e-01, 2.6941e-05, 3.7074e-05,  ..., 2.2545e-03,
           6.7520e-03, 1.7929e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4254e-05, 4.7445e-05, 9.9609e-02,  ..., 3.0746e-02,
           1.1444e-02, 2.3758e-02]],

         [[1.1683e-05, 1.2219e-05, 6.4209e-02,  ..., 5.0507e-03,
           4.8370e-03, 5.7907e-03]],

         [[1.8394e-04, 2.3675e-04, 8.7097e-02,  ..., 1.2604e-02,
           2.8400e-03, 9.1858e-03]],

         ...,

         [[7.2300e-05, 5.2691e-05, 2.0813e-01,  ..., 5.8228e-02,
           7.1350e-02, 1.4307e-01]],

         [[1.8036e-04, 2.1255e-04, 2.6099e-01,  ..., 1.2848e-02,
           3.4580e-03, 7.5226e-03]],

         [[1.1384e-05, 1.2815e-05, 1.6504e-01,  ..., 2.2984e-03,
           3.4308e-04, 1.6174e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1301e-04, 1.1301e-04, 2.9395e-01,  ..., 7.6485e-03,
           1.2293e-03, 3.1033e-03]],

         [[6.2168e-05, 7.6771e-05, 1.0962e-01,  ..., 5.4131e-03,
           1.1683e-04, 2.1877e-03]],

         [[2.7597e-05, 1.1444e-05, 2.7710e-01,  ..., 1.3062e-02,
           1.6510e-04, 3.9673e-03]],

         ...,

         [[1.6975e-04, 6.9559e-05, 2.2290e-01,  ..., 6.7825e-03,
           2.6655e-04, 1.7624e-03]],

         [[4.8280e-06, 1.3530e-05, 2.1387e-01,  ..., 4.7989e-03,
           6.3062e-05, 1.7042e-03]],

         [[3.3140e-05, 5.0843e-05, 2.2449e-01,  ..., 8.0017e-02,
           6.8426e-04, 1.8570e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1325e-05, 1.5020e-05, 1.1218e-01,  ..., 4.7417e-03,
           7.6234e-05, 8.4305e-04]],

         [[9.2328e-05, 6.1393e-06, 2.3730e-01,  ..., 8.9493e-03,
           1.0985e-04, 2.2907e-03]],

         [[4.4227e-05, 6.2764e-05, 2.4084e-01,  ..., 7.7438e-03,
           2.0039e-04, 2.4948e-03]],

         ...,

         [[5.4121e-05, 2.7478e-05, 1.1621e-01,  ..., 1.1780e-02,
           1.4801e-03, 7.0343e-03]],

         [[1.5855e-05, 4.0948e-05, 1.8579e-01,  ..., 6.6910e-03,
           2.8443e-04, 2.9736e-03]],

         [[5.4181e-05, 3.5942e-05, 1.2561e-01,  ..., 3.7498e-03,
           1.9372e-04, 2.5978e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3241e-05, 6.6757e-06, 1.9714e-02,  ..., 1.3466e-03,
           2.4011e-01, 8.6975e-02]],

         [[4.6074e-05, 5.3227e-05, 1.2512e-01,  ..., 1.0406e-02,
           6.3416e-02, 3.1189e-02]],

         [[1.9932e-04, 4.9400e-04, 3.0664e-01,  ..., 5.8517e-03,
           8.9025e-04, 7.5102e-04]],

         ...,

         [[1.7178e-04, 6.0201e-06, 3.9032e-02,  ..., 1.7452e-04,
           3.0098e-03, 5.3635e-03]],

         [[2.2202e-03, 2.2960e-04, 1.1884e-01,  ..., 2.3178e-02,
           7.6721e-02, 8.2947e-02]],

         [[6.3181e-06, 2.3842e-06, 1.9255e-03,  ..., 5.9652e-04,
           2.0190e-01, 1.0553e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1625e-03, 6.1893e-04, 1.7712e-01,  ..., 3.0304e-02,
           2.0416e-02, 6.0486e-02]],

         [[3.6645e-04, 3.4571e-04, 2.2559e-01,  ..., 1.4366e-02,
           1.7883e-02, 5.2124e-02]],

         [[1.6642e-03, 4.9400e-04, 2.6978e-01,  ..., 2.5497e-02,
           7.8354e-03, 7.9041e-03]],

         ...,

         [[4.5037e-04, 3.9995e-05, 2.7710e-01,  ..., 2.8634e-04,
           3.7422e-03, 2.8091e-02]],

         [[1.5802e-03, 1.7767e-03, 1.8408e-01,  ..., 9.4528e-03,
           1.4648e-02, 4.8767e-02]],

         [[1.1355e-04, 5.9366e-05, 4.2023e-02,  ..., 6.9695e-03,
           1.5381e-01, 2.8271e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[3.9458e-04, 1.1063e-03, 1.4801e-02,  ..., 8.7891e-02,
           3.5864e-01, 8.7891e-02]],

         [[1.2577e-04, 2.3496e-04, 1.6317e-03,  ..., 1.2833e-02,
           1.3794e-01, 1.2964e-01]],

         [[0.0000e+00, 5.9605e-08, 1.3709e-06,  ..., 2.9206e-06,
           5.5218e-04, 9.9854e-01]],

         ...,

         [[5.9605e-08, 1.1921e-07, 0.0000e+00,  ..., 1.2279e-04,
           9.0742e-04, 9.9512e-01]],

         [[4.0436e-03, 5.8823e-03, 3.7994e-03,  ..., 2.9877e-02,
           2.4765e-02, 5.8823e-03]],

         [[4.6272e-03, 1.1702e-03, 2.0981e-04,  ..., 2.1393e-02,
           1.0757e-02, 8.2031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5373e-05, 5.0926e-04, 1.6699e-03,  ..., 1.1955e-02,
           8.8379e-02, 5.4102e-01]],

         [[5.9605e-08, 1.2755e-05, 1.9014e-04,  ..., 9.8511e-02,
           7.9932e-01, 8.6914e-02]],

         [[4.4584e-04, 2.4872e-03, 6.5498e-03,  ..., 4.6906e-02,
           2.9358e-02, 4.6906e-02]],

         ...,

         [[1.3423e-04, 7.9679e-04, 6.0606e-04,  ..., 2.0923e-01,
           2.0923e-01, 4.0308e-01]],

         [[1.3075e-03, 6.0921e-03, 2.7206e-02,  ..., 3.1311e-02,
           2.1103e-02, 1.9821e-02]],

         [[7.3385e-04, 2.1400e-03, 1.1911e-03,  ..., 1.0229e-01,
           5.7959e-01, 1.1060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0356, 0.0031, 0.0159,  ..., 0.0487, 0.0373, 0.0433]],

         [[0.0009, 0.0004, 0.0019,  ..., 0.0032, 0.0038, 0.0158]],

         [[0.0034, 0.0007, 0.0008,  ..., 0.0130, 0.0167, 0.0170]],

         ...,

         [[0.0029, 0.0036, 0.0070,  ..., 0.2454, 0.2454, 0.2306]],

         [[0.0028, 0.0010, 0.0012,  ..., 0.0289, 0.0291, 0.1469]],

         [[0.0192, 0.0083, 0.0321,  ..., 0.0375, 0.0350, 0.2372]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[8.9551e-01, 1.4305e-06, 1.1921e-06,  ..., 7.1383e-04,
           1.3599e-03, 1.2428e-02]],

         [[9.3604e-01, 2.1577e-05, 1.6689e-06,  ..., 1.3016e-02,
           2.5635e-02, 1.0498e-02]],

         [[8.3887e-01, 7.9870e-06, 5.3048e-06,  ..., 2.9736e-03,
           4.6844e-03, 4.4830e-02]],

         ...,

         [[1.8213e-01, 5.1260e-05, 3.9625e-04,  ..., 4.1127e-06,
           3.3975e-06, 2.6550e-02]],

         [[5.0586e-01, 2.1231e-04, 4.2605e-04,  ..., 1.7166e-02,
           5.7411e-03, 2.3285e-02]],

         [[4.6021e-01, 3.2091e-04, 2.2554e-04,  ..., 2.6276e-02,
           7.2266e-02, 1.6830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8857e-01, 1.4186e-05, 5.3048e-06,  ..., 7.4506e-05,
           7.2181e-05, 3.8666e-02]],

         [[9.5020e-01, 1.2934e-04, 7.2241e-05,  ..., 4.7302e-03,
           3.1113e-02, 1.2772e-02]],

         [[9.4824e-01, 2.0456e-04, 1.2898e-04,  ..., 3.3531e-03,
           1.7105e-02, 6.8550e-03]],

         ...,

         [[3.8623e-01, 2.8539e-04, 1.2589e-04,  ..., 8.1116e-02,
           1.0559e-01, 3.2196e-02]],

         [[2.5439e-01, 7.5674e-04, 3.5095e-04,  ..., 8.9417e-03,
           4.6692e-03, 4.7646e-03]],

         [[7.8857e-01, 4.7755e-04, 2.0647e-04,  ..., 2.0798e-02,
           2.3422e-02, 5.7587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9592e-01, 2.9516e-04, 7.0393e-05,  ..., 6.8848e-02,
           3.4302e-02, 4.2084e-02]],

         [[7.4707e-01, 4.0054e-05, 1.1206e-05,  ..., 4.6539e-02,
           3.9673e-02, 4.3488e-02]],

         [[6.0596e-01, 1.2636e-05, 3.8624e-05,  ..., 2.5803e-02,
           8.8867e-02, 4.2572e-02]],

         ...,

         [[8.3936e-01, 6.9141e-06, 2.9802e-07,  ..., 1.2517e-06,
           4.4107e-05, 1.5222e-01]],

         [[9.2969e-01, 6.6400e-05, 1.9073e-06,  ..., 0.0000e+00,
           6.5565e-07, 1.9791e-02]],

         [[9.8828e-01, 1.7941e-05, 5.9605e-08,  ..., 0.0000e+00,
           5.9605e-08, 4.9210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 3.1292e-05, 4.7684e-06,  ..., 2.3816e-01,
           1.4685e-01, 4.4739e-02]],

         [[6.6455e-01, 7.8082e-06, 1.8477e-06,  ..., 1.1516e-04,
           2.1875e-05, 7.4402e-02]],

         [[7.2070e-01, 5.7161e-05, 9.8944e-06,  ..., 1.9684e-03,
           3.2776e-02, 1.1642e-02]],

         ...,

         [[6.6797e-01, 5.3835e-04, 2.0742e-05,  ..., 3.6755e-03,
           2.5177e-03, 2.3766e-03]],

         [[5.9717e-01, 2.3687e-04, 4.3631e-05,  ..., 9.3613e-03,
           2.0233e-02, 1.8066e-01]],

         [[8.4570e-01, 1.4663e-04, 1.7345e-04,  ..., 1.0605e-03,
           5.5542e-03, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8271e-01, 1.0550e-05, 1.7107e-05,  ..., 6.7291e-03,
           4.5738e-03, 2.6505e-02]],

         [[6.6260e-01, 4.3511e-05, 3.1352e-05,  ..., 1.0750e-02,
           6.8207e-03, 3.0334e-02]],

         [[9.5508e-01, 1.3590e-05, 5.5432e-06,  ..., 5.6686e-03,
           4.6997e-03, 9.1629e-03]],

         ...,

         [[5.7275e-01, 1.5616e-04, 5.7697e-05,  ..., 4.0192e-02,
           1.8661e-02, 4.1046e-02]],

         [[8.7354e-01, 1.0759e-04, 2.5153e-05,  ..., 5.1918e-03,
           4.2953e-03, 1.3733e-02]],

         [[8.5547e-01, 1.6153e-05, 1.5140e-05,  ..., 1.9745e-02,
           4.1443e-02, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 5.0068e-04, 3.3641e-04,  ..., 8.7929e-04,
           6.6376e-04, 2.7618e-03]],

         [[9.3408e-01, 1.6868e-05, 6.1333e-05,  ..., 8.4152e-03,
           2.7962e-03, 3.4389e-03]],

         [[9.0234e-01, 1.0788e-05, 7.9870e-06,  ..., 8.4829e-04,
           6.4564e-04, 1.9045e-03]],

         ...,

         [[8.5059e-01, 1.6451e-05, 2.4199e-05,  ..., 2.4796e-02,
           1.0529e-02, 1.8463e-02]],

         [[8.1104e-01, 3.2485e-05, 2.3246e-05,  ..., 3.7018e-02,
           2.3422e-02, 9.0820e-02]],

         [[9.3652e-01, 2.0742e-05, 1.7405e-05,  ..., 3.1853e-03,
           1.5516e-03, 1.6037e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3816e-01, 1.7226e-04, 1.0471e-03,  ..., 2.1866e-02,
           1.1009e-02, 1.3794e-02]],

         [[8.6377e-01, 1.7166e-05, 1.6510e-05,  ..., 1.9547e-02,
           3.2104e-02, 2.0889e-02]],

         [[6.3428e-01, 3.2187e-05, 7.8380e-05,  ..., 2.8458e-02,
           1.6602e-02, 2.2247e-02]],

         ...,

         [[7.4951e-01, 3.0756e-05, 1.3173e-04,  ..., 2.7237e-03,
           6.0234e-03, 1.7560e-04]],

         [[7.2510e-01, 4.8637e-05, 3.7909e-05,  ..., 1.1932e-02,
           4.5776e-03, 2.8442e-02]],

         [[8.7109e-01, 3.6001e-05, 4.0352e-05,  ..., 5.5771e-03,
           4.9225e-02, 1.8478e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9375e-01, 4.5002e-05, 3.9697e-05,  ..., 4.1127e-06,
           7.1526e-07, 2.5208e-02]],

         [[8.1934e-01, 1.4126e-05, 3.6955e-06,  ..., 3.3905e-02,
           2.2247e-02, 5.3680e-02]],

         [[7.3096e-01, 1.6689e-06, 2.0862e-06,  ..., 1.3981e-03,
           7.9453e-05, 2.9251e-02]],

         ...,

         [[9.2627e-01, 5.2452e-05, 4.2617e-05,  ..., 1.5993e-03,
           4.5586e-03, 8.0414e-03]],

         [[7.9395e-01, 6.7890e-05, 3.8981e-05,  ..., 2.6154e-02,
           9.0942e-03, 9.8801e-04]],

         [[4.5386e-01, 4.3964e-04, 8.6904e-05,  ..., 4.8294e-03,
           1.5554e-03, 1.3397e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0640e-01, 2.6822e-06, 4.8280e-06,  ..., 2.3117e-03,
           2.1667e-03, 6.7725e-01]],

         [[8.0127e-01, 4.8542e-04, 6.4468e-04,  ..., 5.2452e-04,
           1.3673e-04, 1.7920e-03]],

         [[6.6650e-01, 6.3848e-04, 8.1205e-04,  ..., 4.6997e-03,
           5.7106e-03, 2.1942e-02]],

         ...,

         [[5.8936e-01, 1.5485e-04, 1.5485e-04,  ..., 5.1193e-03,
           6.5947e-04, 7.5626e-04]],

         [[7.7930e-01, 1.1116e-04, 9.0003e-05,  ..., 1.1604e-02,
           1.8873e-03, 1.4816e-02]],

         [[5.3125e-01, 5.5611e-05, 1.6403e-04,  ..., 1.3687e-02,
           1.8433e-02, 3.4473e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3854e-02, 2.2054e-06, 2.0862e-06,  ..., 9.4128e-04,
           1.0910e-03, 9.4336e-01]],

         [[7.2607e-01, 8.1778e-05, 1.1313e-04,  ..., 2.5879e-02,
           5.2734e-02, 7.0801e-03]],

         [[4.5679e-01, 1.4389e-04, 3.4213e-05,  ..., 6.5269e-03,
           5.3072e-04, 2.6352e-02]],

         ...,

         [[4.0359e-03, 3.5763e-07, 1.3709e-06,  ..., 2.9111e-04,
           5.6696e-04, 9.8975e-01]],

         [[4.4019e-01, 2.9027e-05, 4.1544e-05,  ..., 1.1131e-02,
           5.6877e-03, 3.1763e-01]],

         [[8.8574e-01, 7.3254e-05, 1.1206e-05,  ..., 7.1144e-04,
           4.7588e-04, 3.4542e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9336e-01, 3.0279e-04, 3.7241e-04,  ..., 9.8343e-03,
           3.2864e-03, 2.0325e-02]],

         [[6.9482e-01, 4.3809e-05, 1.0717e-04,  ..., 6.6040e-02,
           1.6589e-01, 1.1726e-02]],

         [[5.7520e-01, 1.8239e-05, 8.5592e-05,  ..., 4.1580e-03,
           2.2995e-04, 1.5266e-02]],

         ...,

         [[4.8584e-01, 3.7014e-05, 2.0683e-05,  ..., 7.6843e-02,
           4.1138e-01, 1.9226e-02]],

         [[4.7192e-01, 4.9734e-04, 7.6723e-04,  ..., 1.0597e-02,
           3.0518e-03, 2.7776e-04]],

         [[8.6230e-01, 1.5557e-05, 1.1086e-05,  ..., 2.7725e-02,
           6.5186e-02, 3.9825e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0703e-01, 1.4603e-05, 1.3411e-05,  ..., 2.4948e-02,
           4.0359e-03, 4.5441e-02]],

         [[1.0175e-01, 6.0272e-04, 3.2902e-04,  ..., 1.2108e-02,
           1.4191e-02, 5.5027e-04]],

         [[3.5620e-01, 1.1187e-03, 9.0122e-04,  ..., 2.6840e-02,
           5.1025e-02, 5.8022e-03]],

         ...,

         [[4.8682e-01, 1.0461e-04, 5.6207e-05,  ..., 1.4539e-01,
           1.6541e-01, 8.1863e-03]],

         [[7.5244e-01, 7.5698e-05, 6.0141e-05,  ..., 6.8542e-02,
           7.2937e-02, 6.2622e-02]],

         [[5.6689e-01, 2.4796e-04, 9.8646e-05,  ..., 6.3477e-02,
           4.8004e-02, 8.1940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7811e-02, 1.6093e-06, 1.6093e-06,  ..., 4.3392e-04,
           1.1129e-03, 9.5654e-01]],

         [[1.8103e-01, 8.0466e-06, 7.3314e-06,  ..., 2.1582e-01,
           2.8052e-01, 5.9814e-02]],

         [[1.7297e-01, 4.2176e-04, 4.7541e-04,  ..., 1.9913e-02,
           4.4670e-03, 9.0103e-03]],

         ...,

         [[3.7573e-01, 1.9407e-04, 5.6803e-05,  ..., 2.5452e-02,
           6.2714e-03, 3.4637e-03]],

         [[6.3721e-01, 1.1742e-05, 4.5896e-06,  ..., 1.2428e-02,
           1.6479e-03, 1.1032e-02]],

         [[7.2754e-01, 4.4942e-05, 6.8545e-05,  ..., 3.0251e-03,
           2.1541e-04, 2.5272e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8145e-01, 5.2571e-05, 5.4300e-05,  ..., 3.8147e-02,
           1.5602e-02, 4.8035e-02]],

         [[2.2327e-01, 6.9618e-05, 3.1114e-05,  ..., 1.6754e-02,
           3.1605e-03, 6.7017e-02]],

         [[5.3223e-01, 9.9778e-05, 9.6440e-05,  ..., 1.6983e-02,
           7.9269e-03, 1.9318e-02]],

         ...,

         [[6.8652e-01, 2.3782e-05, 1.4842e-05,  ..., 2.5299e-02,
           1.6144e-02, 1.7807e-02]],

         [[7.4512e-01, 5.7936e-05, 5.9426e-05,  ..., 3.5553e-02,
           7.2632e-02, 1.7670e-02]],

         [[3.9233e-01, 4.9472e-06, 3.8147e-06,  ..., 1.2810e-02,
           6.6719e-03, 1.3016e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 1.0319e-03, 7.7677e-04,  ..., 1.9569e-03,
           8.2970e-04, 3.4294e-03]],

         [[1.7273e-01, 4.6194e-05, 4.1604e-05,  ..., 5.5962e-03,
           9.6560e-04, 2.4078e-02]],

         [[3.0591e-01, 1.3173e-05, 1.2696e-05,  ..., 1.6309e-01,
           2.1619e-01, 4.8218e-02]],

         ...,

         [[5.0830e-01, 1.5891e-04, 4.5598e-05,  ..., 1.8110e-03,
           1.4458e-03, 4.6539e-02]],

         [[5.3418e-01, 2.5094e-05, 3.1352e-05,  ..., 2.0266e-06,
           8.2254e-06, 2.3880e-03]],

         [[6.0010e-01, 8.4758e-05, 1.2398e-04,  ..., 5.3368e-03,
           5.0621e-03, 5.3644e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2781e-01, 4.8578e-05, 2.8551e-05,  ..., 1.9913e-02,
           4.4891e-02, 6.8817e-03]],

         [[1.6992e-01, 2.8825e-04, 1.1247e-04,  ..., 1.3832e-02,
           2.1801e-03, 1.5930e-02]],

         [[2.9199e-01, 1.6630e-05, 7.7486e-06,  ..., 1.1798e-01,
           9.5520e-02, 1.7432e-01]],

         ...,

         [[3.7012e-01, 6.5565e-06, 3.8743e-06,  ..., 2.5787e-02,
           2.1210e-02, 2.0630e-02]],

         [[6.8896e-01, 1.1802e-05, 5.9009e-06,  ..., 4.1687e-02,
           8.8806e-03, 1.1157e-01]],

         [[2.7344e-01, 2.0623e-05, 1.6928e-05,  ..., 1.3123e-01,
           7.1899e-02, 3.9478e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 1.7643e-05, 6.6757e-06,  ..., 3.9291e-03,
           4.0512e-03, 4.0283e-02]],

         [[1.5918e-01, 6.6757e-06, 1.6510e-05,  ..., 2.3157e-01,
           8.5205e-02, 2.9434e-02]],

         [[1.4441e-01, 8.1658e-06, 1.1325e-06,  ..., 4.3602e-03,
           7.4158e-03, 8.0518e-01]],

         ...,

         [[1.7188e-01, 2.0444e-04, 1.4102e-04,  ..., 1.4102e-04,
           2.2805e-04, 2.0859e-02]],

         [[3.7476e-01, 1.5461e-04, 7.5102e-05,  ..., 2.6169e-03,
           2.2545e-03, 2.7802e-02]],

         [[2.0825e-01, 1.3900e-04, 8.1718e-05,  ..., 1.8921e-02,
           3.8147e-03, 6.0959e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2324e-01, 1.9431e-05, 1.1206e-05,  ..., 2.0676e-02,
           6.1569e-03, 6.6223e-02]],

         [[7.7930e-01, 2.1505e-04, 3.3617e-05,  ..., 4.3511e-06,
           1.0729e-05, 7.5459e-05]],

         [[8.7402e-01, 1.9145e-04, 5.6803e-05,  ..., 8.1718e-05,
           1.9598e-04, 8.3494e-04]],

         ...,

         [[1.4673e-01, 5.8603e-04, 1.1206e-03,  ..., 4.1656e-03,
           5.4169e-04, 5.1003e-03]],

         [[6.5332e-01, 1.0312e-05, 1.5974e-05,  ..., 3.5736e-02,
           2.1133e-03, 2.1011e-02]],

         [[3.6401e-01, 5.3453e-04, 1.1768e-03,  ..., 5.0697e-03,
           6.8245e-03, 1.3680e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8428e-01, 7.5161e-05, 8.7917e-05,  ..., 1.0239e-02,
           1.2062e-02, 1.1787e-02]],

         [[2.6270e-01, 2.2328e-04, 2.9802e-04,  ..., 3.1300e-03,
           9.4700e-04, 3.5004e-02]],

         [[3.5034e-01, 4.4584e-05, 6.3896e-05,  ..., 6.2037e-04,
           2.8849e-04, 5.2757e-03]],

         ...,

         [[5.4871e-02, 4.9293e-05, 7.6294e-05,  ..., 3.0956e-03,
           4.1656e-03, 5.4871e-02]],

         [[2.4475e-02, 2.3007e-05, 2.7776e-05,  ..., 4.1461e-04,
           3.5453e-04, 6.5491e-02]],

         [[4.9243e-01, 7.6830e-05, 8.5711e-05,  ..., 3.9406e-03,
           1.2016e-03, 1.0063e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0386e-01, 3.1531e-05, 2.8253e-05,  ..., 2.9297e-03,
           7.7629e-04, 4.2725e-02]],

         [[3.9746e-01, 3.1233e-04, 3.6526e-04,  ..., 4.7722e-03,
           2.6779e-03, 9.0561e-03]],

         [[1.4233e-01, 6.8963e-05, 1.6737e-04,  ..., 1.1963e-02,
           3.9520e-02, 5.9912e-01]],

         ...,

         [[5.4883e-01, 9.6262e-05, 1.8704e-04,  ..., 1.6968e-02,
           8.5297e-03, 7.3730e-02]],

         [[1.3635e-01, 7.6294e-06, 4.7565e-05,  ..., 3.8314e-04,
           2.3234e-04, 6.5430e-02]],

         [[3.6108e-01, 1.9205e-04, 3.4642e-04,  ..., 5.7364e-04,
           4.6086e-04, 9.3307e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6626e-01, 1.1969e-04, 6.9976e-05,  ..., 2.1301e-02,
           1.0468e-02, 1.2164e-01]],

         [[9.9426e-02, 6.3062e-05, 1.3196e-04,  ..., 1.7725e-01,
           1.6113e-02, 1.2352e-02]],

         [[5.2441e-01, 4.0591e-05, 1.1533e-04,  ..., 6.4964e-03,
           8.1177e-03, 1.1230e-02]],

         ...,

         [[6.5576e-01, 1.8775e-05, 2.8729e-05,  ..., 4.8248e-02,
           2.3010e-01, 5.1361e-02]],

         [[5.5225e-01, 1.1683e-05, 2.9325e-05,  ..., 3.5858e-02,
           2.9266e-02, 2.0432e-02]],

         [[2.9907e-01, 8.1658e-06, 1.2815e-05,  ..., 1.1896e-01,
           4.7778e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4250e-02, 1.8477e-06, 1.2398e-05,  ..., 1.0132e-01,
           3.2135e-02, 4.9377e-02]],

         [[4.2480e-02, 1.6689e-06, 5.8413e-06,  ..., 1.6336e-03,
           4.3373e-03, 9.2236e-01]],

         [[4.8523e-02, 2.1100e-05, 6.7234e-05,  ..., 1.2756e-02,
           1.1261e-02, 1.8555e-02]],

         ...,

         [[1.6711e-01, 5.3883e-05, 1.4007e-04,  ..., 3.5583e-02,
           2.3514e-02, 1.8234e-02]],

         [[5.6250e-01, 4.0531e-06, 8.1062e-06,  ..., 5.4626e-02,
           3.3783e-02, 8.3618e-02]],

         [[1.4429e-01, 1.4067e-05, 1.2994e-05,  ..., 1.0315e-01,
           6.2939e-01, 7.6965e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6016e-01, 1.0377e-04, 5.1689e-04,  ..., 3.7746e-03,
           4.5013e-03, 1.4473e-02]],

         [[9.6436e-01, 3.6597e-05, 7.0393e-05,  ..., 2.5570e-05,
           5.3942e-05, 6.5899e-04]],

         [[8.8867e-01, 2.6822e-05, 6.7949e-06,  ..., 6.1274e-05,
           2.2829e-05, 3.8719e-04]],

         ...,

         [[7.7832e-01, 6.9737e-06, 9.8944e-06,  ..., 1.3809e-02,
           9.0561e-03, 3.5828e-02]],

         [[6.0352e-01, 1.8477e-06, 5.9605e-07,  ..., 4.8566e-04,
           2.7251e-04, 6.1989e-03]],

         [[1.0931e-01, 6.4969e-06, 9.2983e-06,  ..., 2.8896e-03,
           1.0490e-02, 7.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3497e-02, 3.2783e-06, 1.0133e-05,  ..., 2.1313e-01,
           1.2830e-01, 9.6069e-02]],

         [[5.6494e-01, 1.5318e-05, 1.0371e-05,  ..., 1.9867e-02,
           1.9958e-02, 2.2217e-01]],

         [[1.2195e-01, 1.5354e-04, 1.3220e-04,  ..., 4.0863e-02,
           7.9803e-03, 7.1716e-02]],

         ...,

         [[5.0146e-01, 1.1981e-05, 5.6684e-05,  ..., 4.2816e-02,
           2.2202e-02, 6.7871e-02]],

         [[5.6787e-01, 1.6868e-05, 3.5167e-05,  ..., 2.8503e-02,
           1.3046e-02, 2.4185e-02]],

         [[6.3135e-01, 6.2585e-06, 1.6034e-05,  ..., 3.7823e-03,
           5.2528e-03, 4.8676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9440e-02, 3.1590e-06, 1.8477e-06,  ..., 4.1168e-02,
           5.8403e-03, 2.0218e-02]],

         [[1.5381e-01, 5.9187e-05, 2.5690e-05,  ..., 1.0166e-03,
           7.2384e-04, 7.3090e-03]],

         [[2.8198e-01, 6.8009e-05, 1.6022e-04,  ..., 6.5804e-03,
           3.4809e-03, 5.4657e-02]],

         ...,

         [[5.6305e-02, 5.4240e-06, 7.0691e-05,  ..., 1.1894e-02,
           3.1586e-02, 3.4668e-02]],

         [[1.1975e-01, 7.9274e-06, 5.5790e-05,  ..., 9.8267e-03,
           1.7807e-02, 9.8267e-03]],

         [[7.7246e-01, 2.5272e-05, 4.8339e-05,  ..., 5.1758e-02,
           8.5327e-02, 4.0314e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0889e-01, 2.1458e-06, 7.2718e-06,  ..., 2.8366e-02,
           8.4045e-02, 2.1619e-01]],

         [[3.7158e-01, 2.7835e-05, 8.5831e-05,  ..., 9.3933e-02,
           1.2396e-01, 1.5369e-01]],

         [[3.4253e-01, 6.6757e-06, 4.8757e-05,  ..., 6.7139e-02,
           4.8291e-01, 7.9468e-02]],

         ...,

         [[4.2358e-01, 3.2961e-05, 3.1471e-05,  ..., 1.3191e-02,
           3.6507e-03, 1.1284e-02]],

         [[1.8604e-01, 5.6028e-06, 1.0252e-05,  ..., 1.8176e-01,
           9.2834e-02, 4.7791e-02]],

         [[6.5979e-02, 2.7359e-05, 8.1599e-05,  ..., 3.3112e-03,
           1.3298e-02, 6.2109e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9121e-01, 1.6868e-05, 1.0830e-04,  ..., 8.0338e-03,
           4.7798e-03, 1.3855e-01]],

         [[2.0337e-01, 2.6047e-05, 8.2672e-05,  ..., 3.4857e-04,
           3.1114e-04, 4.0741e-03]],

         [[2.5195e-01, 6.4254e-05, 2.7275e-04,  ..., 7.6199e-04,
           1.1568e-03, 5.1178e-02]],

         ...,

         [[4.7217e-01, 3.4857e-04, 5.3596e-04,  ..., 1.9180e-02,
           4.1901e-02, 2.8198e-01]],

         [[5.0879e-01, 1.5306e-04, 5.9605e-04,  ..., 3.1052e-02,
           7.3303e-02, 2.4426e-01]],

         [[5.7861e-01, 3.2723e-05, 4.5061e-05,  ..., 2.9888e-03,
           1.5671e-02, 3.6206e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5488e-01, 3.7730e-05, 2.5058e-04,  ..., 5.9357e-02,
           5.4230e-02, 3.3539e-02]],

         [[1.4087e-01, 4.1723e-06, 5.5909e-05,  ..., 2.7618e-02,
           8.4106e-02, 1.0968e-01]],

         [[3.3984e-01, 1.3804e-04, 6.3848e-04,  ..., 3.6682e-02,
           5.2948e-02, 9.2590e-02]],

         ...,

         [[5.3467e-01, 4.1604e-05, 1.5628e-04,  ..., 2.5665e-02,
           6.1676e-02, 9.7733e-03]],

         [[9.0967e-01, 1.0133e-06, 4.6182e-04,  ..., 2.7275e-04,
           1.6356e-04, 3.0804e-03]],

         [[6.4258e-01, 3.2783e-06, 9.7752e-06,  ..., 2.1561e-02,
           1.7120e-02, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7193e-05, 1.7762e-05, 1.7249e-01,  ..., 5.4779e-03,
           1.3405e-02, 2.2186e-02]],

         [[2.3782e-05, 1.6510e-05, 1.4539e-01,  ..., 1.0162e-02,
           1.3466e-02, 1.2070e-02]],

         [[4.0174e-05, 4.9591e-05, 9.8633e-02,  ..., 6.0387e-03,
           1.6739e-02, 3.6560e-02]],

         ...,

         [[6.5506e-05, 7.8082e-05, 2.8174e-01,  ..., 9.2773e-03,
           9.2224e-02, 1.3843e-01]],

         [[2.2209e-04, 1.1826e-04, 2.7710e-01,  ..., 2.6016e-03,
           4.7112e-03, 2.9770e-02]],

         [[4.4465e-05, 1.6093e-05, 3.1299e-01,  ..., 5.9938e-04,
           1.1110e-03, 5.1003e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2493e-04, 6.2108e-05, 3.2227e-01,  ..., 4.8661e-04,
           7.5626e-04, 4.2343e-03]],

         [[5.3465e-05, 4.3213e-05, 2.0825e-01,  ..., 1.8704e-04,
           9.3126e-04, 2.4529e-03]],

         [[1.2398e-05, 4.2319e-06, 2.7002e-01,  ..., 1.0931e-04,
           8.9741e-04, 5.0163e-04]],

         ...,

         [[1.1438e-04, 1.0496e-04, 2.7075e-01,  ..., 5.3835e-04,
           1.8101e-03, 2.9259e-03]],

         [[1.1921e-06, 1.7881e-06, 2.3254e-01,  ..., 1.4353e-04,
           1.0118e-03, 4.8256e-03]],

         [[1.0276e-04, 1.4544e-04, 1.8872e-01,  ..., 1.5163e-03,
           1.2642e-02, 2.4891e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0014e-05, 6.8545e-06, 1.7029e-01,  ..., 9.8288e-05,
           4.3678e-04, 2.1896e-02]],

         [[3.0220e-05, 1.6093e-06, 2.4927e-01,  ..., 1.2803e-04,
           9.4223e-04, 3.7909e-04]],

         [[1.7643e-05, 1.9372e-05, 2.4829e-01,  ..., 1.5986e-04,
           9.4223e-04, 8.2827e-04]],

         ...,

         [[1.5497e-05, 1.6212e-05, 1.2939e-01,  ..., 1.9646e-03,
           7.1869e-03, 3.1787e-01]],

         [[1.0014e-05, 2.1398e-05, 2.2314e-01,  ..., 2.2221e-04,
           1.1349e-03, 2.7657e-03]],

         [[6.3062e-05, 2.6524e-05, 1.6211e-01,  ..., 2.8229e-04,
           1.1272e-03, 1.3306e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1842e-05, 8.7619e-06, 1.2787e-02,  ..., 1.4636e-01,
           3.8867e-01, 1.8652e-01]],

         [[2.0564e-05, 2.3007e-05, 8.1421e-02,  ..., 4.9194e-02,
           3.0914e-02, 1.5099e-02]],

         [[1.4102e-04, 3.8314e-04, 3.0127e-01,  ..., 2.7771e-03,
           1.0662e-03, 1.4420e-03]],

         ...,

         [[5.1308e-04, 2.2292e-05, 1.3660e-01,  ..., 2.4166e-03,
           3.0041e-03, 2.0294e-02]],

         [[7.6532e-04, 1.3793e-04, 4.5837e-02,  ..., 9.9731e-02,
           2.8979e-01, 2.7856e-01]],

         [[9.4950e-05, 2.0802e-05, 1.5839e-02,  ..., 3.2745e-02,
           3.9948e-02, 4.0674e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9083e-03, 1.2045e-03, 2.3621e-01,  ..., 2.5730e-03,
           5.7220e-03, 6.3599e-02]],

         [[1.5986e-04, 2.4462e-04, 2.3853e-01,  ..., 1.8829e-02,
           2.9388e-02, 4.3091e-02]],

         [[1.5335e-03, 4.1366e-04, 2.4353e-01,  ..., 1.5327e-02,
           5.7831e-02, 6.2042e-02]],

         ...,

         [[1.6689e-04, 4.6670e-05, 2.3132e-01,  ..., 1.5495e-02,
           4.2450e-02, 5.5359e-02]],

         [[8.1921e-04, 7.5150e-04, 1.2549e-01,  ..., 9.7504e-03,
           3.8269e-02, 3.2275e-01]],

         [[1.2934e-05, 2.9266e-05, 8.3252e-02,  ..., 7.1777e-02,
           1.9055e-01, 1.4722e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.3828e-05, 1.9670e-04, 1.9255e-03,  ..., 3.0121e-02,
           6.8555e-01, 1.0510e-01]],

         [[4.6074e-05, 3.1650e-05, 1.8227e-04,  ..., 3.4389e-03,
           1.9791e-02, 9.5361e-01]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.6689e-05,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[5.9605e-06, 3.6359e-06, 4.7684e-07,  ..., 5.3835e-04,
           6.5613e-03, 9.7363e-01]],

         [[3.9635e-03, 2.2583e-03, 3.2845e-03,  ..., 5.8228e-02,
           1.1467e-02, 2.2812e-02]],

         [[3.3455e-03, 2.0294e-03, 2.6631e-04,  ..., 4.3396e-02,
           1.1792e-01, 6.5125e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8036e-05, 8.3542e-04, 2.0676e-03,  ..., 8.2581e-02,
           2.9736e-01, 2.0447e-01]],

         [[0.0000e+00, 1.7881e-07, 8.9407e-06,  ..., 1.2164e-01,
           7.8125e-01, 7.7332e-02]],

         [[1.2903e-03, 4.2305e-03, 1.1871e-02,  ..., 2.1500e-02,
           3.8910e-02, 1.2634e-02]],

         ...,

         [[1.7996e-03, 5.0850e-03, 4.0245e-03,  ..., 4.2017e-01,
           1.1487e-01, 1.3397e-02]],

         [[2.5368e-03, 1.1505e-02, 1.9806e-02,  ..., 1.9485e-02,
           1.7334e-02, 7.7820e-03]],

         [[3.4313e-03, 7.2632e-03, 6.9847e-03,  ..., 2.1069e-01,
           7.2205e-02, 5.8960e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3732e-02, 4.5547e-03, 2.3773e-02,  ..., 1.3763e-02,
           1.9867e-02, 1.3924e-02]],

         [[1.2712e-03, 4.5323e-04, 2.2850e-03,  ..., 6.0654e-03,
           3.0090e-02, 1.1421e-02]],

         [[8.6823e-03, 1.5812e-03, 2.4776e-03,  ..., 1.7197e-02,
           2.0981e-02, 1.1772e-02]],

         ...,

         [[2.0638e-03, 1.6203e-03, 8.1024e-03,  ..., 3.7750e-02,
           1.1627e-01, 5.5469e-01]],

         [[1.6632e-03, 4.2367e-04, 1.0824e-03,  ..., 3.0884e-02,
           1.0535e-01, 4.1901e-02]],

         [[8.9121e-04, 4.2748e-04, 5.3329e-03,  ..., 5.3864e-02,
           5.7910e-01, 1.5466e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4863e-01, 2.3842e-06, 2.5630e-06,  ..., 1.5545e-03,
           8.8043e-03, 2.3682e-02]],

         [[9.5215e-01, 1.0788e-05, 3.0994e-06,  ..., 4.7760e-03,
           1.7395e-02, 1.5732e-02]],

         [[9.1357e-01, 4.7684e-06, 1.6689e-06,  ..., 2.2736e-03,
           1.9012e-02, 1.8524e-02]],

         ...,

         [[6.4355e-01, 1.1837e-04, 9.6560e-04,  ..., 1.1384e-05,
           9.4299e-03, 2.2278e-02]],

         [[5.5127e-01, 1.1843e-04, 3.6526e-04,  ..., 1.6510e-02,
           5.0262e-02, 1.3741e-02]],

         [[4.8267e-01, 7.4983e-05, 2.4331e-04,  ..., 2.7130e-02,
           8.0322e-02, 9.8267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1797e-01, 2.1994e-05, 1.5497e-05,  ..., 1.8358e-04,
           2.5043e-03, 1.6205e-02]],

         [[3.0396e-01, 6.5565e-07, 5.3704e-05,  ..., 6.5327e-04,
           6.8604e-01, 7.7515e-03]],

         [[9.6045e-01, 2.4676e-05, 4.3452e-05,  ..., 3.0098e-03,
           7.5684e-03, 1.7319e-03]],

         ...,

         [[3.6060e-01, 2.2662e-04, 2.1482e-04,  ..., 1.3904e-01,
           5.1758e-02, 2.9480e-02]],

         [[2.4817e-01, 8.8120e-04, 4.2653e-04,  ..., 5.2299e-03,
           8.7128e-03, 6.1150e-03]],

         [[6.3965e-01, 6.6280e-04, 8.9741e-04,  ..., 8.4000e-03,
           2.4951e-01, 3.7903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8628e-01, 1.3542e-04, 4.2915e-05,  ..., 4.4861e-02,
           6.2622e-02, 3.9948e-02]],

         [[7.1240e-01, 4.6134e-05, 1.8358e-05,  ..., 2.1255e-02,
           1.9531e-01, 1.4816e-02]],

         [[7.9443e-01, 3.5703e-05, 2.3067e-05,  ..., 3.8635e-02,
           4.8462e-02, 1.1490e-02]],

         ...,

         [[9.5264e-01, 6.1572e-05, 6.5565e-07,  ..., 4.1723e-07,
           2.0111e-04, 3.8055e-02]],

         [[9.5312e-01, 2.1458e-04, 2.5630e-06,  ..., 0.0000e+00,
           1.1325e-06, 9.2773e-03]],

         [[9.8682e-01, 4.7028e-05, 1.1921e-07,  ..., 0.0000e+00,
           5.9605e-08, 7.8964e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3743e-01, 2.3246e-05, 4.5896e-06,  ..., 7.0068e-02,
           4.3994e-01, 2.9617e-02]],

         [[8.0664e-01, 2.3842e-06, 3.5763e-07,  ..., 1.2267e-04,
           2.3788e-02, 4.7211e-02]],

         [[8.4424e-01, 3.4511e-05, 6.7949e-06,  ..., 6.6109e-03,
           5.9509e-03, 8.2474e-03]],

         ...,

         [[7.8174e-01, 3.0780e-04, 1.7345e-05,  ..., 4.2152e-03,
           1.4999e-02, 1.2566e-02]],

         [[5.3906e-01, 3.3438e-05, 1.2159e-05,  ..., 1.3702e-02,
           2.2766e-02, 3.3081e-01]],

         [[3.0762e-01, 9.0420e-05, 6.5625e-05,  ..., 4.3335e-03,
           6.1670e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1152e-01, 3.3081e-05, 1.4484e-05,  ..., 3.0327e-03,
           1.4313e-02, 1.6113e-02]],

         [[7.8857e-01, 3.7253e-05, 1.8239e-05,  ..., 3.3379e-03,
           9.2392e-03, 6.3721e-02]],

         [[7.4805e-01, 4.0531e-06, 3.9935e-06,  ..., 1.2032e-02,
           1.5137e-01, 2.6031e-02]],

         ...,

         [[5.8447e-01, 9.3758e-05, 1.8954e-05,  ..., 8.0261e-03,
           5.8868e-02, 1.6998e-02]],

         [[8.5010e-01, 9.3162e-05, 3.1948e-05,  ..., 2.2583e-03,
           1.1856e-02, 1.3695e-02]],

         [[8.2666e-01, 1.8179e-05, 1.7762e-05,  ..., 2.6260e-02,
           7.6111e-02, 7.2021e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9971e-01, 8.9073e-04, 4.8423e-04,  ..., 5.0163e-04,
           2.7428e-03, 1.9226e-03]],

         [[9.3652e-01, 8.4639e-06, 4.9472e-05,  ..., 1.2178e-03,
           2.7657e-03, 1.1578e-03]],

         [[7.4805e-01, 3.9876e-05, 1.1504e-05,  ..., 3.4943e-03,
           5.5199e-03, 9.3002e-03]],

         ...,

         [[8.1738e-01, 1.4901e-05, 2.6822e-05,  ..., 2.9507e-03,
           1.1589e-02, 9.7656e-02]],

         [[3.1812e-01, 1.4067e-05, 7.8082e-06,  ..., 3.7903e-02,
           5.2637e-01, 4.8859e-02]],

         [[9.0625e-01, 5.0247e-05, 1.9193e-05,  ..., 9.9468e-04,
           8.1406e-03, 2.5986e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0642e-01, 4.6039e-04, 1.9045e-03,  ..., 8.6746e-03,
           1.5686e-02, 1.3145e-02]],

         [[7.0801e-01, 1.6868e-05, 4.3452e-05,  ..., 3.9612e-02,
           1.1304e-01, 3.0899e-02]],

         [[7.3242e-01, 1.5795e-05, 2.9743e-05,  ..., 9.7961e-03,
           1.9821e-02, 1.3573e-02]],

         ...,

         [[8.5400e-01, 3.5465e-05, 8.8394e-05,  ..., 1.4410e-03,
           1.6618e-04, 5.4061e-05]],

         [[6.3232e-01, 1.1241e-04, 7.0095e-05,  ..., 1.1589e-02,
           7.5500e-02, 2.3163e-02]],

         [[9.5703e-01, 8.4043e-06, 1.9133e-05,  ..., 2.6627e-03,
           6.2943e-03, 9.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1172e-01, 2.0599e-04, 1.7345e-04,  ..., 1.1921e-06,
           2.3575e-03, 1.1337e-02]],

         [[7.3779e-01, 4.1127e-06, 1.7285e-06,  ..., 5.2216e-02,
           7.1838e-02, 7.3624e-03]],

         [[7.3975e-01, 5.9009e-06, 5.7817e-06,  ..., 3.6621e-04,
           5.4047e-02, 9.6893e-03]],

         ...,

         [[9.1260e-01, 1.0157e-04, 7.3731e-05,  ..., 3.0155e-03,
           8.3313e-03, 2.5539e-03]],

         [[8.5400e-01, 6.9797e-05, 3.9160e-05,  ..., 9.0332e-03,
           1.0443e-03, 8.2254e-04]],

         [[5.6787e-01, 9.8801e-04, 3.6430e-04,  ..., 1.8034e-03,
           1.5518e-02, 6.5498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5295e-01, 4.7684e-07, 1.3113e-06,  ..., 1.7226e-04,
           3.3875e-02, 8.0469e-01]],

         [[7.9932e-01, 3.0565e-04, 5.6601e-04,  ..., 9.6262e-05,
           1.8215e-03, 1.3895e-03]],

         [[6.0693e-01, 5.9223e-04, 1.0653e-03,  ..., 4.8943e-03,
           1.7319e-02, 8.2092e-03]],

         ...,

         [[6.6260e-01, 2.3592e-04, 1.9407e-04,  ..., 1.6003e-03,
           2.3651e-03, 5.0163e-04]],

         [[8.0762e-01, 7.8619e-05, 6.5446e-05,  ..., 3.5172e-03,
           1.8799e-02, 5.6305e-03]],

         [[3.0518e-01, 2.3603e-05, 1.1224e-04,  ..., 3.2234e-03,
           1.0388e-01, 5.1709e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5098e-01, 9.8348e-06, 9.5963e-06,  ..., 1.5144e-03,
           3.4399e-01, 3.7207e-01]],

         [[7.5732e-01, 5.1320e-05, 6.3598e-05,  ..., 3.9032e-02,
           2.9861e-02, 1.3794e-02]],

         [[6.1182e-01, 2.4319e-04, 4.9233e-05,  ..., 7.1192e-04,
           4.0741e-02, 2.1347e-02]],

         ...,

         [[9.9106e-03, 7.1526e-07, 1.9073e-06,  ..., 2.5630e-05,
           3.8605e-02, 9.4629e-01]],

         [[5.2588e-01, 3.2365e-05, 4.5300e-05,  ..., 1.0786e-03,
           6.5002e-02, 1.6174e-01]],

         [[9.1748e-01, 5.3704e-05, 1.1027e-05,  ..., 5.4121e-04,
           3.5992e-03, 2.3651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-01, 3.1281e-04, 3.0804e-04,  ..., 4.0321e-03,
           1.9547e-02, 1.7914e-02]],

         [[6.8213e-01, 7.6354e-05, 8.1897e-05,  ..., 1.7847e-01,
           1.3954e-02, 4.8256e-03]],

         [[6.2158e-01, 4.2379e-05, 1.3161e-04,  ..., 3.3998e-04,
           6.0234e-03, 6.3629e-03]],

         ...,

         [[7.6416e-01, 1.0550e-05, 3.7968e-05,  ..., 1.1011e-01,
           1.0815e-01, 6.7558e-03]],

         [[5.8447e-01, 1.4627e-04, 2.3019e-04,  ..., 2.5826e-03,
           3.3498e-04, 2.5487e-04]],

         [[5.0537e-01, 8.4639e-06, 4.2975e-05,  ..., 3.4851e-02,
           3.9600e-01, 5.7220e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2520e-01, 1.1086e-05, 7.7486e-06,  ..., 1.5345e-03,
           2.7908e-02, 1.9943e-02]],

         [[6.3477e-02, 7.1907e-04, 3.4237e-04,  ..., 1.3947e-02,
           7.3433e-04, 2.6727e-04]],

         [[4.2432e-01, 6.1083e-04, 4.5133e-04,  ..., 5.7312e-02,
           1.0025e-02, 3.9043e-03]],

         ...,

         [[6.7285e-01, 1.0216e-04, 6.8009e-05,  ..., 1.1926e-01,
           2.1500e-02, 5.1765e-03]],

         [[4.2822e-01, 8.4162e-05, 1.2201e-04,  ..., 2.2980e-02,
           5.0879e-01, 1.2581e-02]],

         [[5.7178e-01, 1.6212e-04, 1.3185e-04,  ..., 6.1554e-02,
           1.7624e-02, 1.6891e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-02, 1.6689e-06, 2.3842e-06,  ..., 7.7963e-05,
           1.3718e-02, 9.3213e-01]],

         [[1.3586e-01, 1.4842e-05, 1.0133e-05,  ..., 2.7002e-01,
           2.6587e-01, 1.9714e-02]],

         [[2.3730e-01, 5.5838e-04, 5.8317e-04,  ..., 7.8583e-03,
           2.0538e-02, 9.4604e-03]],

         ...,

         [[3.8257e-01, 1.3769e-04, 5.2869e-05,  ..., 8.6365e-03,
           9.6512e-03, 2.7275e-03]],

         [[7.5000e-01, 3.0577e-05, 1.5080e-05,  ..., 3.0060e-03,
           1.8997e-02, 4.7302e-03]],

         [[7.4023e-01, 7.4804e-05, 6.7890e-05,  ..., 3.4475e-04,
           1.4061e-02, 4.4098e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9341e-01, 2.3961e-05, 2.4021e-05,  ..., 9.5825e-03,
           1.9702e-01, 8.4412e-02]],

         [[2.5269e-01, 1.0628e-04, 4.6074e-05,  ..., 4.2381e-03,
           9.3994e-02, 3.7262e-02]],

         [[6.0156e-01, 3.5524e-05, 4.9472e-05,  ..., 8.6823e-03,
           9.3323e-02, 2.3407e-02]],

         ...,

         [[7.5635e-01, 1.6093e-05, 1.2755e-05,  ..., 1.1711e-02,
           3.8391e-02, 1.7578e-02]],

         [[8.1055e-01, 2.7597e-05, 2.2411e-05,  ..., 4.8859e-02,
           2.3911e-02, 1.5434e-02]],

         [[4.4116e-01, 2.8610e-06, 2.9206e-06,  ..., 6.8054e-03,
           1.3641e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4341e-01, 7.2289e-04, 9.4128e-04,  ..., 1.7414e-03,
           8.0261e-03, 4.0550e-03]],

         [[1.1395e-01, 5.4002e-05, 4.3571e-05,  ..., 2.4338e-03,
           4.3945e-02, 1.2054e-02]],

         [[3.4717e-01, 1.2577e-05, 9.9540e-06,  ..., 1.5161e-01,
           2.7686e-01, 2.6978e-02]],

         ...,

         [[5.8447e-01, 2.2376e-04, 9.9242e-05,  ..., 7.2956e-04,
           2.5375e-02, 1.1551e-02]],

         [[9.2725e-01, 4.2617e-05, 4.6074e-05,  ..., 2.3842e-07,
           3.2425e-05, 3.4103e-03]],

         [[6.2793e-01, 1.3101e-04, 1.5771e-04,  ..., 4.2610e-03,
           4.0746e-04, 1.8585e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6924e-02, 4.6551e-05, 4.4107e-05,  ..., 8.1604e-02,
           5.5084e-03, 2.5425e-03]],

         [[1.8408e-01, 1.2398e-04, 5.9366e-05,  ..., 5.1193e-03,
           2.1469e-02, 7.3051e-03]],

         [[1.8787e-01, 2.3842e-06, 1.0729e-06,  ..., 3.2135e-02,
           4.3018e-01, 1.8494e-01]],

         ...,

         [[5.1562e-01, 7.3910e-06, 4.3511e-06,  ..., 2.3361e-02,
           5.8289e-02, 1.4061e-02]],

         [[2.7344e-01, 1.5497e-06, 1.6093e-06,  ..., 3.5248e-03,
           6.2598e-01, 5.0598e-02]],

         [[1.5588e-01, 2.5034e-06, 3.2783e-06,  ..., 2.3712e-02,
           6.1621e-01, 1.2622e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1963e-01, 1.5318e-05, 5.7817e-06,  ..., 5.1155e-03,
           4.1534e-02, 1.9928e-02]],

         [[1.3086e-01, 3.3975e-06, 4.3511e-06,  ..., 1.2683e-01,
           2.2266e-01, 1.4015e-02]],

         [[1.1737e-01, 7.3910e-06, 7.1526e-07,  ..., 2.3746e-04,
           2.5787e-02, 8.4033e-01]],

         ...,

         [[8.7952e-02, 5.3644e-05, 3.6836e-05,  ..., 1.0705e-04,
           2.8778e-02, 1.8005e-02]],

         [[3.7720e-01, 1.1748e-04, 4.8459e-05,  ..., 1.4420e-03,
           9.2239e-03, 2.1820e-03]],

         [[2.0581e-01, 1.1975e-04, 6.7472e-05,  ..., 6.6147e-03,
           1.0910e-02, 1.7605e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9372e-05, 1.0848e-05,  ..., 3.9597e-03,
           4.6051e-02, 4.5685e-02]],

         [[9.6582e-01, 1.5914e-05, 1.4305e-06,  ..., 1.1921e-07,
           1.9670e-06, 2.9335e-03]],

         [[8.2666e-01, 5.5909e-05, 6.7353e-06,  ..., 5.4598e-05,
           4.2105e-04, 8.1635e-03]],

         ...,

         [[2.1545e-01, 6.4421e-04, 1.1129e-03,  ..., 1.5574e-03,
           5.9242e-03, 3.1700e-03]],

         [[7.1094e-01, 1.0967e-05, 2.1517e-05,  ..., 1.6556e-03,
           2.5497e-02, 1.7517e-02]],

         [[2.8931e-01, 5.9700e-04, 1.6356e-03,  ..., 7.8964e-03,
           1.1391e-02, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1299e-01, 1.3280e-04, 1.2577e-04,  ..., 1.0468e-02,
           9.7580e-03, 1.4839e-03]],

         [[3.8062e-01, 1.8287e-04, 3.2854e-04,  ..., 3.0937e-03,
           1.9394e-02, 1.4420e-02]],

         [[4.3628e-01, 1.0872e-04, 9.6738e-05,  ..., 2.7776e-04,
           4.2114e-03, 4.2419e-03]],

         ...,

         [[4.3060e-02, 4.3154e-05, 6.7830e-05,  ..., 9.0256e-03,
           1.4575e-01, 3.5706e-02]],

         [[5.5267e-02, 5.4479e-05, 5.3644e-05,  ..., 1.3208e-03,
           1.5991e-01, 4.1718e-02]],

         [[4.8047e-01, 5.8830e-05, 6.8784e-05,  ..., 1.9627e-03,
           2.7969e-02, 1.1124e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0103e-01, 2.0981e-05, 2.3246e-05,  ..., 1.4839e-03,
           5.6580e-02, 2.0172e-02]],

         [[3.2324e-01, 2.2948e-04, 2.9707e-04,  ..., 3.6182e-03,
           1.8524e-02, 8.2169e-03]],

         [[3.0518e-01, 5.3525e-05, 1.1557e-04,  ..., 2.5253e-02,
           1.6467e-01, 2.8223e-01]],

         ...,

         [[5.6104e-01, 1.0937e-04, 1.6415e-04,  ..., 8.6517e-03,
           8.7402e-02, 5.8228e-02]],

         [[1.8640e-01, 3.9935e-06, 3.8862e-05,  ..., 1.9872e-04,
           4.8645e-02, 8.8074e-02]],

         [[2.5830e-01, 1.5450e-04, 2.0790e-04,  ..., 6.2084e-04,
           1.0414e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7004e-01, 7.6771e-05, 3.6061e-05,  ..., 1.2711e-02,
           1.7822e-01, 6.9275e-02]],

         [[1.0144e-01, 1.9073e-05, 3.3259e-05,  ..., 2.2293e-02,
           1.0632e-01, 1.5556e-02]],

         [[4.3652e-01, 5.3346e-05, 1.4877e-04,  ..., 1.5656e-02,
           1.0971e-02, 7.0000e-03]],

         ...,

         [[4.2017e-01, 2.3842e-07, 1.0431e-05,  ..., 3.8483e-02,
           5.1465e-01, 1.7349e-02]],

         [[3.1714e-01, 1.0133e-06, 8.6427e-06,  ..., 7.6538e-02,
           3.2715e-01, 1.6159e-02]],

         [[1.1072e-01, 1.7881e-07, 3.5167e-06,  ..., 3.3752e-02,
           8.3105e-01, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0117e-02, 9.5367e-07, 5.4836e-06,  ..., 9.3750e-02,
           9.8206e-02, 1.7075e-02]],

         [[2.0279e-02, 5.9605e-07, 1.0729e-06,  ..., 1.3185e-04,
           1.0681e-02, 9.5410e-01]],

         [[2.5558e-02, 2.1040e-05, 7.7009e-05,  ..., 1.8402e-02,
           9.5520e-03, 4.3030e-03]],

         ...,

         [[6.4575e-02, 6.4492e-05, 1.4126e-04,  ..., 4.4556e-02,
           3.1097e-02, 7.6523e-03]],

         [[1.0632e-01, 1.8477e-06, 2.8610e-06,  ..., 1.2476e-01,
           6.1182e-01, 4.5349e-02]],

         [[9.7351e-02, 4.8876e-06, 1.4603e-05,  ..., 1.6956e-01,
           6.0596e-01, 8.9722e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 1.1361e-04, 6.2561e-04,  ..., 2.5635e-03,
           1.0216e-02, 1.5457e-02]],

         [[9.7266e-01, 6.3717e-05, 2.7299e-05,  ..., 5.9664e-05,
           4.8208e-04, 1.6117e-03]],

         [[9.0332e-01, 1.2636e-05, 4.4703e-06,  ..., 4.2975e-05,
           3.7861e-04, 8.5220e-03]],

         ...,

         [[5.4834e-01, 5.0664e-06, 6.2585e-06,  ..., 1.7090e-02,
           1.6211e-01, 1.8958e-01]],

         [[4.3311e-01, 1.7881e-06, 7.1526e-07,  ..., 3.8576e-04,
           8.5754e-03, 4.8103e-03]],

         [[1.1932e-01, 2.8014e-06, 3.8743e-06,  ..., 1.3895e-03,
           7.0129e-02, 7.4268e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6627e-02, 4.1723e-07, 1.3709e-06,  ..., 1.5686e-01,
           3.2446e-01, 5.6366e-02]],

         [[2.3242e-01, 1.7881e-06, 1.2517e-06,  ..., 2.4979e-02,
           4.8462e-01, 2.0044e-01]],

         [[1.0431e-01, 9.2030e-05, 9.2089e-05,  ..., 2.0615e-02,
           1.5967e-01, 3.7628e-02]],

         ...,

         [[1.8652e-01, 1.4901e-06, 9.9540e-06,  ..., 7.5928e-02,
           3.3765e-01, 2.0959e-01]],

         [[5.7471e-01, 1.4484e-05, 4.3690e-05,  ..., 1.9196e-02,
           1.0773e-02, 2.9510e-02]],

         [[6.7236e-01, 5.0068e-06, 1.5736e-05,  ..., 9.5901e-03,
           2.4689e-02, 3.3997e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3804e-02, 1.9670e-06, 1.3709e-06,  ..., 7.3738e-03,
           5.7526e-02, 3.0090e-02]],

         [[1.6809e-01, 2.0206e-05, 1.0133e-05,  ..., 2.7013e-04,
           4.8828e-03, 4.3411e-03]],

         [[2.6196e-01, 2.3484e-05, 7.7248e-05,  ..., 3.4828e-03,
           1.0669e-01, 6.6772e-02]],

         ...,

         [[3.9307e-02, 4.2915e-06, 6.4433e-05,  ..., 2.7451e-02,
           2.2049e-02, 1.0017e-02]],

         [[1.1084e-01, 1.2279e-05, 6.9439e-05,  ..., 2.3224e-02,
           9.5367e-03, 6.2523e-03]],

         [[2.7783e-01, 2.2292e-05, 3.5107e-05,  ..., 2.5708e-01,
           1.6077e-01, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0654e-01, 2.3842e-07, 1.0133e-06,  ..., 4.2145e-02,
           6.1426e-01, 1.1774e-01]],

         [[2.8125e-01, 4.0531e-06, 2.8253e-05,  ..., 9.3506e-02,
           2.8027e-01, 1.0675e-01]],

         [[1.9336e-01, 5.3644e-07, 1.0014e-05,  ..., 8.2520e-02,
           6.5918e-01, 4.7211e-02]],

         ...,

         [[4.7607e-01, 3.8505e-05, 4.7386e-05,  ..., 3.9291e-03,
           4.7569e-03, 6.9504e-03]],

         [[7.1594e-02, 2.6226e-06, 3.0398e-06,  ..., 5.7983e-02,
           3.5254e-01, 9.4849e-02]],

         [[4.5715e-02, 2.1756e-05, 6.3419e-05,  ..., 3.0746e-03,
           1.5222e-01, 5.9277e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4751e-01, 3.0220e-05, 9.2506e-05,  ..., 4.3030e-03,
           1.4868e-01, 9.6802e-02]],

         [[9.8083e-02, 9.1195e-06, 3.1531e-05,  ..., 1.0854e-04,
           2.0847e-03, 1.7281e-03]],

         [[2.1118e-01, 8.2970e-05, 1.4770e-04,  ..., 1.2217e-03,
           6.2927e-02, 1.9562e-02]],

         ...,

         [[3.3478e-02, 5.5730e-05, 5.3585e-05,  ..., 2.2659e-02,
           7.5000e-01, 1.5967e-01]],

         [[3.1250e-01, 1.1075e-04, 7.6246e-04,  ..., 2.9984e-02,
           1.0468e-01, 4.2041e-01]],

         [[4.1968e-01, 2.0742e-05, 4.1306e-05,  ..., 2.1225e-02,
           2.6685e-01, 2.5464e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 3.2723e-05, 1.6534e-04,  ..., 3.4943e-02,
           6.7627e-02, 7.1678e-03]],

         [[6.6223e-02, 1.8477e-06, 1.6391e-05,  ..., 1.3684e-01,
           1.5454e-01, 3.2013e-02]],

         [[4.0894e-01, 1.5008e-04, 7.2241e-04,  ..., 3.7964e-02,
           6.0059e-02, 1.5854e-02]],

         ...,

         [[3.0029e-01, 6.7949e-05, 1.5736e-04,  ..., 1.1945e-01,
           6.2675e-03, 6.9389e-03]],

         [[9.0771e-01, 5.9605e-06, 4.9257e-04,  ..., 1.1718e-04,
           1.3447e-03, 1.3260e-02]],

         [[5.1270e-01, 7.3314e-06, 2.5749e-05,  ..., 3.3417e-02,
           2.6031e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5809e-05, 1.2994e-05, 1.0345e-01,  ..., 1.3412e-02,
           1.6891e-02, 1.3840e-02]],

         [[3.0220e-05, 3.8564e-05, 1.7725e-01,  ..., 2.0920e-02,
           1.0040e-02, 9.0332e-03]],

         [[1.4687e-04, 1.0639e-04, 9.6802e-02,  ..., 1.6373e-02,
           8.4412e-02, 1.2000e-01]],

         ...,

         [[7.0572e-05, 1.2100e-05, 1.9080e-01,  ..., 9.0149e-02,
           2.6294e-01, 1.2225e-01]],

         [[7.9536e-04, 1.0157e-03, 2.0764e-01,  ..., 1.2085e-02,
           3.7384e-02, 2.7451e-02]],

         [[2.5558e-04, 6.8247e-05, 2.7588e-01,  ..., 1.1997e-03,
           7.9498e-03, 2.8572e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7241e-04, 1.4079e-04, 3.2812e-01,  ..., 7.5245e-04,
           1.8187e-03, 2.1763e-03]],

         [[2.9707e-04, 1.5628e-04, 2.1533e-01,  ..., 7.6771e-04,
           1.1129e-03, 1.3371e-03]],

         [[6.8545e-05, 1.6689e-05, 2.9272e-01,  ..., 8.5831e-04,
           1.6761e-04, 2.5616e-03]],

         ...,

         [[6.2752e-04, 4.8137e-04, 2.8687e-01,  ..., 1.5240e-03,
           1.7328e-03, 3.1447e-04]],

         [[6.0201e-06, 1.0133e-05, 2.3059e-01,  ..., 9.0981e-04,
           2.0103e-03, 1.8816e-03]],

         [[6.0320e-05, 2.1601e-04, 1.5149e-01,  ..., 1.3130e-02,
           2.6875e-03, 6.2609e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0562e-04, 4.8518e-05, 1.6785e-01,  ..., 4.8399e-04,
           1.5976e-02, 1.3237e-03]],

         [[2.4986e-04, 8.2254e-06, 2.7295e-01,  ..., 8.4829e-04,
           1.1712e-04, 5.0259e-04]],

         [[1.9932e-04, 1.4389e-04, 2.8394e-01,  ..., 1.0166e-03,
           4.0126e-04, 8.7976e-04]],

         ...,

         [[3.6657e-05, 2.2650e-05, 5.8105e-02,  ..., 6.3438e-03,
           4.7534e-01, 8.8623e-02]],

         [[6.9737e-05, 1.1301e-04, 2.4500e-01,  ..., 9.6083e-04,
           2.1877e-03, 2.1496e-03]],

         [[4.4370e-04, 1.1724e-04, 1.7383e-01,  ..., 1.4019e-03,
           1.0239e-02, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3411e-05, 7.7486e-07, 1.0891e-03,  ..., 1.7480e-01,
           7.1338e-01, 1.9608e-02]],

         [[8.5711e-05, 8.2552e-05, 9.7717e-02,  ..., 5.2521e-02,
           1.3863e-02, 6.4850e-03]],

         [[5.3835e-04, 1.5316e-03, 2.3499e-01,  ..., 1.1387e-03,
           1.8806e-03, 1.0691e-03]],

         ...,

         [[1.1826e-03, 3.0756e-05, 7.2205e-02,  ..., 2.3438e-02,
           1.1353e-01, 2.7069e-02]],

         [[4.6659e-04, 6.1154e-05, 2.4948e-02,  ..., 2.4524e-01,
           4.4751e-01, 7.5378e-02]],

         [[1.0252e-05, 3.6359e-06, 1.9779e-03,  ..., 1.1279e-01,
           3.6987e-01, 2.2253e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1656e-03, 1.1091e-03, 2.2510e-01,  ..., 4.6577e-03,
           5.0018e-02, 6.8909e-02]],

         [[3.7622e-04, 6.9475e-04, 2.5879e-01,  ..., 9.1324e-03,
           1.3390e-02, 2.2430e-02]],

         [[1.7729e-03, 1.1063e-03, 2.5342e-01,  ..., 2.7328e-02,
           4.1046e-02, 1.7731e-02]],

         ...,

         [[2.1803e-04, 6.3419e-05, 1.9666e-01,  ..., 7.1777e-02,
           1.4160e-01, 5.9052e-02]],

         [[2.1100e-04, 1.6952e-04, 4.6265e-02,  ..., 2.0218e-02,
           1.8591e-01, 4.8975e-01]],

         [[5.9009e-06, 5.3644e-06, 1.8494e-02,  ..., 1.8103e-01,
           4.1772e-01, 1.3147e-01]]]], device='cuda:0', dtype=torch.float16))...]

For each of the so many tokens generated, I can access the attention state of each of the 28 layers in the model.

(#36) [tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4316e-01, 2.5684e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6530e-02, 5.0439e-01, 4.5923e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7796e-03, 1.6203e-03, 6.6109e-03,  ..., 1.9934e-01,
           0.0000e+00, 0.0000e+00],
          [1.4343e-03, 3.3360e-03, 1.1284e-02,  ..., 3.9771e-01,
           1.0059e-01, 0.0000e+00],
          [9.5129e-05, 2.2335e-03, 7.0953e-03,  ..., 4.6265e-02,
           1.7200e-01, 5.8154e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9141e-04, 2.5940e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8413e-06, 2.3186e-05, 8.0943e-05,  ..., 9.5459e-01,
           0.0000e+00, 0.0000e+00],
          [7.1168e-05, 2.8157e-04, 8.2245e-03,  ..., 5.3650e-02,
           5.0879e-01, 0.0000e+00],
          [1.0729e-06, 1.6928e-05, 6.3515e-04,  ..., 6.7616e-04,
           2.8763e-02, 9.5215e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1124e-04, 9.9902e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5359e-05, 4.3058e-04, 9.9951e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0000e+00,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0986e-02,
           9.8877e-01, 0.0000e+00],
          [6.4969e-06, 2.2650e-05, 4.0985e-02,  ..., 2.2650e-05,
           2.7609e-04, 4.9927e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9312e-02, 9.7070e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0651e-01, 1.9897e-01, 6.9434e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7684e-07, 6.0797e-06, 6.0797e-06,  ..., 9.9268e-01,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 8.3447e-07, 4.5121e-05,  ..., 4.0588e-03,
           9.9316e-01, 0.0000e+00],
          [8.1024e-03, 9.1858e-03, 1.9440e-02,  ..., 1.1795e-02,
           1.7151e-02, 7.6904e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5915e-02, 5.9143e-02, 9.2480e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0719e-02, 9.4604e-03, 1.0071e-02,  ..., 2.7374e-02,
           0.0000e+00, 0.0000e+00],
          [1.1459e-02, 1.0765e-02, 1.0109e-02,  ..., 7.8735e-03,
           1.2985e-02, 0.0000e+00],
          [2.3193e-03, 6.3057e-03, 4.6600e-02,  ..., 2.6283e-03,
           1.9424e-02, 1.5283e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3018e-01, 5.6982e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.5571e-01, 5.8643e-01, 5.8044e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0340e-02, 1.4427e-02, 6.8130e-03,  ..., 2.6947e-02,
           0.0000e+00, 0.0000e+00],
          [2.4475e-02, 1.3947e-02, 4.3221e-03,  ..., 3.9734e-02,
           3.6194e-02, 0.0000e+00],
          [8.5526e-03, 1.2444e-02, 1.3962e-03,  ..., 2.0508e-02,
           4.6234e-02, 3.3493e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 1.9934e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7510e-01, 2.6245e-01, 2.6245e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9380e-04, 1.0414e-03, 7.8125e-03,  ..., 5.6543e-01,
           0.0000e+00, 0.0000e+00],
          [7.8392e-04, 2.8667e-03, 1.6495e-02,  ..., 2.5806e-01,
           4.2554e-01, 0.0000e+00],
          [1.2255e-03, 3.5458e-03, 9.3384e-03,  ..., 4.4556e-02,
           1.1743e-01, 1.4380e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6387e-01, 3.6224e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0004e-01, 6.5234e-01, 2.4756e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0564e-05, 1.6093e-06, 5.9605e-08,  ..., 4.1504e-02,
           0.0000e+00, 0.0000e+00],
          [3.3140e-05, 7.4685e-05, 1.6928e-05,  ..., 7.0752e-01,
           1.8457e-01, 0.0000e+00],
          [2.0862e-06, 7.0572e-05, 1.6165e-04,  ..., 5.5756e-02,
           7.6953e-01, 1.6650e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4180e-01, 1.5820e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.5615e-01, 3.3740e-01, 1.0620e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0942e-03, 1.4763e-02, 3.6530e-02,  ..., 3.1250e-02,
           0.0000e+00, 0.0000e+00],
          [7.3433e-03, 7.5760e-03, 4.1595e-02,  ..., 1.4978e-01,
           2.3206e-01, 0.0000e+00],
          [1.1871e-02, 3.0792e-02, 3.6011e-02,  ..., 3.2776e-02,
           1.4917e-01, 6.9397e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 2.5830e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4263e-01, 2.6025e-01, 2.9712e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1372e-02, 2.9022e-02, 4.3579e-02,  ..., 9.5154e-02,
           0.0000e+00, 0.0000e+00],
          [8.6288e-03, 2.0523e-02, 4.0192e-02,  ..., 1.2769e-01,
           2.3315e-01, 0.0000e+00],
          [1.6680e-03, 7.0229e-03, 1.2039e-02,  ..., 5.1483e-02,
           1.5137e-01, 8.4229e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4697e-01, 3.5303e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9893e-01, 1.9434e-01, 4.0674e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.4289e-03, 1.4206e-02, 1.0095e-01,  ..., 6.1493e-03,
           0.0000e+00, 0.0000e+00],
          [1.7691e-03, 6.5956e-03, 5.2917e-02,  ..., 2.7451e-02,
           1.0944e-01, 0.0000e+00],
          [7.9880e-03, 1.3489e-02, 3.9795e-02,  ..., 2.8549e-02,
           7.0679e-02, 1.5076e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9160e-01, 1.0822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6890e-01, 4.6997e-01, 1.6113e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1887e-02, 3.3600e-02, 1.8692e-02,  ..., 3.4204e-01,
           0.0000e+00, 0.0000e+00],
          [4.3449e-03, 2.4033e-02, 2.5391e-02,  ..., 6.9580e-02,
           4.0649e-01, 0.0000e+00],
          [8.3590e-04, 7.3929e-03, 1.7977e-03,  ..., 1.6431e-01,
           2.7515e-01, 2.1606e-01]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1826e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2510e-01, 2.0776e-01, 6.6956e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0966e-02, 1.4944e-03, 4.0627e-03,  ..., 7.6660e-02,
           0.0000e+00, 0.0000e+00],
          [2.2415e-02, 4.0817e-03, 1.0422e-02,  ..., 4.7455e-02,
           1.1298e-01, 0.0000e+00],
          [1.7786e-04, 7.8321e-05, 1.6844e-04,  ..., 1.3831e-01,
           6.2012e-01, 1.2207e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5146e-01, 2.4866e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5581e-01, 2.3828e-01, 3.0591e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9973e-03, 5.5733e-03, 8.0795e-03,  ..., 2.6321e-03,
           0.0000e+00, 0.0000e+00],
          [1.1795e-02, 5.0926e-03, 1.1040e-02,  ..., 2.2392e-03,
           1.3770e-01, 0.0000e+00],
          [7.9575e-03, 5.2261e-03, 7.1182e-03,  ..., 3.3321e-03,
           1.5173e-01, 4.3030e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6809e-01, 8.3203e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.2864e-01, 2.3962e-01, 5.3174e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9897e-02, 1.0612e-02, 1.5198e-02,  ..., 1.4935e-03,
           0.0000e+00, 0.0000e+00],
          [1.3489e-02, 7.1106e-03, 7.7782e-03,  ..., 3.8643e-03,
           8.2474e-03, 0.0000e+00],
          [3.9291e-03, 1.2903e-03, 1.0471e-03,  ..., 3.5706e-02,
           3.1891e-02, 1.0413e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1143e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0248e-02, 9.2383e-01, 5.5908e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6413e-02, 7.4339e-04, 2.4700e-04,  ..., 4.6460e-01,
           0.0000e+00, 0.0000e+00],
          [4.2763e-03, 1.2445e-03, 1.4544e-03,  ..., 5.0195e-01,
           1.7480e-01, 0.0000e+00],
          [4.2200e-05, 5.6505e-04, 2.3232e-03,  ..., 3.0609e-02,
           7.8955e-01, 1.3928e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1650e-01, 8.3618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2314e-01, 1.6394e-01, 1.1267e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6040e-02, 1.0773e-02, 4.5380e-02,  ..., 3.8818e-02,
           0.0000e+00, 0.0000e+00],
          [3.5645e-02, 7.0724e-03, 1.8051e-02,  ..., 2.4292e-02,
           1.2146e-01, 0.0000e+00],
          [1.6937e-02, 2.5177e-03, 1.3718e-02,  ..., 2.4078e-02,
           3.9062e-02, 1.7932e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7990e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3267e-01, 1.0986e-01, 6.5771e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6642e-02, 1.5879e-03, 3.3360e-03,  ..., 1.8213e-01,
           0.0000e+00, 0.0000e+00],
          [1.7288e-02, 4.7607e-03, 1.5732e-02,  ..., 2.2241e-01,
           2.8101e-01, 0.0000e+00],
          [1.3023e-02, 1.8110e-03, 1.0712e-02,  ..., 4.9530e-02,
           1.0162e-01, 4.6631e-01]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6182e-01, 1.3831e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6216e-01, 1.4137e-02, 5.2393e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1128e-01, 2.6108e-02, 1.1072e-03,  ..., 3.9154e-02,
           0.0000e+00, 0.0000e+00],
          [2.5928e-01, 1.8272e-03, 1.2445e-01,  ..., 3.6001e-05,
           7.5745e-02, 0.0000e+00],
          [7.4072e-01, 4.9448e-04, 2.7351e-03,  ..., 3.5667e-03,
           3.2684e-02, 1.5419e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 2.3327e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 7.9956e-03, 1.2802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 1.0800e-04, 1.5259e-05,  ..., 6.1941e-04,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.0908e-05, 4.1056e-04,  ..., 1.1955e-02,
           2.4429e-02, 0.0000e+00],
          [8.3789e-01, 4.9233e-05, 1.1092e-04,  ..., 1.0651e-02,
           3.4058e-02, 9.6970e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.7275e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.5106e-02, 1.5381e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6846e-01, 1.4937e-04, 7.3969e-05,  ..., 5.6946e-02,
           0.0000e+00, 0.0000e+00],
          [7.7881e-01, 1.3530e-04, 2.4116e-04,  ..., 6.4564e-04,
           1.2840e-02, 0.0000e+00],
          [8.0615e-01, 8.6963e-05, 9.3079e-04,  ..., 1.5926e-04,
           6.0120e-03, 2.1149e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 1.0138e-03, 6.6071e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1885e-01, 4.9591e-03, 1.0330e-02,  ..., 7.1167e-02,
           0.0000e+00, 0.0000e+00],
          [6.8213e-01, 6.4316e-03, 6.5575e-03,  ..., 1.2457e-05,
           4.9408e-02, 0.0000e+00],
          [8.4082e-01, 1.0557e-03, 4.4098e-03,  ..., 4.2140e-05,
           6.8426e-04, 5.7983e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8291e-01, 1.7242e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 1.9501e-02, 2.6443e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0479e-01, 2.0683e-04, 9.8610e-04,  ..., 9.0456e-04,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.7376e-03, 2.8324e-03,  ..., 1.9646e-03,
           2.3438e-02, 0.0000e+00],
          [5.4688e-01, 3.4404e-04, 6.7139e-04,  ..., 2.3544e-04,
           1.5030e-02, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9586e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2334e-01, 2.8275e-02, 4.8370e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3701e-01, 9.6679e-05, 3.9649e-04,  ..., 6.2752e-04,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 1.2379e-03, 3.8052e-03,  ..., 6.2323e-04,
           3.2104e-02, 0.0000e+00],
          [7.5684e-01, 3.4761e-04, 1.7176e-03,  ..., 2.4796e-04,
           6.6757e-03, 2.1469e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1235e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 3.9482e-03, 4.6921e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6187e-01, 1.0431e-05, 2.2113e-05,  ..., 6.0081e-03,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 6.3539e-05, 5.8460e-04,  ..., 5.2452e-06,
           2.0981e-02, 0.0000e+00],
          [6.6162e-01, 1.7941e-05, 1.0166e-03,  ..., 4.7922e-04,
           1.4820e-03, 1.6098e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.0712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3374e-02, 8.1635e-03, 3.5715e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9541e-01, 5.9605e-06, 5.9605e-08,  ..., 2.6875e-03,
           0.0000e+00, 0.0000e+00],
          [9.2822e-01, 2.3007e-04, 1.2040e-05,  ..., 1.7654e-02,
           1.9470e-02, 0.0000e+00],
          [8.7891e-01, 2.4915e-05, 9.0420e-05,  ..., 3.6907e-03,
           1.0187e-01, 1.4641e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 4.0833e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2920e-01, 4.8859e-02, 2.1820e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3018e-01, 2.4247e-04, 5.3704e-05,  ..., 7.7972e-03,
           0.0000e+00, 0.0000e+00],
          [9.0625e-01, 1.1749e-03, 1.2217e-03,  ..., 3.5248e-02,
           1.9638e-02, 0.0000e+00],
          [9.0479e-01, 4.1747e-04, 2.6131e-04,  ..., 5.4359e-03,
           9.6512e-03, 1.0956e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 5.4993e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.1450e-01, 3.4241e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7510e-01, 3.3474e-04, 9.2220e-04,  ..., 3.5706e-03,
           0.0000e+00, 0.0000e+00],
          [7.4512e-01, 4.6825e-04, 6.5136e-04,  ..., 9.2621e-03,
           3.0869e-02, 0.0000e+00],
          [6.6162e-01, 2.3687e-04, 2.0623e-04,  ..., 4.4823e-03,
           3.7903e-02, 8.8806e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7028e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 3.2837e-02, 3.6957e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3994e-01, 5.5046e-03, 1.2650e-02,  ..., 5.8708e-03,
           0.0000e+00, 0.0000e+00],
          [6.9141e-01, 1.0643e-02, 1.0078e-02,  ..., 7.0496e-03,
           3.6865e-02, 0.0000e+00],
          [4.1699e-01, 5.6419e-03, 5.1384e-03,  ..., 5.6915e-03,
           5.7648e-02, 2.2400e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4824e-01, 5.1849e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0127e-01, 1.3257e-01, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.2445e-04, 1.0324e-04,  ..., 4.9927e-02,
           0.0000e+00, 0.0000e+00],
          [6.1768e-01, 2.5797e-04, 2.5535e-04,  ..., 9.6985e-02,
           9.6191e-02, 0.0000e+00],
          [5.4834e-01, 8.7798e-05, 1.6689e-04,  ..., 4.7272e-02,
           1.2732e-01, 1.2976e-01]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3588e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 2.1469e-02, 3.3905e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.4255e-01, 9.2840e-04, 7.3147e-04,  ..., 2.2919e-02,
           0.0000e+00, 0.0000e+00],
          [3.0200e-01, 9.4652e-04, 1.0891e-03,  ..., 1.3756e-02,
           7.5684e-02, 0.0000e+00],
          [4.0796e-01, 1.8358e-03, 7.1335e-04,  ..., 1.2772e-02,
           7.2998e-02, 4.8004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 2.3392e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2568e-01, 1.6895e-01, 5.1270e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1123e-01, 5.3692e-04, 4.4227e-05,  ..., 4.5929e-02,
           0.0000e+00, 0.0000e+00],
          [3.7915e-01, 7.6771e-04, 4.3631e-05,  ..., 4.4525e-02,
           2.7710e-01, 0.0000e+00],
          [2.9761e-01, 1.4448e-04, 4.3511e-06,  ..., 6.6414e-03,
           4.4897e-01, 5.8441e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 3.1860e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 2.5665e-02, 1.9028e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8564e-01, 3.8195e-04, 1.2207e-04,  ..., 2.9812e-03,
           0.0000e+00, 0.0000e+00],
          [7.7637e-01, 1.3866e-03, 3.2234e-03,  ..., 6.1989e-03,
           1.0529e-02, 0.0000e+00],
          [7.8125e-01, 2.8181e-04, 6.3896e-05,  ..., 4.9210e-03,
           2.2171e-02, 3.0945e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2604e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 1.1498e-04, 3.6865e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9111e-01, 7.3552e-05, 1.1325e-06,  ..., 1.0364e-01,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 4.2260e-05, 1.4305e-06,  ..., 5.8949e-05,
           2.2446e-02, 0.0000e+00],
          [9.4385e-01, 1.8537e-05, 3.2544e-05,  ..., 1.0258e-04,
           3.6073e-04, 5.2582e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4014e-01, 1.1917e-02, 0.0000e+00,  ..., 6.0997e-03,
           6.0997e-03, 6.0997e-03],
          [2.8540e-01, 1.5497e-06, 7.4158e-03,  ..., 7.4425e-03,
           7.4425e-03, 0.0000e+00],
          ...,
          [9.3018e-01, 1.3933e-03, 3.7253e-05,  ..., 7.2021e-03,
           0.0000e+00, 0.0000e+00],
          [9.8779e-01, 2.1183e-04, 9.8348e-06,  ..., 4.1723e-07,
           6.9733e-03, 0.0000e+00],
          [9.8096e-01, 5.4538e-05, 4.5514e-04,  ..., 2.3842e-07,
           1.1921e-05, 1.3779e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-02, 4.1628e-04, 0.0000e+00,  ..., 4.4823e-03,
           0.0000e+00, 4.4823e-03],
          [1.7053e-01, 2.3842e-07, 8.7404e-04,  ..., 5.7526e-03,
           5.7526e-03, 0.0000e+00],
          ...,
          [9.9365e-01, 7.3791e-05, 5.3644e-07,  ..., 2.0618e-03,
           0.0000e+00, 0.0000e+00],
          [9.9512e-01, 2.8074e-05, 6.5565e-07,  ..., 4.1723e-07,
           3.9139e-03, 0.0000e+00],
          [9.9805e-01, 4.8876e-06, 7.9274e-06,  ..., 5.9605e-08,
           7.7486e-07, 1.5945e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7914e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0615e-01, 1.9080e-01, 3.0651e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5425e-01, 2.3782e-05, 1.3769e-05,  ..., 4.8859e-02,
           0.0000e+00, 0.0000e+00],
          [5.7080e-01, 1.1081e-04, 9.5487e-05,  ..., 8.8989e-02,
           9.0332e-02, 0.0000e+00],
          [7.5342e-01, 3.3808e-04, 2.2292e-05,  ..., 2.7664e-02,
           1.3538e-01, 1.3496e-02]],

         [[5.9277e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5662e-01, 3.0994e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.4082e-01, 6.5422e-04, 6.8045e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6719e-01, 1.0473e-04, 8.1539e-05,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [7.0752e-01, 3.7422e-03, 7.0286e-04,  ..., 1.1086e-05,
           3.4149e-02, 0.0000e+00],
          [9.2480e-01, 1.4150e-04, 4.0627e-04,  ..., 1.2755e-05,
           9.4843e-04, 1.4572e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 9.0027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.7802e-02, 4.3907e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7612e-01, 4.2038e-03, 8.9455e-04,  ..., 1.2268e-01,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 1.8911e-03, 6.9571e-04,  ..., 6.1531e-03,
           1.8356e-02, 0.0000e+00],
          [8.2227e-01, 1.4400e-03, 2.4021e-04,  ..., 1.5097e-03,
           1.7242e-02, 1.7166e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5106e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0869e-01, 8.5022e-02, 6.3057e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1162e-01, 7.6151e-04, 2.4056e-04,  ..., 6.1951e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 4.3793e-03, 5.4407e-04,  ..., 4.9324e-03,
           2.0966e-02, 0.0000e+00],
          [8.6230e-01, 1.1921e-03, 3.2592e-04,  ..., 4.8943e-03,
           1.5625e-02, 2.9266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.7042e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1309e-01, 3.2806e-02, 5.4230e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6865e-01, 2.5034e-05, 5.9307e-05,  ..., 6.3599e-02,
           0.0000e+00, 0.0000e+00],
          [8.2520e-01, 4.2796e-04, 8.6975e-04,  ..., 1.3519e-02,
           9.7107e-02, 0.0000e+00],
          [8.7354e-01, 5.1320e-05, 1.1891e-04,  ..., 3.2444e-03,
           1.7273e-02, 4.6082e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9471e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2021e-01, 2.6611e-01, 1.3817e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8770e-01, 1.6451e-04, 4.8816e-05,  ..., 7.7698e-02,
           0.0000e+00, 0.0000e+00],
          [8.7500e-01, 3.3140e-04, 1.4877e-04,  ..., 3.1910e-03,
           7.4463e-02, 0.0000e+00],
          [8.7830e-02, 3.7050e-04, 4.4584e-05,  ..., 3.6030e-03,
           8.9355e-01, 3.5515e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9805e-01, 1.9121e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 8.2703e-03, 7.5302e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4893e-01, 5.5313e-05, 1.3638e-04,  ..., 4.8523e-02,
           0.0000e+00, 0.0000e+00],
          [5.1904e-01, 2.7776e-04, 1.0710e-03,  ..., 2.8934e-03,
           1.1650e-02, 0.0000e+00],
          [6.9629e-01, 1.5771e-04, 2.1303e-04,  ..., 1.9398e-03,
           7.5378e-03, 2.8076e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5125e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6914e-01, 7.2327e-02, 5.8502e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0566e-01, 5.9414e-04, 1.3161e-03,  ..., 6.1005e-02,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 7.0047e-04, 2.0945e-04,  ..., 1.6546e-03,
           2.1942e-02, 0.0000e+00],
          [5.6152e-01, 7.2432e-04, 1.9264e-04,  ..., 4.8180e-03,
           3.0151e-02, 6.9458e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 1.0078e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 2.3755e-01, 2.3361e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 2.7001e-05, 5.4359e-05,  ..., 3.2330e-03,
           0.0000e+00, 0.0000e+00],
          [9.5361e-01, 7.1645e-05, 4.4823e-05,  ..., 2.8782e-03,
           9.6970e-03, 0.0000e+00],
          [9.0820e-01, 1.0967e-04, 1.7285e-05,  ..., 3.7174e-03,
           4.5105e-02, 1.1147e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.3030e-02, 6.8016e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5293e-01, 1.0973e-04, 2.9743e-05,  ..., 2.5511e-04,
           0.0000e+00, 0.0000e+00],
          [6.3574e-01, 1.1864e-03, 6.1083e-04,  ..., 1.9722e-03,
           2.6718e-02, 0.0000e+00],
          [7.1436e-01, 3.7956e-04, 1.0496e-04,  ..., 1.6937e-03,
           1.3695e-02, 1.4694e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9932e-01, 1.9336e-01, 7.4959e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1406e-01, 6.3467e-04, 2.1601e-04,  ..., 4.5013e-04,
           0.0000e+00, 0.0000e+00],
          [5.8789e-01, 3.4122e-03, 4.3225e-04,  ..., 7.0457e-03,
           2.9953e-02, 0.0000e+00],
          [4.8145e-01, 5.6763e-03, 2.9516e-04,  ..., 6.5231e-03,
           4.9438e-02, 2.0691e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-01, 1.0638e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9126e-01, 6.4746e-01, 6.1310e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8909e-01, 5.7077e-04, 1.4858e-03,  ..., 2.2385e-02,
           0.0000e+00, 0.0000e+00],
          [3.4790e-01, 3.4618e-03, 1.7891e-03,  ..., 9.7809e-03,
           7.8735e-02, 0.0000e+00],
          [6.1328e-01, 1.4267e-03, 8.6164e-04,  ..., 1.4519e-02,
           7.0190e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6172e-01, 2.3853e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 1.8054e-01, 8.2153e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9775e-01, 8.5211e-04, 6.9284e-04,  ..., 2.2202e-03,
           0.0000e+00, 0.0000e+00],
          [3.3130e-01, 6.7482e-03, 4.0779e-03,  ..., 5.2023e-04,
           1.5572e-02, 0.0000e+00],
          [5.9082e-01, 1.8854e-03, 1.0052e-03,  ..., 5.8413e-04,
           7.5760e-03, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.7449e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1602e-01, 7.2327e-02, 1.1589e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7646e-01, 2.4140e-05, 1.0002e-04,  ..., 1.0023e-03,
           0.0000e+00, 0.0000e+00],
          [1.2634e-02, 5.4240e-06, 1.3173e-05,  ..., 1.8167e-04,
           1.1021e-04, 0.0000e+00],
          [1.3664e-02, 2.1577e-05, 1.5199e-05,  ..., 2.6774e-04,
           2.7370e-04, 5.4550e-04]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 1.7456e-02, 7.4501e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 5.0962e-05, 2.0564e-05,  ..., 1.6670e-03,
           0.0000e+00, 0.0000e+00],
          [8.4082e-01, 3.1257e-04, 1.5771e-04,  ..., 2.6684e-03,
           8.6670e-03, 0.0000e+00],
          [9.1406e-01, 6.3419e-05, 1.3173e-05,  ..., 1.8902e-03,
           2.7943e-03, 3.4904e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7852e-01, 2.1271e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 9.6054e-03, 1.4503e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0176e-01, 5.0783e-04, 9.0170e-04,  ..., 2.2461e-01,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 3.3069e-04, 5.3549e-04,  ..., 4.0321e-03,
           1.2891e-01, 0.0000e+00],
          [2.4939e-01, 2.5196e-03, 7.6828e-03,  ..., 6.1321e-04,
           6.0539e-03, 1.6101e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.0403e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7461e-01, 2.1423e-02, 3.8376e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3291e-01, 2.2054e-05, 2.7478e-05,  ..., 1.3718e-02,
           0.0000e+00, 0.0000e+00],
          [1.6650e-01, 9.7096e-05, 1.0788e-04,  ..., 2.1782e-03,
           1.8740e-03, 0.0000e+00],
          [9.0759e-02, 4.7922e-05, 6.3717e-05,  ..., 1.4668e-03,
           6.3896e-04, 4.7340e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3311e-01, 6.6895e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 6.8262e-01, 7.6065e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9258e-01, 2.5606e-04, 2.4533e-04,  ..., 1.0010e-02,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 6.8617e-04, 4.6253e-05,  ..., 5.0659e-03,
           6.7558e-03, 0.0000e+00],
          [8.6035e-01, 8.8549e-04, 1.8454e-04,  ..., 8.0795e-03,
           3.2898e-02, 3.6564e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1543e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 2.8534e-02, 3.5980e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7246e-01, 4.6086e-04, 2.2736e-03,  ..., 3.1952e-02,
           0.0000e+00, 0.0000e+00],
          [3.1982e-01, 3.9983e-04, 7.0429e-04,  ..., 2.1912e-02,
           1.4880e-01, 0.0000e+00],
          [6.5723e-01, 1.6463e-04, 4.4918e-04,  ..., 6.2370e-03,
           5.2948e-02, 4.5868e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6582e-01, 3.4332e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 3.0884e-01, 1.6479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.0830e-01, 2.9802e-05, 2.9087e-05,  ..., 5.5328e-02,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 4.9353e-05, 2.5809e-05,  ..., 6.5918e-03,
           1.8860e-02, 0.0000e+00],
          [4.5508e-01, 2.8431e-05, 6.5148e-05,  ..., 2.7649e-02,
           2.8003e-01, 2.5620e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 1.0901e-01, 7.2098e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.8672e-01, 1.3340e-04, 3.1638e-04,  ..., 1.2199e-02,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.3328e-04, 4.2701e-04,  ..., 4.9438e-03,
           1.2604e-02, 0.0000e+00],
          [7.1680e-01, 5.5122e-04, 9.0218e-04,  ..., 7.4043e-03,
           2.0370e-02, 1.4412e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 6.3965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1299e-01, 1.5271e-01, 3.4424e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4805e-01, 6.8617e-04, 5.6887e-04,  ..., 7.7400e-03,
           0.0000e+00, 0.0000e+00],
          [5.4590e-01, 7.1716e-04, 4.5061e-04,  ..., 2.5043e-03,
           3.8391e-02, 0.0000e+00],
          [4.1138e-01, 7.5769e-04, 5.5122e-04,  ..., 7.8888e-03,
           3.5950e-02, 2.4463e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8223e-01, 1.9971e-01, 1.7838e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.8535e-01, 3.3903e-04, 5.1141e-05,  ..., 4.3411e-03,
           0.0000e+00, 0.0000e+00],
          [9.0918e-01, 3.2973e-04, 2.6393e-04,  ..., 4.6654e-03,
           8.6975e-03, 0.0000e+00],
          [7.1631e-01, 3.0112e-04, 2.4796e-04,  ..., 1.7456e-02,
           1.3623e-01, 1.1559e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.4027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9609e-01, 2.8114e-03, 9.2840e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8555e-01, 8.1062e-06, 1.2696e-05,  ..., 1.6336e-03,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 3.6955e-06, 1.0550e-05,  ..., 8.4019e-04,
           2.7218e-03, 0.0000e+00],
          [9.8584e-01, 1.8477e-06, 4.7684e-06,  ..., 3.2544e-04,
           1.0405e-03, 9.0170e-04]]]], device='cuda:0', dtype=torch.float16),tensor([[[[3.9673e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1787e-02, 4.0829e-05, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.1765e-01, 2.1935e-05, 1.6966e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1689e-01, 1.3232e-04, 2.7156e-04,  ..., 6.6948e-04,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.3403e-03, 4.0054e-04,  ..., 3.9339e-05,
           8.3069e-02, 0.0000e+00],
          [5.6006e-01, 2.9874e-04, 6.7825e-03,  ..., 9.3222e-05,
           3.6740e-04, 8.0505e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1024e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4043e-01, 5.7312e-02, 2.0084e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2073e-01, 2.0087e-05, 6.2823e-05,  ..., 2.6672e-02,
           0.0000e+00, 0.0000e+00],
          [6.5771e-01, 3.2234e-04, 1.2970e-04,  ..., 2.0618e-03,
           1.0925e-02, 0.0000e+00],
          [6.3770e-01, 1.2201e-04, 4.3511e-05,  ..., 9.6512e-03,
           1.2482e-01, 1.4923e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.8763e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 3.1311e-02, 1.4671e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.3916e-01, 1.1921e-07, 4.7684e-07,  ..., 1.8148e-03,
           0.0000e+00, 0.0000e+00],
          [4.6484e-01, 1.4472e-04, 1.6689e-05,  ..., 3.3200e-05,
           8.6441e-03, 0.0000e+00],
          [2.9907e-01, 6.7949e-06, 5.8651e-05,  ..., 4.6134e-04,
           7.6675e-04, 2.6428e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.3400e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.0236e-01, 2.0950e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 1.7107e-04, 7.5042e-05,  ..., 5.9938e-04,
           0.0000e+00, 0.0000e+00],
          [4.9146e-01, 1.0958e-03, 3.8099e-04,  ..., 4.3106e-03,
           2.7420e-02, 0.0000e+00],
          [7.5195e-01, 8.4496e-04, 4.6468e-04,  ..., 4.1847e-03,
           1.5030e-02, 9.9106e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5736e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 9.3460e-03, 1.6449e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7041e-01, 3.4904e-04, 5.0783e-04,  ..., 6.4453e-02,
           0.0000e+00, 0.0000e+00],
          [4.9292e-01, 8.6308e-04, 1.3876e-03,  ..., 2.7893e-02,
           5.6244e-02, 0.0000e+00],
          [4.6387e-01, 2.5005e-03, 2.8687e-03,  ..., 2.8030e-02,
           4.1016e-02, 4.9316e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7617e-01, 4.2383e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2412e-01, 2.5879e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1604e-02, 2.2449e-03, 8.2397e-04,  ..., 8.5144e-03,
           0.0000e+00, 0.0000e+00],
          [1.9751e-01, 1.2169e-03, 8.7452e-04,  ..., 8.4448e-04,
           1.6495e-02, 0.0000e+00],
          [2.4634e-01, 1.4076e-03, 9.1314e-04,  ..., 4.4250e-03,
           2.7069e-02, 4.2358e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9473e-01, 4.0552e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7139e-01, 4.9591e-02, 2.7881e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5867e-02, 1.7405e-05, 1.7047e-05,  ..., 8.8867e-01,
           0.0000e+00, 0.0000e+00],
          [4.4189e-01, 6.7353e-06, 3.3379e-05,  ..., 1.9093e-03,
           4.9097e-01, 0.0000e+00],
          [4.2578e-01, 5.2452e-06, 3.7134e-05,  ..., 3.0537e-03,
           3.4607e-02, 4.6582e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7725e-01, 3.2251e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3657e-01, 4.0552e-01, 3.5791e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3516e-01, 1.9395e-04, 2.3019e-04,  ..., 2.6646e-03,
           0.0000e+00, 0.0000e+00],
          [5.4077e-02, 8.0681e-04, 3.1223e-03,  ..., 1.2665e-03,
           2.2812e-02, 0.0000e+00],
          [1.4929e-01, 1.8950e-03, 4.1046e-03,  ..., 2.4090e-03,
           2.1851e-02, 3.6896e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9166e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.1879e-02, 3.6087e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4902e-01, 3.9864e-04, 4.1676e-04,  ..., 5.4054e-03,
           0.0000e+00, 0.0000e+00],
          [7.6843e-02, 6.7520e-04, 4.6611e-04,  ..., 8.4534e-03,
           4.6265e-02, 0.0000e+00],
          [4.5288e-01, 1.0509e-03, 8.3017e-04,  ..., 3.6240e-03,
           2.4384e-02, 5.3680e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.5725e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.2573e-02, 1.6556e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7427e-01, 4.9591e-05, 2.1207e-04,  ..., 3.5645e-02,
           0.0000e+00, 0.0000e+00],
          [1.2091e-01, 4.7517e-04, 8.7214e-04,  ..., 6.4354e-03,
           2.5845e-03, 0.0000e+00],
          [3.2642e-01, 3.0828e-04, 7.9775e-04,  ..., 4.4174e-03,
           6.0310e-03, 1.9257e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4549e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6865e-01, 1.1395e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1310e-02, 2.2054e-06, 5.1856e-06,  ..., 4.1046e-03,
           0.0000e+00, 0.0000e+00],
          [2.9834e-01, 1.4734e-04, 2.0862e-04,  ..., 1.5808e-02,
           6.8169e-03, 0.0000e+00],
          [3.1152e-01, 2.2471e-05, 4.5598e-05,  ..., 2.9907e-02,
           1.5686e-02, 1.0612e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1934e-01, 1.8079e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 1.9791e-02, 3.3142e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1350e-02, 4.1366e-05, 1.7226e-04,  ..., 7.2656e-01,
           0.0000e+00, 0.0000e+00],
          [4.2700e-01, 1.0288e-04, 7.7248e-04,  ..., 2.0676e-03,
           4.8267e-01, 0.0000e+00],
          [3.5718e-01, 2.0742e-05, 2.3258e-04,  ..., 2.7523e-03,
           2.9877e-02, 5.6250e-01]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8584e-01, 5.1416e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3628e-01, 1.7786e-01, 3.8574e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4259e-02, 3.5286e-05, 6.6936e-05,  ..., 8.8770e-01,
           0.0000e+00, 0.0000e+00],
          [1.4587e-01, 1.3125e-04, 3.7265e-04,  ..., 2.9016e-04,
           7.6611e-01, 0.0000e+00],
          [1.0394e-01, 1.3053e-04, 2.2340e-04,  ..., 1.4601e-03,
           1.3318e-01, 6.4941e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8926e-01, 1.0590e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 4.0131e-02, 1.2756e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7078e-01, 9.1314e-05, 1.1140e-04,  ..., 6.3610e-04,
           0.0000e+00, 0.0000e+00],
          [8.7891e-01, 3.1328e-04, 1.3189e-03,  ..., 5.3883e-04,
           8.9569e-03, 0.0000e+00],
          [8.1641e-01, 2.6131e-04, 9.4891e-04,  ..., 3.8567e-03,
           1.4427e-02, 8.2245e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1055e-01, 1.8921e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.7930e-01, 1.3416e-01, 8.6365e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2544e-01, 2.0683e-05, 1.8120e-05,  ..., 3.9429e-02,
           0.0000e+00, 0.0000e+00],
          [1.4320e-02, 9.8765e-05, 9.3699e-04,  ..., 4.5598e-05,
           7.5836e-03, 0.0000e+00],
          [1.9165e-01, 1.5199e-04, 5.6887e-04,  ..., 8.0729e-04,
           6.0577e-03, 1.3191e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 2.6465e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5459e-01, 5.9662e-02, 4.8584e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4290e-02, 1.0133e-06, 5.7220e-06,  ..., 9.6143e-01,
           0.0000e+00, 0.0000e+00],
          [1.4404e-01, 1.2696e-05, 8.8394e-05,  ..., 2.3918e-03,
           8.0811e-01, 0.0000e+00],
          [1.2042e-01, 7.8082e-06, 1.8895e-05,  ..., 2.3537e-03,
           3.3752e-02, 8.0566e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 4.4861e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2617e-01, 5.9906e-02, 1.1414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.5625e-01, 4.6611e-04, 1.7691e-04,  ..., 2.2766e-02,
           0.0000e+00, 0.0000e+00],
          [1.6797e-01, 2.1982e-04, 6.8378e-04,  ..., 9.1124e-04,
           4.9951e-01, 0.0000e+00],
          [1.6870e-01, 1.7023e-04, 6.9332e-04,  ..., 2.2087e-03,
           2.1277e-01, 2.6733e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 1.7792e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 9.8877e-03, 5.1994e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4473e-01, 6.8069e-05, 1.7166e-05,  ..., 1.3332e-03,
           0.0000e+00, 0.0000e+00],
          [8.3740e-01, 4.2510e-04, 1.2913e-03,  ..., 1.5039e-03,
           1.1263e-03, 0.0000e+00],
          [8.7988e-01, 1.8942e-04, 4.3869e-04,  ..., 2.4128e-03,
           7.1335e-04, 2.0199e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4351e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 4.6768e-03, 2.6245e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1228e-01, 3.1924e-04, 3.3069e-04,  ..., 7.8735e-03,
           0.0000e+00, 0.0000e+00],
          [7.4707e-01, 4.4227e-04, 7.5531e-04,  ..., 1.3332e-03,
           1.5869e-02, 0.0000e+00],
          [4.3018e-01, 9.6893e-04, 7.5150e-04,  ..., 2.6226e-03,
           3.8025e-02, 2.9739e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3071e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4014e-01, 2.9126e-01, 6.8481e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0605e-01, 2.4617e-05, 1.2982e-04,  ..., 9.9564e-03,
           0.0000e+00, 0.0000e+00],
          [9.3945e-01, 9.3877e-05, 1.0353e-04,  ..., 1.6842e-03,
           1.0338e-02, 0.0000e+00],
          [7.6611e-01, 3.7372e-05, 6.1631e-05,  ..., 1.7910e-03,
           9.2834e-02, 1.8341e-02]],

         [[8.4814e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3756e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0801e-01, 1.7929e-03, 2.6283e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7031e-01, 2.7776e-04, 5.1117e-04,  ..., 1.7838e-02,
           0.0000e+00, 0.0000e+00],
          [6.3477e-01, 8.8072e-04, 2.7943e-03,  ..., 6.0499e-05,
           1.5671e-02, 0.0000e+00],
          [6.3330e-01, 4.7302e-04, 1.1730e-03,  ..., 5.7745e-04,
           2.5520e-03, 3.3173e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8340e-01, 1.6479e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4502e-01, 3.3179e-01, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8467e-01, 2.9385e-05, 1.5497e-06,  ..., 2.1713e-02,
           0.0000e+00, 0.0000e+00],
          [7.5244e-01, 1.4746e-04, 5.6863e-05,  ..., 1.2091e-01,
           5.4993e-02, 0.0000e+00],
          [4.6704e-01, 5.2452e-06, 5.4896e-05,  ..., 9.9411e-03,
           4.4922e-01, 6.9092e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.1569e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 1.1719e-02, 1.2405e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7437e-01, 1.4515e-03, 1.6584e-03,  ..., 5.3635e-03,
           0.0000e+00, 0.0000e+00],
          [6.8262e-01, 3.9024e-03, 5.1613e-03,  ..., 1.5554e-03,
           3.5801e-03, 0.0000e+00],
          [7.0068e-01, 1.6909e-03, 3.3092e-03,  ..., 3.2654e-03,
           7.6065e-03, 1.9516e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 2.0920e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9971e-01, 2.8149e-01, 1.8967e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1826e-01, 3.2663e-05, 3.0696e-05,  ..., 3.0029e-02,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 9.1314e-05, 5.6505e-05,  ..., 5.1453e-02,
           2.7679e-02, 0.0000e+00],
          [7.5781e-01, 3.3617e-05, 6.1095e-05,  ..., 8.0383e-02,
           7.8125e-02, 1.8143e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.3940e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 8.9722e-03, 8.7738e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 6.7890e-05, 1.7107e-05,  ..., 6.9656e-03,
           0.0000e+00, 0.0000e+00],
          [3.7109e-01, 1.6394e-03, 1.2894e-03,  ..., 6.5193e-03,
           1.8753e-02, 0.0000e+00],
          [7.4268e-01, 6.4278e-04, 2.2960e-04,  ..., 5.7411e-03,
           1.1101e-03, 5.6305e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 1.9958e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8145e-01, 9.2239e-03, 9.2010e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8684e-02, 2.3496e-04, 3.6311e-04,  ..., 5.5695e-03,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.6751e-04, 8.2731e-05,  ..., 1.9133e-05,
           1.7529e-03, 0.0000e+00],
          [9.1699e-01, 3.1805e-04, 1.5104e-04,  ..., 2.7847e-04,
           8.2321e-03, 1.6754e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 4.8157e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 3.2043e-02, 1.6205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8833e-01, 2.7061e-04, 1.5426e-04,  ..., 1.3733e-02,
           0.0000e+00, 0.0000e+00],
          [9.1992e-01, 2.3327e-03, 5.4264e-04,  ..., 3.8147e-05,
           4.8370e-03, 0.0000e+00],
          [5.8740e-01, 1.7452e-03, 6.4516e-04,  ..., 4.7016e-04,
           6.7078e-02, 4.1748e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 2.5131e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0322e-01, 1.5112e-01, 4.5593e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.0312e-01, 5.7936e-05, 5.5492e-05,  ..., 3.1143e-02,
           0.0000e+00, 0.0000e+00],
          [1.8848e-01, 2.6608e-04, 2.7466e-04,  ..., 7.1655e-02,
           8.5938e-02, 0.0000e+00],
          [3.4033e-01, 8.7857e-05, 1.3554e-04,  ..., 6.2683e-02,
           6.1646e-02, 6.0150e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9272e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 8.3252e-02, 2.6749e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6113e-01, 2.6166e-05, 1.4365e-05,  ..., 3.9276e-02,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 6.6948e-04, 9.4461e-04,  ..., 7.1594e-02,
           3.6346e-02, 0.0000e+00],
          [7.3877e-01, 3.4642e-04, 2.6774e-04,  ..., 3.2867e-02,
           7.9346e-03, 7.5684e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.2797e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9941e-01, 6.1523e-02, 3.8879e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0400e-01, 4.3631e-05, 1.0425e-04,  ..., 3.0319e-02,
           0.0000e+00, 0.0000e+00],
          [8.1494e-01, 1.2999e-03, 3.9978e-03,  ..., 3.5839e-03,
           1.9257e-02, 0.0000e+00],
          [7.3340e-01, 4.8923e-04, 9.4652e-04,  ..., 3.1338e-03,
           5.0446e-02, 5.6427e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.0303e-01, 3.9697e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4221e-01, 3.9337e-02, 8.1836e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7842e-02, 6.7353e-06, 2.9504e-05,  ..., 8.4814e-01,
           0.0000e+00, 0.0000e+00],
          [4.2786e-02, 1.9729e-05, 7.5281e-05,  ..., 1.4610e-03,
           9.3652e-01, 0.0000e+00],
          [6.3721e-02, 3.8743e-06, 3.0339e-05,  ..., 1.8616e-03,
           2.9404e-02, 8.7988e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 1.8591e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3384e-01, 5.2148e-01, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2583e-01, 7.0751e-05, 6.4909e-05,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [7.2070e-01, 3.2783e-04, 1.1355e-04,  ..., 2.7969e-02,
           5.9753e-02, 0.0000e+00],
          [4.5215e-01, 1.0467e-04, 6.5386e-05,  ..., 1.9348e-02,
           1.2854e-01, 2.0947e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6025e-01, 2.3975e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9600e-01, 3.7933e-02, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3481e-01, 3.3970e-03, 3.4695e-03,  ..., 1.1940e-02,
           0.0000e+00, 0.0000e+00],
          [5.0244e-01, 2.7161e-02, 3.9612e-02,  ..., 2.0828e-03,
           4.4098e-02, 0.0000e+00],
          [5.2881e-01, 8.4763e-03, 1.0414e-02,  ..., 4.0359e-03,
           2.7756e-02, 8.0017e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6826e-01, 3.1921e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6621e-01, 9.6008e-02, 3.7659e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9385e-01, 1.1683e-05, 8.5235e-06,  ..., 1.1147e-02,
           0.0000e+00, 0.0000e+00],
          [2.7637e-01, 7.2975e-03, 4.6005e-03,  ..., 2.8496e-03,
           6.8016e-03, 0.0000e+00],
          [2.9980e-01, 8.8263e-04, 7.1049e-04,  ..., 5.3368e-03,
           1.2199e-02, 3.0777e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8047e-01, 1.9516e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 4.6806e-03, 4.1695e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9590e-01, 2.6882e-05, 3.3140e-05,  ..., 7.3929e-03,
           0.0000e+00, 0.0000e+00],
          [3.5767e-01, 5.5084e-03, 3.3417e-03,  ..., 2.5868e-05,
           2.0798e-02, 0.0000e+00],
          [7.7979e-01, 5.6791e-04, 6.1655e-04,  ..., 5.0449e-04,
           3.3665e-03, 1.4374e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9185e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.5375e-02, 1.3771e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2189e-01, 2.9325e-05, 8.8811e-06,  ..., 2.8305e-03,
           0.0000e+00, 0.0000e+00],
          [2.9831e-02, 4.4918e-04, 4.0293e-04,  ..., 4.7326e-04,
           1.0452e-02, 0.0000e+00],
          [6.4148e-02, 1.5223e-04, 1.0306e-04,  ..., 4.7951e-03,
           4.9400e-04, 5.5199e-03]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0820e-01, 9.2041e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6938e-01, 4.2505e-01, 2.0557e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0015e-01, 1.0192e-05, 6.6757e-06,  ..., 5.6953e-03,
           0.0000e+00, 0.0000e+00],
          [1.7822e-01, 3.3021e-04, 3.8624e-04,  ..., 7.3853e-03,
           3.6346e-02, 0.0000e+00],
          [3.0176e-01, 3.0756e-04, 3.2878e-04,  ..., 5.8861e-03,
           5.2032e-02, 4.1504e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 2.0859e-02, 2.0172e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0654e-01, 1.1480e-04, 6.2346e-05,  ..., 6.5002e-02,
           0.0000e+00, 0.0000e+00],
          [4.4873e-01, 2.7637e-03, 4.4708e-03,  ..., 3.7785e-03,
           1.0254e-01, 0.0000e+00],
          [4.8560e-01, 1.6909e-03, 1.8425e-03,  ..., 7.4310e-03,
           2.7710e-02, 5.9357e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1982e-01, 1.8005e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4075e-01, 6.6895e-01, 1.9019e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.2571e-02, 3.1590e-05, 3.2783e-06,  ..., 2.4323e-02,
           0.0000e+00, 0.0000e+00],
          [2.5955e-02, 4.2772e-04, 3.8195e-04,  ..., 3.9787e-03,
           1.4000e-02, 0.0000e+00],
          [7.0534e-03, 3.0279e-04, 6.9737e-05,  ..., 7.6866e-03,
           6.2752e-03, 1.2383e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4883e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 2.5894e-02, 4.5700e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7876e-01, 8.5831e-06, 1.0669e-05,  ..., 1.1864e-03,
           0.0000e+00, 0.0000e+00],
          [3.5181e-01, 9.8610e-04, 8.5163e-04,  ..., 4.8714e-03,
           7.3280e-03, 0.0000e+00],
          [3.6182e-01, 4.0913e-04, 3.0494e-04,  ..., 4.5395e-03,
           6.0577e-03, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7089e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9512e-01, 3.7671e-01, 1.2817e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8197e-02, 7.2718e-06, 3.8147e-06,  ..., 5.8517e-03,
           0.0000e+00, 0.0000e+00],
          [5.8008e-01, 1.2808e-03, 3.0422e-03,  ..., 7.8821e-04,
           6.6833e-02, 0.0000e+00],
          [1.1591e-01, 8.4591e-04, 8.6594e-04,  ..., 1.4477e-03,
           4.3976e-02, 3.5889e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5205e-01, 1.4807e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 3.8544e-02, 2.7390e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0225e-01, 5.3704e-05, 6.3956e-05,  ..., 3.2654e-03,
           0.0000e+00, 0.0000e+00],
          [3.5474e-01, 2.9697e-03, 1.7807e-02,  ..., 2.8753e-04,
           1.0864e-01, 0.0000e+00],
          [5.7129e-01, 2.0599e-03, 5.7335e-03,  ..., 7.6914e-04,
           1.6785e-02, 4.0283e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1113e-01, 8.8989e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 3.6831e-03, 1.8196e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5854e-01, 1.5478e-03, 2.1458e-03,  ..., 2.2507e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 3.7785e-03, 1.7462e-03,  ..., 3.5524e-05,
           7.5134e-02, 0.0000e+00],
          [1.2891e-01, 2.1935e-03, 2.9202e-03,  ..., 5.0366e-05,
           5.3613e-01, 2.4506e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.2292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.3941e-02, 1.5205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0039e-02, 4.0829e-05, 3.5167e-05,  ..., 9.9659e-04,
           0.0000e+00, 0.0000e+00],
          [6.1914e-01, 4.3945e-03, 2.2850e-03,  ..., 3.5834e-04,
           2.0203e-02, 0.0000e+00],
          [2.9346e-01, 2.2659e-03, 2.2411e-03,  ..., 1.8721e-03,
           6.6650e-02, 5.1392e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0430e-01, 9.5520e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 2.3438e-01, 3.5248e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6450e-01, 1.9193e-05, 1.6570e-05,  ..., 1.6418e-02,
           0.0000e+00, 0.0000e+00],
          [9.5154e-02, 8.0729e-04, 7.1812e-04,  ..., 1.3336e-02,
           2.1072e-02, 0.0000e+00],
          [2.7637e-01, 5.4979e-04, 7.0572e-04,  ..., 1.3496e-02,
           3.4729e-02, 5.2765e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 5.6458e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2578e-01, 4.1870e-02, 3.2440e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8301e-01, 3.9482e-04, 2.3222e-04,  ..., 2.5421e-02,
           0.0000e+00, 0.0000e+00],
          [4.8047e-01, 8.4610e-03, 1.5152e-02,  ..., 2.6627e-03,
           2.4490e-02, 0.0000e+00],
          [6.0059e-01, 2.7161e-03, 3.7785e-03,  ..., 6.0387e-03,
           2.3560e-02, 5.4352e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 1.1539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 7.8142e-05, 2.1954e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7803e-01, 5.6446e-05, 8.7738e-05,  ..., 4.8399e-04,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 9.0599e-05, 3.1495e-04,  ..., 7.8678e-05,
           4.3091e-02, 0.0000e+00],
          [9.9023e-01, 1.3995e-04, 7.5483e-04,  ..., 3.1590e-05,
           5.1594e-04, 1.7796e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.1643e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 3.2711e-03, 2.4147e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9541e-01, 4.4727e-04, 5.5075e-04,  ..., 2.0466e-03,
           0.0000e+00, 0.0000e+00],
          [8.5596e-01, 8.0633e-04, 1.2493e-03,  ..., 1.3151e-03,
           1.7593e-02, 0.0000e+00],
          [8.8428e-01, 2.3603e-04, 4.8137e-04,  ..., 1.2245e-03,
           6.9504e-03, 1.3878e-02]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9062e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4775e-01, 2.2293e-02, 2.9999e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0419e-01, 1.7099e-03, 8.1396e-04,  ..., 2.4353e-02,
           0.0000e+00, 0.0000e+00],
          [8.4814e-01, 5.5389e-03, 4.0512e-03,  ..., 4.6074e-05,
           4.0283e-02, 0.0000e+00],
          [3.5205e-01, 8.5449e-03, 3.7918e-03,  ..., 3.6669e-04,
           1.0248e-01, 9.0454e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0576e-01, 9.3994e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3936e-01, 1.2573e-01, 3.4668e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2227e-01, 2.1210e-03, 8.9788e-04,  ..., 1.0201e-02,
           0.0000e+00, 0.0000e+00],
          [1.3062e-01, 9.1019e-03, 1.8196e-03,  ..., 5.8055e-05,
           6.5674e-02, 0.0000e+00],
          [6.8787e-02, 5.7831e-03, 1.5202e-03,  ..., 4.4250e-04,
           1.1981e-01, 1.3062e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 4.1479e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1812e-01, 2.0068e-01, 4.8120e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1179e-02, 1.8477e-06, 1.4305e-06,  ..., 6.4355e-01,
           0.0000e+00, 0.0000e+00],
          [7.6538e-02, 1.3266e-03, 2.3289e-03,  ..., 3.0384e-03,
           6.7676e-01, 0.0000e+00],
          [3.7598e-02, 8.2910e-05, 1.4901e-04,  ..., 1.6556e-02,
           1.0712e-01, 5.8838e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6436e-01, 3.5553e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0186e-01, 5.6519e-02, 4.1840e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2539e-01, 2.4676e-05, 1.6332e-05,  ..., 2.0142e-02,
           0.0000e+00, 0.0000e+00],
          [2.8516e-01, 1.3571e-03, 2.3270e-03,  ..., 3.5896e-03,
           7.4341e-02, 0.0000e+00],
          [2.5488e-01, 2.4080e-04, 3.0136e-04,  ..., 9.6512e-03,
           5.3864e-02, 6.3477e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2041e-01, 7.9773e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9336e-01, 2.0984e-01, 9.6802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1758e-01, 6.0797e-06, 3.8147e-06,  ..., 1.2070e-02,
           0.0000e+00, 0.0000e+00],
          [2.3022e-01, 6.8009e-05, 8.2672e-05,  ..., 2.5436e-02,
           3.6713e-02, 0.0000e+00],
          [4.2017e-01, 5.3823e-05, 5.3585e-05,  ..., 5.6854e-02,
           1.5549e-02, 6.8604e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0898e-01, 2.9102e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1074e-01, 1.7444e-01, 3.1470e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2852e-01, 1.2159e-05, 6.9737e-06,  ..., 2.7783e-01,
           0.0000e+00, 0.0000e+00],
          [3.7280e-01, 1.2054e-03, 1.2236e-03,  ..., 2.8019e-03,
           3.7280e-01, 0.0000e+00],
          [2.8638e-01, 3.6240e-04, 3.4046e-04,  ..., 3.8147e-02,
           9.0759e-02, 3.8818e-01]]]], device='cuda:0', dtype=torch.float16),tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2104e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 1.5945e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.5059e-01, 3.5703e-05, 1.3351e-05,  ..., 5.2986e-03,
           0.0000e+00, 0.0000e+00],
          [4.9390e-01, 6.7215e-03, 3.2074e-02,  ..., 4.0591e-05,
           1.7615e-01, 0.0000e+00],
          [7.5732e-01, 1.5802e-03, 4.0359e-03,  ..., 2.0730e-04,
           3.2257e-02, 3.1250e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6904e-01, 2.3096e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 2.9419e-01, 1.2073e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8384e-01, 3.8028e-05, 3.7432e-05,  ..., 1.3115e-02,
           0.0000e+00, 0.0000e+00],
          [3.2983e-01, 2.7657e-03, 3.3894e-03,  ..., 4.9591e-04,
           3.2983e-01, 0.0000e+00],
          [1.8567e-01, 5.5552e-04, 4.2582e-04,  ..., 1.7643e-03,
           1.5881e-01, 1.3379e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6982e-01, 4.3018e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9941e-01, 6.3171e-02, 5.3760e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6294e-02, 2.8551e-05, 1.1563e-05,  ..., 7.8271e-01,
           0.0000e+00, 0.0000e+00],
          [1.4319e-01, 1.1282e-03, 1.3189e-03,  ..., 2.8062e-04,
           7.9834e-01, 0.0000e+00],
          [3.3862e-01, 2.2244e-04, 3.6097e-04,  ..., 2.0771e-03,
           1.1340e-01, 4.9268e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4229e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 7.4196e-03, 5.3024e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4153e-02, 1.3828e-05, 2.9802e-06,  ..., 1.2760e-03,
           0.0000e+00, 0.0000e+00],
          [5.6152e-01, 3.8818e-02, 1.7090e-02,  ..., 2.4438e-05,
           1.1145e-01, 0.0000e+00],
          [3.9551e-01, 1.4748e-02, 8.8043e-03,  ..., 6.8998e-04,
           2.8870e-02, 1.3147e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.5596e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 6.2904e-03, 3.7556e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6853e-02, 2.1338e-05, 9.7752e-06,  ..., 7.0333e-04,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 2.2755e-03, 2.8534e-03,  ..., 6.1214e-05,
           3.0930e-02, 0.0000e+00],
          [6.6650e-01, 1.4524e-03, 1.6136e-03,  ..., 4.7874e-04,
           2.4078e-02, 2.7298e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 2.9205e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.2842e-02, 9.3765e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7896e-01, 3.7789e-04, 2.2304e-04,  ..., 5.7068e-03,
           0.0000e+00, 0.0000e+00],
          [4.0869e-01, 9.6054e-03, 9.7580e-03,  ..., 2.3222e-04,
           9.4788e-02, 0.0000e+00],
          [3.8306e-01, 5.2338e-03, 5.8403e-03,  ..., 7.3957e-04,
           6.0608e-02, 2.1606e-02]]]], device='cuda:0', dtype=torch.float16)...]

The input prompt had so many tokens (the second dimension).

torch.Size([1, 410])

Every layer stores the attention for each of the so many tokens.

torch.Size([1, 16, 410, 410])

Looking at the output shape, there are two repeating dimensions. 424 corresponds to the input sequence length. However, it appears twice. The reason for this is that attention is all about relationships.

It’s best to think of this as a grid or embedding of sorts. Each token is mapped to every other token.

I’ll take a simple input sequence as an example ['<image>', 'the', 'cat']. To calculate the attention of this sequence, the model needs to know how important each token is to each other.

· <image> (being looked at) the (being looked at) cat (being looked at)
<image> (looking) score(image, image) score(image, the) score(image, cat)
the (looking) score(the, image) score(the, the) score(the, cat)
cat (looking) score(cat, image) score(cat, the) score(cat, cat)

Specifically speaking, the first sequence length (the rows) is the query dimension–the token that is doing the looking. The second sequence length (the columns) is the key dimension–the token that is being looked at.

So a layer with shape [1, 28, 424, 424] is a layer with 1 batch, with 28 attention heads, and a 424x424 attention matrix. An entry [i,j] tells us how much attention token i is paying to token j.

When the first token is generated, it gets appended to the input prompt. This appended form is then used to generate the next token.

I can see that the sequence length has indeed been increased by 1 after the first token has been generated.

torch.Size([1, 16, 1, 411])

reword???? - what is self attention? - steps0/1/2 don’t seem relevant to the previous sentence

However, the last two dimensions are now not the same. Something smart is happening here: computation is being saved.

Let’s say that the last two dimensions remained at 425x425, that would mean a lot of duplicated, redundant computation.

  • Step 0 (Prompt): compute attention for 424 tokens. Matrix is 424x424.
  • Step 1 (Generate 1st Token): The sequence is now 425 tokens long. Recompute attention for all 425 tokens. Matrix would be 425x425.
  • Step 2 (Generate 2nd Token): Sequence is 426 tokens. Recalculate attention for all 426. Matrix would be 426x426.

The attention for tokens that have already been calculated, are repeatedly being calculated. That’s computationally wasteful. Instead, the model is performing what’s called KV caching. In KV caching, the model knows the attention scores for certain tokens won’t change. They can therefore be cached.

  1. Processing the Prompt
    • The model processes the initial 424 tokens and calculates the query, key, and value vectors for each token.
    • Full self-attention is performed, where every token’s query is compared against every token’s key.
    • The key and value vectors for all 424 tokens are then saved to a KV cache
  2. Generating the First New Token
    • The query vector for the single new token that is about to be generated is calculated
    • This single query is compared against the keys of all 425 tokens (424 cached keys + the key for the new token itself)

Only the single row of the attention matrix that is need to predict the next token is calculated.

I’ll begin by calculating the average attention for the prompt, and for the first layer of the model

torch.Size([1, 16, 410, 410])
torch.Size([16, 410, 410])

I squeeze because the batch size is 1. Now, I’ll average the attention scores across all heads in this layer to produce a single attention map/grid/matrix.

(torch.Size([410, 410]),
 tensor(0., device='cuda:0', dtype=torch.float16),
 tensor(0.9375, device='cuda:0', dtype=torch.float16))

In autoregressive models, the attention for token i is used to predict token i+1. The final token in the prompt doesn’t predict a new token within the prompt itself, so we don’t need to visualize its attention.

torch.Size([409, 410])

The first token is typically a special “beginning of sentence” token. Including it can skew the visualization (we don’t want the other tokens to pay attention to it). In this case, it’s <|im_start|>.

Now since some values are 0, the attention scores for each token no longer sum to 1.

Token 0: 1.0
Token 2: 1.0
Token 4: 1.0
Token 6: 1.0
Token 8: 1.0
Token 0: 1.0
Token 2: 0.85107421875
Token 4: 0.93994140625
Token 6: 0.927734375
Token 8: 0.966796875

I’ll renormalize.

torch.Size([408, 410])
torch.Size([408])
torch.Size([408, 1])
Token 0: 1.0
Token 2: 1.0
Token 4: 0.99951171875
Token 6: 0.99951171875
Token 8: 1.0

And there I have the aggregated attention map for this layer!

Now I’ll aggregate the attention map for every layer, and then take the average aross all layers.

((#36) [tensor([[9.3750e-01, 0.0000e+00, 0.0000e+00,  ..., 1.5509e-04, 1.5509e-04,
         1.5509e-04],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.3022e-01, 7.7002e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 3.6240e-03, 4.5227e-02,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.1763e-03, 2.3079e-03,  ..., 5.6934e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.9507e-03, 6.1569e-03,  ..., 1.5173e-01, 4.4360e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.5005, 0.4993,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0266, 0.0486,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0193, 0.0310,  ..., 0.1383, 0.0000, 0.0000],
        [0.0000, 0.0118, 0.0295,  ..., 0.1365, 0.2205, 0.0000]],
       dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.4321, 0.5679,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0047, 0.0279,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0101, 0.0315,  ..., 0.1324, 0.0000, 0.0000],
        [0.0000, 0.0050, 0.0187,  ..., 0.1065, 0.1547, 0.0000]],
       dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.2546, 0.7456,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0017, 0.0051,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0064, 0.0036,  ..., 0.0943, 0.0000, 0.0000],
        [0.0000, 0.0038, 0.0262,  ..., 0.0109, 0.0842, 0.0000]],
       dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.5645, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.5317, 0.1278,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0036, 0.0082,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0025, 0.0040,  ..., 0.0565, 0.0000, 0.0000],
        [0.0000, 0.0063, 0.0084,  ..., 0.1383, 0.1129, 0.0000]],
       dtype=torch.float16),tensor([[9.8584e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.4756e-01, 0.0000e+00,  ..., 2.3136e-03, 9.6464e-04,
         2.3136e-03],
        [0.0000e+00, 2.0923e-01, 1.2354e-01,  ..., 7.7744e-03, 2.3479e-03,
         0.0000e+00],
        ...,
        [0.0000e+00, 4.7531e-03, 8.2827e-04,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.1795e-02, 4.3893e-04,  ..., 1.0870e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 4.2633e-02, 2.9583e-03,  ..., 3.5858e-02, 2.7051e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[9.7461e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 3.5059e-01, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 5.8398e-01, 1.3818e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 3.9940e-03, 3.4351e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.5430e-03, 7.5197e-04,  ..., 1.8066e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 9.0408e-03, 3.3112e-03,  ..., 3.5156e-02, 1.8225e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 7.5342e-01, 2.4670e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 5.4092e-03, 3.6259e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.4486e-03, 9.3985e-04,  ..., 1.0785e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 5.8098e-03, 8.2779e-03,  ..., 2.0111e-02, 9.5703e-02,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.7437, 0.2563,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0039, 0.0050,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0010, 0.0013,  ..., 0.0941, 0.0000, 0.0000],
        [0.0000, 0.0020, 0.0030,  ..., 0.0086, 0.0482, 0.0000]],
       dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 8.0127e-01, 1.9849e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 4.1161e-03, 3.6125e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 9.3937e-04, 1.6222e-03,  ..., 7.7209e-02, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 3.7823e-03, 2.2488e-03,  ..., 4.8798e-02, 1.9214e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[9.6240e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.5771e-01, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 4.6777e-01, 2.6929e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 4.2343e-03, 7.1182e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 9.3174e-04, 6.3372e-04,  ..., 1.4697e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.5940e-03, 2.4128e-03,  ..., 1.2657e-02, 1.3232e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.6260e-01, 3.3716e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 2.6321e-03, 3.2005e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.3329e-04, 2.7370e-04,  ..., 1.6296e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 8.0824e-04, 1.4486e-03,  ..., 1.0498e-02, 1.3550e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.4553, 0.4863,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0134, 0.0215,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0018, 0.0015,  ..., 0.1836, 0.0000, 0.0000],
        [0.0000, 0.0016, 0.0088,  ..., 0.0011, 0.2212, 0.0000]],
       dtype=torch.float16),tensor([[0.9614, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.3962, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.4507, 0.1093,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0039, 0.0042,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0012, 0.0011,  ..., 0.0604, 0.0000, 0.0000],
        [0.0000, 0.0039, 0.0053,  ..., 0.0582, 0.2627, 0.0000]],
       dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 7.0215e-01, 2.9834e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 2.2697e-03, 1.6098e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 3.6788e-04, 2.6536e-04,  ..., 1.3330e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 2.3022e-03, 2.6283e-03,  ..., 2.4414e-02, 2.9639e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.7627e-01, 3.2349e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 6.1646e-03, 7.9880e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 4.3368e-04, 3.8838e-04,  ..., 1.3025e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0765e-02, 1.0086e-02,  ..., 1.4320e-02, 2.0105e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.6602e-01, 3.3423e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 2.3918e-03, 5.2643e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 5.5075e-05, 5.1320e-05,  ..., 1.2042e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.6098e-03, 4.1618e-03,  ..., 3.7708e-03, 2.4011e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000, 0.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 1.0000, 0.0000,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.5054, 0.4951,  ..., 0.0000, 0.0000, 0.0000],
        ...,
        [0.0000, 0.0186, 0.0322,  ..., 0.0000, 0.0000, 0.0000],
        [0.0000, 0.0012, 0.0013,  ..., 0.0992, 0.0000, 0.0000],
        [0.0000, 0.0077, 0.0092,  ..., 0.0061, 0.2847, 0.0000]],
       dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 5.8398e-01, 4.1626e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 4.8218e-03, 3.8338e-03,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.8951e-04, 4.6182e-04,  ..., 1.0419e-01, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 3.3150e-03, 2.7370e-03,  ..., 1.0651e-02, 2.1912e-01,
         0.0000e+00]], dtype=torch.float16),tensor([[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 6.3379e-01, 3.6694e-01,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 1.1711e-02, 1.7197e-02,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.6582e-04, 8.2612e-05,  ..., 9.2590e-02, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 1.2901e-02, 1.3611e-02,  ..., 1.8188e-02, 3.0884e-01,
         0.0000e+00]], dtype=torch.float16)...],
 torch.Size([409, 410]))
torch.Size([409, 410])

What I’ve just done is that for each layer, I’ve averaged the attention across the 28 heads. Then I’ve averaged the averages across all layers.

Now I want to aggregate the attentions map for generating only the 1st token, the map used to generate only the 2nd token, and so on. Rather than prompt+1st token, prompt+2nd token, and so on. The main difference here is how the vector is created.

torch.Size([410, 410])
(torch.Size([410]), torch.Size([409]))
torch.Size([411])
  1. The first entry is zero, because that’s null attention.
  2. On first generation, there’s a row for each token in the prompt. So I do [-1] to get the most recent token.
  3. The attention for the last token is set to zero. This token never gets any attention since it doesn’t exist.
(#27) [(tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4316e-01, 2.5684e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6530e-02, 5.0439e-01, 4.5923e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7796e-03, 1.6203e-03, 6.6109e-03,  ..., 1.9934e-01,
           0.0000e+00, 0.0000e+00],
          [1.4343e-03, 3.3360e-03, 1.1284e-02,  ..., 3.9771e-01,
           1.0059e-01, 0.0000e+00],
          [9.5129e-05, 2.2335e-03, 7.0953e-03,  ..., 4.6265e-02,
           1.7200e-01, 5.8154e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9141e-04, 2.5940e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8413e-06, 2.3186e-05, 8.0943e-05,  ..., 9.5459e-01,
           0.0000e+00, 0.0000e+00],
          [7.1168e-05, 2.8157e-04, 8.2245e-03,  ..., 5.3650e-02,
           5.0879e-01, 0.0000e+00],
          [1.0729e-06, 1.6928e-05, 6.3515e-04,  ..., 6.7616e-04,
           2.8763e-02, 9.5215e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1124e-04, 9.9902e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5359e-05, 4.3058e-04, 9.9951e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0000e+00,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 1.0986e-02,
           9.8877e-01, 0.0000e+00],
          [6.4969e-06, 2.2650e-05, 4.0985e-02,  ..., 2.2650e-05,
           2.7609e-04, 4.9927e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9312e-02, 9.7070e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0651e-01, 1.9897e-01, 6.9434e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7684e-07, 6.0797e-06, 6.0797e-06,  ..., 9.9268e-01,
           0.0000e+00, 0.0000e+00],
          [0.0000e+00, 8.3447e-07, 4.5121e-05,  ..., 4.0588e-03,
           9.9316e-01, 0.0000e+00],
          [8.1024e-03, 9.1858e-03, 1.9440e-02,  ..., 1.1795e-02,
           1.7151e-02, 7.6904e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9233e-01, 6.0742e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5915e-02, 5.9143e-02, 9.2480e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0719e-02, 9.4604e-03, 1.0071e-02,  ..., 2.7374e-02,
           0.0000e+00, 0.0000e+00],
          [1.1459e-02, 1.0765e-02, 1.0109e-02,  ..., 7.8735e-03,
           1.2985e-02, 0.0000e+00],
          [2.3193e-03, 6.3057e-03, 4.6600e-02,  ..., 2.6283e-03,
           1.9424e-02, 1.5283e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3018e-01, 5.6982e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.5571e-01, 5.8643e-01, 5.8044e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0340e-02, 1.4427e-02, 6.8130e-03,  ..., 2.6947e-02,
           0.0000e+00, 0.0000e+00],
          [2.4475e-02, 1.3947e-02, 4.3221e-03,  ..., 3.9734e-02,
           3.6194e-02, 0.0000e+00],
          [8.5526e-03, 1.2444e-02, 1.3962e-03,  ..., 2.0508e-02,
           4.6234e-02, 3.3493e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 1.9934e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7510e-01, 2.6245e-01, 2.6245e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9380e-04, 1.0414e-03, 7.8125e-03,  ..., 5.6543e-01,
           0.0000e+00, 0.0000e+00],
          [7.8392e-04, 2.8667e-03, 1.6495e-02,  ..., 2.5806e-01,
           4.2554e-01, 0.0000e+00],
          [1.2255e-03, 3.5458e-03, 9.3384e-03,  ..., 4.4556e-02,
           1.1743e-01, 1.4380e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6387e-01, 3.6224e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0004e-01, 6.5234e-01, 2.4756e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0564e-05, 1.6093e-06, 5.9605e-08,  ..., 4.1504e-02,
           0.0000e+00, 0.0000e+00],
          [3.3140e-05, 7.4685e-05, 1.6928e-05,  ..., 7.0752e-01,
           1.8457e-01, 0.0000e+00],
          [2.0862e-06, 7.0572e-05, 1.6165e-04,  ..., 5.5756e-02,
           7.6953e-01, 1.6650e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4180e-01, 1.5820e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.5615e-01, 3.3740e-01, 1.0620e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0942e-03, 1.4763e-02, 3.6530e-02,  ..., 3.1250e-02,
           0.0000e+00, 0.0000e+00],
          [7.3433e-03, 7.5760e-03, 4.1595e-02,  ..., 1.4978e-01,
           2.3206e-01, 0.0000e+00],
          [1.1871e-02, 3.0792e-02, 3.6011e-02,  ..., 3.2776e-02,
           1.4917e-01, 6.9397e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 2.5830e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4263e-01, 2.6025e-01, 2.9712e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1372e-02, 2.9022e-02, 4.3579e-02,  ..., 9.5154e-02,
           0.0000e+00, 0.0000e+00],
          [8.6288e-03, 2.0523e-02, 4.0192e-02,  ..., 1.2769e-01,
           2.3315e-01, 0.0000e+00],
          [1.6680e-03, 7.0229e-03, 1.2039e-02,  ..., 5.1483e-02,
           1.5137e-01, 8.4229e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4697e-01, 3.5303e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9893e-01, 1.9434e-01, 4.0674e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.4289e-03, 1.4206e-02, 1.0095e-01,  ..., 6.1493e-03,
           0.0000e+00, 0.0000e+00],
          [1.7691e-03, 6.5956e-03, 5.2917e-02,  ..., 2.7451e-02,
           1.0944e-01, 0.0000e+00],
          [7.9880e-03, 1.3489e-02, 3.9795e-02,  ..., 2.8549e-02,
           7.0679e-02, 1.5076e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9160e-01, 1.0822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6890e-01, 4.6997e-01, 1.6113e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1887e-02, 3.3600e-02, 1.8692e-02,  ..., 3.4204e-01,
           0.0000e+00, 0.0000e+00],
          [4.3449e-03, 2.4033e-02, 2.5391e-02,  ..., 6.9580e-02,
           4.0649e-01, 0.0000e+00],
          [8.3590e-04, 7.3929e-03, 1.7977e-03,  ..., 1.6431e-01,
           2.7515e-01, 2.1606e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1826e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2510e-01, 2.0776e-01, 6.6956e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0966e-02, 1.4944e-03, 4.0627e-03,  ..., 7.6660e-02,
           0.0000e+00, 0.0000e+00],
          [2.2415e-02, 4.0817e-03, 1.0422e-02,  ..., 4.7455e-02,
           1.1298e-01, 0.0000e+00],
          [1.7786e-04, 7.8321e-05, 1.6844e-04,  ..., 1.3831e-01,
           6.2012e-01, 1.2207e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5146e-01, 2.4866e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5581e-01, 2.3828e-01, 3.0591e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9973e-03, 5.5733e-03, 8.0795e-03,  ..., 2.6321e-03,
           0.0000e+00, 0.0000e+00],
          [1.1795e-02, 5.0926e-03, 1.1040e-02,  ..., 2.2392e-03,
           1.3770e-01, 0.0000e+00],
          [7.9575e-03, 5.2261e-03, 7.1182e-03,  ..., 3.3321e-03,
           1.5173e-01, 4.3030e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6809e-01, 8.3203e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.2864e-01, 2.3962e-01, 5.3174e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9897e-02, 1.0612e-02, 1.5198e-02,  ..., 1.4935e-03,
           0.0000e+00, 0.0000e+00],
          [1.3489e-02, 7.1106e-03, 7.7782e-03,  ..., 3.8643e-03,
           8.2474e-03, 0.0000e+00],
          [3.9291e-03, 1.2903e-03, 1.0471e-03,  ..., 3.5706e-02,
           3.1891e-02, 1.0413e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1143e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0248e-02, 9.2383e-01, 5.5908e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6413e-02, 7.4339e-04, 2.4700e-04,  ..., 4.6460e-01,
           0.0000e+00, 0.0000e+00],
          [4.2763e-03, 1.2445e-03, 1.4544e-03,  ..., 5.0195e-01,
           1.7480e-01, 0.0000e+00],
          [4.2200e-05, 5.6505e-04, 2.3232e-03,  ..., 3.0609e-02,
           7.8955e-01, 1.3928e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1650e-01, 8.3618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2314e-01, 1.6394e-01, 1.1267e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6040e-02, 1.0773e-02, 4.5380e-02,  ..., 3.8818e-02,
           0.0000e+00, 0.0000e+00],
          [3.5645e-02, 7.0724e-03, 1.8051e-02,  ..., 2.4292e-02,
           1.2146e-01, 0.0000e+00],
          [1.6937e-02, 2.5177e-03, 1.3718e-02,  ..., 2.4078e-02,
           3.9062e-02, 1.7932e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7990e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3267e-01, 1.0986e-01, 6.5771e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6642e-02, 1.5879e-03, 3.3360e-03,  ..., 1.8213e-01,
           0.0000e+00, 0.0000e+00],
          [1.7288e-02, 4.7607e-03, 1.5732e-02,  ..., 2.2241e-01,
           2.8101e-01, 0.0000e+00],
          [1.3023e-02, 1.8110e-03, 1.0712e-02,  ..., 4.9530e-02,
           1.0162e-01, 4.6631e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6182e-01, 1.3831e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6216e-01, 1.4137e-02, 5.2393e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1128e-01, 2.6108e-02, 1.1072e-03,  ..., 3.9154e-02,
           0.0000e+00, 0.0000e+00],
          [2.5928e-01, 1.8272e-03, 1.2445e-01,  ..., 3.6001e-05,
           7.5745e-02, 0.0000e+00],
          [7.4072e-01, 4.9448e-04, 2.7351e-03,  ..., 3.5667e-03,
           3.2684e-02, 1.5419e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 2.3327e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 7.9956e-03, 1.2802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 1.0800e-04, 1.5259e-05,  ..., 6.1941e-04,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.0908e-05, 4.1056e-04,  ..., 1.1955e-02,
           2.4429e-02, 0.0000e+00],
          [8.3789e-01, 4.9233e-05, 1.1092e-04,  ..., 1.0651e-02,
           3.4058e-02, 9.6970e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.7275e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.5106e-02, 1.5381e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6846e-01, 1.4937e-04, 7.3969e-05,  ..., 5.6946e-02,
           0.0000e+00, 0.0000e+00],
          [7.7881e-01, 1.3530e-04, 2.4116e-04,  ..., 6.4564e-04,
           1.2840e-02, 0.0000e+00],
          [8.0615e-01, 8.6963e-05, 9.3079e-04,  ..., 1.5926e-04,
           6.0120e-03, 2.1149e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 1.0138e-03, 6.6071e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1885e-01, 4.9591e-03, 1.0330e-02,  ..., 7.1167e-02,
           0.0000e+00, 0.0000e+00],
          [6.8213e-01, 6.4316e-03, 6.5575e-03,  ..., 1.2457e-05,
           4.9408e-02, 0.0000e+00],
          [8.4082e-01, 1.0557e-03, 4.4098e-03,  ..., 4.2140e-05,
           6.8426e-04, 5.7983e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8291e-01, 1.7242e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 1.9501e-02, 2.6443e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0479e-01, 2.0683e-04, 9.8610e-04,  ..., 9.0456e-04,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.7376e-03, 2.8324e-03,  ..., 1.9646e-03,
           2.3438e-02, 0.0000e+00],
          [5.4688e-01, 3.4404e-04, 6.7139e-04,  ..., 2.3544e-04,
           1.5030e-02, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9586e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2334e-01, 2.8275e-02, 4.8370e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3701e-01, 9.6679e-05, 3.9649e-04,  ..., 6.2752e-04,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 1.2379e-03, 3.8052e-03,  ..., 6.2323e-04,
           3.2104e-02, 0.0000e+00],
          [7.5684e-01, 3.4761e-04, 1.7176e-03,  ..., 2.4796e-04,
           6.6757e-03, 2.1469e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1235e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 3.9482e-03, 4.6921e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6187e-01, 1.0431e-05, 2.2113e-05,  ..., 6.0081e-03,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 6.3539e-05, 5.8460e-04,  ..., 5.2452e-06,
           2.0981e-02, 0.0000e+00],
          [6.6162e-01, 1.7941e-05, 1.0166e-03,  ..., 4.7922e-04,
           1.4820e-03, 1.6098e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6543e-01, 1.0712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3374e-02, 8.1635e-03, 3.5715e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9541e-01, 5.9605e-06, 5.9605e-08,  ..., 2.6875e-03,
           0.0000e+00, 0.0000e+00],
          [9.2822e-01, 2.3007e-04, 1.2040e-05,  ..., 1.7654e-02,
           1.9470e-02, 0.0000e+00],
          [8.7891e-01, 2.4915e-05, 9.0420e-05,  ..., 3.6907e-03,
           1.0187e-01, 1.4641e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 4.0833e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2920e-01, 4.8859e-02, 2.1820e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3018e-01, 2.4247e-04, 5.3704e-05,  ..., 7.7972e-03,
           0.0000e+00, 0.0000e+00],
          [9.0625e-01, 1.1749e-03, 1.2217e-03,  ..., 3.5248e-02,
           1.9638e-02, 0.0000e+00],
          [9.0479e-01, 4.1747e-04, 2.6131e-04,  ..., 5.4359e-03,
           9.6512e-03, 1.0956e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 5.4993e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5107e-01, 1.1450e-01, 3.4241e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7510e-01, 3.3474e-04, 9.2220e-04,  ..., 3.5706e-03,
           0.0000e+00, 0.0000e+00],
          [7.4512e-01, 4.6825e-04, 6.5136e-04,  ..., 9.2621e-03,
           3.0869e-02, 0.0000e+00],
          [6.6162e-01, 2.3687e-04, 2.0623e-04,  ..., 4.4823e-03,
           3.7903e-02, 8.8806e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7028e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 3.2837e-02, 3.6957e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3994e-01, 5.5046e-03, 1.2650e-02,  ..., 5.8708e-03,
           0.0000e+00, 0.0000e+00],
          [6.9141e-01, 1.0643e-02, 1.0078e-02,  ..., 7.0496e-03,
           3.6865e-02, 0.0000e+00],
          [4.1699e-01, 5.6419e-03, 5.1384e-03,  ..., 5.6915e-03,
           5.7648e-02, 2.2400e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4824e-01, 5.1849e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0127e-01, 1.3257e-01, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.2445e-04, 1.0324e-04,  ..., 4.9927e-02,
           0.0000e+00, 0.0000e+00],
          [6.1768e-01, 2.5797e-04, 2.5535e-04,  ..., 9.6985e-02,
           9.6191e-02, 0.0000e+00],
          [5.4834e-01, 8.7798e-05, 1.6689e-04,  ..., 4.7272e-02,
           1.2732e-01, 1.2976e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3588e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4482e-01, 2.1469e-02, 3.3905e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.4255e-01, 9.2840e-04, 7.3147e-04,  ..., 2.2919e-02,
           0.0000e+00, 0.0000e+00],
          [3.0200e-01, 9.4652e-04, 1.0891e-03,  ..., 1.3756e-02,
           7.5684e-02, 0.0000e+00],
          [4.0796e-01, 1.8358e-03, 7.1335e-04,  ..., 1.2772e-02,
           7.2998e-02, 4.8004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 2.3392e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2568e-01, 1.6895e-01, 5.1270e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1123e-01, 5.3692e-04, 4.4227e-05,  ..., 4.5929e-02,
           0.0000e+00, 0.0000e+00],
          [3.7915e-01, 7.6771e-04, 4.3631e-05,  ..., 4.4525e-02,
           2.7710e-01, 0.0000e+00],
          [2.9761e-01, 1.4448e-04, 4.3511e-06,  ..., 6.6414e-03,
           4.4897e-01, 5.8441e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 3.1860e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 2.5665e-02, 1.9028e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8564e-01, 3.8195e-04, 1.2207e-04,  ..., 2.9812e-03,
           0.0000e+00, 0.0000e+00],
          [7.7637e-01, 1.3866e-03, 3.2234e-03,  ..., 6.1989e-03,
           1.0529e-02, 0.0000e+00],
          [7.8125e-01, 2.8181e-04, 6.3896e-05,  ..., 4.9210e-03,
           2.2171e-02, 3.0945e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2604e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 1.1498e-04, 3.6865e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9111e-01, 7.3552e-05, 1.1325e-06,  ..., 1.0364e-01,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 4.2260e-05, 1.4305e-06,  ..., 5.8949e-05,
           2.2446e-02, 0.0000e+00],
          [9.4385e-01, 1.8537e-05, 3.2544e-05,  ..., 1.0258e-04,
           3.6073e-04, 5.2582e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4014e-01, 1.1917e-02, 0.0000e+00,  ..., 6.0997e-03,
           6.0997e-03, 6.0997e-03],
          [2.8540e-01, 1.5497e-06, 7.4158e-03,  ..., 7.4425e-03,
           7.4425e-03, 0.0000e+00],
          ...,
          [9.3018e-01, 1.3933e-03, 3.7253e-05,  ..., 7.2021e-03,
           0.0000e+00, 0.0000e+00],
          [9.8779e-01, 2.1183e-04, 9.8348e-06,  ..., 4.1723e-07,
           6.9733e-03, 0.0000e+00],
          [9.8096e-01, 5.4538e-05, 4.5514e-04,  ..., 2.3842e-07,
           1.1921e-05, 1.3779e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-02, 4.1628e-04, 0.0000e+00,  ..., 4.4823e-03,
           0.0000e+00, 4.4823e-03],
          [1.7053e-01, 2.3842e-07, 8.7404e-04,  ..., 5.7526e-03,
           5.7526e-03, 0.0000e+00],
          ...,
          [9.9365e-01, 7.3791e-05, 5.3644e-07,  ..., 2.0618e-03,
           0.0000e+00, 0.0000e+00],
          [9.9512e-01, 2.8074e-05, 6.5565e-07,  ..., 4.1723e-07,
           3.9139e-03, 0.0000e+00],
          [9.9805e-01, 4.8876e-06, 7.9274e-06,  ..., 5.9605e-08,
           7.7486e-07, 1.5945e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 1.7914e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0615e-01, 1.9080e-01, 3.0651e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5425e-01, 2.3782e-05, 1.3769e-05,  ..., 4.8859e-02,
           0.0000e+00, 0.0000e+00],
          [5.7080e-01, 1.1081e-04, 9.5487e-05,  ..., 8.8989e-02,
           9.0332e-02, 0.0000e+00],
          [7.5342e-01, 3.3808e-04, 2.2292e-05,  ..., 2.7664e-02,
           1.3538e-01, 1.3496e-02]],

         [[5.9277e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5662e-01, 3.0994e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.4082e-01, 6.5422e-04, 6.8045e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6719e-01, 1.0473e-04, 8.1539e-05,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [7.0752e-01, 3.7422e-03, 7.0286e-04,  ..., 1.1086e-05,
           3.4149e-02, 0.0000e+00],
          [9.2480e-01, 1.4150e-04, 4.0627e-04,  ..., 1.2755e-05,
           9.4843e-04, 1.4572e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 9.0027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.7802e-02, 4.3907e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7612e-01, 4.2038e-03, 8.9455e-04,  ..., 1.2268e-01,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 1.8911e-03, 6.9571e-04,  ..., 6.1531e-03,
           1.8356e-02, 0.0000e+00],
          [8.2227e-01, 1.4400e-03, 2.4021e-04,  ..., 1.5097e-03,
           1.7242e-02, 1.7166e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5106e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0869e-01, 8.5022e-02, 6.3057e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1162e-01, 7.6151e-04, 2.4056e-04,  ..., 6.1951e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 4.3793e-03, 5.4407e-04,  ..., 4.9324e-03,
           2.0966e-02, 0.0000e+00],
          [8.6230e-01, 1.1921e-03, 3.2592e-04,  ..., 4.8943e-03,
           1.5625e-02, 2.9266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.7042e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1309e-01, 3.2806e-02, 5.4230e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6865e-01, 2.5034e-05, 5.9307e-05,  ..., 6.3599e-02,
           0.0000e+00, 0.0000e+00],
          [8.2520e-01, 4.2796e-04, 8.6975e-04,  ..., 1.3519e-02,
           9.7107e-02, 0.0000e+00],
          [8.7354e-01, 5.1320e-05, 1.1891e-04,  ..., 3.2444e-03,
           1.7273e-02, 4.6082e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9471e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2021e-01, 2.6611e-01, 1.3817e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8770e-01, 1.6451e-04, 4.8816e-05,  ..., 7.7698e-02,
           0.0000e+00, 0.0000e+00],
          [8.7500e-01, 3.3140e-04, 1.4877e-04,  ..., 3.1910e-03,
           7.4463e-02, 0.0000e+00],
          [8.7830e-02, 3.7050e-04, 4.4584e-05,  ..., 3.6030e-03,
           8.9355e-01, 3.5515e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9805e-01, 1.9121e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 8.2703e-03, 7.5302e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4893e-01, 5.5313e-05, 1.3638e-04,  ..., 4.8523e-02,
           0.0000e+00, 0.0000e+00],
          [5.1904e-01, 2.7776e-04, 1.0710e-03,  ..., 2.8934e-03,
           1.1650e-02, 0.0000e+00],
          [6.9629e-01, 1.5771e-04, 2.1303e-04,  ..., 1.9398e-03,
           7.5378e-03, 2.8076e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5125e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6914e-01, 7.2327e-02, 5.8502e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0566e-01, 5.9414e-04, 1.3161e-03,  ..., 6.1005e-02,
           0.0000e+00, 0.0000e+00],
          [7.1387e-01, 7.0047e-04, 2.0945e-04,  ..., 1.6546e-03,
           2.1942e-02, 0.0000e+00],
          [5.6152e-01, 7.2432e-04, 1.9264e-04,  ..., 4.8180e-03,
           3.0151e-02, 6.9458e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 1.0078e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 2.3755e-01, 2.3361e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 2.7001e-05, 5.4359e-05,  ..., 3.2330e-03,
           0.0000e+00, 0.0000e+00],
          [9.5361e-01, 7.1645e-05, 4.4823e-05,  ..., 2.8782e-03,
           9.6970e-03, 0.0000e+00],
          [9.0820e-01, 1.0967e-04, 1.7285e-05,  ..., 3.7174e-03,
           4.5105e-02, 1.1147e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.3030e-02, 6.8016e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5293e-01, 1.0973e-04, 2.9743e-05,  ..., 2.5511e-04,
           0.0000e+00, 0.0000e+00],
          [6.3574e-01, 1.1864e-03, 6.1083e-04,  ..., 1.9722e-03,
           2.6718e-02, 0.0000e+00],
          [7.1436e-01, 3.7956e-04, 1.0496e-04,  ..., 1.6937e-03,
           1.3695e-02, 1.4694e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9932e-01, 1.9336e-01, 7.4959e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.1406e-01, 6.3467e-04, 2.1601e-04,  ..., 4.5013e-04,
           0.0000e+00, 0.0000e+00],
          [5.8789e-01, 3.4122e-03, 4.3225e-04,  ..., 7.0457e-03,
           2.9953e-02, 0.0000e+00],
          [4.8145e-01, 5.6763e-03, 2.9516e-04,  ..., 6.5231e-03,
           4.9438e-02, 2.0691e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9355e-01, 1.0638e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.9126e-01, 6.4746e-01, 6.1310e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8909e-01, 5.7077e-04, 1.4858e-03,  ..., 2.2385e-02,
           0.0000e+00, 0.0000e+00],
          [3.4790e-01, 3.4618e-03, 1.7891e-03,  ..., 9.7809e-03,
           7.8735e-02, 0.0000e+00],
          [6.1328e-01, 1.4267e-03, 8.6164e-04,  ..., 1.4519e-02,
           7.0190e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6172e-01, 2.3853e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 1.8054e-01, 8.2153e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9775e-01, 8.5211e-04, 6.9284e-04,  ..., 2.2202e-03,
           0.0000e+00, 0.0000e+00],
          [3.3130e-01, 6.7482e-03, 4.0779e-03,  ..., 5.2023e-04,
           1.5572e-02, 0.0000e+00],
          [5.9082e-01, 1.8854e-03, 1.0052e-03,  ..., 5.8413e-04,
           7.5760e-03, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.7449e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1602e-01, 7.2327e-02, 1.1589e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7646e-01, 2.4140e-05, 1.0002e-04,  ..., 1.0023e-03,
           0.0000e+00, 0.0000e+00],
          [1.2634e-02, 5.4240e-06, 1.3173e-05,  ..., 1.8167e-04,
           1.1021e-04, 0.0000e+00],
          [1.3664e-02, 2.1577e-05, 1.5199e-05,  ..., 2.6774e-04,
           2.7370e-04, 5.4550e-04]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 1.7456e-02, 7.4501e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 5.0962e-05, 2.0564e-05,  ..., 1.6670e-03,
           0.0000e+00, 0.0000e+00],
          [8.4082e-01, 3.1257e-04, 1.5771e-04,  ..., 2.6684e-03,
           8.6670e-03, 0.0000e+00],
          [9.1406e-01, 6.3419e-05, 1.3173e-05,  ..., 1.8902e-03,
           2.7943e-03, 3.4904e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7852e-01, 2.1271e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 9.6054e-03, 1.4503e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.0176e-01, 5.0783e-04, 9.0170e-04,  ..., 2.2461e-01,
           0.0000e+00, 0.0000e+00],
          [7.3926e-01, 3.3069e-04, 5.3549e-04,  ..., 4.0321e-03,
           1.2891e-01, 0.0000e+00],
          [2.4939e-01, 2.5196e-03, 7.6828e-03,  ..., 6.1321e-04,
           6.0539e-03, 1.6101e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.0403e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7461e-01, 2.1423e-02, 3.8376e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3291e-01, 2.2054e-05, 2.7478e-05,  ..., 1.3718e-02,
           0.0000e+00, 0.0000e+00],
          [1.6650e-01, 9.7096e-05, 1.0788e-04,  ..., 2.1782e-03,
           1.8740e-03, 0.0000e+00],
          [9.0759e-02, 4.7922e-05, 6.3717e-05,  ..., 1.4668e-03,
           6.3896e-04, 4.7340e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3311e-01, 6.6895e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 6.8262e-01, 7.6065e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.9258e-01, 2.5606e-04, 2.4533e-04,  ..., 1.0010e-02,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 6.8617e-04, 4.6253e-05,  ..., 5.0659e-03,
           6.7558e-03, 0.0000e+00],
          [8.6035e-01, 8.8549e-04, 1.8454e-04,  ..., 8.0795e-03,
           3.2898e-02, 3.6564e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1543e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 2.8534e-02, 3.5980e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7246e-01, 4.6086e-04, 2.2736e-03,  ..., 3.1952e-02,
           0.0000e+00, 0.0000e+00],
          [3.1982e-01, 3.9983e-04, 7.0429e-04,  ..., 2.1912e-02,
           1.4880e-01, 0.0000e+00],
          [6.5723e-01, 1.6463e-04, 4.4918e-04,  ..., 6.2370e-03,
           5.2948e-02, 4.5868e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6582e-01, 3.4332e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 3.0884e-01, 1.6479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.0830e-01, 2.9802e-05, 2.9087e-05,  ..., 5.5328e-02,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 4.9353e-05, 2.5809e-05,  ..., 6.5918e-03,
           1.8860e-02, 0.0000e+00],
          [4.5508e-01, 2.8431e-05, 6.5148e-05,  ..., 2.7649e-02,
           2.8003e-01, 2.5620e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 1.0901e-01, 7.2098e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.8672e-01, 1.3340e-04, 3.1638e-04,  ..., 1.2199e-02,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.3328e-04, 4.2701e-04,  ..., 4.9438e-03,
           1.2604e-02, 0.0000e+00],
          [7.1680e-01, 5.5122e-04, 9.0218e-04,  ..., 7.4043e-03,
           2.0370e-02, 1.4412e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 6.3965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1299e-01, 1.5271e-01, 3.4424e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4805e-01, 6.8617e-04, 5.6887e-04,  ..., 7.7400e-03,
           0.0000e+00, 0.0000e+00],
          [5.4590e-01, 7.1716e-04, 4.5061e-04,  ..., 2.5043e-03,
           3.8391e-02, 0.0000e+00],
          [4.1138e-01, 7.5769e-04, 5.5122e-04,  ..., 7.8888e-03,
           3.5950e-02, 2.4463e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8223e-01, 1.9971e-01, 1.7838e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.8535e-01, 3.3903e-04, 5.1141e-05,  ..., 4.3411e-03,
           0.0000e+00, 0.0000e+00],
          [9.0918e-01, 3.2973e-04, 2.6393e-04,  ..., 4.6654e-03,
           8.6975e-03, 0.0000e+00],
          [7.1631e-01, 3.0112e-04, 2.4796e-04,  ..., 1.7456e-02,
           1.3623e-01, 1.1559e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.4027e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9609e-01, 2.8114e-03, 9.2840e-04,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8555e-01, 8.1062e-06, 1.2696e-05,  ..., 1.6336e-03,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 3.6955e-06, 1.0550e-05,  ..., 8.4019e-04,
           2.7218e-03, 0.0000e+00],
          [9.8584e-01, 1.8477e-06, 4.7684e-06,  ..., 3.2544e-04,
           1.0405e-03, 9.0170e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9673e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1787e-02, 4.0829e-05, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.1765e-01, 2.1935e-05, 1.6966e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1689e-01, 1.3232e-04, 2.7156e-04,  ..., 6.6948e-04,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.3403e-03, 4.0054e-04,  ..., 3.9339e-05,
           8.3069e-02, 0.0000e+00],
          [5.6006e-01, 2.9874e-04, 6.7825e-03,  ..., 9.3222e-05,
           3.6740e-04, 8.0505e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1024e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4043e-01, 5.7312e-02, 2.0084e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2073e-01, 2.0087e-05, 6.2823e-05,  ..., 2.6672e-02,
           0.0000e+00, 0.0000e+00],
          [6.5771e-01, 3.2234e-04, 1.2970e-04,  ..., 2.0618e-03,
           1.0925e-02, 0.0000e+00],
          [6.3770e-01, 1.2201e-04, 4.3511e-05,  ..., 9.6512e-03,
           1.2482e-01, 1.4923e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.8763e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5410e-01, 3.1311e-02, 1.4671e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.3916e-01, 1.1921e-07, 4.7684e-07,  ..., 1.8148e-03,
           0.0000e+00, 0.0000e+00],
          [4.6484e-01, 1.4472e-04, 1.6689e-05,  ..., 3.3200e-05,
           8.6441e-03, 0.0000e+00],
          [2.9907e-01, 6.7949e-06, 5.8651e-05,  ..., 4.6134e-04,
           7.6675e-04, 2.6428e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.3400e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.0236e-01, 2.0950e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.3213e-01, 1.7107e-04, 7.5042e-05,  ..., 5.9938e-04,
           0.0000e+00, 0.0000e+00],
          [4.9146e-01, 1.0958e-03, 3.8099e-04,  ..., 4.3106e-03,
           2.7420e-02, 0.0000e+00],
          [7.5195e-01, 8.4496e-04, 4.6468e-04,  ..., 4.1847e-03,
           1.5030e-02, 9.9106e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9854e-01, 1.5736e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 9.3460e-03, 1.6449e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7041e-01, 3.4904e-04, 5.0783e-04,  ..., 6.4453e-02,
           0.0000e+00, 0.0000e+00],
          [4.9292e-01, 8.6308e-04, 1.3876e-03,  ..., 2.7893e-02,
           5.6244e-02, 0.0000e+00],
          [4.6387e-01, 2.5005e-03, 2.8687e-03,  ..., 2.8030e-02,
           4.1016e-02, 4.9316e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7617e-01, 4.2383e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.2412e-01, 2.5879e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1604e-02, 2.2449e-03, 8.2397e-04,  ..., 8.5144e-03,
           0.0000e+00, 0.0000e+00],
          [1.9751e-01, 1.2169e-03, 8.7452e-04,  ..., 8.4448e-04,
           1.6495e-02, 0.0000e+00],
          [2.4634e-01, 1.4076e-03, 9.1314e-04,  ..., 4.4250e-03,
           2.7069e-02, 4.2358e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9473e-01, 4.0552e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7139e-01, 4.9591e-02, 2.7881e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5867e-02, 1.7405e-05, 1.7047e-05,  ..., 8.8867e-01,
           0.0000e+00, 0.0000e+00],
          [4.4189e-01, 6.7353e-06, 3.3379e-05,  ..., 1.9093e-03,
           4.9097e-01, 0.0000e+00],
          [4.2578e-01, 5.2452e-06, 3.7134e-05,  ..., 3.0537e-03,
           3.4607e-02, 4.6582e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7725e-01, 3.2251e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3657e-01, 4.0552e-01, 3.5791e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3516e-01, 1.9395e-04, 2.3019e-04,  ..., 2.6646e-03,
           0.0000e+00, 0.0000e+00],
          [5.4077e-02, 8.0681e-04, 3.1223e-03,  ..., 1.2665e-03,
           2.2812e-02, 0.0000e+00],
          [1.4929e-01, 1.8950e-03, 4.1046e-03,  ..., 2.4090e-03,
           2.1851e-02, 3.6896e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 5.9166e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.1879e-02, 3.6087e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.4902e-01, 3.9864e-04, 4.1676e-04,  ..., 5.4054e-03,
           0.0000e+00, 0.0000e+00],
          [7.6843e-02, 6.7520e-04, 4.6611e-04,  ..., 8.4534e-03,
           4.6265e-02, 0.0000e+00],
          [4.5288e-01, 1.0509e-03, 8.3017e-04,  ..., 3.6240e-03,
           2.4384e-02, 5.3680e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9658e-01, 3.5725e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.2573e-02, 1.6556e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7427e-01, 4.9591e-05, 2.1207e-04,  ..., 3.5645e-02,
           0.0000e+00, 0.0000e+00],
          [1.2091e-01, 4.7517e-04, 8.7214e-04,  ..., 6.4354e-03,
           2.5845e-03, 0.0000e+00],
          [3.2642e-01, 3.0828e-04, 7.9775e-04,  ..., 4.4174e-03,
           6.0310e-03, 1.9257e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4549e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6865e-01, 1.1395e-01, 1.7197e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1310e-02, 2.2054e-06, 5.1856e-06,  ..., 4.1046e-03,
           0.0000e+00, 0.0000e+00],
          [2.9834e-01, 1.4734e-04, 2.0862e-04,  ..., 1.5808e-02,
           6.8169e-03, 0.0000e+00],
          [3.1152e-01, 2.2471e-05, 4.5598e-05,  ..., 2.9907e-02,
           1.5686e-02, 1.0612e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1934e-01, 1.8079e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 1.9791e-02, 3.3142e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1350e-02, 4.1366e-05, 1.7226e-04,  ..., 7.2656e-01,
           0.0000e+00, 0.0000e+00],
          [4.2700e-01, 1.0288e-04, 7.7248e-04,  ..., 2.0676e-03,
           4.8267e-01, 0.0000e+00],
          [3.5718e-01, 2.0742e-05, 2.3258e-04,  ..., 2.7523e-03,
           2.9877e-02, 5.6250e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8584e-01, 5.1416e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3628e-01, 1.7786e-01, 3.8574e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4259e-02, 3.5286e-05, 6.6936e-05,  ..., 8.8770e-01,
           0.0000e+00, 0.0000e+00],
          [1.4587e-01, 1.3125e-04, 3.7265e-04,  ..., 2.9016e-04,
           7.6611e-01, 0.0000e+00],
          [1.0394e-01, 1.3053e-04, 2.2340e-04,  ..., 1.4601e-03,
           1.3318e-01, 6.4941e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8926e-01, 1.0590e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 4.0131e-02, 1.2756e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7078e-01, 9.1314e-05, 1.1140e-04,  ..., 6.3610e-04,
           0.0000e+00, 0.0000e+00],
          [8.7891e-01, 3.1328e-04, 1.3189e-03,  ..., 5.3883e-04,
           8.9569e-03, 0.0000e+00],
          [8.1641e-01, 2.6131e-04, 9.4891e-04,  ..., 3.8567e-03,
           1.4427e-02, 8.2245e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1055e-01, 1.8921e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.7930e-01, 1.3416e-01, 8.6365e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2544e-01, 2.0683e-05, 1.8120e-05,  ..., 3.9429e-02,
           0.0000e+00, 0.0000e+00],
          [1.4320e-02, 9.8765e-05, 9.3699e-04,  ..., 4.5598e-05,
           7.5836e-03, 0.0000e+00],
          [1.9165e-01, 1.5199e-04, 5.6887e-04,  ..., 8.0729e-04,
           6.0577e-03, 1.3191e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 2.6465e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5459e-01, 5.9662e-02, 4.8584e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4290e-02, 1.0133e-06, 5.7220e-06,  ..., 9.6143e-01,
           0.0000e+00, 0.0000e+00],
          [1.4404e-01, 1.2696e-05, 8.8394e-05,  ..., 2.3918e-03,
           8.0811e-01, 0.0000e+00],
          [1.2042e-01, 7.8082e-06, 1.8895e-05,  ..., 2.3537e-03,
           3.3752e-02, 8.0566e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5508e-01, 4.4861e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2617e-01, 5.9906e-02, 1.1414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.5625e-01, 4.6611e-04, 1.7691e-04,  ..., 2.2766e-02,
           0.0000e+00, 0.0000e+00],
          [1.6797e-01, 2.1982e-04, 6.8378e-04,  ..., 9.1124e-04,
           4.9951e-01, 0.0000e+00],
          [1.6870e-01, 1.7023e-04, 6.9332e-04,  ..., 2.2087e-03,
           2.1277e-01, 2.6733e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 1.7792e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 9.8877e-03, 5.1994e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4473e-01, 6.8069e-05, 1.7166e-05,  ..., 1.3332e-03,
           0.0000e+00, 0.0000e+00],
          [8.3740e-01, 4.2510e-04, 1.2913e-03,  ..., 1.5039e-03,
           1.1263e-03, 0.0000e+00],
          [8.7988e-01, 1.8942e-04, 4.3869e-04,  ..., 2.4128e-03,
           7.1335e-04, 2.0199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4351e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 4.6768e-03, 2.6245e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1228e-01, 3.1924e-04, 3.3069e-04,  ..., 7.8735e-03,
           0.0000e+00, 0.0000e+00],
          [7.4707e-01, 4.4227e-04, 7.5531e-04,  ..., 1.3332e-03,
           1.5869e-02, 0.0000e+00],
          [4.3018e-01, 9.6893e-04, 7.5150e-04,  ..., 2.6226e-03,
           3.8025e-02, 2.9739e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3071e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4014e-01, 2.9126e-01, 6.8481e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0605e-01, 2.4617e-05, 1.2982e-04,  ..., 9.9564e-03,
           0.0000e+00, 0.0000e+00],
          [9.3945e-01, 9.3877e-05, 1.0353e-04,  ..., 1.6842e-03,
           1.0338e-02, 0.0000e+00],
          [7.6611e-01, 3.7372e-05, 6.1631e-05,  ..., 1.7910e-03,
           9.2834e-02, 1.8341e-02]],

         [[8.4814e-01, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3756e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0801e-01, 1.7929e-03, 2.6283e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7031e-01, 2.7776e-04, 5.1117e-04,  ..., 1.7838e-02,
           0.0000e+00, 0.0000e+00],
          [6.3477e-01, 8.8072e-04, 2.7943e-03,  ..., 6.0499e-05,
           1.5671e-02, 0.0000e+00],
          [6.3330e-01, 4.7302e-04, 1.1730e-03,  ..., 5.7745e-04,
           2.5520e-03, 3.3173e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8340e-01, 1.6479e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4502e-01, 3.3179e-01, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.8467e-01, 2.9385e-05, 1.5497e-06,  ..., 2.1713e-02,
           0.0000e+00, 0.0000e+00],
          [7.5244e-01, 1.4746e-04, 5.6863e-05,  ..., 1.2091e-01,
           5.4993e-02, 0.0000e+00],
          [4.6704e-01, 5.2452e-06, 5.4896e-05,  ..., 9.9411e-03,
           4.4922e-01, 6.9092e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9365e-01, 6.1569e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 1.1719e-02, 1.2405e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7437e-01, 1.4515e-03, 1.6584e-03,  ..., 5.3635e-03,
           0.0000e+00, 0.0000e+00],
          [6.8262e-01, 3.9024e-03, 5.1613e-03,  ..., 1.5554e-03,
           3.5801e-03, 0.0000e+00],
          [7.0068e-01, 1.6909e-03, 3.3092e-03,  ..., 3.2654e-03,
           7.6065e-03, 1.9516e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7900e-01, 2.0920e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9971e-01, 2.8149e-01, 1.8967e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.1826e-01, 3.2663e-05, 3.0696e-05,  ..., 3.0029e-02,
           0.0000e+00, 0.0000e+00],
          [8.8379e-01, 9.1314e-05, 5.6505e-05,  ..., 5.1453e-02,
           2.7679e-02, 0.0000e+00],
          [7.5781e-01, 3.3617e-05, 6.1095e-05,  ..., 8.0383e-02,
           7.8125e-02, 1.8143e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.3940e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8242e-01, 8.9722e-03, 8.7738e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 6.7890e-05, 1.7107e-05,  ..., 6.9656e-03,
           0.0000e+00, 0.0000e+00],
          [3.7109e-01, 1.6394e-03, 1.2894e-03,  ..., 6.5193e-03,
           1.8753e-02, 0.0000e+00],
          [7.4268e-01, 6.4278e-04, 2.2960e-04,  ..., 5.7411e-03,
           1.1101e-03, 5.6305e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 1.9958e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8145e-01, 9.2239e-03, 9.2010e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.8684e-02, 2.3496e-04, 3.6311e-04,  ..., 5.5695e-03,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 2.6751e-04, 8.2731e-05,  ..., 1.9133e-05,
           1.7529e-03, 0.0000e+00],
          [9.1699e-01, 3.1805e-04, 1.5104e-04,  ..., 2.7847e-04,
           8.2321e-03, 1.6754e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 4.8157e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5166e-01, 3.2043e-02, 1.6205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8833e-01, 2.7061e-04, 1.5426e-04,  ..., 1.3733e-02,
           0.0000e+00, 0.0000e+00],
          [9.1992e-01, 2.3327e-03, 5.4264e-04,  ..., 3.8147e-05,
           4.8370e-03, 0.0000e+00],
          [5.8740e-01, 1.7452e-03, 6.4516e-04,  ..., 4.7016e-04,
           6.7078e-02, 4.1748e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 2.5131e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0322e-01, 1.5112e-01, 4.5593e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.0312e-01, 5.7936e-05, 5.5492e-05,  ..., 3.1143e-02,
           0.0000e+00, 0.0000e+00],
          [1.8848e-01, 2.6608e-04, 2.7466e-04,  ..., 7.1655e-02,
           8.5938e-02, 0.0000e+00],
          [3.4033e-01, 8.7857e-05, 1.3554e-04,  ..., 6.2683e-02,
           6.1646e-02, 6.0150e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9272e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 8.3252e-02, 2.6749e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.6113e-01, 2.6166e-05, 1.4365e-05,  ..., 3.9276e-02,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 6.6948e-04, 9.4461e-04,  ..., 7.1594e-02,
           3.6346e-02, 0.0000e+00],
          [7.3877e-01, 3.4642e-04, 2.6774e-04,  ..., 3.2867e-02,
           7.9346e-03, 7.5684e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.2797e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9941e-01, 6.1523e-02, 3.8879e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0400e-01, 4.3631e-05, 1.0425e-04,  ..., 3.0319e-02,
           0.0000e+00, 0.0000e+00],
          [8.1494e-01, 1.2999e-03, 3.9978e-03,  ..., 3.5839e-03,
           1.9257e-02, 0.0000e+00],
          [7.3340e-01, 4.8923e-04, 9.4652e-04,  ..., 3.1338e-03,
           5.0446e-02, 5.6427e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.0303e-01, 3.9697e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4221e-01, 3.9337e-02, 8.1836e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.7842e-02, 6.7353e-06, 2.9504e-05,  ..., 8.4814e-01,
           0.0000e+00, 0.0000e+00],
          [4.2786e-02, 1.9729e-05, 7.5281e-05,  ..., 1.4610e-03,
           9.3652e-01, 0.0000e+00],
          [6.3721e-02, 3.8743e-06, 3.0339e-05,  ..., 1.8616e-03,
           2.9404e-02, 8.7988e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 1.8591e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3384e-01, 5.2148e-01, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2583e-01, 7.0751e-05, 6.4909e-05,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [7.2070e-01, 3.2783e-04, 1.1355e-04,  ..., 2.7969e-02,
           5.9753e-02, 0.0000e+00],
          [4.5215e-01, 1.0467e-04, 6.5386e-05,  ..., 1.9348e-02,
           1.2854e-01, 2.0947e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6025e-01, 2.3975e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9600e-01, 3.7933e-02, 6.6162e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.3481e-01, 3.3970e-03, 3.4695e-03,  ..., 1.1940e-02,
           0.0000e+00, 0.0000e+00],
          [5.0244e-01, 2.7161e-02, 3.9612e-02,  ..., 2.0828e-03,
           4.4098e-02, 0.0000e+00],
          [5.2881e-01, 8.4763e-03, 1.0414e-02,  ..., 4.0359e-03,
           2.7756e-02, 8.0017e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6826e-01, 3.1921e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6621e-01, 9.6008e-02, 3.7659e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9385e-01, 1.1683e-05, 8.5235e-06,  ..., 1.1147e-02,
           0.0000e+00, 0.0000e+00],
          [2.7637e-01, 7.2975e-03, 4.6005e-03,  ..., 2.8496e-03,
           6.8016e-03, 0.0000e+00],
          [2.9980e-01, 8.8263e-04, 7.1049e-04,  ..., 5.3368e-03,
           1.2199e-02, 3.0777e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8047e-01, 1.9516e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9121e-01, 4.6806e-03, 4.1695e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9590e-01, 2.6882e-05, 3.3140e-05,  ..., 7.3929e-03,
           0.0000e+00, 0.0000e+00],
          [3.5767e-01, 5.5084e-03, 3.3417e-03,  ..., 2.5868e-05,
           2.0798e-02, 0.0000e+00],
          [7.7979e-01, 5.6791e-04, 6.1655e-04,  ..., 5.0449e-04,
           3.3665e-03, 1.4374e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9185e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.5375e-02, 1.3771e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2189e-01, 2.9325e-05, 8.8811e-06,  ..., 2.8305e-03,
           0.0000e+00, 0.0000e+00],
          [2.9831e-02, 4.4918e-04, 4.0293e-04,  ..., 4.7326e-04,
           1.0452e-02, 0.0000e+00],
          [6.4148e-02, 1.5223e-04, 1.0306e-04,  ..., 4.7951e-03,
           4.9400e-04, 5.5199e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0820e-01, 9.2041e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6938e-01, 4.2505e-01, 2.0557e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0015e-01, 1.0192e-05, 6.6757e-06,  ..., 5.6953e-03,
           0.0000e+00, 0.0000e+00],
          [1.7822e-01, 3.3021e-04, 3.8624e-04,  ..., 7.3853e-03,
           3.6346e-02, 0.0000e+00],
          [3.0176e-01, 3.0756e-04, 3.2878e-04,  ..., 5.8861e-03,
           5.2032e-02, 4.1504e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6875e-01, 3.1250e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 2.0859e-02, 2.0172e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.0654e-01, 1.1480e-04, 6.2346e-05,  ..., 6.5002e-02,
           0.0000e+00, 0.0000e+00],
          [4.4873e-01, 2.7637e-03, 4.4708e-03,  ..., 3.7785e-03,
           1.0254e-01, 0.0000e+00],
          [4.8560e-01, 1.6909e-03, 1.8425e-03,  ..., 7.4310e-03,
           2.7710e-02, 5.9357e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.1982e-01, 1.8005e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.4075e-01, 6.6895e-01, 1.9019e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.2571e-02, 3.1590e-05, 3.2783e-06,  ..., 2.4323e-02,
           0.0000e+00, 0.0000e+00],
          [2.5955e-02, 4.2772e-04, 3.8195e-04,  ..., 3.9787e-03,
           1.4000e-02, 0.0000e+00],
          [7.0534e-03, 3.0279e-04, 6.9737e-05,  ..., 7.6866e-03,
           6.2752e-03, 1.2383e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9268e-01, 7.4883e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 2.5894e-02, 4.5700e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.7876e-01, 8.5831e-06, 1.0669e-05,  ..., 1.1864e-03,
           0.0000e+00, 0.0000e+00],
          [3.5181e-01, 9.8610e-04, 8.5163e-04,  ..., 4.8714e-03,
           7.3280e-03, 0.0000e+00],
          [3.6182e-01, 4.0913e-04, 3.0494e-04,  ..., 4.5395e-03,
           6.0577e-03, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5312e-01, 4.7089e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9512e-01, 3.7671e-01, 1.2817e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8197e-02, 7.2718e-06, 3.8147e-06,  ..., 5.8517e-03,
           0.0000e+00, 0.0000e+00],
          [5.8008e-01, 1.2808e-03, 3.0422e-03,  ..., 7.8821e-04,
           6.6833e-02, 0.0000e+00],
          [1.1591e-01, 8.4591e-04, 8.6594e-04,  ..., 1.4477e-03,
           4.3976e-02, 3.5889e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5205e-01, 1.4807e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 3.8544e-02, 2.7390e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0225e-01, 5.3704e-05, 6.3956e-05,  ..., 3.2654e-03,
           0.0000e+00, 0.0000e+00],
          [3.5474e-01, 2.9697e-03, 1.7807e-02,  ..., 2.8753e-04,
           1.0864e-01, 0.0000e+00],
          [5.7129e-01, 2.0599e-03, 5.7335e-03,  ..., 7.6914e-04,
           1.6785e-02, 4.0283e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1113e-01, 8.8989e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 3.6831e-03, 1.8196e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5854e-01, 1.5478e-03, 2.1458e-03,  ..., 2.2507e-03,
           0.0000e+00, 0.0000e+00],
          [8.1396e-01, 3.7785e-03, 1.7462e-03,  ..., 3.5524e-05,
           7.5134e-02, 0.0000e+00],
          [1.2891e-01, 2.1935e-03, 2.9202e-03,  ..., 5.0366e-05,
           5.3613e-01, 2.4506e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7695e-01, 1.2292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 2.3941e-02, 1.5205e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.0039e-02, 4.0829e-05, 3.5167e-05,  ..., 9.9659e-04,
           0.0000e+00, 0.0000e+00],
          [6.1914e-01, 4.3945e-03, 2.2850e-03,  ..., 3.5834e-04,
           2.0203e-02, 0.0000e+00],
          [2.9346e-01, 2.2659e-03, 2.2411e-03,  ..., 1.8721e-03,
           6.6650e-02, 5.1392e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0430e-01, 9.5520e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 2.3438e-01, 3.5248e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6450e-01, 1.9193e-05, 1.6570e-05,  ..., 1.6418e-02,
           0.0000e+00, 0.0000e+00],
          [9.5154e-02, 8.0729e-04, 7.1812e-04,  ..., 1.3336e-02,
           2.1072e-02, 0.0000e+00],
          [2.7637e-01, 5.4979e-04, 7.0572e-04,  ..., 1.3496e-02,
           3.4729e-02, 5.2765e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 5.6458e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2578e-01, 4.1870e-02, 3.2440e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8301e-01, 3.9482e-04, 2.3222e-04,  ..., 2.5421e-02,
           0.0000e+00, 0.0000e+00],
          [4.8047e-01, 8.4610e-03, 1.5152e-02,  ..., 2.6627e-03,
           2.4490e-02, 0.0000e+00],
          [6.0059e-01, 2.7161e-03, 3.7785e-03,  ..., 6.0387e-03,
           2.3560e-02, 5.4352e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 1.1539e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 7.8142e-05, 2.1954e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7803e-01, 5.6446e-05, 8.7738e-05,  ..., 4.8399e-04,
           0.0000e+00, 0.0000e+00],
          [9.4336e-01, 9.0599e-05, 3.1495e-04,  ..., 7.8678e-05,
           4.3091e-02, 0.0000e+00],
          [9.9023e-01, 1.3995e-04, 7.5483e-04,  ..., 3.1590e-05,
           5.1594e-04, 1.7796e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9707e-01, 3.1643e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9414e-01, 3.2711e-03, 2.4147e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9541e-01, 4.4727e-04, 5.5075e-04,  ..., 2.0466e-03,
           0.0000e+00, 0.0000e+00],
          [8.5596e-01, 8.0633e-04, 1.2493e-03,  ..., 1.3151e-03,
           1.7593e-02, 0.0000e+00],
          [8.8428e-01, 2.3603e-04, 4.8137e-04,  ..., 1.2245e-03,
           6.9504e-03, 1.3878e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6094e-01, 3.9062e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4775e-01, 2.2293e-02, 2.9999e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0419e-01, 1.7099e-03, 8.1396e-04,  ..., 2.4353e-02,
           0.0000e+00, 0.0000e+00],
          [8.4814e-01, 5.5389e-03, 4.0512e-03,  ..., 4.6074e-05,
           4.0283e-02, 0.0000e+00],
          [3.5205e-01, 8.5449e-03, 3.7918e-03,  ..., 3.6669e-04,
           1.0248e-01, 9.0454e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0576e-01, 9.3994e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3936e-01, 1.2573e-01, 3.4668e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.2227e-01, 2.1210e-03, 8.9788e-04,  ..., 1.0201e-02,
           0.0000e+00, 0.0000e+00],
          [1.3062e-01, 9.1019e-03, 1.8196e-03,  ..., 5.8055e-05,
           6.5674e-02, 0.0000e+00],
          [6.8787e-02, 5.7831e-03, 1.5202e-03,  ..., 4.4250e-04,
           1.1981e-01, 1.3062e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 4.1479e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1812e-01, 2.0068e-01, 4.8120e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1179e-02, 1.8477e-06, 1.4305e-06,  ..., 6.4355e-01,
           0.0000e+00, 0.0000e+00],
          [7.6538e-02, 1.3266e-03, 2.3289e-03,  ..., 3.0384e-03,
           6.7676e-01, 0.0000e+00],
          [3.7598e-02, 8.2910e-05, 1.4901e-04,  ..., 1.6556e-02,
           1.0712e-01, 5.8838e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6436e-01, 3.5553e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0186e-01, 5.6519e-02, 4.1840e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2539e-01, 2.4676e-05, 1.6332e-05,  ..., 2.0142e-02,
           0.0000e+00, 0.0000e+00],
          [2.8516e-01, 1.3571e-03, 2.3270e-03,  ..., 3.5896e-03,
           7.4341e-02, 0.0000e+00],
          [2.5488e-01, 2.4080e-04, 3.0136e-04,  ..., 9.6512e-03,
           5.3864e-02, 6.3477e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2041e-01, 7.9773e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9336e-01, 2.0984e-01, 9.6802e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1758e-01, 6.0797e-06, 3.8147e-06,  ..., 1.2070e-02,
           0.0000e+00, 0.0000e+00],
          [2.3022e-01, 6.8009e-05, 8.2672e-05,  ..., 2.5436e-02,
           3.6713e-02, 0.0000e+00],
          [4.2017e-01, 5.3823e-05, 5.3585e-05,  ..., 5.6854e-02,
           1.5549e-02, 6.8604e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.0898e-01, 2.9102e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1074e-01, 1.7444e-01, 3.1470e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2852e-01, 1.2159e-05, 6.9737e-06,  ..., 2.7783e-01,
           0.0000e+00, 0.0000e+00],
          [3.7280e-01, 1.2054e-03, 1.2236e-03,  ..., 2.8019e-03,
           3.7280e-01, 0.0000e+00],
          [2.8638e-01, 3.6240e-04, 3.4046e-04,  ..., 3.8147e-02,
           9.0759e-02, 3.8818e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2104e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 1.5945e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.5059e-01, 3.5703e-05, 1.3351e-05,  ..., 5.2986e-03,
           0.0000e+00, 0.0000e+00],
          [4.9390e-01, 6.7215e-03, 3.2074e-02,  ..., 4.0591e-05,
           1.7615e-01, 0.0000e+00],
          [7.5732e-01, 1.5802e-03, 4.0359e-03,  ..., 2.0730e-04,
           3.2257e-02, 3.1250e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6904e-01, 2.3096e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.8496e-01, 2.9419e-01, 1.2073e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8384e-01, 3.8028e-05, 3.7432e-05,  ..., 1.3115e-02,
           0.0000e+00, 0.0000e+00],
          [3.2983e-01, 2.7657e-03, 3.3894e-03,  ..., 4.9591e-04,
           3.2983e-01, 0.0000e+00],
          [1.8567e-01, 5.5552e-04, 4.2582e-04,  ..., 1.7643e-03,
           1.5881e-01, 1.3379e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.6982e-01, 4.3018e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9941e-01, 6.3171e-02, 5.3760e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6294e-02, 2.8551e-05, 1.1563e-05,  ..., 7.8271e-01,
           0.0000e+00, 0.0000e+00],
          [1.4319e-01, 1.1282e-03, 1.3189e-03,  ..., 2.8062e-04,
           7.9834e-01, 0.0000e+00],
          [3.3862e-01, 2.2244e-04, 3.6097e-04,  ..., 2.0771e-03,
           1.1340e-01, 4.9268e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 1.4229e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 7.4196e-03, 5.3024e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.4153e-02, 1.3828e-05, 2.9802e-06,  ..., 1.2760e-03,
           0.0000e+00, 0.0000e+00],
          [5.6152e-01, 3.8818e-02, 1.7090e-02,  ..., 2.4438e-05,
           1.1145e-01, 0.0000e+00],
          [3.9551e-01, 1.4748e-02, 8.8043e-03,  ..., 6.8998e-04,
           2.8870e-02, 1.3147e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9023e-01, 9.5596e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 6.2904e-03, 3.7556e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6853e-02, 2.1338e-05, 9.7752e-06,  ..., 7.0333e-04,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 2.2755e-03, 2.8534e-03,  ..., 6.1214e-05,
           3.0930e-02, 0.0000e+00],
          [6.6650e-01, 1.4524e-03, 1.6136e-03,  ..., 4.7874e-04,
           2.4078e-02, 2.7298e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 2.9205e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 2.2842e-02, 9.3765e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7896e-01, 3.7789e-04, 2.2304e-04,  ..., 5.7068e-03,
           0.0000e+00, 0.0000e+00],
          [4.0869e-01, 9.6054e-03, 9.7580e-03,  ..., 2.3222e-04,
           9.4788e-02, 0.0000e+00],
          [3.8306e-01, 5.2338e-03, 5.8403e-03,  ..., 7.3957e-04,
           6.0608e-02, 2.1606e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5654e-01, 4.3518e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 2.9694e-02, 2.4429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.6895e-01, 1.2517e-05, 1.1802e-05,  ..., 6.3110e-02,
           0.0000e+00, 0.0000e+00],
          [4.5386e-01, 3.1972e-04, 3.5954e-04,  ..., 1.2383e-02,
           6.3843e-02, 0.0000e+00],
          [2.8516e-01, 3.5262e-04, 2.5296e-04,  ..., 1.8097e-02,
           9.4788e-02, 1.7163e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5195e-01, 2.4805e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7266e-01, 1.7960e-02, 9.2468e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.6533e-01, 3.9935e-05, 2.2769e-05,  ..., 2.9087e-03,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 2.7790e-03, 1.3542e-03,  ..., 4.8370e-03,
           1.8402e-02, 0.0000e+00],
          [9.3652e-01, 2.0485e-03, 5.7364e-04,  ..., 6.4125e-03,
           1.2848e-02, 6.9847e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5488e-01, 2.4512e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 1.5312e-02, 7.0686e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7188e-01, 3.9637e-05, 1.4961e-05,  ..., 1.7761e-02,
           0.0000e+00, 0.0000e+00],
          [8.4229e-01, 3.8395e-03, 3.4161e-03,  ..., 1.6037e-02,
           3.7018e-02, 0.0000e+00],
          [8.4668e-01, 9.9945e-04, 5.5170e-04,  ..., 1.1986e-02,
           1.6769e-02, 1.4458e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4727e-01, 5.2612e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 5.1788e-02, 4.4312e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.7126e-01, 2.5005e-03, 3.7251e-03,  ..., 1.6174e-02,
           0.0000e+00, 0.0000e+00],
          [2.4719e-01, 4.8943e-03, 5.7220e-03,  ..., 1.2350e-04,
           2.2864e-01, 0.0000e+00],
          [1.9580e-01, 1.8167e-03, 2.6436e-03,  ..., 5.9462e-04,
           3.4888e-01, 1.5491e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2285e-01, 7.6965e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9014e-01, 7.0801e-02, 3.9093e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1562e-01, 3.0518e-05, 4.6909e-05,  ..., 3.3508e-02,
           0.0000e+00, 0.0000e+00],
          [1.8628e-01, 1.5869e-03, 9.0408e-04,  ..., 1.2878e-02,
           4.1565e-02, 0.0000e+00],
          [3.2349e-01, 1.3506e-04, 1.0270e-04,  ..., 6.1737e-02,
           6.2561e-03, 8.1787e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8682e-01, 1.3222e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5850e-01, 1.9882e-02, 2.1835e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3076e-01, 1.5745e-03, 3.5477e-03,  ..., 3.1548e-03,
           0.0000e+00, 0.0000e+00],
          [4.9268e-01, 6.1035e-03, 1.0147e-02,  ..., 5.8126e-04,
           2.7573e-02, 0.0000e+00],
          [3.7793e-01, 3.9749e-03, 8.4152e-03,  ..., 9.2936e-04,
           7.0984e-02, 6.1676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 1.5778e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7510e-01, 9.3384e-03, 1.5396e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7810e-02, 1.3888e-05, 1.4484e-05,  ..., 1.5610e-02,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 7.9584e-04, 2.1973e-03,  ..., 2.3532e-04,
           1.1450e-01, 0.0000e+00],
          [4.2407e-01, 8.5115e-04, 2.0103e-03,  ..., 2.5215e-03,
           2.6489e-02, 3.8849e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3359e-01, 6.6589e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3457e-01, 3.5675e-02, 2.9800e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1536e-01, 4.5598e-05, 7.8797e-05,  ..., 7.1449e-03,
           0.0000e+00, 0.0000e+00],
          [4.4360e-01, 1.9302e-03, 3.6621e-03,  ..., 4.4346e-05,
           1.2512e-01, 0.0000e+00],
          [1.9507e-01, 7.9107e-04, 2.2888e-03,  ..., 5.5647e-04,
           4.8553e-02, 1.2018e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.8982e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 2.3163e-02, 2.9968e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.9580e-02, 4.7684e-06, 1.0788e-05,  ..., 4.1771e-03,
           0.0000e+00, 0.0000e+00],
          [5.6396e-01, 8.4782e-04, 2.0504e-03,  ..., 4.6086e-04,
           1.2988e-01, 0.0000e+00],
          [4.4458e-01, 6.0415e-04, 2.2621e-03,  ..., 1.9951e-03,
           5.5664e-02, 2.8656e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9590e-01, 2.0435e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4170e-01, 1.4832e-01, 1.1023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.5969e-02, 5.1022e-05, 3.8505e-05,  ..., 1.6541e-02,
           0.0000e+00, 0.0000e+00],
          [9.4238e-02, 1.5717e-03, 2.9812e-03,  ..., 4.1628e-04,
           3.8666e-02, 0.0000e+00],
          [9.1629e-03, 1.8430e-04, 3.9029e-04,  ..., 2.7943e-03,
           8.6060e-03, 9.1125e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.3697e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1064e-01, 3.6438e-02, 5.3009e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.9820e-03, 9.4771e-06, 2.1458e-06,  ..., 9.8133e-04,
           0.0000e+00, 0.0000e+00],
          [6.6064e-01, 1.0242e-03, 1.2951e-03,  ..., 1.7250e-04,
           3.0411e-02, 0.0000e+00],
          [5.1758e-01, 1.3876e-03, 1.4544e-03,  ..., 5.3177e-03,
           6.9389e-03, 1.0034e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7705e-01, 2.3148e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7119e-01, 1.2321e-02, 1.6708e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4834e-01, 4.3333e-05, 4.2319e-05,  ..., 3.9940e-03,
           0.0000e+00, 0.0000e+00],
          [5.9619e-01, 1.0309e-03, 2.2526e-03,  ..., 3.0971e-04,
           6.7253e-03, 0.0000e+00],
          [6.5918e-01, 1.6994e-03, 2.9125e-03,  ..., 2.6679e-04,
           6.5651e-03, 1.5747e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.5693e-01, 8.1604e-02, 6.1584e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1462e-02, 1.0133e-06, 7.7486e-07,  ..., 1.9150e-03,
           0.0000e+00, 0.0000e+00],
          [2.4402e-01, 7.3528e-04, 7.2432e-04,  ..., 1.7047e-05,
           2.3056e-02, 0.0000e+00],
          [2.7441e-01, 2.4629e-04, 2.7800e-04,  ..., 1.2407e-03,
           1.5358e-02, 1.2366e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8535e-01, 1.4618e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 2.4612e-02, 3.7231e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5288e-01, 1.4126e-05, 2.5272e-05,  ..., 6.4621e-03,
           0.0000e+00, 0.0000e+00],
          [3.1274e-01, 8.4496e-04, 1.1816e-03,  ..., 6.9141e-06,
           1.0559e-01, 0.0000e+00],
          [1.7444e-01, 3.1638e-04, 6.9666e-04,  ..., 1.7071e-04,
           1.7310e-01, 5.1971e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3643e-01, 1.6345e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 4.1565e-02, 1.6821e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4749e-02, 2.3961e-05, 3.7551e-05,  ..., 3.5962e-01,
           0.0000e+00, 0.0000e+00],
          [5.1367e-01, 2.5272e-04, 6.5041e-04,  ..., 7.4148e-05,
           2.3157e-01, 0.0000e+00],
          [4.0527e-01, 9.2685e-05, 3.5524e-04,  ..., 1.0281e-03,
           5.4413e-02, 4.0527e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6973e-01, 3.0334e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2139e-01, 2.2186e-02, 5.6213e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1558e-01, 1.1325e-05, 9.8348e-06,  ..., 5.4398e-03,
           0.0000e+00, 0.0000e+00],
          [4.7632e-01, 4.8280e-04, 1.6270e-03,  ..., 1.4210e-04,
           5.8228e-02, 0.0000e+00],
          [7.3438e-01, 2.7370e-04, 6.4421e-04,  ..., 1.5631e-03,
           2.0035e-02, 4.8065e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5752e-01, 4.2389e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.0071e-02, 2.4933e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7295e-01, 9.0599e-06, 3.4213e-05,  ..., 8.5068e-03,
           0.0000e+00, 0.0000e+00],
          [6.2012e-01, 5.7888e-04, 3.1281e-03,  ..., 7.9572e-05,
           4.2175e-02, 0.0000e+00],
          [7.6953e-01, 2.3699e-04, 2.1458e-03,  ..., 4.3583e-04,
           6.2065e-03, 1.7548e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7754e-01, 2.2293e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 1.7273e-02, 4.6600e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5332e-01, 4.1723e-06, 5.0664e-06,  ..., 2.9888e-03,
           0.0000e+00, 0.0000e+00],
          [3.0103e-01, 3.6163e-03, 1.8951e-02,  ..., 1.1998e-04,
           5.0720e-02, 0.0000e+00],
          [5.7227e-01, 3.3779e-03, 8.6899e-03,  ..., 9.3079e-04,
           1.0239e-02, 3.9276e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5459e-01, 4.5349e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1699e-01, 4.1901e-02, 4.0924e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.9651e-02, 2.4974e-05, 1.5259e-05,  ..., 9.6664e-03,
           0.0000e+00, 0.0000e+00],
          [3.6499e-01, 1.8272e-03, 3.1204e-03,  ..., 1.0090e-03,
           7.5928e-02, 0.0000e+00],
          [4.6411e-01, 6.3515e-04, 5.6934e-04,  ..., 9.3689e-03,
           1.6388e-02, 4.4525e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 3.2837e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5898e-01, 1.8631e-02, 2.2385e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.6365e-02, 1.5736e-05, 7.3910e-06,  ..., 4.3091e-02,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 4.6635e-04, 9.4604e-04,  ..., 3.8362e-04,
           1.4122e-02, 0.0000e+00],
          [9.8511e-02, 9.9421e-05, 9.1255e-05,  ..., 2.4139e-02,
           3.2158e-03, 2.1805e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6289e-01, 3.6896e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5605e-01, 1.5640e-02, 2.8427e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9241e-02, 2.9802e-07, 1.1921e-07,  ..., 8.9765e-05,
           0.0000e+00, 0.0000e+00],
          [5.4541e-01, 2.1172e-03, 4.0665e-03,  ..., 3.0637e-04,
           4.6082e-03, 0.0000e+00],
          [5.3662e-01, 2.1420e-03, 3.4637e-03,  ..., 2.3441e-03,
           6.9733e-03, 2.0981e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9219e-01, 7.6370e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8232e-01, 1.0956e-01, 8.1253e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0625e-01, 1.9312e-05, 2.3842e-06,  ..., 2.0340e-02,
           0.0000e+00, 0.0000e+00],
          [6.5332e-01, 6.7055e-05, 4.8161e-05,  ..., 9.4116e-02,
           2.2354e-02, 0.0000e+00],
          [6.1279e-01, 6.9737e-06, 1.1003e-04,  ..., 9.6924e-02,
           1.0986e-01, 3.1219e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5801e-01, 4.2084e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 6.5430e-02, 5.9113e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9766e-01, 3.5524e-05, 1.2350e-04,  ..., 6.7627e-02,
           0.0000e+00, 0.0000e+00],
          [5.2002e-01, 3.7503e-04, 1.2884e-03,  ..., 2.0157e-02,
           3.5370e-02, 0.0000e+00],
          [4.6948e-01, 8.4937e-05, 2.5558e-04,  ..., 7.3120e-02,
           6.7627e-02, 1.6449e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3799e-01, 6.1890e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7095e-01, 4.7461e-01, 5.4504e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4551e-01, 1.3769e-04, 5.2869e-05,  ..., 8.4656e-02,
           0.0000e+00, 0.0000e+00],
          [5.7861e-01, 2.4533e-04, 2.4343e-04,  ..., 2.3132e-02,
           9.3689e-02, 0.0000e+00],
          [5.0049e-01, 5.1618e-05, 3.6693e-04,  ..., 3.5431e-02,
           2.8516e-01, 9.7046e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9443e-01, 2.0557e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9395e-01, 7.4158e-02, 1.3171e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8687e-01, 4.1127e-06, 7.3910e-06,  ..., 2.2995e-02,
           0.0000e+00, 0.0000e+00],
          [4.9683e-02, 4.5395e-04, 6.8426e-04,  ..., 2.4676e-04,
           3.7781e-02, 0.0000e+00],
          [2.8351e-02, 8.2493e-05, 3.2616e-04,  ..., 4.1466e-03,
           6.5369e-02, 8.8684e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.3579e-01, 5.6396e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1798e-01, 3.7109e-02, 8.4473e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.5732e-02, 2.0266e-05, 5.9903e-05,  ..., 5.3320e-01,
           0.0000e+00, 0.0000e+00],
          [7.4951e-02, 1.6975e-04, 4.6806e-03,  ..., 9.7504e-03,
           7.2803e-01, 0.0000e+00],
          [1.4099e-02, 9.6560e-06, 1.3709e-04,  ..., 4.1656e-03,
           1.9424e-02, 8.9990e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9746e-01, 5.1025e-02, 5.1422e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.3450e-01, 8.0645e-05, 3.5667e-04,  ..., 2.0523e-03,
           0.0000e+00, 0.0000e+00],
          [5.3271e-01, 1.9369e-03, 5.4741e-03,  ..., 1.0133e-06,
           1.1147e-02, 0.0000e+00],
          [1.1969e-01, 3.5357e-04, 1.4715e-03,  ..., 2.9802e-06,
           4.9286e-02, 1.5762e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3652e-01, 6.3721e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7451e-01, 8.0078e-02, 4.5349e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4150e-01, 4.5943e-04, 7.5245e-04,  ..., 3.0975e-03,
           0.0000e+00, 0.0000e+00],
          [5.3906e-01, 2.9011e-03, 4.7455e-03,  ..., 3.1986e-03,
           7.6721e-02, 0.0000e+00],
          [2.6514e-01, 3.3703e-03, 6.2599e-03,  ..., 2.6245e-03,
           2.7246e-01, 1.1182e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 2.4231e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6729e-01, 1.9073e-02, 1.3664e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7051e-01, 3.7551e-06, 7.1526e-06,  ..., 3.5065e-02,
           0.0000e+00, 0.0000e+00],
          [3.8818e-01, 2.1946e-04, 3.0231e-04,  ..., 1.4992e-02,
           2.9465e-02, 0.0000e+00],
          [1.4966e-01, 8.6665e-05, 1.0997e-04,  ..., 1.2871e-02,
           1.1009e-02, 6.6406e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 5.9326e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.7480e-01, 2.3877e-01, 8.6182e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.8350e-01, 7.1645e-05, 5.6565e-05,  ..., 1.6455e-01,
           0.0000e+00, 0.0000e+00],
          [3.2397e-01, 4.9639e-04, 5.0449e-04,  ..., 7.2815e-02,
           1.6675e-01, 0.0000e+00],
          [1.5173e-01, 4.5180e-04, 7.4053e-04,  ..., 1.3260e-02,
           4.8779e-01, 1.7053e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 2.1744e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4385e-01, 2.0721e-02, 3.5614e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.0918e-01, 1.2314e-04, 4.3511e-04,  ..., 2.9516e-04,
           0.0000e+00, 0.0000e+00],
          [8.8428e-01, 2.8133e-04, 7.0190e-04,  ..., 3.1567e-04,
           2.4796e-02, 0.0000e+00],
          [8.0322e-01, 1.5533e-04, 7.5722e-04,  ..., 6.3133e-04,
           5.1361e-02, 4.0009e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9902e-01, 9.3651e-04, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9756e-01, 3.5620e-04, 1.9112e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.8877e-01, 6.7353e-06, 2.6643e-05,  ..., 9.9599e-05,
           0.0000e+00, 0.0000e+00],
          [9.7607e-01, 2.5010e-04, 1.6756e-03,  ..., 2.5105e-04,
           2.6569e-03, 0.0000e+00],
          [9.7803e-01, 8.9347e-05, 5.0402e-04,  ..., 1.9288e-04,
           5.1308e-03, 3.3646e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2672e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 1.0696e-02, 1.5381e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2666e-01, 2.4533e-04, 1.6367e-04,  ..., 1.1768e-03,
           0.0000e+00, 0.0000e+00],
          [7.2168e-01, 1.3199e-03, 6.1989e-03,  ..., 3.9406e-03,
           2.2842e-02, 0.0000e+00],
          [3.6719e-01, 5.2691e-04, 2.2964e-03,  ..., 8.4381e-03,
           1.5396e-02, 1.4050e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8633e-01, 1.3634e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8770e-01, 5.2887e-02, 5.9479e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2422e-01, 3.0279e-05, 2.2173e-05,  ..., 4.9782e-03,
           0.0000e+00, 0.0000e+00],
          [5.3467e-01, 4.6158e-04, 8.5592e-04,  ..., 3.6316e-03,
           5.2948e-02, 0.0000e+00],
          [2.7100e-01, 1.8227e-04, 2.4724e-04,  ..., 8.9874e-03,
           3.0899e-02, 2.9251e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8096e-01, 1.9119e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7998e-01, 8.5449e-03, 1.1497e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5498e-01, 1.2517e-06, 3.5763e-07,  ..., 5.7373e-03,
           0.0000e+00, 0.0000e+00],
          [3.4717e-01, 6.1941e-04, 9.7466e-04,  ..., 1.0727e-02,
           3.8025e-02, 0.0000e+00],
          [3.4277e-01, 1.3447e-04, 6.9201e-05,  ..., 4.0314e-02,
           1.1368e-02, 3.5004e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.4756e-01, 2.5244e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.7249e-01, 7.8369e-02, 7.4902e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.3691e-01, 5.5492e-05, 2.1112e-04,  ..., 2.7930e-01,
           0.0000e+00, 0.0000e+00],
          [1.8079e-01, 1.1806e-03, 1.4381e-02,  ..., 2.0714e-03,
           7.0410e-01, 0.0000e+00],
          [1.0321e-01, 2.8086e-04, 1.3199e-03,  ..., 5.4283e-03,
           1.3049e-01, 7.0508e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7549e-01, 1.2463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5977e-01, 1.5930e-01, 8.0750e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1270e-01, 5.6028e-05, 6.8963e-05,  ..., 3.0548e-02,
           0.0000e+00, 0.0000e+00],
          [4.4849e-01, 9.7930e-05, 1.9217e-04,  ..., 5.1727e-03,
           5.0934e-02, 0.0000e+00],
          [1.6614e-01, 2.4796e-05, 7.1108e-05,  ..., 3.2837e-02,
           9.5032e-02, 1.1420e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5703e-01, 4.2816e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0674e-01, 6.1218e-02, 3.1952e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.4033e-01, 1.4997e-04, 6.3002e-05,  ..., 7.7858e-03,
           0.0000e+00, 0.0000e+00],
          [5.2393e-01, 8.6963e-05, 2.1589e-04,  ..., 1.8860e-02,
           3.4943e-02, 0.0000e+00],
          [5.3027e-01, 4.5538e-05, 4.5717e-05,  ..., 4.3854e-02,
           1.8936e-02, 2.3941e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4629e-01, 5.3802e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3604e-01, 3.6652e-02, 2.7557e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.5708e-01, 5.0831e-04, 2.7394e-04,  ..., 2.8748e-02,
           0.0000e+00, 0.0000e+00],
          [7.5391e-01, 2.4211e-04, 6.2275e-04,  ..., 2.6722e-03,
           1.8082e-02, 0.0000e+00],
          [5.7715e-01, 3.7193e-04, 9.0504e-04,  ..., 7.2098e-03,
           2.7206e-02, 3.2074e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.2432e-01, 7.5867e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8672e-01, 4.6631e-02, 6.6772e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3193e-01, 6.4850e-05, 2.0719e-04,  ..., 9.5444e-03,
           0.0000e+00, 0.0000e+00],
          [6.8945e-01, 1.8442e-04, 2.1954e-03,  ..., 9.2087e-03,
           1.7670e-02, 0.0000e+00],
          [2.2400e-01, 2.2697e-04, 1.3962e-03,  ..., 7.8506e-03,
           3.7140e-02, 6.5735e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4580e-01, 5.3986e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8721e-01, 2.6581e-02, 8.6487e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.3711e-01, 4.2200e-05, 7.4446e-05,  ..., 1.3451e-02,
           0.0000e+00, 0.0000e+00],
          [7.9297e-01, 4.9686e-04, 4.7493e-03,  ..., 1.4048e-03,
           4.2999e-02, 0.0000e+00],
          [6.9482e-01, 4.0126e-04, 5.1613e-03,  ..., 5.8250e-03,
           4.3396e-02, 7.2083e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6777e-01, 3.2471e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6484e-01, 1.4198e-02, 2.0981e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4639e-01, 5.7220e-06, 2.1517e-05,  ..., 2.6566e-02,
           0.0000e+00, 0.0000e+00],
          [7.1533e-01, 4.3440e-04, 2.2240e-03,  ..., 8.8654e-03,
           8.9539e-02, 0.0000e+00],
          [5.4541e-01, 1.9026e-04, 7.2384e-04,  ..., 9.1629e-03,
           3.9825e-02, 1.5259e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0527e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8975e-01, 5.4626e-03, 4.7493e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.6234e-02, 1.6093e-06, 5.3644e-07,  ..., 1.6617e-02,
           0.0000e+00, 0.0000e+00],
          [1.4441e-01, 6.4039e-04, 8.5831e-04,  ..., 2.6550e-03,
           6.5193e-03, 0.0000e+00],
          [4.7989e-03, 1.9372e-05, 1.5616e-05,  ..., 1.6098e-02,
           3.9077e-04, 2.2144e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7363e-01, 2.6154e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 1.6129e-02, 1.3260e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.1104e-01, 5.8413e-06, 1.7881e-06,  ..., 1.7136e-02,
           0.0000e+00, 0.0000e+00],
          [3.1006e-01, 5.2691e-04, 1.3542e-03,  ..., 2.5902e-03,
           4.5815e-03, 0.0000e+00],
          [2.5073e-01, 7.8559e-05, 1.1277e-04,  ..., 3.2898e-02,
           2.7752e-03, 1.4656e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4678e-01, 5.3192e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 1.7670e-02, 3.2623e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.2681e-01, 2.6345e-05, 4.5359e-05,  ..., 1.5556e-02,
           0.0000e+00, 0.0000e+00],
          [3.1494e-01, 1.6088e-03, 7.0953e-03,  ..., 3.2997e-03,
           3.9124e-02, 0.0000e+00],
          [1.0626e-01, 4.4107e-04, 1.8139e-03,  ..., 1.4267e-02,
           2.1088e-02, 8.0200e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.9463e-01, 5.4283e-03, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7803e-01, 3.8147e-03, 1.7914e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.4053e-01, 5.4002e-05, 4.2820e-04,  ..., 1.5907e-03,
           0.0000e+00, 0.0000e+00],
          [8.0078e-01, 8.9455e-04, 1.7786e-03,  ..., 4.8637e-04,
           3.4088e-02, 0.0000e+00],
          [4.0918e-01, 1.9348e-04, 1.2426e-03,  ..., 6.6519e-04,
           1.8152e-01, 1.1359e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.0178e-01, 7.9834e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.1455e-01, 3.5461e-02, 5.0018e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.9814e-01, 1.0085e-04, 2.5749e-04,  ..., 1.0948e-02,
           0.0000e+00, 0.0000e+00],
          [4.1309e-01, 3.9253e-03, 3.8643e-03,  ..., 5.4359e-04,
           1.0944e-01, 0.0000e+00],
          [1.2842e-01, 7.5197e-04, 2.4261e-03,  ..., 6.4325e-04,
           1.4331e-01, 1.9897e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6680e-01, 3.3081e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8193e-01, 6.8245e-03, 1.1429e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0029e-01, 1.9479e-04, 3.3140e-04,  ..., 4.5052e-03,
           0.0000e+00, 0.0000e+00],
          [7.3047e-01, 8.2254e-04, 1.0080e-03,  ..., 2.0199e-03,
           1.8295e-02, 0.0000e+00],
          [5.8594e-01, 1.0633e-03, 1.4753e-03,  ..., 3.1738e-03,
           5.1208e-02, 4.0192e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0381e-01, 9.6191e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5049e-01, 2.7878e-02, 2.2144e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.1133e-01, 2.6643e-05, 9.9838e-05,  ..., 2.5482e-02,
           0.0000e+00, 0.0000e+00],
          [7.3730e-01, 8.0407e-05, 1.0598e-04,  ..., 8.1940e-03,
           8.4351e-02, 0.0000e+00],
          [7.1582e-01, 5.2452e-05, 7.2598e-05,  ..., 7.6752e-03,
           7.2815e-02, 9.5703e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8486e-01, 1.5259e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4092e-01, 2.7115e-02, 3.2013e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [4.5386e-01, 6.5327e-05, 1.9968e-04,  ..., 4.7546e-02,
           0.0000e+00, 0.0000e+00],
          [6.0596e-01, 1.8382e-04, 2.3282e-04,  ..., 2.2736e-02,
           1.1841e-01, 0.0000e+00],
          [3.6572e-01, 9.9838e-05, 2.0587e-04,  ..., 2.1545e-02,
           1.5918e-01, 1.8896e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5947e-01, 4.0619e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.8164e-01, 1.5454e-01, 1.6382e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.9565e-01, 2.8396e-04, 3.0339e-05,  ..., 9.0837e-04,
           0.0000e+00, 0.0000e+00],
          [6.0254e-01, 2.1160e-04, 1.4949e-04,  ..., 1.0925e-02,
           2.1484e-01, 0.0000e+00],
          [1.5747e-01, 3.8922e-05, 1.6475e-04,  ..., 1.1314e-02,
           6.0840e-01, 1.8701e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.6338e-01, 3.6774e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3018e-01, 2.5375e-02, 4.4525e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.6221e-01, 2.5809e-05, 6.7949e-05,  ..., 9.8495e-03,
           0.0000e+00, 0.0000e+00],
          [7.9736e-01, 8.5688e-04, 1.6909e-03,  ..., 1.2913e-03,
           2.2263e-02, 0.0000e+00],
          [4.0845e-01, 9.1076e-04, 2.4853e-03,  ..., 1.4534e-02,
           2.6520e-02, 3.5400e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.0723e-01, 9.2712e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5264e-01, 2.9221e-02, 1.8066e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.7842e-01, 1.2875e-05, 1.0252e-05,  ..., 6.8665e-03,
           0.0000e+00, 0.0000e+00],
          [6.1719e-01, 7.2122e-05, 1.3208e-04,  ..., 2.2621e-03,
           1.2611e-02, 0.0000e+00],
          [2.1655e-01, 8.5592e-05, 1.8477e-04,  ..., 1.5686e-02,
           2.8412e-02, 3.5339e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2935e-01, 6.7041e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3538e-01, 5.4688e-02, 8.1006e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8356e-02, 4.1187e-05, 1.0747e-04,  ..., 1.8835e-01,
           0.0000e+00, 0.0000e+00],
          [8.8684e-02, 1.6756e-03, 4.2953e-03,  ..., 7.8735e-03,
           5.0635e-01, 0.0000e+00],
          [2.4338e-02, 4.1223e-04, 8.3590e-04,  ..., 1.8120e-03,
           2.6108e-02, 8.0566e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8877e-01, 1.1070e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8438e-01, 3.5572e-03, 1.2054e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.2080e-01, 7.5519e-05, 2.0230e-04,  ..., 7.9575e-03,
           0.0000e+00, 0.0000e+00],
          [3.7134e-01, 4.2152e-04, 1.0185e-03,  ..., 6.4125e-03,
           2.6538e-01, 0.0000e+00],
          [2.1777e-01, 5.1641e-04, 1.3180e-03,  ..., 1.0391e-02,
           2.9761e-01, 1.6565e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3555e-01, 6.4636e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7070e-01, 9.7046e-03, 1.9760e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.7295e-01, 2.4891e-04, 3.6287e-04,  ..., 1.7786e-03,
           0.0000e+00, 0.0000e+00],
          [1.2457e-01, 7.4501e-03, 8.4763e-03,  ..., 1.0262e-02,
           4.9164e-02, 0.0000e+00],
          [8.9600e-02, 5.9128e-03, 7.1297e-03,  ..., 1.5640e-02,
           3.5919e-02, 6.0638e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3408e-01, 6.6101e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.3164e-01, 2.3682e-02, 4.4769e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1790e-01, 4.9412e-05, 1.5533e-04,  ..., 1.0071e-02,
           0.0000e+00, 0.0000e+00],
          [2.7563e-01, 1.5154e-03, 2.8095e-03,  ..., 2.0920e-02,
           1.2016e-02, 0.0000e+00],
          [1.7615e-01, 6.8426e-04, 1.2054e-03,  ..., 2.2659e-02,
           7.6180e-03, 3.1342e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8584e-01, 5.7259e-03, 8.5983e-03,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5244e-01, 1.5459e-03, 1.4296e-03,  ..., 5.2299e-03,
           0.0000e+00, 0.0000e+00],
          [7.2266e-01, 2.4109e-03, 1.1568e-03,  ..., 1.1749e-03,
           3.5431e-02, 0.0000e+00],
          [6.1523e-01, 2.9802e-04, 3.5930e-04,  ..., 2.2011e-03,
           2.0569e-02, 1.6266e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.0000e-01, 5.0000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7783e-01, 2.0325e-01, 5.1904e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.8931e-01, 5.6744e-04, 1.5421e-03,  ..., 2.7173e-01,
           0.0000e+00, 0.0000e+00],
          [2.0471e-01, 1.7710e-03, 3.8071e-03,  ..., 3.1891e-02,
           4.8340e-01, 0.0000e+00],
          [1.4502e-01, 1.0405e-03, 2.6569e-03,  ..., 1.9623e-02,
           4.4220e-02, 6.4014e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.4971e-01, 5.0323e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7559e-01, 1.2646e-03, 2.3300e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.6318e-01, 2.1219e-04, 6.0940e-04,  ..., 6.3972e-03,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1711e-03, 1.8721e-03,  ..., 2.6398e-03,
           2.5848e-02, 0.0000e+00],
          [5.6934e-01, 4.9162e-04, 1.6499e-03,  ..., 9.2010e-03,
           4.5319e-02, 2.4246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8828e-01, 1.1948e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.4961e-01, 4.9095e-03, 1.4539e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.5020e-01, 6.7353e-06, 5.0247e-05,  ..., 2.2709e-04,
           0.0000e+00, 0.0000e+00],
          [8.1641e-01, 1.1849e-04, 6.1274e-04,  ..., 4.6196e-03,
           1.4488e-02, 0.0000e+00],
          [6.5820e-01, 3.7789e-05, 6.6280e-04,  ..., 4.9362e-03,
           1.2032e-02, 1.0803e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.4404e-01, 3.5571e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9131e-01, 5.2490e-02, 3.5596e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.7656e-01, 6.4969e-06, 5.3048e-05,  ..., 1.5879e-04,
           0.0000e+00, 0.0000e+00],
          [6.2500e-01, 2.0123e-04, 9.0075e-04,  ..., 4.9362e-03,
           4.8370e-02, 0.0000e+00],
          [6.8909e-02, 9.5665e-05, 9.2685e-05,  ..., 1.5344e-01,
           1.3206e-02, 3.6041e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.8730e-01, 1.2650e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 1.1169e-02, 1.5039e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [9.4287e-01, 2.4140e-05, 1.2863e-04,  ..., 7.5877e-05,
           0.0000e+00, 0.0000e+00],
          [6.5625e-01, 7.9298e-04, 2.2221e-03,  ..., 1.3018e-03,
           1.5222e-01, 0.0000e+00],
          [2.4585e-01, 1.4963e-03, 1.7529e-03,  ..., 1.8845e-02,
           1.4062e-01, 9.5520e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.9004e-01, 2.1008e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7412e-01, 7.6752e-03, 1.8021e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.1904e-01, 1.3628e-03, 6.5956e-03,  ..., 1.3062e-02,
           0.0000e+00, 0.0000e+00],
          [8.3838e-01, 2.8276e-04, 4.9286e-03,  ..., 3.7117e-03,
           8.3313e-02, 0.0000e+00],
          [7.6221e-01, 2.5392e-04, 8.5449e-04,  ..., 4.5967e-03,
           2.2232e-02, 1.2665e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.8613e-01, 2.1399e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.6270e-01, 4.6967e-02, 1.9055e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8018e-01, 2.0623e-05, 2.5196e-03,  ..., 4.9362e-03,
           0.0000e+00, 0.0000e+00],
          [6.9824e-01, 5.7793e-04, 2.3766e-03,  ..., 1.3794e-02,
           1.6260e-01, 0.0000e+00],
          [8.5742e-01, 3.8981e-05, 8.1253e-04,  ..., 5.4283e-03,
           1.7593e-02, 9.6207e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.5020e-01, 4.9866e-02, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [9.7656e-01, 7.7744e-03, 1.5717e-02,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.8262e-01, 4.0913e-04, 3.6359e-04,  ..., 1.0757e-02,
           0.0000e+00, 0.0000e+00],
          [7.3535e-01, 1.3466e-03, 9.7227e-04,  ..., 6.9122e-03,
           4.9164e-02, 0.0000e+00],
          [5.0635e-01, 1.9236e-03, 1.8082e-03,  ..., 8.9111e-03,
           5.0049e-02, 2.6520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-02, 9.1064e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3428e-02, 1.3351e-02, 9.7314e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1062e-06, 1.1265e-05, 3.2397e-01,  ..., 9.3174e-04,
           0.0000e+00, 0.0000e+00],
          [8.9836e-04, 3.8414e-03, 1.4026e-01,  ..., 1.5793e-02,
           8.2458e-02, 0.0000e+00],
          [5.6839e-04, 1.0252e-03, 1.7236e-01,  ..., 2.5314e-02,
           3.1769e-02, 3.9978e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8296e-01, 7.1680e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9276e-02, 3.0838e-02, 9.2969e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0121e-04, 2.0444e-04, 2.9248e-01,  ..., 8.4229e-03,
           0.0000e+00, 0.0000e+00],
          [4.5538e-04, 1.4811e-03, 2.6465e-01,  ..., 4.6806e-03,
           6.3843e-02, 0.0000e+00],
          [5.2738e-04, 2.9683e-04, 1.6028e-01,  ..., 9.5139e-03,
           1.1604e-02, 1.2947e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.7424e-02, 9.5264e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.0131e-02, 4.1473e-02, 9.1846e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.7115e-05, 4.3809e-05, 7.9041e-03,  ..., 1.5087e-03,
           0.0000e+00, 0.0000e+00],
          [1.3237e-03, 7.2670e-03, 8.8867e-02,  ..., 1.1566e-02,
           2.4927e-01, 0.0000e+00],
          [1.0719e-03, 2.0542e-03, 1.6162e-01,  ..., 1.2657e-02,
           4.2145e-02, 3.6041e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7178e-01, 4.2822e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5845e-01, 8.5510e-02, 7.5586e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2214e-04, 6.2990e-04, 1.2335e-01,  ..., 5.9631e-02,
           0.0000e+00, 0.0000e+00],
          [2.5978e-03, 2.4986e-03, 2.4036e-01,  ..., 2.1164e-02,
           2.0557e-01, 0.0000e+00],
          [3.3307e-04, 1.8907e-04, 9.3811e-02,  ..., 1.1584e-01,
           2.5708e-01, 2.3035e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.1567e-01, 6.8408e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1864e-02, 4.6005e-03, 9.8340e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1501e-03, 5.1260e-04, 3.1519e-01,  ..., 9.6130e-04,
           0.0000e+00, 0.0000e+00],
          [2.1517e-04, 7.9155e-04, 3.2300e-01,  ..., 5.7030e-04,
           6.8237e-02, 0.0000e+00],
          [6.0797e-04, 8.7070e-04, 2.7783e-01,  ..., 7.6866e-04,
           4.1321e-02, 5.1819e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9600e-01, 6.0400e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.2267e-03, 1.5068e-03, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.4392e-05, 6.9737e-05, 3.6182e-01,  ..., 2.8564e-02,
           0.0000e+00, 0.0000e+00],
          [2.2042e-04, 1.8132e-04, 2.3608e-01,  ..., 1.7242e-02,
           4.4037e-02, 0.0000e+00],
          [1.7631e-04, 2.9147e-05, 1.1548e-01,  ..., 1.9455e-02,
           7.5569e-03, 3.1830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7598e-01, 1.2421e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.0109e-03, 2.2697e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6716e-05, 8.6427e-06, 4.1968e-01,  ..., 1.0258e-04,
           0.0000e+00, 0.0000e+00],
          [7.7188e-05, 7.0572e-05, 3.6914e-01,  ..., 1.9836e-03,
           8.6823e-03, 0.0000e+00],
          [3.5357e-04, 1.2058e-04, 2.2266e-01,  ..., 8.6670e-03,
           4.5166e-03, 2.2049e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.9531e-01, 3.0493e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.7308e-03, 1.6651e-03, 9.9463e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.7578e-05, 1.5676e-05, 2.8076e-01,  ..., 7.0953e-04,
           0.0000e+00, 0.0000e+00],
          [1.5128e-04, 8.7261e-05, 2.8662e-01,  ..., 1.7899e-02,
           4.7546e-02, 0.0000e+00],
          [1.8680e-04, 1.0645e-04, 1.6724e-01,  ..., 1.8188e-02,
           1.2596e-02, 1.1925e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2373e-01, 1.7615e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2567e-03, 1.0767e-03, 9.9365e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9339e-05, 5.7220e-06, 3.9648e-01,  ..., 1.5335e-03,
           0.0000e+00, 0.0000e+00],
          [6.9559e-05, 1.3769e-05, 3.2861e-01,  ..., 1.5137e-02,
           4.6997e-02, 0.0000e+00],
          [9.7513e-05, 2.0146e-05, 3.1543e-01,  ..., 1.1139e-02,
           9.2697e-03, 2.0416e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6975e-03, 1.5202e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.0320e-04, 3.0398e-05, 1.4526e-01,  ..., 5.1785e-04,
           0.0000e+00, 0.0000e+00],
          [2.1303e-04, 9.0718e-05, 3.5400e-01,  ..., 1.4582e-03,
           1.4221e-02, 0.0000e+00],
          [7.7152e-04, 1.9884e-04, 1.8652e-01,  ..., 5.6992e-03,
           2.2678e-03, 1.1421e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.9111e-01, 1.0895e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.7023e-04, 9.5034e-04, 9.9805e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.2994e-05, 8.5235e-06, 3.7646e-01,  ..., 3.2258e-04,
           0.0000e+00, 0.0000e+00],
          [2.4855e-05, 2.6584e-05, 3.2129e-01,  ..., 1.5266e-02,
           6.0959e-03, 0.0000e+00],
          [1.9252e-05, 2.9683e-05, 2.4817e-01,  ..., 7.5531e-03,
           1.2617e-03, 5.5237e-03]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8086e-01, 1.1920e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3733e-04, 5.1260e-04, 9.9854e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7893e-02, 1.0166e-03, 1.9067e-01,  ..., 1.3390e-02,
           0.0000e+00, 0.0000e+00],
          [4.9055e-05, 2.8789e-05, 3.1885e-01,  ..., 4.9210e-03,
           1.9800e-01, 0.0000e+00],
          [6.2764e-05, 2.2829e-05, 2.4365e-01,  ..., 5.0774e-03,
           5.3925e-02, 2.0203e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.8843e-01, 6.1182e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6564e-03, 1.2646e-03, 9.9512e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.3195e-05, 4.5478e-05, 3.8770e-01,  ..., 8.4925e-04,
           0.0000e+00, 0.0000e+00],
          [4.3035e-05, 3.6001e-05, 3.4351e-01,  ..., 3.8086e-02,
           3.1395e-03, 0.0000e+00],
          [6.4552e-05, 4.0233e-05, 1.2573e-01,  ..., 1.9089e-02,
           3.9315e-04, 1.2337e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.8281e-01, 1.1737e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.7068e-03, 1.6725e-04, 9.9414e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.1683e-04, 4.5300e-06, 3.9160e-01,  ..., 8.9073e-04,
           0.0000e+00, 0.0000e+00],
          [8.7202e-05, 5.1260e-06, 2.9126e-01,  ..., 1.6953e-02,
           1.7624e-02, 0.0000e+00],
          [1.5831e-04, 3.9935e-06, 1.7432e-01,  ..., 1.7532e-02,
           6.3248e-03, 1.0384e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5024e-01, 7.5000e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.1498e-03, 3.7670e-03, 9.9121e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.0711e-04, 1.0902e-04, 4.2725e-01,  ..., 1.9703e-03,
           0.0000e+00, 0.0000e+00],
          [1.5020e-04, 1.5020e-04, 3.1665e-01,  ..., 3.9642e-02,
           4.3449e-03, 0.0000e+00],
          [1.6618e-04, 1.5664e-04, 2.2070e-01,  ..., 3.2684e-02,
           9.4175e-04, 7.0419e-03]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.9707e-01, 5.0293e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.7298e-02, 2.9907e-03, 9.6973e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.5024e-04, 3.7861e-04, 2.2754e-01,  ..., 3.0861e-03,
           0.0000e+00, 0.0000e+00],
          [1.0262e-03, 1.1187e-03, 2.7417e-01,  ..., 1.2871e-02,
           4.0894e-02, 0.0000e+00],
          [1.8728e-04, 1.8537e-04, 1.7029e-01,  ..., 2.0248e-02,
           1.4534e-02, 1.8738e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.5537e-01, 7.4463e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.5106e-03, 1.4343e-03, 9.9707e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.8716e-05, 3.5584e-05, 2.7148e-01,  ..., 1.2331e-03,
           0.0000e+00, 0.0000e+00],
          [4.7505e-05, 4.1664e-05, 2.3096e-01,  ..., 1.7807e-02,
           1.3817e-02, 0.0000e+00],
          [6.5148e-05, 5.7399e-05, 7.5745e-02,  ..., 1.6312e-02,
           6.2675e-03, 1.9455e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.9375e-01, 4.0649e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.6136e-03, 1.5936e-03, 9.8975e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.9935e-05, 2.2054e-05, 3.7598e-01,  ..., 7.6389e-04,
           0.0000e+00, 0.0000e+00],
          [1.0240e-04, 3.3379e-05, 2.6514e-01,  ..., 2.4658e-02,
           1.7212e-02, 0.0000e+00],
          [2.1017e-04, 3.4332e-05, 1.2720e-01,  ..., 2.9297e-02,
           4.1885e-03, 1.4618e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [7.5684e-01, 2.4292e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.5002e-02, 4.4342e-02, 8.9062e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.1024e-03, 3.2926e-04, 3.1250e-01,  ..., 7.9651e-02,
           0.0000e+00, 0.0000e+00],
          [1.7290e-03, 2.0897e-04, 2.4536e-02,  ..., 3.2446e-01,
           2.6880e-01, 0.0000e+00],
          [8.2397e-04, 7.2718e-06, 4.3335e-03,  ..., 5.9521e-01,
           4.6692e-03, 9.3445e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.1230e-01, 3.8770e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.2986e-03, 1.5945e-02, 9.7852e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1875e-04, 1.3947e-04, 2.6978e-01,  ..., 6.3038e-04,
           0.0000e+00, 0.0000e+00],
          [1.2836e-03, 3.6168e-04, 3.4570e-01,  ..., 1.5545e-04,
           3.9558e-03, 0.0000e+00],
          [9.1648e-04, 3.2330e-04, 2.7930e-01,  ..., 1.9062e-04,
           7.9041e-03, 1.7792e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.5435e-01, 5.4590e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.6959e-03, 4.8561e-03, 9.9023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.7132e-04, 9.8324e-04, 1.0468e-01,  ..., 2.0134e-04,
           0.0000e+00, 0.0000e+00],
          [8.6725e-05, 1.0185e-03, 3.0884e-01,  ..., 2.6941e-05,
           1.0345e-01, 0.0000e+00],
          [1.4329e-04, 3.3188e-04, 2.4072e-01,  ..., 1.3065e-04,
           1.1731e-01, 5.0049e-02]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.3727e-02, 9.7607e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.4189e-02, 3.4363e-02, 9.2139e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [3.6335e-03, 9.5892e-04, 2.2986e-01,  ..., 2.3441e-03,
           0.0000e+00, 0.0000e+00],
          [2.5487e-04, 2.5010e-04, 9.4238e-02,  ..., 4.4327e-03,
           1.7615e-01, 0.0000e+00],
          [6.7472e-04, 1.3363e-04, 2.5177e-02,  ..., 4.1840e-02,
           1.6223e-01, 6.7078e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.0248e-01, 8.9746e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.2769e-01, 6.8604e-02, 8.0371e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.0185e-03, 3.2806e-03, 2.5879e-01,  ..., 8.4351e-02,
           0.0000e+00, 0.0000e+00],
          [8.9264e-04, 1.9007e-03, 7.1594e-02,  ..., 3.6285e-02,
           5.8350e-01, 0.0000e+00],
          [4.6206e-04, 8.5950e-05, 1.1154e-02,  ..., 2.4219e-01,
           4.4897e-01, 1.9312e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [2.8052e-01, 7.1924e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.3696e-01, 3.3630e-02, 8.2959e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [2.1114e-03, 5.3120e-04, 1.9543e-01,  ..., 8.9478e-02,
           0.0000e+00, 0.0000e+00],
          [1.4889e-04, 2.3186e-05, 1.9119e-02,  ..., 5.0598e-02,
           6.6895e-01, 0.0000e+00],
          [7.9334e-05, 9.7156e-06, 1.0818e-02,  ..., 5.9387e-02,
           5.1208e-02, 5.4504e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.1475e-01, 8.8525e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.6072e-02, 6.0425e-02, 9.0332e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [5.2528e-03, 1.1911e-03, 1.3135e-01,  ..., 1.1932e-02,
           0.0000e+00, 0.0000e+00],
          [6.0539e-03, 5.4054e-03, 2.0850e-01,  ..., 1.7586e-03,
           9.0393e-02, 0.0000e+00],
          [6.7711e-03, 1.8911e-03, 1.4539e-01,  ..., 1.3924e-03,
           2.5253e-02, 7.4524e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [4.8676e-02, 9.5117e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [5.3864e-02, 2.4368e-02, 9.2188e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.9188e-03, 3.9825e-03, 1.6772e-01,  ..., 3.8483e-02,
           0.0000e+00, 0.0000e+00],
          [2.9087e-03, 3.9291e-03, 1.7236e-01,  ..., 1.2100e-02,
           1.9995e-01, 0.0000e+00],
          [1.2283e-03, 4.0207e-03, 2.7222e-01,  ..., 2.1267e-03,
           1.2703e-02, 4.5654e-02]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.2373e-01, 6.7627e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [3.9703e-02, 5.1483e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [8.3923e-03, 3.3703e-03, 1.7322e-01,  ..., 3.3447e-02,
           0.0000e+00, 0.0000e+00],
          [3.0594e-03, 3.2177e-03, 1.1591e-01,  ..., 1.0071e-01,
           1.8884e-01, 0.0000e+00],
          [2.7275e-03, 1.0405e-03, 1.5308e-01,  ..., 1.9302e-02,
           2.8198e-02, 1.2891e-01]],

         ...,

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [6.6699e-01, 3.3276e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9006e-01, 6.9946e-02, 7.4023e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [6.4135e-04, 1.6797e-04, 2.1960e-01,  ..., 5.9113e-02,
           0.0000e+00, 0.0000e+00],
          [2.3746e-03, 1.8895e-04, 1.2659e-01,  ..., 3.6011e-02,
           3.1769e-02, 0.0000e+00],
          [1.0061e-03, 1.4222e-04, 4.8859e-02,  ..., 1.6663e-01,
           1.8152e-01, 2.5000e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.2092e-02, 9.1797e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.6251e-02, 7.5195e-02, 9.0869e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [7.5483e-04, 3.7594e-03, 2.4182e-01,  ..., 6.6101e-02,
           0.0000e+00, 0.0000e+00],
          [3.5119e-04, 2.6298e-04, 2.9541e-01,  ..., 1.5900e-02,
           8.2031e-02, 0.0000e+00],
          [4.9448e-04, 8.6117e-04, 2.0422e-01,  ..., 6.8245e-03,
           1.0086e-02, 3.1128e-01]],

         [[1.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [8.3105e-01, 1.6882e-01, 0.0000e+00,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          [1.9458e-01, 3.6353e-01, 4.4189e-01,  ..., 0.0000e+00,
           0.0000e+00, 0.0000e+00],
          ...,
          [1.3742e-03, 1.2512e-03, 8.4351e-02,  ..., 2.5391e-01,
           0.0000e+00, 0.0000e+00],
          [3.9101e-03, 8.9645e-04, 5.3558e-02,  ..., 6.2103e-02,
           2.4756e-01, 0.0000e+00],
          [1.8728e-04, 4.7326e-05, 1.7059e-02,  ..., 9.8145e-02,
           3.0957e-01, 2.1948e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.9605e-07, 4.2319e-06, 6.4611e-04,  ..., 2.5803e-02,
           8.0273e-01, 1.3525e-01]],

         [[1.9014e-05, 1.9014e-05, 2.6531e-03,  ..., 2.3651e-02,
           2.8809e-01, 4.1919e-01]],

         [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 2.9802e-07,
           1.0986e-02, 9.8877e-01]],

         ...,

         [[2.5177e-03, 3.2310e-03, 4.1504e-03,  ..., 3.0670e-02,
           2.2656e-01, 5.0568e-02]],

         [[4.7569e-03, 1.2146e-02, 5.1147e-02,  ..., 3.9825e-02,
           1.3904e-01, 1.0175e-01]],

         [[2.0248e-02, 1.6785e-02, 2.5749e-03,  ..., 4.2877e-02,
           6.5765e-03, 6.2378e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1215e-03, 1.3323e-03, 1.0483e-02,  ..., 5.0781e-02,
           1.9775e-01, 2.6611e-01]],

         [[5.9605e-08, 2.0266e-06, 4.6670e-05,  ..., 9.4116e-02,
           7.7588e-01, 1.2671e-01]],

         [[3.3493e-03, 5.5237e-03, 1.7273e-02,  ..., 1.3586e-01,
           1.9775e-01, 1.9165e-01]],

         ...,

         [[4.5319e-03, 1.5450e-02, 1.5450e-02,  ..., 1.2250e-01,
           1.8494e-02, 9.6863e-02]],

         [[1.4267e-03, 4.0588e-03, 4.0192e-02,  ..., 6.7566e-02,
           3.6230e-01, 4.1443e-02]],

         [[1.0452e-03, 3.9444e-03, 1.8372e-02,  ..., 3.2324e-01,
           1.0577e-01, 1.8274e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1133e-02, 4.7874e-03, 1.4076e-02,  ..., 9.2529e-02,
           8.0383e-02, 1.0162e-01]],

         [[3.3569e-03, 1.0214e-03, 2.3251e-03,  ..., 2.9980e-01,
           3.9490e-02, 1.2817e-02]],

         [[1.7456e-02, 7.6904e-03, 7.7782e-03,  ..., 1.5114e-02,
           5.4413e-02, 2.8671e-02]],

         ...,

         [[4.0936e-04, 3.4332e-04, 1.5507e-03,  ..., 2.8540e-01,
           3.8379e-01, 1.4917e-01]],

         [[1.3947e-02, 2.5902e-03, 4.1885e-03,  ..., 6.7627e-02,
           1.6345e-01, 5.3070e-02]],

         [[9.6359e-03, 2.0123e-03, 2.0325e-02,  ..., 5.6549e-02,
           4.2114e-01, 9.5459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8955e-01, 2.2829e-05, 5.0485e-05,  ..., 1.4982e-03,
           3.7003e-03, 1.1658e-02]],

         [[9.6143e-01, 3.6359e-06, 3.6001e-05,  ..., 4.9477e-03,
           9.5749e-03, 4.0779e-03]],

         [[8.5010e-01, 3.7551e-05, 2.2292e-04,  ..., 7.1716e-03,
           3.6335e-03, 2.0660e-02]],

         ...,

         [[6.3428e-01, 4.7040e-04, 1.8635e-03,  ..., 1.5821e-03,
           4.9400e-03, 3.3112e-02]],

         [[2.6343e-01, 4.1771e-04, 8.1348e-04,  ..., 4.1870e-02,
           1.2917e-02, 3.8574e-02]],

         [[5.7227e-01, 3.1590e-04, 7.3385e-04,  ..., 2.0355e-02,
           3.3813e-02, 4.4128e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 6.8545e-06, 1.6212e-05,  ..., 1.7285e-04,
           3.8338e-04, 3.4790e-02]],

         [[8.7207e-01, 1.1921e-06, 1.5259e-05,  ..., 9.9945e-03,
           1.0199e-01, 1.2650e-02]],

         [[9.1650e-01, 3.8385e-05, 1.3304e-04,  ..., 6.9427e-04,
           2.2110e-02, 1.7593e-02]],

         ...,

         [[2.1680e-01, 1.5008e-04, 7.3016e-05,  ..., 1.0278e-01,
           2.5244e-01, 3.2715e-02]],

         [[9.7839e-02, 2.9011e-03, 2.0599e-03,  ..., 1.5100e-01,
           4.3335e-01, 2.5024e-02]],

         [[4.9292e-01, 8.5294e-05, 1.0532e-04,  ..., 7.5317e-02,
           1.1896e-01, 9.3506e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1807e-01, 7.0715e-04, 2.7585e-04,  ..., 5.5145e-02,
           6.5002e-02, 3.3905e-02]],

         [[3.4277e-01, 3.1829e-05, 3.4571e-05,  ..., 9.7473e-02,
           3.9355e-01, 4.2053e-02]],

         [[7.5586e-01, 9.7990e-05, 1.8477e-05,  ..., 8.3389e-03,
           4.4189e-02, 6.0242e-02]],

         ...,

         [[9.2383e-01, 4.1604e-05, 1.9073e-06,  ..., 1.8835e-05,
           3.4428e-04, 7.1960e-02]],

         [[9.6826e-01, 1.5640e-04, 3.6001e-05,  ..., 2.6822e-06,
           1.0061e-04, 1.0414e-02]],

         [[9.9609e-01, 2.3901e-05, 7.1526e-07,  ..., 5.9605e-08,
           8.7619e-06, 2.3708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1270e-01, 1.6809e-05, 8.7619e-06,  ..., 8.7830e-02,
           2.1753e-01, 2.4368e-02]],

         [[9.6045e-01, 1.4126e-05, 7.3910e-06,  ..., 7.3338e-04,
           1.0042e-03, 1.6953e-02]],

         [[8.2617e-01, 3.5930e-04, 2.9862e-05,  ..., 1.5083e-02,
           5.7335e-03, 5.0507e-02]],

         ...,

         [[9.5605e-01, 1.3530e-04, 4.3571e-05,  ..., 2.7504e-03,
           1.9112e-03, 9.3384e-03]],

         [[9.6533e-01, 7.5996e-05, 1.5736e-05,  ..., 5.0583e-03,
           1.7986e-03, 1.2878e-02]],

         [[6.2012e-01, 2.9850e-04, 1.1092e-04,  ..., 2.3486e-01,
           5.9929e-03, 1.0544e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0264e-01, 3.0339e-05, 3.5048e-05,  ..., 4.9629e-03,
           5.9052e-03, 1.4191e-02]],

         [[7.6318e-01, 7.7486e-05, 6.7592e-05,  ..., 8.9722e-03,
           7.9803e-03, 4.7882e-02]],

         [[8.7695e-01, 1.1253e-04, 3.6240e-05,  ..., 1.0391e-02,
           3.6469e-03, 8.0414e-03]],

         ...,

         [[6.9141e-01, 2.7156e-04, 4.2498e-05,  ..., 1.6449e-02,
           8.6365e-03, 6.1279e-02]],

         [[6.9775e-01, 6.0272e-04, 9.4295e-05,  ..., 2.7939e-02,
           3.9703e-02, 1.7212e-02]],

         [[7.1045e-01, 2.6655e-04, 1.6093e-04,  ..., 3.3722e-02,
           8.2214e-02, 2.9587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2212e-01, 2.8057e-03, 1.1206e-03,  ..., 1.3977e-02,
           3.4119e-02, 1.1032e-02]],

         [[8.1738e-01, 1.5521e-04, 3.4547e-04,  ..., 2.4452e-03,
           1.0376e-02, 5.0774e-03]],

         [[8.7305e-01, 5.6088e-05, 3.4809e-05,  ..., 2.0008e-03,
           3.9005e-03, 6.4049e-03]],

         ...,

         [[7.2168e-01, 4.3488e-04, 5.3930e-04,  ..., 3.4008e-03,
           1.4114e-02, 1.0901e-01]],

         [[8.6328e-01, 1.6832e-04, 6.0976e-05,  ..., 2.5539e-03,
           5.8632e-03, 2.8198e-02]],

         [[9.0771e-01, 2.8706e-04, 1.5128e-04,  ..., 2.2873e-02,
           4.1313e-03, 1.2688e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5610e-01, 2.9087e-04, 7.6056e-04,  ..., 7.0312e-02,
           1.8982e-01, 1.3977e-02]],

         [[8.1689e-01, 2.5988e-05, 1.7273e-04,  ..., 3.4851e-02,
           3.3264e-02, 2.6764e-02]],

         [[8.1006e-01, 7.9095e-05, 2.4843e-04,  ..., 6.9962e-03,
           8.7051e-03, 6.3171e-02]],

         ...,

         [[9.8511e-02, 2.1958e-04, 4.1199e-04,  ..., 3.0518e-02,
           7.4805e-01, 4.4403e-03]],

         [[7.4902e-01, 4.5776e-04, 1.7202e-04,  ..., 7.1167e-02,
           5.9204e-03, 1.0147e-02]],

         [[8.5596e-01, 1.1563e-05, 1.1444e-05,  ..., 4.6158e-03,
           5.9128e-03, 6.5002e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 3.1173e-05, 1.1539e-04,  ..., 9.3102e-05,
           3.3264e-03, 2.9358e-02]],

         [[4.8975e-01, 8.9169e-05, 7.7188e-05,  ..., 6.6162e-02,
           3.3508e-02, 1.0797e-01]],

         [[3.4814e-01, 5.8413e-06, 6.9737e-06,  ..., 9.9468e-04,
           3.7384e-03, 3.9368e-02]],

         ...,

         [[9.1748e-01, 1.8966e-04, 8.4162e-05,  ..., 5.2261e-03,
           5.4817e-03, 9.8724e-03]],

         [[4.1772e-01, 4.2820e-04, 2.6226e-04,  ..., 4.6692e-02,
           2.9633e-02, 2.6196e-01]],

         [[1.8042e-01, 5.5552e-04, 3.6287e-04,  ..., 2.5024e-02,
           4.9316e-02, 3.9520e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6367e-01, 4.7684e-06, 7.8082e-06,  ..., 4.7379e-03,
           3.2288e-02, 6.7041e-01]],

         [[7.5537e-01, 6.5231e-04, 8.3590e-04,  ..., 1.6632e-02,
           1.1604e-02, 2.5043e-03]],

         [[8.7988e-01, 4.5228e-04, 5.7936e-04,  ..., 1.0300e-02,
           1.3832e-02, 2.6566e-02]],

         ...,

         [[4.0430e-01, 1.7905e-04, 4.5037e-04,  ..., 1.1139e-02,
           1.9394e-02, 3.8300e-02]],

         [[6.2939e-01, 4.6968e-05, 5.5134e-05,  ..., 9.7580e-03,
           3.8509e-03, 1.9577e-02]],

         [[2.9810e-01, 2.4974e-05, 8.3506e-05,  ..., 5.6610e-03,
           7.0862e-02, 5.5322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.5184e-04, 8.8930e-04,  ..., 1.2741e-02,
           8.3252e-02, 3.8208e-01]],

         [[5.4150e-01, 2.5821e-04, 6.7472e-04,  ..., 3.4851e-02,
           4.5227e-02, 6.7871e-02]],

         [[4.8608e-01, 5.8603e-04, 1.0443e-03,  ..., 4.8790e-03,
           1.3260e-02, 3.3539e-02]],

         ...,

         [[8.2947e-02, 2.1458e-06, 3.8147e-06,  ..., 5.6744e-04,
           6.9847e-03, 8.9844e-01]],

         [[4.8633e-01, 2.9421e-04, 4.8590e-04,  ..., 2.7054e-02,
           5.2246e-02, 1.3940e-01]],

         [[8.7842e-01, 1.3733e-04, 1.0735e-04,  ..., 1.7948e-03,
           3.3016e-03, 3.2940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 3.0899e-04, 3.9220e-04,  ..., 4.8447e-03,
           8.0719e-03, 1.4458e-02]],

         [[9.0088e-01, 2.5570e-05, 5.0724e-05,  ..., 7.8125e-03,
           1.6266e-02, 2.4902e-02]],

         [[5.2734e-01, 1.1134e-04, 2.8324e-04,  ..., 2.5291e-03,
           3.0136e-02, 3.1555e-02]],

         ...,

         [[3.9575e-01, 0.0000e+00, 3.9935e-06,  ..., 2.9099e-02,
           5.2344e-01, 4.5013e-02]],

         [[6.2744e-01, 1.0300e-03, 1.2751e-03,  ..., 1.7365e-02,
           2.9160e-02, 2.2629e-02]],

         [[9.1748e-01, 1.9670e-06, 7.9870e-06,  ..., 9.9945e-03,
           1.9119e-02, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8281e-01, 3.4809e-04, 8.9586e-05,  ..., 7.8011e-04,
           2.2030e-03, 6.1340e-03]],

         [[2.4976e-01, 7.1859e-04, 4.8018e-04,  ..., 1.5417e-01,
           2.8076e-01, 2.7130e-02]],

         [[4.8022e-01, 7.6246e-04, 4.2534e-04,  ..., 5.2643e-02,
           9.0576e-02, 3.1189e-02]],

         ...,

         [[3.5913e-01, 1.0192e-05, 4.1008e-05,  ..., 6.3232e-02,
           2.2168e-01, 8.7769e-02]],

         [[7.6318e-01, 5.7220e-06, 1.8120e-05,  ..., 3.1769e-02,
           8.1543e-02, 6.1340e-02]],

         [[1.6931e-01, 4.3452e-05, 1.0747e-04,  ..., 2.5903e-01,
           3.9795e-01, 2.2171e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7373e-02, 1.2517e-06, 2.5034e-06,  ..., 2.3365e-03,
           4.6112e-02, 8.5645e-01]],

         [[2.1960e-01, 1.0848e-05, 2.0385e-05,  ..., 1.2036e-01,
           4.8730e-01, 6.2195e-02]],

         [[5.0781e-01, 8.9550e-04, 1.1225e-03,  ..., 2.6245e-02,
           6.1493e-02, 1.1353e-02]],

         ...,

         [[1.9678e-01, 2.2161e-04, 2.4390e-04,  ..., 1.5198e-02,
           3.7170e-02, 1.0994e-02]],

         [[4.7925e-01, 5.7518e-05, 4.3273e-05,  ..., 2.8858e-03,
           1.4221e-02, 2.2858e-02]],

         [[9.9365e-02, 3.9959e-04, 2.0254e-04,  ..., 1.1683e-03,
           3.5477e-03, 6.1607e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 2.5487e-04, 3.8195e-04,  ..., 2.2308e-02,
           2.6901e-02, 2.1698e-02]],

         [[2.4841e-01, 4.2319e-04, 3.9911e-04,  ..., 2.3376e-02,
           6.0852e-02, 6.1829e-02]],

         [[1.2061e-01, 2.9659e-04, 1.5235e-04,  ..., 5.6648e-03,
           1.1620e-02, 1.0620e-02]],

         ...,

         [[7.2510e-01, 4.1246e-05, 2.8074e-05,  ..., 1.3485e-03,
           1.0712e-02, 2.8442e-02]],

         [[5.4883e-01, 7.1096e-04, 4.9305e-04,  ..., 6.9504e-03,
           2.4414e-02, 3.5919e-02]],

         [[8.9844e-01, 3.9554e-04, 1.8942e-04,  ..., 2.0199e-03,
           1.0712e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8726e-02, 1.6499e-03, 1.4124e-03,  ..., 5.4053e-01,
           1.1243e-01, 3.4218e-03]],

         [[3.9526e-01, 3.1967e-03, 1.6804e-03,  ..., 2.5818e-02,
           1.9150e-02, 3.1372e-02]],

         [[4.9341e-01, 1.4496e-04, 1.8835e-04,  ..., 1.4755e-02,
           6.8115e-02, 5.9174e-02]],

         ...,

         [[4.9219e-01, 2.8634e-04, 3.5477e-04,  ..., 1.0117e-02,
           2.2141e-02, 1.0400e-01]],

         [[9.8584e-01, 3.3975e-05, 1.0395e-04,  ..., 6.8545e-06,
           2.1577e-04, 2.1572e-03]],

         [[8.6963e-01, 1.3053e-04, 1.2374e-04,  ..., 8.7051e-03,
           1.0895e-02, 2.1179e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1689e-01, 3.5763e-03, 2.9640e-03,  ..., 1.8628e-01,
           1.8054e-01, 7.8857e-02]],

         [[7.9834e-02, 6.8169e-03, 3.1948e-03,  ..., 1.3477e-01,
           1.0095e-01, 2.5131e-02]],

         [[1.0645e-01, 6.4135e-05, 8.6308e-05,  ..., 1.8204e-02,
           1.1688e-01, 4.7314e-01]],

         ...,

         [[1.4807e-01, 5.7995e-05, 8.9586e-05,  ..., 5.6610e-02,
           1.6382e-01, 1.2199e-02]],

         [[7.6953e-01, 1.5020e-05, 2.3246e-05,  ..., 1.0391e-02,
           4.2419e-02, 1.1963e-02]],

         [[2.9590e-01, 1.7631e-04, 2.5058e-04,  ..., 5.1422e-02,
           3.3789e-01, 1.6467e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7061e-01, 2.7180e-05, 2.8014e-05,  ..., 1.1921e-03,
           5.5122e-03, 8.2275e-02]],

         [[2.5269e-01, 1.6081e-04, 7.2539e-05,  ..., 8.8684e-02,
           2.2656e-01, 3.7567e-02]],

         [[3.9526e-01, 6.2644e-05, 6.9857e-05,  ..., 9.0103e-03,
           7.3120e-02, 4.9194e-01]],

         ...,

         [[3.3423e-01, 3.9215e-03, 2.4929e-03,  ..., 6.9885e-03,
           1.9455e-02, 2.8305e-02]],

         [[5.2393e-01, 1.1330e-03, 1.0519e-03,  ..., 2.3743e-02,
           6.9824e-02, 2.4887e-02]],

         [[2.8296e-01, 2.0695e-03, 2.4872e-03,  ..., 4.6570e-02,
           4.9561e-02, 6.8787e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9248e-01, 3.3069e-04, 1.8263e-04,  ..., 1.2006e-01,
           1.2195e-01, 5.8502e-02]],

         [[8.0273e-01, 3.1972e-04, 4.7147e-05,  ..., 1.0674e-02,
           2.5349e-03, 3.6957e-02]],

         [[8.6719e-01, 7.4983e-05, 2.4855e-05,  ..., 1.0338e-02,
           3.7727e-03, 1.0666e-02]],

         ...,

         [[5.8716e-02, 1.7462e-03, 3.2616e-03,  ..., 4.6191e-01,
           1.8372e-01, 1.6571e-02]],

         [[3.8916e-01, 1.1164e-04, 1.0651e-04,  ..., 3.4733e-03,
           2.7740e-02, 2.9541e-02]],

         [[3.0933e-01, 1.5736e-03, 5.0392e-03,  ..., 1.0693e-01,
           1.6052e-01, 4.0253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9907e-01, 4.1914e-04, 9.2983e-04,  ..., 1.7960e-02,
           7.3303e-02, 8.3069e-02]],

         [[1.6553e-01, 5.6982e-04, 1.8387e-03,  ..., 2.8763e-02,
           1.0858e-01, 6.7932e-02]],

         [[2.7661e-01, 1.2004e-04, 5.7745e-04,  ..., 4.2755e-02,
           7.6782e-02, 5.5328e-02]],

         ...,

         [[4.4785e-03, 2.7061e-05, 6.0022e-05,  ..., 2.7599e-03,
           4.4525e-02, 3.8319e-03]],

         [[4.9585e-01, 4.4870e-04, 6.8951e-04,  ..., 4.2229e-03,
           4.5410e-02, 6.6040e-02]],

         [[5.9521e-01, 4.5013e-04, 6.9189e-04,  ..., 6.1188e-03,
           6.7871e-02, 3.5767e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3928e-01, 1.1158e-04, 1.0610e-04,  ..., 2.6932e-03,
           6.9458e-02, 1.4819e-01]],

         [[2.6367e-01, 4.3559e-04, 6.5899e-04,  ..., 6.3171e-02,
           9.4055e-02, 3.5706e-02]],

         [[3.4131e-01, 9.0957e-05, 3.2377e-04,  ..., 1.9257e-02,
           1.2268e-01, 4.0527e-01]],

         ...,

         [[8.3398e-01, 9.0480e-05, 2.4509e-04,  ..., 5.7983e-03,
           1.8005e-02, 1.8723e-02]],

         [[6.3818e-01, 2.4652e-04, 1.9703e-03,  ..., 3.9482e-03,
           1.7548e-02, 5.7098e-02]],

         [[4.1846e-01, 3.3665e-04, 3.5548e-04,  ..., 4.2305e-03,
           1.3344e-02, 2.8473e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9250e-01, 2.5129e-04, 3.4499e-04,  ..., 1.0773e-02,
           8.6731e-02, 1.4526e-01]],

         [[2.4826e-02, 2.6762e-05, 4.7743e-05,  ..., 1.1253e-03,
           1.4366e-02, 1.0109e-02]],

         [[5.1514e-01, 4.6802e-04, 1.1358e-03,  ..., 1.0735e-02,
           4.4312e-02, 4.2969e-02]],

         ...,

         [[6.3525e-01, 1.1921e-07, 1.6153e-05,  ..., 4.0619e-02,
           2.6904e-01, 2.0416e-02]],

         [[3.1763e-01, 1.0371e-05, 7.9811e-05,  ..., 1.3245e-01,
           1.1871e-01, 4.4006e-02]],

         [[5.3320e-01, 7.1526e-07, 8.1241e-05,  ..., 1.0834e-01,
           3.2349e-01, 1.4664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1067e-02, 3.6120e-05, 1.0407e-04,  ..., 2.6367e-02,
           1.9946e-01, 2.8076e-02]],

         [[7.7133e-03, 1.3113e-06, 8.1062e-06,  ..., 4.5967e-04,
           7.9575e-03, 9.6387e-01]],

         [[1.6504e-01, 3.9816e-04, 1.3733e-03,  ..., 6.4636e-02,
           3.9978e-02, 2.6733e-02]],

         ...,

         [[2.7783e-01, 6.9761e-04, 1.2894e-03,  ..., 1.4136e-01,
           2.3035e-01, 2.4857e-02]],

         [[5.9961e-01, 1.3471e-05, 2.3305e-05,  ..., 1.0689e-02,
           7.9895e-02, 7.0496e-03]],

         [[2.1973e-01, 5.6684e-05, 4.4751e-04,  ..., 2.4707e-01,
           4.2358e-01, 6.9946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1885e-01, 1.5497e-04, 5.8937e-04,  ..., 5.2765e-02,
           3.4332e-02, 1.8234e-02]],

         [[9.8340e-01, 2.5094e-05, 1.6844e-04,  ..., 2.4471e-03,
           8.2922e-04, 2.7990e-04]],

         [[2.2925e-01, 5.7042e-05, 2.9254e-04,  ..., 4.4518e-03,
           6.9771e-03, 2.3556e-03]],

         ...,

         [[6.6162e-01, 4.4405e-05, 7.0989e-05,  ..., 5.6801e-03,
           1.5442e-02, 3.2196e-02]],

         [[1.0175e-01, 1.2875e-05, 1.0431e-05,  ..., 2.3384e-03,
           1.0727e-02, 2.4551e-02]],

         [[1.4062e-01, 3.1173e-05, 5.9605e-05,  ..., 1.6785e-02,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5759e-01, 5.1498e-05, 1.0884e-04,  ..., 6.9092e-02,
           1.8567e-01, 7.6782e-02]],

         [[6.9873e-01, 7.7665e-05, 1.1289e-04,  ..., 8.6212e-03,
           1.5587e-02, 3.3722e-02]],

         [[2.4121e-01, 1.0526e-04, 4.7612e-04,  ..., 6.2927e-02,
           1.3745e-01, 3.3740e-01]],

         ...,

         [[5.8252e-01, 1.0622e-04, 4.8256e-04,  ..., 3.6957e-02,
           6.2347e-02, 5.1697e-02]],

         [[3.8867e-01, 7.2718e-05, 6.4945e-04,  ..., 2.1255e-02,
           1.0223e-01, 1.7798e-01]],

         [[5.7129e-01, 1.6034e-05, 9.0897e-05,  ..., 4.4327e-03,
           2.2507e-02, 6.0211e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4465e-02, 9.8348e-06, 1.0252e-05,  ..., 3.3760e-04,
           3.8338e-03, 1.3275e-02]],

         [[2.6642e-02, 3.7849e-05, 5.5194e-05,  ..., 1.1940e-03,
           1.5793e-02, 4.5319e-02]],

         [[1.0657e-01, 1.5163e-04, 6.7711e-04,  ..., 2.6321e-02,
           1.0992e-01, 2.6782e-01]],

         ...,

         [[1.1658e-01, 3.7313e-05, 3.3021e-04,  ..., 2.7979e-01,
           2.7979e-01, 9.3689e-02]],

         [[2.2241e-01, 1.7345e-04, 1.1673e-03,  ..., 1.0504e-01,
           2.0251e-01, 1.6272e-01]],

         [[9.4775e-01, 4.6909e-05, 1.2851e-04,  ..., 5.5504e-03,
           7.1259e-03, 1.1566e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.4617e-05, 3.8505e-05,  ..., 2.0401e-02,
           2.9739e-02, 2.5732e-01]],

         [[1.9263e-01, 5.2333e-05, 1.1748e-04,  ..., 1.1548e-01,
           4.5142e-01, 1.0229e-01]],

         [[1.0815e-01, 2.5034e-06, 2.3603e-05,  ..., 9.5764e-02,
           6.8359e-01, 8.4229e-02]],

         ...,

         [[3.1421e-01, 2.6655e-04, 1.6613e-03,  ..., 1.4465e-02,
           3.0869e-02, 7.2876e-02]],

         [[3.3569e-01, 2.7537e-04, 5.0068e-04,  ..., 4.5074e-02,
           9.3201e-02, 1.9531e-02]],

         [[3.7018e-02, 1.0234e-04, 1.0520e-04,  ..., 3.1338e-03,
           2.9739e-02, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4297e-01, 6.4254e-05, 4.7040e-04,  ..., 8.4229e-02,
           8.8623e-02, 1.0565e-01]],

         [[1.6418e-01, 3.1147e-03, 4.4098e-03,  ..., 1.3420e-02,
           3.7506e-02, 4.7211e-02]],

         [[2.8702e-02, 3.5524e-05, 1.2290e-04,  ..., 2.1973e-03,
           1.5610e-02, 9.0332e-03]],

         ...,

         [[8.9795e-01, 1.2648e-04, 1.7846e-04,  ..., 5.0926e-03,
           4.8599e-03, 1.3008e-02]],

         [[1.8799e-01, 1.1420e-04, 2.3067e-04,  ..., 8.1329e-03,
           4.8279e-02, 7.0947e-01]],

         [[6.0303e-01, 4.1068e-05, 1.9598e-04,  ..., 3.7933e-02,
           9.1003e-02, 1.6748e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 9.4414e-05, 1.9455e-03,  ..., 1.9379e-02,
           1.5602e-02, 4.4952e-02]],

         [[6.6772e-02, 6.5863e-05, 2.9874e-04,  ..., 7.7591e-03,
           1.5434e-02, 3.8055e-02]],

         [[2.1533e-01, 2.0103e-03, 4.0092e-03,  ..., 1.9763e-01,
           1.0956e-01, 8.3374e-02]],

         ...,

         [[7.7490e-01, 2.6464e-04, 7.2718e-04,  ..., 1.3756e-02,
           1.3786e-02, 3.1799e-02]],

         [[7.5879e-01, 1.0073e-05, 1.9302e-03,  ..., 1.6891e-02,
           1.4732e-02, 1.6464e-02]],

         [[6.7041e-01, 6.9261e-05, 1.6987e-04,  ..., 1.4671e-02,
           1.4503e-02, 5.2826e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9888e-03, 2.8400e-03, 6.5918e-02,  ..., 4.1260e-02,
           2.8122e-02, 4.6021e-02]],

         [[2.5344e-04, 2.6870e-04, 1.3086e-01,  ..., 8.4991e-03,
           1.4069e-02, 5.0873e-02]],

         [[5.5075e-04, 5.0402e-04, 7.9651e-02,  ..., 1.6769e-02,
           2.7740e-02, 1.0803e-01]],

         ...,

         [[1.6069e-04, 9.0122e-05, 1.5186e-01,  ..., 5.1270e-02,
           1.6174e-01, 3.0688e-01]],

         [[6.6757e-04, 6.5470e-04, 2.4475e-01,  ..., 4.3365e-02,
           5.0110e-02, 6.6895e-02]],

         [[1.4663e-05, 2.5630e-06, 1.0864e-02,  ..., 4.3130e-04,
           1.1816e-03, 7.4654e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3188e-04, 1.4496e-04, 2.1484e-01,  ..., 3.5973e-03,
           1.5564e-02, 1.0309e-01]],

         [[2.3603e-04, 1.3721e-04, 1.0962e-01,  ..., 1.4717e-02,
           9.2850e-03, 4.7150e-02]],

         [[1.4353e-04, 2.9266e-05, 2.9468e-01,  ..., 4.8370e-03,
           1.3405e-02, 7.5073e-02]],

         ...,

         [[1.0796e-03, 4.0269e-04, 1.4172e-01,  ..., 1.4839e-03,
           9.7961e-03, 5.5695e-02]],

         [[8.3447e-05, 9.7752e-05, 2.5317e-01,  ..., 2.7485e-03,
           9.1553e-03, 7.1411e-02]],

         [[4.3690e-05, 4.1425e-05, 2.5781e-01,  ..., 3.5736e-02,
           1.4026e-01, 1.0339e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-05, 9.8944e-05, 1.6516e-01,  ..., 5.2214e-04,
           1.5945e-02, 6.2073e-02]],

         [[1.6975e-04, 7.3910e-06, 7.6721e-02,  ..., 4.3983e-03,
           6.9695e-03, 4.4067e-02]],

         [[3.8743e-04, 4.8423e-04, 1.7200e-01,  ..., 1.1520e-03,
           9.2621e-03, 6.1829e-02]],

         ...,

         [[5.0664e-06, 1.5676e-05, 1.7871e-01,  ..., 4.5090e-03,
           1.4328e-02, 1.5527e-01]],

         [[4.2796e-05, 1.1128e-04, 3.9398e-02,  ..., 3.4409e-03,
           8.9951e-03, 3.1281e-02]],

         [[6.0797e-04, 1.4114e-04, 1.0101e-01,  ..., 2.3479e-03,
           8.5220e-03, 5.1605e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3358e-04, 1.0800e-04, 1.9638e-02,  ..., 9.9564e-03,
           1.3525e-01, 5.3516e-01]],

         [[5.8830e-05, 2.2531e-04, 2.3608e-01,  ..., 6.7383e-02,
           1.3196e-01, 6.2561e-02]],

         [[5.5850e-05, 3.8290e-04, 2.9468e-01,  ..., 8.1848e-02,
           4.2938e-02, 5.5046e-03]],

         ...,

         [[7.4053e-04, 5.3835e-04, 8.2458e-02,  ..., 1.1316e-01,
           2.7084e-02, 8.7769e-02]],

         [[1.6088e-03, 3.5715e-04, 4.0619e-02,  ..., 2.0703e-01,
           1.8701e-01, 3.6060e-01]],

         [[1.6534e-04, 3.7074e-05, 5.3558e-02,  ..., 2.2507e-02,
           7.1228e-02, 1.6174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8553e-03, 9.3699e-04, 1.3599e-01,  ..., 1.5015e-02,
           3.7323e-02, 7.0801e-02]],

         [[2.8276e-04, 9.8038e-04, 2.4902e-01,  ..., 5.4779e-03,
           2.7496e-02, 7.7759e-02]],

         [[1.3762e-03, 5.2786e-04, 1.8665e-01,  ..., 1.8478e-02,
           4.2725e-02, 4.0039e-02]],

         ...,

         [[2.0313e-03, 2.7919e-04, 1.2372e-01,  ..., 1.3458e-02,
           2.0203e-02, 3.7524e-01]],

         [[6.6805e-04, 1.4706e-03, 2.5732e-01,  ..., 8.2493e-04,
           4.6005e-03, 1.4771e-01]],

         [[9.2125e-04, 7.6962e-04, 1.5979e-01,  ..., 4.4037e-02,
           4.7607e-02, 1.6235e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7180e-05, 6.6757e-06, 2.6054e-03,  ..., 2.2729e-01,
           4.5215e-01, 6.7200e-02]],

         [[5.5122e-04, 2.9516e-04, 4.3373e-03,  ..., 1.2671e-01,
           2.2241e-01, 3.9038e-01]],

         [[0.0000e+00, 0.0000e+00, 4.7684e-07,  ..., 2.4395e-03,
           1.0933e-02, 9.8438e-01]],

         ...,

         [[1.6510e-05, 3.6955e-06, 4.4882e-05,  ..., 2.4509e-03,
           2.4509e-03, 9.8877e-01]],

         [[6.0043e-03, 8.2016e-03, 5.2986e-03,  ..., 2.3743e-02,
           1.6479e-01, 1.0536e-02]],

         [[9.8495e-03, 3.5114e-03, 8.8787e-04,  ..., 5.9738e-03,
           2.9388e-02, 7.7454e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7980e-04, 6.5613e-04, 6.0349e-03,  ..., 7.2388e-02,
           1.2311e-01, 5.5176e-01]],

         [[0.0000e+00, 0.0000e+00, 9.5367e-07,  ..., 1.3989e-01,
           7.8027e-01, 7.2571e-02]],

         [[7.8440e-04, 1.3552e-03, 1.2077e-02,  ..., 1.4954e-01,
           1.4038e-01, 9.3567e-02]],

         ...,

         [[2.1458e-03, 3.8261e-03, 7.0381e-03,  ..., 2.7405e-02,
           1.5161e-01, 5.1709e-01]],

         [[2.3270e-03, 9.8572e-03, 3.6194e-02,  ..., 1.6040e-01,
           1.5002e-01, 2.2919e-02]],

         [[1.3390e-03, 2.7905e-03, 1.1841e-02,  ..., 7.4280e-02,
           3.2764e-01, 2.1155e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8441e-02, 7.9727e-03, 2.9617e-02,  ..., 4.4464e-02,
           8.0505e-02, 4.6234e-02]],

         [[8.7976e-04, 4.2868e-04, 1.7023e-03,  ..., 1.9943e-02,
           5.8517e-03, 2.2907e-03]],

         [[1.5053e-02, 5.0011e-03, 6.7825e-03,  ..., 1.3098e-01,
           5.9998e-02, 3.3661e-02]],

         ...,

         [[2.4490e-03, 1.9407e-04, 1.0328e-03,  ..., 1.6003e-01,
           2.4585e-01, 3.2080e-01]],

         [[1.9760e-02, 3.8166e-03, 4.8637e-03,  ..., 1.4270e-01,
           1.7896e-01, 7.8186e-02]],

         [[7.3776e-03, 7.3051e-04, 1.2070e-02,  ..., 1.5173e-01,
           2.8564e-01, 1.7188e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 2.8014e-06, 9.3877e-05,  ..., 4.7035e-03,
           3.1033e-03, 1.6052e-02]],

         [[9.2188e-01, 4.7684e-07, 2.8551e-05,  ..., 2.7115e-02,
           2.1561e-02, 1.1696e-02]],

         [[9.7314e-01, 1.2338e-05, 4.2915e-05,  ..., 1.2150e-03,
           3.4161e-03, 1.8320e-03]],

         ...,

         [[2.9114e-02, 1.9670e-05, 1.2422e-04,  ..., 1.6713e-04,
           1.6129e-04, 2.3708e-03]],

         [[6.5869e-01, 4.5204e-04, 1.1892e-03,  ..., 1.1505e-02,
           1.3557e-02, 7.7095e-03]],

         [[3.0103e-01, 3.4165e-04, 3.7813e-04,  ..., 1.4084e-02,
           2.0721e-02, 2.6718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9575e-01, 1.5497e-06, 4.8280e-06,  ..., 2.8706e-03,
           5.1928e-04, 9.1782e-03]],

         [[8.4668e-01, 0.0000e+00, 5.9605e-08,  ..., 6.9275e-03,
           9.5093e-02, 1.6403e-02]],

         [[9.2285e-01, 3.2485e-05, 1.3232e-05,  ..., 1.8244e-03,
           3.4668e-02, 3.6697e-03]],

         ...,

         [[2.8174e-01, 6.0129e-04, 1.8942e-04,  ..., 2.4573e-01,
           9.2529e-02, 6.6490e-03]],

         [[8.8745e-02, 9.9087e-04, 4.1199e-04,  ..., 3.1226e-01,
           8.8074e-02, 5.6343e-03]],

         [[6.3086e-01, 2.8551e-05, 8.0347e-05,  ..., 3.4576e-02,
           1.9543e-01, 5.5450e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7456e-01, 2.7227e-04, 1.5688e-04,  ..., 1.2793e-01,
           1.0730e-01, 7.1228e-02]],

         [[7.1680e-01, 1.7166e-05, 2.2590e-05,  ..., 7.8613e-02,
           1.3110e-01, 1.5747e-02]],

         [[8.8135e-01, 2.1160e-05, 1.2159e-05,  ..., 7.8583e-03,
           6.3232e-02, 1.5236e-02]],

         ...,

         [[4.9756e-01, 1.0133e-06, 2.9802e-07,  ..., 3.5524e-05,
           5.1117e-04, 4.8145e-01]],

         [[6.7773e-01, 8.7023e-06, 8.7023e-06,  ..., 8.1658e-06,
           2.0742e-05, 7.0953e-03]],

         [[9.7559e-01, 8.3447e-07, 5.9605e-07,  ..., 2.3842e-07,
           7.1526e-07, 1.2865e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7578e-01, 1.4842e-05, 5.0068e-06,  ..., 4.9683e-02,
           1.0162e-01, 1.1285e-01]],

         [[8.7891e-01, 1.5736e-05, 3.4571e-06,  ..., 2.4986e-04,
           5.3072e-04, 1.1620e-02]],

         [[7.2998e-01, 2.4629e-04, 5.7578e-05,  ..., 2.6531e-03,
           1.4641e-02, 9.8877e-02]],

         ...,

         [[8.9600e-01, 3.7122e-04, 8.9884e-05,  ..., 1.6317e-03,
           6.7444e-03, 1.5335e-02]],

         [[9.4385e-01, 6.4075e-05, 1.6570e-05,  ..., 1.7319e-03,
           1.5617e-02, 2.3590e-02]],

         [[9.1406e-01, 1.4281e-04, 1.8275e-04,  ..., 1.1349e-03,
           7.4196e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3838e-01, 6.2227e-05, 8.5771e-05,  ..., 2.3003e-03,
           2.0180e-03, 1.6739e-02]],

         [[7.6416e-01, 2.6441e-04, 8.2552e-05,  ..., 4.5433e-03,
           5.7220e-03, 2.9053e-02]],

         [[9.8486e-01, 1.0729e-06, 1.0133e-06,  ..., 4.5252e-04,
           1.5545e-03, 6.7673e-03]],

         ...,

         [[6.1963e-01, 1.0067e-04, 3.0339e-05,  ..., 1.0246e-02,
           1.1011e-01, 3.1403e-02]],

         [[9.1553e-01, 2.5439e-04, 2.8074e-05,  ..., 8.0032e-03,
           7.4348e-03, 2.8439e-03]],

         [[8.8232e-01, 1.6034e-05, 4.2558e-05,  ..., 5.9814e-03,
           2.6443e-02, 5.1758e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2837e-01, 9.7132e-04, 2.5535e-04,  ..., 1.2352e-02,
           6.6223e-03, 3.1357e-03]],

         [[9.8145e-01, 7.9274e-06, 1.3828e-05,  ..., 1.1263e-03,
           3.5496e-03, 2.0027e-04]],

         [[9.3018e-01, 8.3447e-06, 1.1265e-05,  ..., 3.2663e-04,
           3.5596e-04, 8.4734e-04]],

         ...,

         [[8.0420e-01, 2.9027e-05, 2.4438e-05,  ..., 1.2474e-03,
           1.6068e-02, 9.5581e-02]],

         [[9.6484e-01, 1.6093e-06, 2.7418e-06,  ..., 1.5621e-03,
           1.3039e-02, 6.5460e-03]],

         [[9.2334e-01, 1.1444e-05, 1.0967e-05,  ..., 3.9864e-04,
           1.2112e-03, 1.6281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9712e-02, 3.3200e-05, 2.7776e-04,  ..., 2.6880e-01,
           1.4612e-01, 1.7319e-02]],

         [[9.5410e-01, 4.3511e-06, 1.4246e-05,  ..., 6.6147e-03,
           6.6414e-03, 4.3030e-03]],

         [[7.9980e-01, 1.3232e-05, 5.0843e-05,  ..., 2.2202e-03,
           2.3132e-02, 2.6321e-02]],

         ...,

         [[9.0088e-01, 2.5034e-05, 4.3392e-05,  ..., 3.8757e-02,
           2.3117e-03, 1.0548e-03]],

         [[8.0859e-01, 9.6381e-05, 4.3750e-05,  ..., 1.0500e-03,
           1.6127e-03, 1.8921e-02]],

         [[8.5645e-01, 5.6028e-06, 7.3314e-06,  ..., 2.6264e-03,
           3.3478e-02, 2.3289e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6992e-01, 8.3447e-07, 5.3644e-06,  ..., 7.6294e-04,
           1.4853e-04, 9.4986e-03]],

         [[7.8125e-01, 4.4048e-05, 5.0724e-05,  ..., 2.4155e-02,
           5.7617e-02, 7.3242e-03]],

         [[7.2998e-01, 0.0000e+00, 1.7881e-07,  ..., 2.7161e-03,
           1.4200e-03, 1.5480e-02]],

         ...,

         [[7.8760e-01, 1.9073e-04, 8.5950e-05,  ..., 1.7471e-02,
           1.7242e-02, 6.6299e-03]],

         [[6.2939e-01, 7.6771e-05, 9.4414e-05,  ..., 2.6073e-03,
           5.6580e-02, 4.6844e-02]],

         [[3.2715e-01, 5.6839e-04, 1.0914e-04,  ..., 3.5339e-02,
           1.3832e-02, 1.0971e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1667e-01, 7.7486e-06, 5.7220e-06,  ..., 6.3944e-04,
           3.8280e-03, 7.6807e-01]],

         [[4.8730e-01, 3.8123e-04, 4.8780e-04,  ..., 1.5793e-02,
           1.8492e-03, 1.6098e-02]],

         [[7.8076e-01, 2.1708e-04, 5.3883e-04,  ..., 2.4475e-02,
           1.6266e-02, 5.2605e-03]],

         ...,

         [[6.6260e-01, 2.0468e-04, 2.2054e-04,  ..., 3.3493e-03,
           7.5188e-03, 2.8934e-03]],

         [[8.5107e-01, 1.6749e-05, 2.8253e-05,  ..., 2.0771e-03,
           9.2850e-03, 5.4138e-02]],

         [[4.7046e-01, 8.9407e-06, 3.8803e-05,  ..., 6.8893e-03,
           5.1849e-02, 3.6035e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9038e-01, 6.0797e-05, 8.5413e-05,  ..., 2.2812e-03,
           9.2010e-03, 3.3887e-01]],

         [[3.8428e-01, 1.8537e-04, 1.5795e-04,  ..., 1.6037e-02,
           1.2170e-01, 3.8109e-03]],

         [[4.4287e-01, 9.9778e-05, 5.0962e-05,  ..., 1.1377e-03,
           4.0016e-03, 7.9422e-03]],

         ...,

         [[1.6113e-02, 2.5630e-06, 2.1458e-06,  ..., 3.6061e-05,
           9.1219e-04, 8.0713e-01]],

         [[4.7095e-01, 5.9366e-05, 9.9421e-05,  ..., 1.8692e-03,
           2.1103e-02, 5.7983e-02]],

         [[5.0879e-01, 9.1195e-05, 1.1802e-05,  ..., 5.9223e-04,
           1.1482e-03, 7.0496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3564e-01, 1.0605e-03, 5.5695e-04,  ..., 6.6910e-03,
           1.0475e-02, 3.4103e-03]],

         [[6.5967e-01, 1.0806e-04, 3.9387e-04,  ..., 5.1636e-02,
           1.7249e-01, 5.6648e-03]],

         [[4.5825e-01, 1.4007e-05, 6.0081e-05,  ..., 1.2360e-02,
           1.1948e-02, 6.9466e-03]],

         ...,

         [[3.6035e-01, 0.0000e+00, 5.9605e-08,  ..., 5.8197e-02,
           5.0342e-01, 4.3701e-02]],

         [[4.0845e-01, 7.9441e-04, 5.9414e-04,  ..., 2.4597e-02,
           1.6693e-02, 8.3542e-03]],

         [[8.9014e-01, 2.9802e-07, 6.5565e-07,  ..., 2.5043e-03,
           1.4832e-02, 5.3955e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4180e-01, 6.5267e-05, 1.2159e-05,  ..., 7.4673e-04,
           3.5686e-03, 9.0942e-03]],

         [[6.9031e-02, 1.2791e-04, 1.0622e-04,  ..., 2.9590e-01,
           3.1250e-01, 1.1663e-03]],

         [[1.5662e-01, 1.3053e-04, 6.0618e-05,  ..., 1.9684e-02,
           1.8530e-01, 2.0103e-03]],

         ...,

         [[7.0996e-01, 6.5565e-07, 2.8014e-06,  ..., 4.4922e-02,
           1.0400e-01, 6.3843e-02]],

         [[5.9180e-01, 1.7881e-07, 4.1723e-07,  ..., 1.6464e-02,
           1.8066e-01, 1.7554e-01]],

         [[2.9761e-01, 3.0398e-06, 2.8014e-06,  ..., 2.6025e-01,
           2.7734e-01, 1.4610e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7415e-02, 5.3644e-07, 8.3447e-07,  ..., 2.3594e-03,
           4.6272e-03, 9.4287e-01]],

         [[1.0089e-01, 1.1981e-05, 3.6478e-05,  ..., 2.7759e-01,
           4.5581e-01, 3.5980e-02]],

         [[4.3066e-01, 3.6931e-04, 4.5919e-04,  ..., 9.7229e-02,
           2.3102e-02, 8.4457e-03]],

         ...,

         [[2.4072e-01, 8.1062e-05, 4.0591e-05,  ..., 4.2534e-03,
           6.6757e-03, 9.6436e-03]],

         [[5.9180e-01, 1.2994e-05, 1.2994e-05,  ..., 1.0086e-02,
           1.1627e-02, 1.0040e-02]],

         [[4.3213e-01, 1.0967e-04, 6.2227e-05,  ..., 9.2554e-04,
           4.5276e-04, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1025e-01, 5.8353e-05, 9.2328e-05,  ..., 8.8120e-03,
           4.5624e-02, 4.4739e-02]],

         [[3.5400e-01, 9.4318e-04, 4.1389e-04,  ..., 6.8207e-03,
           1.6495e-02, 1.7151e-02]],

         [[2.5488e-01, 4.0650e-05, 1.6510e-05,  ..., 3.0746e-03,
           1.0483e-02, 1.9653e-02]],

         ...,

         [[6.9775e-01, 1.8895e-05, 1.4246e-05,  ..., 1.6966e-03,
           1.2489e-02, 9.8801e-03]],

         [[5.0977e-01, 3.7909e-05, 3.4928e-05,  ..., 4.8599e-03,
           5.2063e-02, 8.8348e-03]],

         [[7.3584e-01, 5.2452e-05, 3.2842e-05,  ..., 5.7831e-03,
           6.3667e-03, 1.7105e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8174e-01, 6.5708e-04, 4.8828e-04,  ..., 9.0027e-02,
           1.3939e-02, 2.7847e-03]],

         [[3.3618e-01, 1.2314e-04, 6.6519e-05,  ..., 7.1669e-04,
           5.6553e-04, 6.6681e-03]],

         [[6.1230e-01, 1.3947e-05, 1.7226e-05,  ..., 7.7362e-03,
           2.4872e-02, 1.5656e-02]],

         ...,

         [[1.6382e-01, 2.7990e-04, 6.9797e-05,  ..., 1.8959e-03,
           7.4272e-03, 6.2927e-02]],

         [[9.5398e-02, 3.6538e-05, 3.0458e-05,  ..., 3.2187e-05,
           3.7372e-05, 6.7520e-03]],

         [[8.5059e-01, 1.7929e-04, 8.3387e-05,  ..., 4.0665e-03,
           7.0496e-03, 7.6818e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2441e-01, 2.6393e-04, 3.2592e-04,  ..., 5.6122e-02,
           2.5562e-01, 7.1907e-03]],

         [[2.5317e-01, 7.7820e-04, 3.5620e-04,  ..., 1.3741e-02,
           1.0658e-02, 1.6647e-02]],

         [[2.2253e-01, 1.2279e-05, 1.1265e-05,  ..., 4.7646e-03,
           3.6041e-02, 2.4060e-01]],

         ...,

         [[5.0195e-01, 1.6868e-05, 1.7285e-05,  ..., 4.1687e-02,
           3.7079e-02, 1.4580e-02]],

         [[8.2031e-01, 9.4771e-06, 1.1563e-05,  ..., 4.4403e-03,
           7.6141e-03, 3.4119e-02]],

         [[2.0007e-01, 3.3438e-05, 2.7061e-05,  ..., 8.7891e-03,
           1.3298e-02, 3.0518e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 8.3089e-05, 4.5180e-05,  ..., 2.0771e-03,
           1.5106e-02, 1.5587e-02]],

         [[2.0837e-01, 3.3021e-05, 4.8041e-05,  ..., 1.3245e-01,
           6.8726e-02, 1.3321e-02]],

         [[1.2262e-01, 7.2718e-06, 3.4571e-06,  ..., 2.5864e-03,
           3.0231e-03, 7.9980e-01]],

         ...,

         [[1.6931e-01, 2.3723e-04, 1.3208e-04,  ..., 1.1148e-03,
           3.8958e-04, 2.3453e-02]],

         [[4.9487e-01, 3.7575e-04, 2.0945e-04,  ..., 5.0468e-03,
           6.1569e-03, 9.1705e-03]],

         [[3.5010e-01, 6.6280e-04, 5.2023e-04,  ..., 4.3907e-03,
           8.3313e-03, 4.5471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7920e-01, 5.1260e-05, 2.3842e-05,  ..., 1.9653e-02,
           1.7883e-02, 5.1758e-02]],

         [[2.4976e-01, 2.1338e-05, 4.5300e-06,  ..., 6.2523e-03,
           9.0075e-04, 7.5417e-03]],

         [[8.0615e-01, 5.3644e-05, 2.4676e-05,  ..., 8.7509e-03,
           1.8196e-03, 2.9297e-03]],

         ...,

         [[1.3147e-01, 1.3943e-03, 1.9665e-03,  ..., 1.6882e-01,
           1.7120e-02, 7.8354e-03]],

         [[8.1201e-01, 2.8014e-05, 5.7101e-05,  ..., 1.6680e-03,
           1.0443e-03, 2.6505e-02]],

         [[4.9976e-01, 9.0647e-04, 1.6289e-03,  ..., 8.5510e-02,
           3.1464e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9321e-01, 3.0303e-04, 5.8842e-04,  ..., 4.8615e-02,
           4.2908e-02, 1.3924e-02]],

         [[1.9165e-01, 3.5286e-04, 4.4966e-04,  ..., 5.3192e-02,
           2.5925e-02, 3.3295e-02]],

         [[2.3865e-01, 5.0664e-05, 1.4043e-04,  ..., 2.2537e-02,
           1.9745e-02, 2.6978e-02]],

         ...,

         [[5.1208e-02, 8.5890e-05, 1.5068e-04,  ..., 2.4185e-02,
           9.9258e-03, 1.2352e-02]],

         [[6.4087e-02, 4.7684e-05, 4.8459e-05,  ..., 7.1192e-04,
           1.3514e-03, 5.7434e-02]],

         [[3.8501e-01, 2.5702e-04, 5.0688e-04,  ..., 1.7731e-02,
           1.1269e-02, 9.9487e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1924e-01, 2.1756e-05, 2.0921e-05,  ..., 3.8319e-03,
           4.5166e-03, 3.7811e-02]],

         [[3.2227e-01, 1.3459e-04, 2.7084e-04,  ..., 2.0599e-02,
           3.1647e-02, 3.5248e-03]],

         [[2.5122e-01, 2.5439e-04, 5.8031e-04,  ..., 6.9702e-02,
           9.0942e-02, 3.1006e-01]],

         ...,

         [[4.9463e-01, 1.6403e-04, 2.8014e-04,  ..., 3.7262e-02,
           1.7319e-02, 2.7039e-02]],

         [[1.1444e-01, 4.7088e-06, 5.0843e-05,  ..., 6.5422e-04,
           8.3351e-04, 7.0496e-02]],

         [[1.9141e-01, 2.8563e-04, 5.3358e-04,  ..., 4.9057e-03,
           4.0665e-03, 1.0777e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6636e-01, 2.3234e-04, 1.7917e-04,  ..., 1.3741e-02,
           9.5139e-03, 9.4238e-02]],

         [[1.0718e-01, 8.6427e-05, 1.2851e-04,  ..., 1.4503e-02,
           8.7280e-03, 7.2510e-02]],

         [[3.5571e-01, 2.5153e-04, 5.8508e-04,  ..., 2.6688e-02,
           1.9516e-02, 3.3356e-02]],

         ...,

         [[7.1680e-01, 0.0000e+00, 1.1921e-07,  ..., 6.2164e-02,
           1.3257e-01, 7.3792e-02]],

         [[4.2236e-01, 9.5367e-06, 4.2677e-05,  ..., 1.7883e-01,
           1.7200e-01, 3.1067e-02]],

         [[2.0715e-01, 1.7881e-07, 3.5763e-06,  ..., 5.4016e-02,
           7.0068e-01, 1.8677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0742e-02, 2.3842e-06, 9.5367e-06,  ..., 5.1239e-02,
           1.7853e-02, 4.0527e-02]],

         [[9.8114e-03, 3.5763e-07, 2.3842e-07,  ..., 8.5533e-05,
           1.9779e-03, 9.7754e-01]],

         [[5.3864e-02, 1.6809e-04, 7.6246e-04,  ..., 1.5222e-01,
           6.3965e-02, 2.5578e-03]],

         ...,

         [[1.6907e-01, 3.0899e-04, 7.9298e-04,  ..., 3.4277e-01,
           5.4688e-02, 7.5455e-03]],

         [[4.7681e-01, 1.0848e-05, 3.5822e-05,  ..., 7.6660e-02,
           1.2537e-01, 1.3928e-01]],

         [[1.0913e-01, 8.9407e-07, 2.1279e-05,  ..., 1.7993e-01,
           4.4873e-01, 1.8555e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 7.6294e-05, 3.8791e-04,  ..., 4.9042e-02,
           2.1515e-02, 1.2253e-02]],

         [[9.5264e-01, 2.7895e-05, 1.1194e-04,  ..., 2.8248e-03,
           3.9005e-04, 5.7411e-04]],

         [[7.7271e-02, 8.8811e-06, 4.3094e-05,  ..., 2.9964e-03,
           9.0647e-04, 1.2989e-03]],

         ...,

         [[7.6709e-01, 2.2471e-05, 2.6941e-05,  ..., 9.8114e-03,
           6.7017e-02, 4.3976e-02]],

         [[2.3840e-01, 3.8147e-06, 2.3246e-06,  ..., 5.1880e-03,
           4.3335e-03, 7.3700e-03]],

         [[7.8308e-02, 1.5557e-05, 3.9756e-05,  ..., 6.8970e-03,
           9.2087e-03, 8.2910e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9873e-01, 1.4663e-05, 2.6286e-05,  ..., 2.1826e-01,
           2.0178e-01, 3.8086e-02]],

         [[7.8467e-01, 2.1338e-05, 1.6749e-05,  ..., 4.7417e-03,
           1.9196e-02, 6.4148e-02]],

         [[2.4585e-01, 5.3823e-05, 9.3400e-05,  ..., 8.3984e-02,
           1.5503e-01, 2.7100e-01]],

         ...,

         [[5.6836e-01, 1.0788e-05, 6.8665e-05,  ..., 1.8997e-02,
           1.5784e-01, 5.5847e-02]],

         [[5.9717e-01, 2.8586e-04, 1.0414e-03,  ..., 6.3904e-02,
           6.3416e-02, 1.3504e-02]],

         [[6.3867e-01, 2.4617e-05, 1.5676e-04,  ..., 3.4393e-02,
           1.5417e-01, 1.1345e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7942e-02, 6.2227e-05, 3.7074e-05,  ..., 2.6684e-03,
           8.5449e-03, 5.6610e-02]],

         [[1.1206e-01, 4.6313e-05, 3.8505e-05,  ..., 7.1335e-03,
           8.9874e-03, 1.6144e-02]],

         [[2.4146e-01, 1.9741e-04, 6.1750e-04,  ..., 4.0985e-02,
           1.1407e-01, 1.1139e-01]],

         ...,

         [[6.8298e-02, 2.2948e-05, 1.9038e-04,  ..., 3.6377e-01,
           8.5022e-02, 1.8349e-03]],

         [[1.5430e-01, 7.8917e-05, 3.4285e-04,  ..., 2.2095e-01,
           3.7012e-01, 3.8033e-03]],

         [[5.8936e-01, 1.4019e-04, 2.8110e-04,  ..., 4.1382e-02,
           1.7554e-01, 7.5500e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2090e-01, 1.7583e-05, 5.3525e-05,  ..., 1.0727e-02,
           1.0400e-01, 3.6157e-01]],

         [[3.3594e-01, 6.7294e-05, 7.1347e-05,  ..., 1.3416e-01,
           7.9468e-02, 2.4866e-01]],

         [[5.4779e-02, 8.9407e-07, 1.1921e-06,  ..., 2.3889e-01,
           6.3428e-01, 2.9205e-02]],

         ...,

         [[5.9424e-01, 1.1760e-04, 2.6703e-04,  ..., 6.7291e-03,
           1.3916e-02, 1.0674e-02]],

         [[4.2188e-01, 2.9683e-05, 7.0632e-05,  ..., 3.7750e-02,
           6.5186e-02, 7.3738e-03]],

         [[2.1667e-02, 2.9802e-05, 1.9729e-05,  ..., 3.0022e-03,
           3.4882e-02, 8.0664e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2051e-01, 3.0041e-05, 3.3259e-04,  ..., 6.1676e-02,
           9.1858e-02, 1.5625e-01]],

         [[6.1401e-02, 3.2163e-04, 1.0405e-03,  ..., 1.1009e-02,
           1.5526e-02, 1.1673e-02]],

         [[2.5562e-01, 1.2159e-04, 4.8184e-04,  ..., 3.9185e-02,
           2.1301e-02, 3.9490e-02]],

         ...,

         [[8.3203e-01, 3.6407e-04, 2.5225e-04,  ..., 1.2245e-02,
           3.2257e-02, 2.2522e-02]],

         [[1.2756e-01, 1.2195e-04, 1.2577e-04,  ..., 8.8196e-03,
           4.3427e-02, 7.4609e-01]],

         [[6.8848e-01, 1.9073e-05, 1.6499e-04,  ..., 1.9836e-02,
           2.2351e-01, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5996e-01, 6.1154e-05, 1.1148e-03,  ..., 5.3864e-02,
           3.4180e-01, 3.1738e-02]],

         [[9.9304e-02, 1.9789e-05, 7.0274e-05,  ..., 7.5226e-03,
           1.2924e-02, 3.2471e-02]],

         [[3.4302e-01, 3.1877e-04, 2.2469e-03,  ..., 1.2280e-01,
           6.9946e-02, 3.3966e-02]],

         ...,

         [[5.8838e-01, 1.4472e-04, 1.6508e-03,  ..., 4.6814e-02,
           4.4312e-02, 6.9695e-03]],

         [[6.9727e-01, 7.8678e-06, 1.0366e-03,  ..., 3.3188e-03,
           5.9395e-03, 9.9640e-03]],

         [[6.9092e-01, 5.0485e-05, 1.0335e-04,  ..., 6.0364e-02,
           6.2042e-02, 4.5563e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1478e-04, 5.4789e-04, 1.1786e-01,  ..., 1.9699e-02,
           8.6243e-02, 6.6490e-03]],

         [[1.8024e-04, 1.1700e-04, 1.2421e-01,  ..., 2.5940e-02,
           6.9702e-02, 1.1761e-01]],

         [[6.0272e-04, 3.4213e-04, 8.1848e-02,  ..., 3.9429e-02,
           1.4880e-01, 3.9276e-02]],

         ...,

         [[1.8609e-04, 4.5240e-05, 5.5786e-02,  ..., 1.0101e-01,
           5.9961e-01, 7.3303e-02]],

         [[2.1672e-04, 8.6737e-04, 3.0396e-01,  ..., 3.6316e-02,
           2.7191e-02, 1.8654e-03]],

         [[4.9829e-05, 7.4804e-05, 4.3579e-02,  ..., 4.4518e-03,
           7.1144e-03, 1.9913e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2541e-04, 2.2972e-04, 3.0029e-01,  ..., 1.3504e-02,
           3.2928e-02, 2.9335e-03]],

         [[2.1017e-04, 3.9577e-04, 1.7126e-01,  ..., 1.0696e-02,
           2.3361e-02, 1.1475e-02]],

         [[3.6299e-05, 4.4227e-05, 2.6636e-01,  ..., 1.7166e-02,
           2.7634e-02, 1.2767e-04]],

         ...,

         [[1.4925e-03, 1.0691e-03, 2.2412e-01,  ..., 2.9526e-02,
           3.4241e-02, 6.1646e-03]],

         [[3.1888e-05, 8.3983e-05, 2.4939e-01,  ..., 4.0169e-03,
           3.0731e-02, 5.8746e-04]],

         [[6.5446e-05, 1.7107e-04, 1.1652e-01,  ..., 3.7915e-01,
           5.3772e-02, 8.0681e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6904e-04, 2.5082e-04, 1.5552e-01,  ..., 1.8356e-02,
           2.4414e-02, 9.0837e-05]],

         [[5.9938e-04, 5.7876e-05, 2.6147e-01,  ..., 1.4412e-02,
           2.0477e-02, 9.1248e-03]],

         [[4.3416e-04, 1.4772e-03, 2.4646e-01,  ..., 1.2039e-02,
           1.9608e-02, 1.5783e-03]],

         ...,

         [[1.4377e-04, 2.4331e-04, 1.4807e-01,  ..., 1.3771e-02,
           7.8918e-02, 5.7526e-02]],

         [[1.2469e-04, 7.6294e-04, 2.4207e-01,  ..., 2.0996e-02,
           1.4145e-02, 1.0513e-02]],

         [[3.6788e-04, 3.4571e-04, 1.3916e-01,  ..., 7.8201e-03,
           1.2207e-02, 6.3591e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8777e-04, 4.5359e-05, 2.9419e-02,  ..., 1.3501e-01,
           4.0283e-01, 1.9946e-01]],

         [[8.3864e-05, 2.3949e-04, 1.6138e-01,  ..., 1.7517e-01,
           5.4474e-02, 5.0964e-02]],

         [[7.0763e-04, 3.0746e-03, 1.6626e-01,  ..., 4.0100e-02,
           2.6913e-03, 2.0008e-03]],

         ...,

         [[7.0715e-04, 5.0306e-05, 1.0193e-02,  ..., 1.5535e-03,
           1.2732e-03, 8.1253e-03]],

         [[4.2076e-03, 1.4496e-04, 6.9824e-02,  ..., 1.3562e-01,
           1.5308e-01, 6.4575e-02]],

         [[1.0669e-05, 3.8147e-06, 3.3245e-03,  ..., 3.9581e-02,
           2.2552e-02, 1.2878e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3618e-03, 2.9869e-03, 1.9702e-01,  ..., 9.6970e-03,
           5.6992e-03, 4.9835e-02]],

         [[1.2531e-03, 2.1744e-03, 2.7539e-01,  ..., 1.4885e-02,
           2.1912e-02, 2.2171e-02]],

         [[2.7199e-03, 1.8635e-03, 1.8103e-01,  ..., 5.8655e-02,
           6.8176e-02, 2.5665e-02]],

         ...,

         [[4.6501e-03, 6.3658e-04, 1.7310e-01,  ..., 2.0523e-02,
           3.5187e-02, 9.4849e-02]],

         [[3.6182e-03, 1.2169e-03, 1.5503e-01,  ..., 1.8524e-02,
           1.1615e-01, 2.8735e-01]],

         [[5.0187e-05, 1.5640e-04, 5.3345e-02,  ..., 2.8549e-02,
           6.8652e-01, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2994e-05, 4.7684e-07, 2.1636e-04,  ..., 1.2695e-01,
           7.7783e-01, 4.9713e-02]],

         [[7.5340e-05, 2.4438e-05, 4.9114e-04,  ..., 9.2697e-03,
           3.4103e-03, 9.4531e-01]],

         [[5.9605e-08, 0.0000e+00, 6.1393e-06,  ..., 2.0325e-04,
           2.0325e-04, 9.9854e-01]],

         ...,

         [[3.5591e-03, 5.7335e-03, 8.1482e-03,  ..., 4.9530e-02,
           1.1856e-02, 3.6890e-01]],

         [[9.0256e-03, 1.2329e-02, 1.6861e-02,  ..., 9.1125e-02,
           4.2610e-03, 8.5571e-02]],

         [[8.6136e-03, 3.5915e-03, 1.0948e-03,  ..., 1.0828e-01,
           7.2144e-02, 2.6535e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1052e-04, 1.2302e-03, 5.3444e-03,  ..., 1.4673e-01,
           1.5869e-01, 2.7856e-01]],

         [[0.0000e+00, 0.0000e+00, 1.7881e-07,  ..., 1.5881e-01,
           7.5732e-01, 7.1533e-02]],

         [[1.0567e-03, 1.5135e-03, 7.6866e-03,  ..., 1.7224e-01,
           6.9580e-02, 4.1534e-02]],

         ...,

         [[2.0921e-05, 6.8784e-05, 1.1015e-03,  ..., 6.5674e-01,
           2.2705e-01, 2.1957e-02]],

         [[1.3304e-03, 5.8479e-03, 2.2156e-02,  ..., 1.8542e-01,
           3.6652e-02, 4.7089e-02]],

         [[1.3371e-03, 5.2071e-03, 8.3160e-03,  ..., 4.3677e-01,
           7.0190e-02, 3.2654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6711e-02, 7.9041e-03, 3.3783e-02,  ..., 8.5632e-02,
           5.7495e-02, 6.8787e-02]],

         [[3.2482e-03, 9.1600e-04, 2.8763e-03,  ..., 1.1604e-02,
           6.7177e-03, 8.0414e-03]],

         [[1.4313e-02, 4.1885e-03, 4.2458e-03,  ..., 7.0984e-02,
           4.3060e-02, 5.9326e-02]],

         ...,

         [[2.4605e-03, 1.9252e-05, 3.1590e-05,  ..., 6.0577e-02,
           7.4951e-01, 1.4197e-01]],

         [[5.4016e-03, 1.2636e-03, 1.0471e-03,  ..., 2.4451e-01,
           1.0516e-01, 1.0681e-01]],

         [[1.6571e-02, 1.3027e-03, 1.3954e-02,  ..., 9.3872e-02,
           1.2244e-01, 2.5317e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1914e-01, 1.7285e-06, 5.9009e-06,  ..., 1.3485e-03,
           9.6054e-03, 4.9706e-03]],

         [[8.6963e-01, 5.4836e-06, 8.6427e-06,  ..., 4.7150e-02,
           5.4260e-02, 4.4861e-03]],

         [[8.7988e-01, 8.0466e-06, 3.3796e-05,  ..., 1.1330e-02,
           2.1881e-02, 3.0289e-02]],

         ...,

         [[9.0454e-02, 7.4089e-05, 2.2960e-04,  ..., 8.6308e-04,
           3.1376e-04, 8.0261e-03]],

         [[6.2305e-01, 1.3232e-04, 1.1463e-03,  ..., 1.1726e-02,
           4.3396e-02, 1.4061e-02]],

         [[4.1943e-01, 2.0730e-04, 5.1260e-04,  ..., 7.6660e-02,
           2.7557e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1748e-01, 7.6890e-06, 1.2159e-05,  ..., 9.1124e-04,
           1.5259e-05, 1.6052e-02]],

         [[8.5889e-01, 0.0000e+00, 0.0000e+00,  ..., 1.7715e-02,
           1.0193e-01, 1.7578e-02]],

         [[9.4580e-01, 7.2896e-05, 7.2718e-06,  ..., 3.4580e-03,
           6.6681e-03, 1.1612e-02]],

         ...,

         [[4.1626e-01, 3.2854e-04, 2.7442e-04,  ..., 7.5073e-02,
           5.1994e-03, 5.5656e-03]],

         [[1.8970e-01, 1.0004e-03, 5.1689e-04,  ..., 7.5928e-02,
           2.6436e-03, 2.7733e-03]],

         [[7.3633e-01, 1.5438e-05, 8.7857e-05,  ..., 9.4116e-02,
           9.3933e-02, 4.0741e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 4.1056e-04, 2.0003e-04,  ..., 8.1360e-02,
           5.1636e-02, 3.0991e-02]],

         [[6.8262e-01, 5.3644e-06, 9.3579e-06,  ..., 9.3140e-02,
           1.0120e-01, 1.5480e-02]],

         [[7.4268e-01, 2.5153e-05, 4.0531e-06,  ..., 9.9426e-02,
           1.2103e-01, 2.5387e-03]],

         ...,

         [[2.3694e-01, 2.3842e-07, 4.7684e-07,  ..., 9.1851e-05,
           1.6570e-04, 7.3828e-01]],

         [[1.5894e-01, 1.7881e-07, 5.1260e-06,  ..., 8.3447e-07,
           1.1921e-07, 1.2886e-02]],

         [[8.3691e-01, 1.1921e-07, 8.3447e-07,  ..., 7.7486e-07,
           5.9605e-08, 2.9266e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1948e-01, 1.5616e-05, 1.6153e-05,  ..., 6.7688e-02,
           6.4502e-01, 1.3397e-02]],

         [[9.1113e-01, 2.6822e-06, 1.7881e-06,  ..., 8.8787e-04,
           2.9922e-04, 3.6983e-03]],

         [[7.4951e-01, 3.2806e-04, 9.6977e-05,  ..., 9.6893e-03,
           8.4305e-03, 5.4413e-02]],

         ...,

         [[8.3447e-01, 7.2718e-05, 1.2934e-05,  ..., 2.6001e-02,
           1.0651e-01, 5.7755e-03]],

         [[9.2969e-01, 6.8784e-05, 2.7120e-05,  ..., 1.2390e-02,
           1.5884e-02, 2.7679e-02]],

         [[8.9551e-01, 1.2279e-04, 7.3075e-05,  ..., 1.0460e-02,
           6.9122e-03, 2.2659e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0078e-01, 2.8014e-05, 3.7193e-05,  ..., 5.4512e-03,
           2.3148e-02, 3.2440e-02]],

         [[5.3174e-01, 5.3024e-04, 1.4842e-04,  ..., 1.8311e-02,
           3.6896e-02, 1.0724e-01]],

         [[5.3467e-02, 1.4305e-06, 7.7486e-07,  ..., 1.8021e-02,
           8.7500e-01, 4.1412e-02]],

         ...,

         [[6.0449e-01, 1.0878e-04, 6.2704e-05,  ..., 9.1248e-02,
           2.5925e-02, 1.0925e-02]],

         [[9.3701e-01, 2.0814e-04, 3.1650e-05,  ..., 4.5013e-03,
           3.7327e-03, 1.3733e-03]],

         [[7.1094e-01, 3.8743e-06, 1.8179e-05,  ..., 4.5349e-02,
           1.8970e-01, 1.6235e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6997e-01, 1.7958e-03, 8.5831e-04,  ..., 6.7215e-03,
           1.4372e-03, 8.3494e-04]],

         [[9.6436e-01, 1.1623e-05, 3.2067e-05,  ..., 3.5095e-03,
           8.6355e-04, 4.2915e-04]],

         [[8.9502e-01, 8.1658e-06, 7.5102e-06,  ..., 1.9989e-03,
           3.1204e-03, 7.3576e-04]],

         ...,

         [[6.5186e-01, 9.9361e-05, 9.4056e-05,  ..., 3.1235e-02,
           5.2834e-03, 2.1252e-01]],

         [[8.9502e-01, 2.9802e-07, 1.7285e-06,  ..., 5.2216e-02,
           1.8967e-02, 1.7258e-02]],

         [[9.4043e-01, 3.5763e-06, 7.7486e-06,  ..., 7.0877e-03,
           2.2446e-02, 8.2245e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4194e-01, 2.6846e-04, 1.1940e-03,  ..., 5.6458e-02,
           8.5144e-03, 3.9368e-03]],

         [[6.7432e-01, 1.0848e-05, 4.2260e-05,  ..., 3.0319e-02,
           2.2424e-01, 1.1444e-02]],

         [[8.3496e-01, 6.2585e-06, 6.1870e-05,  ..., 6.6101e-02,
           2.8351e-02, 2.6169e-02]],

         ...,

         [[7.3438e-01, 6.8665e-05, 2.0790e-04,  ..., 1.2337e-02,
           1.2074e-03, 3.8075e-04]],

         [[7.0898e-01, 1.8096e-04, 4.8041e-05,  ..., 2.1790e-02,
           9.7656e-02, 5.3291e-03]],

         [[8.4961e-01, 5.3644e-06, 7.4506e-06,  ..., 1.1591e-01,
           7.2861e-04, 2.9907e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8867e-02, 3.0398e-06, 2.0385e-05,  ..., 3.6407e-04,
           6.0618e-05, 4.5662e-03]],

         [[8.3643e-01, 6.3181e-06, 9.1195e-06,  ..., 6.7444e-02,
           6.1302e-03, 6.3744e-03]],

         [[8.0957e-01, 4.7684e-07, 4.0531e-06,  ..., 5.8098e-03,
           7.1049e-04, 2.4986e-03]],

         ...,

         [[9.3066e-01, 7.5936e-05, 3.8028e-05,  ..., 9.1400e-03,
           1.2350e-03, 1.4019e-03]],

         [[8.6816e-01, 1.2165e-04, 6.0439e-05,  ..., 2.6520e-02,
           4.5776e-03, 6.9466e-03]],

         [[7.3389e-01, 5.6362e-04, 2.0409e-04,  ..., 1.0605e-02,
           3.2234e-04, 4.8018e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 1.0192e-05, 4.9472e-06,  ..., 2.3331e-02,
           3.2074e-02, 5.5664e-01]],

         [[8.0713e-01, 4.2748e-04, 3.3426e-04,  ..., 4.5242e-03,
           6.1493e-03, 3.3379e-03]],

         [[8.0762e-01, 6.1035e-04, 5.6934e-04,  ..., 5.1025e-02,
           2.7771e-03, 2.6798e-03]],

         ...,

         [[5.6787e-01, 4.7565e-05, 1.4102e-04,  ..., 1.2993e-02,
           2.9312e-02, 5.0201e-03]],

         [[8.7128e-03, 1.0133e-06, 1.0729e-06,  ..., 6.5756e-04,
           8.7256e-01, 5.6152e-03]],

         [[6.9763e-02, 9.1791e-06, 2.2709e-05,  ..., 5.3131e-02,
           2.0065e-02, 8.1787e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0381e-01, 2.0373e-04, 3.3069e-04,  ..., 3.3905e-02,
           3.0014e-02, 4.2310e-01]],

         [[6.1621e-01, 8.0943e-05, 1.3494e-04,  ..., 6.2988e-02,
           2.0325e-02, 5.3520e-03]],

         [[3.7109e-01, 1.2815e-04, 1.0496e-04,  ..., 5.1880e-03,
           7.5867e-02, 8.3847e-03]],

         ...,

         [[1.0681e-02, 1.0729e-06, 2.7418e-06,  ..., 8.5163e-04,
           7.9811e-05, 9.8242e-01]],

         [[7.8125e-01, 1.7083e-04, 1.4675e-04,  ..., 1.1475e-02,
           1.3268e-02, 9.8267e-03]],

         [[8.4180e-01, 7.9811e-05, 2.8789e-05,  ..., 1.9951e-03,
           7.1259e-03, 1.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5986e-01, 6.2609e-04, 1.2922e-03,  ..., 1.3962e-02,
           7.2441e-03, 5.9891e-03]],

         [[7.5488e-01, 1.5974e-05, 6.1691e-05,  ..., 1.5417e-01,
           6.1569e-03, 9.9945e-03]],

         [[3.1152e-01, 2.6584e-05, 6.0380e-05,  ..., 2.6367e-02,
           6.0844e-04, 2.5654e-03]],

         ...,

         [[6.6211e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1346e-01,
           1.9836e-01, 2.0416e-02]],

         [[5.2051e-01, 4.9543e-04, 5.5504e-04,  ..., 1.9897e-02,
           1.6418e-02, 5.8517e-03]],

         [[1.5210e-01, 1.1921e-07, 1.1921e-07,  ..., 1.7960e-02,
           8.0518e-01, 1.6983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9111e-01, 4.2081e-05, 1.3471e-05,  ..., 5.5084e-03,
           5.5122e-03, 3.9482e-03]],

         [[7.0496e-02, 5.8603e-04, 5.3835e-04,  ..., 2.0679e-01,
           3.0231e-04, 6.6137e-04]],

         [[5.0049e-01, 2.2113e-04, 1.6177e-04,  ..., 1.3867e-01,
           1.0777e-04, 2.9397e-04]],

         ...,

         [[4.1577e-01, 5.9605e-07, 2.3842e-06,  ..., 3.5767e-01,
           6.2073e-02, 5.9967e-03]],

         [[1.9751e-01, 5.9605e-08, 5.9605e-08,  ..., 3.2788e-01,
           4.4141e-01, 8.3389e-03]],

         [[8.9294e-02, 2.5630e-06, 6.1393e-06,  ..., 2.2949e-01,
           1.3351e-02, 2.6417e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6294e-02, 6.5565e-07, 1.2517e-06,  ..., 5.2643e-03,
           2.4765e-02, 8.0127e-01]],

         [[1.2238e-01, 9.4771e-06, 2.3007e-05,  ..., 4.0918e-01,
           5.4535e-02, 1.7395e-02]],

         [[2.5781e-01, 2.8563e-04, 3.9768e-04,  ..., 7.3303e-02,
           1.2810e-02, 8.7738e-03]],

         ...,

         [[3.3887e-01, 5.4538e-05, 8.3506e-05,  ..., 4.1840e-02,
           1.4626e-02, 4.8180e-03]],

         [[5.6152e-01, 3.0398e-06, 1.0490e-05,  ..., 2.0905e-02,
           1.7662e-03, 3.6488e-03]],

         [[1.8799e-01, 1.3876e-04, 1.6773e-04,  ..., 1.2379e-03,
           1.1467e-02, 8.0185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4902e-01, 2.1935e-05, 5.3167e-05,  ..., 3.5858e-02,
           1.0277e-02, 1.5251e-02]],

         [[2.2607e-01, 2.2376e-04, 3.6240e-04,  ..., 7.0007e-02,
           2.6169e-02, 1.8478e-02]],

         [[2.2430e-02, 2.6822e-06, 2.0862e-06,  ..., 8.2855e-03,
           6.0156e-01, 1.1818e-02]],

         ...,

         [[7.0898e-01, 4.9233e-05, 3.8207e-05,  ..., 1.9257e-02,
           3.7193e-03, 1.7746e-02]],

         [[5.5225e-01, 3.2425e-05, 5.4955e-05,  ..., 1.0254e-01,
           7.3128e-03, 1.4153e-02]],

         [[7.4854e-01, 9.1314e-05, 6.3121e-05,  ..., 1.1383e-02,
           1.1276e-02, 6.8130e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8809e-01, 1.9608e-03, 8.5688e-04,  ..., 1.2123e-02,
           2.7823e-04, 5.0688e-04]],

         [[1.7908e-01, 2.8658e-04, 2.5630e-04,  ..., 1.2817e-02,
           1.4641e-02, 1.5373e-02]],

         [[5.1807e-01, 2.3663e-05, 4.4167e-05,  ..., 7.8186e-02,
           4.5441e-02, 1.2306e-02]],

         ...,

         [[2.1021e-01, 1.1116e-04, 4.1544e-05,  ..., 6.1646e-02,
           2.7145e-02, 8.1635e-03]],

         [[8.9893e-01, 1.0598e-04, 1.4210e-04,  ..., 2.7728e-04,
           5.5432e-06, 8.7128e-03]],

         [[9.1553e-01, 7.7784e-05, 6.8009e-05,  ..., 1.8341e-02,
           4.4518e-03, 1.0881e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8799e-01, 5.7125e-04, 1.3075e-03,  ..., 8.4778e-02,
           1.8435e-03, 3.8818e-02]],

         [[2.8931e-01, 4.2915e-03, 3.1166e-03,  ..., 2.0477e-02,
           1.7338e-03, 2.5635e-03]],

         [[1.9812e-01, 1.6809e-05, 2.3901e-05,  ..., 1.3000e-01,
           1.1200e-01, 1.6040e-01]],

         ...,

         [[1.7334e-01, 1.4484e-05, 6.0618e-05,  ..., 6.6833e-02,
           1.0872e-02, 2.2869e-03]],

         [[3.6035e-01, 1.2100e-05, 3.1173e-05,  ..., 8.2947e-02,
           2.2559e-01, 3.1738e-02]],

         [[2.0984e-01, 9.2387e-06, 1.9848e-05,  ..., 1.7676e-01,
           2.3962e-01, 1.5710e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3252e-01, 1.1462e-04, 1.8609e-04,  ..., 2.4750e-02,
           8.9645e-03, 2.5543e-02]],

         [[1.9507e-01, 5.2571e-05, 1.2422e-04,  ..., 2.2107e-01,
           9.5032e-02, 6.8855e-03]],

         [[5.3284e-02, 4.1127e-06, 1.8477e-06,  ..., 7.4816e-04,
           8.3008e-03, 9.3018e-01]],

         ...,

         [[4.0680e-02, 1.2898e-04, 3.1304e-04,  ..., 1.7452e-03,
           2.9926e-03, 1.3306e-02]],

         [[3.0933e-01, 2.7657e-04, 4.4560e-04,  ..., 1.2085e-02,
           2.5391e-02, 5.5908e-02]],

         [[2.0410e-01, 1.7729e-03, 2.5806e-03,  ..., 3.2806e-02,
           4.7073e-03, 6.7177e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7715e-01, 1.1790e-04, 1.0085e-04,  ..., 5.3680e-02,
           5.0049e-02, 1.8265e-02]],

         [[3.7594e-03, 9.5367e-07, 1.3709e-06,  ..., 1.6868e-05,
           2.0027e-05, 1.5891e-04]],

         [[8.9404e-01, 2.5988e-05, 2.1815e-05,  ..., 8.0919e-04,
           4.6825e-04, 2.7537e-04]],

         ...,

         [[1.1536e-01, 1.1663e-03, 2.3746e-03,  ..., 2.9617e-02,
           1.6876e-02, 3.8635e-02]],

         [[3.3960e-01, 3.7551e-05, 8.1420e-05,  ..., 8.1100e-03,
           2.1582e-01, 4.4525e-02]],

         [[5.5762e-01, 1.2579e-03, 3.8452e-03,  ..., 2.0950e-02,
           1.0368e-02, 2.8641e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7437e-01, 1.4722e-04, 6.9666e-04,  ..., 5.7556e-02,
           4.2786e-02, 3.4912e-02]],

         [[6.7444e-02, 1.3757e-04, 6.0225e-04,  ..., 2.2949e-02,
           2.3823e-03, 4.7089e-02]],

         [[2.4500e-01, 3.7909e-05, 1.9717e-04,  ..., 1.5182e-02,
           4.9667e-03, 7.1289e-02]],

         ...,

         [[8.5449e-02, 2.5558e-04, 3.3855e-04,  ..., 1.1625e-03,
           6.3858e-03, 3.8513e-02]],

         [[2.5482e-02, 3.5405e-05, 6.3181e-05,  ..., 3.1590e-04,
           3.4485e-03, 1.5701e-02]],

         [[2.3108e-01, 1.0514e-04, 2.2781e-04,  ..., 2.2171e-02,
           2.0828e-02, 1.5236e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8396e-01, 1.5318e-05, 2.5034e-05,  ..., 2.0325e-02,
           7.0374e-02, 1.4783e-01]],

         [[4.8242e-01, 1.6439e-04, 4.9257e-04,  ..., 6.4270e-02,
           5.1956e-03, 1.7578e-02]],

         [[2.6611e-01, 1.7607e-04, 4.7851e-04,  ..., 7.2144e-02,
           1.0559e-02, 4.7778e-01]],

         ...,

         [[6.9971e-01, 1.4687e-04, 3.7503e-04,  ..., 2.0966e-02,
           9.1629e-03, 4.5074e-02]],

         [[2.9688e-01, 1.1462e-04, 2.2447e-04,  ..., 2.1133e-03,
           3.2978e-03, 3.6285e-02]],

         [[1.0602e-01, 7.3552e-05, 2.1279e-04,  ..., 1.5974e-03,
           1.0719e-03, 1.5396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3391e-01, 9.8884e-05, 1.4329e-04,  ..., 5.1239e-02,
           1.2482e-01, 1.2683e-01]],

         [[1.0815e-01, 4.3511e-05, 5.2392e-05,  ..., 5.9586e-03,
           3.1952e-02, 4.6143e-02]],

         [[4.5020e-01, 1.7798e-04, 3.1853e-04,  ..., 1.1360e-02,
           5.3772e-02, 8.0078e-02]],

         ...,

         [[5.1270e-01, 0.0000e+00, 0.0000e+00,  ..., 5.7526e-02,
           4.0552e-01, 1.4206e-02]],

         [[2.0154e-01, 7.6890e-06, 4.3452e-05,  ..., 2.8857e-01,
           1.3428e-01, 7.2823e-03]],

         [[2.5024e-01, 1.1325e-06, 1.3113e-06,  ..., 6.4270e-02,
           6.5918e-01, 1.4229e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1331e-02, 2.2411e-05, 5.7340e-05,  ..., 2.5818e-02,
           7.2998e-02, 9.7198e-03]],

         [[2.3590e-02, 5.4240e-06, 3.7551e-06,  ..., 3.3989e-03,
           4.8401e-02, 8.8525e-01]],

         [[3.7378e-01, 1.3371e-03, 3.6507e-03,  ..., 2.8824e-02,
           1.5821e-03, 1.2688e-02]],

         ...,

         [[2.7417e-01, 3.9744e-04, 7.8535e-04,  ..., 4.5471e-02,
           1.4198e-02, 3.6240e-03]],

         [[1.2952e-01, 1.0729e-06, 6.1393e-06,  ..., 1.0132e-01,
           6.5283e-01, 2.9831e-02]],

         [[7.3059e-02, 5.9605e-07, 3.5167e-06,  ..., 2.9810e-01,
           4.8779e-01, 2.8732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 1.5974e-04, 4.9686e-04,  ..., 8.1940e-03,
           5.1270e-03, 6.7368e-03]],

         [[9.6289e-01, 2.2471e-05, 8.3268e-05,  ..., 7.2896e-05,
           6.3241e-05, 1.2022e-04]],

         [[3.5669e-01, 6.4790e-05, 1.2589e-04,  ..., 5.0354e-04,
           6.4945e-04, 1.1091e-03]],

         ...,

         [[8.4619e-01, 1.5736e-05, 1.4126e-05,  ..., 1.2260e-02,
           3.8361e-02, 9.5444e-03]],

         [[4.6826e-01, 2.1458e-05, 1.7464e-05,  ..., 3.3321e-03,
           4.7379e-03, 1.0841e-02]],

         [[1.1554e-01, 1.6809e-05, 5.6386e-05,  ..., 9.3384e-03,
           6.3782e-02, 7.1875e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2559e-01, 1.4067e-05, 1.8716e-05,  ..., 1.3257e-01,
           1.2262e-01, 3.8147e-02]],

         [[7.4268e-01, 2.4498e-05, 1.8537e-05,  ..., 1.4305e-02,
           4.7089e-02, 1.3817e-02]],

         [[6.4026e-02, 1.5795e-05, 3.0458e-05,  ..., 5.7831e-02,
           2.8687e-01, 2.3636e-02]],

         ...,

         [[6.3428e-01, 1.6570e-05, 7.5459e-05,  ..., 8.5876e-02,
           8.4534e-02, 3.9001e-02]],

         [[6.5967e-01, 7.9215e-05, 6.0606e-04,  ..., 7.0068e-02,
           1.7303e-02, 8.1100e-03]],

         [[6.5674e-01, 1.5676e-05, 9.7394e-05,  ..., 2.6276e-02,
           5.5962e-03, 2.1118e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9438e-02, 1.1444e-05, 6.7353e-06,  ..., 4.2534e-03,
           3.8818e-02, 3.4981e-03]],

         [[2.5171e-01, 7.7486e-05, 6.8903e-05,  ..., 1.0674e-02,
           7.5111e-03, 5.9433e-03]],

         [[1.7200e-01, 1.1790e-04, 3.1424e-04,  ..., 6.8420e-02,
           7.3975e-02, 7.5745e-02]],

         ...,

         [[5.6213e-02, 1.6391e-05, 1.6940e-04,  ..., 8.5022e-02,
           3.8853e-03, 7.6065e-03]],

         [[1.2390e-01, 2.3127e-05, 1.6963e-04,  ..., 2.4646e-01,
           9.7046e-03, 3.7201e-02]],

         [[9.1650e-01, 6.7949e-06, 1.7345e-05,  ..., 2.7679e-02,
           1.9318e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0789e-01, 1.7285e-06, 5.3644e-06,  ..., 1.3525e-01,
           5.9424e-01, 2.8351e-02]],

         [[5.2344e-01, 1.2398e-05, 1.0550e-05,  ..., 1.8738e-02,
           1.7944e-01, 1.4819e-01]],

         [[8.8440e-02, 2.9802e-07, 4.1723e-07,  ..., 2.4915e-01,
           5.9277e-01, 2.9068e-02]],

         ...,

         [[3.7671e-01, 1.4460e-04, 4.3249e-04,  ..., 1.0811e-02,
           1.0239e-02, 2.9968e-02]],

         [[3.5742e-01, 1.0604e-04, 1.1396e-04,  ..., 1.0815e-01,
           6.9824e-02, 3.0029e-02]],

         [[4.7211e-02, 8.7976e-05, 7.4267e-05,  ..., 4.5410e-02,
           1.4099e-01, 5.2393e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9756e-01, 1.0729e-05, 1.1909e-04,  ..., 6.7322e-02,
           1.4038e-01, 1.5906e-01]],

         [[3.5400e-01, 6.7282e-04, 2.0084e-03,  ..., 1.9135e-02,
           5.5656e-03, 5.1689e-03]],

         [[1.2024e-01, 4.0054e-05, 2.0015e-04,  ..., 9.2010e-03,
           3.9902e-03, 1.5900e-02]],

         ...,

         [[6.5283e-01, 1.0109e-04, 1.2577e-04,  ..., 1.5839e-02,
           1.4575e-01, 5.6183e-02]],

         [[1.7175e-01, 5.9009e-05, 1.4377e-04,  ..., 7.7858e-03,
           4.9225e-02, 7.3438e-01]],

         [[7.4268e-01, 9.7573e-05, 3.8576e-04,  ..., 1.2978e-02,
           1.5656e-02, 2.0618e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1211e-01, 1.8609e-04, 9.5224e-04,  ..., 4.1199e-02,
           1.6077e-01, 7.8674e-02]],

         [[1.9800e-01, 5.2631e-05, 1.0973e-04,  ..., 1.1986e-02,
           5.8075e-02, 4.9683e-02]],

         [[1.7236e-01, 1.3638e-04, 2.5730e-03,  ..., 1.4465e-01,
           1.2817e-01, 2.7710e-02]],

         ...,

         [[8.1445e-01, 6.0141e-05, 1.8752e-04,  ..., 1.4664e-02,
           3.5896e-03, 3.6652e-02]],

         [[7.0801e-01, 1.7285e-06, 1.6093e-04,  ..., 1.1654e-03,
           1.0109e-03, 1.4524e-03]],

         [[6.1963e-01, 4.4167e-05, 4.0591e-05,  ..., 5.1453e-02,
           6.9214e-02, 1.3184e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6955e-04, 6.3610e-04, 1.6125e-01,  ..., 1.1792e-01,
           1.5078e-03, 1.4534e-02]],

         [[2.9087e-04, 2.7323e-04, 1.7761e-01,  ..., 1.1377e-01,
           9.0027e-02, 7.9269e-03]],

         [[2.0826e-04, 2.8610e-04, 6.7322e-02,  ..., 1.3330e-01,
           8.3801e-02, 1.3867e-01]],

         ...,

         [[2.7013e-04, 7.4446e-05, 1.0730e-01,  ..., 1.7419e-01,
           2.1167e-01, 1.3989e-01]],

         [[2.4652e-04, 5.4264e-04, 2.3938e-01,  ..., 4.9042e-02,
           3.0022e-03, 1.8753e-02]],

         [[4.8459e-05, 5.9009e-06, 2.7786e-02,  ..., 1.8616e-03,
           1.2183e-04, 2.8839e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4894e-04, 6.4909e-05, 2.7515e-01,  ..., 1.0498e-02,
           9.2125e-04, 1.7281e-03]],

         [[3.1090e-04, 1.3959e-04, 2.7197e-01,  ..., 9.4452e-03,
           2.4948e-03, 7.7581e-04]],

         [[4.3988e-05, 6.5565e-06, 2.9175e-01,  ..., 9.2316e-03,
           1.6034e-05, 8.8596e-04]],

         ...,

         [[2.9049e-03, 4.6778e-04, 2.7295e-01,  ..., 2.7130e-02,
           2.4757e-03, 8.0967e-04]],

         [[8.8930e-05, 6.3777e-05, 3.0640e-01,  ..., 1.7151e-02,
           2.5725e-04, 9.4509e-04]],

         [[2.2411e-05, 8.8215e-06, 2.0251e-01,  ..., 2.9175e-02,
           1.2493e-04, 2.5845e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2149e-04, 5.9187e-05, 2.1240e-01,  ..., 1.0933e-02,
           1.6630e-05, 8.4591e-04]],

         [[5.5313e-04, 1.0490e-05, 2.8320e-01,  ..., 1.1780e-02,
           2.7447e-03, 7.3290e-04]],

         [[4.4942e-04, 2.2376e-04, 2.5977e-01,  ..., 8.6517e-03,
           3.8815e-04, 1.1501e-03]],

         ...,

         [[2.6226e-05, 5.0664e-05, 2.1765e-01,  ..., 8.1970e-02,
           3.7689e-02, 3.3386e-02]],

         [[1.5008e-04, 9.3639e-05, 2.3108e-01,  ..., 9.0637e-03,
           5.9433e-03, 1.3418e-03]],

         [[6.8474e-04, 5.6446e-05, 2.1899e-01,  ..., 7.0648e-03,
           1.5106e-03, 7.8201e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9855e-03, 5.9128e-04, 4.9225e-02,  ..., 2.6416e-01,
           1.8433e-01, 4.0741e-03]],

         [[1.2577e-04, 2.6846e-04, 1.3220e-01,  ..., 4.0649e-02,
           6.9702e-02, 3.7292e-02]],

         [[4.9353e-04, 8.7976e-04, 7.7026e-02,  ..., 1.4715e-03,
           7.8869e-04, 2.0075e-04]],

         ...,

         [[2.4967e-03, 5.3263e-04, 3.9215e-02,  ..., 7.9041e-03,
           4.4365e-03, 5.5084e-03]],

         [[1.1711e-03, 2.1994e-04, 8.6548e-02,  ..., 5.0110e-02,
           6.5369e-02, 1.8723e-02]],

         [[1.1528e-04, 3.8266e-05, 2.4338e-02,  ..., 4.6021e-02,
           1.0699e-01, 7.8857e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7656e-03, 1.7033e-03, 2.1130e-01,  ..., 8.6517e-03,
           7.2144e-02, 2.2446e-02]],

         [[1.0262e-03, 1.2312e-03, 2.9590e-01,  ..., 1.7426e-02,
           1.3573e-02, 3.8361e-02]],

         [[3.2711e-03, 8.4972e-04, 1.9812e-01,  ..., 2.5894e-02,
           6.3843e-02, 2.9739e-02]],

         ...,

         [[1.8752e-04, 1.1921e-04, 3.6865e-02,  ..., 7.3364e-02,
           6.5869e-01, 8.8440e-02]],

         [[1.1320e-03, 4.5919e-04, 2.4646e-01,  ..., 3.8086e-02,
           3.1586e-02, 1.1914e-01]],

         [[1.3173e-04, 4.0746e-04, 2.0618e-01,  ..., 1.7358e-01,
           2.9480e-02, 1.2093e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2817e-03, 7.7009e-05, 5.8699e-04,  ..., 4.3793e-02,
           2.9468e-01, 4.7095e-01]],

         [[2.2488e-03, 1.2808e-03, 1.2154e-02,  ..., 2.7122e-03,
           2.2705e-02, 4.5605e-01]],

         [[3.1829e-05, 2.6226e-06, 1.1104e-04,  ..., 1.1104e-04,
           8.9941e-01, 9.4788e-02]],

         ...,

         [[2.6054e-03, 3.3455e-03, 5.5161e-03,  ..., 2.8015e-02,
           3.0127e-01, 4.2953e-03]],

         [[3.1338e-03, 3.5515e-03, 1.1642e-02,  ..., 1.3905e-03,
           7.5928e-02, 1.8213e-01]],

         [[8.3084e-03, 5.8899e-03, 1.1969e-03,  ..., 7.1777e-02,
           2.2583e-02, 2.1881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8065e-04, 2.2926e-03, 6.5308e-03,  ..., 1.0059e-01,
           7.3608e-02, 1.5332e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 2.1997e-01,
           6.9873e-01, 7.4829e-02]],

         [[1.8339e-03, 2.1439e-03, 6.6032e-03,  ..., 2.9587e-02,
           2.6520e-02, 1.6083e-02]],

         ...,

         [[6.2895e-04, 1.6823e-03, 6.5002e-03,  ..., 6.6711e-02,
           6.4011e-03, 2.0981e-02]],

         [[4.4136e-03, 9.5673e-03, 4.2694e-02,  ..., 2.5696e-02,
           2.3666e-02, 2.5208e-02]],

         [[4.2200e-04, 9.1457e-04, 2.8172e-03,  ..., 2.3462e-01,
           5.8984e-01, 5.4840e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7618e-02, 4.0092e-03, 1.7548e-02,  ..., 8.5083e-02,
           1.0022e-01, 7.8674e-02]],

         [[2.4605e-03, 7.0477e-04, 2.5673e-03,  ..., 2.7122e-03,
           3.9444e-03, 8.1110e-04]],

         [[8.9798e-03, 2.4261e-03, 1.9875e-03,  ..., 2.7878e-02,
           3.5645e-02, 3.6926e-02]],

         ...,

         [[4.9286e-03, 3.0422e-04, 4.3917e-04,  ..., 3.3765e-01,
           3.1226e-01, 2.3755e-01]],

         [[3.0708e-03, 6.1893e-04, 1.3256e-03,  ..., 1.8555e-01,
           4.1412e-02, 3.0777e-02]],

         [[1.9333e-02, 1.8196e-03, 9.2773e-03,  ..., 5.5084e-02,
           4.0698e-01, 8.4656e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4551e-01, 2.8014e-06, 9.4771e-06,  ..., 9.0576e-02,
           1.4076e-02, 1.8982e-02]],

         [[9.0283e-01, 8.9407e-06, 1.4842e-05,  ..., 4.7607e-02,
           8.3771e-03, 9.3689e-03]],

         [[4.3164e-01, 1.1325e-06, 1.2100e-05,  ..., 1.1438e-01,
           3.7659e-02, 1.1002e-02]],

         ...,

         [[4.5288e-01, 3.6144e-04, 2.0924e-03,  ..., 5.6458e-04,
           2.9678e-03, 8.0719e-03]],

         [[5.1465e-01, 2.4915e-04, 1.3351e-03,  ..., 2.7267e-02,
           5.6854e-02, 1.5244e-02]],

         [[4.3481e-01, 1.2934e-04, 3.6740e-04,  ..., 2.9358e-02,
           4.1901e-02, 1.9104e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9766e-01, 5.1856e-06, 9.2983e-06,  ..., 2.9325e-04,
           5.5885e-04, 7.3738e-03]],

         [[1.3110e-01, 1.1921e-07, 0.0000e+00,  ..., 9.8648e-03,
           8.4521e-01, 1.3084e-02]],

         [[9.5117e-01, 1.0216e-04, 2.7835e-05,  ..., 1.1425e-03,
           7.6675e-03, 3.0689e-03]],

         ...,

         [[3.6108e-01, 1.3781e-04, 3.4904e-04,  ..., 1.1131e-02,
           7.2365e-03, 3.5954e-03]],

         [[1.5125e-01, 1.3914e-03, 6.4230e-04,  ..., 3.8986e-03,
           3.9711e-03, 1.6613e-03]],

         [[6.5381e-01, 1.2398e-05, 8.6248e-05,  ..., 6.7322e-02,
           4.9713e-02, 1.3000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3105e-01, 3.4046e-04, 1.8656e-04,  ..., 8.2397e-02,
           3.3630e-02, 2.3087e-02]],

         [[6.2305e-01, 8.3447e-06, 1.2755e-05,  ..., 7.7332e-02,
           1.4771e-01, 2.3834e-02]],

         [[8.2031e-01, 1.1742e-05, 3.1590e-06,  ..., 1.0278e-01,
           4.5471e-03, 8.6288e-03]],

         ...,

         [[2.8882e-01, 4.7684e-07, 2.9802e-06,  ..., 2.9278e-04,
           1.3031e-02, 6.8359e-01]],

         [[3.4814e-01, 4.7684e-07, 3.3402e-04,  ..., 2.9802e-07,
           1.1367e-04, 7.0984e-02]],

         [[8.9844e-01, 8.3447e-07, 2.4319e-05,  ..., 5.3644e-07,
           2.8312e-05, 4.6539e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2876e-01, 1.1683e-05, 4.1127e-06,  ..., 4.3750e-01,
           2.5879e-01, 1.4908e-02]],

         [[9.1406e-01, 7.6890e-06, 3.8743e-06,  ..., 3.9062e-03,
           1.2512e-03, 6.1951e-03]],

         [[6.9922e-01, 4.1795e-04, 2.5058e-04,  ..., 1.1650e-02,
           2.5970e-02, 2.6901e-02]],

         ...,

         [[8.5840e-01, 3.0100e-05, 1.2040e-05,  ..., 7.4707e-02,
           1.4839e-02, 8.0338e-03]],

         [[6.9971e-01, 3.5882e-05, 7.3314e-06,  ..., 2.2324e-02,
           1.9312e-01, 6.7383e-02]],

         [[8.9502e-01, 1.2231e-04, 2.8908e-05,  ..., 4.0512e-03,
           3.1372e-02, 2.6764e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0312e-01, 1.5914e-05, 1.0192e-05,  ..., 7.4707e-02,
           8.4778e-02, 5.0903e-02]],

         [[5.3369e-01, 9.8050e-05, 2.1672e-04,  ..., 4.4586e-02,
           6.3904e-02, 1.3245e-01]],

         [[1.7297e-01, 2.9802e-06, 4.2319e-06,  ..., 6.7578e-01,
           1.1322e-01, 7.5760e-03]],

         ...,

         [[6.6943e-01, 8.3685e-05, 4.6730e-05,  ..., 3.1555e-02,
           1.2543e-02, 1.0178e-02]],

         [[8.6621e-01, 1.7929e-04, 3.8624e-05,  ..., 6.8855e-03,
           4.2076e-03, 9.2459e-04]],

         [[8.0371e-01, 6.9737e-06, 8.4043e-06,  ..., 4.1565e-02,
           7.0557e-02, 1.3153e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.5936e-03, 1.3008e-03,  ..., 2.1954e-03,
           6.0701e-04, 5.5504e-04]],

         [[9.6094e-01, 1.1981e-05, 2.8968e-05,  ..., 9.3651e-04,
           8.8310e-04, 5.4407e-04]],

         [[8.7646e-01, 1.8656e-05, 1.2040e-05,  ..., 2.4223e-03,
           1.5631e-03, 1.9760e-03]],

         ...,

         [[6.0840e-01, 6.5744e-05, 1.0067e-04,  ..., 1.7319e-02,
           1.4600e-01, 8.2520e-02]],

         [[9.2969e-01, 6.5565e-07, 1.7285e-06,  ..., 8.3160e-03,
           2.4292e-02, 9.5749e-03]],

         [[9.5801e-01, 2.0862e-06, 2.1458e-06,  ..., 3.0499e-03,
           1.2657e-02, 1.1452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 4.9353e-04, 1.3733e-03,  ..., 5.1727e-03,
           4.8866e-03, 1.9951e-03]],

         [[5.9717e-01, 2.6345e-05, 3.2544e-05,  ..., 8.2458e-02,
           2.3315e-01, 2.9648e-02]],

         [[9.4775e-01, 8.4639e-06, 2.1815e-05,  ..., 9.5062e-03,
           5.3444e-03, 3.5458e-03]],

         ...,

         [[6.1035e-01, 1.3340e-04, 2.0754e-04,  ..., 1.9360e-03,
           5.2357e-04, 2.0826e-04]],

         [[7.7783e-01, 5.6505e-04, 7.6830e-05,  ..., 2.5955e-02,
           1.1581e-02, 4.9210e-03]],

         [[3.0469e-01, 1.1086e-05, 1.2815e-05,  ..., 7.8659e-03,
           5.2246e-01, 1.0083e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7981e-01, 4.8876e-06, 2.3723e-05,  ..., 2.1410e-04,
           6.1703e-04, 3.1395e-03]],

         [[7.5146e-01, 8.2850e-06, 1.4603e-05,  ..., 4.1473e-02,
           3.2166e-02, 3.7022e-03]],

         [[6.6309e-01, 5.9605e-07, 6.6161e-06,  ..., 3.7136e-03,
           2.0275e-03, 1.9045e-03]],

         ...,

         [[9.5703e-01, 9.3997e-05, 5.4181e-05,  ..., 5.0402e-04,
           6.7282e-04, 7.5817e-05]],

         [[5.3223e-01, 9.4831e-05, 3.7551e-05,  ..., 5.6992e-03,
           4.1565e-02, 3.0151e-01]],

         [[6.4990e-01, 4.0007e-04, 1.3912e-04,  ..., 3.4428e-04,
           1.5640e-03, 6.9809e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5283e-01, 2.3067e-05, 8.9407e-06,  ..., 2.1000e-03,
           1.6266e-02, 8.1348e-01]],

         [[8.5254e-01, 3.5262e-04, 2.6274e-04,  ..., 3.5744e-03,
           1.7557e-03, 7.6008e-04]],

         [[8.2129e-01, 3.8266e-04, 2.7466e-04,  ..., 5.8784e-03,
           7.0229e-03, 1.6394e-03]],

         ...,

         [[5.1904e-01, 5.9903e-05, 2.0742e-04,  ..., 3.3966e-02,
           3.9825e-02, 3.5419e-03]],

         [[1.8726e-01, 4.1723e-06, 5.1260e-06,  ..., 4.7729e-01,
           1.0242e-01, 6.4507e-03]],

         [[1.4392e-01, 2.3842e-05, 3.2842e-05,  ..., 1.1536e-02,
           1.1163e-01, 6.6992e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6709e-01, 2.2972e-04, 2.2793e-04,  ..., 2.6810e-02,
           1.1792e-01, 4.6973e-01]],

         [[6.4355e-01, 5.5611e-05, 5.3465e-05,  ..., 2.2064e-02,
           4.7638e-02, 3.7060e-03]],

         [[4.6460e-01, 1.8823e-04, 3.7718e-04,  ..., 4.1290e-02,
           2.6489e-02, 1.4801e-03]],

         ...,

         [[1.3786e-02, 8.3447e-07, 2.2650e-06,  ..., 7.3910e-06,
           5.2528e-03, 9.7803e-01]],

         [[7.7295e-01, 1.5545e-04, 1.9121e-04,  ..., 5.5618e-03,
           4.3144e-03, 3.4912e-02]],

         [[7.8711e-01, 1.4138e-04, 4.8637e-05,  ..., 1.9867e-02,
           2.0798e-02, 7.2098e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9624e-01, 4.5586e-04, 1.0376e-03,  ..., 6.6261e-03,
           3.0899e-03, 1.7338e-03]],

         [[8.6035e-01, 1.7226e-05, 5.0068e-05,  ..., 4.0169e-03,
           3.2776e-02, 7.9575e-03]],

         [[3.6646e-01, 3.0756e-05, 9.1195e-05,  ..., 7.6628e-04,
           2.4490e-03, 5.2681e-03]],

         ...,

         [[3.8989e-01, 3.5763e-07, 0.0000e+00,  ..., 6.8542e-02,
           5.2393e-01, 1.4458e-02]],

         [[5.4492e-01, 8.4209e-04, 9.8038e-04,  ..., 1.3222e-02,
           4.8370e-03, 6.3801e-04]],

         [[6.3770e-01, 2.9206e-06, 1.3709e-06,  ..., 2.9028e-01,
           3.6926e-02, 4.2267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9648e-01, 8.1241e-05, 2.4498e-05,  ..., 5.1193e-03,
           2.2373e-03, 4.7588e-04]],

         [[1.3904e-01, 3.0684e-04, 2.6536e-04,  ..., 1.1177e-03,
           1.4839e-03, 4.8780e-04]],

         [[7.7832e-01, 9.7573e-05, 4.3988e-05,  ..., 3.2067e-04,
           5.9891e-04, 3.2878e-04]],

         ...,

         [[6.5527e-01, 2.3842e-07, 1.7881e-07,  ..., 7.6172e-02,
           9.0332e-02, 1.2856e-02]],

         [[8.1836e-01, 5.9605e-08, 5.9605e-08,  ..., 7.3059e-02,
           5.3467e-02, 1.4977e-02]],

         [[5.0293e-01, 3.2783e-06, 6.8545e-06,  ..., 4.0771e-02,
           3.2745e-02, 8.5144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8167e-02, 1.0133e-06, 1.7285e-06,  ..., 1.0004e-03,
           1.0887e-02, 8.9941e-01]],

         [[1.4319e-01, 3.8743e-06, 9.1195e-06,  ..., 1.7200e-01,
           2.2791e-01, 3.7628e-02]],

         [[4.2773e-01, 1.8287e-04, 2.0754e-04,  ..., 1.2413e-02,
           7.1068e-03, 2.0294e-03]],

         ...,

         [[4.3359e-01, 1.0377e-04, 1.6844e-04,  ..., 2.8229e-02,
           1.0490e-02, 5.6686e-03]],

         [[4.9854e-01, 3.7551e-06, 2.1815e-05,  ..., 1.2039e-02,
           1.2688e-02, 1.1185e-02]],

         [[1.8750e-01, 2.0826e-04, 2.4915e-04,  ..., 2.3098e-03,
           3.8834e-03, 1.6222e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2539e-01, 2.7120e-05, 5.5492e-05,  ..., 1.7838e-02,
           1.0323e-02, 6.0921e-03]],

         [[3.5059e-01, 3.1900e-04, 2.3162e-04,  ..., 8.1360e-02,
           3.2745e-02, 1.3329e-02]],

         [[1.6553e-01, 3.1471e-05, 3.8981e-05,  ..., 2.5732e-01,
           3.0029e-02, 8.6899e-03]],

         ...,

         [[7.7637e-01, 5.4836e-06, 7.9870e-06,  ..., 1.4275e-02,
           7.1487e-03, 4.8828e-04]],

         [[6.4600e-01, 5.2273e-05, 7.6115e-05,  ..., 1.1925e-02,
           2.2903e-02, 1.5764e-03]],

         [[8.8477e-01, 4.3154e-05, 5.1439e-05,  ..., 8.4381e-03,
           1.0424e-03, 6.3705e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8979e-01, 2.7294e-03, 1.2264e-03,  ..., 7.8058e-04,
           1.2732e-03, 5.8603e-04]],

         [[2.9956e-01, 1.0729e-03, 7.4434e-04,  ..., 6.5575e-03,
           2.8381e-03, 4.1924e-03]],

         [[3.7695e-01, 2.0146e-05, 2.0504e-05,  ..., 2.0984e-01,
           1.3806e-01, 1.5022e-02]],

         ...,

         [[3.5156e-01, 2.1899e-04, 1.3340e-04,  ..., 4.6577e-03,
           1.4896e-03, 2.2171e-02]],

         [[9.5020e-01, 6.2585e-05, 2.7823e-04,  ..., 1.7881e-07,
           2.7657e-05, 2.3937e-03]],

         [[8.2422e-01, 5.6744e-05, 6.6936e-05,  ..., 1.5427e-02,
           4.4785e-03, 1.5640e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9990e-01, 4.3988e-04, 8.0919e-04,  ..., 1.1978e-02,
           2.9419e-02, 7.6752e-03]],

         [[4.5386e-01, 1.9894e-03, 1.4496e-03,  ..., 1.2474e-02,
           5.1994e-03, 2.0695e-03]],

         [[1.5442e-01, 2.5451e-05, 2.1219e-05,  ..., 1.0262e-02,
           3.5620e-01, 2.4475e-01]],

         ...,

         [[1.9727e-01, 1.0550e-05, 3.1114e-05,  ..., 2.3560e-02,
           1.1528e-02, 5.4979e-04]],

         [[4.2676e-01, 3.7968e-05, 4.3094e-05,  ..., 1.1395e-01,
           7.2998e-02, 2.3788e-02]],

         [[4.2920e-01, 1.1480e-04, 1.2082e-04,  ..., 4.5959e-02,
           1.0480e-01, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4131e-01, 1.5831e-04, 2.5702e-04,  ..., 6.7711e-04,
           2.5558e-03, 1.5900e-02]],

         [[1.1365e-01, 9.0599e-06, 1.0729e-05,  ..., 2.4426e-01,
           7.4524e-02, 7.8506e-03]],

         [[2.6465e-01, 5.8174e-05, 1.9789e-05,  ..., 2.3975e-03,
           1.0522e-01, 5.8691e-01]],

         ...,

         [[7.9041e-02, 3.1066e-04, 7.8726e-04,  ..., 6.7329e-04,
           3.4199e-03, 1.0048e-02]],

         [[3.3984e-01, 4.1699e-04, 5.1880e-04,  ..., 4.2610e-03,
           2.7039e-02, 2.7679e-02]],

         [[2.1509e-01, 1.9817e-03, 2.0046e-03,  ..., 3.8033e-03,
           2.8267e-03, 3.8490e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6655e-01, 4.4823e-05, 5.1498e-05,  ..., 6.7749e-02,
           3.6530e-02, 8.8120e-03]],

         [[2.1152e-03, 2.9802e-07, 5.9605e-08,  ..., 1.5199e-04,
           1.0610e-05, 3.7708e-03]],

         [[8.1543e-01, 1.5855e-05, 3.1590e-06,  ..., 2.8496e-03,
           1.3971e-04, 4.2033e-04]],

         ...,

         [[9.1003e-02, 8.4496e-04, 1.6031e-03,  ..., 1.5327e-02,
           1.6830e-02, 2.7275e-03]],

         [[5.2588e-01, 2.5809e-05, 4.9770e-05,  ..., 1.8164e-01,
           2.0065e-02, 8.9035e-03]],

         [[4.8169e-01, 6.0034e-04, 1.3638e-03,  ..., 9.8419e-03,
           1.6876e-02, 1.4973e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6265e-01, 5.0008e-05, 1.8728e-04,  ..., 3.4576e-02,
           1.8799e-02, 6.6650e-02]],

         [[8.6365e-02, 4.6670e-05, 1.3828e-04,  ..., 7.4310e-03,
           3.6011e-02, 2.5543e-02]],

         [[2.2742e-01, 3.7193e-05, 1.1367e-04,  ..., 3.0003e-03,
           1.9882e-02, 2.8687e-02]],

         ...,

         [[1.2988e-01, 7.0751e-05, 7.5281e-05,  ..., 6.6719e-03,
           3.3875e-02, 3.9612e-02]],

         [[7.8430e-02, 7.8559e-05, 6.5148e-05,  ..., 4.0932e-03,
           1.1841e-02, 2.2476e-02]],

         [[4.4678e-01, 4.5359e-05, 8.2731e-05,  ..., 2.0889e-02,
           2.3666e-02, 1.3916e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6150e-01, 1.2398e-05, 1.1802e-05,  ..., 2.4368e-02,
           7.2205e-02, 6.4697e-02]],

         [[4.7363e-01, 2.3031e-04, 4.3201e-04,  ..., 7.2479e-03,
           1.4992e-02, 2.1469e-02]],

         [[1.9617e-01, 2.7061e-04, 6.8045e-04,  ..., 4.9095e-03,
           2.5977e-01, 3.0151e-01]],

         ...,

         [[6.0352e-01, 6.1750e-05, 2.0885e-04,  ..., 7.9575e-03,
           7.5562e-02, 5.3131e-02]],

         [[1.9617e-01, 5.0247e-05, 2.5415e-04,  ..., 1.3962e-03,
           8.4839e-03, 2.9846e-02]],

         [[1.7834e-01, 3.6418e-05, 1.4019e-04,  ..., 3.9625e-04,
           6.5956e-03, 1.7654e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6675e-01, 7.0035e-05, 1.2934e-04,  ..., 5.4565e-02,
           7.1167e-02, 7.9407e-02]],

         [[1.5686e-01, 2.2113e-05, 2.0862e-05,  ..., 1.2192e-02,
           6.7329e-03, 1.3077e-02]],

         [[6.2061e-01, 1.1545e-04, 1.9491e-04,  ..., 2.0111e-02,
           1.3290e-02, 8.6823e-03]],

         ...,

         [[6.4893e-01, 5.3644e-07, 1.1921e-07,  ..., 1.4490e-01,
           1.9495e-01, 3.4065e-03]],

         [[3.9575e-01, 6.4373e-06, 2.8610e-05,  ..., 1.7285e-01,
           9.1064e-02, 2.9736e-03]],

         [[9.6191e-02, 1.7881e-06, 8.3447e-07,  ..., 4.8370e-02,
           8.4424e-01, 5.1384e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4209e-01, 7.4089e-05, 1.8847e-04,  ..., 7.3120e-02,
           1.9836e-02, 3.7270e-03]],

         [[7.9834e-02, 9.9540e-06, 8.4043e-06,  ..., 1.1414e-02,
           9.2590e-02, 6.8408e-01]],

         [[3.0249e-01, 9.9945e-04, 2.6073e-03,  ..., 1.2655e-03,
           7.6027e-03, 1.0576e-03]],

         ...,

         [[2.7197e-01, 2.6298e-04, 5.4169e-04,  ..., 2.5192e-02,
           9.8267e-03, 2.4509e-03]],

         [[8.4839e-02, 4.1723e-07, 1.3709e-06,  ..., 6.5723e-01,
           2.1667e-01, 1.7303e-02]],

         [[1.5112e-01, 2.9802e-07, 1.0133e-06,  ..., 4.0112e-01,
           3.5669e-01, 1.7349e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 1.4782e-04, 4.5228e-04,  ..., 3.7193e-03,
           4.2496e-03, 5.7936e-04]],

         [[9.7998e-01, 1.5795e-05, 6.3956e-05,  ..., 3.3855e-05,
           3.3557e-05, 1.0067e-04]],

         [[7.8271e-01, 5.3525e-05, 1.4734e-04,  ..., 6.1989e-04,
           3.2163e-04, 5.8270e-04]],

         ...,

         [[8.4863e-01, 2.0325e-05, 2.5272e-05,  ..., 2.4460e-02,
           1.9043e-02, 1.4603e-02]],

         [[5.7471e-01, 8.2850e-06, 8.5235e-06,  ..., 3.6106e-03,
           5.7678e-03, 1.6434e-02]],

         [[1.2408e-01, 3.0220e-05, 9.0182e-05,  ..., 5.3673e-03,
           4.3549e-02, 7.3682e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6111e-02, 9.5367e-06, 9.0003e-06,  ..., 1.1432e-01,
           9.5520e-02, 3.0460e-03]],

         [[7.6660e-01, 2.3305e-05, 2.2233e-05,  ..., 3.1647e-02,
           1.3138e-02, 1.4046e-02]],

         [[1.3977e-01, 4.6790e-05, 7.8797e-05,  ..., 1.4026e-01,
           4.3304e-02, 4.2114e-02]],

         ...,

         [[6.7578e-01, 1.1563e-05, 3.9995e-05,  ..., 4.7089e-02,
           9.1492e-02, 2.5208e-02]],

         [[6.2744e-01, 1.3888e-04, 8.9979e-04,  ..., 1.6205e-02,
           8.2397e-03, 2.5894e-02]],

         [[6.0254e-01, 1.1444e-05, 6.1989e-05,  ..., 6.4354e-03,
           2.9297e-02, 5.9631e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6724e-01, 5.6982e-05, 6.8486e-05,  ..., 3.4790e-02,
           5.2528e-03, 1.7365e-02]],

         [[2.0459e-01, 6.1870e-05, 5.7459e-05,  ..., 1.6922e-02,
           1.3542e-02, 1.3916e-02]],

         [[7.8857e-02, 1.0610e-04, 2.0134e-04,  ..., 5.7251e-02,
           7.0129e-02, 7.0129e-02]],

         ...,

         [[3.7048e-02, 6.9141e-06, 1.0169e-04,  ..., 5.6381e-03,
           1.0956e-02, 1.2102e-03]],

         [[1.0004e-01, 7.8678e-06, 5.6624e-05,  ..., 1.1223e-02,
           2.8671e-02, 1.2520e-02]],

         [[9.0088e-01, 1.6391e-05, 2.0087e-05,  ..., 1.3054e-02,
           4.9255e-02, 3.7403e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8257e-01, 2.4438e-06, 1.1027e-05,  ..., 2.1289e-01,
           1.7786e-01, 1.4868e-01]],

         [[3.9233e-01, 2.8014e-05, 3.2306e-05,  ..., 3.8391e-02,
           3.6377e-02, 2.9053e-01]],

         [[5.5122e-03, 1.7881e-07, 1.1921e-07,  ..., 4.7339e-01,
           4.1455e-01, 3.8552e-04]],

         ...,

         [[1.6272e-01, 1.2970e-04, 3.1042e-04,  ..., 1.6357e-02,
           3.5461e-02, 6.5247e-02]],

         [[3.4766e-01, 9.4235e-05, 4.9055e-05,  ..., 4.2175e-02,
           1.6553e-01, 1.8417e-02]],

         [[2.0691e-02, 2.6226e-05, 6.5207e-05,  ..., 3.2043e-02,
           2.3270e-02, 7.4707e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9370e-01, 3.2187e-06, 5.2333e-05,  ..., 9.9121e-02,
           1.6211e-01, 3.0054e-01]],

         [[5.8984e-01, 3.1757e-04, 8.6164e-04,  ..., 3.4943e-03,
           9.9564e-03, 2.6642e-02]],

         [[8.7952e-02, 7.5698e-06, 4.1187e-05,  ..., 4.7722e-03,
           2.8564e-02, 3.2349e-02]],

         ...,

         [[6.8555e-01, 7.1228e-05, 7.7009e-05,  ..., 8.8562e-02,
           7.2266e-02, 2.4200e-02]],

         [[9.6313e-02, 6.3241e-05, 1.6415e-04,  ..., 9.8419e-03,
           1.5732e-02, 8.4521e-01]],

         [[1.3477e-01, 2.4557e-05, 7.6354e-05,  ..., 6.6071e-03,
           7.8809e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8027e-01, 4.5240e-05, 4.3774e-04,  ..., 3.6499e-02,
           4.8920e-02, 6.1829e-02]],

         [[7.3059e-02, 1.7703e-05, 3.8922e-05,  ..., 2.9068e-02,
           7.9346e-02, 6.5491e-02]],

         [[5.3833e-02, 1.2481e-04, 5.8603e-04,  ..., 1.2085e-01,
           2.7344e-01, 3.0441e-02]],

         ...,

         [[8.3887e-01, 1.3196e-04, 1.3459e-04,  ..., 7.1182e-03,
           1.6006e-02, 4.0970e-03]],

         [[7.5391e-01, 1.3709e-06, 7.3814e-04,  ..., 4.2343e-03,
           1.5278e-03, 5.2917e-02]],

         [[5.7959e-01, 8.3685e-05, 8.0645e-05,  ..., 3.6743e-02,
           7.2510e-02, 1.0388e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1192e-04, 4.9019e-04, 1.4417e-01,  ..., 1.6651e-03,
           9.0714e-03, 3.3295e-02]],

         [[1.0842e-04, 9.5189e-05, 1.3232e-01,  ..., 9.9915e-02,
           3.6392e-03, 1.9669e-02]],

         [[2.2089e-04, 1.9765e-04, 4.7974e-02,  ..., 3.9917e-02,
           8.9600e-02, 1.0522e-01]],

         ...,

         [[3.8505e-05, 1.7285e-05, 2.9587e-02,  ..., 2.9199e-01,
           3.1079e-01, 6.9336e-02]],

         [[8.5020e-04, 9.6369e-04, 1.7456e-01,  ..., 1.5976e-02,
           3.0334e-02, 1.7548e-02]],

         [[3.5763e-06, 4.7684e-07, 1.0300e-02,  ..., 6.6757e-06,
           8.9109e-05, 8.0633e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7680e-04, 5.1618e-05, 2.9663e-01,  ..., 7.3814e-04,
           9.7752e-04, 3.4882e-02]],

         [[1.1408e-04, 6.8426e-05, 2.4597e-01,  ..., 4.6310e-03,
           2.7800e-04, 2.2522e-02]],

         [[2.0087e-05, 6.8545e-06, 2.5366e-01,  ..., 1.7643e-05,
           6.3658e-04, 2.2171e-02]],

         ...,

         [[9.4891e-04, 2.0885e-04, 2.7832e-01,  ..., 4.5700e-03,
           3.6359e-04, 6.1989e-03]],

         [[3.9637e-05, 4.5478e-05, 3.1079e-01,  ..., 2.0099e-04,
           3.3259e-04, 3.1494e-02]],

         [[6.4492e-05, 4.4942e-05, 2.2302e-01,  ..., 6.3133e-04,
           1.0948e-03, 8.9188e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7716e-05, 3.2783e-05, 2.1777e-01,  ..., 2.2829e-05,
           4.8327e-04, 3.0640e-02]],

         [[7.0989e-05, 3.8147e-06, 2.6538e-01,  ..., 1.1307e-02,
           3.6073e-04, 2.2491e-02]],

         [[6.3598e-05, 1.5819e-04, 2.6294e-01,  ..., 1.6451e-03,
           7.8344e-04, 1.8616e-02]],

         ...,

         [[1.8597e-05, 3.9697e-05, 1.6895e-01,  ..., 6.6650e-02,
           1.1627e-02, 1.3257e-01]],

         [[1.7285e-05, 4.5896e-05, 1.6028e-01,  ..., 2.1103e-02,
           9.1314e-04, 1.5991e-02]],

         [[1.1545e-04, 4.3094e-05, 1.6296e-01,  ..., 3.6411e-03,
           4.2486e-04, 9.1171e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3631e-05, 2.6107e-05, 4.6844e-03,  ..., 1.2372e-01,
           6.5332e-01, 4.6600e-02]],

         [[5.5850e-05, 2.7919e-04, 1.2585e-01,  ..., 9.1370e-02,
           5.2277e-02, 9.4681e-03]],

         [[8.8334e-05, 6.6614e-04, 9.3506e-02,  ..., 5.9175e-04,
           1.8060e-04, 3.0613e-04]],

         ...,

         [[3.0861e-03, 5.7411e-04, 2.7084e-02,  ..., 4.3602e-03,
           5.0087e-03, 6.6223e-03]],

         [[9.1124e-04, 3.0470e-04, 1.6541e-02,  ..., 6.6956e-02,
           3.0249e-01, 2.7481e-02]],

         [[2.3711e-04, 6.1154e-05, 4.1534e-02,  ..., 3.3508e-02,
           1.0056e-02, 3.4973e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0055, 0.0010, 0.1638,  ..., 0.1348, 0.0202, 0.0223]],

         [[0.0004, 0.0008, 0.2350,  ..., 0.0208, 0.0168, 0.1542]],

         [[0.0034, 0.0011, 0.1915,  ..., 0.0368, 0.0211, 0.0542]],

         ...,

         [[0.0019, 0.0007, 0.0858,  ..., 0.2012, 0.0757, 0.1299]],

         [[0.0008, 0.0014, 0.1866,  ..., 0.0077, 0.0101, 0.2759]],

         [[0.0003, 0.0008, 0.1636,  ..., 0.0365, 0.0151, 0.0424]]]],
       device='cuda:0', dtype=torch.float16)),(tensor([[[[5.7888e-04, 1.2910e-04, 7.9107e-04,  ..., 7.1924e-01,
           1.5552e-01, 6.6895e-02]],

         [[7.1335e-04, 2.9731e-04, 4.6539e-03,  ..., 3.0334e-02,
           3.9355e-01, 1.7288e-02]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.3709e-06,
           2.0337e-04, 9.9951e-01]],

         ...,

         [[1.6272e-05, 2.6822e-05, 5.3835e-04,  ..., 3.9787e-03,
           6.5575e-03, 9.7314e-01]],

         [[6.8817e-03, 6.4659e-03, 5.3596e-03,  ..., 2.7222e-02,
           4.7791e-02, 8.3008e-03]],

         [[1.5556e-02, 3.3646e-03, 9.3412e-04,  ..., 1.8188e-02,
           1.2115e-02, 4.0985e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5736e-03, 2.2202e-03, 9.9487e-03,  ..., 4.3884e-02,
           1.6821e-01, 2.6050e-01]],

         [[5.9605e-08, 5.9605e-08, 0.0000e+00,  ..., 1.5369e-01,
           7.6855e-01, 7.3792e-02]],

         [[8.1587e-04, 1.8673e-03, 7.1602e-03,  ..., 2.4994e-02,
           2.4216e-02, 1.1436e-02]],

         ...,

         [[1.7233e-03, 2.1610e-03, 8.0948e-03,  ..., 1.9882e-02,
           7.0496e-02, 1.0663e-01]],

         [[4.7569e-03, 1.2535e-02, 3.8177e-02,  ..., 1.3290e-02,
           1.7410e-02, 1.2932e-02]],

         [[1.9894e-03, 3.4103e-03, 7.1106e-03,  ..., 2.1106e-01,
           2.5208e-02, 6.0944e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7148e-02, 7.7286e-03, 4.2938e-02,  ..., 5.2216e-02,
           5.0201e-02, 4.2938e-02]],

         [[2.8381e-03, 1.0900e-03, 4.2114e-03,  ..., 7.3624e-03,
           2.3708e-03, 1.0216e-02]],

         [[7.5951e-03, 2.0599e-03, 2.0676e-03,  ..., 3.2471e-02,
           3.6530e-02, 2.3300e-02]],

         ...,

         [[3.8452e-03, 5.0449e-04, 1.7881e-03,  ..., 1.7542e-01,
           2.3059e-01, 1.9263e-01]],

         [[3.3493e-03, 1.7548e-04, 5.8222e-04,  ..., 7.6843e-02,
           3.5187e-02, 1.0101e-01]],

         [[4.4975e-03, 1.0147e-03, 2.1744e-03,  ..., 4.8828e-01,
           1.6614e-01, 7.7881e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8418e-01, 3.2187e-06, 4.0531e-06,  ..., 1.9054e-03,
           2.4185e-03, 1.2573e-02]],

         [[9.4336e-01, 4.3511e-06, 9.1195e-06,  ..., 2.8648e-03,
           2.3636e-02, 5.1155e-03]],

         [[6.9189e-01, 5.1856e-06, 2.0862e-05,  ..., 4.4403e-03,
           1.6460e-03, 2.1606e-02]],

         ...,

         [[8.4351e-02, 4.4441e-04, 6.4659e-04,  ..., 2.6011e-04,
           7.3862e-04, 8.9264e-03]],

         [[5.4199e-01, 1.5676e-04, 1.1539e-03,  ..., 2.0538e-02,
           1.7563e-02, 1.1841e-02]],

         [[3.4692e-01, 2.2840e-04, 4.8971e-04,  ..., 6.2439e-02,
           2.9251e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1226e-01, 2.9206e-06, 3.5167e-06,  ..., 4.4346e-05,
           3.9160e-05, 6.6986e-03]],

         [[3.5767e-01, 5.0664e-06, 5.9605e-08,  ..., 3.8965e-01,
           2.3083e-01, 1.8829e-02]],

         [[8.9453e-01, 2.7323e-04, 1.7524e-05,  ..., 3.5126e-02,
           1.8188e-02, 4.8447e-03]],

         ...,

         [[3.5596e-01, 1.1045e-04, 2.2840e-04,  ..., 8.4457e-03,
           7.8506e-03, 2.2110e-02]],

         [[1.4429e-01, 1.9522e-03, 1.6851e-03,  ..., 3.0327e-03,
           1.3723e-03, 6.9885e-03]],

         [[1.9751e-01, 2.5868e-05, 7.9453e-05,  ..., 2.6587e-01,
           8.4656e-02, 4.9164e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 2.8729e-04, 1.3578e-04,  ..., 4.1870e-02,
           2.1301e-02, 2.9984e-02]],

         [[6.3330e-01, 1.1504e-05, 1.1444e-05,  ..., 1.2286e-01,
           9.5520e-03, 1.5078e-03]],

         [[8.6963e-01, 2.4259e-05, 5.6028e-06,  ..., 6.2714e-03,
           3.7903e-02, 2.3804e-03]],

         ...,

         [[4.2822e-01, 4.7684e-07, 1.7881e-06,  ..., 6.5279e-04,
           8.2350e-04, 5.4443e-01]],

         [[4.6582e-01, 1.3113e-06, 5.2750e-05,  ..., 2.5034e-06,
           2.3961e-05, 1.8799e-02]],

         [[9.3311e-01, 1.0729e-06, 8.4043e-06,  ..., 2.9802e-07,
           3.1590e-06, 1.9073e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4595e-01, 8.8274e-05, 1.2994e-05,  ..., 1.6260e-01,
           1.2830e-01, 2.5116e-02]],

         [[8.9746e-01, 4.4107e-06, 1.8477e-06,  ..., 5.4061e-05,
           1.6749e-05, 5.7907e-03]],

         [[7.8125e-01, 1.7500e-04, 5.4836e-05,  ..., 5.7487e-03,
           3.8509e-03, 3.3752e-02]],

         ...,

         [[9.0234e-01, 2.7227e-04, 1.8597e-05,  ..., 9.6359e-03,
           7.9117e-03, 1.2451e-02]],

         [[7.6807e-01, 1.2118e-04, 3.9518e-05,  ..., 6.7627e-02,
           4.4098e-02, 6.1615e-02]],

         [[8.8818e-01, 6.1989e-05, 3.4094e-05,  ..., 4.1565e-02,
           2.5497e-02, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7588e-01, 5.4955e-05, 3.1590e-05,  ..., 2.3239e-02,
           1.4923e-02, 5.0537e-02]],

         [[5.5713e-01, 9.4354e-05, 1.3733e-04,  ..., 1.0559e-01,
           2.2095e-02, 8.3557e-02]],

         [[8.3435e-02, 2.1458e-06, 3.5167e-06,  ..., 3.5254e-01,
           6.7749e-03, 2.3174e-03]],

         ...,

         [[5.8740e-01, 4.8995e-05, 1.7524e-05,  ..., 2.2003e-02,
           1.7975e-02, 2.8137e-02]],

         [[8.7939e-01, 2.9278e-04, 7.4267e-05,  ..., 2.7885e-03,
           1.0662e-03, 7.1602e-03]],

         [[3.9331e-01, 1.0669e-05, 9.0599e-06,  ..., 1.5881e-01,
           1.6052e-01, 6.4148e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 3.4161e-03, 1.2865e-03,  ..., 8.2731e-04,
           5.9128e-04, 8.2779e-03]],

         [[9.6045e-01, 9.9540e-06, 3.5346e-05,  ..., 1.7424e-03,
           5.5695e-04, 4.7827e-04]],

         [[8.7012e-01, 5.7757e-05, 5.2989e-05,  ..., 2.1324e-03,
           1.3142e-03, 1.1482e-02]],

         ...,

         [[5.7422e-01, 1.4770e-04, 3.4070e-04,  ..., 1.8265e-02,
           3.2501e-02, 1.6870e-01]],

         [[8.4570e-01, 4.4107e-06, 1.8418e-05,  ..., 4.9713e-02,
           1.3885e-02, 6.1684e-03]],

         [[8.7793e-01, 8.7619e-06, 1.3828e-05,  ..., 1.3329e-02,
           1.5434e-02, 6.6101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1899e-01, 3.8242e-04, 8.1873e-04,  ..., 2.1152e-03,
           1.4257e-03, 1.2505e-02]],

         [[5.1465e-01, 2.8491e-05, 2.4974e-05,  ..., 2.3413e-01,
           7.1045e-02, 3.0945e-02]],

         [[8.8086e-01, 2.2590e-05, 2.4021e-05,  ..., 4.0802e-02,
           1.5373e-02, 9.5444e-03]],

         ...,

         [[7.5000e-01, 9.8288e-05, 1.3304e-04,  ..., 2.6560e-04,
           8.3923e-05, 4.2033e-04]],

         [[7.8027e-01, 1.6356e-04, 4.5478e-05,  ..., 3.3630e-02,
           1.7685e-02, 6.6101e-02]],

         [[5.5518e-01, 1.2934e-05, 1.4484e-05,  ..., 1.9116e-01,
           1.2390e-01, 1.9638e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3416e-02, 4.1723e-06, 1.6153e-05,  ..., 7.8678e-06,
           9.4175e-06, 2.5673e-03]],

         [[4.8779e-01, 1.1265e-05, 2.0266e-05,  ..., 6.1035e-02,
           1.1276e-02, 4.8584e-02]],

         [[8.3887e-01, 8.3447e-07, 2.2054e-06,  ..., 2.0754e-04,
           5.9664e-05, 1.8644e-03]],

         ...,

         [[9.4238e-01, 5.8711e-05, 5.6267e-05,  ..., 2.9526e-03,
           2.8110e-04, 1.7567e-03]],

         [[7.3340e-01, 1.6546e-04, 6.3539e-05,  ..., 3.3951e-03,
           9.0820e-02, 1.3596e-02]],

         [[6.2939e-01, 7.2289e-04, 1.3268e-04,  ..., 7.9679e-04,
           2.1112e-04, 7.9651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1431e-01, 1.5199e-05, 1.6689e-05,  ..., 3.0575e-03,
           5.7373e-03, 5.4883e-01]],

         [[5.0732e-01, 8.8549e-04, 8.9073e-04,  ..., 3.2043e-03,
           2.8849e-04, 1.7487e-02]],

         [[6.0840e-01, 3.2377e-04, 3.1090e-04,  ..., 3.6373e-03,
           7.8058e-04, 7.8354e-03]],

         ...,

         [[3.2642e-01, 9.2506e-05, 3.0684e-04,  ..., 2.1713e-02,
           1.9932e-03, 6.7291e-03]],

         [[4.0796e-01, 3.5703e-05, 4.3213e-05,  ..., 1.2262e-01,
           9.5825e-03, 3.2532e-02]],

         [[3.3960e-01, 4.5598e-05, 1.1283e-04,  ..., 1.0315e-02,
           3.1860e-02, 5.4395e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4717e-01, 1.6046e-04, 2.0921e-04,  ..., 1.6510e-02,
           5.7678e-02, 5.0000e-01]],

         [[6.9141e-01, 8.1420e-05, 1.1623e-04,  ..., 1.8753e-02,
           3.8185e-03, 1.5762e-02]],

         [[4.1602e-01, 3.4261e-04, 6.1798e-04,  ..., 6.1684e-03,
           9.7227e-04, 9.3002e-03]],

         ...,

         [[9.3323e-02, 3.6895e-05, 8.4937e-05,  ..., 4.1628e-04,
           8.7967e-03, 8.4473e-01]],

         [[7.4512e-01, 3.2282e-04, 2.7823e-04,  ..., 2.7294e-03,
           1.4511e-02, 2.1484e-02]],

         [[7.3633e-01, 1.0198e-04, 3.8981e-05,  ..., 2.7924e-02,
           7.8821e-04, 1.1475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1616e-01, 6.3324e-04, 1.1854e-03,  ..., 1.4137e-02,
           6.4964e-03, 8.0505e-02]],

         [[6.9385e-01, 1.9014e-05, 5.2571e-05,  ..., 3.4058e-02,
           1.3351e-02, 1.1432e-01]],

         [[1.8774e-01, 5.1260e-06, 1.6391e-05,  ..., 1.8013e-04,
           7.7963e-04, 1.0307e-02]],

         ...,

         [[8.1836e-01, 1.3113e-06, 6.5565e-07,  ..., 7.7454e-02,
           5.6366e-02, 2.3499e-02]],

         [[5.4785e-01, 7.2813e-04, 1.2035e-03,  ..., 3.8986e-03,
           7.8440e-04, 1.4259e-02]],

         [[6.3330e-01, 2.3842e-06, 2.5630e-06,  ..., 5.5847e-02,
           2.1255e-02, 9.4238e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5010e-01, 5.4598e-05, 1.9848e-05,  ..., 4.7607e-03,
           6.2275e-04, 3.6182e-03]],

         [[1.4880e-01, 3.7384e-04, 3.1281e-04,  ..., 6.0892e-04,
           5.1022e-04, 1.7471e-03]],

         [[6.4746e-01, 2.2197e-04, 1.3304e-04,  ..., 1.6365e-03,
           5.6458e-04, 1.2779e-02]],

         ...,

         [[5.2783e-01, 4.1723e-07, 4.1723e-07,  ..., 3.1158e-02,
           3.7460e-03, 1.5526e-02]],

         [[3.4595e-01, 5.9605e-08, 5.9605e-08,  ..., 1.6284e-01,
           6.6589e-02, 8.2153e-02]],

         [[3.0811e-01, 3.3975e-06, 8.5235e-06,  ..., 5.3253e-03,
           2.0695e-03, 1.5823e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6721e-02, 2.3842e-07, 6.5565e-07,  ..., 7.2908e-04,
           2.6393e-04, 8.9551e-01]],

         [[1.0455e-01, 8.4043e-06, 2.5332e-05,  ..., 1.6968e-01,
           6.3538e-02, 6.3782e-02]],

         [[2.9590e-01, 2.3162e-04, 4.9400e-04,  ..., 9.9182e-03,
           1.7424e-03, 3.2501e-02]],

         ...,

         [[4.3457e-01, 7.1347e-05, 1.3804e-04,  ..., 2.3911e-02,
           3.3150e-03, 2.5391e-02]],

         [[7.2461e-01, 1.0133e-06, 1.2875e-05,  ..., 7.5264e-03,
           2.2335e-03, 6.8169e-03]],

         [[3.9941e-01, 1.2612e-04, 1.1754e-04,  ..., 1.0696e-02,
           1.7891e-03, 1.0017e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8643e-01, 2.4557e-05, 4.0352e-05,  ..., 1.7914e-02,
           8.5449e-03, 1.5002e-01]],

         [[4.7803e-01, 3.9887e-04, 5.7888e-04,  ..., 1.0941e-02,
           4.5166e-03, 3.8483e-02]],

         [[5.2185e-02, 7.7486e-06, 1.1921e-05,  ..., 6.9153e-02,
           6.9275e-03, 4.0344e-02]],

         ...,

         [[6.8213e-01, 2.4557e-05, 2.0325e-05,  ..., 2.6978e-02,
           2.7561e-03, 8.3740e-02]],

         [[5.6006e-01, 1.2994e-05, 2.6584e-05,  ..., 1.0559e-01,
           3.4351e-03, 1.1688e-01]],

         [[8.2520e-01, 4.6015e-05, 4.9412e-05,  ..., 1.4467e-03,
           4.6134e-04, 9.8190e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8882e-01, 1.7090e-03, 7.8917e-04,  ..., 4.2963e-04,
           2.1935e-04, 4.8294e-03]],

         [[3.7598e-01, 6.1226e-04, 4.5180e-04,  ..., 1.6222e-03,
           2.1420e-03, 7.8735e-03]],

         [[3.5400e-01, 2.2471e-05, 3.2306e-05,  ..., 1.6333e-01,
           1.3374e-02, 4.1931e-02]],

         ...,

         [[3.7280e-01, 2.3663e-04, 9.1434e-05,  ..., 7.8430e-03,
           7.8583e-03, 4.4006e-02]],

         [[8.3203e-01, 6.4194e-05, 6.0987e-04,  ..., 3.2783e-06,
           3.5214e-04, 5.8861e-03]],

         [[8.4570e-01, 1.5640e-04, 1.3089e-04,  ..., 2.3880e-03,
           7.0143e-04, 2.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3789e-01, 2.4557e-04, 3.5739e-04,  ..., 8.5449e-02,
           1.1925e-02, 3.9734e-02]],

         [[2.6709e-01, 1.3475e-03, 7.4148e-04,  ..., 9.0637e-03,
           2.5177e-03, 1.1024e-02]],

         [[9.7778e-02, 1.2815e-05, 1.2279e-05,  ..., 4.3030e-02,
           3.0289e-02, 4.3823e-01]],

         ...,

         [[3.6548e-01, 2.4796e-05, 5.2035e-05,  ..., 1.0612e-02,
           8.6689e-04, 1.3214e-02]],

         [[5.3320e-01, 8.7976e-05, 8.1480e-05,  ..., 2.3621e-02,
           1.0857e-02, 5.7770e-02]],

         [[2.8101e-01, 3.9279e-05, 3.3855e-05,  ..., 6.4148e-02,
           3.8025e-02, 2.6611e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3340e-01, 3.1292e-05, 3.2783e-05,  ..., 7.1907e-03,
           1.8646e-02, 8.0185e-03]],

         [[1.2451e-01, 1.7405e-05, 1.1444e-05,  ..., 9.5459e-02,
           1.0880e-02, 3.8574e-02]],

         [[1.6406e-01, 3.9637e-05, 2.2233e-05,  ..., 9.2545e-03,
           9.1095e-03, 7.4658e-01]],

         ...,

         [[3.9520e-02, 5.2631e-05, 8.2791e-05,  ..., 6.0940e-04,
           1.4734e-03, 2.3788e-02]],

         [[1.4221e-01, 2.8110e-04, 2.5702e-04,  ..., 2.2339e-02,
           3.6255e-02, 3.6888e-03]],

         [[2.4426e-01, 1.2321e-03, 9.0885e-04,  ..., 4.2496e-03,
           7.7896e-03, 2.6398e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5425e-01, 3.9935e-05, 3.6538e-05,  ..., 2.7527e-02,
           6.4888e-03, 4.1016e-02]],

         [[3.2910e-01, 9.6679e-05, 1.9312e-05,  ..., 5.7411e-04,
           1.0866e-04, 9.4833e-03]],

         [[8.1885e-01, 3.2961e-05, 7.4506e-06,  ..., 1.4057e-03,
           7.9036e-05, 6.5002e-03]],

         ...,

         [[4.5685e-02, 5.4455e-04, 1.0252e-03,  ..., 3.8090e-03,
           9.4843e-04, 1.0857e-02]],

         [[5.3906e-01, 5.6922e-05, 1.2338e-04,  ..., 3.4122e-03,
           2.5368e-03, 1.7059e-02]],

         [[4.2676e-01, 6.7759e-04, 2.1362e-03,  ..., 3.6133e-02,
           3.1090e-03, 1.6418e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4438e-01, 3.8266e-05, 9.0718e-05,  ..., 5.1498e-03,
           9.7733e-03, 1.0490e-02]],

         [[7.9529e-02, 1.4532e-04, 4.3058e-04,  ..., 1.1276e-02,
           1.2680e-02, 2.5421e-02]],

         [[1.4587e-01, 9.8050e-05, 2.0444e-04,  ..., 9.1095e-03,
           1.1879e-02, 1.7014e-02]],

         ...,

         [[3.2104e-02, 5.9128e-05, 8.7380e-05,  ..., 4.6692e-02,
           3.1113e-02, 5.0476e-02]],

         [[1.0370e-01, 7.8380e-05, 7.5936e-05,  ..., 6.3248e-03,
           1.1101e-02, 3.8147e-02]],

         [[5.3369e-01, 8.5890e-05, 1.2493e-04,  ..., 1.4214e-02,
           7.7286e-03, 1.4000e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3623e-01, 2.0325e-05, 1.8656e-05,  ..., 5.2094e-02,
           1.1177e-02, 3.8116e-02]],

         [[4.0454e-01, 1.9670e-04, 4.1461e-04,  ..., 1.2024e-02,
           1.3412e-02, 1.7639e-02]],

         [[1.2323e-01, 7.4863e-05, 2.0993e-04,  ..., 4.3243e-02,
           5.0964e-02, 6.3086e-01]],

         ...,

         [[5.6348e-01, 8.2254e-05, 1.4555e-04,  ..., 1.8539e-02,
           1.1787e-02, 3.8055e-02]],

         [[2.0740e-01, 2.1696e-05, 2.9993e-04,  ..., 2.9125e-03,
           1.9287e-02, 5.0446e-02]],

         [[4.0430e-01, 4.0340e-04, 5.8031e-04,  ..., 3.0403e-03,
           3.0155e-03, 7.3471e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6760e-01, 2.0730e-04, 1.6665e-04,  ..., 2.7802e-02,
           2.4719e-02, 8.3008e-02]],

         [[5.9723e-02, 5.2452e-05, 7.9572e-05,  ..., 1.7242e-02,
           1.7929e-02, 3.1235e-02]],

         [[2.0996e-01, 4.9353e-05, 7.4565e-05,  ..., 4.0627e-03,
           1.4830e-03, 1.6870e-03]],

         ...,

         [[7.4121e-01, 1.0729e-05, 1.3292e-05,  ..., 4.7394e-02,
           8.5144e-02, 5.4108e-02]],

         [[3.1982e-01, 1.9372e-05, 8.9645e-05,  ..., 2.4673e-02,
           4.9744e-03, 2.8168e-02]],

         [[1.9409e-01, 2.0981e-05, 2.8551e-05,  ..., 4.3066e-01,
           1.4307e-01, 2.4872e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2253e-02, 6.6757e-06, 3.1948e-05,  ..., 2.7332e-03,
           5.2977e-04, 1.3145e-02]],

         [[1.5900e-02, 5.3644e-07, 1.1921e-06,  ..., 4.4751e-04,
           3.0117e-03, 9.7607e-01]],

         [[2.7145e-02, 8.6546e-05, 4.4918e-04,  ..., 2.7943e-03,
           7.6234e-05, 4.5991e-04]],

         ...,

         [[1.0669e-01, 3.0708e-04, 9.4128e-04,  ..., 8.9645e-03,
           1.1549e-03, 9.0332e-03]],

         [[8.7280e-02, 7.1526e-07, 1.8477e-06,  ..., 6.3904e-02,
           7.0267e-03, 2.3224e-02]],

         [[2.9114e-02, 4.1723e-07, 1.9670e-06,  ..., 4.2285e-01,
           2.0129e-01, 1.1932e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0664e-01, 2.0564e-04, 7.5150e-04,  ..., 7.0343e-03,
           8.7166e-04, 1.1238e-02]],

         [[9.7900e-01, 1.4865e-04, 2.9111e-04,  ..., 2.9862e-05,
           1.9550e-05, 4.4727e-04]],

         [[6.6260e-01, 2.6941e-05, 2.5797e-04,  ..., 9.4318e-04,
           1.1024e-03, 2.9114e-02]],

         ...,

         [[2.9492e-01, 1.8895e-05, 1.6153e-05,  ..., 3.5571e-01,
           1.1731e-01, 4.5929e-02]],

         [[4.4458e-01, 2.1458e-06, 5.9605e-07,  ..., 1.0357e-03,
           1.0519e-03, 5.9586e-03]],

         [[6.4636e-02, 2.3425e-05, 7.6890e-05,  ..., 1.2026e-03,
           4.2648e-03, 8.7842e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8054e-02, 7.6890e-06, 1.1921e-05,  ..., 1.3512e-02,
           1.8282e-03, 2.1667e-02]],

         [[5.7959e-01, 5.4538e-05, 2.5928e-05,  ..., 7.9346e-02,
           1.6449e-02, 6.2286e-02]],

         [[3.1006e-01, 3.3307e-04, 3.9458e-04,  ..., 7.8369e-02,
           6.8115e-02, 7.9041e-02]],

         ...,

         [[3.3813e-01, 1.7107e-05, 8.8334e-05,  ..., 8.8867e-02,
           2.1439e-02, 8.0261e-02]],

         [[6.8506e-01, 5.3763e-05, 1.3387e-04,  ..., 2.3766e-03,
           1.8806e-03, 1.0567e-02]],

         [[5.0586e-01, 4.8041e-05, 1.9228e-04,  ..., 1.1261e-02,
           7.5607e-03, 2.8992e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9662e-03, 3.1590e-06, 1.2517e-06,  ..., 5.2223e-03,
           3.4714e-04, 8.8644e-04]],

         [[4.8401e-02, 5.5432e-06, 1.7881e-06,  ..., 6.6423e-04,
           2.5105e-04, 1.0653e-03]],

         [[1.1908e-01, 7.3195e-05, 1.4961e-04,  ..., 8.7830e-02,
           1.6632e-02, 1.4819e-01]],

         ...,

         [[5.6000e-02, 5.2691e-05, 3.8028e-04,  ..., 7.8201e-03,
           1.4238e-03, 6.4812e-03]],

         [[1.2366e-01, 7.2837e-05, 3.5834e-04,  ..., 1.6998e-02,
           1.1044e-03, 1.1147e-02]],

         [[1.6980e-01, 5.2869e-05, 5.2273e-05,  ..., 2.6099e-01,
           2.5238e-02, 1.7075e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1123e-01, 1.0848e-05, 1.4544e-05,  ..., 1.4087e-01,
           4.2786e-02, 8.9539e-02]],

         [[3.2861e-01, 7.9691e-05, 1.4412e-04,  ..., 1.5282e-02,
           7.3166e-03, 2.6929e-01]],

         [[4.5679e-01, 5.4240e-06, 1.5438e-05,  ..., 1.0938e-01,
           1.1823e-01, 1.6736e-01]],

         ...,

         [[4.7192e-01, 1.4114e-04, 1.4138e-04,  ..., 1.6403e-02,
           3.7613e-03, 1.3283e-02]],

         [[1.7908e-01, 8.7023e-06, 7.8082e-06,  ..., 1.7357e-03,
           3.9792e-04, 1.4587e-02]],

         [[7.6904e-03, 2.5868e-05, 7.2241e-05,  ..., 1.0033e-02,
           2.7054e-02, 8.6816e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6025e-01, 5.5194e-05, 2.0671e-04,  ..., 5.9021e-02,
           2.9785e-02, 1.7212e-01]],

         [[1.1749e-01, 1.3733e-03, 1.4992e-03,  ..., 6.2027e-03,
           6.5231e-03, 1.8082e-02]],

         [[1.6223e-01, 2.0456e-04, 3.3593e-04,  ..., 1.3847e-02,
           2.0905e-03, 6.8283e-03]],

         ...,

         [[6.7578e-01, 6.1131e-04, 1.9693e-04,  ..., 7.1228e-02,
           1.9165e-02, 3.0151e-02]],

         [[6.4453e-02, 2.2531e-04, 6.3181e-04,  ..., 2.8763e-03,
           1.0193e-02, 8.3545e-01]],

         [[3.2642e-01, 3.3379e-05, 1.6177e-04,  ..., 2.5415e-01,
           3.3667e-01, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5469e-01, 2.7776e-04, 1.0977e-03,  ..., 9.8267e-02,
           2.7686e-01, 5.9662e-03]],

         [[1.2793e-01, 1.5914e-05, 3.3498e-05,  ..., 7.0763e-03,
           5.4779e-03, 1.0645e-01]],

         [[5.4102e-01, 3.9840e-04, 2.1820e-03,  ..., 8.2626e-03,
           6.4163e-03, 1.2802e-02]],

         ...,

         [[6.5918e-01, 2.7323e-04, 4.5586e-03,  ..., 2.3605e-02,
           2.9602e-03, 1.3496e-02]],

         [[8.9355e-01, 2.1374e-04, 2.0027e-03,  ..., 1.6413e-03,
           8.6927e-04, 1.1833e-02]],

         [[4.0015e-01, 1.6832e-04, 1.7810e-04,  ..., 2.8711e-01,
           3.1342e-02, 5.8319e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2343e-04, 5.9843e-04, 2.2729e-01,  ..., 1.2817e-02,
           1.0880e-02, 1.8330e-03]],

         [[8.0287e-05, 2.5058e-04, 1.9556e-01,  ..., 2.4815e-03,
           7.2365e-03, 2.9583e-03]],

         [[1.0395e-03, 1.0748e-03, 1.2030e-01,  ..., 2.7054e-02,
           1.2726e-02, 2.8793e-02]],

         ...,

         [[2.8849e-04, 4.6730e-05, 5.6519e-02,  ..., 7.7820e-02,
           7.0312e-02, 2.4170e-01]],

         [[2.8825e-04, 2.1629e-03, 1.9617e-01,  ..., 1.5915e-02,
           2.3193e-03, 5.5885e-03]],

         [[3.5644e-04, 4.1676e-04, 2.5635e-01,  ..., 9.5596e-03,
           5.4474e-03, 1.8005e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5125e-03, 2.0447e-03, 3.0225e-01,  ..., 3.5591e-03,
           3.0804e-03, 7.8201e-04]],

         [[1.7462e-03, 1.5774e-03, 2.3230e-01,  ..., 1.2426e-03,
           2.2945e-03, 1.2379e-03]],

         [[1.1902e-03, 2.4748e-04, 3.3789e-01,  ..., 9.8324e-04,
           1.3123e-03, 1.9321e-03]],

         ...,

         [[2.8076e-03, 5.9891e-04, 2.4878e-01,  ..., 5.3167e-04,
           9.0075e-04, 1.0509e-03]],

         [[1.6892e-04, 3.3069e-04, 3.4082e-01,  ..., 9.9945e-04,
           4.7836e-03, 2.8014e-04]],

         [[2.9981e-05, 9.8825e-05, 1.0590e-01,  ..., 1.2970e-03,
           7.9250e-04, 1.6165e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3390e-03, 1.1101e-03, 2.3682e-01,  ..., 5.5552e-04,
           1.9302e-03, 9.6512e-04]],

         [[5.2032e-03, 2.0409e-04, 2.7954e-01,  ..., 1.4277e-03,
           3.3188e-03, 1.1450e-04]],

         [[1.3933e-03, 1.8969e-03, 3.2275e-01,  ..., 1.2102e-03,
           1.8597e-03, 8.9788e-04]],

         ...,

         [[3.5763e-04, 7.7677e-04, 1.6968e-01,  ..., 3.5980e-02,
           2.2079e-02, 1.1963e-02]],

         [[7.0238e-04, 1.5078e-03, 2.5513e-01,  ..., 1.3180e-03,
           1.2922e-03, 5.0640e-04]],

         [[2.8439e-03, 1.4162e-03, 1.5906e-01,  ..., 1.3838e-03,
           1.2255e-03, 8.1682e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3355e-04, 3.8207e-05, 2.1042e-02,  ..., 2.6596e-02,
           3.4698e-02, 1.8604e-01]],

         [[1.0699e-04, 2.5272e-04, 1.7285e-01,  ..., 8.0490e-03,
           3.8033e-03, 3.2593e-02]],

         [[2.7704e-04, 1.0939e-03, 3.1421e-01,  ..., 1.5628e-04,
           1.8358e-04, 1.2636e-03]],

         ...,

         [[9.6321e-05, 1.1027e-05, 1.6510e-02,  ..., 1.1396e-04,
           1.9026e-04, 1.5747e-02]],

         [[5.7678e-03, 4.6015e-04, 1.5112e-01,  ..., 1.0948e-02,
           4.1473e-02, 9.3445e-02]],

         [[1.5378e-05, 4.5896e-06, 4.3983e-03,  ..., 5.0545e-04,
           7.7629e-04, 2.5977e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0117e-03, 1.3189e-03, 2.0801e-01,  ..., 6.7635e-03,
           1.2863e-02, 4.1779e-02]],

         [[1.0357e-03, 1.1921e-03, 2.9077e-01,  ..., 1.0748e-03,
           6.3858e-03, 1.9257e-02]],

         [[4.8676e-03, 2.1591e-03, 2.0190e-01,  ..., 1.9455e-02,
           4.5410e-02, 2.7481e-02]],

         ...,

         [[3.1319e-03, 3.3808e-04, 2.2839e-01,  ..., 1.0090e-03,
           4.9667e-03, 1.2779e-02]],

         [[1.5345e-03, 3.1605e-03, 1.1816e-01,  ..., 5.2856e-02,
           4.0833e-02, 3.7549e-01]],

         [[5.9748e-04, 8.9109e-05, 2.9694e-02,  ..., 8.5266e-02,
           1.2402e-01, 2.7954e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.1231e-04, 2.7275e-04, 6.9618e-04,  ..., 1.7041e-01,
           4.7778e-01, 2.7222e-01]],

         [[4.0436e-03, 7.9632e-04, 1.4122e-02,  ..., 2.0752e-01,
           2.9892e-02, 2.4780e-02]],

         [[5.9605e-08, 5.9605e-08, 7.4506e-05,  ..., 5.5027e-04,
           3.3379e-04, 9.9512e-01]],

         ...,

         [[2.8610e-06, 9.9540e-06, 2.0027e-04,  ..., 2.0027e-04,
           6.6338e-03, 9.8438e-01]],

         [[4.9629e-03, 6.3744e-03, 1.9638e-02,  ..., 2.0905e-02,
           4.3793e-03, 1.2024e-01]],

         [[9.9869e-03, 4.4327e-03, 5.9986e-04,  ..., 5.7465e-02,
           3.2745e-02, 1.6464e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3341e-04, 1.6203e-03, 4.7607e-03,  ..., 1.2280e-01,
           1.2091e-01, 1.4807e-01]],

         [[1.1921e-07, 2.3842e-07, 1.1921e-07,  ..., 1.5063e-01,
           7.4170e-01, 1.0193e-01]],

         [[2.3174e-03, 3.0231e-03, 1.1772e-02,  ..., 1.5114e-02,
           1.8799e-02, 1.9409e-02]],

         ...,

         [[5.4777e-05, 8.2850e-05, 1.3905e-03,  ..., 6.7017e-02,
           3.1226e-01, 1.3013e-01]],

         [[3.6449e-03, 6.8626e-03, 2.6306e-02,  ..., 1.2527e-02,
           1.3702e-02, 4.9744e-02]],

         [[1.4496e-04, 3.3975e-04, 1.5230e-03,  ..., 2.6367e-02,
           7.2205e-02, 1.0266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2077e-02, 1.4648e-03, 2.6321e-03,  ..., 2.3328e-01,
           1.4380e-01, 6.9519e-02]],

         [[3.6907e-03, 9.6655e-04, 3.5362e-03,  ..., 2.0237e-03,
           8.8577e-03, 2.0111e-02]],

         [[6.3362e-03, 1.5354e-03, 1.0910e-03,  ..., 3.3081e-02,
           2.5955e-02, 4.5044e-02]],

         ...,

         [[1.1808e-04, 1.3151e-03, 4.5166e-03,  ..., 9.5825e-02,
           5.3027e-01, 2.6660e-01]],

         [[2.3193e-03, 2.6011e-04, 4.2081e-04,  ..., 4.1443e-02,
           1.1896e-01, 1.3477e-01]],

         [[4.6295e-02, 7.8583e-03, 1.8707e-02,  ..., 3.9276e-02,
           2.8290e-02, 8.1848e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2920e-01, 1.6689e-06, 1.7881e-07,  ..., 5.6229e-03,
           3.6449e-03, 2.6360e-03]],

         [[8.8477e-01, 5.0068e-06, 1.1861e-05,  ..., 2.7390e-02,
           4.6448e-02, 2.2411e-03]],

         [[8.9062e-01, 7.7486e-07, 1.6689e-06,  ..., 4.5662e-03,
           1.2217e-03, 4.2023e-02]],

         ...,

         [[5.2979e-01, 2.1887e-04, 1.1368e-03,  ..., 2.6226e-03,
           1.0853e-03, 3.6072e-02]],

         [[5.5811e-01, 2.4819e-04, 5.1165e-04,  ..., 1.7593e-02,
           1.6663e-02, 1.2718e-02]],

         [[7.0654e-01, 3.8743e-04, 4.8733e-04,  ..., 5.7449e-03,
           5.2109e-03, 2.3438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0273e-01, 2.2113e-05, 8.7619e-06,  ..., 2.7027e-03,
           3.2091e-04, 2.7802e-02]],

         [[9.4922e-01, 7.4506e-06, 4.1723e-07,  ..., 5.5237e-03,
           3.0090e-02, 1.4740e-02]],

         [[9.5117e-01, 2.7251e-04, 2.9624e-05,  ..., 2.8610e-03,
           3.2291e-03, 8.4457e-03]],

         ...,

         [[4.5239e-01, 9.4593e-05, 1.2934e-04,  ..., 4.9248e-03,
           3.6438e-02, 2.4460e-02]],

         [[1.4270e-01, 4.0054e-03, 1.6546e-03,  ..., 1.6270e-03,
           4.7264e-03, 1.2581e-02]],

         [[6.5820e-01, 6.8784e-05, 2.1148e-04,  ..., 8.2092e-02,
           5.9937e-02, 1.6113e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8296e-01, 2.7776e-04, 1.5342e-04,  ..., 2.1118e-02,
           3.1219e-02, 2.8503e-02]],

         [[6.7090e-01, 1.1086e-05, 7.3910e-06,  ..., 7.8087e-03,
           2.4376e-03, 2.2110e-02]],

         [[2.8125e-01, 1.1146e-05, 5.6624e-06,  ..., 5.3894e-02,
           1.9226e-02, 1.7807e-02]],

         ...,

         [[6.2402e-01, 2.0325e-05, 1.9073e-06,  ..., 5.6803e-05,
           1.0443e-04, 3.7183e-01]],

         [[9.5166e-01, 1.3280e-04, 5.3227e-05,  ..., 2.9802e-07,
           1.1921e-07, 1.4023e-02]],

         [[9.9268e-01, 1.8477e-05, 2.3246e-06,  ..., 0.0000e+00,
           0.0000e+00, 3.9101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0063e-01, 1.2755e-05, 6.0201e-06,  ..., 5.2368e-02,
           2.2217e-01, 1.6006e-02]],

         [[9.5654e-01, 1.1742e-05, 5.4240e-06,  ..., 1.2696e-05,
           2.8658e-04, 1.0208e-02]],

         [[8.0420e-01, 3.3832e-04, 6.7413e-05,  ..., 2.2144e-03,
           3.6736e-03, 6.1707e-02]],

         ...,

         [[7.2217e-01, 1.9097e-04, 3.2663e-05,  ..., 7.9498e-03,
           1.2100e-02, 2.4292e-02]],

         [[4.2725e-01, 9.4891e-05, 1.6570e-05,  ..., 7.1831e-03,
           7.9651e-02, 2.4323e-02]],

         [[8.9844e-01, 1.1200e-04, 8.7202e-05,  ..., 4.0169e-03,
           9.3994e-03, 1.1902e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0625e-01, 5.2452e-06, 6.2585e-06,  ..., 6.1127e-02,
           2.9834e-01, 9.0515e-02]],

         [[7.3877e-01, 1.1396e-04, 5.1975e-05,  ..., 2.2449e-03,
           3.4981e-03, 2.9037e-02]],

         [[4.2627e-01, 1.2577e-05, 4.7088e-06,  ..., 1.2917e-02,
           7.5012e-02, 1.7532e-02]],

         ...,

         [[7.0361e-01, 1.4913e-04, 5.1558e-05,  ..., 1.7776e-02,
           2.6077e-02, 2.6779e-02]],

         [[8.6279e-01, 3.6860e-04, 1.0562e-04,  ..., 1.1854e-03,
           1.7593e-02, 3.9635e-03]],

         [[3.9307e-01, 1.3590e-05, 2.2411e-05,  ..., 5.8899e-02,
           1.0120e-01, 3.5706e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3232e-01, 2.2793e-03, 1.8320e-03,  ..., 1.7583e-04,
           3.6144e-03, 1.9817e-03]],

         [[9.5020e-01, 3.8803e-05, 6.7651e-05,  ..., 1.6003e-03,
           1.1530e-03, 3.8090e-03]],

         [[8.8232e-01, 7.0870e-05, 4.8339e-05,  ..., 1.8635e-03,
           1.9958e-02, 1.8415e-03]],

         ...,

         [[8.3447e-01, 2.8205e-04, 3.0375e-04,  ..., 1.6718e-03,
           1.0967e-03, 6.2988e-02]],

         [[9.3164e-01, 4.0531e-06, 7.5698e-06,  ..., 3.0060e-03,
           5.6076e-03, 3.0869e-02]],

         [[8.5742e-01, 4.0293e-05, 3.4213e-05,  ..., 4.9057e-03,
           6.5674e-02, 3.8185e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4158e-01, 7.4387e-04, 8.2684e-04,  ..., 8.0442e-04,
           1.7471e-02, 7.7057e-03]],

         [[8.1299e-01, 8.2076e-05, 1.0455e-04,  ..., 2.8591e-03,
           2.6978e-02, 6.5956e-03]],

         [[3.0347e-01, 1.8299e-05, 4.2200e-05,  ..., 4.6844e-02,
           3.5010e-01, 8.0338e-03]],

         ...,

         [[1.9543e-01, 6.2323e-04, 1.3161e-03,  ..., 1.7428e-04,
           2.3327e-03, 7.1754e-03]],

         [[6.0791e-01, 8.8751e-05, 5.7876e-05,  ..., 1.1581e-02,
           2.7515e-01, 5.4893e-03]],

         [[8.9209e-01, 4.0531e-06, 8.6427e-06,  ..., 3.1929e-03,
           3.4199e-03, 4.1718e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.9426e-05, 2.2161e-04,  ..., 2.9254e-04,
           1.0052e-03, 1.7365e-02]],

         [[4.4604e-01, 5.3227e-05, 1.0425e-04,  ..., 1.5167e-02,
           1.1066e-01, 2.1179e-02]],

         [[6.9824e-01, 1.0550e-05, 3.6538e-05,  ..., 9.0265e-04,
           2.7580e-03, 2.4979e-02]],

         ...,

         [[9.5166e-01, 4.7743e-05, 2.5332e-05,  ..., 6.4135e-04,
           2.5463e-03, 3.0861e-03]],

         [[7.4170e-01, 4.2176e-04, 1.3018e-04,  ..., 4.1504e-02,
           6.9351e-03, 2.4780e-02]],

         [[2.3315e-01, 1.5812e-03, 5.0116e-04,  ..., 1.9360e-04,
           3.3913e-03, 1.6815e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9453e-01, 3.8147e-06, 6.7353e-06,  ..., 4.3011e-04,
           9.4986e-03, 5.3955e-01]],

         [[6.1670e-01, 8.0633e-04, 1.4725e-03,  ..., 3.7050e-04,
           1.3412e-02, 1.0862e-03]],

         [[7.6074e-01, 4.6921e-04, 5.1641e-04,  ..., 1.1520e-03,
           1.5068e-02, 1.9608e-02]],

         ...,

         [[5.6885e-01, 1.2589e-04, 2.4748e-04,  ..., 2.4700e-03,
           1.2375e-02, 8.2932e-03]],

         [[8.0994e-02, 2.1040e-05, 1.2755e-05,  ..., 3.8719e-04,
           7.5226e-03, 4.0649e-02]],

         [[1.4270e-01, 2.6464e-05, 7.7844e-05,  ..., 8.5602e-03,
           9.0561e-03, 7.1436e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7810e-01, 1.1140e-04, 1.7059e-04,  ..., 7.6332e-03,
           5.4741e-03, 5.8984e-01]],

         [[5.3369e-01, 2.1195e-04, 3.6073e-04,  ..., 4.8327e-04,
           3.2806e-02, 8.9951e-03]],

         [[6.7529e-01, 1.3280e-04, 1.5402e-04,  ..., 2.0564e-04,
           2.6627e-03, 9.5901e-03]],

         ...,

         [[2.5162e-02, 2.9206e-06, 1.0967e-05,  ..., 7.3671e-04,
           3.7212e-03, 9.5898e-01]],

         [[7.6807e-01, 1.1253e-04, 1.4508e-04,  ..., 4.3907e-03,
           2.2163e-03, 9.4833e-03]],

         [[7.6855e-01, 1.4806e-04, 7.7069e-05,  ..., 7.9060e-04,
           1.9867e-02, 7.3586e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1919e-01, 5.2071e-04, 7.6914e-04,  ..., 3.9597e-03,
           2.6367e-01, 1.0849e-02]],

         [[7.2070e-01, 7.6234e-05, 1.5223e-04,  ..., 2.3346e-03,
           4.9377e-02, 2.9724e-02]],

         [[3.7671e-01, 7.2837e-05, 1.8597e-04,  ..., 2.8920e-04,
           9.2459e-04, 1.2733e-02]],

         ...,

         [[7.8271e-01, 8.9407e-07, 1.9670e-06,  ..., 5.7831e-03,
           1.6736e-01, 3.1342e-02]],

         [[5.0195e-01, 9.1887e-04, 9.4414e-04,  ..., 5.6601e-04,
           1.4656e-02, 1.3916e-02]],

         [[5.2490e-01, 4.7684e-07, 1.0729e-06,  ..., 1.8295e-02,
           3.4692e-01, 6.1951e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0180e-04, 2.3901e-05,  ..., 7.2050e-04,
           3.4218e-03, 5.2147e-03]],

         [[1.5161e-01, 6.0701e-04, 6.1512e-04,  ..., 3.3474e-04,
           5.0011e-03, 1.3130e-02]],

         [[4.5557e-01, 8.5735e-04, 8.5735e-04,  ..., 1.8466e-04,
           2.4063e-02, 1.4534e-02]],

         ...,

         [[5.0098e-01, 2.9802e-06, 3.8743e-06,  ..., 1.7490e-03,
           4.0955e-02, 5.0049e-02]],

         [[2.9858e-01, 1.4305e-06, 1.2517e-06,  ..., 4.7798e-03,
           5.7068e-02, 2.8564e-02]],

         [[1.7041e-01, 3.0041e-05, 5.2512e-05,  ..., 3.7766e-04,
           1.3268e-02, 1.1101e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0171e-02, 8.3447e-07, 1.0729e-06,  ..., 1.3065e-04,
           6.6338e-03, 8.7207e-01]],

         [[3.0615e-01, 6.7353e-06, 1.9312e-05,  ..., 1.1978e-02,
           7.1777e-02, 6.5857e-02]],

         [[4.4312e-01, 1.0128e-03, 1.0872e-03,  ..., 4.3273e-04,
           9.3307e-03, 1.1795e-02]],

         ...,

         [[2.1509e-01, 4.3452e-05, 5.5313e-05,  ..., 3.9330e-03,
           8.4045e-02, 2.9602e-02]],

         [[5.8594e-01, 5.7220e-06, 1.5616e-05,  ..., 1.4782e-03,
           3.6907e-03, 1.4816e-02]],

         [[8.9722e-02, 7.8201e-05, 7.4625e-05,  ..., 7.9679e-04,
           1.2428e-02, 8.3008e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0928e-01, 4.6194e-05, 7.8380e-05,  ..., 6.5880e-03,
           1.7603e-01, 8.2321e-03]],

         [[2.2754e-01, 4.1771e-04, 4.1032e-04,  ..., 4.6692e-03,
           5.5115e-02, 4.3427e-02]],

         [[7.3181e-02, 7.7248e-05, 2.4319e-05,  ..., 5.8985e-04,
           3.4546e-02, 4.7607e-02]],

         ...,

         [[6.7627e-01, 7.8857e-05, 2.3901e-05,  ..., 1.8377e-03,
           5.8411e-02, 2.5513e-02]],

         [[3.1201e-01, 1.0449e-04, 1.3804e-04,  ..., 3.3450e-04,
           6.1920e-02, 4.4067e-02]],

         [[6.8506e-01, 2.4092e-04, 1.1784e-04,  ..., 6.5470e-04,
           2.8717e-02, 2.2812e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9854e-02, 1.4505e-03, 1.3819e-03,  ..., 6.9916e-05,
           3.7956e-03, 3.0975e-03]],

         [[1.7029e-01, 4.9925e-04, 3.0589e-04,  ..., 1.6460e-03,
           4.1901e-02, 1.2291e-02]],

         [[5.2051e-01, 2.8849e-05, 5.4836e-05,  ..., 1.0939e-03,
           1.0406e-01, 1.9775e-02]],

         ...,

         [[5.0000e-01, 3.1114e-04, 1.4615e-04,  ..., 1.9894e-03,
           9.8267e-03, 3.8422e-02]],

         [[9.6143e-01, 8.8871e-05, 1.9515e-04,  ..., 1.0252e-05,
           1.9372e-05, 3.8395e-03]],

         [[9.1357e-01, 1.1140e-04, 1.5104e-04,  ..., 5.0259e-04,
           1.6308e-03, 3.8414e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9048e-01, 5.7888e-04, 7.5483e-04,  ..., 3.5667e-04,
           1.0185e-02, 2.5192e-02]],

         [[2.0630e-01, 2.9888e-03, 1.3847e-03,  ..., 7.7677e-04,
           1.7334e-02, 1.6937e-02]],

         [[3.1204e-02, 7.2122e-06, 8.0466e-06,  ..., 6.4392e-03,
           2.0309e-02, 6.1719e-01]],

         ...,

         [[3.0566e-01, 2.5630e-05, 2.9325e-05,  ..., 5.2357e-04,
           3.3875e-02, 1.3062e-02]],

         [[5.1562e-01, 1.1504e-05, 8.4639e-06,  ..., 2.4834e-03,
           2.3242e-01, 5.7297e-03]],

         [[1.4270e-01, 2.1636e-05, 1.7762e-05,  ..., 3.2024e-03,
           7.4585e-02, 3.7012e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8135e-01, 4.6074e-05, 2.3901e-05,  ..., 1.4553e-03,
           1.0481e-03, 1.0925e-02]],

         [[6.4331e-02, 4.3631e-05, 2.6464e-05,  ..., 1.5612e-03,
           8.5266e-02, 2.2583e-02]],

         [[1.7957e-01, 4.0770e-05, 2.3246e-05,  ..., 4.0507e-04,
           1.0124e-02, 7.4414e-01]],

         ...,

         [[1.2927e-01, 4.8828e-04, 2.2447e-04,  ..., 6.5231e-04,
           3.4981e-03, 1.0941e-02]],

         [[3.5010e-01, 8.6784e-04, 5.3883e-04,  ..., 6.8817e-03,
           5.1537e-03, 7.3242e-03]],

         [[3.6841e-01, 1.1015e-03, 1.0843e-03,  ..., 1.1169e-02,
           9.8190e-03, 4.6921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4585e-01, 2.2507e-04, 9.9480e-05,  ..., 5.2643e-03,
           6.8848e-02, 3.9825e-02]],

         [[9.2578e-01, 7.8499e-05, 1.1444e-05,  ..., 2.1505e-04,
           2.2411e-03, 9.2163e-03]],

         [[8.9600e-01, 3.9876e-05, 9.6560e-06,  ..., 8.2970e-04,
           7.8125e-03, 2.6779e-03]],

         ...,

         [[8.0750e-02, 2.0218e-03, 3.4389e-03,  ..., 2.9802e-04,
           4.9248e-03, 4.8866e-03]],

         [[5.1465e-01, 3.0935e-05, 3.5048e-05,  ..., 1.8997e-03,
           5.1758e-02, 6.2790e-03]],

         [[2.0142e-01, 5.7030e-04, 1.7157e-03,  ..., 1.9217e-03,
           1.6403e-02, 3.0884e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0820e-01, 2.1040e-04, 2.2221e-04,  ..., 7.8354e-03,
           4.9400e-03, 1.2917e-02]],

         [[3.3228e-01, 2.8920e-04, 6.9904e-04,  ..., 3.1586e-03,
           1.3718e-02, 3.1891e-02]],

         [[2.9028e-01, 7.9215e-05, 1.9073e-04,  ..., 1.1986e-02,
           3.5797e-02, 1.1345e-02]],

         ...,

         [[1.3245e-02, 5.9485e-05, 9.5010e-05,  ..., 1.9388e-03,
           1.5244e-02, 2.2888e-02]],

         [[1.3245e-01, 1.5998e-04, 1.0008e-04,  ..., 6.2275e-04,
           1.9178e-03, 2.3880e-03]],

         [[6.4893e-01, 2.4104e-04, 3.9124e-04,  ..., 2.7580e-03,
           4.5815e-03, 2.6951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1542e-01, 1.7881e-05, 1.7107e-05,  ..., 2.0981e-03,
           1.6251e-02, 9.1858e-03]],

         [[2.6440e-01, 1.0127e-04, 1.8120e-04,  ..., 5.5084e-03,
           1.8127e-02, 6.3896e-03]],

         [[2.0337e-01, 4.7088e-05, 1.5676e-04,  ..., 6.9046e-03,
           1.8051e-02, 6.2158e-01]],

         ...,

         [[7.0361e-01, 3.5465e-05, 6.5267e-05,  ..., 4.1351e-03,
           3.9703e-02, 2.7924e-02]],

         [[3.9160e-01, 2.5868e-05, 1.3876e-04,  ..., 3.7193e-03,
           7.6332e-03, 2.5620e-02]],

         [[4.8657e-01, 8.4341e-05, 9.3699e-05,  ..., 5.3930e-04,
           1.9121e-03, 1.2962e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3684e-01, 1.6236e-04, 8.1360e-05,  ..., 3.2425e-03,
           2.3224e-02, 1.5228e-02]],

         [[6.1340e-02, 2.4915e-05, 2.6405e-05,  ..., 8.4991e-03,
           1.3391e-01, 1.6373e-02]],

         [[4.9756e-01, 1.5318e-04, 1.3280e-04,  ..., 1.5955e-03,
           5.4207e-03, 1.3893e-02]],

         ...,

         [[2.6636e-01, 4.4703e-06, 2.9266e-05,  ..., 1.7303e-02,
           6.3916e-01, 2.8290e-02]],

         [[4.0601e-01, 3.3498e-05, 7.9751e-05,  ..., 3.9787e-03,
           3.5187e-02, 4.3457e-02]],

         [[4.9951e-01, 2.2352e-05, 7.5817e-05,  ..., 3.5267e-03,
           3.3276e-01, 7.3059e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6865e-02, 1.1683e-05, 5.7042e-05,  ..., 2.0542e-03,
           3.8025e-02, 2.7817e-02]],

         [[1.1864e-02, 4.7684e-06, 9.8944e-06,  ..., 2.6631e-04,
           6.3477e-03, 9.6484e-01]],

         [[2.6520e-02, 2.2829e-05, 1.1903e-04,  ..., 1.6046e-04,
           1.2407e-03, 2.7618e-03]],

         ...,

         [[9.7595e-02, 2.5415e-04, 9.0599e-04,  ..., 6.9332e-04,
           1.5869e-02, 1.1566e-02]],

         [[4.4727e-01, 3.9935e-06, 9.7156e-06,  ..., 8.5020e-04,
           1.3123e-01, 4.0802e-02]],

         [[2.5488e-01, 7.7486e-07, 3.6955e-06,  ..., 2.0432e-02,
           3.4961e-01, 1.4294e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9395e-01, 2.3758e-04, 1.1396e-03,  ..., 5.2452e-04,
           8.0032e-03, 6.7635e-03]],

         [[9.8486e-01, 3.1948e-05, 1.9491e-04,  ..., 1.8716e-05,
           3.3689e-04, 1.0376e-03]],

         [[8.1934e-01, 1.8883e-04, 2.1994e-04,  ..., 3.1815e-03,
           1.1871e-02, 6.1569e-03]],

         ...,

         [[6.5332e-01, 6.1333e-05, 6.8426e-05,  ..., 1.8814e-02,
           5.7983e-02, 1.1063e-02]],

         [[4.8413e-01, 1.2517e-06, 2.9802e-07,  ..., 9.4175e-04,
           1.4820e-03, 4.8981e-03]],

         [[1.9873e-01, 2.5272e-05, 5.7459e-05,  ..., 2.3997e-04,
           4.5044e-02, 6.6162e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2074e-02, 9.0003e-06, 1.2636e-05,  ..., 3.9368e-03,
           1.7334e-01, 8.8501e-02]],

         [[6.1914e-01, 1.1981e-04, 6.1274e-05,  ..., 1.2848e-02,
           7.0557e-02, 2.8503e-02]],

         [[1.8530e-01, 5.5122e-04, 3.5477e-04,  ..., 2.3743e-02,
           6.0638e-02, 1.8280e-02]],

         ...,

         [[4.3701e-01, 2.5630e-05, 1.0747e-04,  ..., 3.0212e-02,
           2.0972e-01, 1.6953e-02]],

         [[6.4551e-01, 2.3007e-05, 7.4208e-05,  ..., 5.6915e-03,
           2.9251e-02, 6.2408e-02]],

         [[3.9014e-01, 1.8835e-05, 6.1929e-05,  ..., 5.9204e-03,
           2.6749e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7863e-03, 1.9670e-06, 1.6093e-06,  ..., 5.3167e-04,
           2.5768e-03, 4.4861e-03]],

         [[8.2520e-02, 8.4639e-06, 4.4107e-06,  ..., 6.6566e-04,
           5.2376e-03, 5.8861e-03]],

         [[1.8958e-01, 9.5308e-05, 3.0708e-04,  ..., 1.7227e-02,
           1.2537e-01, 3.4821e-02]],

         ...,

         [[8.3374e-02, 1.1683e-05, 1.4198e-04,  ..., 1.7996e-03,
           1.0277e-02, 6.4941e-02]],

         [[2.0288e-01, 3.9995e-05, 3.4571e-04,  ..., 1.5488e-03,
           1.7456e-02, 5.4626e-02]],

         [[7.9199e-01, 2.4915e-05, 7.7367e-05,  ..., 4.8599e-03,
           3.2196e-02, 3.2196e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4316e-01, 1.3769e-05, 2.9385e-05,  ..., 1.0681e-02,
           3.2074e-02, 4.1840e-02]],

         [[3.2617e-01, 3.7551e-05, 1.5211e-04,  ..., 1.8509e-02,
           1.7468e-01, 5.4321e-02]],

         [[1.5942e-01, 2.8610e-06, 1.1146e-05,  ..., 2.7370e-03,
           6.6357e-01, 1.0956e-01]],

         ...,

         [[4.5703e-01, 7.6950e-05, 1.4067e-04,  ..., 5.7755e-03,
           1.1436e-02, 1.0132e-02]],

         [[2.9590e-01, 2.5809e-05, 1.5795e-05,  ..., 1.1444e-03,
           5.2216e-02, 4.3304e-02]],

         [[5.9021e-02, 1.0747e-04, 4.1842e-04,  ..., 1.5884e-02,
           6.4819e-02, 5.9619e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4546e-01, 7.5340e-05, 1.7405e-04,  ..., 1.7136e-02,
           1.0870e-01, 1.4259e-02]],

         [[3.3496e-01, 3.9530e-04, 4.8256e-04,  ..., 4.1847e-03,
           6.1836e-03, 3.1109e-03]],

         [[2.0679e-01, 8.7023e-05, 1.8466e-04,  ..., 3.2654e-03,
           4.6616e-03, 9.4910e-03]],

         ...,

         [[7.0996e-01, 4.5896e-04, 4.6635e-04,  ..., 6.5880e-03,
           2.0782e-02, 3.9429e-02]],

         [[1.4392e-01, 1.6844e-04, 4.9496e-04,  ..., 3.0327e-03,
           2.2064e-02, 7.5391e-01]],

         [[8.8184e-01, 2.4259e-05, 4.9829e-05,  ..., 1.3390e-02,
           6.1302e-03, 2.4628e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4766e-01, 1.0097e-04, 1.1864e-03,  ..., 1.7792e-02,
           1.3863e-02, 7.3700e-03]],

         [[4.8645e-02, 4.6492e-06, 2.7835e-05,  ..., 9.5797e-04,
           5.5786e-02, 1.8463e-02]],

         [[1.5686e-01, 8.0884e-05, 3.6645e-04,  ..., 2.8122e-02,
           7.1838e-02, 1.3904e-01]],

         ...,

         [[6.9922e-01, 1.6582e-04, 1.0080e-03,  ..., 4.5166e-03,
           3.2959e-02, 4.0863e-02]],

         [[8.0518e-01, 1.7881e-06, 3.2544e-04,  ..., 2.5272e-03,
           6.6261e-03, 2.7409e-03]],

         [[2.7588e-01, 1.1235e-04, 1.0341e-04,  ..., 4.3121e-02,
           1.2189e-01, 4.4312e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8394e-05, 1.2207e-04, 1.5515e-01,  ..., 5.0850e-03,
           7.7200e-04, 4.1748e-02]],

         [[2.9862e-05, 2.2411e-05, 4.1595e-02,  ..., 6.4945e-04,
           1.4019e-03, 8.3542e-03]],

         [[5.8222e-04, 8.4305e-04, 8.8013e-02,  ..., 3.3417e-03,
           1.1147e-02, 1.8738e-02]],

         ...,

         [[4.6206e-04, 1.6224e-04, 2.1545e-01,  ..., 1.0315e-02,
           4.3762e-02, 1.4465e-01]],

         [[2.7132e-04, 2.9278e-04, 3.2861e-01,  ..., 9.1934e-04,
           5.3749e-03, 8.3923e-03]],

         [[1.4782e-05, 1.5616e-05, 1.4355e-01,  ..., 6.6996e-04,
           5.4359e-03, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0969e-04, 1.3959e-04, 2.8052e-01,  ..., 1.1873e-03,
           2.7618e-03, 1.5343e-02]],

         [[1.3328e-04, 1.3196e-04, 1.2646e-01,  ..., 1.8616e-03,
           1.2302e-03, 1.0139e-02]],

         [[9.5546e-05, 2.7180e-05, 2.8418e-01,  ..., 1.7920e-03,
           4.0245e-03, 2.2263e-02]],

         ...,

         [[4.1771e-04, 6.7532e-05, 3.7018e-02,  ..., 2.2447e-04,
           6.0797e-04, 1.6136e-03]],

         [[1.9312e-05, 2.2113e-05, 2.2534e-01,  ..., 3.0670e-03,
           2.0957e-04, 8.9798e-03]],

         [[1.0610e-04, 1.3387e-04, 2.1741e-01,  ..., 2.0256e-03,
           5.2643e-04, 1.6769e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0637e-05, 3.2425e-05, 8.7585e-02,  ..., 8.1825e-04,
           8.0776e-04, 3.0575e-03]],

         [[2.5201e-04, 1.1802e-05, 2.0142e-01,  ..., 1.8396e-03,
           8.5711e-05, 1.1360e-02]],

         [[1.0622e-04, 1.2219e-04, 1.9727e-01,  ..., 9.7656e-04,
           1.0729e-03, 7.4158e-03]],

         ...,

         [[1.5414e-04, 7.8619e-05, 1.3635e-01,  ..., 5.5618e-03,
           3.8528e-03, 2.3499e-02]],

         [[1.9073e-05, 4.5121e-05, 7.1045e-02,  ..., 3.8767e-04,
           5.0449e-04, 4.5242e-03]],

         [[6.9559e-05, 2.9445e-05, 1.1188e-01,  ..., 7.3910e-04,
           6.5994e-04, 4.8409e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9605e-05, 5.0068e-06, 1.4328e-02,  ..., 2.3479e-03,
           1.1011e-01, 2.0093e-01]],

         [[3.0518e-05, 8.3923e-05, 1.5942e-01,  ..., 5.7030e-03,
           7.5256e-02, 3.5828e-02]],

         [[1.7166e-04, 5.4932e-04, 2.8955e-01,  ..., 6.0177e-04,
           8.3733e-04, 5.8699e-04]],

         ...,

         [[3.4142e-04, 4.9293e-05, 6.7871e-02,  ..., 4.0352e-05,
           2.4338e-03, 9.0714e-03]],

         [[4.0665e-03, 2.0480e-04, 1.2030e-01,  ..., 2.6684e-03,
           4.6204e-02, 2.5171e-01]],

         [[7.2122e-06, 1.3113e-06, 1.5020e-03,  ..., 1.6463e-04,
           8.5327e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8321e-03, 1.4305e-03, 1.2891e-01,  ..., 1.4130e-02,
           1.7578e-02, 7.2876e-02]],

         [[2.3098e-03, 1.4591e-03, 2.4158e-01,  ..., 7.2098e-03,
           1.3161e-02, 2.7008e-02]],

         [[8.6288e-03, 1.1053e-03, 2.2351e-01,  ..., 5.2223e-03,
           7.4806e-03, 1.1208e-02]],

         ...,

         [[1.6441e-03, 9.9123e-05, 2.3840e-01,  ..., 1.5812e-03,
           6.3515e-03, 1.4453e-01]],

         [[2.3174e-03, 2.9526e-03, 1.9751e-01,  ..., 4.2389e-02,
           5.7373e-03, 5.5298e-02]],

         [[1.5950e-04, 7.7426e-05, 3.7231e-02,  ..., 7.5645e-03,
           2.5439e-01, 3.8501e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7001e-05, 6.4790e-05, 1.2608e-03,  ..., 8.8379e-02,
           4.7803e-01, 3.2837e-01]],

         [[2.2125e-03, 1.1835e-03, 1.7395e-02,  ..., 3.8815e-03,
           6.8787e-02, 4.2139e-01]],

         [[9.5367e-07, 1.1921e-06, 2.2087e-03,  ..., 8.5592e-05,
           5.6946e-02, 8.9062e-01]],

         ...,

         [[1.7881e-07, 8.3447e-07, 7.4267e-05,  ..., 1.2243e-04,
           2.4586e-03, 9.9170e-01]],

         [[2.7199e-03, 2.3994e-03, 1.4702e-02,  ..., 6.1302e-03,
           2.4246e-02, 2.0300e-01]],

         [[1.1749e-02, 4.3221e-03, 9.6464e-04,  ..., 3.1952e-02,
           2.6489e-02, 2.8198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0118e-03, 4.6082e-03, 1.6327e-02,  ..., 7.3181e-02,
           1.0651e-01, 1.4551e-01]],

         [[1.7881e-07, 3.5763e-07, 4.1723e-07,  ..., 1.7346e-01,
           7.4219e-01, 7.8186e-02]],

         [[2.5635e-03, 4.7150e-03, 1.7517e-02,  ..., 2.5879e-02,
           4.2664e-02, 1.9852e-02]],

         ...,

         [[1.2808e-03, 6.8521e-04, 6.8130e-03,  ..., 7.7393e-02,
           1.2634e-02, 7.3242e-02]],

         [[4.8714e-03, 1.4091e-02, 4.4617e-02,  ..., 1.2581e-02,
           3.6560e-02, 1.0933e-02]],

         [[1.8797e-03, 1.2070e-02, 2.8732e-02,  ..., 5.4932e-02,
           6.7810e-02, 3.0106e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2490e-02, 6.6757e-03, 1.5396e-02,  ..., 6.6895e-02,
           4.5959e-02, 4.3518e-02]],

         [[3.8414e-03, 8.4352e-04, 3.3245e-03,  ..., 4.8370e-03,
           8.6899e-03, 3.0270e-03]],

         [[4.9629e-03, 1.3208e-03, 1.2264e-03,  ..., 2.0264e-02,
           3.7994e-02, 2.2949e-02]],

         ...,

         [[2.9812e-03, 1.4639e-03, 2.3575e-03,  ..., 3.5254e-01,
           1.0754e-01, 1.5405e-01]],

         [[3.0689e-03, 5.2929e-04, 3.3236e-04,  ..., 2.8711e-01,
           1.2952e-01, 5.2307e-02]],

         [[2.6566e-02, 6.7711e-03, 1.7975e-02,  ..., 2.6566e-02,
           3.8135e-01, 9.4910e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6035e-01, 1.7285e-06, 1.1921e-06,  ..., 2.1915e-03,
           2.1458e-03, 1.0811e-02]],

         [[9.3359e-01, 4.8876e-06, 1.1086e-05,  ..., 8.8882e-03,
           1.2314e-02, 4.8208e-04]],

         [[8.8232e-01, 1.1921e-06, 4.7684e-06,  ..., 6.9504e-03,
           1.8280e-02, 1.9165e-02]],

         ...,

         [[6.3379e-01, 8.7798e-05, 8.4114e-04,  ..., 5.1618e-05,
           2.6684e-03, 2.4200e-02]],

         [[5.0342e-01, 1.3387e-04, 3.1519e-04,  ..., 2.8061e-02,
           2.4002e-02, 1.0948e-02]],

         [[6.2158e-01, 3.1114e-04, 4.3416e-04,  ..., 2.0050e-02,
           4.9622e-02, 2.2278e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6182e-01, 4.4703e-06, 1.7881e-06,  ..., 7.7200e-04,
           2.3499e-03, 3.6652e-02]],

         [[1.0907e-01, 5.3644e-07, 1.7881e-07,  ..., 2.5597e-03,
           8.7842e-01, 9.5596e-03]],

         [[9.4824e-01, 1.0818e-04, 5.7936e-05,  ..., 1.2007e-03,
           1.5656e-02, 4.8294e-03]],

         ...,

         [[4.4360e-01, 2.5415e-04, 1.2094e-04,  ..., 5.8624e-02,
           5.4016e-02, 3.1311e-02]],

         [[1.3232e-01, 9.4175e-04, 6.2084e-04,  ..., 5.9929e-03,
           1.7319e-02, 7.3738e-03]],

         [[4.4824e-01, 8.7202e-05, 1.4317e-04,  ..., 1.1877e-01,
           6.4819e-02, 1.2711e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1348e-01, 3.6001e-04, 1.5426e-04,  ..., 4.5898e-02,
           5.9296e-02, 3.0640e-02]],

         [[6.4502e-01, 1.9372e-05, 1.2517e-05,  ..., 1.2398e-02,
           1.2891e-01, 1.8326e-02]],

         [[5.4980e-01, 1.7405e-05, 4.3511e-06,  ..., 1.7868e-02,
           6.3721e-02, 1.2474e-02]],

         ...,

         [[6.1670e-01, 1.7881e-06, 4.7684e-07,  ..., 6.2585e-06,
           5.4779e-03, 3.7402e-01]],

         [[9.2383e-01, 1.9789e-05, 2.9981e-05,  ..., 0.0000e+00,
           1.3483e-04, 4.3243e-02]],

         [[9.8389e-01, 2.3842e-06, 7.1526e-07,  ..., 0.0000e+00,
           2.7180e-05, 9.4223e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 1.9431e-05, 2.4438e-06,  ..., 1.2018e-01,
           1.3232e-01, 1.0727e-02]],

         [[9.5947e-01, 1.0967e-05, 2.6822e-06,  ..., 3.1614e-04,
           1.5564e-03, 1.1902e-02]],

         [[7.7197e-01, 5.7459e-04, 6.1989e-05,  ..., 9.2888e-04,
           2.3941e-02, 5.9906e-02]],

         ...,

         [[8.0273e-01, 4.7505e-05, 1.1981e-05,  ..., 6.9962e-03,
           1.8890e-02, 1.8280e-02]],

         [[7.9932e-01, 1.5581e-04, 4.9233e-05,  ..., 3.6865e-02,
           2.7878e-02, 3.1647e-02]],

         [[9.2285e-01, 6.2644e-05, 4.0948e-05,  ..., 6.5727e-03,
           9.7809e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 6.1393e-06, 6.9737e-06,  ..., 2.6221e-01,
           3.3417e-02, 3.4821e-02]],

         [[7.6025e-01, 2.9564e-05, 9.3997e-05,  ..., 5.6725e-03,
           1.8433e-02, 5.0507e-02]],

         [[9.3652e-01, 3.0160e-05, 2.5034e-06,  ..., 1.1932e-02,
           6.6795e-03, 1.7529e-03]],

         ...,

         [[6.7920e-01, 1.9467e-04, 8.2731e-05,  ..., 1.1627e-02,
           2.7252e-02, 1.9485e-02]],

         [[8.7793e-01, 1.4353e-04, 3.6597e-05,  ..., 5.9624e-03,
           9.1629e-03, 4.3602e-03]],

         [[7.5635e-01, 3.4571e-05, 1.7524e-05,  ..., 2.0599e-02,
           6.6101e-02, 2.8351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6055e-01, 1.7023e-03, 8.2970e-04,  ..., 2.9411e-03,
           1.9445e-03, 3.4523e-03]],

         [[9.1064e-01, 8.7082e-05, 1.2863e-04,  ..., 9.7752e-04,
           7.9956e-03, 9.7961e-03]],

         [[8.7842e-01, 6.3598e-05, 2.9564e-05,  ..., 4.7874e-03,
           7.5436e-04, 2.4338e-03]],

         ...,

         [[5.2295e-01, 3.8505e-05, 1.3447e-04,  ..., 1.9188e-03,
           3.4607e-02, 3.0249e-01]],

         [[5.4102e-01, 4.7684e-06, 6.3181e-06,  ..., 3.3665e-03,
           2.9224e-01, 1.0883e-01]],

         [[9.0576e-01, 2.7418e-05, 2.1696e-05,  ..., 2.7893e-02,
           5.4398e-03, 1.0880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5405e-01, 6.1035e-04, 1.1740e-03,  ..., 1.6129e-02,
           1.6174e-02, 3.7251e-03]],

         [[8.8818e-01, 4.5359e-05, 8.0884e-05,  ..., 7.3662e-03,
           3.7994e-02, 1.5358e-02]],

         [[5.8740e-01, 4.2796e-05, 6.1810e-05,  ..., 1.5955e-01,
           1.9180e-02, 3.2257e-02]],

         ...,

         [[3.4180e-01, 6.5756e-04, 7.1764e-04,  ..., 2.9240e-03,
           1.0925e-02, 1.6832e-03]],

         [[7.6172e-01, 7.1943e-05, 4.4048e-05,  ..., 1.2457e-01,
           7.4539e-03, 6.5346e-03]],

         [[1.6016e-01, 1.7285e-06, 6.5565e-06,  ..., 2.1992e-03,
           7.3486e-01, 8.5754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9824e-01, 6.3598e-05, 1.1116e-04,  ..., 1.4663e-04,
           7.8964e-04, 1.5961e-02]],

         [[5.5908e-01, 9.8348e-05, 9.3460e-05,  ..., 7.6416e-02,
           7.7881e-02, 1.6861e-02]],

         [[7.5342e-01, 4.5300e-06, 9.1791e-06,  ..., 1.6556e-03,
           3.8357e-03, 1.6022e-02]],

         ...,

         [[9.4922e-01, 4.9889e-05, 2.0802e-05,  ..., 2.7180e-03,
           4.6005e-03, 1.6909e-03]],

         [[3.8745e-01, 2.1100e-04, 6.7949e-05,  ..., 5.8441e-03,
           5.5962e-03, 5.1709e-01]],

         [[2.4194e-01, 3.5524e-04, 1.8680e-04,  ..., 2.3773e-02,
           4.4617e-02, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6182e-01, 1.7285e-06, 5.4240e-06,  ..., 6.6710e-04,
           7.2365e-03, 6.0840e-01]],

         [[6.2744e-01, 5.0688e-04, 9.1457e-04,  ..., 7.1945e-03,
           8.8215e-04, 1.7529e-03]],

         [[6.4502e-01, 6.0272e-04, 6.9714e-04,  ..., 1.6602e-02,
           3.6957e-02, 1.6571e-02]],

         ...,

         [[6.0498e-01, 1.3793e-04, 2.3556e-04,  ..., 3.8834e-03,
           3.0365e-03, 5.3482e-03]],

         [[2.2571e-01, 9.7215e-05, 5.9187e-05,  ..., 4.3259e-03,
           7.6904e-02, 2.5192e-02]],

         [[2.1545e-01, 3.0696e-05, 6.8605e-05,  ..., 3.2616e-03,
           6.0516e-02, 6.5918e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 5.8413e-05, 9.1851e-05,  ..., 3.7136e-03,
           7.6904e-02, 3.7720e-01]],

         [[3.3252e-01, 4.8161e-04, 4.8542e-04,  ..., 2.9282e-02,
           1.8542e-01, 4.1107e-02]],

         [[4.2432e-01, 2.1553e-04, 1.5771e-04,  ..., 2.1191e-03,
           4.4403e-03, 3.6526e-03]],

         ...,

         [[2.8488e-02, 1.3113e-06, 1.7285e-06,  ..., 8.5175e-05,
           5.1308e-03, 9.6191e-01]],

         [[8.4961e-01, 5.5671e-05, 4.8757e-05,  ..., 1.2856e-03,
           4.1733e-03, 2.8763e-02]],

         [[6.9141e-01, 2.3329e-04, 7.6056e-05,  ..., 1.1208e-02,
           4.7836e-03, 6.4659e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3516e-01, 4.3631e-04, 5.4741e-04,  ..., 8.8684e-02,
           1.3275e-02, 9.2010e-03]],

         [[7.7979e-01, 4.3869e-05, 1.6499e-04,  ..., 2.2858e-02,
           5.0812e-02, 3.9154e-02]],

         [[4.1699e-01, 1.6260e-04, 1.3793e-04,  ..., 1.2636e-03,
           4.6501e-03, 1.6068e-02]],

         ...,

         [[2.8027e-01, 5.9605e-08, 8.9407e-07,  ..., 4.0985e-02,
           6.4453e-01, 2.7740e-02]],

         [[3.8745e-01, 5.5361e-04, 7.1335e-04,  ..., 1.7059e-02,
           1.1269e-02, 4.8561e-03]],

         [[7.6514e-01, 1.3709e-06, 2.1458e-06,  ..., 5.0507e-02,
           9.0942e-02, 2.3468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1484e-01, 6.7890e-05, 1.9014e-05,  ..., 3.7365e-03,
           3.0708e-03, 7.8735e-03]],

         [[1.1322e-01, 4.3678e-04, 4.7469e-04,  ..., 6.2370e-03,
           1.7380e-02, 4.1428e-03]],

         [[3.9697e-01, 2.5558e-04, 2.5606e-04,  ..., 6.4819e-02,
           1.0394e-01, 1.5762e-02]],

         ...,

         [[3.2544e-01, 1.6093e-06, 1.4901e-06,  ..., 5.0049e-02,
           4.4873e-01, 1.7365e-02]],

         [[7.7734e-01, 1.7881e-07, 1.1921e-07,  ..., 2.0676e-02,
           8.9539e-02, 1.6113e-02]],

         [[4.0771e-01, 1.6332e-05, 3.4690e-05,  ..., 1.8295e-02,
           6.2378e-02, 7.2708e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9773e-02, 4.1723e-07, 5.9605e-07,  ..., 8.4448e-04,
           7.0915e-03, 8.8818e-01]],

         [[1.7615e-01, 8.4043e-06, 2.0444e-05,  ..., 8.6670e-02,
           2.6245e-01, 3.5919e-02]],

         [[3.4058e-01, 1.3103e-03, 1.1044e-03,  ..., 2.2278e-02,
           1.9119e-02, 8.5678e-03]],

         ...,

         [[2.8198e-01, 5.0306e-05, 2.7180e-05,  ..., 8.5144e-02,
           2.1469e-02, 1.3817e-02]],

         [[6.1230e-01, 5.5432e-06, 8.1062e-06,  ..., 4.2267e-03,
           6.6338e-03, 1.3954e-02]],

         [[9.5459e-02, 8.5056e-05, 4.2081e-05,  ..., 5.3358e-04,
           3.5739e-04, 2.0027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0352e-01, 1.9133e-04, 8.8990e-05,  ..., 3.3997e-02,
           7.9498e-03, 3.3722e-02]],

         [[3.3008e-01, 4.7565e-04, 1.4567e-04,  ..., 7.6904e-02,
           3.1555e-02, 2.9770e-02]],

         [[8.6441e-03, 2.2173e-05, 2.5034e-06,  ..., 1.2238e-02,
           2.7252e-02, 3.5877e-03]],

         ...,

         [[7.0117e-01, 5.8830e-05, 2.2709e-05,  ..., 5.6915e-02,
           1.9287e-02, 1.0239e-02]],

         [[2.6416e-01, 2.1183e-04, 1.0926e-04,  ..., 1.3184e-01,
           7.0312e-02, 4.6631e-02]],

         [[8.9893e-01, 2.4176e-04, 6.8486e-05,  ..., 7.3357e-03,
           3.0384e-03, 9.0179e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3342e-01, 1.0586e-03, 1.7157e-03,  ..., 5.5161e-03,
           2.9755e-03, 3.0875e-04]],

         [[2.6489e-01, 4.4131e-04, 2.3580e-04,  ..., 2.8702e-02,
           7.1220e-03, 8.4610e-03]],

         [[2.7002e-01, 4.1127e-05, 6.1393e-05,  ..., 1.0162e-01,
           1.4502e-01, 5.9998e-02]],

         ...,

         [[7.3730e-01, 2.1100e-04, 1.0806e-04,  ..., 3.0632e-03,
           1.6479e-02, 2.5681e-02]],

         [[9.7461e-01, 1.7107e-05, 2.4676e-05,  ..., 3.3975e-06,
           4.6372e-05, 2.7084e-03]],

         [[7.8223e-01, 2.6870e-04, 3.2043e-04,  ..., 5.5046e-03,
           1.0902e-02, 4.9362e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0358e-01, 1.1039e-04, 8.6010e-05,  ..., 7.7400e-03,
           6.1829e-02, 8.2397e-03]],

         [[1.7688e-01, 4.3602e-03, 2.1076e-03,  ..., 1.8646e-02,
           1.6586e-02, 1.1757e-02]],

         [[6.2622e-02, 5.4240e-06, 6.1393e-06,  ..., 2.1637e-02,
           2.3071e-01, 5.2393e-01]],

         ...,

         [[2.6636e-01, 2.5094e-05, 1.9431e-05,  ..., 6.7566e-02,
           4.9438e-02, 5.7030e-03]],

         [[7.1777e-01, 2.5332e-05, 1.8597e-05,  ..., 5.4504e-02,
           1.4328e-02, 1.3046e-02]],

         [[2.4475e-01, 1.5104e-04, 1.1086e-04,  ..., 1.8143e-02,
           1.8323e-01, 2.6465e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3545e-01, 4.6432e-05, 2.1935e-05,  ..., 1.0910e-03,
           4.1161e-03, 3.0426e-02]],

         [[5.6274e-02, 3.9339e-05, 2.4617e-05,  ..., 2.6025e-01,
           1.0681e-01, 2.6169e-02]],

         [[1.9421e-01, 1.6212e-05, 1.8060e-05,  ..., 1.0509e-03,
           4.6112e-02, 7.3291e-01]],

         ...,

         [[1.5320e-01, 7.3195e-04, 2.1219e-04,  ..., 6.2370e-04,
           1.8549e-03, 8.1863e-03]],

         [[3.7964e-01, 4.4298e-04, 1.8978e-04,  ..., 4.1046e-03,
           5.4359e-03, 1.2016e-02]],

         [[3.9819e-01, 1.1635e-03, 1.2531e-03,  ..., 1.0910e-02,
           7.1526e-03, 1.2360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8933e-01, 1.2189e-04, 6.7592e-05,  ..., 8.5327e-02,
           5.6396e-02, 1.9333e-02]],

         [[9.6533e-01, 5.4836e-06, 5.9605e-07,  ..., 2.8191e-03,
           9.6560e-06, 1.2226e-03]],

         [[8.4961e-01, 4.9353e-05, 6.6757e-06,  ..., 1.4618e-02,
           1.7023e-04, 1.7872e-03]],

         ...,

         [[1.0632e-01, 1.9627e-03, 2.3670e-03,  ..., 7.0114e-03,
           5.3749e-03, 2.0237e-03]],

         [[6.4258e-01, 5.8889e-05, 6.5207e-05,  ..., 3.7384e-02,
           7.7171e-03, 1.4870e-02]],

         [[2.9614e-01, 1.1644e-03, 2.4452e-03,  ..., 9.7504e-03,
           2.2308e-02, 1.9531e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6450e-01, 1.8060e-04, 2.3377e-04,  ..., 1.0178e-02,
           2.9678e-02, 4.8920e-02]],

         [[2.7222e-01, 2.6417e-04, 5.3358e-04,  ..., 1.4771e-02,
           2.0996e-02, 2.5314e-02]],

         [[2.7490e-01, 7.6711e-05, 1.0693e-04,  ..., 1.7563e-02,
           5.3177e-03, 7.3242e-03]],

         ...,

         [[3.5889e-02, 2.2030e-04, 2.9635e-04,  ..., 3.2196e-02,
           2.5452e-02, 1.8341e-02]],

         [[5.5725e-02, 1.9479e-04, 9.9480e-05,  ..., 7.5817e-04,
           3.8743e-04, 8.0261e-03]],

         [[5.5908e-01, 3.5048e-04, 4.3273e-04,  ..., 3.4294e-03,
           2.7771e-03, 9.4681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4661e-01, 1.3292e-05, 1.1444e-05,  ..., 7.8278e-03,
           3.3150e-03, 8.8043e-03]],

         [[3.5034e-01, 1.7512e-04, 2.8205e-04,  ..., 1.9928e-02,
           8.0185e-03, 8.1787e-03]],

         [[2.5073e-01, 5.1022e-05, 1.1951e-04,  ..., 1.1818e-02,
           1.6443e-01, 3.8525e-01]],

         ...,

         [[5.2588e-01, 7.4446e-05, 1.2517e-04,  ..., 6.5857e-02,
           5.1666e-02, 3.6072e-02]],

         [[7.1777e-01, 3.0458e-05, 2.6226e-04,  ..., 2.2144e-03,
           1.1330e-02, 1.9882e-02]],

         [[2.4927e-01, 3.9935e-05, 8.2970e-05,  ..., 1.6537e-03,
           8.6594e-03, 3.2959e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0813e-01, 2.1040e-04, 1.7929e-04,  ..., 1.3992e-02,
           1.2947e-02, 2.9053e-02]],

         [[1.3623e-01, 1.7047e-04, 2.4748e-04,  ..., 7.5867e-02,
           2.8122e-02, 1.4702e-02]],

         [[4.3408e-01, 1.7595e-04, 1.3316e-04,  ..., 5.4207e-03,
           1.7166e-02, 6.4514e-02]],

         ...,

         [[4.6533e-01, 8.3447e-07, 1.4067e-05,  ..., 1.5588e-01,
           3.2983e-01, 2.6245e-02]],

         [[3.7817e-01, 4.1842e-05, 6.9261e-05,  ..., 1.1444e-01,
           1.7310e-01, 3.8330e-02]],

         [[3.5059e-01, 7.7486e-07, 9.6560e-06,  ..., 9.2224e-02,
           4.8682e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9520e-02, 7.9870e-06, 3.1590e-05,  ..., 4.4434e-02,
           5.0751e-02, 1.0803e-02]],

         [[1.6281e-02, 1.0848e-05, 1.4544e-05,  ..., 6.2132e-04,
           2.1912e-02, 9.3164e-01]],

         [[3.5583e-02, 4.1962e-05, 1.4591e-04,  ..., 3.0251e-03,
           6.8169e-03, 3.6049e-03]],

         ...,

         [[1.7627e-01, 3.1638e-04, 1.2674e-03,  ..., 1.8295e-02,
           3.4058e-02, 8.9874e-03]],

         [[2.9932e-01, 6.4969e-06, 8.1062e-06,  ..., 1.1768e-01,
           2.4524e-01, 3.8361e-02]],

         [[2.9395e-01, 1.1921e-07, 8.9407e-07,  ..., 1.0565e-01,
           4.6240e-01, 5.7892e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1055e-01, 1.9360e-04, 8.8835e-04,  ..., 7.5531e-03,
           5.7678e-03, 1.0529e-02]],

         [[9.8438e-01, 2.1040e-05, 6.2168e-05,  ..., 2.7704e-04,
           3.2640e-04, 8.8024e-04]],

         [[9.0186e-01, 1.7035e-04, 8.7857e-05,  ..., 3.4084e-03,
           1.1778e-03, 1.3351e-03]],

         ...,

         [[7.7881e-01, 3.9160e-05, 3.7909e-05,  ..., 3.4760e-02,
           8.2550e-03, 1.1826e-02]],

         [[3.5498e-01, 1.4901e-06, 4.7684e-07,  ..., 1.6308e-03,
           2.8629e-03, 1.0887e-02]],

         [[1.4478e-01, 1.1981e-05, 1.7881e-05,  ..., 5.8365e-03,
           5.1605e-02, 7.2363e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9669e-02, 7.5102e-06, 1.0729e-05,  ..., 1.6333e-01,
           1.8359e-01, 8.2764e-02]],

         [[7.0703e-01, 6.2168e-05, 4.1962e-05,  ..., 3.7323e-02,
           2.9770e-02, 4.1016e-02]],

         [[1.8665e-01, 7.8106e-04, 5.4312e-04,  ..., 4.3488e-02,
           2.2293e-02, 5.7793e-03]],

         ...,

         [[5.9814e-01, 1.6749e-05, 7.1406e-05,  ..., 1.2250e-01,
           2.5085e-02, 4.6196e-03]],

         [[6.3379e-01, 1.6809e-05, 4.8757e-05,  ..., 1.9287e-02,
           2.9648e-02, 2.9877e-02]],

         [[2.8320e-01, 7.5698e-06, 3.4392e-05,  ..., 1.8524e-02,
           1.0828e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5053e-02, 3.8147e-06, 4.7684e-06,  ..., 2.2717e-03,
           7.0534e-03, 1.0506e-02]],

         [[1.5710e-01, 1.6630e-05, 8.9407e-06,  ..., 7.8201e-03,
           8.9340e-03, 5.3436e-02]],

         [[2.3853e-01, 1.3375e-04, 4.1103e-04,  ..., 6.5186e-02,
           2.8488e-02, 7.2144e-02]],

         ...,

         [[6.1066e-02, 3.4571e-06, 4.5419e-05,  ..., 1.5198e-02,
           9.3079e-02, 1.5833e-01]],

         [[8.9172e-02, 3.8743e-06, 5.2512e-05,  ..., 1.4786e-02,
           9.0576e-02, 9.4971e-02]],

         [[9.4189e-01, 6.3777e-06, 1.4544e-05,  ..., 8.6746e-03,
           1.5945e-02, 4.6425e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 2.0862e-06, 1.1086e-05,  ..., 2.1164e-02,
           1.6266e-02, 6.8176e-02]],

         [[3.2422e-01, 2.0683e-05, 6.3837e-05,  ..., 9.2163e-02,
           2.5439e-01, 4.8584e-02]],

         [[8.8501e-02, 4.1723e-07, 1.9073e-06,  ..., 7.0312e-02,
           7.7344e-01, 4.9072e-02]],

         ...,

         [[6.2354e-01, 7.4148e-05, 1.6642e-04,  ..., 5.6725e-03,
           1.0597e-02, 9.8038e-03]],

         [[3.8184e-01, 2.1875e-05, 1.2100e-05,  ..., 3.9307e-02,
           1.0358e-01, 2.5375e-02]],

         [[7.0374e-02, 5.1141e-05, 2.1040e-04,  ..., 2.2491e-02,
           6.5613e-02, 6.6797e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 8.7440e-05, 1.6320e-04,  ..., 4.2786e-02,
           2.4658e-02, 3.2654e-02]],

         [[4.3481e-01, 3.3188e-04, 2.0695e-04,  ..., 1.2999e-03,
           1.3313e-03, 1.2726e-02]],

         [[3.0713e-01, 8.6010e-05, 2.0015e-04,  ..., 2.5253e-03,
           7.2212e-03, 5.0140e-02]],

         ...,

         [[8.2861e-01, 1.6212e-04, 2.3055e-04,  ..., 1.2985e-02,
           1.2985e-02, 9.6512e-03]],

         [[1.7407e-01, 7.4983e-05, 2.8753e-04,  ..., 4.9400e-03,
           3.7079e-02, 7.3291e-01]],

         [[5.0488e-01, 2.4974e-05, 6.8963e-05,  ..., 1.4107e-02,
           2.7466e-01, 1.2378e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1309e-01, 3.8862e-05, 5.5647e-04,  ..., 9.2697e-03,
           1.0880e-02, 1.5350e-02]],

         [[7.6843e-02, 6.4373e-06, 5.2273e-05,  ..., 1.4549e-02,
           4.2450e-02, 4.4464e-02]],

         [[2.3804e-01, 7.7903e-05, 5.6076e-04,  ..., 8.1299e-02,
           1.2158e-01, 1.2549e-01]],

         ...,

         [[8.6475e-01, 7.1943e-05, 3.5644e-04,  ..., 7.9956e-03,
           1.0590e-02, 2.4780e-02]],

         [[9.0039e-01, 8.9407e-07, 3.1090e-04,  ..., 2.5043e-03,
           3.0661e-04, 7.4158e-03]],

         [[5.4395e-01, 5.8651e-05, 1.1730e-04,  ..., 1.0834e-01,
           6.2943e-03, 1.4099e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3663e-05, 2.7180e-05, 1.7798e-01,  ..., 4.1199e-04,
           2.3254e-02, 1.0339e-01]],

         [[5.5134e-05, 1.7822e-05, 1.0384e-02,  ..., 1.0948e-03,
           2.9068e-03, 7.9041e-03]],

         [[2.9325e-04, 4.6730e-04, 1.0724e-01,  ..., 4.9744e-03,
           2.8519e-02, 5.6519e-02]],

         ...,

         [[3.9172e-04, 5.1618e-05, 1.9751e-01,  ..., 3.0289e-02,
           1.2854e-01, 1.7297e-01]],

         [[1.1021e-04, 5.7638e-05, 2.9224e-01,  ..., 2.9221e-03,
           6.4583e-03, 1.3779e-02]],

         [[4.1127e-06, 2.9206e-06, 2.5879e-01,  ..., 3.2845e-03,
           4.1199e-03, 3.8483e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3956e-05, 5.6624e-05, 2.8125e-01,  ..., 1.6203e-03,
           6.2103e-03, 5.2826e-02]],

         [[5.3287e-05, 1.9968e-05, 1.7651e-01,  ..., 5.4216e-04,
           3.8223e-03, 3.8452e-02]],

         [[6.8307e-05, 7.2718e-06, 3.1519e-01,  ..., 9.3508e-04,
           6.4659e-03, 4.8340e-02]],

         ...,

         [[1.9836e-04, 2.4676e-05, 2.3941e-02,  ..., 5.4359e-04,
           9.5034e-04, 9.9411e-03]],

         [[1.1027e-05, 5.9605e-06, 2.4927e-01,  ..., 1.1677e-04,
           3.5553e-03, 3.7628e-02]],

         [[4.8816e-05, 3.6001e-05, 2.5391e-01,  ..., 3.1161e-04,
           8.5602e-03, 3.2562e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8716e-05, 6.3777e-06, 1.0327e-01,  ..., 3.3545e-04,
           1.6146e-03, 1.4565e-02]],

         [[6.3062e-05, 1.3113e-06, 2.2290e-01,  ..., 3.0994e-05,
           3.5744e-03, 2.9465e-02]],

         [[5.1200e-05, 1.5795e-05, 2.3291e-01,  ..., 2.6941e-04,
           3.1452e-03, 2.4170e-02]],

         ...,

         [[5.1141e-05, 1.6928e-05, 1.4893e-01,  ..., 1.2245e-03,
           1.3115e-02, 4.2664e-02]],

         [[1.0073e-05, 1.1086e-05, 9.4971e-02,  ..., 2.8896e-04,
           2.7370e-03, 1.4801e-02]],

         [[4.0472e-05, 5.1856e-06, 1.5356e-01,  ..., 3.1233e-04,
           2.4242e-03, 1.7624e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6134e-05, 2.6822e-06, 1.1559e-02,  ..., 1.1284e-02,
           3.9795e-01, 2.2144e-01]],

         [[9.6560e-06, 4.9472e-05, 9.8633e-02,  ..., 1.3269e-01,
           4.4617e-02, 3.8330e-02]],

         [[3.8862e-05, 2.7871e-04, 3.1860e-01,  ..., 2.0161e-03,
           7.9250e-04, 7.7057e-04]],

         ...,

         [[5.9891e-04, 5.6386e-05, 1.8567e-01,  ..., 1.3237e-03,
           4.4365e-03, 1.4572e-02]],

         [[1.1988e-03, 9.5844e-05, 4.3945e-02,  ..., 1.8753e-02,
           2.4805e-01, 3.1348e-01]],

         [[1.8501e-04, 1.7166e-05, 4.2450e-02,  ..., 1.1650e-02,
           1.1816e-01, 1.4758e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1324e-03, 4.3726e-04, 9.3689e-02,  ..., 1.0971e-02,
           5.9082e-02, 9.0088e-02]],

         [[7.8630e-04, 1.2426e-03, 2.4292e-01,  ..., 9.9106e-03,
           1.3977e-02, 1.4816e-02]],

         [[2.9640e-03, 3.4666e-04, 1.4819e-01,  ..., 4.9706e-03,
           5.3680e-02, 6.8665e-02]],

         ...,

         [[3.8052e-04, 8.2314e-05, 2.4292e-02,  ..., 6.1417e-03,
           7.3828e-01, 1.3550e-01]],

         [[1.0805e-03, 3.2520e-03, 2.9053e-01,  ..., 1.0157e-03,
           6.0768e-03, 3.1097e-02]],

         [[4.1103e-04, 3.3021e-04, 2.1973e-01,  ..., 2.8595e-02,
           2.9739e-02, 3.4760e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[3.8123e-04, 1.7452e-04, 6.9695e-03,  ..., 1.1249e-01,
           5.3662e-01, 7.9773e-02]],

         [[3.8147e-04, 5.5504e-04, 6.7635e-03,  ..., 1.4320e-02,
           1.0577e-01, 7.8174e-01]],

         [[1.0014e-05, 2.1279e-05, 7.4208e-05,  ..., 1.5712e-04,
           2.4567e-03, 9.9121e-01]],

         ...,

         [[2.2054e-06, 6.0797e-06, 7.3731e-05,  ..., 2.0039e-04,
           4.0245e-03, 9.8535e-01]],

         [[3.4237e-03, 2.5043e-03, 2.3518e-03,  ..., 3.6804e-02,
           7.7881e-02, 2.6917e-02]],

         [[1.9806e-02, 5.3291e-03, 1.6251e-03,  ..., 1.9196e-02,
           3.1647e-02, 5.7281e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0041e-04, 8.8310e-04, 3.6602e-03,  ..., 3.3661e-02,
           7.5867e-02, 6.7627e-01]],

         [[2.3842e-07, 2.9802e-07, 1.1921e-06,  ..., 1.0779e-01,
           7.7197e-01, 1.1475e-01]],

         [[9.3174e-04, 2.6970e-03, 1.8417e-02,  ..., 6.7383e-02,
           4.2175e-02, 1.8323e-01]],

         ...,

         [[2.6684e-03, 6.7635e-03, 1.2245e-02,  ..., 6.3515e-03,
           2.8915e-02, 4.1846e-01]],

         [[2.6913e-03, 1.0918e-02, 3.9612e-02,  ..., 2.7649e-02,
           3.1586e-02, 2.6382e-02]],

         [[5.0087e-03, 1.2108e-02, 1.9043e-02,  ..., 2.0007e-01,
           4.9774e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0797, 0.0073, 0.0407,  ..., 0.0317, 0.0271, 0.0395]],

         [[0.0009, 0.0004, 0.0023,  ..., 0.0082, 0.0031, 0.0075]],

         [[0.0091, 0.0022, 0.0018,  ..., 0.0525, 0.0319, 0.0242]],

         ...,

         [[0.0175, 0.0009, 0.0030,  ..., 0.0721, 0.1962, 0.3416]],

         [[0.0207, 0.0023, 0.0026,  ..., 0.0334, 0.0279, 0.0581]],

         [[0.0115, 0.0041, 0.0156,  ..., 0.0721, 0.2273, 0.3181]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[6.9189e-01, 1.6093e-06, 3.2365e-05,  ..., 9.1858e-03,
           2.7786e-02, 6.3858e-03]],

         [[8.4766e-01, 2.9802e-06, 7.8499e-05,  ..., 5.4283e-03,
           5.3444e-03, 1.3863e-02]],

         [[9.6875e-01, 2.1458e-06, 9.8348e-06,  ..., 3.0842e-03,
           1.3208e-03, 4.2801e-03]],

         ...,

         [[3.5156e-01, 2.0087e-04, 1.2732e-03,  ..., 3.9935e-05,
           2.5392e-05, 9.3307e-03]],

         [[7.3633e-01, 1.7548e-04, 4.4608e-04,  ..., 6.4316e-03,
           2.6932e-03, 5.2299e-03]],

         [[2.7832e-01, 1.6439e-04, 3.0708e-04,  ..., 3.1616e-02,
           2.9922e-02, 1.1360e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1895e-01, 1.6809e-05, 5.9605e-07,  ..., 1.9445e-03,
           4.1723e-04, 2.1225e-02]],

         [[8.6328e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8275e-02,
           5.8380e-02, 4.9011e-02]],

         [[9.1846e-01, 3.4308e-04, 1.0836e-04,  ..., 5.8136e-03,
           1.0818e-02, 4.9133e-03]],

         ...,

         [[1.7554e-01, 6.1703e-04, 2.3746e-04,  ..., 1.2396e-01,
           1.5015e-01, 1.9760e-02]],

         [[1.6113e-01, 1.4229e-03, 4.4441e-04,  ..., 2.5055e-02,
           7.5455e-03, 7.0190e-03]],

         [[4.9121e-01, 1.1736e-04, 3.5143e-04,  ..., 2.5833e-02,
           3.5370e-02, 1.9470e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7505e-01, 2.7704e-04, 9.3937e-05,  ..., 9.0027e-02,
           5.2521e-02, 1.1346e-01]],

         [[6.6406e-01, 1.2374e-04, 4.1783e-05,  ..., 6.8237e-02,
           8.1848e-02, 9.5825e-02]],

         [[8.0811e-01, 8.3447e-05, 2.1577e-05,  ..., 1.0361e-02,
           1.5808e-02, 4.0588e-02]],

         ...,

         [[9.0527e-01, 3.4988e-05, 5.9605e-07,  ..., 4.7684e-05,
           5.0068e-06, 8.9905e-02]],

         [[8.7109e-01, 4.3678e-04, 6.7949e-06,  ..., 2.3246e-06,
           1.7881e-07, 3.0640e-02]],

         [[9.4678e-01, 3.3951e-04, 1.0133e-06,  ..., 1.4901e-06,
           5.9605e-08, 1.2749e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9380e-01, 1.6212e-05, 7.0333e-06,  ..., 7.4036e-02,
           2.6709e-01, 1.8042e-01]],

         [[9.0088e-01, 2.8253e-05, 6.4373e-06,  ..., 1.8358e-04,
           2.0492e-04, 3.7659e-02]],

         [[7.1240e-01, 9.1195e-05, 2.2948e-05,  ..., 1.6804e-03,
           6.7377e-04, 5.3375e-02]],

         ...,

         [[6.8896e-01, 2.8777e-04, 2.6107e-05,  ..., 3.0594e-03,
           5.8327e-03, 2.1500e-02]],

         [[7.5293e-01, 1.3599e-03, 1.1909e-04,  ..., 7.3318e-03,
           1.0315e-02, 8.5815e-02]],

         [[9.3799e-01, 1.1265e-04, 4.9412e-05,  ..., 1.1263e-03,
           2.8210e-03, 1.3809e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1260e-01, 2.7239e-05, 4.2856e-05,  ..., 6.0425e-03,
           2.4796e-03, 8.0948e-03]],

         [[9.6094e-01, 1.8120e-05, 7.5102e-06,  ..., 7.0524e-04,
           6.6805e-04, 1.1612e-02]],

         [[9.8486e-01, 3.4571e-06, 1.6689e-06,  ..., 7.3576e-04,
           1.8902e-03, 6.5079e-03]],

         ...,

         [[8.1787e-01, 1.6010e-04, 1.1146e-05,  ..., 8.4229e-03,
           7.8201e-03, 2.1271e-02]],

         [[8.9258e-01, 2.2078e-04, 4.1306e-05,  ..., 2.0294e-03,
           7.6151e-04, 1.2581e-02]],

         [[9.6973e-01, 9.0003e-06, 6.3777e-06,  ..., 1.2074e-03,
           5.3368e-03, 6.6948e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4741e-01, 9.2888e-04, 4.5085e-04,  ..., 4.9162e-04,
           2.3441e-03, 2.5803e-02]],

         [[9.3652e-01, 2.5213e-05, 5.2154e-05,  ..., 1.0395e-03,
           2.1248e-03, 1.3943e-03]],

         [[9.3066e-01, 3.1650e-05, 1.0252e-05,  ..., 1.5092e-04,
           1.8682e-03, 3.0685e-02]],

         ...,

         [[9.6338e-01, 1.5855e-05, 1.6153e-05,  ..., 2.5272e-04,
           1.6375e-03, 1.2520e-02]],

         [[9.0869e-01, 2.3842e-06, 1.1086e-05,  ..., 1.3199e-02,
           4.3640e-02, 3.5191e-03]],

         [[8.4961e-01, 3.5763e-06, 6.1393e-06,  ..., 5.3978e-04,
           4.3678e-03, 1.0321e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0999e-01, 2.1839e-04, 7.7438e-04,  ..., 3.4515e-02,
           1.8204e-02, 1.3763e-02]],

         [[9.1406e-01, 2.8849e-05, 4.0710e-05,  ..., 3.8261e-03,
           1.1536e-02, 1.4526e-02]],

         [[7.1436e-01, 2.9981e-05, 5.3048e-05,  ..., 1.0185e-02,
           1.3879e-01, 4.9561e-02]],

         ...,

         [[7.0898e-01, 1.6093e-04, 1.7536e-04,  ..., 2.2583e-03,
           1.1148e-03, 9.9335e-03]],

         [[7.8906e-01, 1.0824e-04, 4.3392e-05,  ..., 4.7016e-04,
           2.0866e-03, 5.9082e-02]],

         [[7.5293e-01, 1.4663e-05, 2.3782e-05,  ..., 7.7942e-02,
           4.9164e-02, 1.6479e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 7.3612e-05, 8.1062e-06,  ..., 3.5167e-06,
           4.2915e-06, 1.3733e-02]],

         [[8.9160e-01, 7.0989e-05, 2.1029e-04,  ..., 5.0545e-03,
           5.6839e-03, 8.0643e-03]],

         [[8.9111e-01, 1.7285e-05, 5.5432e-06,  ..., 4.0460e-04,
           2.1327e-04, 1.9531e-02]],

         ...,

         [[8.1006e-01, 3.2973e-04, 1.5819e-04,  ..., 1.7071e-03,
           2.2354e-03, 2.7557e-02]],

         [[7.3389e-01, 8.8632e-05, 1.0002e-04,  ..., 4.8943e-03,
           3.5004e-02, 1.0437e-02]],

         [[3.6743e-01, 1.0185e-03, 1.8835e-04,  ..., 1.1765e-02,
           1.9394e-02, 4.4586e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0527e-01, 7.7486e-06, 6.6757e-06,  ..., 4.6110e-04,
           1.9045e-03, 5.8057e-01]],

         [[5.1123e-01, 5.1069e-04, 7.5674e-04,  ..., 9.0742e-04,
           1.4334e-03, 4.5197e-02]],

         [[7.9688e-01, 2.4772e-04, 4.8232e-04,  ..., 6.7635e-03,
           6.3286e-03, 1.7441e-02]],

         ...,

         [[7.3584e-01, 4.8041e-05, 9.0420e-05,  ..., 5.6505e-04,
           2.6741e-03, 1.5478e-03]],

         [[7.9297e-01, 5.2810e-05, 3.6001e-05,  ..., 1.2001e-02,
           5.7587e-02, 1.9913e-02]],

         [[2.7319e-01, 1.4901e-04, 2.5845e-04,  ..., 1.7838e-02,
           3.5370e-02, 5.5469e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1396e-01, 6.6042e-05, 4.8518e-05,  ..., 3.1624e-03,
           2.0203e-02, 5.6494e-01]],

         [[7.4072e-01, 8.7738e-05, 8.2076e-05,  ..., 1.0361e-02,
           8.3435e-02, 8.8654e-03]],

         [[2.2693e-01, 2.8038e-04, 1.3936e-04,  ..., 1.3485e-03,
           3.4904e-03, 5.8838e-02]],

         ...,

         [[4.1870e-02, 5.0664e-06, 3.3975e-06,  ..., 4.7112e-04,
           5.0621e-03, 9.4043e-01]],

         [[8.1885e-01, 1.9610e-05, 3.2842e-05,  ..., 2.4052e-03,
           9.6817e-03, 2.6001e-02]],

         [[8.4717e-01, 2.2352e-04, 4.5776e-05,  ..., 1.8272e-03,
           2.6520e-02, 1.0246e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2129e-01, 7.5054e-04, 5.6648e-04,  ..., 3.0327e-03,
           3.0098e-03, 1.6460e-03]],

         [[7.8223e-01, 5.3346e-05, 1.8036e-04,  ..., 4.7546e-02,
           6.7932e-02, 1.4771e-02]],

         [[2.8101e-01, 8.5652e-05, 7.5877e-05,  ..., 3.3879e-04,
           8.9598e-04, 2.3300e-02]],

         ...,

         [[6.4502e-01, 0.0000e+00, 1.7881e-07,  ..., 3.2349e-02,
           2.7905e-01, 3.6072e-02]],

         [[3.4082e-01, 3.7146e-04, 7.9679e-04,  ..., 6.0272e-03,
           4.1847e-03, 4.9973e-04]],

         [[9.2969e-01, 5.9605e-08, 5.9605e-08,  ..., 9.2936e-04,
           8.3389e-03, 5.8167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7891e-01, 5.1618e-05, 1.2636e-05,  ..., 1.7643e-03,
           5.5618e-03, 1.7624e-02]],

         [[1.2408e-01, 3.7169e-04, 4.6253e-04,  ..., 4.9744e-02,
           3.6896e-02, 2.9984e-03]],

         [[4.0381e-01, 5.7161e-05, 8.0884e-05,  ..., 3.7476e-02,
           2.7539e-01, 3.5522e-02]],

         ...,

         [[6.8701e-01, 4.0531e-06, 3.0398e-06,  ..., 8.7708e-02,
           5.3528e-02, 7.0068e-02]],

         [[6.5820e-01, 2.9802e-07, 3.5763e-07,  ..., 2.8553e-03,
           2.6108e-02, 3.0591e-01]],

         [[6.8701e-01, 1.6093e-05, 2.3603e-05,  ..., 5.8868e-02,
           5.7068e-02, 6.5552e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0472e-02, 2.3842e-07, 2.3842e-07,  ..., 1.0723e-04,
           8.3160e-04, 9.6680e-01]],

         [[7.0740e-02, 8.2850e-06, 1.4961e-05,  ..., 2.2046e-01,
           3.0493e-01, 5.7983e-02]],

         [[2.3425e-01, 4.5848e-04, 9.4461e-04,  ..., 1.8890e-02,
           1.7105e-02, 6.0028e-02]],

         ...,

         [[5.2344e-01, 4.4286e-05, 4.6074e-05,  ..., 2.6398e-03,
           8.3694e-03, 1.2573e-02]],

         [[6.4258e-01, 1.2398e-05, 9.3579e-06,  ..., 3.4904e-03,
           5.2376e-03, 2.4429e-02]],

         [[9.2236e-01, 2.2650e-05, 1.3351e-05,  ..., 1.5378e-04,
           3.8815e-04, 5.0735e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4072e-01, 2.4199e-05, 2.0504e-05,  ..., 1.5507e-03,
           1.1581e-02, 1.5283e-01]],

         [[1.5283e-01, 3.7217e-04, 6.5207e-05,  ..., 2.1076e-03,
           2.2995e-02, 1.0028e-01]],

         [[2.5415e-01, 2.2650e-05, 1.6391e-05,  ..., 9.7733e-03,
           1.5442e-02, 8.9539e-02]],

         ...,

         [[8.7646e-01, 3.6657e-05, 2.7955e-05,  ..., 2.6512e-03,
           1.8225e-03, 1.0735e-02]],

         [[7.0996e-01, 8.0049e-05, 3.9995e-05,  ..., 1.1208e-02,
           7.0251e-02, 6.7566e-02]],

         [[6.2744e-01, 1.4722e-05, 7.6294e-06,  ..., 6.8331e-04,
           4.6043e-03, 1.8762e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9531e-01, 7.1716e-04, 7.9107e-04,  ..., 1.8494e-02,
           1.2413e-02, 1.8280e-02]],

         [[2.7100e-01, 6.2227e-05, 4.6611e-05,  ..., 8.1396e-04,
           9.8724e-03, 4.4586e-02]],

         [[6.8164e-01, 2.8789e-05, 8.1360e-05,  ..., 1.7426e-02,
           7.5256e-02, 9.0454e-02]],

         ...,

         [[4.9194e-01, 3.8815e-04, 6.1631e-05,  ..., 9.9087e-04,
           3.8776e-03, 8.0627e-02]],

         [[6.2061e-01, 1.0786e-03, 5.9748e-04,  ..., 2.5630e-06,
           1.1981e-05, 1.4999e-02]],

         [[3.9844e-01, 2.2602e-04, 1.7059e-04,  ..., 1.0735e-02,
           1.0025e-02, 1.2016e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.4971e-02, 1.4281e-04, 1.0204e-04,  ..., 9.2041e-02,
           1.5173e-01, 7.9803e-03]],

         [[3.7012e-01, 1.6904e-04, 1.1688e-04,  ..., 5.3864e-03,
           5.5023e-02, 3.6652e-02]],

         [[2.9370e-01, 3.0696e-05, 1.6212e-05,  ..., 8.2703e-03,
           1.2927e-01, 4.1431e-01]],

         ...,

         [[5.6201e-01, 1.2577e-05, 2.0206e-05,  ..., 2.8320e-02,
           8.3557e-02, 8.3557e-02]],

         [[8.2031e-01, 1.1861e-05, 1.7643e-05,  ..., 2.6321e-03,
           3.3531e-03, 9.4971e-02]],

         [[5.4980e-01, 1.1122e-04, 1.0407e-04,  ..., 8.3160e-03,
           4.3732e-02, 2.3462e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.1128e-04, 1.9968e-05,  ..., 3.0255e-04,
           6.8855e-03, 3.4973e-02]],

         [[1.7480e-01, 4.6372e-05, 3.3438e-05,  ..., 8.1299e-02,
           2.0764e-01, 1.1829e-01]],

         [[3.1836e-01, 2.2411e-05, 1.0252e-05,  ..., 2.4068e-04,
           9.0790e-04, 6.6357e-01]],

         ...,

         [[6.4148e-02, 1.0592e-04, 3.2306e-05,  ..., 1.3232e-05,
           3.7313e-05, 2.0020e-02]],

         [[2.7905e-01, 3.5191e-04, 1.6367e-04,  ..., 3.6964e-03,
           1.4114e-02, 8.5602e-03]],

         [[3.1738e-01, 2.9397e-04, 1.3638e-04,  ..., 5.2071e-03,
           2.6047e-02, 2.0126e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8696e-01, 1.1867e-04, 7.5102e-05,  ..., 7.1449e-03,
           8.7509e-03, 3.9520e-02]],

         [[9.7461e-01, 1.0532e-04, 4.4703e-06,  ..., 1.2517e-06,
           4.4703e-06, 7.2956e-05]],

         [[9.2822e-01, 6.0034e-04, 1.0931e-04,  ..., 1.8883e-04,
           5.8711e-05, 1.4172e-03]],

         ...,

         [[1.8787e-01, 5.1165e-04, 9.3365e-04,  ..., 9.3536e-03,
           6.8436e-03, 3.5309e-02]],

         [[8.3398e-01, 2.7061e-05, 2.9266e-05,  ..., 1.0891e-03,
           1.6613e-03, 7.7576e-02]],

         [[1.7285e-01, 1.0195e-03, 1.2989e-03,  ..., 7.9575e-03,
           8.7357e-03, 3.2227e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0010e-01, 1.7631e-04, 2.2995e-04,  ..., 5.5275e-03,
           1.6251e-02, 2.5955e-02]],

         [[3.7695e-01, 1.5736e-04, 2.9874e-04,  ..., 3.8433e-03,
           6.9046e-03, 2.0935e-02]],

         [[4.9048e-01, 2.3568e-04, 2.8658e-04,  ..., 6.5088e-04,
           1.7824e-03, 3.1830e-02]],

         ...,

         [[8.4686e-03, 4.2379e-05, 2.5690e-05,  ..., 5.2185e-03,
           3.9764e-02, 1.2909e-02]],

         [[3.4180e-03, 4.1127e-06, 1.0729e-06,  ..., 7.9870e-06,
           8.1658e-05, 1.0696e-02]],

         [[3.4058e-01, 1.4222e-04, 9.6202e-05,  ..., 5.9891e-04,
           2.6016e-03, 3.1677e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4004e-01, 2.9027e-05, 4.9114e-05,  ..., 1.6527e-03,
           1.0445e-02, 6.2012e-02]],

         [[4.8560e-01, 1.6105e-04, 2.0599e-04,  ..., 2.2018e-02,
           6.2225e-02, 6.4583e-03]],

         [[2.6562e-01, 4.5002e-05, 1.2183e-04,  ..., 1.1609e-01,
           1.3049e-01, 2.8271e-01]],

         ...,

         [[5.7910e-01, 4.0483e-04, 3.9697e-04,  ..., 3.2166e-02,
           2.9282e-02, 5.9174e-02]],

         [[1.8274e-01, 4.5538e-05, 3.3855e-05,  ..., 1.5414e-04,
           3.9053e-04, 1.7712e-01]],

         [[4.6704e-01, 1.8752e-04, 1.9503e-04,  ..., 8.6212e-03,
           1.6235e-02, 3.6865e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5742e-01, 2.0790e-04, 2.0957e-04,  ..., 7.2746e-03,
           2.9800e-02, 8.8928e-02]],

         [[2.8003e-01, 9.6560e-05, 1.7822e-04,  ..., 5.5008e-03,
           3.1921e-02, 6.5979e-02]],

         [[5.0195e-01, 2.8205e-04, 3.0041e-04,  ..., 1.0338e-02,
           8.0078e-02, 3.0396e-02]],

         ...,

         [[7.2998e-01, 5.9605e-08, 5.3644e-07,  ..., 3.1586e-02,
           1.9043e-01, 2.8748e-02]],

         [[3.4277e-01, 3.9399e-05, 3.0935e-05,  ..., 9.2957e-02,
           2.2302e-01, 7.9529e-02]],

         [[5.9033e-01, 4.7684e-07, 3.1590e-06,  ..., 1.9730e-02,
           3.1104e-01, 6.1249e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 1.8656e-05, 1.5259e-05,  ..., 4.7089e-02,
           1.6431e-01, 1.6266e-02]],

         [[1.2964e-01, 1.6272e-05, 1.8835e-05,  ..., 3.9749e-03,
           1.5717e-02, 7.6953e-01]],

         [[1.8787e-01, 1.2267e-04, 2.2209e-04,  ..., 3.1525e-02,
           6.2439e-02, 4.8714e-03]],

         ...,

         [[1.6968e-01, 1.0812e-04, 3.0494e-04,  ..., 1.0492e-01,
           4.2389e-02, 3.1616e-02]],

         [[5.4736e-01, 1.4663e-05, 8.7023e-06,  ..., 1.0126e-01,
           2.1851e-01, 4.2358e-02]],

         [[3.3276e-01, 0.0000e+00, 1.7881e-07,  ..., 8.2520e-02,
           4.5483e-01, 7.3975e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4229e-01, 7.7069e-05, 2.1052e-04,  ..., 2.2488e-03,
           3.1586e-03, 3.1036e-02]],

         [[9.8926e-01, 1.3828e-05, 1.7166e-05,  ..., 4.3273e-05,
           2.5570e-05, 3.0947e-04]],

         [[9.5801e-01, 1.2106e-04, 2.1935e-05,  ..., 5.2166e-04,
           3.8457e-04, 2.6913e-03]],

         ...,

         [[8.4033e-01, 5.4240e-05, 5.7757e-05,  ..., 3.6564e-03,
           2.2400e-02, 1.0522e-01]],

         [[9.1406e-01, 8.5235e-06, 3.0994e-06,  ..., 7.4148e-04,
           1.6069e-03, 6.4545e-03]],

         [[4.5825e-01, 9.1791e-05, 1.1879e-04,  ..., 4.5624e-03,
           1.6434e-02, 4.5825e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4023e-01, 1.5259e-05, 2.7180e-05,  ..., 1.2366e-01,
           2.4207e-01, 7.6477e-02]],

         [[8.1055e-01, 1.8179e-05, 9.5367e-06,  ..., 7.8201e-03,
           2.1606e-02, 1.2482e-01]],

         [[3.1860e-01, 1.8954e-04, 2.3341e-04,  ..., 5.2032e-02,
           2.9068e-02, 2.2778e-01]],

         ...,

         [[8.2959e-01, 5.3644e-06, 1.7107e-05,  ..., 3.4515e-02,
           5.6458e-02, 3.0457e-02]],

         [[8.4033e-01, 2.1696e-05, 4.6492e-05,  ..., 5.4054e-03,
           6.8855e-03, 5.0079e-02]],

         [[4.8291e-01, 2.6703e-05, 3.9697e-05,  ..., 4.1229e-02,
           9.3628e-02, 1.0364e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6499e-01, 4.7028e-05, 2.7657e-05,  ..., 9.6512e-03,
           6.7566e-02, 5.1392e-02]],

         [[3.4692e-01, 3.9577e-05, 1.7822e-05,  ..., 1.7433e-03,
           8.6823e-03, 6.2683e-02]],

         [[5.1221e-01, 1.2493e-04, 2.2149e-04,  ..., 1.2238e-02,
           3.4576e-02, 9.0393e-02]],

         ...,

         [[3.9337e-02, 1.3113e-06, 2.1100e-05,  ..., 1.2903e-01,
           2.1106e-01, 1.8738e-02]],

         [[4.9780e-01, 1.1086e-05, 6.8009e-05,  ..., 5.0079e-02,
           9.5032e-02, 2.6531e-03]],

         [[5.8252e-01, 4.9829e-05, 4.9829e-05,  ..., 1.2793e-01,
           1.1835e-01, 5.8594e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0088e-01, 5.3644e-07, 5.3644e-07,  ..., 1.1070e-02,
           5.5542e-02, 5.0488e-01]],

         [[7.2070e-01, 7.8678e-06, 2.0981e-05,  ..., 2.6352e-02,
           7.0801e-02, 2.7893e-02]],

         [[4.2798e-01, 2.9802e-07, 7.7486e-07,  ..., 5.0720e-02,
           2.7856e-01, 2.2473e-01]],

         ...,

         [[8.9453e-01, 1.4341e-04, 1.9038e-04,  ..., 1.0977e-03,
           3.2139e-03, 2.8748e-02]],

         [[8.4814e-01, 1.5199e-05, 8.3447e-06,  ..., 9.2392e-03,
           3.6682e-02, 1.9485e-02]],

         [[1.7151e-01, 1.5736e-04, 2.2852e-04,  ..., 1.3756e-02,
           3.1982e-02, 6.1768e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 3.0518e-05, 8.1301e-05,  ..., 2.8214e-02,
           3.4302e-02, 1.6052e-01]],

         [[3.9014e-01, 4.0555e-04, 3.0541e-04,  ..., 7.5912e-04,
           4.8180e-03, 2.2705e-02]],

         [[2.3730e-01, 4.4942e-05, 1.4997e-04,  ..., 1.4534e-02,
           8.4595e-02, 8.3313e-02]],

         ...,

         [[8.7695e-01, 2.0218e-04, 5.3692e-04,  ..., 1.6571e-02,
           2.4490e-02, 1.6830e-02]],

         [[2.2180e-01, 1.9300e-04, 2.6798e-04,  ..., 5.5542e-03,
           1.7105e-02, 6.7236e-01]],

         [[6.1572e-01, 2.4498e-05, 6.9201e-05,  ..., 7.5867e-02,
           2.1606e-01, 6.1920e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0632e-01, 1.1623e-05, 1.4126e-04,  ..., 6.0364e-02,
           7.1240e-01, 3.8208e-02]],

         [[1.7969e-01, 1.6332e-05, 8.0228e-05,  ..., 1.1131e-02,
           8.9294e-02, 4.5074e-02]],

         [[4.9487e-01, 2.9850e-04, 1.8482e-03,  ..., 3.9581e-02,
           1.1255e-01, 6.9885e-02]],

         ...,

         [[8.4375e-01, 9.5427e-05, 2.4247e-04,  ..., 3.6926e-02,
           2.9327e-02, 8.0490e-03]],

         [[9.6094e-01, 2.2650e-06, 1.4830e-04,  ..., 3.4308e-04,
           1.0262e-03, 5.6343e-03]],

         [[7.7637e-01, 5.5373e-05, 3.8326e-05,  ..., 1.1818e-02,
           6.1417e-03, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4809e-05, 5.1916e-05, 1.8958e-01,  ..., 3.3966e-02,
           1.7395e-01, 1.4671e-02]],

         [[9.8050e-05, 4.6730e-05, 2.9556e-02,  ..., 1.1353e-02,
           3.2227e-02, 1.2549e-01]],

         [[1.4770e-04, 1.9062e-04, 1.0907e-01,  ..., 4.6539e-02,
           7.0984e-02, 1.2659e-01]],

         ...,

         [[3.0065e-04, 1.5557e-05, 6.7505e-02,  ..., 3.7292e-02,
           7.0361e-01, 5.8655e-02]],

         [[5.0545e-04, 2.3592e-04, 3.0737e-01,  ..., 4.1504e-03,
           5.2452e-03, 9.2957e-02]],

         [[7.1585e-05, 2.2757e-04, 2.9639e-01,  ..., 1.3535e-02,
           2.2675e-02, 2.7130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6606e-04, 2.2435e-04, 2.6196e-01,  ..., 1.6235e-02,
           6.6223e-02, 8.1635e-03]],

         [[1.4269e-04, 1.8394e-04, 2.7563e-01,  ..., 1.0689e-02,
           3.0685e-02, 9.5825e-03]],

         [[3.3832e-04, 1.4782e-04, 3.3252e-01,  ..., 7.5798e-03,
           2.3163e-02, 1.4467e-03]],

         ...,

         [[6.5899e-04, 1.6057e-04, 1.4453e-01,  ..., 3.1815e-03,
           1.8234e-02, 2.5366e-01]],

         [[1.3685e-04, 1.0717e-04, 3.2764e-01,  ..., 8.3618e-03,
           2.8412e-02, 4.2229e-03]],

         [[3.5405e-05, 9.8705e-05, 3.0664e-01,  ..., 7.0152e-03,
           1.0490e-02, 9.6464e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6192e-04, 2.0516e-04, 2.8003e-01,  ..., 7.8812e-03,
           2.0004e-02, 8.5974e-04]],

         [[3.7813e-04, 1.4484e-05, 2.9736e-01,  ..., 6.0768e-03,
           1.3275e-02, 2.3483e-02]],

         [[8.0681e-04, 4.2844e-04, 3.2812e-01,  ..., 6.9427e-03,
           1.5465e-02, 7.5378e-03]],

         ...,

         [[4.8995e-05, 2.3484e-05, 1.7273e-01,  ..., 1.6327e-02,
           3.2715e-02, 3.2788e-01]],

         [[1.1295e-04, 2.0814e-04, 2.1606e-01,  ..., 1.0971e-02,
           1.5961e-02, 1.6861e-02]],

         [[4.1032e-04, 7.3135e-05, 2.9395e-01,  ..., 6.2447e-03,
           1.5213e-02, 1.8066e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6473e-05, 6.4373e-06, 2.0966e-02,  ..., 6.8176e-02,
           5.1562e-01, 2.6343e-01]],

         [[1.6451e-05, 5.2214e-05, 2.0813e-01,  ..., 3.1052e-02,
           4.5380e-02, 1.9821e-02]],

         [[5.5790e-05, 3.1662e-04, 2.3218e-01,  ..., 1.4839e-03,
           1.3628e-03, 9.7504e-03]],

         ...,

         [[2.2411e-04, 1.4067e-05, 2.8564e-01,  ..., 6.3241e-05,
           2.9397e-04, 1.1873e-03]],

         [[1.0118e-03, 9.0003e-05, 5.3772e-02,  ..., 6.2866e-02,
           1.8262e-01, 4.2969e-01]],

         [[1.6546e-04, 1.2398e-05, 5.7495e-02,  ..., 5.6824e-02,
           1.5259e-01, 3.9746e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2173e-04, 5.5885e-04, 2.0740e-01,  ..., 6.2866e-03,
           1.0612e-02, 1.5649e-01]],

         [[2.2829e-04, 3.0518e-04, 2.8589e-01,  ..., 6.7253e-03,
           9.9411e-03, 5.7861e-02]],

         [[1.2398e-03, 5.6791e-04, 1.7395e-01,  ..., 7.1106e-02,
           1.9702e-01, 8.3435e-02]],

         ...,

         [[1.0605e-03, 1.7309e-04, 1.4783e-01,  ..., 4.5441e-02,
           1.3354e-01, 2.7173e-01]],

         [[2.0409e-04, 6.2370e-04, 2.9883e-01,  ..., 4.9820e-03,
           8.1726e-02, 5.3589e-02]],

         [[2.3901e-04, 5.1677e-05, 1.3281e-01,  ..., 1.1908e-01,
           1.7322e-01, 6.4758e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.7351e-03, 1.1053e-03, 1.7288e-02,  ..., 1.6406e-01,
           2.7905e-01, 1.9788e-01]],

         [[2.5928e-05, 2.4629e-04, 1.8196e-03,  ..., 1.1864e-02,
           1.9562e-02, 9.4238e-01]],

         [[2.3842e-07, 3.5346e-05, 1.0133e-05,  ..., 1.0133e-05,
           4.3058e-04, 9.9951e-01]],

         ...,

         [[0.0000e+00, 1.7881e-07, 6.1393e-06,  ..., 2.7478e-05,
           9.1028e-04, 9.9854e-01]],

         [[3.2711e-03, 2.8858e-03, 3.9444e-03,  ..., 8.7036e-02,
           4.2419e-02, 5.1178e-02]],

         [[1.5305e-02, 4.1847e-03, 2.0065e-03,  ..., 1.4153e-02,
           5.9570e-02, 6.3416e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6253e-05, 2.2066e-04, 1.9665e-03,  ..., 2.9816e-02,
           2.3450e-01, 5.6250e-01]],

         [[0.0000e+00, 5.9605e-08, 4.1723e-07,  ..., 8.2642e-02,
           8.6133e-01, 5.3375e-02]],

         [[2.9278e-04, 8.2111e-04, 6.9847e-03,  ..., 6.7322e-02,
           3.5254e-01, 4.2114e-02]],

         ...,

         [[2.1801e-03, 6.3591e-03, 1.1879e-02,  ..., 3.0334e-02,
           3.0884e-01, 8.9905e-02]],

         [[1.6146e-03, 1.1032e-02, 3.1921e-02,  ..., 2.4292e-02,
           2.3727e-02, 1.9745e-02]],

         [[6.5804e-03, 1.8173e-02, 1.9196e-02,  ..., 1.7471e-02,
           1.0876e-01, 1.0620e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5378e-02, 7.2632e-03, 4.0344e-02,  ..., 3.3966e-02,
           6.1035e-02, 3.5339e-02]],

         [[2.6627e-03, 1.0386e-03, 5.8174e-03,  ..., 6.0005e-03,
           1.7090e-02, 1.2802e-02]],

         [[4.3945e-03, 8.2541e-04, 9.4652e-04,  ..., 3.1952e-02,
           2.7557e-02, 1.9852e-02]],

         ...,

         [[1.3618e-02, 4.7803e-05, 1.2159e-04,  ..., 1.3647e-01,
           5.8838e-01, 2.0178e-01]],

         [[7.4654e-03, 1.0881e-03, 1.7452e-03,  ..., 8.9539e-02,
           5.4718e-02, 1.3440e-01]],

         [[5.2032e-03, 8.3590e-04, 3.1052e-03,  ..., 1.1475e-01,
           5.7373e-01, 1.9519e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.0381e-01, 1.4341e-04, 8.3923e-05,  ..., 4.5547e-03,
           8.4457e-03, 2.9469e-03]],

         [[7.6123e-01, 9.1791e-06, 1.7703e-05,  ..., 1.2306e-02,
           5.8594e-02, 1.4209e-01]],

         [[7.9590e-01, 1.6093e-06, 1.6093e-06,  ..., 4.5319e-03,
           1.5610e-02, 1.6235e-01]],

         ...,

         [[5.2686e-01, 1.0264e-04, 6.6566e-04,  ..., 2.4092e-04,
           2.5809e-05, 3.4218e-03]],

         [[6.5967e-01, 3.1662e-04, 8.8549e-04,  ..., 1.7105e-02,
           2.0004e-02, 1.7441e-02]],

         [[6.6406e-01, 1.1134e-04, 3.5334e-04,  ..., 4.9072e-02,
           1.4557e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6084e-01, 6.1989e-06, 8.9407e-07,  ..., 3.4637e-03,
           8.4877e-04, 9.3994e-02]],

         [[5.5371e-01, 5.9605e-08, 0.0000e+00,  ..., 2.8954e-03,
           3.7256e-01, 6.9824e-02]],

         [[7.9443e-01, 9.8407e-05, 2.4498e-05,  ..., 7.3586e-03,
           1.6431e-01, 1.7334e-02]],

         ...,

         [[5.9863e-01, 5.4932e-04, 2.8300e-04,  ..., 5.6549e-02,
           8.3389e-03, 1.1826e-02]],

         [[3.7109e-01, 1.4200e-03, 6.9284e-04,  ..., 1.1002e-02,
           1.6068e-02, 1.6846e-02]],

         [[3.5156e-01, 3.9876e-05, 4.4942e-05,  ..., 3.6835e-02,
           5.5957e-01, 1.4313e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6099e-01, 3.0088e-04, 1.1426e-04,  ..., 6.5369e-02,
           1.2402e-01, 8.9233e-02]],

         [[6.7920e-01, 8.1599e-05, 3.8981e-05,  ..., 7.3486e-02,
           1.6565e-01, 2.1332e-02]],

         [[7.7344e-01, 4.1366e-05, 2.5868e-05,  ..., 1.0757e-02,
           1.1316e-01, 7.7698e-02]],

         ...,

         [[7.9736e-01, 1.7881e-07, 5.9605e-08,  ..., 1.4246e-05,
           2.6524e-05, 1.9641e-01]],

         [[9.4092e-01, 2.6822e-06, 1.0729e-06,  ..., 0.0000e+00,
           0.0000e+00, 1.7578e-02]],

         [[9.7461e-01, 1.7881e-07, 5.9605e-08,  ..., 0.0000e+00,
           0.0000e+00, 2.1820e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 4.2379e-05, 5.1260e-06,  ..., 2.5406e-02,
           2.7686e-01, 4.3671e-02]],

         [[9.2578e-01, 1.0073e-05, 5.9605e-07,  ..., 1.4818e-04,
           2.8896e-03, 5.7800e-02]],

         [[8.7939e-01, 6.3181e-05, 2.4140e-05,  ..., 3.4771e-03,
           5.2185e-03, 8.6365e-03]],

         ...,

         [[1.4697e-01, 4.5002e-05, 7.3314e-06,  ..., 3.1757e-03,
           8.2324e-01, 5.2376e-03]],

         [[6.8506e-01, 7.4005e-04, 1.3292e-04,  ..., 2.3010e-02,
           5.9967e-02, 1.4636e-01]],

         [[4.4702e-01, 2.4962e-04, 5.0843e-05,  ..., 4.1237e-03,
           9.7229e-02, 4.4360e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.7119e-01, 7.5102e-06, 9.3579e-06,  ..., 5.6839e-04,
           5.9586e-03, 6.8588e-03]],

         [[9.1260e-01, 1.0312e-05, 2.6822e-06,  ..., 2.3804e-03,
           6.5575e-03, 5.0751e-02]],

         [[9.7266e-01, 3.3975e-06, 5.3644e-07,  ..., 4.7565e-04,
           2.0966e-02, 3.7861e-03]],

         ...,

         [[8.2812e-01, 8.1003e-05, 1.6630e-05,  ..., 2.8324e-03,
           2.1576e-02, 5.8533e-02]],

         [[9.4873e-01, 8.1539e-05, 5.0664e-06,  ..., 6.4373e-04,
           9.1476e-03, 9.8343e-03]],

         [[9.0576e-01, 1.2577e-05, 3.0994e-06,  ..., 2.0866e-03,
           2.4979e-02, 5.6091e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7695e-01, 1.1835e-03, 2.8110e-04,  ..., 9.8267e-03,
           4.6112e-02, 3.9856e-02]],

         [[9.7461e-01, 3.9935e-06, 1.2279e-05,  ..., 1.0490e-03,
           1.7910e-03, 2.2106e-03]],

         [[7.4561e-01, 4.5002e-05, 2.8014e-06,  ..., 6.6614e-04,
           1.6382e-01, 5.5573e-02]],

         ...,

         [[5.2490e-01, 1.1384e-05, 1.7285e-05,  ..., 1.0376e-03,
           1.9409e-02, 4.3945e-01]],

         [[9.6680e-01, 1.3709e-06, 3.5167e-06,  ..., 6.8092e-03,
           1.2703e-03, 1.3596e-02]],

         [[4.5605e-01, 2.0266e-06, 4.2915e-06,  ..., 1.1845e-03,
           5.2002e-01, 5.4321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5415e-01, 1.2720e-04, 8.7595e-04,  ..., 5.1758e-02,
           2.5635e-02, 1.0559e-02]],

         [[7.1680e-01, 1.7345e-05, 5.4836e-06,  ..., 5.6267e-03,
           2.4524e-01, 1.6632e-02]],

         [[9.4092e-01, 1.0252e-05, 9.7752e-06,  ..., 9.6893e-03,
           1.6708e-02, 1.0712e-02]],

         ...,

         [[9.0967e-01, 3.1412e-05, 2.7955e-05,  ..., 8.2874e-04,
           1.3107e-02, 2.8563e-04]],

         [[5.4834e-01, 1.8477e-05, 5.7817e-06,  ..., 1.8463e-03,
           3.6963e-01, 4.0497e-02]],

         [[8.9502e-01, 2.0266e-05, 2.7061e-05,  ..., 2.0798e-02,
           7.2975e-03, 3.5370e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8301e-01, 1.5259e-04, 3.3379e-05,  ..., 7.7486e-07,
           6.5327e-05, 1.6079e-03]],

         [[9.3115e-01, 2.0325e-05, 2.8908e-05,  ..., 1.7319e-03,
           2.6932e-02, 1.2527e-02]],

         [[9.6729e-01, 1.7285e-05, 1.9908e-05,  ..., 4.7159e-04,
           5.3978e-03, 6.6376e-03]],

         ...,

         [[8.8574e-01, 3.0136e-04, 1.7846e-04,  ..., 1.2808e-03,
           2.6566e-02, 4.2648e-03]],

         [[8.7451e-01, 6.3419e-05, 6.0737e-05,  ..., 1.0849e-02,
           4.7569e-03, 6.1913e-03]],

         [[7.4121e-01, 2.4929e-03, 4.5562e-04,  ..., 1.8097e-02,
           8.1253e-03, 9.5749e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5576e-01, 1.9133e-05, 1.9312e-05,  ..., 5.8556e-04,
           1.1902e-02, 3.2739e-01]],

         [[5.0537e-01, 9.4509e-04, 1.7004e-03,  ..., 1.7099e-03,
           5.8044e-02, 3.9581e-02]],

         [[8.2471e-01, 2.6512e-04, 4.9829e-04,  ..., 3.9043e-03,
           6.8207e-03, 1.1177e-02]],

         ...,

         [[6.5137e-01, 2.0468e-04, 2.7871e-04,  ..., 5.8556e-03,
           1.1223e-02, 1.3504e-03]],

         [[7.4414e-01, 7.9811e-05, 7.1228e-05,  ..., 6.5857e-02,
           6.3354e-02, 8.2169e-03]],

         [[5.5127e-01, 9.3102e-05, 1.6475e-04,  ..., 1.1276e-02,
           2.7969e-02, 3.6694e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3843e-02, 3.3557e-05, 1.1683e-05,  ..., 1.6870e-03,
           5.8441e-02, 8.5449e-01]],

         [[6.3965e-01, 5.0664e-05, 2.3901e-05,  ..., 1.6687e-01,
           4.8279e-02, 1.0498e-02]],

         [[1.4270e-01, 5.9128e-04, 1.1694e-04,  ..., 1.2306e-02,
           2.1997e-01, 6.1249e-02]],

         ...,

         [[2.0874e-01, 1.1384e-05, 1.2994e-05,  ..., 3.1910e-03,
           4.8981e-02, 6.8311e-01]],

         [[8.7402e-01, 4.7326e-05, 2.7180e-05,  ..., 4.7302e-03,
           2.0889e-02, 1.5602e-02]],

         [[8.6816e-01, 1.3626e-04, 2.1577e-05,  ..., 1.7975e-02,
           4.2816e-02, 2.4277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.9199e-01, 4.2439e-04, 9.4175e-04,  ..., 8.9722e-03,
           1.6144e-02, 3.6812e-03]],

         [[8.2959e-01, 1.0014e-05, 4.8637e-05,  ..., 4.4952e-02,
           3.5522e-02, 1.3252e-02]],

         [[3.5107e-01, 3.0696e-05, 3.1292e-05,  ..., 2.4071e-03,
           2.9526e-02, 3.6221e-03]],

         ...,

         [[2.4536e-01, 0.0000e+00, 0.0000e+00,  ..., 3.6743e-02,
           7.0166e-01, 1.4389e-02]],

         [[4.0137e-01, 5.4312e-04, 8.1062e-04,  ..., 5.9166e-03,
           2.7142e-03, 1.2457e-04]],

         [[5.8643e-01, 1.7881e-07, 0.0000e+00,  ..., 3.1738e-03,
           3.9453e-01, 1.2199e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3975e-01, 4.8637e-05, 7.3314e-06,  ..., 1.2077e-02,
           4.1107e-02, 2.4933e-02]],

         [[4.0436e-02, 9.8610e-04, 4.1842e-04,  ..., 5.1727e-02,
           1.8501e-03, 6.4754e-04]],

         [[2.9980e-01, 6.4194e-05, 9.7871e-05,  ..., 3.1543e-01,
           4.2877e-02, 7.7133e-03]],

         ...,

         [[6.1279e-01, 1.7285e-06, 7.1526e-07,  ..., 6.4087e-02,
           1.8860e-01, 2.2430e-02]],

         [[7.2632e-02, 5.9605e-08, 0.0000e+00,  ..., 1.1147e-02,
           9.0918e-01, 4.1962e-03]],

         [[5.1709e-01, 1.7464e-05, 3.4690e-05,  ..., 9.4543e-02,
           1.0236e-01, 1.1963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6987e-02, 5.3644e-07, 7.7486e-07,  ..., 1.8752e-04,
           2.0508e-02, 9.3896e-01]],

         [[5.0812e-02, 5.5432e-06, 1.3351e-05,  ..., 3.8892e-01,
           1.2927e-01, 3.6530e-02]],

         [[1.3562e-01, 5.6648e-04, 7.9679e-04,  ..., 2.7039e-02,
           3.4302e-02, 1.0445e-02]],

         ...,

         [[6.6895e-01, 7.9870e-05, 2.7061e-05,  ..., 1.0033e-02,
           2.5818e-02, 1.6289e-03]],

         [[6.3965e-01, 6.3181e-06, 1.3709e-06,  ..., 3.8872e-03,
           1.1951e-01, 5.9570e-02]],

         [[6.0059e-01, 3.3438e-05, 1.6212e-05,  ..., 2.3377e-04,
           1.0358e-01, 2.1149e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7974e-02, 2.1458e-06, 3.5763e-07,  ..., 2.3060e-03,
           9.1553e-01, 6.2180e-03]],

         [[2.0740e-01, 3.2973e-04, 6.3479e-05,  ..., 4.3152e-02,
           1.6992e-01, 9.7595e-02]],

         [[4.1919e-01, 3.5644e-05, 3.7432e-05,  ..., 2.8976e-02,
           1.0071e-01, 1.0638e-01]],

         ...,

         [[9.0332e-01, 3.1173e-05, 2.2948e-05,  ..., 3.7346e-03,
           2.2003e-02, 1.2733e-02]],

         [[5.2490e-01, 2.9743e-05, 1.7226e-05,  ..., 5.4474e-02,
           2.6392e-01, 2.6978e-02]],

         [[7.3584e-01, 1.3351e-05, 8.2850e-06,  ..., 3.3817e-03,
           1.5857e-01, 2.7451e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6609e-02, 1.5383e-03, 2.3708e-03,  ..., 6.9695e-03,
           5.3024e-03, 4.9133e-03]],

         [[5.6305e-02, 4.3809e-05, 2.0146e-05,  ..., 2.6398e-02,
           6.2317e-02, 2.9434e-02]],

         [[3.4229e-01, 1.5020e-05, 2.3723e-05,  ..., 1.5186e-01,
           3.2397e-01, 8.4229e-02]],

         ...,

         [[4.3359e-01, 4.1437e-04, 4.1485e-05,  ..., 9.9373e-04,
           6.9153e-02, 2.4353e-02]],

         [[8.3984e-01, 2.9349e-04, 1.9693e-04,  ..., 3.5763e-05,
           4.5252e-04, 1.0780e-02]],

         [[5.6934e-01, 1.0842e-04, 1.0228e-04,  ..., 1.7059e-02,
           2.2755e-03, 1.2026e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-02, 7.9095e-05, 9.1016e-05,  ..., 1.1853e-01,
           1.9007e-03, 1.1925e-02]],

         [[3.1323e-01, 4.3726e-04, 2.9683e-04,  ..., 6.5613e-02,
           2.9587e-02, 1.8951e-02]],

         [[6.1920e-02, 2.7418e-06, 1.4305e-06,  ..., 5.4230e-02,
           3.1689e-01, 4.7217e-01]],

         ...,

         [[3.1665e-01, 6.7353e-06, 8.4043e-06,  ..., 2.8833e-01,
           1.0858e-01, 1.7715e-02]],

         [[2.4036e-01, 4.9472e-06, 5.9009e-06,  ..., 2.5883e-03,
           6.6895e-01, 5.8899e-02]],

         [[3.6401e-01, 3.8981e-05, 2.2531e-05,  ..., 8.7280e-03,
           3.0884e-01, 2.4060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7119e-01, 6.5863e-05, 1.1265e-05,  ..., 3.4668e-02,
           1.4368e-01, 7.0007e-02]],

         [[4.0497e-02, 6.4969e-06, 2.5034e-06,  ..., 1.9019e-01,
           4.1528e-01, 7.4463e-02]],

         [[1.4001e-01, 4.2319e-06, 9.5367e-07,  ..., 2.8777e-04,
           2.2156e-02, 8.3105e-01]],

         ...,

         [[3.7872e-02, 2.8610e-05, 1.5914e-05,  ..., 1.8239e-04,
           3.9291e-03, 6.0387e-03]],

         [[2.6880e-01, 2.6298e-04, 1.0830e-04,  ..., 1.4809e-02,
           3.2104e-02, 3.1860e-02]],

         [[1.7468e-01, 2.2376e-04, 9.1255e-05,  ..., 4.8126e-02,
           2.7206e-02, 1.1253e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7305e-01, 1.0419e-04, 4.1306e-05,  ..., 5.5351e-03,
           5.2917e-02, 7.5195e-02]],

         [[9.2334e-01, 2.9564e-04, 2.0266e-06,  ..., 5.3644e-07,
           4.8995e-05, 1.5249e-03]],

         [[9.1504e-01, 3.1424e-04, 2.0087e-05,  ..., 2.7359e-05,
           1.6994e-03, 5.5275e-03]],

         ...,

         [[1.6919e-01, 4.9829e-04, 1.0138e-03,  ..., 5.0316e-03,
           1.7014e-02, 3.1799e-02]],

         [[8.1104e-01, 2.3186e-05, 2.5094e-05,  ..., 9.0599e-04,
           8.2825e-02, 4.5044e-02]],

         [[1.8188e-01, 7.7629e-04, 1.3304e-03,  ..., 1.7319e-02,
           4.6722e-02, 5.6366e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8135e-01, 1.6296e-04, 2.1935e-04,  ..., 2.6779e-02,
           5.0812e-02, 4.4159e-02]],

         [[2.3132e-01, 1.1384e-04, 2.5249e-04,  ..., 1.1429e-02,
           2.0065e-02, 4.6265e-02]],

         [[3.4473e-01, 1.1927e-04, 1.6046e-04,  ..., 3.6812e-03,
           5.6274e-02, 3.7170e-02]],

         ...,

         [[1.9287e-02, 7.0632e-05, 5.0902e-05,  ..., 8.7769e-02,
           3.3844e-02, 3.9551e-02]],

         [[7.2823e-03, 7.6294e-06, 2.3842e-06,  ..., 2.6941e-04,
           1.4038e-02, 3.9978e-02]],

         [[5.0439e-01, 4.7684e-05, 4.7684e-05,  ..., 8.2779e-03,
           1.8262e-01, 5.8380e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9102e-01, 2.7537e-05, 3.0398e-05,  ..., 1.2993e-02,
           1.9238e-01, 1.4526e-01]],

         [[3.5425e-01, 1.3781e-04, 2.4581e-04,  ..., 9.6069e-02,
           1.8768e-02, 2.4292e-02]],

         [[1.9373e-01, 2.9504e-05, 9.3460e-05,  ..., 5.2948e-02,
           1.3110e-01, 3.7061e-01]],

         ...,

         [[5.2441e-01, 2.0480e-04, 2.2316e-04,  ..., 4.4403e-02,
           8.7585e-02, 7.9773e-02]],

         [[3.6224e-02, 7.5102e-06, 8.8811e-06,  ..., 8.3685e-05,
           4.8256e-03, 4.0405e-02]],

         [[2.1960e-01, 1.1319e-04, 1.1635e-04,  ..., 2.3880e-02,
           1.2985e-02, 1.9485e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7285e-01, 9.3043e-05, 6.8605e-05,  ..., 3.2501e-02,
           2.6562e-01, 1.0162e-01]],

         [[5.6396e-02, 1.0073e-05, 1.7107e-05,  ..., 4.9011e-02,
           3.8843e-01, 4.4281e-02]],

         [[3.3911e-01, 9.5427e-05, 5.2571e-05,  ..., 7.9285e-02,
           2.3071e-02, 1.4107e-02]],

         ...,

         [[6.1182e-01, 2.9802e-07, 5.9605e-08,  ..., 5.0995e-02,
           2.8442e-01, 4.2297e-02]],

         [[2.1509e-01, 1.5080e-05, 9.8348e-06,  ..., 1.7017e-01,
           2.2546e-01, 4.9896e-02]],

         [[1.5906e-01, 7.1526e-07, 5.9605e-07,  ..., 8.6487e-02,
           7.1289e-01, 3.5217e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1858e-03, 1.1921e-06, 1.8477e-06,  ..., 9.5703e-02,
           8.0414e-03, 1.4786e-02]],

         [[2.0599e-02, 1.1921e-06, 1.9073e-06,  ..., 2.6131e-04,
           1.1375e-02, 9.6191e-01]],

         [[6.0486e-02, 6.0499e-05, 9.8944e-05,  ..., 5.2155e-02,
           2.0695e-03, 5.5389e-03]],

         ...,

         [[1.2830e-01, 1.1438e-04, 3.9339e-04,  ..., 3.1921e-02,
           2.2293e-02, 7.2937e-03]],

         [[4.3335e-02, 1.7881e-07, 1.1921e-07,  ..., 6.3428e-01,
           2.7115e-02, 7.8278e-03]],

         [[1.7749e-01, 5.9605e-08, 5.9605e-08,  ..., 1.7615e-01,
           5.4883e-01, 6.5308e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 8.9526e-05, 4.1723e-04,  ..., 5.2147e-03,
           2.7969e-02, 4.9835e-02]],

         [[9.7949e-01, 8.1599e-05, 1.4365e-05,  ..., 1.0079e-04,
           6.5613e-04, 3.1776e-03]],

         [[8.7988e-01, 1.0693e-04, 9.3997e-05,  ..., 1.1635e-03,
           1.4221e-02, 2.2385e-02]],

         ...,

         [[4.2944e-01, 6.5565e-06, 6.5565e-06,  ..., 6.1859e-02,
           3.8477e-01, 7.9407e-02]],

         [[7.0508e-01, 6.2585e-06, 2.6226e-06,  ..., 6.1989e-03,
           6.4964e-03, 7.4768e-03]],

         [[2.5879e-01, 1.5676e-05, 4.0054e-05,  ..., 1.6632e-03,
           7.3395e-03, 6.9238e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4551e-01, 1.2577e-05, 1.8418e-05,  ..., 1.9128e-01,
           1.1292e-01, 7.3792e-02]],

         [[4.2407e-01, 9.6560e-06, 9.2983e-06,  ..., 4.2999e-02,
           3.2275e-01, 1.4539e-01]],

         [[2.3669e-01, 1.6999e-04, 2.7418e-04,  ..., 1.1559e-02,
           2.4707e-01, 2.3401e-01]],

         ...,

         [[6.0596e-01, 5.4836e-06, 1.7881e-05,  ..., 6.1890e-02,
           7.0129e-02, 1.0608e-01]],

         [[7.5293e-01, 2.7835e-05, 6.2764e-05,  ..., 8.3313e-03,
           1.5503e-02, 1.7029e-02]],

         [[3.0225e-01, 2.9445e-05, 7.1645e-05,  ..., 1.4844e-01,
           2.5208e-02, 3.0396e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8665e-02, 7.6294e-06, 6.9141e-06,  ..., 7.9651e-02,
           3.1128e-03, 5.7716e-03]],

         [[4.2017e-01, 2.4378e-05, 1.3530e-05,  ..., 1.6296e-02,
           9.8419e-03, 7.4272e-03]],

         [[2.6392e-01, 1.1843e-04, 2.7490e-04,  ..., 1.7960e-02,
           1.0577e-01, 1.0828e-01]],

         ...,

         [[2.5940e-02, 7.7486e-06, 1.0604e-04,  ..., 6.7322e-02,
           5.6992e-03, 2.2888e-02]],

         [[1.3257e-01, 2.0027e-05, 1.1003e-04,  ..., 1.1511e-01,
           3.2158e-03, 8.6060e-03]],

         [[5.3613e-01, 2.4259e-05, 2.6345e-05,  ..., 1.0394e-01,
           3.9764e-02, 7.9712e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1118e-02, 1.1921e-07, 0.0000e+00,  ..., 6.6986e-03,
           9.5947e-01, 9.4452e-03]],

         [[5.1318e-01, 7.2718e-06, 2.4915e-05,  ..., 5.3436e-02,
           1.7593e-02, 2.2400e-02]],

         [[3.0493e-01, 1.0133e-06, 1.0729e-06,  ..., 7.8613e-02,
           5.0830e-01, 6.6162e-02]],

         ...,

         [[7.9150e-01, 2.4414e-04, 3.1352e-04,  ..., 3.1223e-03,
           5.9013e-03, 4.4556e-03]],

         [[4.9976e-01, 4.7684e-06, 2.3842e-06,  ..., 2.1997e-01,
           3.1204e-02, 1.0292e-02]],

         [[4.6387e-02, 3.4869e-05, 8.4102e-05,  ..., 3.2692e-03,
           1.1749e-01, 7.6025e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7100e-01, 2.0742e-05, 7.7128e-05,  ..., 2.6520e-02,
           1.8921e-01, 1.6309e-01]],

         [[2.8345e-01, 4.8876e-04, 3.8815e-04,  ..., 5.7945e-03,
           2.7634e-02, 1.9073e-02]],

         [[1.1023e-01, 3.4988e-05, 1.2171e-04,  ..., 4.3823e-02,
           2.6581e-02, 1.5625e-02]],

         ...,

         [[6.4014e-01, 2.2864e-04, 3.0756e-04,  ..., 1.4145e-02,
           3.6682e-02, 1.6968e-01]],

         [[1.2335e-01, 1.1247e-04, 1.4663e-04,  ..., 1.2405e-02,
           8.6121e-02, 7.2070e-01]],

         [[6.4758e-02, 7.0333e-06, 7.0930e-06,  ..., 4.7852e-01,
           6.4758e-02, 5.0446e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7463e-02, 3.8743e-06, 6.9559e-05,  ..., 6.6699e-01,
           7.5989e-02, 2.0294e-02]],

         [[1.1041e-01, 1.1563e-05, 7.8619e-05,  ..., 1.4807e-01,
           4.3427e-02, 4.7119e-02]],

         [[2.8882e-01, 2.6536e-04, 1.3523e-03,  ..., 8.1482e-02,
           4.7150e-02, 1.0461e-01]],

         ...,

         [[7.8418e-01, 2.4188e-04, 5.9652e-04,  ..., 3.5126e-02,
           1.0445e-02, 5.5504e-03]],

         [[9.3213e-01, 2.1458e-06, 5.6362e-04,  ..., 6.1655e-04,
           4.2496e-03, 1.0170e-02]],

         [[3.1470e-01, 5.1498e-05, 4.0889e-05,  ..., 6.9809e-03,
           4.6417e-02, 1.2866e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5923e-05, 4.0829e-05, 1.2524e-01,  ..., 1.7664e-01,
           1.0902e-02, 1.7151e-02]],

         [[7.3791e-05, 1.5378e-05, 7.2098e-03,  ..., 9.1858e-03,
           3.3875e-02, 1.0574e-02]],

         [[7.3969e-05, 1.5533e-04, 5.7159e-02,  ..., 2.7115e-02,
           2.1484e-01, 7.3425e-02]],

         ...,

         [[2.9397e-04, 1.7107e-05, 6.8420e-02,  ..., 4.8218e-01,
           6.1310e-02, 5.8502e-02]],

         [[5.8985e-04, 2.4867e-04, 1.8970e-01,  ..., 1.3214e-02,
           8.6853e-02, 1.8066e-02]],

         [[2.3544e-05, 4.6492e-05, 1.5942e-01,  ..., 1.2589e-02,
           1.0193e-02, 2.7428e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4046e-04, 3.3784e-04, 2.8833e-01,  ..., 5.0903e-02,
           2.2373e-03, 6.0806e-03]],

         [[1.7166e-04, 1.4865e-04, 1.8396e-01,  ..., 3.1464e-02,
           4.6768e-03, 7.2289e-04]],

         [[9.4604e-04, 2.4772e-04, 3.5034e-01,  ..., 2.0386e-02,
           4.6301e-04, 1.1803e-02]],

         ...,

         [[4.5609e-04, 8.9645e-05, 5.9723e-02,  ..., 1.4465e-02,
           1.9739e-01, 1.9699e-02]],

         [[8.2731e-05, 5.6267e-05, 2.7661e-01,  ..., 3.2288e-02,
           9.7847e-04, 1.6136e-03]],

         [[4.8697e-05, 7.5281e-05, 3.0151e-01,  ..., 1.9882e-02,
           1.6451e-03, 5.3101e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6367e-04, 6.9022e-05, 2.3047e-01,  ..., 1.3206e-02,
           2.6369e-04, 3.9711e-03]],

         [[3.8385e-04, 1.1563e-05, 2.4548e-01,  ..., 1.5823e-02,
           3.4027e-02, 5.9700e-04]],

         [[3.8671e-04, 1.6153e-04, 2.7808e-01,  ..., 1.8707e-02,
           5.4626e-03, 6.7616e-04]],

         ...,

         [[1.8895e-05, 9.5963e-06, 8.7402e-02,  ..., 2.6230e-02,
           4.0723e-01, 8.6731e-02]],

         [[4.9591e-05, 7.6473e-05, 1.2341e-01,  ..., 1.4236e-02,
           6.7253e-03, 4.9400e-03]],

         [[2.1505e-04, 3.4213e-05, 1.9373e-01,  ..., 1.2978e-02,
           1.1581e-02, 4.4327e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3910e-06, 3.5763e-07, 1.3256e-03,  ..., 7.2510e-01,
           3.2867e-02, 5.0934e-02]],

         [[2.2709e-05, 7.5281e-05, 1.3611e-01,  ..., 4.2328e-02,
           3.1647e-02, 3.0304e-02]],

         [[1.1551e-04, 5.9795e-04, 3.1030e-01,  ..., 8.3017e-04,
           4.0627e-03, 1.5697e-03]],

         ...,

         [[1.8275e-04, 5.7817e-06, 2.6782e-01,  ..., 2.0004e-02,
           4.2267e-03, 4.3869e-03]],

         [[6.9618e-04, 3.2842e-05, 1.6281e-02,  ..., 2.8760e-01,
           3.2568e-01, 1.9910e-01]],

         [[7.1943e-05, 5.1856e-06, 3.6316e-02,  ..., 1.5906e-01,
           4.6265e-02, 1.5771e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4277e-03, 2.8110e-04, 1.2286e-01,  ..., 1.1299e-02,
           5.4535e-02, 1.0931e-01]],

         [[4.4680e-04, 8.7738e-04, 2.9517e-01,  ..., 1.9932e-03,
           3.2520e-03, 1.5945e-02]],

         [[8.2779e-04, 2.3139e-04, 1.4697e-01,  ..., 6.9946e-02,
           1.1853e-01, 7.2754e-02]],

         ...,

         [[5.1451e-04, 1.0294e-04, 7.6965e-02,  ..., 1.8896e-01,
           1.1377e-01, 3.0200e-01]],

         [[3.9506e-04, 1.0824e-03, 2.6270e-01,  ..., 3.6682e-02,
           4.2908e-02, 3.7842e-02]],

         [[1.1569e-04, 1.0616e-04, 1.1023e-01,  ..., 1.5918e-01,
           1.1646e-01, 7.7576e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.0862e-03, 3.3140e-04, 1.4542e-02,  ..., 8.6304e-02,
           3.8696e-01, 8.1116e-02]],

         [[1.7500e-04, 1.8816e-03, 1.3908e-02,  ..., 1.0826e-02,
           9.6497e-02, 4.3262e-01]],

         [[0.0000e+00, 2.2650e-06, 7.4208e-05,  ..., 4.7684e-07,
           9.0456e-04, 9.9170e-01]],

         ...,

         [[5.9605e-08, 1.3709e-06, 4.4823e-05,  ..., 1.6510e-05,
           4.0321e-03, 9.8682e-01]],

         [[2.7122e-03, 4.2000e-03, 3.9482e-03,  ..., 1.0727e-02,
           2.7390e-02, 1.0727e-02]],

         [[1.7120e-02, 2.8839e-03, 5.1737e-04,  ..., 5.1117e-02,
           1.6083e-02, 5.7953e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2648e-04, 3.7169e-04, 4.1885e-03,  ..., 6.9763e-02,
           9.8389e-02, 5.6592e-01]],

         [[0.0000e+00, 0.0000e+00, 2.3842e-07,  ..., 4.2480e-02,
           9.3701e-01, 1.8265e-02]],

         [[7.1669e-04, 1.6155e-03, 8.3313e-03,  ..., 2.0825e-01,
           3.3997e-02, 2.8183e-02]],

         ...,

         [[2.0618e-03, 5.3062e-03, 6.2485e-03,  ..., 1.7163e-01,
           3.9490e-02, 2.6172e-01]],

         [[2.5940e-03, 1.3725e-02, 2.9510e-02,  ..., 2.2537e-02,
           2.7512e-02, 7.7286e-03]],

         [[1.7061e-03, 6.6414e-03, 6.3400e-03,  ..., 2.7808e-01,
           2.4927e-01, 5.6061e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7791e-02, 5.1575e-03, 2.9449e-02,  ..., 7.4585e-02,
           5.9479e-02, 4.3854e-02]],

         [[1.1368e-03, 5.9652e-04, 2.5711e-03,  ..., 1.1795e-02,
           7.7362e-03, 1.1816e-03]],

         [[3.8681e-03, 6.9618e-04, 8.5640e-04,  ..., 2.4353e-02,
           1.6220e-02, 1.2939e-02]],

         ...,

         [[1.4229e-02, 6.1560e-04, 3.4261e-04,  ..., 1.3708e-01,
           5.2588e-01, 2.6025e-01]],

         [[3.4103e-03, 4.3702e-04, 8.1015e-04,  ..., 5.0507e-02,
           5.4626e-02, 1.8585e-02]],

         [[9.8801e-03, 1.6909e-03, 1.0117e-02,  ..., 8.4045e-02,
           3.5376e-01, 1.0876e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0801e-01, 3.5763e-06, 6.5327e-05,  ..., 5.5771e-03,
           2.6657e-02, 1.6083e-02]],

         [[6.7090e-01, 2.2054e-06, 4.3035e-05,  ..., 7.7477e-03,
           2.1521e-01, 1.5022e-02]],

         [[8.1885e-01, 5.5432e-06, 1.1146e-05,  ..., 8.4778e-02,
           5.9235e-02, 9.8038e-03]],

         ...,

         [[3.2715e-02, 7.3314e-06, 3.3677e-05,  ..., 1.5032e-04,
           2.1398e-05, 6.0501e-03]],

         [[6.9141e-01, 1.1659e-04, 3.9959e-04,  ..., 8.5983e-03,
           1.2268e-02, 1.4015e-02]],

         [[3.4863e-01, 2.0278e-04, 2.3139e-04,  ..., 1.4786e-02,
           3.3722e-02, 3.8055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4570e-01, 1.7285e-06, 8.3447e-07,  ..., 7.0076e-03,
           7.1478e-04, 1.8570e-02]],

         [[7.2559e-01, 7.7486e-07, 5.9605e-08,  ..., 6.9519e-02,
           1.9324e-01, 7.9193e-03]],

         [[9.4385e-01, 1.7977e-04, 9.8884e-05,  ..., 5.3062e-03,
           7.3204e-03, 1.0712e-02]],

         ...,

         [[2.3926e-01, 6.6471e-04, 1.4734e-04,  ..., 3.1738e-02,
           4.0283e-02, 1.2238e-02]],

         [[1.2408e-01, 1.4324e-03, 6.9618e-04,  ..., 4.4060e-03,
           4.4518e-03, 2.9831e-03]],

         [[4.8145e-01, 4.8876e-04, 2.1648e-04,  ..., 1.8347e-01,
           2.2217e-01, 4.4434e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6321e-01, 2.0540e-04, 8.3983e-05,  ..., 9.4910e-02,
           1.0529e-01, 5.4901e-02]],

         [[4.6997e-01, 8.3745e-05, 1.5080e-05,  ..., 2.0667e-01,
           1.3538e-01, 3.8490e-03]],

         [[5.1465e-01, 1.1086e-05, 7.6294e-06,  ..., 2.3181e-01,
           1.2274e-01, 4.5593e-02]],

         ...,

         [[7.7637e-01, 2.2650e-06, 5.9605e-08,  ..., 7.6234e-05,
           1.6093e-06, 2.1045e-01]],

         [[9.0430e-01, 2.8312e-05, 1.1921e-07,  ..., 1.9073e-06,
           0.0000e+00, 1.1688e-02]],

         [[9.6924e-01, 1.0610e-05, 0.0000e+00,  ..., 1.1921e-07,
           0.0000e+00, 4.4785e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2878e-01, 1.9073e-05, 1.1921e-06,  ..., 2.1411e-01,
           5.2588e-01, 2.7039e-02]],

         [[8.3887e-01, 2.6226e-06, 1.2517e-06,  ..., 9.1248e-03,
           1.1480e-04, 2.2598e-02]],

         [[6.6943e-01, 2.2995e-04, 4.9531e-05,  ..., 1.8158e-02,
           2.8076e-03, 2.3575e-02]],

         ...,

         [[5.9375e-01, 4.3750e-04, 3.8028e-05,  ..., 2.8369e-01,
           3.6133e-02, 3.1738e-02]],

         [[8.9355e-01, 2.0051e-04, 4.0591e-05,  ..., 1.4595e-02,
           2.8290e-02, 2.5269e-02]],

         [[2.8320e-01, 2.2137e-04, 3.0100e-05,  ..., 1.7761e-01,
           5.1172e-01, 6.6681e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 4.3511e-06, 2.8014e-06,  ..., 3.5614e-02,
           2.1655e-01, 2.1896e-02]],

         [[9.1016e-01, 6.9320e-05, 1.3053e-05,  ..., 9.9945e-03,
           1.2054e-02, 1.7090e-02]],

         [[9.5654e-01, 1.7881e-07, 1.1921e-07,  ..., 2.2278e-03,
           1.7639e-02, 2.0615e-02]],

         ...,

         [[6.3428e-01, 2.3901e-05, 6.5565e-06,  ..., 4.0283e-02,
           1.4746e-01, 3.3661e-02]],

         [[8.8330e-01, 1.7571e-04, 2.6524e-05,  ..., 8.8043e-03,
           5.5969e-02, 3.1261e-03]],

         [[6.8506e-01, 2.0742e-05, 4.6492e-06,  ..., 8.6365e-02,
           1.3196e-01, 3.0502e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 8.9216e-04, 1.6177e-04,  ..., 1.4885e-02,
           1.1147e-02, 7.3967e-03]],

         [[9.7949e-01, 5.0068e-06, 1.5378e-05,  ..., 8.0919e-04,
           2.9602e-03, 1.7428e-04]],

         [[9.1162e-01, 8.5235e-06, 5.5432e-06,  ..., 1.5656e-02,
           6.0242e-02, 9.1982e-04]],

         ...,

         [[7.6074e-01, 4.6730e-05, 1.3113e-05,  ..., 4.1351e-02,
           5.1880e-02, 9.5947e-02]],

         [[8.3203e-01, 4.1723e-07, 4.1127e-06,  ..., 1.0433e-03,
           5.1880e-03, 3.6163e-03]],

         [[7.4219e-01, 1.1325e-06, 1.5497e-06,  ..., 1.7322e-01,
           4.7424e-02, 1.5167e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3862e-02, 8.5354e-05, 6.8426e-04,  ..., 1.2787e-02,
           8.4839e-03, 3.3234e-02]],

         [[5.5713e-01, 7.0930e-06, 1.7226e-05,  ..., 5.0903e-02,
           3.5059e-01, 2.3300e-02]],

         [[4.7925e-01, 1.3471e-05, 5.0962e-05,  ..., 6.1279e-02,
           2.7490e-01, 1.7654e-02]],

         ...,

         [[8.3936e-01, 4.0889e-05, 8.3923e-05,  ..., 7.5493e-03,
           4.4918e-04, 2.9430e-03]],

         [[4.0723e-01, 3.5644e-05, 2.8729e-05,  ..., 1.8066e-01,
           3.0737e-01, 4.2267e-02]],

         [[6.2695e-01, 2.3842e-06, 5.3048e-06,  ..., 6.5765e-03,
           6.8176e-02, 5.7869e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3008e-01, 6.7949e-06, 8.6427e-06,  ..., 1.7965e-04,
           3.4273e-05, 6.6910e-03]],

         [[9.0479e-01, 9.0599e-06, 1.4961e-05,  ..., 5.7373e-03,
           1.3443e-02, 1.7061e-03]],

         [[5.8105e-01, 2.3842e-07, 1.1921e-06,  ..., 1.0956e-02,
           1.0014e-03, 1.3855e-02]],

         ...,

         [[8.4668e-01, 1.3411e-04, 6.3360e-05,  ..., 2.6688e-02,
           1.6479e-02, 1.4740e-02]],

         [[5.4688e-01, 1.1736e-04, 9.5010e-05,  ..., 7.1045e-02,
           4.0283e-03, 4.1229e-02]],

         [[3.1763e-01, 5.4646e-04, 1.0848e-04,  ..., 1.9470e-02,
           6.2866e-03, 3.3173e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6270e-01, 5.4836e-06, 2.2352e-05,  ..., 1.6928e-03,
           2.4014e-03, 7.2559e-01]],

         [[3.9600e-01, 5.5075e-04, 1.3485e-03,  ..., 4.7668e-02,
           1.0742e-01, 6.7200e-02]],

         [[7.0068e-01, 3.3545e-04, 8.0109e-04,  ..., 2.0065e-02,
           3.1647e-02, 1.4008e-02]],

         ...,

         [[6.0596e-01, 2.1231e-04, 1.8084e-04,  ..., 1.8082e-02,
           7.9651e-02, 8.2855e-03]],

         [[7.5732e-01, 4.9055e-05, 4.6074e-05,  ..., 1.1505e-02,
           6.4758e-02, 6.5430e-02]],

         [[3.2178e-01, 2.8014e-05, 5.6624e-05,  ..., 5.0293e-02,
           2.4071e-03, 4.4043e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2139e-01, 8.2135e-05, 9.2328e-05,  ..., 8.2458e-02,
           4.9774e-02, 2.0654e-01]],

         [[4.9365e-01, 2.2018e-04, 1.4555e-04,  ..., 3.2043e-02,
           1.3293e-01, 1.0551e-02]],

         [[4.5410e-01, 1.3316e-04, 6.7472e-05,  ..., 6.2408e-02,
           3.7659e-02, 2.5360e-02]],

         ...,

         [[4.6600e-02, 2.0266e-06, 9.8348e-06,  ..., 8.3590e-04,
           1.9016e-03, 8.4570e-01]],

         [[5.5420e-01, 8.2493e-05, 9.3460e-05,  ..., 1.6068e-02,
           2.7542e-03, 1.6296e-02]],

         [[5.6348e-01, 6.3777e-05, 7.8678e-06,  ..., 9.3750e-02,
           2.1045e-01, 1.1932e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8066e-01, 1.3371e-03, 8.3685e-04,  ..., 3.3783e-02,
           3.4698e-02, 1.8816e-03]],

         [[4.0991e-01, 1.5736e-04, 3.8648e-04,  ..., 3.0106e-02,
           3.5492e-02, 6.9466e-03]],

         [[2.8369e-01, 2.0921e-05, 7.5698e-05,  ..., 1.0262e-02,
           8.2970e-04, 1.3634e-02]],

         ...,

         [[4.6094e-01, 3.3379e-06, 1.7881e-07,  ..., 5.0354e-02,
           4.6802e-01, 1.8448e-02]],

         [[3.4082e-01, 6.7759e-04, 8.2207e-04,  ..., 1.1795e-02,
           1.8482e-03, 7.6904e-03]],

         [[8.0078e-01, 2.5034e-06, 3.5763e-07,  ..., 3.3020e-02,
           1.3086e-01, 3.3264e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7734e-01, 5.2154e-05, 1.2338e-05,  ..., 2.0584e-02,
           1.3039e-02, 3.4241e-02]],

         [[9.7290e-02, 4.8566e-04, 4.1866e-04,  ..., 2.5463e-04,
           1.5569e-04, 3.4389e-03]],

         [[3.6743e-01, 9.0539e-05, 7.3314e-05,  ..., 5.7869e-03,
           8.8959e-03, 7.5912e-03]],

         ...,

         [[4.5093e-01, 2.0266e-06, 1.6689e-06,  ..., 1.2817e-01,
           3.8177e-02, 2.7863e-02]],

         [[3.1250e-01, 7.1526e-07, 1.1921e-07,  ..., 5.5225e-01,
           3.2349e-02, 6.5041e-03]],

         [[4.7314e-01, 1.6570e-05, 2.0742e-05,  ..., 1.0933e-02,
           7.3166e-03, 6.3210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9011e-02, 8.9407e-07, 4.5300e-06,  ..., 1.2007e-03,
           1.6193e-03, 9.3945e-01]],

         [[1.0059e-01, 6.3777e-06, 1.3471e-05,  ..., 6.2317e-02,
           6.5979e-02, 1.9913e-02]],

         [[2.0544e-01, 3.6550e-04, 7.3385e-04,  ..., 2.2125e-02,
           7.0457e-03, 1.1169e-02]],

         ...,

         [[2.9224e-01, 4.6790e-05, 1.5616e-05,  ..., 1.4026e-01,
           1.8079e-01, 2.7863e-02]],

         [[5.1709e-01, 1.1921e-06, 5.3644e-07,  ..., 1.1681e-02,
           2.0969e-04, 6.9656e-03]],

         [[5.7617e-01, 7.1645e-05, 2.6345e-05,  ..., 3.6865e-02,
           5.6877e-03, 1.1429e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0071e-01, 5.9009e-06, 1.9670e-06,  ..., 3.1128e-01,
           5.3809e-01, 1.4282e-02]],

         [[1.9360e-01, 2.2447e-04, 5.8293e-05,  ..., 4.5074e-02,
           5.9021e-02, 2.8214e-02]],

         [[1.5393e-01, 1.1504e-05, 3.3975e-06,  ..., 2.2705e-02,
           2.2446e-02, 8.2703e-02]],

         ...,

         [[6.9482e-01, 2.6226e-05, 1.4067e-05,  ..., 2.6016e-02,
           8.3008e-02, 1.8448e-02]],

         [[4.4922e-01, 1.3053e-05, 1.0669e-05,  ..., 8.2458e-02,
           8.5388e-02, 3.4912e-02]],

         [[6.2939e-01, 7.2896e-05, 2.5213e-05,  ..., 8.8928e-02,
           6.3538e-02, 2.7237e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9775e-01, 9.7227e-04, 1.3180e-03,  ..., 2.4915e-04,
           1.1265e-04, 7.8630e-04]],

         [[3.2275e-01, 6.2525e-05, 7.4267e-05,  ..., 1.6088e-03,
           1.3866e-03, 1.3908e-02]],

         [[3.8940e-01, 2.7418e-06, 4.8876e-06,  ..., 6.5308e-02,
           4.9500e-02, 1.5884e-02]],

         ...,

         [[2.9102e-01, 2.2030e-04, 4.0233e-05,  ..., 2.4078e-02,
           7.8583e-03, 2.3056e-02]],

         [[4.6069e-01, 7.1824e-05, 6.3777e-05,  ..., 5.8234e-05,
           4.8280e-06, 1.3382e-02]],

         [[7.5000e-01, 1.0866e-04, 1.2052e-04,  ..., 1.7500e-03,
           3.1776e-03, 8.5735e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1248e-02, 4.5955e-05, 4.0233e-05,  ..., 3.6144e-04,
           6.0034e-04, 4.0703e-03]],

         [[1.7676e-01, 3.5763e-04, 2.2125e-04,  ..., 3.8910e-03,
           1.9875e-03, 2.3926e-02]],

         [[6.5491e-02, 7.5698e-06, 4.6492e-06,  ..., 2.7954e-02,
           2.1271e-02, 2.4915e-01]],

         ...,

         [[2.0972e-01, 5.5432e-06, 5.7220e-06,  ..., 3.3051e-02,
           1.0675e-01, 1.4275e-02]],

         [[1.7749e-01, 1.7285e-06, 2.2054e-06,  ..., 3.8159e-01,
           3.1641e-01, 5.6274e-02]],

         [[1.1700e-01, 1.1444e-05, 8.1658e-06,  ..., 1.9409e-02,
           9.9304e-02, 3.3862e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9326e-01, 2.0623e-05, 4.2319e-06,  ..., 3.5614e-02,
           1.7365e-02, 4.3640e-02]],

         [[2.8488e-02, 5.4240e-06, 6.2585e-06,  ..., 3.6011e-02,
           1.0002e-02, 1.3046e-02]],

         [[1.4624e-01, 8.6427e-06, 2.1458e-06,  ..., 3.4409e-03,
           3.1143e-02, 7.5439e-01]],

         ...,

         [[1.3940e-01, 2.3842e-04, 8.8751e-05,  ..., 3.1776e-03,
           2.3422e-03, 1.4351e-02]],

         [[2.4829e-01, 1.7357e-04, 9.5129e-05,  ..., 1.7166e-02,
           1.5747e-02, 3.5217e-02]],

         [[3.9380e-01, 3.7050e-04, 3.0470e-04,  ..., 1.2484e-03,
           7.6008e-04, 1.8597e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4905e-01, 2.4736e-05, 1.5914e-05,  ..., 2.0660e-02,
           2.7359e-02, 4.5471e-02]],

         [[7.0654e-01, 8.8573e-05, 6.8545e-06,  ..., 3.8457e-04,
           4.5052e-03, 2.1782e-03]],

         [[9.2383e-01, 7.3910e-05, 2.7955e-05,  ..., 5.6553e-04,
           9.4748e-04, 1.6117e-03]],

         ...,

         [[1.6028e-01, 1.4086e-03, 2.4338e-03,  ..., 1.3153e-02,
           1.0406e-02, 1.6113e-02]],

         [[3.6206e-01, 7.0333e-06, 9.5367e-06,  ..., 1.7102e-01,
           3.0957e-01, 4.7485e-02]],

         [[3.5522e-01, 6.5947e-04, 1.7242e-03,  ..., 2.1011e-02,
           3.2776e-02, 4.5532e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4180e-01, 9.3937e-05, 1.6952e-04,  ..., 5.1498e-03,
           9.3231e-03, 1.7014e-02]],

         [[2.7222e-01, 1.2875e-04, 2.2602e-04,  ..., 3.3741e-03,
           5.9204e-03, 2.8915e-02]],

         [[2.1790e-01, 2.1338e-05, 4.1306e-05,  ..., 7.0038e-03,
           1.2009e-02, 3.1372e-02]],

         ...,

         [[4.5441e-02, 3.5465e-05, 6.3181e-05,  ..., 2.1708e-04,
           1.4460e-04, 2.2842e-02]],

         [[9.3140e-02, 2.7120e-05, 1.9848e-05,  ..., 2.9011e-03,
           1.3704e-03, 5.3070e-02]],

         [[5.4297e-01, 1.7512e-04, 2.3568e-04,  ..., 2.6199e-02,
           2.7023e-02, 3.0151e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3667e-01, 6.7949e-06, 7.9870e-06,  ..., 1.7029e-02,
           2.8961e-02, 5.0812e-02]],

         [[3.0493e-01, 4.6611e-05, 1.1957e-04,  ..., 1.1215e-03,
           1.8196e-03, 4.5586e-03]],

         [[1.3660e-01, 3.6538e-05, 1.0777e-04,  ..., 5.1308e-03,
           3.6469e-02, 3.8916e-01]],

         ...,

         [[3.6841e-01, 4.2379e-05, 5.4181e-05,  ..., 4.8706e-02,
           1.5002e-01, 1.1237e-01]],

         [[2.3621e-01, 2.6822e-06, 1.4067e-05,  ..., 9.0647e-04,
           3.0851e-04, 6.2561e-02]],

         [[2.9321e-01, 1.3185e-04, 2.2864e-04,  ..., 3.0182e-02,
           3.4729e-02, 4.3144e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2058e-01, 1.0014e-04, 5.6982e-05,  ..., 3.9551e-02,
           4.8828e-02, 5.3192e-02]],

         [[5.5603e-02, 3.0398e-05, 2.0266e-05,  ..., 8.2825e-02,
           6.0577e-02, 4.6448e-02]],

         [[3.0591e-01, 1.1975e-04, 1.4782e-04,  ..., 6.3972e-03,
           1.3390e-02, 2.5513e-02]],

         ...,

         [[5.9570e-01, 7.2122e-06, 2.3842e-07,  ..., 2.2751e-02,
           2.9492e-01, 7.9407e-02]],

         [[1.9958e-01, 2.1040e-05, 3.6418e-05,  ..., 4.8157e-02,
           2.7008e-02, 7.5745e-02]],

         [[3.0737e-01, 2.0564e-05, 1.2577e-05,  ..., 5.7739e-02,
           5.4785e-01, 3.8452e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2392e-03, 8.3447e-07, 2.7418e-06,  ..., 4.8399e-04,
           2.2469e-03, 1.9714e-02]],

         [[2.5040e-02, 1.5497e-06, 4.7088e-06,  ..., 7.9823e-04,
           3.8483e-02, 9.1064e-01]],

         [[2.6962e-02, 2.7239e-05, 1.0455e-04,  ..., 1.8239e-04,
           6.4659e-04, 2.0466e-03]],

         ...,

         [[5.1910e-02, 1.1736e-04, 3.4237e-04,  ..., 7.9956e-03,
           8.0566e-03, 1.1490e-02]],

         [[2.2278e-01, 5.9605e-07, 4.1723e-07,  ..., 1.9089e-02,
           5.6976e-02, 2.8992e-02]],

         [[1.3171e-01, 7.1526e-07, 7.1526e-07,  ..., 2.4695e-01,
           3.2983e-01, 8.0505e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8945e-01, 1.6201e-04, 9.1791e-04,  ..., 1.3977e-02,
           9.3842e-03, 1.3702e-02]],

         [[9.8291e-01, 6.7532e-05, 5.7757e-05,  ..., 3.6645e-04,
           1.8203e-04, 9.9182e-04]],

         [[8.7695e-01, 3.5703e-05, 1.2958e-04,  ..., 1.4290e-02,
           4.3221e-03, 1.1749e-02]],

         ...,

         [[1.4612e-01, 1.4305e-06, 3.4571e-06,  ..., 1.9971e-01,
           2.6465e-01, 5.8105e-02]],

         [[3.9697e-01, 1.9670e-06, 8.9407e-07,  ..., 1.1873e-03,
           2.0676e-03, 1.2836e-03]],

         [[1.3574e-01, 1.1325e-05, 3.5167e-05,  ..., 4.4227e-04,
           1.1497e-02, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1504e-02, 2.8014e-06, 1.2040e-05,  ..., 3.4149e-02,
           4.8157e-02, 4.3152e-02]],

         [[8.0750e-02, 1.0133e-06, 1.1325e-06,  ..., 2.2473e-01,
           6.0156e-01, 4.1077e-02]],

         [[2.4890e-01, 2.1136e-04, 1.9538e-04,  ..., 7.0435e-02,
           1.4062e-01, 7.1838e-02]],

         ...,

         [[3.2007e-01, 2.8014e-06, 1.2279e-05,  ..., 3.1677e-02,
           1.9409e-01, 7.7820e-02]],

         [[6.2402e-01, 1.0848e-05, 3.9697e-05,  ..., 1.6754e-02,
           8.3740e-02, 4.3793e-02]],

         [[4.2773e-01, 5.9605e-06, 2.7716e-05,  ..., 3.4302e-02,
           8.4839e-02, 2.5482e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5221e-02, 3.5763e-06, 1.4305e-06,  ..., 1.6975e-04,
           2.9445e-04, 2.5681e-02]],

         [[2.8491e-01, 6.9141e-06, 2.9206e-06,  ..., 1.0063e-02,
           3.4302e-02, 1.0063e-02]],

         [[2.4829e-01, 4.6372e-05, 1.0163e-04,  ..., 3.7476e-02,
           1.2683e-01, 4.8492e-02]],

         ...,

         [[1.5213e-02, 9.4771e-06, 1.2457e-04,  ..., 3.0112e-04,
           2.0580e-03, 9.4223e-04]],

         [[7.6965e-02, 1.8001e-05, 1.3530e-04,  ..., 1.6844e-04,
           6.1607e-04, 2.9392e-03]],

         [[1.7310e-01, 2.1040e-05, 1.9729e-05,  ..., 1.1597e-02,
           9.7107e-02, 6.3660e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6549e-02, 5.9605e-07, 9.5367e-07,  ..., 8.5742e-01,
           3.6224e-02, 4.2763e-03]],

         [[3.4595e-01, 4.6492e-06, 3.6001e-05,  ..., 4.8370e-03,
           4.6768e-03, 9.3811e-02]],

         [[1.6626e-01, 8.9407e-07, 1.0729e-06,  ..., 9.1064e-02,
           6.7285e-01, 5.0690e-02]],

         ...,

         [[5.7471e-01, 5.1081e-05, 6.2883e-05,  ..., 2.3766e-03,
           5.7487e-03, 1.3206e-02]],

         [[1.5039e-01, 1.7881e-06, 3.2783e-06,  ..., 5.3520e-03,
           5.0278e-03, 5.1460e-03]],

         [[6.0577e-03, 1.0788e-05, 2.4021e-05,  ..., 7.4806e-03,
           1.5381e-01, 7.5732e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0483e-01, 3.0518e-05, 1.3244e-04,  ..., 4.4647e-02,
           2.4512e-01, 5.3009e-02]],

         [[2.8735e-01, 1.4877e-04, 1.8907e-04,  ..., 1.3069e-02,
           8.6746e-03, 1.2375e-02]],

         [[1.3086e-01, 1.9181e-04, 2.9135e-04,  ..., 1.7605e-03,
           8.7967e-03, 3.5686e-03]],

         ...,

         [[5.8887e-01, 6.1321e-04, 3.7766e-04,  ..., 1.0132e-02,
           1.2341e-01, 1.0452e-02]],

         [[9.4299e-02, 1.3745e-04, 4.9496e-04,  ..., 1.8570e-02,
           2.1378e-02, 7.3047e-01]],

         [[6.6895e-01, 1.0252e-05, 2.5749e-05,  ..., 2.8931e-02,
           5.5756e-02, 7.4310e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9854e-01, 4.6432e-05, 4.2772e-04,  ..., 8.4961e-02,
           2.0142e-01, 3.2959e-02]],

         [[4.9774e-02, 3.7551e-06, 1.8299e-05,  ..., 1.3290e-02,
           7.9346e-03, 8.5999e-02]],

         [[4.2798e-01, 3.0684e-04, 7.2050e-04,  ..., 3.9429e-02,
           4.2877e-02, 3.6896e-02]],

         ...,

         [[4.1626e-01, 1.2112e-04, 7.0620e-04,  ..., 2.8038e-03,
           4.8637e-03, 5.8899e-03]],

         [[8.0225e-01, 2.2650e-05, 1.3828e-03,  ..., 4.3564e-03,
           5.8784e-03, 3.8853e-03]],

         [[2.9321e-01, 4.1962e-05, 1.3065e-04,  ..., 3.8376e-03,
           5.0354e-02, 1.6663e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2201e-04, 8.8274e-05, 8.8501e-02,  ..., 1.0023e-03,
           4.3716e-03, 4.7455e-03]],

         [[2.0444e-05, 2.3901e-05, 7.1167e-02,  ..., 1.0460e-02,
           6.8855e-03, 8.5205e-02]],

         [[2.3770e-04, 2.0850e-04, 4.9927e-02,  ..., 2.0981e-02,
           1.4502e-01, 2.7908e-02]],

         ...,

         [[4.7374e-04, 7.7009e-05, 7.7515e-02,  ..., 9.8755e-02,
           1.9800e-01, 1.8591e-01]],

         [[4.1890e-04, 1.7366e-03, 2.3462e-01,  ..., 8.5449e-03,
           1.5717e-02, 3.7632e-03]],

         [[1.9431e-05, 8.7082e-05, 1.6113e-01,  ..., 9.8896e-04,
           8.1301e-04, 1.1740e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5739e-04, 7.9632e-04, 3.0762e-01,  ..., 4.9829e-04,
           2.9240e-03, 2.6302e-03]],

         [[1.7941e-04, 2.7347e-04, 1.0242e-01,  ..., 2.3758e-04,
           1.1092e-04, 1.3723e-03]],

         [[1.6749e-04, 1.0711e-04, 2.8833e-01,  ..., 5.9605e-05,
           2.1172e-03, 7.2598e-05]],

         ...,

         [[4.6277e-04, 4.7374e-04, 2.6440e-01,  ..., 1.1124e-02,
           2.4071e-03, 8.9931e-04]],

         [[2.8670e-05, 1.0788e-04, 2.4902e-01,  ..., 7.4744e-05,
           4.9591e-04, 3.7730e-05]],

         [[8.3387e-05, 3.9220e-04, 1.0626e-01,  ..., 1.0520e-04,
           7.5006e-04, 4.6778e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8203e-04, 1.2445e-04, 1.1975e-01,  ..., 1.0312e-05,
           2.5806e-03, 8.9407e-06]],

         [[7.0333e-04, 3.6895e-05, 2.7393e-01,  ..., 3.2024e-03,
           1.7035e-04, 1.5545e-03]],

         [[8.5449e-04, 7.2956e-04, 2.7441e-01,  ..., 3.6621e-04,
           2.7537e-04, 1.2755e-04]],

         ...,

         [[1.1927e-04, 1.0705e-04, 1.3074e-01,  ..., 9.4681e-03,
           4.8462e-02, 1.1467e-02]],

         [[1.0026e-04, 3.2067e-04, 2.4915e-01,  ..., 5.3358e-04,
           9.8896e-04, 1.8730e-03]],

         [[2.7132e-04, 1.8501e-04, 1.3464e-01,  ..., 1.0767e-03,
           1.2064e-03, 1.2789e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2034e-04, 2.2709e-05, 1.8600e-02,  ..., 8.7402e-02,
           6.3904e-02, 8.4045e-02]],

         [[8.3387e-05, 1.7464e-04, 9.8816e-02,  ..., 7.3700e-03,
           2.2659e-02, 2.4399e-02]],

         [[7.8678e-04, 3.8433e-03, 1.9336e-01,  ..., 1.1158e-03,
           7.6103e-04, 1.7185e-03]],

         ...,

         [[1.7166e-04, 2.8014e-06, 3.1006e-02,  ..., 2.8553e-03,
           5.5218e-04, 2.6520e-02]],

         [[3.0537e-03, 2.9349e-04, 6.2622e-02,  ..., 1.8701e-01,
           2.3096e-01, 5.1331e-02]],

         [[5.7817e-06, 2.4438e-06, 1.5039e-03,  ..., 2.6722e-03,
           1.7109e-03, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2120e-03, 1.3618e-03, 2.2388e-01,  ..., 2.9202e-03,
           3.8185e-03, 4.5135e-02]],

         [[1.2531e-03, 1.3628e-03, 2.5317e-01,  ..., 6.5947e-04,
           1.3628e-03, 2.1698e-02]],

         [[2.8515e-03, 1.4219e-03, 2.7637e-01,  ..., 5.2071e-03,
           9.2239e-03, 7.0457e-03]],

         ...,

         [[1.4629e-03, 1.7965e-04, 2.2766e-01,  ..., 1.7548e-02,
           4.1046e-03, 9.6375e-02]],

         [[3.1972e-04, 8.2922e-04, 1.0864e-01,  ..., 5.3345e-02,
           1.0779e-01, 2.2644e-01]],

         [[3.1590e-05, 3.4273e-05, 1.5747e-02,  ..., 1.1011e-01,
           3.8135e-01, 4.3457e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[7.7546e-05, 5.2166e-04, 6.7635e-03,  ..., 7.7393e-02,
           5.5420e-01, 1.6394e-01]],

         [[1.0413e-04, 5.6314e-04, 7.7705e-03,  ..., 1.3641e-02,
           4.4739e-02, 2.4182e-01]],

         [[0.0000e+00, 1.7881e-07, 2.7061e-05,  ..., 3.4750e-05,
           8.5068e-03, 9.8340e-01]],

         ...,

         [[0.0000e+00, 2.3842e-07, 4.7088e-06,  ..., 2.0146e-04,
           2.4529e-03, 9.8975e-01]],

         [[4.3373e-03, 5.2299e-03, 8.6212e-03,  ..., 1.7151e-02,
           1.5137e-02, 2.2018e-02]],

         [[1.9958e-02, 8.8577e-03, 2.4605e-03,  ..., 2.6443e-02,
           3.5034e-02, 2.9053e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6652e-05, 1.4329e-04, 4.3182e-03,  ..., 1.6556e-02,
           6.5491e-02, 4.1382e-01]],

         [[0.0000e+00, 1.1921e-07, 1.3113e-06,  ..., 1.5173e-01,
           7.7051e-01, 6.8359e-02]],

         [[9.6083e-04, 2.9583e-03, 1.4114e-02,  ..., 7.7972e-03,
           1.0994e-02, 2.2552e-02]],

         ...,

         [[1.3113e-05, 3.0613e-04, 6.3801e-04,  ..., 2.1851e-01,
           4.4849e-01, 2.3999e-01]],

         [[2.7447e-03, 8.3542e-03, 2.5436e-02,  ..., 1.1734e-02,
           7.3433e-03, 9.2850e-03]],

         [[1.6861e-03, 5.7487e-03, 6.7711e-03,  ..., 8.2520e-02,
           2.5970e-02, 2.3636e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1768e-02, 4.8370e-03, 1.3466e-02,  ..., 6.7871e-02,
           4.6997e-02, 5.8502e-02]],

         [[1.3371e-03, 4.9019e-04, 1.9379e-03,  ..., 6.2332e-03,
           6.2418e-04, 3.9291e-03]],

         [[4.6539e-03, 9.3460e-04, 1.0843e-03,  ..., 1.6891e-02,
           1.2260e-02, 1.5259e-02]],

         ...,

         [[4.9257e-04, 2.2087e-03, 3.8452e-03,  ..., 6.9763e-02,
           7.2705e-01, 1.7273e-01]],

         [[3.4165e-04, 4.4823e-05, 4.4107e-05,  ..., 8.3618e-02,
           2.0966e-02, 2.7145e-02]],

         [[3.3951e-03, 9.5034e-04, 1.7380e-02,  ..., 1.8646e-02,
           2.4506e-02, 3.5449e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.7354e-01, 1.8477e-06, 1.4305e-06,  ..., 2.2202e-03,
           7.9346e-03, 1.8740e-03]],

         [[8.6914e-01, 3.3379e-06, 3.6359e-06,  ..., 1.3893e-02,
           2.9297e-02, 7.0620e-04]],

         [[6.0352e-01, 1.9670e-06, 7.9870e-06,  ..., 6.5613e-02,
           2.4147e-03, 1.3293e-01]],

         ...,

         [[4.7803e-01, 2.3949e-04, 8.7357e-04,  ..., 1.0365e-04,
           1.3649e-05, 1.0834e-01]],

         [[4.2871e-01, 5.3024e-04, 6.4993e-04,  ..., 1.8051e-02,
           1.4389e-02, 5.4718e-02]],

         [[4.8560e-01, 4.0245e-04, 8.5354e-04,  ..., 1.2787e-02,
           7.7286e-03, 1.6678e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6377e-01, 3.7432e-04, 9.4175e-06,  ..., 2.7065e-03,
           1.5855e-05, 4.4647e-02]],

         [[9.2920e-01, 9.7871e-05, 1.6689e-06,  ..., 1.5198e-02,
           2.1362e-02, 3.2654e-02]],

         [[9.4434e-01, 2.8300e-04, 2.3818e-04,  ..., 4.0531e-04,
           9.7942e-04, 6.0349e-03]],

         ...,

         [[4.0112e-01, 1.9801e-04, 1.2034e-04,  ..., 1.4877e-02,
           4.2381e-03, 1.7212e-02]],

         [[1.7151e-01, 1.6737e-03, 5.4455e-04,  ..., 5.5199e-03,
           3.7403e-03, 4.8027e-03]],

         [[8.0273e-01, 1.2579e-03, 3.8671e-04,  ..., 6.8115e-02,
           3.4973e-02, 1.5015e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2695e-01, 1.9729e-04, 8.6546e-05,  ..., 1.0858e-01,
           7.5256e-02, 3.2990e-02]],

         [[6.5332e-01, 9.0897e-05, 1.0252e-05,  ..., 7.1594e-02,
           2.3300e-02, 3.9154e-02]],

         [[4.4580e-01, 1.7464e-05, 3.1590e-05,  ..., 2.6779e-02,
           4.7424e-02, 4.5288e-02]],

         ...,

         [[7.9590e-01, 1.1665e-04, 1.1325e-06,  ..., 4.7684e-07,
           1.6809e-05, 1.9702e-01]],

         [[8.6279e-01, 5.3453e-04, 5.6028e-06,  ..., 0.0000e+00,
           2.9802e-07, 3.8391e-02]],

         [[9.7168e-01, 3.8266e-04, 2.3842e-07,  ..., 0.0000e+00,
           5.9605e-08, 8.5754e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 8.9407e-06, 1.8477e-06,  ..., 1.9287e-01,
           4.2700e-01, 1.8784e-02]],

         [[9.4727e-01, 9.5963e-06, 1.8477e-06,  ..., 3.1590e-04,
           6.1750e-04, 1.5167e-02]],

         [[7.7539e-01, 1.3232e-04, 5.0843e-05,  ..., 1.8845e-03,
           4.7836e-03, 1.4397e-02]],

         ...,

         [[8.8672e-01, 2.8992e-04, 2.4199e-05,  ..., 2.3518e-03,
           2.9999e-02, 6.9122e-03]],

         [[6.5674e-01, 4.3392e-04, 6.8247e-05,  ..., 5.0201e-02,
           1.5312e-02, 1.3110e-01]],

         [[8.3984e-01, 8.5592e-05, 7.7665e-05,  ..., 2.2869e-03,
           1.5297e-03, 1.3318e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0732e-01, 8.8215e-06, 3.6955e-06,  ..., 2.2424e-01,
           4.0497e-02, 9.8999e-02]],

         [[6.7578e-01, 4.6194e-05, 1.3113e-05,  ..., 6.6490e-03,
           2.5635e-02, 1.0828e-01]],

         [[7.4658e-01, 6.8545e-06, 1.2517e-06,  ..., 7.7591e-03,
           1.2634e-01, 7.5195e-02]],

         ...,

         [[4.5386e-01, 5.1916e-05, 3.7372e-05,  ..., 1.4221e-01,
           7.3059e-02, 4.7729e-02]],

         [[8.3594e-01, 1.3900e-04, 9.2208e-05,  ..., 1.7502e-02,
           4.4441e-03, 9.5520e-03]],

         [[7.4609e-01, 3.6716e-05, 1.5557e-05,  ..., 6.8970e-02,
           2.1912e-02, 2.4536e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3760e-01, 1.4038e-03, 6.4993e-04,  ..., 4.6005e-03,
           4.3907e-03, 4.5128e-03]],

         [[9.1211e-01, 4.9591e-05, 8.4341e-05,  ..., 5.5771e-03,
           1.2989e-03, 4.1122e-03]],

         [[7.8711e-01, 4.4107e-05, 1.1683e-05,  ..., 5.6488e-02,
           3.0945e-02, 1.2405e-02]],

         ...,

         [[9.1162e-01, 4.0174e-05, 2.0444e-05,  ..., 3.0403e-03,
           2.8458e-03, 1.9852e-02]],

         [[6.4795e-01, 4.1723e-06, 3.3200e-05,  ..., 2.0046e-03,
           5.5389e-03, 7.8003e-02]],

         [[9.6289e-01, 5.6624e-06, 4.5896e-06,  ..., 3.4313e-03,
           2.7905e-03, 7.6981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3184e-01, 1.7512e-04, 7.2575e-04,  ..., 1.1301e-03,
           1.6975e-03, 1.5869e-02]],

         [[8.8477e-01, 1.3769e-05, 1.8537e-05,  ..., 1.8066e-02,
           1.9577e-02, 5.2368e-02]],

         [[4.7827e-01, 2.8610e-05, 3.5810e-04,  ..., 6.6833e-02,
           2.3911e-02, 1.5587e-02]],

         ...,

         [[4.0259e-01, 1.3697e-04, 2.2399e-04,  ..., 1.0204e-03,
           3.1929e-03, 1.4563e-03]],

         [[7.9443e-01, 8.5175e-05, 3.9577e-05,  ..., 1.2039e-02,
           1.1337e-02, 6.7139e-02]],

         [[6.2793e-01, 1.4961e-05, 1.7583e-05,  ..., 7.1678e-03,
           1.0109e-03, 1.8848e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2471e-01, 1.0419e-04, 1.4985e-04,  ..., 3.8028e-05,
           4.5896e-06, 1.1719e-02]],

         [[8.6084e-01, 2.2888e-05, 3.6538e-05,  ..., 9.7275e-03,
           5.7983e-03, 1.5366e-02]],

         [[8.9160e-01, 1.5318e-05, 2.2471e-05,  ..., 6.0349e-03,
           7.9823e-04, 1.1253e-02]],

         ...,

         [[9.2969e-01, 7.7367e-05, 5.0545e-05,  ..., 3.0117e-03,
           3.2959e-03, 7.5722e-03]],

         [[8.0762e-01, 1.2183e-04, 1.2088e-04,  ..., 9.9277e-04,
           2.4529e-03, 2.6302e-03]],

         [[6.2891e-01, 9.8324e-04, 3.1304e-04,  ..., 2.9011e-03,
           3.3321e-03, 1.0254e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2290e-01, 6.5565e-07, 1.3828e-05,  ..., 7.4482e-04,
           1.5137e-02, 7.5391e-01]],

         [[8.8281e-01, 3.3212e-04, 5.7316e-04,  ..., 3.6793e-03,
           3.1643e-03, 7.0572e-04]],

         [[7.1191e-01, 5.1641e-04, 7.3957e-04,  ..., 1.0872e-02,
           1.7685e-02, 1.3580e-02]],

         ...,

         [[6.3037e-01, 9.6321e-05, 1.9836e-04,  ..., 6.8436e-03,
           7.3166e-03, 1.9951e-03]],

         [[7.7197e-01, 7.7605e-05, 3.5226e-05,  ..., 1.2138e-02,
           5.7678e-02, 5.1849e-02]],

         [[2.4426e-01, 1.2922e-04, 1.8871e-04,  ..., 2.2812e-03,
           1.9821e-02, 5.9521e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3711e-01, 3.7193e-05, 4.2498e-05,  ..., 6.8321e-03,
           1.3245e-02, 3.8647e-01]],

         [[7.3828e-01, 9.5189e-05, 1.1486e-04,  ..., 9.5978e-03,
           1.9852e-02, 7.1045e-02]],

         [[3.0469e-01, 7.9775e-04, 1.4472e-04,  ..., 3.6530e-02,
           4.7760e-02, 1.9699e-02]],

         ...,

         [[9.5215e-02, 3.4571e-06, 7.9870e-06,  ..., 1.3947e-05,
           2.1279e-04, 8.9795e-01]],

         [[8.1445e-01, 6.5804e-05, 5.6922e-05,  ..., 7.8726e-04,
           5.0545e-03, 3.3142e-02]],

         [[9.1016e-01, 1.4532e-04, 3.9577e-05,  ..., 1.7319e-02,
           1.9684e-03, 1.1055e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2695e-01, 1.4067e-03, 1.4009e-03,  ..., 1.6510e-02,
           3.4885e-03, 2.7649e-02]],

         [[8.5840e-01, 7.4863e-05, 1.1772e-04,  ..., 6.0272e-03,
           5.1231e-03, 3.3905e-02]],

         [[6.4600e-01, 3.0971e-04, 1.1003e-04,  ..., 3.3512e-03,
           1.4696e-03, 1.0208e-02]],

         ...,

         [[6.1768e-01, 5.1498e-05, 8.6427e-06,  ..., 1.6953e-02,
           3.0103e-01, 6.1218e-02]],

         [[3.8940e-01, 5.2071e-04, 5.6982e-04,  ..., 5.5962e-03,
           1.4549e-02, 1.4629e-03]],

         [[8.1152e-01, 4.1068e-05, 1.4007e-05,  ..., 2.1667e-02,
           1.4062e-01, 2.3407e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5439e-01, 7.7784e-05, 1.5914e-05,  ..., 5.7755e-03,
           2.1820e-02, 1.8723e-02]],

         [[8.1360e-02, 1.7233e-03, 1.5736e-03,  ..., 1.6856e-04,
           1.7166e-03, 4.7455e-03]],

         [[6.2256e-01, 2.5249e-04, 2.1517e-04,  ..., 2.6779e-03,
           1.1047e-02, 2.6215e-02]],

         ...,

         [[4.6875e-01, 1.8775e-05, 5.3048e-06,  ..., 3.5583e-02,
           7.1350e-02, 1.4380e-01]],

         [[7.4414e-01, 7.1526e-06, 2.4438e-06,  ..., 2.5055e-02,
           5.1483e-02, 7.8674e-02]],

         [[3.4204e-01, 5.6803e-05, 5.9783e-05,  ..., 7.2060e-03,
           1.9516e-02, 3.5736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0444e-02, 1.1921e-07, 4.7684e-07,  ..., 1.6439e-04,
           2.2697e-03, 9.1016e-01]],

         [[3.5132e-01, 2.9325e-05, 4.9710e-05,  ..., 6.6650e-02,
           9.0210e-02, 8.2642e-02]],

         [[1.8896e-01, 4.5991e-04, 7.5245e-04,  ..., 6.6910e-03,
           1.8173e-02, 3.3020e-02]],

         ...,

         [[4.3311e-01, 1.7190e-04, 1.3340e-04,  ..., 8.5938e-02,
           9.1309e-02, 6.3553e-03]],

         [[4.3311e-01, 7.6294e-06, 1.4305e-06,  ..., 1.5710e-01,
           2.6260e-02, 3.5858e-02]],

         [[1.7578e-01, 1.2076e-04, 7.2122e-06,  ..., 4.1771e-03,
           2.9144e-03, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2646e-01, 2.5332e-05, 2.1636e-05,  ..., 7.7515e-02,
           1.4877e-02, 4.0039e-02]],

         [[2.3157e-01, 1.1045e-04, 4.2319e-05,  ..., 3.1543e-01,
           8.7585e-02, 2.6291e-02]],

         [[7.1436e-01, 7.4267e-05, 9.0659e-05,  ..., 1.0071e-02,
           1.5594e-02, 4.6753e-02]],

         ...,

         [[8.7256e-01, 3.8564e-05, 1.6272e-05,  ..., 2.2186e-02,
           2.0279e-02, 7.8201e-03]],

         [[8.1738e-01, 2.6643e-05, 5.4896e-05,  ..., 2.9716e-03,
           7.5722e-03, 4.4861e-02]],

         [[5.3271e-01, 1.0490e-04, 3.1650e-05,  ..., 1.0327e-01,
           1.1084e-01, 1.0391e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5747e-01, 8.1015e-04, 1.1702e-03,  ..., 4.0650e-04,
           4.7779e-04, 1.9970e-03]],

         [[1.5698e-01, 1.2231e-04, 1.0985e-04,  ..., 6.5088e-04,
           7.1831e-03, 3.9307e-02]],

         [[6.6260e-01, 1.4126e-05, 3.7968e-05,  ..., 2.5101e-02,
           7.7576e-02, 2.4796e-02]],

         ...,

         [[8.3789e-01, 6.9022e-05, 1.6749e-05,  ..., 4.0588e-03,
           8.5926e-04, 2.2720e-02]],

         [[9.2578e-01, 2.6643e-05, 1.7703e-05,  ..., 4.7684e-07,
           4.7684e-07, 1.5495e-02]],

         [[5.9473e-01, 2.2876e-04, 4.1580e-04,  ..., 2.8992e-02,
           1.3115e-02, 5.6915e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8196e-02, 1.7571e-04, 1.0574e-04,  ..., 4.3907e-03,
           1.3847e-02, 2.3376e-02]],

         [[1.5723e-01, 3.4380e-04, 1.5497e-04,  ..., 5.1308e-03,
           4.6814e-02, 4.7180e-02]],

         [[5.1562e-01, 4.2915e-05, 2.7597e-05,  ..., 1.1215e-02,
           1.5091e-02, 1.8530e-01]],

         ...,

         [[2.5732e-01, 5.1260e-06, 7.6294e-06,  ..., 2.2180e-01,
           7.1472e-02, 6.1722e-03]],

         [[7.0557e-01, 9.0599e-06, 6.3777e-06,  ..., 1.8799e-02,
           2.7145e-02, 1.2854e-01]],

         [[1.9397e-01, 1.2398e-05, 8.9407e-06,  ..., 2.4460e-02,
           5.3436e-02, 6.5088e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-01, 6.2168e-05, 1.4305e-05,  ..., 1.1780e-01,
           4.9103e-02, 2.6276e-02]],

         [[2.0850e-01, 3.5167e-05, 6.6757e-05,  ..., 4.5105e-02,
           1.0651e-01, 4.8004e-02]],

         [[2.5708e-01, 1.6987e-05, 3.5763e-06,  ..., 1.3494e-03,
           3.3913e-03, 7.2119e-01]],

         ...,

         [[7.8979e-02, 2.2447e-04, 2.1219e-05,  ..., 2.4498e-05,
           7.2002e-05, 5.2612e-02]],

         [[3.4521e-01, 8.4400e-05, 3.7491e-05,  ..., 1.2138e-02,
           2.5009e-02, 4.8294e-03]],

         [[1.4075e-01, 7.4744e-05, 4.2498e-05,  ..., 1.9501e-02,
           6.5979e-02, 1.4725e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5254e-01, 2.6166e-05, 1.4067e-05,  ..., 2.2369e-02,
           3.4637e-02, 3.8635e-02]],

         [[9.7803e-01, 4.9531e-05, 1.1027e-05,  ..., 5.1260e-06,
           5.0068e-06, 1.7560e-04]],

         [[8.3350e-01, 2.6679e-04, 1.6630e-04,  ..., 2.7943e-04,
           1.1864e-03, 3.2234e-03]],

         ...,

         [[9.8999e-02, 6.6710e-04, 9.6321e-04,  ..., 2.7710e-02,
           1.9501e-02, 3.6407e-02]],

         [[6.5820e-01, 3.2842e-05, 3.2604e-05,  ..., 3.0792e-02,
           3.8940e-02, 2.4750e-02]],

         [[2.6904e-01, 5.6362e-04, 1.0958e-03,  ..., 3.8452e-02,
           4.7485e-02, 3.9062e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9175e-01, 1.2279e-04, 1.0175e-04,  ..., 8.3557e-02,
           8.8989e-02, 2.1973e-02]],

         [[2.7563e-01, 7.7844e-05, 1.1969e-04,  ..., 8.2626e-03,
           1.3939e-02, 7.1899e-02]],

         [[2.1960e-01, 4.5002e-05, 1.0633e-04,  ..., 1.1253e-03,
           3.5496e-03, 1.1543e-02]],

         ...,

         [[3.2227e-02, 2.3842e-04, 2.6608e-04,  ..., 6.5851e-04,
           7.1907e-03, 2.2842e-02]],

         [[8.0200e-02, 2.4140e-05, 1.0371e-05,  ..., 5.1856e-05,
           4.7982e-05, 1.6980e-01]],

         [[4.9634e-01, 1.3280e-04, 1.1903e-04,  ..., 1.3863e-02,
           8.4076e-03, 2.2858e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0391e-01, 3.5286e-05, 2.6941e-05,  ..., 1.1574e-02,
           2.0950e-02, 3.3478e-02]],

         [[2.1912e-01, 4.4048e-05, 7.2062e-05,  ..., 4.8790e-03,
           2.2034e-02, 8.4305e-03]],

         [[4.0649e-01, 9.2268e-05, 1.7238e-04,  ..., 3.4084e-03,
           2.3483e-02, 2.8833e-01]],

         ...,

         [[4.0430e-01, 8.1122e-05, 6.2943e-05,  ..., 4.1962e-02,
           1.0201e-02, 1.9821e-02]],

         [[7.8320e-01, 8.6010e-05, 7.8321e-05,  ..., 7.0155e-05,
           1.0782e-04, 4.0558e-02]],

         [[7.1143e-01, 7.2777e-05, 2.1470e-04,  ..., 9.5291e-03,
           3.2673e-03, 1.4198e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6953e-01, 1.6880e-04, 6.9976e-05,  ..., 2.8061e-02,
           4.0680e-02, 5.3894e-02]],

         [[2.8516e-01, 1.6117e-04, 1.5807e-04,  ..., 1.0490e-03,
           2.4338e-02, 8.3618e-02]],

         [[4.2334e-01, 1.8525e-04, 1.0681e-04,  ..., 3.8177e-02,
           3.4760e-02, 1.6937e-02]],

         ...,

         [[5.2637e-01, 8.6069e-05, 1.5438e-05,  ..., 5.4199e-02,
           3.5083e-01, 5.6366e-02]],

         [[2.9395e-01, 1.3053e-05, 4.6730e-05,  ..., 8.4229e-02,
           2.5732e-01, 1.0236e-01]],

         [[6.1377e-01, 1.4496e-04, 8.3566e-05,  ..., 5.0018e-02,
           1.8433e-01, 1.3599e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4399e-02, 4.6492e-06, 1.2398e-05,  ..., 8.8043e-03,
           9.8022e-02, 2.6382e-02]],

         [[3.4839e-01, 8.5175e-05, 1.3888e-04,  ..., 1.2253e-02,
           2.4094e-02, 3.0762e-01]],

         [[7.5989e-02, 2.7120e-05, 1.3745e-04,  ..., 3.1376e-03,
           1.0406e-02, 2.3727e-02]],

         ...,

         [[1.0077e-01, 1.6022e-04, 5.5075e-04,  ..., 8.7357e-03,
           1.9089e-02, 1.5396e-02]],

         [[4.8926e-01, 7.2718e-06, 2.2709e-05,  ..., 5.1361e-02,
           1.3013e-01, 1.7236e-01]],

         [[3.1299e-01, 2.9206e-06, 7.8678e-06,  ..., 8.4900e-02,
           3.5596e-01, 1.6101e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 7.9751e-05, 3.5524e-04,  ..., 4.8248e-02,
           3.4485e-02, 3.7262e-02]],

         [[9.6045e-01, 2.1219e-05, 1.1760e-04,  ..., 5.5027e-04,
           1.7557e-03, 1.2798e-03]],

         [[8.8525e-01, 1.6856e-04, 2.3866e-04,  ..., 6.8665e-03,
           1.3130e-02, 7.3662e-03]],

         ...,

         [[8.5400e-01, 7.9274e-06, 1.2279e-05,  ..., 3.9337e-02,
           1.9165e-02, 2.6611e-02]],

         [[9.1309e-01, 4.3511e-06, 4.4107e-06,  ..., 1.3094e-03,
           8.3256e-04, 6.2485e-03]],

         [[2.7930e-01, 1.0490e-05, 2.3305e-05,  ..., 4.2725e-03,
           2.5574e-02, 6.1963e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3196e-01, 5.4240e-06, 2.4319e-05,  ..., 4.1534e-02,
           7.8857e-02, 2.7295e-01]],

         [[7.2070e-01, 2.5868e-05, 1.6987e-05,  ..., 1.0065e-01,
           1.9440e-02, 1.6632e-02]],

         [[1.2988e-01, 1.3173e-04, 7.8261e-05,  ..., 9.2102e-02,
           6.2561e-02, 3.3234e-02]],

         ...,

         [[6.4307e-01, 1.3530e-05, 5.1439e-05,  ..., 2.6123e-02,
           4.7699e-02, 4.3762e-02]],

         [[4.9634e-01, 1.7166e-05, 6.6996e-05,  ..., 1.3159e-01,
           9.9304e-02, 3.1006e-02]],

         [[7.2998e-01, 8.7023e-06, 4.0472e-05,  ..., 4.1199e-02,
           1.9455e-02, 3.1082e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9785e-01, 1.2279e-05, 3.5167e-06,  ..., 5.9433e-03,
           2.8580e-02, 2.2656e-01]],

         [[6.0938e-01, 9.1195e-06, 2.5034e-06,  ..., 4.2725e-03,
           7.2765e-04, 1.1188e-01]],

         [[2.9443e-01, 5.8830e-05, 5.4955e-05,  ..., 3.4332e-02,
           3.3813e-02, 1.5039e-01]],

         ...,

         [[1.0437e-01, 2.5630e-06, 5.5492e-05,  ..., 8.0017e-02,
           2.0874e-02, 6.9519e-02]],

         [[1.2213e-01, 3.8147e-06, 3.2902e-05,  ..., 4.3549e-02,
           3.4454e-02, 4.0894e-02]],

         [[8.4473e-01, 1.7583e-05, 2.7001e-05,  ..., 1.1147e-02,
           1.3229e-02, 1.1864e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3887e-01, 2.8014e-06, 9.5367e-06,  ..., 1.4183e-02,
           2.0920e-02, 5.3223e-02]],

         [[5.3027e-01, 4.8935e-05, 2.2078e-04,  ..., 1.9394e-02,
           1.8909e-01, 9.0027e-02]],

         [[6.5332e-01, 1.6928e-05, 3.1531e-05,  ..., 1.2077e-02,
           1.0748e-01, 2.1875e-01]],

         ...,

         [[4.2505e-01, 2.5034e-05, 1.9252e-05,  ..., 2.8702e-02,
           6.2225e-02, 4.3678e-03]],

         [[5.6543e-01, 1.2755e-05, 6.8545e-06,  ..., 2.1408e-02,
           2.7481e-02, 1.1572e-01]],

         [[1.6833e-01, 9.6262e-05, 3.1567e-04,  ..., 9.5947e-02,
           5.7281e-02, 2.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3037e-01, 7.4506e-06, 1.2815e-05,  ..., 1.1955e-02,
           3.6888e-03, 2.9541e-01]],

         [[7.1582e-01, 9.1612e-05, 7.3314e-05,  ..., 1.8673e-03,
           1.1024e-03, 1.0980e-01]],

         [[4.8779e-01, 2.7239e-05, 6.2346e-05,  ..., 6.0692e-03,
           1.2817e-03, 1.7944e-01]],

         ...,

         [[5.8301e-01, 1.0383e-04, 3.2496e-04,  ..., 5.4230e-02,
           1.9028e-02, 1.0785e-01]],

         [[5.5908e-01, 3.6144e-04, 1.0796e-03,  ..., 1.5617e-02,
           3.4119e-02, 2.1228e-01]],

         [[6.0840e-01, 5.9903e-05, 1.2290e-04,  ..., 1.8372e-02,
           3.4332e-02, 2.4207e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8262e-01, 3.4392e-05, 2.5082e-04,  ..., 1.2527e-02,
           9.9121e-02, 7.5378e-02]],

         [[2.2961e-01, 1.9670e-06, 3.2127e-05,  ..., 5.0735e-03,
           5.1613e-03, 4.4604e-01]],

         [[2.5708e-01, 1.1420e-04, 7.8201e-04,  ..., 1.1060e-01,
           1.3818e-01, 7.2815e-02]],

         ...,

         [[8.3154e-01, 7.1645e-05, 4.1699e-04,  ..., 1.4847e-02,
           1.0925e-02, 4.4060e-03]],

         [[9.5068e-01, 1.7881e-07, 1.5771e-04,  ..., 5.5170e-04,
           1.2836e-03, 7.9489e-04]],

         [[7.3730e-01, 1.6868e-05, 6.4254e-05,  ..., 1.6632e-02,
           1.0010e-02, 8.0795e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1100e-05, 1.1921e-05, 2.1741e-01,  ..., 3.0785e-03,
           7.2098e-03, 5.3284e-02]],

         [[7.0333e-06, 5.0664e-06, 2.0737e-02,  ..., 3.8643e-03,
           1.2402e-01, 5.6229e-03]],

         [[1.5116e-04, 1.3387e-04, 6.7078e-02,  ..., 2.9190e-02,
           3.0838e-02, 1.4626e-02]],

         ...,

         [[5.1796e-05, 9.6560e-06, 1.4612e-01,  ..., 1.4816e-02,
           1.5320e-01, 3.1421e-01]],

         [[2.0134e-04, 4.1902e-05, 3.0518e-01,  ..., 6.0692e-03,
           7.5531e-03, 1.3733e-02]],

         [[1.1206e-05, 1.2517e-06, 2.4475e-01,  ..., 7.4911e-04,
           7.8535e-04, 6.4880e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8413e-05, 1.1504e-05, 2.5977e-01,  ..., 1.1806e-03,
           2.8877e-03, 2.9617e-02]],

         [[3.0518e-05, 8.6427e-06, 2.7222e-01,  ..., 1.2636e-04,
           5.1975e-04, 3.2257e-02]],

         [[2.1875e-05, 3.1590e-06, 3.2251e-01,  ..., 1.3866e-03,
           8.3745e-05, 1.4305e-03]],

         ...,

         [[6.8367e-05, 1.4424e-05, 9.4421e-02,  ..., 2.3365e-03,
           1.2503e-03, 5.4131e-03]],

         [[3.6955e-06, 2.0266e-06, 2.8516e-01,  ..., 5.4836e-04,
           2.1815e-04, 7.3357e-03]],

         [[1.2815e-05, 1.0431e-05, 2.9712e-01,  ..., 1.0931e-04,
           7.1406e-05, 5.2643e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0192e-05, 1.7881e-06, 2.0081e-01,  ..., 1.9045e-03,
           1.2219e-05, 3.0914e-02]],

         [[1.7643e-05, 2.9802e-07, 2.3828e-01,  ..., 5.4169e-04,
           1.0738e-03, 1.3514e-03]],

         [[2.0385e-05, 4.0531e-06, 2.2229e-01,  ..., 4.7231e-04,
           1.6069e-04, 3.3112e-03]],

         ...,

         [[2.5868e-04, 3.0518e-05, 2.1802e-01,  ..., 2.3697e-02,
           1.0765e-02, 1.3428e-01]],

         [[8.3447e-06, 2.9206e-06, 1.0846e-01,  ..., 1.0490e-03,
           6.8521e-04, 1.7227e-02]],

         [[3.6836e-05, 2.9206e-06, 2.0776e-01,  ..., 1.5678e-03,
           6.4611e-04, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.8651e-05, 4.2319e-06, 2.8549e-02,  ..., 5.9814e-03,
           1.2466e-02, 6.4941e-01]],

         [[8.1658e-06, 3.3498e-05, 1.3940e-01,  ..., 5.3131e-02,
           7.1228e-02, 1.1200e-02]],

         [[9.1612e-05, 2.7108e-04, 3.2861e-01,  ..., 2.4438e-04,
           4.7898e-04, 1.0252e-04]],

         ...,

         [[8.0061e-04, 1.3232e-05, 1.8530e-01,  ..., 8.1158e-04,
           4.5586e-04, 5.0926e-03]],

         [[4.5848e-04, 1.5616e-05, 9.2773e-02,  ..., 1.1040e-02,
           1.4915e-02, 5.5518e-01]],

         [[3.0351e-04, 1.2279e-05, 7.6172e-02,  ..., 1.4481e-02,
           1.0269e-02, 4.5215e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6011e-04, 1.4663e-04, 1.1298e-01,  ..., 5.1544e-02,
           1.4999e-02, 6.4392e-02]],

         [[7.7629e-04, 4.7827e-04, 2.4707e-01,  ..., 5.7869e-03,
           2.0844e-02, 6.9153e-02]],

         [[3.4523e-04, 9.9242e-05, 1.8079e-01,  ..., 1.3695e-02,
           1.2421e-02, 5.5786e-02]],

         ...,

         [[1.1581e-04, 4.4286e-05, 1.6821e-01,  ..., 3.6957e-02,
           8.5938e-02, 1.5808e-01]],

         [[3.6407e-04, 3.9196e-04, 2.7759e-01,  ..., 1.1276e-02,
           1.0996e-03, 7.8308e-02]],

         [[5.2166e-04, 1.4377e-04, 2.2412e-01,  ..., 1.9882e-02,
           3.4912e-02, 1.6785e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.6028e-06, 8.2493e-05, 6.6948e-04,  ..., 7.7393e-02,
           4.0552e-01, 4.0552e-01]],

         [[3.4475e-04, 2.2256e-04, 3.7060e-03,  ..., 4.4708e-03,
           2.9144e-02, 4.0234e-01]],

         [[5.9605e-08, 5.9605e-08, 8.9347e-05,  ..., 3.2842e-05,
           2.9583e-03, 9.2969e-01]],

         ...,

         [[0.0000e+00, 4.7684e-07, 6.0201e-06,  ..., 2.7120e-05,
           2.0027e-04, 9.8438e-01]],

         [[3.1185e-03, 2.0123e-03, 7.4806e-03,  ..., 1.5678e-03,
           4.8294e-03, 1.8115e-01]],

         [[1.0628e-02, 3.9101e-03, 5.6314e-04,  ..., 1.1426e-01,
           7.8491e-02, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3167e-04, 1.4906e-03, 1.2283e-02,  ..., 5.5939e-02,
           1.2408e-01, 1.0126e-01]],

         [[0.0000e+00, 0.0000e+00, 3.5763e-07,  ..., 8.5022e-02,
           8.4570e-01, 6.2225e-02]],

         [[3.8319e-03, 5.9357e-03, 1.6388e-02,  ..., 1.0414e-02,
           2.1042e-02, 1.4694e-02]],

         ...,

         [[4.2439e-04, 3.0632e-03, 4.4556e-03,  ..., 3.7988e-01,
           4.7180e-02, 5.7800e-02]],

         [[4.5204e-03, 1.3763e-02, 3.3661e-02,  ..., 1.2100e-02,
           1.1459e-02, 8.7509e-03]],

         [[2.0866e-03, 1.0353e-02, 9.4223e-03,  ..., 1.1481e-01,
           7.3547e-02, 5.3406e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7292e-02, 3.9597e-03, 1.6541e-02,  ..., 3.9093e-02,
           7.6538e-02, 5.0964e-02]],

         [[2.1992e-03, 6.2513e-04, 2.7905e-03,  ..., 8.0919e-04,
           3.9520e-03, 3.8891e-03]],

         [[3.4599e-03, 6.0606e-04, 6.7902e-04,  ..., 1.2367e-02,
           1.4572e-02, 1.3901e-02]],

         ...,

         [[6.3300e-05, 1.5526e-03, 1.8463e-02,  ..., 2.1301e-01,
           4.6167e-01, 2.1301e-01]],

         [[1.1063e-03, 2.5964e-04, 4.3321e-04,  ..., 6.6345e-02,
           3.8391e-02, 1.9608e-02]],

         [[2.7752e-03, 8.6641e-04, 8.8882e-03,  ..., 3.5156e-02,
           3.8696e-01, 1.5881e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3203e-01, 1.1325e-06, 1.6689e-06,  ..., 5.0354e-03,
           3.4389e-03, 6.1073e-03]],

         [[9.4727e-01, 4.8876e-06, 5.6028e-06,  ..., 1.9638e-02,
           1.0666e-02, 4.7226e-03]],

         [[4.3311e-01, 1.0729e-06, 1.1921e-06,  ..., 4.9324e-03,
           7.7324e-03, 5.5580e-03]],

         ...,

         [[6.4990e-01, 1.9169e-04, 1.3828e-03,  ..., 5.2929e-04,
           8.0538e-04, 2.7252e-02]],

         [[5.9766e-01, 1.0711e-04, 2.3782e-04,  ..., 1.9363e-02,
           7.3914e-02, 9.9258e-03]],

         [[5.6787e-01, 2.2542e-04, 4.7064e-04,  ..., 1.1162e-02,
           1.0144e-01, 2.1255e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2163e-01, 6.6161e-06, 6.1989e-06,  ..., 2.6155e-04,
           1.3208e-03, 1.9806e-02]],

         [[1.0132e-01, 3.6955e-06, 5.3048e-06,  ..., 2.3346e-03,
           8.8770e-01, 8.0795e-03]],

         [[9.2773e-01, 7.5519e-05, 4.9174e-05,  ..., 1.5173e-03,
           4.6272e-03, 2.9354e-03]],

         ...,

         [[3.0200e-01, 1.2779e-04, 1.1212e-04,  ..., 1.1284e-02,
           9.7656e-02, 1.4717e-02]],

         [[1.6919e-01, 5.2071e-04, 5.0449e-04,  ..., 4.9210e-03,
           4.3259e-03, 2.2354e-03]],

         [[7.8320e-01, 2.5153e-04, 5.1069e-04,  ..., 1.3084e-02,
           1.1084e-01, 7.4921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0569e-01, 2.3878e-04, 9.8765e-05,  ..., 7.6355e-02,
           7.7271e-02, 1.9211e-02]],

         [[6.8555e-01, 7.3135e-05, 4.0233e-05,  ..., 3.1525e-02,
           1.8518e-01, 1.4923e-02]],

         [[7.9785e-01, 3.6120e-05, 1.6689e-05,  ..., 1.2009e-02,
           1.5182e-02, 2.5215e-03]],

         ...,

         [[8.3643e-01, 1.1921e-05, 3.4571e-06,  ..., 2.0504e-05,
           5.6982e-05, 1.5857e-01]],

         [[9.5947e-01, 9.1672e-05, 1.3554e-04,  ..., 5.9605e-08,
           4.1723e-07, 1.9989e-02]],

         [[9.8242e-01, 3.8505e-05, 4.4703e-06,  ..., 1.7881e-07,
           1.1921e-07, 1.1383e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9568e-01, 2.9206e-06, 6.5565e-07,  ..., 6.1157e-02,
           6.7383e-01, 2.0203e-02]],

         [[8.7988e-01, 4.7088e-06, 1.4305e-06,  ..., 4.3106e-03,
           3.4237e-03, 5.0011e-03]],

         [[7.0557e-01, 5.1403e-04, 3.3450e-04,  ..., 8.5907e-03,
           9.3460e-03, 1.7960e-02]],

         ...,

         [[9.0527e-01, 5.1737e-05, 1.4365e-05,  ..., 2.4673e-02,
           3.2234e-03, 1.7380e-02]],

         [[5.3809e-01, 2.7657e-05, 2.1696e-05,  ..., 1.0948e-02,
           3.6621e-01, 2.3941e-02]],

         [[4.9268e-01, 5.8830e-05, 6.8784e-05,  ..., 6.7902e-03,
           4.5117e-01, 8.9951e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.1045e-01, 8.1658e-06, 9.1791e-06,  ..., 4.7180e-02,
           4.5837e-02, 4.7684e-03]],

         [[9.0771e-01, 4.6074e-05, 2.4915e-05,  ..., 1.6144e-02,
           1.5327e-02, 6.5613e-03]],

         [[9.5898e-01, 1.6689e-05, 4.6492e-06,  ..., 3.7632e-03,
           2.7985e-02, 1.5936e-03]],

         ...,

         [[7.1240e-01, 2.1446e-04, 7.3850e-05,  ..., 1.5244e-02,
           3.5034e-02, 5.3101e-03]],

         [[8.8232e-01, 1.6189e-04, 2.8133e-05,  ..., 3.0308e-03,
           9.6664e-03, 1.1368e-03]],

         [[9.0137e-01, 1.5616e-05, 1.1861e-05,  ..., 7.4158e-03,
           4.1962e-02, 3.9139e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6069e-01, 9.6369e-04, 6.7282e-04,  ..., 7.4043e-03,
           4.5815e-03, 2.5005e-03]],

         [[9.3799e-01, 7.1406e-05, 8.4162e-05,  ..., 9.1505e-04,
           9.9754e-04, 9.4032e-04]],

         [[8.5352e-01, 4.5896e-05, 1.6510e-05,  ..., 1.0323e-02,
           9.7198e-03, 3.2692e-03]],

         ...,

         [[6.8066e-01, 1.8120e-05, 2.3961e-05,  ..., 6.1989e-03,
           1.6800e-02, 2.2632e-01]],

         [[6.7139e-01, 1.6689e-06, 7.0333e-06,  ..., 1.0056e-02,
           2.7393e-01, 1.8906e-02]],

         [[9.5312e-01, 1.2100e-05, 1.4901e-05,  ..., 3.8013e-03,
           1.0956e-02, 8.8501e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8127e-01, 3.9005e-04, 1.0624e-03,  ..., 1.2306e-02,
           4.2755e-02, 3.8719e-03]],

         [[7.2070e-01, 1.3113e-06, 3.5167e-06,  ..., 9.3231e-03,
           2.4719e-01, 1.0162e-02]],

         [[8.3154e-01, 1.2279e-05, 2.6941e-05,  ..., 1.0445e-02,
           1.5282e-02, 5.8708e-03]],

         ...,

         [[6.9580e-01, 2.4939e-04, 2.6298e-04,  ..., 4.7646e-03,
           2.6798e-03, 5.3120e-04]],

         [[7.7930e-01, 1.3828e-04, 7.6354e-05,  ..., 1.0803e-02,
           4.7058e-02, 1.6165e-03]],

         [[1.7426e-02, 3.5763e-07, 2.9802e-07,  ..., 4.4489e-04,
           9.7363e-01, 6.6147e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3301e-01, 5.4061e-05, 5.5313e-05,  ..., 4.0472e-05,
           3.9935e-04, 5.3024e-03]],

         [[8.5449e-01, 3.0339e-05, 3.6895e-05,  ..., 7.9575e-03,
           4.5166e-02, 1.0786e-03]],

         [[9.3213e-01, 2.0862e-06, 9.1195e-06,  ..., 3.5248e-03,
           1.9236e-03, 4.6883e-03]],

         ...,

         [[9.5801e-01, 9.9719e-05, 8.1062e-05,  ..., 2.5578e-03,
           3.9978e-03, 3.5501e-04]],

         [[3.2300e-01, 1.3781e-04, 1.0484e-04,  ..., 2.0714e-03,
           1.6693e-02, 5.8057e-01]],

         [[7.0947e-01, 2.6870e-04, 1.0985e-04,  ..., 4.9248e-03,
           2.8473e-02, 8.7433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6907e-01, 2.9802e-07, 2.3842e-06,  ..., 9.3746e-04,
           4.7569e-03, 8.1885e-01]],

         [[7.5781e-01, 2.2495e-04, 3.2377e-04,  ..., 4.5509e-03,
           7.4196e-04, 5.0354e-04]],

         [[6.9189e-01, 7.5340e-04, 1.3399e-03,  ..., 1.4023e-02,
           1.2016e-02, 1.7757e-03]],

         ...,

         [[4.0674e-01, 1.9670e-04, 2.9063e-04,  ..., 2.0187e-02,
           4.0512e-03, 2.5654e-03]],

         [[9.0674e-01, 4.5359e-05, 4.0650e-05,  ..., 4.4365e-03,
           1.6647e-02, 1.6022e-02]],

         [[1.2445e-01, 3.3557e-05, 3.3021e-05,  ..., 2.8057e-03,
           4.8706e-02, 7.8809e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1448e-01, 8.5235e-06, 7.0930e-06,  ..., 9.2545e-03,
           3.7720e-02, 6.9775e-01]],

         [[5.1855e-01, 5.5790e-05, 6.2943e-05,  ..., 8.2626e-03,
           2.7979e-01, 1.9699e-02]],

         [[3.8550e-01, 2.2256e-04, 4.5359e-05,  ..., 5.3223e-02,
           2.2583e-02, 1.0872e-02]],

         ...,

         [[9.7839e-02, 1.8477e-06, 2.7418e-06,  ..., 3.8087e-05,
           8.3923e-03, 8.7891e-01]],

         [[8.8135e-01, 2.9206e-05, 4.8757e-05,  ..., 2.2182e-03,
           6.0349e-03, 1.6541e-02]],

         [[8.1104e-01, 1.3769e-04, 3.7313e-05,  ..., 3.0365e-03,
           2.2644e-02, 3.0537e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 1.2159e-03, 9.8419e-04,  ..., 2.8553e-03,
           2.1957e-02, 6.2218e-03]],

         [[8.6914e-01, 5.3287e-05, 9.8825e-05,  ..., 6.2332e-03,
           3.3752e-02, 1.0895e-02]],

         [[5.4395e-01, 3.0375e-04, 1.3483e-04,  ..., 4.1695e-03,
           1.1810e-02, 6.4087e-03]],

         ...,

         [[3.2495e-01, 1.1027e-05, 2.0087e-05,  ..., 3.7689e-02,
           6.1621e-01, 1.8127e-02]],

         [[4.3042e-01, 3.3760e-04, 6.6614e-04,  ..., 1.7746e-02,
           1.9550e-03, 4.3511e-04]],

         [[8.9844e-01, 7.8082e-06, 1.1444e-05,  ..., 1.1841e-02,
           6.3293e-02, 2.3285e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.4609e-01, 1.9729e-05, 1.1921e-05,  ..., 1.8539e-02,
           1.0117e-02, 3.0994e-03]],

         [[9.7107e-02, 1.2846e-03, 9.5940e-04,  ..., 5.2452e-03,
           1.3466e-02, 1.2741e-03]],

         [[4.3701e-01, 2.7061e-04, 2.2054e-04,  ..., 3.3966e-02,
           1.1786e-01, 6.1188e-03]],

         ...,

         [[3.3789e-01, 3.8087e-05, 1.8477e-05,  ..., 4.1840e-02,
           5.2490e-01, 3.3508e-02]],

         [[8.7939e-01, 9.9540e-06, 3.2783e-06,  ..., 1.4343e-02,
           7.8003e-02, 1.9150e-02]],

         [[6.7529e-01, 3.0935e-05, 2.3186e-05,  ..., 2.1896e-02,
           3.9001e-02, 3.5210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2642e-02, 1.1921e-07, 2.3842e-07,  ..., 1.2314e-04,
           9.6817e-03, 9.0234e-01]],

         [[1.7065e-01, 6.6161e-06, 1.1206e-05,  ..., 8.2214e-02,
           3.7720e-01, 3.7781e-02]],

         [[1.8262e-01, 3.1662e-04, 5.8079e-04,  ..., 2.6718e-02,
           4.6539e-02, 4.0321e-03]],

         ...,

         [[5.1270e-01, 3.0017e-04, 1.8001e-04,  ..., 4.2175e-02,
           8.8348e-03, 1.5421e-03]],

         [[5.8008e-01, 1.4246e-05, 5.3644e-06,  ..., 7.3547e-02,
           4.6417e-02, 1.4000e-02]],

         [[9.3933e-02, 5.6684e-05, 1.5676e-05,  ..., 1.5812e-03,
           6.4964e-03, 1.6632e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6445e-01, 3.2485e-05, 3.0696e-05,  ..., 1.4771e-02,
           2.8473e-02, 6.4575e-02]],

         [[2.5903e-01, 1.1766e-04, 5.0843e-05,  ..., 1.5283e-01,
           5.6427e-02, 4.9622e-02]],

         [[5.7129e-01, 2.1863e-04, 1.2982e-04,  ..., 4.9133e-03,
           7.6965e-02, 3.8849e-02]],

         ...,

         [[7.8223e-01, 1.8418e-05, 1.5199e-05,  ..., 2.3987e-02,
           3.6865e-02, 8.1940e-03]],

         [[5.8203e-01, 2.4855e-05, 4.4703e-05,  ..., 1.4519e-02,
           2.0837e-01, 8.0185e-03]],

         [[6.8896e-01, 3.4153e-05, 1.2755e-05,  ..., 6.8726e-02,
           5.3024e-03, 2.4738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.8662e-01, 1.3447e-03, 1.5888e-03,  ..., 1.0138e-03,
           3.7651e-03, 2.6512e-04]],

         [[2.3267e-01, 1.5700e-04, 1.2994e-04,  ..., 7.5912e-03,
           2.3636e-02, 1.1337e-02]],

         [[2.4829e-01, 5.0664e-06, 1.7166e-05,  ..., 9.8755e-02,
           3.8745e-01, 2.8625e-02]],

         ...,

         [[7.8564e-01, 5.2691e-05, 2.8431e-05,  ..., 1.1778e-03,
           2.7725e-02, 2.5955e-02]],

         [[9.1895e-01, 4.0352e-05, 5.6565e-05,  ..., 2.1458e-06,
           4.5943e-04, 1.0193e-02]],

         [[5.9229e-01, 1.7333e-04, 3.7026e-04,  ..., 6.9351e-03,
           6.5155e-03, 3.3283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5552e-02, 2.4772e-04, 1.2660e-04,  ..., 4.4342e-02,
           3.2715e-02, 2.5616e-03]],

         [[5.3558e-02, 5.4598e-04, 2.0361e-04,  ..., 2.1652e-02,
           3.0289e-02, 3.6736e-03]],

         [[4.5923e-01, 2.7955e-05, 1.7047e-05,  ..., 3.8815e-03,
           2.4207e-01, 1.6638e-01]],

         ...,

         [[2.9932e-01, 4.5896e-06, 5.9009e-06,  ..., 4.8096e-02,
           1.8036e-02, 2.1381e-03]],

         [[5.9912e-01, 6.4373e-06, 2.9206e-06,  ..., 4.3640e-03,
           2.2217e-01, 9.1187e-02]],

         [[1.2622e-01, 5.9605e-06, 2.9802e-06,  ..., 9.8877e-03,
           5.1904e-01, 3.0054e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0342e-01, 1.7524e-05, 4.1723e-06,  ..., 4.8309e-02,
           3.1189e-02, 2.3926e-02]],

         [[1.5002e-01, 5.1916e-05, 8.9705e-05,  ..., 8.4106e-02,
           3.8306e-01, 5.6915e-02]],

         [[2.5977e-01, 7.7486e-06, 2.2054e-06,  ..., 7.1812e-04,
           1.0492e-01, 6.2305e-01]],

         ...,

         [[4.4098e-02, 8.2791e-05, 1.1683e-05,  ..., 1.1683e-05,
           2.9774e-03, 1.0887e-02]],

         [[4.0112e-01, 8.1599e-05, 3.7611e-05,  ..., 3.7323e-02,
           9.9258e-03, 3.4447e-03]],

         [[1.7944e-01, 7.3314e-05, 4.8757e-05,  ..., 8.3435e-02,
           4.2686e-03, 6.0921e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.5337e-01, 1.7881e-05, 1.4961e-05,  ..., 1.7990e-02,
           3.6621e-02, 3.7811e-02]],

         [[9.3262e-01, 1.2636e-05, 2.5034e-06,  ..., 4.9472e-06,
           9.0361e-05, 1.3512e-02]],

         [[6.8994e-01, 9.4235e-05, 1.1802e-05,  ..., 2.4300e-03,
           4.2992e-03, 2.0996e-02]],

         ...,

         [[2.4829e-01, 7.6008e-04, 1.0147e-03,  ..., 1.1986e-02,
           1.8707e-02, 5.8403e-03]],

         [[7.9395e-01, 2.5570e-05, 2.3663e-05,  ..., 1.1688e-02,
           1.4099e-02, 1.5732e-02]],

         [[2.1973e-01, 3.7575e-04, 5.0354e-04,  ..., 5.0568e-02,
           4.6051e-02, 1.0277e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4644e-01, 1.0961e-04, 1.4293e-04,  ..., 8.0994e-02,
           1.0880e-02, 1.1398e-02]],

         [[2.8198e-01, 6.1333e-05, 1.2827e-04,  ..., 1.2985e-02,
           4.6387e-02, 4.0955e-02]],

         [[2.0984e-01, 5.7220e-05, 7.6413e-05,  ..., 1.5774e-03,
           7.1220e-03, 6.3362e-03]],

         ...,

         [[1.0107e-01, 6.7043e-04, 6.7043e-04,  ..., 1.0002e-02,
           5.4932e-02, 2.0859e-02]],

         [[6.1829e-02, 7.2420e-05, 2.4259e-05,  ..., 1.8775e-04,
           5.1270e-02, 8.0627e-02]],

         [[5.1123e-01, 1.5736e-04, 1.2946e-04,  ..., 2.1423e-02,
           9.9640e-03, 7.5226e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9395e-01, 4.3094e-05, 2.5034e-05,  ..., 1.3855e-02,
           9.8450e-02, 6.9275e-02]],

         [[2.7612e-01, 9.3758e-05, 1.3220e-04,  ..., 5.1086e-02,
           3.6224e-02, 1.6846e-02]],

         [[3.8403e-01, 1.9121e-04, 2.8706e-04,  ..., 2.1835e-02,
           1.4026e-01, 1.9165e-01]],

         ...,

         [[4.7095e-01, 9.5069e-05, 9.0182e-05,  ..., 1.2070e-02,
           4.0192e-02, 6.7871e-02]],

         [[6.4307e-01, 3.8981e-05, 1.4949e-04,  ..., 6.5947e-04,
           2.9617e-02, 1.0419e-01]],

         [[4.6826e-01, 5.1022e-05, 1.0842e-04,  ..., 7.3967e-03,
           1.4824e-02, 4.9744e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9958e-01, 1.3340e-04, 5.3644e-05,  ..., 1.8784e-02,
           5.7617e-02, 4.9286e-02]],

         [[4.2480e-01, 2.4772e-04, 2.0099e-04,  ..., 1.1772e-02,
           1.1084e-01, 4.5502e-02]],

         [[4.2261e-01, 3.5930e-04, 2.6655e-04,  ..., 2.8320e-02,
           2.4887e-02, 1.3428e-02]],

         ...,

         [[5.9668e-01, 4.8280e-05, 3.6240e-04,  ..., 3.8422e-02,
           3.1934e-01, 3.3661e-02]],

         [[2.6367e-01, 9.1195e-06, 6.5744e-05,  ..., 8.1665e-02,
           4.1162e-01, 5.6580e-02]],

         [[2.0813e-01, 7.0930e-06, 9.0361e-05,  ..., 1.1925e-02,
           7.3779e-01, 3.5065e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9007e-02, 5.8413e-06, 9.0599e-06,  ..., 9.8145e-02,
           2.2461e-01, 1.1035e-01]],

         [[1.7432e-01, 2.8729e-05, 2.3007e-05,  ..., 3.1185e-03,
           3.4332e-02, 6.6797e-01]],

         [[5.6641e-02, 2.5928e-05, 1.2183e-04,  ..., 1.0078e-02,
           2.9266e-02, 6.1111e-03]],

         ...,

         [[1.5076e-01, 1.7929e-04, 7.3147e-04,  ..., 4.1870e-02,
           3.8574e-02, 1.5343e-02]],

         [[1.7029e-01, 1.7881e-06, 4.1127e-06,  ..., 8.9569e-03,
           7.3096e-01, 5.8167e-02]],

         [[4.7192e-01, 1.0729e-06, 3.3379e-06,  ..., 1.0529e-01,
           3.3203e-01, 4.3732e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5381e-01, 6.8605e-05, 2.6512e-04,  ..., 2.3438e-02,
           3.6163e-02, 6.9313e-03]],

         [[9.5947e-01, 3.5763e-05, 1.3006e-04,  ..., 1.0805e-03,
           3.6192e-04, 3.9434e-04]],

         [[8.4863e-01, 2.3520e-04, 1.8895e-04,  ..., 1.3405e-02,
           4.3869e-03, 2.1038e-03]],

         ...,

         [[9.1797e-01, 1.0192e-05, 1.5676e-05,  ..., 9.1400e-03,
           2.1591e-02, 5.0468e-03]],

         [[8.9551e-01, 3.1590e-06, 4.6492e-06,  ..., 6.1131e-04,
           2.6760e-03, 4.0817e-03]],

         [[2.7295e-01, 5.9605e-06, 1.6034e-05,  ..., 3.2520e-03,
           4.0558e-02, 6.2451e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1957e-01, 2.6226e-06, 1.1921e-05,  ..., 6.0120e-02,
           3.6816e-01, 1.5112e-01]],

         [[8.0420e-01, 2.5094e-05, 1.9073e-05,  ..., 1.2794e-02,
           2.0935e-02, 2.2552e-02]],

         [[1.8079e-01, 2.8944e-04, 1.3626e-04,  ..., 5.6427e-02,
           3.5034e-02, 1.1307e-02]],

         ...,

         [[6.5869e-01, 1.2040e-05, 4.5061e-05,  ..., 1.7548e-02,
           6.4209e-02, 3.1036e-02]],

         [[3.8037e-01, 1.7583e-05, 8.7917e-05,  ..., 5.8350e-02,
           1.8143e-02, 3.5126e-02]],

         [[6.9238e-01, 8.1658e-06, 5.1856e-05,  ..., 1.3397e-02,
           3.0670e-02, 5.4230e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9185e-01, 2.8253e-05, 1.5855e-05,  ..., 1.0124e-02,
           1.3440e-01, 1.9861e-01]],

         [[5.4150e-01, 8.8811e-06, 2.1458e-06,  ..., 2.4819e-04,
           2.6535e-02, 1.9763e-01]],

         [[2.1570e-01, 6.3598e-05, 5.2333e-05,  ..., 3.5767e-02,
           6.2286e-02, 1.6931e-01]],

         ...,

         [[8.8196e-02, 1.2517e-06, 2.6345e-05,  ..., 2.1286e-02,
           1.2250e-01, 4.5776e-02]],

         [[9.6619e-02, 4.1127e-06, 3.7014e-05,  ..., 2.8122e-02,
           4.6356e-02, 1.4587e-02]],

         [[9.5312e-01, 6.0201e-06, 1.5676e-05,  ..., 4.6272e-03,
           8.7814e-03, 9.1124e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3594e-01, 1.0133e-06, 8.8811e-06,  ..., 7.3318e-03,
           1.5587e-02, 8.2458e-02]],

         [[4.4019e-01, 1.6689e-05, 1.6630e-04,  ..., 1.0492e-01,
           1.7920e-01, 1.0455e-01]],

         [[1.5759e-01, 2.1458e-06, 6.5923e-05,  ..., 4.0924e-02,
           7.5732e-01, 3.4332e-02]],

         ...,

         [[4.2651e-01, 5.3465e-05, 4.0352e-05,  ..., 6.0486e-02,
           4.4670e-03, 2.1408e-02]],

         [[3.5913e-01, 1.0550e-05, 7.5698e-06,  ..., 1.6037e-02,
           3.2959e-01, 7.9529e-02]],

         [[1.1731e-01, 6.8247e-05, 2.4629e-04,  ..., 1.8066e-02,
           3.3081e-02, 3.5010e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4160e-01, 1.1683e-05, 1.8537e-05,  ..., 5.5733e-03,
           1.1505e-01, 1.2830e-01]],

         [[6.8604e-01, 4.8220e-05, 4.8816e-05,  ..., 3.8776e-03,
           3.3508e-02, 9.0332e-02]],

         [[3.0469e-01, 2.6405e-05, 5.4240e-05,  ..., 3.7460e-03,
           4.5654e-02, 1.9360e-01]],

         ...,

         [[6.7432e-01, 4.3869e-05, 1.2398e-04,  ..., 5.6534e-03,
           8.5754e-02, 3.1052e-02]],

         [[3.8892e-01, 1.6749e-04, 3.9554e-04,  ..., 9.5825e-03,
           1.4389e-02, 4.3384e-01]],

         [[2.9028e-01, 5.0128e-05, 7.7605e-05,  ..., 1.7700e-02,
           4.8608e-01, 1.6028e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2285e-01, 1.1206e-05, 1.1319e-04,  ..., 8.9951e-03,
           8.1863e-03, 3.0487e-02]],

         [[1.2756e-01, 1.3709e-06, 2.6584e-05,  ..., 3.5820e-03,
           3.1567e-01, 1.5564e-01]],

         [[2.7710e-01, 2.3544e-04, 9.5463e-04,  ..., 8.5815e-02,
           6.1798e-02, 8.3191e-02]],

         ...,

         [[7.9639e-01, 5.3525e-05, 3.4785e-04,  ..., 1.8112e-02,
           5.3711e-03, 1.0239e-02]],

         [[8.8184e-01, 2.3842e-07, 2.3127e-04,  ..., 2.9087e-03,
           1.3733e-03, 1.3977e-02]],

         [[5.6152e-01, 2.8670e-05, 1.1957e-04,  ..., 1.3580e-02,
           3.3932e-03, 9.9335e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6093e-05, 7.9870e-06, 1.5479e-01,  ..., 8.5678e-03,
           5.2490e-02, 3.0542e-01]],

         [[6.5565e-06, 2.3246e-06, 1.3466e-02,  ..., 1.2189e-01,
           3.3131e-03, 9.1782e-03]],

         [[1.4305e-04, 1.0079e-04, 5.8319e-02,  ..., 1.3527e-02,
           1.2131e-02, 1.3452e-01]],

         ...,

         [[3.7789e-05, 1.0014e-05, 1.3318e-01,  ..., 9.3689e-02,
           2.4683e-01, 1.7371e-01]],

         [[1.1557e-04, 1.7822e-05, 2.7783e-01,  ..., 6.6109e-03,
           7.4615e-03, 1.6556e-02]],

         [[3.2187e-06, 1.7881e-07, 3.2812e-01,  ..., 5.7697e-04,
           2.6932e-02, 7.6111e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3896e-05, 1.0252e-05, 2.6953e-01,  ..., 1.2436e-03,
           1.4977e-02, 1.3660e-01]],

         [[3.7491e-05, 3.8147e-06, 1.6736e-01,  ..., 2.1992e-03,
           4.0070e-02, 2.9370e-01]],

         [[1.7762e-05, 6.5565e-07, 2.9785e-01,  ..., 1.3649e-04,
           1.1168e-03, 7.2937e-02]],

         ...,

         [[2.9862e-05, 2.5630e-06, 6.4209e-02,  ..., 2.3956e-03,
           2.3308e-03, 5.1208e-02]],

         [[2.3246e-06, 7.1526e-07, 2.5708e-01,  ..., 4.6992e-04,
           5.8174e-03, 1.3550e-01]],

         [[2.7895e-05, 1.1802e-05, 3.0078e-01,  ..., 2.3544e-04,
           8.3828e-04, 4.6844e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9009e-06, 8.3447e-07, 1.5015e-01,  ..., 5.2035e-05,
           2.3026e-02, 1.2744e-01]],

         [[7.7486e-06, 5.9605e-08, 1.6296e-01,  ..., 5.7106e-03,
           6.4325e-04, 9.1431e-02]],

         [[8.7619e-06, 2.0862e-06, 1.5698e-01,  ..., 5.5742e-04,
           1.9760e-03, 9.9792e-02]],

         ...,

         [[9.3222e-05, 5.2452e-06, 1.8616e-01,  ..., 7.7744e-03,
           9.5825e-02, 2.0605e-01]],

         [[3.8147e-06, 1.2517e-06, 7.6416e-02,  ..., 2.5654e-03,
           5.7564e-03, 1.1298e-01]],

         [[2.2709e-05, 1.3113e-06, 1.6699e-01,  ..., 2.9526e-03,
           3.1250e-02, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2338e-05, 1.2517e-06, 1.8723e-02,  ..., 4.5166e-03,
           7.1387e-01, 1.4734e-01]],

         [[7.3314e-06, 7.1049e-05, 8.3740e-02,  ..., 9.4116e-02,
           2.2186e-02, 1.9775e-02]],

         [[3.6716e-05, 4.0388e-04, 3.2886e-01,  ..., 1.1263e-03,
           4.3297e-04, 5.3596e-04]],

         ...,

         [[1.2903e-03, 4.3988e-05, 2.5684e-01,  ..., 1.8525e-04,
           1.0004e-03, 2.7599e-03]],

         [[9.3222e-04, 5.3704e-05, 6.5613e-02,  ..., 6.3820e-03,
           2.3633e-01, 4.5728e-01]],

         [[2.1017e-04, 1.7405e-05, 6.6223e-02,  ..., 2.3022e-03,
           2.0569e-01, 3.6523e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4211e-04, 7.9453e-05, 8.9294e-02,  ..., 1.6129e-02,
           1.9989e-02, 4.6844e-02]],

         [[6.9666e-04, 3.6168e-04, 2.0789e-01,  ..., 2.7054e-02,
           3.2135e-02, 6.8787e-02]],

         [[3.5334e-04, 1.5354e-04, 1.4697e-01,  ..., 9.0179e-03,
           3.5797e-02, 6.4331e-02]],

         ...,

         [[1.0347e-04, 2.1040e-05, 3.7109e-02,  ..., 5.5618e-03,
           5.8545e-01, 2.4402e-01]],

         [[3.1710e-04, 5.6076e-04, 3.3960e-01,  ..., 3.5930e-04,
           4.8065e-03, 3.9001e-02]],

         [[4.2248e-04, 1.3614e-04, 2.8540e-01,  ..., 2.9793e-03,
           1.8539e-02, 3.1525e-02]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[7.8082e-06, 3.2961e-05, 2.0390e-03,  ..., 6.1462e-02,
           5.1465e-01, 1.1487e-01]],

         [[1.1110e-04, 3.3855e-05, 7.7105e-04,  ..., 2.8931e-02,
           1.5637e-01, 5.1270e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 8.3447e-07,
           2.4700e-03, 9.9658e-01]],

         ...,

         [[0.0000e+00, 5.9605e-08, 1.7881e-07,  ..., 2.0015e-04,
           6.6261e-03, 9.8340e-01]],

         [[9.1705e-03, 7.6065e-03, 3.8242e-03,  ..., 2.0676e-02,
           2.6550e-02, 4.3335e-03]],

         [[8.9722e-03, 3.6259e-03, 1.1768e-03,  ..., 1.6769e-02,
           1.6769e-02, 5.0049e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5048e-04, 3.5048e-04, 3.0270e-03,  ..., 2.6154e-02,
           9.4177e-02, 5.4199e-01]],

         [[0.0000e+00, 0.0000e+00, 5.9605e-08,  ..., 1.4624e-01,
           7.9053e-01, 5.3802e-02]],

         [[5.2929e-04, 1.8759e-03, 1.2238e-02,  ..., 3.8879e-02,
           3.6530e-02, 7.2632e-02]],

         ...,

         [[5.6725e-03, 7.8735e-03, 1.6281e-02,  ..., 8.6517e-03,
           1.8448e-02, 8.9417e-02]],

         [[2.1458e-03, 6.3667e-03, 2.9907e-02,  ..., 1.0788e-02,
           2.1790e-02, 1.1002e-02]],

         [[2.9049e-03, 7.7744e-03, 1.4076e-02,  ..., 5.7434e-02,
           6.7139e-02, 1.1786e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6519e-02, 5.5084e-03, 2.9541e-02,  ..., 2.6901e-02,
           1.9836e-02, 2.6901e-02]],

         [[8.1158e-04, 3.4904e-04, 1.8291e-03,  ..., 3.8719e-03,
           4.7836e-03, 3.5820e-03]],

         [[3.1414e-03, 5.6791e-04, 6.5327e-04,  ..., 1.9547e-02,
           1.8723e-02, 1.6586e-02]],

         ...,

         [[1.0204e-04, 1.9670e-04, 1.0170e-02,  ..., 1.3708e-01,
           3.7866e-01, 1.5784e-01]],

         [[4.5776e-03, 6.2180e-04, 2.1877e-03,  ..., 6.7810e-02,
           1.3458e-02, 2.0996e-02]],

         [[2.7542e-02, 5.5962e-03, 1.8494e-02,  ..., 8.8928e-02,
           9.2468e-02, 1.3245e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.5791e-01, 5.9605e-08, 4.1723e-07,  ..., 3.7766e-03,
           5.1918e-03, 2.8625e-02]],

         [[8.8818e-01, 1.3113e-06, 5.3048e-06,  ..., 1.3306e-02,
           5.1788e-02, 1.4694e-02]],

         [[9.4629e-01, 1.6689e-06, 9.7752e-06,  ..., 6.9351e-03,
           1.2875e-03, 1.2100e-02]],

         ...,

         [[7.2754e-01, 4.9114e-04, 1.0986e-03,  ..., 2.1946e-04,
           1.1164e-04, 1.9623e-02]],

         [[6.5039e-01, 1.3351e-04, 6.0225e-04,  ..., 5.0354e-02,
           5.3062e-03, 1.5472e-02]],

         [[2.5439e-01, 1.4508e-04, 5.4884e-04,  ..., 4.2261e-01,
           2.9663e-02, 1.8280e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9453e-01, 5.6505e-05, 2.8193e-05,  ..., 7.7391e-04,
           3.0577e-05, 4.7699e-02]],

         [[8.0957e-01, 1.1921e-06, 9.6560e-06,  ..., 6.1554e-02,
           9.1064e-02, 3.5767e-02]],

         [[8.5986e-01, 5.1796e-05, 1.7583e-05,  ..., 8.2016e-04,
           1.3580e-02, 4.0512e-03]],

         ...,

         [[2.8613e-01, 2.2459e-04, 3.4809e-04,  ..., 1.2549e-01,
           4.2053e-02, 2.3682e-02]],

         [[2.0691e-01, 1.9855e-03, 9.0981e-04,  ..., 4.9286e-03,
           2.1172e-03, 3.9139e-03]],

         [[4.8145e-01, 8.8334e-05, 4.9400e-04,  ..., 1.1328e-01,
           3.9764e-02, 1.2512e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0337e-01, 1.9586e-04, 7.2598e-05,  ..., 7.1472e-02,
           3.1708e-02, 3.5553e-02]],

         [[6.9189e-01, 4.9591e-05, 3.2246e-05,  ..., 9.6924e-02,
           5.3162e-02, 8.8562e-02]],

         [[9.1602e-01, 1.4603e-04, 5.3287e-05,  ..., 3.1719e-03,
           7.0496e-03, 1.4214e-02]],

         ...,

         [[9.3311e-01, 6.7949e-06, 2.1458e-06,  ..., 1.1402e-04,
           1.7583e-05, 6.4392e-02]],

         [[9.3359e-01, 3.4571e-05, 3.0160e-05,  ..., 4.2319e-06,
           9.5367e-07, 2.2903e-02]],

         [[9.5508e-01, 9.0003e-06, 1.7881e-06,  ..., 3.6955e-06,
           7.7486e-07, 2.2537e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6792e-01, 4.7088e-06, 3.1590e-06,  ..., 1.9165e-01,
           2.5000e-01, 8.0383e-02]],

         [[8.2275e-01, 2.3246e-06, 1.3709e-06,  ..., 9.7322e-04,
           5.1880e-04, 2.4338e-02]],

         [[8.1006e-01, 1.0234e-04, 4.3988e-05,  ..., 1.1444e-03,
           1.4925e-03, 1.5160e-02]],

         ...,

         [[6.9531e-01, 2.0337e-04, 3.2306e-05,  ..., 1.3466e-02,
           5.5275e-03, 1.9763e-01]],

         [[5.1953e-01, 1.7798e-04, 5.4300e-05,  ..., 6.7322e-02,
           3.6865e-02, 1.7358e-01]],

         [[8.5010e-01, 3.7313e-05, 5.9187e-05,  ..., 6.1707e-02,
           1.0445e-02, 4.5685e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4424e-01, 1.4126e-05, 1.0431e-05,  ..., 3.0533e-02,
           3.9940e-03, 1.5060e-02]],

         [[9.0820e-01, 6.4969e-05, 1.5557e-05,  ..., 1.0170e-02,
           2.8191e-03, 2.0264e-02]],

         [[9.7656e-01, 1.3113e-06, 5.9605e-07,  ..., 9.3002e-03,
           4.8599e-03, 4.4708e-03]],

         ...,

         [[6.4111e-01, 2.6798e-04, 5.2154e-05,  ..., 3.4454e-02,
           9.9411e-03, 2.5864e-02]],

         [[8.6523e-01, 1.3316e-04, 2.0087e-05,  ..., 5.4131e-03,
           2.1696e-04, 6.1226e-03]],

         [[9.5557e-01, 1.5378e-05, 4.2915e-06,  ..., 6.6299e-03,
           4.8904e-03, 9.1858e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2446e-01, 2.4605e-03, 1.2093e-03,  ..., 6.9427e-03,
           1.5459e-03, 1.2283e-02]],

         [[9.2920e-01, 2.0862e-05, 6.0916e-05,  ..., 1.9321e-03,
           2.5196e-03, 4.9133e-03]],

         [[8.8574e-01, 3.4392e-05, 1.1086e-05,  ..., 1.7500e-03,
           1.5440e-03, 1.4061e-02]],

         ...,

         [[8.0566e-01, 3.4571e-05, 3.1233e-05,  ..., 4.3564e-03,
           2.5650e-02, 1.0498e-01]],

         [[7.9688e-01, 4.5300e-06, 9.7156e-06,  ..., 9.7290e-02,
           6.8787e-02, 6.7444e-03]],

         [[8.5791e-01, 5.6028e-06, 1.8001e-05,  ..., 7.1945e-03,
           9.9411e-03, 8.2886e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1792e-01, 1.9073e-04, 7.5865e-04,  ..., 1.0767e-01,
           1.0033e-02, 1.0674e-02]],

         [[8.0322e-01, 1.4901e-06, 7.2718e-06,  ..., 6.2988e-02,
           7.2449e-02, 1.7456e-02]],

         [[8.5840e-01, 3.2187e-06, 8.2254e-06,  ..., 1.4458e-02,
           2.3148e-02, 2.8915e-02]],

         ...,

         [[7.5879e-01, 1.2314e-04, 1.2410e-04,  ..., 2.4090e-03,
           5.2118e-04, 1.1120e-03]],

         [[6.6504e-01, 4.8220e-05, 1.9908e-05,  ..., 4.9896e-02,
           4.2572e-03, 9.2712e-02]],

         [[1.3245e-01, 2.6822e-06, 1.6093e-06,  ..., 7.7441e-01,
           6.2561e-02, 1.4130e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6216e-01, 1.2505e-04, 7.3195e-05,  ..., 1.7345e-05,
           4.9472e-06, 3.7117e-03]],

         [[9.0186e-01, 1.4126e-05, 3.1114e-05,  ..., 2.4246e-02,
           1.2722e-03, 8.3113e-04]],

         [[9.2871e-01, 1.0729e-05, 5.9485e-05,  ..., 1.7529e-03,
           1.3411e-04, 1.7868e-02]],

         ...,

         [[8.2520e-01, 2.2554e-04, 1.5080e-04,  ..., 7.9193e-03,
           2.5215e-03, 3.8357e-03]],

         [[7.8271e-01, 6.7472e-05, 5.9962e-05,  ..., 6.8903e-04,
           9.4482e-02, 9.5596e-03]],

         [[6.2402e-01, 7.9632e-04, 2.6989e-04,  ..., 4.9988e-02,
           1.4465e-02, 2.0081e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1863e-01, 1.7881e-07, 1.4901e-06,  ..., 6.6662e-04,
           6.7444e-03, 7.6611e-01]],

         [[5.3320e-01, 6.9475e-04, 1.2875e-03,  ..., 2.3918e-03,
           8.2731e-04, 5.9853e-03]],

         [[6.9043e-01, 5.8889e-04, 7.3671e-04,  ..., 9.5825e-03,
           2.7008e-03, 1.2001e-02]],

         ...,

         [[6.6309e-01, 1.5581e-04, 1.9395e-04,  ..., 2.8286e-03,
           2.4090e-03, 7.8201e-04]],

         [[4.0771e-01, 8.4043e-06, 9.6560e-06,  ..., 1.2537e-01,
           4.0405e-01, 1.8997e-02]],

         [[4.1382e-01, 1.2195e-04, 1.8167e-04,  ..., 2.3254e-02,
           6.6101e-02, 4.1968e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9810e-01, 1.0139e-04, 3.4511e-05,  ..., 5.2299e-03,
           8.7891e-03, 5.1514e-01]],

         [[4.4434e-01, 5.4240e-06, 7.5698e-06,  ..., 1.9177e-01,
           2.8394e-01, 9.8038e-03]],

         [[3.0908e-01, 1.4782e-04, 8.1956e-05,  ..., 4.4830e-02,
           2.2003e-02, 2.0416e-02]],

         ...,

         [[7.8064e-02, 3.9935e-06, 1.9670e-06,  ..., 2.0278e-04,
           2.1782e-03, 9.1113e-01]],

         [[8.7842e-01, 3.0577e-05, 2.4259e-05,  ..., 2.2907e-03,
           2.0409e-03, 8.2245e-03]],

         [[7.8809e-01, 5.4955e-05, 2.6882e-05,  ..., 3.3905e-02,
           3.0457e-02, 3.2288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1230e-01, 1.2293e-03, 8.4782e-04,  ..., 9.2239e-03,
           2.4967e-03, 2.6379e-03]],

         [[8.9355e-01, 5.4479e-05, 1.0794e-04,  ..., 1.7014e-02,
           1.3954e-02, 1.2283e-02]],

         [[5.0928e-01, 1.4162e-04, 6.5565e-05,  ..., 6.4182e-04,
           8.8453e-04, 8.3923e-03]],

         ...,

         [[5.5273e-01, 5.9605e-08, 3.1590e-06,  ..., 8.3618e-02,
           3.2300e-01, 3.5767e-02]],

         [[4.7168e-01, 6.3944e-04, 8.3733e-04,  ..., 4.9133e-03,
           1.5821e-03, 5.2404e-04]],

         [[8.9453e-01, 2.3842e-07, 4.2319e-06,  ..., 1.7044e-02,
           6.2408e-02, 2.2339e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.7095e-01, 2.0361e-04, 6.7174e-05,  ..., 4.9927e-02,
           2.3819e-02, 8.2153e-02]],

         [[1.2103e-01, 2.5578e-03, 1.1511e-03,  ..., 5.2910e-03,
           3.2806e-04, 1.7481e-03]],

         [[7.6758e-01, 9.1136e-05, 7.5221e-05,  ..., 3.2013e-02,
           7.0152e-03, 4.7150e-02]],

         ...,

         [[7.1777e-01, 3.1352e-05, 1.5199e-05,  ..., 8.3984e-02,
           9.3323e-02, 6.6528e-02]],

         [[8.1543e-01, 4.5300e-06, 3.6955e-06,  ..., 2.5604e-02,
           5.7983e-02, 9.1309e-02]],

         [[8.1445e-01, 4.7386e-05, 2.5988e-05,  ..., 5.2155e-02,
           1.9943e-02, 1.6754e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1500e-02, 0.0000e+00, 0.0000e+00,  ..., 7.4291e-04,
           1.8177e-03, 9.7314e-01]],

         [[1.2024e-01, 1.5318e-05, 1.8895e-05,  ..., 3.1934e-01,
           1.8616e-01, 8.5571e-02]],

         [[2.0825e-01, 2.6417e-04, 4.9591e-04,  ..., 4.8309e-02,
           4.8828e-03, 1.4824e-02]],

         ...,

         [[4.4263e-01, 2.7680e-04, 2.7466e-04,  ..., 2.1484e-02,
           1.4084e-02, 1.7128e-03]],

         [[5.9180e-01, 1.2636e-05, 6.3777e-06,  ..., 2.5803e-02,
           5.4436e-03, 1.0773e-02]],

         [[3.2324e-01, 4.9472e-05, 2.1279e-05,  ..., 2.5368e-03,
           5.9090e-03, 1.0475e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0303e-01, 9.8348e-06, 6.3777e-06,  ..., 3.1342e-02,
           2.4078e-02, 2.0996e-01]],

         [[3.7036e-01, 2.7323e-04, 1.4567e-04,  ..., 1.4160e-02,
           3.6530e-02, 5.4840e-02]],

         [[1.1218e-01, 1.4842e-05, 6.9737e-06,  ..., 5.8411e-02,
           6.6064e-01, 4.5502e-02]],

         ...,

         [[8.3252e-01, 1.5557e-05, 3.1829e-05,  ..., 3.4241e-02,
           1.4275e-02, 2.2903e-02]],

         [[6.9775e-01, 1.4603e-05, 2.3067e-05,  ..., 1.1615e-01,
           2.9251e-02, 5.1941e-02]],

         [[8.0176e-01, 1.9848e-05, 8.7023e-06,  ..., 8.7051e-03,
           7.2975e-03, 4.4006e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5491e-01, 2.2068e-03, 2.0370e-03,  ..., 3.7518e-03,
           5.4646e-04, 2.0618e-03]],

         [[3.3130e-01, 1.2374e-04, 6.5625e-05,  ..., 4.3549e-02,
           5.0629e-02, 1.6663e-02]],

         [[4.8315e-01, 2.8610e-06, 5.1856e-06,  ..., 1.6565e-01,
           1.5137e-01, 9.9304e-02]],

         ...,

         [[3.1519e-01, 4.5300e-05, 1.5378e-05,  ..., 1.1492e-03,
           3.3226e-03, 3.6716e-03]],

         [[1.3374e-02, 2.9802e-06, 4.1723e-06,  ..., 2.9802e-07,
           1.7881e-06, 1.3342e-03]],

         [[4.4092e-01, 1.5318e-04, 3.6693e-04,  ..., 3.4428e-03,
           4.4823e-03, 4.3917e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5328e-02, 1.2386e-04, 7.9989e-05,  ..., 5.6641e-02,
           8.3542e-03, 5.9814e-02]],

         [[1.7822e-01, 3.5620e-04, 8.9705e-05,  ..., 6.6589e-02,
           1.4061e-02, 1.3527e-02]],

         [[4.4971e-01, 3.4988e-05, 1.4007e-05,  ..., 3.6072e-02,
           9.7290e-02, 3.3179e-01]],

         ...,

         [[5.3809e-01, 1.6689e-06, 4.9472e-06,  ..., 4.5593e-02,
           3.8391e-02, 1.4572e-02]],

         [[7.8711e-01, 2.3842e-06, 3.2187e-06,  ..., 8.4900e-02,
           4.8004e-02, 2.1133e-02]],

         [[1.5784e-01, 4.5300e-06, 2.2054e-06,  ..., 4.1504e-02,
           9.1370e-02, 6.2891e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4453e-01, 3.9995e-05, 9.9540e-06,  ..., 5.5771e-03,
           9.7885e-03, 4.3884e-02]],

         [[8.9539e-02, 2.7418e-06, 6.0797e-06,  ..., 4.4775e-01,
           3.1738e-01, 1.8768e-02]],

         [[1.1060e-01, 6.7353e-06, 1.7881e-06,  ..., 3.6678e-03,
           7.5264e-03, 8.6963e-01]],

         ...,

         [[4.3701e-02, 1.0294e-04, 5.3644e-06,  ..., 1.8034e-03,
           1.0767e-03, 4.8767e-02]],

         [[2.1667e-01, 1.9872e-04, 1.1867e-04,  ..., 3.2959e-02,
           2.4887e-02, 3.0243e-02]],

         [[2.0276e-01, 1.0997e-04, 4.7982e-05,  ..., 7.5195e-02,
           3.7506e-02, 4.2847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0771e-01, 1.8418e-05, 2.7299e-05,  ..., 3.0716e-02,
           3.0716e-02, 1.4429e-01]],

         [[1.6089e-01, 2.6822e-06, 5.3644e-07,  ..., 2.2542e-04,
           2.4962e-04, 4.9362e-03]],

         [[5.1514e-01, 8.5235e-05, 1.0133e-05,  ..., 2.9697e-03,
           6.0797e-04, 1.2306e-02]],

         ...,

         [[1.8738e-01, 9.6846e-04, 1.0967e-03,  ..., 3.5767e-02,
           1.0406e-02, 1.2169e-02]],

         [[6.4258e-01, 1.4901e-05, 1.3351e-05,  ..., 7.3242e-02,
           6.8787e-02, 6.8787e-02]],

         [[2.1289e-01, 4.3750e-04, 5.1165e-04,  ..., 2.4078e-02,
           1.0513e-02, 3.2379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8457e-01, 9.2804e-05, 2.0111e-04,  ..., 1.1780e-02,
           1.7548e-02, 5.5328e-02]],

         [[8.9355e-02, 1.7881e-05, 3.0458e-05,  ..., 1.3702e-02,
           1.7044e-02, 1.3069e-02]],

         [[3.4106e-01, 2.5153e-05, 2.4915e-05,  ..., 2.9736e-03,
           4.4289e-03, 5.0293e-02]],

         ...,

         [[3.1036e-02, 8.1897e-05, 2.6584e-05,  ..., 1.1713e-01,
           9.4116e-02, 1.1780e-02]],

         [[1.3159e-01, 1.5974e-05, 3.8743e-06,  ..., 1.3657e-02,
           1.9257e-02, 9.4788e-02]],

         [[8.2764e-01, 1.0139e-04, 7.9572e-05,  ..., 1.1810e-02,
           8.7738e-03, 2.4231e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3257e-01, 5.4240e-06, 2.3842e-06,  ..., 7.1533e-02,
           1.1255e-01, 1.4001e-01]],

         [[1.4050e-01, 2.9862e-05, 3.3438e-05,  ..., 1.4380e-01,
           2.0923e-01, 3.3112e-02]],

         [[2.4463e-01, 1.1712e-04, 1.7238e-04,  ..., 5.5450e-02,
           7.0068e-02, 5.1367e-01]],

         ...,

         [[4.5215e-01, 1.8656e-04, 1.2875e-04,  ..., 2.8458e-02,
           3.8605e-02, 6.5125e-02]],

         [[2.7344e-01, 1.6987e-05, 1.3137e-04,  ..., 7.9250e-04,
           4.0550e-03, 1.4417e-01]],

         [[3.8794e-01, 5.1200e-05, 5.5552e-05,  ..., 1.1444e-02,
           8.2642e-02, 1.5526e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0996e-01, 4.4525e-05, 2.2531e-05,  ..., 6.8726e-02,
           1.3245e-01, 1.4893e-01]],

         [[4.6948e-01, 4.1068e-05, 9.2983e-06,  ..., 1.0236e-01,
           8.8196e-02, 1.0724e-01]],

         [[4.9487e-01, 3.5906e-04, 2.8801e-04,  ..., 1.4320e-02,
           8.2245e-03, 8.0948e-03]],

         ...,

         [[6.3428e-01, 1.1921e-07, 3.7551e-05,  ..., 4.7028e-02,
           2.6855e-01, 4.0863e-02]],

         [[3.2593e-01, 1.1921e-06, 2.4676e-05,  ..., 1.9922e-01,
           2.5781e-01, 1.8539e-02]],

         [[5.2783e-01, 1.9670e-06, 8.8990e-05,  ..., 7.2510e-02,
           2.8687e-01, 8.4778e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1765e-02, 1.2517e-06, 8.9407e-07,  ..., 2.8296e-01,
           5.8057e-01, 2.6108e-02]],

         [[2.6709e-01, 7.3314e-06, 7.5102e-06,  ..., 2.4395e-03,
           1.7883e-02, 6.6113e-01]],

         [[1.2317e-01, 6.1393e-05, 1.4925e-04,  ..., 1.2317e-01,
           4.3762e-02, 2.5223e-02]],

         ...,

         [[9.9365e-02, 1.2982e-04, 4.3750e-04,  ..., 4.5135e-02,
           3.9520e-02, 3.1525e-02]],

         [[8.8501e-02, 1.7881e-06, 2.6226e-06,  ..., 4.2896e-01,
           3.9062e-01, 7.9346e-02]],

         [[3.0078e-01, 5.9605e-07, 7.6294e-06,  ..., 1.5662e-01,
           4.4775e-01, 7.5134e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2988e-01, 5.5194e-05, 1.8680e-04,  ..., 3.0624e-02,
           9.3765e-03, 6.5857e-02]],

         [[9.6436e-01, 2.8372e-05, 4.7922e-05,  ..., 1.8513e-04,
           5.1916e-05, 1.9140e-03]],

         [[9.5215e-01, 7.6175e-05, 3.4869e-05,  ..., 2.0027e-03,
           4.9067e-04, 2.3975e-03]],

         ...,

         [[9.1943e-01, 4.3511e-06, 4.8876e-06,  ..., 7.9575e-03,
           7.3586e-03, 5.5206e-02]],

         [[9.6387e-01, 1.5497e-06, 1.0133e-06,  ..., 3.4976e-04,
           1.4725e-03, 3.1166e-03]],

         [[2.3108e-01, 3.2187e-06, 6.4373e-06,  ..., 6.2561e-03,
           1.3664e-02, 7.3438e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6309e-01, 6.5565e-07, 2.6822e-06,  ..., 1.0992e-01,
           1.8628e-01, 4.6460e-01]],

         [[9.4727e-01, 7.2122e-06, 5.0068e-06,  ..., 5.3520e-03,
           7.7896e-03, 1.8188e-02]],

         [[3.6182e-01, 1.7297e-04, 2.0552e-04,  ..., 8.1055e-02,
           3.6377e-02, 4.7272e-02]],

         ...,

         [[7.0312e-01, 9.4175e-06, 2.1935e-05,  ..., 7.8918e-02,
           6.9092e-02, 3.1891e-02]],

         [[8.5645e-01, 9.9540e-06, 3.9816e-05,  ..., 8.2703e-03,
           6.0005e-03, 1.9226e-02]],

         [[7.9492e-01, 6.2585e-06, 2.1994e-05,  ..., 2.7847e-02,
           8.4473e-02, 1.9745e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 3.5167e-06, 1.4305e-06,  ..., 2.2864e-01,
           4.4067e-01, 7.4829e-02]],

         [[6.6064e-01, 3.1590e-06, 7.1526e-07,  ..., 7.5722e-03,
           1.2805e-01, 1.4062e-01]],

         [[5.4541e-01, 5.8472e-05, 3.4571e-05,  ..., 1.4534e-02,
           6.0242e-02, 4.7302e-02]],

         ...,

         [[8.1726e-02, 5.3644e-07, 1.3769e-05,  ..., 6.8787e-02,
           8.5632e-02, 1.8701e-01]],

         [[1.5881e-01, 2.3246e-06, 2.3842e-05,  ..., 9.7839e-02,
           6.7200e-02, 1.9257e-02]],

         [[9.3311e-01, 2.5094e-05, 3.2246e-05,  ..., 1.6830e-02,
           4.6005e-03, 3.2368e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6543e-01, 1.4305e-06, 2.3365e-05,  ..., 9.3384e-03,
           4.2419e-02, 3.4692e-01]],

         [[6.5234e-01, 1.9073e-06, 6.0141e-05,  ..., 4.9133e-02,
           1.1029e-01, 5.5237e-02]],

         [[5.3662e-01, 8.9407e-07, 3.5346e-05,  ..., 4.3060e-02,
           1.3733e-01, 2.6172e-01]],

         ...,

         [[4.9097e-01, 1.4591e-04, 8.0585e-05,  ..., 1.4145e-02,
           4.1260e-02, 2.4246e-02]],

         [[5.4535e-02, 4.1723e-07, 4.1723e-07,  ..., 4.3555e-01,
           4.2212e-01, 5.1208e-02]],

         [[6.7993e-02, 5.1498e-05, 1.7321e-04,  ..., 6.6528e-03,
           2.6413e-02, 6.7578e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7354e-01, 1.4067e-05, 1.2755e-05,  ..., 1.1566e-01,
           1.9067e-01, 1.8054e-01]],

         [[7.9004e-01, 1.4305e-05, 1.3471e-05,  ..., 1.5343e-02,
           5.7678e-02, 4.4556e-02]],

         [[1.6101e-01, 1.2577e-05, 3.0935e-05,  ..., 5.1056e-02,
           2.5732e-01, 1.3879e-01]],

         ...,

         [[6.5283e-01, 3.7789e-05, 1.0103e-04,  ..., 1.3684e-01,
           4.1748e-02, 4.1077e-02]],

         [[2.5293e-01, 1.2350e-04, 2.8706e-04,  ..., 1.1721e-03,
           1.2794e-02, 6.6650e-01]],

         [[1.4343e-01, 5.4240e-06, 9.8348e-06,  ..., 4.6289e-01,
           3.1323e-01, 5.3589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1655e-01, 1.4722e-05, 9.1672e-05,  ..., 4.1351e-02,
           5.9570e-01, 1.0852e-01]],

         [[5.5450e-02, 5.9605e-07, 4.8280e-06,  ..., 1.1829e-01,
           1.7480e-01, 2.6562e-01]],

         [[4.7217e-01, 1.9479e-04, 6.1321e-04,  ..., 1.4374e-02,
           4.9103e-02, 8.8013e-02]],

         ...,

         [[9.1357e-01, 4.3929e-05, 1.1456e-04,  ..., 4.1656e-03,
           5.2452e-03, 1.6510e-02]],

         [[9.6436e-01, 1.7881e-07, 9.4175e-05,  ..., 1.8823e-04,
           6.0558e-04, 8.7814e-03]],

         [[6.6992e-01, 2.2531e-05, 3.8624e-05,  ..., 3.0556e-03,
           1.1131e-02, 6.2805e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0788e-05, 6.6161e-06, 1.8677e-01,  ..., 3.1219e-02,
           2.6343e-01, 1.9302e-02]],

         [[2.9802e-06, 8.9407e-07, 7.5912e-03,  ..., 2.8362e-03,
           1.1940e-02, 2.7485e-03]],

         [[9.1195e-05, 8.9169e-05, 6.0394e-02,  ..., 1.0872e-02,
           6.4819e-02, 4.4458e-01]],

         ...,

         [[1.0967e-05, 1.4901e-06, 7.5623e-02,  ..., 7.1594e-02,
           4.8169e-01, 1.6785e-01]],

         [[1.1927e-04, 3.3796e-05, 3.3594e-01,  ..., 6.4507e-03,
           8.0566e-03, 9.4910e-03]],

         [[7.2718e-06, 5.3644e-07, 3.0103e-01,  ..., 1.3123e-02,
           5.5664e-02, 5.5664e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3273e-05, 2.5511e-05, 2.6465e-01,  ..., 5.4283e-03,
           5.4169e-02, 1.7868e-02]],

         [[3.8207e-05, 9.8944e-06, 2.3279e-01,  ..., 3.1250e-02,
           1.4575e-01, 8.6136e-03]],

         [[4.9829e-05, 7.0930e-06, 2.8076e-01,  ..., 1.6365e-03,
           4.3365e-02, 1.0345e-02]],

         ...,

         [[1.2875e-05, 4.0531e-06, 2.4841e-01,  ..., 1.7004e-03,
           2.7863e-02, 1.3794e-02]],

         [[2.3842e-06, 1.4901e-06, 2.8149e-01,  ..., 6.5956e-03,
           7.0618e-02, 4.3602e-03]],

         [[1.2457e-05, 1.7047e-05, 3.1714e-01,  ..., 6.5422e-04,
           2.8366e-02, 5.3520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8944e-06, 2.7418e-06, 1.6943e-01,  ..., 1.9318e-02,
           1.1292e-01, 9.0027e-03]],

         [[1.5736e-05, 4.1723e-07, 2.4414e-01,  ..., 3.1877e-04,
           4.2755e-02, 1.4656e-02]],

         [[2.2113e-05, 7.6890e-06, 2.5122e-01,  ..., 1.9407e-03,
           6.5552e-02, 1.0864e-02]],

         ...,

         [[3.9101e-05, 1.8477e-06, 1.6418e-01,  ..., 1.1548e-01,
           2.1411e-01, 1.6162e-01]],

         [[8.5235e-06, 4.2319e-06, 2.6514e-01,  ..., 4.3716e-03,
           4.6997e-02, 2.8503e-02]],

         [[4.3511e-05, 3.7551e-06, 2.2449e-01,  ..., 2.2308e-02,
           7.5195e-02, 2.4033e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-06, 2.3842e-07, 8.5220e-03,  ..., 2.4329e-01,
           5.5273e-01, 1.5466e-01]],

         [[3.1769e-05, 1.2100e-04, 7.6965e-02,  ..., 1.6830e-02,
           1.5747e-02, 1.4771e-02]],

         [[2.0921e-05, 2.3735e-04, 3.1396e-01,  ..., 5.4264e-04,
           4.6468e-04, 6.1369e-04]],

         ...,

         [[2.8586e-04, 9.0003e-06, 2.8540e-01,  ..., 5.2261e-04,
           1.7233e-03, 3.8967e-03]],

         [[2.9755e-04, 9.8348e-06, 2.9099e-02,  ..., 2.4124e-02,
           1.9958e-01, 6.6211e-01]],

         [[6.6936e-05, 7.3314e-06, 3.3722e-02,  ..., 8.6792e-02,
           3.9673e-01, 2.8906e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1505e-04, 8.4519e-05, 1.5125e-01,  ..., 1.3847e-02,
           2.6794e-02, 5.7861e-02]],

         [[1.3423e-04, 1.7035e-04, 1.6260e-01,  ..., 2.4368e-02,
           4.3762e-02, 1.9617e-01]],

         [[5.5742e-04, 1.2922e-04, 1.5149e-01,  ..., 2.4918e-02,
           7.5256e-02, 4.9927e-02]],

         ...,

         [[8.7142e-05, 2.0206e-05, 5.1147e-02,  ..., 6.5674e-02,
           4.0234e-01, 2.9443e-01]],

         [[2.0337e-04, 5.2357e-04, 3.7061e-01,  ..., 7.6332e-03,
           4.0619e-02, 4.7852e-02]],

         [[1.4532e-04, 6.8903e-05, 1.8188e-01,  ..., 6.7444e-02,
           1.5088e-01, 1.6956e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.2696e-05, 3.3975e-06, 4.7588e-04,  ..., 1.6943e-01,
           4.4653e-01, 8.2581e-02]],

         [[9.0551e-04, 2.2900e-04, 2.4624e-03,  ..., 1.0468e-01,
           4.9438e-02, 5.3174e-01]],

         [[2.9802e-07, 0.0000e+00, 8.3447e-07,  ..., 9.0456e-04,
           4.0550e-03, 9.9219e-01]],

         ...,

         [[1.7881e-07, 5.9605e-08, 2.9802e-07,  ..., 7.4446e-05,
           9.0694e-04, 9.9463e-01]],

         [[6.6071e-03, 8.7509e-03, 4.2648e-03,  ..., 7.1045e-02,
           1.5358e-02, 2.3788e-02]],

         [[4.8370e-03, 1.6203e-03, 5.7745e-04,  ..., 1.1971e-02,
           2.0370e-02, 1.6876e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2171e-04, 3.9911e-04, 3.7270e-03,  ..., 4.6143e-02,
           1.6101e-01, 4.9585e-01]],

         [[2.3842e-07, 0.0000e+00, 0.0000e+00,  ..., 1.0016e-01,
           8.1299e-01, 7.8003e-02]],

         [[1.7393e-04, 7.6723e-04, 1.0933e-02,  ..., 5.2155e-02,
           1.1749e-01, 6.9092e-02]],

         ...,

         [[2.1915e-03, 2.3499e-03, 5.5084e-03,  ..., 1.5479e-01,
           1.1414e-01, 2.7588e-01]],

         [[1.4372e-03, 7.4158e-03, 3.0975e-02,  ..., 4.1992e-02,
           1.0658e-02, 1.8417e-02]],

         [[7.6294e-04, 3.7861e-03, 4.3564e-03,  ..., 6.6040e-02,
           1.0229e-01, 5.1453e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9347e-02, 5.6686e-03, 3.1616e-02,  ..., 3.2867e-02,
           4.4922e-02, 3.9978e-02]],

         [[1.7757e-03, 8.0681e-04, 3.9101e-03,  ..., 5.2185e-03,
           9.5978e-03, 1.0544e-02]],

         [[7.3853e-03, 1.2197e-03, 1.4772e-03,  ..., 2.2568e-02,
           2.3376e-02, 2.1454e-02]],

         ...,

         [[2.8074e-05, 3.0994e-06, 2.0123e-04,  ..., 3.0334e-02,
           4.0576e-01, 4.1553e-01]],

         [[2.8954e-03, 9.6941e-04, 2.5940e-03,  ..., 4.2877e-02,
           2.8778e-02, 4.3549e-02]],

         [[2.3670e-03, 5.1355e-04, 2.4986e-03,  ..., 1.4185e-01,
           1.1310e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6904e-01, 1.7822e-05, 1.7405e-05,  ..., 2.2202e-03,
           4.9438e-03, 8.0872e-03]],

         [[9.2041e-01, 6.3181e-06, 5.2452e-06,  ..., 1.4114e-02,
           2.1988e-02, 1.6815e-02]],

         [[9.3750e-01, 1.7881e-07, 5.5432e-06,  ..., 5.4779e-03,
           1.2634e-02, 4.6387e-03]],

         ...,

         [[5.0586e-01, 2.2960e-04, 1.3962e-03,  ..., 2.5606e-04,
           3.5596e-04, 1.8494e-02]],

         [[5.3076e-01, 8.3804e-05, 1.2522e-03,  ..., 5.9319e-03,
           4.4250e-02, 1.0780e-02]],

         [[3.8818e-01, 2.3401e-04, 4.6921e-04,  ..., 4.1412e-02,
           8.2474e-03, 1.0468e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.3350e-01, 4.7684e-06, 1.2100e-05,  ..., 8.2970e-04,
           8.1837e-05, 1.0269e-02]],

         [[9.2188e-01, 0.0000e+00, 0.0000e+00,  ..., 2.9392e-03,
           2.6077e-02, 4.5288e-02]],

         [[9.1504e-01, 1.5974e-05, 1.6093e-06,  ..., 2.9068e-03,
           1.8677e-02, 1.3496e-02]],

         ...,

         [[3.5742e-01, 1.8167e-04, 2.0897e-04,  ..., 8.0688e-02,
           7.2876e-02, 6.6566e-03]],

         [[2.4536e-01, 5.3358e-04, 2.6298e-04,  ..., 8.7128e-03,
           4.6577e-03, 8.7051e-03]],

         [[6.9629e-01, 1.6451e-05, 1.2136e-04,  ..., 1.1665e-02,
           1.6992e-01, 5.8838e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7383e-01, 1.2094e-04, 5.8234e-05,  ..., 2.8320e-02,
           5.9631e-02, 4.9744e-02]],

         [[6.5967e-01, 1.0252e-05, 9.0003e-06,  ..., 1.6739e-02,
           7.0435e-02, 1.5884e-02]],

         [[6.6797e-01, 2.8789e-05, 4.7028e-05,  ..., 4.8584e-02,
           1.5076e-01, 1.0474e-01]],

         ...,

         [[8.3691e-01, 2.3842e-07, 4.1723e-07,  ..., 2.2769e-05,
           6.0797e-06, 1.5442e-01]],

         [[8.8135e-01, 2.9802e-07, 3.8147e-06,  ..., 5.9605e-07,
           2.3842e-07, 3.5614e-02]],

         [[9.8438e-01, 5.9605e-08, 2.9802e-07,  ..., 2.9802e-07,
           5.9605e-08, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7563e-01, 4.0531e-06, 2.5034e-06,  ..., 1.7371e-01,
           3.5742e-01, 2.5681e-02]],

         [[9.4385e-01, 1.4305e-06, 1.9670e-06,  ..., 1.9097e-04,
           2.9802e-04, 3.9406e-03]],

         [[7.7783e-01, 1.1218e-04, 2.0087e-05,  ..., 1.4324e-03,
           7.7133e-03, 1.5778e-02]],

         ...,

         [[5.8789e-01, 3.1638e-04, 9.3162e-05,  ..., 2.0660e-02,
           1.9128e-01, 5.0293e-02]],

         [[3.9868e-01, 4.1187e-05, 1.1086e-05,  ..., 8.6288e-03,
           3.2422e-01, 6.8176e-02]],

         [[8.3545e-01, 3.5644e-05, 6.7353e-05,  ..., 8.6136e-03,
           2.3438e-02, 2.0325e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6338e-01, 2.9802e-06, 3.0994e-06,  ..., 7.2670e-04,
           4.0855e-03, 3.6335e-03]],

         [[9.6289e-01, 7.7486e-06, 1.6093e-06,  ..., 9.6798e-04,
           2.1515e-03, 1.5274e-02]],

         [[9.5508e-01, 2.0266e-06, 4.7684e-07,  ..., 7.0534e-03,
           9.6436e-03, 1.1055e-02]],

         ...,

         [[7.9395e-01, 1.8239e-04, 5.2631e-05,  ..., 5.4016e-03,
           2.6215e-02, 1.5022e-02]],

         [[8.9355e-01, 1.1694e-04, 1.2636e-05,  ..., 2.2721e-04,
           8.0566e-03, 1.3733e-02]],

         [[9.4824e-01, 6.1989e-06, 2.2054e-06,  ..., 1.4915e-02,
           1.2932e-02, 3.9520e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7031e-01, 1.6203e-03, 1.2817e-03,  ..., 2.2316e-03,
           1.2329e-02, 8.2626e-03]],

         [[9.3945e-01, 1.1683e-05, 2.0504e-05,  ..., 5.4893e-03,
           2.3060e-03, 1.5974e-03]],

         [[8.8232e-01, 1.5855e-05, 8.1658e-06,  ..., 2.5578e-03,
           2.1851e-02, 7.7896e-03]],

         ...,

         [[7.2607e-01, 6.2048e-05, 2.3842e-05,  ..., 8.9569e-03,
           1.3336e-02, 1.9312e-01]],

         [[7.4463e-01, 2.7418e-06, 5.9605e-06,  ..., 5.9998e-02,
           2.8057e-03, 3.0289e-03]],

         [[7.0264e-01, 5.9009e-06, 6.9141e-06,  ..., 1.1787e-02,
           2.1692e-01, 2.4048e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9226e-01, 1.6105e-04, 7.7724e-04,  ..., 1.9302e-02,
           1.8097e-02, 4.7340e-03]],

         [[7.7441e-01, 4.7684e-07, 1.9670e-06,  ..., 2.3453e-02,
           7.9712e-02, 2.3453e-02]],

         [[7.6953e-01, 1.0014e-05, 2.1517e-05,  ..., 5.1178e-02,
           9.2468e-02, 2.5955e-02]],

         ...,

         [[8.7158e-01, 6.2764e-05, 4.9651e-05,  ..., 8.6975e-04,
           3.1872e-03, 1.5364e-03]],

         [[6.0059e-01, 2.2948e-05, 1.3292e-05,  ..., 4.2305e-03,
           1.7078e-01, 9.6924e-02]],

         [[4.7046e-01, 5.4836e-06, 8.1658e-06,  ..., 1.0999e-01,
           6.2195e-02, 2.2705e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6504e-01, 1.0842e-04, 2.7180e-05,  ..., 2.1875e-05,
           2.1064e-04, 1.7715e-02]],

         [[9.5361e-01, 6.9737e-06, 4.8280e-06,  ..., 1.5421e-03,
           3.5305e-03, 5.2986e-03]],

         [[7.9297e-01, 3.3975e-06, 1.6212e-05,  ..., 1.2112e-03,
           1.4275e-02, 1.5617e-02]],

         ...,

         [[8.3887e-01, 1.4365e-04, 8.8871e-05,  ..., 1.1215e-03,
           9.0485e-03, 5.7297e-03]],

         [[8.5938e-01, 7.1764e-05, 7.5221e-05,  ..., 2.1271e-02,
           1.3206e-02, 5.5962e-03]],

         [[7.5439e-01, 1.1101e-03, 4.8494e-04,  ..., 9.8114e-03,
           1.2016e-02, 9.5596e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2783e-01, 1.2517e-06, 9.5367e-07,  ..., 1.8990e-04,
           2.3499e-03, 4.6191e-01]],

         [[3.8940e-01, 4.6515e-04, 9.2936e-04,  ..., 1.4391e-03,
           1.3123e-02, 5.6549e-02]],

         [[7.3730e-01, 3.4523e-04, 5.2595e-04,  ..., 2.3861e-03,
           7.7095e-03, 2.7542e-02]],

         ...,

         [[4.4849e-01, 1.6379e-04, 1.9991e-04,  ..., 1.5228e-02,
           5.0545e-03, 8.6164e-04]],

         [[3.8403e-01, 4.1246e-05, 2.8491e-05,  ..., 7.5806e-02,
           2.3071e-01, 3.9185e-02]],

         [[2.9907e-01, 4.2260e-05, 3.8922e-05,  ..., 1.0231e-02,
           1.5030e-02, 6.2598e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9473e-01, 3.0279e-05, 4.2558e-05,  ..., 9.1171e-03,
           1.0826e-02, 3.4912e-01]],

         [[4.5776e-01, 4.0412e-05, 7.2300e-05,  ..., 1.0992e-01,
           1.2341e-01, 5.1697e-02]],

         [[9.8145e-02, 9.3281e-05, 1.1927e-04,  ..., 3.6407e-02,
           3.9917e-02, 6.9275e-02]],

         ...,

         [[2.0508e-01, 3.5763e-06, 8.6427e-06,  ..., 5.1498e-04,
           2.6646e-03, 7.8125e-01]],

         [[7.3438e-01, 2.5928e-05, 5.9545e-05,  ..., 2.6535e-02,
           6.9466e-03, 2.0950e-02]],

         [[6.9336e-01, 7.9751e-05, 1.6868e-05,  ..., 4.4250e-02,
           7.5134e-02, 3.6499e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0146e-01, 3.9554e-04, 1.2722e-03,  ..., 1.7929e-02,
           2.2583e-02, 1.4868e-01]],

         [[6.9043e-01, 3.8683e-05, 1.4722e-04,  ..., 4.3884e-02,
           1.4709e-01, 2.6459e-02]],

         [[6.6553e-01, 1.2445e-04, 1.2994e-04,  ..., 1.3176e-02,
           3.3259e-05, 9.3231e-03]],

         ...,

         [[3.3862e-01, 0.0000e+00, 0.0000e+00,  ..., 1.4519e-02,
           6.0986e-01, 2.9083e-02]],

         [[5.7373e-01, 4.9496e-04, 8.4209e-04,  ..., 1.7214e-03,
           1.6651e-03, 2.3854e-04]],

         [[7.3193e-01, 0.0000e+00, 5.9605e-08,  ..., 8.5068e-03,
           6.1951e-02, 1.9019e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2969e-01, 5.4717e-05, 2.4557e-05,  ..., 2.9968e-02,
           2.4261e-02, 5.5389e-02]],

         [[9.1125e-02, 1.2445e-03, 1.3647e-03,  ..., 1.6327e-03,
           2.8687e-03, 2.2488e-03]],

         [[1.9873e-01, 2.2411e-04, 8.1658e-05,  ..., 3.6743e-02,
           2.2607e-01, 2.2888e-02]],

         ...,

         [[7.5732e-01, 2.6941e-05, 2.0802e-05,  ..., 4.2145e-02,
           6.5796e-02, 1.2169e-02]],

         [[5.7178e-01, 1.7881e-06, 2.9206e-06,  ..., 1.0815e-01,
           1.8286e-01, 6.2927e-02]],

         [[7.0117e-01, 4.3213e-05, 2.7895e-05,  ..., 2.2797e-02,
           4.1321e-02, 7.0007e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6821e-01, 9.5367e-07, 2.3842e-07,  ..., 5.1022e-04,
           6.6261e-03, 8.0566e-01]],

         [[1.3025e-01, 1.4305e-06, 2.5630e-06,  ..., 1.6724e-01,
           2.3218e-01, 2.0676e-02]],

         [[1.1017e-01, 3.6740e-04, 5.3120e-04,  ..., 1.1406e-02,
           1.8265e-02, 2.6260e-02]],

         ...,

         [[5.9717e-01, 8.4817e-05, 8.1599e-05,  ..., 3.9864e-03,
           1.3281e-01, 1.9419e-04]],

         [[4.9561e-01, 5.3048e-06, 2.3246e-06,  ..., 1.0266e-01,
           3.0880e-03, 3.6133e-02]],

         [[6.9922e-01, 8.6963e-05, 1.6248e-04,  ..., 9.1505e-04,
           5.6458e-04, 3.9093e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5808e-01, 1.0729e-06, 2.1458e-06,  ..., 8.2169e-03,
           7.1680e-01, 4.8401e-02]],

         [[2.1057e-01, 5.4479e-05, 2.4915e-05,  ..., 1.5527e-01,
           8.9172e-02, 9.7168e-02]],

         [[7.9834e-01, 3.3975e-06, 1.3411e-05,  ..., 5.8861e-03,
           1.2006e-01, 1.1749e-02]],

         ...,

         [[7.1289e-01, 2.2650e-06, 1.0312e-05,  ..., 2.0233e-02,
           7.4829e-02, 1.5091e-02]],

         [[6.6650e-01, 1.0133e-06, 3.8147e-06,  ..., 2.1801e-03,
           2.5098e-01, 2.2629e-02]],

         [[6.2598e-01, 3.7074e-05, 3.3617e-05,  ..., 2.8038e-03,
           2.5940e-02, 4.4983e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2825e-02, 4.2176e-04, 9.8228e-04,  ..., 1.0357e-03,
           1.0815e-03, 1.0612e-02]],

         [[1.8115e-01, 5.6207e-05, 1.1462e-04,  ..., 2.2141e-02,
           2.2316e-03, 1.6296e-01]],

         [[7.8674e-02, 5.9605e-07, 7.7486e-07,  ..., 1.0339e-01,
           5.5908e-01, 1.0626e-01]],

         ...,

         [[7.7051e-01, 1.2422e-04, 4.7147e-05,  ..., 5.6419e-03,
           2.6035e-03, 5.6366e-02]],

         [[7.9883e-01, 1.4663e-05, 9.0003e-06,  ..., 1.3053e-05,
           5.0664e-06, 9.0103e-03]],

         [[4.8267e-01, 1.9038e-04, 3.4475e-04,  ..., 7.0038e-03,
           1.6422e-03, 1.8559e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9713e-02, 3.6788e-04, 1.6975e-04,  ..., 6.8359e-03,
           2.0096e-02, 4.5891e-03]],

         [[2.5757e-01, 1.9121e-04, 6.3956e-05,  ..., 6.1760e-03,
           4.1275e-03, 4.3106e-03]],

         [[2.4048e-01, 4.2915e-06, 2.5630e-06,  ..., 2.0898e-01,
           1.8433e-01, 2.1558e-01]],

         ...,

         [[3.8379e-01, 2.3842e-07, 2.9802e-07,  ..., 1.4877e-02,
           3.4607e-02, 3.2177e-03]],

         [[4.4800e-01, 4.1723e-07, 3.5763e-07,  ..., 2.3926e-02,
           3.2764e-01, 6.5063e-02]],

         [[2.9907e-01, 1.4901e-06, 1.6689e-06,  ..., 6.5674e-02,
           1.0663e-01, 4.7046e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6826e-01, 5.3465e-05, 2.3723e-05,  ..., 5.3375e-02,
           1.2482e-02, 2.5208e-02]],

         [[2.2491e-02, 1.0133e-06, 2.3246e-06,  ..., 1.8530e-01,
           2.6978e-01, 6.0181e-02]],

         [[2.3413e-01, 7.3910e-06, 2.3246e-06,  ..., 4.0894e-03,
           5.8594e-03, 7.4365e-01]],

         ...,

         [[1.3757e-01, 3.7026e-04, 9.0361e-05,  ..., 3.7599e-04,
           1.7834e-04, 3.8433e-03]],

         [[4.7974e-01, 2.6679e-04, 1.5783e-04,  ..., 6.4392e-02,
           1.0269e-02, 2.1744e-02]],

         [[1.2408e-01, 7.3910e-05, 9.3460e-05,  ..., 2.4246e-02,
           3.1929e-03, 1.3603e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4790e-01, 6.6757e-06, 8.4639e-06,  ..., 2.2949e-02,
           4.9347e-02, 2.0142e-01]],

         [[6.9385e-01, 4.7088e-06, 7.1526e-07,  ..., 9.6464e-04,
           7.6294e-04, 1.3180e-03]],

         [[4.9731e-01, 1.2887e-04, 2.8968e-05,  ..., 2.9633e-02,
           2.1347e-02, 3.3760e-03]],

         ...,

         [[8.4595e-02, 2.6917e-04, 3.3498e-04,  ..., 7.7438e-03,
           4.6959e-03, 7.3891e-03]],

         [[6.8799e-01, 1.6093e-05, 1.4067e-05,  ..., 1.4961e-02,
           3.4790e-02, 9.7534e-02]],

         [[1.8054e-01, 4.2224e-04, 9.7370e-04,  ..., 7.9346e-03,
           2.0584e-02, 5.3802e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4424e-01, 4.4703e-05, 5.6744e-05,  ..., 3.2013e-02,
           6.4209e-02, 1.4575e-01]],

         [[2.1509e-01, 1.4007e-05, 4.8459e-05,  ..., 4.7241e-02,
           2.2675e-02, 7.5500e-02]],

         [[3.9990e-01, 2.4438e-05, 3.5822e-05,  ..., 7.6752e-03,
           1.6113e-02, 9.6436e-02]],

         ...,

         [[3.1067e-02, 5.0485e-05, 4.6730e-05,  ..., 7.1106e-02,
           5.2338e-03, 4.8126e-02]],

         [[3.0566e-01, 1.0252e-04, 7.0453e-05,  ..., 2.3575e-02,
           1.5221e-02, 1.5613e-01]],

         [[6.4160e-01, 7.1526e-05, 1.0091e-04,  ..., 1.9379e-02,
           1.4854e-02, 6.6589e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9789e-05, 1.4544e-05,  ..., 4.0344e-02,
           5.1392e-02, 3.1470e-01]],

         [[6.6846e-01, 1.8811e-04, 3.9673e-04,  ..., 6.8665e-03,
           5.6038e-03, 1.5480e-02]],

         [[3.8477e-01, 1.3065e-04, 2.0790e-04,  ..., 6.7383e-02,
           6.1371e-02, 1.0437e-01]],

         ...,

         [[5.8008e-01, 1.3328e-04, 8.9109e-05,  ..., 3.9795e-02,
           5.9265e-02, 1.0730e-01]],

         [[4.1431e-01, 2.8133e-05, 1.2553e-04,  ..., 3.6469e-02,
           1.4038e-03, 1.8970e-01]],

         [[4.7144e-01, 2.0385e-04, 2.1279e-04,  ..., 2.3102e-02,
           4.3518e-02, 8.9966e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 8.0526e-05, 6.2525e-05,  ..., 5.5481e-02,
           1.2598e-01, 1.1121e-01]],

         [[1.7786e-01, 4.0054e-05, 2.0742e-05,  ..., 1.7932e-01,
           2.2668e-01, 1.5210e-01]],

         [[4.2480e-01, 2.0897e-04, 1.7095e-04,  ..., 3.7415e-02,
           5.4550e-03, 1.8234e-02]],

         ...,

         [[4.0698e-01, 0.0000e+00, 8.9407e-07,  ..., 2.9251e-02,
           4.8315e-01, 6.9092e-02]],

         [[1.9678e-01, 5.9605e-07, 3.9935e-06,  ..., 6.1920e-02,
           2.3880e-02, 1.8738e-02]],

         [[1.7822e-01, 0.0000e+00, 5.9605e-07,  ..., 6.3538e-02,
           7.0459e-01, 3.7354e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2619e-02, 1.4901e-06, 1.6093e-06,  ..., 5.6122e-02,
           1.7834e-03, 1.7517e-02]],

         [[1.8262e-01, 4.8280e-06, 2.0862e-06,  ..., 7.3624e-03,
           1.3863e-02, 7.3975e-01]],

         [[4.9835e-02, 3.8028e-05, 2.1851e-04,  ..., 3.0994e-04,
           2.6512e-04, 1.0757e-03]],

         ...,

         [[9.4910e-03, 3.1233e-05, 1.2046e-04,  ..., 1.7490e-03,
           2.9755e-03, 3.4523e-03]],

         [[1.8768e-02, 0.0000e+00, 0.0000e+00,  ..., 7.5439e-02,
           2.2125e-02, 1.9073e-02]],

         [[5.4077e-02, 0.0000e+00, 5.9605e-08,  ..., 1.7871e-01,
           5.7227e-01, 7.6233e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.6738e-01, 1.2767e-04, 5.2977e-04,  ..., 7.0038e-03,
           1.9882e-02, 2.4536e-02]],

         [[9.6924e-01, 1.8096e-04, 1.3137e-04,  ..., 5.5969e-05,
           2.4021e-04, 6.0987e-04]],

         [[9.3652e-01, 6.1989e-06, 1.5438e-05,  ..., 3.1490e-03,
           3.4313e-03, 4.6883e-03]],

         ...,

         [[3.2324e-01, 4.1723e-07, 4.7684e-07,  ..., 1.3062e-01,
           1.7041e-01, 8.4351e-02]],

         [[8.4668e-01, 6.3181e-06, 9.1791e-06,  ..., 5.3883e-04,
           8.4782e-04, 4.3411e-03]],

         [[1.3220e-01, 4.6492e-06, 1.0371e-05,  ..., 5.4026e-04,
           1.4687e-03, 8.4863e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8958e-01, 2.3842e-07, 8.9407e-07,  ..., 3.6896e-02,
           1.4307e-01, 3.0298e-01]],

         [[4.2822e-01, 6.5565e-07, 7.1526e-07,  ..., 6.6162e-02,
           1.5747e-01, 1.2366e-01]],

         [[3.3252e-01, 3.1531e-05, 4.3750e-05,  ..., 2.6886e-02,
           7.7148e-02, 5.7129e-02]],

         ...,

         [[1.6113e-01, 6.5565e-07, 2.5034e-06,  ..., 1.0321e-01,
           5.5664e-02, 4.2694e-02]],

         [[4.2993e-01, 4.5896e-06, 2.3365e-05,  ..., 4.6051e-02,
           3.5309e-02, 3.1403e-02]],

         [[7.0410e-01, 3.2783e-06, 1.7762e-05,  ..., 4.0985e-02,
           5.1270e-03, 1.0361e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4053e-01, 1.9789e-05, 1.2755e-05,  ..., 1.0565e-01,
           8.6670e-03, 1.9128e-01]],

         [[5.1465e-01, 6.7949e-06, 5.1856e-06,  ..., 4.4983e-02,
           1.0513e-02, 1.7517e-01]],

         [[4.2896e-01, 5.4538e-05, 1.4532e-04,  ..., 3.0838e-02,
           2.0065e-02, 6.4758e-02]],

         ...,

         [[1.0645e-01, 3.4571e-06, 4.4823e-05,  ..., 1.4183e-02,
           1.8219e-02, 6.0539e-03]],

         [[1.8750e-01, 3.3379e-05, 2.1613e-04,  ..., 1.7538e-03,
           1.2150e-03, 6.9189e-04]],

         [[2.7161e-02, 2.7418e-06, 6.3181e-06,  ..., 9.0942e-03,
           6.1569e-03, 1.3031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0828e-01, 5.9605e-08, 3.5763e-07,  ..., 2.1484e-02,
           8.0615e-01, 4.1901e-02]],

         [[5.0232e-02, 1.7881e-07, 1.6093e-06,  ..., 3.1605e-03,
           2.3670e-03, 1.5198e-02]],

         [[3.6084e-01, 3.5763e-07, 2.6822e-06,  ..., 1.6846e-02,
           4.2847e-01, 1.7651e-01]],

         ...,

         [[6.5918e-01, 5.6446e-05, 2.6345e-05,  ..., 1.5625e-02,
           1.5869e-02, 4.3152e-02]],

         [[2.6392e-01, 4.4703e-06, 5.8413e-06,  ..., 1.5759e-01,
           7.2144e-02, 8.4351e-02]],

         [[1.0876e-01, 2.3067e-05, 7.6652e-05,  ..., 5.6854e-02,
           1.1945e-01, 3.8867e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.3887e-01, 3.8147e-06, 1.6749e-05,  ..., 4.9591e-02,
           1.0834e-01, 3.5522e-01]],

         [[7.8174e-01, 8.7023e-06, 3.0756e-05,  ..., 3.5034e-02,
           2.5131e-02, 4.5166e-02]],

         [[4.8779e-01, 6.3956e-05, 1.2898e-04,  ..., 1.0468e-01,
           1.7288e-02, 9.9854e-02]],

         ...,

         [[1.0498e-01, 1.1921e-05, 6.8545e-06,  ..., 3.4088e-02,
           1.0239e-02, 1.8539e-02]],

         [[4.9072e-02, 1.8954e-05, 1.1975e-04,  ..., 1.4503e-02,
           4.8309e-02, 8.1689e-01]],

         [[2.3120e-01, 7.4506e-06, 1.7464e-05,  ..., 1.2189e-01,
           1.5894e-01, 6.1279e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6875e-01, 5.7518e-05, 2.1303e-04,  ..., 9.3750e-02,
           2.1289e-01, 7.5317e-02]],

         [[2.9761e-01, 1.9610e-05, 5.5969e-05,  ..., 3.3783e-02,
           1.8909e-01, 1.7224e-01]],

         [[6.5479e-01, 1.5008e-04, 5.3596e-04,  ..., 1.6449e-02,
           1.9852e-02, 1.7624e-02]],

         ...,

         [[5.1123e-01, 4.8220e-05, 3.0303e-04,  ..., 3.0106e-02,
           1.7029e-02, 9.7733e-03]],

         [[8.8965e-01, 6.0797e-06, 4.3654e-04,  ..., 2.5768e-03,
           1.4023e-02, 8.4076e-03]],

         [[5.7080e-01, 5.4240e-06, 4.7266e-05,  ..., 1.8387e-02,
           1.8173e-02, 2.6077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9743e-05, 2.6286e-05, 1.7261e-01,  ..., 1.0974e-01,
           4.6444e-04, 5.3177e-03]],

         [[1.3292e-05, 7.0333e-06, 5.0598e-02,  ..., 4.1290e-02,
           5.7220e-03, 3.2864e-03]],

         [[9.1672e-05, 4.7445e-05, 6.9092e-02,  ..., 1.0699e-01,
           2.3364e-01, 1.1212e-01]],

         ...,

         [[8.9109e-05, 7.7486e-06, 6.7688e-02,  ..., 7.7271e-02,
           1.3892e-01, 1.6235e-01]],

         [[4.4560e-04, 1.3046e-03, 2.9517e-01,  ..., 1.0460e-02,
           4.3297e-03, 6.6490e-03]],

         [[1.6272e-04, 8.7440e-05, 3.9258e-01,  ..., 4.8294e-03,
           8.3923e-04, 1.4496e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.6921e-04, 4.7493e-04, 3.1104e-01,  ..., 7.4844e-03,
           2.5320e-04, 2.2926e-03]],

         [[2.2364e-04, 2.8944e-04, 2.7246e-01,  ..., 2.4567e-02,
           4.3273e-04, 4.0741e-03]],

         [[1.9777e-04, 9.9242e-05, 3.2617e-01,  ..., 6.2599e-03,
           5.6219e-04, 2.7514e-04]],

         ...,

         [[1.5509e-04, 3.4404e-04, 3.0371e-01,  ..., 3.8548e-03,
           1.5574e-03, 5.9366e-04]],

         [[9.9540e-06, 5.8830e-05, 3.0688e-01,  ..., 7.8659e-03,
           8.7500e-05, 4.0007e-04]],

         [[1.9073e-05, 1.1712e-04, 1.9128e-01,  ..., 5.2605e-03,
           1.4186e-04, 3.6502e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1127e-04, 2.1541e-04, 2.3218e-01,  ..., 1.1269e-02,
           1.0663e-04, 2.3098e-03]],

         [[4.0865e-04, 1.7524e-05, 3.1421e-01,  ..., 5.3825e-03,
           6.8712e-04, 2.0385e-05]],

         [[5.1451e-04, 4.2415e-04, 3.1104e-01,  ..., 9.5444e-03,
           4.1509e-04, 4.7922e-05]],

         ...,

         [[1.2267e-04, 2.7061e-05, 1.2695e-01,  ..., 1.0199e-01,
           6.6895e-02, 7.4890e-02]],

         [[7.7903e-05, 1.4806e-04, 3.1348e-01,  ..., 5.6076e-03,
           2.0046e-03, 4.9067e-04]],

         [[5.1260e-04, 2.0289e-04, 2.2046e-01,  ..., 5.9433e-03,
           9.4795e-04, 6.5279e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6061e-05, 1.4305e-06, 2.5902e-03,  ..., 7.1106e-02,
           2.1228e-01, 3.7750e-02]],

         [[4.1056e-04, 7.3814e-04, 1.2805e-01,  ..., 3.5706e-02,
           3.7354e-02, 3.6133e-02]],

         [[6.1226e-04, 3.0079e-03, 1.2201e-01,  ..., 6.5923e-05,
           7.1824e-05, 5.2452e-04]],

         ...,

         [[4.1847e-03, 5.6148e-05, 1.9153e-01,  ..., 1.2169e-02,
           1.8600e-02, 5.2765e-02]],

         [[1.1444e-03, 3.4332e-05, 1.1353e-02,  ..., 8.1299e-02,
           6.1523e-01, 1.1920e-01]],

         [[4.1127e-06, 8.3447e-07, 6.1083e-04,  ..., 9.9411e-03,
           1.1261e-02, 3.0136e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6670e-03, 9.6035e-04, 2.3938e-01,  ..., 1.1482e-03,
           3.2845e-03, 4.7882e-02]],

         [[7.7057e-04, 1.0767e-03, 2.6318e-01,  ..., 3.3989e-03,
           1.5621e-03, 1.4168e-02]],

         [[3.7766e-03, 8.3637e-04, 2.1362e-01,  ..., 1.7746e-02,
           5.7281e-02, 2.8915e-02]],

         ...,

         [[2.9302e-04, 3.7372e-05, 8.6487e-02,  ..., 4.9934e-03,
           1.2169e-02, 2.4765e-02]],

         [[1.8120e-04, 8.3780e-04, 1.0309e-01,  ..., 6.0608e-02,
           7.6599e-02, 4.4775e-01]],

         [[3.3081e-05, 3.9935e-06, 4.0009e-02,  ..., 1.2915e-01,
           3.6206e-01, 1.8494e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[5.7280e-05, 2.4438e-06, 1.5569e-04,  ..., 2.9678e-02,
           1.8750e-01, 6.1475e-01]],

         [[1.3527e-02, 2.5826e-03, 7.7095e-03,  ..., 9.0103e-03,
           3.6774e-02, 2.0837e-01]],

         [[3.8147e-05, 3.8147e-05, 1.5396e-02,  ..., 1.7107e-04,
           1.3316e-04, 5.0977e-01]],

         ...,

         [[1.7853e-03, 1.3905e-03, 1.3065e-03,  ..., 9.0637e-03,
           3.4760e-02, 7.0610e-03]],

         [[6.9466e-03, 5.4092e-03, 3.7537e-02,  ..., 3.0823e-03,
           1.5656e-02, 2.0300e-01]],

         [[4.2686e-03, 4.0131e-03, 1.1379e-04,  ..., 2.4567e-02,
           4.4495e-02, 8.2321e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4717e-05, 4.8018e-04, 2.2202e-03,  ..., 3.5828e-02,
           7.7026e-02, 8.5938e-02]],

         [[8.3208e-05, 7.3910e-06, 2.0862e-06,  ..., 1.2561e-01,
           5.2100e-01, 3.3105e-01]],

         [[3.6201e-03, 5.2681e-03, 2.1164e-02,  ..., 4.2343e-03,
           3.9787e-03, 9.3918e-03]],

         ...,

         [[6.6423e-04, 4.5300e-04, 2.6894e-03,  ..., 2.6535e-02,
           1.3657e-02, 5.8868e-02]],

         [[3.8528e-03, 5.6953e-03, 5.0964e-02,  ..., 2.8973e-03,
           6.6566e-03, 1.7471e-02]],

         [[5.9223e-04, 4.6043e-03, 3.2387e-03,  ..., 4.3335e-02,
           6.5063e-02, 2.0349e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1244e-03, 9.2268e-05, 1.7107e-04,  ..., 1.6943e-01,
           1.8909e-01, 1.2695e-01]],

         [[4.9973e-03, 1.9188e-03, 3.8013e-03,  ..., 1.5511e-02,
           1.4343e-02, 1.2032e-02]],

         [[1.3475e-03, 1.2100e-04, 1.0151e-04,  ..., 9.0332e-03,
           7.9727e-03, 4.6204e-02]],

         ...,

         [[1.2112e-04, 2.9802e-07, 3.5763e-06,  ..., 2.2202e-02,
           2.8125e-01, 6.4355e-01]],

         [[4.1618e-03, 6.4325e-04, 1.5678e-03,  ..., 1.3123e-02,
           6.2981e-03, 5.6122e-02]],

         [[1.7914e-02, 1.6794e-03, 1.0132e-02,  ..., 1.7227e-02,
           3.2196e-02, 1.5601e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8574e-01, 5.9605e-07, 4.2915e-06,  ..., 2.5730e-03,
           2.3556e-03, 1.7120e-02]],

         [[9.3799e-01, 1.6689e-05, 3.0398e-06,  ..., 6.5079e-03,
           9.3002e-03, 1.7075e-02]],

         [[8.5156e-01, 3.5763e-07, 1.6093e-06,  ..., 8.0414e-03,
           1.4000e-03, 2.5986e-02]],

         ...,

         [[8.1641e-01, 2.7609e-04, 1.2865e-03,  ..., 2.6608e-04,
           2.0862e-04, 3.2215e-03]],

         [[1.3391e-01, 5.9187e-05, 1.5533e-04,  ..., 3.1052e-02,
           2.0691e-02, 5.5786e-02]],

         [[8.2764e-01, 1.8513e-04, 6.0415e-04,  ..., 3.3360e-03,
           8.2932e-03, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4082e-01, 1.2815e-05, 4.9472e-06,  ..., 9.9421e-05,
           2.2471e-05, 9.0408e-03]],

         [[9.6875e-01, 0.0000e+00, 0.0000e+00,  ..., 1.1683e-03,
           1.8860e-02, 9.5520e-03]],

         [[8.7158e-01, 7.8392e-04, 7.6115e-05,  ..., 4.6997e-03,
           1.8311e-02, 8.7814e-03]],

         ...,

         [[3.6353e-01, 1.9562e-04, 8.5950e-05,  ..., 6.6986e-03,
           7.6408e-03, 1.4946e-02]],

         [[7.4890e-02, 2.3842e-03, 1.1606e-03,  ..., 3.6550e-04,
           8.7166e-04, 3.4447e-03]],

         [[7.0264e-01, 7.9274e-06, 1.0967e-05,  ..., 8.3740e-02,
           8.8623e-02, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3315e-01, 3.6645e-04, 1.4246e-04,  ..., 4.4434e-02,
           3.3478e-02, 4.4617e-02]],

         [[6.6602e-01, 1.9121e-04, 1.7107e-05,  ..., 5.7755e-03,
           2.4891e-03, 2.5085e-02]],

         [[4.2334e-01, 5.2691e-05, 2.7478e-05,  ..., 1.9196e-02,
           4.5380e-02, 1.4610e-02]],

         ...,

         [[8.7305e-01, 1.2100e-05, 6.5565e-07,  ..., 4.5598e-05,
           3.8862e-05, 1.1383e-01]],

         [[8.6523e-01, 2.8193e-05, 3.9935e-06,  ..., 6.5565e-07,
           2.9802e-07, 1.7044e-02]],

         [[9.6240e-01, 9.3579e-06, 4.7684e-07,  ..., 1.1921e-07,
           5.9605e-08, 1.4771e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.7812e-01, 6.4969e-06, 7.1526e-07,  ..., 7.7271e-02,
           9.5703e-02, 1.1299e-02]],

         [[9.1650e-01, 1.2040e-05, 2.0266e-06,  ..., 2.2674e-04,
           1.6081e-04, 8.0490e-03]],

         [[7.5879e-01, 7.8630e-04, 9.1732e-05,  ..., 7.1573e-04,
           6.4898e-04, 3.1403e-02]],

         ...,

         [[8.0469e-01, 8.6963e-05, 2.8670e-05,  ..., 1.8072e-03,
           3.9434e-04, 1.5358e-02]],

         [[8.3740e-01, 3.0398e-05, 1.4424e-05,  ..., 1.1307e-02,
           1.6975e-03, 3.2318e-02]],

         [[9.2773e-01, 9.7334e-05, 6.3837e-05,  ..., 1.0424e-03,
           7.3910e-04, 1.2993e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4443e-01, 1.6391e-05, 1.1265e-05,  ..., 2.1469e-02,
           1.5762e-02, 5.6305e-02]],

         [[7.0654e-01, 9.2864e-05, 4.8876e-06,  ..., 2.9984e-03,
           3.9711e-03, 1.4355e-01]],

         [[4.2651e-01, 2.7418e-06, 7.1526e-07,  ..., 5.5771e-03,
           1.3336e-02, 4.0527e-02]],

         ...,

         [[2.9297e-01, 8.6188e-05, 3.4273e-05,  ..., 8.6609e-02,
           8.1970e-02, 2.0233e-02]],

         [[5.6543e-01, 7.2384e-04, 6.8963e-05,  ..., 1.4122e-02,
           3.4882e-02, 2.1454e-02]],

         [[1.4136e-01, 9.6560e-06, 1.8299e-05,  ..., 2.2308e-02,
           1.3725e-02, 8.2642e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9536e-01, 2.8591e-03, 2.5635e-03,  ..., 1.0271e-03,
           1.7681e-03, 4.7874e-03]],

         [[8.0664e-01, 6.0141e-05, 3.2759e-04,  ..., 6.3286e-03,
           3.4008e-03, 1.0979e-02]],

         [[4.4458e-01, 2.7323e-04, 6.2883e-05,  ..., 1.5918e-01,
           3.9490e-02, 2.3514e-02]],

         ...,

         [[8.4961e-01, 3.2234e-04, 7.1347e-05,  ..., 1.4639e-04,
           6.4731e-05, 4.7333e-02]],

         [[2.6172e-01, 2.5868e-05, 7.2479e-05,  ..., 1.2083e-03,
           5.4598e-04, 6.4354e-03]],

         [[9.5801e-01, 4.0770e-05, 3.9220e-05,  ..., 8.4000e-03,
           6.1560e-04, 5.9700e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8853e-01, 9.3555e-04, 2.1496e-03,  ..., 1.5211e-04,
           1.0455e-04, 1.2993e-02]],

         [[9.1357e-01, 7.6294e-06, 2.8610e-05,  ..., 5.3864e-03,
           5.1804e-03, 5.4512e-03]],

         [[3.0151e-01, 1.9312e-05, 2.1601e-04,  ..., 2.2888e-02,
           1.3130e-02, 1.9470e-02]],

         ...,

         [[1.5793e-02, 8.1718e-05, 3.1018e-04,  ..., 3.4094e-05,
           3.3021e-05, 5.3825e-03]],

         [[8.9990e-01, 7.9811e-05, 7.9334e-05,  ..., 1.4259e-02,
           1.4442e-02, 2.6951e-03]],

         [[5.7129e-01, 2.6047e-05, 3.0696e-05,  ..., 1.5116e-03,
           4.1628e-04, 1.4488e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7720e-01, 9.1672e-05, 9.0075e-04,  ..., 1.0967e-04,
           8.6427e-06, 6.0028e-02]],

         [[3.0298e-01, 6.5207e-05, 6.9678e-05,  ..., 4.1199e-04,
           8.4877e-04, 3.2715e-02]],

         [[7.0605e-01, 1.5497e-06, 2.4915e-05,  ..., 3.8195e-04,
           4.1068e-05, 4.6631e-02]],

         ...,

         [[8.5889e-01, 1.9276e-04, 7.3433e-05,  ..., 2.1629e-03,
           4.2496e-03, 7.8278e-03]],

         [[4.6973e-01, 8.4066e-04, 2.6608e-04,  ..., 1.0147e-02,
           5.7945e-03, 2.1820e-02]],

         [[3.1128e-01, 8.1301e-04, 2.5797e-04,  ..., 2.4414e-04,
           2.5082e-04, 2.2827e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0618e-01, 3.9339e-06, 6.5565e-06,  ..., 1.2434e-04,
           7.9060e-04, 7.6611e-01]],

         [[6.3818e-01, 1.8282e-03, 2.4261e-03,  ..., 3.2902e-03,
           8.8196e-03, 2.1000e-03]],

         [[6.1670e-01, 5.5790e-04, 5.7840e-04,  ..., 6.8398e-03,
           1.7365e-02, 5.8746e-02]],

         ...,

         [[5.0830e-01, 1.7643e-04, 2.5988e-04,  ..., 1.0155e-02,
           5.8556e-03, 9.6497e-02]],

         [[5.0000e-01, 2.0683e-05, 2.4140e-05,  ..., 2.7679e-02,
           6.4270e-02, 5.3650e-02]],

         [[4.3237e-01, 1.1754e-04, 1.7786e-04,  ..., 3.0017e-04,
           1.4486e-03, 4.4678e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9233e-01, 3.7122e-04, 5.2166e-04,  ..., 1.6127e-03,
           2.5272e-03, 3.5107e-01]],

         [[8.3252e-01, 2.0719e-04, 5.4359e-04,  ..., 4.6692e-03,
           7.7286e-03, 1.3199e-02]],

         [[8.3984e-01, 3.4285e-04, 3.6788e-04,  ..., 1.1511e-03,
           3.3360e-03, 2.1912e-02]],

         ...,

         [[1.4355e-01, 3.6716e-05, 3.5703e-05,  ..., 4.1604e-05,
           4.0650e-04, 7.7783e-01]],

         [[4.8242e-01, 2.0981e-04, 1.9789e-04,  ..., 4.9257e-04,
           8.2636e-04, 3.7598e-02]],

         [[7.2754e-01, 6.4552e-05, 2.9325e-05,  ..., 2.1942e-02,
           1.0010e-01, 8.9645e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1953e-01, 2.2781e-04, 2.0337e-04,  ..., 3.0792e-02,
           1.0431e-01, 2.5482e-02]],

         [[7.8223e-01, 1.9908e-05, 2.1040e-05,  ..., 3.9642e-02,
           7.9193e-03, 2.4765e-02]],

         [[4.1602e-01, 1.0413e-04, 2.2662e-04,  ..., 2.2769e-05,
           9.7156e-06, 2.8061e-02]],

         ...,

         [[9.5801e-01, 0.0000e+00, 0.0000e+00,  ..., 2.2926e-03,
           2.3743e-02, 1.3809e-02]],

         [[5.8936e-01, 1.4191e-03, 1.9321e-03,  ..., 8.5926e-04,
           1.9131e-03, 1.3786e-02]],

         [[5.0635e-01, 2.3842e-07, 0.0000e+00,  ..., 1.7944e-02,
           3.6377e-01, 5.6244e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8809e-01, 1.0598e-04, 3.6776e-05,  ..., 2.9510e-02,
           8.1329e-03, 1.7822e-02]],

         [[3.1763e-01, 1.5974e-03, 1.0738e-03,  ..., 5.0640e-04,
           5.1641e-04, 1.2878e-01]],

         [[7.7588e-01, 3.0327e-04, 2.0754e-04,  ..., 1.9140e-03,
           4.5252e-04, 2.1988e-02]],

         ...,

         [[4.7754e-01, 2.5034e-06, 2.0266e-06,  ..., 5.0507e-03,
           1.7662e-03, 2.6337e-02]],

         [[5.5469e-01, 7.6294e-06, 3.9339e-06,  ..., 2.0325e-02,
           5.1498e-03, 3.0716e-02]],

         [[2.7930e-01, 8.8811e-05, 1.5771e-04,  ..., 4.7278e-04,
           4.8018e-04, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8877e-02, 6.3777e-06, 5.9009e-06,  ..., 1.0204e-04,
           2.9254e-04, 8.7109e-01]],

         [[6.2842e-01, 1.8477e-05, 3.2187e-05,  ..., 1.1650e-02,
           6.2866e-03, 5.1880e-02]],

         [[2.3035e-01, 1.1425e-03, 2.0084e-03,  ..., 7.0238e-04,
           1.5736e-03, 3.1052e-02]],

         ...,

         [[3.4741e-01, 9.5308e-05, 1.0884e-04,  ..., 6.3667e-03,
           4.8180e-03, 5.5023e-02]],

         [[4.9194e-01, 4.2915e-06, 4.7088e-06,  ..., 8.1873e-04,
           1.7917e-04, 5.8197e-02]],

         [[9.8450e-02, 4.2558e-05, 1.8895e-05,  ..., 2.4979e-02,
           1.1151e-01, 6.0760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9792e-02, 5.4836e-06, 1.4901e-06,  ..., 2.5879e-01,
           3.2471e-01, 3.1158e-02]],

         [[3.2910e-01, 4.3917e-04, 5.2595e-04,  ..., 2.9488e-03,
           3.1757e-03, 8.9294e-02]],

         [[2.7148e-01, 1.9789e-05, 8.5235e-06,  ..., 1.1696e-02,
           6.0120e-02, 6.9763e-02]],

         ...,

         [[4.3481e-01, 1.3828e-05, 9.4175e-06,  ..., 4.4952e-02,
           3.0411e-02, 1.0248e-01]],

         [[6.1475e-01, 1.2994e-05, 2.1338e-05,  ..., 2.6917e-02,
           3.0853e-02, 2.2049e-02]],

         [[6.6895e-01, 1.7452e-04, 1.2839e-04,  ..., 3.7003e-03,
           2.9160e-02, 5.7770e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1261e-01, 1.0166e-03, 1.3332e-03,  ..., 2.7061e-05,
           1.0765e-04, 6.2561e-03]],

         [[3.0322e-01, 1.0195e-03, 8.7166e-04,  ..., 1.6546e-04,
           1.3266e-03, 2.8931e-02]],

         [[4.0015e-01, 8.9407e-07, 1.5497e-06,  ..., 8.3923e-03,
           1.0582e-02, 1.5221e-02]],

         ...,

         [[6.2939e-01, 1.8084e-04, 1.9193e-04,  ..., 3.6430e-04,
           1.1501e-03, 2.6550e-02]],

         [[9.7900e-01, 3.3438e-05, 6.6698e-05,  ..., 9.5367e-07,
           1.2517e-06, 3.7403e-03]],

         [[6.7236e-01, 2.1315e-04, 3.2210e-04,  ..., 1.6289e-03,
           2.1229e-03, 1.0941e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2424e-01, 4.6802e-04, 7.2479e-04,  ..., 8.4066e-04,
           2.4283e-04, 6.2256e-02]],

         [[3.5913e-01, 4.2458e-03, 2.2564e-03,  ..., 6.6423e-04,
           6.8808e-04, 6.7993e-02]],

         [[3.9856e-02, 2.8074e-05, 2.9325e-05,  ..., 1.0872e-03,
           6.2943e-04, 6.1377e-01]],

         ...,

         [[1.2152e-01, 9.9540e-06, 1.8299e-05,  ..., 1.4362e-03,
           1.9836e-03, 3.2959e-02]],

         [[4.5776e-02, 6.5565e-07, 1.7881e-07,  ..., 2.2192e-01,
           3.2544e-01, 3.5950e-02]],

         [[1.9116e-01, 1.0014e-05, 6.6757e-06,  ..., 1.1040e-02,
           2.7328e-02, 5.3613e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0439e-01, 1.1098e-04, 5.0008e-05,  ..., 6.1569e-03,
           3.2444e-03, 4.8431e-02]],

         [[4.4952e-02, 1.2875e-05, 2.3365e-05,  ..., 3.4122e-03,
           4.1809e-03, 2.9480e-02]],

         [[1.9238e-01, 3.2425e-05, 1.0073e-05,  ..., 1.0252e-03,
           1.2688e-02, 7.4902e-01]],

         ...,

         [[1.3367e-01, 1.4858e-03, 3.5977e-04,  ..., 1.8454e-04,
           4.9973e-04, 1.4091e-02]],

         [[3.4399e-01, 1.0786e-03, 7.0143e-04,  ..., 1.7233e-03,
           7.5293e-04, 1.4709e-02]],

         [[1.9922e-01, 7.3814e-04, 9.3699e-04,  ..., 7.9823e-04,
           2.2125e-03, 2.1317e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1934e-01, 3.1650e-05, 2.9385e-05,  ..., 1.4702e-02,
           2.0416e-02, 1.3953e-01]],

         [[9.7363e-01, 1.0848e-05, 3.5167e-06,  ..., 1.5795e-04,
           3.5226e-05, 2.9335e-03]],

         [[6.6992e-01, 1.4913e-04, 7.9811e-05,  ..., 7.0457e-03,
           1.8024e-03, 2.1713e-02]],

         ...,

         [[1.2366e-01, 1.3113e-03, 2.3727e-03,  ..., 1.0529e-03,
           1.8482e-03, 3.7140e-02]],

         [[1.4880e-01, 2.2173e-05, 2.0504e-05,  ..., 4.3976e-02,
           1.8225e-01, 3.4790e-02]],

         [[3.2642e-01, 1.9703e-03, 6.1188e-03,  ..., 1.7262e-03,
           6.3629e-03, 7.7515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7908e-01, 1.1140e-04, 2.3401e-04,  ..., 1.3466e-03,
           3.4122e-03, 1.7319e-02]],

         [[3.1421e-01, 3.9995e-05, 2.2662e-04,  ..., 7.7248e-04,
           3.5458e-03, 3.3112e-02]],

         [[3.1567e-01, 4.0174e-05, 1.1998e-04,  ..., 2.7523e-03,
           1.0887e-02, 8.0261e-03]],

         ...,

         [[1.3132e-03, 3.9339e-06, 1.0192e-05,  ..., 1.1921e-06,
           4.6492e-06, 1.0223e-03]],

         [[5.1758e-01, 1.5318e-04, 1.5080e-04,  ..., 2.8019e-03,
           2.0386e-02, 1.7065e-01]],

         [[7.2998e-01, 1.1843e-04, 1.6308e-04,  ..., 5.8403e-03,
           6.4163e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5625e-01, 6.2764e-05, 8.3447e-05,  ..., 4.7660e-04,
           9.6817e-03, 4.7668e-02]],

         [[5.8740e-01, 2.9945e-04, 5.9986e-04,  ..., 5.4836e-04,
           2.8744e-03, 1.5656e-02]],

         [[8.8928e-02, 1.1504e-05, 4.2379e-05,  ..., 2.0428e-03,
           1.0139e-02, 7.9883e-01]],

         ...,

         [[2.2510e-01, 3.7670e-05, 2.5749e-05,  ..., 4.7913e-02,
           1.5710e-01, 1.9556e-01]],

         [[6.3770e-01, 8.1837e-05, 3.0184e-04,  ..., 7.8869e-04,
           7.8869e-04, 2.2171e-02]],

         [[2.6807e-01, 7.5912e-04, 6.1464e-04,  ..., 6.0272e-04,
           4.1199e-03, 1.3351e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9121e-02, 7.1466e-05, 3.6597e-05,  ..., 5.9738e-03,
           2.2995e-02, 1.0065e-01]],

         [[2.6047e-02, 1.5020e-05, 1.3828e-05,  ..., 5.7220e-03,
           3.7231e-03, 5.2216e-02]],

         [[2.1338e-01, 1.0663e-04, 1.3578e-04,  ..., 3.0258e-02,
           2.7557e-02, 4.9896e-02]],

         ...,

         [[3.2056e-01, 0.0000e+00, 5.9605e-08,  ..., 2.1301e-02,
           4.3799e-01, 1.5625e-01]],

         [[3.7085e-01, 1.8537e-05, 8.5533e-05,  ..., 6.4659e-04,
           2.5787e-03, 8.4045e-02]],

         [[8.7451e-01, 4.1723e-07, 1.4901e-06,  ..., 3.5172e-03,
           3.0151e-02, 5.0903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0933e-02, 4.5896e-06, 1.8537e-05,  ..., 1.2338e-05,
           1.4677e-03, 9.4421e-02]],

         [[2.9572e-02, 2.0385e-05, 5.0366e-05,  ..., 4.4894e-04,
           3.9711e-03, 9.1992e-01]],

         [[1.7383e-01, 1.4019e-04, 9.6750e-04,  ..., 1.6093e-05,
           1.6201e-04, 1.8906e-02]],

         ...,

         [[1.5955e-01, 5.6219e-04, 1.7376e-03,  ..., 8.4686e-04,
           2.9621e-03, 5.2002e-02]],

         [[3.8403e-01, 1.5497e-06, 5.3644e-06,  ..., 7.8487e-04,
           7.1869e-03, 6.9153e-02]],

         [[2.4927e-01, 1.4901e-06, 1.6689e-06,  ..., 3.9581e-02,
           2.3145e-01, 1.6345e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8506e-01, 1.7536e-04, 8.5068e-04,  ..., 6.0692e-03,
           6.8245e-03, 2.8503e-02]],

         [[9.8389e-01, 1.2982e-04, 2.1899e-04,  ..., 3.7551e-05,
           4.4584e-05, 6.8283e-04]],

         [[7.9688e-01, 2.3234e-04, 7.8583e-04,  ..., 3.0594e-03,
           3.4008e-03, 4.3335e-03]],

         ...,

         [[3.0487e-02, 5.9605e-07, 1.7285e-06,  ..., 5.2147e-03,
           4.7211e-02, 2.1622e-02]],

         [[3.1152e-01, 5.6624e-06, 5.2452e-06,  ..., 9.7454e-05,
           2.7537e-04, 1.7138e-03]],

         [[1.7444e-01, 3.3081e-05, 1.0431e-04,  ..., 3.1638e-04,
           5.4779e-03, 7.2266e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1151e-01, 6.1989e-06, 2.0683e-05,  ..., 3.5839e-03,
           1.6388e-02, 1.3550e-01]],

         [[6.2103e-02, 1.5497e-06, 1.5497e-06,  ..., 3.8574e-02,
           2.4561e-01, 5.5237e-02]],

         [[1.7932e-01, 7.2479e-05, 1.0580e-04,  ..., 2.5345e-02,
           9.4543e-02, 2.4368e-02]],

         ...,

         [[1.4075e-01, 1.0908e-05, 6.7472e-05,  ..., 3.3760e-03,
           1.6296e-02, 8.5388e-02]],

         [[3.7280e-01, 9.0003e-06, 1.0014e-04,  ..., 2.0905e-03,
           7.7400e-03, 7.7515e-02]],

         [[5.3809e-01, 5.7220e-06, 2.9087e-05,  ..., 1.0033e-03,
           7.2670e-03, 1.3501e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.2953e-03, 2.3842e-07, 2.3842e-07,  ..., 3.0577e-05,
           1.5342e-04, 4.1962e-03]],

         [[1.8384e-01, 3.5167e-06, 2.1458e-06,  ..., 2.4643e-03,
           2.1622e-02, 9.8190e-03]],

         [[6.1676e-02, 1.5914e-05, 5.3048e-05,  ..., 5.6038e-03,
           2.1820e-02, 5.9326e-02]],

         ...,

         [[1.4087e-01, 8.2970e-05, 5.3692e-04,  ..., 1.7185e-03,
           1.5049e-03, 1.3867e-01]],

         [[1.6028e-01, 5.8270e-04, 2.0504e-03,  ..., 5.9652e-04,
           4.1008e-04, 1.2292e-01]],

         [[1.2085e-01, 3.2902e-04, 9.0122e-04,  ..., 2.2469e-03,
           1.4320e-02, 8.1116e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2939e-01, 4.3511e-06, 1.5795e-05,  ..., 1.7078e-01,
           2.8870e-02, 7.9880e-03]],

         [[1.0748e-01, 5.8413e-06, 3.3557e-05,  ..., 1.7614e-03,
           8.0200e-02, 3.1592e-01]],

         [[6.9141e-01, 3.5763e-07, 6.5565e-07,  ..., 9.5215e-03,
           9.8450e-02, 1.7139e-01]],

         ...,

         [[5.4736e-01, 1.8799e-04, 5.2738e-04,  ..., 1.9064e-03,
           4.2305e-03, 2.2415e-02]],

         [[4.3921e-01, 8.5235e-06, 4.5180e-05,  ..., 2.4338e-03,
           4.8027e-03, 3.3875e-02]],

         [[7.0923e-02, 3.2210e-04, 4.0555e-04,  ..., 2.7344e-02,
           1.5343e-02, 5.2832e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6711e-01, 4.8697e-05, 2.1243e-04,  ..., 1.5915e-02,
           1.4046e-02, 5.9143e-02]],

         [[1.5857e-01, 5.3177e-03, 5.3177e-03,  ..., 2.9030e-03,
           3.1147e-03, 3.8391e-02]],

         [[3.3911e-01, 9.5272e-04, 1.1339e-03,  ..., 1.3000e-02,
           1.8616e-02, 6.8909e-02]],

         ...,

         [[6.8799e-01, 3.8052e-04, 4.8494e-04,  ..., 2.4433e-03,
           4.8218e-03, 1.4389e-02]],

         [[2.3108e-01, 7.0190e-04, 1.2131e-03,  ..., 3.3493e-03,
           4.3427e-02, 5.6299e-01]],

         [[7.7295e-01, 3.7718e-04, 1.4687e-03,  ..., 2.3289e-03,
           3.8109e-03, 4.1595e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0354e-02, 6.1393e-06, 5.9128e-05,  ..., 6.0368e-04,
           7.8297e-04, 1.3904e-01]],

         [[5.7617e-01, 2.1732e-04, 2.6679e-04,  ..., 2.2430e-03,
           8.1558e-03, 2.2125e-02]],

         [[5.2393e-01, 7.0858e-04, 1.7052e-03,  ..., 1.4099e-02,
           5.3314e-02, 4.6417e-02]],

         ...,

         [[9.4482e-02, 1.0193e-02, 8.6212e-03,  ..., 5.7364e-04,
           3.5405e-04, 9.9182e-03]],

         [[8.0566e-01, 4.3213e-05, 4.1771e-04,  ..., 1.0902e-02,
           2.4815e-03, 5.8289e-03]],

         [[3.4058e-01, 2.4140e-04, 3.7026e-04,  ..., 7.9346e-04,
           1.3056e-03, 2.6459e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9397e-04, 2.3060e-03, 1.8689e-01,  ..., 2.5535e-04,
           8.9121e-04, 8.8623e-02]],

         [[4.3678e-04, 9.9182e-04, 1.7249e-01,  ..., 4.7798e-03,
           1.9159e-03, 3.8177e-02]],

         [[5.0783e-04, 6.8855e-04, 1.2402e-01,  ..., 2.6001e-02,
           4.1466e-03, 5.3986e-02]],

         ...,

         [[3.3832e-04, 2.4366e-04, 1.2878e-01,  ..., 1.9245e-03,
           1.4305e-03, 2.4817e-01]],

         [[3.3398e-03, 1.5442e-02, 4.1321e-02,  ..., 1.0568e-04,
           8.9347e-05, 8.8167e-04]],

         [[1.0319e-03, 6.9857e-04, 4.1870e-02,  ..., 1.0080e-03,
           5.9509e-04, 6.7825e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8501e-03, 2.8157e-04, 1.9543e-01,  ..., 7.3576e-04,
           1.1902e-03, 1.8188e-02]],

         [[2.3766e-03, 2.5196e-03, 1.9336e-01,  ..., 5.0211e-04,
           4.0131e-03, 2.8732e-02]],

         [[2.5787e-03, 4.0865e-04, 2.9102e-01,  ..., 1.3959e-04,
           3.0756e-05, 3.9703e-02]],

         ...,

         [[1.6449e-02, 1.9875e-03, 1.6266e-02,  ..., 1.3294e-03,
           5.5313e-05, 2.5425e-03]],

         [[2.9588e-04, 6.6948e-04, 2.7173e-01,  ..., 4.7088e-04,
           3.9577e-04, 1.1932e-02]],

         [[2.1954e-03, 5.0278e-03, 1.3451e-02,  ..., 3.2187e-06,
           9.1195e-06, 6.6071e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5191e-03, 1.0977e-03, 1.3599e-01,  ..., 1.8358e-03,
           2.1011e-02, 4.7150e-02]],

         [[3.2768e-03, 4.3511e-05, 1.2732e-01,  ..., 8.7051e-03,
           3.1769e-05, 2.6077e-02]],

         [[8.6670e-03, 4.3068e-03, 2.0911e-01,  ..., 4.1084e-03,
           6.0260e-05, 5.3711e-02]],

         ...,

         [[1.5907e-03, 8.6164e-04, 1.8396e-01,  ..., 8.8730e-03,
           1.7071e-03, 4.1504e-02]],

         [[8.2111e-04, 8.5258e-04, 1.4880e-01,  ..., 4.5746e-02,
           1.0529e-03, 2.1271e-02]],

         [[7.8964e-03, 1.0672e-03, 5.7251e-02,  ..., 4.6997e-03,
           3.6573e-04, 6.1035e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.6594e-04, 9.5069e-05, 2.5909e-02,  ..., 4.5204e-03,
           7.3242e-02, 3.6035e-01]],

         [[1.1196e-03, 5.0974e-04, 2.8784e-01,  ..., 1.1078e-02,
           3.9101e-03, 6.7711e-03]],

         [[5.1918e-03, 1.6251e-02, 4.3221e-03,  ..., 2.2948e-05,
           9.0241e-05, 7.6294e-02]],

         ...,

         [[2.1935e-03, 1.2970e-03, 2.7298e-02,  ..., 3.7527e-04,
           4.5090e-03, 4.8022e-01]],

         [[1.7710e-03, 2.6131e-04, 1.3344e-02,  ..., 1.2178e-03,
           1.2733e-02, 8.7207e-01]],

         [[6.8626e-03, 1.2598e-03, 2.2156e-02,  ..., 2.4471e-03,
           1.5900e-02, 6.3770e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5907e-03, 1.4849e-03, 1.7029e-01,  ..., 1.8097e-02,
           4.2236e-02, 5.3406e-02]],

         [[2.3117e-03, 6.4754e-04, 2.6733e-01,  ..., 8.4066e-04,
           3.0899e-03, 4.9286e-02]],

         [[4.5204e-03, 1.2493e-03, 2.6099e-01,  ..., 3.4294e-03,
           7.0381e-03, 5.9601e-02]],

         ...,

         [[6.8188e-04, 2.6083e-04, 1.3940e-01,  ..., 1.1778e-03,
           1.3069e-02, 3.2153e-01]],

         [[3.9601e-04, 2.8477e-03, 2.4268e-01,  ..., 4.7302e-03,
           1.1501e-03, 1.3501e-01]],

         [[2.8229e-04, 3.7611e-05, 4.3549e-02,  ..., 4.4136e-03,
           8.9874e-03, 6.1084e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[2.2812e-03, 1.2088e-04, 9.8133e-04,  ..., 1.0010e-01,
           6.3232e-01, 1.1340e-01]],

         [[7.6246e-04, 1.1688e-04, 6.7282e-04,  ..., 1.3244e-04,
           2.8610e-02, 9.4727e-01]],

         [[8.3447e-07, 1.3709e-06, 2.2650e-06,  ..., 2.2650e-06,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[3.1686e-04, 6.7091e-04, 4.0698e-04,  ..., 3.8605e-03,
           4.7028e-02, 7.3584e-01]],

         [[5.2719e-03, 4.6539e-03, 9.2545e-03,  ..., 2.6779e-02,
           7.7454e-02, 7.7454e-02]],

         [[6.7291e-03, 3.2806e-03, 2.0969e-04,  ..., 1.4702e-02,
           5.5809e-03, 2.2064e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.5310e-04, 1.9388e-03, 5.8784e-03,  ..., 2.9861e-02,
           2.4231e-01, 1.2183e-01]],

         [[2.7955e-05, 4.9829e-05, 1.0610e-05,  ..., 1.0376e-01,
           8.2861e-01, 6.0974e-02]],

         [[1.3132e-03, 3.1013e-03, 1.0651e-02,  ..., 9.6970e-03,
           5.0842e-02, 2.6367e-02]],

         ...,

         [[1.3857e-03, 1.7653e-03, 6.7673e-03,  ..., 6.8359e-02,
           1.0260e-01, 2.8717e-02]],

         [[3.3550e-03, 9.0790e-03, 3.3081e-02,  ..., 2.0706e-02,
           6.2317e-02, 2.9541e-02]],

         [[2.3210e-04, 1.7290e-03, 4.8757e-04,  ..., 5.3772e-02,
           8.3447e-01, 3.3661e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2163e-03, 8.3780e-04, 2.6207e-03,  ..., 1.0468e-01,
           1.5234e-01, 8.6121e-02]],

         [[1.3533e-03, 3.4213e-04, 1.8787e-03,  ..., 3.3245e-03,
           3.7956e-03, 3.2082e-03]],

         [[2.8172e-03, 4.1389e-04, 4.6515e-04,  ..., 1.2627e-02,
           9.3994e-02, 1.0468e-02]],

         ...,

         [[3.9253e-03, 7.9930e-05, 1.2875e-04,  ..., 7.5867e-02,
           6.0107e-01, 1.9373e-01]],

         [[5.8889e-04, 1.1420e-04, 2.7609e-04,  ..., 5.3329e-03,
           7.7591e-03, 6.4316e-03]],

         [[3.1342e-02, 4.9973e-03, 1.1002e-02,  ..., 2.4033e-02,
           2.9980e-01, 4.8553e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1836e-01, 3.2783e-06, 1.1921e-06,  ..., 4.2677e-04,
           5.3864e-03, 1.8682e-03]],

         [[9.5703e-01, 4.2319e-06, 5.3644e-07,  ..., 1.5926e-03,
           2.8259e-02, 4.2381e-03]],

         [[7.4023e-01, 5.3644e-07, 2.3842e-07,  ..., 1.1909e-02,
           1.1253e-02, 2.4765e-02]],

         ...,

         [[1.7749e-01, 1.7941e-05, 1.7500e-04,  ..., 1.5998e-04,
           1.0826e-02, 1.1253e-02]],

         [[1.1304e-01, 2.8014e-06, 3.9101e-05,  ..., 4.5929e-03,
           2.3605e-02, 6.9237e-03]],

         [[7.3291e-01, 3.3903e-04, 2.4939e-04,  ..., 2.4071e-03,
           1.4519e-02, 5.8317e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0469e-01, 2.5034e-06, 1.5497e-06,  ..., 4.7469e-04,
           8.2855e-03, 5.2460e-02]],

         [[2.9590e-01, 5.3644e-07, 0.0000e+00,  ..., 1.4969e-02,
           6.7236e-01, 1.6357e-02]],

         [[9.6094e-01, 1.2034e-04, 5.5432e-06,  ..., 4.3182e-03,
           1.2863e-02, 4.5319e-03]],

         ...,

         [[5.7764e-01, 5.3883e-05, 2.5630e-05,  ..., 3.4580e-03,
           5.2368e-02, 1.5366e-02]],

         [[1.2732e-01, 4.9448e-04, 4.5753e-04,  ..., 9.5272e-04,
           1.4511e-02, 4.2343e-03]],

         [[2.3584e-01, 1.4305e-05, 7.6890e-06,  ..., 5.1422e-02,
           5.5469e-01, 3.7170e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2690e-01, 2.1076e-04, 8.1897e-05,  ..., 3.6316e-02,
           6.1981e-02, 2.3117e-02]],

         [[6.4600e-01, 4.5061e-05, 9.3579e-06,  ..., 6.8092e-03,
           1.2610e-01, 2.7267e-02]],

         [[5.2344e-01, 6.3181e-06, 7.5102e-06,  ..., 1.2062e-02,
           4.9225e-02, 1.2917e-02]],

         ...,

         [[8.4863e-01, 1.2159e-05, 7.1526e-06,  ..., 1.2636e-05,
           1.7748e-03, 1.4331e-01]],

         [[9.4727e-01, 2.0444e-05, 2.2006e-04,  ..., 1.7881e-07,
           3.2365e-05, 2.4902e-02]],

         [[9.8877e-01, 2.0266e-06, 2.3961e-05,  ..., 0.0000e+00,
           5.3048e-06, 8.7738e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8379e-01, 1.4246e-05, 2.6822e-06,  ..., 7.4829e-02,
           3.3154e-01, 2.0752e-02]],

         [[6.2549e-01, 2.7418e-06, 7.7486e-07,  ..., 2.5129e-04,
           4.7073e-03, 5.6458e-03]],

         [[9.0430e-01, 2.5940e-04, 3.1829e-05,  ..., 1.6880e-04,
           1.1078e-02, 4.2534e-03]],

         ...,

         [[6.3135e-01, 6.9082e-05, 9.7156e-06,  ..., 7.0858e-04,
           6.4011e-03, 2.6642e-02]],

         [[2.3303e-01, 7.3314e-05, 6.9141e-06,  ..., 3.9635e-03,
           4.7112e-03, 4.6173e-02]],

         [[9.0479e-01, 9.0122e-05, 6.9320e-05,  ..., 2.7542e-03,
           1.6159e-02, 5.1117e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6685e-01, 8.3447e-07, 9.5367e-07,  ..., 7.3181e-02,
           9.3201e-02, 1.3901e-02]],

         [[4.1064e-01, 4.1366e-05, 2.3305e-05,  ..., 1.3687e-02,
           9.1370e-02, 1.0242e-01]],

         [[7.7783e-01, 1.5378e-05, 2.5034e-06,  ..., 4.3869e-03,
           2.5101e-02, 5.9280e-03]],

         ...,

         [[6.9629e-01, 4.3535e-04, 9.6381e-05,  ..., 1.1658e-02,
           9.3842e-03, 2.5482e-02]],

         [[8.4131e-01, 1.9455e-04, 5.1141e-05,  ..., 3.8166e-03,
           1.1665e-02, 4.9210e-03]],

         [[7.3193e-01, 3.8147e-06, 6.3777e-06,  ..., 8.7967e-03,
           5.7495e-02, 3.2990e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5811e-01, 1.1663e-03, 9.9754e-04,  ..., 1.7786e-03,
           5.6992e-03, 9.8991e-04]],

         [[9.1650e-01, 6.6221e-05, 2.0158e-04,  ..., 1.7967e-03,
           5.8022e-03, 5.0812e-03]],

         [[7.0801e-01, 4.1664e-05, 1.7941e-05,  ..., 2.8931e-02,
           4.2191e-03, 4.3869e-03]],

         ...,

         [[6.3721e-01, 5.2392e-05, 1.1265e-05,  ..., 1.8275e-04,
           9.3994e-03, 2.4414e-01]],

         [[8.7207e-01, 1.4722e-05, 4.5180e-05,  ..., 1.1683e-03,
           3.3207e-03, 3.6407e-02]],

         [[9.0820e-01, 1.3769e-05, 1.9491e-05,  ..., 4.2076e-03,
           1.8740e-03, 7.2556e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0764e-01, 4.5228e-04, 1.3475e-03,  ..., 2.0638e-03,
           1.2219e-01, 6.3324e-03]],

         [[8.7891e-01, 3.2783e-06, 6.7949e-06,  ..., 2.6150e-03,
           5.7129e-02, 2.7039e-02]],

         [[3.5498e-01, 1.7285e-06, 1.9073e-05,  ..., 7.5951e-03,
           7.3586e-03, 1.9623e-02]],

         ...,

         [[5.2979e-01, 8.4281e-05, 1.4448e-04,  ..., 1.1683e-03,
           6.1096e-02, 1.5545e-03]],

         [[8.6279e-01, 6.8605e-05, 3.9458e-05,  ..., 7.8087e-03,
           6.0806e-03, 7.3013e-03]],

         [[8.7354e-01, 5.2452e-06, 1.0192e-05,  ..., 1.9112e-03,
           4.8103e-03, 1.0963e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2891e-01, 1.9431e-05, 3.3975e-05,  ..., 1.7786e-04,
           1.8204e-02, 6.8398e-03]],

         [[7.5537e-01, 1.1444e-05, 1.0848e-05,  ..., 1.8097e-02,
           4.8248e-02, 1.9272e-02]],

         [[1.1560e-01, 0.0000e+00, 1.7881e-07,  ..., 2.0046e-03,
           6.7993e-02, 5.6496e-03]],

         ...,

         [[9.2480e-01, 4.9174e-05, 2.2054e-05,  ..., 3.1452e-03,
           4.4708e-03, 4.3068e-03]],

         [[4.7559e-01, 8.1837e-05, 2.1303e-04,  ..., 2.2736e-03,
           1.5594e-02, 2.7710e-01]],

         [[5.4932e-01, 2.9731e-04, 7.4625e-05,  ..., 2.6512e-03,
           6.6948e-03, 4.8981e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2329e-01, 8.9407e-07, 1.7285e-06,  ..., 2.4629e-04,
           5.5237e-03, 8.5938e-01]],

         [[7.9004e-01, 8.5545e-04, 7.7820e-04,  ..., 1.7868e-02,
           2.0561e-03, 5.5742e-04]],

         [[6.1328e-01, 4.6372e-04, 6.3705e-04,  ..., 2.4353e-02,
           5.6824e-02, 1.4824e-02]],

         ...,

         [[5.0098e-01, 7.5817e-05, 1.5867e-04,  ..., 1.3222e-02,
           2.7527e-02, 1.3077e-02]],

         [[1.6199e-01, 4.0293e-05, 3.6657e-05,  ..., 6.8092e-03,
           1.0139e-02, 4.4128e-02]],

         [[1.8250e-01, 3.3677e-05, 1.1617e-04,  ..., 3.7789e-04,
           7.5378e-02, 6.5332e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9983e-01, 2.4259e-05, 4.6253e-05,  ..., 2.7733e-03,
           4.0741e-02, 6.1328e-01]],

         [[6.5869e-01, 3.1495e-04, 3.0541e-04,  ..., 2.9053e-02,
           6.1340e-02, 5.0659e-03]],

         [[6.1572e-01, 1.3328e-04, 6.9141e-05,  ..., 7.9117e-03,
           4.4136e-03, 1.2039e-02]],

         ...,

         [[9.5444e-03, 5.9605e-07, 4.2915e-06,  ..., 1.9789e-05,
           7.6294e-04, 9.8291e-01]],

         [[6.2158e-01, 6.4850e-05, 5.1320e-05,  ..., 3.3875e-03,
           6.6147e-03, 1.2866e-01]],

         [[7.1094e-01, 5.1200e-05, 1.5497e-05,  ..., 6.7444e-02,
           2.4319e-03, 2.7027e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1069e-01, 3.5000e-04, 4.9543e-04,  ..., 8.6914e-02,
           3.6865e-02, 8.6288e-03]],

         [[8.5254e-01, 6.2585e-05, 1.0520e-04,  ..., 4.4556e-03,
           5.7007e-02, 1.2695e-02]],

         [[2.8931e-01, 7.0333e-06, 2.7478e-05,  ..., 1.3895e-03,
           2.5345e-02, 1.1719e-02]],

         ...,

         [[2.1570e-01, 5.9605e-08, 0.0000e+00,  ..., 9.9487e-03,
           7.5293e-01, 2.0203e-02]],

         [[4.7241e-01, 5.7745e-04, 6.9618e-04,  ..., 1.0422e-02,
           2.1652e-02, 8.3847e-03]],

         [[8.9502e-01, 1.7285e-06, 2.9802e-07,  ..., 1.9791e-02,
           3.1219e-02, 3.4668e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1445e-01, 4.6909e-05, 1.6451e-05,  ..., 5.1575e-03,
           1.9512e-03, 7.4501e-03]],

         [[1.2988e-01, 3.9005e-04, 3.7003e-04,  ..., 3.0780e-04,
           2.6123e-01, 4.5471e-03]],

         [[4.2432e-01, 2.6393e-04, 2.4164e-04,  ..., 6.9427e-03,
           1.2744e-01, 1.4198e-02]],

         ...,

         [[4.1724e-01, 7.9274e-06, 2.9206e-06,  ..., 9.4452e-03,
           4.1260e-01, 1.0382e-01]],

         [[7.6367e-01, 2.1458e-06, 1.0133e-06,  ..., 6.4611e-04,
           1.4636e-01, 6.5735e-02]],

         [[4.6704e-01, 2.1398e-05, 1.2279e-05,  ..., 1.2112e-03,
           3.4229e-01, 5.9052e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5054e-02, 2.5034e-06, 5.1856e-06,  ..., 5.9605e-05,
           1.5465e-02, 8.9551e-01]],

         [[4.2554e-01, 4.5300e-06, 5.8413e-06,  ..., 9.7275e-03,
           3.4204e-01, 1.3440e-01]],

         [[2.5952e-01, 1.6391e-04, 2.0516e-04,  ..., 2.2602e-03,
           4.8096e-02, 9.1248e-03]],

         ...,

         [[1.9678e-01, 1.6093e-05, 6.7949e-06,  ..., 3.8300e-02,
           1.0889e-01, 1.3206e-02]],

         [[2.6904e-01, 1.4305e-06, 5.9605e-07,  ..., 2.0993e-04,
           2.2925e-01, 1.6495e-02]],

         [[3.1470e-01, 1.3423e-04, 4.8816e-05,  ..., 9.3155e-03,
           9.8419e-03, 2.8946e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1519e-01, 3.4332e-05, 2.3425e-05,  ..., 4.7760e-02,
           9.4604e-02, 8.4000e-03]],

         [[9.7778e-02, 6.3539e-05, 1.6809e-05,  ..., 9.0942e-03,
           1.6504e-01, 1.2924e-02]],

         [[9.9426e-02, 6.0201e-05, 5.1260e-06,  ..., 5.2071e-03,
           2.6962e-02, 2.7176e-02]],

         ...,

         [[4.8560e-01, 1.0610e-05, 7.6294e-06,  ..., 1.7822e-02,
           1.3074e-01, 1.9196e-02]],

         [[5.4639e-01, 1.5199e-05, 1.7285e-05,  ..., 5.6488e-02,
           6.2744e-02, 3.1067e-02]],

         [[2.7222e-01, 2.7180e-05, 1.5557e-05,  ..., 1.2659e-01,
           2.2388e-01, 5.9433e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4355e-01, 5.6648e-04, 7.8249e-04,  ..., 6.2108e-05,
           1.1971e-02, 1.4162e-03]],

         [[1.9238e-01, 7.9870e-05, 1.4853e-04,  ..., 1.1635e-03,
           3.1647e-02, 8.2703e-03]],

         [[4.6680e-01, 4.2319e-06, 4.4703e-06,  ..., 4.2992e-03,
           1.4624e-01, 3.4607e-02]],

         ...,

         [[6.4062e-01, 4.9472e-05, 2.3067e-05,  ..., 1.5078e-03,
           1.9272e-02, 2.8763e-03]],

         [[9.8535e-01, 1.8299e-05, 2.4736e-05,  ..., 1.9073e-06,
           2.0313e-04, 2.8057e-03]],

         [[6.9629e-01, 1.0759e-04, 1.2779e-04,  ..., 9.3613e-03,
           2.8183e-02, 3.4237e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8408e-01, 9.5606e-05, 1.1444e-04,  ..., 6.8474e-04,
           1.2067e-01, 1.1406e-02]],

         [[1.5649e-01, 8.0872e-04, 5.0020e-04,  ..., 1.3786e-02,
           1.0510e-01, 1.8402e-02]],

         [[9.5703e-02, 4.3511e-06, 3.9339e-06,  ..., 1.9703e-03,
           7.3364e-02, 5.1318e-01]],

         ...,

         [[2.4561e-01, 3.9935e-06, 3.5167e-06,  ..., 8.3252e-02,
           2.4756e-01, 1.6586e-02]],

         [[4.1943e-01, 7.3910e-06, 6.1393e-06,  ..., 1.7517e-02,
           5.3314e-02, 2.4597e-02]],

         [[1.6187e-01, 2.0266e-06, 2.8014e-06,  ..., 5.7602e-03,
           1.9080e-01, 2.4683e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4873e-01, 9.5963e-06, 6.0201e-06,  ..., 1.2146e-02,
           6.4636e-02, 7.7209e-03]],

         [[2.1765e-01, 1.9073e-05, 2.9504e-05,  ..., 1.7414e-03,
           4.9347e-02, 3.6102e-02]],

         [[1.0931e-01, 4.5896e-06, 2.0862e-06,  ..., 4.5657e-05,
           1.8997e-02, 8.2031e-01]],

         ...,

         [[8.6670e-02, 7.9632e-05, 2.8849e-05,  ..., 4.9055e-05,
           2.8515e-03, 6.4278e-03]],

         [[3.5498e-01, 3.3665e-04, 1.7321e-04,  ..., 4.7379e-03,
           1.2581e-02, 3.6755e-03]],

         [[2.5513e-01, 2.0456e-04, 1.3053e-04,  ..., 1.4397e-02,
           7.6050e-02, 1.3847e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9365e-02, 2.4319e-05, 1.1086e-05,  ..., 6.5674e-02,
           1.0333e-01, 2.2690e-02]],

         [[8.9893e-01, 1.1718e-04, 2.2292e-05,  ..., 4.5300e-06,
           7.7009e-04, 3.9406e-03]],

         [[9.1016e-01, 7.4804e-05, 1.7822e-05,  ..., 2.9862e-05,
           7.0429e-04, 2.4204e-03]],

         ...,

         [[7.7332e-02, 5.8556e-04, 1.2016e-03,  ..., 5.5580e-03,
           5.1514e-02, 3.8490e-03]],

         [[4.2896e-01, 1.0610e-05, 2.0564e-05,  ..., 4.8126e-02,
           3.3081e-02, 2.2736e-02]],

         [[3.9526e-01, 5.1212e-04, 1.5650e-03,  ..., 1.9669e-02,
           5.7831e-02, 7.9498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9004e-01, 3.2246e-05, 7.1228e-05,  ..., 5.7030e-03,
           1.2848e-02, 1.7151e-02]],

         [[1.4648e-01, 3.1710e-05, 1.0008e-04,  ..., 2.0580e-03,
           2.1774e-02, 3.8208e-02]],

         [[1.4307e-01, 1.2755e-05, 3.4034e-05,  ..., 6.2370e-03,
           6.9008e-03, 4.9324e-03]],

         ...,

         [[1.5747e-02, 1.3053e-05, 3.0875e-05,  ..., 1.7583e-05,
           3.6259e-03, 5.5313e-03]],

         [[1.9165e-01, 4.7028e-05, 4.6313e-05,  ..., 2.5034e-04,
           1.2451e-02, 9.2468e-03]],

         [[5.5859e-01, 7.2300e-05, 1.2010e-04,  ..., 9.4681e-03,
           7.8506e-03, 9.7656e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.2632e-02, 5.4240e-06, 7.5698e-06,  ..., 1.9817e-03,
           1.0971e-02, 2.0493e-02]],

         [[2.0801e-01, 8.6486e-05, 1.4102e-04,  ..., 1.7042e-03,
           3.9635e-03, 7.5760e-03]],

         [[1.6565e-01, 2.9385e-05, 1.5521e-04,  ..., 1.4219e-03,
           9.6619e-02, 4.9463e-01]],

         ...,

         [[2.7295e-01, 1.8477e-05, 1.2696e-05,  ..., 2.1713e-02,
           1.0120e-01, 5.4893e-03]],

         [[1.6248e-01, 1.7107e-05, 7.6592e-05,  ..., 2.8348e-04,
           7.2517e-03, 4.4434e-02]],

         [[2.9004e-01, 2.4796e-05, 3.2842e-05,  ..., 4.4990e-04,
           6.9275e-03, 1.8072e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.5063e-02, 3.5703e-05, 2.2948e-05,  ..., 2.8362e-03,
           2.3560e-02, 3.6774e-02]],

         [[1.2863e-02, 4.1723e-06, 7.7486e-06,  ..., 1.9681e-04,
           9.7046e-03, 2.1530e-02]],

         [[2.3523e-01, 1.4722e-05, 2.1040e-05,  ..., 1.4830e-03,
           8.6365e-03, 3.2520e-03]],

         ...,

         [[6.3232e-01, 1.1921e-07, 0.0000e+00,  ..., 1.0321e-01,
           1.8408e-01, 4.2023e-02]],

         [[5.3516e-01, 7.9870e-06, 1.9610e-05,  ..., 2.9953e-02,
           1.6321e-01, 2.6230e-02]],

         [[4.3066e-01, 1.1325e-06, 1.0729e-06,  ..., 1.3313e-02,
           4.9561e-01, 4.9438e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.2232e-02, 2.3842e-06, 1.7464e-05,  ..., 1.7548e-03,
           5.1270e-02, 2.4033e-02]],

         [[9.9106e-03, 4.7684e-06, 2.8849e-05,  ..., 8.1182e-05,
           3.2940e-03, 9.7998e-01]],

         [[1.0052e-01, 2.9087e-05, 2.0778e-04,  ..., 4.2558e-04,
           2.8137e-02, 6.0463e-04]],

         ...,

         [[1.9568e-01, 8.0884e-05, 3.4952e-04,  ..., 3.6888e-03,
           1.0760e-01, 8.1711e-03]],

         [[5.4541e-01, 5.9605e-07, 1.3113e-06,  ..., 8.3771e-03,
           9.9670e-02, 4.1565e-02]],

         [[2.0288e-01, 4.2915e-06, 2.7418e-06,  ..., 2.9572e-02,
           5.6250e-01, 1.3513e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2793e-01, 1.0943e-04, 6.6423e-04,  ..., 1.5472e-02,
           4.1565e-02, 6.5269e-03]],

         [[9.7266e-01, 4.9293e-05, 3.4595e-04,  ..., 1.5354e-04,
           6.4373e-04, 3.1519e-04]],

         [[8.1494e-01, 4.2260e-05, 1.0258e-04,  ..., 8.1787e-03,
           1.4580e-02, 1.3990e-03]],

         ...,

         [[3.2690e-01, 1.7285e-06, 3.0994e-06,  ..., 2.0889e-02,
           2.7252e-02, 1.1009e-02]],

         [[4.0112e-01, 4.7684e-07, 2.3842e-07,  ..., 3.8314e-04,
           1.8864e-03, 2.0714e-03]],

         [[1.2561e-01, 1.0192e-05, 3.2842e-05,  ..., 2.0266e-04,
           2.1835e-02, 7.8174e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3025e-01, 1.8477e-06, 6.9737e-06,  ..., 1.8326e-02,
           3.3789e-01, 6.3232e-02]],

         [[2.8174e-01, 5.0664e-06, 3.8147e-06,  ..., 1.4966e-01,
           4.3701e-02, 4.2542e-02]],

         [[2.5195e-01, 1.4687e-04, 2.1279e-04,  ..., 2.2369e-02,
           1.2650e-02, 8.3923e-03]],

         ...,

         [[2.4707e-01, 1.1563e-05, 5.1856e-05,  ..., 3.8177e-02,
           9.0881e-02, 1.6678e-02]],

         [[5.3516e-01, 1.4782e-05, 7.3731e-05,  ..., 1.1734e-02,
           7.4158e-02, 9.0714e-03]],

         [[4.9780e-01, 2.0266e-06, 1.0014e-05,  ..., 7.3280e-03,
           1.6040e-01, 1.2077e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6591e-02, 4.3511e-06, 3.9339e-06,  ..., 7.7903e-05,
           1.4191e-03, 1.1162e-02]],

         [[3.1396e-01, 3.9697e-05, 1.7881e-05,  ..., 3.0609e-02,
           9.9030e-03, 2.2745e-04]],

         [[2.7637e-01, 3.5167e-05, 7.2777e-05,  ..., 1.8372e-02,
           3.6804e-02, 4.2114e-03]],

         ...,

         [[3.0518e-02, 1.1802e-05, 1.1265e-04,  ..., 9.9659e-04,
           1.3782e-01, 6.7024e-03]],

         [[8.9355e-02, 5.4240e-05, 2.9802e-04,  ..., 1.8942e-04,
           2.3169e-01, 1.0674e-02]],

         [[6.7090e-01, 4.1127e-05, 4.9055e-05,  ..., 7.8125e-03,
           8.0139e-02, 4.5288e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4248e-01, 1.7881e-05, 3.3140e-05,  ..., 2.0706e-02,
           1.7609e-02, 1.1237e-01]],

         [[2.4939e-01, 6.3777e-06, 3.7432e-05,  ..., 1.0236e-01,
           2.8809e-01, 1.3135e-01]],

         [[2.3352e-01, 1.8477e-06, 1.8477e-06,  ..., 1.4984e-02,
           6.4746e-01, 8.9355e-02]],

         ...,

         [[5.2783e-01, 8.0585e-05, 1.3816e-04,  ..., 4.1428e-03,
           1.9989e-02, 1.1299e-02]],

         [[1.5808e-01, 3.5167e-06, 7.6890e-06,  ..., 6.1798e-03,
           1.0291e-01, 8.8013e-02]],

         [[4.0436e-02, 9.1612e-05, 3.1471e-04,  ..., 2.8000e-02,
           5.3558e-02, 4.2139e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1567e-01, 4.2915e-05, 1.8322e-04,  ..., 3.9337e-02,
           3.5126e-02, 8.3771e-03]],

         [[6.6406e-01, 4.7064e-04, 1.0376e-03,  ..., 1.7118e-03,
           4.3869e-03, 1.6985e-03]],

         [[7.2266e-02, 8.4817e-05, 1.4651e-04,  ..., 1.9569e-03,
           4.7874e-03, 1.0767e-03]],

         ...,

         [[4.2603e-01, 4.9114e-04, 3.3212e-04,  ..., 1.4572e-02,
           1.7853e-02, 2.1881e-02]],

         [[7.5684e-02, 6.1870e-05, 2.0933e-04,  ..., 1.6113e-02,
           3.4119e-02, 7.8906e-01]],

         [[8.9795e-01, 3.5167e-05, 8.2374e-05,  ..., 5.7755e-03,
           1.4519e-02, 5.3406e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.9746e-01, 8.9347e-05, 3.8981e-04,  ..., 1.8711e-03,
           2.2858e-02, 1.1909e-02]],

         [[1.3623e-01, 2.2531e-05, 6.2823e-05,  ..., 7.2899e-03,
           3.4851e-02, 3.4851e-02]],

         [[2.5635e-01, 2.7728e-04, 8.0156e-04,  ..., 1.5356e-01,
           5.8777e-02, 2.5818e-02]],

         ...,

         [[4.2871e-01, 1.4198e-04, 1.6222e-03,  ..., 2.8324e-03,
           2.4292e-02, 2.0447e-02]],

         [[8.6963e-01, 2.3484e-05, 6.9904e-04,  ..., 1.0386e-03,
           3.3321e-03, 1.8082e-03]],

         [[3.7573e-01, 3.8147e-05, 1.1253e-04,  ..., 2.1801e-03,
           1.1452e-02, 4.7760e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.0902e-05, 1.1158e-04, 1.7419e-01,  ..., 1.6813e-03,
           5.2734e-02, 1.4137e-02]],

         [[2.5213e-05, 9.3877e-05, 1.7822e-01,  ..., 5.0688e-04,
           1.7181e-02, 1.2230e-02]],

         [[1.4997e-04, 1.7262e-04, 1.0297e-01,  ..., 1.3039e-02,
           2.8580e-02, 1.1192e-02]],

         ...,

         [[2.0671e-04, 1.5295e-04, 1.6846e-01,  ..., 4.9286e-03,
           7.2449e-02, 2.2314e-01]],

         [[4.4751e-04, 9.3603e-04, 1.8921e-01,  ..., 2.4471e-03,
           1.4137e-02, 2.2598e-02]],

         [[2.3961e-05, 6.9559e-05, 1.6589e-01,  ..., 4.0710e-05,
           4.3182e-03, 1.8148e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1195e-05, 1.8930e-04, 2.8369e-01,  ..., 1.3530e-04,
           1.3802e-02, 2.5330e-03]],

         [[1.6189e-04, 1.9526e-04, 1.7554e-01,  ..., 1.1511e-03,
           1.0704e-02, 7.4863e-04]],

         [[7.0274e-05, 3.2544e-05, 3.1641e-01,  ..., 2.7776e-05,
           1.8265e-02, 5.8031e-04]],

         ...,

         [[4.0627e-04, 2.4438e-04, 2.8052e-01,  ..., 2.8133e-04,
           2.5284e-02, 1.2589e-03]],

         [[2.8670e-05, 1.3173e-04, 2.9102e-01,  ..., 1.2153e-04,
           8.8959e-03, 3.9244e-04]],

         [[1.4961e-05, 7.1764e-05, 6.4758e-02,  ..., 1.0127e-04,
           1.2976e-01, 2.3956e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0341e-04, 1.1975e-04, 1.8665e-01,  ..., 1.0738e-03,
           9.4376e-03, 4.4799e-04]],

         [[4.8923e-04, 3.4392e-05, 2.8149e-01,  ..., 1.7881e-05,
           1.0963e-02, 4.9496e-04]],

         [[1.5295e-04, 2.4796e-04, 3.1470e-01,  ..., 5.4836e-06,
           1.1826e-02, 1.0376e-03]],

         ...,

         [[1.6189e-04, 1.8084e-04, 1.4673e-01,  ..., 4.1466e-03,
           1.5656e-02, 8.6441e-03]],

         [[7.4744e-05, 2.1899e-04, 2.6489e-01,  ..., 2.5678e-04,
           1.1421e-02, 1.3189e-03]],

         [[4.5490e-04, 3.3855e-04, 1.7004e-01,  ..., 1.2720e-04,
           6.3171e-03, 1.2531e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7214e-04, 1.5378e-05, 3.3722e-02,  ..., 1.0864e-02,
           6.3362e-03, 3.3521e-01]],

         [[5.1856e-05, 9.5904e-05, 1.0199e-01,  ..., 4.5990e-02,
           1.2665e-02, 7.0374e-02]],

         [[2.2662e-04, 1.0223e-03, 3.5840e-01,  ..., 2.1267e-03,
           6.6795e-03, 1.5802e-03]],

         ...,

         [[2.4939e-04, 1.8656e-05, 3.3630e-02,  ..., 1.5610e-02,
           7.9870e-04, 3.6804e-02]],

         [[6.1455e-03, 8.9836e-04, 2.0667e-01,  ..., 4.0619e-02,
           3.8544e-02, 6.1829e-02]],

         [[2.6584e-05, 1.5259e-05, 1.3115e-02,  ..., 2.2232e-02,
           2.2449e-03, 3.0322e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.8515e-04, 8.1348e-04, 2.0020e-01,  ..., 1.4526e-02,
           4.5380e-02, 7.0801e-02]],

         [[3.7193e-04, 4.9496e-04, 2.4146e-01,  ..., 3.1185e-03,
           1.9745e-02, 3.3844e-02]],

         [[1.5402e-03, 1.3371e-03, 2.3975e-01,  ..., 6.9313e-03,
           4.4281e-02, 1.1971e-02]],

         ...,

         [[5.8651e-04, 9.4235e-05, 2.5586e-01,  ..., 2.2316e-03,
           2.2089e-04, 9.0332e-03]],

         [[1.7223e-03, 2.4109e-03, 1.2317e-01,  ..., 2.8954e-03,
           1.7609e-02, 2.8638e-01]],

         [[1.9121e-04, 6.4552e-05, 5.2612e-02,  ..., 2.3842e-03,
           2.7084e-02, 2.2498e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.3571e-03, 1.6880e-03, 2.2373e-03,  ..., 2.7515e-01,
           1.0126e-01, 4.5386e-01]],

         [[2.3537e-03, 8.1348e-04, 4.9820e-03,  ..., 1.5503e-01,
           7.3242e-02, 1.9714e-02]],

         [[2.9802e-07, 2.2650e-06, 1.2255e-04,  ..., 2.4605e-03,
           9.0504e-04, 9.9268e-01]],

         ...,

         [[6.0201e-06, 2.0981e-05, 3.4630e-05,  ..., 6.9523e-04,
           6.5956e-03, 9.7900e-01]],

         [[3.2806e-03, 3.2806e-03, 1.6663e-02,  ..., 2.9251e-02,
           2.2781e-02, 9.0088e-02]],

         [[4.3221e-03, 2.9716e-03, 2.4390e-04,  ..., 7.5874e-03,
           4.9469e-02, 1.6068e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2783e-04, 2.5005e-03, 4.8180e-03,  ..., 1.7249e-01,
           1.2427e-01, 1.4307e-01]],

         [[3.1590e-06, 1.5748e-04, 1.7846e-04,  ..., 1.5247e-01,
           7.1582e-01, 1.1871e-01]],

         [[3.2578e-03, 3.3607e-03, 8.3237e-03,  ..., 7.7698e-02,
           3.8483e-02, 8.7204e-03]],

         ...,

         [[3.2663e-04, 5.4693e-04, 4.3964e-04,  ..., 1.4221e-02,
           7.2363e-01, 2.1387e-01]],

         [[4.0092e-03, 1.0078e-02, 3.0579e-02,  ..., 6.3477e-02,
           3.4119e-02, 1.8250e-02]],

         [[1.5473e-04, 1.0490e-03, 1.3554e-04,  ..., 7.6074e-01,
           1.2512e-01, 1.0597e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8295e-02, 1.7357e-03, 5.4092e-03,  ..., 1.8335e-01,
           7.7637e-02, 5.3772e-02]],

         [[2.1400e-03, 4.2486e-04, 1.8673e-03,  ..., 1.7748e-03,
           2.1400e-03, 1.9722e-03]],

         [[2.8286e-03, 4.8208e-04, 5.3120e-04,  ..., 8.3984e-02,
           1.0887e-02, 1.2672e-02]],

         ...,

         [[2.7809e-03, 4.1652e-04, 2.7752e-04,  ..., 2.6855e-01,
           4.3579e-01, 1.7334e-01]],

         [[8.0872e-04, 2.0611e-04, 5.3024e-04,  ..., 1.5869e-02,
           3.2318e-02, 1.3573e-02]],

         [[1.7685e-02, 3.4561e-03, 7.7286e-03,  ..., 2.2766e-01,
           9.1248e-02, 4.8462e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.6660e-01, 3.5763e-07, 2.9802e-07,  ..., 8.8654e-03,
           1.8091e-03, 5.3482e-03]],

         [[9.7559e-01, 1.7405e-05, 2.3246e-06,  ..., 5.3444e-03,
           1.9274e-03, 6.0081e-04]],

         [[6.4258e-01, 6.5565e-07, 5.3644e-07,  ..., 6.2370e-03,
           5.0201e-03, 1.7029e-02]],

         ...,

         [[2.0813e-01, 1.3053e-05, 1.3220e-04,  ..., 9.3613e-03,
           2.2945e-03, 1.0567e-02]],

         [[2.9224e-01, 2.9206e-05, 1.6415e-04,  ..., 4.7333e-02,
           7.5493e-03, 4.4899e-03]],

         [[6.9189e-01, 1.5616e-04, 3.1257e-04,  ..., 2.6932e-02,
           1.2941e-03, 3.0746e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0576e-01, 1.4305e-06, 1.5497e-06,  ..., 9.2010e-03,
           3.0499e-03, 2.9968e-02]],

         [[2.4146e-01, 4.5359e-05, 1.7881e-07,  ..., 4.7668e-02,
           6.9434e-01, 1.5503e-02]],

         [[9.7168e-01, 4.0507e-04, 2.0862e-05,  ..., 8.0109e-03,
           5.6992e-03, 1.2283e-03]],

         ...,

         [[5.5908e-01, 5.4777e-05, 3.2842e-05,  ..., 1.6138e-01,
           2.4216e-02, 1.4381e-02]],

         [[7.3486e-02, 1.1444e-03, 6.1131e-04,  ..., 1.1742e-02,
           2.1858e-03, 2.6836e-03]],

         [[6.5137e-01, 3.8385e-05, 9.9540e-06,  ..., 1.7590e-01,
           3.8788e-02, 2.1301e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.1204e-01, 1.2910e-04, 6.9618e-05,  ..., 9.3323e-02,
           3.2166e-02, 1.2115e-02]],

         [[5.3906e-01, 2.1458e-05, 5.6028e-06,  ..., 2.7856e-01,
           1.1554e-01, 2.0828e-02]],

         [[7.9932e-01, 1.1742e-05, 6.1989e-06,  ..., 2.0584e-02,
           5.1819e-02, 2.5421e-02]],

         ...,

         [[7.0117e-01, 8.2254e-06, 8.9407e-07,  ..., 2.0647e-04,
           1.2302e-03, 2.9346e-01]],

         [[9.5410e-01, 4.8161e-05, 3.8803e-05,  ..., 8.3447e-06,
           2.7275e-04, 2.9510e-02]],

         [[9.9072e-01, 2.4617e-05, 2.5630e-06,  ..., 7.7486e-07,
           3.3796e-05, 7.2937e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3613e-01, 2.1100e-05, 9.5367e-07,  ..., 8.3557e-02,
           2.8271e-01, 1.3374e-02]],

         [[4.2065e-01, 1.7881e-06, 1.3113e-06,  ..., 3.1776e-03,
           2.0337e-04, 4.1695e-03]],

         [[8.4766e-01, 9.1934e-04, 4.8161e-05,  ..., 1.7014e-02,
           5.6877e-03, 4.7607e-03]],

         ...,

         [[8.3105e-01, 3.4356e-04, 2.6882e-05,  ..., 1.0025e-02,
           2.0111e-02, 1.5686e-02]],

         [[5.9375e-01, 1.6069e-04, 1.8597e-05,  ..., 2.1301e-02,
           1.1115e-01, 7.9895e-02]],

         [[9.3701e-01, 7.1764e-05, 6.6400e-05,  ..., 1.5350e-02,
           3.0670e-03, 8.2397e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.9722e-01, 6.7353e-06, 1.0967e-05,  ..., 1.5381e-01,
           1.0139e-02, 1.3763e-02]],

         [[3.2275e-01, 5.5671e-05, 9.8348e-06,  ..., 9.3445e-02,
           4.7241e-02, 7.8308e-02]],

         [[9.0967e-01, 3.6538e-05, 1.0729e-05,  ..., 9.0408e-03,
           2.2232e-02, 5.6152e-03]],

         ...,

         [[5.7373e-01, 5.8651e-04, 2.5821e-04,  ..., 1.2543e-02,
           3.3051e-02, 8.0185e-03]],

         [[8.1250e-01, 3.9887e-04, 4.9114e-05,  ..., 6.1417e-03,
           5.9776e-03, 1.2627e-02]],

         [[3.9697e-01, 5.3048e-06, 5.2452e-06,  ..., 1.0669e-01,
           2.9150e-01, 3.6072e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.4883e-01, 1.1597e-03, 1.0767e-03,  ..., 9.1934e-03,
           1.2388e-03, 4.3154e-04]],

         [[8.6328e-01, 4.1068e-05, 9.9301e-05,  ..., 6.9618e-03,
           1.8021e-02, 1.1436e-02]],

         [[5.8838e-01, 8.4281e-05, 3.3021e-05,  ..., 5.1537e-03,
           1.0887e-02, 8.4991e-03]],

         ...,

         [[7.8662e-01, 3.5346e-05, 2.0564e-05,  ..., 3.0251e-03,
           4.3396e-02, 3.0655e-02]],

         [[7.3291e-01, 5.3644e-06, 6.7949e-06,  ..., 2.9411e-03,
           1.9958e-01, 4.3030e-02]],

         [[9.3359e-01, 1.4842e-05, 4.8280e-06,  ..., 1.0824e-03,
           2.9221e-02, 5.4550e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.0796e-01, 8.0252e-04, 1.2722e-03,  ..., 8.3618e-02,
           2.8057e-03, 1.8291e-03]],

         [[8.2812e-01, 1.6689e-06, 1.4305e-06,  ..., 2.9007e-02,
           1.2067e-01, 1.3832e-02]],

         [[6.7236e-01, 3.2783e-06, 8.5831e-06,  ..., 1.1345e-02,
           2.6276e-02, 1.2238e-02]],

         ...,

         [[3.0591e-01, 2.8777e-04, 3.8147e-04,  ..., 1.0907e-01,
           1.0862e-03, 1.7300e-03]],

         [[9.4629e-01, 2.4676e-05, 1.7285e-05,  ..., 3.3875e-03,
           1.2817e-02, 4.7646e-03]],

         [[9.4043e-01, 6.4969e-06, 7.0333e-06,  ..., 2.9697e-03,
           4.7226e-03, 2.2736e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.9639e-01, 5.1677e-05, 9.6560e-05,  ..., 6.6956e-02,
           1.5526e-03, 7.6103e-03]],

         [[7.8418e-01, 1.6332e-05, 1.1861e-05,  ..., 3.0487e-02,
           2.0752e-02, 1.2756e-02]],

         [[2.3975e-01, 7.7486e-07, 2.0266e-06,  ..., 1.1108e-01,
           7.9155e-04, 3.3836e-03]],

         ...,

         [[8.4570e-01, 1.2541e-04, 9.1016e-05,  ..., 8.2474e-03,
           8.9111e-03, 2.5574e-02]],

         [[4.1992e-01, 2.6083e-04, 3.2091e-04,  ..., 1.5411e-02,
           1.3672e-01, 1.9751e-01]],

         [[4.1113e-01, 3.8981e-04, 2.7061e-04,  ..., 7.7324e-03,
           2.2125e-03, 2.8553e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.0017e-02, 3.5763e-07, 1.4901e-06,  ..., 4.7946e-04,
           1.2909e-02, 9.0137e-01]],

         [[6.8311e-01, 1.8139e-03, 2.0866e-03,  ..., 2.4414e-03,
           8.4686e-04, 2.2829e-04]],

         [[5.4541e-01, 5.9891e-04, 4.3130e-04,  ..., 9.8572e-02,
           1.9638e-02, 3.8147e-02]],

         ...,

         [[6.1670e-01, 1.2362e-04, 1.8334e-04,  ..., 3.0533e-02,
           1.5671e-02, 3.6049e-03]],

         [[2.0129e-01, 4.4227e-05, 7.2062e-05,  ..., 3.7048e-02,
           2.5537e-01, 4.2114e-02]],

         [[1.2421e-01, 4.4167e-05, 2.7180e-04,  ..., 1.1559e-02,
           5.9082e-02, 7.4170e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2471e-01, 1.9038e-04, 2.0182e-04,  ..., 2.6260e-02,
           1.1462e-01, 4.0161e-01]],

         [[5.7959e-01, 1.6820e-04, 2.1601e-04,  ..., 9.9548e-02,
           4.0039e-02, 2.9251e-02]],

         [[4.9756e-01, 1.8835e-04, 9.3579e-05,  ..., 1.0307e-02,
           4.2786e-02, 4.3640e-03]],

         ...,

         [[1.7441e-02, 1.7881e-07, 2.6226e-06,  ..., 7.0214e-05,
           4.5509e-03, 9.7510e-01]],

         [[8.8330e-01, 5.1677e-05, 4.6313e-05,  ..., 2.2812e-03,
           1.3924e-02, 1.4694e-02]],

         [[6.7578e-01, 8.6546e-05, 4.9293e-05,  ..., 2.9964e-03,
           1.0391e-02, 2.5330e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 6.1178e-04, 6.1178e-04,  ..., 2.7740e-02,
           6.2904e-03, 7.3166e-03]],

         [[8.9551e-01, 4.2439e-05, 9.1970e-05,  ..., 2.4811e-02,
           2.0447e-02, 3.0212e-02]],

         [[4.2334e-01, 1.8060e-05, 2.9445e-05,  ..., 4.7760e-02,
           4.2534e-03, 2.1229e-03]],

         ...,

         [[4.4507e-01, 3.6955e-06, 1.7881e-07,  ..., 1.2164e-01,
           3.9893e-01, 3.1494e-02]],

         [[4.7217e-01, 5.9891e-04, 7.1383e-04,  ..., 1.4183e-02,
           9.3765e-03, 3.7556e-03]],

         [[6.6553e-01, 6.9737e-06, 1.2517e-06,  ..., 2.3376e-02,
           2.6367e-01, 3.4515e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7041e-01, 9.1195e-05, 2.8372e-05,  ..., 2.6741e-03,
           1.5762e-02, 5.4474e-03]],

         [[1.8958e-01, 4.8566e-04, 5.2977e-04,  ..., 1.6724e-01,
           2.9430e-03, 2.5120e-03]],

         [[5.8789e-01, 7.4291e-04, 8.7881e-04,  ..., 6.5063e-02,
           5.1613e-03, 8.1177e-03]],

         ...,

         [[3.8892e-01, 9.5367e-06, 5.5432e-06,  ..., 4.0186e-01,
           1.3855e-01, 2.0538e-02]],

         [[4.4360e-01, 6.4373e-06, 1.4305e-06,  ..., 1.3525e-01,
           3.8428e-01, 2.8336e-02]],

         [[4.9365e-01, 4.3988e-05, 2.8133e-05,  ..., 2.7832e-01,
           5.2002e-02, 1.5640e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0660e-02, 6.5565e-07, 2.5034e-06,  ..., 1.0223e-03,
           1.5297e-02, 9.5361e-01]],

         [[3.3276e-01, 1.4305e-06, 2.0266e-06,  ..., 3.2642e-01,
           2.2437e-01, 4.9866e-02]],

         [[2.1619e-01, 2.7609e-04, 3.2210e-04,  ..., 4.1107e-02,
           6.4316e-03, 2.8229e-03]],

         ...,

         [[2.0923e-01, 3.3319e-05, 1.1683e-05,  ..., 1.4209e-01,
           2.5085e-02, 7.2250e-03]],

         [[2.1912e-01, 2.8014e-06, 1.7881e-06,  ..., 2.6489e-01,
           5.9967e-03, 6.7215e-03]],

         [[3.0469e-01, 1.5736e-04, 5.2691e-05,  ..., 1.2711e-02,
           2.4719e-02, 9.2926e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.2637e-01, 5.9843e-05, 3.9399e-05,  ..., 7.4341e-02,
           6.7291e-03, 5.5008e-03]],

         [[7.9956e-02, 7.7009e-05, 1.7822e-05,  ..., 2.2253e-01,
           8.3313e-03, 3.5133e-03]],

         [[1.6064e-01, 9.7394e-05, 1.0133e-05,  ..., 7.1838e-02,
           6.0516e-02, 1.3611e-02]],

         ...,

         [[3.9941e-01, 1.8179e-05, 9.1195e-06,  ..., 1.9934e-01,
           1.6235e-02, 1.3832e-02]],

         [[7.0801e-01, 3.3319e-05, 5.3406e-05,  ..., 3.2990e-02,
           3.9642e-02, 4.6692e-02]],

         [[3.2397e-01, 4.7684e-05, 3.4213e-05,  ..., 1.9043e-01,
           7.3547e-03, 1.0132e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5564e-01, 1.6403e-03, 2.0981e-03,  ..., 2.0050e-02,
           9.1887e-04, 9.2649e-04]],

         [[6.7200e-02, 7.6652e-05, 1.2016e-04,  ..., 6.1401e-02,
           1.4359e-02, 6.5613e-03]],

         [[4.5874e-01, 6.9737e-06, 9.1791e-06,  ..., 1.4893e-01,
           1.0480e-01, 2.7328e-02]],

         ...,

         [[6.2109e-01, 7.8261e-05, 4.9293e-05,  ..., 1.5900e-02,
           6.4087e-03, 9.4376e-03]],

         [[9.7412e-01, 1.0073e-05, 2.6464e-05,  ..., 2.0981e-04,
           4.0984e-04, 6.3248e-03]],

         [[6.1377e-01, 8.2195e-05, 1.9908e-04,  ..., 4.2175e-02,
           7.0877e-03, 1.0056e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5928e-01, 1.7858e-04, 2.8086e-04,  ..., 1.2445e-01,
           7.1869e-03, 1.0216e-02]],

         [[2.3181e-01, 1.3571e-03, 7.2050e-04,  ..., 1.2598e-01,
           1.0841e-02, 6.0806e-03]],

         [[1.6064e-01, 6.6161e-06, 6.4373e-06,  ..., 2.6428e-02,
           2.1277e-01, 4.8706e-01]],

         ...,

         [[2.4915e-01, 4.5896e-06, 4.6492e-06,  ..., 2.7588e-01,
           1.9211e-02, 1.1208e-02]],

         [[5.3662e-01, 6.9141e-06, 4.7088e-06,  ..., 9.1797e-02,
           1.1334e-01, 2.2324e-02]],

         [[1.9690e-01, 3.9339e-06, 4.4703e-06,  ..., 1.1218e-01,
           3.7354e-01, 1.2915e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.4131e-01, 7.8082e-06, 4.1723e-06,  ..., 1.4233e-01,
           3.6182e-03, 8.4152e-03]],

         [[1.6895e-01, 1.6749e-05, 1.9908e-05,  ..., 7.8552e-02,
           7.8552e-02, 1.9547e-02]],

         [[1.4099e-01, 4.8280e-06, 1.3709e-06,  ..., 5.5542e-03,
           4.4373e-02, 7.8662e-01]],

         ...,

         [[1.4014e-01, 4.9353e-04, 1.8156e-04,  ..., 1.2634e-02,
           7.0877e-03, 4.2000e-03]],

         [[4.1968e-01, 4.1699e-04, 2.7514e-04,  ..., 1.1765e-02,
           3.3436e-03, 4.0817e-03]],

         [[2.3889e-01, 3.2687e-04, 1.9991e-04,  ..., 8.2520e-02,
           1.7166e-02, 8.5678e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.4319e-01, 6.7472e-05, 2.4557e-05,  ..., 6.6589e-02,
           2.8198e-02, 9.0866e-03]],

         [[8.8037e-01, 1.8489e-04, 1.0490e-05,  ..., 3.2246e-05,
           2.3782e-05, 4.5753e-04]],

         [[8.6816e-01, 6.7592e-05, 7.0333e-06,  ..., 2.9826e-04,
           1.2243e-04, 1.2655e-03]],

         ...,

         [[6.2988e-02, 9.9468e-04, 2.0256e-03,  ..., 4.6814e-02,
           2.6207e-03, 4.5538e-04]],

         [[4.4263e-01, 1.5497e-05, 2.2054e-05,  ..., 3.5217e-02,
           5.5389e-02, 4.0741e-03]],

         [[3.4497e-01, 6.3038e-04, 1.7958e-03,  ..., 7.2876e-02,
           9.6359e-03, 2.1545e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8062e-01, 1.3268e-04, 2.6393e-04,  ..., 7.4196e-03,
           1.7792e-02, 2.1469e-02]],

         [[3.0981e-01, 1.6224e-04, 3.9554e-04,  ..., 1.1200e-02,
           1.2794e-02, 4.1542e-03]],

         [[2.4011e-01, 4.4644e-05, 1.0341e-04,  ..., 1.1765e-02,
           6.3477e-03, 2.3727e-03]],

         ...,

         [[1.5839e-02, 2.6584e-05, 6.1750e-05,  ..., 1.0281e-03,
           3.0231e-03, 2.7523e-03]],

         [[1.1975e-01, 7.7426e-05, 8.1122e-05,  ..., 1.8356e-02,
           4.3602e-03, 3.2921e-03]],

         [[5.6885e-01, 9.3758e-05, 1.1945e-04,  ..., 8.7738e-03,
           7.1640e-03, 1.9283e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-02, 1.2517e-05, 1.1742e-05,  ..., 1.5381e-02,
           1.9897e-02, 3.5686e-03]],

         [[1.9897e-01, 9.1910e-05, 1.2910e-04,  ..., 9.2316e-03,
           1.1398e-02, 4.7874e-03]],

         [[2.6025e-01, 9.1851e-05, 3.9744e-04,  ..., 5.2887e-02,
           3.4668e-02, 4.8242e-01]],

         ...,

         [[4.5239e-01, 1.9848e-05, 2.6524e-05,  ..., 7.0984e-02,
           6.7596e-03, 4.5395e-03]],

         [[4.8779e-01, 1.0359e-04, 2.9397e-04,  ..., 6.6910e-03,
           4.1962e-02, 1.8326e-02]],

         [[3.6035e-01, 4.0174e-05, 5.8651e-05,  ..., 2.5043e-03,
           9.8038e-04, 2.8820e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.9182e-02, 5.5790e-05, 3.0696e-05,  ..., 2.7985e-02,
           2.7756e-02, 1.0826e-02]],

         [[2.0645e-02, 5.6624e-06, 9.0599e-06,  ..., 2.7130e-02,
           7.3181e-02, 1.6232e-03]],

         [[3.5132e-01, 4.8280e-05, 5.8651e-05,  ..., 1.7014e-02,
           7.2632e-03, 8.4534e-03]],

         ...,

         [[4.7656e-01, 2.0266e-06, 3.5763e-07,  ..., 8.2825e-02,
           3.4863e-01, 2.0935e-02]],

         [[4.8462e-01, 1.1802e-05, 2.4259e-05,  ..., 1.8250e-01,
           7.0923e-02, 2.0309e-02]],

         [[4.0112e-01, 3.6955e-06, 3.5763e-06,  ..., 2.1143e-01,
           3.3789e-01, 3.7018e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6495e-02, 1.6689e-06, 1.2517e-05,  ..., 3.6896e-02,
           3.3081e-02, 1.0818e-02]],

         [[1.1261e-02, 2.3246e-06, 2.8014e-05,  ..., 7.9679e-04,
           1.7853e-02, 9.5947e-01]],

         [[9.7473e-02, 2.3127e-05, 1.3876e-04,  ..., 3.3417e-02,
           1.1482e-03, 1.4620e-03]],

         ...,

         [[2.9932e-01, 1.0496e-04, 3.4237e-04,  ..., 1.1450e-01,
           1.0246e-02, 6.1035e-03]],

         [[4.6362e-01, 6.5565e-07, 1.6093e-06,  ..., 7.2510e-02,
           5.3467e-02, 2.9877e-02]],

         [[2.3547e-01, 1.4246e-05, 8.5831e-06,  ..., 1.9226e-01,
           4.3311e-01, 1.0614e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.2451e-01, 1.5986e-04, 9.2983e-04,  ..., 3.3630e-02,
           4.4975e-03, 5.4893e-03]],

         [[9.8389e-01, 2.1160e-05, 8.7440e-05,  ..., 4.6897e-04,
           6.9976e-05, 3.2735e-04]],

         [[8.0273e-01, 3.0994e-05, 2.8014e-05,  ..., 9.7961e-03,
           7.0095e-04, 4.8184e-04]],

         ...,

         [[3.6597e-01, 2.9802e-06, 3.3379e-06,  ..., 2.9114e-02,
           1.5106e-02, 6.1035e-03]],

         [[3.3569e-01, 9.5367e-07, 3.5763e-07,  ..., 1.0767e-03,
           1.2207e-03, 2.1935e-03]],

         [[1.2207e-01, 7.6890e-06, 3.0220e-05,  ..., 5.0392e-03,
           3.4973e-02, 7.5928e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7627e-02, 1.5497e-06, 4.9472e-06,  ..., 2.6538e-01,
           1.6345e-01, 6.3049e-02]],

         [[3.2031e-01, 1.0073e-05, 6.9737e-06,  ..., 3.6346e-02,
           3.2990e-02, 2.2400e-02]],

         [[1.5039e-01, 1.8501e-04, 1.7929e-04,  ..., 9.8038e-03,
           9.7275e-03, 2.6455e-03]],

         ...,

         [[2.5635e-01, 8.1062e-06, 4.0352e-05,  ..., 1.0767e-01,
           1.6769e-02, 6.0959e-03]],

         [[5.2441e-01, 9.1791e-06, 4.8280e-05,  ..., 7.4402e-02,
           1.3130e-02, 2.3041e-02]],

         [[5.1270e-01, 4.4703e-06, 1.4782e-05,  ..., 8.5022e-02,
           1.0315e-02, 2.5528e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8654e-03, 8.3447e-07, 5.3644e-07,  ..., 2.3136e-03,
           3.5065e-02, 3.6392e-03]],

         [[2.6758e-01, 2.5749e-05, 1.0669e-05,  ..., 3.7727e-03,
           2.5463e-04, 4.1676e-04]],

         [[2.1655e-01, 5.2929e-05, 9.0361e-05,  ..., 2.3361e-02,
           2.8896e-03, 2.9125e-03]],

         ...,

         [[6.1554e-02, 8.0466e-06, 1.2648e-04,  ..., 1.0803e-01,
           1.2901e-02, 2.5269e-02]],

         [[2.7686e-01, 2.0742e-05, 1.4615e-04,  ..., 1.0028e-01,
           1.0910e-02, 1.9135e-02]],

         [[8.8574e-01, 2.4021e-05, 3.4153e-05,  ..., 2.5131e-02,
           1.8387e-02, 1.7273e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8613e-01, 1.0014e-05, 1.4603e-05,  ..., 1.5266e-02,
           4.3854e-02, 7.0862e-02]],

         [[3.4619e-01, 8.8215e-06, 5.0843e-05,  ..., 1.4539e-01,
           1.3037e-01, 6.1340e-02]],

         [[1.1584e-01, 8.2254e-06, 7.3910e-06,  ..., 2.9590e-01,
           5.1318e-01, 5.5603e-02]],

         ...,

         [[4.4824e-01, 4.0650e-05, 7.7248e-05,  ..., 9.1248e-03,
           1.1223e-02, 4.5891e-03]],

         [[7.7271e-02, 1.9670e-06, 2.6226e-06,  ..., 5.7404e-02,
           2.1497e-01, 1.6449e-02]],

         [[4.7089e-02, 9.4295e-05, 7.1764e-04,  ..., 6.7558e-03,
           1.1810e-02, 5.7373e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2983e-01, 3.0339e-05, 1.2636e-04,  ..., 3.0685e-02,
           6.7139e-03, 5.6534e-03]],

         [[5.3760e-01, 1.0180e-04, 2.9778e-04,  ..., 3.9940e-03,
           1.6012e-03, 1.4277e-03]],

         [[9.3323e-02, 7.0512e-05, 1.4448e-04,  ..., 3.1300e-03,
           7.9489e-04, 1.7090e-03]],

         ...,

         [[5.6689e-01, 2.9683e-04, 3.2353e-04,  ..., 9.4528e-03,
           1.7380e-02, 6.0669e-02]],

         [[7.4463e-02, 3.2604e-05, 1.9038e-04,  ..., 9.6130e-03,
           3.1036e-02, 8.1299e-01]],

         [[8.6035e-01, 2.2411e-05, 4.7445e-05,  ..., 1.0826e-02,
           3.4599e-03, 4.0222e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.2139e-01, 8.5473e-05, 5.6171e-04,  ..., 7.2021e-03,
           5.2986e-03, 4.6654e-03]],

         [[5.7831e-02, 6.5565e-06, 3.9279e-05,  ..., 1.8997e-02,
           6.0120e-02, 2.0538e-02]],

         [[1.2805e-01, 1.2791e-04, 3.2163e-04,  ..., 6.0974e-02,
           6.8054e-02, 5.8655e-02]],

         ...,

         [[6.8945e-01, 4.5121e-05, 3.5238e-04,  ..., 2.1942e-02,
           4.7455e-03, 2.4719e-02]],

         [[8.4277e-01, 1.1921e-06, 2.1207e-04,  ..., 3.3245e-03,
           8.3351e-04, 9.6941e-04]],

         [[3.2275e-01, 2.6941e-05, 3.7074e-05,  ..., 2.2545e-03,
           6.7520e-03, 1.7929e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.4254e-05, 4.7445e-05, 9.9609e-02,  ..., 3.0746e-02,
           1.1444e-02, 2.3758e-02]],

         [[1.1683e-05, 1.2219e-05, 6.4209e-02,  ..., 5.0507e-03,
           4.8370e-03, 5.7907e-03]],

         [[1.8394e-04, 2.3675e-04, 8.7097e-02,  ..., 1.2604e-02,
           2.8400e-03, 9.1858e-03]],

         ...,

         [[7.2300e-05, 5.2691e-05, 2.0813e-01,  ..., 5.8228e-02,
           7.1350e-02, 1.4307e-01]],

         [[1.8036e-04, 2.1255e-04, 2.6099e-01,  ..., 1.2848e-02,
           3.4580e-03, 7.5226e-03]],

         [[1.1384e-05, 1.2815e-05, 1.6504e-01,  ..., 2.2984e-03,
           3.4308e-04, 1.6174e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1301e-04, 1.1301e-04, 2.9395e-01,  ..., 7.6485e-03,
           1.2293e-03, 3.1033e-03]],

         [[6.2168e-05, 7.6771e-05, 1.0962e-01,  ..., 5.4131e-03,
           1.1683e-04, 2.1877e-03]],

         [[2.7597e-05, 1.1444e-05, 2.7710e-01,  ..., 1.3062e-02,
           1.6510e-04, 3.9673e-03]],

         ...,

         [[1.6975e-04, 6.9559e-05, 2.2290e-01,  ..., 6.7825e-03,
           2.6655e-04, 1.7624e-03]],

         [[4.8280e-06, 1.3530e-05, 2.1387e-01,  ..., 4.7989e-03,
           6.3062e-05, 1.7042e-03]],

         [[3.3140e-05, 5.0843e-05, 2.2449e-01,  ..., 8.0017e-02,
           6.8426e-04, 1.8570e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1325e-05, 1.5020e-05, 1.1218e-01,  ..., 4.7417e-03,
           7.6234e-05, 8.4305e-04]],

         [[9.2328e-05, 6.1393e-06, 2.3730e-01,  ..., 8.9493e-03,
           1.0985e-04, 2.2907e-03]],

         [[4.4227e-05, 6.2764e-05, 2.4084e-01,  ..., 7.7438e-03,
           2.0039e-04, 2.4948e-03]],

         ...,

         [[5.4121e-05, 2.7478e-05, 1.1621e-01,  ..., 1.1780e-02,
           1.4801e-03, 7.0343e-03]],

         [[1.5855e-05, 4.0948e-05, 1.8579e-01,  ..., 6.6910e-03,
           2.8443e-04, 2.9736e-03]],

         [[5.4181e-05, 3.5942e-05, 1.2561e-01,  ..., 3.7498e-03,
           1.9372e-04, 2.5978e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.3241e-05, 6.6757e-06, 1.9714e-02,  ..., 1.3466e-03,
           2.4011e-01, 8.6975e-02]],

         [[4.6074e-05, 5.3227e-05, 1.2512e-01,  ..., 1.0406e-02,
           6.3416e-02, 3.1189e-02]],

         [[1.9932e-04, 4.9400e-04, 3.0664e-01,  ..., 5.8517e-03,
           8.9025e-04, 7.5102e-04]],

         ...,

         [[1.7178e-04, 6.0201e-06, 3.9032e-02,  ..., 1.7452e-04,
           3.0098e-03, 5.3635e-03]],

         [[2.2202e-03, 2.2960e-04, 1.1884e-01,  ..., 2.3178e-02,
           7.6721e-02, 8.2947e-02]],

         [[6.3181e-06, 2.3842e-06, 1.9255e-03,  ..., 5.9652e-04,
           2.0190e-01, 1.0553e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.1625e-03, 6.1893e-04, 1.7712e-01,  ..., 3.0304e-02,
           2.0416e-02, 6.0486e-02]],

         [[3.6645e-04, 3.4571e-04, 2.2559e-01,  ..., 1.4366e-02,
           1.7883e-02, 5.2124e-02]],

         [[1.6642e-03, 4.9400e-04, 2.6978e-01,  ..., 2.5497e-02,
           7.8354e-03, 7.9041e-03]],

         ...,

         [[4.5037e-04, 3.9995e-05, 2.7710e-01,  ..., 2.8634e-04,
           3.7422e-03, 2.8091e-02]],

         [[1.5802e-03, 1.7767e-03, 1.8408e-01,  ..., 9.4528e-03,
           1.4648e-02, 4.8767e-02]],

         [[1.1355e-04, 5.9366e-05, 4.2023e-02,  ..., 6.9695e-03,
           1.5381e-01, 2.8271e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[3.9458e-04, 1.1063e-03, 1.4801e-02,  ..., 8.7891e-02,
           3.5864e-01, 8.7891e-02]],

         [[1.2577e-04, 2.3496e-04, 1.6317e-03,  ..., 1.2833e-02,
           1.3794e-01, 1.2964e-01]],

         [[0.0000e+00, 5.9605e-08, 1.3709e-06,  ..., 2.9206e-06,
           5.5218e-04, 9.9854e-01]],

         ...,

         [[5.9605e-08, 1.1921e-07, 0.0000e+00,  ..., 1.2279e-04,
           9.0742e-04, 9.9512e-01]],

         [[4.0436e-03, 5.8823e-03, 3.7994e-03,  ..., 2.9877e-02,
           2.4765e-02, 5.8823e-03]],

         [[4.6272e-03, 1.1702e-03, 2.0981e-04,  ..., 2.1393e-02,
           1.0757e-02, 8.2031e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.5373e-05, 5.0926e-04, 1.6699e-03,  ..., 1.1955e-02,
           8.8379e-02, 5.4102e-01]],

         [[5.9605e-08, 1.2755e-05, 1.9014e-04,  ..., 9.8511e-02,
           7.9932e-01, 8.6914e-02]],

         [[4.4584e-04, 2.4872e-03, 6.5498e-03,  ..., 4.6906e-02,
           2.9358e-02, 4.6906e-02]],

         ...,

         [[1.3423e-04, 7.9679e-04, 6.0606e-04,  ..., 2.0923e-01,
           2.0923e-01, 4.0308e-01]],

         [[1.3075e-03, 6.0921e-03, 2.7206e-02,  ..., 3.1311e-02,
           2.1103e-02, 1.9821e-02]],

         [[7.3385e-04, 2.1400e-03, 1.1911e-03,  ..., 1.0229e-01,
           5.7959e-01, 1.1060e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[0.0356, 0.0031, 0.0159,  ..., 0.0487, 0.0373, 0.0433]],

         [[0.0009, 0.0004, 0.0019,  ..., 0.0032, 0.0038, 0.0158]],

         [[0.0034, 0.0007, 0.0008,  ..., 0.0130, 0.0167, 0.0170]],

         ...,

         [[0.0029, 0.0036, 0.0070,  ..., 0.2454, 0.2454, 0.2306]],

         [[0.0028, 0.0010, 0.0012,  ..., 0.0289, 0.0291, 0.1469]],

         [[0.0192, 0.0083, 0.0321,  ..., 0.0375, 0.0350, 0.2372]]]],
       device='cuda:0', dtype=torch.float16), tensor([[[[8.9551e-01, 1.4305e-06, 1.1921e-06,  ..., 7.1383e-04,
           1.3599e-03, 1.2428e-02]],

         [[9.3604e-01, 2.1577e-05, 1.6689e-06,  ..., 1.3016e-02,
           2.5635e-02, 1.0498e-02]],

         [[8.3887e-01, 7.9870e-06, 5.3048e-06,  ..., 2.9736e-03,
           4.6844e-03, 4.4830e-02]],

         ...,

         [[1.8213e-01, 5.1260e-05, 3.9625e-04,  ..., 4.1127e-06,
           3.3975e-06, 2.6550e-02]],

         [[5.0586e-01, 2.1231e-04, 4.2605e-04,  ..., 1.7166e-02,
           5.7411e-03, 2.3285e-02]],

         [[4.6021e-01, 3.2091e-04, 2.2554e-04,  ..., 2.6276e-02,
           7.2266e-02, 1.6830e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8857e-01, 1.4186e-05, 5.3048e-06,  ..., 7.4506e-05,
           7.2181e-05, 3.8666e-02]],

         [[9.5020e-01, 1.2934e-04, 7.2241e-05,  ..., 4.7302e-03,
           3.1113e-02, 1.2772e-02]],

         [[9.4824e-01, 2.0456e-04, 1.2898e-04,  ..., 3.3531e-03,
           1.7105e-02, 6.8550e-03]],

         ...,

         [[3.8623e-01, 2.8539e-04, 1.2589e-04,  ..., 8.1116e-02,
           1.0559e-01, 3.2196e-02]],

         [[2.5439e-01, 7.5674e-04, 3.5095e-04,  ..., 8.9417e-03,
           4.6692e-03, 4.7646e-03]],

         [[7.8857e-01, 4.7755e-04, 2.0647e-04,  ..., 2.0798e-02,
           2.3422e-02, 5.7587e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9592e-01, 2.9516e-04, 7.0393e-05,  ..., 6.8848e-02,
           3.4302e-02, 4.2084e-02]],

         [[7.4707e-01, 4.0054e-05, 1.1206e-05,  ..., 4.6539e-02,
           3.9673e-02, 4.3488e-02]],

         [[6.0596e-01, 1.2636e-05, 3.8624e-05,  ..., 2.5803e-02,
           8.8867e-02, 4.2572e-02]],

         ...,

         [[8.3936e-01, 6.9141e-06, 2.9802e-07,  ..., 1.2517e-06,
           4.4107e-05, 1.5222e-01]],

         [[9.2969e-01, 6.6400e-05, 1.9073e-06,  ..., 0.0000e+00,
           6.5565e-07, 1.9791e-02]],

         [[9.8828e-01, 1.7941e-05, 5.9605e-08,  ..., 0.0000e+00,
           5.9605e-08, 4.9210e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.6816e-01, 3.1292e-05, 4.7684e-06,  ..., 2.3816e-01,
           1.4685e-01, 4.4739e-02]],

         [[6.6455e-01, 7.8082e-06, 1.8477e-06,  ..., 1.1516e-04,
           2.1875e-05, 7.4402e-02]],

         [[7.2070e-01, 5.7161e-05, 9.8944e-06,  ..., 1.9684e-03,
           3.2776e-02, 1.1642e-02]],

         ...,

         [[6.6797e-01, 5.3835e-04, 2.0742e-05,  ..., 3.6755e-03,
           2.5177e-03, 2.3766e-03]],

         [[5.9717e-01, 2.3687e-04, 4.3631e-05,  ..., 9.3613e-03,
           2.0233e-02, 1.8066e-01]],

         [[8.4570e-01, 1.4663e-04, 1.7345e-04,  ..., 1.0605e-03,
           5.5542e-03, 1.0358e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.8271e-01, 1.0550e-05, 1.7107e-05,  ..., 6.7291e-03,
           4.5738e-03, 2.6505e-02]],

         [[6.6260e-01, 4.3511e-05, 3.1352e-05,  ..., 1.0750e-02,
           6.8207e-03, 3.0334e-02]],

         [[9.5508e-01, 1.3590e-05, 5.5432e-06,  ..., 5.6686e-03,
           4.6997e-03, 9.1629e-03]],

         ...,

         [[5.7275e-01, 1.5616e-04, 5.7697e-05,  ..., 4.0192e-02,
           1.8661e-02, 4.1046e-02]],

         [[8.7354e-01, 1.0759e-04, 2.5153e-05,  ..., 5.1918e-03,
           4.2953e-03, 1.3733e-02]],

         [[8.5547e-01, 1.6153e-05, 1.5140e-05,  ..., 1.9745e-02,
           4.1443e-02, 2.1729e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.7930e-01, 5.0068e-04, 3.3641e-04,  ..., 8.7929e-04,
           6.6376e-04, 2.7618e-03]],

         [[9.3408e-01, 1.6868e-05, 6.1333e-05,  ..., 8.4152e-03,
           2.7962e-03, 3.4389e-03]],

         [[9.0234e-01, 1.0788e-05, 7.9870e-06,  ..., 8.4829e-04,
           6.4564e-04, 1.9045e-03]],

         ...,

         [[8.5059e-01, 1.6451e-05, 2.4199e-05,  ..., 2.4796e-02,
           1.0529e-02, 1.8463e-02]],

         [[8.1104e-01, 3.2485e-05, 2.3246e-05,  ..., 3.7018e-02,
           2.3422e-02, 9.0820e-02]],

         [[9.3652e-01, 2.0742e-05, 1.7405e-05,  ..., 3.1853e-03,
           1.5516e-03, 1.6037e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3816e-01, 1.7226e-04, 1.0471e-03,  ..., 2.1866e-02,
           1.1009e-02, 1.3794e-02]],

         [[8.6377e-01, 1.7166e-05, 1.6510e-05,  ..., 1.9547e-02,
           3.2104e-02, 2.0889e-02]],

         [[6.3428e-01, 3.2187e-05, 7.8380e-05,  ..., 2.8458e-02,
           1.6602e-02, 2.2247e-02]],

         ...,

         [[7.4951e-01, 3.0756e-05, 1.3173e-04,  ..., 2.7237e-03,
           6.0234e-03, 1.7560e-04]],

         [[7.2510e-01, 4.8637e-05, 3.7909e-05,  ..., 1.1932e-02,
           4.5776e-03, 2.8442e-02]],

         [[8.7109e-01, 3.6001e-05, 4.0352e-05,  ..., 5.5771e-03,
           4.9225e-02, 1.8478e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.9375e-01, 4.5002e-05, 3.9697e-05,  ..., 4.1127e-06,
           7.1526e-07, 2.5208e-02]],

         [[8.1934e-01, 1.4126e-05, 3.6955e-06,  ..., 3.3905e-02,
           2.2247e-02, 5.3680e-02]],

         [[7.3096e-01, 1.6689e-06, 2.0862e-06,  ..., 1.3981e-03,
           7.9453e-05, 2.9251e-02]],

         ...,

         [[9.2627e-01, 5.2452e-05, 4.2617e-05,  ..., 1.5993e-03,
           4.5586e-03, 8.0414e-03]],

         [[7.9395e-01, 6.7890e-05, 3.8981e-05,  ..., 2.6154e-02,
           9.0942e-03, 9.8801e-04]],

         [[4.5386e-01, 4.3964e-04, 8.6904e-05,  ..., 4.8294e-03,
           1.5554e-03, 1.3397e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0640e-01, 2.6822e-06, 4.8280e-06,  ..., 2.3117e-03,
           2.1667e-03, 6.7725e-01]],

         [[8.0127e-01, 4.8542e-04, 6.4468e-04,  ..., 5.2452e-04,
           1.3673e-04, 1.7920e-03]],

         [[6.6650e-01, 6.3848e-04, 8.1205e-04,  ..., 4.6997e-03,
           5.7106e-03, 2.1942e-02]],

         ...,

         [[5.8936e-01, 1.5485e-04, 1.5485e-04,  ..., 5.1193e-03,
           6.5947e-04, 7.5626e-04]],

         [[7.7930e-01, 1.1116e-04, 9.0003e-05,  ..., 1.1604e-02,
           1.8873e-03, 1.4816e-02]],

         [[5.3125e-01, 5.5611e-05, 1.6403e-04,  ..., 1.3687e-02,
           1.8433e-02, 3.4473e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3854e-02, 2.2054e-06, 2.0862e-06,  ..., 9.4128e-04,
           1.0910e-03, 9.4336e-01]],

         [[7.2607e-01, 8.1778e-05, 1.1313e-04,  ..., 2.5879e-02,
           5.2734e-02, 7.0801e-03]],

         [[4.5679e-01, 1.4389e-04, 3.4213e-05,  ..., 6.5269e-03,
           5.3072e-04, 2.6352e-02]],

         ...,

         [[4.0359e-03, 3.5763e-07, 1.3709e-06,  ..., 2.9111e-04,
           5.6696e-04, 9.8975e-01]],

         [[4.4019e-01, 2.9027e-05, 4.1544e-05,  ..., 1.1131e-02,
           5.6877e-03, 3.1763e-01]],

         [[8.8574e-01, 7.3254e-05, 1.1206e-05,  ..., 7.1144e-04,
           4.7588e-04, 3.4542e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9336e-01, 3.0279e-04, 3.7241e-04,  ..., 9.8343e-03,
           3.2864e-03, 2.0325e-02]],

         [[6.9482e-01, 4.3809e-05, 1.0717e-04,  ..., 6.6040e-02,
           1.6589e-01, 1.1726e-02]],

         [[5.7520e-01, 1.8239e-05, 8.5592e-05,  ..., 4.1580e-03,
           2.2995e-04, 1.5266e-02]],

         ...,

         [[4.8584e-01, 3.7014e-05, 2.0683e-05,  ..., 7.6843e-02,
           4.1138e-01, 1.9226e-02]],

         [[4.7192e-01, 4.9734e-04, 7.6723e-04,  ..., 1.0597e-02,
           3.0518e-03, 2.7776e-04]],

         [[8.6230e-01, 1.5557e-05, 1.1086e-05,  ..., 2.7725e-02,
           6.5186e-02, 3.9825e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.0703e-01, 1.4603e-05, 1.3411e-05,  ..., 2.4948e-02,
           4.0359e-03, 4.5441e-02]],

         [[1.0175e-01, 6.0272e-04, 3.2902e-04,  ..., 1.2108e-02,
           1.4191e-02, 5.5027e-04]],

         [[3.5620e-01, 1.1187e-03, 9.0122e-04,  ..., 2.6840e-02,
           5.1025e-02, 5.8022e-03]],

         ...,

         [[4.8682e-01, 1.0461e-04, 5.6207e-05,  ..., 1.4539e-01,
           1.6541e-01, 8.1863e-03]],

         [[7.5244e-01, 7.5698e-05, 6.0141e-05,  ..., 6.8542e-02,
           7.2937e-02, 6.2622e-02]],

         [[5.6689e-01, 2.4796e-04, 9.8646e-05,  ..., 6.3477e-02,
           4.8004e-02, 8.1940e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7811e-02, 1.6093e-06, 1.6093e-06,  ..., 4.3392e-04,
           1.1129e-03, 9.5654e-01]],

         [[1.8103e-01, 8.0466e-06, 7.3314e-06,  ..., 2.1582e-01,
           2.8052e-01, 5.9814e-02]],

         [[1.7297e-01, 4.2176e-04, 4.7541e-04,  ..., 1.9913e-02,
           4.4670e-03, 9.0103e-03]],

         ...,

         [[3.7573e-01, 1.9407e-04, 5.6803e-05,  ..., 2.5452e-02,
           6.2714e-03, 3.4637e-03]],

         [[6.3721e-01, 1.1742e-05, 4.5896e-06,  ..., 1.2428e-02,
           1.6479e-03, 1.1032e-02]],

         [[7.2754e-01, 4.4942e-05, 6.8545e-05,  ..., 3.0251e-03,
           2.1541e-04, 2.5272e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8145e-01, 5.2571e-05, 5.4300e-05,  ..., 3.8147e-02,
           1.5602e-02, 4.8035e-02]],

         [[2.2327e-01, 6.9618e-05, 3.1114e-05,  ..., 1.6754e-02,
           3.1605e-03, 6.7017e-02]],

         [[5.3223e-01, 9.9778e-05, 9.6440e-05,  ..., 1.6983e-02,
           7.9269e-03, 1.9318e-02]],

         ...,

         [[6.8652e-01, 2.3782e-05, 1.4842e-05,  ..., 2.5299e-02,
           1.6144e-02, 1.7807e-02]],

         [[7.4512e-01, 5.7936e-05, 5.9426e-05,  ..., 3.5553e-02,
           7.2632e-02, 1.7670e-02]],

         [[3.9233e-01, 4.9472e-06, 3.8147e-06,  ..., 1.2810e-02,
           6.6719e-03, 1.3016e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.7441e-01, 1.0319e-03, 7.7677e-04,  ..., 1.9569e-03,
           8.2970e-04, 3.4294e-03]],

         [[1.7273e-01, 4.6194e-05, 4.1604e-05,  ..., 5.5962e-03,
           9.6560e-04, 2.4078e-02]],

         [[3.0591e-01, 1.3173e-05, 1.2696e-05,  ..., 1.6309e-01,
           2.1619e-01, 4.8218e-02]],

         ...,

         [[5.0830e-01, 1.5891e-04, 4.5598e-05,  ..., 1.8110e-03,
           1.4458e-03, 4.6539e-02]],

         [[5.3418e-01, 2.5094e-05, 3.1352e-05,  ..., 2.0266e-06,
           8.2254e-06, 2.3880e-03]],

         [[6.0010e-01, 8.4758e-05, 1.2398e-04,  ..., 5.3368e-03,
           5.0621e-03, 5.3644e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2781e-01, 4.8578e-05, 2.8551e-05,  ..., 1.9913e-02,
           4.4891e-02, 6.8817e-03]],

         [[1.6992e-01, 2.8825e-04, 1.1247e-04,  ..., 1.3832e-02,
           2.1801e-03, 1.5930e-02]],

         [[2.9199e-01, 1.6630e-05, 7.7486e-06,  ..., 1.1798e-01,
           9.5520e-02, 1.7432e-01]],

         ...,

         [[3.7012e-01, 6.5565e-06, 3.8743e-06,  ..., 2.5787e-02,
           2.1210e-02, 2.0630e-02]],

         [[6.8896e-01, 1.1802e-05, 5.9009e-06,  ..., 4.1687e-02,
           8.8806e-03, 1.1157e-01]],

         [[2.7344e-01, 2.0623e-05, 1.6928e-05,  ..., 1.3123e-01,
           7.1899e-02, 3.9478e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1465e-01, 1.7643e-05, 6.6757e-06,  ..., 3.9291e-03,
           4.0512e-03, 4.0283e-02]],

         [[1.5918e-01, 6.6757e-06, 1.6510e-05,  ..., 2.3157e-01,
           8.5205e-02, 2.9434e-02]],

         [[1.4441e-01, 8.1658e-06, 1.1325e-06,  ..., 4.3602e-03,
           7.4158e-03, 8.0518e-01]],

         ...,

         [[1.7188e-01, 2.0444e-04, 1.4102e-04,  ..., 1.4102e-04,
           2.2805e-04, 2.0859e-02]],

         [[3.7476e-01, 1.5461e-04, 7.5102e-05,  ..., 2.6169e-03,
           2.2545e-03, 2.7802e-02]],

         [[2.0825e-01, 1.3900e-04, 8.1718e-05,  ..., 1.8921e-02,
           3.8147e-03, 6.0959e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.2324e-01, 1.9431e-05, 1.1206e-05,  ..., 2.0676e-02,
           6.1569e-03, 6.6223e-02]],

         [[7.7930e-01, 2.1505e-04, 3.3617e-05,  ..., 4.3511e-06,
           1.0729e-05, 7.5459e-05]],

         [[8.7402e-01, 1.9145e-04, 5.6803e-05,  ..., 8.1718e-05,
           1.9598e-04, 8.3494e-04]],

         ...,

         [[1.4673e-01, 5.8603e-04, 1.1206e-03,  ..., 4.1656e-03,
           5.4169e-04, 5.1003e-03]],

         [[6.5332e-01, 1.0312e-05, 1.5974e-05,  ..., 3.5736e-02,
           2.1133e-03, 2.1011e-02]],

         [[3.6401e-01, 5.3453e-04, 1.1768e-03,  ..., 5.0697e-03,
           6.8245e-03, 1.3680e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.8428e-01, 7.5161e-05, 8.7917e-05,  ..., 1.0239e-02,
           1.2062e-02, 1.1787e-02]],

         [[2.6270e-01, 2.2328e-04, 2.9802e-04,  ..., 3.1300e-03,
           9.4700e-04, 3.5004e-02]],

         [[3.5034e-01, 4.4584e-05, 6.3896e-05,  ..., 6.2037e-04,
           2.8849e-04, 5.2757e-03]],

         ...,

         [[5.4871e-02, 4.9293e-05, 7.6294e-05,  ..., 3.0956e-03,
           4.1656e-03, 5.4871e-02]],

         [[2.4475e-02, 2.3007e-05, 2.7776e-05,  ..., 4.1461e-04,
           3.5453e-04, 6.5491e-02]],

         [[4.9243e-01, 7.6830e-05, 8.5711e-05,  ..., 3.9406e-03,
           1.2016e-03, 1.0063e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0386e-01, 3.1531e-05, 2.8253e-05,  ..., 2.9297e-03,
           7.7629e-04, 4.2725e-02]],

         [[3.9746e-01, 3.1233e-04, 3.6526e-04,  ..., 4.7722e-03,
           2.6779e-03, 9.0561e-03]],

         [[1.4233e-01, 6.8963e-05, 1.6737e-04,  ..., 1.1963e-02,
           3.9520e-02, 5.9912e-01]],

         ...,

         [[5.4883e-01, 9.6262e-05, 1.8704e-04,  ..., 1.6968e-02,
           8.5297e-03, 7.3730e-02]],

         [[1.3635e-01, 7.6294e-06, 4.7565e-05,  ..., 3.8314e-04,
           2.3234e-04, 6.5430e-02]],

         [[3.6108e-01, 1.9205e-04, 3.4642e-04,  ..., 5.7364e-04,
           4.6086e-04, 9.3307e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.6626e-01, 1.1969e-04, 6.9976e-05,  ..., 2.1301e-02,
           1.0468e-02, 1.2164e-01]],

         [[9.9426e-02, 6.3062e-05, 1.3196e-04,  ..., 1.7725e-01,
           1.6113e-02, 1.2352e-02]],

         [[5.2441e-01, 4.0591e-05, 1.1533e-04,  ..., 6.4964e-03,
           8.1177e-03, 1.1230e-02]],

         ...,

         [[6.5576e-01, 1.8775e-05, 2.8729e-05,  ..., 4.8248e-02,
           2.3010e-01, 5.1361e-02]],

         [[5.5225e-01, 1.1683e-05, 2.9325e-05,  ..., 3.5858e-02,
           2.9266e-02, 2.0432e-02]],

         [[2.9907e-01, 8.1658e-06, 1.2815e-05,  ..., 1.1896e-01,
           4.7778e-01, 8.8379e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4250e-02, 1.8477e-06, 1.2398e-05,  ..., 1.0132e-01,
           3.2135e-02, 4.9377e-02]],

         [[4.2480e-02, 1.6689e-06, 5.8413e-06,  ..., 1.6336e-03,
           4.3373e-03, 9.2236e-01]],

         [[4.8523e-02, 2.1100e-05, 6.7234e-05,  ..., 1.2756e-02,
           1.1261e-02, 1.8555e-02]],

         ...,

         [[1.6711e-01, 5.3883e-05, 1.4007e-04,  ..., 3.5583e-02,
           2.3514e-02, 1.8234e-02]],

         [[5.6250e-01, 4.0531e-06, 8.1062e-06,  ..., 5.4626e-02,
           3.3783e-02, 8.3618e-02]],

         [[1.4429e-01, 1.4067e-05, 1.2994e-05,  ..., 1.0315e-01,
           6.2939e-01, 7.6965e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.6016e-01, 1.0377e-04, 5.1689e-04,  ..., 3.7746e-03,
           4.5013e-03, 1.4473e-02]],

         [[9.6436e-01, 3.6597e-05, 7.0393e-05,  ..., 2.5570e-05,
           5.3942e-05, 6.5899e-04]],

         [[8.8867e-01, 2.6822e-05, 6.7949e-06,  ..., 6.1274e-05,
           2.2829e-05, 3.8719e-04]],

         ...,

         [[7.7832e-01, 6.9737e-06, 9.8944e-06,  ..., 1.3809e-02,
           9.0561e-03, 3.5828e-02]],

         [[6.0352e-01, 1.8477e-06, 5.9605e-07,  ..., 4.8566e-04,
           2.7251e-04, 6.1989e-03]],

         [[1.0931e-01, 6.4969e-06, 9.2983e-06,  ..., 2.8896e-03,
           1.0490e-02, 7.5879e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.3497e-02, 3.2783e-06, 1.0133e-05,  ..., 2.1313e-01,
           1.2830e-01, 9.6069e-02]],

         [[5.6494e-01, 1.5318e-05, 1.0371e-05,  ..., 1.9867e-02,
           1.9958e-02, 2.2217e-01]],

         [[1.2195e-01, 1.5354e-04, 1.3220e-04,  ..., 4.0863e-02,
           7.9803e-03, 7.1716e-02]],

         ...,

         [[5.0146e-01, 1.1981e-05, 5.6684e-05,  ..., 4.2816e-02,
           2.2202e-02, 6.7871e-02]],

         [[5.6787e-01, 1.6868e-05, 3.5167e-05,  ..., 2.8503e-02,
           1.3046e-02, 2.4185e-02]],

         [[6.3135e-01, 6.2585e-06, 1.6034e-05,  ..., 3.7823e-03,
           5.2528e-03, 4.8676e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9440e-02, 3.1590e-06, 1.8477e-06,  ..., 4.1168e-02,
           5.8403e-03, 2.0218e-02]],

         [[1.5381e-01, 5.9187e-05, 2.5690e-05,  ..., 1.0166e-03,
           7.2384e-04, 7.3090e-03]],

         [[2.8198e-01, 6.8009e-05, 1.6022e-04,  ..., 6.5804e-03,
           3.4809e-03, 5.4657e-02]],

         ...,

         [[5.6305e-02, 5.4240e-06, 7.0691e-05,  ..., 1.1894e-02,
           3.1586e-02, 3.4668e-02]],

         [[1.1975e-01, 7.9274e-06, 5.5790e-05,  ..., 9.8267e-03,
           1.7807e-02, 9.8267e-03]],

         [[7.7246e-01, 2.5272e-05, 4.8339e-05,  ..., 5.1758e-02,
           8.5327e-02, 4.0314e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.0889e-01, 2.1458e-06, 7.2718e-06,  ..., 2.8366e-02,
           8.4045e-02, 2.1619e-01]],

         [[3.7158e-01, 2.7835e-05, 8.5831e-05,  ..., 9.3933e-02,
           1.2396e-01, 1.5369e-01]],

         [[3.4253e-01, 6.6757e-06, 4.8757e-05,  ..., 6.7139e-02,
           4.8291e-01, 7.9468e-02]],

         ...,

         [[4.2358e-01, 3.2961e-05, 3.1471e-05,  ..., 1.3191e-02,
           3.6507e-03, 1.1284e-02]],

         [[1.8604e-01, 5.6028e-06, 1.0252e-05,  ..., 1.8176e-01,
           9.2834e-02, 4.7791e-02]],

         [[6.5979e-02, 2.7359e-05, 8.1599e-05,  ..., 3.3112e-03,
           1.3298e-02, 6.2109e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9121e-01, 1.6868e-05, 1.0830e-04,  ..., 8.0338e-03,
           4.7798e-03, 1.3855e-01]],

         [[2.0337e-01, 2.6047e-05, 8.2672e-05,  ..., 3.4857e-04,
           3.1114e-04, 4.0741e-03]],

         [[2.5195e-01, 6.4254e-05, 2.7275e-04,  ..., 7.6199e-04,
           1.1568e-03, 5.1178e-02]],

         ...,

         [[4.7217e-01, 3.4857e-04, 5.3596e-04,  ..., 1.9180e-02,
           4.1901e-02, 2.8198e-01]],

         [[5.0879e-01, 1.5306e-04, 5.9605e-04,  ..., 3.1052e-02,
           7.3303e-02, 2.4426e-01]],

         [[5.7861e-01, 3.2723e-05, 4.5061e-05,  ..., 2.9888e-03,
           1.5671e-02, 3.6206e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.5488e-01, 3.7730e-05, 2.5058e-04,  ..., 5.9357e-02,
           5.4230e-02, 3.3539e-02]],

         [[1.4087e-01, 4.1723e-06, 5.5909e-05,  ..., 2.7618e-02,
           8.4106e-02, 1.0968e-01]],

         [[3.3984e-01, 1.3804e-04, 6.3848e-04,  ..., 3.6682e-02,
           5.2948e-02, 9.2590e-02]],

         ...,

         [[5.3467e-01, 4.1604e-05, 1.5628e-04,  ..., 2.5665e-02,
           6.1676e-02, 9.7733e-03]],

         [[9.0967e-01, 1.0133e-06, 4.6182e-04,  ..., 2.7275e-04,
           1.6356e-04, 3.0804e-03]],

         [[6.4258e-01, 3.2783e-06, 9.7752e-06,  ..., 2.1561e-02,
           1.7120e-02, 2.5513e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7193e-05, 1.7762e-05, 1.7249e-01,  ..., 5.4779e-03,
           1.3405e-02, 2.2186e-02]],

         [[2.3782e-05, 1.6510e-05, 1.4539e-01,  ..., 1.0162e-02,
           1.3466e-02, 1.2070e-02]],

         [[4.0174e-05, 4.9591e-05, 9.8633e-02,  ..., 6.0387e-03,
           1.6739e-02, 3.6560e-02]],

         ...,

         [[6.5506e-05, 7.8082e-05, 2.8174e-01,  ..., 9.2773e-03,
           9.2224e-02, 1.3843e-01]],

         [[2.2209e-04, 1.1826e-04, 2.7710e-01,  ..., 2.6016e-03,
           4.7112e-03, 2.9770e-02]],

         [[4.4465e-05, 1.6093e-05, 3.1299e-01,  ..., 5.9938e-04,
           1.1110e-03, 5.1003e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.2493e-04, 6.2108e-05, 3.2227e-01,  ..., 4.8661e-04,
           7.5626e-04, 4.2343e-03]],

         [[5.3465e-05, 4.3213e-05, 2.0825e-01,  ..., 1.8704e-04,
           9.3126e-04, 2.4529e-03]],

         [[1.2398e-05, 4.2319e-06, 2.7002e-01,  ..., 1.0931e-04,
           8.9741e-04, 5.0163e-04]],

         ...,

         [[1.1438e-04, 1.0496e-04, 2.7075e-01,  ..., 5.3835e-04,
           1.8101e-03, 2.9259e-03]],

         [[1.1921e-06, 1.7881e-06, 2.3254e-01,  ..., 1.4353e-04,
           1.0118e-03, 4.8256e-03]],

         [[1.0276e-04, 1.4544e-04, 1.8872e-01,  ..., 1.5163e-03,
           1.2642e-02, 2.4891e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0014e-05, 6.8545e-06, 1.7029e-01,  ..., 9.8288e-05,
           4.3678e-04, 2.1896e-02]],

         [[3.0220e-05, 1.6093e-06, 2.4927e-01,  ..., 1.2803e-04,
           9.4223e-04, 3.7909e-04]],

         [[1.7643e-05, 1.9372e-05, 2.4829e-01,  ..., 1.5986e-04,
           9.4223e-04, 8.2827e-04]],

         ...,

         [[1.5497e-05, 1.6212e-05, 1.2939e-01,  ..., 1.9646e-03,
           7.1869e-03, 3.1787e-01]],

         [[1.0014e-05, 2.1398e-05, 2.2314e-01,  ..., 2.2221e-04,
           1.1349e-03, 2.7657e-03]],

         [[6.3062e-05, 2.6524e-05, 1.6211e-01,  ..., 2.8229e-04,
           1.1272e-03, 1.3306e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1842e-05, 8.7619e-06, 1.2787e-02,  ..., 1.4636e-01,
           3.8867e-01, 1.8652e-01]],

         [[2.0564e-05, 2.3007e-05, 8.1421e-02,  ..., 4.9194e-02,
           3.0914e-02, 1.5099e-02]],

         [[1.4102e-04, 3.8314e-04, 3.0127e-01,  ..., 2.7771e-03,
           1.0662e-03, 1.4420e-03]],

         ...,

         [[5.1308e-04, 2.2292e-05, 1.3660e-01,  ..., 2.4166e-03,
           3.0041e-03, 2.0294e-02]],

         [[7.6532e-04, 1.3793e-04, 4.5837e-02,  ..., 9.9731e-02,
           2.8979e-01, 2.7856e-01]],

         [[9.4950e-05, 2.0802e-05, 1.5839e-02,  ..., 3.2745e-02,
           3.9948e-02, 4.0674e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.9083e-03, 1.2045e-03, 2.3621e-01,  ..., 2.5730e-03,
           5.7220e-03, 6.3599e-02]],

         [[1.5986e-04, 2.4462e-04, 2.3853e-01,  ..., 1.8829e-02,
           2.9388e-02, 4.3091e-02]],

         [[1.5335e-03, 4.1366e-04, 2.4353e-01,  ..., 1.5327e-02,
           5.7831e-02, 6.2042e-02]],

         ...,

         [[1.6689e-04, 4.6670e-05, 2.3132e-01,  ..., 1.5495e-02,
           4.2450e-02, 5.5359e-02]],

         [[8.1921e-04, 7.5150e-04, 1.2549e-01,  ..., 9.7504e-03,
           3.8269e-02, 3.2275e-01]],

         [[1.2934e-05, 2.9266e-05, 8.3252e-02,  ..., 7.1777e-02,
           1.9055e-01, 1.4722e-01]]]], device='cuda:0', dtype=torch.float16)),(tensor([[[[1.3828e-05, 1.9670e-04, 1.9255e-03,  ..., 3.0121e-02,
           6.8555e-01, 1.0510e-01]],

         [[4.6074e-05, 3.1650e-05, 1.8227e-04,  ..., 3.4389e-03,
           1.9791e-02, 9.5361e-01]],

         [[0.0000e+00, 0.0000e+00, 1.3709e-06,  ..., 1.6689e-05,
           1.5011e-03, 9.9805e-01]],

         ...,

         [[5.9605e-06, 3.6359e-06, 4.7684e-07,  ..., 5.3835e-04,
           6.5613e-03, 9.7363e-01]],

         [[3.9635e-03, 2.2583e-03, 3.2845e-03,  ..., 5.8228e-02,
           1.1467e-02, 2.2812e-02]],

         [[3.3455e-03, 2.0294e-03, 2.6631e-04,  ..., 4.3396e-02,
           1.1792e-01, 6.5125e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.8036e-05, 8.3542e-04, 2.0676e-03,  ..., 8.2581e-02,
           2.9736e-01, 2.0447e-01]],

         [[0.0000e+00, 1.7881e-07, 8.9407e-06,  ..., 1.2164e-01,
           7.8125e-01, 7.7332e-02]],

         [[1.2903e-03, 4.2305e-03, 1.1871e-02,  ..., 2.1500e-02,
           3.8910e-02, 1.2634e-02]],

         ...,

         [[1.7996e-03, 5.0850e-03, 4.0245e-03,  ..., 4.2017e-01,
           1.1487e-01, 1.3397e-02]],

         [[2.5368e-03, 1.1505e-02, 1.9806e-02,  ..., 1.9485e-02,
           1.7334e-02, 7.7820e-03]],

         [[3.4313e-03, 7.2632e-03, 6.9847e-03,  ..., 2.1069e-01,
           7.2205e-02, 5.8960e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3732e-02, 4.5547e-03, 2.3773e-02,  ..., 1.3763e-02,
           1.9867e-02, 1.3924e-02]],

         [[1.2712e-03, 4.5323e-04, 2.2850e-03,  ..., 6.0654e-03,
           3.0090e-02, 1.1421e-02]],

         [[8.6823e-03, 1.5812e-03, 2.4776e-03,  ..., 1.7197e-02,
           2.0981e-02, 1.1772e-02]],

         ...,

         [[2.0638e-03, 1.6203e-03, 8.1024e-03,  ..., 3.7750e-02,
           1.1627e-01, 5.5469e-01]],

         [[1.6632e-03, 4.2367e-04, 1.0824e-03,  ..., 3.0884e-02,
           1.0535e-01, 4.1901e-02]],

         [[8.9121e-04, 4.2748e-04, 5.3329e-03,  ..., 5.3864e-02,
           5.7910e-01, 1.5466e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.4863e-01, 2.3842e-06, 2.5630e-06,  ..., 1.5545e-03,
           8.8043e-03, 2.3682e-02]],

         [[9.5215e-01, 1.0788e-05, 3.0994e-06,  ..., 4.7760e-03,
           1.7395e-02, 1.5732e-02]],

         [[9.1357e-01, 4.7684e-06, 1.6689e-06,  ..., 2.2736e-03,
           1.9012e-02, 1.8524e-02]],

         ...,

         [[6.4355e-01, 1.1837e-04, 9.6560e-04,  ..., 1.1384e-05,
           9.4299e-03, 2.2278e-02]],

         [[5.5127e-01, 1.1843e-04, 3.6526e-04,  ..., 1.6510e-02,
           5.0262e-02, 1.3741e-02]],

         [[4.8267e-01, 7.4983e-05, 2.4331e-04,  ..., 2.7130e-02,
           8.0322e-02, 9.8267e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.1797e-01, 2.1994e-05, 1.5497e-05,  ..., 1.8358e-04,
           2.5043e-03, 1.6205e-02]],

         [[3.0396e-01, 6.5565e-07, 5.3704e-05,  ..., 6.5327e-04,
           6.8604e-01, 7.7515e-03]],

         [[9.6045e-01, 2.4676e-05, 4.3452e-05,  ..., 3.0098e-03,
           7.5684e-03, 1.7319e-03]],

         ...,

         [[3.6060e-01, 2.2662e-04, 2.1482e-04,  ..., 1.3904e-01,
           5.1758e-02, 2.9480e-02]],

         [[2.4817e-01, 8.8120e-04, 4.2653e-04,  ..., 5.2299e-03,
           8.7128e-03, 6.1150e-03]],

         [[6.3965e-01, 6.6280e-04, 8.9741e-04,  ..., 8.4000e-03,
           2.4951e-01, 3.7903e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.8628e-01, 1.3542e-04, 4.2915e-05,  ..., 4.4861e-02,
           6.2622e-02, 3.9948e-02]],

         [[7.1240e-01, 4.6134e-05, 1.8358e-05,  ..., 2.1255e-02,
           1.9531e-01, 1.4816e-02]],

         [[7.9443e-01, 3.5703e-05, 2.3067e-05,  ..., 3.8635e-02,
           4.8462e-02, 1.1490e-02]],

         ...,

         [[9.5264e-01, 6.1572e-05, 6.5565e-07,  ..., 4.1723e-07,
           2.0111e-04, 3.8055e-02]],

         [[9.5312e-01, 2.1458e-04, 2.5630e-06,  ..., 0.0000e+00,
           1.1325e-06, 9.2773e-03]],

         [[9.8682e-01, 4.7028e-05, 1.1921e-07,  ..., 0.0000e+00,
           5.9605e-08, 7.8964e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3743e-01, 2.3246e-05, 4.5896e-06,  ..., 7.0068e-02,
           4.3994e-01, 2.9617e-02]],

         [[8.0664e-01, 2.3842e-06, 3.5763e-07,  ..., 1.2267e-04,
           2.3788e-02, 4.7211e-02]],

         [[8.4424e-01, 3.4511e-05, 6.7949e-06,  ..., 6.6109e-03,
           5.9509e-03, 8.2474e-03]],

         ...,

         [[7.8174e-01, 3.0780e-04, 1.7345e-05,  ..., 4.2152e-03,
           1.4999e-02, 1.2566e-02]],

         [[5.3906e-01, 3.3438e-05, 1.2159e-05,  ..., 1.3702e-02,
           2.2766e-02, 3.3081e-01]],

         [[3.0762e-01, 9.0420e-05, 6.5625e-05,  ..., 4.3335e-03,
           6.1670e-01, 4.0588e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.1152e-01, 3.3081e-05, 1.4484e-05,  ..., 3.0327e-03,
           1.4313e-02, 1.6113e-02]],

         [[7.8857e-01, 3.7253e-05, 1.8239e-05,  ..., 3.3379e-03,
           9.2392e-03, 6.3721e-02]],

         [[7.4805e-01, 4.0531e-06, 3.9935e-06,  ..., 1.2032e-02,
           1.5137e-01, 2.6031e-02]],

         ...,

         [[5.8447e-01, 9.3758e-05, 1.8954e-05,  ..., 8.0261e-03,
           5.8868e-02, 1.6998e-02]],

         [[8.5010e-01, 9.3162e-05, 3.1948e-05,  ..., 2.2583e-03,
           1.1856e-02, 1.3695e-02]],

         [[8.2666e-01, 1.8179e-05, 1.7762e-05,  ..., 2.6260e-02,
           7.6111e-02, 7.2021e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.9971e-01, 8.9073e-04, 4.8423e-04,  ..., 5.0163e-04,
           2.7428e-03, 1.9226e-03]],

         [[9.3652e-01, 8.4639e-06, 4.9472e-05,  ..., 1.2178e-03,
           2.7657e-03, 1.1578e-03]],

         [[7.4805e-01, 3.9876e-05, 1.1504e-05,  ..., 3.4943e-03,
           5.5199e-03, 9.3002e-03]],

         ...,

         [[8.1738e-01, 1.4901e-05, 2.6822e-05,  ..., 2.9507e-03,
           1.1589e-02, 9.7656e-02]],

         [[3.1812e-01, 1.4067e-05, 7.8082e-06,  ..., 3.7903e-02,
           5.2637e-01, 4.8859e-02]],

         [[9.0625e-01, 5.0247e-05, 1.9193e-05,  ..., 9.9468e-04,
           8.1406e-03, 2.5986e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0642e-01, 4.6039e-04, 1.9045e-03,  ..., 8.6746e-03,
           1.5686e-02, 1.3145e-02]],

         [[7.0801e-01, 1.6868e-05, 4.3452e-05,  ..., 3.9612e-02,
           1.1304e-01, 3.0899e-02]],

         [[7.3242e-01, 1.5795e-05, 2.9743e-05,  ..., 9.7961e-03,
           1.9821e-02, 1.3573e-02]],

         ...,

         [[8.5400e-01, 3.5465e-05, 8.8394e-05,  ..., 1.4410e-03,
           1.6618e-04, 5.4061e-05]],

         [[6.3232e-01, 1.1241e-04, 7.0095e-05,  ..., 1.1589e-02,
           7.5500e-02, 2.3163e-02]],

         [[9.5703e-01, 8.4043e-06, 1.9133e-05,  ..., 2.6627e-03,
           6.2943e-03, 9.2621e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[5.1172e-01, 2.0599e-04, 1.7345e-04,  ..., 1.1921e-06,
           2.3575e-03, 1.1337e-02]],

         [[7.3779e-01, 4.1127e-06, 1.7285e-06,  ..., 5.2216e-02,
           7.1838e-02, 7.3624e-03]],

         [[7.3975e-01, 5.9009e-06, 5.7817e-06,  ..., 3.6621e-04,
           5.4047e-02, 9.6893e-03]],

         ...,

         [[9.1260e-01, 1.0157e-04, 7.3731e-05,  ..., 3.0155e-03,
           8.3313e-03, 2.5539e-03]],

         [[8.5400e-01, 6.9797e-05, 3.9160e-05,  ..., 9.0332e-03,
           1.0443e-03, 8.2254e-04]],

         [[5.6787e-01, 9.8801e-04, 3.6430e-04,  ..., 1.8034e-03,
           1.5518e-02, 6.5498e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.5295e-01, 4.7684e-07, 1.3113e-06,  ..., 1.7226e-04,
           3.3875e-02, 8.0469e-01]],

         [[7.9932e-01, 3.0565e-04, 5.6601e-04,  ..., 9.6262e-05,
           1.8215e-03, 1.3895e-03]],

         [[6.0693e-01, 5.9223e-04, 1.0653e-03,  ..., 4.8943e-03,
           1.7319e-02, 8.2092e-03]],

         ...,

         [[6.6260e-01, 2.3592e-04, 1.9407e-04,  ..., 1.6003e-03,
           2.3651e-03, 5.0163e-04]],

         [[8.0762e-01, 7.8619e-05, 6.5446e-05,  ..., 3.5172e-03,
           1.8799e-02, 5.6305e-03]],

         [[3.0518e-01, 2.3603e-05, 1.1224e-04,  ..., 3.2234e-03,
           1.0388e-01, 5.1709e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5098e-01, 9.8348e-06, 9.5963e-06,  ..., 1.5144e-03,
           3.4399e-01, 3.7207e-01]],

         [[7.5732e-01, 5.1320e-05, 6.3598e-05,  ..., 3.9032e-02,
           2.9861e-02, 1.3794e-02]],

         [[6.1182e-01, 2.4319e-04, 4.9233e-05,  ..., 7.1192e-04,
           4.0741e-02, 2.1347e-02]],

         ...,

         [[9.9106e-03, 7.1526e-07, 1.9073e-06,  ..., 2.5630e-05,
           3.8605e-02, 9.4629e-01]],

         [[5.2588e-01, 3.2365e-05, 4.5300e-05,  ..., 1.0786e-03,
           6.5002e-02, 1.6174e-01]],

         [[9.1748e-01, 5.3704e-05, 1.1027e-05,  ..., 5.4121e-04,
           3.5992e-03, 2.3651e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.8359e-01, 3.1281e-04, 3.0804e-04,  ..., 4.0321e-03,
           1.9547e-02, 1.7914e-02]],

         [[6.8213e-01, 7.6354e-05, 8.1897e-05,  ..., 1.7847e-01,
           1.3954e-02, 4.8256e-03]],

         [[6.2158e-01, 4.2379e-05, 1.3161e-04,  ..., 3.3998e-04,
           6.0234e-03, 6.3629e-03]],

         ...,

         [[7.6416e-01, 1.0550e-05, 3.7968e-05,  ..., 1.1011e-01,
           1.0815e-01, 6.7558e-03]],

         [[5.8447e-01, 1.4627e-04, 2.3019e-04,  ..., 2.5826e-03,
           3.3498e-04, 2.5487e-04]],

         [[5.0537e-01, 8.4639e-06, 4.2975e-05,  ..., 3.4851e-02,
           3.9600e-01, 5.7220e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[8.2520e-01, 1.1086e-05, 7.7486e-06,  ..., 1.5345e-03,
           2.7908e-02, 1.9943e-02]],

         [[6.3477e-02, 7.1907e-04, 3.4237e-04,  ..., 1.3947e-02,
           7.3433e-04, 2.6727e-04]],

         [[4.2432e-01, 6.1083e-04, 4.5133e-04,  ..., 5.7312e-02,
           1.0025e-02, 3.9043e-03]],

         ...,

         [[6.7285e-01, 1.0216e-04, 6.8009e-05,  ..., 1.1926e-01,
           2.1500e-02, 5.1765e-03]],

         [[4.2822e-01, 8.4162e-05, 1.2201e-04,  ..., 2.2980e-02,
           5.0879e-01, 1.2581e-02]],

         [[5.7178e-01, 1.6212e-04, 1.3185e-04,  ..., 6.1554e-02,
           1.7624e-02, 1.6891e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.8828e-02, 1.6689e-06, 2.3842e-06,  ..., 7.7963e-05,
           1.3718e-02, 9.3213e-01]],

         [[1.3586e-01, 1.4842e-05, 1.0133e-05,  ..., 2.7002e-01,
           2.6587e-01, 1.9714e-02]],

         [[2.3730e-01, 5.5838e-04, 5.8317e-04,  ..., 7.8583e-03,
           2.0538e-02, 9.4604e-03]],

         ...,

         [[3.8257e-01, 1.3769e-04, 5.2869e-05,  ..., 8.6365e-03,
           9.6512e-03, 2.7275e-03]],

         [[7.5000e-01, 3.0577e-05, 1.5080e-05,  ..., 3.0060e-03,
           1.8997e-02, 4.7302e-03]],

         [[7.4023e-01, 7.4804e-05, 6.7890e-05,  ..., 3.4475e-04,
           1.4061e-02, 4.4098e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.9341e-01, 2.3961e-05, 2.4021e-05,  ..., 9.5825e-03,
           1.9702e-01, 8.4412e-02]],

         [[2.5269e-01, 1.0628e-04, 4.6074e-05,  ..., 4.2381e-03,
           9.3994e-02, 3.7262e-02]],

         [[6.0156e-01, 3.5524e-05, 4.9472e-05,  ..., 8.6823e-03,
           9.3323e-02, 2.3407e-02]],

         ...,

         [[7.5635e-01, 1.6093e-05, 1.2755e-05,  ..., 1.1711e-02,
           3.8391e-02, 1.7578e-02]],

         [[8.1055e-01, 2.7597e-05, 2.2411e-05,  ..., 4.8859e-02,
           2.3911e-02, 1.5434e-02]],

         [[4.4116e-01, 2.8610e-06, 2.9206e-06,  ..., 6.8054e-03,
           1.3641e-02, 8.1787e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.4341e-01, 7.2289e-04, 9.4128e-04,  ..., 1.7414e-03,
           8.0261e-03, 4.0550e-03]],

         [[1.1395e-01, 5.4002e-05, 4.3571e-05,  ..., 2.4338e-03,
           4.3945e-02, 1.2054e-02]],

         [[3.4717e-01, 1.2577e-05, 9.9540e-06,  ..., 1.5161e-01,
           2.7686e-01, 2.6978e-02]],

         ...,

         [[5.8447e-01, 2.2376e-04, 9.9242e-05,  ..., 7.2956e-04,
           2.5375e-02, 1.1551e-02]],

         [[9.2725e-01, 4.2617e-05, 4.6074e-05,  ..., 2.3842e-07,
           3.2425e-05, 3.4103e-03]],

         [[6.2793e-01, 1.3101e-04, 1.5771e-04,  ..., 4.2610e-03,
           4.0746e-04, 1.8585e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[9.6924e-02, 4.6551e-05, 4.4107e-05,  ..., 8.1604e-02,
           5.5084e-03, 2.5425e-03]],

         [[1.8408e-01, 1.2398e-04, 5.9366e-05,  ..., 5.1193e-03,
           2.1469e-02, 7.3051e-03]],

         [[1.8787e-01, 2.3842e-06, 1.0729e-06,  ..., 3.2135e-02,
           4.3018e-01, 1.8494e-01]],

         ...,

         [[5.1562e-01, 7.3910e-06, 4.3511e-06,  ..., 2.3361e-02,
           5.8289e-02, 1.4061e-02]],

         [[2.7344e-01, 1.5497e-06, 1.6093e-06,  ..., 3.5248e-03,
           6.2598e-01, 5.0598e-02]],

         [[1.5588e-01, 2.5034e-06, 3.2783e-06,  ..., 2.3712e-02,
           6.1621e-01, 1.2622e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.1963e-01, 1.5318e-05, 5.7817e-06,  ..., 5.1155e-03,
           4.1534e-02, 1.9928e-02]],

         [[1.3086e-01, 3.3975e-06, 4.3511e-06,  ..., 1.2683e-01,
           2.2266e-01, 1.4015e-02]],

         [[1.1737e-01, 7.3910e-06, 7.1526e-07,  ..., 2.3746e-04,
           2.5787e-02, 8.4033e-01]],

         ...,

         [[8.7952e-02, 5.3644e-05, 3.6836e-05,  ..., 1.0705e-04,
           2.8778e-02, 1.8005e-02]],

         [[3.7720e-01, 1.1748e-04, 4.8459e-05,  ..., 1.4420e-03,
           9.2239e-03, 2.1820e-03]],

         [[2.0581e-01, 1.1975e-04, 6.7472e-05,  ..., 6.6147e-03,
           1.0910e-02, 1.7605e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.3018e-01, 1.9372e-05, 1.0848e-05,  ..., 3.9597e-03,
           4.6051e-02, 4.5685e-02]],

         [[9.6582e-01, 1.5914e-05, 1.4305e-06,  ..., 1.1921e-07,
           1.9670e-06, 2.9335e-03]],

         [[8.2666e-01, 5.5909e-05, 6.7353e-06,  ..., 5.4598e-05,
           4.2105e-04, 8.1635e-03]],

         ...,

         [[2.1545e-01, 6.4421e-04, 1.1129e-03,  ..., 1.5574e-03,
           5.9242e-03, 3.1700e-03]],

         [[7.1094e-01, 1.0967e-05, 2.1517e-05,  ..., 1.6556e-03,
           2.5497e-02, 1.7517e-02]],

         [[2.8931e-01, 5.9700e-04, 1.6356e-03,  ..., 7.8964e-03,
           1.1391e-02, 1.5335e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.1299e-01, 1.3280e-04, 1.2577e-04,  ..., 1.0468e-02,
           9.7580e-03, 1.4839e-03]],

         [[3.8062e-01, 1.8287e-04, 3.2854e-04,  ..., 3.0937e-03,
           1.9394e-02, 1.4420e-02]],

         [[4.3628e-01, 1.0872e-04, 9.6738e-05,  ..., 2.7776e-04,
           4.2114e-03, 4.2419e-03]],

         ...,

         [[4.3060e-02, 4.3154e-05, 6.7830e-05,  ..., 9.0256e-03,
           1.4575e-01, 3.5706e-02]],

         [[5.5267e-02, 5.4479e-05, 5.3644e-05,  ..., 1.3208e-03,
           1.5991e-01, 4.1718e-02]],

         [[4.8047e-01, 5.8830e-05, 6.8784e-05,  ..., 1.9627e-03,
           2.7969e-02, 1.1124e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.0103e-01, 2.0981e-05, 2.3246e-05,  ..., 1.4839e-03,
           5.6580e-02, 2.0172e-02]],

         [[3.2324e-01, 2.2948e-04, 2.9707e-04,  ..., 3.6182e-03,
           1.8524e-02, 8.2169e-03]],

         [[3.0518e-01, 5.3525e-05, 1.1557e-04,  ..., 2.5253e-02,
           1.6467e-01, 2.8223e-01]],

         ...,

         [[5.6104e-01, 1.0937e-04, 1.6415e-04,  ..., 8.6517e-03,
           8.7402e-02, 5.8228e-02]],

         [[1.8640e-01, 3.9935e-06, 3.8862e-05,  ..., 1.9872e-04,
           4.8645e-02, 8.8074e-02]],

         [[2.5830e-01, 1.5450e-04, 2.0790e-04,  ..., 6.2084e-04,
           1.0414e-02, 5.1575e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.7004e-01, 7.6771e-05, 3.6061e-05,  ..., 1.2711e-02,
           1.7822e-01, 6.9275e-02]],

         [[1.0144e-01, 1.9073e-05, 3.3259e-05,  ..., 2.2293e-02,
           1.0632e-01, 1.5556e-02]],

         [[4.3652e-01, 5.3346e-05, 1.4877e-04,  ..., 1.5656e-02,
           1.0971e-02, 7.0000e-03]],

         ...,

         [[4.2017e-01, 2.3842e-07, 1.0431e-05,  ..., 3.8483e-02,
           5.1465e-01, 1.7349e-02]],

         [[3.1714e-01, 1.0133e-06, 8.6427e-06,  ..., 7.6538e-02,
           3.2715e-01, 1.6159e-02]],

         [[1.1072e-01, 1.7881e-07, 3.5167e-06,  ..., 3.3752e-02,
           8.3105e-01, 1.9852e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0117e-02, 9.5367e-07, 5.4836e-06,  ..., 9.3750e-02,
           9.8206e-02, 1.7075e-02]],

         [[2.0279e-02, 5.9605e-07, 1.0729e-06,  ..., 1.3185e-04,
           1.0681e-02, 9.5410e-01]],

         [[2.5558e-02, 2.1040e-05, 7.7009e-05,  ..., 1.8402e-02,
           9.5520e-03, 4.3030e-03]],

         ...,

         [[6.4575e-02, 6.4492e-05, 1.4126e-04,  ..., 4.4556e-02,
           3.1097e-02, 7.6523e-03]],

         [[1.0632e-01, 1.8477e-06, 2.8610e-06,  ..., 1.2476e-01,
           6.1182e-01, 4.5349e-02]],

         [[9.7351e-02, 4.8876e-06, 1.4603e-05,  ..., 1.6956e-01,
           6.0596e-01, 8.9722e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[6.7529e-01, 1.1361e-04, 6.2561e-04,  ..., 2.5635e-03,
           1.0216e-02, 1.5457e-02]],

         [[9.7266e-01, 6.3717e-05, 2.7299e-05,  ..., 5.9664e-05,
           4.8208e-04, 1.6117e-03]],

         [[9.0332e-01, 1.2636e-05, 4.4703e-06,  ..., 4.2975e-05,
           3.7861e-04, 8.5220e-03]],

         ...,

         [[5.4834e-01, 5.0664e-06, 6.2585e-06,  ..., 1.7090e-02,
           1.6211e-01, 1.8958e-01]],

         [[4.3311e-01, 1.7881e-06, 7.1526e-07,  ..., 3.8576e-04,
           8.5754e-03, 4.8103e-03]],

         [[1.1932e-01, 2.8014e-06, 3.8743e-06,  ..., 1.3895e-03,
           7.0129e-02, 7.4268e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.6627e-02, 4.1723e-07, 1.3709e-06,  ..., 1.5686e-01,
           3.2446e-01, 5.6366e-02]],

         [[2.3242e-01, 1.7881e-06, 1.2517e-06,  ..., 2.4979e-02,
           4.8462e-01, 2.0044e-01]],

         [[1.0431e-01, 9.2030e-05, 9.2089e-05,  ..., 2.0615e-02,
           1.5967e-01, 3.7628e-02]],

         ...,

         [[1.8652e-01, 1.4901e-06, 9.9540e-06,  ..., 7.5928e-02,
           3.3765e-01, 2.0959e-01]],

         [[5.7471e-01, 1.4484e-05, 4.3690e-05,  ..., 1.9196e-02,
           1.0773e-02, 2.9510e-02]],

         [[6.7236e-01, 5.0068e-06, 1.5736e-05,  ..., 9.5901e-03,
           2.4689e-02, 3.3997e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.3804e-02, 1.9670e-06, 1.3709e-06,  ..., 7.3738e-03,
           5.7526e-02, 3.0090e-02]],

         [[1.6809e-01, 2.0206e-05, 1.0133e-05,  ..., 2.7013e-04,
           4.8828e-03, 4.3411e-03]],

         [[2.6196e-01, 2.3484e-05, 7.7248e-05,  ..., 3.4828e-03,
           1.0669e-01, 6.6772e-02]],

         ...,

         [[3.9307e-02, 4.2915e-06, 6.4433e-05,  ..., 2.7451e-02,
           2.2049e-02, 1.0017e-02]],

         [[1.1084e-01, 1.2279e-05, 6.9439e-05,  ..., 2.3224e-02,
           9.5367e-03, 6.2523e-03]],

         [[2.7783e-01, 2.2292e-05, 3.5107e-05,  ..., 2.5708e-01,
           1.6077e-01, 6.5979e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.0654e-01, 2.3842e-07, 1.0133e-06,  ..., 4.2145e-02,
           6.1426e-01, 1.1774e-01]],

         [[2.8125e-01, 4.0531e-06, 2.8253e-05,  ..., 9.3506e-02,
           2.8027e-01, 1.0675e-01]],

         [[1.9336e-01, 5.3644e-07, 1.0014e-05,  ..., 8.2520e-02,
           6.5918e-01, 4.7211e-02]],

         ...,

         [[4.7607e-01, 3.8505e-05, 4.7386e-05,  ..., 3.9291e-03,
           4.7569e-03, 6.9504e-03]],

         [[7.1594e-02, 2.6226e-06, 3.0398e-06,  ..., 5.7983e-02,
           3.5254e-01, 9.4849e-02]],

         [[4.5715e-02, 2.1756e-05, 6.3419e-05,  ..., 3.0746e-03,
           1.5222e-01, 5.9277e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.4751e-01, 3.0220e-05, 9.2506e-05,  ..., 4.3030e-03,
           1.4868e-01, 9.6802e-02]],

         [[9.8083e-02, 9.1195e-06, 3.1531e-05,  ..., 1.0854e-04,
           2.0847e-03, 1.7281e-03]],

         [[2.1118e-01, 8.2970e-05, 1.4770e-04,  ..., 1.2217e-03,
           6.2927e-02, 1.9562e-02]],

         ...,

         [[3.3478e-02, 5.5730e-05, 5.3585e-05,  ..., 2.2659e-02,
           7.5000e-01, 1.5967e-01]],

         [[3.1250e-01, 1.1075e-04, 7.6246e-04,  ..., 2.9984e-02,
           1.0468e-01, 4.2041e-01]],

         [[4.1968e-01, 2.0742e-05, 4.1306e-05,  ..., 2.1225e-02,
           2.6685e-01, 2.5464e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[7.3535e-01, 3.2723e-05, 1.6534e-04,  ..., 3.4943e-02,
           6.7627e-02, 7.1678e-03]],

         [[6.6223e-02, 1.8477e-06, 1.6391e-05,  ..., 1.3684e-01,
           1.5454e-01, 3.2013e-02]],

         [[4.0894e-01, 1.5008e-04, 7.2241e-04,  ..., 3.7964e-02,
           6.0059e-02, 1.5854e-02]],

         ...,

         [[3.0029e-01, 6.7949e-05, 1.5736e-04,  ..., 1.1945e-01,
           6.2675e-03, 6.9389e-03]],

         [[9.0771e-01, 5.9605e-06, 4.9257e-04,  ..., 1.1718e-04,
           1.3447e-03, 1.3260e-02]],

         [[5.1270e-01, 7.3314e-06, 2.5749e-05,  ..., 3.3417e-02,
           2.6031e-02, 6.6467e-02]]]], device='cuda:0', dtype=torch.float16), tensor([[[[2.5809e-05, 1.2994e-05, 1.0345e-01,  ..., 1.3412e-02,
           1.6891e-02, 1.3840e-02]],

         [[3.0220e-05, 3.8564e-05, 1.7725e-01,  ..., 2.0920e-02,
           1.0040e-02, 9.0332e-03]],

         [[1.4687e-04, 1.0639e-04, 9.6802e-02,  ..., 1.6373e-02,
           8.4412e-02, 1.2000e-01]],

         ...,

         [[7.0572e-05, 1.2100e-05, 1.9080e-01,  ..., 9.0149e-02,
           2.6294e-01, 1.2225e-01]],

         [[7.9536e-04, 1.0157e-03, 2.0764e-01,  ..., 1.2085e-02,
           3.7384e-02, 2.7451e-02]],

         [[2.5558e-04, 6.8247e-05, 2.7588e-01,  ..., 1.1997e-03,
           7.9498e-03, 2.8572e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[3.7241e-04, 1.4079e-04, 3.2812e-01,  ..., 7.5245e-04,
           1.8187e-03, 2.1763e-03]],

         [[2.9707e-04, 1.5628e-04, 2.1533e-01,  ..., 7.6771e-04,
           1.1129e-03, 1.3371e-03]],

         [[6.8545e-05, 1.6689e-05, 2.9272e-01,  ..., 8.5831e-04,
           1.6761e-04, 2.5616e-03]],

         ...,

         [[6.2752e-04, 4.8137e-04, 2.8687e-01,  ..., 1.5240e-03,
           1.7328e-03, 3.1447e-04]],

         [[6.0201e-06, 1.0133e-05, 2.3059e-01,  ..., 9.0981e-04,
           2.0103e-03, 1.8816e-03]],

         [[6.0320e-05, 2.1601e-04, 1.5149e-01,  ..., 1.3130e-02,
           2.6875e-03, 6.2609e-04]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.0562e-04, 4.8518e-05, 1.6785e-01,  ..., 4.8399e-04,
           1.5976e-02, 1.3237e-03]],

         [[2.4986e-04, 8.2254e-06, 2.7295e-01,  ..., 8.4829e-04,
           1.1712e-04, 5.0259e-04]],

         [[1.9932e-04, 1.4389e-04, 2.8394e-01,  ..., 1.0166e-03,
           4.0126e-04, 8.7976e-04]],

         ...,

         [[3.6657e-05, 2.2650e-05, 5.8105e-02,  ..., 6.3438e-03,
           4.7534e-01, 8.8623e-02]],

         [[6.9737e-05, 1.1301e-04, 2.4500e-01,  ..., 9.6083e-04,
           2.1877e-03, 2.1496e-03]],

         [[4.4370e-04, 1.1724e-04, 1.7383e-01,  ..., 1.4019e-03,
           1.0239e-02, 6.2103e-03]]]], device='cuda:0', dtype=torch.float16), tensor([[[[1.3411e-05, 7.7486e-07, 1.0891e-03,  ..., 1.7480e-01,
           7.1338e-01, 1.9608e-02]],

         [[8.5711e-05, 8.2552e-05, 9.7717e-02,  ..., 5.2521e-02,
           1.3863e-02, 6.4850e-03]],

         [[5.3835e-04, 1.5316e-03, 2.3499e-01,  ..., 1.1387e-03,
           1.8806e-03, 1.0691e-03]],

         ...,

         [[1.1826e-03, 3.0756e-05, 7.2205e-02,  ..., 2.3438e-02,
           1.1353e-01, 2.7069e-02]],

         [[4.6659e-04, 6.1154e-05, 2.4948e-02,  ..., 2.4524e-01,
           4.4751e-01, 7.5378e-02]],

         [[1.0252e-05, 3.6359e-06, 1.9779e-03,  ..., 1.1279e-01,
           3.6987e-01, 2.2253e-01]]]], device='cuda:0', dtype=torch.float16), tensor([[[[4.1656e-03, 1.1091e-03, 2.2510e-01,  ..., 4.6577e-03,
           5.0018e-02, 6.8909e-02]],

         [[3.7622e-04, 6.9475e-04, 2.5879e-01,  ..., 9.1324e-03,
           1.3390e-02, 2.2430e-02]],

         [[1.7729e-03, 1.1063e-03, 2.5342e-01,  ..., 2.7328e-02,
           4.1046e-02, 1.7731e-02]],

         ...,

         [[2.1803e-04, 6.3419e-05, 1.9666e-01,  ..., 7.1777e-02,
           1.4160e-01, 5.9052e-02]],

         [[2.1100e-04, 1.6952e-04, 4.6265e-02,  ..., 2.0218e-02,
           1.8591e-01, 4.8975e-01]],

         [[5.9009e-06, 5.3644e-06, 1.8494e-02,  ..., 1.8103e-01,
           4.1772e-01, 1.3147e-01]]]], device='cuda:0', dtype=torch.float16))...]
(#27) [tensor([0.0000e+00, 2.0995e-03, 2.7405e-02, 1.8757e-03, 1.6518e-02, 9.7674e-03,
        2.0832e-02, 1.4429e-02, 2.1065e-04, 7.7037e-05, 5.8581e-04, 1.4250e-03,
        2.9177e-04, 1.5615e-04, 1.3919e-04, 2.5103e-04, 1.2524e-04, 2.4706e-04,
        1.2310e-04, 6.6666e-05, 7.8322e-05, 1.1446e-04, 6.5871e-05, 4.1962e-04,
        1.1645e-03, 4.3152e-04, 9.8872e-03, 2.4708e-03, 1.4412e-04, 2.0080e-04,
        1.9734e-04, 3.4057e-04, 8.2279e-05, 3.3434e-04, 2.6524e-04, 1.2731e-04,
        2.9733e-04, 1.8690e-04, 4.2251e-04, 2.4262e-04, 8.3429e-05, 1.8682e-04,
        1.4841e-04, 9.4306e-05, 1.1355e-04, 5.8052e-05, 1.6379e-04, 3.6455e-04,
        6.7753e-05, 1.3450e-04, 1.1614e-03, 2.2301e-04, 2.3734e-04, 1.2042e-04,
        8.7030e-05, 1.1398e-04, 1.2669e-04, 1.7507e-04, 2.9081e-04, 2.1285e-04,
        2.0025e-04, 1.2710e-04, 1.4154e-04, 1.3130e-04, 9.9964e-05, 5.5416e-04,
        5.4445e-04, 1.5767e-04, 1.6429e-04, 5.0369e-05, 4.8870e-05, 7.7485e-05,
        1.0013e-04, 8.9446e-04, 1.7793e-04, 2.5433e-04, 7.0215e-05, 5.6076e-05,
        1.2590e-04, 1.6106e-04, 4.3351e-04, 2.5718e-04, 4.7299e-04, 1.2339e-04,
        1.4607e-04, 2.3589e-04, 8.7904e-05, 2.9971e-04, 2.5707e-04, 2.0849e-04,
        5.4416e-04, 2.8961e-04, 4.8098e-05, 3.3206e-04, 4.5613e-05, 9.0711e-05,
        1.2050e-04, 1.0690e-04, 1.5388e-04, 4.3277e-04, 1.7042e-04, 1.7599e-04,
        8.3460e-04, 3.5083e-04, 2.1440e-04, 2.5117e-04, 5.9451e-04, 8.2867e-05,
        1.3558e-04, 9.2849e-05, 3.9647e-04, 7.3888e-05, 3.6669e-04, 1.9151e-04,
        3.0562e-04, 8.1717e-05, 5.8042e-05, 4.5635e-04, 1.1870e-04, 1.6051e-04,
        1.3402e-04, 4.1933e-04, 1.1864e-04, 1.2055e-04, 6.3635e-04, 7.4235e-05,
        7.9467e-05, 1.8167e-04, 2.1731e-04, 2.9772e-04, 1.0390e-04, 1.4710e-04,
        4.6704e-04, 3.2933e-04, 9.6793e-05, 6.0079e-05, 1.7493e-04, 5.4936e-05,
        2.3987e-04, 6.5725e-05, 1.7009e-04, 7.9013e-04, 5.8277e-04, 1.1043e-04,
        1.6312e-04, 8.4647e-05, 1.2604e-04, 1.3627e-04, 4.3728e-04, 3.9532e-04,
        6.1274e-04, 2.0146e-04, 9.5331e-05, 1.3595e-04, 1.5405e-04, 1.3751e-04,
        1.7501e-04, 1.4274e-04, 7.0240e-05, 9.5953e-05, 2.0510e-04, 7.6978e-05,
        4.0205e-04, 3.9932e-04, 1.7846e-04, 5.4029e-04, 3.7338e-04, 3.5644e-04,
        4.9995e-04, 3.3334e-04, 1.7340e-04, 1.3439e-04, 1.1004e-04, 3.7257e-04,
        8.1710e-05, 1.0376e-04, 2.4555e-04, 2.1710e-04, 1.9823e-04, 2.4076e-04,
        2.6112e-04, 8.2458e-05, 3.4772e-04, 1.5195e-04, 1.4768e-04, 1.6743e-04,
        1.9006e-04, 7.7818e-05, 4.8230e-04, 1.8233e-04, 2.2664e-04, 2.4231e-04,
        1.5073e-04, 5.5789e-05, 2.2686e-04, 1.8810e-04, 3.0044e-04, 2.7613e-04,
        1.5855e-04, 8.2310e-05, 1.4121e-04, 1.1345e-04, 4.8753e-04, 3.0198e-04,
        1.0925e-04, 1.1696e-04, 4.9969e-05, 1.3454e-04, 4.7630e-04, 5.6139e-05,
        1.4386e-04, 3.7507e-04, 3.5993e-04, 2.8365e-04, 3.7485e-04, 1.4938e-04,
        2.2019e-04, 1.1113e-04, 4.8371e-04, 1.2875e-04, 2.7149e-04, 7.7015e-05,
        3.5546e-04, 8.1440e-05, 1.0748e-04, 3.1628e-04, 5.7318e-04, 2.2540e-04,
        1.8270e-04, 5.4165e-05, 3.3378e-05, 8.8320e-05, 1.2913e-04, 5.0935e-05,
        1.0670e-03, 1.9158e-04, 1.2707e-04, 1.3573e-04, 1.9572e-04, 2.5263e-04,
        1.6311e-04, 2.4661e-04, 1.4102e-04, 1.6365e-04, 1.0914e-04, 1.0444e-04,
        1.0842e-04, 9.3341e-05, 4.7737e-05, 4.5232e-05, 7.5839e-05, 1.4865e-04,
        3.6155e-05, 3.2668e-05, 7.9449e-05, 1.1816e-04, 1.7056e-04, 9.0694e-04,
        1.8932e-04, 2.3635e-04, 1.8588e-04, 1.2854e-04, 4.9513e-05, 1.5092e-04,
        1.3086e-04, 1.7316e-04, 1.0433e-04, 1.5422e-04, 3.5250e-04, 1.0327e-04,
        1.3559e-04, 2.0887e-04, 2.4345e-04, 6.6416e-05, 3.8511e-04, 1.6539e-04,
        6.1245e-05, 6.1404e-05, 7.7612e-05, 1.1043e-04, 3.2745e-04, 2.3557e-04,
        6.3927e-04, 1.0744e-04, 8.0083e-05, 1.0547e-04, 1.0222e-04, 2.2153e-04,
        1.0574e-04, 6.4695e-04, 1.0208e-04, 1.1976e-04, 1.0704e-04, 9.5863e-05,
        8.7803e-05, 5.1131e-05, 5.0665e-05, 2.5373e-04, 1.6662e-04, 6.6702e-05,
        6.4874e-05, 8.0515e-05, 1.4666e-04, 1.6256e-04, 1.9190e-04, 3.4264e-04,
        2.7973e-04, 7.9241e-05, 2.3414e-04, 7.1643e-05, 1.5972e-04, 9.3743e-05,
        5.0138e-05, 1.1735e-04, 1.3944e-04, 1.3560e-04, 5.6217e-04, 1.4262e-04,
        9.1778e-05, 5.0265e-04, 8.5356e-05, 5.0947e-05, 9.0430e-05, 1.3529e-04,
        9.6027e-05, 3.8390e-05, 4.2231e-04, 2.4103e-04, 2.0410e-04, 1.7589e-04,
        8.2773e-04, 3.0008e-05, 1.4848e-04, 1.0857e-03, 3.9088e-04, 1.2605e-04,
        5.4416e-05, 3.9584e-04, 1.3449e-04, 1.0714e-04, 3.8068e-04, 1.2090e-04,
        3.5730e-04, 2.9159e-04, 8.2976e-05, 1.6814e-04, 2.2068e-04, 1.4559e-04,
        6.0188e-05, 1.2359e-03, 1.6058e-04, 1.8238e-04, 1.8631e-04, 9.5567e-05,
        2.1929e-04, 1.7106e-04, 1.2618e-04, 1.3433e-04, 8.1627e-05, 1.3603e-04,
        1.0694e-04, 8.9566e-05, 2.7342e-04, 2.7877e-04, 1.3195e-04, 6.0129e-04,
        1.9970e-04, 7.0297e-05, 2.6676e-04, 3.1597e-04, 2.6740e-03, 5.2099e-04,
        1.4511e-03, 2.8406e-04, 2.3990e-04, 1.6956e-04, 5.7075e-05, 2.2302e-04,
        4.8397e-05, 1.6891e-04, 3.1918e-04, 2.3269e-04, 1.1079e-04, 1.9822e-04,
        5.6211e-05, 1.4769e-04, 6.7083e-05, 1.4303e-04, 1.0037e-04, 3.1506e-04,
        2.5585e-04, 1.3075e-04, 3.3099e-04, 6.9556e-04, 7.5481e-04, 1.0352e-02,
        5.5527e-02, 3.4235e-02, 2.9242e-02, 3.1150e-02, 1.1207e-02, 4.2191e-02,
        1.0118e-01, 4.8331e-02, 6.4940e-02, 1.8898e-02, 4.7815e-02, 2.4592e-02,
        1.2309e-01, 1.6283e-01, 0.0000e+00]),tensor([0.0000e+00, 1.1428e-03, 2.3274e-02, 1.1907e-03, 8.9420e-03, 6.5407e-03,
        2.0408e-02, 4.3832e-03, 3.1063e-04, 1.0249e-04, 8.4341e-04, 1.9057e-03,
        4.0820e-04, 2.6501e-04, 1.9689e-04, 2.7995e-04, 2.0532e-04, 4.4914e-04,
        1.7689e-04, 9.0413e-05, 9.0322e-05, 1.2604e-04, 8.4000e-05, 4.7947e-04,
        1.6504e-03, 3.8463e-04, 3.5429e-03, 2.2946e-03, 1.7889e-04, 3.4931e-04,
        2.8291e-04, 5.2981e-04, 1.1950e-04, 4.2238e-04, 3.7901e-04, 1.9096e-04,
        4.2443e-04, 2.3981e-04, 6.9420e-04, 3.9606e-04, 1.2204e-04, 3.2763e-04,
        1.5902e-04, 1.1275e-04, 1.1827e-04, 7.6164e-05, 3.2146e-04, 3.7329e-04,
        9.3480e-05, 1.1990e-04, 9.1333e-04, 4.9621e-04, 3.1537e-04, 1.7254e-04,
        1.1908e-04, 1.4859e-04, 1.5738e-04, 2.6356e-04, 3.3557e-04, 2.9307e-04,
        2.7339e-04, 1.7810e-04, 2.0775e-04, 2.7207e-04, 1.4190e-04, 5.0055e-04,
        6.9121e-04, 2.6799e-04, 1.9556e-04, 5.1097e-05, 4.6475e-05, 7.5740e-05,
        9.1364e-05, 1.3119e-03, 2.3765e-04, 3.3759e-04, 7.8370e-05, 6.9448e-05,
        1.7582e-04, 2.1065e-04, 3.7060e-04, 3.3075e-04, 3.9961e-04, 1.3346e-04,
        1.4844e-04, 3.2793e-04, 1.1029e-04, 3.9100e-04, 3.4012e-04, 2.4977e-04,
        8.2243e-04, 3.8835e-04, 4.9788e-05, 4.7690e-04, 4.6502e-05, 8.0739e-05,
        1.8790e-04, 1.3386e-04, 1.9380e-04, 5.6071e-04, 2.1159e-04, 2.7116e-04,
        1.0572e-03, 3.9170e-04, 2.8231e-04, 3.0235e-04, 4.6741e-04, 1.0664e-04,
        1.9242e-04, 1.0609e-04, 5.9688e-04, 8.7854e-05, 3.5421e-04, 2.3595e-04,
        2.1416e-04, 9.0704e-05, 5.3948e-05, 4.0097e-04, 1.0132e-04, 1.6391e-04,
        1.7380e-04, 5.1520e-04, 1.0464e-04, 1.5229e-04, 5.2985e-04, 8.3928e-05,
        1.0074e-04, 1.9376e-04, 2.0115e-04, 3.1249e-04, 1.6797e-04, 2.1608e-04,
        4.1185e-04, 2.7122e-04, 1.1074e-04, 8.6168e-05, 2.6278e-04, 5.7617e-05,
        3.5066e-04, 8.3157e-05, 2.7624e-04, 7.5434e-04, 7.3077e-04, 1.4125e-04,
        1.4431e-04, 1.1018e-04, 1.1807e-04, 2.2023e-04, 4.1189e-04, 3.4933e-04,
        9.4902e-04, 2.3166e-04, 1.2925e-04, 2.1159e-04, 1.9866e-04, 1.9730e-04,
        2.6248e-04, 2.4181e-04, 9.4549e-05, 1.1723e-04, 2.1668e-04, 1.0529e-04,
        3.4569e-04, 4.0233e-04, 3.1571e-04, 5.7224e-04, 4.2273e-04, 3.9981e-04,
        6.4458e-04, 3.9109e-04, 1.6246e-04, 1.5736e-04, 1.3749e-04, 2.9217e-04,
        1.1631e-04, 1.0899e-04, 2.5793e-04, 2.7019e-04, 2.7818e-04, 3.0057e-04,
        1.9653e-04, 1.0447e-04, 3.4721e-04, 2.1752e-04, 2.0327e-04, 1.6898e-04,
        2.5641e-04, 1.0682e-04, 5.7060e-04, 1.9673e-04, 2.0262e-04, 2.5019e-04,
        1.9895e-04, 6.9616e-05, 1.8872e-04, 3.3736e-04, 4.4242e-04, 2.6674e-04,
        2.0270e-04, 8.9850e-05, 2.0539e-04, 1.4989e-04, 4.5766e-04, 2.1001e-04,
        8.8902e-05, 2.1756e-04, 5.5733e-05, 1.5160e-04, 6.0036e-04, 7.3456e-05,
        1.6657e-04, 4.2415e-04, 3.2946e-04, 3.4496e-04, 2.8357e-04, 1.3462e-04,
        2.4563e-04, 1.5374e-04, 7.1286e-04, 1.2245e-04, 3.0415e-04, 1.2173e-04,
        4.1646e-04, 9.2782e-05, 1.9499e-04, 4.0677e-04, 5.2127e-04, 2.5313e-04,
        1.6872e-04, 5.7713e-05, 4.2190e-05, 1.0021e-04, 1.8036e-04, 6.8958e-05,
        7.9758e-04, 2.5371e-04, 9.6299e-05, 1.3192e-04, 2.0231e-04, 2.9497e-04,
        2.1062e-04, 2.8255e-04, 2.0349e-04, 2.0787e-04, 1.9647e-04, 1.2826e-04,
        1.4647e-04, 1.1400e-04, 4.9530e-05, 6.0109e-05, 7.6078e-05, 1.8782e-04,
        3.9765e-05, 3.1769e-05, 7.4599e-05, 1.1686e-04, 2.3437e-04, 9.4765e-04,
        2.2985e-04, 3.2091e-04, 1.3955e-04, 1.1344e-04, 5.5189e-05, 2.1087e-04,
        1.4978e-04, 2.2272e-04, 1.6644e-04, 2.3552e-04, 4.8570e-04, 1.2146e-04,
        1.3137e-04, 2.1900e-04, 2.3866e-04, 8.5194e-05, 3.0123e-04, 1.8123e-04,
        7.5671e-05, 7.5437e-05, 5.6762e-05, 1.4442e-04, 2.8319e-04, 1.8096e-04,
        5.6301e-04, 1.0835e-04, 8.8881e-05, 1.0820e-04, 1.4147e-04, 1.9324e-04,
        1.3809e-04, 5.6590e-04, 1.4434e-04, 2.1957e-04, 1.1988e-04, 1.2004e-04,
        1.1065e-04, 5.8625e-05, 5.2683e-05, 1.8368e-04, 2.5346e-04, 6.3669e-05,
        5.6535e-05, 1.0549e-04, 1.7528e-04, 2.3184e-04, 1.8204e-04, 3.4046e-04,
        2.0013e-04, 9.1819e-05, 2.4560e-04, 1.1775e-04, 1.6442e-04, 1.3183e-04,
        7.6755e-05, 1.1127e-04, 1.4755e-04, 2.1237e-04, 5.8895e-04, 1.9800e-04,
        1.0498e-04, 6.6457e-04, 1.1452e-04, 6.8602e-05, 8.3460e-05, 1.2257e-04,
        1.4771e-04, 5.2702e-05, 4.5888e-04, 2.4035e-04, 1.9363e-04, 2.2607e-04,
        9.3730e-04, 3.0023e-05, 1.7685e-04, 1.1332e-03, 5.6494e-04, 1.9593e-04,
        6.3419e-05, 4.4880e-04, 1.6253e-04, 1.0711e-04, 4.7416e-04, 1.4422e-04,
        4.8662e-04, 2.8576e-04, 1.0311e-04, 2.3134e-04, 2.6105e-04, 1.9422e-04,
        5.7233e-05, 8.3108e-04, 1.6177e-04, 1.6653e-04, 2.4872e-04, 1.2288e-04,
        2.4279e-04, 1.6256e-04, 1.2991e-04, 1.2907e-04, 9.6348e-05, 1.4485e-04,
        1.2896e-04, 1.1279e-04, 2.1468e-04, 4.7341e-04, 1.0918e-04, 7.2936e-04,
        1.8292e-04, 7.5761e-05, 2.3603e-04, 4.6083e-04, 2.5521e-03, 4.9356e-04,
        7.7311e-04, 2.4729e-04, 2.5329e-04, 1.4164e-04, 7.0850e-05, 1.9658e-04,
        6.6195e-05, 2.1890e-04, 3.9471e-04, 3.5494e-04, 1.3964e-04, 1.3762e-04,
        7.5412e-05, 1.2458e-04, 9.0459e-05, 1.5883e-04, 1.3026e-04, 2.3306e-04,
        3.0765e-04, 9.8926e-05, 3.8125e-04, 6.9503e-04, 7.9650e-04, 6.6652e-03,
        4.0395e-02, 4.1115e-02, 5.7639e-02, 4.3746e-02, 1.5590e-02, 5.7824e-02,
        9.0219e-02, 2.7566e-02, 3.4977e-02, 9.6270e-03, 2.5740e-02, 1.6211e-02,
        6.8687e-02, 1.3654e-01, 1.5380e-01, 0.0000e+00]),tensor([0.0000e+00, 5.7286e-04, 2.2771e-02, 7.8036e-04, 5.0904e-03, 4.0628e-03,
        2.2771e-02, 2.0210e-03, 3.4597e-04, 1.9295e-04, 2.0095e-03, 3.5531e-03,
        1.4617e-03, 4.3770e-04, 2.4688e-04, 6.6279e-04, 2.8111e-04, 6.6326e-04,
        4.3322e-04, 1.3499e-04, 7.9949e-05, 8.4731e-05, 1.0391e-04, 7.1528e-04,
        4.7657e-03, 7.0813e-04, 1.7315e-03, 1.2592e-03, 1.4907e-04, 7.1563e-04,
        2.9813e-04, 9.0370e-04, 2.9700e-04, 1.3158e-03, 1.4413e-03, 3.3546e-04,
        1.0328e-03, 3.3509e-04, 1.0053e-03, 1.1073e-03, 1.3359e-04, 8.8873e-04,
        2.1641e-04, 2.0189e-04, 1.9193e-04, 2.2559e-04, 5.8184e-04, 1.1853e-03,
        1.4145e-04, 1.0639e-04, 5.4079e-04, 6.1079e-04, 6.6427e-04, 3.0749e-04,
        1.1385e-04, 3.5702e-04, 1.0318e-04, 7.0429e-04, 9.0967e-04, 1.1044e-03,
        5.2027e-04, 3.6607e-04, 1.5530e-04, 3.7167e-04, 5.4024e-04, 1.5100e-03,
        4.9952e-04, 5.2123e-04, 4.8523e-04, 1.2589e-04, 1.0769e-04, 1.7123e-04,
        1.5657e-04, 2.9112e-03, 1.4274e-04, 9.2593e-04, 1.1881e-04, 1.1784e-04,
        3.3345e-04, 5.8137e-04, 7.6516e-04, 1.2876e-03, 1.0177e-03, 4.6127e-04,
        1.9935e-04, 7.6349e-04, 1.2809e-04, 1.3534e-03, 3.7585e-04, 2.6480e-04,
        3.1385e-03, 1.6537e-03, 1.1008e-04, 2.2643e-03, 1.0168e-04, 1.1584e-04,
        1.7667e-04, 6.9013e-05, 4.2950e-04, 1.3012e-03, 4.5368e-04, 7.1628e-04,
        3.5682e-03, 3.2466e-04, 7.5486e-04, 1.2307e-03, 1.3831e-03, 1.9176e-04,
        5.7249e-04, 1.5523e-04, 2.1078e-03, 1.0426e-04, 2.9957e-04, 3.3148e-04,
        4.7481e-04, 2.8876e-04, 5.3722e-05, 1.5839e-03, 1.1489e-04, 1.1523e-04,
        2.7448e-04, 1.0228e-03, 1.8513e-04, 1.8792e-04, 1.2832e-03, 2.0447e-04,
        2.6721e-04, 4.8275e-04, 4.7587e-04, 1.5166e-03, 5.3572e-04, 7.4356e-04,
        4.1895e-04, 3.4964e-04, 2.3567e-04, 1.3682e-04, 1.1626e-03, 9.8989e-05,
        8.4973e-04, 1.2917e-04, 3.0071e-04, 7.3699e-04, 1.0170e-03, 3.3464e-04,
        2.6129e-04, 2.3809e-04, 2.8780e-04, 5.8627e-04, 4.8401e-04, 1.5403e-03,
        2.3263e-03, 1.2760e-03, 4.4957e-04, 4.1345e-04, 2.5254e-04, 4.1435e-04,
        8.4981e-04, 3.7469e-04, 4.5488e-04, 1.2687e-04, 1.0005e-03, 3.1337e-04,
        9.7149e-04, 7.8572e-04, 5.7810e-04, 7.8672e-04, 4.9994e-04, 3.9274e-04,
        2.1916e-03, 6.2847e-04, 5.9373e-04, 3.7892e-04, 3.5926e-04, 9.9718e-04,
        5.9089e-04, 2.9871e-04, 3.4261e-04, 2.6734e-04, 8.0426e-04, 7.7091e-04,
        3.6582e-04, 4.0403e-04, 7.5744e-04, 4.0997e-04, 2.1212e-04, 2.8835e-04,
        1.6691e-03, 1.1856e-04, 5.1556e-04, 3.0496e-04, 3.3542e-04, 5.8674e-04,
        5.1542e-04, 1.7525e-04, 6.0836e-04, 6.1852e-04, 1.5029e-03, 9.2553e-04,
        3.3863e-04, 9.0054e-05, 9.5206e-04, 5.1345e-04, 6.1509e-04, 4.2728e-04,
        1.7250e-04, 7.9851e-04, 1.3385e-04, 2.6999e-04, 3.1086e-03, 2.3381e-04,
        1.8874e-04, 4.4699e-04, 4.1366e-04, 9.0383e-04, 1.0457e-03, 2.8331e-04,
        6.4531e-04, 4.3844e-04, 1.2674e-03, 3.0840e-04, 1.0514e-03, 3.7315e-04,
        7.3908e-04, 1.1800e-04, 5.1913e-04, 3.4878e-04, 1.1709e-03, 6.7962e-04,
        3.6661e-04, 1.1643e-04, 1.1054e-04, 2.0499e-04, 4.7668e-04, 1.7427e-04,
        6.4090e-04, 4.5731e-04, 1.7207e-04, 3.7586e-04, 7.2387e-04, 9.3499e-04,
        3.5878e-04, 2.8177e-04, 3.0024e-04, 4.7741e-04, 5.3967e-04, 1.5226e-04,
        2.1269e-04, 6.6179e-04, 9.5943e-05, 1.8566e-04, 2.1449e-04, 7.4755e-04,
        8.0497e-05, 6.7491e-05, 1.0145e-04, 1.7014e-04, 6.5325e-04, 1.0062e-03,
        4.3598e-04, 3.5394e-04, 3.6851e-04, 3.9281e-04, 1.3553e-04, 5.0908e-04,
        6.0978e-04, 3.9194e-04, 2.9274e-04, 5.6815e-04, 6.0545e-04, 3.8791e-04,
        2.7629e-04, 4.2858e-04, 7.4567e-04, 1.8318e-04, 6.2996e-04, 6.4295e-04,
        2.6214e-04, 1.8812e-04, 4.9992e-05, 2.2472e-04, 3.3139e-04, 2.7765e-04,
        5.2500e-04, 1.7058e-04, 1.9808e-04, 1.6276e-04, 4.6012e-04, 5.5175e-04,
        4.0947e-04, 1.1434e-03, 3.7058e-04, 3.8472e-04, 1.3055e-04, 3.2940e-04,
        1.8623e-04, 1.4078e-04, 1.0142e-04, 7.4190e-04, 5.5490e-04, 1.2860e-04,
        7.9703e-05, 2.1560e-04, 2.7285e-04, 4.8222e-04, 3.0027e-04, 1.0278e-03,
        9.7898e-04, 9.7094e-05, 5.3036e-04, 1.4270e-04, 2.6548e-04, 3.6470e-04,
        1.8014e-04, 2.6951e-04, 1.5222e-04, 4.4396e-04, 1.2483e-03, 7.5976e-04,
        2.0897e-04, 2.9657e-03, 2.8246e-04, 2.1327e-04, 9.0057e-05, 1.4178e-04,
        1.5418e-04, 1.1074e-04, 5.9121e-04, 4.0356e-04, 5.1514e-04, 3.9053e-04,
        1.5060e-03, 8.8354e-05, 2.7425e-04, 1.5179e-03, 6.8717e-04, 1.9061e-04,
        1.5174e-04, 4.5486e-04, 2.3127e-04, 1.3264e-04, 9.9868e-04, 2.3923e-04,
        6.9992e-04, 4.3032e-04, 2.2215e-04, 5.1844e-04, 4.1179e-04, 1.9535e-04,
        5.4195e-05, 4.0967e-04, 2.9737e-04, 3.0641e-04, 5.3301e-04, 3.3999e-04,
        6.9112e-04, 3.0314e-04, 3.1266e-04, 1.8599e-04, 1.4755e-04, 3.8114e-04,
        2.5808e-04, 2.7141e-04, 4.2897e-04, 3.9054e-04, 1.8030e-04, 9.9781e-04,
        2.6165e-04, 9.1624e-05, 2.7464e-04, 5.3114e-04, 1.4732e-03, 4.7182e-04,
        4.1303e-04, 3.4679e-04, 4.7495e-04, 3.2505e-04, 1.7402e-04, 1.7786e-04,
        1.0989e-04, 3.1978e-04, 1.2162e-03, 6.1983e-04, 1.5997e-04, 1.9523e-04,
        1.0782e-04, 1.0111e-04, 1.2507e-04, 2.0885e-04, 3.4807e-04, 7.4781e-04,
        3.3165e-04, 1.8589e-04, 2.9035e-04, 7.3176e-04, 6.4205e-04, 5.8082e-03,
        1.2972e-02, 2.3586e-02, 7.1743e-02, 5.4822e-02, 1.8954e-02, 8.0957e-02,
        4.8488e-02, 1.0542e-02, 1.4404e-02, 6.6976e-03, 1.3116e-02, 1.4983e-02,
        3.0827e-02, 7.1765e-02, 1.1551e-01, 1.3362e-01, 0.0000e+00]),tensor([0.0000e+00, 5.0560e-04, 2.4907e-02, 5.9794e-04, 5.7207e-03, 3.9906e-03,
        2.3623e-02, 1.9364e-03, 1.6044e-04, 1.0710e-04, 7.4690e-04, 8.6671e-04,
        3.0832e-04, 3.7043e-04, 1.7013e-04, 4.1428e-04, 1.2319e-04, 7.0174e-04,
        3.1063e-04, 8.7601e-05, 9.3652e-05, 1.3985e-04, 1.4444e-04, 8.1586e-04,
        8.8254e-04, 3.4756e-04, 1.7341e-03, 1.9200e-03, 1.6553e-04, 6.1113e-04,
        2.3463e-04, 3.3853e-04, 2.1676e-04, 1.3183e-03, 6.3246e-04, 1.9232e-04,
        7.7389e-04, 1.5505e-04, 1.7569e-03, 1.0698e-03, 1.1771e-04, 9.3174e-04,
        1.6524e-04, 1.6390e-04, 1.1159e-04, 1.6503e-04, 5.3374e-04, 1.3343e-03,
        9.1649e-05, 1.1583e-04, 5.6533e-04, 3.5947e-04, 2.0863e-04, 1.8373e-04,
        6.1552e-05, 1.1930e-04, 1.0209e-04, 3.5084e-04, 2.3737e-04, 2.8266e-04,
        2.4148e-04, 2.9818e-04, 1.0630e-04, 5.7750e-04, 2.2468e-04, 8.9098e-04,
        1.4880e-04, 9.3051e-05, 3.4723e-04, 7.3378e-05, 5.6421e-05, 7.4537e-05,
        8.8305e-05, 1.4003e-03, 1.0339e-04, 2.4873e-04, 4.8633e-05, 5.8433e-05,
        1.5521e-04, 3.4960e-04, 1.6988e-04, 4.8279e-04, 1.8946e-04, 1.2474e-04,
        1.0644e-04, 2.7726e-04, 9.0514e-05, 2.1190e-04, 7.9803e-05, 6.8589e-05,
        4.5893e-04, 6.9676e-04, 4.2232e-05, 3.4051e-04, 6.0073e-05, 7.8445e-05,
        1.6542e-04, 6.1519e-05, 2.0696e-04, 2.0426e-03, 2.8766e-04, 3.7185e-04,
        5.2092e-04, 9.7843e-05, 1.2730e-04, 3.7216e-04, 6.2907e-04, 1.7693e-04,
        2.0401e-04, 1.3536e-04, 6.7779e-04, 4.6137e-05, 7.7977e-05, 1.2638e-04,
        1.9802e-04, 2.5627e-04, 6.0400e-05, 8.1680e-04, 8.4465e-05, 1.1403e-04,
        1.9693e-04, 1.0758e-03, 7.4589e-05, 1.2809e-04, 5.5207e-04, 5.8950e-05,
        1.4802e-04, 1.4528e-04, 1.2315e-04, 6.6652e-04, 3.0166e-04, 2.2394e-04,
        1.4713e-04, 8.6702e-05, 1.0466e-04, 3.6694e-05, 2.4116e-04, 4.6804e-05,
        1.3056e-03, 9.0508e-05, 3.1939e-04, 7.3786e-04, 7.9237e-04, 2.1344e-04,
        1.1388e-04, 8.7805e-05, 1.4214e-04, 2.3875e-04, 1.1133e-04, 4.9381e-04,
        9.3284e-04, 5.1671e-04, 1.7765e-04, 3.8369e-04, 1.7615e-04, 6.0754e-04,
        4.3931e-04, 4.2072e-04, 7.1281e-05, 5.3558e-05, 7.7907e-04, 2.2559e-04,
        6.4287e-04, 1.0322e-03, 6.0066e-04, 7.0885e-04, 6.5587e-04, 3.0060e-04,
        2.9256e-04, 4.3491e-04, 2.6596e-04, 1.9127e-04, 2.8223e-04, 8.5382e-04,
        1.7655e-04, 9.5883e-05, 2.6232e-04, 2.8586e-04, 6.5996e-04, 4.2432e-04,
        2.3206e-04, 2.6273e-04, 4.0223e-04, 1.5376e-04, 2.5311e-04, 2.6551e-04,
        2.4048e-04, 8.4578e-05, 3.4703e-04, 1.7431e-04, 7.9560e-05, 2.9025e-04,
        1.7535e-04, 7.4190e-05, 4.8380e-04, 5.1374e-04, 2.7525e-04, 5.1383e-04,
        2.0320e-04, 1.0459e-04, 6.4590e-04, 1.8505e-04, 2.3482e-04, 3.2049e-04,
        5.5179e-05, 3.5015e-04, 8.6455e-05, 1.6435e-04, 4.1835e-04, 1.8874e-04,
        1.2186e-04, 7.3104e-04, 1.9934e-04, 1.2212e-04, 4.9441e-04, 9.3493e-05,
        6.9139e-04, 3.9989e-04, 5.7169e-04, 1.5529e-04, 1.1267e-03, 1.3518e-04,
        1.3415e-03, 7.4966e-05, 2.1131e-04, 1.2384e-04, 5.1240e-04, 3.8072e-04,
        7.6797e-05, 4.2901e-05, 4.3434e-05, 1.0013e-04, 3.8891e-04, 8.9513e-05,
        4.0144e-04, 1.8561e-04, 5.5144e-05, 1.7890e-04, 4.0266e-04, 3.2046e-04,
        1.0659e-04, 1.4071e-04, 1.7440e-04, 7.3211e-04, 3.0608e-04, 1.3003e-04,
        2.4371e-04, 7.5745e-05, 3.5547e-05, 6.5328e-05, 5.3354e-05, 2.7164e-04,
        2.9711e-05, 4.0123e-05, 6.8339e-05, 1.5684e-04, 6.0292e-04, 1.0540e-03,
        3.2664e-04, 1.2090e-04, 1.8934e-04, 1.9113e-04, 6.1808e-05, 1.3313e-04,
        2.9849e-04, 2.6359e-04, 2.5873e-04, 2.7975e-04, 4.3305e-04, 2.0824e-04,
        8.1823e-05, 9.4749e-05, 2.6496e-04, 7.8413e-05, 7.4421e-04, 2.8434e-04,
        1.0209e-04, 1.1152e-04, 5.8379e-05, 1.1257e-04, 2.6644e-04, 8.2569e-05,
        1.0535e-04, 7.4938e-05, 8.8261e-05, 1.0307e-04, 3.9501e-04, 4.9528e-04,
        1.4626e-04, 1.1519e-03, 3.6512e-04, 2.4853e-04, 8.2291e-05, 1.1803e-04,
        1.1270e-04, 4.9954e-05, 4.7462e-05, 4.8065e-04, 4.8209e-04, 5.7760e-05,
        5.3046e-05, 1.6207e-04, 1.3905e-04, 4.8406e-04, 1.3140e-04, 3.4219e-04,
        5.9760e-04, 1.0328e-04, 6.3653e-04, 1.1117e-04, 1.5986e-04, 1.1243e-04,
        1.4890e-04, 1.6010e-04, 2.1941e-04, 2.3705e-04, 1.5912e-03, 3.9739e-04,
        8.0440e-05, 4.8411e-04, 3.6257e-04, 1.6729e-04, 7.0955e-05, 1.2455e-04,
        1.6998e-04, 5.4474e-05, 5.0895e-04, 2.4246e-04, 3.2065e-04, 2.3654e-04,
        1.4836e-03, 4.8037e-05, 1.4867e-04, 1.6537e-03, 3.0352e-04, 2.5264e-04,
        7.3929e-05, 4.9423e-04, 1.7635e-04, 1.3637e-04, 6.6043e-04, 1.1426e-04,
        3.1581e-04, 2.5359e-04, 2.6444e-04, 1.6958e-04, 2.6751e-04, 2.8831e-04,
        3.8473e-05, 5.1326e-04, 1.3637e-04, 1.4975e-04, 2.0700e-04, 1.1449e-04,
        4.3965e-04, 1.9117e-04, 1.4123e-04, 1.1446e-04, 6.9182e-05, 2.1697e-04,
        1.0949e-04, 1.0318e-04, 5.3154e-04, 4.1174e-04, 1.2318e-04, 1.0324e-03,
        1.7001e-04, 6.2784e-05, 2.4483e-04, 3.3958e-04, 1.9040e-03, 3.7068e-04,
        4.0664e-04, 2.0174e-04, 2.4119e-04, 2.1556e-04, 1.0449e-04, 1.6731e-04,
        7.0051e-05, 1.5815e-04, 2.8986e-04, 5.1298e-04, 1.5077e-04, 1.3680e-04,
        8.0977e-05, 8.8358e-05, 7.6454e-05, 1.5895e-04, 1.0968e-04, 3.4210e-04,
        4.1466e-04, 1.6475e-04, 2.5017e-04, 1.1022e-03, 6.5474e-04, 4.0235e-03,
        1.3452e-02, 2.4685e-02, 2.4127e-02, 5.9290e-02, 3.5230e-02, 1.0587e-01,
        5.2090e-02, 1.0535e-02, 1.5337e-02, 5.8788e-03, 1.3643e-02, 1.2935e-02,
        3.3666e-02, 6.4544e-02, 1.0011e-01, 1.2372e-01, 1.1555e-01, 0.0000e+00]),tensor([0.0000e+00, 5.6587e-04, 2.1728e-02, 6.4390e-04, 5.3344e-03, 5.4775e-03,
        2.0594e-02, 1.9041e-03, 2.4622e-04, 1.2700e-04, 6.5077e-04, 8.7766e-04,
        2.7479e-04, 5.6153e-04, 4.1424e-04, 3.5786e-04, 1.6605e-04, 1.0071e-03,
        4.9308e-04, 1.0711e-04, 2.0170e-04, 2.8380e-04, 2.4784e-04, 9.8361e-04,
        8.1432e-04, 3.3000e-04, 1.7475e-03, 2.2247e-03, 2.7640e-04, 8.8908e-04,
        5.9152e-04, 4.5025e-04, 3.3113e-04, 9.9922e-04, 5.8304e-04, 2.0086e-04,
        7.7473e-04, 2.2817e-04, 2.4296e-03, 9.9786e-04, 2.4603e-04, 1.1048e-03,
        1.9549e-04, 1.9012e-04, 1.3130e-04, 1.6557e-04, 8.4809e-04, 1.0356e-03,
        2.0907e-04, 2.0224e-04, 8.4439e-04, 4.9405e-04, 2.7472e-04, 1.8501e-04,
        9.6171e-05, 1.6940e-04, 1.3569e-04, 4.4078e-04, 3.0053e-04, 4.1276e-04,
        2.9627e-04, 6.3410e-04, 1.3635e-04, 6.1891e-04, 2.1252e-04, 6.9927e-04,
        2.2333e-04, 1.0761e-04, 2.6187e-04, 8.4520e-05, 6.6649e-05, 8.4827e-05,
        1.0496e-04, 1.0960e-03, 1.5168e-04, 3.1859e-04, 7.1641e-05, 9.1000e-05,
        2.4104e-04, 4.3892e-04, 2.4185e-04, 4.7790e-04, 2.7277e-04, 1.5732e-04,
        1.2464e-04, 3.1475e-04, 1.1374e-04, 1.9919e-04, 1.1859e-04, 1.2150e-04,
        4.2331e-04, 6.1282e-04, 4.8111e-05, 2.9624e-04, 7.1726e-05, 9.8735e-05,
        2.9744e-04, 8.6664e-05, 2.6206e-04, 2.2843e-03, 3.5984e-04, 4.5547e-04,
        4.8896e-04, 1.4766e-04, 1.6910e-04, 5.8769e-04, 5.9476e-04, 3.8623e-04,
        2.2545e-04, 1.5948e-04, 6.6835e-04, 6.4234e-05, 1.2157e-04, 1.7009e-04,
        2.4222e-04, 3.1016e-04, 1.0619e-04, 6.0031e-04, 1.1119e-04, 1.8521e-04,
        3.2523e-04, 1.1773e-03, 9.7629e-05, 2.1939e-04, 5.3017e-04, 8.7205e-05,
        2.2334e-04, 2.1724e-04, 1.9499e-04, 5.4406e-04, 3.6878e-04, 2.5517e-04,
        1.9966e-04, 1.1607e-04, 1.2379e-04, 4.9170e-05, 2.5723e-04, 6.0543e-05,
        1.3468e-03, 1.7689e-04, 4.3313e-04, 7.6452e-04, 7.4241e-04, 3.5395e-04,
        1.5620e-04, 1.0666e-04, 1.3953e-04, 3.0363e-04, 1.8548e-04, 5.7995e-04,
        1.0432e-03, 4.8750e-04, 2.0524e-04, 5.0646e-04, 2.2309e-04, 7.7283e-04,
        4.7349e-04, 6.5462e-04, 8.1106e-05, 9.0265e-05, 6.8594e-04, 2.9431e-04,
        4.1012e-04, 1.0230e-03, 9.2447e-04, 7.0214e-04, 1.0623e-03, 4.6516e-04,
        2.6592e-04, 4.2171e-04, 2.6117e-04, 2.0140e-04, 3.8571e-04, 6.8391e-04,
        2.0890e-04, 1.0493e-04, 2.8365e-04, 3.3232e-04, 8.0910e-04, 5.5881e-04,
        2.0743e-04, 1.6737e-04, 3.2628e-04, 1.7086e-04, 4.3619e-04, 2.8912e-04,
        2.2860e-04, 1.5576e-04, 3.7983e-04, 2.7056e-04, 1.4220e-04, 2.3764e-04,
        2.0924e-04, 1.2657e-04, 3.5853e-04, 7.6382e-04, 2.4748e-04, 3.9013e-04,
        2.3199e-04, 2.2573e-04, 5.8524e-04, 2.4246e-04, 2.1586e-04, 2.6119e-04,
        1.0111e-04, 4.4994e-04, 9.1588e-05, 1.9342e-04, 3.4309e-04, 2.1724e-04,
        1.6826e-04, 1.1293e-03, 3.2779e-04, 1.6531e-04, 4.2406e-04, 1.6534e-04,
        9.9418e-04, 5.9898e-04, 6.0460e-04, 1.6294e-04, 8.8663e-04, 2.4508e-04,
        2.6165e-03, 9.6251e-05, 3.5032e-04, 1.7943e-04, 4.4331e-04, 2.7317e-04,
        1.1727e-04, 7.1055e-05, 5.2733e-05, 1.2715e-04, 3.5418e-04, 1.2659e-04,
        6.3827e-04, 2.6763e-04, 9.2715e-05, 2.0151e-04, 4.6411e-04, 4.0667e-04,
        1.5050e-04, 2.0945e-04, 4.2391e-04, 1.1801e-03, 6.1381e-04, 2.6092e-04,
        3.2495e-04, 1.1779e-04, 4.9670e-05, 8.5088e-05, 6.6986e-05, 2.9470e-04,
        4.7206e-05, 4.4794e-05, 8.1875e-05, 2.2147e-04, 6.0944e-04, 9.9779e-04,
        5.5961e-04, 1.8293e-04, 2.3659e-04, 1.9826e-04, 6.8821e-05, 1.9575e-04,
        2.7925e-04, 2.6616e-04, 3.5473e-04, 4.3095e-04, 9.3177e-04, 2.2859e-04,
        1.4401e-04, 1.6999e-04, 2.5330e-04, 1.2693e-04, 1.1100e-03, 3.1386e-04,
        1.2211e-04, 1.4504e-04, 9.4892e-05, 2.2153e-04, 4.5841e-04, 1.6269e-04,
        1.9385e-04, 1.1468e-04, 2.0212e-04, 1.2334e-04, 5.6796e-04, 5.3492e-04,
        1.9614e-04, 8.7095e-04, 4.4796e-04, 4.7899e-04, 9.4437e-05, 1.3899e-04,
        1.6191e-04, 7.2428e-05, 6.8920e-05, 2.7837e-04, 8.2154e-04, 6.6855e-05,
        6.0044e-05, 2.0067e-04, 1.8293e-04, 6.4519e-04, 3.0010e-04, 3.0557e-04,
        3.7227e-04, 2.2444e-04, 8.7540e-04, 1.6664e-04, 1.7920e-04, 1.2072e-04,
        1.8490e-04, 2.0434e-04, 4.1192e-04, 2.7412e-04, 1.4256e-03, 3.7010e-04,
        1.0384e-04, 4.0003e-04, 3.8594e-04, 1.9334e-04, 9.1531e-05, 1.6476e-04,
        2.6514e-04, 8.0804e-05, 8.5703e-04, 5.4871e-04, 3.5303e-04, 4.4951e-04,
        1.1536e-03, 5.1881e-05, 2.9081e-04, 1.7934e-03, 3.5977e-04, 5.4299e-04,
        9.5877e-05, 4.8924e-04, 2.1003e-04, 2.0400e-04, 9.0676e-04, 1.9863e-04,
        2.8336e-04, 2.5281e-04, 3.6081e-04, 1.8291e-04, 3.1815e-04, 3.8958e-04,
        6.8811e-05, 6.1120e-04, 2.1186e-04, 2.1937e-04, 2.5127e-04, 1.4577e-04,
        3.8199e-04, 2.3699e-04, 1.5524e-04, 1.4927e-04, 1.2502e-04, 3.9728e-04,
        1.3382e-04, 1.1477e-04, 7.0574e-04, 6.8766e-04, 1.3475e-04, 8.3511e-04,
        1.9183e-04, 1.2302e-04, 2.6622e-04, 3.1808e-04, 1.7854e-03, 3.9139e-04,
        5.0411e-04, 2.8310e-04, 3.4985e-04, 2.5098e-04, 1.0711e-04, 2.9261e-04,
        8.3960e-05, 3.2284e-04, 2.4991e-04, 7.3996e-04, 2.2538e-04, 1.3990e-04,
        1.0301e-04, 1.2580e-04, 1.3637e-04, 2.0873e-04, 1.2807e-04, 4.3750e-04,
        4.4813e-04, 1.6818e-04, 2.9523e-04, 1.6918e-03, 7.0121e-04, 4.0012e-03,
        9.4421e-03, 2.3547e-02, 1.8002e-02, 1.5052e-02, 3.3740e-02, 1.2836e-01,
        3.3576e-02, 8.4985e-03, 1.3387e-02, 5.4478e-03, 1.3300e-02, 1.2014e-02,
        2.6997e-02, 5.3274e-02, 7.9752e-02, 7.9804e-02, 1.3109e-01, 1.0764e-01,
        0.0000e+00]),tensor([0.0000e+00, 6.3286e-04, 2.8344e-02, 7.4669e-04, 4.9134e-03, 4.7843e-03,
        2.6535e-02, 1.9334e-03, 1.8244e-04, 9.2724e-05, 7.0789e-04, 1.2057e-03,
        3.3101e-04, 2.6285e-04, 1.5653e-04, 5.6781e-04, 1.1955e-04, 3.8828e-04,
        2.0950e-04, 6.9085e-05, 1.1896e-04, 1.9087e-04, 1.1603e-04, 4.4188e-04,
        1.0924e-03, 7.6542e-04, 1.6830e-03, 1.8126e-03, 1.7481e-04, 3.6879e-04,
        1.9885e-04, 3.2965e-04, 2.1268e-04, 8.9330e-04, 5.5135e-04, 2.3684e-04,
        9.4243e-04, 2.7256e-04, 3.2533e-03, 7.9451e-04, 1.5562e-04, 1.0822e-03,
        2.2303e-04, 1.8382e-04, 1.2666e-04, 1.4170e-04, 2.6427e-04, 1.2553e-03,
        1.0828e-04, 1.4891e-04, 8.3338e-04, 4.0488e-04, 2.4312e-04, 1.9322e-04,
        7.9144e-05, 1.4054e-04, 1.0988e-04, 6.6440e-04, 6.9790e-04, 6.2720e-04,
        4.0283e-04, 2.8995e-04, 1.3689e-04, 4.1698e-04, 2.8359e-04, 1.3595e-03,
        8.5284e-04, 2.4847e-04, 3.2513e-04, 9.2235e-05, 6.7095e-05, 8.6156e-05,
        9.6391e-05, 1.3202e-03, 1.2071e-04, 2.8115e-04, 5.8285e-05, 6.7980e-05,
        1.6563e-04, 8.7252e-04, 5.4744e-04, 7.6433e-04, 5.2643e-04, 2.5936e-04,
        1.7386e-04, 3.3449e-04, 9.9881e-05, 3.2552e-04, 3.6449e-04, 2.5871e-04,
        6.3532e-04, 8.0350e-04, 7.1280e-05, 4.4557e-04, 7.4039e-05, 1.0229e-04,
        1.8254e-04, 7.2930e-05, 2.8467e-04, 1.1377e-03, 8.4251e-04, 8.3980e-04,
        8.2625e-04, 8.6104e-04, 3.5923e-04, 1.0415e-03, 1.2717e-03, 1.7222e-04,
        2.4256e-04, 2.7272e-04, 9.0554e-04, 1.7313e-04, 9.5809e-04, 2.7046e-04,
        6.8124e-04, 3.1346e-04, 7.4373e-05, 1.3666e-03, 1.1819e-04, 1.1925e-04,
        1.7263e-04, 9.7686e-04, 1.5566e-04, 9.2376e-05, 1.0453e-03, 1.8048e-04,
        7.8395e-04, 3.7661e-04, 3.8920e-04, 9.0771e-04, 2.1728e-04, 2.4501e-04,
        8.9501e-04, 2.8641e-04, 1.1064e-04, 8.9264e-05, 4.2349e-04, 8.6318e-05,
        5.9470e-04, 8.9815e-05, 2.1756e-04, 5.7217e-04, 8.6432e-04, 1.7940e-04,
        2.7702e-04, 1.6701e-04, 2.8885e-04, 4.2887e-04, 5.4545e-04, 1.2938e-03,
        1.3899e-03, 7.6891e-04, 2.8955e-04, 3.7188e-04, 1.8005e-04, 1.0182e-03,
        7.2527e-04, 7.6923e-04, 8.2132e-05, 1.6295e-04, 9.3479e-04, 1.7122e-04,
        9.4348e-04, 1.1085e-03, 3.0806e-04, 7.3285e-04, 8.7663e-04, 6.0036e-04,
        4.5730e-04, 5.7535e-04, 5.0316e-04, 2.7086e-04, 2.2273e-04, 9.9034e-04,
        2.6237e-04, 1.9463e-04, 4.0748e-04, 4.5529e-04, 1.0515e-03, 3.1951e-04,
        4.5861e-04, 1.8048e-04, 5.6917e-04, 1.3622e-04, 1.3977e-04, 2.8330e-04,
        2.5503e-04, 1.1002e-04, 3.9179e-04, 3.7779e-04, 3.5387e-04, 3.6615e-04,
        1.7562e-04, 1.7665e-04, 5.0252e-04, 3.0097e-04, 4.0413e-04, 6.2347e-04,
        3.4528e-04, 1.3307e-04, 6.0570e-04, 3.4914e-04, 3.9676e-04, 4.8174e-04,
        1.2912e-04, 5.6122e-04, 1.1931e-04, 1.3258e-04, 4.7179e-04, 1.3637e-04,
        1.0008e-04, 7.7976e-04, 5.3888e-04, 4.2901e-04, 9.1829e-04, 2.5887e-04,
        7.5225e-04, 3.2573e-04, 6.1592e-04, 2.1353e-04, 8.1801e-04, 1.6942e-04,
        1.6241e-03, 1.6208e-04, 4.8965e-04, 3.8499e-04, 8.7303e-04, 3.3915e-04,
        1.8622e-04, 1.1603e-04, 6.1861e-05, 1.8245e-04, 3.5812e-04, 7.5868e-05,
        7.5945e-04, 3.6356e-04, 4.4100e-04, 3.5319e-04, 4.5444e-04, 5.7086e-04,
        2.1990e-04, 2.8375e-04, 4.3559e-04, 6.6483e-04, 3.3047e-04, 1.6176e-04,
        4.5697e-04, 1.8994e-04, 7.4990e-05, 9.5033e-05, 9.4127e-05, 5.5241e-04,
        6.3940e-05, 5.1166e-05, 8.5430e-05, 2.2791e-04, 3.6097e-04, 1.0275e-03,
        5.0502e-04, 3.3720e-04, 4.5696e-04, 3.6177e-04, 7.8180e-05, 2.7127e-04,
        3.5114e-04, 2.5893e-04, 1.7022e-04, 2.9986e-04, 3.4589e-04, 3.2005e-04,
        3.1638e-04, 2.3300e-04, 5.4394e-04, 1.8524e-04, 9.1177e-04, 5.2430e-04,
        1.3551e-04, 1.2634e-04, 7.6206e-05, 1.2928e-04, 4.2528e-04, 3.8348e-04,
        7.0151e-04, 1.9917e-04, 2.0588e-04, 1.3909e-04, 4.8881e-04, 4.8068e-04,
        2.4165e-04, 1.1843e-03, 2.0883e-04, 2.0863e-04, 1.2491e-04, 1.7951e-04,
        2.1743e-04, 1.1378e-04, 9.6332e-05, 5.8714e-04, 8.9435e-04, 7.8429e-05,
        7.1948e-05, 1.4046e-04, 1.3270e-04, 3.7904e-04, 3.3974e-04, 6.3742e-04,
        7.6417e-04, 1.1866e-04, 6.8809e-04, 8.6310e-05, 1.7217e-04, 1.2606e-04,
        1.1050e-04, 2.9065e-04, 2.7286e-04, 2.7143e-04, 1.2306e-03, 4.4224e-04,
        1.3827e-04, 5.4372e-04, 2.6249e-04, 1.0087e-04, 1.0355e-04, 1.7259e-04,
        1.1584e-04, 5.8742e-05, 4.2463e-04, 4.4790e-04, 4.6862e-04, 2.1014e-04,
        1.4108e-03, 3.4461e-05, 1.6348e-04, 2.0685e-03, 3.5081e-04, 1.8719e-04,
        1.3023e-04, 4.7983e-04, 1.7782e-04, 1.3652e-04, 1.1159e-03, 1.2815e-04,
        3.2279e-04, 3.0963e-04, 1.5199e-04, 1.6591e-04, 3.2981e-04, 2.6399e-04,
        5.2005e-05, 4.5939e-04, 2.4483e-04, 3.0518e-04, 2.3487e-04, 1.1954e-04,
        4.8615e-04, 1.8268e-04, 2.0209e-04, 1.8560e-04, 8.3022e-05, 3.9276e-04,
        1.2296e-04, 9.0415e-05, 4.5763e-04, 2.9408e-04, 1.1905e-04, 1.0830e-03,
        1.8653e-04, 7.8334e-05, 2.0160e-04, 3.7564e-04, 1.9189e-03, 4.4163e-04,
        5.4364e-04, 2.9713e-04, 3.6541e-04, 1.7233e-04, 1.0446e-04, 1.8553e-04,
        7.0259e-05, 1.7320e-04, 3.4709e-04, 3.7814e-04, 1.4184e-04, 2.6858e-04,
        5.8869e-05, 1.0857e-04, 8.4274e-05, 1.1228e-04, 3.3906e-04, 1.2267e-03,
        3.5784e-04, 1.5962e-04, 2.5540e-04, 1.4652e-03, 7.5597e-04, 4.8135e-03,
        1.2215e-02, 2.3541e-02, 1.9255e-02, 1.6958e-02, 1.0263e-02, 7.8072e-02,
        5.6227e-02, 1.0312e-02, 1.8119e-02, 6.7988e-03, 1.4151e-02, 1.4261e-02,
        3.1512e-02, 6.6336e-02, 6.9060e-02, 7.2894e-02, 7.9846e-02, 4.4743e-02,
        1.2309e-01, 0.0000e+00]),tensor([0.0000e+00, 5.9693e-04, 2.2647e-02, 5.3573e-04, 4.7362e-03, 4.0698e-03,
        2.2828e-02, 2.2536e-03, 2.4892e-04, 8.7895e-05, 9.6114e-04, 2.1132e-03,
        4.6408e-04, 2.8292e-04, 2.2627e-04, 5.8962e-04, 1.6090e-04, 3.5472e-04,
        1.8693e-04, 8.3841e-05, 6.8733e-05, 7.7609e-05, 6.0508e-05, 4.7188e-04,
        1.8148e-03, 9.0462e-04, 1.7508e-03, 1.4314e-03, 1.0324e-04, 2.5306e-04,
        2.9809e-04, 3.9192e-04, 8.1624e-05, 7.1688e-04, 4.0593e-04, 2.1005e-04,
        1.0875e-03, 3.2666e-04, 8.2403e-04, 5.2926e-04, 9.7445e-05, 4.7183e-04,
        2.4163e-04, 2.2228e-04, 2.0092e-04, 1.2255e-04, 2.2440e-04, 8.4359e-04,
        8.4777e-05, 9.3319e-05, 5.8336e-04, 3.1637e-04, 2.5827e-04, 1.8683e-04,
        1.0209e-04, 1.3196e-04, 1.3057e-04, 5.6205e-04, 1.1497e-03, 8.9678e-04,
        6.3399e-04, 1.4820e-04, 1.4251e-04, 2.9416e-04, 2.9944e-04, 1.6795e-03,
        2.7199e-03, 8.4012e-04, 3.5257e-04, 9.5824e-05, 7.7179e-05, 1.1863e-04,
        9.4477e-05, 1.4762e-03, 1.4063e-04, 2.9812e-04, 7.5034e-05, 7.4351e-05,
        1.7820e-04, 6.2898e-04, 2.5887e-03, 1.4395e-03, 2.3611e-03, 3.3300e-04,
        2.1972e-04, 2.9456e-04, 9.7722e-05, 5.5426e-04, 1.6079e-03, 1.2326e-03,
        1.0875e-03, 5.6768e-04, 1.9241e-04, 7.0924e-04, 7.6351e-05, 1.0270e-04,
        1.0813e-04, 8.5119e-05, 2.6797e-04, 9.2541e-04, 9.0058e-04, 6.9941e-04,
        1.5473e-03, 1.8087e-03, 1.1983e-03, 1.4604e-03, 1.8596e-03, 9.5773e-05,
        2.1004e-04, 1.9293e-04, 6.2530e-04, 3.2650e-04, 2.5056e-03, 6.0385e-04,
        1.0301e-03, 2.5551e-04, 3.9690e-05, 1.2918e-03, 1.2349e-04, 8.6277e-05,
        1.5201e-04, 1.2344e-03, 4.0676e-04, 1.2883e-04, 1.4850e-03, 5.7103e-04,
        3.4470e-04, 1.2019e-03, 1.1954e-03, 1.0671e-03, 1.9301e-04, 2.3542e-04,
        1.7858e-03, 1.1136e-03, 1.7777e-04, 2.2770e-04, 4.2162e-04, 1.4646e-04,
        4.4177e-04, 7.8982e-05, 2.3997e-04, 5.8945e-04, 7.9191e-04, 1.7358e-04,
        7.7278e-04, 2.3953e-04, 5.8914e-04, 5.7579e-04, 2.9937e-03, 2.4683e-03,
        2.4912e-03, 8.3097e-04, 3.2267e-04, 2.1354e-04, 1.9896e-04, 4.6693e-04,
        3.8031e-04, 3.3994e-04, 1.5821e-04, 3.4198e-04, 7.9033e-04, 1.1288e-04,
        7.1609e-04, 5.4319e-04, 2.5619e-04, 8.6957e-04, 6.5859e-04, 6.7385e-04,
        8.3713e-04, 5.2838e-04, 7.4930e-04, 3.2252e-04, 2.2587e-04, 6.1478e-04,
        2.4786e-04, 2.9528e-04, 4.8826e-04, 3.1443e-04, 6.9325e-04, 3.5684e-04,
        6.3120e-04, 1.6369e-04, 7.4813e-04, 2.0098e-04, 1.6759e-04, 3.3036e-04,
        4.0659e-04, 1.1844e-04, 5.7079e-04, 5.6002e-04, 1.7532e-03, 4.0978e-04,
        1.8978e-04, 1.7205e-04, 3.8841e-04, 2.4394e-04, 5.7824e-04, 4.4823e-04,
        3.6152e-04, 6.6332e-05, 3.7222e-04, 3.5766e-04, 6.3474e-04, 3.5720e-04,
        3.0614e-04, 3.0599e-04, 8.7614e-05, 1.2947e-04, 7.3954e-04, 7.3086e-05,
        1.3060e-04, 5.9312e-04, 8.5819e-04, 1.4565e-03, 1.1351e-03, 9.4590e-04,
        5.5653e-04, 2.2153e-04, 7.3437e-04, 2.4718e-04, 5.3931e-04, 1.6805e-04,
        5.5430e-04, 1.3016e-04, 2.3567e-04, 9.8150e-04, 1.0950e-03, 3.2431e-04,
        6.2775e-04, 2.1499e-04, 6.4049e-05, 1.7833e-04, 3.6137e-04, 6.0325e-05,
        1.1985e-03, 7.0613e-04, 1.2387e-03, 4.4416e-04, 8.1461e-04, 5.6005e-04,
        5.0254e-04, 4.5273e-04, 3.2807e-04, 3.0969e-04, 2.5954e-04, 1.2445e-04,
        1.9823e-04, 4.1197e-04, 9.8808e-05, 1.0846e-04, 1.1185e-04, 4.4922e-04,
        1.2085e-04, 4.4708e-05, 8.9021e-05, 1.6043e-04, 3.1102e-04, 8.6136e-04,
        5.5902e-04, 9.1583e-04, 6.4306e-04, 4.3155e-04, 1.0427e-04, 5.2125e-04,
        3.6981e-04, 2.9079e-04, 1.9074e-04, 3.0887e-04, 7.4862e-04, 2.2419e-04,
        7.2756e-04, 8.3405e-04, 6.6319e-04, 1.4663e-04, 4.2517e-04, 6.9969e-04,
        1.1926e-04, 1.1737e-04, 4.5641e-05, 1.5798e-04, 3.0281e-04, 1.4011e-03,
        2.8397e-03, 2.9315e-04, 2.8723e-04, 1.9283e-04, 4.7903e-04, 4.6140e-04,
        2.5968e-04, 8.4224e-04, 1.5314e-04, 1.8054e-04, 1.2559e-04, 1.8939e-04,
        1.6454e-04, 1.0637e-04, 9.6794e-05, 4.3623e-04, 3.3993e-04, 9.5644e-05,
        8.5074e-05, 1.1790e-04, 1.4837e-04, 2.7220e-04, 6.2243e-04, 6.8429e-04,
        5.2932e-04, 7.4414e-05, 4.5611e-04, 8.7306e-05, 2.2474e-04, 1.8087e-04,
        9.0210e-05, 3.5770e-04, 1.3186e-04, 2.2647e-04, 7.4589e-04, 3.1692e-04,
        1.5677e-04, 8.8654e-04, 1.4297e-04, 6.3705e-05, 1.1216e-04, 1.6954e-04,
        1.0582e-04, 5.4418e-05, 5.0904e-04, 5.1116e-04, 5.8544e-04, 2.7680e-04,
        9.3428e-04, 2.7535e-05, 1.7455e-04, 9.2755e-04, 5.6681e-04, 1.3318e-04,
        1.2676e-04, 4.5225e-04, 1.8640e-04, 8.5729e-05, 4.9409e-04, 1.4463e-04,
        5.3883e-04, 3.8729e-04, 9.5506e-05, 3.1937e-04, 3.2117e-04, 1.4456e-04,
        4.1269e-05, 5.1284e-04, 3.1135e-04, 4.0776e-04, 2.4739e-04, 1.0263e-04,
        3.7540e-04, 2.3643e-04, 2.2988e-04, 2.1033e-04, 9.0627e-05, 4.9704e-04,
        1.0379e-04, 8.9055e-05, 4.8246e-04, 4.7033e-04, 1.3136e-04, 8.3246e-04,
        2.2956e-04, 5.9252e-05, 1.8225e-04, 3.6606e-04, 1.5056e-03, 5.8092e-04,
        4.6636e-04, 3.8039e-04, 5.2132e-04, 2.0028e-04, 7.0941e-05, 1.3507e-04,
        4.4625e-05, 2.7729e-04, 4.4681e-04, 3.8243e-04, 1.2953e-04, 3.4309e-04,
        8.5492e-05, 8.3276e-05, 9.4735e-05, 1.3363e-04, 2.7379e-04, 1.2327e-03,
        2.5730e-04, 1.2274e-04, 2.6663e-04, 5.5855e-04, 5.9781e-04, 3.3433e-03,
        2.0108e-02, 2.4625e-02, 2.3767e-02, 2.1388e-02, 6.9276e-03, 2.6863e-02,
        5.4393e-02, 1.2100e-02, 2.0605e-02, 5.0308e-03, 1.5178e-02, 1.4513e-02,
        3.2578e-02, 6.7738e-02, 6.2118e-02, 6.1771e-02, 4.2208e-02, 2.0408e-02,
        9.3147e-02, 1.2580e-01, 0.0000e+00]),tensor([0.0000e+00, 5.8029e-04, 2.3704e-02, 5.7186e-04, 5.0922e-03, 3.7940e-03,
        2.5015e-02, 2.2238e-03, 3.5003e-04, 1.4214e-04, 1.5627e-03, 3.5431e-03,
        8.6810e-04, 3.0572e-04, 2.8047e-04, 4.0468e-04, 2.2382e-04, 3.9701e-04,
        2.6650e-04, 1.2135e-04, 6.1949e-05, 6.0772e-05, 6.7615e-05, 5.0533e-04,
        3.3458e-03, 5.2904e-04, 1.8180e-03, 1.2689e-03, 1.0508e-04, 3.1718e-04,
        3.6248e-04, 5.9249e-04, 1.1323e-04, 6.3235e-04, 6.3793e-04, 1.8003e-04,
        6.0163e-04, 3.6083e-04, 7.3409e-04, 5.4402e-04, 1.0008e-04, 5.5406e-04,
        3.2597e-04, 2.7662e-04, 2.6253e-04, 1.4566e-04, 2.8199e-04, 5.1637e-04,
        1.2097e-04, 9.2988e-05, 6.2147e-04, 3.7196e-04, 4.2189e-04, 2.0560e-04,
        1.3569e-04, 2.1320e-04, 1.2735e-04, 4.9804e-04, 1.6435e-03, 1.1111e-03,
        6.9290e-04, 1.7144e-04, 1.7992e-04, 3.1724e-04, 5.8279e-04, 8.8916e-04,
        4.0119e-03, 1.6085e-03, 5.5564e-04, 1.1049e-04, 9.0839e-05, 1.6846e-04,
        1.0469e-04, 2.0185e-03, 1.6686e-04, 5.1221e-04, 1.2181e-04, 1.1706e-04,
        2.5807e-04, 5.2536e-04, 3.5486e-03, 1.4546e-03, 5.1732e-03, 3.2052e-04,
        1.8889e-04, 4.2722e-04, 1.3208e-04, 1.0876e-03, 2.3855e-03, 1.5747e-03,
        2.4506e-03, 8.8245e-04, 2.3896e-04, 1.4231e-03, 6.9685e-05, 9.0421e-05,
        1.1391e-04, 8.1134e-05, 2.7928e-04, 4.8032e-04, 7.8325e-04, 7.4082e-04,
        2.9943e-03, 2.5036e-03, 3.0869e-03, 9.5058e-04, 9.7454e-04, 1.0301e-04,
        3.1739e-04, 2.2123e-04, 1.0676e-03, 4.1830e-04, 2.8014e-03, 9.6429e-04,
        7.1934e-04, 1.4728e-04, 3.4125e-05, 6.3995e-04, 1.1252e-04, 9.5665e-05,
        1.8703e-04, 1.0416e-03, 3.7951e-04, 2.0878e-04, 9.8918e-04, 7.6240e-04,
        3.1037e-04, 1.7575e-03, 1.9203e-03, 6.0975e-04, 2.7211e-04, 3.8895e-04,
        2.6088e-03, 2.3905e-03, 1.8715e-04, 4.6478e-04, 9.2714e-04, 1.5862e-04,
        3.3641e-04, 8.4282e-05, 2.5894e-04, 6.1624e-04, 6.4987e-04, 2.0508e-04,
        7.2227e-04, 1.9606e-04, 6.2840e-04, 6.6039e-04, 3.9175e-03, 1.2833e-03,
        3.3800e-03, 5.2009e-04, 2.8087e-04, 2.2137e-04, 2.6443e-04, 3.3216e-04,
        6.2264e-04, 4.0777e-04, 2.9624e-04, 4.6377e-04, 4.4237e-04, 1.1182e-04,
        3.5131e-04, 4.9147e-04, 2.9217e-04, 6.4730e-04, 4.0664e-04, 7.6334e-04,
        1.5246e-03, 6.6334e-04, 4.9710e-04, 2.4438e-04, 2.0954e-04, 3.7114e-04,
        2.4273e-04, 3.6406e-04, 3.7228e-04, 2.7726e-04, 4.5240e-04, 5.4565e-04,
        3.7803e-04, 1.7722e-04, 5.4165e-04, 2.7953e-04, 2.5511e-04, 2.3110e-04,
        8.3725e-04, 1.2492e-04, 6.2998e-04, 4.4515e-04, 2.5187e-03, 3.3833e-04,
        3.2170e-04, 1.8375e-04, 2.6676e-04, 2.8464e-04, 1.0461e-03, 3.3772e-04,
        3.3487e-04, 6.5122e-05, 3.3615e-04, 7.5416e-04, 1.2051e-03, 2.6776e-04,
        5.2692e-04, 6.2036e-04, 8.6595e-05, 1.9463e-04, 1.4665e-03, 7.7044e-05,
        2.0161e-04, 3.2791e-04, 6.1717e-04, 3.1338e-03, 6.2164e-04, 1.0763e-03,
        3.7689e-04, 1.9152e-04, 1.1623e-03, 1.9281e-04, 3.9260e-04, 2.5595e-04,
        4.4666e-04, 1.2346e-04, 2.9269e-04, 1.5800e-03, 7.5963e-04, 3.2974e-04,
        1.6600e-03, 3.4044e-04, 8.3801e-05, 1.7364e-04, 3.0067e-04, 7.5269e-05,
        8.3457e-04, 9.0119e-04, 1.5623e-03, 3.5418e-04, 6.0872e-04, 6.0032e-04,
        6.4152e-04, 4.5665e-04, 3.1265e-04, 2.3279e-04, 2.6441e-04, 1.2684e-04,
        1.3473e-04, 7.5920e-04, 1.0407e-04, 1.4595e-04, 1.2149e-04, 3.5993e-04,
        1.6648e-04, 4.1439e-05, 1.0382e-04, 1.4025e-04, 2.9884e-04, 6.2513e-04,
        3.6457e-04, 1.2780e-03, 4.2139e-04, 2.7428e-04, 1.0665e-04, 6.9383e-04,
        2.3917e-04, 3.4543e-04, 2.0566e-04, 3.2528e-04, 9.2002e-04, 1.8538e-04,
        1.0862e-03, 2.0429e-03, 4.4215e-04, 1.4517e-04, 2.8877e-04, 4.9045e-04,
        1.2167e-04, 1.0990e-04, 3.9678e-05, 2.0137e-04, 2.0245e-04, 1.8789e-03,
        5.2510e-03, 2.3836e-04, 2.2712e-04, 1.6457e-04, 2.7201e-04, 2.1761e-04,
        2.8171e-04, 6.4293e-04, 1.5490e-04, 2.2158e-04, 1.1756e-04, 2.5144e-04,
        1.5535e-04, 1.3682e-04, 9.9378e-05, 2.3252e-04, 3.5873e-04, 1.0505e-04,
        8.0935e-05, 1.4116e-04, 2.1357e-04, 2.0153e-04, 5.0479e-04, 5.9041e-04,
        2.7580e-04, 6.2444e-05, 3.4663e-04, 8.7285e-05, 2.3017e-04, 2.6979e-04,
        7.2395e-05, 1.8278e-04, 1.0571e-04, 3.4101e-04, 5.1783e-04, 2.8073e-04,
        1.4727e-04, 1.5605e-03, 1.0457e-04, 8.0415e-05, 8.5001e-05, 1.1629e-04,
        1.1389e-04, 6.6582e-05, 5.7877e-04, 3.1750e-04, 4.5262e-04, 3.2546e-04,
        1.0546e-03, 3.8150e-05, 2.2898e-04, 1.2328e-03, 8.3327e-04, 1.2530e-04,
        1.0670e-04, 3.9112e-04, 1.5971e-04, 7.9294e-05, 7.9230e-04, 2.0218e-04,
        6.7912e-04, 3.4703e-04, 1.0338e-04, 4.5805e-04, 2.7381e-04, 1.1500e-04,
        4.7447e-05, 5.9206e-04, 2.6242e-04, 3.0111e-04, 3.4504e-04, 1.7701e-04,
        2.6078e-04, 3.0580e-04, 2.4106e-04, 1.7937e-04, 1.2744e-04, 2.3285e-04,
        1.3748e-04, 1.4276e-04, 2.5709e-04, 4.9023e-04, 1.5757e-04, 8.9405e-04,
        2.3431e-04, 7.6940e-05, 2.5261e-04, 5.6950e-04, 1.3634e-03, 4.6902e-04,
        6.2157e-04, 2.7901e-04, 4.2769e-04, 2.3858e-04, 7.7094e-05, 1.1824e-04,
        6.3633e-05, 4.1168e-04, 6.5679e-04, 4.8840e-04, 1.1856e-04, 1.8560e-04,
        1.1759e-04, 8.1303e-05, 9.3594e-05, 1.9174e-04, 2.1427e-04, 4.9653e-04,
        1.7976e-04, 9.0349e-05, 2.7227e-04, 5.1195e-04, 7.2869e-04, 3.0969e-03,
        1.2214e-02, 2.1946e-02, 3.4249e-02, 1.4290e-02, 5.8906e-03, 2.1785e-02,
        3.2364e-02, 9.2108e-03, 1.3527e-02, 3.9889e-03, 9.9190e-03, 1.3239e-02,
        2.2894e-02, 4.6142e-02, 4.7495e-02, 6.3423e-02, 2.8093e-02, 1.4253e-02,
        5.9657e-02, 1.1686e-01, 1.2694e-01, 0.0000e+00]),tensor([0.0000e+00, 7.2978e-04, 3.2048e-02, 8.9254e-04, 5.3547e-03, 4.9462e-03,
        3.5940e-02, 1.8575e-03, 4.8233e-04, 3.2828e-04, 1.8475e-03, 5.8225e-03,
        1.5733e-03, 5.4251e-04, 3.6338e-04, 4.5019e-04, 3.4835e-04, 5.5556e-04,
        5.2081e-04, 2.2522e-04, 5.5971e-05, 4.5263e-05, 6.0443e-05, 7.2481e-04,
        5.7321e-03, 6.0221e-04, 1.7327e-03, 1.0919e-03, 1.1272e-04, 6.3966e-04,
        5.0810e-04, 1.0421e-03, 2.4813e-04, 5.6569e-04, 1.0525e-03, 1.9118e-04,
        4.9261e-04, 3.5904e-04, 3.8159e-04, 5.2128e-04, 1.0407e-04, 3.5173e-04,
        2.9223e-04, 2.3334e-04, 4.2234e-04, 1.1967e-04, 6.2437e-04, 3.9172e-04,
        1.7902e-04, 8.5294e-05, 6.4121e-04, 3.6424e-04, 8.5139e-04, 2.8661e-04,
        2.3064e-04, 5.0970e-04, 1.3502e-04, 4.2411e-04, 1.7815e-03, 1.3681e-03,
        8.4009e-04, 2.1886e-04, 2.4337e-04, 2.4244e-04, 6.5627e-04, 6.7893e-04,
        7.1380e-03, 2.0541e-03, 3.7035e-04, 1.0259e-04, 1.0249e-04, 2.8666e-04,
        1.3098e-04, 1.8667e-03, 1.9078e-04, 1.1455e-03, 3.2638e-04, 2.1778e-04,
        4.1928e-04, 3.8581e-04, 3.5217e-03, 1.0555e-03, 5.7421e-03, 4.1170e-04,
        1.6571e-04, 8.6000e-04, 2.3838e-04, 1.5453e-03, 3.3279e-03, 2.9476e-03,
        4.2608e-03, 8.2396e-04, 2.4424e-04, 2.0881e-03, 8.9599e-05, 1.0015e-04,
        1.1916e-04, 8.1412e-05, 2.9064e-04, 2.8600e-04, 5.6576e-04, 5.7930e-04,
        4.4145e-03, 3.8447e-03, 4.0939e-03, 7.5404e-04, 7.0048e-04, 1.1920e-04,
        6.8446e-04, 2.3950e-04, 1.1049e-03, 7.0015e-04, 6.0652e-03, 1.1606e-03,
        5.3136e-04, 1.1707e-04, 3.7942e-05, 6.2373e-04, 1.2856e-04, 1.1911e-04,
        2.4562e-04, 4.7989e-04, 4.3675e-04, 3.0718e-04, 7.6364e-04, 7.2344e-04,
        1.9368e-04, 1.8017e-03, 2.8302e-03, 6.8690e-04, 4.7354e-04, 9.3076e-04,
        5.7165e-03, 4.2797e-03, 3.4838e-04, 6.1982e-04, 1.2071e-03, 1.6187e-04,
        3.6491e-04, 1.0883e-04, 3.4439e-04, 6.1216e-04, 5.5634e-04, 3.0747e-04,
        6.7692e-04, 3.2898e-04, 3.5448e-04, 4.9104e-04, 5.5980e-03, 8.4780e-04,
        1.7649e-03, 6.6944e-04, 3.7602e-04, 2.0885e-04, 3.1135e-04, 2.5876e-04,
        6.4730e-04, 2.8429e-04, 6.8534e-04, 7.3370e-04, 3.2868e-04, 1.3573e-04,
        2.5354e-04, 2.6958e-04, 5.4167e-04, 3.8731e-04, 1.9442e-04, 3.1289e-04,
        1.9778e-03, 3.7377e-04, 3.9669e-04, 3.2970e-04, 2.7308e-04, 2.7129e-04,
        5.9479e-04, 5.8082e-04, 2.9518e-04, 2.1042e-04, 3.3682e-04, 9.2516e-04,
        1.8121e-04, 2.8446e-04, 3.7937e-04, 4.2624e-04, 2.8975e-04, 2.0427e-04,
        1.5547e-03, 1.5511e-04, 4.6689e-04, 3.4690e-04, 3.9545e-03, 3.8174e-04,
        6.3805e-04, 2.0168e-04, 2.3254e-04, 4.9513e-04, 1.1869e-03, 3.9207e-04,
        3.5649e-04, 7.6367e-05, 3.5545e-04, 1.9019e-03, 1.1715e-03, 1.7344e-04,
        1.6341e-03, 7.5392e-04, 1.0760e-04, 4.2226e-04, 1.8053e-03, 1.4591e-04,
        3.0049e-04, 1.6444e-04, 5.3070e-04, 5.9401e-03, 5.3796e-04, 1.2870e-03,
        2.2265e-04, 2.3548e-04, 1.2886e-03, 2.6018e-04, 3.1590e-04, 4.1083e-04,
        2.6333e-04, 1.6958e-04, 4.3184e-04, 1.5110e-03, 6.8272e-04, 4.7987e-04,
        3.4902e-03, 1.0888e-03, 1.9230e-04, 2.7947e-04, 2.3638e-04, 1.2759e-04,
        4.3281e-04, 6.6140e-04, 3.8085e-03, 3.0140e-04, 3.9779e-04, 1.2252e-03,
        9.3506e-04, 3.8142e-04, 1.8075e-04, 1.4913e-04, 3.4521e-04, 1.4279e-04,
        1.0931e-04, 4.9133e-03, 1.6999e-04, 2.6751e-04, 2.1553e-04, 3.9639e-04,
        4.7490e-04, 6.0877e-05, 1.6013e-04, 1.1655e-04, 4.5233e-04, 3.6777e-04,
        2.1374e-04, 1.9445e-03, 2.5891e-04, 2.4193e-04, 1.7770e-04, 9.8809e-04,
        2.5076e-04, 2.9746e-04, 3.1875e-04, 5.2244e-04, 1.3762e-03, 1.9045e-04,
        3.1363e-03, 3.8046e-03, 4.3697e-04, 1.8040e-04, 1.8343e-04, 3.7345e-04,
        2.2100e-04, 1.7834e-04, 3.8341e-05, 3.6835e-04, 1.4084e-04, 3.2123e-03,
        1.3053e-02, 2.4449e-04, 2.1151e-04, 2.0551e-04, 2.0689e-04, 1.5436e-04,
        5.5526e-04, 3.9608e-04, 2.2782e-04, 4.1353e-04, 1.4068e-04, 4.6921e-04,
        1.7849e-04, 2.6403e-04, 1.4730e-04, 1.9438e-04, 2.3982e-04, 2.0319e-04,
        1.1343e-04, 2.1443e-04, 3.3495e-04, 1.9036e-04, 5.4452e-04, 6.7544e-04,
        2.6043e-04, 5.4926e-05, 1.7554e-04, 1.0817e-04, 2.1727e-04, 5.0157e-04,
        9.1394e-05, 1.8084e-04, 9.3647e-05, 3.1626e-04, 3.6331e-04, 2.9527e-04,
        2.4974e-04, 1.9447e-03, 9.8596e-05, 1.2402e-04, 8.1181e-05, 7.9469e-05,
        1.5882e-04, 1.2456e-04, 6.2964e-04, 2.7133e-04, 3.1773e-04, 5.6084e-04,
        6.1367e-04, 7.2954e-05, 4.2575e-04, 6.8146e-04, 8.0474e-04, 1.1226e-04,
        1.5863e-04, 3.0533e-04, 1.8534e-04, 8.3612e-05, 5.0799e-04, 4.5051e-04,
        4.3929e-04, 2.6917e-04, 1.5993e-04, 3.3312e-04, 2.4369e-04, 7.9917e-05,
        6.1298e-05, 4.8541e-04, 3.2164e-04, 3.7566e-04, 4.9948e-04, 3.5073e-04,
        2.4774e-04, 3.4533e-04, 1.7673e-04, 2.0080e-04, 2.7490e-04, 1.9108e-04,
        2.0519e-04, 2.5944e-04, 1.6803e-04, 5.8035e-04, 1.9426e-04, 5.4312e-04,
        2.1386e-04, 1.1955e-04, 2.9697e-04, 3.5882e-04, 1.0298e-03, 2.8023e-04,
        4.4389e-04, 3.1080e-04, 3.6277e-04, 2.9259e-04, 1.0707e-04, 1.1113e-04,
        1.0036e-04, 6.7906e-04, 5.8888e-04, 5.4969e-04, 1.2506e-04, 1.0360e-04,
        1.2601e-04, 7.7282e-05, 1.6626e-04, 1.9163e-04, 2.5319e-04, 2.7737e-04,
        1.4584e-04, 9.0506e-05, 1.9169e-04, 2.9509e-04, 4.2229e-04, 3.7445e-03,
        6.9048e-03, 2.2529e-02, 3.6315e-02, 6.7420e-03, 3.9564e-03, 1.4272e-02,
        1.4846e-02, 5.2658e-03, 8.2982e-03, 3.4920e-03, 7.2903e-03, 8.6175e-03,
        1.3708e-02, 2.6260e-02, 2.5849e-02, 4.3048e-02, 1.2934e-02, 7.1001e-03,
        2.5195e-02, 4.6229e-02, 1.0532e-01, 1.8813e-01, 0.0000e+00]),tensor([0.0000e+00, 5.8700e-04, 2.5324e-02, 5.4487e-04, 3.6975e-03, 3.8973e-03,
        2.5402e-02, 1.5034e-03, 3.2061e-04, 1.5535e-04, 1.1639e-03, 4.4474e-03,
        7.9467e-04, 4.5067e-04, 2.7771e-04, 3.9817e-04, 2.2373e-04, 5.1167e-04,
        3.2362e-04, 1.3275e-04, 5.1165e-05, 4.3091e-05, 5.2871e-05, 5.5168e-04,
        4.2311e-03, 4.5041e-04, 1.4315e-03, 7.6311e-04, 8.5427e-05, 4.3179e-04,
        3.5560e-04, 5.7920e-04, 1.3612e-04, 3.6003e-04, 4.3948e-04, 1.6675e-04,
        3.2642e-04, 2.1688e-04, 2.3705e-04, 2.8517e-04, 8.6768e-05, 2.8994e-04,
        2.4141e-04, 2.4002e-04, 2.8591e-04, 1.1757e-04, 4.8704e-04, 3.4665e-04,
        1.2418e-04, 7.3638e-05, 4.4577e-04, 2.2886e-04, 3.9037e-04, 2.0545e-04,
        1.5835e-04, 2.2633e-04, 1.5406e-04, 3.5785e-04, 1.1364e-03, 8.6917e-04,
        5.3028e-04, 1.7317e-04, 2.1120e-04, 1.8980e-04, 5.2123e-04, 7.2058e-04,
        5.4013e-03, 1.3546e-03, 3.2281e-04, 9.8902e-05, 8.3817e-05, 2.6671e-04,
        1.1301e-04, 1.2805e-03, 1.2297e-04, 5.3079e-04, 1.8824e-04, 1.3586e-04,
        2.6471e-04, 2.9308e-04, 2.3082e-03, 7.8099e-04, 3.1760e-03, 2.6052e-04,
        1.4655e-04, 4.6281e-04, 2.0307e-04, 1.0632e-03, 2.0449e-03, 1.9888e-03,
        3.0205e-03, 8.4136e-04, 3.1082e-04, 1.5143e-03, 9.8294e-05, 8.2151e-05,
        1.0224e-04, 6.2566e-05, 1.8886e-04, 3.6061e-04, 5.0540e-04, 5.9994e-04,
        3.1605e-03, 2.5957e-03, 2.2262e-03, 5.0605e-04, 7.4121e-04, 1.1378e-04,
        3.6002e-04, 2.0493e-04, 1.0228e-03, 8.4359e-04, 4.5074e-03, 9.6245e-04,
        5.3256e-04, 1.1619e-04, 3.7762e-05, 6.6708e-04, 9.3910e-05, 8.5134e-05,
        1.9545e-04, 5.4771e-04, 3.5919e-04, 1.9997e-04, 7.1748e-04, 5.9692e-04,
        1.7430e-04, 1.7695e-03, 3.0807e-03, 4.9774e-04, 2.6671e-04, 4.6756e-04,
        6.8516e-03, 2.3563e-03, 2.0997e-04, 6.2822e-04, 1.2792e-03, 2.0555e-04,
        3.7181e-04, 9.8805e-05, 3.3327e-04, 4.8443e-04, 3.7725e-04, 2.0516e-04,
        5.7678e-04, 2.0651e-04, 3.5381e-04, 6.3604e-04, 3.6278e-03, 6.9662e-04,
        1.7890e-03, 4.0704e-04, 2.8185e-04, 1.7145e-04, 2.5414e-04, 2.3777e-04,
        7.3069e-04, 3.7814e-04, 3.6990e-04, 7.3000e-04, 3.1910e-04, 1.2025e-04,
        2.4441e-04, 2.6551e-04, 4.4321e-04, 3.8308e-04, 1.6562e-04, 3.8243e-04,
        1.3991e-03, 3.5286e-04, 3.0332e-04, 2.4504e-04, 2.0325e-04, 1.9853e-04,
        3.8686e-04, 4.8428e-04, 2.1284e-04, 1.5316e-04, 2.9330e-04, 5.7369e-04,
        2.1470e-04, 1.6205e-04, 3.6459e-04, 2.6532e-04, 2.8743e-04, 2.1854e-04,
        8.9823e-04, 1.3023e-04, 4.3643e-04, 2.7456e-04, 2.7602e-03, 2.6729e-04,
        3.2900e-04, 1.5525e-04, 1.9425e-04, 4.1444e-04, 9.0142e-04, 2.2779e-04,
        3.1023e-04, 8.6922e-05, 2.1818e-04, 1.0013e-03, 8.6580e-04, 1.4973e-04,
        1.0193e-03, 7.6801e-04, 8.9015e-05, 2.5379e-04, 1.1830e-03, 9.3232e-05,
        2.3152e-04, 1.3761e-04, 4.2042e-04, 3.4913e-03, 3.8439e-04, 1.1416e-03,
        2.0013e-04, 1.7694e-04, 9.0871e-04, 1.9580e-04, 2.3146e-04, 3.4083e-04,
        2.6241e-04, 1.2826e-04, 3.0678e-04, 1.5082e-03, 5.7634e-04, 2.9200e-04,
        2.0506e-03, 1.3195e-03, 1.4753e-04, 2.5094e-04, 2.1401e-04, 1.1270e-04,
        3.9769e-04, 7.1251e-04, 4.2304e-03, 2.2635e-04, 3.9558e-04, 8.0519e-04,
        7.3015e-04, 4.3963e-04, 2.0956e-04, 1.4436e-04, 3.5083e-04, 1.4892e-04,
        7.9091e-05, 3.4323e-03, 1.6085e-04, 2.1834e-04, 2.3135e-04, 3.2769e-04,
        9.1762e-04, 5.0814e-05, 1.3598e-04, 1.5207e-04, 3.5421e-04, 3.4345e-04,
        1.9469e-04, 2.0434e-03, 3.2800e-04, 1.9249e-04, 1.4478e-04, 7.2088e-04,
        1.7806e-04, 2.2135e-04, 2.5775e-04, 4.0951e-04, 9.6435e-04, 1.6275e-04,
        5.2445e-03, 2.6786e-03, 3.4503e-04, 1.7906e-04, 2.1813e-04, 3.9107e-04,
        2.4114e-04, 1.6136e-04, 3.9489e-05, 2.3949e-04, 1.1147e-04, 4.1225e-03,
        7.0544e-03, 2.3299e-04, 2.5815e-04, 1.6322e-04, 2.3717e-04, 1.6239e-04,
        3.6815e-04, 3.2384e-04, 1.8505e-04, 3.9883e-04, 1.0281e-04, 2.9075e-04,
        1.2968e-04, 2.0095e-04, 1.4218e-04, 1.8748e-04, 2.7056e-04, 1.5134e-04,
        8.1434e-05, 1.7616e-04, 2.1193e-04, 1.6925e-04, 5.7646e-04, 5.1106e-04,
        1.9635e-04, 5.0596e-05, 1.6885e-04, 8.5044e-05, 1.7488e-04, 2.9131e-04,
        7.6794e-05, 2.3540e-04, 9.6468e-05, 2.2612e-04, 2.8577e-04, 1.9193e-04,
        1.8676e-04, 1.2103e-03, 8.7405e-05, 9.1073e-05, 7.2815e-05, 7.1142e-05,
        1.9839e-04, 9.5557e-05, 4.4340e-04, 2.5489e-04, 2.9035e-04, 3.3175e-04,
        5.5275e-04, 4.9836e-05, 3.2355e-04, 5.3539e-04, 8.2202e-04, 1.2297e-04,
        1.2729e-04, 2.6575e-04, 1.4388e-04, 9.3536e-05, 5.3339e-04, 3.0624e-04,
        5.0186e-04, 2.5777e-04, 1.2917e-04, 3.3091e-04, 1.9706e-04, 7.6847e-05,
        4.9676e-05, 3.3166e-04, 2.4856e-04, 2.8363e-04, 2.8585e-04, 1.7720e-04,
        1.9109e-04, 2.5642e-04, 1.4346e-04, 1.3652e-04, 1.8813e-04, 1.7389e-04,
        1.2564e-04, 1.4350e-04, 1.5161e-04, 4.5642e-04, 1.4046e-04, 3.9809e-04,
        2.0166e-04, 1.0429e-04, 2.4152e-04, 4.4321e-04, 8.9175e-04, 2.4384e-04,
        3.3205e-04, 2.3338e-04, 2.9393e-04, 2.1523e-04, 6.6598e-05, 9.2948e-05,
        5.8719e-05, 4.2741e-04, 4.2938e-04, 5.9072e-04, 1.2067e-04, 9.6547e-05,
        8.9108e-05, 7.7645e-05, 1.1400e-04, 1.7083e-04, 1.4661e-04, 2.5489e-04,
        1.2451e-04, 7.4220e-05, 1.7395e-04, 3.6074e-04, 4.7535e-04, 2.6067e-03,
        5.9086e-03, 1.8470e-02, 2.6302e-02, 6.2804e-03, 3.7625e-03, 1.2251e-02,
        1.6212e-02, 4.6255e-03, 7.2224e-03, 2.8809e-03, 6.0669e-03, 8.6736e-03,
        1.1559e-02, 2.4213e-02, 2.3510e-02, 3.2653e-02, 9.0895e-03, 5.9759e-03,
        2.0512e-02, 3.8580e-02, 8.2316e-02, 1.8781e-01, 1.6319e-01, 0.0000e+00]),tensor([0.0000e+00, 4.7308e-04, 2.3106e-02, 4.6992e-04, 2.8572e-03, 2.9065e-03,
        2.3495e-02, 1.2624e-03, 2.8598e-04, 1.4751e-04, 1.4550e-03, 2.6541e-03,
        1.0865e-03, 4.1310e-04, 1.8012e-04, 1.1019e-03, 2.0888e-04, 4.4280e-04,
        2.9072e-04, 9.8998e-05, 4.6782e-05, 4.8517e-05, 4.4172e-05, 5.6738e-04,
        3.1423e-03, 8.2146e-04, 1.0530e-03, 8.8496e-04, 7.0214e-05, 4.4790e-04,
        2.5537e-04, 6.5254e-04, 1.4817e-04, 1.1129e-03, 8.2921e-04, 3.2530e-04,
        1.3704e-03, 2.4010e-04, 4.7625e-04, 6.2229e-04, 7.8616e-05, 4.8971e-04,
        1.7405e-04, 1.5479e-04, 1.7772e-04, 1.5961e-04, 4.1691e-04, 1.5231e-03,
        8.7247e-05, 6.2248e-05, 3.7809e-04, 3.8601e-04, 5.1313e-04, 2.4363e-04,
        1.1081e-04, 2.7328e-04, 8.4113e-05, 5.6823e-04, 1.6953e-03, 1.1610e-03,
        5.7197e-04, 1.7880e-04, 9.4751e-05, 1.9964e-04, 5.3663e-04, 2.4084e-03,
        7.6450e-04, 7.6898e-04, 3.5658e-04, 9.0569e-05, 8.8003e-05, 1.7075e-04,
        1.1720e-04, 2.0548e-03, 9.6955e-05, 6.4211e-04, 9.7108e-05, 1.0729e-04,
        2.1712e-04, 5.7525e-04, 1.2067e-03, 1.3778e-03, 1.6109e-03, 4.5151e-04,
        1.5462e-04, 4.8442e-04, 8.1274e-05, 1.0848e-03, 5.3536e-04, 3.8827e-04,
        2.2643e-03, 1.3459e-03, 1.1266e-04, 1.6738e-03, 8.7638e-05, 8.3257e-05,
        8.4977e-05, 4.8000e-05, 4.2324e-04, 1.2060e-03, 4.7332e-04, 6.4307e-04,
        2.6234e-03, 5.0280e-04, 1.1058e-03, 2.0810e-03, 3.8171e-03, 1.0583e-04,
        3.7841e-04, 1.5507e-04, 1.5406e-03, 1.6447e-04, 4.3146e-04, 4.4626e-04,
        8.5060e-04, 2.0071e-04, 2.7992e-05, 3.2821e-03, 8.2246e-05, 6.2864e-05,
        1.6815e-04, 1.1789e-03, 2.0644e-04, 1.5772e-04, 2.4872e-03, 4.4134e-04,
        3.2414e-04, 8.0875e-04, 8.9019e-04, 2.2718e-03, 3.8822e-04, 4.7233e-04,
        6.2441e-04, 5.5346e-04, 2.0690e-04, 2.0970e-04, 1.1728e-03, 8.7939e-05,
        7.5492e-04, 7.4918e-05, 2.4941e-04, 4.9785e-04, 8.9125e-04, 2.2197e-04,
        3.1471e-04, 2.1449e-04, 4.0319e-04, 6.4160e-04, 8.8962e-04, 3.3946e-03,
        2.9987e-03, 1.8953e-03, 4.6085e-04, 2.3078e-04, 1.5176e-04, 2.8349e-04,
        8.6899e-04, 3.7693e-04, 3.9720e-04, 2.0176e-04, 9.7587e-04, 1.5863e-04,
        1.1377e-03, 5.8494e-04, 4.8550e-04, 7.4442e-04, 3.9395e-04, 4.8206e-04,
        1.3859e-03, 6.0578e-04, 8.3268e-04, 4.4801e-04, 3.4225e-04, 9.9119e-04,
        6.0041e-04, 4.0804e-04, 3.8467e-04, 1.9047e-04, 6.3097e-04, 6.6185e-04,
        4.7146e-04, 2.2463e-04, 7.9628e-04, 2.8594e-04, 1.6151e-04, 2.8993e-04,
        1.3551e-03, 1.1223e-04, 3.5923e-04, 2.9117e-04, 7.0674e-04, 6.6318e-04,
        3.7108e-04, 1.8737e-04, 6.5147e-04, 4.3049e-04, 1.2400e-03, 8.1791e-04,
        3.4810e-04, 5.0930e-05, 4.7940e-04, 6.3320e-04, 7.2651e-04, 3.3460e-04,
        2.7548e-04, 7.9220e-04, 9.0909e-05, 1.9209e-04, 2.0439e-03, 1.4890e-04,
        1.4131e-04, 3.3187e-04, 3.8738e-04, 1.1436e-03, 1.9459e-03, 4.9180e-04,
        7.1477e-04, 3.9968e-04, 7.4993e-04, 3.3619e-04, 9.4681e-04, 2.6759e-04,
        5.4265e-04, 8.6065e-05, 4.0453e-04, 5.7000e-04, 1.4008e-03, 4.4107e-04,
        5.4674e-04, 1.5349e-04, 1.0126e-04, 2.9283e-04, 4.0324e-04, 9.3244e-05,
        5.3578e-04, 6.2757e-04, 4.6361e-04, 5.5996e-04, 8.1900e-04, 9.6496e-04,
        3.9529e-04, 3.1781e-04, 2.6668e-04, 3.7335e-04, 4.3528e-04, 1.0278e-04,
        1.1530e-04, 7.7013e-04, 7.5263e-05, 1.5855e-04, 1.6183e-04, 7.0770e-04,
        1.0211e-04, 4.8082e-05, 7.9044e-05, 1.2242e-04, 4.8442e-04, 8.4170e-04,
        5.5521e-04, 5.8919e-04, 9.7329e-04, 7.4278e-04, 1.5599e-04, 5.3542e-04,
        7.1475e-04, 2.6200e-04, 2.6157e-04, 4.9837e-04, 6.4170e-04, 2.7242e-04,
        3.8476e-04, 6.4888e-04, 8.7224e-04, 1.5064e-04, 4.8384e-04, 6.7052e-04,
        1.9353e-04, 1.8057e-04, 2.8805e-05, 1.7360e-04, 2.0310e-04, 5.1643e-04,
        9.1578e-04, 2.0376e-04, 2.5088e-04, 2.5331e-04, 7.8008e-04, 7.7073e-04,
        4.7815e-04, 1.0001e-03, 2.9515e-04, 2.8745e-04, 8.9188e-05, 2.5290e-04,
        1.2274e-04, 1.1970e-04, 8.5439e-05, 8.5052e-04, 3.9975e-04, 9.7796e-05,
        6.5997e-05, 1.3289e-04, 1.9675e-04, 4.0723e-04, 3.5462e-04, 1.0397e-03,
        1.1549e-03, 6.6289e-05, 4.7628e-04, 1.0111e-04, 2.2290e-04, 2.9327e-04,
        1.3711e-04, 5.1053e-04, 9.4947e-05, 2.6973e-04, 9.9037e-04, 3.8767e-04,
        1.9265e-04, 2.0281e-03, 2.1085e-04, 1.1656e-04, 6.1608e-05, 8.4912e-05,
        1.0932e-04, 8.4714e-05, 4.6472e-04, 4.5443e-04, 6.6821e-04, 3.2778e-04,
        9.2302e-04, 4.7068e-05, 2.1541e-04, 7.9696e-04, 5.2209e-04, 1.4619e-04,
        2.1228e-04, 4.3561e-04, 2.3485e-04, 7.1473e-05, 7.5037e-04, 1.6956e-04,
        4.3798e-04, 4.9390e-04, 1.6011e-04, 3.5915e-04, 4.0806e-04, 1.4511e-04,
        3.2986e-05, 2.6545e-04, 3.5805e-04, 4.4296e-04, 4.1254e-04, 2.0316e-04,
        6.6048e-04, 3.2325e-04, 2.8025e-04, 1.3801e-04, 1.0807e-04, 7.6810e-04,
        1.6739e-04, 1.5237e-04, 3.8361e-04, 3.4590e-04, 1.6500e-04, 5.9248e-04,
        3.2498e-04, 5.6993e-05, 2.2589e-04, 3.6292e-04, 9.7805e-04, 4.3533e-04,
        2.8909e-04, 4.1932e-04, 5.0445e-04, 3.0395e-04, 1.1389e-04, 1.0119e-04,
        6.7646e-05, 2.6343e-04, 8.7966e-04, 4.9434e-04, 1.2409e-04, 1.8972e-04,
        6.3846e-05, 5.9662e-05, 1.0544e-04, 1.7264e-04, 2.5168e-04, 1.5337e-03,
        2.5518e-04, 1.3506e-04, 1.9757e-04, 4.5517e-04, 4.5387e-04, 3.0232e-03,
        6.1448e-03, 1.7371e-02, 2.8280e-02, 1.0536e-02, 4.7549e-03, 2.0307e-02,
        2.1830e-02, 5.1086e-03, 7.8630e-03, 3.6716e-03, 6.5033e-03, 8.9638e-03,
        1.5841e-02, 3.2612e-02, 2.3957e-02, 4.0878e-02, 1.5505e-02, 7.6802e-03,
        3.4136e-02, 5.5514e-02, 5.7958e-02, 7.4644e-02, 1.2484e-01, 1.1215e-01,
        0.0000e+00]),tensor([0.0000e+00, 5.1638e-04, 2.9503e-02, 5.8240e-04, 3.8813e-03, 5.0845e-03,
        3.4423e-02, 1.5459e-03, 1.2557e-03, 2.6576e-04, 1.3130e-03, 2.6107e-03,
        6.6788e-04, 4.3990e-04, 2.7585e-04, 7.2658e-04, 6.0051e-04, 4.7810e-04,
        2.9425e-04, 1.9653e-04, 6.7922e-05, 6.2385e-05, 6.4206e-05, 5.8088e-04,
        2.3212e-03, 7.0434e-04, 1.3903e-03, 1.1675e-03, 9.8309e-05, 3.9460e-04,
        3.3701e-04, 1.7001e-03, 1.2497e-04, 9.9842e-04, 6.6896e-04, 6.9654e-04,
        1.0926e-03, 6.4087e-04, 5.3664e-04, 8.2522e-04, 1.0811e-04, 4.1953e-04,
        1.9996e-04, 1.9397e-04, 1.9051e-04, 1.5353e-04, 3.6498e-04, 7.2296e-04,
        1.0664e-04, 1.0216e-04, 5.8706e-04, 1.2981e-03, 1.4553e-03, 4.4985e-04,
        5.4120e-04, 8.5992e-04, 4.2940e-04, 8.9035e-04, 3.0562e-02, 2.4323e-03,
        2.8850e-03, 2.3021e-04, 4.1832e-04, 7.1763e-04, 3.7831e-04, 1.0803e-03,
        6.6010e-04, 3.7038e-04, 3.7497e-04, 1.1387e-04, 9.4857e-05, 1.8061e-04,
        1.3983e-04, 1.5986e-03, 5.2431e-04, 1.7441e-03, 4.2346e-04, 4.1856e-04,
        9.9372e-04, 7.4473e-04, 9.1241e-04, 1.4495e-03, 1.6461e-03, 1.7605e-03,
        4.7570e-04, 1.4541e-03, 3.5617e-04, 8.3472e-04, 4.3500e-04, 3.4837e-04,
        1.2917e-03, 6.1705e-04, 2.8915e-04, 7.8281e-04, 2.0328e-04, 1.5959e-04,
        1.4887e-04, 2.3534e-04, 8.8561e-04, 5.9728e-04, 4.8879e-04, 4.5367e-04,
        1.9392e-03, 6.2338e-04, 7.5240e-04, 2.1178e-03, 1.7425e-03, 1.6215e-04,
        1.1571e-03, 1.3575e-04, 5.6989e-04, 2.5392e-04, 4.6028e-04, 2.2950e-04,
        6.0411e-04, 1.9048e-04, 4.2532e-05, 1.0416e-03, 1.6216e-04, 1.1179e-04,
        3.0733e-04, 6.4232e-04, 2.2688e-04, 3.8711e-04, 1.1386e-03, 3.8756e-04,
        2.7773e-04, 6.2210e-04, 6.7038e-04, 1.6220e-03, 5.4123e-04, 1.3166e-03,
        6.0649e-04, 3.2223e-04, 2.8417e-04, 1.5756e-04, 3.9202e-04, 1.0684e-04,
        4.6372e-04, 9.7685e-05, 4.3843e-04, 7.9739e-04, 8.0355e-04, 3.6871e-04,
        3.4291e-04, 4.8811e-04, 4.5442e-04, 3.8745e-04, 1.1069e-03, 2.3646e-03,
        2.1701e-03, 1.6449e-03, 7.9537e-04, 6.1879e-04, 8.9270e-04, 3.1727e-04,
        3.1682e-04, 2.4867e-04, 2.8889e-04, 1.9539e-04, 7.2416e-04, 2.0994e-04,
        5.0128e-04, 4.4847e-04, 3.4334e-04, 6.1144e-04, 3.0276e-04, 2.7313e-04,
        6.6428e-04, 3.0075e-04, 6.3892e-04, 4.6897e-04, 2.5971e-04, 5.5040e-04,
        1.6075e-03, 2.7567e-03, 8.0828e-04, 2.6235e-04, 5.3966e-04, 6.5000e-04,
        2.4176e-04, 4.1493e-04, 5.3368e-04, 7.7685e-04, 3.3176e-04, 3.3116e-04,
        7.8318e-04, 2.3557e-04, 5.7168e-04, 2.5495e-04, 9.2811e-04, 4.2728e-04,
        7.5621e-04, 1.4607e-04, 3.9857e-04, 2.9777e-04, 7.2635e-04, 4.7855e-04,
        1.0942e-03, 9.3050e-05, 6.8358e-04, 2.5128e-04, 3.6507e-04, 2.2425e-04,
        1.8341e-04, 2.5268e-04, 1.2318e-04, 3.1368e-04, 6.9833e-04, 8.9636e-05,
        3.3937e-04, 2.8944e-04, 4.4219e-04, 1.2359e-03, 8.1693e-04, 9.3307e-04,
        4.6087e-04, 3.0155e-04, 6.8739e-04, 3.9297e-04, 6.4112e-04, 6.9322e-04,
        3.9807e-04, 1.7426e-04, 2.8901e-04, 2.6572e-04, 8.1452e-04, 3.8805e-04,
        2.6997e-04, 2.8709e-04, 1.3038e-04, 5.2580e-04, 5.0986e-04, 1.9245e-04,
        7.0143e-04, 3.7204e-04, 7.4964e-04, 6.2739e-04, 7.4992e-04, 1.7279e-03,
        1.1693e-03, 6.3753e-04, 4.2062e-04, 3.9403e-04, 7.4455e-04, 1.5303e-04,
        1.6821e-04, 3.7576e-04, 2.8444e-04, 4.7912e-04, 3.7388e-04, 6.5029e-04,
        8.9485e-05, 9.4641e-05, 1.7297e-04, 1.6765e-04, 5.2684e-04, 8.6280e-04,
        2.8234e-04, 2.9285e-04, 8.3370e-04, 5.2320e-04, 6.1494e-04, 1.1521e-03,
        6.3043e-04, 3.0784e-04, 4.7542e-04, 8.8198e-04, 1.0501e-03, 4.1482e-04,
        5.7471e-04, 3.5366e-04, 5.6045e-04, 1.9594e-04, 3.4469e-04, 6.6546e-04,
        3.9249e-04, 2.5994e-04, 4.8587e-05, 1.8540e-04, 2.0121e-04, 4.8332e-04,
        5.7309e-04, 2.7246e-04, 2.0015e-04, 4.3148e-04, 5.2378e-04, 3.2665e-04,
        7.0027e-04, 5.5362e-04, 1.8033e-04, 2.6833e-04, 1.7868e-04, 4.5971e-04,
        1.5848e-04, 1.0867e-04, 1.0855e-04, 3.5013e-04, 3.3651e-04, 2.2618e-04,
        1.3734e-04, 2.1633e-04, 4.5687e-04, 3.2358e-04, 3.8259e-04, 7.2004e-04,
        5.2827e-04, 9.1191e-05, 3.3154e-04, 1.3158e-04, 3.0794e-04, 3.4629e-04,
        1.2234e-04, 5.0942e-04, 1.3245e-04, 4.6428e-04, 5.5322e-04, 5.5311e-04,
        4.1899e-04, 9.9409e-04, 1.5219e-04, 1.3750e-04, 1.3902e-04, 1.4893e-04,
        1.3108e-04, 1.3935e-04, 5.6582e-04, 3.7343e-04, 5.1294e-04, 3.1925e-04,
        5.7184e-04, 1.1166e-04, 2.2172e-04, 5.7977e-04, 8.1813e-04, 1.5548e-04,
        3.7613e-04, 7.4352e-04, 3.3889e-04, 1.2403e-04, 3.7102e-04, 1.8184e-04,
        4.3309e-04, 3.0773e-04, 1.3842e-04, 4.4137e-04, 6.6017e-04, 1.0997e-04,
        4.5323e-05, 4.0169e-04, 5.2619e-04, 5.5767e-04, 8.4468e-04, 4.2172e-04,
        5.6827e-04, 3.9300e-04, 1.9881e-04, 2.6191e-04, 1.2594e-04, 6.8315e-04,
        3.9365e-04, 4.4103e-04, 3.4558e-04, 4.2210e-04, 2.2990e-04, 5.3716e-04,
        2.0387e-04, 6.7458e-05, 2.9301e-04, 2.4722e-04, 9.1737e-04, 3.4625e-04,
        4.0763e-04, 5.3288e-04, 4.4019e-04, 2.6751e-04, 1.4613e-04, 1.1231e-04,
        1.2632e-04, 4.9234e-04, 3.7545e-04, 6.4246e-04, 1.5561e-04, 1.5984e-04,
        2.4404e-04, 9.2896e-05, 7.5857e-05, 1.7412e-04, 2.9306e-04, 5.4384e-04,
        1.4128e-04, 1.3293e-04, 1.9935e-04, 3.6057e-04, 3.2455e-04, 2.7233e-03,
        6.2994e-03, 2.1752e-02, 1.5559e-02, 7.4162e-03, 3.9447e-03, 1.2210e-02,
        1.8378e-02, 4.9444e-03, 1.0161e-02, 2.5957e-03, 7.7328e-03, 1.0015e-02,
        1.5255e-02, 3.3748e-02, 2.5860e-02, 2.1207e-02, 1.1967e-02, 7.1857e-03,
        1.8799e-02, 4.2884e-02, 4.6989e-02, 3.9140e-02, 5.3972e-02, 8.3470e-02,
        1.6604e-01, 0.0000e+00]),tensor([0.0000e+00, 4.4916e-04, 2.5991e-02, 5.7746e-04, 4.0501e-03, 4.8323e-03,
        2.9392e-02, 1.6959e-03, 1.5046e-03, 3.2473e-04, 1.3546e-03, 2.8714e-03,
        8.2072e-04, 4.7359e-04, 3.5620e-04, 5.5874e-04, 7.5748e-04, 5.3164e-04,
        3.9531e-04, 2.3827e-04, 7.3746e-05, 6.2526e-05, 8.7949e-05, 6.4583e-04,
        2.6308e-03, 6.0940e-04, 1.5266e-03, 1.4176e-03, 1.1731e-04, 4.8714e-04,
        4.2478e-04, 2.0829e-03, 1.6692e-04, 8.0376e-04, 6.8137e-04, 6.9273e-04,
        8.7651e-04, 6.7475e-04, 4.4654e-04, 7.1220e-04, 1.2089e-04, 4.1153e-04,
        2.2174e-04, 1.8056e-04, 2.0767e-04, 1.3395e-04, 4.4654e-04, 5.1245e-04,
        1.5173e-04, 1.0426e-04, 6.0468e-04, 1.7213e-03, 1.8534e-03, 4.8594e-04,
        6.6708e-04, 1.1724e-03, 6.0734e-04, 6.4025e-04, 4.6464e-02, 1.5795e-03,
        1.8761e-03, 2.7647e-04, 8.9078e-04, 7.8568e-04, 3.9329e-04, 8.1367e-04,
        9.5838e-04, 5.0625e-04, 3.3099e-04, 9.0547e-05, 7.8455e-05, 1.6662e-04,
        1.2714e-04, 1.4570e-03, 7.3146e-04, 2.0453e-03, 5.7682e-04, 5.4452e-04,
        1.2025e-03, 4.8588e-04, 9.5715e-04, 1.1446e-03, 1.8390e-03, 9.9038e-04,
        3.8242e-04, 2.5446e-03, 6.5159e-04, 9.5950e-04, 5.4718e-04, 4.4709e-04,
        1.7132e-03, 5.9122e-04, 2.6557e-04, 9.0601e-04, 1.4527e-04, 1.2625e-04,
        1.7007e-04, 2.8931e-04, 9.6693e-04, 4.5150e-04, 3.4761e-04, 3.1939e-04,
        2.0904e-03, 6.6336e-04, 9.2066e-04, 1.4743e-03, 1.2733e-03, 1.9797e-04,
        2.0363e-03, 1.5995e-04, 5.5876e-04, 2.6991e-04, 5.8170e-04, 2.5467e-04,
        4.6802e-04, 1.3687e-04, 5.0154e-05, 7.6288e-04, 1.4464e-04, 1.4905e-04,
        4.1405e-04, 4.7101e-04, 1.9251e-04, 5.5585e-04, 9.0021e-04, 3.2275e-04,
        1.7400e-04, 6.3431e-04, 6.7892e-04, 1.1194e-03, 6.8973e-04, 2.2883e-03,
        6.8325e-04, 4.1625e-04, 2.6024e-04, 2.0582e-04, 4.3822e-04, 1.0462e-04,
        4.3030e-04, 1.2382e-04, 3.9513e-04, 7.3869e-04, 7.7328e-04, 4.4321e-04,
        2.6786e-04, 3.7474e-04, 3.0525e-04, 2.9032e-04, 1.0581e-03, 1.6462e-03,
        1.9591e-03, 1.1515e-03, 4.6585e-04, 7.9193e-04, 1.5517e-03, 2.4553e-04,
        3.3377e-04, 2.4929e-04, 3.6857e-04, 2.3244e-04, 4.5416e-04, 2.2510e-04,
        3.2825e-04, 3.4584e-04, 3.8967e-04, 5.0334e-04, 2.1656e-04, 2.4087e-04,
        7.1913e-04, 2.3636e-04, 4.7830e-04, 3.6581e-04, 2.9126e-04, 3.6094e-04,
        1.3108e-03, 1.5802e-03, 4.5606e-04, 2.5337e-04, 3.8517e-04, 7.8416e-04,
        1.8390e-04, 3.6606e-04, 4.1248e-04, 9.6568e-04, 3.7742e-04, 2.4160e-04,
        9.4422e-04, 2.2718e-04, 5.6854e-04, 2.0319e-04, 8.6569e-04, 3.5409e-04,
        8.6475e-04, 1.1370e-04, 2.7646e-04, 3.4282e-04, 7.1984e-04, 3.5907e-04,
        5.3674e-04, 1.0964e-04, 6.1886e-04, 2.8630e-04, 3.8713e-04, 1.7632e-04,
        2.5628e-04, 2.5964e-04, 9.5457e-05, 3.4698e-04, 7.6387e-04, 1.1320e-04,
        4.6423e-04, 2.1829e-04, 3.0457e-04, 1.1407e-03, 5.7109e-04, 5.7147e-04,
        2.7408e-04, 2.8740e-04, 8.3476e-04, 2.9868e-04, 4.5103e-04, 8.0316e-04,
        3.2799e-04, 1.9309e-04, 2.6278e-04, 3.0451e-04, 6.8318e-04, 3.5874e-04,
        3.8543e-04, 3.0790e-04, 1.1606e-04, 4.4091e-04, 3.5146e-04, 1.8702e-04,
        5.2851e-04, 2.8144e-04, 6.3276e-04, 4.9852e-04, 4.8616e-04, 1.6474e-03,
        9.8360e-04, 5.0323e-04, 3.6095e-04, 2.7534e-04, 6.8445e-04, 1.8136e-04,
        1.2376e-04, 4.8927e-04, 3.2016e-04, 4.8111e-04, 3.4946e-04, 5.0686e-04,
        1.1122e-04, 6.7375e-05, 1.7920e-04, 1.5233e-04, 4.9808e-04, 6.6845e-04,
        1.8376e-04, 3.0804e-04, 4.1740e-04, 3.3401e-04, 4.0418e-04, 1.0472e-03,
        4.1542e-04, 3.1743e-04, 4.5219e-04, 8.4364e-04, 1.2745e-03, 3.6756e-04,
        6.4783e-04, 4.7054e-04, 5.0168e-04, 1.8990e-04, 2.3538e-04, 4.9238e-04,
        3.0154e-04, 2.2478e-04, 5.1625e-05, 2.6932e-04, 1.4311e-04, 4.7079e-04,
        7.6833e-04, 1.7525e-04, 1.2261e-04, 2.5195e-04, 4.2988e-04, 2.3931e-04,
        6.3745e-04, 4.4474e-04, 1.9900e-04, 2.9545e-04, 1.8693e-04, 5.1536e-04,
        1.4091e-04, 1.0747e-04, 9.4246e-05, 2.6047e-04, 2.5261e-04, 1.9478e-04,
        1.2396e-04, 2.3938e-04, 5.2473e-04, 3.1555e-04, 2.5937e-04, 5.2386e-04,
        3.4822e-04, 9.2672e-05, 2.3556e-04, 1.4835e-04, 2.3738e-04, 3.7144e-04,
        1.2619e-04, 3.1362e-04, 1.2335e-04, 6.0586e-04, 4.4700e-04, 4.0516e-04,
        3.2688e-04, 1.1642e-03, 1.4279e-04, 1.5012e-04, 1.2648e-04, 1.2061e-04,
        1.4183e-04, 1.5685e-04, 6.6454e-04, 2.2316e-04, 3.6643e-04, 4.5010e-04,
        4.2193e-04, 1.1375e-04, 3.1988e-04, 5.5134e-04, 8.4993e-04, 1.6674e-04,
        2.8829e-04, 5.7609e-04, 3.1357e-04, 1.2116e-04, 3.2089e-04, 2.8876e-04,
        3.8116e-04, 2.9499e-04, 1.4698e-04, 4.8080e-04, 6.0711e-04, 1.0715e-04,
        5.7581e-05, 4.6450e-04, 3.2754e-04, 3.1714e-04, 1.0129e-03, 5.3158e-04,
        5.1855e-04, 4.0112e-04, 1.9134e-04, 2.2034e-04, 1.9522e-04, 4.5873e-04,
        5.9866e-04, 6.8720e-04, 3.0099e-04, 4.8271e-04, 2.1609e-04, 5.0390e-04,
        2.2491e-04, 1.0007e-04, 3.1840e-04, 2.1620e-04, 8.2180e-04, 3.3643e-04,
        4.5489e-04, 3.0859e-04, 3.2193e-04, 2.7326e-04, 1.8720e-04, 1.0954e-04,
        2.0320e-04, 6.5975e-04, 4.1446e-04, 7.1906e-04, 1.4045e-04, 1.2046e-04,
        3.0315e-04, 9.8250e-05, 1.0036e-04, 1.9042e-04, 2.2842e-04, 3.7793e-04,
        1.3675e-04, 1.4516e-04, 1.8689e-04, 2.7742e-04, 2.8345e-04, 2.6995e-03,
        5.7405e-03, 1.9910e-02, 1.5135e-02, 4.7848e-03, 3.1830e-03, 1.0147e-02,
        1.2430e-02, 4.9041e-03, 7.0413e-03, 2.3820e-03, 5.8078e-03, 7.1574e-03,
        1.2925e-02, 2.4234e-02, 1.9909e-02, 1.7331e-02, 6.8311e-03, 5.3175e-03,
        1.3303e-02, 2.0347e-02, 2.7967e-02, 3.7541e-02, 3.6424e-02, 4.7379e-02,
        1.8261e-01, 1.3718e-01, 0.0000e+00]),tensor([0.0000e+00, 4.6420e-04, 2.7601e-02, 6.7861e-04, 4.4264e-03, 4.8542e-03,
        3.4108e-02, 1.9332e-03, 1.0991e-03, 2.8238e-04, 1.1456e-03, 2.7098e-03,
        8.1606e-04, 4.2034e-04, 6.6273e-04, 3.6799e-04, 5.2389e-04, 5.0148e-04,
        5.0632e-04, 1.8907e-04, 8.3171e-05, 5.4791e-05, 1.0090e-04, 4.9933e-04,
        2.5142e-03, 4.4306e-04, 1.7951e-03, 9.7764e-04, 1.7229e-04, 4.3238e-04,
        7.9487e-04, 1.4786e-03, 1.8268e-04, 4.3937e-04, 4.7167e-04, 3.4104e-04,
        5.3005e-04, 6.8233e-04, 2.2960e-04, 3.8096e-04, 1.8287e-04, 2.2065e-04,
        2.1426e-04, 1.5882e-04, 2.4856e-04, 9.2762e-05, 4.4293e-04, 2.8602e-04,
        3.3862e-04, 1.3874e-04, 5.8149e-04, 9.7878e-04, 1.3882e-03, 2.8936e-04,
        1.2904e-03, 9.2276e-04, 6.3246e-04, 5.0235e-04, 7.6530e-02, 1.7102e-03,
        1.9195e-03, 4.5175e-04, 1.3213e-03, 3.6873e-04, 3.0032e-04, 4.7750e-04,
        1.0885e-03, 5.1212e-04, 2.0000e-04, 8.0598e-05, 8.0685e-05, 2.2750e-04,
        1.4229e-04, 1.1020e-03, 7.5919e-04, 1.9670e-03, 1.8181e-03, 6.6693e-04,
        1.2148e-03, 3.0961e-04, 8.6510e-04, 7.0919e-04, 2.3186e-03, 1.3977e-03,
        4.4833e-04, 1.8734e-03, 1.2982e-03, 7.7180e-04, 4.8411e-04, 5.7922e-04,
        1.6250e-03, 3.8606e-04, 2.3533e-04, 9.0714e-04, 1.2429e-04, 1.2635e-04,
        2.1594e-04, 2.8283e-04, 4.5498e-04, 1.9866e-04, 2.3075e-04, 2.0021e-04,
        1.7474e-03, 5.2845e-04, 8.6690e-04, 1.0824e-03, 6.8731e-04, 1.9214e-04,
        1.6815e-03, 1.2592e-04, 3.9179e-04, 3.2111e-04, 5.5804e-04, 1.9892e-04,
        3.1513e-04, 8.6952e-05, 5.6316e-05, 4.6481e-04, 1.5106e-04, 1.3186e-04,
        5.3627e-04, 1.9611e-04, 1.6378e-04, 4.3573e-04, 5.2078e-04, 2.7279e-04,
        1.2556e-04, 7.1311e-04, 7.8542e-04, 6.3018e-04, 4.7270e-04, 2.0098e-03,
        5.9624e-04, 4.1297e-04, 2.2593e-04, 2.0427e-04, 4.0820e-04, 1.1896e-04,
        2.5747e-04, 2.1657e-04, 2.9619e-04, 5.4197e-04, 4.1396e-04, 5.5310e-04,
        2.1110e-04, 4.4327e-04, 1.8411e-04, 1.8971e-04, 9.7763e-04, 7.7600e-04,
        1.0236e-03, 6.3202e-04, 3.8719e-04, 5.1449e-04, 2.1916e-03, 1.4286e-04,
        2.2932e-04, 1.3798e-04, 3.0629e-04, 3.8122e-04, 2.5438e-04, 2.1659e-04,
        1.6825e-04, 1.8025e-04, 3.9041e-04, 2.5455e-04, 8.5594e-05, 1.1138e-04,
        5.2049e-04, 1.2476e-04, 2.4617e-04, 1.9666e-04, 2.1398e-04, 1.7996e-04,
        8.9694e-04, 2.0829e-03, 3.5937e-04, 1.6963e-04, 1.9873e-04, 6.2171e-04,
        1.0760e-04, 1.9679e-04, 2.4723e-04, 7.3084e-04, 2.5563e-04, 1.4669e-04,
        7.2573e-04, 1.8806e-04, 4.0373e-04, 1.2285e-04, 7.9114e-04, 2.4046e-04,
        6.4733e-04, 9.1297e-05, 1.5575e-04, 2.8807e-04, 5.5708e-04, 2.1985e-04,
        5.6576e-04, 1.3082e-04, 2.5859e-04, 3.0986e-04, 2.6905e-04, 1.0224e-04,
        4.8514e-04, 2.3945e-04, 7.7240e-05, 3.0679e-04, 6.4169e-04, 9.4048e-05,
        4.5097e-04, 8.6132e-05, 1.7084e-04, 9.4554e-04, 3.2345e-04, 4.7110e-04,
        1.1993e-04, 2.2664e-04, 6.3902e-04, 1.8362e-04, 2.1186e-04, 8.8596e-04,
        1.8160e-04, 1.3844e-04, 1.9630e-04, 2.1402e-04, 4.0249e-04, 2.5360e-04,
        4.5752e-04, 5.1859e-04, 1.1836e-04, 3.7935e-04, 2.0142e-04, 1.8605e-04,
        2.3204e-04, 1.5082e-04, 6.9826e-04, 3.1911e-04, 2.3672e-04, 1.1823e-03,
        1.2917e-03, 4.8054e-04, 1.8720e-04, 1.4607e-04, 4.2629e-04, 2.7186e-04,
        8.5837e-05, 6.7149e-04, 3.6678e-04, 4.7841e-04, 3.6696e-04, 2.8718e-04,
        1.8400e-04, 6.4948e-05, 2.0328e-04, 9.2081e-05, 3.3035e-04, 3.5100e-04,
        9.0357e-05, 2.5872e-04, 1.8961e-04, 1.7663e-04, 2.4297e-04, 1.5083e-03,
        2.1023e-04, 1.8134e-04, 3.0286e-04, 5.5214e-04, 1.3563e-03, 1.8003e-04,
        8.8648e-04, 7.5009e-04, 3.0005e-04, 1.4837e-04, 1.2450e-04, 2.8870e-04,
        2.2531e-04, 1.5583e-04, 5.3833e-05, 5.9645e-04, 7.5094e-05, 4.6331e-04,
        7.3766e-04, 1.2364e-04, 9.9984e-05, 2.0635e-04, 1.9101e-04, 1.1174e-04,
        4.6297e-04, 2.0021e-04, 1.4144e-04, 2.3178e-04, 1.7856e-04, 4.8578e-04,
        1.3129e-04, 1.5302e-04, 1.0885e-04, 1.3452e-04, 1.3881e-04, 2.4584e-04,
        1.3708e-04, 1.5105e-04, 4.0122e-04, 1.6124e-04, 1.6017e-04, 3.8146e-04,
        1.7628e-04, 7.1402e-05, 1.1367e-04, 9.1657e-05, 1.5487e-04, 2.6692e-04,
        7.8707e-05, 2.4587e-04, 1.2197e-04, 3.5680e-04, 2.4349e-04, 2.2368e-04,
        2.8927e-04, 8.9963e-04, 9.3496e-05, 1.1137e-04, 1.2021e-04, 9.1417e-05,
        1.0305e-04, 1.4788e-04, 4.9846e-04, 1.2955e-04, 1.8876e-04, 7.8314e-04,
        2.2880e-04, 9.5030e-05, 8.6343e-04, 3.1754e-04, 7.1560e-04, 1.2459e-04,
        2.3186e-04, 3.0081e-04, 1.9981e-04, 1.3784e-04, 1.7819e-04, 7.8261e-04,
        2.2444e-04, 1.8845e-04, 1.2673e-04, 2.4462e-04, 3.4631e-04, 7.3989e-05,
        1.3409e-04, 4.1644e-04, 2.4256e-04, 2.3471e-04, 6.3597e-04, 4.4563e-04,
        3.1351e-04, 3.4918e-04, 1.0107e-04, 1.6515e-04, 5.1822e-04, 2.6503e-04,
        3.8353e-04, 4.5636e-04, 1.9188e-04, 4.5837e-04, 1.9417e-04, 2.6168e-04,
        1.7634e-04, 3.2838e-04, 2.7208e-04, 1.4003e-04, 6.1498e-04, 1.9117e-04,
        4.9632e-04, 3.0725e-04, 2.9450e-04, 2.0835e-04, 1.3339e-04, 1.2854e-04,
        1.8102e-04, 7.6404e-04, 2.9273e-04, 8.4897e-04, 1.5512e-04, 1.2940e-04,
        2.3836e-04, 1.0633e-04, 2.9788e-04, 1.5195e-04, 1.3501e-04, 1.9973e-04,
        1.8963e-04, 1.1767e-04, 1.3323e-04, 1.7664e-04, 2.4062e-04, 2.3692e-03,
        4.7437e-03, 2.0254e-02, 1.1708e-02, 3.1725e-03, 2.6520e-03, 7.1124e-03,
        9.2844e-03, 4.2011e-03, 6.2596e-03, 2.0844e-03, 5.8452e-03, 4.3659e-03,
        1.0988e-02, 2.0563e-02, 1.5458e-02, 1.2060e-02, 4.1935e-03, 3.6134e-03,
        8.8506e-03, 1.1856e-02, 1.4490e-02, 3.7781e-02, 3.4241e-02, 3.1387e-02,
        1.0167e-01, 1.2474e-01, 1.6752e-01, 0.0000e+00]),tensor([0.0000e+00, 4.1872e-04, 2.7311e-02, 6.6148e-04, 3.6564e-03, 3.9930e-03,
        3.0026e-02, 1.5699e-03, 7.6481e-04, 2.7331e-04, 7.5465e-04, 1.7052e-03,
        6.0440e-04, 4.1017e-04, 2.0455e-04, 6.2101e-04, 4.6063e-04, 5.1971e-04,
        3.2570e-04, 1.8387e-04, 6.6620e-05, 7.0409e-05, 8.4303e-05, 5.0251e-04,
        1.8389e-03, 5.5615e-04, 1.3923e-03, 8.8404e-04, 1.0803e-04, 5.4611e-04,
        2.3728e-04, 1.2994e-03, 2.2145e-04, 6.3838e-04, 4.4052e-04, 5.8877e-04,
        7.2243e-04, 5.1519e-04, 2.4581e-04, 4.2794e-04, 1.0054e-04, 2.5361e-04,
        1.4611e-04, 1.3479e-04, 1.4004e-04, 1.0175e-04, 4.0539e-04, 6.3967e-04,
        1.2441e-04, 9.1250e-05, 4.8952e-04, 1.1786e-03, 1.1568e-03, 3.5580e-04,
        4.7497e-04, 7.2608e-04, 3.8667e-04, 5.2761e-04, 3.1787e-02, 1.2579e-03,
        1.1158e-03, 2.5689e-04, 6.7007e-04, 5.3910e-04, 2.6650e-04, 7.8594e-04,
        5.2348e-04, 3.7735e-04, 1.8398e-04, 7.1574e-05, 5.5992e-05, 1.4054e-04,
        1.1809e-04, 8.3107e-04, 5.8720e-04, 1.4495e-03, 4.4013e-04, 3.0563e-04,
        5.4687e-04, 3.3900e-04, 4.4882e-04, 5.4557e-04, 1.0112e-03, 9.0017e-04,
        3.5928e-04, 1.4623e-03, 4.2327e-04, 5.6064e-04, 2.2923e-04, 1.9900e-04,
        1.2671e-03, 3.9853e-04, 1.3137e-04, 7.1921e-04, 1.2236e-04, 1.0446e-04,
        1.6518e-04, 1.8640e-04, 5.2636e-04, 4.3518e-04, 2.4278e-04, 2.4176e-04,
        1.4412e-03, 3.4170e-04, 6.0996e-04, 1.0232e-03, 1.4230e-03, 1.4969e-04,
        1.3467e-03, 8.9727e-05, 3.8963e-04, 1.5723e-04, 3.2458e-04, 1.4076e-04,
        3.5191e-04, 9.3488e-05, 4.4589e-05, 1.1397e-03, 1.0841e-04, 1.7652e-04,
        2.2025e-04, 2.5582e-04, 1.1597e-04, 4.9869e-04, 8.0593e-04, 1.2211e-04,
        1.1977e-04, 3.6012e-04, 4.2097e-04, 1.1340e-03, 5.3265e-04, 1.5257e-03,
        3.4596e-04, 2.9575e-04, 1.5105e-04, 9.9453e-05, 3.0982e-04, 6.1453e-05,
        4.0665e-04, 7.7732e-05, 3.1466e-04, 6.0120e-04, 6.4332e-04, 2.1130e-04,
        1.4536e-04, 3.2564e-04, 1.5468e-04, 2.3025e-04, 3.3353e-04, 1.0144e-03,
        8.1310e-04, 9.7441e-04, 3.7274e-04, 7.2914e-04, 1.9723e-03, 1.2524e-04,
        2.0452e-04, 1.1784e-04, 2.2179e-04, 1.0571e-04, 3.3473e-04, 1.6903e-04,
        3.8503e-04, 2.0886e-04, 3.4296e-04, 3.5739e-04, 1.4876e-04, 8.8457e-05,
        6.3778e-04, 1.0151e-04, 2.1786e-04, 2.5398e-04, 1.9279e-04, 3.7409e-04,
        1.0203e-03, 1.2110e-03, 3.9519e-04, 1.3255e-04, 2.1650e-04, 4.8168e-04,
        1.0826e-04, 2.1257e-04, 2.4979e-04, 5.6022e-04, 2.3030e-04, 1.9840e-04,
        5.5670e-04, 1.4435e-04, 2.9543e-04, 1.1914e-04, 2.2057e-04, 2.7846e-04,
        5.3374e-04, 6.6724e-05, 2.3646e-04, 3.3534e-04, 4.0633e-04, 3.2928e-04,
        4.8572e-04, 8.2785e-05, 2.7100e-04, 2.0618e-04, 2.0592e-04, 9.6710e-05,
        2.3139e-04, 1.8594e-04, 6.3634e-05, 2.5555e-04, 6.4525e-04, 1.3401e-04,
        3.4720e-04, 1.2140e-04, 1.8711e-04, 5.3693e-04, 4.3789e-04, 2.0659e-04,
        1.9106e-04, 2.2739e-04, 4.3538e-04, 2.4524e-04, 3.9040e-04, 4.3060e-04,
        2.2970e-04, 1.0506e-04, 1.7243e-04, 1.6198e-04, 5.3102e-04, 2.6829e-04,
        2.7059e-04, 2.2953e-04, 9.2195e-05, 4.3036e-04, 2.4879e-04, 2.0376e-04,
        2.2466e-04, 1.2659e-04, 2.7900e-04, 3.6194e-04, 3.0051e-04, 1.5216e-03,
        4.3963e-04, 3.0029e-04, 1.8220e-04, 2.2205e-04, 3.9161e-04, 9.2182e-05,
        7.0600e-05, 5.6333e-04, 2.1893e-04, 4.4254e-04, 2.1531e-04, 2.9549e-04,
        1.0766e-04, 4.6745e-05, 1.2035e-04, 9.4923e-05, 4.4304e-04, 6.1964e-04,
        1.3135e-04, 1.4491e-04, 3.6388e-04, 3.0059e-04, 1.9144e-04, 4.7632e-04,
        3.6127e-04, 1.6544e-04, 3.0598e-04, 4.8260e-04, 4.4474e-04, 1.8280e-04,
        4.7564e-04, 3.6586e-04, 3.4039e-04, 1.0607e-04, 1.5467e-04, 3.3950e-04,
        2.4923e-04, 1.7547e-04, 3.8504e-05, 1.6825e-04, 9.2286e-05, 2.1738e-04,
        4.3021e-04, 1.0490e-04, 7.4580e-05, 1.7144e-04, 3.3028e-04, 2.7868e-04,
        5.9960e-04, 3.3881e-04, 2.0382e-04, 2.7534e-04, 1.1131e-04, 4.9632e-04,
        8.7462e-05, 7.7012e-05, 5.9198e-05, 2.9846e-04, 1.5035e-04, 1.5910e-04,
        9.1291e-05, 1.6653e-04, 3.8410e-04, 2.3702e-04, 1.2860e-04, 4.3775e-04,
        4.1204e-04, 6.2171e-05, 1.6844e-04, 1.1361e-04, 1.5734e-04, 2.3830e-04,
        1.0376e-04, 4.4959e-04, 8.3132e-05, 3.5184e-04, 4.1621e-04, 2.3069e-04,
        2.8978e-04, 8.1339e-04, 1.1857e-04, 1.5569e-04, 8.0872e-05, 7.2629e-05,
        1.1229e-04, 1.2675e-04, 4.1097e-04, 1.6754e-04, 3.0025e-04, 2.2663e-04,
        2.3772e-04, 7.9622e-05, 1.7870e-04, 2.9052e-04, 6.0163e-04, 1.5068e-04,
        3.3029e-04, 3.3870e-04, 2.4537e-04, 8.5291e-05, 1.7462e-04, 1.6528e-04,
        2.7394e-04, 1.8676e-04, 1.7152e-04, 3.5137e-04, 4.5190e-04, 9.4251e-05,
        4.2479e-05, 2.9277e-04, 2.6796e-04, 2.6736e-04, 6.5391e-04, 4.5909e-04,
        5.8929e-04, 2.1576e-04, 1.2428e-04, 1.3740e-04, 1.1378e-04, 4.0683e-04,
        3.9947e-04, 3.8460e-04, 2.4801e-04, 2.5119e-04, 1.5826e-04, 2.4780e-04,
        1.5607e-04, 7.0514e-05, 2.3380e-04, 1.5345e-04, 6.4810e-04, 2.1939e-04,
        3.2903e-04, 2.4317e-04, 2.2598e-04, 1.9484e-04, 1.6412e-04, 8.2904e-05,
        1.7266e-04, 2.1812e-04, 2.5008e-04, 3.9437e-04, 1.1193e-04, 8.3787e-05,
        2.0890e-04, 7.6414e-05, 6.2875e-05, 1.2730e-04, 1.3821e-04, 4.1864e-04,
        1.1216e-04, 1.4261e-04, 1.2503e-04, 1.8850e-04, 1.8943e-04, 2.7888e-03,
        4.7136e-03, 1.8813e-02, 1.2850e-02, 4.5034e-03, 2.8708e-03, 7.2887e-03,
        1.4010e-02, 4.7907e-03, 6.7473e-03, 3.3234e-03, 5.6170e-03, 5.5292e-03,
        1.2904e-02, 2.6567e-02, 1.7534e-02, 1.6086e-02, 5.2396e-03, 3.6076e-03,
        1.2447e-02, 1.8522e-02, 1.5857e-02, 2.4431e-02, 2.4823e-02, 5.5346e-02,
        9.6783e-02, 5.7615e-02, 1.1917e-01, 1.6314e-01, 0.0000e+00]),tensor([0.0000e+00, 1.0949e-03, 2.2268e-02, 1.1909e-03, 6.1347e-03, 5.5105e-03,
        2.0418e-02, 3.7353e-03, 1.7860e-04, 1.0229e-04, 4.2585e-04, 8.0200e-04,
        2.2014e-04, 2.3369e-04, 1.1447e-04, 6.0185e-04, 1.1010e-04, 2.4263e-04,
        1.1053e-04, 7.1013e-05, 5.9521e-05, 7.2603e-05, 5.3813e-05, 2.7666e-04,
        7.2489e-04, 5.2404e-04, 2.7959e-03, 1.2947e-03, 1.0351e-04, 1.8226e-04,
        1.6065e-04, 2.9492e-04, 6.3887e-05, 4.5323e-04, 2.3096e-04, 3.4796e-04,
        4.4736e-04, 2.5728e-04, 3.4096e-04, 2.6677e-04, 8.5702e-05, 2.5801e-04,
        1.2527e-04, 1.1406e-04, 1.1387e-04, 7.1841e-05, 1.6698e-04, 6.3209e-04,
        6.8096e-05, 1.0639e-04, 6.8063e-04, 2.9763e-04, 2.1413e-04, 2.9773e-04,
        1.0751e-04, 1.2064e-04, 2.1027e-04, 2.9638e-04, 4.1738e-03, 4.2352e-04,
        3.3513e-04, 1.3001e-04, 1.4728e-04, 2.4933e-04, 1.4664e-04, 8.6384e-04,
        1.8721e-04, 9.0722e-05, 1.2541e-04, 6.1203e-05, 4.6490e-05, 9.9221e-05,
        1.1112e-04, 7.2872e-04, 1.1611e-04, 2.6052e-04, 6.6158e-05, 6.6312e-05,
        1.3783e-04, 2.7386e-04, 2.3723e-04, 4.4923e-04, 3.5560e-04, 3.8352e-04,
        2.5392e-04, 2.8530e-04, 9.2668e-05, 2.0202e-04, 1.0826e-04, 9.9107e-05,
        3.2974e-04, 3.2466e-04, 4.6943e-05, 2.5105e-04, 5.8856e-05, 9.7441e-05,
        1.0354e-04, 7.6333e-05, 3.3386e-04, 6.5099e-04, 1.5275e-04, 1.9416e-04,
        4.4679e-04, 1.8347e-04, 2.1011e-04, 8.6664e-04, 1.4200e-03, 9.3851e-05,
        1.7073e-04, 7.3264e-05, 4.0448e-04, 7.4957e-05, 1.5846e-04, 1.2160e-04,
        4.4864e-04, 7.0287e-05, 3.9632e-05, 1.1218e-03, 1.0826e-04, 8.1369e-05,
        9.9515e-05, 3.9228e-04, 6.4086e-05, 9.5641e-05, 1.2261e-03, 6.4610e-05,
        1.2837e-04, 2.1838e-04, 2.8487e-04, 6.0688e-04, 1.8279e-04, 1.7392e-04,
        3.1756e-04, 1.0667e-04, 1.2743e-04, 3.9163e-05, 1.7824e-04, 4.2241e-05,
        4.6391e-04, 5.0295e-05, 1.7875e-04, 5.6323e-04, 6.2423e-04, 9.4622e-05,
        7.7773e-05, 1.3758e-04, 1.1967e-04, 1.7378e-04, 2.0248e-04, 1.0258e-03,
        8.5239e-04, 5.1087e-04, 2.4355e-04, 1.9823e-04, 1.8374e-04, 1.5085e-04,
        2.3796e-04, 1.8776e-04, 5.7465e-05, 6.3790e-05, 3.4165e-04, 9.5112e-05,
        5.6243e-04, 4.7405e-04, 2.2266e-04, 5.7562e-04, 2.8875e-04, 1.7785e-04,
        2.9791e-04, 2.0216e-04, 2.6998e-04, 2.3697e-04, 1.6637e-04, 4.7745e-04,
        3.3174e-04, 3.0976e-04, 5.2784e-04, 1.6161e-04, 2.6251e-04, 2.7448e-04,
        2.3382e-04, 8.4209e-05, 4.6084e-04, 1.4839e-04, 1.1032e-04, 2.1559e-04,
        1.7987e-04, 8.6988e-05, 4.0492e-04, 1.1255e-04, 1.3571e-04, 3.2855e-04,
        1.5784e-04, 6.5382e-05, 3.0732e-04, 1.8185e-04, 2.7701e-04, 3.1875e-04,
        4.4437e-04, 7.8454e-05, 1.7837e-04, 9.9173e-05, 1.8530e-04, 2.1987e-04,
        7.0112e-05, 1.3694e-04, 5.2456e-05, 1.0818e-04, 3.1824e-04, 6.8720e-05,
        9.0436e-05, 2.7449e-04, 1.8330e-04, 1.4412e-04, 6.6453e-04, 1.1288e-04,
        2.9804e-04, 2.3326e-04, 2.8965e-04, 1.9440e-04, 4.1709e-04, 1.0536e-04,
        3.7711e-04, 6.1980e-05, 1.1848e-04, 1.4644e-04, 7.9953e-04, 1.7353e-04,
        8.2381e-05, 7.7113e-05, 5.5912e-05, 3.5328e-04, 2.5313e-04, 5.3672e-05,
        3.4913e-04, 1.4515e-04, 1.6307e-04, 3.7487e-04, 2.8077e-04, 4.8290e-04,
        2.3139e-04, 3.2798e-04, 2.4286e-04, 2.6545e-04, 2.3980e-04, 9.7660e-05,
        7.7318e-05, 7.5507e-05, 5.2555e-05, 7.9114e-05, 1.2696e-04, 2.6206e-04,
        5.4338e-05, 3.8443e-05, 9.2988e-05, 1.1244e-04, 2.0776e-04, 1.0503e-03,
        2.2755e-04, 1.4249e-04, 5.7393e-04, 3.6983e-04, 7.9156e-05, 2.3932e-04,
        2.8100e-04, 1.5972e-04, 1.4647e-04, 2.4448e-04, 2.6540e-04, 1.4567e-04,
        1.5511e-04, 9.7427e-05, 3.5339e-04, 7.1898e-05, 3.5075e-04, 3.3920e-04,
        1.0057e-04, 1.8501e-04, 4.9861e-05, 8.5567e-05, 1.8675e-04, 1.4912e-04,
        1.9388e-04, 1.1504e-04, 1.0337e-04, 1.5736e-04, 3.7247e-04, 3.1613e-04,
        2.6666e-04, 5.8089e-04, 1.5098e-04, 1.2832e-04, 8.6187e-05, 1.0091e-04,
        7.1094e-05, 4.7238e-05, 4.9482e-05, 3.7051e-04, 2.1852e-04, 7.0067e-05,
        6.0004e-05, 6.6218e-05, 1.5442e-04, 2.0778e-04, 1.3617e-04, 5.8878e-04,
        4.3551e-04, 7.4310e-05, 2.6892e-04, 5.4615e-05, 2.0807e-04, 9.4929e-05,
        7.9522e-05, 4.2362e-04, 1.3691e-04, 1.2595e-04, 5.2284e-04, 1.5313e-04,
        1.5114e-04, 3.8704e-04, 1.3326e-04, 4.0430e-05, 7.0828e-05, 9.6144e-05,
        6.5870e-05, 3.8407e-05, 2.4950e-04, 2.2867e-04, 4.1355e-04, 1.3742e-04,
        5.4959e-04, 2.7902e-05, 1.1873e-04, 5.2440e-04, 2.9745e-04, 1.3770e-04,
        1.8021e-04, 4.8587e-04, 2.1502e-04, 9.1963e-05, 2.8663e-04, 9.0168e-05,
        2.7836e-04, 3.0480e-04, 9.1988e-05, 1.3644e-04, 3.5886e-04, 1.3022e-04,
        3.8889e-05, 4.9729e-04, 1.9043e-04, 2.4287e-04, 2.6940e-04, 1.0348e-04,
        4.5335e-04, 2.1408e-04, 1.5841e-04, 1.4191e-04, 6.0559e-05, 4.7868e-04,
        1.4095e-04, 8.9189e-05, 2.8438e-04, 2.4094e-04, 1.0519e-04, 4.1664e-04,
        1.9094e-04, 5.3068e-05, 2.0951e-04, 2.0569e-04, 1.3674e-03, 4.2856e-04,
        5.9888e-04, 2.7372e-04, 3.0171e-04, 1.5462e-04, 9.1959e-05, 1.4482e-04,
        5.4701e-05, 1.1784e-04, 2.3292e-04, 2.3724e-04, 1.0747e-04, 1.9464e-04,
        3.6030e-05, 8.8649e-05, 5.8610e-05, 1.1001e-04, 1.3293e-04, 1.0297e-03,
        2.3060e-04, 1.3170e-04, 2.1036e-04, 5.1488e-04, 4.3239e-04, 9.9392e-03,
        1.4367e-02, 2.0108e-02, 9.8894e-03, 7.9582e-03, 4.2576e-03, 1.5099e-02,
        4.1690e-02, 1.2864e-02, 2.5238e-02, 2.0747e-02, 2.0148e-02, 8.0638e-03,
        3.2781e-02, 6.8054e-02, 3.7513e-02, 2.1162e-02, 1.1307e-02, 6.4015e-03,
        2.8696e-02, 4.2048e-02, 2.8832e-02, 1.8949e-02, 2.0812e-02, 3.3878e-02,
        4.9674e-02, 1.6057e-02, 2.1273e-02, 4.0634e-02, 1.5071e-01, 0.0000e+00]),tensor([0.0000e+00, 4.2060e-04, 2.7249e-02, 4.1054e-04, 2.4433e-03, 3.2430e-03,
        2.8838e-02, 1.3305e-03, 1.7891e-04, 8.6006e-05, 7.1169e-04, 9.5287e-04,
        3.2337e-04, 3.0699e-04, 8.9637e-05, 1.3074e-03, 1.2483e-04, 3.0692e-04,
        1.3847e-04, 6.3775e-05, 4.0433e-05, 5.2081e-05, 4.3867e-05, 2.7600e-04,
        9.4803e-04, 9.2797e-04, 1.0391e-03, 7.4376e-04, 5.6053e-05, 2.6068e-04,
        1.1224e-04, 3.1956e-04, 6.7880e-05, 1.1460e-03, 4.5306e-04, 2.8462e-04,
        1.5703e-03, 3.2979e-04, 6.0307e-04, 6.1995e-04, 5.8713e-05, 4.3819e-04,
        1.1506e-04, 1.1443e-04, 8.8435e-05, 1.0214e-04, 2.4447e-04, 1.7050e-03,
        4.3945e-05, 5.6966e-05, 3.5541e-04, 3.4597e-04, 2.1469e-04, 1.6546e-04,
        5.5874e-05, 1.1529e-04, 6.7767e-05, 6.7624e-04, 9.0435e-04, 1.0851e-03,
        4.5336e-04, 1.3454e-04, 1.0257e-04, 2.9813e-04, 2.1343e-04, 2.8424e-03,
        1.6695e-04, 1.2503e-04, 2.5446e-04, 6.4761e-05, 5.2022e-05, 7.8469e-05,
        7.3382e-05, 1.0934e-03, 7.9515e-05, 2.3040e-04, 4.8085e-05, 4.9091e-05,
        1.3258e-04, 5.2467e-04, 2.7016e-04, 1.4381e-03, 4.3614e-04, 5.5467e-04,
        2.5107e-04, 2.8280e-04, 6.7227e-05, 2.8695e-04, 1.1109e-04, 9.0896e-05,
        5.1782e-04, 5.1085e-04, 5.5754e-05, 3.7964e-04, 9.2865e-05, 9.1824e-05,
        6.9973e-05, 4.4561e-05, 2.6164e-04, 1.0494e-03, 3.9324e-04, 5.4143e-04,
        7.2856e-04, 1.3723e-04, 2.1383e-04, 3.0702e-03, 3.9873e-03, 8.1585e-05,
        2.2001e-04, 1.0584e-04, 5.5653e-04, 6.7158e-05, 1.1776e-04, 1.0752e-04,
        9.0718e-04, 2.0732e-04, 2.6484e-05, 3.8660e-03, 9.6285e-05, 4.5805e-05,
        9.9480e-05, 1.0701e-03, 8.2516e-05, 9.4671e-05, 3.2825e-03, 9.4048e-05,
        2.6736e-04, 2.0692e-04, 1.8463e-04, 2.3471e-03, 2.0421e-04, 2.4151e-04,
        1.8280e-04, 9.0846e-05, 1.8263e-04, 4.9192e-05, 2.9081e-04, 5.6009e-05,
        7.9600e-04, 4.5031e-05, 1.8591e-04, 4.4400e-04, 6.7099e-04, 1.1663e-04,
        1.2950e-04, 2.0803e-04, 2.7308e-04, 4.3303e-04, 1.9867e-04, 3.5440e-03,
        2.0245e-03, 1.7924e-03, 4.6189e-04, 2.9261e-04, 1.7267e-04, 3.7946e-04,
        3.8280e-04, 3.1310e-04, 9.0837e-05, 6.8468e-05, 1.1200e-03, 1.2148e-04,
        1.3718e-03, 6.6210e-04, 3.1876e-04, 8.2756e-04, 4.3849e-04, 3.0718e-04,
        3.2229e-04, 3.8937e-04, 9.6512e-04, 6.1787e-04, 2.2840e-04, 1.1359e-03,
        6.9575e-04, 3.9178e-04, 5.6546e-04, 1.9649e-04, 1.0171e-03, 3.4643e-04,
        4.9023e-04, 2.6331e-04, 1.2093e-03, 1.4640e-04, 1.1825e-04, 3.2637e-04,
        2.7620e-04, 7.8934e-05, 3.6657e-04, 2.0678e-04, 1.4076e-04, 8.9573e-04,
        1.7473e-04, 1.3585e-04, 6.9298e-04, 2.6390e-04, 3.7604e-04, 7.8342e-04,
        4.6195e-04, 5.2355e-05, 6.7439e-04, 1.7265e-04, 2.0450e-04, 4.0719e-04,
        6.7488e-05, 2.7259e-04, 8.4556e-05, 1.2821e-04, 4.0493e-04, 1.0289e-04,
        8.3628e-05, 3.8056e-04, 2.4758e-04, 1.9417e-04, 1.8703e-03, 1.6276e-04,
        6.7864e-04, 4.1596e-04, 4.0298e-04, 4.2389e-04, 1.0116e-03, 1.4418e-04,
        5.9414e-04, 8.4650e-05, 3.9561e-04, 1.3550e-04, 2.1900e-03, 3.8816e-04,
        8.6872e-05, 6.2416e-05, 6.6581e-05, 3.3356e-04, 7.3368e-04, 6.0507e-05,
        3.5399e-04, 2.9948e-04, 1.6001e-04, 6.5147e-04, 8.0276e-04, 7.1972e-04,
        2.1711e-04, 2.5430e-04, 3.5202e-04, 5.0438e-04, 4.0344e-04, 9.3231e-05,
        1.6351e-04, 1.3067e-04, 6.3649e-05, 1.0891e-04, 1.1672e-04, 9.0019e-04,
        3.1776e-05, 4.7534e-05, 7.9796e-05, 1.7639e-04, 3.7727e-04, 9.2799e-04,
        5.7627e-04, 1.2339e-04, 1.0351e-03, 8.4108e-04, 1.3402e-04, 2.9678e-04,
        6.9140e-04, 2.1272e-04, 1.7226e-04, 3.5096e-04, 2.7039e-04, 4.7621e-04,
        1.2966e-04, 1.0222e-04, 1.2587e-03, 1.5697e-04, 8.0507e-04, 8.4869e-04,
        2.0772e-04, 2.5682e-04, 2.8726e-05, 8.6992e-05, 2.1589e-04, 1.1797e-04,
        1.4146e-04, 1.3871e-04, 1.5580e-04, 2.2477e-04, 8.3410e-04, 8.9376e-04,
        3.3604e-04, 1.0185e-03, 2.3174e-04, 1.9771e-04, 1.1133e-04, 1.7582e-04,
        1.3754e-04, 7.7418e-05, 6.5000e-05, 1.0147e-03, 6.3889e-04, 9.3972e-05,
        6.7491e-05, 9.1608e-05, 1.2099e-04, 3.4740e-04, 1.8469e-04, 1.0468e-03,
        1.1155e-03, 8.8343e-05, 5.0511e-04, 6.5139e-05, 2.9942e-04, 1.3623e-04,
        1.2759e-04, 5.6835e-04, 1.3032e-04, 1.7648e-04, 9.3604e-04, 6.1635e-04,
        1.6859e-04, 5.4113e-04, 2.2101e-04, 6.3554e-05, 7.8487e-05, 1.1405e-04,
        8.0118e-05, 4.7174e-05, 2.4367e-04, 3.7766e-04, 7.2807e-04, 1.6799e-04,
        8.2139e-04, 2.4086e-05, 1.3238e-04, 6.2056e-04, 2.8808e-04, 1.5390e-04,
        2.7103e-04, 4.5504e-04, 2.0439e-04, 7.3735e-05, 5.1905e-04, 8.6357e-05,
        2.7765e-04, 6.3623e-04, 1.2673e-04, 1.7151e-04, 3.6295e-04, 1.6857e-04,
        2.4597e-05, 3.4627e-04, 3.1375e-04, 3.8185e-04, 2.3383e-04, 1.0169e-04,
        7.1858e-04, 2.6019e-04, 2.7431e-04, 1.5166e-04, 5.2080e-05, 9.1655e-04,
        1.1029e-04, 7.8739e-05, 5.0568e-04, 2.0025e-04, 1.3027e-04, 5.6676e-04,
        3.3016e-04, 3.6457e-05, 1.7855e-04, 2.1921e-04, 1.0581e-03, 4.9534e-04,
        2.8172e-04, 4.2503e-04, 4.8705e-04, 2.0166e-04, 9.7248e-05, 1.1247e-04,
        4.7022e-05, 1.0141e-04, 3.4137e-04, 3.0115e-04, 1.1300e-04, 2.7711e-04,
        4.0372e-05, 6.0829e-05, 6.2403e-05, 1.1513e-04, 2.2440e-04, 1.9583e-03,
        2.6638e-04, 1.7474e-04, 1.8563e-04, 6.3889e-04, 3.3094e-04, 2.4940e-03,
        7.1113e-03, 2.0011e-02, 1.5842e-02, 9.6649e-03, 4.2133e-03, 1.7576e-02,
        2.2077e-02, 5.1815e-03, 8.0842e-03, 3.0675e-03, 6.5999e-03, 7.9617e-03,
        1.7432e-02, 3.2461e-02, 2.6683e-02, 2.9138e-02, 1.2891e-02, 5.7870e-03,
        2.5858e-02, 4.7216e-02, 2.9208e-02, 1.8949e-02, 1.9840e-02, 3.5891e-02,
        4.8322e-02, 1.2835e-02, 1.4040e-02, 2.5309e-02, 1.2818e-01, 1.1693e-01,
        0.0000e+00]),tensor([0.0000e+00, 4.7537e-04, 2.3211e-02, 3.9616e-04, 2.7091e-03, 3.3435e-03,
        2.5505e-02, 1.4001e-03, 2.4320e-04, 9.4811e-05, 7.5131e-04, 9.6056e-04,
        3.1400e-04, 4.1263e-04, 1.5058e-04, 1.5138e-03, 1.6434e-04, 4.6231e-04,
        2.0629e-04, 6.4556e-05, 6.1658e-05, 7.4263e-05, 5.8559e-05, 4.1573e-04,
        9.0738e-04, 1.1613e-03, 1.1230e-03, 9.5610e-04, 9.7429e-05, 3.8447e-04,
        2.0151e-04, 4.0358e-04, 9.0803e-05, 1.4066e-03, 4.1729e-04, 3.2648e-04,
        1.5673e-03, 2.8384e-04, 6.9729e-04, 7.0972e-04, 9.7203e-05, 4.1857e-04,
        1.5523e-04, 1.5257e-04, 1.1306e-04, 1.1394e-04, 3.7202e-04, 2.0021e-03,
        6.4125e-05, 8.1827e-05, 5.0950e-04, 4.1389e-04, 2.4346e-04, 1.9370e-04,
        7.9742e-05, 1.2600e-04, 1.0128e-04, 5.3487e-04, 6.7352e-04, 7.3845e-04,
        4.7073e-04, 2.0744e-04, 1.2690e-04, 3.6231e-04, 1.6489e-04, 3.9018e-03,
        2.7686e-04, 1.2845e-04, 2.4086e-04, 8.4403e-05, 5.5353e-05, 8.1959e-05,
        8.1164e-05, 1.1134e-03, 1.1836e-04, 2.5803e-04, 6.0170e-05, 6.0962e-05,
        1.6247e-04, 4.7042e-04, 3.4879e-04, 1.4163e-03, 4.5568e-04, 4.1383e-04,
        2.3060e-04, 2.9810e-04, 8.8209e-05, 2.7196e-04, 1.4387e-04, 1.3545e-04,
        4.9400e-04, 4.2429e-04, 8.2870e-05, 3.4185e-04, 1.0974e-04, 1.1212e-04,
        1.1184e-04, 7.0451e-05, 3.0059e-04, 1.3061e-03, 3.5326e-04, 4.0253e-04,
        7.3576e-04, 1.8888e-04, 1.9790e-04, 2.4525e-03, 4.6567e-03, 1.2100e-04,
        2.2755e-04, 1.0780e-04, 4.1249e-04, 8.8007e-05, 1.9086e-04, 1.1623e-04,
        1.2661e-03, 2.6053e-04, 3.9427e-05, 4.4755e-03, 1.3507e-04, 7.0983e-05,
        1.5466e-04, 1.3165e-03, 9.1459e-05, 1.1182e-04, 3.5817e-03, 8.8099e-05,
        2.2984e-04, 1.9921e-04, 2.0928e-04, 2.7731e-03, 2.1309e-04, 2.4120e-04,
        2.3023e-04, 1.2714e-04, 1.7906e-04, 4.8060e-05, 2.1045e-04, 7.7277e-05,
        1.1147e-03, 6.5627e-05, 2.4656e-04, 4.5039e-04, 7.9299e-04, 1.8548e-04,
        1.4557e-04, 1.6662e-04, 3.8000e-04, 3.3197e-04, 3.1396e-04, 4.8172e-03,
        2.0560e-03, 1.9450e-03, 4.6342e-04, 3.1284e-04, 2.0388e-04, 4.2189e-04,
        2.4036e-04, 2.3739e-04, 7.9962e-05, 9.4468e-05, 1.3876e-03, 1.6873e-04,
        1.5506e-03, 6.0007e-04, 4.9099e-04, 9.8811e-04, 5.4197e-04, 2.3187e-04,
        3.1338e-04, 2.3905e-04, 1.2721e-03, 7.5995e-04, 3.8351e-04, 1.1277e-03,
        5.1636e-04, 3.8928e-04, 5.9161e-04, 2.1003e-04, 9.2930e-04, 3.9564e-04,
        5.7635e-04, 2.4652e-04, 1.3963e-03, 1.8213e-04, 1.4009e-04, 4.2384e-04,
        2.2053e-04, 1.0750e-04, 3.2427e-04, 2.5800e-04, 1.6666e-04, 1.0389e-03,
        1.8966e-04, 1.1695e-04, 7.5651e-04, 4.1208e-04, 3.4893e-04, 7.4061e-04,
        4.4831e-04, 7.7389e-05, 6.1946e-04, 1.1033e-04, 1.7701e-04, 3.5250e-04,
        6.6523e-05, 1.7196e-04, 9.2805e-05, 1.1162e-04, 3.2304e-04, 1.4504e-04,
        1.0141e-04, 5.4672e-04, 3.1438e-04, 1.8932e-04, 2.8079e-03, 2.1999e-04,
        7.8711e-04, 6.1051e-04, 3.8290e-04, 4.1671e-04, 1.1327e-03, 2.1093e-04,
        8.3494e-04, 1.0836e-04, 2.5141e-04, 1.3798e-04, 2.4392e-03, 4.1596e-04,
        9.8080e-05, 6.6859e-05, 6.7805e-05, 2.6492e-04, 9.2841e-04, 7.9827e-05,
        5.1954e-04, 1.7095e-04, 1.7118e-04, 6.5589e-04, 1.1475e-03, 5.9685e-04,
        2.2887e-04, 2.9108e-04, 2.9070e-04, 5.6130e-04, 6.3453e-04, 1.4673e-04,
        1.7750e-04, 9.4298e-05, 7.6372e-05, 9.0986e-05, 1.2449e-04, 7.7684e-04,
        3.5427e-05, 5.6923e-05, 9.7485e-05, 2.1252e-04, 4.8196e-04, 9.3855e-04,
        7.0638e-04, 1.3971e-04, 1.0610e-03, 8.0212e-04, 1.3518e-04, 2.8429e-04,
        6.6146e-04, 1.9997e-04, 2.2495e-04, 5.2502e-04, 4.2805e-04, 4.0482e-04,
        1.4276e-04, 1.1720e-04, 1.4290e-03, 1.6252e-04, 7.1765e-04, 1.3084e-03,
        1.7492e-04, 2.2014e-04, 4.2336e-05, 1.3100e-04, 2.6450e-04, 1.7292e-04,
        2.1728e-04, 1.5711e-04, 1.8668e-04, 2.3756e-04, 1.2451e-03, 8.1770e-04,
        2.6262e-04, 1.0282e-03, 3.4176e-04, 2.7407e-04, 1.4077e-04, 1.7366e-04,
        1.5844e-04, 8.4095e-05, 6.7269e-05, 1.2261e-03, 4.5479e-04, 1.3747e-04,
        1.1660e-04, 1.1801e-04, 1.2810e-04, 5.6447e-04, 2.5133e-04, 1.0020e-03,
        1.0960e-03, 1.2839e-04, 6.0211e-04, 9.6926e-05, 2.7034e-04, 1.1333e-04,
        1.9018e-04, 5.3064e-04, 1.9785e-04, 1.9335e-04, 1.0615e-03, 4.8443e-04,
        1.2472e-04, 4.9696e-04, 3.0004e-04, 8.3568e-05, 1.3104e-04, 1.6101e-04,
        1.0329e-04, 6.6356e-05, 3.3753e-04, 5.1069e-04, 8.8272e-04, 2.8210e-04,
        6.8076e-04, 3.0748e-05, 1.9294e-04, 6.2359e-04, 2.5342e-04, 1.9458e-04,
        1.9431e-04, 4.6151e-04, 2.3024e-04, 1.0800e-04, 3.2155e-04, 1.2668e-04,
        2.6433e-04, 8.1907e-04, 1.9457e-04, 1.7954e-04, 3.9635e-04, 2.3225e-04,
        3.0219e-05, 3.5964e-04, 3.7324e-04, 5.0657e-04, 2.8715e-04, 1.0521e-04,
        5.9467e-04, 3.0817e-04, 2.2735e-04, 1.4964e-04, 7.1298e-05, 8.5715e-04,
        1.3040e-04, 8.8502e-05, 5.5611e-04, 2.5231e-04, 1.4381e-04, 5.2584e-04,
        4.8891e-04, 4.9786e-05, 1.8976e-04, 1.8979e-04, 9.7093e-04, 5.4557e-04,
        3.4213e-04, 5.3045e-04, 5.7059e-04, 2.5879e-04, 9.3965e-05, 1.6978e-04,
        5.0281e-05, 1.3145e-04, 3.5745e-04, 3.4194e-04, 1.6057e-04, 3.5374e-04,
        5.0741e-05, 8.4216e-05, 1.0567e-04, 1.4495e-04, 2.3393e-04, 2.0654e-03,
        3.3613e-04, 1.5676e-04, 1.9216e-04, 5.7461e-04, 3.1416e-04, 2.4143e-03,
        8.3381e-03, 1.9419e-02, 1.2676e-02, 9.6917e-03, 4.1632e-03, 1.6357e-02,
        2.2358e-02, 5.1326e-03, 8.6788e-03, 2.5680e-03, 6.5913e-03, 6.7565e-03,
        1.5793e-02, 3.0095e-02, 2.3197e-02, 2.1699e-02, 1.1134e-02, 5.6284e-03,
        2.3073e-02, 4.7138e-02, 2.6831e-02, 1.4500e-02, 1.7102e-02, 2.7290e-02,
        3.6285e-02, 1.1003e-02, 1.0829e-02, 1.6817e-02, 9.7902e-02, 9.3981e-02,
        1.1158e-01, 0.0000e+00]),tensor([0.0000e+00, 4.1023e-04, 2.6558e-02, 5.2357e-04, 3.1328e-03, 3.8198e-03,
        2.8066e-02, 1.2411e-03, 2.5367e-04, 1.7442e-04, 8.2506e-04, 1.0780e-03,
        4.4361e-04, 9.3459e-04, 2.5862e-04, 1.8330e-03, 1.9825e-04, 1.1806e-03,
        5.2943e-04, 1.1831e-04, 8.3614e-05, 8.1378e-05, 6.5538e-05, 7.1469e-04,
        1.2825e-03, 1.1487e-03, 1.0709e-03, 8.8922e-04, 1.4195e-04, 1.0697e-03,
        3.6392e-04, 5.6337e-04, 2.4796e-04, 3.3592e-03, 8.2860e-04, 4.4267e-04,
        2.6193e-03, 2.6704e-04, 1.0348e-03, 1.6733e-03, 1.2324e-04, 4.8070e-04,
        1.1610e-04, 1.2545e-04, 9.2864e-05, 1.2102e-04, 1.0734e-03, 3.1680e-03,
        1.0299e-04, 9.6662e-05, 5.5295e-04, 5.8520e-04, 3.1526e-04, 3.1546e-04,
        6.8098e-05, 1.7162e-04, 1.3314e-04, 5.2917e-04, 4.1702e-04, 5.5826e-04,
        4.6587e-04, 4.1505e-04, 1.2576e-04, 5.3043e-04, 1.6428e-04, 5.4852e-03,
        1.8117e-04, 1.1375e-04, 3.0612e-04, 9.1851e-05, 4.8010e-05, 6.9130e-05,
        7.5977e-05, 1.2440e-03, 1.0597e-04, 3.6206e-04, 5.8452e-05, 6.0083e-05,
        1.9480e-04, 4.1230e-04, 2.1925e-04, 1.3241e-03, 2.3780e-04, 2.9650e-04,
        1.7059e-04, 4.6889e-04, 1.0684e-04, 3.3534e-04, 8.5721e-05, 8.3469e-05,
        7.8952e-04, 5.5355e-04, 7.2447e-05, 4.8094e-04, 1.7050e-04, 1.2768e-04,
        1.4694e-04, 5.8543e-05, 3.8534e-04, 1.9139e-03, 3.4181e-04, 4.6283e-04,
        1.0072e-03, 1.0372e-04, 1.3530e-04, 2.0416e-03, 4.3746e-03, 3.1104e-04,
        3.7339e-04, 1.0349e-04, 5.6537e-04, 5.8897e-05, 1.1245e-04, 8.5935e-05,
        1.4030e-03, 4.7453e-04, 8.4339e-05, 6.4769e-03, 1.4807e-04, 1.7262e-04,
        4.1559e-04, 1.8651e-03, 7.8925e-05, 1.4742e-04, 3.3138e-03, 6.6371e-05,
        2.5041e-04, 1.3573e-04, 1.5374e-04, 5.8856e-03, 4.8682e-04, 4.2764e-04,
        1.7521e-04, 8.9178e-05, 2.1308e-04, 3.7362e-05, 2.6316e-04, 6.6276e-05,
        2.6721e-03, 1.4006e-04, 4.3890e-04, 5.8330e-04, 1.0714e-03, 4.5381e-04,
        1.0970e-04, 2.0364e-04, 2.9930e-04, 3.7717e-04, 1.7047e-04, 5.2378e-03,
        1.9940e-03, 3.7923e-03, 7.0829e-04, 5.5359e-04, 2.4166e-04, 5.7558e-04,
        2.9311e-04, 2.5182e-04, 1.3878e-04, 7.0825e-05, 2.9827e-03, 5.1492e-04,
        2.9917e-03, 8.0812e-04, 1.1228e-03, 1.2668e-03, 6.4095e-04, 2.2063e-04,
        3.6503e-04, 2.2886e-04, 1.6049e-03, 1.3590e-03, 5.7641e-04, 2.1487e-03,
        7.2547e-04, 3.6092e-04, 6.0120e-04, 1.8739e-04, 1.1249e-03, 6.7967e-04,
        4.4262e-04, 7.2630e-04, 1.6781e-03, 2.5882e-04, 2.3473e-04, 6.8454e-04,
        3.8802e-04, 1.9346e-04, 3.0368e-04, 2.4101e-04, 1.0729e-04, 1.3567e-03,
        3.1660e-04, 1.1865e-04, 1.5587e-03, 9.8870e-04, 3.9063e-04, 1.5378e-03,
        5.1698e-04, 1.8293e-04, 1.6502e-03, 9.2984e-05, 1.2523e-04, 4.9950e-04,
        5.1643e-05, 2.0715e-04, 1.1227e-04, 1.6636e-04, 5.3169e-04, 3.0719e-04,
        1.3117e-04, 7.1095e-04, 2.1649e-04, 1.4805e-04, 3.7067e-03, 1.6240e-04,
        1.2460e-03, 1.3877e-03, 5.9621e-04, 5.9777e-04, 2.9048e-03, 4.3819e-04,
        1.9231e-03, 8.8665e-05, 2.4072e-04, 8.7117e-05, 2.6943e-03, 8.2753e-04,
        7.7080e-05, 5.3668e-05, 7.3884e-05, 3.5513e-04, 1.4622e-03, 1.7088e-04,
        3.6715e-04, 1.3315e-04, 9.2575e-05, 1.0509e-03, 1.7843e-03, 8.4304e-04,
        2.2582e-04, 2.5549e-04, 2.8500e-04, 1.5569e-03, 2.0650e-03, 3.2312e-04,
        1.9285e-04, 1.1453e-04, 6.1484e-05, 9.5875e-05, 1.3459e-04, 1.2110e-03,
        3.1457e-05, 8.0780e-05, 1.2596e-04, 3.3376e-04, 1.3478e-03, 1.1570e-03,
        6.7824e-04, 9.4116e-05, 1.0801e-03, 1.0898e-03, 1.3956e-04, 3.1984e-04,
        1.0389e-03, 3.3070e-04, 5.1141e-04, 1.3665e-03, 6.7899e-04, 6.8978e-04,
        1.1446e-04, 9.0436e-05, 1.4004e-03, 1.5454e-04, 1.5317e-03, 2.3414e-03,
        3.1671e-04, 3.1810e-04, 6.5882e-05, 3.2460e-04, 2.4943e-04, 1.3033e-04,
        1.3272e-04, 1.3922e-04, 1.6508e-04, 2.4418e-04, 2.9172e-03, 1.0683e-03,
        4.2079e-04, 1.4216e-03, 7.1419e-04, 6.8535e-04, 1.3751e-04, 2.0863e-04,
        1.5738e-04, 6.9977e-05, 5.9529e-05, 1.8920e-03, 7.1294e-04, 1.4796e-04,
        1.3041e-04, 3.2959e-04, 1.9759e-04, 1.1747e-03, 1.6905e-04, 1.0899e-03,
        1.7641e-03, 3.5038e-04, 1.1065e-03, 2.2375e-04, 3.0736e-04, 1.9999e-04,
        4.2383e-04, 4.7999e-04, 3.4956e-04, 4.5811e-04, 1.7354e-03, 1.4954e-03,
        1.8510e-04, 7.0948e-04, 7.5143e-04, 2.9702e-04, 1.4247e-04, 2.2035e-04,
        2.3948e-04, 1.4767e-04, 4.8845e-04, 3.4665e-04, 1.1177e-03, 7.8492e-04,
        7.8270e-04, 7.9637e-05, 5.7615e-04, 8.1523e-04, 2.4836e-04, 3.1072e-04,
        2.2041e-04, 6.6232e-04, 3.4220e-04, 2.5615e-04, 3.7871e-04, 2.8322e-04,
        2.4299e-04, 7.1584e-04, 5.8687e-04, 1.8075e-04, 5.5251e-04, 4.5818e-04,
        4.3124e-05, 3.0386e-04, 2.6972e-04, 3.3971e-04, 4.2237e-04, 1.7660e-04,
        8.2643e-04, 3.3631e-04, 2.1249e-04, 1.6315e-04, 1.3292e-04, 6.8085e-04,
        1.6889e-04, 1.3154e-04, 8.1449e-04, 2.9455e-04, 1.3064e-04, 6.6002e-04,
        4.7170e-04, 9.3362e-05, 2.4353e-04, 1.7564e-04, 9.9970e-04, 5.5488e-04,
        2.8446e-04, 3.9922e-04, 4.9377e-04, 3.6857e-04, 1.4351e-04, 3.0424e-04,
        7.6145e-05, 1.5652e-04, 3.5473e-04, 5.4779e-04, 3.3218e-04, 3.0211e-04,
        7.3397e-05, 1.0218e-04, 1.9741e-04, 1.9042e-04, 2.9118e-04, 1.6886e-03,
        4.7342e-04, 2.2418e-04, 2.1301e-04, 7.6505e-04, 2.6668e-04, 2.8254e-03,
        5.7248e-03, 1.8683e-02, 1.5817e-02, 7.0450e-03, 4.7154e-03, 1.7829e-02,
        1.3807e-02, 3.4434e-03, 5.2408e-03, 1.6935e-03, 5.2450e-03, 5.3243e-03,
        1.1547e-02, 2.0314e-02, 1.6576e-02, 2.3031e-02, 7.8227e-03, 5.3065e-03,
        2.1014e-02, 1.7095e-02, 1.2505e-02, 1.4595e-02, 1.2080e-02, 2.3408e-02,
        2.2562e-02, 9.3217e-03, 8.5321e-03, 1.3974e-02, 5.5444e-02, 5.5628e-02,
        7.9437e-02, 1.5291e-01, 0.0000e+00]),tensor([0.0000e+00, 3.7527e-04, 2.4209e-02, 5.1165e-04, 3.0273e-03, 3.5504e-03,
        2.5006e-02, 1.1472e-03, 2.5603e-04, 1.9944e-04, 7.1503e-04, 9.2345e-04,
        3.8045e-04, 9.0290e-04, 2.3770e-04, 1.6494e-03, 2.0125e-04, 1.0046e-03,
        4.6564e-04, 1.0641e-04, 8.0486e-05, 7.5734e-05, 7.2251e-05, 6.4995e-04,
        1.0731e-03, 1.0192e-03, 9.9633e-04, 7.7063e-04, 1.4491e-04, 9.7124e-04,
        3.1882e-04, 6.0490e-04, 2.4949e-04, 2.5298e-03, 7.6902e-04, 4.4151e-04,
        2.1893e-03, 2.4148e-04, 8.0253e-04, 1.2984e-03, 1.2308e-04, 4.1167e-04,
        1.0404e-04, 1.1836e-04, 8.6788e-05, 1.1437e-04, 9.0635e-04, 2.6253e-03,
        9.5166e-05, 9.0361e-05, 5.1893e-04, 5.8391e-04, 3.6964e-04, 3.7541e-04,
        1.1868e-04, 2.1837e-04, 1.5633e-04, 4.5735e-04, 3.5071e-04, 4.8452e-04,
        3.8972e-04, 3.7500e-04, 1.3175e-04, 4.2382e-04, 1.4688e-04, 4.2541e-03,
        1.2211e-04, 1.0060e-04, 2.3876e-04, 7.9750e-05, 4.3144e-05, 7.0366e-05,
        7.8075e-05, 1.0050e-03, 9.8792e-05, 5.1078e-04, 7.1544e-05, 7.1420e-05,
        2.2981e-04, 3.6446e-04, 1.6555e-04, 1.0677e-03, 1.8881e-04, 2.9045e-04,
        1.6101e-04, 5.2993e-04, 1.0249e-04, 2.7795e-04, 6.1587e-05, 6.4802e-05,
        6.3160e-04, 4.9293e-04, 5.6933e-05, 4.0785e-04, 1.9307e-04, 1.2847e-04,
        1.6575e-04, 5.6759e-05, 4.6591e-04, 1.5419e-03, 2.6877e-04, 3.5281e-04,
        8.1498e-04, 8.5017e-05, 1.1198e-04, 1.7777e-03, 3.9791e-03, 2.6939e-04,
        4.1262e-04, 8.4331e-05, 4.5109e-04, 5.6750e-05, 8.7777e-05, 6.8834e-05,
        1.1661e-03, 3.7372e-04, 8.8364e-05, 5.2448e-03, 1.4234e-04, 1.9669e-04,
        3.8911e-04, 1.3621e-03, 6.9545e-05, 1.4971e-04, 3.0259e-03, 5.3685e-05,
        2.3050e-04, 1.1250e-04, 1.4293e-04, 3.7542e-03, 4.9643e-04, 4.8548e-04,
        1.7088e-04, 6.6156e-05, 2.0813e-04, 2.5902e-05, 2.0650e-04, 5.9097e-05,
        2.7199e-03, 1.3259e-04, 4.9225e-04, 6.3547e-04, 1.0103e-03, 4.4151e-04,
        8.4881e-05, 1.9613e-04, 2.6079e-04, 2.9123e-04, 1.2946e-04, 3.7397e-03,
        1.5665e-03, 2.5166e-03, 6.2825e-04, 4.3050e-04, 2.3572e-04, 4.3754e-04,
        2.5971e-04, 2.2584e-04, 1.0998e-04, 5.8023e-05, 2.3458e-03, 4.7330e-04,
        2.1404e-03, 7.7155e-04, 1.0774e-03, 1.1461e-03, 4.8787e-04, 1.6936e-04,
        3.3003e-04, 1.9101e-04, 1.3928e-03, 1.2441e-03, 5.5680e-04, 1.6892e-03,
        7.6208e-04, 3.6399e-04, 6.0176e-04, 1.4857e-04, 8.7735e-04, 6.6970e-04,
        3.6637e-04, 5.5491e-04, 1.2795e-03, 2.5400e-04, 2.2441e-04, 7.1092e-04,
        3.4440e-04, 2.4999e-04, 2.9965e-04, 1.7470e-04, 8.7149e-05, 1.0946e-03,
        3.2034e-04, 1.0773e-04, 1.1192e-03, 9.2782e-04, 3.2664e-04, 1.1236e-03,
        5.5237e-04, 1.8262e-04, 1.1805e-03, 7.8023e-05, 9.8014e-05, 3.6418e-04,
        4.5365e-05, 1.5890e-04, 1.0501e-04, 1.5864e-04, 4.3963e-04, 2.9711e-04,
        1.1857e-04, 5.6256e-04, 1.6926e-04, 1.2072e-04, 2.6202e-03, 1.3534e-04,
        1.0277e-03, 1.2934e-03, 5.4015e-04, 5.9550e-04, 2.2230e-03, 4.6109e-04,
        1.4062e-03, 6.8042e-05, 1.9278e-04, 7.3828e-05, 2.1232e-03, 6.2548e-04,
        6.3179e-05, 4.8099e-05, 6.8123e-05, 4.0996e-04, 1.3153e-03, 1.8938e-04,
        2.5884e-04, 1.1130e-04, 1.1368e-04, 9.8072e-04, 1.3468e-03, 9.0222e-04,
        2.2533e-04, 2.7207e-04, 2.8836e-04, 1.4193e-03, 2.1463e-03, 3.1447e-04,
        1.4726e-04, 1.0032e-04, 5.5164e-05, 1.0315e-04, 1.2429e-04, 8.8718e-04,
        3.0066e-05, 7.3808e-05, 1.2546e-04, 2.5880e-04, 1.3401e-03, 1.2035e-03,
        5.0160e-04, 8.8902e-05, 9.8793e-04, 9.5747e-04, 1.4193e-04, 3.0159e-04,
        9.4564e-04, 2.9823e-04, 5.3440e-04, 1.6080e-03, 7.1295e-04, 5.5620e-04,
        1.1345e-04, 7.7348e-05, 1.1189e-03, 1.2966e-04, 1.0232e-03, 1.6987e-03,
        3.2163e-04, 3.4240e-04, 6.2325e-05, 3.5986e-04, 1.9843e-04, 1.1652e-04,
        1.1408e-04, 1.2543e-04, 1.4517e-04, 2.1074e-04, 2.2967e-03, 9.1596e-04,
        4.5366e-04, 1.1387e-03, 7.6343e-04, 6.2907e-04, 1.0968e-04, 2.0712e-04,
        1.2284e-04, 5.8052e-05, 4.7473e-05, 1.3672e-03, 5.1857e-04, 1.4392e-04,
        1.1859e-04, 2.9894e-04, 2.0434e-04, 1.2070e-03, 1.3819e-04, 1.0212e-03,
        1.3608e-03, 3.2752e-04, 1.0328e-03, 2.5823e-04, 3.2021e-04, 1.8210e-04,
        4.4925e-04, 5.8280e-04, 3.2790e-04, 3.5436e-04, 1.7055e-03, 9.0044e-04,
        1.6526e-04, 5.8447e-04, 7.2780e-04, 2.4172e-04, 1.2404e-04, 1.6451e-04,
        2.1554e-04, 1.8300e-04, 4.4454e-04, 3.1063e-04, 1.0828e-03, 9.0842e-04,
        6.6755e-04, 7.8261e-05, 6.9944e-04, 6.6808e-04, 2.4395e-04, 2.8201e-04,
        2.7105e-04, 6.6973e-04, 3.4161e-04, 2.1703e-04, 3.2073e-04, 3.1784e-04,
        2.5546e-04, 6.9493e-04, 5.5594e-04, 1.7716e-04, 5.5889e-04, 4.3985e-04,
        4.0694e-05, 2.8725e-04, 2.4686e-04, 3.1270e-04, 4.8695e-04, 2.0807e-04,
        9.0556e-04, 4.1130e-04, 2.1399e-04, 1.6982e-04, 1.4916e-04, 6.4548e-04,
        1.8126e-04, 1.4033e-04, 6.6153e-04, 2.9769e-04, 1.4084e-04, 5.2591e-04,
        5.1674e-04, 9.3715e-05, 2.6210e-04, 1.5229e-04, 8.1169e-04, 4.6605e-04,
        3.1195e-04, 3.5210e-04, 4.4225e-04, 4.2908e-04, 1.5262e-04, 2.5398e-04,
        8.4870e-05, 1.4592e-04, 3.2431e-04, 5.4743e-04, 3.4397e-04, 3.1975e-04,
        6.2429e-05, 1.0179e-04, 2.0036e-04, 1.8307e-04, 2.7642e-04, 1.3787e-03,
        4.5245e-04, 2.8214e-04, 2.1180e-04, 6.1089e-04, 2.2577e-04, 2.5793e-03,
        4.8214e-03, 1.6668e-02, 1.3033e-02, 5.5357e-03, 3.9850e-03, 1.5119e-02,
        1.2765e-02, 3.3684e-03, 4.8305e-03, 2.2698e-03, 4.3634e-03, 4.8306e-03,
        1.0727e-02, 1.7805e-02, 1.3445e-02, 1.7157e-02, 5.7789e-03, 4.4375e-03,
        1.6813e-02, 1.2969e-02, 9.4222e-03, 1.3461e-02, 1.1035e-02, 2.0561e-02,
        1.8760e-02, 8.0980e-03, 8.6777e-03, 1.3407e-02, 4.8067e-02, 4.1227e-02,
        5.6328e-02, 1.6958e-01, 1.1816e-01, 0.0000e+00])...]

Now I’ll perform a heterogenous stack on the prompt attention and the response attention. In other words, shorter vectors are padded to equal the length of the longest vector.

437
(437,
 [tensor([1]),
  tensor([9.9316e-01, 4.2319e-06, 4.2319e-06, 8.5831e-06, 8.5831e-06, 4.2319e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.7142e-05, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 2.5558e-04,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.7142e-05, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          1.2374e-04, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 2.7037e-04, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 2.5558e-04, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          1.2374e-04, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.7142e-05,
          8.7142e-05, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 1.2374e-04,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 7.3731e-05, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 1.2374e-04, 8.5831e-06,
          4.5061e-04, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 1.5235e-04,
          7.3731e-05, 8.7142e-05, 8.5831e-06, 8.5831e-06, 8.5831e-06, 1.5235e-04,
          3.4881e-04, 3.3545e-04, 8.5831e-06, 8.7142e-05, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 4.2319e-06, 8.5831e-06, 4.2319e-06, 4.2319e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06, 8.5831e-06,
          8.5831e-06, 8.5831e-06], dtype=torch.float16),
  tensor([0.0000e+00, 9.1357e-01, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 2.2173e-05, 5.9724e-05, 5.9724e-05, 4.1842e-05, 4.1842e-05,
          4.1842e-05, 2.2173e-05, 4.1842e-05, 2.2173e-05, 7.8619e-05, 4.1842e-05,
          4.1842e-05, 1.0544e-04, 2.2173e-05, 2.2173e-05, 1.4234e-04, 2.2173e-05,
          7.8619e-05, 4.1127e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 4.1842e-05,
          5.9724e-05, 5.9724e-05, 5.9724e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 2.2173e-05, 5.9724e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          1.4234e-04, 2.2173e-05, 2.2173e-05, 2.2173e-05, 4.0412e-04, 2.2173e-05,
          4.1842e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 4.1842e-05, 2.2173e-05, 2.2173e-05, 5.9724e-05, 2.2173e-05,
          1.0544e-04, 1.0544e-04, 5.9724e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 4.1842e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 4.1842e-05, 1.0544e-04, 5.9724e-05, 8.6427e-05, 2.2173e-05,
          2.2173e-05, 4.1127e-05, 2.2173e-05, 1.0544e-04, 1.0544e-04, 1.0544e-04,
          4.4084e-04, 4.1842e-05, 4.1842e-05, 8.6427e-05, 2.2173e-05, 2.2173e-05,
          2.2173e-05, 2.2173e-05, 1.0544e-04, 4.1127e-05, 2.2173e-05, 4.1842e-05,
          4.2200e-04, 1.0544e-04, 4.4084e-04, 2.2173e-05, 2.2173e-05, 4.1842e-05,
          4.1127e-05, 1.6022e-04, 1.0544e-04, 2.2173e-05, 1.0544e-04, 1.0544e-04,
          2.2173e-05, 1.4234e-04, 1.4234e-04, 3.5763e-04, 1.4234e-04, 2.2173e-05,
          7.8619e-05, 7.8619e-05, 7.8619e-05, 3.5763e-04, 3.5763e-04, 1.0544e-04,
          4.1842e-05, 1.0544e-04, 1.0544e-04, 2.2173e-05, 1.0544e-04, 2.2173e-05,
          1.0544e-04, 1.0544e-04, 1.0544e-04, 3.8767e-04, 1.0544e-04, 1.4234e-04,
          1.0544e-04, 1.9872e-04, 4.1127e-05, 2.2173e-05, 2.2173e-05, 4.1413e-04,
          1.0544e-04, 1.0544e-04, 5.9724e-05, 5.9724e-05, 4.4084e-04, 3.7551e-04,
          5.9724e-05, 3.9995e-05, 5.9724e-05, 4.1842e-05, 2.2173e-05, 1.4234e-04,
          1.0544e-04, 1.4234e-04, 4.4084e-04, 5.9724e-05, 1.4234e-04, 1.0544e-04,
          2.2173e-05, 3.9995e-05, 1.0544e-04, 2.2173e-05, 4.1127e-05, 1.0544e-04,
          4.4084e-04, 1.0544e-04, 3.5763e-04, 3.9625e-04, 4.4084e-04, 2.2173e-05,
          3.9995e-05, 2.2173e-05, 2.2173e-05, 1.4234e-04, 1.4234e-04, 2.2173e-05,
          1.9872e-04, 2.2173e-05, 2.2173e-05, 4.4084e-04, 2.2173e-05, 2.2173e-05,
          4.4084e-04, 2.2173e-05, 4.1127e-05, 1.0544e-04, 4.4084e-04, 3.7670e-04,
          1.0544e-04, 7.8619e-05, 7.8619e-05, 6.9809e-04, 4.4084e-04, 5.9724e-05,
          2.2173e-05, 5.3120e-04, 1.4234e-04, 5.0783e-04, 1.0544e-04, 1.6129e-04,
          1.0544e-04, 3.8767e-04, 1.4234e-04, 4.1127e-05, 1.0544e-04, 1.0544e-04,
          5.8949e-05, 4.3797e-04, 7.8619e-05, 4.4084e-04, 3.9434e-04, 1.0544e-04,
          5.8949e-05, 1.0544e-04, 2.2173e-05, 4.4084e-04, 3.9995e-05, 8.6689e-04,
          5.9724e-05, 8.5783e-04, 4.6134e-04, 2.2554e-04, 3.5763e-04, 2.2173e-05,
          1.0544e-04, 1.0544e-04, 1.9872e-04, 4.1127e-05, 2.2173e-05, 3.9995e-05,
          4.3797e-04, 1.0544e-04, 1.0544e-04, 5.8949e-05, 3.9995e-05, 3.7551e-04,
          4.7493e-04, 2.2173e-05, 7.8619e-05, 7.8619e-05, 8.6689e-04, 5.3120e-04,
          5.5790e-04, 5.0783e-04, 1.4234e-04, 1.6129e-04, 1.4234e-04, 1.4234e-04,
          2.2554e-04, 1.4234e-04, 1.4234e-04, 1.4234e-04, 2.2173e-05, 1.0544e-04,
          5.8949e-05, 4.4084e-04, 5.8949e-05, 5.8949e-05, 2.2173e-05, 4.9257e-04,
          5.8949e-05, 4.1127e-05, 1.0544e-04, 4.4084e-04, 4.4084e-04, 1.4234e-04,
          5.5790e-04, 1.0544e-04, 1.4234e-04, 4.2462e-04, 1.4234e-04, 1.6022e-04,
          1.6022e-04, 1.0544e-04, 1.4234e-04, 4.1413e-04, 4.3797e-04, 1.0544e-04,
          4.4084e-04, 2.2173e-05, 2.2173e-05, 2.2173e-05, 1.0544e-04, 1.0544e-04,
          3.7670e-04, 7.8619e-05, 3.6287e-04, 1.0544e-04, 1.4234e-04, 1.4234e-04,
          1.4234e-04, 1.7917e-04, 1.6129e-04, 1.6129e-04, 5.0783e-04, 1.4234e-04,
          1.4234e-04, 1.0544e-04, 1.0544e-04, 5.2118e-04, 4.9448e-04, 4.4084e-04,
          5.8949e-05, 4.9448e-04, 1.0544e-04, 1.0544e-04, 5.8949e-05, 4.4084e-04,
          1.7405e-03, 1.0544e-04, 1.0672e-03, 7.8619e-05, 4.1127e-05, 1.4234e-04,
          1.4234e-04, 1.0544e-04, 5.0783e-04, 2.2554e-04, 1.4234e-04, 1.4234e-04,
          5.0783e-04, 4.8113e-04, 7.8619e-05, 4.5586e-04, 5.2118e-04, 8.9359e-04,
          1.0544e-04, 1.5202e-03, 8.9359e-04, 7.8619e-05, 2.2173e-05, 1.0214e-03,
          7.8619e-05, 5.8949e-05, 6.4135e-04, 1.0281e-03, 5.0783e-04, 8.9359e-04,
          1.0544e-04, 1.6129e-04, 6.7806e-04, 4.4084e-04, 1.6129e-04, 4.6134e-04,
          1.6136e-03, 1.0544e-04, 4.7469e-04, 3.9995e-05, 4.4084e-04, 4.4084e-04,
          1.0544e-04, 8.5711e-05, 4.4084e-04, 1.0214e-03, 8.9359e-04, 1.0544e-04,
          4.4084e-04, 4.4084e-04, 2.2554e-04, 4.1127e-05, 1.0544e-04, 4.1127e-05,
          4.1127e-05, 6.5136e-04, 4.1127e-05, 1.0544e-04, 2.2173e-05, 1.6129e-04,
          1.0544e-04, 5.8949e-05, 5.8949e-05, 5.8949e-05, 4.4084e-04, 6.1417e-04,
          8.1539e-04, 4.4084e-04, 4.4084e-04, 7.8619e-05, 5.9462e-04, 5.0507e-03,
          6.4135e-04, 5.9462e-04, 1.0138e-03, 5.8949e-05, 5.0783e-04, 1.1415e-03,
          6.5136e-04, 1.8978e-03, 1.6129e-04, 1.7917e-04, 1.7917e-04, 2.2173e-05,
          8.6427e-05, 8.6427e-05, 6.8665e-05, 8.6427e-05, 1.0544e-04, 8.6427e-05,
          1.0109e-03, 8.6427e-05, 8.6427e-05, 8.6427e-05, 8.6427e-05, 8.6427e-05,
          4.8995e-05, 8.6427e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.0405e-01, 5.2148e-01, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 2.4819e-04, 9.5010e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 9.5010e-05, 1.4782e-04, 4.2737e-05, 2.2447e-04, 9.5010e-05,
          7.1168e-05, 2.2066e-04, 9.5010e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 1.7166e-04, 9.5606e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 2.7680e-04, 4.2737e-05,
          9.5010e-05, 9.5010e-05, 9.5606e-05, 4.2737e-05, 1.7166e-04, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 9.5606e-05, 4.2737e-05, 4.2737e-05, 9.5606e-05, 4.2737e-05,
          1.6844e-04, 1.6844e-04, 9.5606e-05, 1.1945e-04, 9.5010e-05, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 1.7166e-04, 1.1945e-04, 9.5010e-05,
          4.2737e-05, 4.2737e-05, 1.3232e-04, 4.2737e-05, 1.3232e-04, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 1.6844e-04, 1.6844e-04, 2.2066e-04,
          1.3232e-04, 7.1168e-05, 9.5010e-05, 1.3232e-04, 9.5010e-05, 9.5010e-05,
          4.2737e-05, 4.2737e-05, 9.5010e-05, 9.5010e-05, 4.2737e-05, 6.7174e-05,
          4.2737e-05, 2.2066e-04, 1.6844e-04, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          4.2737e-05, 1.6844e-04, 9.5606e-05, 1.8466e-04, 1.6844e-04, 2.2066e-04,
          9.5010e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05,
          6.7174e-05, 4.2737e-05, 9.5010e-05, 2.2388e-04, 1.7166e-04, 2.2066e-04,
          1.1945e-04, 2.2066e-04, 1.6844e-04, 4.2737e-05, 1.9598e-04, 4.2737e-05,
          3.4690e-04, 1.6844e-04, 1.3232e-04, 2.9469e-04, 1.6844e-04, 9.5010e-05,
          1.4782e-04, 1.4782e-04, 9.5010e-05, 4.2737e-05, 4.2737e-05, 1.9598e-04,
          1.3161e-04, 9.5010e-05, 1.4782e-04, 1.1945e-04, 1.6844e-04, 1.7166e-04,
          4.2737e-05, 4.2737e-05, 9.5010e-05, 4.2737e-05, 4.2737e-05, 9.5606e-05,
          2.9469e-04, 2.2066e-04, 2.2066e-04, 1.4782e-04, 6.7174e-05, 9.5606e-05,
          4.2737e-05, 1.1945e-04, 2.2447e-04, 4.2737e-05, 9.5010e-05, 2.2066e-04,
          3.4952e-04, 2.2066e-04, 1.7166e-04, 2.4819e-04, 2.0003e-04, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 4.2737e-05, 6.7174e-05, 4.2737e-05,
          2.2066e-04, 4.2737e-05, 2.2447e-04, 2.2388e-04, 9.5010e-05, 9.5010e-05,
          1.9622e-04, 9.5010e-05, 9.5010e-05, 1.6844e-04, 2.2066e-04, 1.7166e-04,
          2.8920e-04, 1.4782e-04, 4.2737e-05, 2.7680e-04, 3.4952e-04, 4.2737e-05,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 1.6844e-04, 1.6844e-04, 9.5606e-05,
          1.6844e-04, 1.6844e-04, 2.0325e-04, 9.5010e-05, 1.3232e-04, 3.4952e-04,
          3.4952e-04, 4.2737e-05, 1.8394e-04, 2.6131e-04, 1.7166e-04, 2.2066e-04,
          9.5010e-05, 9.5606e-05, 4.2737e-05, 2.2388e-04, 4.2737e-05, 1.4410e-03,
          4.2737e-05, 2.2769e-04, 2.9230e-04, 1.6844e-04, 6.7282e-04, 4.2737e-05,
          1.6844e-04, 2.2066e-04, 3.0828e-04, 9.5010e-05, 4.2737e-05, 1.4782e-04,
          4.2737e-05, 2.2066e-04, 2.2066e-04, 9.5010e-05, 4.2737e-05, 1.7166e-04,
          4.2737e-05, 4.2737e-05, 4.2737e-05, 9.5010e-05, 1.3123e-03, 6.7174e-05,
          2.0325e-04, 1.6844e-04, 4.2737e-05, 2.5558e-04, 4.2737e-05, 6.7174e-05,
          2.2066e-04, 2.0325e-04, 4.2737e-05, 2.5558e-04, 9.5010e-05, 1.4782e-04,
          1.4782e-04, 2.2066e-04, 9.5010e-05, 9.5010e-05, 9.5010e-05, 6.7174e-05,
          9.5010e-05, 9.5010e-05, 9.5010e-05, 3.4952e-04, 1.7166e-04, 2.0325e-04,
          1.6844e-04, 1.6844e-04, 4.2737e-05, 2.0325e-04, 2.5558e-04, 6.7174e-05,
          2.5558e-04, 2.2388e-04, 2.5606e-04, 6.7282e-04, 9.5010e-05, 2.2066e-04,
          2.9731e-04, 9.5010e-05, 9.5010e-05, 9.5010e-05, 1.1945e-04, 9.5010e-05,
          2.2388e-04, 4.2737e-05, 1.8466e-04, 2.2066e-04, 4.2737e-05, 4.2737e-05,
          2.0325e-04, 2.5558e-04, 2.5558e-04, 4.2737e-05, 2.7990e-04, 2.0325e-04,
          2.5558e-04, 1.3113e-04, 3.4952e-04, 2.2066e-04, 1.8394e-04, 2.2066e-04,
          9.5010e-05, 1.8394e-04, 9.5010e-05, 2.2066e-04, 9.5010e-05, 2.2388e-04,
          2.2066e-04, 1.4782e-04, 2.2769e-04, 1.7166e-04, 4.2737e-05, 6.7174e-05,
          9.5010e-05, 1.6844e-04, 1.5812e-03, 1.5612e-03, 2.5558e-04, 2.5558e-04,
          1.6174e-03, 3.4451e-04, 4.2737e-05, 1.1945e-04, 1.1945e-04, 1.7166e-04,
          1.4782e-04, 3.4952e-04, 1.9598e-04, 4.2737e-05, 9.5010e-05, 1.1945e-04,
          9.5010e-05, 9.5010e-05, 1.9228e-04, 2.5606e-04, 2.2066e-04, 6.9714e-04,
          1.3113e-04, 9.5010e-05, 2.0103e-03, 2.5988e-04, 1.7099e-03, 3.4451e-04,
          3.4451e-04, 1.3232e-04, 9.5010e-05, 9.5010e-05, 9.7370e-04, 2.9683e-04,
          9.5010e-05, 1.3113e-04, 1.3113e-04, 1.3113e-04, 1.7166e-04, 1.3113e-04,
          3.4952e-04, 9.7370e-04, 3.4690e-04, 9.5010e-05, 2.2066e-04, 4.2737e-05,
          2.5988e-04, 6.7174e-05, 9.5010e-05, 2.2066e-04, 4.2737e-05, 9.5010e-05,
          9.5606e-05, 6.7174e-05, 1.1945e-04, 9.5010e-05, 9.7370e-04, 9.2554e-04,
          2.1896e-03, 2.2388e-04, 3.4952e-04, 9.5010e-05, 3.1590e-04, 1.5545e-04,
          3.4952e-04, 1.0624e-03, 7.4911e-04, 9.5010e-05, 2.9159e-04, 3.8123e-04,
          3.4451e-04, 1.5631e-03, 2.9159e-04, 3.1590e-04, 2.5558e-04, 7.1168e-05,
          1.6844e-04, 1.3232e-04, 1.0800e-04, 1.0800e-04, 7.1168e-05, 7.1168e-05,
          4.2737e-05, 1.8466e-04, 1.8466e-04, 1.3232e-04, 1.3232e-04, 2.5868e-04,
          1.0800e-04, 4.2737e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.1313e-01, 3.6475e-01, 3.7573e-01, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.6296e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.6296e-04, 1.6296e-04,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          2.3830e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 9.4235e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          9.4235e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 2.9683e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 2.3830e-04, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.6296e-04, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 9.4235e-05, 1.8954e-05, 1.8954e-05,
          9.4235e-05, 1.8954e-05, 1.8954e-05, 2.9683e-05, 1.8954e-05, 2.9683e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 2.7037e-04,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 9.4235e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.6296e-04, 3.0661e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 5.2261e-04,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          4.1437e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.6296e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 2.9683e-05,
          1.8954e-05, 1.8954e-05, 9.4235e-05, 1.8954e-05, 5.2261e-04, 1.8954e-05,
          1.6296e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 5.2261e-04, 1.8954e-05, 1.6296e-04,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05, 9.4235e-05,
          5.2261e-04, 1.8954e-05, 1.8954e-05, 9.4235e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 4.6873e-04, 1.8954e-05, 3.0661e-04, 1.8954e-05,
          1.8954e-05, 9.4235e-05, 9.4235e-05, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.6296e-04, 1.8954e-05, 9.4235e-05, 9.4235e-05, 9.4235e-05,
          9.4235e-05, 1.6296e-04, 2.7037e-04, 2.7037e-04, 1.8954e-05, 1.8954e-05,
          2.7037e-04, 2.3830e-04, 5.4407e-04, 9.4235e-05, 1.8954e-05, 1.8954e-05,
          2.3830e-04, 3.4571e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 5.2261e-04,
          1.6296e-04, 3.5715e-04, 7.7391e-04, 1.8954e-05, 9.4235e-05, 4.8971e-04,
          2.7037e-04, 1.8954e-05, 5.4407e-04, 4.1437e-04, 5.4407e-04, 7.7391e-04,
          9.4235e-05, 1.8954e-05, 4.8971e-04, 9.4235e-05, 9.4235e-05, 8.3160e-04,
          8.3160e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05, 7.4196e-04, 9.4235e-05,
          9.4235e-05, 9.4235e-05, 9.4235e-05, 9.4235e-05, 7.7391e-04, 3.4571e-04,
          3.4571e-04, 5.9795e-04, 4.8971e-04, 9.4235e-05, 9.4235e-05, 1.8954e-05,
          9.4235e-05, 6.8808e-04, 1.8954e-05, 2.3830e-04, 1.8954e-05, 9.4235e-05,
          1.8954e-05, 1.8954e-05, 1.8954e-05, 9.4235e-05, 9.4235e-05, 4.6873e-04,
          5.3978e-03, 7.7391e-04, 6.3562e-04, 9.4235e-05, 5.4407e-04, 4.3678e-04,
          4.8971e-04, 6.8808e-04, 1.9722e-03, 9.4235e-05, 2.9278e-04, 7.2432e-04,
          6.8808e-04, 1.4896e-03, 5.8031e-04, 8.3160e-04, 8.3160e-04, 1.8954e-05,
          1.8954e-05, 1.6296e-04, 1.8954e-05, 5.6076e-04, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05, 5.9128e-04, 1.8954e-05, 1.8954e-05, 1.8954e-05,
          1.8954e-05, 1.8954e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.6965e-02, 2.0618e-01, 1.3904e-01, 5.3467e-01, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05,
          9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05,
          3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05,
          3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05,
          3.2604e-05, 9.1076e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 6.2990e-04, 9.1076e-05, 9.1076e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 5.0592e-04,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05,
          6.8808e-04, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 2.6283e-03, 9.1076e-05, 3.2604e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          8.9025e-04, 3.2604e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 9.1076e-05,
          9.1076e-05, 6.8808e-04, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          6.2990e-04, 9.1076e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          5.9509e-04, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05,
          3.2604e-05, 5.0592e-04, 9.1076e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          6.2990e-04, 3.2604e-05, 9.1076e-05, 3.2604e-05, 6.2990e-04, 9.1076e-05,
          3.2604e-05, 9.1076e-05, 9.1076e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 9.1076e-05,
          9.1076e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          5.0592e-04, 5.0592e-04, 6.2990e-04, 3.2604e-05, 9.1076e-05, 3.2604e-05,
          9.1076e-05, 3.2604e-05, 3.2604e-05, 4.1652e-04, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 9.1076e-05, 5.0592e-04, 3.2604e-05,
          1.1034e-03, 9.1076e-05, 9.1076e-05, 3.2604e-05, 6.2990e-04, 3.2604e-05,
          1.1034e-03, 3.2604e-05, 9.1076e-05, 3.2604e-05, 3.2604e-05, 1.8692e-03,
          6.2990e-04, 5.6648e-03, 3.2604e-05, 3.2604e-05, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 2.1803e-04, 5.3644e-04, 3.2604e-05,
          3.2604e-05, 3.2604e-05, 3.2604e-05, 6.4135e-04, 3.2604e-05, 3.2604e-05,
          3.2604e-05, 3.2604e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9551e-02, 1.6736e-01, 7.7209e-02, 1.2939e-01, 5.5371e-01,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 6.5279e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 6.5279e-04, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.5596e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.5596e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.5596e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 6.5279e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.5596e-04, 3.5596e-04, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.5596e-04, 3.5596e-04, 3.5596e-04, 6.5279e-04, 3.4034e-05,
          3.5596e-04, 3.5596e-04, 3.5596e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.5596e-04, 3.4034e-05, 3.4034e-05, 3.5596e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.0422e-03, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 9.7513e-04,
          3.5596e-04, 3.4034e-05, 3.5596e-04, 3.5596e-04, 3.5596e-04, 1.0500e-03,
          2.4223e-03, 3.0422e-03, 3.5596e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 2.6631e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 4.0102e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.7018e-02, 1.2720e-01, 8.0383e-02, 1.4319e-01, 2.4414e-01,
          3.1616e-01, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 2.1863e-04, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 2.1863e-04, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 2.1863e-04, 3.0994e-05, 2.1863e-04, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 2.1863e-04,
          3.0994e-05, 2.1863e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 4.9210e-04, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 1.8203e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          1.7023e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.2115e-04,
          3.0994e-05, 2.1863e-04, 3.0994e-05, 3.0994e-05, 4.5085e-04, 3.0994e-05,
          3.0994e-05, 2.1863e-04, 2.1863e-04, 3.0994e-05, 3.0994e-05, 4.9925e-04,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 1.8203e-04,
          2.1863e-04, 3.0994e-05, 3.6955e-04, 3.6955e-04, 3.0994e-05, 3.0994e-05,
          2.1863e-04, 2.1863e-04, 2.1863e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.8203e-04, 3.0994e-05,
          3.0994e-05, 3.0994e-05, 3.0994e-05, 1.7023e-04, 1.7023e-04, 3.0994e-05,
          3.2115e-04, 1.7023e-04, 3.1161e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          2.1863e-04, 3.0994e-05, 3.6955e-04, 1.7023e-04, 3.0994e-05, 3.2115e-04,
          1.7023e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.8203e-04, 3.0994e-05,
          1.7023e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 1.8203e-04,
          3.0994e-05, 3.0994e-05, 2.1863e-04, 3.0994e-05, 2.1863e-04, 2.1863e-04,
          2.1863e-04, 3.0994e-05, 3.6883e-04, 3.0994e-05, 1.8203e-04, 3.0994e-05,
          3.0994e-05, 1.0462e-03, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 1.8203e-04, 3.6955e-04, 1.7023e-04, 3.0994e-05, 3.0994e-05,
          1.8203e-04, 3.0994e-05, 1.8568e-03, 3.6955e-04, 2.1863e-04, 2.1863e-04,
          1.9035e-03, 6.5041e-04, 3.0994e-05, 1.8203e-04, 3.0994e-05, 3.2115e-04,
          3.0994e-05, 3.2115e-04, 3.2902e-04, 1.7023e-04, 3.0994e-05, 1.8203e-04,
          1.8203e-04, 3.0994e-05, 2.1863e-04, 1.8203e-04, 3.0994e-05, 6.0177e-04,
          3.0994e-05, 3.0994e-05, 1.5059e-03, 3.0994e-05, 7.0715e-04, 3.6955e-04,
          1.8568e-03, 3.0994e-05, 1.8203e-04, 1.8203e-04, 1.7023e-04, 1.7023e-04,
          3.2115e-04, 2.2948e-04, 3.0994e-05, 3.0994e-05, 3.2115e-04, 1.8203e-04,
          5.1975e-04, 5.1975e-04, 3.8052e-04, 3.0994e-05, 2.2948e-04, 3.0994e-05,
          2.2948e-04, 6.5041e-04, 3.0994e-05, 2.2948e-04, 3.0994e-05, 2.1863e-04,
          3.0994e-05, 1.8203e-04, 1.8203e-04, 3.0994e-05, 1.1435e-03, 1.8203e-04,
          1.1435e-03, 3.2115e-04, 3.0994e-05, 3.0994e-05, 3.6955e-04, 1.3895e-03,
          5.1975e-04, 3.6955e-04, 6.0177e-04, 2.2948e-04, 3.0994e-05, 5.6791e-04,
          1.3418e-03, 5.6791e-04, 4.1699e-04, 5.6791e-04, 3.6955e-04, 3.0994e-05,
          4.9210e-04, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 4.9210e-04, 3.0994e-05, 8.2016e-04, 3.0994e-05, 3.0994e-05,
          3.0994e-05, 3.0994e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.3732e-02, 1.1676e-01, 7.8064e-02, 2.0361e-01, 1.7615e-01,
          7.4280e-02, 2.6489e-01, 7.6532e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05,
          3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05,
          7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 7.5388e-04, 7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.3140e-04, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05,
          7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05,
          3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05,
          3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 7.5388e-04, 7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05,
          3.4511e-05, 7.6532e-05, 3.4511e-05, 7.9584e-04, 7.6532e-05, 7.6532e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 2.8944e-04, 7.6532e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 1.3952e-03,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05,
          6.8235e-04, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 4.2748e-04,
          3.8552e-04, 7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          1.3952e-03, 3.4511e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 7.6532e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05,
          7.6532e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05,
          7.6532e-05, 1.4019e-03, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.8552e-04, 7.6532e-05, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          3.3140e-04, 7.6532e-05, 3.4511e-05, 3.4511e-05, 3.8552e-04, 7.6532e-05,
          3.4511e-05, 4.2748e-04, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05,
          4.2748e-04, 3.4511e-05, 7.6532e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05,
          3.4511e-05, 7.6532e-05, 3.3140e-04, 7.6532e-05, 9.1124e-04, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.3140e-04, 7.6532e-05,
          1.2617e-03, 3.4511e-05, 7.6532e-05, 3.4511e-05, 7.6532e-05, 1.1473e-03,
          7.6532e-05, 7.6532e-05, 3.8552e-04, 3.4511e-05, 7.6532e-05, 3.4511e-05,
          3.0971e-04, 3.4511e-05, 3.4511e-05, 1.2617e-03, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05,
          1.4019e-03, 7.6532e-05, 7.6532e-05, 3.4511e-05, 3.8552e-04, 2.6774e-04,
          9.3269e-04, 3.4511e-05, 7.6532e-05, 7.6532e-05, 3.4511e-05, 7.7295e-04,
          1.1053e-03, 1.8854e-03, 3.4511e-05, 3.4511e-05, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 2.0146e-04, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05, 3.4511e-05, 4.5228e-04, 3.4511e-05, 3.4511e-05,
          3.4511e-05, 3.4511e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.9989e-02, 9.8633e-02, 5.9296e-02, 6.8542e-02, 1.7383e-01,
          1.8079e-01, 3.9612e-02, 2.9321e-01, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.3532e-04, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          6.2943e-05, 2.8372e-05, 6.7174e-05, 6.2943e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 1.0598e-04,
          2.3532e-04, 1.0598e-04, 2.3532e-04, 1.0598e-04, 6.7174e-05, 2.8372e-05,
          2.8372e-05, 1.0598e-04, 6.7174e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 1.0598e-04, 1.4043e-04,
          6.2943e-05, 2.8372e-05, 2.8372e-05, 1.0598e-04, 1.0598e-04, 1.0598e-04,
          1.0598e-04, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05, 6.2943e-05,
          6.2943e-05, 2.8372e-05, 4.3488e-04, 6.2943e-05, 1.0598e-04, 1.0598e-04,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          6.2943e-05, 1.0169e-04, 6.2943e-05, 2.8372e-05, 6.2943e-05, 6.2943e-05,
          2.8372e-05, 1.0598e-04, 2.8372e-05, 1.0598e-04, 6.2943e-05, 2.8372e-05,
          1.4043e-04, 1.0598e-04, 1.0598e-04, 6.2943e-05, 1.0598e-04, 2.8372e-05,
          6.2943e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05, 6.2943e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05, 2.8372e-05,
          1.0169e-04, 6.2943e-05, 2.8372e-05, 1.0598e-04, 2.8372e-05, 1.0598e-04,
          6.7174e-05, 4.0436e-04, 2.8372e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05,
          2.8372e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05, 6.2943e-05, 6.2943e-05,
          2.8372e-05, 6.2943e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05, 2.0468e-04,
          1.0169e-04, 2.0468e-04, 6.2943e-05, 4.3488e-04, 1.0598e-04, 3.0541e-04,
          1.0598e-04, 2.0468e-04, 6.7174e-05, 2.8372e-05, 2.8372e-05, 2.8372e-05,
          1.0169e-04, 2.8372e-05, 6.2943e-05, 6.2943e-05, 1.0169e-04, 6.2943e-05,
          6.7174e-05, 2.8372e-05, 2.8372e-05, 1.0598e-04, 2.4354e-04, 2.8372e-05,
          2.0468e-04, 6.2943e-05, 6.2943e-05, 6.7174e-05, 2.8372e-05, 2.0468e-04,
          1.0169e-04, 1.4043e-04, 2.8372e-05, 2.8372e-05, 6.8903e-05, 6.2943e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 3.7766e-04, 1.0169e-04, 6.2943e-05,
          2.8372e-05, 1.4043e-04, 1.7929e-04, 2.7800e-04, 6.7174e-05, 2.0468e-04,
          1.4043e-04, 1.4472e-04, 2.0468e-04, 2.8372e-05, 1.0169e-04, 6.7174e-05,
          2.8372e-05, 2.8372e-05, 2.8372e-05, 1.0765e-04, 6.2943e-05, 6.8903e-05,
          2.8372e-05, 1.0169e-04, 2.8372e-05, 6.2943e-05, 6.2943e-05, 6.2943e-05,
          1.4043e-04, 3.3236e-04, 3.7098e-04, 4.0555e-04, 2.6679e-04, 1.0169e-04,
          2.3103e-04, 4.4298e-04, 4.9734e-04, 2.4354e-04, 2.8372e-05, 3.0541e-04,
          2.8372e-05, 2.8372e-05, 6.2943e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05,
          2.8372e-05, 2.8372e-05, 6.7174e-05, 6.7174e-05, 6.2943e-05, 1.7929e-04,
          3.3236e-04, 3.2520e-04, 1.0598e-04, 2.3925e-04, 1.0598e-04, 2.4354e-04,
          6.0129e-04, 2.0468e-04, 1.0598e-04, 2.0468e-04, 2.8372e-05, 1.9479e-04,
          2.8372e-05, 1.0169e-04, 2.8372e-05, 2.8372e-05, 2.8372e-05, 1.0169e-04,
          6.7174e-05, 2.8372e-05, 2.8372e-05, 6.2943e-05, 4.7708e-04, 4.0293e-04,
          3.4428e-04, 6.5863e-05, 5.0354e-04, 3.3236e-04, 5.3024e-04, 3.6979e-04,
          3.6979e-04, 1.0169e-04, 4.6158e-04, 2.6679e-04, 2.8372e-05, 1.0765e-04,
          1.0461e-04, 5.1260e-05, 1.5593e-04, 2.8372e-05, 2.6107e-04, 2.2650e-04,
          1.0169e-04, 6.7174e-05, 6.7174e-05, 3.7766e-04, 1.0598e-04, 6.7174e-05,
          3.3236e-04, 3.3236e-04, 3.3236e-04, 2.7800e-04, 5.3024e-04, 1.4472e-04,
          1.0598e-04, 2.4354e-04, 6.7174e-05, 1.0169e-04, 2.8372e-05, 1.0169e-04,
          6.7174e-05, 7.6389e-04, 6.7174e-05, 2.8372e-05, 2.8372e-05, 1.0169e-04,
          4.7708e-04, 6.7174e-05, 2.7227e-04, 1.0169e-04, 6.7174e-05, 5.8842e-04,
          4.0293e-04, 4.7708e-04, 3.7098e-04, 4.0555e-04, 2.4354e-04, 2.3353e-04,
          6.1893e-04, 3.0541e-04, 1.9479e-04, 1.5593e-04, 1.9479e-04, 4.7708e-04,
          1.9479e-04, 1.0169e-04, 4.7994e-04, 1.0169e-04, 2.8372e-05, 2.7227e-04,
          1.0169e-04, 1.0598e-04, 1.2720e-04, 3.7098e-04, 6.8140e-04, 6.4182e-04,
          3.9482e-04, 2.3925e-04, 5.8270e-04, 2.5296e-04, 1.4472e-04, 5.3596e-04,
          3.7098e-04, 3.8862e-04, 3.5405e-04, 1.5593e-04, 6.7174e-05, 2.8372e-05,
          2.8372e-05, 2.8372e-05, 4.1938e-04, 3.8481e-04, 6.0463e-04, 2.1839e-04,
          6.7174e-05, 6.7174e-05, 1.4472e-04, 4.3821e-04, 2.8372e-05, 1.0598e-04,
          4.2939e-04, 6.0129e-04, 2.8372e-05, 4.8971e-04, 2.8372e-05, 1.0598e-04,
          7.2956e-04, 3.5405e-04, 3.5405e-04, 2.8372e-05, 2.8372e-05, 6.0892e-04,
          6.7174e-05, 4.7708e-04, 6.2847e-04, 1.9479e-04, 2.3353e-04, 9.3222e-04,
          3.1734e-04, 5.3024e-04, 8.4734e-04, 2.5725e-04, 9.3222e-04, 5.5695e-04,
          7.8106e-04, 2.4354e-04, 3.5596e-04, 6.8140e-04, 5.8270e-04, 2.8372e-05,
          3.0375e-04, 2.6917e-04, 3.0375e-04, 2.6917e-04, 5.7983e-04, 3.0375e-04,
          2.8372e-05, 8.3828e-04, 2.6917e-04, 1.0023e-03, 8.0013e-04, 1.4172e-03,
          2.6917e-04, 2.6917e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.2659e-02, 9.6497e-02, 5.3467e-02, 8.5327e-02, 2.0374e-01,
          1.8494e-01, 4.6234e-02, 9.8450e-02, 1.4795e-01, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 5.5313e-04, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 1.1212e-04, 6.3896e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          1.1611e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 1.1611e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04, 6.3896e-05,
          6.3896e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04, 1.1611e-04, 1.1611e-04,
          1.1611e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05,
          6.3896e-05, 2.1040e-05, 1.5903e-04, 2.1040e-05, 1.1611e-04, 1.1611e-04,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 6.3896e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05,
          2.1040e-05, 1.1611e-04, 2.1040e-05, 1.1611e-04, 6.3896e-05, 2.1040e-05,
          2.1040e-05, 1.1611e-04, 3.7694e-04, 6.3896e-05, 1.1611e-04, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05,
          6.3896e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04,
          1.1212e-04, 4.1986e-04, 2.8181e-04, 2.1040e-05, 2.1040e-05, 6.3896e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05,
          2.1040e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04,
          2.1040e-05, 1.1611e-04, 6.3896e-05, 1.5903e-04, 1.1611e-04, 4.1986e-04,
          2.1040e-05, 1.1611e-04, 2.7800e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05, 6.3896e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04, 1.1611e-04, 2.1040e-05,
          1.1611e-04, 6.3896e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05, 1.1611e-04,
          6.3896e-05, 3.7694e-04, 2.1040e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 4.4394e-04, 6.3896e-05, 6.3896e-05,
          2.1040e-05, 1.1611e-04, 6.3896e-05, 1.1611e-04, 2.1040e-05, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 1.1611e-04, 2.1040e-05, 6.3896e-05, 7.0524e-04,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05, 6.3896e-05,
          2.1040e-05, 6.3896e-05, 2.1040e-05, 6.3896e-05, 6.3896e-05, 6.3896e-05,
          1.5903e-04, 1.1611e-04, 1.1611e-04, 2.1040e-05, 3.5334e-04, 6.3896e-05,
          2.1040e-05, 1.5903e-04, 1.1611e-04, 2.1040e-05, 2.1040e-05, 6.1798e-04,
          2.1040e-05, 2.1040e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 3.8314e-04,
          1.1611e-04, 2.1040e-05, 1.1611e-04, 1.5903e-04, 1.1611e-04, 2.1040e-05,
          1.1611e-04, 1.1611e-04, 1.1611e-04, 1.1611e-04, 2.1040e-05, 2.4533e-04,
          2.1040e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 1.5497e-04, 1.1611e-04,
          1.5903e-04, 6.3896e-05, 2.6226e-04, 1.1611e-04, 1.1611e-04, 6.3896e-05,
          1.1611e-04, 6.3896e-05, 3.7694e-04, 4.4847e-04, 2.1040e-05, 6.3896e-05,
          6.3896e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05, 2.1040e-05,
          6.3896e-05, 2.1040e-05, 2.7800e-04, 4.4394e-04, 1.1611e-04, 2.1040e-05,
          1.1611e-04, 1.1611e-04, 3.4046e-04, 1.5903e-04, 1.1611e-04, 1.1611e-04,
          1.1611e-04, 3.7694e-04, 2.1040e-05, 6.3896e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 5.4502e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 6.3896e-05,
          3.7694e-04, 6.3896e-05, 3.4046e-04, 6.3896e-05, 2.7823e-04, 2.5010e-04,
          1.1212e-04, 1.5497e-04, 6.9237e-04, 6.0129e-04, 1.1611e-04, 1.1611e-04,
          4.6802e-04, 3.8314e-04, 6.6805e-04, 2.1040e-05, 2.1040e-05, 1.2760e-03,
          2.1040e-05, 2.8801e-04, 5.4502e-04, 1.5497e-04, 2.1040e-05, 3.4046e-04,
          3.2115e-04, 1.1611e-04, 1.1611e-04, 5.0640e-04, 3.4046e-04, 8.3828e-04,
          2.1040e-05, 6.3896e-05, 7.6342e-04, 6.3896e-05, 1.1611e-04, 8.5783e-04,
          6.0129e-04, 5.4502e-04, 2.4533e-04, 2.1040e-05, 6.3896e-05, 2.1040e-05,
          2.1040e-05, 1.8728e-04, 6.3896e-05, 1.1611e-04, 1.6661e-03, 2.7823e-04,
          1.1212e-04, 6.3896e-05, 1.1611e-04, 6.3896e-05, 2.1040e-05, 6.3896e-05,
          3.2520e-04, 6.7282e-04, 2.1040e-05, 4.8685e-04, 2.1040e-05, 1.1611e-04,
          6.6805e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 3.4046e-04,
          1.1212e-04, 1.6670e-03, 1.5497e-04, 1.8728e-04, 3.4046e-04, 2.1458e-03,
          2.1040e-05, 5.9700e-04, 2.1267e-03, 2.8229e-04, 2.1040e-05, 1.0242e-03,
          2.1458e-03, 2.8229e-04, 1.1611e-04, 6.7234e-04, 6.7234e-04, 2.1040e-05,
          5.7745e-04, 3.6883e-04, 3.6883e-04, 1.1212e-04, 3.6883e-04, 6.3896e-05,
          2.1040e-05, 4.1175e-04, 3.6883e-04, 1.4448e-03, 5.3453e-04, 1.6108e-03,
          5.7745e-04, 3.6883e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.6077e-02, 1.1115e-01, 5.1300e-02, 8.8989e-02, 1.7590e-01,
          2.0764e-01, 4.3976e-02, 3.3508e-02, 2.5848e-02, 1.6956e-01, 3.2246e-05,
          3.2246e-05, 6.3896e-05, 6.3896e-05, 3.2246e-05, 1.3661e-04, 6.3896e-05,
          6.3896e-05, 6.3896e-05, 3.2246e-05, 3.2246e-05, 1.4794e-04, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05,
          6.3896e-05, 6.3896e-05, 1.3661e-04, 6.3896e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 1.3661e-04, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05, 3.2246e-05,
          3.2246e-05, 6.3896e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 6.3896e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          6.3896e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 8.1182e-05, 1.8072e-04, 1.1283e-04, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 1.1283e-04, 3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05,
          3.2246e-05, 3.2246e-05, 6.3896e-05, 3.2246e-05, 1.4794e-04, 3.2246e-05,
          8.1182e-05, 8.1182e-05, 1.1283e-04, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05,
          1.1283e-04, 3.2246e-05, 3.2246e-05, 1.4794e-04, 6.3896e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 1.9693e-04, 6.3896e-05, 3.2246e-05, 8.1182e-05,
          1.1283e-04, 3.2246e-05, 3.2246e-05, 1.8072e-04, 6.3896e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05, 1.1283e-04,
          1.1283e-04, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 1.4794e-04,
          1.3661e-04, 6.4898e-04, 2.7919e-04, 3.2246e-05, 3.2246e-05, 1.1283e-04,
          3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05,
          3.2246e-05, 6.3896e-05, 3.2246e-05, 6.3896e-05, 1.9693e-04, 3.2246e-05,
          3.2246e-05, 1.4794e-04, 3.2246e-05, 1.4794e-04, 3.2246e-05, 6.4898e-04,
          3.2246e-05, 3.2246e-05, 3.1137e-04, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05, 1.3661e-04, 3.2246e-05,
          8.1182e-05, 8.1182e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 6.3896e-05, 6.3896e-05, 1.8072e-04, 6.3896e-05, 3.2246e-05,
          3.2246e-05, 5.7650e-04, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05,
          6.3896e-05, 3.2246e-05, 3.2246e-05, 3.1137e-04, 3.2246e-05, 3.2246e-05,
          3.2246e-05, 3.0136e-04, 6.3896e-05, 3.2246e-05, 3.2246e-05, 3.2246e-05,
          8.1182e-05, 3.2246e-05, 3.2246e-05, 6.3896e-05, 3.2246e-05, 4.1223e-04,
          2.8157e-04, 3.2246e-05, 3.2246e-05, 6.3896e-05, 6.3896e-05, 3.2246e-05,
          3.2246e-05, 1.8561e-04, 3.2246e-05, 6.3896e-05, 6.3896e-05, 3.0231e-04,
          6.3896e-05, 1.9693e-04, 3.2246e-05, 8.1182e-05, 1.6475e-04, 6.3896e-05,
          3.2246e-05, 2.2852e-04, 1.4794e-04, 8.1182e-05, 3.2246e-05, 6.9332e-04,
          3.2246e-05, 3.2246e-05, 1.1283e-04, 6.3896e-05, 6.3896e-05, 2.2960e-04,
          8.1182e-05, 8.1182e-05, 8.1182e-05, 8.1182e-05, 3.0231e-04, 3.0136e-04,
          1.4794e-04, 3.2246e-05, 1.9693e-04, 4.0221e-04, 1.9693e-04, 3.2246e-05,
          1.9693e-04, 1.4794e-04, 1.4794e-04, 3.2246e-05, 6.3896e-05, 8.1182e-05,
          3.2246e-05, 1.1283e-04, 8.1182e-05, 8.1182e-05, 8.1182e-05, 8.1182e-05,
          1.1283e-04, 3.2246e-05, 6.3896e-05, 4.0340e-04, 2.3758e-04, 1.4794e-04,
          1.1283e-04, 6.3896e-05, 1.6475e-04, 1.9693e-04, 1.9693e-04, 1.3316e-04,
          2.9778e-04, 1.6475e-04, 5.4455e-04, 4.0221e-04, 8.1182e-05, 1.1283e-04,
          1.1283e-04, 8.1182e-05, 8.1182e-05, 8.1182e-05, 3.0231e-04, 1.1283e-04,
          2.2960e-04, 3.2246e-05, 3.1137e-04, 1.3661e-04, 1.9693e-04, 8.1182e-05,
          1.4794e-04, 3.2246e-05, 1.9693e-04, 6.3896e-05, 1.4794e-04, 1.9693e-04,
          1.9693e-04, 1.6475e-04, 4.2629e-04, 3.0231e-04, 8.1182e-05, 3.2246e-05,
          6.3896e-05, 1.5020e-03, 1.5390e-04, 3.0231e-04, 3.2246e-05, 3.2246e-05,
          9.8419e-04, 1.1283e-04, 1.9693e-04, 2.2960e-04, 3.2246e-05, 6.3896e-05,
          3.2806e-04, 3.2246e-05, 8.6784e-04, 8.2397e-04, 1.9693e-04, 1.9693e-04,
          9.6846e-04, 5.7650e-04, 1.8561e-04, 8.1182e-05, 8.1182e-05, 3.0231e-04,
          3.2246e-05, 1.4515e-03, 3.1328e-04, 6.3896e-05, 3.2246e-05, 6.1846e-04,
          1.8561e-04, 6.3896e-05, 1.2207e-03, 7.7486e-04, 3.2246e-05, 5.1928e-04,
          3.2246e-05, 3.2246e-05, 7.2145e-04, 6.3896e-05, 5.6553e-04, 6.1750e-04,
          5.7650e-04, 2.2960e-04, 8.1182e-05, 8.1182e-05, 4.6587e-04, 4.6587e-04,
          4.6587e-04, 1.7679e-04, 3.2246e-05, 3.2246e-05, 5.1928e-04, 4.2200e-04,
          1.0748e-03, 7.8392e-04, 4.6492e-04, 6.3896e-05, 2.5749e-04, 3.2246e-05,
          2.0850e-04, 6.3181e-04, 6.3896e-05, 2.8110e-04, 3.2246e-05, 1.4794e-04,
          1.1283e-04, 8.1182e-05, 8.1182e-05, 6.3896e-05, 6.1035e-04, 1.9693e-04,
          8.5783e-04, 1.1616e-03, 3.2246e-05, 3.2246e-05, 3.4523e-04, 1.4191e-03,
          7.1144e-04, 6.0129e-04, 1.2770e-03, 6.3896e-05, 3.2246e-05, 8.3590e-04,
          1.4811e-03, 1.1044e-03, 1.4794e-04, 5.8889e-04, 3.4142e-04, 3.2246e-05,
          4.5562e-04, 3.2246e-05, 3.2246e-05, 3.2246e-05, 1.0496e-04, 6.3896e-05,
          3.2246e-05, 1.3661e-04, 1.0496e-04, 7.9632e-04, 4.2391e-04, 5.9843e-04,
          1.0496e-04, 3.2246e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.3788e-02, 1.0364e-01, 4.5532e-02, 7.9102e-02, 1.5417e-01,
          1.7749e-01, 4.3945e-02, 2.3514e-02, 2.1149e-02, 8.1909e-02, 1.8384e-01,
          2.9027e-05, 6.9737e-05, 1.0467e-04, 2.9027e-05, 1.0467e-04, 1.0467e-04,
          6.3956e-05, 2.9027e-05, 2.9027e-05, 6.3956e-05, 1.2147e-04, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 1.0467e-04,
          6.9737e-05, 6.9737e-05, 1.1516e-04, 6.9737e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 1.0467e-04, 2.9027e-05,
          6.3956e-05, 1.2147e-04, 2.9027e-05, 2.9027e-05, 6.9737e-05, 2.9027e-05,
          2.9027e-05, 6.9737e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 1.1516e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 1.2147e-04, 1.2147e-04, 1.2147e-04,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 6.9737e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 6.9737e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05,
          2.9027e-05, 2.9027e-05, 1.6212e-04, 2.9027e-05, 1.2147e-04, 1.2147e-04,
          2.9027e-05, 2.9027e-05, 1.0282e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 1.0467e-04,
          2.9027e-05, 1.2147e-04, 2.9027e-05, 1.2147e-04, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 1.2147e-04, 1.2147e-04, 6.9737e-05, 1.2147e-04, 2.9027e-05,
          6.9737e-05, 2.9027e-05, 2.9027e-05, 1.9455e-04, 6.9737e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 1.0467e-04, 2.9027e-05,
          1.1516e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 1.2147e-04,
          1.5593e-04, 8.2493e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05, 1.5593e-04,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 2.9027e-05, 1.0467e-04,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 1.2147e-04, 1.2147e-04,
          2.9027e-05, 1.2147e-04, 2.9027e-05, 1.2147e-04, 1.8895e-04, 4.1151e-04,
          2.9027e-05, 2.9027e-05, 1.5593e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 6.9737e-05, 6.9737e-05, 1.5593e-04, 2.9027e-05,
          6.3956e-05, 2.9027e-05, 2.9027e-05, 1.2147e-04, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 6.9737e-05, 6.9737e-05, 1.0467e-04, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 1.2147e-04, 2.9027e-05, 2.9027e-05, 6.9737e-05, 1.9455e-04,
          1.0467e-04, 2.9027e-05, 2.9027e-05, 1.5593e-04, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 4.2701e-04, 1.1516e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 1.2147e-04, 6.9737e-05, 2.9027e-05, 1.5593e-04,
          1.9455e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 2.9027e-05,
          2.9027e-05, 1.5593e-04, 2.9027e-05, 6.9737e-05, 1.0467e-04, 2.8062e-04,
          1.9705e-04, 1.2147e-04, 2.9027e-05, 2.9027e-05, 3.1900e-04, 6.9737e-05,
          2.9027e-05, 2.0766e-04, 1.2147e-04, 2.9027e-05, 2.9027e-05, 4.8542e-04,
          2.9027e-05, 2.9027e-05, 2.9027e-05, 6.9737e-05, 2.9027e-05, 1.9455e-04,
          2.9027e-05, 2.9027e-05, 1.1516e-04, 1.8895e-04, 3.5453e-04, 4.6778e-04,
          1.2147e-04, 2.9027e-05, 2.4652e-04, 3.2210e-04, 1.2147e-04, 6.3956e-05,
          1.5640e-04, 1.2147e-04, 1.2147e-04, 1.2147e-04, 6.3956e-05, 2.9027e-05,
          2.9027e-05, 2.9027e-05, 1.0282e-04, 1.0282e-04, 2.9027e-05, 2.9027e-05,
          1.8895e-04, 2.9027e-05, 6.9737e-05, 2.8062e-04, 1.5593e-04, 2.0766e-04,
          2.7084e-04, 6.9737e-05, 1.5593e-04, 1.2147e-04, 2.8133e-04, 2.9027e-05,
          1.2147e-04, 1.0467e-04, 1.1129e-03, 3.1900e-04, 2.9027e-05, 6.9737e-05,
          6.9737e-05, 1.0282e-04, 1.0282e-04, 2.9027e-05, 3.5453e-04, 1.8895e-04,
          1.9455e-04, 6.3956e-05, 1.5593e-04, 1.0467e-04, 1.2147e-04, 2.9027e-05,
          1.2147e-04, 1.2147e-04, 1.2147e-04, 2.2972e-04, 1.2147e-04, 1.2147e-04,
          1.2147e-04, 6.3956e-05, 1.9455e-04, 5.2786e-04, 9.0218e-04, 2.9027e-05,
          1.8895e-04, 1.8263e-03, 1.8895e-04, 3.5453e-04, 2.9027e-05, 2.9027e-05,
          2.2972e-04, 2.2972e-04, 4.2701e-04, 2.2948e-04, 6.3956e-05, 6.3956e-05,
          1.1516e-04, 2.9027e-05, 8.5831e-04, 4.8542e-04, 2.8133e-04, 2.8133e-04,
          1.0214e-03, 4.8542e-04, 1.1516e-04, 2.9027e-05, 1.0282e-04, 5.7888e-04,
          2.9027e-05, 6.5279e-04, 7.6103e-04, 2.2948e-04, 2.9027e-05, 2.4652e-04,
          2.2972e-04, 6.9737e-05, 3.5453e-04, 4.2701e-04, 2.9027e-05, 4.6468e-04,
          2.9027e-05, 2.9027e-05, 4.8542e-04, 6.9737e-05, 1.1864e-03, 8.5831e-04,
          1.1568e-03, 1.9455e-04, 2.9027e-05, 2.9027e-05, 3.0327e-04, 2.6846e-04,
          1.5669e-03, 6.3956e-05, 2.9027e-05, 2.9027e-05, 5.9986e-04, 1.3781e-04,
          5.0020e-04, 5.0020e-04, 2.3019e-04, 6.9737e-05, 2.2972e-04, 2.9027e-05,
          6.9737e-05, 7.4005e-04, 6.9737e-05, 1.7846e-04, 2.9027e-05, 1.2147e-04,
          2.9027e-05, 1.5414e-04, 1.1516e-04, 2.9027e-05, 4.1389e-04, 2.4652e-04,
          1.9407e-03, 5.7888e-04, 2.9027e-05, 6.3956e-05, 5.7983e-04, 5.7983e-04,
          3.0327e-04, 4.2701e-04, 9.2888e-04, 6.9737e-05, 2.9027e-05, 5.0974e-04,
          5.7983e-04, 2.0199e-03, 1.2147e-04, 5.0974e-04, 5.0974e-04, 2.9027e-05,
          3.5572e-04, 2.9027e-05, 2.9027e-05, 2.9027e-05, 3.1495e-04, 2.2948e-04,
          2.9027e-05, 3.5572e-04, 6.3956e-05, 8.1491e-04, 3.1495e-04, 3.6597e-04,
          6.3956e-05, 2.9027e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.3972e-02, 1.0095e-01, 4.8462e-02, 7.4341e-02, 1.6760e-01,
          1.6089e-01, 4.0192e-02, 2.7359e-02, 2.3087e-02, 8.3252e-02, 1.2659e-01,
          6.1798e-02, 5.5134e-05, 1.0812e-04, 2.4736e-05, 1.0812e-04, 5.5134e-05,
          5.5134e-05, 5.5134e-05, 2.4736e-05, 2.4736e-05, 2.6584e-04, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05,
          5.5134e-05, 5.5134e-05, 1.0812e-04, 5.5134e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 1.0812e-04, 2.4736e-05,
          2.4736e-05, 1.4234e-04, 2.4736e-05, 5.5134e-05, 5.5134e-05, 2.4736e-05,
          2.4736e-05, 5.5134e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05,
          5.5134e-05, 2.4736e-05, 2.4736e-05, 1.4234e-04, 1.4234e-04, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 1.0949e-04, 2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05,
          2.4736e-05, 2.4736e-05, 5.5134e-05, 2.4736e-05, 1.4234e-04, 1.4234e-04,
          2.4736e-05, 2.4736e-05, 1.0949e-04, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05,
          1.0949e-04, 2.4736e-05, 2.4736e-05, 1.4234e-04, 5.5134e-05, 2.4736e-05,
          2.4736e-05, 1.4234e-04, 1.4234e-04, 5.5134e-05, 1.4234e-04, 2.4736e-05,
          5.5134e-05, 2.4736e-05, 2.4736e-05, 1.8024e-04, 5.5134e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05, 1.0812e-04, 5.5134e-05,
          1.0949e-04, 2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 1.4234e-04,
          1.0812e-04, 1.0223e-03, 2.4736e-05, 2.4736e-05, 2.4736e-05, 1.0812e-04,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05,
          2.4736e-05, 5.5134e-05, 2.4736e-05, 5.5134e-05, 2.4736e-05, 1.4234e-04,
          2.4736e-05, 1.4234e-04, 2.4736e-05, 1.4234e-04, 1.0949e-04, 3.7885e-04,
          2.4736e-05, 2.4736e-05, 1.9288e-04, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05, 1.0812e-04, 5.5134e-05,
          2.4736e-05, 2.4736e-05, 2.4736e-05, 1.4234e-04, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 5.5134e-05, 5.5134e-05, 1.0812e-04, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 1.7273e-04, 2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05,
          5.5134e-05, 2.4736e-05, 2.4736e-05, 1.9288e-04, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 3.6478e-04, 5.5134e-05, 2.4736e-05, 2.4736e-05, 2.4736e-05,
          2.4736e-05, 2.4736e-05, 1.4234e-04, 5.5134e-05, 2.4736e-05, 1.9288e-04,
          1.8024e-04, 2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05, 5.5134e-05,
          2.4736e-05, 1.0812e-04, 2.4736e-05, 5.5134e-05, 5.5134e-05, 3.7241e-04,
          5.5134e-05, 1.9670e-04, 2.4736e-05, 7.9095e-05, 2.6107e-04, 5.5134e-05,
          2.4736e-05, 1.7273e-04, 1.4234e-04, 2.4736e-05, 2.4736e-05, 3.8004e-04,
          2.4736e-05, 2.4736e-05, 5.5134e-05, 5.5134e-05, 1.8024e-04, 2.3460e-04,
          2.4736e-05, 2.4736e-05, 1.3208e-04, 7.9095e-05, 3.7241e-04, 3.6478e-04,
          1.4234e-04, 2.4736e-05, 1.9670e-04, 2.2709e-04, 1.4234e-04, 2.4736e-05,
          1.9670e-04, 1.4234e-04, 1.4234e-04, 1.4234e-04, 5.5134e-05, 7.9095e-05,
          2.4736e-05, 5.5134e-05, 7.9095e-05, 7.9095e-05, 7.9095e-05, 2.4736e-05,
          1.0949e-04, 2.4736e-05, 5.5134e-05, 3.7241e-04, 1.0812e-04, 1.9670e-04,
          1.6248e-04, 5.5134e-05, 2.6107e-04, 1.4234e-04, 2.4962e-04, 5.5134e-05,
          1.4234e-04, 5.5134e-05, 8.7881e-04, 2.6107e-04, 7.9095e-05, 5.5134e-05,
          5.5134e-05, 7.9095e-05, 7.9095e-05, 2.4736e-05, 3.7241e-04, 2.1684e-04,
          2.3460e-04, 2.4736e-05, 1.9288e-04, 1.0812e-04, 1.4234e-04, 2.4736e-05,
          1.4234e-04, 2.4736e-05, 1.4234e-04, 1.6248e-04, 1.4234e-04, 1.9670e-04,
          1.4234e-04, 1.0812e-04, 1.8024e-04, 6.1655e-04, 2.4736e-05, 2.4736e-05,
          1.0949e-04, 1.7605e-03, 2.1684e-04, 7.4148e-04, 2.4736e-05, 2.4736e-05,
          2.4724e-04, 1.0949e-04, 3.3450e-04, 2.3460e-04, 2.4736e-05, 5.5134e-05,
          2.4736e-05, 2.4736e-05, 1.0567e-03, 1.0767e-03, 1.9670e-04, 1.9670e-04,
          1.3933e-03, 5.1785e-04, 1.6248e-04, 7.9095e-05, 7.9095e-05, 6.8712e-04,
          2.4736e-05, 7.4148e-04, 3.7551e-04, 2.3329e-04, 2.4736e-05, 1.6391e-04,
          1.6248e-04, 5.5134e-05, 5.3644e-04, 3.3450e-04, 2.4736e-05, 4.0030e-04,
          2.4736e-05, 5.5134e-05, 6.8188e-04, 5.5134e-05, 9.4461e-04, 8.9359e-04,
          8.9359e-04, 2.3329e-04, 7.9095e-05, 2.4736e-05, 2.3460e-04, 2.3460e-04,
          2.1362e-03, 1.6248e-04, 2.4736e-05, 2.4736e-05, 1.9288e-04, 7.9095e-05,
          9.0551e-04, 5.3644e-04, 7.9095e-05, 5.5134e-05, 4.1103e-04, 2.4736e-05,
          5.5134e-05, 5.1785e-04, 5.5134e-05, 1.0812e-04, 2.4736e-05, 1.4234e-04,
          1.6248e-04, 7.9095e-05, 7.9095e-05, 5.5134e-05, 2.3460e-04, 2.8157e-04,
          2.0905e-03, 8.4019e-04, 2.4736e-05, 2.4736e-05, 2.8157e-04, 8.6737e-04,
          2.8753e-04, 3.3450e-04, 8.8692e-04, 5.5134e-05, 2.4736e-05, 7.4387e-04,
          8.6737e-04, 2.0828e-03, 1.4234e-04, 4.9829e-04, 4.9829e-04, 2.4736e-05,
          3.5357e-04, 2.4736e-05, 2.4736e-05, 2.4736e-05, 4.7660e-04, 1.8024e-04,
          2.4736e-05, 1.0812e-04, 7.7724e-05, 8.8358e-04, 3.2306e-04, 8.8358e-04,
          2.4736e-05, 7.7724e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.9958e-02, 9.8083e-02, 5.7312e-02, 5.5603e-02, 1.7029e-01,
          1.5137e-01, 3.0823e-02, 3.4302e-02, 3.1586e-02, 4.8401e-02, 5.4016e-02,
          2.5421e-02, 1.6235e-01, 8.3745e-05, 2.5868e-05, 8.3745e-05, 2.5868e-05,
          2.5868e-05, 8.3745e-05, 2.5868e-05, 2.5868e-05, 1.8191e-04, 2.5868e-05,
          5.8830e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 1.1677e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          1.8072e-04, 2.5868e-05, 2.5868e-05, 1.8191e-04, 1.8191e-04, 1.8191e-04,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05, 5.8830e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 6.0081e-05, 2.5868e-05, 6.0081e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 1.1677e-04, 1.8072e-04, 5.8830e-05,
          1.1677e-04, 2.5868e-05, 5.8830e-05, 5.8830e-05, 2.5868e-05, 1.8191e-04,
          2.5868e-05, 2.5868e-05, 8.3745e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          5.8830e-05, 1.8072e-04, 1.1677e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 1.8191e-04, 2.5868e-05, 1.8191e-04, 2.5487e-04, 1.4770e-04,
          2.5868e-05, 1.8191e-04, 1.8191e-04, 2.5868e-05, 1.8191e-04, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05, 5.8830e-05, 2.5868e-05,
          2.5868e-05, 2.7120e-05, 6.0081e-05, 2.5868e-05, 1.4770e-04, 2.5868e-05,
          2.5487e-04, 1.8072e-04, 2.5868e-05, 1.4770e-04, 8.3745e-05, 1.8191e-04,
          2.5868e-05, 2.3985e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05,
          2.5868e-05, 2.5868e-05, 5.8830e-05, 2.5868e-05, 6.0081e-05, 5.8830e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 8.3745e-05, 2.5868e-05, 1.8191e-04,
          8.3745e-05, 1.8191e-04, 1.1677e-04, 2.9755e-04, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 8.3745e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          1.1677e-04, 2.5868e-05, 5.8830e-05, 2.5868e-05, 2.5868e-05, 2.5868e-05,
          5.8830e-05, 1.2958e-04, 2.5868e-05, 1.8191e-04, 2.5868e-05, 2.5868e-05,
          2.5868e-05, 5.8830e-05, 5.8830e-05, 1.8072e-04, 2.5868e-05, 2.5868e-05,
          1.1677e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 1.2410e-04, 5.8830e-05,
          1.4770e-04, 2.5868e-05, 2.5868e-05, 8.3745e-05, 1.1677e-04, 2.5868e-05,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 2.2197e-04, 2.5487e-04, 2.5868e-05,
          3.5858e-04, 8.3745e-05, 1.8191e-04, 5.8830e-05, 1.1677e-04, 8.3745e-05,
          1.2279e-04, 2.5868e-05, 2.5868e-05, 1.1802e-04, 5.8830e-05, 1.2410e-04,
          2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05, 2.5868e-05, 5.8830e-05,
          5.8830e-05, 3.6001e-04, 3.6001e-04, 3.5858e-04, 5.8830e-05, 5.8830e-05,
          3.3760e-04, 8.1682e-04, 3.6812e-04, 2.7847e-04, 2.5868e-05, 5.8830e-05,
          2.5868e-05, 8.9824e-05, 3.2616e-04, 5.8830e-05, 2.5868e-05, 2.5082e-04,
          4.2868e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 5.8830e-05, 1.2279e-04,
          2.8563e-04, 2.5487e-04, 3.6001e-04, 3.8218e-04, 2.3675e-04, 2.5868e-05,
          8.1682e-04, 1.8191e-04, 1.8191e-04, 2.5868e-05, 2.5868e-05, 1.4770e-04,
          2.5868e-05, 4.4680e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 4.2987e-04,
          2.5868e-05, 2.5868e-05, 5.8830e-05, 5.8830e-05, 1.1677e-04, 2.5868e-05,
          7.7152e-04, 3.6144e-04, 5.8830e-05, 1.8191e-04, 2.5868e-05, 2.5868e-05,
          5.8830e-05, 5.8830e-05, 1.8191e-04, 1.1677e-04, 2.5868e-05, 5.5313e-04,
          2.5487e-04, 1.2958e-04, 1.2958e-04, 2.5868e-05, 5.8830e-05, 2.5868e-05,
          2.5082e-04, 2.5868e-05, 2.5868e-05, 2.5868e-05, 3.6001e-04, 2.3675e-04,
          2.8563e-04, 3.4952e-04, 2.8563e-04, 2.5868e-05, 2.5868e-05, 2.8563e-04,
          1.8191e-04, 2.5868e-05, 4.0936e-04, 2.5868e-05, 1.9348e-04, 5.8830e-05,
          2.5868e-05, 8.9824e-05, 2.5868e-05, 5.8830e-05, 2.5868e-05, 5.8830e-05,
          2.5868e-05, 5.8830e-05, 3.4952e-04, 2.5082e-04, 2.5868e-05, 5.8830e-05,
          2.8443e-04, 1.8072e-04, 2.5868e-05, 2.5868e-05, 2.8563e-04, 1.8191e-04,
          2.5868e-05, 1.2958e-04, 8.3745e-05, 1.2958e-04, 2.5868e-05, 1.8072e-04,
          2.5868e-05, 1.6248e-04, 1.8072e-04, 5.8830e-05, 2.5868e-05, 1.2958e-04,
          4.2868e-04, 5.8830e-05, 5.8830e-05, 4.0746e-04, 4.8018e-04, 1.8072e-04,
          4.0936e-04, 5.8830e-05, 2.5868e-05, 4.0936e-04, 9.1696e-04, 3.4952e-04,
          4.0746e-04, 1.8072e-04, 1.9348e-04, 1.2958e-04, 4.0936e-04, 1.8072e-04,
          1.6248e-04, 8.9824e-05, 5.8830e-05, 5.8830e-05, 7.3910e-04, 5.8830e-05,
          1.1454e-03, 5.8746e-04, 5.1308e-04, 5.8830e-05, 3.7646e-04, 2.5868e-05,
          5.8830e-05, 4.4036e-04, 5.8830e-05, 1.2827e-03, 8.9824e-05, 8.9824e-05,
          1.4770e-04, 3.8528e-04, 1.9348e-04, 2.5868e-05, 2.8443e-04, 1.9348e-04,
          3.2806e-03, 7.3910e-04, 1.8072e-04, 2.5868e-05, 1.2958e-04, 1.0386e-03,
          5.8746e-04, 4.0746e-04, 1.0624e-03, 3.1853e-04, 5.8830e-05, 1.8191e-04,
          9.6560e-04, 4.8018e-04, 8.9824e-05, 3.4952e-04, 3.4952e-04, 2.5868e-05,
          4.6015e-04, 4.2725e-04, 3.6311e-04, 3.6311e-04, 4.2725e-04, 5.8830e-05,
          2.5868e-05, 4.6015e-04, 3.6311e-04, 7.5197e-04, 3.0518e-04, 1.1530e-03,
          4.2725e-04, 4.2725e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.3300e-02, 1.0059e-01, 6.6833e-02, 6.4270e-02, 2.0044e-01,
          1.0632e-01, 3.5706e-02, 1.5305e-02, 1.0872e-02, 1.7807e-02, 2.0737e-02,
          1.0437e-02, 5.5908e-02, 2.0483e-01, 4.3154e-05, 6.1452e-05, 1.9789e-05,
          1.9789e-05, 9.1255e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          6.1452e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 4.3154e-05, 1.9789e-05, 4.3154e-05, 4.3154e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 3.8087e-05, 4.3154e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 3.8087e-05, 4.3154e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 9.2447e-05, 4.3154e-05, 4.3154e-05,
          1.9789e-05, 4.3154e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 4.3154e-05,
          6.1452e-05, 3.8087e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 4.3154e-05, 4.3154e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 2.0087e-05, 4.3154e-05, 8.7798e-05, 1.9789e-05,
          1.9789e-05, 9.2447e-05, 1.9789e-05, 1.7643e-04, 6.1452e-05, 4.3154e-05,
          1.1843e-04, 1.9789e-05, 1.9789e-05, 1.1843e-04, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 4.3154e-05, 1.9789e-05, 1.9789e-05, 6.9082e-05,
          6.1452e-05, 1.6346e-03, 1.6820e-04, 1.9789e-05, 6.1452e-05, 1.9789e-05,
          9.2447e-05, 1.9789e-05, 3.8087e-05, 1.9789e-05, 6.1452e-05, 3.8087e-05,
          1.5616e-04, 1.9789e-05, 1.9789e-05, 4.3154e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.1075e-04, 2.0087e-05, 4.3154e-05, 4.3154e-05, 2.0087e-05,
          1.9789e-05, 8.7798e-05, 8.7798e-05, 6.1452e-05, 6.1452e-05, 9.2447e-05,
          1.6880e-04, 2.0242e-04, 6.1452e-05, 9.6262e-05, 3.8087e-05, 1.9789e-05,
          3.8087e-05, 1.9789e-05, 1.9789e-05, 4.3154e-05, 1.9789e-05, 1.9789e-05,
          6.9499e-05, 9.1255e-05, 4.3154e-05, 6.9082e-05, 4.3571e-05, 1.7166e-03,
          1.1116e-04, 1.1075e-04, 8.7380e-05, 6.9082e-05, 6.9082e-05, 6.3801e-04,
          1.2493e-04, 1.3280e-04, 1.7643e-04, 1.9789e-05, 2.8563e-04, 1.9789e-05,
          4.3154e-05, 3.8087e-05, 9.5069e-05, 1.9789e-05, 1.9789e-05, 4.9591e-05,
          2.7943e-04, 1.4544e-04, 6.1452e-05, 6.1452e-05, 1.9789e-05, 6.1452e-05,
          1.3912e-04, 8.7798e-05, 1.9789e-05, 1.9789e-05, 2.5845e-04, 1.9789e-05,
          1.1957e-04, 8.9765e-05, 2.9445e-04, 9.1255e-05, 3.8087e-05, 3.8087e-05,
          1.7643e-04, 1.9789e-05, 1.9789e-05, 6.9380e-05, 2.5892e-04, 1.6403e-04,
          4.3154e-05, 6.9082e-05, 4.3154e-05, 9.1255e-05, 2.7919e-04, 1.1075e-04,
          8.7380e-05, 1.9789e-05, 1.3912e-04, 2.0242e-04, 2.0242e-04, 2.3258e-04,
          1.4019e-04, 3.1567e-04, 6.3801e-04, 1.1957e-04, 1.4830e-04, 3.8087e-05,
          7.2956e-05, 1.9789e-05, 6.9082e-05, 1.6546e-03, 8.9765e-05, 9.2864e-05,
          2.4796e-04, 1.9789e-05, 4.3154e-05, 1.1957e-04, 1.6403e-04, 4.3154e-05,
          6.1452e-05, 6.3801e-04, 7.1430e-04, 3.3903e-04, 2.9445e-04, 1.1957e-04,
          2.0313e-04, 3.1400e-04, 2.2876e-04, 3.9864e-04, 6.7890e-05, 4.3154e-05,
          3.8087e-05, 2.0242e-04, 2.8706e-04, 1.9503e-04, 2.8682e-04, 2.4021e-04,
          2.1708e-04, 6.9082e-05, 6.9082e-05, 3.8087e-05, 4.3154e-05, 1.9789e-05,
          6.5613e-04, 3.1567e-04, 1.8227e-04, 4.5681e-04, 2.5225e-04, 2.8205e-04,
          3.4380e-04, 6.5613e-04, 6.9201e-05, 1.8215e-04, 6.1452e-05, 1.9789e-05,
          1.1569e-04, 3.1662e-04, 6.9082e-05, 1.3912e-04, 6.9499e-05, 2.4056e-04,
          1.3912e-04, 1.4567e-04, 6.1452e-05, 6.1452e-05, 4.3154e-05, 2.8181e-04,
          4.0102e-04, 1.4019e-04, 6.0558e-04, 7.1430e-04, 3.6120e-04, 4.5681e-04,
          4.2701e-04, 2.2209e-04, 1.9789e-05, 1.9789e-05, 1.4811e-03, 1.7452e-03,
          1.3947e-04, 2.4533e-04, 9.7811e-05, 6.9082e-05, 1.3912e-04, 1.9872e-04,
          2.9349e-04, 2.7132e-04, 1.6403e-04, 2.7895e-04, 3.3021e-04, 2.2388e-04,
          7.1430e-04, 2.5868e-04, 6.6614e-04, 3.8457e-04, 7.7105e-04, 1.1688e-04,
          6.9201e-05, 2.2209e-04, 1.7643e-04, 6.1452e-05, 1.1611e-04, 2.7156e-04,
          2.7132e-04, 1.1569e-04, 2.4796e-04, 1.6403e-04, 1.7560e-04, 1.9383e-04,
          1.4067e-04, 4.1914e-04, 6.9082e-05, 4.1914e-04, 2.5558e-04, 5.0306e-04,
          3.3903e-04, 2.7919e-04, 1.9789e-05, 3.8087e-05, 1.1569e-04, 1.1569e-04,
          1.9872e-04, 1.6880e-04, 3.8087e-05, 1.1569e-04, 2.1839e-04, 1.9789e-05,
          2.4796e-04, 2.7156e-04, 1.9789e-05, 2.4164e-04, 1.7560e-04, 6.9082e-05,
          4.1914e-04, 2.2209e-04, 2.4319e-04, 6.9082e-05, 4.1819e-04, 1.9789e-05,
          3.3522e-04, 3.9697e-04, 1.4544e-04, 2.7132e-04, 2.7132e-04, 1.1569e-04,
          1.9789e-05, 6.1452e-05, 1.1581e-04, 1.1581e-04, 4.7588e-04, 3.1161e-04,
          2.1338e-04, 4.3154e-05, 2.7132e-04, 2.7132e-04, 1.9789e-05, 3.2830e-04,
          6.7997e-04, 4.1914e-04, 2.7132e-04, 1.9789e-05, 1.7643e-04, 2.2209e-04,
          1.7643e-04, 1.9789e-05, 7.2956e-05, 3.2830e-04, 1.9789e-05, 7.2956e-05,
          3.8087e-05, 2.6345e-04, 1.4544e-04, 4.3154e-05, 3.9077e-04, 6.9082e-05,
          4.7588e-04, 4.3154e-05, 3.2830e-04, 1.9872e-04, 6.9082e-05, 7.2956e-05,
          3.2830e-04, 9.7394e-05, 1.9789e-05, 7.2956e-05, 8.9693e-04, 1.2722e-03,
          6.9082e-05, 3.2830e-04, 7.2956e-05, 1.4544e-04, 7.7903e-05, 1.9789e-05,
          1.7643e-04, 3.1548e-03, 1.5306e-04, 1.5306e-04, 3.1233e-04, 1.4675e-04,
          3.8087e-05, 1.7643e-04, 1.5306e-04, 6.0606e-04, 9.5069e-05, 9.5069e-05,
          1.2338e-04, 1.5306e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.5106e-02, 9.1431e-02, 3.9825e-02, 5.3467e-02, 1.5234e-01,
          1.4246e-01, 3.2257e-02, 3.3173e-02, 2.7344e-02, 5.2734e-02, 5.8655e-02,
          2.8152e-02, 5.0568e-02, 1.0193e-02, 1.6125e-01, 2.1994e-05, 7.3731e-05,
          7.3731e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 3.1424e-04, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 7.3731e-05,
          1.4389e-04, 2.1994e-05, 1.5831e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 7.3731e-05, 2.1994e-05,
          7.3731e-05, 1.9979e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 7.3731e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 1.9979e-04, 1.9979e-04, 1.9979e-04,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.4736e-05, 2.1994e-05, 2.4736e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.9979e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 1.9979e-04, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.9979e-04, 2.1994e-05, 1.9979e-04, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 1.4389e-04, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          1.4389e-04, 2.1994e-05, 2.1994e-05, 7.3731e-05, 2.1994e-05, 1.9979e-04,
          7.3731e-05, 2.5153e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.9562e-04,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.4736e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 1.9979e-04, 2.1994e-05, 1.9979e-04, 2.1994e-05, 7.3731e-05,
          2.1994e-05, 2.1994e-05, 4.5514e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          1.9562e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.5831e-04, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 7.3731e-05, 2.1994e-05, 2.1994e-05,
          7.3731e-05, 1.9979e-04, 2.1994e-05, 2.1994e-05, 1.4663e-04, 2.1994e-05,
          7.3731e-05, 2.1994e-05, 2.1994e-05, 5.7697e-04, 7.3731e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.9979e-04, 2.1994e-05, 7.3731e-05, 4.5514e-04,
          1.4389e-04, 2.1994e-05, 2.1994e-05, 1.4663e-04, 2.1994e-05, 2.4736e-05,
          2.1994e-05, 1.5831e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.9562e-04,
          2.1994e-05, 1.9979e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.2221e-04, 1.9979e-04, 1.9979e-04, 2.1994e-05, 2.1994e-05, 1.9979e-04,
          2.1994e-05, 2.1994e-05, 1.4389e-04, 2.1994e-05, 2.1994e-05, 1.4389e-04,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.8014e-04,
          1.9979e-04, 7.3731e-05, 2.0230e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          1.9979e-04, 1.9979e-04, 1.9979e-04, 2.1994e-05, 2.1994e-05, 3.3879e-04,
          2.1994e-05, 1.4389e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.9562e-04, 2.1994e-05,
          1.9562e-04, 1.4389e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.0586e-03, 1.9562e-04, 2.1994e-05, 1.9836e-04,
          1.4389e-04, 2.1994e-05, 2.0230e-04, 2.1994e-05, 7.3731e-05, 2.1994e-05,
          1.4389e-04, 2.1994e-05, 4.5514e-04, 6.4611e-04, 2.1994e-05, 2.1994e-05,
          1.9979e-04, 2.1994e-05, 3.8004e-04, 2.1994e-05, 2.1994e-05, 1.9979e-04,
          2.1994e-05, 2.1994e-05, 1.4389e-04, 1.5831e-04, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 3.3879e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          1.5831e-04, 2.1994e-05, 3.3879e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 7.3731e-05, 1.1454e-03, 1.5831e-04, 1.9979e-04, 2.1994e-05,
          1.1568e-03, 5.1641e-04, 3.3879e-04, 2.0230e-04, 2.1994e-05, 2.8014e-04,
          2.1994e-05, 3.3879e-04, 4.6062e-04, 2.1994e-05, 2.1994e-05, 2.8706e-04,
          2.1994e-05, 2.1994e-05, 1.0657e-04, 3.3879e-04, 4.6778e-04, 6.6090e-04,
          2.1994e-05, 2.1994e-05, 4.6778e-04, 1.5116e-04, 2.1994e-05, 1.1950e-03,
          1.1950e-03, 6.2799e-04, 2.0230e-04, 2.0230e-04, 1.9562e-04, 1.4389e-04,
          2.1994e-05, 2.8753e-04, 2.1994e-05, 2.1994e-05, 6.6090e-04, 2.1994e-05,
          7.9012e-04, 3.9577e-04, 2.0230e-04, 2.1994e-05, 1.5116e-04, 2.1994e-05,
          2.1994e-05, 9.5320e-04, 2.1994e-05, 7.3731e-05, 2.1994e-05, 2.1994e-05,
          4.6778e-04, 2.0230e-04, 2.0230e-04, 2.1994e-05, 3.2425e-04, 2.8706e-04,
          4.6778e-04, 4.6062e-04, 1.9562e-04, 2.8753e-04, 2.8706e-04, 1.6499e-03,
          5.0497e-04, 6.4564e-04, 1.2016e-03, 2.1994e-05, 2.1994e-05, 3.2902e-04,
          6.4564e-04, 3.8338e-04, 2.3568e-04, 6.4564e-04, 5.9366e-04, 2.1994e-05,
          2.4116e-04, 7.3731e-05, 2.4116e-04, 7.3731e-05, 3.2592e-04, 3.6287e-04,
          7.3731e-05, 2.4116e-04, 7.3731e-05, 1.0786e-03, 1.8954e-04, 1.1625e-03,
          2.4116e-04, 2.4116e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.4778e-02, 8.8623e-02, 4.9835e-02, 5.2429e-02, 1.5100e-01,
          1.2311e-01, 3.2776e-02, 1.0510e-01, 4.8248e-02, 2.9007e-02, 3.1525e-02,
          1.4114e-02, 3.0106e-02, 8.3771e-03, 2.9633e-02, 1.2622e-01, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          1.0014e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 7.1287e-05,
          5.2929e-05, 2.4080e-05, 7.1287e-05, 5.2929e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 7.1287e-05, 5.2929e-05,
          2.4080e-05, 1.7643e-04, 7.1287e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 5.2929e-05,
          5.2929e-05, 5.2929e-05, 2.4080e-05, 1.7643e-04, 1.7643e-04, 1.7643e-04,
          1.7643e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 5.2929e-05, 5.2929e-05, 5.2929e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 5.2929e-05, 1.7130e-04, 5.2929e-05,
          5.2929e-05, 2.4080e-05, 2.0528e-04, 5.2929e-05, 1.7643e-04, 1.7643e-04,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          5.2929e-05, 1.5736e-04, 5.2929e-05, 2.4080e-05, 5.2929e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 1.7643e-04, 1.7130e-04, 5.2929e-05,
          5.2929e-05, 1.7643e-04, 1.7643e-04, 5.2929e-05, 1.7643e-04, 2.4080e-05,
          5.2929e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 5.2929e-05, 5.2929e-05,
          2.4080e-05, 2.4080e-05, 5.2929e-05, 5.2929e-05, 2.4080e-05, 2.4080e-05,
          1.5736e-04, 5.2929e-05, 2.4080e-05, 2.4080e-05, 5.2929e-05, 1.7643e-04,
          7.1287e-05, 5.2929e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.0014e-04,
          5.2929e-05, 2.4080e-05, 5.2929e-05, 2.4080e-05, 5.2929e-05, 1.0014e-04,
          2.4080e-05, 5.2929e-05, 1.2851e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          1.0014e-04, 1.7643e-04, 5.2929e-05, 3.2353e-04, 5.2929e-05, 5.2929e-05,
          2.4080e-05, 2.4080e-05, 1.2851e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          3.0780e-04, 2.0826e-04, 1.1015e-04, 5.2929e-05, 1.0014e-04, 5.2929e-05,
          1.2851e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.0377e-04, 2.4080e-05,
          8.1301e-05, 5.2929e-05, 1.5736e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          1.0014e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.0210e-04, 5.2929e-05,
          2.4080e-05, 4.8518e-04, 5.2929e-05, 1.2851e-04, 3.6502e-04, 1.0014e-04,
          2.4080e-05, 2.4080e-05, 1.1015e-04, 2.3711e-04, 1.5736e-04, 2.4080e-05,
          1.7130e-04, 2.0826e-04, 1.7643e-04, 2.4080e-05, 1.0014e-04, 1.2851e-04,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 1.0210e-04, 5.2929e-05, 5.9557e-04,
          1.2851e-04, 1.0014e-04, 2.4080e-05, 1.0014e-04, 1.0014e-04, 1.0014e-04,
          5.2929e-05, 2.5606e-04, 1.6093e-04, 2.3711e-04, 7.7009e-04, 5.2929e-05,
          2.7585e-04, 4.6420e-04, 2.5606e-04, 5.2929e-05, 2.4080e-05, 5.2929e-05,
          2.4080e-05, 3.3617e-04, 3.1781e-04, 2.4080e-05, 2.4080e-05, 6.1131e-04,
          2.4080e-05, 2.4080e-05, 5.6505e-04, 7.1287e-05, 5.2929e-05, 1.7989e-04,
          2.5606e-04, 2.3711e-04, 1.0377e-04, 1.3268e-04, 2.4080e-05, 1.2851e-04,
          5.0783e-04, 1.7643e-04, 1.7643e-04, 2.4080e-05, 2.4080e-05, 4.1580e-04,
          2.4080e-05, 3.6502e-04, 2.4080e-05, 8.1301e-05, 2.4080e-05, 1.8978e-04,
          7.1287e-05, 2.4080e-05, 2.4080e-05, 5.2929e-05, 3.0780e-04, 1.6093e-04,
          2.3711e-04, 5.5432e-05, 4.1914e-04, 3.1328e-04, 2.0826e-04, 1.3268e-04,
          2.4080e-05, 2.6059e-04, 2.5606e-04, 3.0780e-04, 5.6505e-04, 7.1764e-04,
          1.0014e-04, 8.2970e-04, 6.2227e-04, 2.4080e-05, 1.5736e-04, 1.2851e-04,
          1.4095e-03, 7.1287e-05, 7.1287e-05, 4.1318e-04, 1.0377e-04, 2.4080e-05,
          3.1328e-04, 3.1328e-04, 7.7438e-04, 1.5736e-04, 3.6049e-04, 3.1328e-04,
          2.5606e-04, 2.4080e-05, 2.4080e-05, 5.2929e-05, 6.2227e-04, 4.4465e-04,
          7.1287e-05, 8.7690e-04, 2.0826e-04, 2.4080e-05, 2.4080e-05, 1.0014e-04,
          2.7895e-04, 1.2851e-04, 6.6948e-04, 1.0014e-04, 1.2851e-04, 5.1165e-04,
          4.4465e-04, 3.6502e-04, 1.5104e-04, 1.5104e-04, 2.5606e-04, 3.1328e-04,
          5.6410e-04, 2.3711e-04, 2.0826e-04, 5.6505e-04, 1.0377e-04, 3.6502e-04,
          2.0826e-04, 2.3711e-04, 4.4680e-04, 1.5736e-04, 8.1301e-05, 1.6093e-04,
          2.0826e-04, 2.4080e-05, 2.4080e-05, 2.0826e-04, 6.9046e-04, 4.4465e-04,
          5.2929e-05, 3.1781e-04, 2.0826e-04, 1.3268e-04, 1.5104e-04, 3.6049e-04,
          5.6839e-04, 6.4993e-04, 2.0826e-04, 1.0377e-04, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 5.4073e-04, 3.9744e-04, 4.4465e-04, 2.0826e-04,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 3.4022e-04, 2.4080e-05, 1.3268e-04,
          4.3201e-04, 5.6314e-04, 2.4080e-05, 1.5173e-03, 2.4080e-05, 2.4080e-05,
          4.1580e-04, 3.6860e-04, 2.0826e-04, 2.4080e-05, 2.4080e-05, 4.1580e-04,
          2.4080e-05, 4.4465e-04, 4.4465e-04, 2.0826e-04, 2.0826e-04, 7.1955e-04,
          2.4080e-05, 3.6049e-04, 7.0667e-04, 2.4080e-05, 7.1955e-04, 8.4305e-04,
          4.1580e-04, 2.4080e-05, 2.8896e-04, 3.6049e-04, 6.3515e-04, 2.4080e-05,
          3.0518e-04, 2.7633e-04, 3.0518e-04, 2.7633e-04, 3.3355e-04, 3.0518e-04,
          2.4080e-05, 5.1308e-04, 2.7633e-04, 9.0218e-04, 8.5497e-04, 1.1101e-03,
          2.7633e-04, 2.7633e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.1194e-02, 1.0034e-01, 5.9814e-02, 5.8807e-02, 1.6309e-01,
          1.1670e-01, 3.1525e-02, 1.4626e-02, 1.2108e-02, 2.4933e-02, 2.3895e-02,
          1.1948e-02, 5.2124e-02, 4.8615e-02, 1.8082e-02, 1.3733e-02, 1.5356e-01,
          2.2411e-05, 9.9838e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05,
          6.3896e-05, 3.9339e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05,
          3.9339e-05, 6.3896e-05, 2.2411e-05, 4.6909e-05, 2.2411e-05, 3.9339e-05,
          2.2411e-05, 2.2411e-05, 3.9339e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 4.6909e-05, 2.2411e-05,
          2.2411e-05, 3.9339e-05, 2.2411e-05, 2.2411e-05, 6.6400e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 2.2411e-05, 3.9339e-05, 2.2411e-05, 6.3896e-05,
          2.4438e-04, 1.3185e-04, 3.9339e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 2.2411e-05, 6.3896e-05, 9.0957e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 6.4492e-05, 4.6909e-05, 6.4492e-05, 2.2411e-05,
          2.2411e-05, 3.9339e-05, 2.2411e-05, 1.9193e-04, 8.8930e-05, 2.3377e-04,
          1.3137e-04, 2.2411e-05, 4.6909e-05, 6.3896e-05, 2.2411e-05, 2.2411e-05,
          2.2411e-05, 2.2411e-05, 6.4433e-05, 2.2411e-05, 2.2411e-05, 3.9339e-05,
          6.3896e-05, 1.9193e-04, 1.5211e-04, 2.2411e-05, 4.6909e-05, 2.2411e-05,
          8.3387e-05, 3.9339e-05, 3.9339e-05, 2.2411e-05, 2.9850e-04, 8.8930e-05,
          4.6909e-05, 2.2411e-05, 2.2411e-05, 4.6909e-05, 2.2411e-05, 2.2411e-05,
          2.2411e-05, 3.9339e-05, 2.2411e-05, 6.3896e-05, 4.6909e-05, 3.9995e-05,
          2.2411e-05, 1.3363e-04, 1.3363e-04, 4.6909e-05, 1.9193e-04, 3.9339e-05,
          2.9850e-04, 2.5511e-04, 8.8930e-05, 9.9838e-05, 8.8930e-05, 2.2411e-05,
          2.2411e-05, 3.9339e-05, 3.9339e-05, 3.9339e-05, 3.9339e-05, 4.6909e-05,
          1.0908e-04, 9.9838e-05, 6.3896e-05, 1.3494e-03, 8.9586e-05, 6.3896e-05,
          3.9339e-05, 6.3896e-05, 6.6400e-05, 3.9339e-05, 6.6400e-05, 3.9339e-05,
          1.2434e-04, 8.3387e-05, 1.9193e-04, 4.6909e-05, 6.3896e-05, 2.2411e-05,
          3.9339e-05, 3.9339e-05, 2.2411e-05, 2.2411e-05, 2.2411e-05, 9.9838e-05,
          8.8930e-05, 1.4534e-03, 8.8930e-05, 8.8930e-05, 2.2411e-05, 6.3896e-05,
          1.0788e-04, 1.5354e-04, 2.2411e-05, 2.2411e-05, 6.4433e-05, 2.2411e-05,
          9.9838e-05, 8.8930e-05, 1.2434e-04, 1.9193e-04, 2.2411e-05, 2.2411e-05,
          1.2434e-04, 2.2411e-05, 2.2411e-05, 6.4433e-05, 1.8382e-04, 1.2434e-04,
          1.3185e-04, 1.4181e-03, 3.9339e-05, 2.2411e-05, 1.5640e-04, 6.3896e-05,
          6.6400e-05, 6.6400e-05, 6.3896e-05, 2.3103e-04, 1.8692e-04, 1.2434e-04,
          1.9550e-04, 6.4433e-05, 3.7575e-04, 1.2434e-04, 6.3896e-05, 2.2411e-05,
          1.9193e-04, 2.2411e-05, 6.4433e-05, 1.4620e-03, 8.8930e-05, 1.5373e-03,
          9.9838e-05, 2.2411e-05, 2.2411e-05, 1.9193e-04, 6.3896e-05, 6.3896e-05,
          4.6909e-05, 4.9925e-04, 5.2452e-04, 2.9850e-04, 3.5238e-04, 1.2434e-04,
          3.2425e-04, 6.2513e-04, 4.1962e-04, 5.7697e-04, 3.9339e-05, 4.6909e-05,
          3.9339e-05, 1.4095e-03, 5.4359e-04, 8.8930e-05, 1.8251e-04, 2.0313e-04,
          2.4772e-04, 1.2898e-04, 3.9339e-05, 3.9339e-05, 6.3896e-05, 8.8930e-05,
          4.3654e-04, 2.3103e-04, 1.2898e-04, 4.3392e-04, 1.9550e-04, 8.8930e-05,
          5.6219e-04, 4.1962e-04, 6.6400e-05, 2.2411e-05, 2.2411e-05, 3.9339e-05,
          6.4433e-05, 1.9550e-03, 1.4181e-03, 1.0842e-04, 6.6400e-05, 2.6464e-04,
          1.0788e-04, 4.6909e-05, 1.2434e-04, 6.3896e-05, 6.3896e-05, 1.6844e-04,
          8.1682e-04, 1.9670e-04, 3.5691e-04, 1.4849e-03, 1.4389e-04, 8.8930e-05,
          2.6083e-04, 1.5748e-04, 3.7575e-04, 6.3896e-05, 2.2411e-05, 3.9434e-04,
          2.9039e-04, 2.0206e-04, 1.5268e-03, 6.6400e-05, 6.3896e-05, 6.4433e-05,
          3.9458e-04, 3.9339e-05, 2.2411e-05, 2.2411e-05, 1.2898e-04, 4.3440e-04,
          4.6182e-04, 4.6182e-04, 4.6182e-04, 1.2434e-04, 4.6182e-04, 4.8232e-04,
          6.6400e-05, 9.9838e-05, 1.9193e-04, 2.2411e-05, 1.0908e-04, 8.8930e-05,
          6.3896e-05, 1.0842e-04, 1.0842e-04, 1.3292e-04, 1.2434e-04, 9.9361e-05,
          2.2411e-05, 2.2662e-04, 1.0842e-04, 2.6202e-04, 3.9339e-05, 1.6844e-04,
          1.3218e-03, 1.9193e-04, 3.9339e-05, 3.9339e-05, 4.1962e-04, 4.1962e-04,
          3.9339e-05, 4.6909e-05, 2.2411e-05, 1.0842e-04, 6.4433e-05, 6.3896e-05,
          6.4433e-05, 2.0146e-04, 5.8603e-04, 8.8930e-05, 9.9838e-05, 1.4389e-04,
          1.0309e-03, 8.8930e-05, 1.6844e-04, 1.0842e-04, 1.4389e-04, 1.3292e-04,
          4.3607e-04, 1.2434e-04, 2.2411e-05, 2.3592e-04, 4.2200e-04, 2.2411e-05,
          7.0572e-04, 1.5748e-04, 1.0842e-04, 1.7703e-04, 2.3592e-04, 1.6844e-04,
          8.1348e-04, 3.4404e-04, 2.6202e-04, 4.1246e-04, 8.2922e-04, 3.2949e-04,
          1.8978e-03, 9.9468e-04, 4.8923e-04, 3.4356e-04, 4.1151e-04, 6.3896e-05,
          2.8563e-04, 1.6844e-04, 1.2434e-04, 7.5293e-04, 4.7445e-05, 1.2434e-04,
          6.4433e-05, 2.0206e-04, 1.4389e-04, 2.2411e-05, 9.3222e-04, 1.0842e-04,
          9.9468e-04, 3.3307e-04, 4.3607e-04, 2.2411e-05, 1.0842e-04, 3.6860e-04,
          5.4264e-04, 1.0842e-04, 1.0347e-03, 4.1151e-04, 4.1246e-04, 5.4502e-04,
          3.8815e-04, 9.0790e-04, 3.2711e-04, 6.6400e-05, 4.9734e-04, 2.2411e-05,
          1.5640e-04, 1.6737e-04, 1.0687e-04, 3.9339e-05, 3.9339e-05, 4.6909e-05,
          3.9339e-05, 1.5640e-04, 1.3185e-04, 7.2813e-04, 7.2813e-04, 3.9339e-05,
          3.9339e-05, 1.0687e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.8173e-02, 9.1614e-02, 5.7648e-02, 6.0669e-02, 1.5149e-01,
          1.0883e-01, 3.1433e-02, 1.4297e-02, 1.0323e-02, 2.4185e-02, 2.3575e-02,
          1.1444e-02, 4.5227e-02, 4.5532e-02, 1.5671e-02, 1.4252e-02, 1.1090e-01,
          8.5876e-02, 1.5116e-04, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          6.4909e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 4.1723e-05, 2.1815e-05, 4.1723e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 4.1723e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 4.1723e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 6.4909e-05,
          1.7250e-04, 1.3602e-04, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 4.1723e-05, 4.1723e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 4.2319e-05, 4.1723e-05, 6.5446e-05, 2.1815e-05,
          2.1815e-05, 4.1723e-05, 2.1815e-05, 1.9097e-04, 6.4909e-05, 1.4937e-04,
          1.3602e-04, 2.1815e-05, 4.1723e-05, 6.4909e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 4.4942e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          6.4909e-05, 2.4223e-04, 6.5446e-05, 2.1815e-05, 4.1723e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 4.4942e-05, 4.1723e-05, 3.8052e-04, 6.4909e-05,
          4.1723e-05, 2.1815e-05, 2.1815e-05, 4.1723e-05, 2.1815e-05, 2.1815e-05,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 4.1723e-05, 7.6890e-05, 4.2319e-05,
          2.2888e-03, 1.0538e-04, 1.0538e-04, 4.1723e-05, 1.1629e-04, 2.1815e-05,
          3.4308e-04, 4.7731e-04, 6.4909e-05, 6.4909e-05, 6.4909e-05, 3.6168e-04,
          2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          1.4055e-04, 6.4909e-05, 6.4909e-05, 1.4277e-03, 1.0061e-04, 1.0008e-04,
          4.1723e-05, 1.0008e-04, 6.1750e-05, 4.1723e-05, 2.1815e-05, 6.1750e-05,
          6.4909e-05, 4.4942e-05, 1.9097e-04, 4.1723e-05, 6.4909e-05, 4.1723e-05,
          4.1723e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05, 4.4942e-05,
          6.4909e-05, 1.8463e-03, 1.0008e-04, 1.1629e-04, 2.1815e-05, 6.4909e-05,
          1.3995e-04, 1.1683e-04, 2.1815e-05, 2.1815e-05, 1.0484e-04, 2.1815e-05,
          1.7118e-04, 1.1629e-04, 1.7118e-04, 2.4223e-04, 2.1815e-05, 2.1815e-05,
          6.4909e-05, 2.1815e-05, 2.1815e-05, 1.3626e-04, 1.0097e-04, 1.0008e-04,
          6.4909e-05, 1.4677e-03, 6.4909e-05, 2.1815e-05, 1.3602e-04, 6.4909e-05,
          9.6917e-05, 2.1815e-05, 6.4909e-05, 2.8205e-04, 2.8205e-04, 1.7118e-04,
          2.1243e-04, 8.4877e-05, 4.0174e-04, 1.1975e-04, 6.4909e-05, 2.1815e-05,
          6.4909e-05, 2.1815e-05, 8.4877e-05, 1.9789e-04, 1.0008e-04, 1.5459e-03,
          1.9109e-04, 2.1815e-05, 4.1723e-05, 2.4223e-04, 6.4909e-05, 4.1723e-05,
          4.1723e-05, 4.9877e-04, 1.0509e-03, 7.3671e-04, 2.0766e-04, 6.4909e-05,
          3.8004e-04, 5.0116e-04, 5.2929e-04, 4.9543e-04, 4.4942e-05, 4.1723e-05,
          2.1815e-05, 3.1734e-04, 4.1151e-04, 1.5152e-04, 1.9133e-04, 2.1374e-04,
          2.1434e-04, 9.6917e-05, 9.6917e-05, 4.4942e-05, 7.6890e-05, 2.1815e-05,
          4.0174e-04, 3.0804e-04, 1.5867e-04, 3.4451e-04, 2.8825e-04, 1.5616e-04,
          4.8757e-04, 4.0174e-04, 6.1750e-05, 2.1815e-05, 2.1815e-05, 2.1815e-05,
          8.4877e-05, 4.6229e-04, 1.5421e-03, 4.4107e-04, 9.1672e-05, 2.3782e-04,
          1.3995e-04, 4.1723e-05, 6.4909e-05, 7.6890e-05, 4.1723e-05, 2.1100e-04,
          6.9666e-04, 1.4091e-04, 3.1424e-04, 1.1015e-03, 1.9109e-04, 2.6393e-04,
          3.8624e-04, 4.3631e-04, 2.1815e-05, 4.1723e-05, 9.2363e-04, 7.1287e-04,
          2.3782e-04, 2.4395e-03, 9.7513e-04, 9.6917e-05, 7.6890e-05, 4.4942e-05,
          2.9111e-04, 4.4942e-05, 2.1815e-05, 4.4942e-05, 1.9383e-04, 3.4475e-04,
          4.7612e-04, 4.7612e-04, 1.0509e-03, 1.1975e-04, 1.1015e-03, 4.2486e-04,
          4.0174e-04, 5.5313e-04, 2.4223e-04, 2.1815e-05, 1.0338e-03, 1.3995e-04,
          6.4909e-05, 5.6982e-05, 5.1785e-04, 1.3995e-04, 9.6560e-05, 4.9829e-04,
          2.1815e-05, 1.9133e-04, 6.1750e-05, 1.9133e-04, 4.4942e-05, 3.8981e-04,
          1.2245e-03, 2.4223e-04, 2.1815e-05, 2.1815e-05, 4.0174e-04, 4.0174e-04,
          4.4942e-05, 8.1658e-05, 2.1815e-05, 1.7142e-04, 1.2004e-04, 4.1723e-05,
          8.4877e-05, 1.1683e-04, 4.1723e-05, 1.0008e-04, 1.1975e-04, 8.4877e-05,
          7.6723e-04, 6.4909e-05, 1.5962e-04, 6.1750e-05, 1.9109e-04, 4.1723e-05,
          6.0415e-04, 5.8794e-04, 2.1815e-05, 1.5962e-04, 3.8505e-04, 2.1815e-05,
          6.1750e-05, 7.6890e-05, 1.7142e-04, 1.2004e-04, 2.7752e-04, 1.0008e-04,
          5.6362e-04, 2.1815e-05, 2.4629e-04, 6.7949e-04, 3.1471e-04, 1.0004e-03,
          1.8377e-03, 4.1437e-04, 7.0572e-04, 4.1723e-05, 2.4223e-04, 6.4909e-05,
          2.7752e-04, 4.4942e-05, 4.1723e-05, 5.9891e-04, 2.1815e-05, 1.4794e-04,
          2.1815e-05, 1.7142e-04, 1.7142e-04, 2.1815e-05, 1.1854e-03, 6.1750e-05,
          1.0977e-03, 3.1471e-04, 7.1526e-04, 2.1815e-05, 6.1750e-05, 5.5313e-04,
          7.4100e-04, 1.3626e-04, 4.6229e-04, 1.5116e-04, 7.1526e-04, 7.1526e-04,
          3.3474e-04, 9.4509e-04, 7.6652e-05, 8.4877e-05, 5.0783e-04, 2.1815e-05,
          1.8752e-04, 2.2233e-04, 1.1611e-04, 1.1611e-04, 1.1611e-04, 1.0008e-04,
          2.1815e-05, 4.2105e-04, 1.1611e-04, 7.1955e-04, 1.1611e-04, 1.1611e-04,
          1.1611e-04, 1.1611e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.3901e-02, 8.4106e-02, 4.3060e-02, 5.2429e-02, 1.2732e-01,
          1.1176e-01, 3.5339e-02, 6.3782e-02, 4.7577e-02, 3.2288e-02, 4.8248e-02,
          1.7807e-02, 3.8330e-02, 9.7275e-03, 2.7466e-02, 8.0566e-02, 3.6133e-02,
          2.1149e-02, 7.1350e-02, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 9.7871e-05,
          1.6570e-05, 1.6570e-05, 1.8036e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 9.7871e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          9.7871e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 5.9962e-05, 5.9962e-05, 1.6570e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 1.8036e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.8036e-04, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 5.9962e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 4.1866e-04, 5.9962e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 4.1866e-04,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 5.9962e-05, 5.9962e-05,
          1.6570e-05, 9.7871e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 4.6206e-04,
          5.9962e-05, 1.9467e-04, 1.9467e-04, 1.6570e-05, 5.9962e-05, 1.6570e-05,
          1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05,
          1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05,
          4.6730e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 1.4246e-04,
          1.9467e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 3.5858e-04,
          1.6570e-05, 5.9962e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 5.9962e-05, 7.7057e-04, 1.6570e-05,
          5.9962e-05, 5.9962e-05, 1.4246e-04, 1.9467e-04, 1.9467e-04, 1.6570e-05,
          1.6570e-05, 5.9962e-05, 1.1644e-03, 2.2376e-04, 1.9467e-04, 5.9962e-05,
          5.9962e-05, 4.6730e-05, 1.9467e-04, 1.6570e-05, 1.8036e-04, 1.6570e-05,
          5.9962e-05, 1.6570e-05, 1.8036e-04, 4.1866e-04, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.9467e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 9.9063e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 3.5858e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          9.9063e-05, 9.9063e-05, 3.5858e-04, 5.9962e-05, 1.8036e-04, 9.9063e-05,
          9.9063e-05, 1.4126e-04, 3.5858e-04, 4.3273e-04, 1.6570e-05, 2.7728e-04,
          4.1866e-04, 2.7728e-04, 3.5858e-04, 1.9467e-04, 1.6570e-05, 7.7057e-04,
          1.9467e-04, 4.0197e-04, 2.2376e-04, 1.4126e-04, 1.6570e-05, 2.7728e-04,
          1.8036e-04, 1.6570e-05, 1.6570e-05, 3.5858e-04, 2.7728e-04, 7.7057e-04,
          1.6570e-05, 5.9962e-05, 3.5858e-04, 5.9962e-05, 1.6570e-05, 3.5858e-04,
          2.7728e-04, 4.0197e-04, 1.9467e-04, 1.6570e-05, 1.6570e-05, 1.6570e-05,
          1.6570e-05, 1.6570e-05, 5.9962e-05, 1.6570e-05, 9.4843e-04, 9.7871e-05,
          1.6570e-05, 1.6570e-05, 1.6570e-05, 6.0654e-04, 1.6570e-05, 1.6570e-05,
          5.9962e-05, 4.0197e-04, 1.6570e-05, 2.2376e-04, 1.6570e-05, 1.6570e-05,
          3.5858e-04, 9.9063e-05, 9.9063e-05, 1.6570e-05, 1.6570e-05, 2.7728e-04,
          1.6570e-05, 7.7057e-04, 6.8808e-04, 3.5858e-04, 2.7728e-04, 9.0551e-04,
          1.6570e-05, 3.5858e-04, 9.4843e-04, 1.6570e-05, 1.9467e-04, 3.5858e-04,
          9.0551e-04, 1.6570e-05, 1.6570e-05, 2.7728e-04, 3.5858e-04, 1.6570e-05,
          3.3331e-04, 1.6570e-05, 3.3331e-04, 9.7871e-05, 3.3331e-04, 1.4126e-04,
          1.6570e-05, 3.7646e-04, 9.7871e-05, 1.0939e-03, 3.3331e-04, 4.1580e-04,
          3.3331e-04, 3.3331e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.2461e-02, 9.1003e-02, 6.4758e-02, 4.5380e-02, 2.1594e-01,
          7.1411e-02, 4.1565e-02, 7.7896e-03, 5.7373e-03, 7.2632e-03, 9.2545e-03,
          4.4403e-03, 1.9257e-02, 6.7322e-02, 8.1635e-03, 6.6795e-03, 6.6833e-02,
          3.0350e-02, 3.5667e-03, 1.7139e-01, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          5.4359e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 8.7857e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05,
          4.7266e-05, 5.4359e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.4305e-05, 1.3769e-05, 1.4901e-05, 1.3769e-05,
          1.3769e-05, 4.7266e-05, 1.3769e-05, 1.4746e-04, 8.7857e-05, 4.7266e-05,
          8.7857e-05, 1.3769e-05, 1.3769e-05, 8.7857e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 5.4359e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          8.7857e-05, 5.4359e-05, 8.8990e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 5.4359e-05, 1.3769e-05, 8.7857e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05, 4.7266e-05, 1.4305e-05,
          1.3769e-05, 1.4901e-05, 1.4901e-05, 4.7266e-05, 8.7857e-05, 1.3769e-05,
          1.8096e-04, 1.8215e-04, 4.7266e-05, 5.4955e-05, 5.4359e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.4305e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8459e-05, 4.7266e-05,
          1.4901e-05, 4.7266e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          5.4359e-05, 1.3769e-05, 1.8096e-04, 4.7266e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 5.4359e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.8215e-04, 3.0365e-03, 4.7266e-05, 4.7266e-05, 1.3769e-05, 1.3769e-05,
          8.7857e-05, 1.4305e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 8.7857e-05, 4.7922e-05, 8.7857e-05, 1.3769e-05, 1.3769e-05,
          3.2353e-04, 1.3769e-05, 1.3769e-05, 1.4305e-05, 4.8459e-05, 8.7857e-05,
          8.7857e-05, 1.3769e-05, 1.3769e-05, 5.4359e-05, 3.2401e-04, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.4806e-04, 5.4359e-05, 1.3769e-05,
          8.8990e-05, 5.4955e-05, 1.3769e-05, 1.8096e-04, 2.3103e-04, 5.4359e-05,
          4.7266e-05, 1.3769e-05, 1.3769e-05, 8.8990e-05, 8.7857e-05, 4.8459e-05,
          1.3769e-05, 1.3769e-05, 4.7266e-05, 1.8096e-04, 1.3769e-05, 4.7266e-05,
          1.3769e-05, 1.3769e-05, 5.4359e-05, 1.8096e-04, 8.8453e-05, 8.7857e-05,
          5.5492e-05, 4.7922e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05, 4.7266e-05,
          1.3769e-05, 1.4746e-04, 4.7266e-05, 4.7266e-05, 1.3769e-05, 4.7266e-05,
          1.4305e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05,
          1.3769e-05, 1.4806e-04, 1.3769e-05, 4.7266e-05, 5.5492e-05, 8.8453e-05,
          8.8990e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.8096e-04, 1.3769e-05, 1.3769e-05, 1.4305e-05, 4.8459e-05,
          1.3769e-05, 1.3769e-05, 8.7857e-05, 4.7266e-05, 4.7266e-05, 1.3769e-05,
          1.8215e-04, 4.8459e-05, 4.7922e-05, 1.4365e-05, 1.3769e-05, 8.7857e-05,
          4.7266e-05, 4.7266e-05, 1.3769e-05, 8.7857e-05, 1.3769e-05, 1.8215e-04,
          8.8990e-05, 1.4901e-05, 1.4901e-05, 1.3769e-05, 5.4359e-05, 5.4359e-05,
          4.7922e-05, 1.3769e-05, 5.4359e-05, 5.4359e-05, 1.3769e-05, 1.4365e-05,
          1.3769e-05, 1.4901e-05, 1.3769e-05, 4.7266e-05, 1.4746e-04, 1.4901e-05,
          1.3769e-05, 5.4359e-05, 3.2353e-04, 1.3769e-05, 1.0800e-04, 4.7922e-05,
          5.4359e-05, 5.4359e-05, 5.4359e-05, 8.7857e-05, 4.7266e-05, 8.7857e-05,
          5.4359e-05, 1.8096e-04, 1.3769e-05, 1.8096e-04, 5.4359e-05, 1.8167e-04,
          1.8167e-04, 5.4693e-04, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          5.4359e-05, 4.7266e-05, 1.3769e-05, 1.3769e-05, 5.4955e-05, 8.7857e-05,
          5.4359e-05, 8.8453e-05, 8.8394e-05, 1.3769e-05, 4.7266e-05, 1.3769e-05,
          1.8167e-04, 1.3769e-05, 4.7266e-05, 1.3769e-05, 5.1260e-04, 8.7857e-05,
          1.8096e-04, 4.7266e-05, 5.4359e-05, 4.0364e-04, 8.7857e-05, 1.3769e-05,
          1.3769e-05, 5.4359e-05, 1.4746e-04, 5.4955e-05, 1.8096e-04, 8.7857e-05,
          4.7266e-05, 4.7266e-05, 1.8096e-04, 4.0364e-04, 8.7857e-05, 3.2353e-04,
          3.2353e-04, 3.2401e-04, 1.4746e-04, 4.7266e-05, 3.2353e-04, 1.3769e-05,
          5.4598e-04, 1.3769e-05, 4.7266e-05, 5.1260e-04, 1.3769e-05, 4.7266e-05,
          5.4359e-05, 1.4806e-04, 1.4746e-04, 4.7266e-05, 8.7857e-05, 1.3769e-05,
          3.2353e-04, 8.7857e-05, 1.7433e-03, 1.3769e-05, 1.3769e-05, 8.7857e-05,
          5.2376e-03, 1.3769e-05, 8.7857e-05, 1.4043e-04, 4.1656e-03, 5.1260e-04,
          5.4359e-05, 5.1260e-04, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.9681e-04, 2.8992e-04, 1.9681e-04, 2.8992e-04, 2.8992e-04, 1.9681e-04,
          1.3769e-05, 2.3043e-04, 2.8992e-04, 8.1491e-04, 5.4359e-05, 1.9681e-04,
          5.4359e-05, 1.9681e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.8259e-02, 9.4482e-02, 7.6233e-02, 5.2917e-02, 2.3669e-01,
          7.0679e-02, 4.5135e-02, 7.1602e-03, 5.5923e-03, 9.7122e-03, 1.3313e-02,
          6.0997e-03, 1.5083e-02, 3.3234e-02, 7.7629e-03, 7.2136e-03, 4.9194e-02,
          2.2263e-02, 4.2725e-03, 7.1533e-02, 9.1492e-02, 1.4246e-05, 1.4246e-05,
          6.3419e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          7.6652e-05, 6.3419e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 7.5996e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4901e-05, 1.4246e-05, 1.6868e-05, 1.4246e-05,
          1.4246e-05, 7.5996e-05, 1.4246e-05, 1.3697e-04, 2.0134e-04, 7.8619e-05,
          6.3419e-05, 1.4246e-05, 1.4246e-05, 6.3419e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 6.3419e-05, 1.4246e-05, 1.4246e-05, 4.8256e-04,
          6.3419e-05, 1.3697e-04, 6.6042e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.2779e-04, 6.6042e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4901e-05, 7.5996e-05, 7.5996e-05, 4.8327e-04,
          4.8256e-04, 1.6868e-05, 1.4901e-05, 1.4246e-05, 6.3419e-05, 1.4246e-05,
          1.9872e-04, 2.0134e-04, 6.3419e-05, 1.3959e-04, 1.3697e-04, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.2517e-04,
          4.8518e-04, 1.4246e-05, 4.8256e-04, 4.8256e-04, 1.2779e-04, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.3697e-04, 1.4246e-05, 1.9872e-04, 7.5996e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 8.7738e-05,
          1.3757e-04, 6.0511e-04, 1.4246e-05, 5.3167e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 7.5996e-05, 7.5996e-05, 1.2517e-04, 1.4246e-05, 1.4246e-05,
          1.3697e-04, 1.4246e-05, 1.4246e-05, 4.8327e-04, 2.0134e-04, 1.2517e-04,
          1.9872e-04, 4.8327e-04, 1.4246e-05, 6.3419e-05, 4.5490e-04, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.3959e-04, 8.7738e-05, 1.4246e-05,
          1.2779e-04, 6.3419e-05, 1.4246e-05, 1.9872e-04, 1.3697e-04, 6.3419e-05,
          7.5996e-05, 1.4246e-05, 1.4246e-05, 6.6948e-04, 6.3419e-05, 6.6948e-04,
          4.8256e-04, 1.4246e-05, 1.4246e-05, 6.0511e-04, 1.4246e-05, 1.9872e-04,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.9872e-04, 7.5996e-05, 1.4246e-05,
          2.0134e-04, 7.5996e-05, 1.4246e-05, 7.5996e-05, 1.4246e-05, 7.5996e-05,
          1.4246e-05, 6.0511e-04, 7.6652e-05, 7.5996e-05, 4.8256e-04, 7.5996e-05,
          1.4901e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.2517e-04, 1.4246e-05,
          1.4246e-05, 1.3888e-04, 1.4246e-05, 7.5996e-05, 1.4901e-05, 1.4246e-05,
          2.0134e-04, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 6.6710e-04, 4.8256e-04, 1.4246e-05, 1.4901e-05, 7.8619e-05,
          1.4246e-05, 1.4246e-05, 1.9872e-04, 1.2517e-04, 1.2517e-04, 1.4246e-05,
          2.0134e-04, 7.8619e-05, 7.5996e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.9872e-04, 1.4246e-05, 2.0134e-04,
          1.2779e-04, 1.4901e-05, 4.8327e-04, 1.4246e-05, 6.3419e-05, 1.4246e-05,
          7.5996e-05, 1.4246e-05, 6.3419e-05, 6.3419e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.6868e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.9872e-04, 1.4246e-05, 1.4901e-05, 4.8256e-04,
          1.4246e-05, 6.3419e-05, 1.4246e-05, 6.3419e-05, 1.4246e-05, 1.2517e-04,
          6.3419e-05, 1.3697e-04, 1.4246e-05, 1.9872e-04, 1.4246e-05, 1.4246e-05,
          9.7275e-04, 3.9315e-04, 1.4246e-05, 6.3419e-05, 1.4246e-05, 1.4246e-05,
          6.3419e-05, 7.5996e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.9872e-04,
          6.3419e-05, 1.2708e-04, 1.9932e-04, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.0347e-03, 1.4246e-05, 1.2517e-04, 1.4246e-05, 3.7551e-04, 1.9872e-04,
          1.3697e-04, 7.5996e-05, 6.3419e-05, 4.3726e-04, 1.2517e-04, 1.4246e-05,
          1.3697e-04, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.9872e-04, 1.9872e-04,
          7.5996e-05, 8.4400e-04, 1.3697e-04, 8.4400e-04, 1.9872e-04, 1.2293e-03,
          2.3613e-03, 1.9872e-04, 9.7275e-04, 1.2517e-04, 1.9872e-04, 1.4246e-05,
          6.9332e-04, 1.9872e-04, 7.5996e-05, 6.3181e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 2.8286e-03, 1.4246e-05,
          1.0347e-03, 1.9872e-04, 6.9332e-04, 1.4246e-05, 9.2363e-04, 8.4400e-04,
          1.2903e-03, 8.7738e-05, 1.0347e-03, 1.9872e-04, 3.7551e-04, 3.7551e-04,
          9.7275e-04, 1.4677e-03, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          6.3419e-05, 1.3697e-04, 3.1972e-04, 1.5342e-04, 6.3419e-05, 6.3419e-05,
          1.4246e-05, 1.4246e-05, 6.3419e-05, 7.8201e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.2415e-02, 8.5388e-02, 7.2510e-02, 4.1748e-02, 1.6089e-01,
          7.2083e-02, 3.9490e-02, 5.6305e-03, 4.5395e-03, 7.7019e-03, 1.0704e-02,
          4.8599e-03, 1.8036e-02, 4.0955e-02, 6.3934e-03, 5.9242e-03, 6.5002e-02,
          3.2837e-02, 3.3588e-03, 6.2683e-02, 4.8218e-02, 1.5491e-01, 1.7345e-05,
          6.5506e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05,
          5.1022e-05, 6.5506e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7762e-05, 1.7345e-05, 1.7762e-05, 1.7345e-05,
          1.7345e-05, 5.1022e-05, 1.7345e-05, 6.5506e-05, 5.1022e-05, 5.1022e-05,
          6.5506e-05, 1.7345e-05, 1.7345e-05, 6.5506e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          6.5506e-05, 6.5506e-05, 6.5863e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 9.9123e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 5.1022e-05, 1.7345e-05,
          1.2560e-03, 1.7762e-05, 1.7762e-05, 5.1022e-05, 6.5506e-05, 1.7345e-05,
          9.9123e-05, 6.6161e-05, 5.1022e-05, 6.5506e-05, 6.5506e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7762e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1379e-05, 9.9123e-05,
          1.7345e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          6.5506e-05, 1.7345e-05, 9.9123e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          6.5506e-05, 6.5506e-05, 5.1022e-05, 9.9123e-05, 1.7345e-05, 1.7345e-05,
          9.9123e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          5.1022e-05, 9.9123e-05, 5.1022e-05, 9.9123e-05, 1.7345e-05, 1.7345e-05,
          9.9123e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 9.9838e-05, 9.9123e-05,
          9.9123e-05, 1.2560e-03, 1.7345e-05, 1.7345e-05, 9.9123e-05, 1.7345e-05,
          1.7405e-05, 1.7345e-05, 5.1022e-05, 6.5506e-05, 6.5506e-05, 9.9123e-05,
          5.1677e-05, 6.5506e-05, 1.7345e-05, 5.1022e-05, 9.9123e-05, 1.7345e-05,
          5.1022e-05, 1.7345e-05, 1.7345e-05, 9.9838e-05, 9.9123e-05, 5.1677e-05,
          1.3037e-03, 1.7345e-05, 1.7345e-05, 2.5225e-04, 1.7345e-05, 9.9123e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 9.9123e-05, 9.9123e-05, 9.9123e-05,
          6.6161e-05, 5.1022e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05, 5.1022e-05,
          1.7345e-05, 1.7345e-05, 5.1022e-05, 5.1022e-05, 1.7345e-05, 5.1022e-05,
          5.1022e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05,
          1.7345e-05, 6.5804e-05, 1.7345e-05, 5.1022e-05, 9.9838e-05, 9.9421e-05,
          9.9838e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          1.7345e-05, 9.9123e-05, 1.7345e-05, 1.7345e-05, 1.2560e-03, 5.1677e-05,
          1.7345e-05, 1.7345e-05, 9.9123e-05, 5.1022e-05, 9.9123e-05, 5.1022e-05,
          9.9838e-05, 5.1677e-05, 9.9421e-05, 1.7703e-05, 6.5506e-05, 9.9123e-05,
          5.1022e-05, 5.1022e-05, 1.7345e-05, 9.9123e-05, 1.7345e-05, 9.9838e-05,
          9.9838e-05, 1.7762e-05, 1.7762e-05, 1.2560e-03, 5.1022e-05, 1.7345e-05,
          5.1022e-05, 1.7345e-05, 1.7345e-05, 6.5506e-05, 1.7405e-05, 5.1320e-05,
          1.7345e-05, 1.8060e-05, 1.7405e-05, 5.1022e-05, 1.7345e-05, 1.8060e-05,
          1.7405e-05, 1.7345e-05, 2.5225e-04, 1.7345e-05, 1.7762e-05, 1.7345e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 1.7345e-05, 5.1022e-05,
          1.7345e-05, 9.9123e-05, 1.7345e-05, 9.9123e-05, 1.7345e-05, 9.9421e-05,
          3.3712e-04, 4.8971e-04, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          6.5506e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05, 9.9123e-05,
          1.7345e-05, 9.9123e-05, 9.9123e-05, 5.1022e-05, 6.5506e-05, 1.7345e-05,
          2.4166e-03, 5.1022e-05, 5.1022e-05, 1.7345e-05, 6.5506e-05, 9.9123e-05,
          6.5506e-05, 5.1022e-05, 1.7345e-05, 9.9123e-05, 9.9123e-05, 1.7345e-05,
          1.7345e-05, 5.1022e-05, 1.7345e-05, 1.7345e-05, 2.5225e-04, 9.9123e-05,
          5.1022e-05, 5.1022e-05, 5.1022e-05, 4.8971e-04, 9.9123e-05, 4.8971e-04,
          2.5225e-04, 9.9123e-05, 6.5506e-05, 5.1022e-05, 9.9123e-05, 1.7345e-05,
          7.4625e-04, 6.5506e-05, 5.1022e-05, 7.4625e-04, 1.7345e-05, 5.1022e-05,
          1.7345e-05, 1.7345e-05, 1.7345e-05, 5.1022e-05, 9.9123e-05, 1.7345e-05,
          9.9123e-05, 9.9123e-05, 4.8971e-04, 1.7345e-05, 1.7345e-05, 9.9123e-05,
          4.8971e-04, 1.7345e-05, 9.9123e-05, 5.1022e-05, 2.5225e-04, 2.5225e-04,
          6.5506e-05, 2.5368e-03, 1.7345e-05, 1.7345e-05, 1.7345e-05, 1.7345e-05,
          2.1851e-04, 4.5609e-04, 2.1851e-04, 1.6117e-04, 6.5506e-05, 6.5506e-05,
          1.7345e-05, 2.5225e-04, 2.1851e-04, 9.1219e-04, 1.7345e-05, 1.7345e-05,
          6.5506e-05, 2.1851e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.5717e-02, 7.6599e-02, 5.6122e-02, 6.3171e-02, 1.1584e-01,
          1.2305e-01, 3.7201e-02, 1.8738e-02, 1.2711e-02, 3.3783e-02, 3.8544e-02,
          1.7334e-02, 3.7506e-02, 1.6083e-02, 2.3361e-02, 2.2171e-02, 5.8136e-02,
          4.8553e-02, 1.2833e-02, 3.8872e-03, 6.0844e-03, 8.4763e-03, 1.1926e-01,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          6.0678e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.8466e-04, 6.0678e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 2.3007e-04, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.8466e-04, 1.5259e-05,
          4.7827e-04, 6.0678e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          6.0678e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 6.0678e-05, 6.0678e-05, 2.3007e-04, 1.5259e-05, 1.5259e-05,
          6.0678e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05,
          6.0678e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05,
          6.0678e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 2.9087e-04, 7.5388e-04, 6.0678e-05, 6.0678e-05,
          6.0678e-05, 2.3007e-04, 1.5259e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 2.9206e-04, 6.0678e-05, 1.5259e-05, 2.9206e-04,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 2.9087e-04, 1.5259e-05,
          7.5388e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 5.4026e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05, 6.0678e-05, 1.5259e-05,
          1.0471e-03, 6.0678e-05, 3.0899e-04, 1.5259e-05, 1.5259e-05, 6.0678e-05,
          2.9087e-04, 6.0678e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 9.8801e-04,
          6.0678e-05, 1.5259e-05, 2.9087e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          2.9206e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 2.9087e-04, 2.9087e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 6.0678e-05, 1.5259e-05, 2.9087e-04, 6.0678e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 6.0678e-05,
          1.5259e-05, 3.0899e-04, 1.5259e-05, 2.3007e-04, 1.5259e-05, 3.0899e-04,
          3.3641e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05, 1.5259e-05, 2.9087e-04, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 6.0678e-05, 6.0678e-05, 6.0678e-05, 1.5259e-05, 1.5259e-05,
          7.0953e-04, 1.5259e-05, 6.0678e-05, 1.5259e-05, 2.9087e-04, 6.0678e-05,
          1.5259e-05, 6.0678e-05, 1.5259e-05, 6.0678e-05, 4.8137e-04, 1.5259e-05,
          2.9087e-04, 1.5259e-05, 2.9087e-04, 2.9087e-04, 2.9206e-04, 2.9206e-04,
          4.8137e-04, 1.5259e-05, 6.0678e-05, 6.0678e-05, 6.0678e-05, 1.4963e-03,
          7.0953e-04, 4.6158e-04, 1.0757e-03, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          8.0013e-04, 1.8466e-04, 1.5259e-05, 8.9884e-04, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 2.9087e-04, 1.5259e-05, 1.5259e-05, 2.9206e-04, 1.5259e-05,
          1.3075e-03, 6.0678e-05, 6.0678e-05, 1.5259e-05, 2.9087e-04, 1.5259e-05,
          1.4057e-03, 2.9087e-04, 5.0592e-04, 1.5259e-05, 6.0678e-05, 7.5483e-04,
          4.6039e-04, 1.0757e-03, 1.5259e-05, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.8466e-04, 1.8466e-04, 1.5259e-05, 1.5259e-05, 1.8466e-04, 1.5259e-05,
          1.5259e-05, 1.8466e-04, 1.8466e-04, 1.5259e-05, 1.5259e-05, 1.5259e-05,
          1.5259e-05, 1.5259e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.9547e-02, 9.4666e-02, 4.2725e-02, 6.9275e-02, 1.2598e-01,
          1.3025e-01, 4.1870e-02, 1.3184e-02, 1.0612e-02, 4.8553e-02, 9.6619e-02,
          3.1921e-02, 1.2840e-02, 5.4703e-03, 1.5472e-02, 1.3458e-02, 1.2634e-02,
          9.0485e-03, 9.3460e-03, 1.8921e-03, 3.1509e-03, 2.8954e-03, 1.3260e-02,
          1.2817e-01, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05,
          6.6578e-05, 6.6578e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05,
          2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 6.6578e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05,
          2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.0862e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 6.6578e-05,
          6.6578e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05,
          6.6578e-05, 2.8670e-05, 2.8670e-05, 1.8573e-04, 6.6578e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 6.6578e-05, 6.6578e-05,
          7.8380e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          1.3280e-04, 1.1629e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05, 1.1629e-04,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 1.4448e-04, 1.1629e-04,
          2.8670e-05, 2.8670e-05, 1.8239e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 6.6578e-05, 6.6578e-05, 1.1629e-04, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 6.6578e-05, 1.8239e-04, 1.8239e-04, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 6.6578e-05,
          6.6578e-05, 2.8670e-05, 2.8670e-05, 1.8239e-04, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.3818e-04, 1.4448e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05, 1.8239e-04,
          1.8573e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05, 6.6578e-05, 2.8670e-05,
          7.8380e-05, 1.1629e-04, 2.8670e-05, 6.6578e-05, 6.6578e-05, 3.0160e-04,
          1.1629e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05, 1.8239e-04, 6.6578e-05,
          2.8670e-05, 7.8380e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.0862e-04,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 1.1629e-04, 2.8670e-05, 1.8573e-04,
          2.8670e-05, 2.8670e-05, 7.8380e-05, 1.4448e-04, 3.9387e-04, 1.8370e-04,
          2.8670e-05, 2.8670e-05, 7.8380e-05, 2.7466e-04, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 7.8380e-05, 2.8670e-05, 2.8670e-05,
          7.8380e-05, 2.8670e-05, 6.6578e-05, 2.3544e-04, 1.8239e-04, 7.8380e-05,
          2.7466e-04, 6.6578e-05, 1.8239e-04, 2.8670e-05, 2.3687e-04, 2.8670e-05,
          2.8670e-05, 6.6578e-05, 7.0381e-04, 1.8239e-04, 2.8670e-05, 6.7890e-05,
          6.6578e-05, 2.8670e-05, 2.8670e-05, 2.8670e-05, 3.9387e-04, 2.3687e-04,
          1.8573e-04, 2.8670e-05, 1.8239e-04, 6.6578e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 2.8670e-05, 1.8239e-04, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 7.8380e-05, 1.8573e-04, 4.7660e-04, 2.8670e-05, 2.8670e-05,
          1.4448e-04, 1.7271e-03, 2.3687e-04, 3.2783e-04, 2.8670e-05, 2.8670e-05,
          2.0862e-04, 2.0862e-04, 6.7854e-04, 2.5201e-04, 2.8670e-05, 1.4448e-04,
          7.8380e-05, 2.8670e-05, 2.3687e-04, 2.7466e-04, 6.1226e-04, 1.7071e-04,
          1.4448e-04, 2.0862e-04, 1.4448e-04, 2.8670e-05, 2.8670e-05, 3.0279e-04,
          2.8670e-05, 6.6185e-04, 3.0279e-04, 3.0160e-04, 2.8670e-05, 1.7071e-04,
          2.0862e-04, 6.6578e-05, 3.2783e-04, 8.1968e-04, 2.8670e-05, 1.8239e-04,
          2.8670e-05, 2.8670e-05, 2.7466e-04, 6.6578e-05, 1.2178e-03, 6.1226e-04,
          9.4652e-04, 1.8573e-04, 2.8670e-05, 2.8670e-05, 1.8573e-04, 2.7800e-04,
          1.4887e-03, 2.8670e-05, 2.8670e-05, 2.8670e-05, 1.8370e-04, 1.8728e-04,
          6.6185e-04, 8.0299e-04, 1.2100e-04, 6.6578e-05, 1.1629e-04, 2.8670e-05,
          6.6578e-05, 4.1604e-04, 6.6578e-05, 6.6578e-05, 2.8670e-05, 2.8670e-05,
          2.8670e-05, 2.8670e-05, 7.8380e-05, 6.6578e-05, 2.7800e-04, 1.7071e-04,
          1.8225e-03, 5.0306e-04, 2.8670e-05, 2.8670e-05, 1.7071e-04, 3.4618e-04,
          2.7800e-04, 8.1968e-04, 5.4264e-04, 6.6578e-05, 2.8670e-05, 4.2772e-04,
          5.0449e-04, 1.8559e-03, 2.8670e-05, 4.2772e-04, 8.0299e-04, 2.8670e-05,
          2.7227e-04, 2.8670e-05, 2.8670e-05, 2.8670e-05, 4.2987e-04, 1.8573e-04,
          2.8670e-05, 1.3280e-04, 2.8670e-05, 8.1396e-04, 2.7227e-04, 2.7227e-04,
          2.8670e-05, 2.8670e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.6266e-02, 8.1177e-02, 4.2633e-02, 6.0059e-02, 1.3794e-01,
          1.2225e-01, 4.0192e-02, 1.1497e-02, 1.3153e-02, 3.3997e-02, 3.8849e-02,
          1.9333e-02, 1.7471e-02, 4.7760e-03, 5.3345e-02, 1.3832e-02, 1.5656e-02,
          9.0027e-03, 1.3512e-02, 3.0231e-03, 5.1117e-03, 4.0359e-03, 2.0187e-02,
          6.9763e-02, 1.2720e-01, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1672e-04, 2.1160e-05, 1.7989e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 1.7989e-04, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 5.5695e-04, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 3.6120e-04,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 4.2272e-04, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 3.6120e-04, 2.1160e-05,
          2.6393e-04, 2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.0278e-04, 2.0278e-04, 2.1160e-05, 2.1160e-05,
          3.9816e-04, 2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1160e-05, 6.1131e-04,
          2.1160e-05, 2.1160e-05, 5.7077e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.0278e-04, 2.1160e-05, 6.0415e-04,
          2.1160e-05, 2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.6393e-04, 2.6393e-04,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 4.3011e-04, 2.1160e-05,
          2.6393e-04, 2.6393e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 3.6120e-04, 2.1160e-05, 2.1672e-04, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.6393e-04, 2.1160e-05,
          3.6120e-04, 2.6393e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          2.1160e-05, 2.0278e-04, 6.0415e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05,
          3.1528e-03, 2.1160e-05, 2.1160e-05, 2.0278e-04, 2.0278e-04, 2.1160e-05,
          4.5729e-04, 2.1160e-05, 2.1160e-05, 2.1160e-05, 4.5729e-04, 2.1160e-05,
          2.1160e-05, 4.5729e-04, 2.1160e-05, 1.1740e-03, 2.9850e-04, 1.1740e-03,
          2.1160e-05, 2.1160e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.3722e-02, 9.9731e-02, 6.3843e-02, 1.7944e-01, 1.5295e-01,
          6.2500e-02, 1.7358e-01, 3.2578e-03, 2.2736e-03, 7.3471e-03, 8.6670e-03,
          3.6869e-03, 3.4294e-03, 3.9864e-03, 3.1166e-03, 2.8801e-03, 6.6223e-03,
          4.0970e-03, 1.5745e-03, 2.0962e-03, 2.5177e-03, 2.3689e-03, 4.5853e-03,
          8.9645e-03, 3.9101e-03, 1.2549e-01, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05,
          7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05,
          2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          7.4923e-05, 1.0614e-03, 7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05,
          2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05,
          9.1553e-04, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          1.3895e-03, 2.6941e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 7.4923e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05,
          7.4923e-05, 6.6233e-04, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          6.1417e-04, 7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05,
          3.2806e-04, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          2.6941e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          6.1417e-04, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05,
          2.6941e-05, 7.4923e-05, 3.2806e-04, 7.4923e-05, 8.5306e-04, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05, 3.2806e-04, 7.4923e-05,
          7.4923e-05, 2.6941e-05, 7.4923e-05, 2.6941e-05, 7.4923e-05, 1.6489e-03,
          7.4923e-05, 7.4923e-05, 6.1417e-04, 2.6941e-05, 7.4923e-05, 2.6941e-05,
          7.4923e-05, 2.6941e-05, 2.6941e-05, 8.5306e-04, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 7.4923e-05, 7.4923e-05, 2.6941e-05,
          1.9503e-03, 7.4923e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          3.2806e-04, 2.6941e-05, 7.4923e-05, 2.6941e-05, 2.6941e-05, 7.1526e-04,
          1.6489e-03, 2.3365e-03, 2.6941e-05, 2.6941e-05, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 1.8382e-04, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05, 2.6941e-05, 4.1175e-04, 2.6941e-05, 2.6941e-05,
          2.6941e-05, 2.6941e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.7039e-02, 9.0454e-02, 6.2988e-02, 1.2036e-01, 1.4136e-01,
          1.0486e-01, 7.3914e-02, 1.0231e-02, 7.4654e-03, 1.9547e-02, 1.7273e-02,
          9.0332e-03, 1.0551e-02, 9.5291e-03, 9.9106e-03, 8.9264e-03, 1.9714e-02,
          1.4397e-02, 4.4479e-03, 4.7073e-03, 6.4545e-03, 4.7722e-03, 1.8967e-02,
          1.7242e-02, 9.4299e-03, 5.8685e-02, 9.2224e-02, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 2.2650e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 2.2650e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05,
          8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          2.2650e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.7976e-05, 8.5235e-05,
          1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 2.3639e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          8.5235e-05, 1.9908e-05, 8.5235e-05, 8.5235e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 8.5235e-05, 8.5235e-05, 2.3639e-04, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.7976e-05, 8.5235e-05,
          8.5235e-05, 1.9908e-05, 1.9908e-05, 2.3639e-04, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 2.3639e-04,
          8.5235e-05, 1.9908e-05, 1.9908e-05, 8.7976e-05, 8.5235e-05, 2.2650e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 8.7976e-05, 8.5235e-05, 1.9908e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          2.5868e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 2.2650e-05,
          1.9908e-05, 1.9908e-05, 8.5235e-05, 8.5235e-05, 2.3639e-04, 1.9908e-05,
          1.0242e-03, 9.1195e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.0242e-03,
          8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.7107e-04, 1.9908e-05,
          1.1501e-03, 1.9908e-05, 1.7107e-04, 2.3639e-04, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05,
          1.7107e-04, 8.5235e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.7107e-04, 1.9908e-05, 1.9908e-05,
          1.7107e-04, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 2.3639e-04,
          1.9908e-05, 8.5235e-05, 2.3639e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          2.3639e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 2.3639e-04,
          1.9908e-05, 8.5235e-05, 2.3639e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 8.5235e-05, 8.5235e-05,
          8.5235e-05, 1.9908e-05, 8.5235e-05, 1.9908e-05, 2.3639e-04, 1.7107e-04,
          1.1663e-03, 2.3639e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          8.5235e-05, 1.7107e-04, 1.9908e-05, 1.2360e-03, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.0147e-03, 1.9908e-05,
          1.1663e-03, 8.5235e-05, 8.5235e-05, 1.9908e-05, 1.9908e-05, 1.7107e-04,
          2.3639e-04, 1.9908e-05, 2.3639e-04, 1.9908e-05, 1.9908e-05, 1.7107e-04,
          1.7107e-04, 1.1015e-03, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.7107e-04, 1.9908e-05,
          1.9908e-05, 1.7107e-04, 1.9908e-05, 1.9908e-05, 1.9908e-05, 1.9908e-05,
          1.9908e-05, 1.9908e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.9119e-02, 7.7026e-02, 5.7190e-02, 3.9062e-02, 1.3574e-01,
          6.0181e-02, 3.3081e-02, 8.7280e-03, 5.5504e-03, 7.4310e-03, 9.8038e-03,
          4.7722e-03, 1.8372e-02, 5.4932e-02, 6.3171e-03, 5.5122e-03, 6.4148e-02,
          2.5131e-02, 3.1261e-03, 5.2307e-02, 2.4796e-02, 3.6102e-02, 1.0864e-02,
          8.9035e-03, 6.6109e-03, 2.4704e-02, 1.9745e-02, 1.4990e-01, 1.2636e-05,
          1.2636e-05, 5.2691e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.7459e-05, 1.2636e-05,
          1.2636e-05, 5.7459e-05, 1.2636e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2994e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.0180e-04, 1.2636e-05, 5.2691e-05,
          5.7042e-05, 5.7459e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 5.7042e-05, 5.7220e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05,
          1.2636e-05, 5.7459e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05, 5.7459e-05,
          5.7459e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 5.2691e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 5.2691e-05, 1.2636e-05,
          9.7096e-05, 1.2755e-05, 1.2636e-05, 5.7459e-05, 5.7459e-05, 5.7459e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          1.2994e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 9.7096e-05,
          1.2755e-05, 5.2691e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.7459e-05,
          5.7459e-05, 5.7459e-05, 1.0180e-04, 5.7459e-05, 5.7459e-05, 1.2636e-05,
          5.7459e-05, 5.7459e-05, 5.7042e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          5.7042e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          9.7096e-05, 9.7454e-05, 1.2636e-05, 1.2636e-05, 5.7459e-05, 1.2636e-05,
          5.7459e-05, 5.2691e-05, 5.2691e-05, 1.4186e-04, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.3108e-05, 5.2691e-05,
          1.2636e-05, 1.2696e-05, 1.2636e-05, 5.7042e-05, 9.7096e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 5.7459e-05, 1.0192e-04, 5.7459e-05, 5.7459e-05,
          9.7573e-05, 1.0180e-04, 5.7459e-05, 5.2691e-05, 5.7042e-05, 5.7042e-05,
          5.2691e-05, 1.2636e-05, 1.2636e-05, 9.7573e-05, 5.2691e-05, 5.2810e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 9.7096e-05, 1.2636e-05, 5.2691e-05,
          1.2636e-05, 5.7518e-05, 4.3011e-04, 1.0180e-04, 4.2725e-04, 5.2691e-05,
          1.0192e-04, 1.4186e-04, 4.3011e-04, 9.7454e-05, 5.7459e-05, 1.2636e-05,
          1.2636e-05, 1.2636e-05, 5.2691e-05, 5.2691e-05, 1.2636e-05, 5.2691e-05,
          5.2750e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05,
          5.7459e-05, 2.6703e-04, 5.7518e-05, 3.8290e-04, 1.0186e-04, 5.7459e-05,
          1.0204e-04, 5.7518e-05, 1.2696e-05, 5.7459e-05, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 9.7096e-05, 1.2636e-05, 1.2636e-05, 1.2696e-05, 5.3108e-05,
          5.2691e-05, 1.2636e-05, 5.2691e-05, 5.2691e-05, 5.2691e-05, 9.7454e-05,
          3.0708e-04, 5.3108e-05, 4.2748e-04, 5.7518e-05, 1.0180e-04, 5.7459e-05,
          5.7459e-05, 1.4186e-04, 1.2636e-05, 5.2691e-05, 1.2636e-05, 9.7573e-05,
          9.7215e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 5.7042e-05, 5.7459e-05,
          5.2691e-05, 1.2636e-05, 5.7042e-05, 3.1066e-04, 1.2696e-05, 5.2750e-05,
          5.7518e-05, 5.7638e-05, 5.7518e-05, 9.7454e-05, 4.7445e-04, 5.7638e-05,
          5.7518e-05, 2.1863e-04, 5.2691e-05, 1.2636e-05, 1.2696e-05, 5.2691e-05,
          1.2636e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05, 5.2691e-05,
          1.2636e-05, 9.7096e-05, 1.2636e-05, 5.2691e-05, 1.2636e-05, 9.7215e-05,
          3.0708e-04, 3.5548e-04, 5.7459e-05, 5.7459e-05, 5.7518e-05, 5.7459e-05,
          1.0180e-04, 1.2636e-05, 1.2636e-05, 1.2636e-05, 1.2636e-05, 9.7096e-05,
          1.2636e-05, 5.2691e-05, 5.2691e-05, 5.2691e-05, 1.2636e-05, 5.7459e-05,
          1.4186e-04, 5.2691e-05, 1.4186e-04, 1.2636e-05, 3.0975e-03, 9.7096e-05,
          1.7428e-04, 9.7454e-05, 5.7042e-05, 2.1422e-04, 9.7454e-05, 5.7459e-05,
          1.2636e-05, 5.7042e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 5.2691e-05,
          5.2691e-05, 1.2636e-05, 2.5868e-04, 2.1863e-04, 5.2691e-05, 3.5071e-04,
          1.1568e-03, 5.2691e-05, 9.7454e-05, 5.2691e-05, 5.2691e-05, 1.2636e-05,
          5.6028e-04, 1.2636e-05, 5.2691e-05, 5.6028e-04, 1.2636e-05, 9.7454e-05,
          5.7042e-05, 1.2636e-05, 1.2636e-05, 5.2691e-05, 9.7561e-04, 1.2636e-05,
          2.3377e-04, 5.2691e-05, 1.2903e-03, 1.2636e-05, 1.2636e-05, 2.1863e-04,
          5.6028e-04, 1.2636e-05, 9.7096e-05, 5.2691e-05, 7.2384e-04, 5.2023e-04,
          5.7042e-05, 5.2023e-04, 1.7428e-04, 1.2636e-05, 1.2636e-05, 1.2636e-05,
          5.7042e-05, 5.7042e-05, 1.9372e-04, 5.7042e-05, 3.5882e-04, 5.7042e-05,
          1.2636e-05, 3.9887e-04, 5.7042e-05, 4.9543e-04, 1.2636e-05, 1.2636e-05,
          1.2636e-05, 5.7042e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.4870e-02, 8.1238e-02, 3.9307e-02, 4.6417e-02, 1.0760e-01,
          9.1003e-02, 2.3117e-02, 1.0422e-02, 7.9803e-03, 2.1820e-02, 1.8753e-02,
          9.7046e-03, 3.3081e-02, 2.2491e-02, 1.2985e-02, 9.6588e-03, 8.8562e-02,
          4.0100e-02, 4.4823e-03, 1.0796e-02, 8.2474e-03, 1.1841e-02, 2.2018e-02,
          1.9730e-02, 9.0103e-03, 1.9531e-02, 2.9083e-02, 1.5671e-02, 1.0107e-01,
          1.8716e-05, 5.9485e-05, 1.8716e-05, 4.0770e-05, 1.8716e-05, 3.7372e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 4.0770e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 4.4823e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 4.0770e-05,
          2.7180e-04, 1.1998e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 4.4823e-05, 4.0770e-05, 6.6936e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 6.0081e-05, 4.0770e-05, 6.0081e-05, 1.8716e-05,
          1.8716e-05, 3.7372e-05, 1.8716e-05, 1.3137e-04, 1.2171e-04, 2.0814e-04,
          5.9485e-05, 1.8716e-05, 1.8942e-04, 5.9485e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 4.4823e-05, 3.7372e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          5.9485e-05, 1.9193e-04, 5.9485e-05, 1.8716e-05, 4.0770e-05, 1.8716e-05,
          6.3539e-05, 1.4186e-04, 3.7372e-05, 6.6936e-05, 6.5422e-04, 2.4605e-04,
          4.0770e-05, 1.8716e-05, 1.8716e-05, 4.0770e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 4.0770e-05, 4.0770e-05, 6.3539e-05,
          4.4823e-05, 8.6248e-05, 8.6248e-05, 4.0770e-05, 1.9193e-04, 1.8716e-05,
          4.4441e-04, 2.4605e-04, 5.9485e-05, 1.7142e-04, 5.9485e-05, 1.4186e-04,
          1.8716e-05, 3.7372e-05, 3.7372e-05, 1.8716e-05, 1.8716e-05, 4.0770e-05,
          6.3539e-05, 3.7372e-05, 5.9485e-05, 4.4823e-05, 1.4114e-04, 1.1438e-04,
          1.8716e-05, 5.9485e-05, 4.4823e-05, 5.9485e-05, 4.4823e-05, 1.4186e-04,
          5.9485e-05, 1.8668e-04, 1.9193e-04, 5.1689e-04, 5.9485e-05, 4.0770e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.3542e-04,
          5.9485e-05, 1.5497e-03, 5.9485e-05, 5.9485e-05, 1.8716e-05, 5.9485e-05,
          1.2183e-04, 1.2183e-04, 1.8716e-05, 1.8716e-05, 3.7372e-05, 1.8716e-05,
          1.3137e-04, 5.9485e-05, 5.9485e-05, 1.9193e-04, 1.8716e-05, 1.8716e-05,
          5.9485e-05, 1.8716e-05, 1.8716e-05, 1.0115e-04, 1.4114e-04, 1.1438e-04,
          9.7871e-05, 1.4772e-03, 5.9485e-05, 1.8716e-05, 5.9485e-05, 4.0770e-05,
          9.9719e-05, 4.4823e-05, 5.9485e-05, 1.5748e-04, 1.5748e-04, 1.3137e-04,
          3.9506e-04, 6.3539e-05, 1.6797e-04, 9.3699e-05, 5.9485e-05, 1.8716e-05,
          1.3137e-04, 1.8716e-05, 1.0115e-04, 6.0463e-04, 5.9485e-05, 1.5926e-03,
          6.3539e-05, 1.8716e-05, 1.8716e-05, 1.3137e-04, 4.0770e-05, 5.9485e-05,
          4.0770e-05, 4.2415e-04, 3.6478e-04, 6.6805e-04, 2.0075e-04, 5.9485e-05,
          3.5810e-04, 8.5545e-04, 4.9496e-04, 3.3092e-04, 1.8716e-05, 4.0770e-05,
          1.8716e-05, 1.3542e-04, 3.3426e-04, 1.8620e-04, 1.4055e-04, 3.1495e-04,
          3.1495e-04, 1.2267e-04, 4.4823e-05, 3.7372e-05, 4.0770e-05, 8.5652e-05,
          1.6797e-04, 2.9755e-04, 4.2415e-04, 7.8344e-04, 7.6628e-04, 3.8695e-04,
          7.4673e-04, 1.6797e-04, 1.6797e-04, 1.4186e-04, 1.8716e-05, 1.8716e-05,
          6.3539e-05, 1.8196e-03, 1.4772e-03, 8.5652e-05, 4.4823e-05, 2.1887e-04,
          8.5652e-05, 1.8716e-05, 5.9485e-05, 9.5665e-05, 5.9485e-05, 4.2462e-04,
          8.2541e-04, 2.1052e-04, 3.2401e-04, 3.6478e-04, 2.2435e-04, 1.8346e-04,
          5.7030e-04, 3.5048e-04, 1.4186e-04, 1.4591e-04, 4.4823e-05, 6.7711e-04,
          6.6853e-04, 1.5516e-03, 1.2960e-03, 9.9719e-05, 1.2183e-04, 3.7372e-05,
          2.3723e-04, 3.7372e-05, 1.8716e-05, 1.8716e-05, 3.0065e-04, 3.0494e-04,
          3.4642e-04, 4.0269e-04, 3.4642e-04, 5.9485e-05, 3.6478e-04, 4.0841e-04,
          1.6797e-04, 7.1585e-05, 1.9193e-04, 1.8716e-05, 1.0115e-04, 1.4055e-04,
          5.9485e-05, 1.1843e-04, 6.3539e-05, 1.4055e-04, 9.3699e-05, 9.3699e-05,
          1.8716e-05, 1.4055e-04, 4.4823e-05, 2.1243e-04, 1.8716e-05, 2.4390e-04,
          7.2098e-04, 1.9193e-04, 1.8716e-05, 1.8716e-05, 1.6797e-04, 1.6797e-04,
          3.7372e-05, 4.0770e-05, 1.8716e-05, 4.4823e-05, 1.1843e-04, 5.9485e-05,
          6.3539e-05, 5.3358e-04, 2.9564e-04, 5.9485e-05, 7.1585e-05, 1.3542e-04,
          5.4359e-04, 5.9485e-05, 1.5748e-04, 4.4823e-05, 2.5845e-04, 3.8195e-04,
          4.9305e-04, 9.3699e-05, 1.8716e-05, 3.6693e-04, 3.5834e-04, 1.6797e-04,
          2.2435e-04, 9.5665e-05, 4.4823e-05, 1.7750e-04, 5.4359e-04, 1.1986e-04,
          4.1103e-04, 7.1585e-05, 2.1243e-04, 5.2309e-04, 5.3787e-04, 3.8838e-04,
          1.2751e-03, 1.2751e-03, 4.5657e-04, 8.5652e-05, 3.1900e-04, 4.0770e-05,
          3.7003e-04, 5.4264e-04, 4.0770e-05, 5.9795e-04, 1.8716e-05, 2.3234e-04,
          4.4823e-05, 1.7750e-04, 1.0115e-04, 1.8716e-05, 1.2522e-03, 4.4823e-05,
          9.1839e-04, 5.9271e-04, 4.6062e-04, 1.8716e-05, 4.4823e-05, 5.2309e-04,
          6.9904e-04, 2.0564e-04, 7.8154e-04, 1.6975e-04, 5.2309e-04, 3.1495e-04,
          7.1096e-04, 7.7152e-04, 2.1362e-04, 1.6797e-04, 2.3973e-04, 1.8716e-05,
          1.5771e-04, 1.6975e-04, 3.7372e-05, 3.7372e-05, 3.7372e-05, 4.0770e-05,
          1.8716e-05, 1.7488e-04, 9.7871e-05, 4.7779e-04, 1.8716e-05, 3.7372e-05,
          1.8716e-05, 9.7871e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3977e-02, 7.7087e-02, 4.4067e-02, 4.8859e-02, 1.1029e-01,
          7.1533e-02, 2.7008e-02, 8.5144e-03, 5.2299e-03, 9.5444e-03, 1.1993e-02,
          5.7945e-03, 2.4460e-02, 7.2571e-02, 6.2637e-03, 6.1989e-03, 8.4900e-02,
          3.6896e-02, 3.0270e-03, 1.2741e-02, 7.0343e-03, 1.1856e-02, 1.4755e-02,
          1.1887e-02, 6.1798e-03, 2.3529e-02, 2.4673e-02, 3.0441e-02, 4.5410e-02,
          9.8389e-02, 3.3915e-05, 1.0312e-05, 3.3915e-05, 3.3915e-05, 1.0312e-05,
          1.0312e-05, 1.0312e-05, 1.0312e-05, 3.3915e-05, 1.0312e-05, 1.0312e-05,
          1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 3.2008e-05, 3.3915e-05,
          1.0312e-05, 1.0312e-05, 1.0312e-05, 5.7161e-05, 3.3915e-05, 1.0312e-05,
          5.7161e-05, 3.3915e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05,
          1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 6.0499e-05, 1.2207e-04,
          1.0240e-04, 3.2008e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05,
          1.0312e-05, 1.0312e-05, 5.7161e-05, 3.3915e-05, 8.0764e-05, 5.7161e-05,
          5.7161e-05, 1.0312e-05, 5.7578e-05, 3.3915e-05, 5.7578e-05, 1.0312e-05,
          1.0312e-05, 8.0764e-05, 1.0312e-05, 6.3121e-05, 5.5611e-05, 1.2207e-04,
          5.5611e-05, 1.0312e-05, 1.0312e-05, 5.5611e-05, 1.0312e-05, 1.0312e-05,
          1.0312e-05, 5.7161e-05, 1.0312e-05, 1.0312e-05, 5.7161e-05, 5.7161e-05,
          3.3915e-05, 1.8430e-04, 1.0282e-04, 1.0312e-05, 3.3915e-05, 1.0312e-05,
          5.7161e-05, 6.0499e-05, 1.0312e-05, 1.0312e-05, 1.9062e-04, 7.8857e-05,
          8.4043e-05, 6.0499e-05, 1.0312e-05, 3.3915e-05, 1.0312e-05, 1.0312e-05,
          1.0312e-05, 5.7161e-05, 1.0312e-05, 3.3915e-05, 3.3915e-05, 1.0312e-05,
          1.0312e-05, 5.7161e-05, 5.7161e-05, 3.3915e-05, 5.5611e-05, 8.0764e-05,
          1.6665e-04, 7.8857e-05, 5.5611e-05, 1.9324e-04, 8.2195e-05, 6.0499e-05,
          3.2008e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05,
          5.7578e-05, 1.0312e-05, 3.3915e-05, 5.7161e-05, 8.1122e-05, 1.8525e-04,
          8.0764e-05, 1.0240e-04, 7.8857e-05, 5.7161e-05, 5.7161e-05, 1.0729e-04,
          1.1325e-04, 6.0499e-05, 1.3685e-04, 6.0499e-05, 1.9062e-04, 1.0312e-05,
          3.3915e-05, 1.0312e-05, 3.2008e-05, 1.0312e-05, 1.0312e-05, 1.0312e-05,
          1.5533e-04, 1.4305e-04, 5.5611e-05, 3.3915e-05, 1.0312e-05, 3.3915e-05,
          1.3554e-04, 9.0301e-05, 1.0312e-05, 1.0312e-05, 1.0729e-04, 1.0312e-05,
          1.3685e-04, 1.4377e-04, 1.4389e-04, 8.6725e-05, 1.0312e-05, 3.2008e-05,
          8.6725e-05, 1.0312e-05, 1.0312e-05, 5.7161e-05, 1.6403e-04, 2.3818e-04,
          3.3915e-05, 5.7161e-05, 3.3915e-05, 9.6202e-05, 1.5533e-04, 8.0764e-05,
          7.8857e-05, 1.0312e-05, 1.0240e-04, 1.9324e-04, 1.9324e-04, 2.2507e-04,
          1.9109e-04, 1.9324e-04, 3.1686e-04, 6.5029e-05, 8.6725e-05, 9.6202e-05,
          3.3915e-05, 1.0312e-05, 5.7161e-05, 1.0282e-04, 5.5611e-05, 8.1182e-05,
          1.4305e-04, 1.0312e-05, 3.3915e-05, 8.6725e-05, 3.3915e-05, 3.3915e-05,
          3.3915e-05, 3.1686e-04, 3.6979e-04, 2.1684e-04, 4.0960e-04, 5.5611e-05,
          1.9324e-04, 2.4080e-04, 4.5991e-04, 3.5167e-04, 1.0312e-05, 3.3915e-05,
          3.2008e-05, 1.4305e-04, 1.6367e-04, 1.8048e-04, 2.1839e-04, 1.6356e-04,
          1.6356e-04, 5.7161e-05, 5.7161e-05, 1.0312e-05, 3.3915e-05, 1.0312e-05,
          3.1686e-04, 1.9324e-04, 1.0729e-04, 2.4080e-04, 1.8573e-04, 4.8351e-04,
          2.7394e-04, 3.1686e-04, 2.6679e-04, 2.7013e-04, 3.3915e-05, 1.0312e-05,
          7.8857e-05, 1.3554e-04, 5.7161e-05, 1.3554e-04, 5.7161e-05, 1.6403e-04,
          8.0764e-05, 3.3915e-05, 3.3915e-05, 3.3915e-05, 1.0312e-05, 4.2629e-04,
          3.0494e-04, 1.0282e-04, 4.3774e-04, 4.0293e-04, 4.0293e-04, 2.7394e-04,
          6.1607e-04, 3.5787e-04, 1.0312e-05, 1.0312e-05, 5.7161e-05, 2.1875e-04,
          1.0246e-04, 9.0301e-05, 5.7161e-05, 5.7161e-05, 1.3554e-04, 1.7428e-04,
          1.1683e-04, 1.3363e-04, 3.2008e-05, 2.8396e-04, 1.0729e-04, 2.1911e-04,
          3.5000e-04, 1.6212e-04, 3.1686e-04, 2.2507e-04, 4.0293e-04, 3.7169e-04,
          3.1686e-04, 2.6965e-04, 8.6725e-05, 1.0312e-05, 1.1194e-04, 2.0218e-04,
          1.8048e-04, 1.1194e-04, 2.1172e-04, 2.0218e-04, 6.5029e-05, 1.6475e-04,
          1.0312e-05, 3.1805e-04, 5.7161e-05, 3.0661e-04, 9.6202e-05, 3.0494e-04,
          3.0494e-04, 3.0756e-04, 6.0499e-05, 1.0312e-05, 3.7169e-04, 3.1686e-04,
          1.4639e-04, 1.2136e-04, 1.0312e-05, 1.1194e-04, 1.1194e-04, 1.0312e-05,
          1.8799e-04, 2.6631e-04, 1.0312e-05, 2.3520e-04, 6.5029e-05, 1.2136e-04,
          2.4962e-04, 1.1981e-04, 2.8348e-04, 5.7161e-05, 5.9080e-04, 1.0312e-05,
          2.4796e-04, 3.0303e-04, 1.4305e-04, 1.6201e-04, 3.5453e-04, 1.7154e-04,
          1.0312e-05, 3.2008e-05, 1.1194e-04, 1.1194e-04, 2.8491e-04, 1.3363e-04,
          1.8048e-04, 3.3915e-05, 2.4462e-04, 2.6631e-04, 1.0312e-05, 2.3329e-04,
          2.3329e-04, 3.1614e-04, 4.9305e-04, 1.0312e-05, 2.1946e-04, 2.1946e-04,
          2.0802e-04, 1.0312e-05, 3.3915e-05, 3.0494e-04, 1.0312e-05, 6.5029e-05,
          9.6202e-05, 1.1194e-04, 1.4305e-04, 3.3915e-05, 2.9445e-04, 9.0301e-05,
          2.3329e-04, 1.0312e-05, 3.5453e-04, 1.9598e-04, 5.7161e-05, 4.1425e-05,
          4.0460e-04, 1.9324e-04, 1.0312e-05, 6.5029e-05, 5.2595e-04, 6.1417e-04,
          1.0312e-05, 5.0259e-04, 1.2469e-04, 3.5262e-04, 3.3092e-04, 1.0312e-05,
          2.0790e-04, 1.8430e-04, 1.8430e-04, 1.8430e-04, 2.8658e-04, 1.7691e-04,
          1.0312e-05, 1.7691e-04, 1.2010e-04, 5.2404e-04, 1.2010e-04, 1.2010e-04,
          1.8430e-04, 1.8430e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.0551e-02, 8.2092e-02, 3.1738e-02, 4.2480e-02, 1.1029e-01,
          9.3933e-02, 2.2614e-02, 7.2327e-02, 2.5803e-02, 2.0813e-02, 1.9928e-02,
          1.0155e-02, 1.5717e-02, 7.5874e-03, 1.7334e-02, 5.4840e-02, 1.6174e-02,
          9.6207e-03, 1.5518e-02, 2.0580e-03, 2.3918e-03, 2.0332e-03, 1.1703e-02,
          1.9333e-02, 1.1620e-02, 2.0447e-02, 2.4063e-02, 6.4545e-03, 1.8204e-02,
          9.4223e-03, 1.2842e-01, 7.5877e-05, 4.4763e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 9.9599e-05, 2.0742e-05, 7.5877e-05, 4.4763e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 4.4763e-05,
          1.3089e-04, 4.4763e-05, 2.0742e-05, 1.3208e-04, 1.3208e-04, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 6.9618e-05, 4.4763e-05, 6.9618e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.9897e-05, 1.5569e-04, 2.2507e-04,
          9.9897e-05, 4.4763e-05, 3.2902e-04, 4.4763e-05, 1.3208e-04, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 4.5657e-05,
          4.4763e-05, 9.9897e-05, 3.5691e-04, 2.0742e-05, 4.4763e-05, 4.4763e-05,
          2.0742e-05, 1.3208e-04, 2.0742e-05, 1.3208e-04, 2.5010e-04, 6.9618e-05,
          4.4763e-05, 1.3208e-04, 1.3208e-04, 4.4763e-05, 1.3208e-04, 2.0742e-05,
          4.4763e-05, 4.5657e-05, 4.5657e-05, 2.0742e-05, 4.4763e-05, 6.9618e-05,
          4.5657e-05, 6.9618e-05, 6.9618e-05, 4.4763e-05, 2.0742e-05, 2.0742e-05,
          1.2469e-04, 6.9618e-05, 2.0742e-05, 4.5657e-05, 1.2469e-04, 1.3208e-04,
          2.0742e-05, 1.6391e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 3.0184e-04,
          6.9618e-05, 2.0742e-05, 6.9618e-05, 4.5657e-05, 1.5569e-04, 4.4763e-05,
          6.9618e-05, 4.4763e-05, 1.0073e-04, 2.0742e-05, 2.0742e-05, 1.5688e-04,
          1.2469e-04, 1.5688e-04, 4.4763e-05, 4.4012e-04, 1.6391e-04, 1.6391e-04,
          4.4763e-05, 2.0742e-05, 9.9897e-05, 2.0742e-05, 2.0742e-05, 4.5657e-05,
          5.3167e-04, 3.3283e-04, 4.4763e-05, 9.9897e-05, 6.9618e-05, 4.4763e-05,
          2.0742e-05, 4.5657e-05, 2.0742e-05, 1.5688e-04, 1.0073e-04, 2.0742e-05,
          2.1195e-04, 9.9897e-05, 2.5487e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          9.9897e-05, 6.9618e-05, 2.0742e-05, 4.5657e-05, 7.0632e-05, 2.5725e-04,
          2.0742e-05, 6.7902e-04, 6.9618e-05, 2.5487e-04, 2.9945e-04, 4.4763e-05,
          2.0742e-05, 6.9618e-05, 1.2469e-04, 2.3603e-04, 1.2469e-04, 1.0073e-04,
          3.6120e-04, 2.1195e-04, 1.5688e-04, 2.0742e-05, 1.2469e-04, 1.2469e-04,
          2.0742e-05, 2.0742e-05, 4.5657e-05, 3.5787e-04, 4.4763e-05, 1.0929e-03,
          2.5558e-04, 6.9618e-05, 2.0742e-05, 9.9897e-05, 4.4763e-05, 6.9618e-05,
          6.9618e-05, 1.5688e-04, 2.1195e-04, 2.7966e-04, 7.3528e-04, 9.9897e-05,
          2.1088e-04, 4.1628e-04, 2.5129e-04, 4.9353e-04, 2.0742e-05, 1.3900e-04,
          2.0742e-05, 2.9945e-04, 7.3147e-04, 2.0742e-05, 4.5657e-05, 7.3147e-04,
          6.9618e-05, 4.5657e-05, 4.5657e-05, 4.5657e-05, 6.9618e-05, 6.9618e-05,
          1.5688e-04, 2.7966e-04, 4.5657e-05, 1.6391e-04, 4.5657e-05, 3.5000e-04,
          4.9543e-04, 1.5688e-04, 1.5688e-04, 1.5688e-04, 2.0742e-05, 5.0783e-04,
          4.5657e-05, 4.5681e-04, 4.5657e-05, 1.0073e-04, 4.5657e-05, 2.2697e-04,
          1.0073e-04, 2.0742e-05, 2.0742e-05, 2.0206e-04, 5.3167e-04, 3.6693e-04,
          7.9203e-04, 1.5688e-04, 6.0320e-04, 1.5688e-04, 3.6693e-04, 1.6391e-04,
          2.7514e-04, 2.7466e-04, 3.2187e-04, 6.2609e-04, 4.5657e-05, 4.5896e-04,
          1.2469e-04, 4.5657e-05, 6.7902e-04, 2.0742e-05, 1.2469e-04, 1.0073e-04,
          7.0667e-04, 1.0073e-04, 1.0073e-04, 2.4283e-04, 2.4295e-04, 6.9618e-05,
          1.5688e-04, 1.5688e-04, 1.5688e-04, 2.1899e-04, 2.1195e-04, 1.5688e-04,
          1.5688e-04, 2.0742e-05, 2.0742e-05, 6.9618e-05, 6.7902e-04, 4.5443e-04,
          1.2469e-04, 1.0900e-03, 1.0073e-04, 2.0742e-05, 2.0742e-05, 2.7466e-04,
          2.4438e-04, 6.9618e-05, 1.5688e-04, 9.9897e-05, 4.5657e-05, 5.1689e-04,
          2.9945e-04, 2.9945e-04, 4.5657e-05, 6.9618e-05, 1.5688e-04, 1.5688e-04,
          2.7537e-04, 1.6391e-04, 4.5657e-05, 6.7902e-04, 4.5657e-05, 5.3167e-04,
          2.7537e-04, 1.0633e-03, 5.3263e-04, 1.2469e-04, 2.0742e-05, 4.5657e-05,
          4.5443e-04, 2.0742e-05, 2.0742e-05, 1.5688e-04, 6.8474e-04, 7.1192e-04,
          4.4763e-05, 2.4438e-04, 1.2469e-04, 1.6391e-04, 4.5657e-05, 2.5129e-04,
          1.5688e-04, 6.8665e-04, 4.5657e-05, 4.5657e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 6.9952e-04, 4.9877e-04, 5.3167e-04, 5.7364e-04,
          2.0742e-05, 2.0742e-05, 4.5657e-05, 2.4438e-04, 2.0742e-05, 6.9618e-05,
          5.3406e-04, 7.2765e-04, 2.0742e-05, 1.0214e-03, 4.5657e-05, 2.0742e-05,
          4.3035e-04, 2.0301e-04, 4.5657e-05, 2.0742e-05, 2.0742e-05, 1.5688e-04,
          2.0742e-05, 7.8297e-04, 8.2970e-04, 4.5657e-05, 4.5657e-05, 7.8583e-04,
          2.0742e-05, 3.9792e-04, 6.2609e-04, 2.0742e-05, 5.5361e-04, 5.3406e-04,
          7.9775e-04, 2.0742e-05, 2.9993e-04, 1.5688e-04, 4.5466e-04, 2.0742e-05,
          2.9659e-04, 2.7251e-04, 2.9659e-04, 7.5877e-05, 2.7251e-04, 4.4763e-05,
          2.0742e-05, 6.2847e-04, 2.7251e-04, 7.1335e-04, 6.5851e-04, 8.8787e-04,
          2.0742e-05, 2.0742e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.7792e-02, 7.8247e-02, 4.4312e-02, 4.4495e-02, 1.0980e-01,
          6.8665e-02, 2.5909e-02, 7.0419e-03, 5.0735e-03, 1.1566e-02, 1.1101e-02,
          5.9814e-03, 2.1851e-02, 3.1982e-02, 7.7286e-03, 5.8784e-03, 8.0750e-02,
          3.2990e-02, 2.7103e-03, 2.3148e-02, 1.3847e-02, 1.9852e-02, 1.3596e-02,
          1.1375e-02, 6.3438e-03, 2.1393e-02, 2.2446e-02, 3.0487e-02, 7.0923e-02,
          2.8198e-02, 1.2749e-02, 5.4993e-02, 3.6657e-05, 1.4007e-05, 2.7180e-05,
          1.4007e-05, 1.4007e-05, 2.7180e-05, 2.7180e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 3.6657e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.0962e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 2.7180e-05, 1.4007e-05, 4.9829e-05,
          2.1935e-04, 6.1691e-05, 2.7180e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 3.6657e-05, 3.6657e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 5.0247e-05, 1.4007e-05, 2.7597e-05, 1.4007e-05,
          1.4007e-05, 4.9829e-05, 1.4007e-05, 1.1200e-04, 1.1963e-04, 1.5855e-04,
          8.4341e-05, 1.4007e-05, 1.0645e-04, 4.9829e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 2.7180e-05, 1.4007e-05, 1.4007e-05, 2.7180e-05,
          4.9829e-05, 3.7098e-04, 8.4341e-05, 1.4007e-05, 3.6657e-05, 1.4007e-05,
          2.7180e-05, 1.4007e-05, 2.7180e-05, 1.4007e-05, 4.5586e-04, 9.0003e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 4.9829e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 2.7180e-05, 1.4007e-05, 4.9829e-05, 4.9829e-05, 2.7180e-05,
          1.4007e-05, 6.4552e-05, 6.4552e-05, 4.9829e-05, 1.1265e-04, 1.4007e-05,
          2.4140e-04, 4.5586e-04, 8.4341e-05, 1.8179e-04, 2.7180e-05, 1.9825e-04,
          1.4007e-05, 2.7180e-05, 2.7180e-05, 2.7180e-05, 1.4007e-05, 3.6657e-05,
          3.2926e-04, 7.7426e-05, 4.9829e-05, 7.2145e-04, 5.0247e-05, 4.9829e-05,
          2.7180e-05, 4.9829e-05, 1.4007e-05, 4.9829e-05, 1.4007e-05, 2.1148e-04,
          6.1691e-05, 2.1148e-04, 1.3459e-04, 2.9063e-04, 4.9829e-05, 3.6657e-05,
          2.7180e-05, 2.7180e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 7.7426e-05,
          1.1265e-04, 8.4305e-04, 4.9829e-05, 4.9829e-05, 1.4007e-05, 4.9829e-05,
          8.6784e-05, 7.3612e-05, 1.4007e-05, 1.4007e-05, 2.7180e-05, 1.4007e-05,
          7.7426e-05, 4.9829e-05, 4.9829e-05, 1.3459e-04, 1.4007e-05, 2.7180e-05,
          1.0633e-04, 1.4007e-05, 1.4007e-05, 9.2447e-05, 8.1956e-05, 4.9829e-05,
          8.4341e-05, 7.8678e-04, 2.7180e-05, 1.4007e-05, 1.1265e-04, 4.9829e-05,
          6.4135e-05, 2.7180e-05, 4.9829e-05, 1.4889e-04, 1.3459e-04, 1.0008e-04,
          1.9550e-04, 2.7180e-05, 1.9825e-04, 1.3459e-04, 4.9829e-05, 1.4007e-05,
          1.3459e-04, 1.4007e-05, 5.5492e-05, 1.1313e-04, 4.9829e-05, 8.4162e-04,
          5.5492e-05, 1.4007e-05, 1.4007e-05, 1.3459e-04, 4.9829e-05, 4.9829e-05,
          3.6657e-05, 3.9601e-04, 3.9601e-04, 3.8886e-04, 1.5855e-04, 7.8082e-05,
          2.1875e-04, 6.2227e-04, 4.6587e-04, 4.2844e-04, 2.7180e-05, 3.6657e-05,
          2.7180e-05, 8.4305e-04, 4.3535e-04, 7.8082e-05, 1.1504e-04, 2.3162e-04,
          4.0698e-04, 5.0962e-05, 2.7180e-05, 2.7180e-05, 3.6657e-05, 4.9829e-05,
          3.9601e-04, 3.6621e-04, 5.2071e-04, 7.5340e-04, 4.2224e-04, 2.8205e-04,
          5.3453e-04, 3.8290e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          6.4135e-05, 1.0605e-03, 7.5865e-04, 1.1504e-04, 7.4530e-04, 2.4390e-04,
          4.9829e-05, 1.4007e-05, 4.9829e-05, 3.6657e-05, 4.9829e-05, 4.3201e-04,
          9.0647e-04, 1.1551e-04, 1.8680e-04, 3.9601e-04, 7.7426e-05, 7.8082e-05,
          5.9700e-04, 3.7527e-04, 1.4007e-05, 4.9829e-05, 1.4007e-05, 4.4012e-04,
          1.4997e-04, 8.1873e-04, 9.2447e-05, 5.0962e-05, 4.9829e-05, 2.7180e-05,
          2.4486e-04, 2.7180e-05, 1.4007e-05, 1.4007e-05, 3.7336e-04, 3.2043e-04,
          3.9601e-04, 4.2439e-04, 3.9601e-04, 7.1824e-05, 4.2439e-04, 4.6587e-04,
          3.8290e-04, 1.0008e-04, 1.3459e-04, 1.4007e-05, 9.2864e-05, 4.9829e-05,
          4.9829e-05, 2.7180e-05, 6.4135e-05, 8.6784e-05, 7.7426e-05, 1.0633e-04,
          3.6657e-05, 1.4675e-04, 5.0962e-05, 1.6880e-04, 2.7180e-05, 1.3459e-04,
          5.9080e-04, 1.3459e-04, 1.4007e-05, 2.7180e-05, 3.9601e-04, 6.4135e-05,
          2.7180e-05, 3.6657e-05, 1.4007e-05, 6.4135e-05, 2.7180e-05, 4.9829e-05,
          2.7180e-05, 6.8367e-05, 5.0247e-05, 4.9829e-05, 7.7426e-05, 9.2447e-05,
          7.2813e-04, 4.9829e-05, 1.3697e-04, 5.0962e-05, 3.3331e-04, 4.9829e-05,
          6.1750e-04, 7.1824e-05, 1.4007e-05, 3.5572e-04, 3.3474e-04, 1.4007e-05,
          6.4135e-05, 8.1539e-05, 9.2447e-05, 9.5904e-05, 3.4523e-04, 8.4341e-05,
          5.2071e-04, 8.3685e-05, 1.7154e-04, 3.4523e-04, 2.8610e-04, 6.1369e-04,
          1.1501e-03, 1.0767e-03, 5.1880e-04, 4.9829e-05, 1.3459e-04, 2.7180e-05,
          1.3459e-04, 8.6784e-05, 4.9829e-05, 3.3474e-04, 1.4007e-05, 1.1200e-04,
          2.7180e-05, 1.2422e-04, 9.2447e-05, 2.7180e-05, 1.1501e-03, 5.0962e-05,
          7.4482e-04, 2.2364e-04, 3.0851e-04, 1.4007e-05, 6.4135e-05, 3.0851e-04,
          1.2026e-03, 2.4843e-04, 6.0558e-04, 1.1200e-04, 5.2977e-04, 3.1900e-04,
          2.5988e-04, 1.4381e-03, 2.6178e-04, 1.9825e-04, 2.9874e-04, 1.4007e-05,
          9.0003e-05, 1.1200e-04, 6.1691e-05, 6.1691e-05, 6.1691e-05, 3.6657e-05,
          2.7180e-05, 1.1265e-04, 9.0003e-05, 4.1771e-04, 1.4007e-05, 2.7180e-05,
          2.7180e-05, 6.1691e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3184e-02, 8.1787e-02, 3.5614e-02, 4.1687e-02, 1.2585e-01,
          1.0101e-01, 2.3239e-02, 1.4763e-02, 1.0963e-02, 3.5492e-02, 2.6199e-02,
          1.5129e-02, 2.0294e-02, 7.1144e-03, 3.5248e-02, 1.3336e-02, 3.0350e-02,
          1.3374e-02, 5.8784e-03, 3.5210e-03, 4.0054e-03, 2.7714e-03, 1.3977e-02,
          2.8549e-02, 1.8875e-02, 1.9806e-02, 2.5467e-02, 7.3357e-03, 3.1250e-02,
          8.7738e-03, 2.3651e-02, 1.1642e-02, 9.9426e-02, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05,
          9.6798e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.9681e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 6.2168e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 6.2168e-05,
          1.4031e-04, 1.8716e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          6.2168e-05, 6.2168e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.9681e-04, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          6.2168e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05,
          2.7490e-04, 6.2168e-05, 1.8716e-05, 9.6798e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 5.0783e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.4031e-04,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 6.2168e-05, 2.8658e-04, 1.8716e-05, 1.4031e-04,
          1.8716e-05, 1.8716e-05, 1.4031e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.4031e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.4031e-04, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05,
          1.4031e-04, 3.8624e-04, 1.8716e-05, 1.8716e-05, 6.4552e-05, 1.8716e-05,
          1.4031e-04, 1.8716e-05, 1.8716e-05, 1.4031e-04, 1.4031e-04, 1.8716e-05,
          1.8716e-05, 1.1832e-04, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05,
          2.9635e-04, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.4031e-04, 1.4031e-04,
          6.2168e-05, 1.8716e-05, 1.8716e-05, 6.4552e-05, 1.8716e-05, 1.1832e-04,
          1.8716e-05, 1.4031e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 1.8716e-05, 1.8716e-05,
          1.9681e-04, 2.9635e-04, 2.8658e-04, 1.5342e-04, 1.8716e-05, 1.6177e-04,
          1.8716e-05, 1.8716e-05, 2.5201e-04, 1.8716e-05, 1.8716e-05, 2.5201e-04,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05, 2.3985e-04,
          1.8716e-05, 6.2168e-05, 1.1832e-04, 2.9635e-04, 2.9635e-04, 1.8716e-05,
          2.9635e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.9646e-04,
          1.8716e-05, 3.5167e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 6.2168e-05, 6.2168e-05, 2.7490e-04, 1.8716e-05,
          2.9635e-04, 6.4433e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.5342e-04, 1.8716e-05, 3.8624e-04, 3.7456e-04, 1.8716e-05, 3.5620e-04,
          3.5167e-04, 1.8716e-05, 1.1832e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          5.1594e-04, 1.8716e-05, 9.6798e-05, 3.1400e-04, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.1832e-04, 1.1832e-04, 1.8716e-05, 1.8716e-05, 1.1832e-04,
          1.8716e-05, 1.8716e-05, 3.1400e-04, 1.8716e-05, 1.1832e-04, 6.2168e-05,
          1.8716e-05, 1.1832e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 6.2168e-05,
          1.1832e-04, 1.1832e-04, 1.1832e-04, 6.2168e-05, 1.8716e-05, 1.8716e-05,
          2.9635e-04, 1.4031e-04, 1.6832e-03, 3.6478e-04, 1.8716e-05, 1.8716e-05,
          3.6478e-04, 5.6410e-04, 1.4031e-04, 1.1832e-04, 1.8716e-05, 2.3985e-04,
          1.8716e-05, 1.6177e-04, 2.4223e-04, 1.8716e-05, 1.8716e-05, 1.1832e-04,
          4.2558e-04, 1.8716e-05, 4.2558e-04, 1.1832e-04, 1.1832e-04, 3.7456e-04,
          1.8716e-05, 1.8716e-05, 3.7456e-04, 3.1400e-04, 4.2558e-04, 4.6420e-04,
          9.8705e-04, 4.2963e-04, 1.1832e-04, 1.1832e-04, 6.9380e-04, 6.9380e-04,
          4.2558e-04, 2.7037e-04, 1.8716e-05, 1.9240e-04, 4.2963e-04, 3.8218e-04,
          1.7996e-03, 7.3814e-04, 3.8218e-04, 6.2168e-05, 3.1400e-04, 1.8716e-05,
          6.2168e-05, 9.8705e-04, 6.2168e-05, 5.7793e-04, 1.8716e-05, 1.8716e-05,
          4.2963e-04, 1.1832e-04, 1.1832e-04, 6.2168e-05, 1.5450e-03, 1.1832e-04,
          8.6737e-04, 5.6458e-04, 3.1400e-04, 1.8716e-05, 1.1832e-04, 9.8228e-04,
          8.6737e-04, 7.3099e-04, 1.3914e-03, 2.3580e-04, 1.8716e-05, 1.9240e-04,
          9.8228e-04, 2.2449e-03, 1.9240e-04, 3.8624e-04, 3.8624e-04, 1.8716e-05,
          9.6798e-05, 1.8716e-05, 5.4264e-04, 1.8716e-05, 9.6798e-05, 1.8716e-05,
          1.8716e-05, 1.4031e-04, 9.6798e-05, 1.8716e-05, 1.8716e-05, 8.8739e-04,
          9.6798e-05, 1.8716e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1459e-02, 7.7271e-02, 3.0594e-02, 4.4922e-02, 9.2285e-02,
          1.0834e-01, 2.3041e-02, 1.4000e-02, 9.9182e-03, 4.4189e-02, 3.6865e-02,
          1.9073e-02, 1.9211e-02, 5.3024e-03, 3.4637e-02, 1.6525e-02, 2.0508e-02,
          1.5152e-02, 7.8583e-03, 1.5650e-03, 1.9932e-03, 1.9932e-03, 2.0294e-02,
          4.4586e-02, 1.5335e-02, 1.9821e-02, 2.7954e-02, 3.3169e-03, 2.5543e-02,
          7.1907e-03, 2.5909e-02, 6.0539e-03, 3.8361e-02, 7.9224e-02, 1.4544e-05,
          1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05,
          1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05,
          4.1604e-05, 1.1003e-04, 1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05,
          1.4544e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05,
          1.4544e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 4.1604e-05,
          4.1604e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.1003e-04, 1.4544e-05,
          1.4544e-05, 1.4544e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05, 1.4544e-05,
          1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05,
          1.4544e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05,
          1.4544e-05, 1.4544e-05, 1.3709e-04, 1.4544e-05, 1.1003e-04, 1.1003e-04,
          1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05,
          4.1604e-05, 4.1604e-05, 4.1604e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05,
          1.4544e-05, 1.4544e-05, 1.4544e-05, 1.1003e-04, 4.1604e-05, 4.1604e-05,
          4.1604e-05, 1.4544e-05, 2.8086e-04, 4.1604e-05, 1.1003e-04, 9.3639e-05,
          4.1604e-05, 1.4544e-05, 1.4544e-05, 1.6880e-04, 4.1604e-05, 4.1604e-05,
          1.4544e-05, 1.4544e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05, 1.4544e-05,
          2.2709e-04, 4.1604e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.1003e-04,
          4.1604e-05, 4.4131e-04, 1.4544e-05, 1.4544e-05, 1.4544e-05, 2.2709e-04,
          1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05,
          1.4544e-05, 4.1604e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05,
          1.4544e-05, 1.1003e-04, 4.1604e-05, 1.8907e-04, 4.1604e-05, 3.0804e-04,
          1.4544e-05, 1.4544e-05, 1.7500e-04, 1.4544e-05, 1.4544e-05, 1.4544e-05,
          4.1604e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05, 1.7500e-04, 4.1604e-05,
          1.4544e-05, 6.6578e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4544e-05,
          1.4544e-05, 4.1604e-05, 4.1604e-05, 4.1604e-05, 1.4544e-05, 1.4544e-05,
          1.4544e-05, 3.0804e-04, 1.4544e-05, 1.4544e-05, 4.1604e-05, 4.1604e-05,
          4.1604e-05, 1.4544e-05, 4.1604e-05, 1.7500e-04, 4.1604e-05, 4.1604e-05,
          1.4544e-05, 3.2258e-04, 1.4544e-05, 1.4544e-05, 4.1604e-05, 1.4544e-05,
          1.8907e-04, 1.4544e-05, 1.1003e-04, 4.1604e-05, 1.4544e-05, 1.7500e-04,
          1.6880e-04, 1.4544e-05, 6.6578e-05, 4.1604e-05, 4.1604e-05, 9.3639e-05,
          1.4544e-05, 2.2709e-04, 1.4544e-05, 4.1604e-05, 4.1604e-05, 3.4952e-04,
          4.1604e-05, 1.6212e-04, 1.4544e-05, 9.3639e-05, 1.6046e-04, 4.1604e-05,
          4.1604e-05, 3.0804e-04, 1.6212e-04, 4.3511e-04, 1.4544e-05, 3.0804e-04,
          1.4544e-05, 6.6578e-05, 2.1625e-04, 1.6880e-04, 4.1604e-05, 2.1625e-04,
          2.1625e-04, 6.6578e-05, 6.6578e-05, 1.4544e-05, 3.4952e-04, 3.2258e-04,
          1.6212e-04, 9.3639e-05, 2.8181e-04, 1.8907e-04, 9.3639e-05, 1.4544e-05,
          3.0804e-04, 1.1003e-04, 1.6212e-04, 1.4544e-05, 4.1604e-05, 6.6578e-05,
          1.4544e-05, 2.1625e-04, 1.4544e-05, 1.4544e-05, 6.6578e-05, 2.1625e-04,
          4.1604e-05, 1.4544e-05, 4.1604e-05, 4.6849e-04, 1.7500e-04, 1.4544e-05,
          5.6124e-04, 4.1604e-05, 1.6046e-04, 1.6212e-04, 1.4544e-05, 4.1604e-05,
          4.0817e-04, 1.6880e-04, 3.8505e-04, 4.4131e-04, 6.6578e-05, 2.1625e-04,
          9.3639e-05, 6.6578e-05, 6.6578e-05, 6.6578e-05, 2.1625e-04, 9.3639e-05,
          2.9135e-04, 1.4544e-05, 1.7500e-04, 1.7500e-04, 2.8181e-04, 1.8632e-04,
          1.6212e-04, 1.6212e-04, 1.6212e-04, 4.1604e-05, 1.1003e-04, 1.6212e-04,
          1.6212e-04, 4.1604e-05, 3.0208e-04, 2.1625e-04, 6.6578e-05, 4.1604e-05,
          4.1604e-05, 9.4366e-04, 6.6578e-05, 2.1625e-04, 1.4544e-05, 4.1604e-05,
          4.4513e-04, 9.3639e-05, 1.6212e-04, 4.1604e-05, 1.4544e-05, 4.1604e-05,
          3.3975e-04, 1.4544e-05, 3.2258e-04, 4.4131e-04, 1.6212e-04, 1.6212e-04,
          4.4131e-04, 3.0804e-04, 6.6578e-05, 6.6578e-05, 6.6578e-05, 3.4952e-04,
          1.4544e-05, 5.5790e-04, 4.6945e-04, 4.1604e-05, 1.4544e-05, 7.5674e-04,
          4.7684e-04, 1.4544e-05, 3.1614e-04, 5.6124e-04, 1.4544e-05, 5.6124e-04,
          1.4544e-05, 1.4544e-05, 4.4131e-04, 4.1604e-05, 2.2078e-04, 4.4131e-04,
          4.4131e-04, 9.3639e-05, 6.6578e-05, 1.8919e-04, 4.7684e-04, 3.4332e-04,
          3.4332e-04, 1.4544e-05, 1.4544e-05, 1.4544e-05, 5.6505e-04, 2.2078e-04,
          4.7684e-04, 4.7684e-04, 2.2078e-04, 4.1604e-05, 1.6880e-04, 1.4544e-05,
          4.1604e-05, 6.3801e-04, 4.1604e-05, 4.4966e-04, 1.4544e-05, 1.1003e-04,
          9.3639e-05, 3.0899e-04, 6.6578e-05, 1.4544e-05, 4.7684e-04, 1.6212e-04,
          9.6226e-04, 1.2054e-03, 4.1604e-05, 1.4544e-05, 3.1161e-04, 1.6270e-03,
          4.7684e-04, 4.0078e-04, 1.4200e-03, 1.4544e-05, 1.4544e-05, 4.2272e-04,
          1.0595e-03, 9.3508e-04, 1.1003e-04, 1.6212e-04, 1.6212e-04, 1.4544e-05,
          7.8678e-04, 1.4544e-05, 1.4544e-05, 1.4544e-05, 1.4794e-04, 1.4544e-05,
          1.4544e-05, 1.7500e-04, 1.4544e-05, 6.2943e-04, 1.4544e-05, 1.4794e-04,
          1.4544e-05, 1.4544e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3710e-02, 7.2449e-02, 3.2745e-02, 3.1952e-02, 1.0760e-01,
          7.3792e-02, 1.9165e-02, 4.8859e-02, 1.9714e-02, 1.5854e-02, 1.2932e-02,
          8.1863e-03, 1.4305e-02, 8.1482e-03, 1.5671e-02, 3.8147e-02, 2.4094e-02,
          1.0216e-02, 9.2621e-03, 4.3640e-03, 4.4556e-03, 3.1624e-03, 1.0391e-02,
          1.3741e-02, 1.3603e-02, 1.6479e-02, 2.0096e-02, 1.0468e-02, 2.1149e-02,
          9.4681e-03, 8.8989e-02, 1.1101e-02, 2.6947e-02, 8.1558e-03, 1.1737e-01,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 7.4029e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          8.3506e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 7.4029e-05, 8.3506e-05, 8.3506e-05,
          8.3506e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          8.3506e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          8.3506e-05, 8.6427e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 3.3069e-04,
          7.4029e-05, 7.4029e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.4389e-04,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          8.3506e-05, 3.3069e-04, 1.4389e-04, 3.3069e-04, 1.3590e-05, 1.4389e-04,
          1.3590e-05, 1.3590e-05, 2.0182e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.4389e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.4389e-04, 1.3590e-05,
          7.4029e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.4389e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.9824e-05, 8.3506e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 3.8528e-04, 1.4389e-04, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.9681e-04, 8.3506e-05, 1.3590e-05,
          8.3506e-05, 1.9681e-04, 3.3069e-04, 1.3590e-05, 1.4389e-04, 2.0182e-04,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 8.9824e-05, 1.3590e-05, 2.7299e-04,
          1.3590e-05, 7.4029e-05, 1.3590e-05, 7.4029e-05, 1.3590e-05, 1.4389e-04,
          1.3590e-05, 1.3590e-05, 5.1403e-04, 2.6679e-04, 1.4389e-04, 1.3590e-05,
          8.6427e-05, 8.3506e-05, 3.3069e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05, 1.3590e-05, 8.3506e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.3506e-05, 3.2711e-04,
          5.1403e-04, 7.4029e-05, 1.3590e-05, 7.4029e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 3.3069e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 2.5725e-04,
          1.3590e-05, 8.3506e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.4389e-04, 1.4389e-04, 1.3590e-05,
          8.3506e-05, 8.6427e-05, 7.4029e-05, 3.3069e-04, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 7.4029e-05, 3.3069e-04, 1.4389e-04, 1.3590e-05, 8.9824e-05,
          8.6427e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 7.4029e-05, 7.4029e-05,
          1.4389e-04, 1.3590e-05, 7.4029e-05, 6.7997e-04, 1.3590e-05, 1.3590e-05,
          3.3069e-04, 1.9681e-04, 5.1403e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 7.4029e-05, 1.3590e-05, 8.3506e-05,
          1.3590e-05, 2.0182e-04, 7.4029e-05, 1.3590e-05, 1.3590e-05, 7.4029e-05,
          7.4029e-05, 1.3590e-05, 3.8505e-04, 1.3590e-05, 1.3590e-05, 7.4029e-05,
          7.4029e-05, 1.4389e-04, 3.8505e-04, 7.4029e-05, 1.3590e-05, 1.3590e-05,
          2.0182e-04, 7.4029e-05, 3.8505e-04, 1.9681e-04, 1.3590e-05, 4.6110e-04,
          1.3590e-05, 7.4029e-05, 4.6110e-04, 8.3506e-05, 1.3590e-05, 3.8505e-04,
          2.0182e-04, 1.3590e-05, 1.3590e-05, 3.8505e-04, 3.8505e-04, 4.5490e-04,
          1.3590e-05, 1.3590e-05, 2.0182e-04, 3.2711e-04, 1.3590e-05, 3.8505e-04,
          7.0190e-04, 4.5490e-04, 1.9681e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 3.8528e-04, 3.2711e-04, 3.2496e-04, 1.1816e-03, 7.4029e-05,
          7.4029e-05, 7.4029e-05, 1.3590e-05, 8.3506e-05, 1.3590e-05, 1.3590e-05,
          3.2711e-04, 4.5490e-04, 1.3590e-05, 3.8528e-04, 1.3590e-05, 1.3590e-05,
          3.8505e-04, 1.9681e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 3.2473e-04,
          7.4029e-05, 1.1749e-03, 1.3590e-03, 7.4029e-05, 3.8505e-04, 1.6060e-03,
          7.4029e-05, 7.0190e-04, 1.7719e-03, 1.3590e-05, 4.4060e-04, 7.0238e-04,
          1.6756e-03, 7.4029e-05, 5.1403e-04, 3.8505e-04, 8.8549e-04, 1.3590e-05,
          7.4029e-05, 3.6836e-04, 3.6836e-04, 1.3590e-05, 4.9591e-04, 1.3590e-05,
          1.3590e-05, 3.6836e-04, 7.4029e-05, 1.3590e-05, 1.3590e-05, 8.7404e-04,
          7.4029e-05, 7.4029e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.6907e-02, 8.2214e-02, 3.3630e-02, 3.7109e-02, 1.4893e-01,
          9.3872e-02, 2.4323e-02, 1.0384e-02, 1.0567e-02, 2.7008e-02, 2.5513e-02,
          1.3779e-02, 1.0597e-02, 8.0338e-03, 1.8555e-02, 6.2828e-03, 1.3901e-02,
          5.9853e-03, 3.4409e-03, 7.2937e-03, 6.3171e-03, 3.5839e-03, 6.0654e-03,
          2.4323e-02, 1.6739e-02, 1.8982e-02, 1.6876e-02, 1.1162e-02, 1.1719e-02,
          8.0338e-03, 1.2581e-02, 7.3891e-03, 3.4851e-02, 1.1986e-02, 1.8951e-02,
          1.4355e-01, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5340e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          7.7248e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 7.7248e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 7.7248e-05, 1.7107e-05, 7.5340e-05, 7.5340e-05, 1.7107e-05,
          7.5340e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 7.5340e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          7.7248e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.3542e-04, 7.7248e-05,
          2.3985e-04, 7.5340e-05, 1.7107e-05, 1.7977e-04, 7.5340e-05, 1.7107e-05,
          1.7107e-05, 4.8470e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.3542e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 1.7107e-05, 1.7107e-05,
          7.5340e-05, 1.7107e-05, 7.5340e-05, 1.7107e-05, 1.7107e-05, 2.2352e-04,
          1.7107e-05, 1.7107e-05, 4.3654e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          7.5340e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.3542e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.3542e-04, 1.7107e-05, 1.7107e-05,
          7.5340e-05, 3.3641e-04, 1.7107e-05, 1.7107e-05, 7.8559e-05, 1.7107e-05,
          1.3542e-04, 1.7107e-05, 1.7107e-05, 3.5739e-04, 7.5340e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.8418e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 7.5340e-05, 2.3985e-04,
          7.7248e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.3542e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.3542e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.0645e-04, 7.7248e-05, 3.3641e-04, 1.7107e-05, 1.7107e-05, 9.2649e-04,
          1.7107e-05, 1.7107e-05, 7.7248e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05,
          1.8954e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.7248e-05, 3.4213e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          3.1042e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.2244e-04,
          1.7107e-05, 7.7248e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.0266e-05,
          1.7107e-05, 1.7107e-05, 7.7248e-05, 6.3276e-04, 1.3542e-04, 1.7107e-05,
          3.4213e-04, 7.8559e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 6.8808e-04, 3.7885e-04, 1.7107e-05, 2.8586e-04,
          2.8396e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.2244e-04, 1.7107e-05,
          7.7248e-05, 1.7107e-05, 1.7977e-04, 5.5408e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 5.5408e-04, 2.2244e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 3.8505e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          2.8062e-04, 1.7107e-05, 3.8505e-04, 7.7248e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 4.9400e-04, 8.2207e-04, 3.8505e-04, 1.7107e-05, 1.7107e-05,
          2.9731e-04, 1.2455e-03, 5.6267e-04, 1.7107e-05, 1.7107e-05, 4.4847e-04,
          1.7107e-05, 8.4162e-04, 4.4847e-04, 1.7107e-05, 1.7107e-05, 2.2244e-04,
          2.8253e-04, 1.7107e-05, 2.8253e-04, 3.8505e-04, 7.5340e-05, 6.8855e-04,
          1.3471e-04, 1.7107e-05, 1.2074e-03, 2.9731e-04, 1.0538e-03, 7.0429e-04,
          8.5258e-04, 3.4070e-04, 2.2244e-04, 1.7107e-05, 1.3542e-04, 1.3542e-04,
          7.7248e-05, 1.9288e-04, 1.7107e-05, 1.3471e-04, 4.4847e-04, 1.7107e-05,
          1.3151e-03, 1.1110e-03, 2.2244e-04, 1.7107e-05, 5.5408e-04, 1.7107e-05,
          1.7107e-05, 1.0080e-03, 7.7248e-05, 1.9288e-04, 1.7107e-05, 1.7107e-05,
          4.4513e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 8.9645e-04, 3.2687e-04,
          7.6056e-04, 4.4656e-04, 3.8934e-04, 1.3471e-04, 2.2244e-04, 8.2207e-04,
          7.5960e-04, 3.8505e-04, 1.1253e-03, 1.3471e-04, 1.7107e-05, 1.3471e-04,
          8.2207e-04, 7.5960e-04, 1.3471e-04, 4.4465e-04, 7.6389e-04, 1.7107e-05,
          7.5340e-05, 1.7107e-05, 2.7180e-04, 7.5340e-05, 7.5340e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 2.7180e-04, 7.8487e-04, 2.7180e-04, 4.9400e-04,
          3.7646e-04, 1.7107e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.0096e-02, 8.0505e-02, 4.0924e-02, 4.0283e-02, 1.8738e-01,
          6.7322e-02, 3.5706e-02, 7.8659e-03, 7.9651e-03, 1.3580e-02, 1.7151e-02,
          9.1629e-03, 6.2637e-03, 8.7738e-03, 8.6899e-03, 5.2795e-03, 7.9269e-03,
          4.1275e-03, 2.6016e-03, 1.0796e-02, 8.2626e-03, 4.6387e-03, 3.6907e-03,
          1.5884e-02, 1.1276e-02, 2.7435e-02, 1.2856e-02, 1.6632e-02, 6.4392e-03,
          7.7744e-03, 1.0338e-02, 5.0888e-03, 1.1543e-02, 4.5128e-03, 1.3206e-02,
          5.7129e-02, 1.6650e-01, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7393e-04, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.4365e-05,
          1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7393e-04, 1.4365e-05,
          1.7393e-04, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05,
          1.4365e-05, 1.4365e-05, 3.7599e-04, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 9.1553e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7393e-04, 1.4365e-05, 1.4365e-05,
          1.0157e-04, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.9193e-05, 1.4365e-05,
          1.7393e-04, 1.4365e-05, 1.4365e-05, 3.7599e-04, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.4365e-05, 1.7881e-04,
          8.6784e-05, 1.4365e-05, 1.4365e-05, 1.9193e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 8.6784e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7393e-04,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.4365e-05,
          1.9193e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.3447e-03,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7881e-04, 9.1553e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 2.4915e-05,
          1.4365e-05, 1.4365e-05, 8.6784e-05, 1.3447e-03, 8.6784e-05, 1.4365e-05,
          1.7393e-04, 1.9193e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 9.1553e-05, 1.4365e-05, 1.9193e-05,
          1.9193e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.0639e-04, 1.4365e-05,
          8.6784e-05, 1.4365e-05, 3.7599e-04, 7.6389e-04, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 4.0483e-04, 1.7393e-04, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 3.0351e-04, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.0157e-04, 1.4365e-05, 1.4365e-05, 1.7881e-04, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.0157e-04, 4.6110e-04, 1.7393e-04, 1.4365e-05, 1.4365e-05,
          4.6110e-04, 7.6008e-04, 1.4365e-05, 1.4365e-05, 1.4365e-05, 9.1553e-05,
          1.4365e-05, 2.1095e-03, 9.7275e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.7393e-04, 6.9284e-04, 1.4365e-05, 1.7881e-04,
          1.4365e-05, 1.4365e-05, 1.2121e-03, 1.7655e-04, 1.4365e-05, 7.7534e-04,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 1.4365e-05, 1.7393e-04, 8.6784e-05,
          8.6784e-05, 1.4365e-05, 1.4365e-05, 1.7655e-04, 1.7881e-04, 1.4365e-05,
          7.6389e-04, 6.0701e-04, 1.4365e-05, 1.4365e-05, 5.3358e-04, 1.4365e-05,
          2.6393e-04, 9.1553e-05, 8.6784e-05, 1.0157e-04, 1.4365e-05, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.4365e-05, 8.6784e-05, 1.7393e-04, 1.4365e-05,
          8.4734e-04, 1.7881e-04, 3.3259e-04, 1.7655e-04, 6.8808e-04, 3.7360e-04,
          7.6389e-04, 6.8808e-04, 8.5211e-04, 1.7655e-04, 1.4365e-05, 2.6393e-04,
          6.8808e-04, 1.4372e-03, 1.4365e-05, 3.7360e-04, 1.4365e-05, 1.4365e-05,
          5.2977e-04, 1.4365e-05, 3.3259e-04, 1.4365e-05, 1.0157e-04, 1.4365e-05,
          1.4365e-05, 1.4365e-05, 1.0157e-04, 5.7268e-04, 1.0157e-04, 5.2977e-04,
          1.0157e-04, 1.4365e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3870e-02, 7.5012e-02, 3.6835e-02, 5.2338e-02, 1.2183e-01,
          7.6050e-02, 2.9602e-02, 9.7046e-03, 6.7482e-03, 3.0640e-02, 2.2217e-02,
          1.2108e-02, 1.1757e-02, 5.7335e-03, 1.3123e-02, 1.1452e-02, 2.6169e-02,
          1.1436e-02, 4.7455e-03, 3.7441e-03, 5.6114e-03, 4.3411e-03, 1.0468e-02,
          2.6978e-02, 1.2726e-02, 2.4292e-02, 2.5269e-02, 5.6534e-03, 2.1927e-02,
          7.3624e-03, 1.9958e-02, 8.5068e-03, 3.0701e-02, 2.0218e-02, 1.9226e-02,
          2.5726e-02, 7.2060e-03, 1.1389e-01, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.7428e-04, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 1.3351e-05,
          1.3351e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 5.0247e-05, 1.3351e-05, 5.0247e-05, 1.3351e-05,
          1.3351e-05, 5.0247e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05,
          5.0247e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05,
          1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05,
          1.3745e-04, 2.8038e-04, 1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05,
          5.0247e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.7428e-04,
          1.3351e-05, 1.3351e-05, 1.7428e-04, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.7428e-04, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05,
          1.7428e-04, 1.3351e-05, 1.3351e-05, 1.7428e-04, 5.0247e-05, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05,
          5.0247e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.7428e-04,
          5.0247e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 5.0247e-05,
          1.3351e-05, 1.7428e-04, 1.3351e-05, 5.0247e-05, 1.3351e-05, 1.7428e-04,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.3351e-05,
          5.0247e-05, 1.5628e-04, 1.3351e-05, 4.5037e-04, 1.3351e-05, 1.5628e-04,
          1.3351e-05, 5.0247e-05, 4.8423e-04, 5.0247e-05, 1.3351e-05, 3.7837e-04,
          1.5628e-04, 1.3351e-05, 1.3351e-05, 1.3351e-05, 5.0247e-05, 2.8038e-04,
          1.3351e-05, 5.0247e-05, 1.1939e-04, 1.5628e-04, 5.0247e-05, 1.3351e-05,
          1.5628e-04, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.1939e-04,
          1.3351e-05, 4.8423e-04, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.5628e-04,
          1.3351e-05, 1.3351e-05, 5.0247e-05, 5.0247e-05, 1.7428e-04, 1.3351e-05,
          1.5628e-04, 5.0247e-05, 5.0247e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 4.5037e-04, 1.1939e-04, 2.8038e-04, 1.3351e-05, 4.8423e-04,
          5.0247e-05, 1.1939e-04, 1.1939e-04, 1.3351e-05, 1.5628e-04, 1.3351e-05,
          7.7820e-04, 1.3351e-05, 1.7428e-04, 1.7428e-04, 1.3351e-05, 1.5628e-04,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.1939e-04,
          1.3351e-05, 1.3351e-05, 1.7428e-04, 5.0247e-05, 1.1939e-04, 5.0247e-05,
          1.3351e-05, 2.4343e-04, 1.3351e-05, 5.0247e-05, 1.3351e-05, 5.0247e-05,
          2.8038e-04, 1.5628e-04, 1.1939e-04, 5.0247e-05, 1.3351e-05, 1.3351e-05,
          5.0247e-05, 5.0247e-05, 2.4343e-04, 2.8038e-04, 1.3351e-05, 1.3351e-05,
          7.7915e-04, 4.5180e-04, 1.3745e-04, 1.1939e-04, 1.3351e-05, 2.8038e-04,
          1.3351e-05, 1.5628e-04, 2.8300e-04, 1.3351e-05, 1.3351e-05, 1.1939e-04,
          6.8045e-04, 1.3351e-05, 5.0247e-05, 1.1939e-04, 1.3351e-05, 2.8038e-04,
          1.3351e-05, 1.3351e-05, 8.8549e-04, 5.0247e-05, 4.5037e-04, 2.4343e-04,
          7.7868e-04, 1.5628e-04, 1.1939e-04, 1.1939e-04, 5.0259e-04, 5.0247e-05,
          5.5647e-04, 1.7428e-04, 5.0247e-05, 1.3351e-05, 2.8038e-04, 5.5647e-04,
          2.8038e-04, 1.7428e-04, 5.1928e-04, 5.0247e-05, 5.0247e-05, 1.3351e-05,
          5.0247e-05, 4.1437e-04, 5.0247e-05, 4.1342e-04, 1.3351e-05, 1.3351e-05,
          1.5628e-04, 1.1939e-04, 1.1939e-04, 5.0247e-05, 6.0844e-04, 1.1939e-04,
          1.2131e-03, 2.8038e-04, 5.0247e-05, 1.3351e-05, 1.1939e-04, 2.4343e-04,
          6.8045e-04, 1.1939e-04, 5.7602e-04, 5.0247e-05, 5.0247e-05, 1.3745e-04,
          2.4343e-04, 1.3323e-03, 1.3351e-05, 2.4343e-04, 1.1939e-04, 1.3351e-05,
          1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05, 1.3351e-05,
          1.3351e-05, 5.0247e-05, 1.3351e-05, 7.0190e-04, 1.3351e-05, 1.3745e-04,
          1.3351e-05, 1.3351e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0551e-02, 7.8003e-02, 3.1555e-02, 4.0436e-02, 1.0901e-01,
          8.7219e-02, 2.2629e-02, 1.5236e-02, 8.4991e-03, 2.6642e-02, 1.9196e-02,
          1.0895e-02, 1.2901e-02, 5.2528e-03, 2.6794e-02, 1.5915e-02, 1.9470e-02,
          1.0353e-02, 5.3024e-03, 2.5921e-03, 3.0842e-03, 2.5234e-03, 1.1337e-02,
          2.2430e-02, 1.5961e-02, 1.9592e-02, 2.2034e-02, 4.5891e-03, 2.0264e-02,
          6.4926e-03, 2.7145e-02, 7.2060e-03, 5.5389e-02, 2.5787e-02, 2.5772e-02,
          3.0289e-02, 5.6458e-03, 4.4708e-02, 5.6519e-02, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 6.3360e-05,
          6.3360e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 6.3360e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 6.3360e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 6.3360e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05,
          1.6391e-05, 6.7520e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.8811e-04,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 6.3360e-05, 6.3360e-05, 1.6391e-05, 6.3360e-05,
          1.6391e-05, 1.6391e-05, 1.8811e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.8811e-04, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 1.6212e-04, 1.6391e-05, 1.6391e-05, 6.5267e-05, 1.6391e-05,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 1.8811e-04, 6.3360e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.8811e-04,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 6.5267e-05, 1.6391e-05, 6.3360e-05,
          1.6391e-05, 1.8811e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.8811e-04,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05,
          2.6059e-04, 2.6059e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 2.0909e-04,
          1.6391e-05, 1.6391e-05, 3.4928e-04, 1.6391e-05, 1.6391e-05, 3.4928e-04,
          6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05, 3.3379e-04,
          1.6391e-05, 6.3360e-05, 1.6212e-04, 2.0909e-04, 1.6391e-05, 1.6391e-05,
          4.0627e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6212e-04,
          1.6391e-05, 3.4928e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.5267e-05,
          1.6391e-05, 1.6391e-05, 6.3360e-05, 1.8811e-04, 1.8811e-04, 1.6391e-05,
          6.3360e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 5.0354e-04, 5.3120e-04, 1.6391e-05, 3.5429e-04,
          6.3360e-05, 1.6391e-05, 1.6212e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          3.4928e-04, 1.6391e-05, 1.4114e-04, 1.8811e-04, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 6.3360e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05,
          1.6391e-05, 2.8682e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 6.3360e-05,
          2.8682e-04, 1.6391e-05, 1.6212e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          6.3360e-05, 1.8811e-04, 2.8682e-04, 2.8682e-04, 1.6391e-05, 1.6391e-05,
          1.4114e-04, 7.4768e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 3.3879e-04,
          1.6391e-05, 2.1207e-04, 3.3879e-04, 1.6391e-05, 1.6391e-05, 1.6212e-04,
          2.0909e-04, 1.6391e-05, 6.3360e-05, 2.8682e-04, 1.6391e-05, 5.3406e-04,
          1.6391e-05, 1.6391e-05, 3.3379e-04, 1.6391e-05, 8.4019e-04, 6.2847e-04,
          6.7520e-04, 2.0909e-04, 1.6212e-04, 1.6391e-05, 1.8811e-04, 6.3360e-05,
          6.3360e-05, 1.4114e-04, 1.6391e-05, 1.6391e-05, 3.3879e-04, 8.4019e-04,
          3.3379e-04, 1.8811e-04, 1.6391e-05, 6.3360e-05, 6.3360e-05, 1.6391e-05,
          6.3360e-05, 8.7261e-04, 1.6391e-05, 9.6512e-04, 1.6391e-05, 1.6391e-05,
          2.0909e-04, 1.6212e-04, 1.6391e-05, 1.6391e-05, 6.1989e-04, 1.6212e-04,
          6.1989e-04, 3.3379e-04, 1.8811e-04, 1.6391e-05, 1.6212e-04, 1.0118e-03,
          3.3379e-04, 1.6212e-04, 5.3120e-04, 1.6391e-05, 1.6391e-05, 5.2404e-04,
          6.2847e-04, 7.1669e-04, 1.6391e-05, 1.6212e-04, 1.6212e-04, 1.6391e-05,
          1.4114e-04, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05,
          1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.6391e-05, 1.4420e-03,
          1.6391e-05, 1.6391e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.4290e-02, 7.0557e-02, 4.0649e-02, 2.8915e-02, 1.1920e-01,
          5.0720e-02, 2.4475e-02, 5.6267e-03, 3.3836e-03, 5.6572e-03, 7.3586e-03,
          3.5629e-03, 1.2947e-02, 5.4718e-02, 5.2834e-03, 4.3983e-03, 5.0262e-02,
          2.1286e-02, 2.2430e-03, 5.2887e-02, 1.8127e-02, 2.7100e-02, 7.4730e-03,
          7.0763e-03, 5.3444e-03, 1.9073e-02, 1.3176e-02, 5.3680e-02, 2.6581e-02,
          3.4973e-02, 8.6288e-03, 2.6306e-02, 8.2397e-03, 3.7479e-03, 8.4686e-03,
          1.4038e-02, 1.1597e-02, 6.6643e-03, 5.3520e-03, 8.8928e-02, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8981e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05,
          6.6519e-05, 3.8981e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0669e-05, 1.0490e-05, 1.0669e-05, 1.0490e-05,
          1.0490e-05, 3.8087e-05, 1.0490e-05, 3.8981e-05, 3.8981e-05, 3.8087e-05,
          3.8981e-05, 1.0490e-05, 1.0490e-05, 3.8981e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          3.8981e-05, 1.4520e-04, 3.9160e-05, 1.0490e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 3.8981e-05, 1.0490e-05, 6.6519e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8087e-05, 6.6519e-05, 3.8087e-05,
          1.5855e-04, 3.9041e-05, 3.8087e-05, 3.8981e-05, 3.8981e-05, 1.0490e-05,
          3.8981e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.0669e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 3.8207e-05, 6.6519e-05,
          3.8981e-05, 6.6519e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          3.8981e-05, 1.0490e-05, 3.0136e-04, 1.0490e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.0490e-05, 3.8981e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          6.6519e-05, 3.8981e-05, 3.8087e-05, 3.8087e-05, 1.0490e-05, 1.0490e-05,
          6.6519e-05, 3.8087e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 6.6519e-05, 6.6519e-05, 6.6519e-05, 1.0490e-05, 1.0490e-05,
          6.6519e-05, 1.0490e-05, 1.0490e-05, 1.0669e-05, 3.8326e-05, 6.6519e-05,
          6.6519e-05, 1.0550e-05, 3.8087e-05, 3.8981e-05, 1.7273e-04, 3.8087e-05,
          1.0490e-05, 1.0490e-05, 3.8087e-05, 3.8981e-05, 3.8981e-05, 3.8087e-05,
          6.6757e-05, 3.8981e-05, 1.0490e-05, 3.8087e-05, 6.6519e-05, 3.8981e-05,
          3.8087e-05, 1.0490e-05, 1.0490e-05, 6.6757e-05, 6.6519e-05, 3.8326e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 6.6519e-05, 6.6519e-05, 3.8087e-05,
          1.0490e-05, 1.0490e-05, 1.0490e-05, 2.8706e-04, 6.6519e-05, 6.6519e-05,
          3.9220e-05, 6.6519e-05, 1.0490e-05, 3.8087e-05, 1.0490e-05, 3.8087e-05,
          1.0490e-05, 1.3101e-04, 6.6519e-05, 3.8087e-05, 1.0490e-05, 3.8087e-05,
          3.8266e-05, 1.0550e-05, 1.0490e-05, 3.8981e-05, 3.8087e-05, 1.0490e-05,
          1.0490e-05, 1.3101e-04, 1.0550e-05, 6.6519e-05, 3.9101e-05, 6.6519e-05,
          6.6638e-05, 1.0550e-05, 1.0550e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.0490e-05, 1.5855e-04, 1.0490e-05, 1.0490e-05, 1.0788e-05, 3.8326e-05,
          3.8087e-05, 1.0490e-05, 6.6519e-05, 6.6519e-05, 3.8087e-05, 3.8087e-05,
          1.5855e-04, 3.8326e-05, 6.6578e-05, 1.0610e-05, 3.8981e-05, 6.6519e-05,
          3.8087e-05, 6.6519e-05, 1.0490e-05, 3.8087e-05, 1.0490e-05, 1.5879e-04,
          6.6578e-05, 1.0788e-05, 1.0788e-05, 1.0550e-05, 6.6519e-05, 3.8981e-05,
          3.8087e-05, 1.0490e-05, 3.8981e-05, 2.3711e-04, 1.0550e-05, 3.8147e-05,
          1.0490e-05, 1.0788e-05, 1.0550e-05, 6.6519e-05, 3.8981e-05, 1.0788e-05,
          1.0610e-05, 6.6519e-05, 1.7273e-04, 1.0490e-05, 1.0276e-04, 6.6519e-05,
          6.6519e-05, 3.8981e-05, 3.8981e-05, 6.6519e-05, 3.8087e-05, 6.6519e-05,
          3.8981e-05, 6.6519e-05, 1.0490e-05, 6.6519e-05, 3.8981e-05, 1.5855e-04,
          1.5726e-03, 3.9315e-04, 1.0490e-05, 3.8981e-05, 1.0550e-05, 1.0490e-05,
          3.8981e-05, 3.8087e-05, 3.8981e-05, 1.0490e-05, 3.9041e-05, 6.6519e-05,
          3.8981e-05, 6.6519e-05, 3.8087e-05, 6.6519e-05, 3.8087e-05, 1.0490e-05,
          2.6798e-03, 3.8087e-05, 6.6519e-05, 1.0490e-05, 3.6573e-04, 3.8087e-05,
          1.3912e-04, 3.8087e-05, 3.8981e-05, 2.8706e-04, 6.6519e-05, 1.0490e-05,
          1.0490e-05, 3.8981e-05, 3.8981e-05, 1.0550e-05, 2.6464e-04, 6.6519e-05,
          3.8087e-05, 3.8087e-05, 1.9515e-04, 9.9659e-04, 6.6519e-05, 1.7273e-04,
          2.6464e-04, 1.5855e-04, 3.8087e-05, 3.8087e-05, 3.9315e-04, 1.0490e-05,
          3.9315e-04, 1.0490e-05, 3.8087e-05, 3.9315e-04, 1.0490e-05, 3.8087e-05,
          3.8981e-05, 1.3101e-04, 1.3101e-04, 3.8087e-05, 6.6519e-05, 1.0490e-05,
          2.6464e-04, 3.8087e-05, 1.1024e-03, 3.8981e-05, 1.0490e-05, 3.8087e-05,
          3.9315e-04, 1.0490e-05, 6.6519e-05, 3.8087e-05, 1.1024e-03, 3.9315e-04,
          1.0490e-05, 3.6573e-04, 1.0490e-05, 1.0490e-05, 1.0490e-05, 1.0490e-05,
          1.3101e-04, 3.6573e-04, 2.3711e-04, 3.5048e-04, 3.6573e-04, 1.4520e-04,
          1.0490e-05, 2.6464e-04, 2.3711e-04, 6.4087e-04, 3.8981e-05, 1.0490e-05,
          3.8981e-05, 1.4520e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.4511e-02, 8.2092e-02, 3.8574e-02, 4.1290e-02, 1.2891e-01,
          7.8430e-02, 2.8793e-02, 5.6190e-03, 5.2643e-03, 2.5909e-02, 2.7771e-02,
          1.4030e-02, 7.7934e-03, 7.0534e-03, 8.4610e-03, 6.1150e-03, 1.4885e-02,
          8.2092e-03, 4.1733e-03, 6.1722e-03, 9.3536e-03, 7.4921e-03, 7.8583e-03,
          3.9307e-02, 1.4893e-02, 2.2842e-02, 1.5884e-02, 7.5150e-03, 1.1063e-02,
          6.7520e-03, 8.2474e-03, 5.9624e-03, 1.4107e-02, 1.0910e-02, 9.6970e-03,
          3.7689e-02, 1.5221e-02, 3.2501e-02, 1.0971e-02, 7.3166e-03, 1.2384e-01,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 2.6703e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.6928e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 2.6703e-04,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.7770e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05,
          6.6519e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04, 1.5676e-05, 6.6519e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 2.6703e-04, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 9.2602e-04, 1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 6.6519e-05,
          1.5676e-05, 1.5676e-05, 2.1625e-04, 2.6703e-04, 1.5676e-05, 1.5676e-05,
          2.6703e-04, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04,
          1.5676e-05, 6.8367e-05, 2.6846e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 6.6519e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04,
          1.5676e-05, 1.5676e-05, 2.6703e-04, 1.5676e-05, 5.8270e-04, 1.5676e-05,
          8.1396e-04, 6.6519e-05, 1.5676e-05, 1.5676e-05, 2.6703e-04, 6.6519e-05,
          5.8270e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 2.7013e-04, 1.5676e-05,
          1.0147e-03, 2.6703e-04, 5.3215e-04, 6.6519e-05, 6.6519e-05, 1.5676e-05,
          6.6519e-05, 1.0157e-03, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 1.5676e-05, 1.5676e-05, 6.6519e-05, 6.6519e-05, 1.5676e-05,
          1.7242e-03, 6.6519e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.6519e-05, 7.6294e-04, 1.0147e-03, 1.5676e-05, 1.5676e-05, 2.1625e-04,
          7.6294e-04, 2.2659e-03, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 2.6703e-04, 1.5676e-05, 6.6519e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.7746e-02, 8.7097e-02, 4.0497e-02, 3.5492e-02, 1.4343e-01,
          6.6101e-02, 3.5553e-02, 3.6888e-03, 3.7403e-03, 1.3008e-02, 1.8814e-02,
          8.7967e-03, 4.5700e-03, 7.8506e-03, 5.1613e-03, 4.0245e-03, 7.8812e-03,
          4.7836e-03, 3.6335e-03, 1.2131e-02, 1.4511e-02, 1.2665e-02, 5.0316e-03,
          2.5635e-02, 1.3626e-02, 2.5970e-02, 1.0811e-02, 1.1124e-02, 5.3215e-03,
          6.5041e-03, 4.9133e-03, 3.8128e-03, 7.1487e-03, 4.5929e-03, 4.7760e-03,
          1.6769e-02, 2.0126e-02, 1.2787e-02, 5.4550e-03, 1.1131e-02, 5.7373e-02,
          1.8079e-01, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          7.3016e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05,
          1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05,
          7.3016e-05, 1.9789e-05, 1.9789e-05, 4.9400e-04, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 2.2173e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 7.3016e-05,
          7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.5400e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 7.3731e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 2.0444e-05, 1.9789e-05,
          7.3016e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 2.2173e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3731e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05,
          7.3016e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3731e-05,
          1.9789e-05, 7.3731e-05, 7.5400e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          4.4060e-04, 1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 7.3731e-05,
          1.9789e-05, 1.9789e-05, 7.3731e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          1.9789e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05,
          7.3016e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 7.5400e-05, 1.9789e-05,
          7.3731e-05, 7.3731e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 1.9789e-05,
          7.3016e-05, 7.3731e-05, 7.3016e-05, 9.2745e-04, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05,
          7.3731e-05, 7.3016e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          7.3016e-05, 1.9789e-05, 7.3016e-05, 7.3016e-05, 1.9789e-05, 9.8515e-04,
          1.9789e-05, 9.8133e-04, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05, 1.9789e-05, 4.3631e-04, 1.9789e-05, 1.9789e-05,
          1.9789e-05, 1.9789e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.7258e-02, 8.4534e-02, 3.9917e-02, 3.4515e-02, 1.3037e-01,
          6.8481e-02, 3.1738e-02, 3.8948e-03, 4.3068e-03, 1.4771e-02, 1.9440e-02,
          9.4681e-03, 5.1460e-03, 7.9346e-03, 6.0577e-03, 4.3068e-03, 8.6288e-03,
          5.1613e-03, 3.7670e-03, 9.4757e-03, 1.0887e-02, 9.4376e-03, 5.5809e-03,
          2.6123e-02, 1.4069e-02, 2.3331e-02, 1.0490e-02, 9.4147e-03, 5.9738e-03,
          7.2517e-03, 5.3101e-03, 4.2419e-03, 8.7051e-03, 5.3062e-03, 5.5313e-03,
          2.3087e-02, 1.9974e-02, 1.4328e-02, 6.4049e-03, 9.6054e-03, 6.0120e-02,
          1.0931e-01, 7.4402e-02, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05,
          1.6689e-05, 1.6689e-05, 3.2258e-04, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          7.7307e-05, 1.6689e-05, 1.6689e-05, 7.7295e-04, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.8299e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 3.2258e-04,
          7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 7.8917e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.8299e-05,
          1.6689e-05, 1.6689e-05, 7.7307e-05, 7.7307e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          7.7307e-05, 1.6689e-05, 3.2258e-04, 3.2258e-04, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 7.7307e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.8299e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7903e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 7.1239e-04, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          7.7295e-04, 7.7307e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 3.2473e-04,
          1.6689e-05, 7.7903e-05, 7.7486e-04, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 3.2473e-04,
          1.6689e-05, 1.6689e-05, 3.2306e-04, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          1.6689e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 7.7307e-05,
          7.7307e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 7.7486e-04, 1.6689e-05,
          3.2258e-04, 7.7903e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 1.6689e-05,
          7.7307e-05, 7.9513e-05, 7.7307e-05, 2.6202e-04, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 7.7307e-05, 7.7307e-05, 1.6689e-05,
          7.7307e-05, 7.7307e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 4.6730e-04,
          7.7307e-05, 1.6689e-05, 7.7343e-04, 7.7307e-05, 1.6689e-05, 1.6689e-05,
          2.1362e-03, 2.4414e-03, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05, 1.6689e-05, 5.8746e-04, 1.6689e-05, 1.6689e-05,
          1.6689e-05, 1.6689e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.5869e-02, 8.1055e-02, 4.0009e-02, 3.2959e-02, 1.3159e-01,
          6.3599e-02, 3.5309e-02, 3.4103e-03, 3.2673e-03, 9.7275e-03, 1.4565e-02,
          6.4888e-03, 4.2152e-03, 6.7902e-03, 4.9553e-03, 3.7174e-03, 6.3744e-03,
          4.2534e-03, 3.1490e-03, 9.6893e-03, 1.0468e-02, 1.1139e-02, 4.5738e-03,
          1.8784e-02, 1.1940e-02, 2.5940e-02, 9.1553e-03, 1.0262e-02, 4.6158e-03,
          5.7259e-03, 4.4441e-03, 3.2539e-03, 6.2141e-03, 3.8757e-03, 4.0131e-03,
          1.1971e-02, 1.4297e-02, 8.2703e-03, 4.5128e-03, 8.7891e-03, 3.2410e-02,
          1.0297e-01, 4.7546e-02, 1.3464e-01, 2.1577e-05, 6.2585e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 6.2585e-05, 9.1410e-04, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 3.5977e-04, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 6.2585e-05, 2.1577e-05, 6.2585e-05, 9.5510e-04, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          6.2585e-05, 2.1577e-05, 2.1577e-05, 3.5977e-04, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.4378e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 3.5977e-04,
          6.2585e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.3419e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 6.5446e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05, 6.2585e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 6.3419e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.3419e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 6.3419e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 3.5977e-04, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.4378e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 2.1577e-05, 6.3419e-05, 2.1577e-05, 6.2585e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 6.2585e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 3.6263e-04,
          2.1577e-05, 6.3419e-05, 3.6263e-04, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 3.6263e-04,
          2.1577e-05, 2.1577e-05, 6.3419e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05,
          2.1577e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 6.3419e-05, 6.3419e-05,
          6.2585e-05, 6.2585e-05, 2.1577e-05, 2.1577e-05, 3.6263e-04, 2.1577e-05,
          6.3419e-05, 6.3419e-05, 6.2585e-05, 2.1577e-05, 6.2585e-05, 2.1577e-05,
          6.2585e-05, 6.3419e-05, 6.2585e-05, 7.8917e-04, 2.1577e-05, 2.1577e-05,
          6.2585e-05, 2.1577e-05, 2.1577e-05, 6.2585e-05, 6.2585e-05, 2.1577e-05,
          6.3419e-05, 6.3419e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05, 1.1597e-03,
          6.2585e-05, 2.1577e-05, 3.6049e-04, 6.2585e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 5.3358e-04, 2.1577e-05, 2.1577e-05, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05, 2.1577e-05, 1.2660e-04, 3.1877e-04, 6.2585e-05,
          2.1577e-05, 6.2585e-05, 2.1577e-05, 5.0926e-04, 2.1577e-05, 2.1577e-05,
          2.1577e-05, 2.1577e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.6724e-02, 8.4961e-02, 4.2786e-02, 3.2928e-02, 1.2054e-01,
          6.5430e-02, 2.9724e-02, 4.6272e-03, 4.2801e-03, 1.5564e-02, 2.0508e-02,
          9.6436e-03, 6.6795e-03, 8.2626e-03, 7.6370e-03, 5.0468e-03, 1.2314e-02,
          6.3477e-03, 3.6411e-03, 8.1711e-03, 1.0681e-02, 1.0742e-02, 7.2060e-03,
          2.7542e-02, 1.4938e-02, 2.2247e-02, 1.0948e-02, 1.0147e-02, 7.6408e-03,
          7.2441e-03, 6.4240e-03, 5.0278e-03, 1.0643e-02, 6.2141e-03, 7.5798e-03,
          2.6199e-02, 1.5549e-02, 1.4587e-02, 7.0076e-03, 9.0790e-03, 5.6854e-02,
          5.2765e-02, 3.7964e-02, 4.0009e-02, 4.4708e-02, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05,
          6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05,
          1.7107e-05, 1.7107e-05, 2.5368e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.5368e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.3360e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05,
          6.3360e-05, 1.7107e-05, 1.7107e-05, 5.3740e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.8358e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 2.5368e-04,
          6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.5487e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.4611e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.8358e-05,
          1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05, 2.5368e-04, 1.7107e-05,
          6.3360e-05, 6.3360e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 2.5368e-04, 1.7107e-05, 1.8358e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          6.3360e-05, 1.7107e-05, 2.5368e-04, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 3.0160e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.4373e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 4.9114e-04, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          6.3360e-05, 6.3360e-05, 5.3740e-04, 1.7107e-05, 1.7107e-05, 2.5487e-04,
          1.7107e-05, 6.5684e-05, 5.3930e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          6.3360e-05, 1.7107e-05, 6.3360e-05, 1.7107e-05, 1.7107e-05, 5.3930e-04,
          1.7107e-05, 1.7107e-05, 2.5368e-04, 1.7107e-05, 6.3360e-05, 1.7107e-05,
          6.3360e-05, 2.5368e-04, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05,
          6.3360e-05, 3.4666e-04, 1.7107e-05, 1.7107e-05, 5.3930e-04, 1.7107e-05,
          5.3740e-04, 5.3835e-04, 6.3360e-05, 6.3360e-05, 6.3360e-05, 1.7107e-05,
          6.3360e-05, 5.3930e-04, 6.3360e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          6.3360e-05, 1.7107e-05, 1.7107e-05, 6.3360e-05, 6.3360e-05, 1.7107e-05,
          5.3740e-04, 5.3740e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 3.0041e-04,
          6.3360e-05, 1.7107e-05, 5.3835e-04, 6.3360e-05, 1.7107e-05, 1.7107e-05,
          2.5539e-03, 2.5997e-03, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 2.0742e-04, 6.0463e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0948e-02, 7.3914e-02, 3.3539e-02, 3.4821e-02, 8.6121e-02,
          7.8796e-02, 1.9806e-02, 8.8348e-03, 6.8741e-03, 1.4397e-02, 1.6525e-02,
          6.8169e-03, 3.0777e-02, 1.2070e-02, 1.1124e-02, 9.6664e-03, 5.5664e-02,
          2.9846e-02, 4.8332e-03, 3.8242e-03, 3.7785e-03, 6.2103e-03, 1.8951e-02,
          1.9119e-02, 8.0185e-03, 1.6663e-02, 2.1332e-02, 5.5504e-03, 5.6885e-02,
          1.4709e-02, 1.5495e-02, 1.9119e-02, 1.9333e-02, 1.1787e-02, 8.6441e-03,
          7.1716e-03, 2.4452e-03, 1.2665e-02, 1.1543e-02, 4.9553e-03, 9.9487e-03,
          3.2711e-03, 2.8477e-03, 2.5234e-03, 3.0708e-03, 8.0872e-02, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 4.0591e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 4.0591e-05,
          1.5998e-04, 6.1870e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 4.0591e-05, 6.1870e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 6.2168e-05, 1.8477e-05, 6.2168e-05, 1.8477e-05,
          1.8477e-05, 3.9756e-05, 1.8477e-05, 1.5402e-04, 1.5974e-04, 1.5199e-04,
          6.1870e-05, 1.8477e-05, 4.0591e-05, 6.1870e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 2.3198e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          6.1870e-05, 1.8489e-04, 6.1870e-05, 1.8477e-05, 4.0591e-05, 1.8477e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 1.8477e-05, 1.5974e-04, 1.5974e-04,
          4.0591e-05, 1.8477e-05, 1.8477e-05, 4.0591e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 6.1870e-05, 7.8261e-05, 1.2646e-03,
          1.2426e-03, 4.0054e-05, 6.2168e-05, 4.0591e-05, 9.1434e-05, 1.8477e-05,
          3.0112e-04, 1.5974e-04, 9.1434e-05, 6.9320e-05, 6.1870e-05, 1.8477e-05,
          1.8477e-05, 6.9320e-05, 3.9756e-05, 1.8477e-05, 3.9756e-05, 3.9756e-05,
          6.9618e-05, 6.9320e-05, 4.0591e-05, 3.9756e-05, 9.1732e-05, 7.8261e-05,
          1.8477e-05, 4.0591e-05, 1.8477e-05, 3.9756e-05, 1.8477e-05, 1.8477e-05,
          9.1434e-05, 6.9320e-05, 1.5402e-04, 4.8566e-04, 6.1870e-05, 1.8477e-05,
          3.9756e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 6.9320e-05,
          9.1434e-05, 1.2932e-03, 4.0591e-05, 9.1434e-05, 3.9756e-05, 1.8477e-05,
          7.8261e-05, 5.9664e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          6.9320e-05, 9.1434e-05, 9.1434e-05, 2.4748e-04, 1.8477e-05, 1.8477e-05,
          9.1434e-05, 3.9756e-05, 1.8477e-05, 6.9320e-05, 1.2934e-04, 9.9540e-05,
          1.6284e-04, 1.7986e-03, 3.9756e-05, 1.8477e-05, 9.1434e-05, 4.0591e-05,
          5.9664e-05, 1.8477e-05, 4.0591e-05, 9.1434e-05, 9.1434e-05, 9.1434e-05,
          2.0087e-04, 6.9320e-05, 1.8477e-05, 9.1434e-05, 6.1870e-05, 1.8477e-05,
          1.5402e-04, 1.8477e-05, 6.9320e-05, 1.2934e-04, 9.1434e-05, 6.3419e-04,
          6.9320e-05, 1.8477e-05, 1.8477e-05, 9.1434e-05, 6.1870e-05, 6.1870e-05,
          4.0591e-05, 1.4925e-04, 8.0943e-05, 2.6345e-04, 1.8966e-04, 6.1870e-05,
          2.0111e-04, 3.2854e-04, 5.9664e-05, 3.9697e-04, 1.8477e-05, 4.0591e-05,
          3.9756e-05, 6.9320e-05, 2.8729e-04, 1.2910e-04, 1.0699e-04, 1.9741e-04,
          1.8764e-04, 1.8477e-05, 1.8477e-05, 3.9756e-05, 6.1870e-05, 6.9320e-05,
          5.9664e-05, 1.5974e-04, 1.2791e-04, 2.9898e-04, 2.3854e-04, 9.1434e-05,
          3.5381e-04, 5.0449e-04, 5.9664e-05, 1.8477e-05, 1.8477e-05, 6.9320e-05,
          6.9320e-05, 1.5402e-03, 1.2722e-03, 6.9320e-05, 5.9664e-05, 1.1945e-04,
          4.0591e-05, 1.8477e-05, 9.1434e-05, 9.9540e-05, 2.5749e-04, 9.1434e-05,
          5.1069e-04, 2.0123e-04, 2.1923e-04, 1.9760e-03, 6.9320e-05, 9.1434e-05,
          2.0099e-04, 2.1875e-04, 1.8477e-05, 9.1434e-05, 5.2309e-04, 2.3890e-04,
          1.9765e-04, 1.8559e-03, 5.9414e-04, 5.9664e-05, 4.0591e-05, 6.9320e-05,
          1.9741e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.2791e-04, 2.0897e-04,
          5.3406e-04, 1.0605e-03, 1.0605e-03, 9.1434e-05, 3.9756e-05, 5.9664e-05,
          5.9664e-05, 6.9320e-05, 4.1008e-04, 1.8477e-05, 6.1512e-04, 6.1870e-05,
          6.1870e-05, 5.7411e-04, 3.9756e-05, 1.7035e-04, 9.1434e-05, 6.1870e-05,
          3.9756e-05, 1.7035e-04, 1.1051e-04, 1.2910e-04, 3.9756e-05, 9.1434e-05,
          3.5191e-04, 2.4748e-04, 3.9756e-05, 3.9756e-05, 1.0099e-03, 5.9664e-05,
          3.9756e-05, 8.1778e-05, 1.8477e-05, 6.1512e-04, 7.7426e-05, 2.8706e-04,
          3.9756e-05, 1.7035e-04, 2.5415e-04, 4.0591e-05, 6.9320e-05, 1.1051e-04,
          5.2023e-04, 4.0591e-05, 1.9515e-04, 1.1051e-04, 1.7309e-04, 2.5415e-04,
          3.4952e-04, 6.1870e-05, 1.8477e-05, 1.9515e-04, 3.1447e-04, 6.9320e-05,
          3.6860e-04, 1.2910e-04, 1.1051e-04, 1.3218e-03, 5.6696e-04, 2.5415e-04,
          3.8505e-04, 1.3185e-04, 1.9157e-04, 8.9216e-04, 4.4966e-04, 6.0797e-04,
          8.7214e-04, 7.6628e-04, 3.8576e-04, 2.5749e-04, 2.6655e-04, 1.8477e-05,
          2.8515e-04, 3.6860e-04, 6.1870e-05, 5.3549e-04, 6.9320e-05, 1.3185e-04,
          6.9320e-05, 1.4818e-04, 1.1051e-04, 1.8477e-05, 8.0395e-04, 1.1051e-04,
          8.0395e-04, 4.8733e-04, 4.4298e-04, 1.8477e-05, 1.1051e-04, 4.2105e-04,
          8.0395e-04, 1.1051e-04, 5.5361e-04, 2.2531e-04, 3.9077e-04, 1.5402e-04,
          5.3120e-04, 7.4387e-04, 3.2759e-04, 1.1051e-04, 6.1798e-04, 1.8477e-05,
          3.8052e-04, 1.6284e-04, 3.9756e-05, 1.8477e-05, 1.8477e-05, 7.8261e-05,
          1.8477e-05, 1.9300e-04, 1.3328e-04, 5.1212e-04, 6.0558e-04, 1.3328e-04,
          3.9756e-05, 1.3328e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.0544e-02, 7.6111e-02, 3.5217e-02, 3.8116e-02, 8.5815e-02,
          7.5073e-02, 2.3315e-02, 6.1646e-03, 4.7302e-03, 1.8356e-02, 1.6434e-02,
          7.8735e-03, 1.5564e-02, 4.1084e-03, 2.5543e-02, 7.5111e-03, 2.2507e-02,
          1.0490e-02, 4.0894e-03, 2.8896e-03, 4.4098e-03, 5.2872e-03, 1.1246e-02,
          2.2629e-02, 1.8600e-02, 2.0142e-02, 1.7365e-02, 4.1695e-03, 1.9104e-02,
          5.2414e-03, 1.0788e-02, 6.1302e-03, 4.3976e-02, 1.4275e-02, 1.1833e-02,
          2.4155e-02, 4.1275e-03, 3.0731e-02, 2.3468e-02, 3.5286e-03, 3.1830e-02,
          8.2397e-03, 7.8964e-03, 5.6000e-03, 9.1782e-03, 2.1179e-02, 9.2773e-02,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05,
          5.9366e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          5.9366e-05, 5.9366e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05,
          5.9366e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.6046e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.6046e-04,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.1832e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.6046e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.6046e-04, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.6046e-04, 1.7166e-05, 1.7166e-05,
          1.6046e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.0380e-05, 1.7166e-05,
          1.6046e-04, 1.7166e-05, 1.7166e-05, 1.6046e-04, 1.6046e-04, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05, 1.6046e-04,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 6.0380e-05, 1.7166e-05, 5.9366e-05,
          1.7166e-05, 1.6046e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.6046e-04,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05,
          5.9366e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05,
          1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 2.2840e-04,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.6046e-04,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05, 1.7166e-05,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 3.0947e-04,
          1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.8239e-05,
          1.7166e-05, 1.7166e-05, 5.9366e-05, 5.9366e-05, 1.6046e-04, 1.7166e-05,
          5.9366e-05, 6.0380e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.6046e-04, 1.7166e-05, 2.2936e-04,
          5.9366e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          2.2840e-04, 1.7166e-05, 1.1832e-04, 1.6046e-04, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.6046e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.9366e-05,
          1.1832e-04, 1.7166e-05, 1.7166e-05, 5.9366e-05, 1.7166e-05, 1.7166e-05,
          5.9366e-05, 1.6046e-04, 1.1832e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.1832e-04, 3.5167e-04, 1.1832e-04, 1.7166e-05, 1.7166e-05, 5.1260e-04,
          1.7166e-05, 3.0947e-04, 5.1355e-04, 1.7166e-05, 1.7166e-05, 3.0947e-04,
          5.9366e-05, 1.7166e-05, 5.9366e-05, 3.0947e-04, 1.7166e-05, 8.0490e-04,
          1.7166e-05, 1.7166e-05, 1.6046e-04, 1.1832e-04, 8.5211e-04, 3.0947e-04,
          4.5276e-04, 4.5276e-04, 1.7166e-05, 1.7166e-05, 3.2949e-04, 2.2840e-04,
          8.5211e-04, 1.1832e-04, 1.7166e-05, 1.7166e-05, 8.0585e-04, 1.7166e-05,
          6.2180e-04, 3.2949e-04, 3.0947e-04, 5.9366e-05, 5.9366e-05, 1.7166e-05,
          5.9366e-05, 8.0490e-04, 5.9366e-05, 1.6046e-04, 1.7166e-05, 1.7166e-05,
          3.0947e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.2180e-04, 1.7166e-05,
          6.2180e-04, 6.8140e-04, 1.6046e-04, 1.7166e-05, 3.0947e-04, 7.6246e-04,
          6.2180e-04, 3.0947e-04, 9.7370e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          7.6246e-04, 4.5276e-04, 1.7166e-05, 3.0947e-04, 3.0947e-04, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.1832e-04, 1.7166e-05, 1.1832e-04, 1.7166e-05,
          1.7166e-05, 1.6046e-04, 1.1832e-04, 5.8413e-04, 1.7166e-05, 1.5049e-03,
          1.7166e-05, 1.7166e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1726e-02, 6.8787e-02, 4.4922e-02, 3.5889e-02, 9.0271e-02,
          5.0964e-02, 2.6581e-02, 4.7569e-03, 2.7599e-03, 5.0163e-03, 7.7171e-03,
          3.3321e-03, 1.3252e-02, 4.7791e-02, 4.1847e-03, 4.7035e-03, 6.0272e-02,
          3.0655e-02, 2.3365e-03, 1.9989e-02, 1.1200e-02, 3.4088e-02, 1.2070e-02,
          8.7585e-03, 5.2681e-03, 2.2156e-02, 1.3771e-02, 2.3087e-02, 2.9617e-02,
          4.3945e-02, 7.5874e-03, 2.2232e-02, 6.4583e-03, 3.6850e-03, 5.2605e-03,
          5.4893e-03, 3.6049e-03, 6.1340e-03, 4.6082e-03, 3.1799e-02, 8.8959e-03,
          7.4692e-03, 5.4550e-03, 6.3820e-03, 4.8447e-03, 2.9968e-02, 1.1864e-02,
          5.8990e-02, 8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05,
          5.6803e-05, 2.9981e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 3.5107e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.5235e-06, 8.2850e-06,
          8.2850e-06, 3.5107e-05, 8.2850e-06, 2.9981e-05, 5.6803e-05, 3.5107e-05,
          5.6803e-05, 8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          5.6803e-05, 1.4412e-04, 5.6803e-05, 8.2850e-06, 3.5107e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06, 5.6803e-05, 2.9981e-05,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 8.2850e-06, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 3.5107e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 5.6803e-05, 8.2850e-06,
          1.3781e-04, 2.9981e-05, 5.6803e-05, 2.9981e-05, 2.9981e-05, 8.2850e-06,
          2.9981e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06, 3.5107e-05, 5.6803e-05,
          2.9981e-05, 5.6803e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          2.9981e-05, 8.2850e-06, 5.6803e-05, 8.2850e-06, 5.6803e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06,
          5.6803e-05, 2.9981e-05, 5.6803e-05, 5.6803e-05, 8.2850e-06, 2.9981e-05,
          2.5678e-04, 2.9981e-05, 8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06,
          2.9981e-05, 5.6803e-05, 5.6803e-05, 5.6803e-05, 8.2850e-06, 8.2850e-06,
          5.6803e-05, 8.2850e-06, 8.2850e-06, 8.2850e-06, 5.6982e-05, 5.6803e-05,
          5.6803e-05, 8.2850e-06, 3.5107e-05, 2.9981e-05, 5.6803e-05, 5.6803e-05,
          8.2850e-06, 8.2850e-06, 5.6803e-05, 2.2995e-04, 2.9981e-05, 5.6803e-05,
          5.6803e-05, 2.9981e-05, 8.2850e-06, 5.6803e-05, 5.6803e-05, 2.9981e-05,
          5.6803e-05, 8.2850e-06, 8.2850e-06, 5.6982e-05, 5.6803e-05, 3.5107e-05,
          2.9981e-05, 8.2850e-06, 8.2850e-06, 1.3781e-04, 5.6803e-05, 8.2850e-06,
          8.2850e-06, 8.2850e-06, 2.2995e-04, 3.3784e-04, 5.6803e-05, 5.6803e-05,
          2.2995e-04, 5.6803e-05, 8.2850e-06, 3.5107e-05, 8.2850e-06, 3.5107e-05,
          2.9981e-05, 1.1104e-04, 5.6803e-05, 3.5107e-05, 2.9981e-05, 7.8797e-05,
          8.2850e-06, 8.2850e-06, 8.2850e-06, 2.9981e-05, 3.5107e-05, 8.2850e-06,
          8.2850e-06, 3.1090e-04, 8.2850e-06, 5.6803e-05, 5.6803e-05, 1.3781e-04,
          3.3784e-04, 8.2850e-06, 8.2850e-06, 8.2850e-06, 2.9981e-05, 8.2850e-06,
          2.9981e-05, 1.3781e-04, 8.2850e-06, 2.2995e-04, 8.3447e-06, 7.9036e-05,
          5.6803e-05, 8.2850e-06, 5.6803e-05, 5.6803e-05, 2.9981e-05, 1.3781e-04,
          3.3784e-04, 2.5702e-04, 1.3781e-04, 2.9981e-05, 1.1104e-04, 3.3784e-04,
          2.5678e-04, 2.8658e-04, 8.2850e-06, 8.2850e-06, 8.2850e-06, 3.8171e-04,
          2.5678e-04, 2.0826e-04, 2.0826e-04, 8.3447e-06, 2.9981e-05, 2.9981e-05,
          7.8797e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 8.3447e-06, 5.6803e-05,
          8.2850e-06, 2.9981e-05, 8.2850e-06, 5.6803e-05, 2.2995e-04, 2.3019e-04,
          8.2850e-06, 2.5964e-04, 1.7095e-04, 8.2850e-06, 1.1104e-04, 5.6803e-05,
          5.6803e-05, 2.9981e-05, 2.9981e-05, 2.5964e-04, 3.5107e-05, 2.8658e-04,
          2.9981e-05, 3.3784e-04, 8.2850e-06, 1.3781e-04, 2.9981e-05, 3.3784e-04,
          3.3784e-04, 4.8161e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 8.2850e-06, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 5.6803e-05, 8.2850e-06, 5.6803e-05, 2.9981e-05, 2.9981e-05,
          3.8147e-04, 5.6803e-05, 5.6803e-05, 2.9981e-05, 4.8113e-04, 2.9981e-05,
          2.5964e-04, 2.8658e-04, 2.9981e-05, 5.6803e-05, 3.3784e-04, 2.9981e-05,
          8.2850e-06, 2.9981e-05, 2.9981e-05, 2.9981e-05, 1.7095e-04, 5.6803e-05,
          5.6803e-05, 8.2850e-06, 2.8658e-04, 5.3787e-04, 2.9981e-05, 6.8140e-04,
          4.8161e-04, 2.8658e-04, 2.2995e-04, 8.2850e-06, 2.2495e-04, 3.6764e-04,
          4.2200e-04, 8.2850e-06, 3.5107e-05, 4.2200e-04, 8.2850e-06, 8.2850e-06,
          2.9981e-05, 3.1090e-04, 1.1104e-04, 8.2850e-06, 5.6803e-05, 8.2850e-06,
          4.8637e-04, 2.9981e-05, 4.8161e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          4.5180e-04, 1.1104e-04, 2.9981e-05, 5.6803e-05, 5.3787e-04, 6.5184e-04,
          2.9981e-05, 8.2493e-04, 8.2850e-06, 1.1104e-04, 8.2850e-06, 8.2850e-06,
          2.2495e-04, 2.2495e-04, 1.4412e-04, 2.2125e-04, 1.4412e-04, 1.4412e-04,
          2.9981e-05, 2.5177e-04, 2.2495e-04, 6.7282e-04, 1.4412e-04, 4.0078e-04,
          1.4412e-04, 2.2495e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.6296e-02, 7.0129e-02, 5.2521e-02, 3.6377e-02, 1.1224e-01,
          4.7394e-02, 3.5828e-02, 3.8662e-03, 2.1706e-03, 3.9825e-03, 6.5689e-03,
          2.7351e-03, 7.9956e-03, 2.8122e-02, 3.3569e-03, 3.6201e-03, 3.1891e-02,
          1.6571e-02, 2.3518e-03, 2.6077e-02, 1.9333e-02, 5.7190e-02, 1.0300e-02,
          7.5798e-03, 5.2795e-03, 2.6962e-02, 1.3733e-02, 2.9358e-02, 1.5022e-02,
          1.9836e-02, 5.1231e-03, 1.4015e-02, 5.3291e-03, 2.7752e-03, 4.1237e-03,
          5.7182e-03, 4.4556e-03, 6.1760e-03, 4.0894e-03, 2.9739e-02, 8.4686e-03,
          1.0483e-02, 7.6218e-03, 1.0490e-02, 6.5193e-03, 1.5823e-02, 1.0040e-02,
          3.9001e-02, 7.7759e-02, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.3697e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1384e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.3697e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1384e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1384e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1206e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 2.3568e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1384e-05, 1.0979e-04,
          1.1086e-05, 1.1206e-05, 1.1086e-05, 1.0979e-04, 1.0979e-04, 1.1086e-05,
          1.1206e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1384e-05, 1.1086e-05, 1.1086e-05, 1.3697e-04, 1.1086e-05, 1.0979e-04,
          1.3697e-04, 1.1086e-05, 1.1086e-05, 1.1021e-04, 1.1086e-05, 1.1384e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.0979e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 2.3568e-04, 1.1086e-05,
          1.1384e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.3697e-04,
          1.1206e-05, 1.1206e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.0979e-04, 1.1206e-05, 1.1086e-05, 1.1206e-05, 1.1086e-05,
          1.1086e-05, 1.1206e-05, 1.1206e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1206e-05, 1.1563e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          5.7936e-04, 1.1504e-05, 1.3709e-04, 1.1206e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1504e-05,
          1.1384e-05, 1.1206e-05, 1.1206e-05, 1.1206e-05, 1.1086e-05, 1.1086e-05,
          1.3697e-04, 1.1086e-05, 1.0979e-04, 1.0979e-04, 1.1206e-05, 1.1206e-05,
          1.1086e-05, 1.1206e-05, 1.1206e-05, 1.1086e-05, 1.1086e-05, 1.1206e-05,
          1.1206e-05, 1.1086e-05, 2.3568e-04, 1.1086e-05, 1.1206e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.0979e-04, 1.1086e-05, 1.3697e-04, 1.1086e-05, 1.0979e-04,
          1.0979e-04, 1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          2.3568e-04, 1.1086e-05, 1.3697e-04, 1.1086e-05, 1.1086e-05, 2.3568e-04,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.3697e-04, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 2.3568e-04, 2.3568e-04,
          1.3697e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.0979e-04, 1.0979e-04,
          2.3568e-04, 2.3568e-04, 1.3697e-04, 1.1086e-05, 1.0979e-04, 1.1086e-05,
          5.7936e-04, 1.1086e-05, 1.3697e-04, 1.0979e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.3695e-03, 1.1086e-05,
          2.3568e-04, 1.1086e-05, 1.0979e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          5.7936e-04, 1.1086e-05, 1.0979e-04, 1.1086e-05, 1.1086e-05, 1.0979e-04,
          1.1086e-05, 4.5347e-04, 1.1086e-05, 1.1086e-05, 1.1086e-05, 1.1086e-05,
          1.0979e-04, 7.8821e-04, 4.4417e-04, 1.2302e-04, 1.0979e-04, 1.1086e-05,
          1.1086e-05, 7.8821e-04, 7.8821e-04, 6.4373e-04, 1.1086e-05, 1.1086e-05,
          1.1086e-05, 1.0979e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.8610e-02, 8.5327e-02, 5.7404e-02, 7.4097e-02, 2.2229e-01,
          6.2408e-02, 7.1045e-02, 4.1809e-03, 2.5921e-03, 4.3030e-03, 6.4774e-03,
          2.7351e-03, 3.8357e-03, 4.6043e-03, 3.1738e-03, 3.4180e-03, 6.0883e-03,
          4.2572e-03, 1.8339e-03, 3.9330e-03, 4.9171e-03, 4.0245e-03, 4.5395e-03,
          7.0000e-03, 4.2763e-03, 4.9805e-02, 1.7075e-02, 7.0114e-03, 4.8294e-03,
          4.5815e-03, 4.7073e-03, 3.2845e-03, 4.7226e-03, 2.5406e-03, 3.1033e-03,
          5.0583e-03, 4.1466e-03, 5.3139e-03, 3.6373e-03, 3.6869e-03, 5.0316e-03,
          6.0539e-03, 4.1351e-03, 5.0011e-03, 2.9297e-03, 3.9978e-03, 5.3825e-03,
          2.7733e-03, 5.9319e-03, 1.1975e-01, 1.6987e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05,
          7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05,
          7.6711e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          7.6711e-05, 1.0395e-03, 7.6711e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.5199e-04, 1.6987e-05,
          7.6711e-05, 1.0986e-03, 7.6711e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05,
          9.2208e-05, 1.6987e-05, 9.2208e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05,
          1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05,
          1.6987e-05, 1.6987e-05, 1.5199e-04, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          7.6711e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05, 9.2208e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 7.6711e-05, 7.6711e-05, 1.5199e-04, 9.2208e-05, 1.6987e-05,
          7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05,
          1.5199e-04, 1.6987e-05, 7.6711e-05, 1.5199e-04, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05, 1.5199e-04,
          7.6711e-05, 1.6987e-05, 1.6987e-05, 8.1003e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 7.6711e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 4.4084e-04, 7.6711e-05, 7.6711e-05,
          1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05,
          1.6987e-05, 1.6987e-05, 7.6711e-05, 7.6711e-05, 1.6987e-05, 7.6711e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 7.6711e-05, 1.5199e-04, 1.5199e-04, 1.6987e-05,
          4.4084e-04, 7.6711e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.5199e-04, 1.6987e-05, 1.5199e-04, 1.6987e-05, 7.6711e-05,
          7.6711e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.5199e-04, 9.2208e-05,
          9.7656e-04, 1.6987e-05, 1.5199e-04, 1.5199e-04, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 7.6711e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 5.0783e-04, 1.5199e-04, 9.2208e-05, 1.6987e-05, 7.6711e-05,
          7.6711e-05, 1.6987e-05, 9.2208e-05, 1.5199e-04, 1.6987e-05, 7.6711e-05,
          9.2208e-05, 9.2208e-05, 1.6987e-05, 5.1594e-04, 1.6987e-05, 1.6987e-05,
          4.5633e-04, 7.6711e-05, 9.2208e-05, 9.2208e-05, 1.6987e-05, 1.6987e-05,
          9.2208e-05, 1.5199e-04, 1.6987e-05, 1.6987e-05, 1.6987e-05, 9.2208e-05,
          1.6987e-05, 1.3208e-03, 9.2208e-05, 1.6987e-05, 1.6987e-05, 9.2208e-05,
          9.2208e-05, 1.6987e-05, 9.8801e-04, 1.6987e-05, 3.8099e-04, 1.5199e-04,
          1.6987e-05, 7.6711e-05, 1.5199e-04, 7.6711e-05, 1.0519e-03, 9.2208e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.5199e-04, 1.5199e-04,
          1.5199e-04, 2.8682e-04, 1.4639e-03, 1.6987e-05, 9.2208e-05, 4.5633e-04,
          9.8801e-04, 4.2987e-04, 4.5633e-04, 1.6987e-05, 3.4642e-04, 1.6987e-05,
          5.6744e-04, 9.2208e-05, 1.6987e-05, 1.6861e-03, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.5199e-04, 1.6987e-05,
          9.8801e-04, 1.5199e-04, 1.4639e-03, 1.6987e-05, 1.6987e-05, 5.0783e-04,
          9.8801e-04, 1.6987e-05, 1.5199e-04, 1.6987e-05, 4.4084e-04, 7.8535e-04,
          9.2208e-05, 2.1725e-03, 1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05, 1.6987e-05, 1.6987e-05, 9.2208e-05, 1.6987e-05,
          1.6987e-05, 9.2208e-05, 1.6987e-05, 5.1403e-04, 1.6987e-05, 1.6987e-05,
          1.6987e-05, 1.6987e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1368e-02, 6.8359e-02, 3.2623e-02, 3.2867e-02, 9.5154e-02,
          6.1035e-02, 2.0004e-02, 3.2379e-02, 1.0567e-02, 8.5297e-03, 8.8654e-03,
          4.5433e-03, 8.0032e-03, 8.3771e-03, 7.3929e-03, 1.7334e-02, 1.9241e-02,
          7.2021e-03, 4.5509e-03, 4.7264e-03, 4.0359e-03, 2.9488e-03, 6.7215e-03,
          8.3084e-03, 6.9542e-03, 1.6830e-02, 1.5320e-02, 1.3527e-02, 1.4175e-02,
          9.5062e-03, 4.8340e-02, 9.1934e-03, 1.3161e-02, 4.7455e-03, 4.1412e-02,
          2.0004e-02, 9.1858e-03, 7.4196e-03, 8.0948e-03, 6.5727e-03, 7.8964e-03,
          4.5433e-03, 3.4218e-03, 2.8973e-03, 3.5095e-03, 7.1716e-03, 9.1629e-03,
          3.5572e-03, 3.9062e-03, 1.3023e-02, 1.6235e-01, 1.3113e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 5.9724e-05, 4.8399e-05,
          2.2769e-04, 9.5010e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 4.8399e-05, 4.8399e-05, 4.8399e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.1081e-04, 9.5010e-05, 2.2769e-04,
          1.1081e-04, 1.3113e-05, 3.8886e-04, 4.8399e-05, 5.9724e-05, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.1081e-04, 1.1081e-04, 4.8935e-05, 1.3113e-05, 4.8399e-05, 4.8399e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 2.2769e-04, 9.5010e-05,
          1.3113e-05, 5.9724e-05, 2.2089e-04, 4.8399e-05, 1.3113e-05, 1.3113e-05,
          4.8399e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05, 4.8399e-05,
          1.3113e-05, 4.8399e-05, 4.8399e-05, 4.8399e-05, 1.3113e-05, 1.3113e-05,
          2.9016e-04, 9.5546e-05, 1.3113e-05, 9.5010e-05, 1.5736e-04, 2.2089e-04,
          1.3113e-05, 5.9986e-04, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05,
          4.8399e-05, 1.3113e-05, 4.8399e-05, 1.3113e-05, 4.8399e-05, 4.8399e-05,
          1.3113e-05, 4.8399e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.7428e-04,
          1.1081e-04, 3.8528e-04, 1.1081e-04, 5.9986e-04, 1.9240e-04, 2.2769e-04,
          1.3113e-05, 1.3113e-05, 1.1081e-04, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.1081e-04, 2.0134e-04, 4.8399e-05, 1.3113e-05, 1.1081e-04, 1.3113e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.3113e-05, 4.8399e-05, 4.8399e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05,
          1.1081e-04, 4.8399e-05, 1.3113e-05, 1.3113e-05, 5.0902e-05, 4.8399e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 2.4259e-04, 1.1134e-04, 4.8399e-05,
          1.3113e-05, 1.3113e-05, 4.8399e-05, 4.4441e-04, 2.8324e-04, 1.3113e-05,
          3.5405e-04, 4.0913e-04, 4.3178e-04, 1.3113e-05, 1.1081e-04, 1.1081e-04,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.7667e-04, 4.8399e-05, 1.7667e-04,
          1.3113e-05, 4.8399e-05, 1.3113e-05, 1.1081e-04, 1.3113e-05, 1.1081e-04,
          1.3113e-05, 1.7428e-04, 3.4666e-04, 6.2704e-04, 2.9016e-04, 4.8399e-05,
          2.8372e-04, 4.3869e-04, 4.3178e-04, 5.9724e-05, 1.3113e-05, 1.8108e-04,
          1.3113e-05, 1.1081e-04, 4.8399e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05,
          4.8399e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05, 3.9768e-04,
          1.7428e-04, 2.8324e-04, 1.3888e-04, 6.2704e-04, 7.5519e-05, 1.2219e-04,
          5.1451e-04, 4.3178e-04, 2.2089e-04, 1.3113e-05, 1.3113e-05, 2.0134e-04,
          1.3113e-05, 4.8399e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05, 2.4343e-04, 1.3113e-05,
          7.7534e-04, 1.7464e-04, 2.9016e-04, 5.1117e-04, 2.2411e-04, 1.9240e-04,
          1.9240e-04, 1.3113e-05, 4.3178e-04, 4.1580e-04, 1.3113e-05, 1.7667e-04,
          1.7464e-04, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05, 1.3113e-05,
          4.8399e-05, 1.3113e-05, 1.1081e-04, 2.4259e-04, 1.3113e-05, 1.3113e-05,
          1.7428e-04, 3.0017e-04, 5.1117e-04, 4.8399e-05, 3.8528e-04, 2.2089e-04,
          1.7428e-04, 2.2411e-04, 1.3113e-05, 4.8399e-05, 1.3113e-05, 2.3651e-04,
          1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3113e-05, 4.8399e-05,
          1.1081e-04, 1.3113e-05, 3.0017e-04, 4.8399e-05, 1.3113e-05, 2.3651e-04,
          9.3937e-04, 1.1081e-04, 5.7364e-04, 2.5940e-04, 1.7428e-04, 1.3113e-05,
          2.8658e-04, 1.8108e-04, 1.7416e-04, 1.3113e-05, 1.3113e-05, 2.3711e-04,
          1.3888e-04, 4.8935e-05, 2.3913e-04, 4.8399e-05, 1.8024e-04, 1.3888e-04,
          1.3113e-05, 1.3113e-05, 2.2411e-04, 3.4666e-04, 5.2977e-04, 4.1652e-04,
          4.8399e-05, 4.8399e-05, 4.1580e-04, 3.1281e-04, 1.3113e-05, 5.1117e-04,
          5.4646e-04, 2.3651e-04, 1.3888e-04, 1.3113e-05, 1.3113e-05, 4.8399e-05,
          1.3113e-05, 2.0730e-04, 3.0589e-04, 1.8024e-04, 2.3711e-04, 3.3307e-04,
          7.5519e-05, 7.5519e-05, 1.3113e-05, 1.7416e-04, 1.3113e-05, 1.3113e-05,
          4.5347e-04, 5.7697e-04, 1.3113e-05, 2.4259e-04, 1.3113e-05, 1.3113e-05,
          2.0134e-04, 1.3888e-04, 1.3113e-05, 1.3113e-05, 1.3113e-05, 1.3888e-04,
          7.5519e-05, 5.0116e-04, 1.9150e-03, 1.3113e-05, 1.3888e-04, 1.0996e-03,
          7.5519e-05, 3.0017e-04, 6.6280e-04, 1.8024e-04, 5.2977e-04, 5.7936e-04,
          2.4433e-03, 2.0730e-04, 3.0613e-04, 3.0017e-04, 4.3178e-04, 1.3113e-05,
          7.5519e-05, 1.3113e-05, 1.1081e-04, 1.3113e-05, 7.5519e-05, 4.8399e-05,
          1.3113e-05, 1.1081e-04, 4.2033e-04, 7.5519e-05, 1.3113e-05, 7.7963e-04,
          7.5519e-05, 1.3113e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.3771e-03, 7.5073e-02, 2.8946e-02, 3.0075e-02, 8.0872e-02,
          6.5796e-02, 1.6006e-02, 3.8086e-02, 1.0689e-02, 9.9258e-03, 1.0902e-02,
          5.4207e-03, 7.8812e-03, 6.2981e-03, 6.1798e-03, 2.3773e-02, 1.3313e-02,
          7.4158e-03, 6.0616e-03, 2.1038e-03, 1.9703e-03, 1.7252e-03, 6.9542e-03,
          1.1078e-02, 5.2834e-03, 1.4610e-02, 1.6647e-02, 6.9160e-03, 1.3031e-02,
          8.9264e-03, 6.7810e-02, 6.0005e-03, 1.0422e-02, 5.8212e-03, 3.4821e-02,
          1.2085e-02, 5.2261e-03, 6.6986e-03, 6.9313e-03, 3.5534e-03, 5.5847e-03,
          2.9507e-03, 2.1362e-03, 2.1343e-03, 1.8597e-03, 7.9803e-03, 6.1531e-03,
          2.3899e-03, 1.9417e-03, 9.4528e-03, 8.7952e-02, 1.1163e-01, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 4.0054e-05, 1.8716e-05, 1.8716e-05, 5.7280e-05, 1.3053e-04,
          1.3053e-04, 7.8619e-05, 1.8716e-05, 1.8716e-05, 1.4710e-04, 5.7280e-05,
          1.8716e-05, 4.0054e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 6.7711e-05, 4.0054e-05, 6.7711e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 4.0054e-05, 7.8619e-05, 1.3053e-04,
          4.0054e-05, 4.0054e-05, 4.3273e-04, 4.0054e-05, 5.7280e-05, 5.7280e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          4.0054e-05, 1.2362e-04, 6.7115e-05, 1.8716e-05, 4.0054e-05, 4.0054e-05,
          1.8716e-05, 1.4710e-04, 1.8716e-05, 5.7280e-05, 1.1909e-04, 1.0568e-04,
          4.0054e-05, 1.8573e-04, 1.8573e-04, 9.2030e-05, 5.7280e-05, 1.8716e-05,
          4.0054e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 4.0054e-05, 6.7115e-05,
          1.8716e-05, 6.7115e-05, 6.7115e-05, 4.0054e-05, 1.8716e-05, 1.8716e-05,
          3.2425e-04, 1.0568e-04, 1.8716e-05, 1.0926e-04, 1.0568e-04, 3.5930e-04,
          4.0054e-05, 4.3273e-04, 1.9240e-04, 1.8716e-05, 1.8716e-05, 3.6979e-04,
          6.7711e-05, 1.8716e-05, 4.0054e-05, 4.5776e-05, 6.7711e-05, 4.0054e-05,
          6.7115e-05, 4.0054e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8573e-04,
          6.7115e-05, 3.4785e-04, 4.0054e-05, 4.5967e-04, 1.3053e-04, 1.3053e-04,
          4.0054e-05, 1.8716e-05, 1.2362e-04, 1.8716e-05, 1.8716e-05, 4.5776e-05,
          6.1750e-04, 4.5776e-05, 4.0054e-05, 4.0054e-05, 6.7115e-05, 4.0054e-05,
          1.0222e-04, 4.5776e-05, 1.8716e-05, 1.4710e-04, 8.4341e-05, 1.8716e-05,
          9.2030e-05, 9.2030e-05, 9.2030e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.2362e-04, 9.2030e-05, 1.8716e-05, 4.5776e-05, 6.8307e-05, 2.2852e-04,
          1.8716e-05, 4.5776e-05, 4.0054e-05, 2.6393e-04, 1.5068e-04, 4.0054e-05,
          1.8716e-05, 6.7115e-05, 9.2030e-05, 3.6955e-04, 1.0568e-04, 9.2030e-05,
          2.8610e-04, 8.4341e-05, 3.8648e-04, 1.8716e-05, 1.5068e-04, 1.2362e-04,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 3.7098e-04, 4.0054e-05, 2.5678e-04,
          4.5776e-05, 6.7115e-05, 1.8716e-05, 1.2362e-04, 4.0054e-05, 4.0054e-05,
          4.0054e-05, 1.7416e-04, 2.1279e-04, 1.1909e-04, 3.2759e-04, 4.0054e-05,
          1.5759e-04, 4.4489e-04, 4.3845e-04, 5.3644e-04, 1.8716e-05, 7.5579e-04,
          1.8716e-05, 6.7115e-05, 2.7227e-04, 4.0054e-05, 6.7115e-05, 4.6086e-04,
          4.5776e-05, 4.5776e-05, 4.5776e-05, 1.8716e-05, 4.0054e-05, 1.9550e-04,
          2.1279e-04, 2.4116e-04, 4.5776e-05, 4.2129e-04, 4.5776e-05, 2.4116e-04,
          2.8610e-04, 3.8648e-04, 1.8573e-04, 3.2091e-04, 1.8716e-05, 4.3201e-04,
          4.5776e-05, 4.6086e-04, 4.5776e-05, 4.5776e-05, 4.5776e-05, 6.7711e-05,
          1.5068e-04, 1.8716e-05, 4.0054e-05, 9.4414e-04, 3.9458e-04, 5.0497e-04,
          4.4656e-04, 1.1963e-04, 3.5477e-04, 3.4785e-04, 5.2214e-04, 1.5759e-04,
          4.5967e-04, 2.4533e-04, 4.3845e-04, 1.5759e-04, 4.5776e-05, 4.6206e-04,
          6.7711e-05, 4.5776e-05, 4.5776e-05, 1.8716e-05, 3.3903e-04, 1.2934e-04,
          4.3368e-04, 6.7115e-05, 1.5068e-04, 2.6393e-04, 1.7416e-04, 1.1909e-04,
          1.7416e-04, 1.7416e-04, 3.4785e-04, 2.0266e-04, 4.3154e-04, 3.8648e-04,
          1.4710e-04, 2.6560e-04, 1.8716e-05, 4.0054e-05, 4.5776e-05, 1.5068e-04,
          1.2362e-04, 2.5558e-04, 4.5776e-05, 4.0054e-05, 1.8716e-05, 4.0054e-05,
          1.5068e-04, 1.5068e-04, 1.7416e-04, 4.0054e-05, 4.5776e-05, 7.3004e-04,
          5.8174e-04, 7.0095e-04, 4.2129e-04, 2.9278e-04, 1.4710e-04, 1.7416e-04,
          2.6560e-04, 1.1909e-04, 6.7115e-05, 4.5776e-05, 6.7115e-05, 4.5347e-04,
          1.2934e-04, 9.7132e-04, 6.1131e-04, 6.7115e-05, 1.8716e-05, 4.5776e-05,
          3.5596e-04, 1.8716e-05, 1.8716e-05, 2.1279e-04, 3.3593e-04, 4.6039e-04,
          4.0054e-05, 2.0552e-04, 2.0266e-04, 1.1909e-04, 4.5776e-05, 3.9983e-04,
          4.5967e-04, 4.5347e-04, 4.5776e-05, 2.3413e-04, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 3.9577e-04, 3.3593e-04, 5.0592e-04, 2.9087e-04,
          1.8716e-05, 1.8716e-05, 4.5776e-05, 6.7115e-05, 1.8716e-05, 4.0054e-05,
          6.7949e-04, 2.8610e-04, 1.8716e-05, 5.4836e-04, 1.8716e-05, 1.8716e-05,
          3.4857e-04, 3.4761e-04, 4.5776e-05, 1.8716e-05, 1.8716e-05, 4.5776e-05,
          1.8716e-05, 9.7179e-04, 8.4114e-04, 4.5776e-05, 6.7115e-05, 1.0595e-03,
          4.5776e-05, 2.8777e-04, 1.1835e-03, 1.8716e-05, 9.3794e-04, 5.9319e-04,
          2.2182e-03, 1.8716e-05, 3.3593e-04, 3.1447e-04, 3.1447e-04, 1.8716e-05,
          1.2362e-04, 1.0222e-04, 1.2362e-04, 1.0222e-04, 4.1175e-04, 4.0054e-05,
          1.8716e-05, 1.2362e-04, 1.0222e-04, 6.0320e-04, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.0222e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.3079e-03, 6.8298e-02, 2.4719e-02, 2.9999e-02, 7.2510e-02,
          6.3965e-02, 1.6968e-02, 3.7292e-02, 1.7639e-02, 1.4473e-02, 1.2802e-02,
          7.6675e-03, 1.0162e-02, 4.6806e-03, 1.1993e-02, 2.8015e-02, 1.5701e-02,
          7.5493e-03, 1.1520e-02, 1.9226e-03, 2.0065e-03, 1.7595e-03, 8.5220e-03,
          1.4771e-02, 9.7198e-03, 1.4915e-02, 1.6327e-02, 4.6501e-03, 1.3779e-02,
          6.1417e-03, 6.6467e-02, 5.1384e-03, 1.8585e-02, 1.0300e-02, 4.2114e-02,
          1.7075e-02, 4.4403e-03, 1.3290e-02, 1.3535e-02, 2.6779e-03, 9.1248e-03,
          2.8038e-03, 2.4948e-03, 2.0142e-03, 2.9163e-03, 1.0254e-02, 1.2833e-02,
          2.3861e-03, 2.1591e-03, 9.9106e-03, 5.5634e-02, 5.1086e-02, 5.2673e-02,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 7.3969e-05,
          7.3969e-05, 1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 7.3969e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 2.2566e-04,
          7.9095e-05, 2.9397e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.4246e-04,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 7.3969e-05, 2.2566e-04, 1.0610e-05, 7.9095e-05,
          1.0610e-05, 1.0610e-05, 2.8729e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.4246e-04, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 2.8729e-04, 1.4246e-04, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 2.2566e-04, 1.0610e-05, 7.3969e-05, 2.8729e-04,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05, 7.3969e-05,
          1.0610e-05, 7.9095e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.4246e-04,
          1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.3969e-05, 1.4246e-04,
          2.2566e-04, 7.9095e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.9095e-05,
          1.0610e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.4246e-04, 1.4246e-04, 1.0610e-05,
          7.3969e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0328e-03, 1.4246e-04, 1.0610e-05, 7.3969e-05,
          7.3969e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 7.9095e-05, 1.0610e-05,
          7.3969e-05, 1.0610e-05, 7.9095e-05, 7.9095e-05, 1.0610e-05, 1.0610e-05,
          2.2566e-04, 1.0610e-05, 2.2566e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 7.9095e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.9095e-05, 1.0610e-05, 2.9397e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 7.9095e-05, 8.2064e-04, 7.9095e-05, 1.0610e-05, 1.0610e-05,
          8.2064e-04, 1.0610e-05, 7.9095e-05, 1.0610e-05, 1.0610e-05, 3.5071e-04,
          1.0610e-05, 7.9095e-05, 3.5071e-04, 7.3969e-05, 1.0610e-05, 1.0610e-05,
          7.9095e-05, 1.0610e-05, 1.0610e-05, 1.0538e-03, 6.2990e-04, 3.5071e-04,
          1.0610e-05, 1.0610e-05, 2.8729e-04, 7.3969e-05, 1.0610e-05, 1.0357e-03,
          1.5869e-03, 3.5071e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 7.9095e-05, 7.3969e-05, 1.0610e-05, 3.5071e-04, 7.9095e-05,
          7.9095e-05, 7.9095e-05, 1.0610e-05, 7.3969e-05, 1.0610e-05, 1.0610e-05,
          1.0610e-05, 9.0837e-04, 1.0610e-05, 1.4246e-04, 1.0610e-05, 1.0610e-05,
          2.8729e-04, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05, 1.0610e-05,
          7.9095e-05, 1.4246e-04, 1.4246e-04, 1.0610e-05, 1.0610e-05, 1.4162e-03,
          7.9095e-05, 1.0538e-03, 9.0170e-04, 1.0610e-05, 1.0610e-05, 2.9397e-04,
          1.0538e-03, 7.9095e-05, 1.0610e-05, 5.0259e-04, 5.0259e-04, 1.0610e-05,
          7.9095e-05, 1.0610e-05, 7.9095e-05, 1.0610e-05, 4.3750e-04, 7.3969e-05,
          1.0610e-05, 7.9095e-05, 7.9095e-05, 4.3750e-04, 1.0610e-05, 4.3750e-04,
          7.9095e-05, 7.9095e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3123e-02, 6.6040e-02, 3.3508e-02, 2.7283e-02, 1.2201e-01,
          5.1727e-02, 2.0279e-02, 2.7405e-02, 7.7515e-03, 4.6234e-03, 6.9923e-03,
          3.2673e-03, 4.2496e-03, 7.4730e-03, 4.3068e-03, 1.0895e-02, 6.4545e-03,
          4.0016e-03, 3.5839e-03, 4.3564e-03, 3.3073e-03, 4.1656e-03, 4.1237e-03,
          6.6833e-03, 5.4207e-03, 1.6495e-02, 9.2239e-03, 1.2756e-02, 6.0120e-03,
          7.9269e-03, 2.8259e-02, 4.3945e-03, 5.3787e-03, 2.4529e-03, 1.7273e-02,
          1.0956e-02, 1.0025e-02, 3.1834e-03, 3.6716e-03, 6.0120e-03, 3.7003e-03,
          4.3945e-03, 3.3855e-03, 4.2610e-03, 2.3060e-03, 3.8109e-03, 4.2686e-03,
          3.8013e-03, 5.6801e-03, 1.9455e-02, 8.1787e-02, 5.4138e-02, 1.0422e-02,
          1.6565e-01, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 8.1003e-05,
          3.4869e-05, 3.4869e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 3.4869e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 3.4869e-05, 3.4869e-05, 1.3292e-05,
          3.4869e-05, 1.3292e-05, 1.3292e-05, 3.4869e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          3.4869e-05, 3.4869e-05, 3.5286e-05, 1.3292e-05, 8.1003e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 3.4869e-05, 3.4869e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 8.1003e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 8.1003e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 8.1003e-05, 3.4869e-05, 1.3292e-05,
          3.4869e-05, 3.5286e-05, 1.3292e-05, 3.4869e-05, 3.4869e-05, 1.3292e-05,
          3.4869e-05, 3.4869e-05, 3.4869e-05, 1.3292e-05, 1.3292e-05, 1.0258e-04,
          1.6928e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 8.1003e-05,
          1.3292e-05, 8.1003e-05, 1.3292e-05, 3.4869e-05, 1.3292e-05, 5.8460e-04,
          3.4869e-05, 3.0684e-04, 1.3423e-04, 1.3292e-05, 1.3292e-05, 1.0258e-04,
          1.3292e-05, 1.3292e-05, 1.8346e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.8346e-04, 3.4869e-05, 8.1003e-05, 8.1003e-05, 1.0300e-04, 8.1003e-05,
          3.4869e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 3.4869e-05, 1.3292e-05,
          1.3292e-05, 8.1003e-05, 8.1420e-05, 3.4869e-05, 3.4869e-05, 1.3292e-05,
          3.4869e-05, 3.4869e-05, 3.4869e-05, 1.7345e-05, 3.5286e-05, 8.1003e-05,
          1.1533e-04, 1.3292e-05, 8.1003e-05, 1.8346e-04, 1.1575e-04, 8.1003e-05,
          1.3292e-05, 3.8922e-05, 1.3292e-05, 3.5286e-05, 3.4869e-05, 1.3292e-05,
          1.3292e-05, 3.4869e-05, 5.8460e-04, 1.3292e-05, 3.5286e-05, 1.8346e-04,
          3.4869e-05, 1.3292e-05, 1.3292e-05, 1.0663e-04, 8.1420e-05, 1.3709e-05,
          1.3292e-05, 1.0300e-04, 1.3292e-05, 1.0258e-04, 8.1003e-05, 1.0300e-04,
          3.4869e-05, 3.4869e-05, 6.0654e-04, 3.1304e-04, 1.0300e-04, 1.0258e-04,
          3.5286e-05, 3.4869e-05, 6.0654e-04, 3.4869e-05, 1.3292e-05, 8.1003e-05,
          3.4869e-05, 3.5286e-05, 3.4869e-05, 1.3292e-05, 1.3709e-05, 1.0258e-04,
          1.6928e-05, 1.3292e-05, 3.5286e-05, 3.4869e-05, 1.0300e-04, 1.0663e-04,
          3.2830e-04, 3.1328e-04, 1.3292e-05, 3.4869e-05, 3.4869e-05, 3.4869e-05,
          3.5286e-05, 3.0684e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05, 3.4869e-05,
          1.3709e-05, 3.5286e-05, 1.3709e-05, 1.3709e-05, 1.3292e-05, 3.8922e-05,
          3.5286e-05, 1.3292e-05, 3.4869e-05, 1.0300e-04, 1.0300e-04, 3.4869e-05,
          3.5286e-05, 1.3709e-05, 1.0300e-04, 3.1304e-04, 3.1304e-04, 1.3709e-05,
          1.3292e-05, 3.4869e-05, 3.0684e-04, 1.0300e-04, 1.3292e-05, 3.8922e-05,
          3.5286e-05, 1.7345e-05, 1.7345e-05, 1.3292e-05, 1.0300e-04, 3.5286e-05,
          1.0300e-04, 1.3292e-05, 1.8346e-04, 2.8253e-04, 1.3292e-05, 3.5286e-05,
          3.0684e-04, 3.5286e-05, 3.0684e-04, 1.0300e-04, 6.0654e-04, 3.5286e-05,
          1.3292e-05, 4.1223e-04, 1.1533e-04, 1.0258e-04, 1.7345e-05, 1.0300e-04,
          3.5286e-05, 3.8922e-05, 3.5286e-05, 3.4869e-05, 3.4869e-05, 3.4869e-05,
          1.0258e-04, 3.4869e-05, 3.5286e-05, 1.0300e-04, 3.4869e-05, 3.5286e-05,
          5.1546e-04, 2.1529e-04, 1.3423e-04, 1.0300e-04, 3.4869e-05, 3.4869e-05,
          1.3423e-04, 3.4869e-05, 3.0494e-04, 1.3292e-05, 3.5286e-05, 1.0663e-04,
          3.4869e-05, 1.0663e-04, 1.0663e-04, 8.1003e-05, 3.4869e-05, 3.4869e-05,
          3.4869e-05, 1.3292e-05, 3.4869e-05, 3.4869e-05, 2.7046e-03, 1.0300e-04,
          1.3423e-04, 8.1003e-05, 2.0230e-04, 4.8494e-04, 3.4869e-05, 3.2830e-04,
          3.4869e-05, 1.1533e-04, 3.5286e-05, 1.3709e-05, 1.1533e-04, 3.4869e-05,
          1.1533e-04, 2.1482e-04, 2.0194e-04, 1.3423e-04, 1.8716e-04, 1.1533e-04,
          2.1482e-04, 1.1533e-04, 3.4869e-05, 1.0258e-04, 2.1482e-04, 1.3292e-05,
          2.0194e-04, 1.0663e-04, 1.3292e-05, 4.1771e-04, 1.3292e-05, 1.3292e-05,
          3.1781e-04, 3.8922e-05, 3.5286e-05, 1.3292e-05, 1.1533e-04, 3.4869e-05,
          2.1482e-04, 1.8346e-04, 2.8300e-04, 1.3423e-04, 3.4869e-05, 4.1723e-04,
          4.1723e-04, 3.4869e-05, 1.8988e-03, 1.3423e-04, 7.8630e-04, 7.0572e-04,
          1.9293e-03, 2.1482e-04, 1.3423e-04, 6.3038e-04, 2.3735e-04, 1.3292e-05,
          3.4869e-05, 1.1533e-04, 1.1533e-04, 1.1533e-04, 3.1781e-04, 1.1533e-04,
          1.3292e-05, 3.1781e-04, 1.1533e-04, 3.4869e-05, 1.3292e-05, 3.4869e-05,
          3.4869e-05, 1.1533e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.1248e-03, 6.8665e-02, 2.7603e-02, 2.7588e-02, 8.5632e-02,
          5.4901e-02, 1.6235e-02, 3.1799e-02, 8.7357e-03, 6.4163e-03, 8.0032e-03,
          4.0436e-03, 5.5809e-03, 6.2790e-03, 5.2147e-03, 1.8387e-02, 9.5978e-03,
          5.3596e-03, 4.9744e-03, 2.2793e-03, 1.9159e-03, 1.6556e-03, 5.6801e-03,
          8.1940e-03, 4.6539e-03, 1.4496e-02, 1.1597e-02, 6.4545e-03, 9.2468e-03,
          7.6485e-03, 5.5420e-02, 5.0087e-03, 7.3738e-03, 4.3755e-03, 2.5009e-02,
          1.2024e-02, 6.1760e-03, 4.4899e-03, 5.1041e-03, 3.8528e-03, 4.1008e-03,
          2.6665e-03, 2.0256e-03, 2.1839e-03, 1.6708e-03, 5.8289e-03, 4.6844e-03,
          2.5158e-03, 2.1553e-03, 1.0201e-02, 8.2703e-02, 8.8989e-02, 1.1063e-02,
          4.3915e-02, 8.2886e-02, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 8.3685e-05, 1.9860e-04,
          1.2767e-04, 3.9697e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 4.0352e-05, 3.9697e-05, 3.9697e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 9.2328e-05, 3.9697e-05, 1.2767e-04,
          3.9697e-05, 1.2875e-05, 1.9860e-04, 3.9697e-05, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          3.9697e-05, 9.2328e-05, 4.0352e-05, 1.2875e-05, 3.9697e-05, 3.9697e-05,
          1.2875e-05, 2.2507e-04, 1.2875e-05, 1.2875e-05, 1.2767e-04, 3.9697e-05,
          3.9697e-05, 6.1274e-04, 2.2507e-04, 1.2767e-04, 1.2875e-05, 1.2875e-05,
          3.9697e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 3.9697e-05, 3.9697e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 3.9697e-05, 1.2875e-05, 1.2875e-05,
          9.2328e-05, 3.9935e-05, 1.2875e-05, 1.2875e-05, 3.9697e-05, 2.2507e-04,
          9.2328e-05, 1.2767e-04, 1.2875e-05, 1.2875e-05, 1.2875e-05, 4.9639e-04,
          3.9697e-05, 1.2875e-05, 3.9697e-05, 1.2875e-05, 4.0352e-05, 3.9697e-05,
          3.9697e-05, 3.9697e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05, 2.2507e-04,
          1.4031e-04, 2.2507e-04, 3.9697e-05, 1.9860e-04, 1.2767e-04, 1.2767e-04,
          3.9697e-05, 1.2875e-05, 6.5506e-05, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          9.2328e-05, 6.0856e-05, 3.9697e-05, 3.9697e-05, 9.2328e-05, 3.9697e-05,
          6.5506e-05, 1.2875e-05, 1.2875e-05, 2.2507e-04, 6.0856e-05, 1.2875e-05,
          1.2875e-05, 3.9697e-05, 1.2767e-04, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          9.2328e-05, 3.9697e-05, 1.2875e-05, 1.2875e-05, 4.0591e-05, 3.9697e-05,
          1.2875e-05, 1.2875e-05, 3.9697e-05, 9.2328e-05, 9.2328e-05, 3.9697e-05,
          1.2875e-05, 3.9697e-05, 3.9697e-05, 1.4031e-04, 1.4031e-04, 3.9697e-05,
          1.7595e-04, 1.8430e-04, 6.1274e-04, 1.2875e-05, 1.4031e-04, 9.2328e-05,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 1.4126e-04, 3.9697e-05, 8.8573e-05,
          1.2875e-05, 9.2328e-05, 1.2875e-05, 9.2328e-05, 3.9697e-05, 3.9697e-05,
          3.9697e-05, 2.7299e-04, 7.1335e-04, 1.4031e-04, 2.5129e-04, 3.9697e-05,
          1.5879e-04, 2.2840e-04, 6.6090e-04, 1.2875e-05, 1.2875e-05, 1.2767e-04,
          1.2875e-05, 8.7678e-05, 5.1689e-04, 1.2875e-05, 6.0856e-05, 3.9697e-05,
          1.2875e-05, 1.2875e-05, 1.1355e-04, 1.2875e-05, 3.9697e-05, 8.7678e-05,
          2.7299e-04, 1.4031e-04, 6.0856e-05, 1.7571e-04, 1.2875e-05, 2.2840e-04,
          1.7571e-04, 6.1274e-04, 1.2875e-05, 1.2875e-05, 1.2875e-05, 5.2214e-04,
          1.2875e-05, 5.6505e-04, 1.2875e-05, 6.0856e-05, 1.2875e-05, 8.8334e-05,
          6.5506e-05, 1.2875e-05, 1.2875e-05, 8.0633e-04, 1.8036e-04, 1.1355e-04,
          2.2840e-04, 1.7667e-04, 2.5153e-04, 6.6090e-04, 5.0116e-04, 1.7571e-04,
          1.0085e-04, 9.2328e-05, 7.3147e-04, 2.2840e-04, 6.0856e-05, 1.4126e-04,
          8.7917e-05, 1.2875e-05, 6.0856e-05, 1.2875e-05, 9.2328e-05, 6.5506e-05,
          5.1689e-04, 1.2875e-05, 9.2328e-05, 3.6597e-04, 1.2875e-05, 1.2875e-05,
          2.7323e-04, 2.7323e-04, 2.7323e-04, 1.8036e-04, 7.1335e-04, 1.2875e-05,
          2.2507e-04, 4.5323e-04, 1.2875e-05, 3.9697e-05, 1.2875e-05, 1.4031e-04,
          6.5506e-05, 1.4031e-04, 1.1355e-04, 1.2875e-05, 1.2875e-05, 9.2328e-05,
          9.2328e-05, 6.5506e-05, 1.1355e-04, 3.9697e-05, 6.5506e-05, 2.2840e-04,
          1.4031e-04, 9.2328e-05, 1.1355e-04, 1.4031e-04, 1.2875e-05, 6.0856e-05,
          6.5506e-05, 8.7678e-05, 1.4031e-04, 6.0856e-05, 8.7917e-05, 5.4979e-04,
          1.1355e-04, 8.0681e-04, 5.4979e-04, 3.9697e-05, 1.2875e-05, 6.0856e-05,
          1.4031e-04, 1.2875e-05, 1.2875e-05, 6.0856e-05, 1.1355e-04, 6.3705e-04,
          3.9697e-05, 3.9697e-05, 2.2840e-04, 8.7678e-05, 1.2875e-05, 7.1335e-04,
          6.6090e-04, 5.4884e-04, 1.1355e-04, 1.2875e-05, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 1.2875e-05, 1.4031e-04, 8.7678e-05, 5.4979e-04, 1.1355e-04,
          1.2875e-05, 1.2875e-05, 1.2875e-05, 8.7678e-05, 1.2875e-05, 1.2875e-05,
          9.2328e-05, 2.2840e-04, 1.2875e-05, 7.8630e-04, 1.2875e-05, 1.2875e-05,
          1.1355e-04, 6.0856e-05, 1.1355e-04, 1.2875e-05, 1.2875e-05, 6.0856e-05,
          6.5506e-05, 1.1883e-03, 6.9189e-04, 1.1355e-04, 8.7678e-05, 9.0837e-04,
          6.5506e-05, 2.7299e-04, 1.1883e-03, 1.2875e-05, 1.4031e-04, 9.5654e-04,
          1.3123e-03, 1.2875e-05, 2.2507e-04, 2.7299e-04, 2.7299e-04, 1.2875e-05,
          9.2328e-05, 6.5506e-05, 3.6597e-04, 3.3927e-04, 3.3927e-04, 9.2328e-05,
          1.2875e-05, 3.6597e-04, 6.5506e-05, 4.1223e-04, 1.2875e-05, 1.2875e-05,
          1.2875e-05, 3.3927e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.1711e-02, 6.3477e-02, 3.2867e-02, 2.8351e-02, 1.3794e-01,
          4.0771e-02, 2.8854e-02, 1.3748e-02, 4.7684e-03, 3.3569e-03, 4.6005e-03,
          2.2316e-03, 3.4828e-03, 5.9166e-03, 4.0474e-03, 6.3629e-03, 5.3062e-03,
          2.9926e-03, 2.1877e-03, 6.0883e-03, 4.0970e-03, 2.5654e-03, 2.6340e-03,
          4.1656e-03, 3.5400e-03, 2.2125e-02, 8.3542e-03, 1.2581e-02, 4.5815e-03,
          5.3368e-03, 1.7639e-02, 3.8242e-03, 5.0430e-03, 1.7748e-03, 1.4389e-02,
          1.0628e-02, 1.2901e-02, 3.1471e-03, 3.0460e-03, 7.0496e-03, 2.8763e-03,
          4.4060e-03, 3.1967e-03, 4.0627e-03, 2.0599e-03, 2.9163e-03, 3.9635e-03,
          2.4719e-03, 4.7150e-03, 2.4094e-02, 6.3110e-02, 3.2532e-02, 7.3547e-03,
          7.0435e-02, 3.3203e-02, 1.5979e-01, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          5.2631e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 6.5804e-05, 6.5804e-05, 5.2631e-05,
          1.0443e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.0443e-04, 6.5804e-05, 1.0449e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05, 1.4007e-05,
          1.0443e-04, 1.0449e-04, 1.4007e-05, 6.5863e-05, 2.2078e-04, 1.4007e-05,
          1.0443e-04, 5.2631e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2750e-05, 5.2631e-05,
          1.4007e-05, 5.2631e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          6.5804e-05, 1.4007e-05, 1.0443e-04, 1.4007e-05, 1.4007e-05, 5.2631e-05,
          1.4007e-05, 1.4007e-05, 3.6240e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.0449e-04, 6.5804e-05, 1.4007e-05, 1.4007e-05, 1.0443e-04, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 5.2631e-05, 5.2631e-05, 6.5804e-05, 6.5804e-05, 1.4007e-05,
          1.0443e-04, 5.2631e-05, 1.4007e-05, 1.4007e-05, 5.3942e-05, 5.2631e-05,
          6.5804e-05, 1.4007e-05, 1.4007e-05, 5.2643e-04, 1.0449e-04, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 6.5804e-05, 1.0443e-04, 1.4007e-05,
          1.4126e-05, 1.8215e-04, 1.4007e-05, 1.4007e-05, 1.0443e-04, 1.0449e-04,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.0574e-04, 5.2631e-05, 5.2750e-05,
          1.4007e-05, 1.0443e-04, 1.4007e-05, 5.2631e-05, 1.4007e-05, 5.2631e-05,
          1.4007e-05, 1.4007e-05, 1.8158e-03, 1.0443e-04, 5.2631e-05, 5.2631e-05,
          1.0449e-04, 5.2631e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05,
          1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05, 5.2631e-05,
          1.4007e-05, 6.5863e-05, 1.4007e-05, 1.4007e-05, 6.5804e-05, 1.4007e-05,
          1.4126e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 5.2750e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.5318e-05,
          1.4007e-05, 1.4007e-05, 5.2631e-05, 5.2631e-05, 5.2631e-05, 1.4007e-05,
          1.0449e-04, 5.3942e-05, 1.0449e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 6.5804e-05, 1.4007e-05, 1.0443e-04, 1.4007e-05, 1.0574e-04,
          5.2750e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.0449e-04, 1.4007e-05,
          5.2750e-05, 1.4007e-05, 1.0449e-04, 5.2643e-04, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 5.2631e-05, 6.5804e-05, 5.2631e-05, 1.4007e-05, 1.0449e-04,
          1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 5.2631e-05,
          1.0443e-04, 1.4007e-05, 1.4007e-05, 5.2631e-05, 1.4007e-05, 6.5804e-05,
          6.5804e-05, 3.6240e-04, 6.5804e-05, 1.0443e-04, 1.4007e-05, 1.4007e-05,
          1.0443e-04, 5.2631e-05, 1.0443e-04, 1.4007e-05, 1.4126e-05, 1.0449e-04,
          1.4007e-05, 1.0449e-04, 1.0574e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          6.5804e-05, 1.4007e-05, 1.4007e-05, 1.4007e-05, 2.2984e-04, 1.0449e-04,
          1.4007e-05, 5.2631e-05, 1.0449e-04, 2.6846e-04, 1.4007e-05, 1.4007e-05,
          1.4007e-05, 6.5804e-05, 1.4007e-05, 1.4007e-05, 1.0443e-04, 1.0443e-04,
          1.4007e-05, 1.4007e-05, 2.1672e-04, 1.7810e-04, 1.0449e-04, 1.4007e-05,
          1.0443e-04, 1.0443e-04, 1.4007e-05, 5.2631e-05, 6.5804e-05, 1.4007e-05,
          2.1672e-04, 1.0449e-04, 1.4007e-05, 2.2984e-04, 1.4007e-05, 1.4007e-05,
          6.5804e-05, 1.4126e-05, 1.4007e-05, 1.4007e-05, 1.0443e-04, 1.4007e-05,
          6.5804e-05, 1.0449e-04, 5.2643e-04, 1.4007e-05, 1.4007e-05, 2.2984e-04,
          4.8780e-04, 1.4007e-05, 1.0449e-04, 2.1672e-04, 2.6846e-04, 1.7810e-04,
          1.0443e-04, 6.5804e-05, 1.7810e-04, 1.4007e-05, 1.4007e-05, 1.4007e-05,
          6.5804e-05, 6.5804e-05, 3.2377e-04, 6.5804e-05, 6.5804e-05, 6.5804e-05,
          1.4007e-05, 3.2377e-04, 6.5804e-05, 6.5804e-05, 1.4007e-05, 3.2377e-04,
          6.5804e-05, 6.5804e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3222e-02, 6.3904e-02, 2.7390e-02, 2.5452e-02, 1.1102e-01,
          6.1859e-02, 2.2186e-02, 5.6915e-03, 7.2403e-03, 1.6693e-02, 2.0523e-02,
          1.0986e-02, 6.6299e-03, 6.1073e-03, 7.9422e-03, 3.3474e-03, 8.2092e-03,
          3.7251e-03, 2.1019e-03, 5.4817e-03, 4.1122e-03, 3.1185e-03, 3.1471e-03,
          1.9867e-02, 9.8419e-03, 1.5732e-02, 7.9956e-03, 9.7351e-03, 6.5308e-03,
          5.8670e-03, 7.2556e-03, 4.6158e-03, 1.0979e-02, 5.0354e-03, 8.3389e-03,
          4.6021e-02, 4.4739e-02, 6.7558e-03, 6.1951e-03, 6.5842e-03, 1.4946e-02,
          1.4267e-02, 1.0147e-02, 1.0139e-02, 9.0942e-03, 4.2992e-03, 1.0216e-02,
          3.5515e-03, 4.5586e-03, 1.6922e-02, 1.9852e-02, 1.1597e-02, 5.9776e-03,
          1.5167e-02, 1.0361e-02, 1.4572e-02, 1.2573e-01, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 8.1718e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.0884e-04, 1.4901e-05, 4.2021e-05, 4.2021e-05, 1.4901e-05,
          4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 9.3937e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 8.1718e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 1.4901e-05, 4.2021e-05, 4.2021e-05, 1.4901e-05,
          4.2021e-05, 5.5265e-04, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.0884e-04,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 8.1718e-05, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.8787e-04,
          1.4901e-05, 1.4901e-05, 1.7917e-04, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.0884e-04, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.7917e-04, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.7917e-04, 1.4901e-05, 1.4901e-05, 1.9813e-04, 4.2021e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 4.2021e-05, 1.4901e-05, 8.1718e-05, 4.2021e-05, 1.0884e-04,
          1.0884e-04, 1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.0884e-04, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.0884e-04,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.7374e-04,
          1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.0884e-04, 1.0884e-04,
          1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 4.2021e-05, 1.4901e-05,
          4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.0884e-04, 1.7917e-04, 1.7917e-04, 1.4901e-05,
          4.2021e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.1241e-04, 4.8614e-04, 2.5821e-04, 1.4901e-05, 4.2021e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.0884e-04, 4.2021e-05,
          8.1718e-05, 1.4901e-05, 1.0884e-04, 1.7917e-04, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.1724e-04, 2.5439e-04, 1.0884e-04, 4.2021e-05, 4.2021e-05,
          1.4901e-05, 1.3137e-04, 1.4901e-05, 1.0884e-04, 1.4901e-05, 1.4901e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 1.0884e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 1.1241e-04, 2.7680e-04, 1.0884e-04, 1.4901e-05, 1.4901e-05,
          2.7680e-04, 7.4196e-04, 1.9813e-04, 1.4901e-05, 1.4901e-05, 1.9896e-04,
          1.4901e-05, 1.5583e-03, 2.7108e-04, 1.4901e-05, 1.4901e-05, 2.0647e-04,
          4.2021e-05, 1.4901e-05, 3.4356e-04, 6.8903e-04, 4.2021e-05, 3.4833e-04,
          1.1724e-04, 1.4901e-05, 3.4451e-04, 1.8752e-04, 4.2021e-05, 2.5444e-03,
          1.2083e-03, 1.0884e-04, 4.2021e-05, 1.4901e-05, 1.7917e-04, 1.0884e-04,
          8.1718e-05, 2.7680e-04, 1.4901e-05, 1.8752e-04, 3.4451e-04, 4.2021e-05,
          3.4356e-04, 3.4356e-04, 1.1724e-04, 1.1241e-04, 2.7680e-04, 1.4901e-05,
          1.8752e-04, 7.1335e-04, 8.1718e-05, 2.7680e-04, 1.4901e-05, 9.0122e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 1.4901e-05, 1.7917e-04, 1.3137e-04,
          9.0122e-04, 3.4451e-04, 1.1241e-04, 1.1724e-04, 7.6437e-04, 2.7680e-04,
          3.4356e-04, 7.6437e-04, 9.8133e-04, 2.7680e-04, 4.2021e-05, 1.8752e-04,
          1.1997e-03, 8.3447e-04, 9.0122e-05, 1.1997e-03, 1.1997e-03, 1.4901e-05,
          1.1241e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          1.4901e-05, 4.2021e-05, 2.0158e-04, 4.9734e-04, 4.9734e-04, 2.0158e-04,
          1.1241e-04, 4.2021e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.9645e-03, 6.4270e-02, 2.6062e-02, 2.4506e-02, 9.5825e-02,
          6.3232e-02, 1.6373e-02, 7.7820e-03, 6.8626e-03, 9.8648e-03, 1.6159e-02,
          7.8011e-03, 4.5128e-03, 4.6883e-03, 5.9853e-03, 5.1689e-03, 5.3673e-03,
          3.0937e-03, 2.6855e-03, 2.3270e-03, 2.1305e-03, 1.7862e-03, 3.2196e-03,
          1.7883e-02, 7.9041e-03, 1.4122e-02, 7.9803e-03, 4.4746e-03, 5.0507e-03,
          4.6310e-03, 1.0338e-02, 2.9755e-03, 6.8398e-03, 3.4561e-03, 1.1452e-02,
          2.1469e-02, 3.0380e-02, 3.7537e-03, 4.1580e-03, 3.9177e-03, 7.2403e-03,
          5.9547e-03, 4.4937e-03, 4.8561e-03, 3.3112e-03, 3.4790e-03, 4.8637e-03,
          2.3155e-03, 2.8839e-03, 1.7975e-02, 2.3544e-02, 1.4557e-02, 6.3477e-03,
          1.8082e-02, 1.1841e-02, 1.3229e-02, 4.1473e-02, 2.1912e-01, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.3995e-04,
          4.0770e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 7.8321e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 4.0770e-05, 1.6510e-05,
          4.0770e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          4.0770e-05, 1.2362e-04, 4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05, 4.0770e-05, 4.0770e-05,
          1.6510e-05, 1.6510e-05, 4.1533e-04, 7.8321e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 7.8321e-05, 7.8321e-05, 4.0770e-05,
          1.6510e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05,
          9.6142e-05, 4.0770e-05, 1.6510e-05, 4.0770e-05, 4.0770e-05, 1.6510e-05,
          4.0770e-05, 2.2602e-04, 2.6369e-04, 1.6510e-05, 1.6510e-05, 1.0288e-04,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 7.8321e-05,
          4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          4.0770e-05, 4.1533e-04, 1.7893e-04, 1.6510e-05, 1.3995e-04, 2.2602e-04,
          1.6510e-05, 1.6510e-05, 1.8573e-04, 1.6510e-05, 1.6510e-05, 4.0770e-05,
          4.0770e-05, 4.0770e-05, 7.8321e-05, 4.0770e-05, 1.0288e-04, 1.6510e-05,
          4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 7.8321e-05, 7.8619e-05, 1.0258e-04, 2.6369e-04, 2.3949e-04,
          1.7893e-04, 4.0770e-05, 1.6510e-05, 4.0770e-05, 1.6868e-05, 1.0258e-04,
          1.8549e-04, 1.6510e-05, 1.6510e-05, 2.4104e-04, 1.7893e-04, 4.0770e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 4.0770e-05, 1.6510e-05,
          1.6510e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 2.4104e-04,
          7.8321e-05, 1.6510e-05, 1.6510e-05, 4.2081e-05, 1.0288e-04, 1.6510e-05,
          4.0770e-05, 1.0288e-04, 1.6510e-05, 4.0770e-05, 4.0770e-05, 1.0288e-04,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 9.6142e-05, 2.2638e-04, 1.6510e-05,
          4.0770e-05, 4.0770e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05, 2.0170e-04,
          1.6510e-05, 4.0770e-05, 4.0770e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05,
          1.7464e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 1.0288e-04, 1.0383e-04,
          1.6510e-05, 4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05,
          4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05, 1.6510e-05,
          1.6510e-05, 1.2362e-04, 1.6510e-05, 4.0770e-05, 1.6510e-05, 4.2081e-05,
          4.0770e-05, 1.6510e-05, 1.5795e-04, 1.0288e-04, 1.0288e-04, 1.6510e-05,
          1.0258e-04, 7.8619e-05, 2.2638e-04, 1.6510e-05, 4.0770e-05, 1.6427e-04,
          4.0770e-05, 9.6142e-05, 4.3964e-04, 2.2638e-04, 1.6510e-05, 4.2081e-05,
          4.1127e-05, 1.7762e-05, 1.6510e-05, 1.6510e-05, 1.0288e-04, 9.6142e-05,
          4.1127e-05, 2.2554e-04, 1.8573e-04, 5.7650e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 2.2602e-04, 4.0770e-05, 1.6570e-05,
          1.6570e-05, 2.4080e-04, 2.4080e-04, 1.5831e-04, 4.0770e-05, 4.1127e-05,
          4.1127e-05, 4.2081e-05, 4.1127e-05, 1.0258e-04, 2.3949e-04, 1.7893e-04,
          1.5795e-04, 2.8086e-04, 2.2590e-04, 1.0288e-04, 4.0770e-05, 4.0770e-05,
          4.0770e-05, 1.7893e-04, 2.8133e-04, 1.5831e-04, 1.6510e-05, 1.6510e-05,
          7.2384e-04, 4.6611e-04, 4.0770e-05, 1.6510e-05, 4.1127e-05, 1.0383e-04,
          2.2554e-04, 3.4404e-04, 1.0383e-04, 4.1127e-05, 7.1943e-05, 4.5538e-04,
          4.0770e-05, 1.6510e-05, 3.1900e-04, 2.2590e-04, 2.8086e-04, 2.2733e-04,
          9.6142e-05, 9.6142e-05, 7.2384e-04, 9.6142e-05, 9.6142e-05, 1.9302e-03,
          6.7997e-04, 1.0258e-04, 4.0770e-05, 1.6510e-05, 2.4080e-04, 1.0258e-04,
          3.0851e-04, 1.7893e-04, 9.6142e-05, 1.7893e-04, 1.0383e-04, 1.7893e-04,
          2.4080e-04, 2.4104e-04, 2.8086e-04, 1.6510e-05, 1.7893e-04, 1.6510e-05,
          6.3944e-04, 4.1103e-04, 7.8321e-05, 3.5381e-04, 1.6510e-05, 7.1943e-05,
          4.0770e-05, 1.6868e-05, 4.0770e-05, 4.0770e-05, 2.4080e-04, 4.5586e-04,
          6.0034e-04, 1.8301e-03, 1.7893e-04, 1.3103e-03, 2.8086e-04, 1.9426e-03,
          5.3835e-04, 4.5586e-04, 5.4932e-04, 1.5795e-04, 5.7793e-04, 2.0142e-03,
          2.1267e-03, 5.3835e-04, 7.1812e-04, 9.3746e-04, 2.8086e-04, 1.6510e-05,
          4.0770e-05, 1.2362e-04, 1.2362e-04, 4.0770e-05, 4.0770e-05, 4.0770e-05,
          2.3949e-04, 4.0770e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 4.0770e-05,
          4.0770e-05, 1.6510e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0712e-02, 6.7017e-02, 2.6657e-02, 2.3682e-02, 9.8877e-02,
          6.4514e-02, 1.8402e-02, 4.3678e-03, 5.0011e-03, 1.3489e-02, 1.9623e-02,
          9.6817e-03, 4.7798e-03, 4.8332e-03, 6.5460e-03, 2.9716e-03, 6.5689e-03,
          2.9945e-03, 1.8559e-03, 3.8795e-03, 3.0880e-03, 2.6035e-03, 2.7275e-03,
          1.8936e-02, 8.5678e-03, 1.4015e-02, 6.8054e-03, 6.0196e-03, 5.0163e-03,
          4.5319e-03, 6.1874e-03, 3.1338e-03, 8.1329e-03, 4.1809e-03, 6.7825e-03,
          2.9785e-02, 4.4678e-02, 5.3291e-03, 4.5395e-03, 5.4169e-03, 1.0536e-02,
          1.0765e-02, 7.4310e-03, 8.5144e-03, 6.0387e-03, 3.7079e-03, 7.1106e-03,
          3.0155e-03, 3.6545e-03, 1.5045e-02, 1.4030e-02, 8.5754e-03, 4.6310e-03,
          1.0803e-02, 7.5989e-03, 1.1307e-02, 6.6528e-02, 3.1555e-02, 1.4075e-01,
          2.2590e-05, 7.3135e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 1.1361e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 6.3002e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05,
          6.3002e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 4.2963e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          7.3135e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 7.3135e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 1.1361e-04, 7.3135e-05,
          6.3002e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          1.1361e-04, 5.2071e-04, 6.3002e-05, 2.2590e-05, 2.2590e-05, 1.1462e-04,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 5.2071e-04,
          2.2590e-05, 2.2590e-05, 2.1458e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 1.1390e-04, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.1434e-04, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 3.3236e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05,
          1.1361e-04, 2.2590e-05, 2.2590e-05, 1.5974e-03, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.3842e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 2.2590e-05, 1.1390e-04,
          1.1361e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 1.1390e-04, 2.2590e-05, 2.2590e-05, 7.3135e-05, 1.1492e-04,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 3.3116e-04,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 1.1390e-04, 1.1492e-04,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.3544e-05,
          2.2590e-05, 2.2590e-05, 7.3135e-05, 1.0061e-03, 2.1458e-04, 2.2590e-05,
          1.1361e-04, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 6.3002e-05, 4.7016e-04, 2.1124e-04, 2.2590e-05, 2.2888e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 1.1390e-04, 6.3002e-05,
          7.3135e-05, 2.2590e-05, 1.1390e-04, 2.7728e-04, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 1.7643e-04, 2.1434e-04, 1.1390e-04, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 1.7715e-04, 6.3002e-05, 1.1361e-04, 2.2590e-05, 2.2590e-05,
          1.1361e-04, 2.2590e-05, 3.7408e-04, 1.1390e-04, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 6.9094e-04, 1.1361e-04, 2.2590e-05, 2.2590e-05,
          3.9029e-04, 5.8508e-04, 2.2650e-04, 2.2590e-05, 2.2590e-05, 1.1492e-04,
          2.2590e-05, 1.3580e-03, 3.2854e-04, 2.2590e-05, 2.2590e-05, 6.3002e-05,
          2.2590e-05, 2.2590e-05, 3.9029e-04, 5.2738e-04, 2.2590e-05, 3.2854e-04,
          2.2590e-05, 2.2590e-05, 1.8511e-03, 8.5473e-05, 6.3002e-05, 8.9979e-04,
          8.2159e-04, 7.3135e-05, 2.2590e-05, 2.2590e-05, 2.1458e-04, 7.3135e-05,
          7.3135e-05, 3.3951e-04, 2.2590e-05, 8.5473e-05, 3.2854e-04, 2.2590e-05,
          7.4148e-04, 3.9053e-04, 6.3002e-05, 2.2590e-05, 2.7728e-04, 2.2590e-05,
          1.2589e-04, 7.5483e-04, 7.3135e-05, 1.6367e-04, 2.2590e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 2.2590e-05, 7.3135e-05, 1.1361e-04, 6.3002e-05,
          7.4148e-04, 3.9053e-04, 2.2590e-05, 1.2589e-04, 4.7708e-04, 3.7899e-03,
          2.7728e-04, 5.2738e-04, 8.6164e-04, 8.5473e-05, 2.2590e-05, 5.3644e-04,
          1.0004e-03, 6.9094e-04, 8.5473e-05, 5.8985e-04, 4.3654e-04, 2.2590e-05,
          6.3002e-05, 6.3002e-05, 2.2590e-05, 3.2115e-04, 6.3002e-05, 2.2590e-05,
          2.2590e-05, 2.2590e-05, 6.3002e-05, 3.2115e-04, 3.6168e-04, 2.7657e-04,
          6.3002e-05, 6.3002e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0696e-02, 6.6650e-02, 2.8687e-02, 2.5284e-02, 1.1047e-01,
          5.3619e-02, 2.3102e-02, 4.4746e-03, 4.2114e-03, 8.2321e-03, 1.2512e-02,
          5.6381e-03, 3.9940e-03, 4.7264e-03, 5.5351e-03, 3.0441e-03, 5.5008e-03,
          2.5921e-03, 1.9217e-03, 4.0932e-03, 3.1414e-03, 2.4471e-03, 2.4757e-03,
          1.1360e-02, 7.5073e-03, 1.7776e-02, 6.3133e-03, 5.9204e-03, 4.4136e-03,
          4.0588e-03, 6.5765e-03, 2.8038e-03, 7.2060e-03, 3.2196e-03, 6.4278e-03,
          2.1988e-02, 4.6295e-02, 4.2458e-03, 4.1656e-03, 5.5962e-03, 6.5422e-03,
          7.6637e-03, 5.5656e-03, 6.4201e-03, 4.0779e-03, 3.2845e-03, 5.7678e-03,
          2.4567e-03, 3.7346e-03, 1.8463e-02, 1.3084e-02, 8.4991e-03, 4.2915e-03,
          1.0780e-02, 7.3776e-03, 1.1902e-02, 4.5532e-02, 3.7659e-02, 9.9915e-02,
          1.0358e-01, 6.3181e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 6.3181e-05, 2.2888e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05,
          5.4061e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05, 6.3181e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 9.4354e-05, 2.2888e-05,
          9.4354e-05, 5.4061e-05, 2.2888e-05, 5.4061e-05, 5.4061e-05, 2.2888e-05,
          6.3181e-05, 6.3181e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3479e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          5.4061e-05, 2.2888e-05, 9.4354e-05, 2.2888e-05, 2.2888e-05, 1.3900e-04,
          2.2888e-05, 2.2888e-05, 1.9383e-04, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          5.4061e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05, 9.4593e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 9.4354e-05, 5.4061e-05, 2.2888e-05,
          5.4061e-05, 6.3181e-05, 2.2888e-05, 2.2888e-05, 2.3186e-05, 6.3181e-05,
          9.4354e-05, 2.2888e-05, 2.2888e-05, 2.4021e-04, 5.4061e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 5.4061e-05, 5.4061e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05, 5.4061e-05, 1.7202e-04,
          9.4354e-05, 2.2888e-05, 2.2888e-05, 9.4593e-05, 6.3181e-05, 2.2888e-05,
          2.2888e-05, 9.4593e-05, 2.2888e-05, 5.4061e-05, 2.2888e-05, 9.6142e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 5.4061e-05, 6.3181e-05, 2.2888e-05,
          5.4300e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 1.7023e-04,
          2.2888e-05, 5.4061e-05, 6.3181e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 9.6142e-05, 9.6142e-05,
          2.2888e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.3186e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05, 2.2888e-05,
          2.2888e-05, 9.4354e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.4736e-05,
          2.2888e-05, 2.2888e-05, 6.3181e-05, 9.4593e-05, 9.4593e-05, 2.2888e-05,
          9.4593e-05, 2.4736e-05, 2.3186e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 5.4061e-05, 5.4169e-04, 1.7047e-04, 2.2888e-05, 9.6142e-05,
          9.4593e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 4.2629e-04, 5.4061e-05,
          6.3181e-05, 2.2888e-05, 9.6142e-05, 3.4666e-04, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 1.4067e-04, 2.3842e-04, 9.4593e-05, 2.2888e-05, 5.4061e-05,
          2.2888e-05, 5.4300e-05, 2.2888e-05, 9.4593e-05, 2.2888e-05, 6.3181e-05,
          4.2486e-04, 2.2888e-05, 3.5334e-04, 9.4593e-05, 2.2888e-05, 5.4061e-05,
          2.2888e-05, 1.5175e-04, 4.7112e-04, 1.4091e-04, 2.2888e-05, 2.2888e-05,
          3.4666e-04, 5.0068e-04, 9.4354e-05, 2.2888e-05, 2.2888e-05, 9.6142e-05,
          2.2888e-05, 1.9512e-03, 9.6142e-05, 2.2888e-05, 2.2888e-05, 5.4061e-05,
          2.2888e-05, 2.2888e-05, 1.4067e-04, 7.1239e-04, 5.4061e-05, 9.6142e-05,
          2.2888e-05, 2.2888e-05, 6.5660e-04, 1.0037e-04, 2.2888e-05, 2.2049e-03,
          5.4216e-04, 2.2888e-05, 2.2888e-05, 2.2888e-05, 9.4593e-05, 9.4593e-05,
          6.3181e-05, 2.3842e-04, 6.9201e-05, 3.0637e-04, 1.9383e-04, 2.2888e-05,
          5.6887e-04, 3.4714e-04, 3.8433e-04, 2.2888e-05, 2.3842e-04, 2.2888e-05,
          2.3842e-04, 1.7047e-04, 6.3181e-05, 1.9813e-04, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 2.2888e-05, 6.3181e-05, 9.4593e-05, 2.2888e-05,
          4.2486e-04, 9.4593e-05, 2.3842e-04, 1.0037e-04, 4.3058e-04, 2.0828e-03,
          5.6839e-04, 5.3930e-04, 3.1447e-04, 1.0949e-04, 1.0037e-04, 1.9813e-04,
          2.2068e-03, 6.7711e-04, 6.9201e-05, 4.3058e-04, 6.9201e-05, 2.2888e-05,
          2.6011e-04, 5.4061e-05, 5.4061e-05, 2.2888e-05, 2.2888e-05, 2.2888e-05,
          2.2888e-05, 2.2888e-05, 5.4061e-05, 3.2353e-04, 5.4061e-05, 5.4061e-05,
          5.4061e-05, 5.4061e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.2718e-02, 6.2286e-02, 3.0060e-02, 2.7466e-02, 9.8267e-02,
          4.8645e-02, 1.9485e-02, 3.8738e-03, 3.0918e-03, 4.6768e-03, 5.2872e-03,
          2.8133e-03, 1.1475e-02, 2.9205e-02, 5.2490e-03, 3.5439e-03, 3.8330e-02,
          1.7334e-02, 1.7376e-03, 1.9012e-02, 1.0246e-02, 1.1444e-02, 8.0872e-03,
          5.4665e-03, 4.4289e-03, 1.5312e-02, 1.0941e-02, 2.1973e-02, 2.2705e-02,
          2.1896e-02, 6.2065e-03, 2.0126e-02, 9.1934e-03, 3.5419e-03, 6.6147e-03,
          1.1887e-02, 7.0915e-03, 6.6795e-03, 5.6725e-03, 2.6886e-02, 7.3128e-03,
          6.4621e-03, 4.2458e-03, 4.1809e-03, 3.4962e-03, 1.4442e-02, 9.3384e-03,
          1.2695e-02, 1.3321e-02, 1.7014e-02, 1.2016e-02, 7.1030e-03, 3.3913e-03,
          6.4278e-03, 6.0310e-03, 8.0261e-03, 1.1345e-02, 7.8659e-03, 1.0025e-02,
          7.9269e-03, 1.1511e-01, 1.3173e-05, 1.3173e-05, 1.3173e-05, 5.0187e-05,
          6.7234e-05, 3.0220e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05,
          1.3173e-05, 1.3173e-05, 1.3173e-05, 5.0187e-05, 1.3173e-05, 1.3173e-05,
          1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 3.0220e-05, 1.3173e-05,
          1.3173e-05, 5.0187e-05, 1.3173e-05, 2.1648e-04, 6.7234e-05, 5.0187e-05,
          1.1814e-04, 1.3173e-05, 1.3173e-05, 6.7234e-05, 1.3173e-05, 1.3173e-05,
          1.3173e-05, 1.3173e-05, 3.0220e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05,
          6.7234e-05, 1.3506e-04, 6.7234e-05, 1.3173e-05, 6.7234e-05, 1.3173e-05,
          1.3173e-05, 1.3173e-05, 3.0220e-05, 1.3173e-05, 6.7234e-05, 3.0220e-05,
          1.3173e-05, 1.3173e-05, 1.3173e-05, 5.0187e-05, 1.3173e-05, 1.3173e-05,
          1.3173e-05, 1.3173e-05, 1.3173e-05, 5.0187e-05, 5.0187e-05, 3.0220e-05,
          1.3173e-05, 1.3173e-05, 3.0220e-05, 5.0187e-05, 3.0220e-05, 1.3173e-05,
          1.2112e-04, 3.0279e-05, 3.0220e-05, 3.0220e-05, 3.0220e-05, 1.3173e-05,
          3.0220e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05,
          3.0220e-05, 3.0220e-05, 1.3173e-05, 1.3173e-05, 5.0187e-05, 6.7234e-05,
          3.0220e-05, 6.7234e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05,
          8.1182e-05, 3.0220e-05, 2.5344e-04, 1.3173e-05, 3.0220e-05, 1.3173e-05,
          1.3173e-05, 3.0220e-05, 3.0220e-05, 1.3173e-05, 1.3173e-05, 3.0220e-05,
          1.1814e-04, 1.4496e-03, 6.7234e-05, 6.7234e-05, 1.3173e-05, 3.0220e-05,
          3.0220e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 3.0220e-05, 1.3173e-05,
          3.0220e-05, 6.7234e-05, 6.7234e-05, 1.1814e-04, 1.3173e-05, 1.3173e-05,
          1.9956e-04, 1.3173e-05, 1.3173e-05, 3.0220e-05, 6.7472e-05, 6.7234e-05,
          6.7234e-05, 1.3275e-03, 1.3173e-05, 3.0220e-05, 1.1814e-04, 6.7234e-05,
          1.3173e-05, 1.3173e-05, 3.0220e-05, 1.3506e-04, 2.1648e-04, 6.7234e-05,
          6.7294e-05, 1.3506e-04, 1.3173e-05, 6.7234e-05, 1.1814e-04, 3.0220e-05,
          6.7234e-05, 1.3173e-05, 3.0220e-05, 1.1837e-04, 6.7234e-05, 6.7234e-05,
          3.0220e-05, 1.3173e-05, 1.3173e-05, 1.1814e-04, 3.0220e-05, 5.0187e-05,
          1.3173e-05, 1.3173e-05, 6.5422e-04, 9.2888e-04, 2.8062e-04, 6.7234e-05,
          8.4162e-05, 6.7234e-05, 3.0220e-05, 6.7234e-05, 1.3173e-05, 1.3173e-05,
          1.3173e-05, 1.3506e-04, 6.7234e-05, 6.7234e-05, 3.0220e-05, 5.0187e-05,
          1.3173e-05, 1.3173e-05, 1.3173e-05, 3.0220e-05, 5.0187e-05, 1.3173e-05,
          1.3173e-05, 1.3506e-04, 3.0220e-05, 2.8062e-04, 1.2112e-04, 1.2112e-04,
          1.0099e-03, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05, 1.3173e-05,
          3.0220e-05, 1.7202e-04, 3.0220e-05, 3.0220e-05, 1.3173e-05, 5.0426e-05,
          3.0220e-05, 3.0220e-05, 6.7234e-05, 6.7234e-05, 5.0187e-05, 8.4102e-05,
          8.4734e-04, 6.7294e-05, 3.3450e-04, 6.5422e-04, 8.4102e-05, 6.7234e-05,
          3.0220e-05, 1.1814e-04, 1.3173e-05, 3.0220e-05, 1.3173e-05, 1.2136e-04,
          6.7294e-05, 1.3411e-05, 1.3232e-05, 1.3173e-05, 3.0220e-05, 3.0220e-05,
          6.7234e-05, 3.0220e-05, 3.0220e-05, 8.1182e-05, 1.3173e-05, 6.7294e-05,
          8.4102e-05, 8.4162e-05, 8.4102e-05, 6.7234e-05, 7.0477e-04, 8.4162e-05,
          1.3173e-05, 3.0220e-05, 1.1814e-04, 3.0220e-05, 8.4102e-05, 6.7234e-05,
          3.0220e-05, 3.0220e-05, 3.0220e-05, 3.0220e-05, 1.3173e-05, 1.1814e-04,
          3.0220e-05, 6.7234e-05, 1.3173e-05, 6.7234e-05, 3.0220e-05, 2.5368e-04,
          8.2350e-04, 2.5344e-04, 3.0220e-05, 3.0220e-05, 3.0220e-05, 3.0220e-05,
          3.0220e-05, 3.0220e-05, 3.0220e-05, 3.0220e-05, 3.0220e-05, 1.3173e-05,
          3.0220e-05, 6.7234e-05, 1.3173e-05, 6.7234e-05, 3.0220e-05, 1.3173e-05,
          8.4734e-04, 3.0220e-05, 6.7234e-05, 1.3173e-05, 8.9169e-04, 3.0220e-05,
          1.6558e-04, 1.4865e-04, 3.0220e-05, 2.5344e-04, 3.0220e-05, 3.0220e-05,
          1.3173e-05, 3.0220e-05, 8.4102e-05, 3.0220e-05, 1.1814e-04, 6.7234e-05,
          6.7234e-05, 1.3173e-05, 2.5344e-04, 2.5344e-04, 3.0220e-05, 2.5344e-04,
          1.1814e-04, 1.1814e-04, 1.6260e-04, 6.7234e-05, 2.1648e-04, 3.0220e-05,
          2.5344e-04, 1.3173e-05, 5.0187e-05, 2.1648e-04, 1.3173e-05, 1.3173e-05,
          3.0220e-05, 7.5912e-04, 8.4102e-05, 1.3173e-05, 6.7234e-05, 1.3173e-05,
          7.5626e-04, 6.7234e-05, 5.5933e-04, 3.0220e-05, 1.3173e-05, 1.3173e-05,
          2.5344e-04, 8.4102e-05, 3.0220e-05, 5.0187e-05, 2.5344e-04, 2.1648e-04,
          3.0220e-05, 2.1648e-04, 1.3173e-05, 8.4102e-05, 6.7115e-05, 1.3173e-05,
          1.3506e-04, 1.3506e-04, 8.1182e-05, 1.3506e-04, 1.3506e-04, 3.0220e-05,
          1.3173e-05, 1.1814e-04, 1.3506e-04, 4.8375e-04, 8.1182e-05, 3.0220e-05,
          8.1182e-05, 8.1182e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0033e-02, 6.7017e-02, 3.0045e-02, 2.5482e-02, 1.2134e-01,
          5.9265e-02, 2.4384e-02, 9.3155e-03, 4.6730e-03, 3.7212e-03, 4.7913e-03,
          2.3880e-03, 3.0270e-03, 3.8776e-03, 4.4212e-03, 9.3536e-03, 4.7226e-03,
          3.0193e-03, 2.9564e-03, 5.1384e-03, 5.3139e-03, 3.7308e-03, 2.9964e-03,
          5.1231e-03, 5.5923e-03, 1.9989e-02, 7.8278e-03, 5.4970e-03, 3.9024e-03,
          3.5362e-03, 1.2032e-02, 2.5902e-03, 5.1460e-03, 2.0695e-03, 1.2154e-02,
          9.4604e-03, 1.0712e-02, 5.1651e-03, 4.2152e-03, 4.8523e-03, 8.5220e-03,
          1.0635e-02, 6.2294e-03, 7.1373e-03, 3.8223e-03, 3.1872e-03, 5.6343e-03,
          2.4109e-03, 4.2992e-03, 2.6230e-02, 2.5650e-02, 1.4252e-02, 5.6076e-03,
          1.4091e-02, 1.1002e-02, 2.0142e-02, 7.9498e-03, 1.5167e-02, 9.8190e-03,
          1.1368e-02, 1.2276e-02, 2.2339e-01, 2.0742e-05, 2.0742e-05, 9.3639e-05,
          9.3639e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 1.2124e-04, 9.3639e-05, 9.3639e-05,
          1.2124e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 1.2124e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3699e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3639e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3639e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3639e-05, 2.0742e-05, 2.0742e-05,
          9.3639e-05, 2.0802e-05, 2.0742e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3699e-05, 9.3639e-05,
          2.0742e-05, 9.3639e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.2124e-04, 2.0742e-05, 1.2124e-04, 9.3639e-05, 9.3639e-05, 9.3639e-05,
          2.0742e-05, 2.0742e-05, 4.5943e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.9407e-04, 1.2124e-04, 9.3639e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 9.3639e-05, 9.3699e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.2124e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3699e-05, 9.3639e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 4.5943e-04, 1.2124e-04, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 1.2130e-04, 2.0742e-05, 2.0742e-05,
          9.3699e-05, 1.2124e-04, 2.0742e-05, 2.0742e-05, 1.2124e-04, 4.5943e-04,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.5546e-05, 9.3639e-05, 9.3699e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 1.2124e-04, 1.9419e-04, 9.3639e-05,
          2.0802e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3639e-05,
          2.0742e-05, 1.2124e-04, 2.0802e-05, 2.0742e-05, 2.0802e-05, 9.3639e-05,
          2.2650e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3699e-05, 2.0802e-05,
          2.0742e-05, 1.2130e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0802e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0802e-05, 2.0802e-05, 2.0802e-05, 2.0742e-05, 2.2650e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3639e-05, 9.3639e-05, 2.0742e-05,
          9.3699e-05, 9.5546e-05, 1.9419e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 9.3699e-05, 2.0742e-05, 2.2650e-05,
          2.0802e-05, 2.2650e-05, 2.2650e-05, 2.0742e-05, 1.2130e-04, 2.0742e-05,
          9.3699e-05, 2.0742e-05, 1.2130e-04, 4.5943e-04, 2.0742e-05, 2.0802e-05,
          2.0742e-05, 2.0802e-05, 2.0742e-05, 9.3639e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0802e-05, 9.3699e-05,
          2.0742e-05, 2.2650e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0802e-05, 9.3699e-05, 2.0742e-05, 1.2130e-04,
          1.2124e-04, 3.8648e-04, 2.0742e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05,
          1.2124e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0802e-05, 9.3699e-05,
          2.0742e-05, 1.2130e-04, 9.5546e-05, 9.3699e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0802e-05, 1.2124e-04, 9.3699e-05,
          2.0742e-05, 9.3639e-05, 1.9419e-04, 9.3639e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 1.2124e-04, 2.0802e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 9.3639e-05, 2.0742e-05, 9.3699e-05, 2.0742e-05,
          1.2124e-04, 1.2124e-04, 2.0742e-05, 9.3639e-05, 2.0742e-05, 2.0742e-05,
          1.9407e-04, 2.0802e-05, 2.0742e-05, 1.2124e-04, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0802e-05, 2.0802e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.2124e-04, 9.3699e-05, 4.5943e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          3.8648e-04, 2.0802e-05, 9.3699e-05, 2.0742e-05, 3.8648e-04, 1.2124e-04,
          2.0742e-05, 1.2124e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.2124e-04, 1.2124e-04, 3.8648e-04, 1.2124e-04, 9.3031e-04, 1.2124e-04,
          2.0742e-05, 1.2124e-04, 1.2124e-04, 5.3263e-04, 2.0742e-05, 1.2124e-04,
          2.0742e-05, 1.2124e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.8343e-03, 6.2500e-02, 2.7161e-02, 2.4490e-02, 9.6985e-02,
          5.7648e-02, 1.8707e-02, 7.9193e-03, 4.9438e-03, 7.6180e-03, 8.0032e-03,
          3.8967e-03, 5.1575e-03, 4.0512e-03, 7.2517e-03, 1.0399e-02, 1.1902e-02,
          5.0964e-03, 3.8834e-03, 3.7556e-03, 5.9204e-03, 4.0817e-03, 6.2485e-03,
          1.0078e-02, 8.6594e-03, 1.5732e-02, 1.1414e-02, 4.6692e-03, 8.1482e-03,
          4.3030e-03, 1.3519e-02, 4.4136e-03, 1.1826e-02, 4.7684e-03, 1.6373e-02,
          1.6617e-02, 7.5493e-03, 1.8036e-02, 1.1154e-02, 4.3411e-03, 2.7344e-02,
          1.3458e-02, 9.5215e-03, 7.0076e-03, 7.6218e-03, 5.9166e-03, 1.4511e-02,
          2.9869e-03, 4.0054e-03, 1.7105e-02, 3.0273e-02, 1.7899e-02, 7.1793e-03,
          7.5073e-03, 1.2482e-02, 1.0101e-02, 1.0185e-02, 1.0750e-02, 9.8190e-03,
          6.8016e-03, 1.6785e-02, 5.4962e-02, 9.9792e-02, 1.4603e-05, 1.4603e-05,
          6.8545e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 6.8545e-05,
          1.4544e-04, 1.4603e-05, 6.8545e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          6.8545e-05, 9.1434e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 6.8545e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          6.8545e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05,
          1.4603e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          6.8545e-05, 1.4603e-05, 1.4544e-04, 6.8545e-05, 1.4603e-05, 6.8545e-05,
          1.4603e-05, 1.4603e-05, 9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4544e-04, 9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4544e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.2002e-05, 6.8545e-05,
          9.1434e-05, 1.4603e-05, 1.4603e-05, 3.5691e-04, 1.4544e-04, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 6.8545e-05, 1.4603e-05,
          6.9916e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04, 9.1434e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.2002e-05, 1.4603e-05, 6.9916e-05,
          1.4603e-05, 9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05, 6.8545e-05, 1.4603e-05,
          6.9916e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05,
          1.4603e-05, 1.4603e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05,
          1.6630e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04, 1.4544e-04,
          1.4603e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.6630e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04, 1.4544e-04, 1.4603e-05,
          6.8545e-05, 7.2002e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4544e-04, 1.4603e-05, 7.2002e-05,
          6.9916e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 9.1434e-05, 1.4603e-05,
          6.8545e-05, 1.4603e-05, 9.1434e-05, 6.1607e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 6.8545e-05,
          9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4544e-04, 9.1434e-05, 9.1434e-05, 1.4603e-05, 1.4603e-05,
          9.1434e-05, 1.4544e-04, 6.8545e-05, 1.4603e-05, 1.4603e-05, 1.4746e-04,
          1.4603e-05, 9.2804e-05, 4.1437e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.1854e-03, 9.1434e-05, 1.4687e-04,
          1.4603e-05, 1.4603e-05, 4.1223e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.1854e-03, 9.1434e-05, 1.4603e-05, 1.4603e-05, 9.1434e-05, 6.8545e-05,
          1.4603e-05, 3.5691e-04, 6.8545e-05, 1.4603e-05, 4.1437e-04, 9.1434e-05,
          1.4544e-04, 1.4544e-04, 1.4603e-05, 6.8545e-05, 1.4603e-05, 1.4603e-05,
          1.4544e-04, 1.4687e-04, 1.4603e-05, 9.1434e-05, 1.4603e-05, 1.4603e-05,
          3.5691e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05, 9.1434e-05, 1.4603e-05,
          9.1434e-05, 4.1080e-04, 4.0436e-04, 1.4603e-05, 1.4603e-05, 1.9102e-03,
          9.1434e-05, 1.1854e-03, 1.5812e-03, 1.4603e-05, 1.4603e-05, 9.1434e-05,
          1.5812e-03, 9.1434e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          9.1434e-05, 9.1434e-05, 3.5024e-04, 1.4603e-05, 9.1434e-05, 1.4603e-05,
          1.4603e-05, 9.1434e-05, 9.1434e-05, 5.3358e-04, 5.3358e-04, 6.1035e-04,
          9.1434e-05, 1.4603e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0773e-02, 7.4707e-02, 2.8137e-02, 2.6627e-02, 9.0881e-02,
          6.6162e-02, 2.1149e-02, 3.2692e-03, 3.7403e-03, 1.7273e-02, 2.5406e-02,
          1.1864e-02, 4.2686e-03, 3.9711e-03, 5.1689e-03, 3.6087e-03, 6.2332e-03,
          3.6278e-03, 2.9621e-03, 2.3766e-03, 3.3970e-03, 3.0537e-03, 4.6234e-03,
          3.7781e-02, 1.2230e-02, 1.6220e-02, 8.3771e-03, 3.0212e-03, 5.1270e-03,
          4.6043e-03, 5.0125e-03, 2.5311e-03, 6.8970e-03, 5.7716e-03, 4.5776e-03,
          1.4175e-02, 9.2163e-03, 1.3611e-02, 5.2986e-03, 3.2291e-03, 5.2948e-02,
          3.9062e-02, 2.5391e-02, 2.4292e-02, 1.6373e-02, 5.0774e-03, 1.1581e-02,
          2.8687e-03, 3.0594e-03, 1.3863e-02, 7.1144e-03, 5.1193e-03, 3.5286e-03,
          3.3035e-03, 3.5515e-03, 2.8400e-03, 1.9424e-02, 1.0498e-02, 1.9623e-02,
          8.7967e-03, 5.9662e-03, 1.0765e-02, 1.2505e-02, 8.7769e-02, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 5.9009e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 5.9009e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 5.9009e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          1.4246e-04, 5.9009e-05, 1.0401e-04, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 1.4246e-04, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 5.9009e-05, 5.9009e-05, 1.4246e-04, 5.9009e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          5.9009e-05, 2.0623e-05, 2.0623e-05, 5.1308e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.1458e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 2.0623e-05, 5.1308e-04,
          5.9009e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 5.9009e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 1.4246e-04,
          5.9009e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 1.4246e-04, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 5.9843e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 5.9009e-05, 5.9009e-05, 1.4246e-04, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 5.9009e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 5.9009e-05, 2.0623e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 1.4246e-04, 2.0623e-05,
          5.9009e-05, 2.0623e-05, 1.4246e-04, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 1.4246e-04, 5.9009e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 4.7541e-04, 2.0623e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          1.4246e-04, 2.0623e-05, 2.0623e-05, 5.9068e-05, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 4.7445e-04, 1.4246e-04, 2.0623e-05, 2.0623e-05,
          5.1308e-04, 1.4246e-04, 1.4246e-04, 2.0623e-05, 2.0623e-05, 1.4329e-04,
          2.0623e-05, 1.4246e-04, 1.4329e-04, 5.9009e-05, 2.0623e-05, 2.0623e-05,
          1.0401e-04, 5.9009e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05, 1.4329e-04,
          2.0623e-05, 2.0623e-05, 5.1308e-04, 2.0623e-05, 5.9009e-05, 1.0401e-04,
          2.0623e-05, 5.9009e-05, 2.0623e-05, 2.0623e-05, 9.5844e-05, 5.9009e-05,
          5.9009e-05, 1.4246e-04, 2.0623e-05, 2.0623e-05, 1.4329e-04, 2.0623e-05,
          2.8419e-03, 3.2687e-04, 2.0623e-05, 2.0623e-05, 5.9009e-05, 2.0623e-05,
          5.9009e-05, 1.4329e-04, 5.9009e-05, 1.0401e-04, 2.0623e-05, 2.0623e-05,
          5.9009e-05, 2.0623e-05, 2.0623e-05, 5.9009e-05, 9.5844e-05, 2.0623e-05,
          1.7929e-04, 1.4246e-04, 2.0623e-05, 1.0401e-04, 2.0623e-05, 1.0401e-04,
          5.9009e-05, 2.0623e-05, 1.4246e-04, 5.9009e-05, 2.0623e-05, 1.0401e-04,
          1.3704e-03, 2.1210e-03, 2.0623e-05, 1.0401e-04, 2.0623e-05, 2.0623e-05,
          2.0623e-05, 2.0623e-05, 2.0623e-05, 2.0623e-05, 1.0401e-04, 2.0623e-05,
          2.0623e-05, 1.4246e-04, 2.0623e-05, 5.5408e-04, 2.0623e-05, 1.0401e-04,
          2.0623e-05, 2.0623e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.6060e-03, 7.5806e-02, 2.8214e-02, 2.8076e-02, 7.5745e-02,
          7.8003e-02, 1.8921e-02, 4.6120e-03, 4.0588e-03, 1.7166e-02, 1.9302e-02,
          8.2016e-03, 8.7738e-03, 2.9526e-03, 1.9043e-02, 5.6801e-03, 1.1597e-02,
          6.5689e-03, 3.6545e-03, 1.7672e-03, 2.8934e-03, 3.1090e-03, 8.0338e-03,
          2.5696e-02, 1.7853e-02, 1.5450e-02, 1.2810e-02, 2.1744e-03, 1.0429e-02,
          3.7384e-03, 8.0566e-03, 3.3875e-03, 2.4460e-02, 1.2283e-02, 6.7520e-03,
          2.3819e-02, 3.3894e-03, 1.8021e-02, 1.3191e-02, 2.1229e-03, 2.9465e-02,
          8.7891e-03, 7.7095e-03, 5.7335e-03, 7.4120e-03, 1.1803e-02, 3.6591e-02,
          2.5349e-03, 2.5043e-03, 1.0254e-02, 7.0839e-03, 5.3368e-03, 5.1193e-03,
          1.8730e-03, 3.0060e-03, 1.8253e-03, 6.5918e-03, 5.0201e-03, 5.8136e-03,
          4.0665e-03, 6.1684e-03, 7.6256e-03, 1.1642e-02, 1.5541e-02, 1.1029e-01,
          5.3763e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 5.3763e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 5.3763e-05, 1.7583e-05, 8.7023e-05, 1.7583e-05, 5.3763e-05,
          1.2314e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          5.3763e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          5.3763e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05,
          5.3763e-05, 1.7583e-05, 1.7583e-05, 8.7023e-05, 8.7023e-05, 1.7583e-05,
          8.7023e-05, 1.2314e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.2314e-04,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          8.7023e-05, 1.7583e-05, 1.2314e-04, 1.7583e-05, 1.7583e-05, 1.2314e-04,
          1.7583e-05, 1.7583e-05, 3.0375e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.2314e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.2314e-04, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          1.2314e-04, 2.6679e-04, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05,
          1.2314e-04, 1.7583e-05, 1.7583e-05, 3.0375e-04, 1.2314e-04, 1.7583e-05,
          1.7583e-05, 1.9741e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.2314e-04, 3.0375e-04,
          5.3763e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.2314e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.2314e-04,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 5.3763e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 5.3763e-05,
          1.7583e-05, 1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05, 1.7321e-04,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 1.2374e-04,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 5.3763e-05, 5.3763e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 2.6679e-04,
          1.7583e-05, 5.3763e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.8179e-05,
          1.7583e-05, 1.7583e-05, 5.3763e-05, 1.2314e-04, 1.2314e-04, 1.7583e-05,
          2.3353e-04, 5.3763e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 5.3763e-05, 1.9741e-04, 1.2314e-04, 1.7583e-05, 1.7381e-04,
          5.3763e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7321e-04, 1.7583e-05, 1.2314e-04, 6.6853e-04, 1.7583e-05, 5.3763e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.7583e-05, 1.9741e-04,
          1.7583e-05, 1.7583e-05, 1.2314e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05,
          1.7583e-05, 1.7583e-05, 1.7583e-05, 1.9741e-04, 1.7583e-05, 5.3763e-05,
          2.6679e-04, 1.7583e-05, 1.9741e-04, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          2.3353e-04, 4.8876e-04, 4.4751e-04, 8.7023e-05, 1.7583e-05, 1.7583e-05,
          2.6774e-04, 3.0303e-04, 4.8351e-04, 1.7583e-05, 1.7583e-05, 4.2391e-04,
          1.7583e-05, 2.3437e-04, 4.2391e-04, 5.3763e-05, 1.7583e-05, 1.9741e-04,
          4.2248e-04, 1.7583e-05, 5.5695e-04, 2.6679e-04, 8.7023e-05, 7.4053e-04,
          1.7583e-05, 1.7583e-05, 3.0375e-04, 1.2314e-04, 5.5695e-04, 4.4751e-04,
          6.2037e-04, 6.0320e-04, 1.7583e-05, 1.7583e-05, 2.4271e-04, 2.4271e-04,
          3.7742e-04, 8.7023e-05, 1.7583e-05, 1.7583e-05, 6.0415e-04, 1.7583e-05,
          5.5933e-04, 8.8310e-04, 5.2118e-04, 5.3763e-05, 1.2314e-04, 1.7583e-05,
          5.3763e-05, 6.2037e-04, 5.3763e-05, 1.2314e-04, 1.7583e-05, 1.7583e-05,
          6.0320e-04, 1.7583e-05, 1.7583e-05, 1.7583e-05, 4.2248e-04, 3.7789e-04,
          9.2697e-04, 4.2343e-04, 1.2314e-04, 1.7583e-05, 1.9741e-04, 9.2411e-04,
          7.4625e-04, 3.7789e-04, 7.4005e-04, 5.3763e-05, 1.7583e-05, 1.7583e-05,
          9.2411e-04, 3.0303e-04, 1.7583e-05, 2.6679e-04, 3.7789e-04, 1.7583e-05,
          5.3763e-05, 1.7583e-05, 4.5252e-04, 1.7583e-05, 8.7023e-05, 5.3763e-05,
          1.7583e-05, 1.2314e-04, 8.7023e-05, 5.4073e-04, 5.4073e-04, 6.3276e-04,
          8.7023e-05, 8.7023e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0399e-02, 8.1360e-02, 2.8412e-02, 3.7750e-02, 8.8928e-02,
          6.8665e-02, 2.4643e-02, 3.4237e-03, 2.9926e-03, 1.1917e-02, 3.0869e-02,
          9.0714e-03, 3.7346e-03, 4.7607e-03, 3.9864e-03, 3.7537e-03, 4.7684e-03,
          3.7193e-03, 2.7618e-03, 1.3742e-03, 1.8082e-03, 2.0828e-03, 4.4098e-03,
          4.0680e-02, 9.7809e-03, 2.0035e-02, 1.0567e-02, 2.0447e-03, 4.6120e-03,
          6.3286e-03, 5.2376e-03, 1.9989e-03, 4.2801e-03, 4.0474e-03, 2.9430e-03,
          5.0850e-03, 4.2763e-03, 6.3515e-03, 3.4199e-03, 2.0199e-03, 2.2629e-02,
          1.9394e-02, 1.2428e-02, 1.6769e-02, 6.4507e-03, 5.2452e-03, 6.8741e-03,
          2.6550e-03, 2.1305e-03, 1.2489e-02, 4.1885e-03, 4.1733e-03, 2.7943e-03,
          2.4624e-03, 2.7390e-03, 1.9913e-03, 5.3749e-03, 7.7133e-03, 8.0261e-03,
          5.4550e-03, 4.3716e-03, 6.8932e-03, 6.5918e-03, 3.5950e-02, 1.6281e-02,
          1.6772e-01, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 5.7757e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 5.7757e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 2.4855e-05,
          1.3483e-04, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 1.1086e-04, 1.1086e-04, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 5.7757e-05, 1.3483e-04,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          1.8287e-04, 7.2777e-05, 7.2777e-05, 2.4855e-05, 2.4855e-05, 1.0574e-04,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 1.1086e-04, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 2.3592e-04,
          2.4855e-05, 2.4855e-05, 3.6812e-04, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 1.1086e-04, 5.7757e-05, 1.0574e-04, 5.7757e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 1.0574e-04, 5.7757e-05, 5.7757e-05, 1.0574e-04, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 1.1086e-04,
          5.7757e-05, 2.4855e-05, 5.7757e-05, 5.2929e-04, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.5213e-05, 1.4985e-04, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 5.7757e-05, 2.4855e-05, 6.0606e-04,
          1.1086e-04, 2.4855e-05, 2.4855e-05, 2.4855e-05, 1.1086e-04, 2.4855e-05,
          2.4855e-05, 1.5879e-04, 2.4855e-05, 5.7757e-05, 1.5879e-04, 1.5879e-04,
          5.7757e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.5511e-04, 5.7757e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 1.1086e-04,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 7.7963e-05, 1.1086e-04,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 7.2777e-05, 1.1086e-04, 1.6952e-04,
          1.0192e-04, 2.4855e-05, 2.4855e-05, 2.7919e-04, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 1.0574e-04, 1.4985e-04,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 5.7757e-05, 2.3592e-04, 1.0574e-04, 2.4629e-04,
          2.4855e-05, 2.4855e-05, 2.7919e-04, 1.0192e-04, 2.4629e-04, 2.4855e-05,
          2.4855e-05, 2.5511e-04, 1.0192e-04, 2.0206e-04, 2.4855e-05, 2.4855e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 2.4855e-05, 2.3592e-04, 7.2777e-05,
          1.1086e-04, 1.4985e-04, 7.8344e-04, 3.6287e-04, 2.4855e-05, 7.2777e-05,
          2.4855e-05, 2.4855e-05, 2.4855e-05, 1.8287e-04, 1.0192e-04, 2.4855e-05,
          2.4855e-05, 1.0574e-04, 1.1086e-04, 5.2071e-04, 2.4855e-05, 2.4855e-05,
          7.2777e-05, 5.3167e-04, 7.2777e-05, 1.5879e-04, 2.4855e-05, 2.4855e-05,
          6.1750e-04, 2.4855e-05, 1.0192e-04, 1.8799e-04, 7.2777e-05, 2.4629e-04,
          2.4629e-04, 2.4855e-05, 4.4012e-04, 3.3236e-04, 1.0192e-04, 2.4629e-04,
          7.7820e-04, 1.8287e-04, 2.4629e-04, 2.4855e-05, 7.7963e-05, 5.4073e-04,
          2.4855e-05, 6.1750e-04, 3.3259e-04, 1.1086e-04, 2.4855e-05, 2.4629e-04,
          2.4629e-04, 5.7757e-05, 2.3592e-04, 2.4629e-04, 2.4855e-05, 2.7966e-04,
          2.4855e-05, 2.4855e-05, 4.7326e-04, 1.3483e-04, 1.3483e-04, 2.4629e-04,
          1.0192e-04, 1.8287e-04, 2.4855e-05, 2.4855e-05, 1.1086e-04, 9.6178e-04,
          1.8787e-04, 1.6916e-04, 2.4855e-05, 1.0192e-04, 5.6458e-04, 1.4985e-04,
          9.6178e-04, 9.6178e-04, 1.3483e-04, 2.4855e-05, 1.3483e-04, 2.4855e-05,
          2.6655e-04, 2.7919e-04, 5.7757e-05, 5.7316e-04, 2.4855e-05, 2.4855e-05,
          1.4985e-04, 1.2118e-04, 2.4855e-05, 5.7757e-05, 9.6178e-04, 1.0192e-04,
          2.3592e-04, 5.4026e-04, 7.2777e-05, 6.9237e-04, 1.0192e-04, 6.1512e-04,
          1.8787e-04, 2.4629e-04, 5.6410e-04, 5.7757e-05, 2.4855e-05, 4.0722e-04,
          5.3120e-04, 8.1539e-04, 2.8181e-04, 4.0722e-04, 4.0722e-04, 2.4855e-05,
          7.2777e-05, 7.2777e-05, 7.2777e-05, 1.5759e-04, 2.3890e-04, 2.7180e-04,
          2.4855e-05, 2.7180e-04, 2.3890e-04, 6.1798e-04, 2.3890e-04, 2.3890e-04,
          7.2777e-05, 2.3890e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.9335e-03, 7.7087e-02, 2.8214e-02, 3.3081e-02, 7.3242e-02,
          6.4148e-02, 2.1271e-02, 4.9744e-03, 3.6163e-03, 1.9241e-02, 4.5197e-02,
          1.3657e-02, 5.2681e-03, 3.3741e-03, 4.9553e-03, 5.5847e-03, 6.7749e-03,
          4.4975e-03, 4.0588e-03, 1.1463e-03, 1.7681e-03, 2.6340e-03, 6.4087e-03,
          6.2134e-02, 1.0490e-02, 1.7578e-02, 1.2009e-02, 1.5354e-03, 5.9967e-03,
          4.4403e-03, 7.8201e-03, 2.1362e-03, 5.1346e-03, 6.6452e-03, 3.5191e-03,
          5.6190e-03, 3.6564e-03, 8.8501e-03, 4.2648e-03, 1.6575e-03, 2.8580e-02,
          1.4481e-02, 1.0208e-02, 1.0727e-02, 7.2517e-03, 8.1100e-03, 8.9340e-03,
          2.4986e-03, 2.0447e-03, 9.6741e-03, 4.5776e-03, 5.4398e-03, 3.6106e-03,
          1.8930e-03, 2.9697e-03, 1.4715e-03, 6.8550e-03, 7.5493e-03, 9.6436e-03,
          4.6768e-03, 3.7174e-03, 5.6114e-03, 7.0267e-03, 3.2257e-02, 2.3697e-02,
          7.1350e-02, 6.5979e-02, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 4.5538e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 4.5538e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 4.5538e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 4.5538e-05,
          1.5283e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 1.7107e-05, 1.7107e-05,
          4.5538e-05, 1.7107e-05, 1.7107e-05, 7.4089e-05, 7.4089e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 4.5538e-05, 1.5283e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.2827e-04, 1.2827e-04, 1.2827e-04, 4.5538e-05, 1.7107e-05, 1.2827e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05,
          1.7107e-05, 4.5538e-05, 1.7107e-05, 7.4089e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 8.0884e-05, 2.3556e-04,
          1.7107e-05, 1.7107e-05, 2.9373e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 4.5538e-05, 4.5538e-05, 1.2827e-04, 4.5538e-05,
          5.2452e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05,
          1.7107e-05, 8.0884e-05, 4.5538e-05, 8.0884e-05, 1.2827e-04, 1.7107e-05,
          1.7107e-05, 1.2827e-04, 5.2452e-05, 1.7107e-05, 1.7107e-05, 7.4089e-05,
          4.5538e-05, 1.7107e-05, 4.5538e-05, 5.7697e-04, 1.7107e-05, 4.5538e-05,
          1.7107e-05, 2.3615e-04, 8.0884e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 1.7107e-05, 5.7697e-04,
          7.4089e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.4089e-05, 1.7107e-05,
          1.7107e-05, 1.2827e-04, 4.5538e-05, 4.5538e-05, 1.0943e-04, 1.5688e-04,
          7.4089e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.5688e-04, 4.5538e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.1672e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 4.5538e-05, 7.4089e-05, 7.4089e-05,
          1.7107e-05, 1.7107e-05, 5.2452e-05, 9.9838e-05, 1.5688e-04, 2.3556e-04,
          1.7107e-05, 1.7107e-05, 6.4492e-05, 2.3556e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 8.0884e-05, 1.2434e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          5.2452e-05, 1.7107e-05, 4.5538e-05, 1.5688e-04, 3.6073e-04, 2.0707e-04,
          1.7107e-05, 1.7107e-05, 1.5688e-04, 1.7107e-05, 2.0707e-04, 4.5538e-05,
          1.7107e-05, 1.2827e-04, 2.0707e-04, 1.2827e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.6417e-04, 9.9838e-05,
          7.4089e-05, 5.2452e-05, 1.2827e-04, 1.2827e-04, 1.7107e-05, 5.2452e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 2.3556e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 8.0884e-05, 7.4089e-05, 4.9639e-04, 1.7107e-05, 1.7107e-05,
          1.2827e-04, 6.0797e-04, 9.9838e-05, 1.5688e-04, 1.7107e-05, 1.7107e-05,
          4.9639e-04, 1.7107e-05, 2.0707e-04, 1.0943e-04, 5.2452e-05, 2.3556e-04,
          9.9838e-05, 1.7107e-05, 6.0797e-04, 2.6417e-04, 2.0707e-04, 2.0707e-04,
          3.7527e-04, 2.3556e-04, 3.4690e-04, 1.7107e-05, 1.7107e-05, 5.3263e-04,
          1.7107e-05, 4.9639e-04, 6.4039e-04, 7.4089e-05, 1.7107e-05, 2.0707e-04,
          2.3556e-04, 7.4089e-05, 1.5688e-04, 2.0707e-04, 1.7107e-05, 4.9639e-04,
          1.7107e-05, 4.5538e-05, 4.0388e-04, 4.5538e-05, 6.9189e-04, 2.0707e-04,
          4.3964e-04, 2.1672e-04, 6.4492e-05, 1.7107e-05, 7.4089e-05, 7.4089e-05,
          6.3801e-04, 1.2827e-04, 1.7107e-05, 1.7107e-05, 4.6802e-04, 5.2452e-05,
          6.7186e-04, 4.3917e-04, 6.0940e-04, 5.2452e-05, 4.5538e-05, 4.5538e-05,
          8.0884e-05, 4.6802e-04, 4.5538e-05, 2.4319e-04, 1.7107e-05, 1.7107e-05,
          1.5962e-04, 6.4492e-05, 9.9838e-05, 4.5538e-05, 1.8132e-04, 6.4492e-05,
          2.6417e-04, 3.8934e-04, 1.7107e-05, 9.9838e-05, 2.0707e-04, 1.0929e-03,
          1.8132e-04, 2.0707e-04, 4.9639e-04, 4.5538e-05, 1.7107e-05, 3.4690e-04,
          5.7936e-04, 4.6802e-04, 6.4492e-05, 3.4690e-04, 3.4690e-04, 1.7107e-05,
          5.2452e-05, 5.2452e-05, 1.7107e-05, 1.7107e-05, 4.0507e-04, 2.4652e-04,
          1.7107e-05, 2.9373e-04, 2.1803e-04, 6.1369e-04, 2.1803e-04, 2.6536e-04,
          5.2452e-05, 5.2452e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.8495e-03, 7.3303e-02, 2.8854e-02, 3.5583e-02, 6.6833e-02,
          6.0760e-02, 2.1790e-02, 4.8180e-03, 3.4389e-03, 2.3804e-02, 3.9185e-02,
          1.1642e-02, 5.5237e-03, 2.7390e-03, 7.5912e-03, 5.5656e-03, 7.4387e-03,
          5.0888e-03, 3.4733e-03, 1.4906e-03, 2.5139e-03, 3.2387e-03, 6.9847e-03,
          4.4037e-02, 1.1490e-02, 1.8509e-02, 1.3878e-02, 2.0008e-03, 7.3738e-03,
          3.6850e-03, 8.1024e-03, 2.6131e-03, 1.2352e-02, 1.2817e-02, 5.2261e-03,
          1.2337e-02, 3.2253e-03, 2.1759e-02, 1.0574e-02, 1.9646e-03, 4.0619e-02,
          1.0033e-02, 9.7580e-03, 7.2670e-03, 8.9035e-03, 9.7122e-03, 1.8951e-02,
          3.2425e-03, 2.8744e-03, 9.7275e-03, 5.9967e-03, 5.6305e-03, 4.8409e-03,
          1.7672e-03, 2.9526e-03, 1.5306e-03, 7.8659e-03, 5.4970e-03, 7.7400e-03,
          3.7804e-03, 4.9477e-03, 5.3291e-03, 8.9798e-03, 2.4948e-02, 4.2786e-02,
          2.6779e-02, 2.2263e-02, 4.1809e-02, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 5.7101e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 5.7101e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 5.7101e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.8930e-04, 5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05,
          1.7107e-05, 1.7107e-05, 1.8930e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.8930e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 5.7101e-05, 1.7107e-05, 5.7101e-05, 1.8930e-04, 1.7107e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 6.7616e-04, 5.7101e-05, 1.7107e-05,
          1.7107e-05, 1.7941e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 6.7616e-04,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.8930e-04, 1.7107e-05, 5.7101e-05, 1.7107e-05, 2.1219e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05,
          1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05,
          1.7107e-05, 1.7107e-05, 5.7101e-05, 5.7101e-05, 1.7107e-05, 2.1219e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.9026e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 3.0017e-04,
          1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 5.7101e-05, 2.1219e-04, 1.8930e-04, 1.7107e-05,
          5.7101e-05, 5.7101e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 2.2626e-04, 1.7107e-05, 1.8930e-04, 1.7107e-05, 5.7101e-05,
          5.7101e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05,
          2.1219e-04, 1.7107e-05, 1.8930e-04, 1.8930e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 2.6631e-04, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7214e-04, 1.7107e-05, 5.7101e-05, 1.7107e-05, 5.7101e-05,
          1.8930e-04, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 6.3610e-04, 1.8930e-04, 1.7107e-05, 1.7107e-05,
          1.8930e-04, 3.4022e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 3.4523e-04,
          1.7107e-05, 4.9639e-04, 3.4523e-04, 1.7107e-05, 1.7107e-05, 4.3249e-04,
          5.4932e-04, 1.7107e-05, 2.6631e-04, 1.7107e-05, 1.7107e-05, 1.9026e-04,
          1.7107e-05, 1.7107e-05, 1.8930e-04, 1.7107e-05, 2.6631e-04, 4.3249e-04,
          3.4022e-04, 3.4022e-04, 1.7107e-05, 1.7107e-05, 4.2129e-04, 1.0452e-03,
          4.2129e-04, 1.4937e-04, 1.7107e-05, 1.7107e-05, 3.4523e-04, 2.2626e-04,
          1.4610e-03, 4.2129e-04, 5.4932e-04, 5.7101e-05, 5.7101e-05, 1.7107e-05,
          5.7101e-05, 1.9026e-04, 5.7101e-05, 1.4937e-04, 1.7107e-05, 1.7107e-05,
          3.0017e-04, 1.7107e-05, 1.7107e-05, 5.7101e-05, 1.3285e-03, 1.7107e-05,
          7.0429e-04, 3.4451e-04, 5.7101e-05, 1.7107e-05, 1.7107e-05, 6.3610e-04,
          4.2129e-04, 3.0017e-04, 6.2752e-04, 5.7101e-05, 1.7107e-05, 1.7107e-05,
          4.3249e-04, 6.8188e-04, 1.7107e-05, 3.0017e-04, 3.0017e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 5.7101e-05, 1.7107e-05, 3.8362e-04, 1.7107e-05, 1.4937e-04,
          1.7107e-05, 1.7107e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3054e-02, 6.8909e-02, 3.6072e-02, 2.5696e-02, 1.0321e-01,
          5.0659e-02, 2.8931e-02, 2.5330e-03, 2.4071e-03, 6.4583e-03, 9.8114e-03,
          4.4785e-03, 3.1528e-03, 4.1084e-03, 3.6736e-03, 2.7580e-03, 5.6114e-03,
          3.1281e-03, 2.2182e-03, 5.5962e-03, 7.5874e-03, 7.1983e-03, 3.9978e-03,
          1.4275e-02, 9.3536e-03, 2.0248e-02, 6.1035e-03, 5.0812e-03, 3.7422e-03,
          4.0703e-03, 3.4523e-03, 2.5101e-03, 4.5433e-03, 2.8934e-03, 2.7752e-03,
          9.6664e-03, 7.5607e-03, 6.7024e-03, 3.2959e-03, 4.7150e-03, 2.6184e-02,
          5.7343e-02, 3.0197e-02, 5.5817e-02, 1.9836e-02, 4.4632e-03, 9.6741e-03,
          5.1842e-03, 9.4528e-03, 2.0081e-02, 4.4136e-03, 3.3607e-03, 2.4490e-03,
          3.5458e-03, 2.7657e-03, 4.2763e-03, 1.1131e-02, 6.5422e-03, 1.0742e-02,
          8.1940e-03, 6.6452e-03, 9.4604e-03, 9.9869e-03, 2.6489e-02, 1.8417e-02,
          1.8265e-02, 9.9487e-03, 7.3776e-03, 6.6101e-02, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05, 1.2696e-05, 9.7513e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04,
          1.2696e-05, 1.2696e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 9.7513e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          8.4162e-05, 1.2696e-05, 1.2696e-05, 5.0783e-04, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.3590e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 1.2696e-05, 1.6892e-04,
          8.4162e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6975e-04,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04, 1.6892e-04, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.6892e-04, 1.6892e-04, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 1.2696e-05, 8.4162e-05, 9.7513e-05, 1.2696e-05, 1.2696e-05,
          1.2696e-05, 9.7513e-05, 1.2696e-05, 9.7513e-05, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 1.2696e-05, 1.2696e-05, 8.4162e-05, 9.7513e-05, 1.2696e-05,
          9.7513e-05, 9.7513e-05, 4.3631e-04, 9.7513e-05, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 8.4162e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05, 5.0879e-04,
          1.2696e-05, 9.7632e-05, 5.0879e-04, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 1.2696e-05, 1.6892e-04, 9.7513e-05, 1.2696e-05, 5.0879e-04,
          1.2696e-05, 1.2696e-05, 1.6892e-04, 1.2696e-05, 8.4162e-05, 9.7513e-05,
          9.7513e-05, 1.6892e-04, 1.2696e-05, 1.2696e-05, 8.4162e-05, 8.4162e-05,
          8.4162e-05, 4.3631e-04, 1.2696e-05, 1.2696e-05, 5.0879e-04, 9.7513e-05,
          1.6892e-04, 1.6892e-04, 1.2696e-05, 1.2696e-05, 8.4162e-05, 1.2696e-05,
          8.4162e-05, 5.0783e-04, 8.4162e-05, 1.0691e-03, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.6892e-04, 1.2696e-05,
          1.6892e-04, 5.0783e-04, 9.7513e-05, 9.7513e-05, 1.2696e-05, 4.3631e-04,
          1.6892e-04, 1.2696e-05, 5.0783e-04, 1.2696e-05, 1.2696e-05, 9.7513e-05,
          4.3631e-04, 9.7513e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05, 1.2696e-05,
          9.7513e-05, 9.7513e-05, 9.7513e-05, 6.5088e-04, 9.7513e-05, 1.2696e-05,
          1.2696e-05, 9.7513e-05, 9.7513e-05, 6.3038e-04, 9.7513e-05, 9.7513e-05,
          1.2696e-05, 1.2696e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1482e-02, 6.2866e-02, 3.4607e-02, 2.5116e-02, 1.0175e-01,
          4.4800e-02, 2.8366e-02, 2.2335e-03, 2.0771e-03, 6.3171e-03, 1.0666e-02,
          4.4098e-03, 2.7351e-03, 4.2267e-03, 3.3016e-03, 2.4128e-03, 4.6577e-03,
          2.9602e-03, 2.1095e-03, 5.3635e-03, 5.9128e-03, 6.5880e-03, 3.5496e-03,
          1.5038e-02, 7.9346e-03, 2.0447e-02, 5.5504e-03, 4.4403e-03, 3.1776e-03,
          4.0054e-03, 2.9812e-03, 2.1381e-03, 3.9825e-03, 2.6913e-03, 2.2602e-03,
          7.4768e-03, 6.8550e-03, 5.9547e-03, 3.0155e-03, 4.2267e-03, 2.3041e-02,
          5.7587e-02, 2.7451e-02, 5.7220e-02, 1.5671e-02, 3.9749e-03, 8.1024e-03,
          4.8447e-03, 8.9645e-03, 2.1713e-02, 3.4981e-03, 2.6608e-03, 2.1839e-03,
          3.6030e-03, 2.4891e-03, 3.9978e-03, 1.0094e-02, 6.7177e-03, 1.0910e-02,
          8.4152e-03, 5.8556e-03, 9.0561e-03, 8.0948e-03, 2.5620e-02, 1.5793e-02,
          1.9363e-02, 9.3842e-03, 6.5002e-03, 4.7668e-02, 6.6833e-02, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.5891e-04, 7.1585e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05,
          1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          7.1585e-05, 1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.5891e-04,
          7.1585e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.5891e-04,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 7.1585e-05, 7.2479e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 7.1585e-05, 1.5891e-04, 1.5891e-04, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.5891e-04, 1.5891e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.0198e-04, 1.5891e-04, 1.0198e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.0198e-04, 1.4603e-05, 1.4603e-05,
          1.0198e-04, 1.4603e-05, 1.4603e-05, 7.1585e-05, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.0198e-04, 1.0198e-04, 1.5891e-04, 1.4603e-05, 1.4603e-05,
          1.5891e-04, 7.1585e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05, 1.5986e-04,
          1.4603e-05, 1.5903e-04, 1.5998e-04, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.0198e-04, 1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05, 1.5998e-04,
          1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05, 7.1585e-05, 1.0198e-04,
          1.4603e-05, 1.5891e-04, 1.4603e-05, 1.4603e-05, 1.5891e-04, 7.1585e-05,
          7.1585e-05, 1.0198e-04, 1.4603e-05, 1.4603e-05, 5.7745e-04, 1.4603e-05,
          5.7650e-04, 1.5903e-04, 1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05,
          7.1585e-05, 1.5986e-04, 7.1585e-05, 1.0198e-04, 1.4603e-05, 1.4603e-05,
          1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.5891e-04, 1.4603e-05,
          5.7650e-04, 1.5891e-04, 1.0198e-04, 1.0198e-04, 1.4603e-05, 5.1928e-04,
          1.5891e-04, 1.4603e-05, 5.7650e-04, 1.4603e-05, 1.4603e-05, 1.0198e-04,
          1.0198e-04, 3.5877e-03, 1.4603e-05, 1.4603e-05, 1.4603e-05, 1.4603e-05,
          1.0198e-04, 1.0198e-04, 1.0198e-04, 5.9605e-04, 1.0198e-04, 1.0198e-04,
          1.4603e-05, 1.0198e-04, 1.0595e-03, 5.9128e-04, 1.0198e-04, 1.0595e-03,
          1.4603e-05, 1.4603e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1070e-02, 6.7200e-02, 3.5309e-02, 2.5467e-02, 9.4116e-02,
          5.1758e-02, 2.5345e-02, 2.5101e-03, 2.0962e-03, 7.0038e-03, 1.2375e-02,
          4.9248e-03, 2.9678e-03, 4.2076e-03, 3.1643e-03, 2.7161e-03, 5.1384e-03,
          3.2883e-03, 2.1172e-03, 4.1237e-03, 4.8370e-03, 7.1182e-03, 3.7041e-03,
          1.6006e-02, 7.7744e-03, 1.9485e-02, 6.1722e-03, 4.2686e-03, 3.5782e-03,
          4.1084e-03, 3.4313e-03, 2.4605e-03, 3.9177e-03, 2.5959e-03, 2.4719e-03,
          5.9586e-03, 5.7716e-03, 4.3144e-03, 2.7199e-03, 4.0131e-03, 1.6312e-02,
          3.9795e-02, 2.1729e-02, 4.5410e-02, 1.3496e-02, 4.5013e-03, 7.4234e-03,
          5.7220e-03, 9.3384e-03, 1.7868e-02, 4.0131e-03, 3.1605e-03, 2.4090e-03,
          3.7003e-03, 2.6360e-03, 3.9253e-03, 7.6447e-03, 7.3471e-03, 1.1108e-02,
          8.4763e-03, 6.0120e-03, 8.2855e-03, 6.8665e-03, 2.2064e-02, 1.3062e-02,
          2.3376e-02, 1.0292e-02, 6.0310e-03, 3.7842e-02, 4.2694e-02, 8.5144e-02,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 8.1360e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 8.1360e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 6.6578e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          6.6578e-05, 6.6578e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05, 6.6578e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04,
          1.5795e-05, 1.5795e-05, 1.3220e-04, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          6.6578e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 6.6578e-05, 1.5795e-05, 1.3220e-04, 6.6578e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.3220e-04, 1.5795e-05, 1.5795e-05, 4.8423e-04, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.6749e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 8.1360e-05, 1.5795e-05, 1.3220e-04,
          1.3220e-04, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.3220e-04, 1.5795e-05, 6.6578e-05, 1.5795e-05, 1.3220e-04,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 6.6578e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 8.1360e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 6.7532e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 8.1360e-05, 1.3220e-04, 1.3220e-04, 6.6578e-05,
          1.5795e-05, 1.5795e-05, 6.6817e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 6.6578e-05, 1.5795e-05, 1.3220e-04, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 6.6578e-05, 6.6578e-05,
          1.5795e-05, 1.5795e-05, 1.3220e-04, 1.3220e-04, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 6.6578e-05, 1.3220e-04, 6.6578e-05, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 6.7532e-05, 6.6578e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05,
          6.6578e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 6.6578e-05, 6.6578e-05,
          6.6578e-05, 6.6578e-05, 4.1866e-04, 1.3220e-04, 1.5795e-05, 1.5795e-05,
          1.3220e-04, 1.3220e-04, 4.1866e-04, 1.5795e-05, 1.5795e-05, 1.3340e-04,
          1.5795e-05, 1.3244e-04, 1.3340e-04, 1.5795e-05, 1.5795e-05, 1.5795e-05,
          6.6578e-05, 1.5795e-05, 1.3220e-04, 6.6578e-05, 1.5795e-05, 4.8542e-04,
          1.5795e-05, 1.5795e-05, 1.3244e-04, 6.6578e-05, 1.3220e-04, 6.6578e-05,
          6.6578e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 8.1360e-05,
          8.1360e-05, 3.8075e-04, 1.5795e-05, 1.5795e-05, 7.9966e-04, 6.6578e-05,
          4.4632e-04, 1.3220e-04, 1.5795e-05, 6.6578e-05, 6.6578e-05, 1.5795e-05,
          1.3220e-04, 1.3316e-04, 1.5795e-05, 3.8075e-04, 1.5795e-05, 1.5795e-05,
          1.5795e-05, 1.5795e-05, 1.5795e-05, 1.5795e-05, 1.3220e-04, 1.5795e-05,
          1.3220e-04, 4.4656e-04, 6.6578e-05, 6.6578e-05, 6.6578e-05, 1.5221e-03,
          1.3220e-04, 6.6578e-05, 4.8447e-04, 6.6578e-05, 1.5795e-05, 6.6578e-05,
          6.6578e-05, 3.8075e-04, 1.5795e-05, 6.6578e-05, 1.5795e-05, 1.5795e-05,
          6.6578e-05, 6.6578e-05, 6.6578e-05, 1.0300e-04, 3.8075e-04, 6.6578e-05,
          1.5795e-05, 6.6578e-05, 6.6578e-05, 5.7268e-04, 4.3488e-04, 4.8542e-04,
          1.5795e-05, 1.5795e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.3191e-02, 6.2042e-02, 4.6082e-02, 2.9312e-02, 1.0260e-01,
          4.4769e-02, 3.2135e-02, 2.6474e-03, 1.9436e-03, 5.4245e-03, 8.6441e-03,
          3.4256e-03, 3.1090e-03, 4.6425e-03, 3.1719e-03, 2.7790e-03, 6.3362e-03,
          3.8185e-03, 1.8940e-03, 5.6000e-03, 6.5727e-03, 1.0742e-02, 4.4174e-03,
          1.0742e-02, 7.1869e-03, 2.3804e-02, 7.2517e-03, 5.4398e-03, 3.8204e-03,
          4.3030e-03, 3.6545e-03, 2.8381e-03, 4.0665e-03, 2.3804e-03, 2.7695e-03,
          6.2637e-03, 5.6305e-03, 4.4441e-03, 2.7943e-03, 4.6577e-03, 1.5533e-02,
          3.3447e-02, 1.6800e-02, 3.2074e-02, 1.3237e-02, 5.0735e-03, 8.0490e-03,
          7.4577e-03, 1.4427e-02, 2.1667e-02, 5.0087e-03, 3.9902e-03, 2.4776e-03,
          4.5204e-03, 3.5744e-03, 5.9471e-03, 7.4463e-03, 6.2408e-03, 8.1558e-03,
          7.4081e-03, 7.3509e-03, 9.3307e-03, 8.8654e-03, 1.2299e-02, 1.2390e-02,
          1.0048e-02, 5.3139e-03, 4.9477e-03, 3.1647e-02, 3.3569e-02, 5.1605e-02,
          6.5491e-02, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.1086e-04, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04,
          1.5914e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 8.1778e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.7679e-04, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04,
          1.7679e-04, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 8.1778e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.7679e-04,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.1086e-04,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.1086e-04, 8.1778e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.7679e-04, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 8.1778e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 8.1778e-05, 1.7679e-04, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 8.1778e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.5914e-05, 1.1086e-04, 8.1778e-05, 8.1778e-05,
          8.1778e-05, 8.1778e-05, 8.1778e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.1086e-04, 8.1778e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04,
          1.5914e-05, 1.7679e-04, 1.7846e-04, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.7679e-04, 8.1778e-05, 1.5914e-05, 1.7679e-04,
          1.5914e-05, 1.5914e-05, 1.7679e-04, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.7679e-04,
          1.5914e-05, 8.1778e-05, 1.5914e-05, 8.1778e-05, 1.7679e-04, 8.1778e-05,
          1.3561e-03, 1.7679e-04, 1.5914e-05, 8.1778e-05, 8.1778e-05, 1.5914e-05,
          8.1778e-05, 8.1778e-05, 1.5914e-05, 8.1778e-05, 1.5914e-05, 1.5914e-05,
          1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.7679e-04, 1.5914e-05,
          8.1778e-05, 1.7679e-04, 8.1778e-05, 8.1778e-05, 1.5914e-05, 8.1778e-05,
          1.7679e-04, 8.1778e-05, 1.7679e-04, 8.1778e-05, 8.1778e-05, 8.1778e-05,
          8.1778e-05, 6.2227e-04, 1.5914e-05, 1.5914e-05, 1.5914e-05, 1.5914e-05,
          8.1778e-05, 8.1778e-05, 8.1778e-05, 1.0657e-04, 8.1778e-05, 1.5914e-05,
          1.5914e-05, 8.1778e-05, 8.1778e-05, 5.2214e-04, 3.6573e-04, 4.3154e-04,
          1.5914e-05, 1.5914e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1940e-02, 7.2144e-02, 2.5253e-02, 4.8004e-02, 6.5063e-02,
          7.7393e-02, 2.5452e-02, 6.1874e-03, 5.0201e-03, 3.5492e-02, 3.8391e-02,
          1.7334e-02, 7.9422e-03, 2.4929e-03, 1.0864e-02, 6.2065e-03, 1.0063e-02,
          6.0272e-03, 3.6430e-03, 9.4271e-04, 1.4315e-03, 1.0767e-03, 7.5531e-03,
          4.3427e-02, 9.4910e-03, 2.1072e-02, 1.9348e-02, 2.0733e-03, 1.1520e-02,
          3.7308e-03, 9.7961e-03, 3.1681e-03, 1.5839e-02, 1.7441e-02, 6.2141e-03,
          1.0597e-02, 3.1624e-03, 1.8539e-02, 1.1848e-02, 1.3437e-03, 1.8219e-02,
          3.4122e-03, 2.9697e-03, 2.2049e-03, 3.5992e-03, 1.0658e-02, 1.8112e-02,
          1.3895e-03, 1.3704e-03, 9.3002e-03, 6.8665e-03, 6.4125e-03, 6.1874e-03,
          1.8530e-03, 3.0537e-03, 1.5640e-03, 5.6686e-03, 4.3068e-03, 5.9280e-03,
          3.3112e-03, 4.0016e-03, 2.9583e-03, 5.2834e-03, 7.4196e-03, 3.1372e-02,
          6.4354e-03, 7.2632e-03, 1.0361e-02, 1.6823e-03, 1.4057e-03, 1.6565e-03,
          1.6232e-03, 7.9956e-02, 1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          6.8843e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05,
          6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 2.7370e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 2.9922e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 6.8843e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05,
          6.8843e-05, 1.7166e-05, 1.7166e-05, 2.9922e-04, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05, 2.9922e-04,
          6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 4.3058e-04,
          1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05,
          6.8843e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 6.8843e-05, 3.7909e-04, 6.8843e-05, 1.7166e-05, 6.8843e-05,
          6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 2.2578e-04, 1.7166e-05,
          6.8843e-05, 1.7166e-05, 2.9922e-04, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7416e-04, 1.7166e-05, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          2.2578e-04, 1.7166e-05, 3.7909e-04, 6.8843e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 2.3518e-03, 2.7370e-04, 2.2209e-04, 1.7166e-05,
          2.7370e-04, 4.3058e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 2.2578e-04, 7.2956e-05, 1.7166e-05, 1.7166e-05, 1.7416e-04,
          2.2578e-04, 1.7166e-05, 6.8843e-05, 3.7909e-04, 1.7166e-05, 6.8843e-05,
          1.7166e-05, 1.7166e-05, 2.9922e-04, 1.7166e-05, 7.4148e-04, 3.7909e-04,
          3.7909e-04, 6.8843e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 6.8843e-05,
          7.4148e-04, 1.7166e-05, 1.7166e-05, 1.7166e-05, 6.8843e-05, 5.3310e-04,
          4.5609e-04, 2.9922e-04, 7.4148e-04, 1.7166e-05, 6.8843e-05, 1.7166e-05,
          6.8843e-05, 6.6090e-04, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 5.8794e-04, 1.7416e-04,
          4.5609e-04, 6.8843e-05, 1.7166e-05, 1.7166e-05, 1.7416e-04, 2.2209e-04,
          5.8460e-04, 3.7909e-04, 4.3058e-04, 1.7166e-05, 1.7166e-05, 2.2209e-04,
          6.0940e-04, 9.8133e-04, 1.7166e-05, 3.7909e-04, 3.7909e-04, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05, 1.7166e-05,
          1.7166e-05, 1.7166e-05, 2.4748e-04, 4.5085e-04, 1.7166e-05, 2.4748e-04,
          1.7166e-05, 1.7166e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0498e-02, 6.5002e-02, 3.6499e-02, 3.2959e-02, 1.1072e-01,
          4.8157e-02, 3.0060e-02, 1.0780e-02, 3.3073e-03, 2.5654e-03, 3.8700e-03,
          1.7262e-03, 2.3232e-03, 3.8223e-03, 2.3804e-03, 4.8790e-03, 3.8757e-03,
          2.2621e-03, 1.7481e-03, 2.7924e-03, 2.2907e-03, 1.9836e-03, 2.3479e-03,
          3.6678e-03, 2.5234e-03, 2.3117e-02, 8.0032e-03, 1.0994e-02, 3.6297e-03,
          4.3678e-03, 1.2962e-02, 2.8515e-03, 3.1605e-03, 1.4477e-03, 8.5297e-03,
          5.4741e-03, 5.5733e-03, 2.2030e-03, 2.1687e-03, 3.7441e-03, 2.0237e-03,
          2.5673e-03, 1.8950e-03, 2.2564e-03, 1.3199e-03, 2.3232e-03, 2.6646e-03,
          1.8425e-03, 3.3340e-03, 1.6678e-02, 4.6234e-02, 2.5742e-02, 5.3558e-03,
          3.6774e-02, 2.1118e-02, 6.6223e-02, 6.3324e-03, 9.6359e-03, 6.1302e-03,
          6.7902e-03, 6.3515e-03, 2.0599e-02, 7.6370e-03, 1.8129e-03, 3.3798e-03,
          2.7084e-03, 1.0891e-03, 1.3056e-03, 4.0016e-03, 2.7504e-03, 4.0894e-03,
          4.2419e-03, 4.0054e-03, 1.4978e-01, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 9.8526e-05, 5.3763e-05, 5.3763e-05,
          5.3763e-05, 9.8526e-05, 5.3763e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          5.3763e-05, 1.5581e-04, 5.3763e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 9.8526e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.3268e-04, 1.9610e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05,
          1.5581e-04, 5.3763e-05, 1.9610e-05, 1.3268e-04, 1.3268e-04, 1.9610e-05,
          1.9610e-05, 1.3268e-04, 9.8526e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05,
          5.3763e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 9.8526e-05,
          1.3268e-04, 9.8526e-05, 1.3268e-04, 1.3268e-04, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 1.5581e-04, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          5.3763e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05, 5.3763e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 9.8526e-05, 1.9610e-05,
          9.8526e-05, 5.3763e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 5.3763e-05, 1.5581e-04, 1.5581e-04, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 9.8526e-05, 1.3268e-04, 1.3268e-04,
          5.3763e-05, 9.8526e-05, 9.8526e-05, 1.9610e-05, 5.3763e-05, 1.5581e-04,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 9.8526e-05, 1.3268e-04, 3.9816e-04, 5.3763e-05,
          5.3763e-05, 1.3268e-04, 5.0449e-04, 9.8526e-05, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 5.3763e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05,
          1.9610e-05, 9.8526e-05, 1.9610e-05, 9.8526e-05, 1.9610e-05, 9.8526e-05,
          1.3268e-04, 9.8526e-05, 1.9610e-05, 9.8526e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 5.3763e-05, 5.3763e-05, 5.3763e-05, 9.8526e-05,
          5.3763e-05, 5.3763e-05, 4.7708e-04, 9.8526e-05, 9.8526e-05, 1.3268e-04,
          9.8526e-05, 9.8526e-05, 9.8526e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05,
          5.3763e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.5581e-04, 1.9610e-05,
          5.3763e-05, 1.9610e-05, 1.5581e-04, 7.4005e-04, 1.9610e-05, 1.9610e-05,
          9.8526e-05, 9.8526e-05, 9.8526e-05, 1.3268e-04, 5.0449e-04, 1.9610e-05,
          1.9610e-05, 9.8526e-05, 1.9610e-05, 5.3763e-05, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05,
          1.5581e-04, 1.5581e-04, 1.9610e-05, 5.3763e-05, 1.9610e-05, 2.3460e-04,
          2.3460e-04, 2.3460e-04, 2.0051e-04, 9.8526e-05, 1.9610e-05, 1.9610e-05,
          2.0051e-04, 1.3268e-04, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.5581e-04,
          1.9610e-05, 1.5581e-04, 1.5581e-04, 1.9610e-05, 1.9610e-05, 9.8526e-05,
          1.9610e-05, 1.9610e-05, 9.8526e-05, 1.9610e-05, 2.5864e-03, 1.5581e-04,
          9.8526e-05, 1.3268e-04, 1.5581e-04, 1.3268e-04, 9.8526e-05, 5.0449e-04,
          9.8526e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 5.3763e-05,
          1.9610e-05, 1.9610e-05, 5.3763e-05, 3.3641e-04, 1.5581e-04, 1.9610e-05,
          1.2165e-04, 1.5581e-04, 9.8526e-05, 5.3763e-05, 1.9610e-05, 1.9610e-05,
          7.4244e-04, 1.5581e-04, 1.9610e-05, 2.3460e-04, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.2165e-04, 1.9610e-05,
          1.2165e-04, 1.5581e-04, 1.5581e-04, 1.9610e-05, 1.9610e-05, 4.3845e-04,
          2.0051e-04, 9.8526e-05, 5.0020e-04, 1.3268e-04, 5.3883e-04, 8.1015e-04,
          2.3460e-04, 1.2165e-04, 9.8526e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05, 1.2165e-04, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.2165e-04, 1.9610e-05, 1.9610e-05, 1.9610e-05, 1.9610e-05,
          1.9610e-05, 1.9610e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.9504e-03, 6.9092e-02, 2.3834e-02, 2.2736e-02, 5.7831e-02,
          5.4260e-02, 1.3359e-02, 2.5726e-02, 6.0272e-03, 5.6572e-03, 7.1754e-03,
          3.3264e-03, 4.4746e-03, 3.8548e-03, 4.2114e-03, 1.6083e-02, 7.7133e-03,
          4.5395e-03, 4.0970e-03, 1.1902e-03, 1.1797e-03, 1.1549e-03, 4.5280e-03,
          7.4539e-03, 3.6316e-03, 1.2077e-02, 1.1345e-02, 3.9291e-03, 8.2626e-03,
          5.4588e-03, 4.6417e-02, 3.7060e-03, 6.5994e-03, 3.5267e-03, 1.8906e-02,
          6.3705e-03, 2.8858e-03, 4.3182e-03, 4.3106e-03, 2.1343e-03, 2.9850e-03,
          1.6394e-03, 1.2827e-03, 1.3218e-03, 1.1120e-03, 5.2185e-03, 4.0703e-03,
          1.6298e-03, 1.3914e-03, 6.3820e-03, 5.5725e-02, 6.5430e-02, 9.2926e-03,
          1.6479e-02, 3.7445e-02, 1.8097e-02, 4.7569e-03, 9.7961e-03, 5.1498e-03,
          4.4022e-03, 6.1760e-03, 1.5228e-02, 8.9569e-03, 2.0466e-03, 5.0125e-03,
          2.8038e-03, 1.5507e-03, 1.8654e-03, 1.0033e-03, 8.1205e-04, 1.2827e-03,
          1.3599e-03, 9.3155e-03, 3.6835e-02, 1.0614e-01, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 6.7830e-05, 4.0412e-05, 6.7830e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 4.0412e-05, 4.0412e-05, 9.5248e-05,
          4.0412e-05, 4.0412e-05, 2.3222e-04, 4.0412e-05, 1.0252e-04, 1.0252e-04,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          4.0412e-05, 1.1814e-04, 6.7592e-05, 1.9968e-05, 4.0412e-05, 4.0412e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 9.5248e-05, 6.7592e-05,
          1.2302e-04, 2.9826e-04, 2.9826e-04, 9.5248e-05, 1.9968e-05, 1.9968e-05,
          4.0412e-05, 4.7147e-05, 1.9968e-05, 1.9968e-05, 4.0412e-05, 4.0412e-05,
          1.9968e-05, 6.7592e-05, 6.7592e-05, 4.0412e-05, 1.9968e-05, 1.9968e-05,
          9.5248e-05, 6.7592e-05, 1.9968e-05, 1.5700e-04, 1.2207e-04, 2.4378e-04,
          4.0412e-05, 2.3222e-04, 1.0252e-04, 1.9968e-05, 1.9968e-05, 3.7885e-04,
          6.7592e-05, 4.0412e-05, 4.0412e-05, 6.7592e-05, 6.7592e-05, 4.0412e-05,
          6.7592e-05, 4.0412e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 3.2544e-04,
          6.7592e-05, 2.4378e-04, 4.0412e-05, 3.7360e-04, 1.4973e-04, 1.4973e-04,
          4.0412e-05, 1.0252e-04, 1.1814e-04, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          6.7592e-05, 6.7592e-05, 4.0412e-05, 4.0412e-05, 6.7592e-05, 4.0412e-05,
          1.9968e-05, 4.7147e-05, 1.9968e-05, 4.7147e-05, 1.0163e-04, 1.9968e-05,
          1.2302e-04, 9.5248e-05, 9.5248e-05, 1.9968e-05, 1.9968e-05, 1.0252e-04,
          1.1814e-04, 1.7786e-04, 1.9968e-05, 4.7147e-05, 6.7830e-05, 4.0412e-05,
          1.9968e-05, 4.7147e-05, 4.0412e-05, 1.1814e-04, 1.4532e-04, 4.0412e-05,
          1.9968e-05, 6.7592e-05, 1.2243e-04, 3.3712e-04, 6.7592e-05, 1.7786e-04,
          1.2243e-04, 1.7929e-04, 2.9826e-04, 1.9968e-05, 2.0015e-04, 1.1814e-04,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 3.7909e-04, 4.0412e-05, 2.9683e-04,
          4.7147e-05, 6.7592e-05, 1.9968e-05, 4.0412e-05, 4.0412e-05, 4.0412e-05,
          4.0412e-05, 2.7084e-04, 3.2544e-04, 2.0504e-04, 2.2745e-04, 4.0412e-05,
          1.2207e-04, 2.5940e-04, 3.8028e-04, 2.0504e-04, 1.9968e-05, 9.5248e-05,
          4.7147e-05, 6.7592e-05, 6.7592e-05, 4.0412e-05, 6.7592e-05, 2.9659e-04,
          6.7592e-05, 4.7147e-05, 4.7147e-05, 1.9968e-05, 4.0412e-05, 6.7592e-05,
          3.2544e-04, 1.4532e-04, 1.2970e-04, 2.0504e-04, 1.2970e-04, 3.3712e-04,
          2.5940e-04, 3.2544e-04, 1.5700e-04, 1.0252e-04, 1.9968e-05, 4.3607e-04,
          4.7147e-05, 2.9659e-04, 4.7147e-05, 4.7147e-05, 1.9968e-05, 2.9683e-04,
          4.7147e-05, 1.9968e-05, 1.9968e-05, 7.5626e-04, 9.5248e-05, 1.2970e-04,
          1.2243e-04, 1.2243e-04, 3.3712e-04, 2.7084e-04, 3.4857e-04, 2.5940e-04,
          2.3222e-04, 2.0063e-04, 3.2544e-04, 1.2243e-04, 4.7147e-05, 2.9683e-04,
          6.7592e-05, 4.7147e-05, 4.7147e-05, 1.9968e-05, 1.4532e-04, 1.2481e-04,
          8.2064e-04, 6.7592e-05, 1.4532e-04, 2.3568e-04, 4.7147e-05, 1.2243e-04,
          2.7084e-04, 2.7084e-04, 2.7084e-04, 1.7786e-04, 3.4857e-04, 1.2970e-04,
          2.4378e-04, 1.2302e-04, 1.9968e-05, 4.0412e-05, 4.7147e-05, 1.4532e-04,
          4.0412e-05, 2.9659e-04, 4.7147e-05, 4.0412e-05, 1.9968e-05, 4.0412e-05,
          6.7592e-05, 6.7592e-05, 1.8835e-04, 4.0412e-05, 1.9968e-05, 3.7217e-04,
          2.8276e-04, 1.4532e-04, 1.2970e-04, 2.0504e-04, 2.4378e-04, 4.7147e-05,
          1.2302e-04, 1.2243e-04, 6.7592e-05, 4.7147e-05, 6.7592e-05, 4.5657e-04,
          6.7592e-05, 7.2861e-04, 4.3392e-04, 6.7592e-05, 1.9968e-05, 4.7147e-05,
          1.4532e-04, 1.9968e-05, 1.0252e-04, 2.7084e-04, 3.2640e-04, 5.1117e-04,
          4.0412e-05, 9.5248e-05, 2.8276e-04, 1.2243e-04, 1.2970e-04, 3.2568e-04,
          2.9135e-04, 4.5657e-04, 4.7147e-05, 4.7147e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 4.1413e-04, 3.2640e-04, 5.1117e-04, 1.4532e-04,
          1.9968e-05, 1.9968e-05, 4.7147e-05, 6.7592e-05, 1.9968e-05, 4.0412e-05,
          3.8195e-04, 3.1805e-04, 1.9968e-05, 8.6927e-04, 1.9968e-05, 4.0412e-05,
          4.3607e-04, 2.7609e-04, 4.7147e-05, 1.9968e-05, 1.9968e-05, 4.7147e-05,
          1.9968e-05, 9.6560e-04, 7.9489e-04, 4.7147e-05, 6.7592e-05, 1.0185e-03,
          4.7147e-05, 2.7084e-04, 5.1117e-04, 4.0412e-05, 9.9564e-04, 4.8661e-04,
          6.8045e-04, 1.9968e-05, 3.8195e-04, 2.7084e-04, 1.8835e-04, 1.9968e-05,
          4.0412e-05, 9.7632e-05, 1.1814e-04, 9.7632e-05, 2.8872e-04, 4.0412e-05,
          1.9968e-05, 1.1814e-04, 9.7632e-05, 3.4070e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 9.7632e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.5073e-03, 5.5847e-02, 2.7832e-02, 1.9714e-02, 7.5439e-02,
          3.9948e-02, 1.6113e-02, 2.0340e-02, 3.8300e-03, 2.5520e-03, 4.3373e-03,
          1.8930e-03, 2.3994e-03, 3.8586e-03, 2.7256e-03, 8.7814e-03, 3.3169e-03,
          2.4300e-03, 2.4433e-03, 1.5402e-03, 1.3361e-03, 1.6222e-03, 2.6836e-03,
          4.3449e-03, 3.0613e-03, 1.3573e-02, 5.4512e-03, 4.5547e-03, 3.5248e-03,
          4.5319e-03, 2.5421e-02, 2.1477e-03, 2.9259e-03, 1.5926e-03, 8.6823e-03,
          4.4022e-03, 3.3054e-03, 2.0275e-03, 2.1782e-03, 2.5520e-03, 1.8158e-03,
          1.9178e-03, 1.6146e-03, 1.9913e-03, 1.0710e-03, 2.5826e-03, 2.5978e-03,
          1.7662e-03, 2.1591e-03, 1.0155e-02, 3.8116e-02, 3.7628e-02, 5.6534e-03,
          5.7800e-02, 2.9282e-02, 3.0914e-02, 3.9005e-03, 1.7365e-02, 5.4893e-03,
          5.8975e-03, 6.3057e-03, 1.7532e-02, 7.6675e-03, 1.9188e-03, 4.0550e-03,
          3.5133e-03, 1.3037e-03, 1.6460e-03, 2.0409e-03, 2.1343e-03, 2.6264e-03,
          2.6188e-03, 4.3716e-03, 6.8054e-02, 6.2927e-02, 1.1743e-01, 1.1861e-05,
          1.1861e-05, 1.1861e-05, 4.5717e-05, 4.4882e-05, 1.1861e-05, 1.1861e-05,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 1.0759e-04, 4.4942e-05, 4.4882e-05,
          7.0810e-05, 1.1861e-05, 2.0576e-04, 4.4882e-05, 1.1861e-05, 1.1861e-05,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05,
          7.0810e-05, 1.2243e-04, 7.1704e-05, 1.1861e-05, 4.4882e-05, 1.1861e-05,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 4.4882e-05, 1.1921e-05,
          1.1462e-04, 8.1658e-05, 1.1861e-05, 4.4882e-05, 1.1861e-05, 1.4458e-03,
          4.4882e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 4.4882e-05, 1.1861e-05,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 4.4882e-05, 3.7789e-05, 1.1861e-05,
          7.0810e-05, 4.4942e-05, 1.1861e-05, 8.1718e-05, 1.1921e-05, 1.1861e-05,
          7.0810e-05, 7.0810e-05, 8.1658e-05, 1.1861e-05, 1.1861e-05, 7.0810e-05,
          1.2755e-05, 1.1861e-05, 4.4882e-05, 1.1861e-05, 4.5776e-05, 7.0810e-05,
          4.4882e-05, 4.4882e-05, 3.7789e-05, 3.7789e-05, 1.1861e-05, 8.1658e-05,
          1.4055e-04, 8.1658e-05, 2.1529e-04, 4.4882e-05, 2.0576e-04, 2.3174e-04,
          4.4882e-05, 8.1658e-05, 1.2243e-04, 1.1861e-05, 1.1861e-05, 1.1861e-05,
          7.1704e-05, 1.1861e-05, 4.4882e-05, 4.4882e-05, 7.0870e-05, 4.4882e-05,
          3.7789e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 8.1658e-05, 1.1861e-05,
          8.1658e-05, 4.4942e-05, 4.4942e-05, 3.7789e-05, 1.0759e-04, 8.1658e-05,
          7.0810e-05, 1.1861e-05, 1.1861e-05, 1.2755e-05, 4.5776e-05, 4.4942e-05,
          3.7789e-05, 1.1861e-05, 4.4882e-05, 1.9789e-04, 1.2338e-04, 4.4942e-05,
          1.1861e-05, 3.7789e-05, 1.1861e-05, 1.0765e-04, 1.0765e-04, 8.1658e-05,
          1.1921e-05, 1.0759e-04, 3.7718e-04, 1.1861e-05, 7.0870e-05, 1.2243e-04,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 7.1704e-05, 4.4942e-05, 4.5776e-05,
          1.1861e-05, 7.0870e-05, 1.1861e-05, 7.0810e-05, 4.4882e-05, 4.4882e-05,
          7.0810e-05, 8.1658e-05, 6.5136e-04, 1.0759e-04, 2.3174e-04, 4.4882e-05,
          3.7849e-05, 3.7789e-05, 1.0759e-04, 1.0759e-04, 1.1861e-05, 2.0576e-04,
          1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 1.1921e-05, 4.4882e-05,
          1.1861e-05, 1.1861e-05, 3.7789e-05, 1.1861e-05, 4.4942e-05, 7.1704e-05,
          8.1658e-05, 2.6679e-04, 1.1861e-05, 1.1861e-05, 3.7789e-05, 1.0759e-04,
          3.7849e-05, 3.0756e-04, 1.1861e-05, 8.1658e-05, 1.1861e-05, 3.7789e-05,
          1.1861e-05, 4.4942e-05, 1.1921e-05, 1.1921e-05, 1.1861e-05, 1.2755e-05,
          1.1921e-05, 1.1861e-05, 1.1861e-05, 4.4942e-05, 7.0870e-05, 8.1658e-05,
          3.7849e-05, 4.5776e-05, 2.3174e-04, 8.1658e-05, 8.1658e-05, 2.3174e-04,
          3.7789e-05, 1.0759e-04, 8.1658e-05, 2.3174e-04, 1.1861e-05, 7.1704e-05,
          7.1704e-05, 1.2755e-05, 1.2755e-05, 1.1861e-05, 7.1704e-05, 3.7849e-05,
          4.5776e-05, 1.1861e-05, 1.5564e-03, 4.2605e-04, 1.1861e-05, 1.1921e-05,
          8.1658e-05, 1.1861e-05, 8.1658e-05, 1.4067e-04, 1.0759e-04, 3.7789e-05,
          1.1861e-05, 2.1529e-04, 3.7789e-05, 4.4882e-05, 1.2755e-05, 4.4942e-05,
          1.1921e-05, 3.8683e-05, 3.7849e-05, 3.7789e-05, 1.1861e-05, 4.4882e-05,
          7.0810e-05, 7.0810e-05, 3.7789e-05, 7.0870e-05, 3.7789e-05, 1.9717e-04,
          1.0759e-04, 1.9705e-04, 1.8227e-04, 1.4067e-04, 1.1861e-05, 1.1861e-05,
          2.1529e-04, 4.4882e-05, 7.0810e-05, 1.1861e-05, 1.2755e-05, 7.1704e-05,
          5.8174e-04, 7.1704e-05, 7.1704e-05, 4.4882e-05, 1.1861e-05, 3.7789e-05,
          3.7789e-05, 8.1658e-05, 1.0759e-04, 5.8174e-04, 9.1124e-04, 2.3258e-04,
          8.1658e-05, 1.1462e-04, 4.6563e-04, 2.1529e-04, 1.1861e-05, 1.0759e-04,
          3.7789e-05, 3.7789e-05, 1.1861e-05, 1.1861e-05, 8.9467e-05, 3.7789e-05,
          3.7789e-05, 1.1861e-05, 2.1529e-04, 1.5616e-04, 7.1704e-05, 3.7789e-05,
          8.9467e-05, 1.2243e-04, 3.7789e-05, 4.4882e-05, 3.7789e-05, 1.1861e-05,
          2.1529e-04, 2.3258e-04, 1.1861e-05, 1.1235e-04, 1.1861e-05, 1.1861e-05,
          3.7789e-05, 1.1861e-05, 1.1861e-05, 1.1861e-05, 3.7789e-05, 1.1861e-05,
          7.3385e-04, 7.0870e-05, 7.6675e-04, 1.1235e-04, 1.1861e-05, 1.8227e-04,
          9.6273e-04, 1.1861e-05, 2.3174e-04, 4.4882e-05, 1.0805e-03, 5.1069e-04,
          6.4039e-04, 1.6403e-04, 1.1462e-04, 4.0793e-04, 1.1861e-05, 1.1861e-05,
          3.7789e-05, 8.9467e-05, 1.2243e-04, 8.9467e-05, 3.2330e-04, 1.2243e-04,
          1.1861e-05, 4.9829e-04, 8.9467e-05, 3.7789e-05, 3.7789e-05, 1.1861e-05,
          8.9467e-05, 2.4891e-04], dtype=torch.float16),
  tensor([0.0000e+00, 8.0643e-03, 5.9509e-02, 2.4307e-02, 1.9196e-02, 7.1411e-02,
          4.1016e-02, 1.3725e-02, 2.1591e-02, 4.5853e-03, 3.1242e-03, 4.9591e-03,
          2.1400e-03, 2.8877e-03, 4.1122e-03, 3.1643e-03, 1.0536e-02, 4.8409e-03,
          2.8133e-03, 3.4847e-03, 1.5564e-03, 1.3056e-03, 1.3790e-03, 3.5191e-03,
          4.9553e-03, 3.2635e-03, 1.1726e-02, 6.6757e-03, 3.9482e-03, 4.3564e-03,
          5.1003e-03, 2.8824e-02, 2.6703e-03, 3.4714e-03, 2.0428e-03, 1.2268e-02,
          5.0240e-03, 3.2654e-03, 2.4529e-03, 2.7237e-03, 2.7714e-03, 2.1076e-03,
          1.6260e-03, 1.3523e-03, 1.5221e-03, 9.9754e-04, 3.1834e-03, 2.9049e-03,
          1.8845e-03, 2.0351e-03, 8.0109e-03, 4.4495e-02, 4.6021e-02, 7.0343e-03,
          4.6143e-02, 3.7781e-02, 3.4790e-02, 4.5433e-03, 1.4603e-02, 5.7487e-03,
          5.5923e-03, 6.6032e-03, 1.4626e-02, 7.2670e-03, 1.7776e-03, 4.0512e-03,
          2.8038e-03, 1.3866e-03, 1.5869e-03, 1.3990e-03, 1.3485e-03, 1.8272e-03,
          2.0046e-03, 5.6000e-03, 5.2612e-02, 6.9519e-02, 3.9612e-02, 6.6772e-02,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          9.5963e-06, 4.1306e-05, 4.1306e-05, 9.5963e-06, 8.2254e-05, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 8.2254e-05, 9.5963e-06, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 8.2254e-05, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 8.2254e-05, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1396e-04,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 8.2254e-05,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06, 1.1396e-04,
          9.5963e-06, 9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06,
          1.1396e-04, 9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1396e-04, 9.5963e-06,
          4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06, 4.1306e-05, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 8.2254e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06,
          4.1306e-05, 9.5963e-06, 4.1306e-05, 9.5963e-06, 8.2254e-05, 8.2254e-05,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1396e-04, 4.1306e-05, 9.5963e-06,
          9.5963e-06, 4.1306e-05, 9.5963e-06, 4.1306e-05, 4.1306e-05, 9.5963e-06,
          9.5963e-06, 9.5963e-06, 7.9393e-04, 9.5963e-06, 4.1306e-05, 4.1306e-05,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1516e-04, 8.2254e-05, 9.5963e-06,
          9.5963e-06, 1.1396e-04, 9.5963e-06, 1.1396e-04, 1.1396e-04, 1.1396e-04,
          4.1306e-05, 9.5963e-06, 3.7479e-04, 4.1306e-05, 1.1396e-04, 4.1306e-05,
          4.1306e-05, 4.1306e-05, 3.4285e-04, 4.1306e-05, 9.5963e-06, 8.2254e-05,
          4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06, 9.5963e-06, 8.2254e-05,
          9.5963e-06, 9.5963e-06, 4.1306e-05, 4.1306e-05, 8.2254e-05, 1.1396e-04,
          4.9210e-04, 3.7479e-04, 9.5963e-06, 4.1306e-05, 4.1306e-05, 3.7479e-04,
          4.1306e-05, 9.5963e-06, 9.5963e-06, 3.4285e-04, 4.1306e-05, 4.1306e-05,
          9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06, 9.5963e-06, 4.1306e-05,
          4.1306e-05, 9.5963e-06, 4.1306e-05, 1.1396e-04, 1.1396e-04, 4.1306e-05,
          4.1306e-05, 9.6560e-06, 1.1396e-04, 3.7479e-04, 3.7479e-04, 4.1306e-05,
          9.5963e-06, 4.1306e-05, 4.6062e-04, 1.1396e-04, 9.5963e-06, 4.1366e-05,
          1.1396e-04, 9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1396e-04, 4.1306e-05,
          1.1402e-04, 4.1306e-05, 4.1306e-05, 2.9516e-04, 9.5963e-06, 4.1306e-05,
          9.5963e-06, 9.5963e-06, 9.5963e-06, 1.1396e-04, 3.7479e-04, 4.1306e-05,
          9.5963e-06, 3.7479e-04, 4.1306e-05, 4.1306e-05, 9.5963e-06, 1.1396e-04,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1306e-05,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 1.1396e-04, 4.1306e-05, 4.1306e-05,
          3.7479e-04, 4.1306e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1366e-05, 1.1396e-04,
          4.1306e-05, 1.1402e-04, 1.1516e-04, 1.1396e-04, 9.5963e-06, 4.1306e-05,
          4.1306e-05, 9.5963e-06, 4.1306e-05, 1.1415e-03, 1.4753e-03, 1.1396e-04,
          4.1306e-05, 8.2254e-05, 1.1396e-04, 1.1396e-04, 4.1306e-05, 3.7479e-04,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1306e-05, 4.1306e-05,
          4.1306e-05, 4.1306e-05, 1.1396e-04, 9.5963e-06, 1.1402e-04, 4.1306e-05,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 1.1396e-04, 4.1306e-05, 9.5963e-06,
          4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1306e-05, 9.5963e-06, 9.5963e-06,
          4.1306e-05, 4.1306e-05, 4.1306e-05, 9.5963e-06, 4.1306e-05, 4.1306e-05,
          4.1306e-05, 1.1396e-04, 1.1396e-04, 4.1306e-05, 4.1306e-05, 1.1320e-03,
          4.1306e-05, 4.1306e-05, 1.2140e-03, 4.1306e-05, 3.7479e-04, 8.2588e-04,
          1.1415e-03, 4.1306e-05, 9.5963e-06, 4.1306e-05, 4.1306e-05, 9.5963e-06,
          9.5963e-06, 2.9516e-04, 4.1306e-05, 2.9516e-04, 8.6069e-04, 2.9516e-04,
          9.5963e-06, 2.9516e-04, 2.9516e-04, 5.6410e-04, 9.5963e-06, 5.6410e-04,
          9.5963e-06, 2.9516e-04], dtype=torch.float16),
  tensor([0.0000e+00, 8.5068e-03, 5.7098e-02, 2.5238e-02, 2.3163e-02, 8.4229e-02,
          3.6865e-02, 1.7365e-02, 1.3641e-02, 3.8910e-03, 3.0441e-03, 4.6959e-03,
          2.0447e-03, 2.9449e-03, 4.6997e-03, 3.0098e-03, 7.8125e-03, 5.4436e-03,
          2.8515e-03, 2.5635e-03, 2.3689e-03, 1.6851e-03, 1.5144e-03, 2.6207e-03,
          4.4441e-03, 2.9106e-03, 1.4610e-02, 6.0959e-03, 5.5695e-03, 5.0087e-03,
          5.0163e-03, 2.4734e-02, 3.1319e-03, 3.8509e-03, 2.0924e-03, 9.6893e-03,
          5.9395e-03, 4.4708e-03, 2.4490e-03, 2.7065e-03, 3.8052e-03, 2.1133e-03,
          1.9722e-03, 1.5306e-03, 1.7681e-03, 1.1091e-03, 3.3131e-03, 2.7981e-03,
          1.9875e-03, 2.3193e-03, 1.0406e-02, 4.3549e-02, 3.9825e-02, 5.4741e-03,
          2.6169e-02, 3.3203e-02, 4.6783e-02, 6.3438e-03, 1.2016e-02, 7.3090e-03,
          6.9504e-03, 7.2594e-03, 1.4160e-02, 6.7177e-03, 1.8826e-03, 3.8643e-03,
          2.5711e-03, 1.1120e-03, 1.4629e-03, 3.0003e-03, 2.9373e-03, 3.1319e-03,
          3.3016e-03, 4.6730e-03, 6.2042e-02, 5.5695e-02, 1.8616e-02, 2.1011e-02,
          7.5867e-02, 1.9372e-05, 6.3300e-05, 1.9372e-05, 1.9610e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 7.1883e-05, 6.3062e-05, 6.3062e-05,
          6.3062e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          6.3062e-05, 1.1557e-04, 6.3300e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05,
          1.1557e-04, 6.3062e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05,
          7.1883e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 6.3062e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 3.1853e-04,
          1.1557e-04, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05, 2.5058e-04,
          1.9372e-05, 1.9372e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.1557e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.1557e-04, 1.9372e-05,
          7.1883e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 3.1853e-04, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.1557e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3300e-05, 6.3062e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 4.2534e-04, 1.1557e-04, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 4.1437e-04, 1.1557e-04, 1.9372e-05,
          6.3062e-05, 7.1883e-05, 3.1853e-04, 1.9372e-05, 1.1557e-04, 1.1557e-04,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3300e-05, 6.3062e-05, 6.3300e-05,
          1.9372e-05, 1.1557e-04, 1.9372e-05, 1.1557e-04, 1.9372e-05, 6.3062e-05,
          1.9372e-05, 1.9372e-05, 6.4564e-04, 1.1557e-04, 3.0303e-04, 1.9372e-05,
          6.3062e-05, 6.3062e-05, 3.1853e-04, 1.9372e-05, 1.9372e-05, 6.3062e-05,
          1.9372e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05,
          1.9610e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 6.3062e-05,
          3.1853e-04, 3.7074e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 3.7074e-04,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 3.4690e-04,
          1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9610e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05, 1.1557e-04, 3.7074e-04,
          1.1557e-04, 6.3300e-05, 3.0303e-04, 3.1853e-04, 6.4564e-04, 2.0695e-04,
          1.9372e-05, 7.1883e-05, 1.9372e-05, 3.0303e-04, 1.9372e-05, 1.1581e-04,
          6.3062e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.1557e-04, 7.1883e-05,
          6.3062e-05, 1.9372e-05, 7.1883e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 7.1883e-05, 3.7074e-04, 1.9372e-05,
          1.9372e-05, 3.7074e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 3.9077e-04,
          1.9372e-05, 1.9372e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05, 6.3062e-05,
          7.1883e-05, 7.1883e-05, 2.9445e-04, 6.3062e-05, 7.1883e-05, 3.4690e-04,
          7.1883e-05, 1.1557e-04, 7.1883e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05,
          7.1883e-05, 6.3062e-05, 3.4690e-04, 1.9372e-05, 1.9372e-05, 1.1581e-04,
          3.4690e-04, 6.3062e-05, 3.9077e-04, 6.3062e-05, 1.9372e-05, 1.9372e-05,
          7.1883e-05, 1.9372e-05, 1.9372e-05, 2.9445e-04, 3.4690e-04, 5.7840e-04,
          1.9372e-05, 6.3062e-05, 1.1557e-04, 6.3062e-05, 7.1883e-05, 3.1853e-04,
          2.9445e-04, 1.1557e-04, 7.1883e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 3.3808e-04, 1.9372e-05, 3.9077e-04, 7.1883e-05,
          7.1883e-05, 7.1883e-05, 1.9372e-05, 6.3062e-05, 1.9372e-05, 1.9372e-05,
          1.1557e-04, 5.7840e-04, 1.9372e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05,
          7.1883e-05, 1.9372e-05, 7.1883e-05, 1.9372e-05, 1.9372e-05, 2.9445e-04,
          7.1883e-05, 3.9077e-04, 9.0170e-04, 1.9372e-05, 1.9372e-05, 5.8317e-04,
          7.1883e-05, 2.9445e-04, 5.7840e-04, 1.9372e-05, 6.8951e-04, 3.7074e-04,
          3.4690e-04, 7.1883e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          7.1883e-05, 7.1883e-05, 7.1883e-05, 7.1883e-05, 3.8147e-04, 7.1883e-05,
          1.9372e-05, 4.2534e-04, 7.1883e-05, 3.2616e-04, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 7.1883e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1208e-02, 6.0944e-02, 2.3209e-02, 2.3010e-02, 9.9304e-02,
          5.5573e-02, 1.9623e-02, 4.0245e-03, 4.9896e-03, 1.3466e-02, 1.6098e-02,
          8.1482e-03, 4.5509e-03, 4.0741e-03, 6.2790e-03, 2.1839e-03, 5.3864e-03,
          2.5234e-03, 1.3924e-03, 3.1052e-03, 2.5311e-03, 1.7872e-03, 2.5196e-03,
          1.4992e-02, 7.4577e-03, 1.4030e-02, 6.7177e-03, 6.8741e-03, 4.8180e-03,
          4.6310e-03, 4.9019e-03, 2.9793e-03, 7.5989e-03, 3.9062e-03, 4.7646e-03,
          2.2598e-02, 1.8967e-02, 6.1760e-03, 3.7880e-03, 3.8166e-03, 1.1787e-02,
          1.0139e-02, 7.3204e-03, 6.7329e-03, 5.7106e-03, 3.3970e-03, 7.7667e-03,
          1.9217e-03, 2.7523e-03, 1.2993e-02, 1.1612e-02, 6.9275e-03, 4.2686e-03,
          8.1558e-03, 5.4550e-03, 8.0109e-03, 5.0934e-02, 1.6144e-02, 4.1931e-02,
          1.9379e-02, 7.0877e-03, 5.7106e-03, 4.3831e-03, 1.2657e-02, 1.6922e-02,
          7.2746e-03, 4.1046e-03, 3.4103e-03, 5.9357e-03, 6.0654e-03, 7.2784e-03,
          5.2109e-03, 1.4748e-02, 1.1551e-02, 7.6752e-03, 4.5166e-03, 4.0627e-03,
          6.7215e-03, 7.9590e-02, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.3685e-04, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 8.7142e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 8.0109e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 8.7142e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 8.0109e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          8.0109e-05, 1.4389e-04, 1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.4389e-04,
          1.9014e-05, 1.9014e-05, 1.3685e-04, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.3685e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.3685e-04, 1.9014e-05, 1.9014e-05, 4.0245e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05, 4.0245e-04,
          7.5758e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 7.5758e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.3685e-04,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.4389e-04,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 7.5758e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 7.5758e-05, 7.5758e-05, 1.3685e-04, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          8.7142e-05, 1.9014e-05, 5.6171e-04, 2.0504e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.3685e-04, 1.9014e-05,
          7.5758e-05, 1.9014e-05, 1.3685e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 8.0109e-05, 1.3685e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 8.0109e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05,
          8.0109e-05, 1.9014e-05, 1.9014e-05, 7.5758e-05, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 7.7200e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05,
          8.2874e-04, 1.3685e-04, 4.0245e-04, 1.9014e-05, 1.9014e-05, 1.3685e-04,
          1.9014e-05, 1.3685e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05, 8.0109e-05,
          8.0109e-05, 1.9014e-05, 8.2874e-04, 8.0109e-05, 1.9014e-05, 2.0504e-04,
          1.9014e-05, 1.9014e-05, 4.7040e-04, 1.9014e-05, 1.9014e-05, 1.2474e-03,
          6.2275e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05, 1.3685e-04, 7.5758e-05,
          7.5758e-05, 8.0109e-05, 1.9014e-05, 1.9014e-05, 1.3685e-04, 1.9014e-05,
          1.7347e-03, 6.3753e-04, 1.9014e-05, 1.9014e-05, 8.0109e-05, 1.9014e-05,
          1.9014e-05, 2.0504e-04, 7.5758e-05, 3.1495e-04, 1.9014e-05, 1.9014e-05,
          1.9014e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 9.5415e-04, 1.9014e-05,
          3.7169e-04, 1.3685e-04, 1.9014e-05, 1.9014e-05, 1.9014e-05, 5.6505e-04,
          1.3685e-04, 8.0109e-05, 5.3787e-04, 1.9014e-05, 1.9014e-05, 2.9969e-04,
          8.0109e-05, 1.4935e-03, 1.9014e-05, 8.0109e-05, 8.0109e-05, 1.9014e-05,
          8.0109e-05, 1.9014e-05, 1.9014e-05, 1.9014e-05, 8.0109e-05, 8.0109e-05,
          1.9014e-05, 8.0109e-05, 8.0109e-05, 6.5565e-04, 3.1495e-04, 5.8079e-04,
          8.0109e-05, 1.9014e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0841e-02, 6.8481e-02, 2.1561e-02, 2.7313e-02, 8.4900e-02,
          5.9784e-02, 1.8127e-02, 3.2749e-03, 4.0207e-03, 1.4565e-02, 2.3468e-02,
          9.8877e-03, 3.6869e-03, 4.3106e-03, 4.7569e-03, 2.2774e-03, 4.9515e-03,
          2.7065e-03, 1.7061e-03, 1.6403e-03, 1.5945e-03, 1.2999e-03, 2.8534e-03,
          2.5146e-02, 6.9733e-03, 1.3710e-02, 7.9346e-03, 2.9678e-03, 4.7607e-03,
          5.6763e-03, 4.8828e-03, 2.3594e-03, 6.5536e-03, 4.5128e-03, 4.3488e-03,
          1.3435e-02, 9.2468e-03, 5.7030e-03, 3.9101e-03, 2.6054e-03, 1.2939e-02,
          1.0887e-02, 7.6370e-03, 9.4528e-03, 4.9438e-03, 3.3970e-03, 5.9967e-03,
          2.0676e-03, 1.9455e-03, 1.0399e-02, 7.8430e-03, 5.7259e-03, 4.0016e-03,
          4.2496e-03, 4.0359e-03, 3.7155e-03, 2.2827e-02, 1.1475e-02, 2.1133e-02,
          1.0468e-02, 5.0926e-03, 4.4479e-03, 3.9749e-03, 1.7853e-02, 1.4381e-02,
          3.3936e-02, 1.6937e-02, 5.8022e-03, 6.7673e-03, 6.3095e-03, 7.7934e-03,
          4.4975e-03, 1.9440e-02, 4.7722e-03, 6.2370e-03, 2.7943e-03, 3.0270e-03,
          4.5242e-03, 3.3081e-02, 8.8806e-02, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 5.6326e-05, 1.8418e-05, 3.6120e-05, 3.6120e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 3.6120e-05, 1.8418e-05, 3.6120e-05, 8.5115e-05,
          1.8418e-05, 1.8418e-05, 5.6326e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05, 3.6120e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          3.6120e-05, 1.8418e-05, 3.2973e-04, 1.8418e-05, 8.5115e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 7.4029e-05, 5.6326e-05, 3.6120e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 5.6326e-05, 7.4029e-05, 5.6326e-05,
          1.8418e-05, 1.8418e-05, 3.6120e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05,
          1.1814e-04, 1.2767e-04, 1.5604e-04, 1.8418e-05, 3.6120e-05, 5.6326e-05,
          1.8418e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 5.6326e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 5.6326e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 3.6120e-05, 3.6120e-05, 3.6120e-05, 1.8418e-05, 1.9443e-04,
          1.8418e-05, 3.6120e-05, 1.9395e-04, 3.6120e-05, 3.6120e-05, 3.6120e-05,
          1.8418e-05, 1.8418e-05, 5.6326e-05, 3.6120e-05, 7.4029e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 3.6120e-05,
          1.8418e-05, 5.6326e-05, 5.6326e-05, 1.1814e-04, 1.1814e-04, 3.6120e-05,
          1.8418e-05, 3.6120e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05, 5.6326e-05,
          7.4029e-05, 1.8418e-05, 1.8418e-05, 2.6894e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.1611e-04, 6.2585e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 7.4029e-05, 1.8418e-05, 2.4772e-04,
          7.4029e-05, 3.6120e-05, 1.8418e-05, 1.8418e-05, 5.6326e-05, 1.8418e-05,
          1.8418e-05, 7.4029e-05, 1.8418e-05, 3.6120e-05, 7.4029e-05, 1.0049e-04,
          7.4029e-05, 2.4533e-04, 1.8418e-05, 1.8418e-05, 1.5414e-04, 5.6326e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 3.6120e-05, 2.0766e-04,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 3.6120e-05, 5.6326e-05, 5.6326e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 3.6120e-05, 5.6326e-05, 1.0049e-04,
          2.4533e-04, 1.8418e-05, 1.1599e-04, 1.6022e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 2.4533e-04, 1.8418e-05, 3.6120e-05, 3.6120e-05, 1.3363e-04,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 3.6120e-05, 7.4029e-05, 1.0049e-04, 1.0049e-04, 3.8719e-04,
          1.8418e-05, 1.8418e-05, 6.2585e-05, 1.8418e-05, 3.8719e-04, 1.8418e-05,
          8.5115e-05, 1.1814e-04, 8.9502e-04, 1.5414e-04, 3.6120e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.9813e-04, 1.3363e-04,
          5.6326e-05, 1.8418e-05, 1.9395e-04, 1.9395e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 6.2585e-05, 2.4533e-04, 1.8418e-05,
          1.8418e-05, 1.5533e-04, 5.6326e-05, 2.9159e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 2.3592e-04, 1.6022e-04, 2.1577e-04, 3.6120e-05, 1.8418e-05,
          2.9159e-04, 1.8418e-05, 3.6073e-04, 1.5390e-04, 3.6120e-05, 1.6022e-04,
          1.6022e-04, 1.8418e-05, 1.1206e-03, 5.7220e-04, 3.4308e-04, 1.1599e-04,
          1.1148e-03, 2.0766e-04, 3.2854e-04, 1.8418e-05, 1.8418e-05, 1.7631e-04,
          1.8418e-05, 7.7772e-04, 2.7394e-04, 5.6326e-05, 3.6120e-05, 2.5368e-04,
          1.6022e-04, 7.4029e-05, 9.3174e-04, 4.8065e-04, 1.8418e-05, 2.2995e-04,
          3.6120e-05, 1.8418e-05, 6.9618e-04, 3.6120e-05, 3.8099e-04, 1.1206e-03,
          3.6073e-04, 1.5390e-04, 1.8418e-05, 1.8418e-05, 5.6028e-04, 7.8392e-04,
          1.7154e-04, 2.3103e-04, 3.6120e-05, 3.6120e-05, 2.7394e-04, 1.3363e-04,
          1.1234e-03, 1.0328e-03, 3.6073e-04, 3.6120e-05, 1.7548e-04, 1.8418e-05,
          1.4901e-04, 3.2759e-04, 5.6326e-05, 2.3103e-04, 1.8418e-05, 3.6120e-05,
          1.1599e-04, 1.8418e-05, 1.8418e-05, 7.4029e-05, 8.1015e-04, 1.3363e-04,
          3.4881e-04, 1.7631e-04, 1.8418e-05, 2.3103e-04, 1.9181e-04, 6.0940e-04,
          1.7154e-04, 4.8065e-04, 4.0269e-04, 7.4029e-05, 1.8418e-05, 5.5552e-04,
          3.2854e-04, 1.2426e-03, 1.8692e-04, 5.3787e-04, 5.5552e-04, 1.8418e-05,
          6.2585e-05, 1.2684e-03, 1.8418e-05, 3.0541e-04, 2.0754e-04, 1.0049e-04,
          3.6120e-05, 1.0049e-04, 2.0754e-04, 4.9448e-04, 2.0754e-04, 2.8348e-04,
          8.0228e-05, 6.2585e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.0790e-03, 7.0190e-02, 1.9485e-02, 2.0935e-02, 7.0618e-02,
          7.1777e-02, 1.3115e-02, 5.5161e-03, 4.3945e-03, 2.3392e-02, 2.8152e-02,
          1.3039e-02, 5.7831e-03, 2.4433e-03, 1.2070e-02, 4.4136e-03, 6.4850e-03,
          3.2940e-03, 2.2564e-03, 9.8419e-04, 1.2150e-03, 7.8917e-04, 4.7874e-03,
          3.0548e-02, 1.0811e-02, 1.0551e-02, 8.2092e-03, 1.8892e-03, 6.6757e-03,
          3.4275e-03, 8.4229e-03, 2.5406e-03, 1.6022e-02, 1.1406e-02, 6.8359e-03,
          2.4796e-02, 5.8098e-03, 9.8495e-03, 9.0027e-03, 1.6184e-03, 1.5961e-02,
          3.6869e-03, 3.4485e-03, 2.6169e-03, 3.8757e-03, 5.2872e-03, 1.2497e-02,
          1.1902e-03, 9.6846e-04, 6.4850e-03, 9.6970e-03, 8.3923e-03, 5.1842e-03,
          3.0365e-03, 4.7760e-03, 2.5330e-03, 1.4320e-02, 1.1734e-02, 1.6937e-02,
          8.0948e-03, 4.3945e-03, 3.5229e-03, 4.8599e-03, 8.5907e-03, 3.6804e-02,
          8.8120e-03, 8.1558e-03, 7.5188e-03, 2.1000e-03, 1.7014e-03, 1.8358e-03,
          1.6394e-03, 2.9678e-02, 3.9864e-03, 9.0485e-03, 2.3022e-03, 2.9106e-03,
          4.6844e-03, 1.6144e-02, 1.9562e-02, 7.0740e-02, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 7.1824e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 7.8797e-05,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 4.5681e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 7.8797e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 1.2243e-04, 7.8797e-05,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 1.9741e-04,
          7.1824e-05, 7.7915e-04, 2.8205e-04, 2.8133e-05, 2.8133e-05, 1.2243e-04,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 7.8797e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 5.5122e-04,
          2.8133e-05, 2.8133e-05, 4.1819e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 1.2243e-04, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 1.2243e-04, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 5.0211e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          1.2243e-04, 2.8133e-05, 2.8133e-05, 5.0879e-04, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.4140e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 7.8797e-05, 2.8133e-05, 6.2847e-04,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 1.2243e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05, 1.2243e-04,
          2.8133e-05, 1.9741e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 1.9741e-04, 2.8133e-05, 2.8133e-05, 7.7915e-04,
          2.8133e-05, 2.8133e-05, 7.8797e-05, 2.8133e-05, 2.8133e-05, 2.5105e-04,
          2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 1.2243e-04, 4.0936e-04,
          1.9741e-04, 2.8133e-05, 1.4603e-04, 1.4603e-04, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 1.9741e-04, 1.9741e-04, 2.8133e-05, 2.8133e-05, 1.4603e-04,
          2.8133e-05, 7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 7.8797e-05, 1.2243e-04, 1.2243e-04, 2.8133e-05,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 7.1764e-04, 2.3270e-04, 2.8133e-05, 7.8797e-05,
          7.8797e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 2.4021e-04, 7.1824e-05,
          2.5105e-04, 2.8133e-05, 2.2173e-04, 1.2243e-04, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 1.9741e-04, 2.8133e-05, 2.8133e-05, 1.9741e-04,
          1.9741e-04, 7.1824e-05, 2.1279e-04, 3.3951e-04, 2.8133e-05, 2.8133e-05,
          2.8133e-05, 3.7909e-04, 7.1824e-05, 2.4021e-04, 2.8133e-05, 2.8133e-05,
          5.4073e-04, 1.4603e-04, 4.5824e-04, 7.8797e-05, 7.1824e-05, 2.8133e-05,
          2.8133e-05, 2.8133e-05, 7.5865e-04, 8.0919e-04, 1.9741e-04, 1.9741e-04,
          7.5865e-04, 7.3004e-04, 4.2963e-04, 2.8133e-05, 2.8133e-05, 3.3975e-04,
          2.8133e-05, 9.6464e-04, 3.4094e-04, 7.8797e-05, 2.8133e-05, 2.8896e-04,
          2.4021e-04, 2.8133e-05, 6.3992e-04, 5.4836e-04, 2.8133e-05, 4.4990e-04,
          2.8133e-05, 2.8133e-05, 7.0906e-04, 7.8797e-05, 6.7616e-04, 7.5865e-04,
          9.6846e-04, 2.4021e-04, 2.8133e-05, 2.8133e-05, 1.2243e-04, 1.2243e-04,
          4.6349e-04, 1.6224e-04, 2.8133e-05, 2.8133e-05, 3.3975e-04, 2.8133e-05,
          6.2656e-04, 4.2963e-04, 4.5633e-04, 2.8133e-05, 2.1279e-04, 2.8133e-05,
          2.8133e-05, 6.6853e-04, 7.8797e-05, 7.1824e-05, 2.8133e-05, 1.9741e-04,
          2.4021e-04, 2.8133e-05, 2.8133e-05, 2.8133e-05, 4.1246e-04, 2.8896e-04,
          6.2656e-04, 2.4045e-04, 2.8133e-05, 1.6224e-04, 3.7909e-04, 1.3695e-03,
          5.9748e-04, 5.4836e-04, 9.0599e-04, 1.1849e-04, 2.8133e-05, 4.4918e-04,
          1.3838e-03, 1.0061e-03, 2.8777e-04, 5.4836e-04, 5.4836e-04, 2.8133e-05,
          7.1824e-05, 2.8133e-05, 2.8133e-05, 2.8133e-05, 4.9210e-04, 2.8133e-05,
          2.8133e-05, 1.2243e-04, 7.1824e-05, 5.4598e-04, 7.1824e-05, 3.6764e-04,
          7.1824e-05, 2.8133e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.2092e-03, 6.9336e-02, 2.1164e-02, 2.3468e-02, 7.5500e-02,
          6.5735e-02, 1.4267e-02, 3.4332e-03, 2.9335e-03, 1.4061e-02, 2.5665e-02,
          9.1476e-03, 3.3607e-03, 3.3112e-03, 4.5853e-03, 2.6684e-03, 3.8738e-03,
          2.4967e-03, 1.8511e-03, 1.1768e-03, 1.2913e-03, 9.9373e-04, 3.2139e-03,
          2.8549e-02, 6.2332e-03, 1.1475e-02, 9.5596e-03, 2.4071e-03, 3.9787e-03,
          4.3755e-03, 5.1994e-03, 1.9608e-03, 5.3291e-03, 4.4937e-03, 4.3335e-03,
          1.1185e-02, 8.8348e-03, 4.9400e-03, 3.5305e-03, 2.0123e-03, 1.0193e-02,
          6.0234e-03, 4.8103e-03, 5.6610e-03, 3.0651e-03, 3.2368e-03, 4.9286e-03,
          1.6031e-03, 1.4572e-03, 9.2239e-03, 6.9046e-03, 5.9128e-03, 3.2272e-03,
          4.1809e-03, 4.2229e-03, 3.0231e-03, 1.6525e-02, 1.7792e-02, 2.5177e-02,
          1.1749e-02, 4.3488e-03, 4.6463e-03, 3.9482e-03, 1.1520e-02, 1.1909e-02,
          2.5604e-02, 1.4381e-02, 5.5161e-03, 3.1738e-03, 3.2539e-03, 3.8071e-03,
          2.5826e-03, 1.8127e-02, 4.8561e-03, 6.6490e-03, 3.6469e-03, 3.6144e-03,
          4.7951e-03, 1.6815e-02, 4.6600e-02, 3.0701e-02, 8.8257e-02, 2.2471e-05,
          2.2471e-05, 5.5552e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 5.5552e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 5.5552e-05,
          5.5552e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 3.8671e-04, 5.5552e-05, 2.2471e-05, 2.2471e-05,
          2.2650e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 5.5552e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 5.5552e-05, 1.3053e-04,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          8.5652e-05, 5.9986e-04, 3.7074e-04, 2.2471e-05, 2.2471e-05, 8.5831e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05,
          2.2471e-05, 5.5552e-05, 2.2471e-05, 5.5552e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 1.6880e-04, 6.8092e-04,
          2.2471e-05, 2.2471e-05, 2.4629e-04, 1.4699e-04, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 5.5552e-05, 5.5552e-05, 8.5831e-05, 5.5552e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 8.5652e-05, 5.5552e-05, 8.5652e-05, 2.1017e-04, 1.4699e-04,
          2.2471e-05, 1.7977e-04, 2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05,
          5.5552e-05, 2.2471e-05, 5.5552e-05, 3.0756e-04, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 1.2779e-04, 5.2571e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 2.2471e-05, 5.1785e-04,
          8.5652e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 2.2471e-05,
          2.2471e-05, 1.6081e-04, 2.2471e-05, 2.1017e-04, 8.5652e-05, 1.6081e-04,
          5.5552e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 1.6880e-04, 5.5552e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 3.7098e-04,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 5.5552e-05, 1.2219e-04, 1.2219e-04,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 5.2571e-05, 1.6069e-04, 2.2745e-04,
          2.5964e-04, 2.2471e-05, 2.2471e-05, 1.2755e-04, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.5964e-04, 8.5652e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05,
          5.5552e-05, 2.2471e-05, 5.5552e-05, 6.9284e-04, 1.7989e-04, 1.2755e-04,
          2.2471e-05, 2.2471e-05, 1.6880e-04, 2.2471e-05, 3.6478e-04, 2.2471e-05,
          2.2471e-05, 2.1017e-04, 5.6458e-04, 8.5652e-05, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2471e-05, 2.2733e-04, 4.1485e-04,
          5.5552e-05, 1.8227e-04, 3.0732e-04, 4.3178e-04, 2.2471e-05, 5.2571e-05,
          2.2471e-05, 2.2471e-05, 2.2471e-05, 1.6069e-04, 2.2471e-05, 2.2471e-05,
          2.2471e-05, 4.6468e-04, 8.5652e-05, 4.4799e-04, 2.2471e-05, 2.2471e-05,
          8.5652e-05, 1.9798e-03, 3.5381e-04, 3.8695e-04, 1.4699e-04, 2.2471e-05,
          5.7507e-04, 9.7513e-05, 5.9128e-04, 2.2733e-04, 5.2571e-05, 1.2755e-04,
          1.9419e-04, 2.2471e-05, 8.7357e-04, 7.3528e-04, 3.6478e-04, 3.6478e-04,
          9.8991e-04, 4.3201e-04, 2.5511e-04, 2.2471e-05, 8.9169e-05, 3.1209e-04,
          2.2471e-05, 1.1320e-03, 4.8137e-04, 5.5552e-05, 2.2471e-05, 4.1485e-04,
          1.6069e-04, 5.5552e-05, 6.9952e-04, 7.1049e-04, 2.2471e-05, 3.1066e-04,
          1.4699e-04, 2.2471e-05, 9.2745e-04, 2.2471e-05, 5.8603e-04, 6.1941e-04,
          4.9210e-04, 5.5552e-05, 2.2471e-05, 2.2471e-05, 1.2219e-04, 7.2908e-04,
          3.8719e-04, 3.0732e-04, 2.2471e-05, 2.2471e-05, 4.8137e-04, 1.2755e-04,
          1.3762e-03, 1.2150e-03, 1.3053e-04, 5.2571e-05, 2.7394e-04, 2.2471e-05,
          2.7394e-04, 5.4789e-04, 5.5552e-05, 1.1349e-04, 2.2471e-05, 5.1117e-04,
          9.7513e-05, 2.2471e-05, 2.2471e-05, 1.8013e-04, 9.1887e-04, 3.5381e-04,
          6.0892e-04, 4.0603e-04, 2.2471e-05, 4.1485e-04, 4.1485e-04, 9.5034e-04,
          2.2161e-04, 5.5838e-04, 6.2513e-04, 2.4378e-04, 2.2471e-05, 7.1383e-04,
          7.7915e-04, 9.0647e-04, 2.0802e-04, 6.1941e-04, 4.9210e-04, 2.2471e-05,
          5.2571e-05, 5.2571e-05, 2.2471e-05, 2.9087e-04, 2.7418e-04, 8.5652e-05,
          2.2471e-05, 1.7989e-04, 5.2571e-05, 3.2115e-04, 1.4675e-04, 2.1327e-04,
          1.7703e-04, 5.2571e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.9956e-03, 6.5918e-02, 2.6672e-02, 1.8723e-02, 9.0637e-02,
          5.6458e-02, 1.5396e-02, 3.0975e-03, 3.0632e-03, 7.8506e-03, 1.2444e-02,
          5.6076e-03, 3.0727e-03, 2.9087e-03, 4.9362e-03, 2.2888e-03, 3.9940e-03,
          1.9369e-03, 1.5039e-03, 2.2488e-03, 1.8883e-03, 1.4238e-03, 2.2736e-03,
          1.2589e-02, 6.7291e-03, 1.2001e-02, 5.1422e-03, 3.3169e-03, 3.1776e-03,
          2.7409e-03, 4.5776e-03, 1.8768e-03, 6.0349e-03, 3.3035e-03, 4.6349e-03,
          1.9928e-02, 2.1164e-02, 3.8719e-03, 3.6297e-03, 3.2310e-03, 6.2294e-03,
          4.7760e-03, 3.5915e-03, 3.7746e-03, 2.6989e-03, 2.5043e-03, 5.3864e-03,
          1.4381e-03, 1.8635e-03, 1.5900e-02, 8.6975e-03, 5.1651e-03, 3.3379e-03,
          6.2866e-03, 4.6387e-03, 7.0915e-03, 2.7130e-02, 2.2217e-02, 6.9336e-02,
          4.5837e-02, 7.2021e-03, 8.1482e-03, 6.1150e-03, 5.1918e-03, 1.1360e-02,
          4.6921e-03, 2.2755e-03, 2.2316e-03, 3.5400e-03, 3.6125e-03, 4.3983e-03,
          4.3907e-03, 8.9951e-03, 9.7733e-03, 5.9395e-03, 4.7073e-03, 3.8414e-03,
          6.8779e-03, 2.0630e-02, 1.2726e-02, 1.8555e-02, 1.3947e-02, 8.8257e-02,
          2.5272e-05, 8.7440e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 9.3520e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 9.3520e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          1.5569e-04, 1.5569e-04, 9.3520e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05,
          2.5272e-05, 2.5272e-05, 1.5593e-04, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 1.5569e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          1.5569e-04, 2.5272e-05, 2.5272e-05, 1.5593e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 1.5593e-04,
          8.7440e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 1.5569e-04, 2.5272e-05, 2.5272e-05, 2.5272e-05, 1.5569e-04,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 8.7440e-05, 1.5628e-04,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5868e-05,
          2.5272e-05, 2.5272e-05, 8.7440e-05, 1.5569e-04, 1.5569e-04, 2.5272e-05,
          2.5272e-05, 2.5392e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 1.5593e-04, 2.5272e-05, 2.5988e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 5.0545e-04, 9.3520e-05,
          8.7440e-05, 2.5272e-05, 1.5593e-04, 2.7823e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.1613e-04, 1.5569e-04, 1.5569e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 7.2002e-04, 2.5272e-05, 1.5569e-04, 2.5272e-05, 2.5272e-05,
          5.0545e-04, 2.5272e-05, 3.7503e-04, 1.5593e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 8.4257e-04, 1.5569e-04, 2.5272e-05, 2.5272e-05,
          5.5504e-04, 5.0545e-04, 1.5569e-04, 2.5272e-05, 2.5272e-05, 1.5652e-04,
          2.5272e-05, 1.8835e-03, 1.5652e-04, 2.5272e-05, 2.5272e-05, 3.7503e-04,
          2.5272e-05, 2.5272e-05, 2.7823e-04, 4.4322e-04, 2.5272e-05, 1.5652e-04,
          2.5272e-05, 2.5272e-05, 9.0551e-04, 2.5272e-05, 2.5272e-05, 4.4322e-04,
          4.4322e-04, 8.7440e-05, 2.5272e-05, 2.5272e-05, 1.5569e-04, 8.7440e-05,
          8.7440e-05, 2.1613e-04, 2.5272e-05, 2.5272e-05, 1.5652e-04, 2.5272e-05,
          1.1711e-03, 8.2111e-04, 9.3520e-05, 2.5272e-05, 2.1613e-04, 2.5272e-05,
          2.5272e-05, 1.5652e-04, 2.5272e-05, 2.1613e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 2.5272e-05, 1.5569e-04, 3.7503e-04,
          1.1711e-03, 1.5593e-04, 2.5272e-05, 1.4782e-04, 3.7503e-04, 1.9360e-03,
          6.2799e-04, 3.7503e-04, 5.0545e-04, 2.5272e-05, 2.5272e-05, 2.1613e-04,
          5.6553e-04, 1.1082e-03, 1.4782e-04, 5.6553e-04, 3.7503e-04, 2.5272e-05,
          9.3520e-05, 9.3520e-05, 2.5272e-05, 4.0197e-04, 2.5272e-05, 2.5272e-05,
          2.5272e-05, 2.5272e-05, 2.5272e-05, 4.0197e-04, 4.7040e-04, 3.5930e-04,
          2.5272e-05, 2.5272e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.1177e-03, 6.2378e-02, 2.8305e-02, 2.2186e-02, 1.1780e-01,
          4.4464e-02, 2.2308e-02, 2.3365e-03, 2.0466e-03, 4.2152e-03, 6.2599e-03,
          2.8515e-03, 2.1667e-03, 2.7027e-03, 3.5477e-03, 1.7691e-03, 2.7466e-03,
          1.5182e-03, 1.1234e-03, 3.0518e-03, 2.5902e-03, 1.6050e-03, 1.7395e-03,
          6.5422e-03, 6.2714e-03, 1.6968e-02, 5.3024e-03, 3.9444e-03, 2.2888e-03,
          2.2583e-03, 3.4428e-03, 1.6441e-03, 4.3335e-03, 1.8005e-03, 3.4904e-03,
          1.5282e-02, 2.6337e-02, 3.3112e-03, 2.8553e-03, 3.7651e-03, 4.8218e-03,
          5.0964e-03, 3.7956e-03, 4.1389e-03, 2.6951e-03, 1.8225e-03, 4.2152e-03,
          1.3304e-03, 2.2907e-03, 2.1622e-02, 7.0000e-03, 4.1885e-03, 2.5558e-03,
          5.7526e-03, 3.7155e-03, 9.4376e-03, 2.8015e-02, 2.0096e-02, 5.6976e-02,
          4.6783e-02, 7.6065e-03, 9.7580e-03, 5.8212e-03, 4.0359e-03, 7.2250e-03,
          3.4790e-03, 1.5116e-03, 1.7242e-03, 4.0016e-03, 4.1084e-03, 4.7836e-03,
          5.1422e-03, 5.2414e-03, 1.0117e-02, 4.7455e-03, 4.1389e-03, 3.0804e-03,
          6.2561e-03, 2.0111e-02, 8.8196e-03, 9.9640e-03, 9.2773e-03, 5.1392e-02,
          8.1787e-02, 9.0182e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05,
          9.0778e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6224e-04, 1.8656e-05,
          9.0778e-05, 1.8656e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05,
          9.0182e-05, 9.0182e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05,
          1.8656e-05, 1.8656e-05, 1.6367e-04, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          9.0778e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6224e-04, 1.8656e-05, 1.8656e-05,
          9.0778e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.6224e-04, 1.8656e-05, 1.8656e-05, 1.6367e-04, 9.0778e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0778e-05, 1.6367e-04,
          9.0182e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.0182e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05, 1.6224e-04,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 9.0182e-05, 9.0182e-05, 9.0182e-05, 1.8656e-05,
          1.8656e-05, 2.0087e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.0778e-05, 1.8656e-05, 9.1612e-05, 1.8656e-05, 1.8656e-05,
          9.0182e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05,
          9.0182e-05, 1.8656e-05, 1.6367e-04, 6.1846e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 2.5153e-04, 1.6224e-04, 1.6224e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.0778e-05, 9.0778e-05, 9.0778e-05, 1.8656e-05, 1.8656e-05,
          1.6224e-04, 1.8656e-05, 1.8656e-05, 1.6224e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.0778e-05, 1.0490e-03, 1.6224e-04, 1.8656e-05, 1.8656e-05,
          7.3767e-04, 1.6224e-04, 1.6224e-04, 1.8656e-05, 1.8656e-05, 9.1612e-05,
          1.8656e-05, 1.6367e-04, 9.3877e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.6224e-04, 8.1587e-04, 1.8656e-05, 9.1612e-05,
          1.8656e-05, 1.8656e-05, 1.5364e-03, 9.0778e-05, 9.0778e-05, 8.8787e-04,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6224e-04, 1.6224e-04,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 2.5153e-04, 1.6367e-04, 9.0778e-05,
          1.6224e-04, 1.6224e-04, 9.0778e-05, 1.8656e-05, 9.0778e-05, 1.8656e-05,
          2.5153e-04, 9.1612e-05, 1.8656e-05, 2.5153e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 9.0182e-05, 1.6224e-04, 8.1587e-04,
          1.6224e-04, 9.1612e-05, 9.0778e-05, 2.5153e-04, 8.1587e-04, 3.1624e-03,
          1.6224e-04, 8.1587e-04, 9.1612e-05, 1.8656e-05, 9.0778e-05, 2.5153e-04,
          8.8787e-04, 1.7586e-03, 1.7929e-04, 1.0490e-03, 1.8656e-05, 1.8656e-05,
          9.0778e-05, 9.0778e-05, 3.8624e-04, 4.5919e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 3.5906e-04, 3.5906e-04, 7.2670e-04,
          1.8656e-05, 1.8656e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.9962e-03, 6.2561e-02, 2.3819e-02, 2.5726e-02, 6.3110e-02,
          5.0934e-02, 1.6235e-02, 1.6068e-02, 4.9706e-03, 4.9629e-03, 6.3553e-03,
          2.7485e-03, 3.5324e-03, 2.8172e-03, 4.2343e-03, 1.4610e-02, 6.3667e-03,
          4.2191e-03, 4.0398e-03, 1.6336e-03, 2.0256e-03, 1.9722e-03, 4.7035e-03,
          7.6103e-03, 4.6997e-03, 1.4519e-02, 8.5602e-03, 2.2602e-03, 5.9776e-03,
          3.0842e-03, 3.2196e-02, 2.6016e-03, 5.6610e-03, 3.3913e-03, 1.3268e-02,
          6.2485e-03, 2.9392e-03, 6.5231e-03, 5.0392e-03, 2.1744e-03, 7.4348e-03,
          4.4250e-03, 3.1719e-03, 2.9888e-03, 2.1973e-03, 4.9477e-03, 6.1913e-03,
          1.7548e-03, 1.8749e-03, 8.6975e-03, 2.1561e-02, 3.4424e-02, 5.8975e-03,
          8.4457e-03, 2.0569e-02, 8.6212e-03, 3.7498e-03, 8.6899e-03, 4.6196e-03,
          3.7594e-03, 6.6566e-03, 3.6194e-02, 2.9953e-02, 5.1994e-03, 1.0063e-02,
          4.2725e-03, 2.0599e-03, 2.6207e-03, 2.5711e-03, 2.3022e-03, 2.5768e-03,
          2.5978e-03, 8.6365e-03, 1.3870e-02, 4.1138e-02, 7.5531e-03, 7.8354e-03,
          1.6098e-02, 3.4142e-03, 2.9449e-03, 4.9095e-03, 3.1929e-03, 3.6831e-03,
          3.0499e-03, 1.2561e-01, 1.8597e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05,
          4.8995e-05, 1.8597e-05, 9.6738e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 4.8995e-05, 4.8995e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05,
          1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05,
          1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05, 4.8995e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 2.0051e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 4.8995e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 8.9943e-05, 1.8597e-05,
          1.8597e-05, 4.8995e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          4.8995e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 4.9233e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 2.3091e-04, 2.3091e-04, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 4.8995e-05, 1.2028e-04, 1.2028e-04, 1.8597e-05,
          4.9233e-05, 8.9943e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 2.0051e-04,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.2058e-04, 4.8995e-05, 4.9233e-05,
          1.8597e-05, 4.8995e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 4.8995e-05,
          4.8995e-05, 1.8597e-05, 8.9943e-05, 1.2028e-04, 2.7871e-04, 4.8995e-05,
          1.2028e-04, 1.2028e-04, 1.8597e-05, 4.8995e-05, 1.8597e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.2028e-04,
          8.9943e-05, 8.9943e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 2.0051e-04,
          1.2028e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 6.6710e-04,
          1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8895e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 6.4135e-04, 4.8995e-05, 1.8597e-05,
          1.2028e-04, 4.9233e-05, 2.7871e-04, 8.9943e-05, 8.9943e-05, 4.8995e-05,
          1.8597e-05, 4.8995e-05, 8.9943e-05, 1.2028e-04, 1.8597e-05, 7.1287e-04,
          4.8995e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05,
          6.6948e-04, 1.8597e-05, 1.8597e-05, 2.0051e-04, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 8.9943e-05, 8.9943e-05, 4.8995e-05, 8.9943e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.2028e-04,
          1.8597e-05, 8.9943e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05,
          1.8597e-05, 1.8597e-05, 8.9943e-05, 4.8995e-05, 1.8597e-05, 3.0231e-04,
          3.0231e-04, 2.3091e-04, 8.9943e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 4.8995e-05, 8.9943e-05, 1.8597e-05, 1.8597e-05, 5.1594e-04,
          8.9943e-05, 6.4135e-04, 5.1594e-04, 4.8995e-05, 1.8597e-05, 8.9943e-05,
          2.0051e-04, 1.8597e-05, 1.8597e-05, 8.9943e-05, 8.9943e-05, 5.1546e-04,
          1.8597e-05, 4.8995e-05, 8.9943e-05, 4.8995e-05, 1.8597e-05, 8.9943e-05,
          8.9943e-05, 6.9761e-04, 8.9943e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.2028e-04, 8.9943e-05, 5.1594e-04, 2.0051e-04,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 4.8995e-05, 1.8597e-05, 1.8597e-05,
          2.3091e-04, 1.2028e-04, 1.8597e-05, 8.5163e-04, 1.8597e-05, 1.8597e-05,
          2.7180e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 8.9943e-05,
          1.8597e-05, 5.1546e-04, 5.0879e-04, 1.8597e-05, 8.9943e-05, 8.9943e-05,
          1.8597e-05, 8.9943e-05, 5.1546e-04, 1.8597e-05, 3.2687e-04, 2.7180e-04,
          5.1546e-04, 1.8597e-05, 1.8597e-05, 8.9943e-05, 8.9943e-05, 1.8597e-05,
          2.0051e-04, 2.0051e-04, 2.3091e-04, 1.8597e-05, 2.0051e-04, 4.8995e-05,
          1.8597e-05, 2.3091e-04, 1.8597e-05, 3.7265e-04, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.5449e-03, 5.7892e-02, 2.6703e-02, 2.4078e-02, 1.0321e-01,
          4.2938e-02, 2.5864e-02, 5.0507e-03, 2.8191e-03, 2.7313e-03, 3.8052e-03,
          1.8158e-03, 2.1019e-03, 3.0003e-03, 2.8191e-03, 5.1613e-03, 3.3054e-03,
          2.3136e-03, 1.8711e-03, 3.1719e-03, 3.1338e-03, 2.7370e-03, 2.4014e-03,
          3.9597e-03, 4.1046e-03, 2.0920e-02, 5.8403e-03, 3.6373e-03, 2.9411e-03,
          3.0479e-03, 7.5951e-03, 1.8797e-03, 3.7575e-03, 1.6508e-03, 6.7940e-03,
          5.4588e-03, 5.8098e-03, 3.8433e-03, 3.0556e-03, 3.4447e-03, 5.5237e-03,
          6.2675e-03, 4.0512e-03, 4.4136e-03, 2.2717e-03, 2.3518e-03, 4.4823e-03,
          2.0409e-03, 3.2177e-03, 1.8188e-02, 1.3039e-02, 9.0637e-03, 3.3493e-03,
          8.3694e-03, 6.5842e-03, 1.2047e-02, 4.1809e-03, 1.0414e-02, 5.7220e-03,
          6.8130e-03, 8.3466e-03, 1.2024e-01, 3.0045e-02, 4.9591e-03, 6.7635e-03,
          4.8561e-03, 1.6422e-03, 2.0008e-03, 4.5929e-03, 4.7417e-03, 4.6158e-03,
          4.9286e-03, 4.3449e-03, 1.2932e-02, 1.1673e-02, 8.5144e-03, 4.7913e-03,
          9.2850e-03, 3.7994e-03, 3.1471e-03, 3.2711e-03, 3.3436e-03, 5.5084e-03,
          6.2866e-03, 6.7749e-02, 8.5571e-02, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          6.8605e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 1.6809e-04, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 2.1887e-04, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 2.1887e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          2.1887e-04, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 6.8605e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 2.1887e-04, 2.1887e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 2.1887e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.9141e-05, 6.8605e-05, 1.8358e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 6.8605e-05, 6.8605e-05,
          6.8605e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.8358e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.8358e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 6.8605e-05, 1.7822e-05,
          6.8605e-05, 6.9141e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.6809e-04, 1.7822e-05, 6.8605e-05, 1.7822e-05, 6.9141e-05,
          6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 1.7822e-05, 2.1887e-04, 4.5896e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.6809e-04,
          1.6809e-04, 4.5896e-04, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 6.9141e-05, 6.9141e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.8605e-05,
          1.7822e-05, 6.8605e-05, 2.1887e-04, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05, 6.8605e-05, 1.7822e-05,
          1.6809e-04, 2.1887e-04, 1.7822e-05, 6.8605e-05, 1.7822e-05, 1.7822e-05,
          6.8605e-05, 1.7822e-05, 1.7822e-05, 1.6809e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 6.8605e-05, 4.5896e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.6809e-04, 1.7822e-05, 6.8605e-05, 6.8605e-05, 6.8605e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.6809e-04, 1.6809e-04, 4.0817e-04, 1.7822e-05, 1.6809e-04, 1.7822e-05,
          1.7822e-05, 4.0817e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.6136e-03, 7.7576e-02, 2.1469e-02, 2.9053e-02, 6.3293e-02,
          6.8909e-02, 1.8890e-02, 6.5689e-03, 4.3907e-03, 2.1896e-02, 5.4657e-02,
          1.5732e-02, 7.1068e-03, 2.1095e-03, 7.4577e-03, 7.6904e-03, 7.6027e-03,
          4.9133e-03, 4.6654e-03, 7.3719e-04, 1.1663e-03, 1.0529e-03, 7.0305e-03,
          6.7261e-02, 8.5831e-03, 1.4969e-02, 1.0925e-02, 1.1549e-03, 7.4921e-03,
          2.9850e-03, 1.0788e-02, 2.0885e-03, 8.1253e-03, 1.4145e-02, 4.1733e-03,
          6.2637e-03, 2.5444e-03, 1.1299e-02, 7.6447e-03, 9.9087e-04, 1.4862e-02,
          4.2267e-03, 3.2139e-03, 2.8591e-03, 3.0365e-03, 9.7046e-03, 9.1553e-03,
          1.2178e-03, 1.0252e-03, 7.6218e-03, 4.3259e-03, 6.2218e-03, 4.5357e-03,
          1.4248e-03, 2.9602e-03, 1.0586e-03, 3.8548e-03, 5.3024e-03, 6.1874e-03,
          3.0613e-03, 2.8191e-03, 3.1834e-03, 5.1460e-03, 9.6130e-03, 1.9791e-02,
          1.0368e-02, 1.4412e-02, 8.7891e-03, 2.0161e-03, 1.8520e-03, 2.4319e-03,
          1.8435e-03, 3.5950e-02, 1.5612e-03, 6.1150e-03, 1.1330e-03, 1.3809e-03,
          2.0714e-03, 3.9520e-03, 9.8114e-03, 1.3710e-02, 8.7967e-03, 2.6932e-03,
          1.7920e-03, 1.1009e-02, 2.2278e-03, 6.5491e-02, 1.7703e-05, 5.2452e-05,
          1.7703e-05, 1.7703e-05, 4.6313e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.2541e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 8.1062e-05,
          5.2452e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 4.6313e-05, 4.6313e-05, 5.2452e-05, 1.7703e-05, 1.7703e-05,
          8.1062e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 5.2452e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 8.1062e-05, 8.1062e-05, 5.2452e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2541e-04, 2.1970e-04, 9.0718e-05, 1.7703e-05, 1.7703e-05, 2.1970e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 8.1062e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 9.0718e-05, 1.2541e-04,
          1.7703e-05, 1.7703e-05, 1.2541e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 5.2452e-05, 5.2452e-05, 2.1970e-04, 1.7703e-05,
          4.6313e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 8.1062e-05, 8.1062e-05, 8.1062e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 4.6313e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 8.1062e-05,
          8.1062e-05, 1.7703e-05, 1.7703e-05, 2.1970e-04, 1.7703e-05, 4.6313e-05,
          1.7703e-05, 1.2541e-04, 9.0718e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 1.7703e-05, 2.1970e-04,
          5.2452e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 8.1062e-05, 1.7703e-05,
          4.6313e-05, 2.1970e-04, 1.7703e-05, 8.1062e-05, 1.2541e-04, 2.2209e-04,
          8.1062e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2541e-04, 8.1062e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 2.9254e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 5.2452e-05, 1.7703e-05, 1.4913e-04,
          1.7703e-05, 1.7703e-05, 9.0718e-05, 9.0718e-05, 2.2209e-04, 1.2541e-04,
          1.7703e-05, 1.7703e-05, 9.0718e-05, 1.2541e-04, 1.7703e-05, 4.6313e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 4.6313e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          9.0718e-05, 1.7703e-05, 5.2452e-05, 2.2209e-04, 2.1970e-04, 9.0718e-05,
          5.2452e-05, 5.2452e-05, 1.2541e-04, 1.7703e-05, 9.0718e-05, 1.7703e-05,
          1.7703e-05, 5.2452e-05, 9.0718e-05, 2.1970e-04, 1.7703e-05, 5.2452e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 2.9254e-04, 9.0718e-05,
          1.7774e-04, 4.6313e-05, 1.2541e-04, 8.1062e-05, 1.7703e-05, 4.6313e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2541e-04, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 4.6313e-05, 5.2452e-05, 4.0698e-04, 1.7703e-05, 1.7703e-05,
          9.0718e-05, 7.0047e-04, 9.0718e-05, 1.2541e-04, 1.7703e-05, 1.7703e-05,
          1.2541e-04, 2.5797e-04, 2.5797e-04, 1.7774e-04, 4.6313e-05, 4.6313e-05,
          9.0718e-05, 1.7703e-05, 9.0718e-05, 2.1970e-04, 9.0718e-05, 2.5797e-04,
          1.8501e-04, 2.9254e-04, 9.0718e-05, 1.7703e-05, 1.7703e-05, 5.0116e-04,
          1.7703e-05, 6.7091e-04, 3.1734e-04, 1.7774e-04, 1.7703e-05, 9.0718e-05,
          2.9254e-04, 5.2452e-05, 4.9305e-04, 3.7360e-04, 1.7703e-05, 2.1970e-04,
          1.7703e-05, 1.7703e-05, 1.2541e-04, 5.2452e-05, 6.1560e-04, 2.5797e-04,
          2.5797e-04, 1.4913e-04, 1.7703e-05, 1.7703e-05, 1.7774e-04, 1.4913e-04,
          6.5994e-04, 9.0718e-05, 1.7703e-05, 1.7703e-05, 2.2066e-04, 4.6313e-05,
          3.8934e-04, 2.2209e-04, 5.5218e-04, 1.7703e-05, 8.1062e-05, 4.6313e-05,
          5.2452e-05, 3.3545e-04, 5.2452e-05, 4.6313e-05, 1.7703e-05, 1.7703e-05,
          8.1062e-05, 6.2168e-05, 9.0718e-05, 5.2452e-05, 3.8934e-04, 6.2168e-05,
          3.8934e-04, 5.0354e-04, 1.7703e-05, 9.0718e-05, 2.5797e-04, 3.7217e-04,
          4.4847e-04, 2.5797e-04, 5.0116e-04, 1.7703e-05, 1.7703e-05, 9.0718e-05,
          5.3930e-04, 5.3930e-04, 1.7703e-05, 2.5797e-04, 2.5797e-04, 1.7703e-05,
          4.6313e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 3.4618e-04, 8.1062e-05,
          1.7703e-05, 3.3641e-04, 4.6313e-05, 6.9046e-04, 2.0742e-04, 2.5201e-04,
          4.6313e-05, 1.7703e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.3084e-03, 7.5989e-02, 2.5787e-02, 3.0426e-02, 6.7688e-02,
          5.8594e-02, 1.9897e-02, 2.4376e-03, 2.3117e-03, 9.3536e-03, 2.1896e-02,
          6.8092e-03, 2.6283e-03, 3.2387e-03, 3.1357e-03, 2.8152e-03, 3.3226e-03,
          2.6703e-03, 2.4643e-03, 7.8392e-04, 1.0300e-03, 1.2712e-03, 4.1237e-03,
          2.9907e-02, 6.9885e-03, 1.6403e-02, 1.0109e-02, 1.3008e-03, 3.4389e-03,
          4.8523e-03, 4.0588e-03, 1.3237e-03, 3.5515e-03, 4.5776e-03, 2.2621e-03,
          3.8052e-03, 2.6417e-03, 6.5460e-03, 3.2692e-03, 1.3065e-03, 1.6403e-02,
          8.3923e-03, 5.8975e-03, 6.8970e-03, 3.3970e-03, 4.1771e-03, 5.6801e-03,
          1.9646e-03, 1.4811e-03, 8.0643e-03, 2.9182e-03, 3.1815e-03, 2.4223e-03,
          1.4496e-03, 2.0084e-03, 1.1168e-03, 3.5610e-03, 5.5542e-03, 5.8289e-03,
          3.3855e-03, 2.9564e-03, 4.4174e-03, 4.4289e-03, 2.4200e-02, 1.2405e-02,
          8.8196e-02, 2.7740e-02, 9.6741e-03, 3.8147e-03, 3.6793e-03, 5.0201e-03,
          2.5482e-03, 1.7914e-02, 1.7004e-03, 3.7117e-03, 1.4515e-03, 1.3676e-03,
          1.8024e-03, 4.5662e-03, 3.8696e-02, 1.0971e-02, 3.4424e-02, 2.6321e-03,
          2.0809e-03, 8.7433e-03, 3.3188e-03, 2.0218e-02, 6.7627e-02, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 4.6730e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05, 1.8179e-05,
          4.6730e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 7.5698e-05, 4.6730e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05, 4.6730e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          9.7036e-05, 9.7036e-05, 6.8486e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05,
          1.8179e-05, 4.6730e-05, 1.8179e-05, 7.5698e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 9.7036e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 9.7036e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 4.6730e-05, 4.6730e-05, 9.7036e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05,
          1.8179e-05, 9.7036e-05, 9.7036e-05, 9.7036e-05, 9.7036e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05,
          4.6730e-05, 1.8179e-05, 4.6730e-05, 2.4939e-04, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 9.7036e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05, 1.8179e-05, 2.4939e-04,
          4.6730e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05, 1.8179e-05,
          1.8179e-05, 9.7036e-05, 1.8179e-05, 4.6730e-05, 9.7036e-05, 7.5698e-05,
          4.6730e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 2.4939e-04, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 4.7147e-05, 4.6730e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 6.8486e-05, 7.5698e-05, 6.8486e-05,
          1.3459e-04, 1.8179e-05, 1.8179e-05, 2.1350e-04, 1.3459e-04, 6.8486e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 9.7036e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 4.6730e-05, 1.2600e-04, 9.7036e-05, 1.8501e-04,
          1.8179e-05, 1.8179e-05, 2.4939e-04, 1.8179e-05, 3.3736e-04, 1.8179e-05,
          1.8179e-05, 9.7036e-05, 1.3459e-04, 9.7036e-05, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 1.8179e-05, 2.4247e-04, 6.8486e-05,
          4.6730e-05, 6.8486e-05, 2.4939e-04, 2.4939e-04, 1.8179e-05, 1.8179e-05,
          1.8179e-05, 1.8179e-05, 1.8179e-05, 3.6597e-04, 1.3459e-04, 1.8179e-05,
          1.8179e-05, 9.7036e-05, 1.8537e-04, 5.1880e-04, 1.8179e-05, 1.8179e-05,
          6.8486e-05, 3.3736e-04, 6.8486e-05, 4.3130e-04, 1.8179e-05, 1.8179e-05,
          2.1350e-04, 1.3459e-04, 1.3459e-04, 1.9217e-04, 6.8486e-05, 3.6597e-04,
          3.3736e-04, 1.8179e-05, 3.3736e-04, 3.6597e-04, 1.3459e-04, 1.3459e-04,
          2.4939e-04, 1.6320e-04, 6.8486e-05, 1.8179e-05, 4.7147e-05, 9.7036e-05,
          1.8179e-05, 5.4789e-04, 9.7573e-05, 1.2600e-04, 1.8179e-05, 3.3736e-04,
          1.8501e-04, 4.6730e-05, 2.7847e-04, 1.8501e-04, 1.8179e-05, 9.7036e-05,
          1.8179e-05, 4.6730e-05, 5.0449e-04, 4.6730e-05, 7.4148e-04, 3.3736e-04,
          1.3459e-04, 1.6320e-04, 1.8179e-05, 1.8179e-05, 1.8537e-04, 7.2813e-04,
          1.9217e-04, 9.7036e-05, 1.8179e-05, 1.8179e-05, 9.7036e-05, 1.8501e-04,
          1.1997e-03, 1.1492e-03, 1.6320e-04, 1.8179e-05, 4.6730e-05, 4.6730e-05,
          9.7036e-05, 2.1350e-04, 4.6730e-05, 1.8501e-04, 1.8179e-05, 1.8179e-05,
          1.6320e-04, 1.8179e-05, 1.8179e-05, 4.6730e-05, 8.4448e-04, 1.3459e-04,
          2.4247e-04, 4.0221e-04, 1.8179e-05, 5.2595e-04, 1.3459e-04, 5.2595e-04,
          1.9217e-04, 3.3736e-04, 3.6597e-04, 4.6730e-05, 1.8179e-05, 5.6267e-04,
          6.4230e-04, 1.2140e-03, 1.8179e-05, 3.3736e-04, 3.3736e-04, 1.8179e-05,
          6.8486e-05, 6.8486e-05, 1.8179e-05, 1.8179e-05, 2.9063e-04, 3.1924e-04,
          1.8179e-05, 2.9063e-04, 6.8486e-05, 5.4979e-04, 2.9063e-04, 6.8486e-05,
          6.8486e-05, 6.8486e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.4915e-03, 6.9824e-02, 2.4521e-02, 2.8793e-02, 7.3669e-02,
          5.0507e-02, 2.1164e-02, 1.9855e-03, 1.8110e-03, 6.7673e-03, 1.6418e-02,
          5.1003e-03, 2.1744e-03, 4.2381e-03, 2.6836e-03, 2.2087e-03, 2.9411e-03,
          2.5330e-03, 1.9169e-03, 9.6893e-04, 1.1730e-03, 1.3790e-03, 3.1796e-03,
          2.1408e-02, 6.2408e-03, 1.6739e-02, 7.2212e-03, 1.4305e-03, 2.9869e-03,
          5.9319e-03, 3.1586e-03, 1.3514e-03, 2.9583e-03, 2.9430e-03, 1.8940e-03,
          3.5419e-03, 2.9469e-03, 4.6501e-03, 2.5234e-03, 1.7099e-03, 1.4282e-02,
          1.1429e-02, 7.2784e-03, 9.6970e-03, 3.7727e-03, 3.6106e-03, 5.1041e-03,
          2.2678e-03, 1.7223e-03, 9.1324e-03, 2.5520e-03, 2.5539e-03, 1.9064e-03,
          1.7586e-03, 1.8082e-03, 1.4210e-03, 3.6831e-03, 6.0196e-03, 5.9662e-03,
          3.9902e-03, 3.5000e-03, 4.8752e-03, 4.1885e-03, 2.6108e-02, 1.1459e-02,
          8.9050e-02, 2.1286e-02, 6.9237e-03, 5.1270e-03, 4.8752e-03, 7.2212e-03,
          3.1948e-03, 1.1536e-02, 1.9951e-03, 3.2158e-03, 1.8072e-03, 1.4143e-03,
          2.1057e-03, 4.9782e-03, 3.8177e-02, 8.3237e-03, 3.1464e-02, 3.1452e-03,
          2.4872e-03, 7.9498e-03, 3.8757e-03, 1.3161e-02, 4.5624e-02, 7.5256e-02,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05,
          4.9114e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 9.5010e-05, 4.9114e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 4.9114e-05, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.1319e-04, 1.9550e-05, 8.3566e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 2.2888e-04,
          1.9550e-05, 1.9550e-05, 1.1319e-04, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 4.9114e-05, 4.9114e-05, 1.1319e-04, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05,
          1.9550e-05, 4.9114e-05, 4.9114e-05, 4.9114e-05, 1.1319e-04, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05,
          4.9114e-05, 1.9550e-05, 4.9114e-05, 3.6645e-04, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.3554e-04, 8.3566e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05, 3.6645e-04,
          9.5010e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 1.9550e-05,
          1.9550e-05, 1.1319e-04, 4.9114e-05, 4.9114e-05, 1.1319e-04, 9.5010e-05,
          4.9114e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.1319e-04, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 4.9114e-05, 4.9114e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 6.5446e-05, 9.5010e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 8.3566e-05, 9.5010e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.3530e-04, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.1319e-04, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 4.9114e-05, 1.5914e-04, 1.1319e-04, 1.9932e-04,
          1.9550e-05, 1.9550e-05, 1.1319e-04, 1.9550e-05, 1.9932e-04, 1.9550e-05,
          1.9550e-05, 1.5914e-04, 1.3530e-04, 1.1319e-04, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 1.9550e-05, 2.4509e-04, 8.3566e-05,
          9.5010e-05, 8.3566e-05, 5.2786e-04, 1.1319e-04, 1.9550e-05, 1.9550e-05,
          1.9550e-05, 1.9550e-05, 1.9550e-05, 2.2888e-04, 1.3530e-04, 1.9550e-05,
          1.9550e-05, 1.1319e-04, 9.5010e-05, 5.5790e-04, 1.9550e-05, 1.9550e-05,
          8.3566e-05, 4.5252e-04, 8.3566e-05, 4.1246e-04, 1.9550e-05, 4.9114e-05,
          5.5790e-04, 1.3530e-04, 1.3530e-04, 2.1076e-04, 8.3566e-05, 1.9932e-04,
          4.5252e-04, 1.9550e-05, 4.5252e-04, 4.8208e-04, 1.3530e-04, 1.3530e-04,
          8.1110e-04, 1.6487e-04, 1.9932e-04, 1.9550e-05, 6.5446e-05, 1.5938e-04,
          1.9550e-05, 6.0368e-04, 1.5938e-04, 9.5010e-05, 1.9550e-05, 1.9932e-04,
          4.5252e-04, 4.9114e-05, 1.5914e-04, 1.3530e-04, 1.9550e-05, 1.1319e-04,
          1.9550e-05, 4.9114e-05, 5.2786e-04, 4.9114e-05, 2.2888e-04, 1.9932e-04,
          1.3530e-04, 1.6487e-04, 1.9550e-05, 1.9550e-05, 1.9157e-04, 1.9157e-04,
          2.1076e-04, 4.9114e-05, 1.9550e-05, 1.9550e-05, 1.1343e-04, 1.9932e-04,
          1.4639e-03, 3.0732e-04, 1.3530e-04, 1.9550e-05, 4.9114e-05, 1.9550e-05,
          4.9114e-05, 1.9932e-04, 4.9114e-05, 3.3689e-04, 1.9550e-05, 1.9550e-05,
          1.3530e-04, 1.9550e-05, 1.9550e-05, 4.9114e-05, 3.0732e-04, 1.3530e-04,
          2.7490e-04, 4.4203e-04, 1.9550e-05, 6.6566e-04, 1.3530e-04, 4.1246e-04,
          2.1076e-04, 1.3530e-04, 2.2888e-04, 4.9114e-05, 1.9550e-05, 7.6437e-04,
          5.2834e-04, 4.8208e-04, 1.9550e-05, 4.5252e-04, 4.5252e-04, 1.9550e-05,
          8.3566e-05, 8.3566e-05, 1.9550e-05, 9.0957e-05, 4.1556e-04, 1.1319e-04,
          1.9550e-05, 4.4513e-04, 8.3566e-05, 7.2765e-04, 8.3566e-05, 8.3566e-05,
          1.9550e-05, 1.9550e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.7885e-03, 7.6416e-02, 2.4033e-02, 3.5645e-02, 6.6895e-02,
          7.2571e-02, 2.0630e-02, 4.9438e-03, 3.2406e-03, 1.8738e-02, 4.0070e-02,
          1.2856e-02, 5.0392e-03, 2.3518e-03, 5.3062e-03, 5.2490e-03, 6.3286e-03,
          4.9171e-03, 2.9869e-03, 7.8487e-04, 1.3161e-03, 1.5707e-03, 7.7362e-03,
          4.9469e-02, 7.2479e-03, 1.6556e-02, 1.2230e-02, 1.2026e-03, 6.3515e-03,
          3.0994e-03, 7.8964e-03, 2.1515e-03, 6.8436e-03, 1.1993e-02, 3.0651e-03,
          4.8981e-03, 2.0599e-03, 1.0719e-02, 6.6872e-03, 1.0347e-03, 1.6632e-02,
          5.7411e-03, 4.1809e-03, 3.7460e-03, 3.3035e-03, 7.5760e-03, 8.4534e-03,
          1.4973e-03, 1.3208e-03, 8.4076e-03, 4.0131e-03, 5.2910e-03, 3.1338e-03,
          1.3809e-03, 2.6283e-03, 1.0061e-03, 3.8261e-03, 5.0392e-03, 5.3024e-03,
          2.7199e-03, 3.1033e-03, 3.4332e-03, 5.6725e-03, 1.0162e-02, 1.7471e-02,
          1.1307e-02, 1.0712e-02, 8.3771e-03, 2.5158e-03, 2.1935e-03, 2.6398e-03,
          2.1706e-03, 2.5223e-02, 1.6747e-03, 5.1155e-03, 1.1387e-03, 1.2226e-03,
          1.8492e-03, 3.4504e-03, 8.1253e-03, 1.0269e-02, 7.8278e-03, 2.1915e-03,
          1.6880e-03, 1.0696e-02, 2.3804e-03, 2.9068e-02, 7.8812e-03, 5.6763e-03,
          7.2449e-02, 2.2054e-05, 8.7678e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 1.0860e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.2452e-05, 8.7678e-05,
          1.0860e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 5.7280e-05, 5.7280e-05, 5.2452e-05, 2.2054e-05, 2.2054e-05,
          5.2452e-05, 2.2054e-05, 2.2054e-05, 1.1355e-04, 1.1355e-04, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 5.2452e-05, 5.2452e-05, 1.0860e-04,
          5.2452e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 6.9082e-05,
          1.3471e-04, 1.3471e-04, 1.0431e-04, 2.2054e-05, 2.2054e-05, 1.3471e-04,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 8.7678e-05,
          2.2054e-05, 5.2452e-05, 2.2054e-05, 1.0860e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.3471e-04, 1.3471e-04,
          2.2054e-05, 2.2054e-05, 1.3471e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 5.2452e-05, 8.0347e-04, 1.3471e-04, 8.7678e-05,
          5.7280e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 5.2452e-05, 8.7678e-05, 1.4877e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 5.7280e-05, 2.2054e-05, 2.2054e-05, 5.2452e-05, 1.1355e-04,
          5.2452e-05, 2.2054e-05, 2.2054e-05, 2.8038e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 1.9121e-04, 1.3471e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 5.2452e-05, 2.2054e-05, 1.3471e-04,
          1.1355e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.1355e-04, 2.2054e-05,
          1.0431e-04, 1.3471e-04, 2.2054e-05, 5.2452e-05, 1.3471e-04, 2.5201e-04,
          1.9574e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.9574e-04, 5.2452e-05,
          2.2054e-05, 5.2452e-05, 2.2054e-05, 7.8261e-05, 2.2054e-05, 2.5201e-04,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 5.2452e-05, 1.1355e-04, 1.1355e-04,
          2.2054e-05, 2.2054e-05, 1.6046e-04, 1.0431e-04, 2.5201e-04, 3.4356e-04,
          2.2054e-05, 2.2054e-05, 1.6046e-04, 1.9085e-04, 2.2054e-05, 5.7280e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 8.7678e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 6.9082e-05, 2.2054e-05, 2.2054e-05,
          1.6046e-04, 2.2054e-05, 5.2452e-05, 4.0436e-04, 1.9574e-04, 1.0431e-04,
          1.0860e-04, 5.2452e-05, 1.9574e-04, 2.2054e-05, 1.6046e-04, 5.2452e-05,
          2.2054e-05, 1.1355e-04, 1.3885e-03, 1.3471e-04, 2.2054e-05, 5.2452e-05,
          5.2452e-05, 6.9082e-05, 2.2054e-05, 2.2054e-05, 4.0436e-04, 1.6046e-04,
          1.4877e-04, 1.0431e-04, 2.8706e-04, 5.2452e-05, 2.2054e-05, 5.7280e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 1.9085e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 8.7678e-05, 1.1355e-04, 4.0436e-04, 2.2054e-05, 2.2054e-05,
          1.9085e-04, 1.0948e-03, 1.6046e-04, 4.0436e-04, 2.2054e-05, 2.2054e-05,
          4.0436e-04, 2.5201e-04, 8.8072e-04, 1.4877e-04, 5.7280e-05, 1.3471e-04,
          1.0431e-04, 2.2054e-05, 4.6539e-04, 3.4332e-04, 1.6046e-04, 1.6046e-04,
          4.9591e-04, 1.9085e-04, 1.0431e-04, 5.8985e-04, 8.3148e-05, 3.4857e-04,
          2.2054e-05, 4.0436e-04, 4.9400e-04, 1.9574e-04, 2.2054e-05, 8.8072e-04,
          2.5201e-04, 5.2452e-05, 4.0436e-04, 8.2874e-04, 2.2054e-05, 3.4809e-04,
          2.2054e-05, 5.2452e-05, 1.9085e-04, 5.2452e-05, 5.9986e-04, 1.6046e-04,
          3.1281e-04, 1.4877e-04, 2.2054e-05, 2.2054e-05, 1.6963e-04, 1.6963e-04,
          1.5650e-03, 1.0431e-04, 2.2054e-05, 2.2054e-05, 3.4857e-04, 1.6046e-04,
          4.0436e-04, 4.0436e-04, 4.8733e-04, 2.2054e-05, 1.0860e-04, 2.2054e-05,
          1.4877e-04, 5.8889e-04, 5.2452e-05, 1.4389e-04, 2.2054e-05, 2.2054e-05,
          5.7280e-05, 1.2529e-04, 1.0431e-04, 5.2452e-05, 3.2210e-04, 6.9332e-04,
          1.1559e-03, 3.4809e-04, 2.2054e-05, 1.0431e-04, 3.1281e-04, 4.0936e-04,
          5.7888e-04, 8.2874e-04, 4.0436e-04, 5.2452e-05, 2.2054e-05, 3.1304e-04,
          4.6539e-04, 1.2474e-03, 2.2054e-05, 3.1304e-04, 3.1304e-04, 2.2054e-05,
          8.7678e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 4.3440e-04, 1.4877e-04,
          2.2054e-05, 3.3307e-04, 5.7280e-05, 6.2799e-04, 2.4164e-04, 2.4164e-04,
          2.2054e-05, 2.2054e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.8882e-03, 7.0435e-02, 2.3560e-02, 3.0075e-02, 5.6824e-02,
          5.6458e-02, 1.8188e-02, 3.3112e-03, 2.8172e-03, 1.4259e-02, 2.1286e-02,
          9.5825e-03, 4.1351e-03, 2.2144e-03, 5.7106e-03, 3.4046e-03, 6.4278e-03,
          4.1428e-03, 2.2354e-03, 1.1225e-03, 2.1439e-03, 2.4796e-03, 5.3406e-03,
          3.1097e-02, 8.7509e-03, 1.4969e-02, 1.0223e-02, 1.5526e-03, 6.1226e-03,
          2.9869e-03, 5.3444e-03, 2.2564e-03, 8.1711e-03, 8.1329e-03, 4.3068e-03,
          9.6741e-03, 3.3417e-03, 1.2901e-02, 6.4812e-03, 1.5345e-03, 2.5818e-02,
          9.6436e-03, 6.5231e-03, 6.0883e-03, 6.0997e-03, 6.7177e-03, 1.3420e-02,
          1.8673e-03, 1.7376e-03, 7.3586e-03, 5.8060e-03, 4.3030e-03, 3.6983e-03,
          1.5907e-03, 2.6150e-03, 1.3609e-03, 9.2850e-03, 6.1226e-03, 9.3307e-03,
          3.7727e-03, 4.1733e-03, 4.4060e-03, 8.4229e-03, 1.6388e-02, 3.1372e-02,
          1.2627e-02, 1.0551e-02, 8.9188e-03, 4.8523e-03, 3.9482e-03, 4.2686e-03,
          3.4103e-03, 2.5146e-02, 2.3060e-03, 4.6196e-03, 1.2093e-03, 1.5326e-03,
          2.3193e-03, 9.9487e-03, 1.1047e-02, 1.9455e-02, 9.9564e-03, 3.3455e-03,
          2.3537e-03, 1.2665e-02, 3.3798e-03, 1.9974e-02, 8.8425e-03, 7.9880e-03,
          4.0436e-02, 4.9011e-02, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05,
          5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05,
          5.3525e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 5.3525e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 5.3525e-05,
          7.7665e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.1474e-04, 3.0756e-04, 7.7665e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04,
          1.6451e-05, 1.6451e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.1474e-04, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 5.3525e-05, 5.3525e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05,
          1.1474e-04, 1.6451e-05, 1.6451e-05, 3.0756e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 2.7108e-04, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 3.0756e-04,
          5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05, 5.3525e-05, 3.0756e-04,
          5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04, 5.3525e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 7.7665e-05, 1.1474e-04, 3.0828e-04,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 5.3525e-05, 1.1474e-04, 1.1474e-04, 1.6451e-05,
          5.3525e-05, 5.3525e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 5.3525e-05, 1.6451e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04, 1.6451e-05,
          5.3525e-05, 1.6451e-05, 1.1474e-04, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.1474e-04, 5.3525e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 2.7061e-04, 7.7665e-05, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.1474e-04, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 2.7061e-04, 1.1474e-04, 1.6451e-05, 1.6451e-05,
          1.1474e-04, 4.5872e-04, 7.7665e-05, 1.6451e-05, 1.6451e-05, 3.0828e-04,
          1.6451e-05, 4.5872e-04, 3.0828e-04, 5.3525e-05, 1.6451e-05, 1.6451e-05,
          3.9744e-04, 5.3525e-05, 1.1474e-04, 6.1464e-04, 1.6451e-05, 3.0828e-04,
          1.6451e-05, 1.6451e-05, 3.0756e-04, 1.6451e-05, 4.6778e-04, 7.7665e-05,
          4.2152e-04, 5.3525e-05, 1.6451e-05, 1.6451e-05, 2.2578e-04, 1.6463e-04,
          4.6778e-04, 7.7665e-05, 1.6451e-05, 1.6451e-05, 3.0828e-04, 1.6451e-05,
          1.1673e-03, 2.2578e-04, 4.3082e-04, 1.6451e-05, 1.1474e-04, 1.6451e-05,
          5.3525e-05, 3.0828e-04, 5.3525e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 5.3525e-05, 1.6463e-04, 1.6451e-05,
          6.5136e-04, 1.1474e-04, 1.6451e-05, 7.7665e-05, 3.6049e-04, 4.8256e-04,
          4.6778e-04, 4.2152e-04, 3.0756e-04, 1.6451e-05, 1.6451e-05, 7.7665e-05,
          1.0195e-03, 1.0195e-03, 1.6451e-05, 7.7665e-05, 7.7665e-05, 1.6451e-05,
          1.6451e-05, 1.6451e-05, 1.6451e-05, 1.6451e-05, 7.7665e-05, 1.6451e-05,
          1.6451e-05, 1.1474e-04, 7.7665e-05, 5.1117e-04, 7.7665e-05, 4.7040e-04,
          1.6451e-05, 1.6451e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.8043e-03, 6.8787e-02, 2.9266e-02, 2.0691e-02, 7.7576e-02,
          5.0323e-02, 1.9806e-02, 2.4090e-03, 2.1000e-03, 6.2141e-03, 1.1192e-02,
          4.1466e-03, 2.8515e-03, 3.9864e-03, 3.8433e-03, 2.7027e-03, 3.8052e-03,
          2.5730e-03, 2.2030e-03, 1.9684e-03, 2.4529e-03, 3.3684e-03, 3.5324e-03,
          1.5808e-02, 9.4833e-03, 1.5404e-02, 4.8599e-03, 2.2755e-03, 2.8439e-03,
          4.5357e-03, 3.2501e-03, 1.4400e-03, 3.6182e-03, 2.7695e-03, 2.2583e-03,
          5.1613e-03, 3.9635e-03, 4.3221e-03, 2.4929e-03, 2.4586e-03, 1.8738e-02,
          2.0676e-02, 1.3680e-02, 1.5411e-02, 7.7934e-03, 3.6907e-03, 6.9122e-03,
          3.1662e-03, 3.9711e-03, 1.3130e-02, 3.0212e-03, 2.4776e-03, 2.0828e-03,
          2.2144e-03, 2.0390e-03, 2.2182e-03, 5.8708e-03, 6.6414e-03, 7.3395e-03,
          5.9814e-03, 4.8370e-03, 6.9351e-03, 6.9923e-03, 2.7527e-02, 2.1103e-02,
          2.6306e-02, 1.0895e-02, 6.0005e-03, 1.4008e-02, 1.4809e-02, 1.4015e-02,
          8.6441e-03, 8.5220e-03, 2.5883e-03, 2.9774e-03, 2.0428e-03, 1.5621e-03,
          2.4529e-03, 5.7106e-03, 1.1497e-02, 7.3891e-03, 8.7128e-03, 4.9591e-03,
          4.1771e-03, 1.2253e-02, 5.4550e-03, 1.0040e-02, 1.1795e-02, 1.8936e-02,
          1.7380e-02, 1.7166e-02, 8.2764e-02, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 9.6321e-05, 9.6321e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 5.7459e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 1.3351e-04, 5.7459e-05,
          9.6321e-05, 9.6321e-05, 5.7459e-05, 9.6321e-05, 9.6321e-05, 2.0206e-05,
          1.3351e-04, 2.0206e-05, 5.7459e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          9.6321e-05, 2.0206e-05, 9.6321e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          2.0206e-05, 2.0206e-05, 1.3351e-04, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          9.6321e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          2.0206e-05, 5.7459e-05, 5.7459e-05, 1.3387e-04, 5.7459e-05, 2.0206e-05,
          1.3351e-04, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          5.7459e-05, 2.0206e-05, 2.0206e-05, 1.3351e-04, 9.6321e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 9.6321e-05, 9.6321e-05, 2.0206e-05,
          2.0206e-05, 9.6321e-05, 2.0206e-05, 5.7459e-05, 9.6321e-05, 1.3351e-04,
          5.7459e-05, 2.0206e-05, 2.0206e-05, 9.6858e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 5.7459e-05, 2.0206e-05, 5.7459e-05, 5.7459e-05, 5.7459e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 9.6321e-05, 5.7459e-05, 5.7459e-05,
          9.6321e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          2.0206e-05, 9.6321e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05,
          2.0206e-05, 6.0463e-04, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 9.6321e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0802e-05,
          2.0206e-05, 2.0206e-05, 5.7459e-05, 5.7459e-05, 5.7459e-05, 2.0206e-05,
          9.7156e-05, 2.1040e-05, 5.7757e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05, 9.7156e-05,
          9.6560e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          5.7459e-05, 2.0206e-05, 1.3351e-04, 1.2541e-03, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 1.3351e-04, 1.3387e-04, 5.7459e-05, 2.0802e-05, 2.0206e-05,
          2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          5.7459e-05, 2.0206e-05, 2.0206e-05, 5.7757e-05, 9.6321e-05, 1.3351e-04,
          2.0206e-05, 1.2169e-03, 2.0206e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          1.3351e-04, 5.7459e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05,
          2.0206e-05, 5.8353e-05, 5.7459e-05, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          2.0206e-05, 5.7459e-05, 5.7459e-05, 2.0206e-05, 9.6607e-04, 5.7459e-05,
          2.0206e-05, 5.7459e-05, 1.3387e-04, 5.7459e-05, 5.7459e-05, 9.6321e-05,
          2.0206e-05, 1.3351e-04, 2.0206e-05, 2.0206e-05, 2.9159e-04, 2.1553e-04,
          2.1529e-04, 5.7459e-05, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.0206e-05,
          1.5049e-03, 2.9230e-04, 5.7459e-05, 2.0206e-05, 1.3351e-04, 5.7459e-05,
          5.7459e-05, 2.0206e-05, 5.7459e-05, 9.6321e-05, 2.0206e-05, 2.0206e-05,
          1.3351e-04, 2.0206e-05, 2.0206e-05, 5.7459e-05, 2.1553e-04, 2.0206e-05,
          1.5049e-03, 5.7459e-05, 3.8457e-04, 5.7459e-05, 2.0206e-05, 2.0206e-05,
          1.3466e-03, 2.0206e-05, 5.7459e-05, 5.7459e-05, 9.6321e-05, 4.5753e-04,
          2.0206e-05, 1.7080e-03, 2.0206e-05, 2.0206e-05, 2.0206e-05, 2.0206e-05,
          8.5545e-04, 1.4696e-03, 3.4714e-04, 1.9467e-04, 3.4714e-04, 9.6321e-05,
          2.0206e-05, 3.8457e-04, 3.4714e-04, 8.3780e-04, 3.4714e-04, 8.5545e-04,
          3.4714e-04, 3.4714e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.6359e-03, 7.2876e-02, 2.4323e-02, 3.2379e-02, 6.3354e-02,
          6.2988e-02, 2.0554e-02, 4.1313e-03, 2.7027e-03, 1.4847e-02, 3.0731e-02,
          1.0086e-02, 3.9902e-03, 2.1133e-03, 4.8828e-03, 4.4785e-03, 5.3825e-03,
          4.2267e-03, 2.5959e-03, 8.0872e-04, 1.3409e-03, 2.1591e-03, 6.2828e-03,
          4.2877e-02, 7.0534e-03, 1.6586e-02, 1.0994e-02, 1.1873e-03, 5.4550e-03,
          2.7905e-03, 6.8169e-03, 1.9550e-03, 5.9433e-03, 8.5449e-03, 3.0823e-03,
          4.8370e-03, 2.1267e-03, 8.3542e-03, 5.3253e-03, 1.1702e-03, 1.5114e-02,
          6.0120e-03, 4.6616e-03, 4.4861e-03, 3.9558e-03, 7.2823e-03, 9.0866e-03,
          1.8692e-03, 1.6823e-03, 7.7782e-03, 3.7994e-03, 4.6082e-03, 3.0289e-03,
          1.2951e-03, 2.4128e-03, 1.0090e-03, 4.2992e-03, 5.2452e-03, 5.7335e-03,
          2.8820e-03, 3.1242e-03, 3.2997e-03, 5.3253e-03, 1.0361e-02, 1.9409e-02,
          1.2222e-02, 1.1681e-02, 8.2092e-03, 3.3302e-03, 2.8248e-03, 3.6163e-03,
          2.7142e-03, 2.4841e-02, 1.6832e-03, 4.8218e-03, 1.1568e-03, 1.2236e-03,
          1.7977e-03, 4.0932e-03, 9.5978e-03, 1.1192e-02, 8.8577e-03, 2.4490e-03,
          1.7872e-03, 1.1856e-02, 2.3975e-03, 2.4551e-02, 8.3847e-03, 6.9275e-03,
          6.0516e-02, 2.6749e-02, 4.1199e-03, 3.9398e-02, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 5.4300e-05,
          5.4300e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05, 2.2769e-05,
          5.4300e-05, 2.2769e-05, 2.2769e-05, 1.2636e-04, 5.4300e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 9.1970e-05, 5.4300e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          1.4722e-04, 1.4722e-04, 1.1569e-04, 2.2769e-05, 2.2769e-05, 1.4722e-04,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 6.0439e-05, 1.4722e-04,
          2.2769e-05, 2.2769e-05, 1.4722e-04, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 5.4300e-05, 5.4300e-05, 1.4722e-04, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 5.4300e-05, 5.4300e-05, 9.1970e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05,
          5.4300e-05, 2.2769e-05, 2.2769e-05, 1.4722e-04, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.3580e-04, 6.0439e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05, 1.4722e-04,
          1.2636e-04, 2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05,
          2.2769e-05, 1.4722e-04, 2.2769e-05, 5.4300e-05, 9.1970e-05, 2.1935e-04,
          5.4300e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.1935e-04, 5.4300e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.5249e-04,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 5.4300e-05, 2.2769e-05, 1.2636e-04,
          2.2769e-05, 2.2769e-05, 1.1569e-04, 1.1569e-04, 2.1935e-04, 1.4746e-04,
          2.2769e-05, 2.2769e-05, 1.1569e-04, 2.3556e-04, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 6.0439e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05,
          6.0439e-05, 2.2769e-05, 5.4300e-05, 2.1935e-04, 1.4722e-04, 1.1569e-04,
          5.4300e-05, 5.4300e-05, 1.4722e-04, 2.2769e-05, 2.0397e-04, 2.2769e-05,
          2.2769e-05, 5.4300e-05, 3.1877e-04, 1.4722e-04, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 3.0756e-04, 1.1569e-04,
          1.2636e-04, 6.0439e-05, 1.4722e-04, 9.1970e-05, 2.2769e-05, 6.0439e-05,
          2.2769e-05, 2.2769e-05, 2.2769e-05, 1.4722e-04, 2.2769e-05, 2.2769e-05,
          2.2769e-05, 9.1970e-05, 1.6403e-04, 4.4799e-04, 2.2769e-05, 2.2769e-05,
          1.1569e-04, 1.6975e-03, 2.0397e-04, 5.2023e-04, 2.2769e-05, 2.2769e-05,
          2.3556e-04, 2.0397e-04, 2.0397e-04, 1.2636e-04, 1.1569e-04, 1.1569e-04,
          1.1569e-04, 2.2769e-05, 2.0397e-04, 2.3556e-04, 2.0397e-04, 2.0397e-04,
          1.4722e-04, 1.8036e-04, 1.1569e-04, 2.2769e-05, 2.2769e-05, 4.3201e-04,
          2.2769e-05, 5.2023e-04, 2.1958e-04, 1.6403e-04, 2.2769e-05, 2.0397e-04,
          2.3556e-04, 5.4300e-05, 5.2023e-04, 3.1877e-04, 2.2769e-05, 3.6001e-04,
          2.2769e-05, 2.2769e-05, 2.3556e-04, 5.4300e-05, 7.4291e-04, 2.0397e-04,
          4.1652e-04, 1.6403e-04, 2.2769e-05, 2.2769e-05, 1.2636e-04, 1.2636e-04,
          1.3828e-03, 1.1569e-04, 2.2769e-05, 2.2769e-05, 3.6001e-04, 6.0439e-05,
          5.2023e-04, 5.2023e-04, 1.1104e-04, 6.0439e-05, 9.1970e-05, 2.2769e-05,
          9.1970e-05, 5.6314e-04, 5.4300e-05, 9.1970e-05, 2.2769e-05, 2.2769e-05,
          6.0439e-05, 1.1569e-04, 1.1569e-04, 5.4300e-05, 4.2725e-04, 7.7963e-05,
          1.4381e-03, 4.3178e-04, 2.2769e-05, 1.1569e-04, 2.0397e-04, 3.2806e-04,
          2.1470e-04, 2.0397e-04, 5.2023e-04, 5.4300e-05, 2.2769e-05, 4.3321e-04,
          4.1652e-04, 1.5955e-03, 2.2769e-05, 2.0397e-04, 4.3321e-04, 2.2769e-05,
          6.0439e-05, 2.2769e-05, 2.2769e-05, 2.2769e-05, 4.6396e-04, 9.1970e-05,
          6.0439e-05, 3.2091e-04, 2.8944e-04, 6.5565e-04, 6.5565e-04, 7.1096e-04,
          6.0439e-05, 2.2769e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.2169e-03, 6.3721e-02, 3.3630e-02, 2.0706e-02, 8.8989e-02,
          4.9683e-02, 2.1820e-02, 2.0142e-03, 1.6346e-03, 3.7937e-03, 7.0038e-03,
          2.3918e-03, 2.5158e-03, 2.5539e-03, 3.4332e-03, 2.2659e-03, 3.9444e-03,
          2.1992e-03, 1.7271e-03, 2.9106e-03, 3.3741e-03, 4.4632e-03, 2.8973e-03,
          8.9874e-03, 9.1400e-03, 1.6846e-02, 4.1580e-03, 3.1948e-03, 2.6703e-03,
          2.4891e-03, 2.6283e-03, 1.5278e-03, 3.8166e-03, 1.8826e-03, 2.0676e-03,
          6.3477e-03, 4.8409e-03, 3.6373e-03, 2.3575e-03, 2.7122e-03, 1.4824e-02,
          2.4490e-02, 1.4610e-02, 2.0203e-02, 1.0971e-02, 3.2692e-03, 8.5983e-03,
          3.1834e-03, 6.1073e-03, 1.6815e-02, 3.4084e-03, 2.2411e-03, 1.8892e-03,
          2.8496e-03, 1.9197e-03, 3.6716e-03, 6.4278e-03, 5.5084e-03, 8.2474e-03,
          8.5068e-03, 4.9820e-03, 6.8283e-03, 6.9656e-03, 1.2932e-02, 1.8600e-02,
          7.6447e-03, 3.9902e-03, 4.2381e-03, 2.6627e-02, 2.8320e-02, 2.5085e-02,
          2.2446e-02, 5.7945e-03, 4.8637e-03, 2.7142e-03, 2.3670e-03, 1.5488e-03,
          2.7809e-03, 5.5771e-03, 4.5929e-03, 4.7836e-03, 3.7785e-03, 6.6490e-03,
          6.7329e-03, 8.6441e-03, 5.5199e-03, 4.7569e-03, 3.0823e-03, 5.5237e-03,
          8.7433e-03, 1.1246e-02, 1.6266e-02, 8.0872e-03, 9.7412e-02, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05,
          1.5461e-04, 9.5487e-05, 1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 7.8440e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05,
          1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05, 1.9312e-05,
          9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.5461e-04, 1.9312e-05, 1.9312e-05, 1.5461e-04, 9.5487e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05, 9.5487e-05, 1.5461e-04,
          7.8440e-05, 1.9312e-05, 1.9312e-05, 1.5569e-04, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05, 1.9312e-05,
          9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05,
          1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05, 9.5487e-05,
          1.9312e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 2.0385e-05,
          1.9312e-05, 1.9312e-05, 7.8440e-05, 7.8440e-05, 1.5461e-04, 1.9312e-05,
          1.5461e-04, 7.8440e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05, 9.6917e-05,
          1.5461e-04, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          7.8440e-05, 1.9312e-05, 1.5461e-04, 1.5461e-04, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05,
          9.5487e-05, 1.9312e-05, 1.9312e-05, 7.8797e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 4.7970e-04, 9.5487e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05,
          9.5487e-05, 7.8440e-05, 1.5461e-04, 1.9312e-05, 1.9312e-05, 1.5569e-04,
          1.9312e-05, 7.8797e-05, 1.5604e-04, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 7.8440e-05, 1.9312e-05, 9.5487e-05, 1.5604e-04,
          1.9312e-05, 1.9312e-05, 1.5461e-04, 9.5487e-05, 1.9312e-05, 9.5487e-05,
          9.5487e-05, 1.5461e-04, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.5461e-04,
          1.9312e-05, 7.8440e-05, 1.9312e-05, 9.5487e-05, 1.5604e-04, 1.9312e-05,
          1.5461e-04, 1.5497e-04, 1.9312e-05, 1.9312e-05, 1.5461e-04, 1.9312e-05,
          1.5461e-04, 1.5569e-04, 7.8440e-05, 9.5487e-05, 1.9312e-05, 1.9312e-05,
          1.9312e-05, 1.9312e-05, 1.9312e-05, 7.8440e-05, 1.5461e-04, 1.9312e-05,
          9.5487e-05, 1.5497e-04, 5.3883e-04, 9.5487e-05, 1.9312e-05, 9.5487e-05,
          1.5461e-04, 1.9312e-05, 1.5497e-04, 7.8440e-05, 9.5487e-05, 9.5487e-05,
          9.5487e-05, 3.9177e-03, 1.9312e-05, 1.9312e-05, 1.9312e-05, 1.9312e-05,
          4.7970e-04, 1.2846e-03, 4.7970e-04, 7.6962e-04, 1.2846e-03, 9.5487e-05,
          1.9312e-05, 5.3883e-04, 4.7970e-04, 7.6962e-04, 5.9795e-04, 1.0586e-03,
          9.5487e-05, 9.5487e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.2239e-03, 6.1920e-02, 4.1138e-02, 2.8442e-02, 9.7778e-02,
          4.0833e-02, 3.4058e-02, 1.6584e-03, 1.1120e-03, 3.0174e-03, 5.6000e-03,
          1.9026e-03, 1.8606e-03, 2.7084e-03, 2.2888e-03, 1.6584e-03, 3.8433e-03,
          2.3232e-03, 1.2016e-03, 3.5763e-03, 4.0398e-03, 7.9346e-03, 2.7199e-03,
          6.8817e-03, 6.0806e-03, 2.4643e-02, 5.1384e-03, 3.4847e-03, 2.4223e-03,
          2.5120e-03, 2.1801e-03, 1.5774e-03, 2.8248e-03, 1.3409e-03, 1.7328e-03,
          4.5052e-03, 4.4441e-03, 2.8973e-03, 1.8215e-03, 2.9736e-03, 1.1818e-02,
          2.4399e-02, 1.2177e-02, 2.0142e-02, 1.0559e-02, 3.2501e-03, 6.2256e-03,
          4.8943e-03, 1.0796e-02, 1.9119e-02, 3.0651e-03, 1.9217e-03, 1.5755e-03,
          2.5291e-03, 1.7023e-03, 3.9005e-03, 5.1308e-03, 5.1117e-03, 6.0387e-03,
          6.5613e-03, 4.5815e-03, 6.3248e-03, 5.6381e-03, 8.3618e-03, 1.1131e-02,
          6.0577e-03, 3.1643e-03, 3.2997e-03, 2.4628e-02, 2.4368e-02, 2.7573e-02,
          3.2593e-02, 4.8141e-03, 4.4174e-03, 2.4281e-03, 2.3193e-03, 1.5326e-03,
          2.9068e-03, 4.5738e-03, 3.3169e-03, 3.0212e-03, 2.8400e-03, 5.0049e-03,
          6.7749e-03, 7.2708e-03, 4.6654e-03, 3.4561e-03, 2.5291e-03, 4.3221e-03,
          6.7139e-03, 8.3542e-03, 8.0719e-03, 6.7787e-03, 4.9591e-02, 7.5806e-02,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 8.4639e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 1.2696e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05,
          2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          8.4639e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05, 1.9026e-04,
          8.4639e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 1.2696e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 1.9026e-04, 2.1219e-05,
          8.4639e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05, 2.1219e-05,
          8.4639e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          8.4639e-05, 2.1219e-05, 1.5888e-03, 1.9026e-04, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          1.2696e-04, 2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 1.2696e-04, 1.2696e-04, 1.9026e-04, 2.1219e-05, 2.1219e-05,
          1.9026e-04, 8.4639e-05, 1.2696e-04, 2.1219e-05, 2.1219e-05, 1.9026e-04,
          2.1219e-05, 2.1219e-05, 1.9324e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 8.4639e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04,
          2.1219e-05, 2.1219e-05, 1.5888e-03, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04, 8.4639e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05,
          1.9026e-04, 1.9026e-04, 2.1219e-05, 2.1219e-05, 1.9026e-04, 2.1219e-05,
          8.4639e-05, 1.9026e-04, 2.1219e-05, 1.2696e-04, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 8.4639e-05, 1.9026e-04, 2.1219e-05,
          1.2696e-04, 1.9026e-04, 1.9026e-04, 2.1219e-05, 2.1219e-05, 1.2696e-04,
          1.9026e-04, 2.1219e-05, 1.9026e-04, 8.4639e-05, 2.1219e-05, 2.1219e-05,
          1.2696e-04, 5.0879e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          1.2696e-04, 1.1559e-03, 5.0879e-04, 4.7565e-04, 2.0254e-04, 2.1219e-05,
          2.1219e-05, 5.7220e-04, 1.2696e-04, 7.8630e-04, 3.6430e-04, 4.7016e-04,
          2.1219e-05, 1.2696e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.3918e-03, 5.6915e-02, 3.3936e-02, 2.5192e-02, 8.4045e-02,
          4.3671e-02, 1.9867e-02, 3.5992e-03, 2.1572e-03, 2.6588e-03, 3.7727e-03,
          1.6966e-03, 8.0032e-03, 2.0187e-02, 2.3575e-03, 2.5539e-03, 3.1494e-02,
          1.1292e-02, 1.2522e-03, 1.2047e-02, 6.8893e-03, 9.4681e-03, 5.4588e-03,
          3.6716e-03, 2.5330e-03, 1.5839e-02, 1.0193e-02, 4.3671e-02, 1.7548e-02,
          2.1240e-02, 5.2757e-03, 1.8433e-02, 3.8586e-03, 1.7633e-03, 3.9368e-03,
          4.4518e-03, 3.0289e-03, 2.8782e-03, 2.4567e-03, 1.5793e-02, 2.5749e-03,
          2.3499e-03, 1.7834e-03, 1.8902e-03, 1.4315e-03, 9.6283e-03, 4.3335e-03,
          8.4534e-03, 1.0109e-02, 1.3908e-02, 1.2413e-02, 7.4806e-03, 2.5539e-03,
          6.9160e-03, 4.9820e-03, 6.8970e-03, 4.8714e-03, 5.0316e-03, 4.0169e-03,
          4.1695e-03, 3.1250e-02, 5.9738e-03, 5.3940e-03, 1.8024e-03, 4.3488e-03,
          2.5349e-03, 1.1301e-03, 1.5841e-03, 2.7409e-03, 2.4071e-03, 3.1643e-03,
          3.8528e-03, 4.5433e-03, 1.4923e-02, 9.3536e-03, 5.2605e-03, 3.9177e-03,
          7.4310e-03, 5.0468e-03, 3.4275e-03, 2.7065e-03, 2.6627e-03, 2.9964e-03,
          3.7136e-03, 8.0109e-03, 3.7708e-03, 1.8072e-03, 1.2512e-03, 1.6346e-03,
          2.6817e-03, 2.3384e-03, 1.4200e-03, 2.2411e-03, 3.2310e-03, 3.9787e-03,
          1.2024e-01, 1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05,
          7.9870e-05, 4.0770e-05, 4.0770e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 6.3598e-05, 8.6427e-05, 6.3598e-05, 4.0770e-05, 6.3598e-05,
          6.3598e-05, 6.3598e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 5.6982e-05, 5.6982e-05, 4.0770e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 5.6982e-05, 1.7941e-05, 1.7941e-05,
          4.0770e-05, 4.0770e-05, 1.7941e-05, 8.6427e-05, 8.6427e-05, 6.3598e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 1.7941e-05, 5.6982e-05, 1.7941e-05, 5.6982e-05, 7.9870e-05,
          4.0770e-05, 5.6982e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 6.3598e-05,
          8.6427e-05, 6.3598e-05, 3.1781e-04, 6.3598e-05, 6.3598e-05, 1.7941e-05,
          1.7941e-05, 6.3598e-05, 4.0770e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05,
          1.6963e-04, 4.0770e-05, 5.6982e-05, 7.9870e-05, 1.7941e-05, 1.7941e-05,
          7.9870e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 6.3598e-05, 1.7941e-05,
          6.3598e-05, 5.6982e-05, 7.9870e-05, 8.6427e-05, 4.0770e-05, 1.7941e-05,
          7.9870e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 7.9870e-05, 7.9870e-05,
          4.0770e-05, 1.7941e-05, 1.7941e-05, 4.0770e-05, 1.6963e-04, 1.7941e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 8.6427e-05, 8.6427e-05, 6.3598e-05,
          8.6427e-05, 8.6427e-05, 6.3598e-05, 1.7941e-05, 7.9870e-05, 4.0770e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 8.0168e-05, 7.9870e-05, 5.6982e-05,
          4.0770e-05, 1.7941e-05, 1.7941e-05, 7.9870e-05, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 6.3598e-05, 3.5238e-04, 8.6427e-05, 1.2553e-04, 7.9870e-05,
          8.6486e-05, 1.2553e-04, 8.6427e-05, 1.0270e-04, 6.3598e-05, 5.6982e-05,
          1.7941e-05, 4.0770e-05, 5.6982e-05, 5.6982e-05, 5.6982e-05, 5.6982e-05,
          5.6982e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 5.6982e-05, 1.7941e-05,
          6.3598e-05, 8.6427e-05, 6.3598e-05, 1.2553e-04, 8.6427e-05, 8.6427e-05,
          8.6427e-05, 6.3598e-05, 1.7941e-05, 6.3598e-05, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 7.9870e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 7.9930e-05,
          1.7941e-05, 1.7941e-05, 7.9870e-05, 7.9870e-05, 7.9870e-05, 8.6427e-05,
          1.2553e-04, 5.7042e-05, 1.2553e-04, 8.6427e-05, 8.6427e-05, 1.2553e-04,
          1.2553e-04, 1.2553e-04, 1.7941e-05, 1.7941e-05, 1.7941e-05, 7.9930e-05,
          7.9930e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 7.9870e-05, 1.7941e-05,
          7.9870e-05, 1.7941e-05, 4.0770e-05, 2.3305e-04, 1.7941e-05, 6.3598e-05,
          6.3598e-05, 6.3598e-05, 6.3598e-05, 1.0270e-04, 3.5238e-04, 8.6427e-05,
          6.3598e-05, 8.6427e-05, 4.0770e-05, 1.7941e-05, 1.7941e-05, 7.9870e-05,
          1.7941e-05, 1.7941e-05, 1.7941e-05, 7.9870e-05, 1.7941e-05, 7.9870e-05,
          4.0770e-05, 7.9870e-05, 1.7941e-05, 5.6982e-05, 1.7941e-05, 2.6178e-04,
          1.2553e-04, 4.5395e-04, 8.6427e-05, 4.0770e-05, 6.3598e-05, 6.3598e-05,
          4.0770e-05, 4.0770e-05, 1.7941e-05, 1.7941e-05, 1.7941e-05, 4.0770e-05,
          4.0770e-05, 7.9870e-05, 4.0770e-05, 5.6982e-05, 1.7941e-05, 4.0770e-05,
          1.2553e-04, 1.7941e-05, 1.2553e-04, 1.7941e-05, 1.9369e-03, 7.9870e-05,
          8.6427e-05, 1.0270e-04, 4.0770e-05, 2.2817e-04, 1.7941e-05, 1.7941e-05,
          1.7941e-05, 4.0770e-05, 4.0770e-05, 1.7941e-05, 1.6963e-04, 7.9870e-05,
          4.0770e-05, 4.0770e-05, 7.6008e-04, 3.1781e-04, 4.0770e-05, 1.6963e-04,
          1.6963e-04, 7.9870e-05, 8.6427e-05, 5.6982e-05, 4.0770e-05, 1.7941e-05,
          4.5395e-04, 4.0770e-05, 5.6982e-05, 4.5395e-04, 1.7941e-05, 1.7941e-05,
          4.0770e-05, 1.7941e-05, 4.0770e-05, 1.7941e-05, 7.9870e-05, 1.7941e-05,
          1.3053e-04, 7.9870e-05, 1.0748e-03, 4.0770e-05, 1.7941e-05, 1.4341e-04,
          4.5395e-04, 4.0770e-05, 7.9870e-05, 7.9870e-05, 7.2002e-04, 4.1485e-04,
          4.0770e-05, 2.7871e-04, 1.6618e-04, 4.0770e-05, 1.7941e-05, 1.7941e-05,
          4.0770e-05, 2.6703e-04, 2.6703e-04, 1.3053e-04, 2.6703e-04, 4.0770e-05,
          1.7941e-05, 1.3053e-04, 1.3053e-04, 4.0770e-05, 1.7941e-05, 1.7941e-05,
          4.0770e-05, 1.3053e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.6741e-03, 5.5847e-02, 3.1433e-02, 2.4826e-02, 1.0602e-01,
          3.6774e-02, 2.5467e-02, 7.8964e-03, 2.2068e-03, 1.8749e-03, 2.7828e-03,
          1.2407e-03, 1.6403e-03, 2.8267e-03, 1.9951e-03, 3.5820e-03, 2.7161e-03,
          1.6031e-03, 1.6718e-03, 2.1000e-03, 1.9083e-03, 1.5383e-03, 1.8177e-03,
          2.5711e-03, 1.9350e-03, 1.9135e-02, 6.2943e-03, 6.9351e-03, 2.5902e-03,
          3.4485e-03, 9.4147e-03, 1.9894e-03, 2.4624e-03, 1.1616e-03, 5.8670e-03,
          3.9673e-03, 3.9062e-03, 2.0313e-03, 1.8368e-03, 2.8954e-03, 1.4801e-03,
          1.9083e-03, 1.4906e-03, 1.7109e-03, 9.4509e-04, 1.7347e-03, 2.1915e-03,
          1.4677e-03, 2.7847e-03, 1.3596e-02, 3.1067e-02, 1.6830e-02, 3.9062e-03,
          2.4460e-02, 1.4854e-02, 5.0873e-02, 4.3526e-03, 7.9498e-03, 4.7836e-03,
          5.7220e-03, 5.1117e-03, 1.3733e-02, 5.9166e-03, 1.4067e-03, 2.5425e-03,
          2.0752e-03, 8.9073e-04, 1.1959e-03, 2.6245e-03, 2.2659e-03, 2.7027e-03,
          2.9716e-03, 3.3150e-03, 1.0358e-01, 2.6672e-02, 1.9531e-02, 1.2909e-02,
          3.3630e-02, 4.8256e-03, 2.9640e-03, 2.3785e-03, 2.6875e-03, 3.6640e-03,
          4.8943e-03, 1.5930e-02, 8.1253e-03, 1.2712e-03, 9.9659e-04, 1.3723e-03,
          1.7872e-03, 1.3371e-03, 1.1063e-03, 1.4467e-03, 2.4109e-03, 3.0403e-03,
          1.3527e-02, 5.8563e-02, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          5.8532e-05, 1.1241e-04, 5.8532e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05, 5.8532e-05,
          1.4091e-04, 1.4091e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05, 5.8532e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.1241e-04, 5.8532e-05, 1.5676e-05, 1.8370e-04, 5.8532e-05, 1.4091e-04,
          6.9618e-05, 1.9491e-04, 1.4091e-04, 1.5676e-05, 1.5676e-05, 1.1241e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05, 5.8532e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.4091e-04,
          1.8370e-04, 1.4091e-04, 2.3770e-04, 1.8370e-04, 1.5676e-05, 1.1241e-04,
          1.5676e-05, 1.5676e-05, 1.1247e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.1241e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.1241e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 5.8532e-05, 1.5676e-05, 6.9618e-05, 1.5676e-05,
          1.1241e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05, 5.8532e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 3.9434e-04, 1.1241e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.8370e-04, 5.8532e-05, 1.4091e-04,
          5.8532e-05, 1.4091e-04, 1.4091e-04, 1.5676e-05, 1.1241e-04, 1.1241e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.1247e-04, 5.8532e-05, 5.8532e-05,
          1.5676e-05, 1.1241e-04, 1.5676e-05, 1.1241e-04, 1.5676e-05, 1.1241e-04,
          1.5676e-05, 1.5676e-05, 1.4091e-04, 1.8370e-04, 1.1241e-04, 1.5676e-05,
          5.8532e-05, 5.8532e-05, 1.4091e-04, 1.4091e-04, 1.5676e-05, 5.8532e-05,
          1.5676e-05, 5.8532e-05, 5.8532e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.1241e-04, 1.1241e-04,
          1.5676e-05, 1.9491e-04, 1.5676e-05, 1.4091e-04, 1.5676e-05, 1.4091e-04,
          5.8532e-05, 1.4091e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 5.8532e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 5.8532e-05, 1.1241e-04, 1.1241e-04, 1.5676e-05,
          5.8532e-05, 5.8532e-05, 1.1241e-04, 1.4091e-04, 1.4091e-04, 1.5676e-05,
          1.4091e-04, 6.9618e-05, 1.5676e-05, 1.1241e-04, 1.5676e-05, 5.8532e-05,
          5.8532e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.1241e-04, 6.9618e-05,
          5.8532e-05, 1.5676e-05, 6.9618e-05, 3.9434e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.4091e-04, 1.5676e-05, 1.5676e-05, 1.4091e-04, 1.5676e-05,
          1.5676e-05, 1.9491e-04, 6.9618e-05, 6.9618e-05, 1.5676e-05, 1.1241e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 6.9618e-05, 1.5676e-05, 1.1241e-04,
          6.9618e-05, 6.9618e-05, 1.5676e-05, 5.8532e-05, 1.5676e-05, 6.9618e-05,
          1.4091e-04, 1.1241e-04, 1.9491e-04, 1.9491e-04, 1.5676e-05, 1.5676e-05,
          1.9491e-04, 2.3770e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.1241e-04,
          1.5676e-05, 1.1247e-04, 1.1241e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          6.9618e-05, 1.5676e-05, 1.9491e-04, 1.5676e-05, 1.4091e-04, 1.1241e-04,
          1.5676e-05, 5.8532e-05, 1.1247e-04, 5.8532e-05, 1.5676e-05, 1.9491e-04,
          1.9491e-04, 6.9618e-05, 1.5676e-05, 1.5676e-05, 6.9618e-05, 5.8532e-05,
          1.5676e-05, 6.9618e-05, 8.6212e-04, 1.5676e-05, 1.1241e-04, 6.9618e-05,
          6.9618e-05, 1.1241e-04, 1.5676e-05, 5.8532e-05, 1.5676e-05, 1.5676e-05,
          1.8370e-04, 1.1241e-04, 1.5676e-05, 6.9618e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 5.8532e-05, 6.9618e-05, 1.5676e-05,
          6.9618e-05, 1.1241e-04, 1.1241e-04, 1.5676e-05, 1.5676e-05, 4.4227e-04,
          6.9618e-05, 1.5676e-05, 1.1241e-04, 5.8532e-05, 1.8370e-04, 1.9491e-04,
          2.6894e-03, 6.9618e-05, 1.4091e-04, 6.9618e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 6.9618e-05, 1.1241e-04, 1.5676e-05, 6.9618e-05, 1.5676e-05,
          1.5676e-05, 6.9618e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6114e-03, 6.2195e-02, 2.1576e-02, 1.9150e-02, 4.9927e-02,
          5.1636e-02, 1.1307e-02, 1.4328e-02, 3.9520e-03, 4.7989e-03, 4.9210e-03,
          2.3804e-03, 5.1575e-03, 3.7289e-03, 4.5815e-03, 1.3687e-02, 1.2100e-02,
          6.0081e-03, 3.5133e-03, 1.2560e-03, 1.1129e-03, 1.0920e-03, 5.1994e-03,
          5.2986e-03, 3.8910e-03, 1.0033e-02, 1.1581e-02, 3.0632e-03, 1.0834e-02,
          5.2147e-03, 3.8544e-02, 4.3907e-03, 8.5907e-03, 3.7804e-03, 2.4368e-02,
          5.9433e-03, 1.7176e-03, 5.4207e-03, 6.2103e-03, 2.2926e-03, 3.1147e-03,
          1.1187e-03, 9.3937e-04, 9.3174e-04, 9.4748e-04, 7.9803e-03, 6.5422e-03,
          1.6518e-03, 1.3304e-03, 5.1346e-03, 3.4027e-02, 3.7872e-02, 1.2497e-02,
          6.3286e-03, 2.0966e-02, 6.7101e-03, 3.0003e-03, 8.3542e-03, 3.7422e-03,
          3.3455e-03, 6.5842e-03, 8.2016e-03, 1.0582e-02, 1.6670e-03, 7.2021e-03,
          2.0847e-03, 1.3113e-03, 1.9798e-03, 7.2908e-04, 5.9462e-04, 9.9182e-04,
          1.1759e-03, 1.1765e-02, 1.4847e-02, 5.1239e-02, 6.9885e-03, 9.0332e-03,
          1.9089e-02, 3.0899e-03, 2.6207e-03, 7.6180e-03, 3.0193e-03, 2.7008e-03,
          2.0714e-03, 5.2979e-02, 5.1537e-03, 3.1242e-03, 1.3237e-03, 1.3123e-03,
          4.9133e-03, 4.8637e-03, 8.4114e-04, 3.3817e-03, 1.1663e-03, 1.2150e-03,
          1.1162e-02, 7.6256e-03, 6.8787e-02, 1.2040e-05, 1.2040e-05, 1.2040e-05,
          4.9591e-05, 9.4295e-05, 9.4295e-05, 1.2040e-05, 4.9591e-05, 1.2040e-05,
          1.2040e-05, 1.2040e-05, 5.6744e-05, 1.2040e-05, 4.9591e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 4.8101e-05, 4.9591e-05, 1.2040e-05, 1.2040e-05,
          4.9591e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 1.2040e-05, 1.2040e-05,
          9.4295e-05, 8.5652e-05, 1.2040e-05, 4.8101e-05, 9.4295e-05, 1.2040e-05,
          5.6744e-05, 8.5652e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05,
          9.4295e-05, 1.2040e-05, 9.4295e-05, 8.5652e-05, 1.2040e-05, 8.5652e-05,
          1.2040e-05, 1.2040e-05, 5.6744e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05,
          3.5906e-04, 8.4758e-05, 1.2040e-05, 1.2040e-05, 9.4295e-05, 1.2040e-05,
          5.6744e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 5.6744e-05, 1.2040e-05,
          1.2040e-05, 1.2040e-05, 9.4295e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05,
          9.4295e-05, 4.9591e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 2.9278e-04, 3.5906e-04, 1.2040e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 1.6546e-04, 1.6701e-04, 1.2040e-05,
          8.5652e-05, 1.6546e-04, 1.2040e-05, 1.2040e-05, 9.4295e-05, 5.6744e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 9.4295e-05, 4.9591e-05, 1.2231e-04,
          1.2040e-05, 9.4295e-05, 1.2040e-05, 9.4295e-05, 1.2040e-05, 9.4295e-05,
          1.2040e-05, 1.2040e-05, 1.2946e-04, 1.6701e-04, 2.8682e-04, 5.6744e-05,
          2.0313e-04, 8.5652e-05, 1.4153e-03, 1.2040e-05, 1.2040e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 4.9591e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 1.6701e-04,
          8.4758e-05, 1.6546e-04, 1.2040e-05, 1.2040e-05, 1.2040e-05, 5.6744e-05,
          1.5843e-04, 1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 5.6171e-04,
          1.2040e-05, 4.9591e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 9.4295e-05, 9.4295e-05, 5.6744e-05,
          9.4295e-05, 4.9591e-05, 3.2282e-04, 1.5326e-03, 1.2946e-04, 1.2040e-05,
          1.2040e-05, 5.6744e-05, 6.6757e-04, 1.6701e-04, 1.2040e-05, 1.2231e-04,
          4.9591e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 9.4295e-05, 5.6744e-05,
          5.7554e-04, 1.2040e-05, 5.6744e-05, 2.5511e-04, 1.2040e-05, 1.2040e-05,
          1.2040e-05, 8.4758e-05, 8.4758e-05, 5.6744e-05, 5.6744e-05, 1.2040e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 4.9591e-05, 1.2040e-05, 1.6701e-04,
          1.2040e-05, 1.2946e-04, 5.6744e-05, 1.2040e-05, 5.6744e-05, 9.4295e-05,
          5.6744e-05, 5.6744e-05, 1.2946e-04, 4.9591e-05, 5.6744e-05, 2.5511e-04,
          5.6744e-05, 3.5906e-04, 1.2946e-04, 5.6744e-05, 1.2040e-05, 1.2040e-05,
          5.6744e-05, 4.9591e-05, 1.2946e-04, 1.2040e-05, 1.2040e-05, 1.6701e-04,
          8.4758e-05, 4.9591e-05, 1.6701e-04, 9.4295e-05, 1.2040e-05, 8.4758e-05,
          5.6744e-05, 1.2040e-05, 1.2040e-05, 1.2779e-04, 5.9271e-04, 1.6701e-04,
          1.2040e-05, 4.9591e-05, 1.2946e-04, 4.9591e-05, 1.2040e-05, 1.2946e-04,
          8.4758e-05, 7.9727e-04, 8.4758e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05,
          1.2040e-05, 1.2040e-05, 1.6701e-04, 8.4758e-05, 4.3178e-04, 5.6744e-05,
          1.2040e-05, 1.2040e-05, 1.2040e-05, 3.8695e-04, 1.2040e-05, 1.2040e-05,
          9.4295e-05, 2.0313e-04, 1.2040e-05, 6.2037e-04, 1.2040e-05, 1.2040e-05,
          3.2783e-04, 1.2040e-05, 1.2040e-05, 1.2040e-05, 1.2040e-05, 8.4758e-05,
          1.2040e-05, 4.3178e-04, 4.3178e-04, 5.6744e-05, 8.4758e-05, 3.9411e-04,
          1.2040e-05, 7.4768e-04, 6.2418e-04, 1.2040e-05, 4.3178e-04, 1.2946e-04,
          4.3178e-04, 1.2040e-05, 1.2040e-05, 1.2946e-04, 1.2946e-04, 1.2040e-05,
          5.6744e-05, 5.6744e-05, 2.7847e-04, 5.6744e-05, 4.3917e-04, 4.9591e-05,
          1.2040e-05, 9.4295e-05, 5.6744e-05, 2.4939e-04, 1.2040e-05, 2.4939e-04,
          1.2040e-05, 5.6744e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1192e-02, 7.0801e-02, 2.7023e-02, 2.7527e-02, 7.2266e-02,
          5.8807e-02, 1.8875e-02, 3.8280e-03, 4.4174e-03, 8.1711e-03, 8.7051e-03,
          4.6577e-03, 8.6365e-03, 4.5433e-03, 6.5575e-03, 3.3245e-03, 1.9196e-02,
          6.6109e-03, 2.4033e-03, 3.1815e-03, 4.9553e-03, 2.5826e-03, 8.4000e-03,
          8.4152e-03, 7.6027e-03, 1.5022e-02, 1.2726e-02, 6.3820e-03, 1.5343e-02,
          5.7831e-03, 6.1493e-03, 7.8125e-03, 1.5541e-02, 5.4932e-03, 7.8583e-03,
          1.5381e-02, 4.2801e-03, 1.4252e-02, 8.9035e-03, 3.7022e-03, 1.2245e-02,
          3.4485e-03, 3.0270e-03, 2.0332e-03, 3.5591e-03, 1.0902e-02, 1.5076e-02,
          2.4147e-03, 2.8629e-03, 1.0643e-02, 1.4183e-02, 6.4697e-03, 6.4392e-03,
          3.4866e-03, 4.0512e-03, 3.7403e-03, 1.0635e-02, 4.7874e-03, 7.9041e-03,
          5.0201e-03, 1.0811e-02, 4.7379e-03, 1.1047e-02, 4.2191e-03, 1.4282e-02,
          3.1185e-03, 2.4509e-03, 3.4351e-03, 2.1572e-03, 1.7815e-03, 1.8034e-03,
          2.4624e-03, 1.5434e-02, 5.9929e-03, 6.9275e-03, 2.4319e-03, 2.5978e-03,
          4.1351e-03, 1.2939e-02, 6.0730e-03, 1.3252e-02, 4.3068e-03, 3.7937e-03,
          3.2940e-03, 8.5602e-03, 3.2940e-03, 4.8103e-03, 2.4452e-03, 2.4281e-03,
          6.9847e-03, 1.0201e-02, 1.8682e-03, 4.9171e-03, 2.4223e-03, 2.3022e-03,
          8.4229e-03, 3.3531e-03, 7.7858e-03, 9.0393e-02, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 2.2376e-04, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 2.2376e-04, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05,
          1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 2.9922e-04, 1.1802e-05,
          2.9922e-04, 2.2376e-04, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 2.9922e-04,
          2.2376e-04, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05,
          1.1802e-05, 8.7321e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05,
          1.1802e-05, 8.7321e-05, 2.9922e-04, 1.1802e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05, 2.9922e-04,
          1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05,
          8.7321e-05, 6.4850e-04, 1.1802e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05,
          1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05, 2.9922e-04, 8.7321e-05,
          6.4850e-04, 8.7321e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05,
          2.9922e-04, 8.7321e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          6.4850e-04, 1.1802e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05,
          8.7321e-05, 2.9922e-04, 8.7321e-05, 1.1802e-05, 1.1802e-05, 6.4850e-04,
          8.7321e-05, 8.7321e-05, 8.6021e-04, 1.1802e-05, 1.1802e-05, 8.7321e-05,
          6.4850e-04, 8.7321e-05, 1.1802e-05, 8.7321e-05, 1.1802e-05, 1.1802e-05,
          8.7321e-05, 1.1802e-05, 1.1802e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05,
          1.1802e-05, 8.7321e-05, 8.7321e-05, 8.7321e-05, 1.1802e-05, 8.7321e-05,
          8.7321e-05, 8.7321e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1436e-02, 5.6030e-02, 2.0187e-02, 2.5696e-02, 8.6792e-02,
          4.5685e-02, 2.2949e-02, 2.6894e-03, 4.1351e-03, 8.8348e-03, 1.1955e-02,
          5.8594e-03, 2.7676e-03, 3.0861e-03, 3.5915e-03, 1.5469e-03, 3.4599e-03,
          1.8368e-03, 1.2960e-03, 2.1248e-03, 1.9188e-03, 1.2770e-03, 2.0695e-03,
          1.0208e-02, 4.8523e-03, 1.5617e-02, 5.6190e-03, 5.0583e-03, 3.2196e-03,
          3.9482e-03, 3.4370e-03, 1.9722e-03, 5.3711e-03, 3.0785e-03, 2.8114e-03,
          1.2207e-02, 1.2772e-02, 4.2152e-03, 2.9335e-03, 2.7237e-03, 8.4381e-03,
          8.9264e-03, 6.6414e-03, 5.5122e-03, 4.1351e-03, 2.2888e-03, 4.8256e-03,
          1.5869e-03, 2.1400e-03, 1.0056e-02, 6.9580e-03, 4.6883e-03, 3.3569e-03,
          4.7569e-03, 3.3131e-03, 4.4670e-03, 2.6276e-02, 1.1284e-02, 2.6337e-02,
          1.3901e-02, 4.4670e-03, 3.6087e-03, 3.1700e-03, 9.7656e-03, 9.0408e-03,
          6.0768e-03, 3.1071e-03, 2.8667e-03, 4.5662e-03, 4.5204e-03, 3.9482e-03,
          3.2597e-03, 1.0620e-02, 7.1144e-03, 5.3673e-03, 3.0231e-03, 2.3212e-03,
          4.3564e-03, 4.2053e-02, 2.7756e-02, 1.2718e-02, 1.2802e-02, 9.8038e-03,
          8.7662e-03, 4.5013e-03, 2.7561e-03, 5.1003e-03, 3.1471e-03, 4.8904e-03,
          7.1182e-03, 8.2626e-03, 5.0278e-03, 5.5466e-03, 5.0850e-03, 3.5076e-03,
          6.7368e-03, 4.0970e-03, 2.9621e-03, 1.0826e-02, 1.0364e-01, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 2.1350e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 7.6652e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          8.1122e-05, 2.7680e-04, 2.1350e-04, 1.7822e-05, 1.7822e-05, 1.3995e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 9.3520e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 4.1127e-04,
          1.7822e-05, 1.7822e-05, 7.0047e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.3995e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 2.1350e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05,
          7.6652e-05, 1.7822e-05, 1.7822e-05, 4.8065e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 1.7822e-05, 8.9693e-04,
          7.6652e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.3995e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 2.7227e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 1.3995e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.3995e-04, 1.3995e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 5.8222e-04, 1.3995e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 8.1122e-05, 1.7822e-05,
          7.6652e-05, 1.7822e-05, 1.3995e-04, 1.3995e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 7.6652e-05, 1.3995e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          8.1122e-05, 8.1122e-05, 1.7822e-05, 7.6652e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 5.7030e-04, 6.2895e-04, 1.7822e-05, 1.7822e-05,
          3.3569e-04, 3.3569e-04, 1.3995e-04, 1.7822e-05, 1.7822e-05, 1.3995e-04,
          1.7822e-05, 7.6652e-05, 1.3995e-04, 7.6652e-05, 1.7822e-05, 8.1122e-05,
          8.1122e-05, 1.7822e-05, 2.7227e-04, 8.1122e-05, 1.7822e-05, 1.3995e-04,
          1.7822e-05, 1.7822e-05, 8.4972e-04, 1.7822e-05, 7.6652e-05, 5.7030e-04,
          5.7030e-04, 1.3995e-04, 1.7822e-05, 1.7822e-05, 7.6652e-05, 7.6652e-05,
          7.6652e-05, 8.1122e-05, 1.7822e-05, 1.7822e-05, 1.3995e-04, 1.7822e-05,
          1.5860e-03, 1.3995e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          7.6652e-05, 5.0879e-04, 7.6652e-05, 3.0160e-04, 1.7822e-05, 1.7822e-05,
          1.3995e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6652e-05, 1.7822e-05,
          3.6049e-04, 1.3995e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 4.8828e-04,
          1.3995e-04, 2.1350e-04, 5.5599e-04, 1.7822e-05, 1.7822e-05, 5.8556e-04,
          2.7680e-04, 7.0810e-04, 1.7822e-05, 3.7432e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 8.1122e-05, 8.1122e-05, 1.7822e-05,
          1.7822e-05, 3.0160e-04, 8.1122e-05, 3.0208e-04, 8.1122e-05, 8.1122e-05,
          1.7822e-05, 1.7822e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0971e-02, 6.0303e-02, 1.9241e-02, 2.6596e-02, 7.3181e-02,
          4.8492e-02, 1.8478e-02, 2.6283e-03, 3.9406e-03, 1.1787e-02, 1.4053e-02,
          8.1558e-03, 3.2539e-03, 2.6417e-03, 5.0812e-03, 1.8482e-03, 4.3869e-03,
          2.2106e-03, 1.5230e-03, 1.5087e-03, 1.7910e-03, 1.1473e-03, 2.8687e-03,
          1.4351e-02, 6.4735e-03, 1.3489e-02, 7.2937e-03, 2.9411e-03, 4.3411e-03,
          3.6278e-03, 3.9711e-03, 2.2812e-03, 8.4457e-03, 4.5662e-03, 3.8280e-03,
          1.6083e-02, 8.2779e-03, 7.1220e-03, 4.6158e-03, 2.2163e-03, 1.1681e-02,
          6.4125e-03, 5.0545e-03, 3.9978e-03, 5.0278e-03, 2.9221e-03, 8.4686e-03,
          1.5354e-03, 1.6203e-03, 7.7934e-03, 7.6027e-03, 4.7188e-03, 3.9330e-03,
          3.1185e-03, 3.2272e-03, 2.8305e-03, 1.7731e-02, 1.0162e-02, 2.1957e-02,
          1.0162e-02, 4.5509e-03, 3.4714e-03, 4.3678e-03, 9.7961e-03, 1.5732e-02,
          6.3744e-03, 4.1962e-03, 3.8166e-03, 3.7327e-03, 3.2120e-03, 2.8877e-03,
          2.5921e-03, 1.6403e-02, 4.6768e-03, 5.3024e-03, 2.0447e-03, 2.1400e-03,
          3.5934e-03, 2.4139e-02, 2.1240e-02, 2.3605e-02, 1.4420e-02, 9.0942e-03,
          5.9395e-03, 6.1188e-03, 2.6073e-03, 6.8512e-03, 3.9749e-03, 4.7035e-03,
          1.1345e-02, 1.4168e-02, 3.8261e-03, 8.4305e-03, 3.1128e-03, 2.7027e-03,
          3.6888e-03, 2.4948e-03, 3.9330e-03, 2.0645e-02, 4.4128e-02, 4.7180e-02,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 3.5858e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05, 7.2002e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.9320e-05, 4.6444e-04, 1.5664e-04, 1.7822e-05, 1.7822e-05, 1.2350e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 2.2435e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 4.6444e-04,
          1.7822e-05, 1.7822e-05, 4.8733e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7011e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 3.0899e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05,
          7.2002e-05, 1.7822e-05, 1.7822e-05, 4.8733e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05, 1.7822e-05, 6.2656e-04,
          7.2002e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.2350e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04,
          1.7822e-05, 1.7011e-04, 1.7822e-05, 1.7822e-05, 7.2002e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7011e-04, 1.7822e-05, 1.7822e-05, 4.1294e-04,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05, 1.2350e-04,
          1.7011e-04, 1.7822e-05, 1.7822e-05, 6.7532e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7011e-04, 1.7011e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04, 1.2350e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          6.7532e-05, 1.7822e-05, 3.5858e-04, 3.7956e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04, 1.7822e-05,
          7.2002e-05, 1.7822e-05, 4.8733e-04, 1.2350e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7011e-04,
          1.7011e-04, 6.9320e-05, 7.2002e-05, 1.2350e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 2.4772e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          2.0814e-04, 1.7822e-05, 1.7011e-04, 7.2002e-05, 6.9320e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 6.8426e-04, 4.1461e-04, 1.7011e-04, 1.7822e-05,
          5.9319e-04, 4.1461e-04, 3.9291e-04, 1.7822e-05, 1.7822e-05, 1.2350e-04,
          1.7822e-05, 1.2350e-04, 1.2386e-04, 7.2002e-05, 1.7822e-05, 6.9320e-05,
          6.9320e-05, 1.7822e-05, 2.6226e-04, 8.4019e-04, 1.7822e-05, 2.4068e-04,
          1.7822e-05, 1.7822e-05, 5.9319e-04, 1.7822e-05, 7.2002e-05, 3.6049e-04,
          9.7942e-04, 1.2350e-04, 1.7822e-05, 1.7822e-05, 7.2002e-05, 7.2002e-05,
          7.2002e-05, 6.9320e-05, 1.7822e-05, 1.7822e-05, 1.2350e-04, 1.7822e-05,
          5.9319e-04, 1.2350e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          7.2002e-05, 1.2007e-03, 7.2002e-05, 6.9320e-05, 1.7822e-05, 1.7822e-05,
          1.2350e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.2002e-05, 1.7822e-05,
          1.0729e-03, 1.2350e-04, 1.7822e-05, 6.9320e-05, 1.7822e-05, 1.0462e-03,
          1.2350e-04, 3.6049e-04, 1.3294e-03, 1.7822e-05, 1.7822e-05, 3.6669e-04,
          1.3027e-03, 1.4887e-03, 1.7822e-05, 6.8426e-04, 6.8426e-04, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.9320e-05, 1.7822e-05,
          1.7822e-05, 6.9320e-05, 6.9320e-05, 5.3215e-04, 2.5463e-04, 2.5463e-04,
          6.9320e-05, 1.7822e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.4147e-03, 6.7139e-02, 1.7960e-02, 3.4576e-02, 6.5918e-02,
          6.8604e-02, 1.7059e-02, 4.9934e-03, 4.0550e-03, 1.7654e-02, 3.5797e-02,
          1.1871e-02, 4.3297e-03, 1.9283e-03, 6.3972e-03, 4.1351e-03, 4.7951e-03,
          3.2978e-03, 2.2392e-03, 5.2643e-04, 7.4291e-04, 5.7745e-04, 5.1804e-03,
          3.7628e-02, 5.7297e-03, 1.3504e-02, 1.2413e-02, 1.3199e-03, 5.8975e-03,
          2.8954e-03, 8.4991e-03, 2.0390e-03, 9.0942e-03, 1.2192e-02, 4.0436e-03,
          8.5220e-03, 3.1700e-03, 7.0000e-03, 6.9962e-03, 8.3637e-04, 6.8588e-03,
          2.2106e-03, 1.7061e-03, 1.6270e-03, 1.7853e-03, 4.9667e-03, 6.6566e-03,
          7.7724e-04, 6.3801e-04, 6.2561e-03, 6.3515e-03, 7.3891e-03, 4.3411e-03,
          1.9493e-03, 3.7918e-03, 1.4324e-03, 6.7024e-03, 6.7787e-03, 7.5340e-03,
          3.9902e-03, 2.3918e-03, 1.9932e-03, 2.9755e-03, 4.2839e-03, 1.1818e-02,
          5.4855e-03, 5.3368e-03, 4.2763e-03, 1.2560e-03, 1.0538e-03, 1.3485e-03,
          1.1196e-03, 2.6535e-02, 2.6283e-03, 7.3204e-03, 1.3638e-03, 1.7338e-03,
          2.6531e-03, 7.7095e-03, 1.1902e-02, 1.5869e-02, 9.3536e-03, 3.0365e-03,
          2.0428e-03, 7.6866e-03, 1.4858e-03, 1.6937e-02, 4.3068e-03, 3.0918e-03,
          3.3020e-02, 1.4801e-02, 1.4009e-03, 1.6724e-02, 1.2007e-03, 1.1292e-03,
          1.8482e-03, 1.2197e-03, 5.0926e-03, 9.1553e-03, 9.3613e-03, 1.1246e-02,
          6.3354e-02, 1.9431e-05, 1.9431e-05, 1.9431e-05, 5.2214e-05, 5.2214e-05,
          1.0240e-04, 9.4056e-05, 1.9431e-05, 9.4056e-05, 1.9431e-05, 1.9431e-05,
          1.9431e-05, 9.4056e-05, 3.3450e-04, 5.2214e-05, 9.4056e-05, 1.9431e-05,
          5.2214e-05, 1.9431e-05, 1.9431e-05, 1.3852e-04, 5.2214e-05, 1.9431e-05,
          1.9431e-05, 1.9431e-05, 1.9431e-05, 5.2214e-05, 8.4221e-05, 1.0240e-04,
          1.0163e-04, 1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05, 9.4056e-05,
          8.4221e-05, 5.9891e-04, 5.1439e-05, 1.9431e-05, 1.9431e-05, 8.4221e-05,
          1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05, 5.2214e-05,
          1.9431e-05, 5.2214e-05, 1.9431e-05, 1.2684e-04, 9.4056e-05, 9.4056e-05,
          1.9431e-05, 9.4056e-05, 1.9431e-05, 9.4056e-05, 2.3341e-04, 5.4836e-04,
          1.9431e-05, 9.4056e-05, 1.3924e-04, 1.9431e-05, 1.9431e-05, 1.9431e-05,
          1.9431e-05, 1.9431e-05, 5.2214e-05, 7.5531e-04, 1.3924e-04, 5.2214e-05,
          1.9431e-05, 1.9431e-05, 1.9431e-05, 9.4056e-05, 1.9431e-05, 1.9431e-05,
          1.9431e-05, 5.2214e-05, 8.4221e-05, 8.4221e-05, 1.9431e-05, 9.4056e-05,
          1.9431e-05, 3.6597e-04, 1.9431e-05, 1.9431e-05, 5.2214e-05, 1.3852e-04,
          5.2214e-05, 1.9431e-05, 1.9431e-05, 1.3924e-04, 1.9431e-05, 1.9431e-05,
          1.9431e-05, 2.6417e-04, 8.4221e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05,
          1.9431e-05, 1.9431e-05, 9.4056e-05, 5.2214e-05, 1.9431e-05, 4.1127e-04,
          1.3852e-04, 1.9431e-05, 1.9431e-05, 1.9431e-05, 5.2214e-05, 1.9431e-05,
          1.9431e-05, 1.3924e-04, 1.9431e-05, 5.2214e-05, 8.4221e-05, 1.8942e-04,
          1.5879e-04, 9.4056e-05, 9.4056e-05, 1.9431e-05, 1.7834e-04, 5.2214e-05,
          1.9431e-05, 2.0897e-04, 9.4056e-05, 9.4056e-05, 1.9431e-05, 4.8089e-04,
          1.9431e-05, 1.9431e-05, 1.9431e-05, 5.2214e-05, 5.2214e-05, 1.8871e-04,
          1.9431e-05, 1.9431e-05, 1.5664e-04, 1.5664e-04, 1.8942e-04, 4.4203e-04,
          9.4056e-05, 1.9431e-05, 1.4424e-04, 2.6417e-04, 9.4056e-05, 1.9431e-05,
          1.4424e-04, 9.4056e-05, 9.4056e-05, 9.4056e-05, 8.4221e-05, 1.9431e-05,
          1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05, 1.9431e-05,
          1.0163e-04, 1.9431e-05, 5.2214e-05, 3.1471e-04, 8.4221e-05, 1.2600e-04,
          2.0897e-04, 5.2214e-05, 1.7834e-04, 9.4056e-05, 2.3127e-04, 9.4056e-05,
          9.4056e-05, 1.3852e-04, 4.7183e-04, 2.5296e-04, 1.9431e-05, 5.2214e-05,
          5.2214e-05, 6.9618e-05, 1.9431e-05, 1.9431e-05, 4.9686e-04, 1.5664e-04,
          1.3852e-04, 5.1439e-05, 1.3924e-04, 8.4221e-05, 9.4056e-05, 9.4056e-05,
          9.4056e-05, 9.4056e-05, 9.4056e-05, 2.9349e-04, 9.4056e-05, 9.4056e-05,
          9.4056e-05, 8.4221e-05, 5.2214e-05, 3.3092e-04, 1.9431e-05, 1.9431e-05,
          1.5664e-04, 1.2112e-03, 2.3627e-04, 4.1723e-04, 1.9431e-05, 1.9431e-05,
          5.0879e-04, 1.8942e-04, 3.7289e-04, 1.7059e-04, 1.2600e-04, 1.2600e-04,
          1.2600e-04, 1.9431e-05, 6.3038e-04, 5.3596e-04, 1.4424e-04, 2.3127e-04,
          6.1560e-04, 5.3596e-04, 1.8609e-04, 1.9431e-05, 6.9618e-05, 2.8086e-04,
          1.9431e-05, 4.1723e-04, 3.3092e-04, 2.2554e-04, 1.9431e-05, 3.7289e-04,
          1.8942e-04, 1.2684e-04, 5.7173e-04, 3.5524e-04, 1.9431e-05, 4.9973e-04,
          1.9431e-05, 5.2214e-05, 4.3774e-04, 5.2214e-05, 8.4925e-04, 4.8876e-04,
          6.3038e-04, 1.7059e-04, 6.9618e-05, 1.9431e-05, 1.8871e-04, 1.8871e-04,
          1.5936e-03, 5.1439e-05, 1.9431e-05, 1.9431e-05, 2.8086e-04, 1.0163e-04,
          5.7173e-04, 5.9128e-04, 2.5582e-04, 1.9431e-05, 2.3854e-04, 1.9431e-05,
          1.2684e-04, 8.2016e-04, 5.2214e-05, 8.4221e-05, 1.9431e-05, 9.4056e-05,
          1.2684e-04, 1.2469e-04, 1.0645e-04, 5.2214e-05, 3.3021e-04, 1.9932e-04,
          1.2007e-03, 2.8086e-04, 1.9431e-05, 5.1439e-05, 4.5252e-04, 6.3038e-04,
          1.8871e-04, 4.8876e-04, 7.5722e-04, 5.2214e-05, 1.9431e-05, 3.1090e-04,
          6.3038e-04, 1.1883e-03, 9.4056e-05, 4.8876e-04, 3.1090e-04, 1.9431e-05,
          8.4221e-05, 1.9431e-05, 1.9431e-05, 5.1439e-05, 4.6396e-04, 8.4221e-05,
          1.9431e-05, 8.4221e-05, 5.1439e-05, 5.4502e-04, 2.3258e-04, 2.3258e-04,
          5.1439e-05, 1.9431e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.7400e-03, 7.2144e-02, 1.6800e-02, 2.5406e-02, 5.6244e-02,
          5.5389e-02, 1.4725e-02, 3.6163e-03, 2.6035e-03, 8.3694e-03, 2.0462e-02,
          6.3591e-03, 3.1528e-03, 2.1763e-03, 3.4275e-03, 2.6150e-03, 2.8687e-03,
          2.0466e-03, 2.0123e-03, 6.0558e-04, 6.7186e-04, 7.0810e-04, 3.4599e-03,
          2.3071e-02, 5.2681e-03, 1.2657e-02, 8.5678e-03, 1.2493e-03, 3.6335e-03,
          3.3875e-03, 5.6686e-03, 1.6689e-03, 3.8643e-03, 4.5052e-03, 2.5158e-03,
          4.6043e-03, 3.1223e-03, 3.2902e-03, 2.8610e-03, 1.1148e-03, 3.8128e-03,
          2.8801e-03, 2.2736e-03, 2.8000e-03, 1.4296e-03, 3.4122e-03, 2.9659e-03,
          1.0557e-03, 7.9012e-04, 6.1188e-03, 3.8471e-03, 4.8714e-03, 3.2139e-03,
          2.2011e-03, 3.0518e-03, 1.5335e-03, 4.6577e-03, 8.2855e-03, 7.5684e-03,
          4.4746e-03, 2.5749e-03, 2.3308e-03, 2.1152e-03, 6.9389e-03, 5.2528e-03,
          7.1838e-02, 1.7990e-02, 3.4542e-03, 1.5163e-03, 1.4687e-03, 2.9392e-03,
          1.2188e-03, 1.5259e-02, 2.1496e-03, 5.6496e-03, 2.3022e-03, 2.2755e-03,
          2.8553e-03, 5.3062e-03, 2.7481e-02, 1.0918e-02, 2.8442e-02, 3.4943e-03,
          2.1839e-03, 5.0125e-03, 1.7986e-03, 1.2810e-02, 2.7893e-02, 2.3422e-02,
          2.4811e-02, 8.6441e-03, 3.0289e-03, 1.2741e-02, 1.0586e-03, 9.5034e-04,
          1.6088e-03, 1.1339e-03, 4.1161e-03, 5.7449e-03, 5.8632e-03, 8.1635e-03,
          3.6316e-02, 7.6721e-02, 1.4424e-05, 1.4424e-05, 1.4424e-05, 3.8385e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 4.1485e-04, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          7.1824e-05, 1.4424e-05, 1.4424e-05, 8.3625e-05, 8.3625e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.0759e-04, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          7.1824e-05, 1.6189e-04, 7.1824e-05, 1.4424e-05, 1.4424e-05, 7.1824e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 3.8385e-05, 7.1824e-05,
          1.4424e-05, 1.4424e-05, 2.4557e-04, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 7.1824e-05, 1.4424e-05,
          3.8385e-05, 7.1824e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 3.8385e-05, 1.4424e-05, 1.4102e-04, 7.1824e-05, 1.4424e-05,
          1.4424e-05, 7.1824e-05, 3.8385e-05, 1.4424e-05, 1.4424e-05, 8.3625e-05,
          1.0759e-04, 1.4424e-05, 1.4424e-05, 3.4261e-04, 1.4424e-05, 1.4424e-05,
          3.8385e-05, 7.1824e-05, 3.8385e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 8.3625e-05, 1.4424e-05, 2.7347e-04,
          1.0759e-04, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 7.1824e-05, 1.4424e-05, 3.8385e-05, 7.1824e-05, 1.4102e-04,
          1.4424e-05, 3.5739e-04, 1.4424e-05, 1.4424e-05, 1.0759e-04, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 7.1824e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 7.1824e-05, 7.1824e-05, 1.4102e-04, 7.1824e-05,
          3.5739e-04, 1.4424e-05, 7.1824e-05, 7.1824e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 3.8385e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05,
          7.1824e-05, 1.4424e-05, 1.4424e-05, 1.4102e-04, 2.4557e-04, 2.0278e-04,
          1.4424e-05, 1.4424e-05, 7.1824e-05, 1.4424e-05, 2.0278e-04, 1.4424e-05,
          1.4424e-05, 1.4102e-04, 9.9564e-04, 1.6892e-04, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 1.4424e-05, 2.0278e-04, 7.1824e-05,
          1.0759e-04, 3.8385e-05, 2.4557e-04, 7.1824e-05, 1.4424e-05, 3.8385e-05,
          1.4424e-05, 1.4424e-05, 1.4424e-05, 7.1824e-05, 1.4424e-05, 1.4424e-05,
          1.4424e-05, 7.1824e-05, 1.4102e-04, 4.8494e-04, 1.4424e-05, 1.4424e-05,
          7.1824e-05, 5.7507e-04, 7.1824e-05, 7.1824e-05, 1.4424e-05, 3.8385e-05,
          6.7949e-04, 7.1824e-05, 5.4550e-04, 1.0759e-04, 7.1824e-05, 1.6940e-04,
          7.1824e-05, 1.4424e-05, 8.7452e-04, 2.0278e-04, 4.1485e-04, 2.0278e-04,
          8.4066e-04, 2.0278e-04, 1.6189e-04, 1.4424e-05, 3.8385e-05, 6.2466e-04,
          1.4424e-05, 5.5408e-04, 8.4543e-04, 8.3625e-05, 1.4424e-05, 7.1824e-05,
          7.1824e-05, 1.4424e-05, 1.4102e-04, 5.4550e-04, 1.4424e-05, 8.4543e-04,
          1.4424e-05, 1.4424e-05, 4.0960e-04, 1.4424e-05, 2.7204e-04, 1.2169e-03,
          5.4550e-04, 3.8385e-05, 7.1824e-05, 3.8385e-05, 2.2423e-04, 2.2423e-04,
          2.7204e-04, 1.7631e-04, 1.4424e-05, 1.4424e-05, 6.8665e-04, 3.8385e-05,
          3.8862e-04, 2.5773e-04, 1.6940e-04, 7.1824e-05, 3.8385e-05, 1.4424e-05,
          1.0759e-04, 3.9005e-04, 8.3625e-05, 1.6892e-04, 1.4424e-05, 4.7922e-05,
          7.1824e-05, 7.1824e-05, 7.1824e-05, 1.4424e-05, 2.5773e-04, 7.1824e-05,
          4.7874e-04, 5.2738e-04, 1.4424e-05, 7.1824e-05, 2.0278e-04, 5.4836e-04,
          1.4102e-04, 5.4550e-04, 8.4543e-04, 1.4424e-05, 1.4424e-05, 1.6189e-04,
          6.7186e-04, 4.7874e-04, 7.1824e-05, 6.3562e-04, 6.3562e-04, 1.4424e-05,
          3.8385e-05, 1.4424e-05, 1.4424e-05, 3.8385e-05, 1.7631e-04, 3.8385e-05,
          3.8385e-05, 1.4293e-04, 3.8385e-05, 4.8065e-04, 1.7631e-04, 1.7631e-04,
          3.8385e-05, 3.8385e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.1635e-03, 6.6589e-02, 1.9241e-02, 2.2476e-02, 6.3293e-02,
          5.8594e-02, 1.2436e-02, 3.6526e-03, 2.5253e-03, 1.0735e-02, 1.9516e-02,
          6.9771e-03, 3.3894e-03, 2.4166e-03, 3.5400e-03, 2.8629e-03, 3.7498e-03,
          2.8458e-03, 1.6422e-03, 6.3086e-04, 7.3338e-04, 6.1893e-04, 4.3526e-03,
          2.1912e-02, 4.2267e-03, 1.0506e-02, 9.8801e-03, 1.2970e-03, 4.5280e-03,
          3.3379e-03, 5.7716e-03, 1.9016e-03, 4.2801e-03, 5.3177e-03, 3.0708e-03,
          6.3248e-03, 3.7632e-03, 3.9520e-03, 3.1700e-03, 1.1120e-03, 5.5161e-03,
          2.5959e-03, 2.0180e-03, 2.2869e-03, 1.4458e-03, 4.1924e-03, 3.6068e-03,
          1.0271e-03, 7.0095e-04, 5.7487e-03, 5.1880e-03, 5.9929e-03, 2.6207e-03,
          2.3766e-03, 3.6030e-03, 1.6613e-03, 7.7820e-03, 1.2108e-02, 1.1032e-02,
          5.7259e-03, 3.0060e-03, 2.7885e-03, 2.7294e-03, 6.7558e-03, 7.4272e-03,
          2.1332e-02, 1.4442e-02, 4.0321e-03, 1.4000e-03, 1.4009e-03, 1.7881e-03,
          1.1320e-03, 1.5594e-02, 3.0003e-03, 6.4812e-03, 2.4071e-03, 2.5291e-03,
          3.1757e-03, 8.8730e-03, 2.4841e-02, 1.8661e-02, 4.0192e-02, 4.7112e-03,
          3.0308e-03, 6.8741e-03, 2.2449e-03, 1.1787e-02, 1.3885e-02, 1.0895e-02,
          3.0319e-02, 1.2428e-02, 2.4376e-03, 1.5007e-02, 9.9087e-04, 9.4652e-04,
          2.0943e-03, 1.4467e-03, 3.9825e-03, 7.3586e-03, 9.7733e-03, 1.3512e-02,
          3.5309e-02, 1.3275e-02, 4.6814e-02, 3.2485e-05, 5.9843e-05, 8.3566e-05,
          1.1647e-04, 3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05,
          3.2485e-05, 1.4472e-04, 3.1686e-04, 5.9843e-05, 3.2485e-05, 3.2485e-05,
          5.6207e-05, 3.2485e-05, 3.2485e-05, 9.2983e-05, 9.2983e-05, 3.2485e-05,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 5.9843e-05, 8.3566e-05, 9.2745e-05,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 1.4472e-04,
          8.3566e-05, 5.3501e-04, 2.9159e-04, 3.2485e-05, 3.2485e-05, 3.7909e-04,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 2.8396e-04,
          3.2485e-05, 5.9843e-05, 3.2485e-05, 9.2745e-05, 6.5446e-05, 1.4472e-04,
          3.2485e-05, 1.4472e-04, 3.2485e-05, 3.2485e-05, 1.8728e-04, 4.8637e-04,
          3.2485e-05, 3.2485e-05, 2.0730e-04, 3.2485e-05, 3.2485e-05, 3.2485e-05,
          3.2485e-05, 3.2485e-05, 9.2983e-05, 2.8396e-04, 1.2219e-04, 5.9843e-05,
          8.9169e-05, 8.9169e-05, 3.2485e-05, 1.4472e-04, 3.2485e-05, 3.2485e-05,
          3.2485e-05, 8.3566e-05, 1.3065e-04, 2.0194e-04, 8.3566e-05, 3.2485e-05,
          3.2485e-05, 3.5548e-04, 5.6207e-05, 3.2485e-05, 3.2485e-05, 9.2983e-05,
          8.3566e-05, 3.2485e-05, 5.9843e-05, 2.5558e-04, 3.2485e-05, 5.9843e-05,
          5.6207e-05, 1.5521e-04, 1.1647e-04, 3.2485e-05, 3.2485e-05, 3.2485e-05,
          3.2485e-05, 3.2485e-05, 1.4472e-04, 5.9843e-05, 3.2485e-05, 3.2640e-04,
          8.3566e-05, 3.2485e-05, 3.2485e-05, 3.2485e-05, 3.1710e-04, 3.2485e-05,
          3.2485e-05, 2.4652e-04, 3.2485e-05, 8.3566e-05, 1.4961e-04, 1.8823e-04,
          1.7202e-04, 1.4472e-04, 3.2485e-05, 3.2485e-05, 4.1151e-04, 5.9843e-05,
          3.2485e-05, 3.2485e-05, 1.4472e-04, 3.2485e-05, 3.2485e-05, 4.8637e-04,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 5.9843e-05, 9.2983e-05, 1.2589e-04,
          6.5446e-05, 6.5446e-05, 8.9169e-05, 2.4652e-04, 1.8823e-04, 4.0674e-04,
          1.4472e-04, 3.2485e-05, 1.2779e-04, 1.2779e-04, 6.5446e-05, 5.6207e-05,
          3.2485e-05, 1.4472e-04, 1.4472e-04, 1.4472e-04, 8.3566e-05, 3.2485e-05,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 5.9843e-05, 6.5446e-05, 6.5446e-05,
          1.1647e-04, 3.2485e-05, 8.3566e-05, 2.7966e-04, 2.9874e-04, 2.0134e-04,
          3.2485e-05, 3.2485e-05, 2.2590e-04, 2.0134e-04, 2.8825e-04, 8.3566e-05,
          3.2485e-05, 8.3566e-05, 3.8838e-04, 2.9874e-04, 3.2485e-05, 3.2485e-05,
          3.2485e-05, 3.2485e-05, 3.2485e-05, 6.5446e-05, 5.0354e-04, 2.9469e-04,
          1.1671e-04, 1.8048e-04, 3.3164e-04, 1.2219e-04, 1.7762e-04, 8.9169e-05,
          1.4472e-04, 3.2485e-05, 1.4472e-04, 3.6550e-04, 1.7762e-04, 1.7762e-04,
          1.7762e-04, 3.1877e-04, 2.5010e-04, 2.4652e-04, 3.2485e-05, 3.2485e-05,
          2.4652e-04, 1.2035e-03, 2.6727e-04, 2.7966e-04, 3.2485e-05, 5.6207e-05,
          5.2786e-04, 1.2779e-04, 3.3140e-04, 2.4104e-04, 5.6207e-05, 2.6727e-04,
          1.2779e-04, 3.2485e-05, 6.1274e-04, 6.2609e-04, 2.3997e-04, 2.3997e-04,
          6.2609e-04, 5.3501e-04, 1.7595e-04, 3.2485e-05, 6.5446e-05, 3.3164e-04,
          3.2485e-05, 3.2783e-04, 6.0415e-04, 1.1671e-04, 3.2485e-05, 1.2035e-03,
          2.0325e-04, 1.7202e-04, 5.5552e-04, 6.4611e-04, 3.2485e-05, 4.5061e-04,
          3.2485e-05, 5.9843e-05, 5.0449e-04, 9.2745e-05, 3.1543e-04, 4.3654e-04,
          5.2786e-04, 8.3566e-05, 1.2779e-04, 5.6207e-05, 3.5930e-04, 3.5930e-04,
          5.0354e-04, 3.4690e-04, 3.2485e-05, 6.5446e-05, 6.0368e-04, 1.3733e-04,
          7.5531e-04, 5.1689e-04, 3.6812e-04, 1.9574e-04, 2.6178e-04, 5.9843e-05,
          2.8253e-04, 4.7779e-04, 1.5116e-04, 1.7595e-04, 3.2485e-05, 4.0364e-04,
          8.9169e-05, 1.2779e-04, 1.2779e-04, 5.9843e-05, 6.6996e-04, 1.2674e-03,
          7.5531e-04, 3.7980e-04, 3.2485e-05, 2.6727e-04, 3.7956e-04, 1.1463e-03,
          2.3651e-04, 5.2786e-04, 6.7472e-04, 1.7035e-04, 3.2485e-05, 6.1274e-04,
          6.1274e-04, 8.6403e-04, 2.5535e-04, 5.2786e-04, 3.7956e-04, 3.2485e-05,
          8.3566e-05, 3.2485e-05, 3.2485e-05, 5.6207e-05, 3.7766e-04, 1.6856e-04,
          5.6207e-05, 1.6856e-04, 1.4126e-04, 5.7983e-04, 1.4126e-04, 1.8001e-04,
          5.6207e-05, 5.6207e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.5455e-03, 6.3171e-02, 2.0279e-02, 1.6891e-02, 7.2693e-02,
          5.3040e-02, 1.3100e-02, 2.6073e-03, 2.6913e-03, 7.8735e-03, 8.7585e-03,
          4.6768e-03, 3.4370e-03, 2.3155e-03, 7.6981e-03, 1.8587e-03, 4.4518e-03,
          2.0657e-03, 1.2951e-03, 1.6537e-03, 1.6050e-03, 1.0014e-03, 2.6665e-03,
          8.9111e-03, 7.7858e-03, 9.9716e-03, 6.2485e-03, 2.8782e-03, 4.3259e-03,
          2.6665e-03, 3.8967e-03, 1.9951e-03, 1.0208e-02, 3.8719e-03, 4.8141e-03,
          2.5696e-02, 1.0803e-02, 5.6419e-03, 5.3215e-03, 2.3441e-03, 7.2746e-03,
          3.0098e-03, 2.7275e-03, 2.3518e-03, 2.2697e-03, 3.2673e-03, 9.6359e-03,
          1.1454e-03, 1.5306e-03, 9.7046e-03, 8.7128e-03, 4.8943e-03, 3.2864e-03,
          4.3373e-03, 4.1504e-03, 5.0507e-03, 2.0660e-02, 1.3062e-02, 3.1097e-02,
          1.5099e-02, 4.8485e-03, 4.2953e-03, 4.7722e-03, 4.0779e-03, 1.8677e-02,
          3.1261e-03, 1.8148e-03, 2.5864e-03, 2.2755e-03, 2.2964e-03, 2.3499e-03,
          2.6894e-03, 1.0208e-02, 5.9738e-03, 5.4703e-03, 3.1223e-03, 3.4313e-03,
          5.5962e-03, 1.7715e-02, 1.0567e-02, 2.4506e-02, 1.1894e-02, 2.0432e-02,
          1.0468e-02, 5.9128e-03, 3.4561e-03, 3.5267e-03, 1.6298e-03, 2.4624e-03,
          5.8975e-03, 7.4692e-03, 2.2049e-03, 4.3297e-03, 3.8223e-03, 3.0060e-03,
          3.9406e-03, 3.8929e-03, 4.7035e-03, 1.5045e-02, 1.7746e-02, 1.5274e-02,
          1.1429e-02, 3.5095e-03, 5.1918e-03, 7.8064e-02, 2.3723e-05, 8.1837e-05,
          6.6519e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 1.8489e-04, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          8.1837e-05, 2.3723e-05, 2.3723e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 6.6519e-05, 2.3723e-05,
          1.2457e-04, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          6.6519e-05, 7.4434e-04, 2.2769e-04, 2.3723e-05, 2.3723e-05, 1.2457e-04,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 6.6519e-05, 2.3723e-05, 2.3723e-05, 7.4434e-04,
          2.3723e-05, 2.3723e-05, 2.7084e-04, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 6.6519e-05, 2.3723e-05,
          2.3723e-05, 5.9938e-04, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          1.2457e-04, 2.3723e-05, 2.3723e-05, 2.7084e-04, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 8.1837e-05, 2.3723e-05, 2.8586e-04,
          1.2457e-04, 2.3723e-05, 2.3723e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 1.2457e-04, 2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 8.3208e-04,
          2.3723e-05, 2.3723e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05, 8.1837e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 1.2517e-04,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 8.1837e-05, 6.7091e-04, 1.2457e-04, 2.3723e-05,
          8.1837e-05, 8.1837e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 6.6519e-05, 6.8617e-04, 1.2469e-04, 2.3723e-05, 8.1837e-05,
          8.1837e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 6.6519e-05,
          8.1837e-05, 2.3723e-05, 2.7084e-04, 5.6696e-04, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.2769e-04, 2.7084e-04, 1.2457e-04, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 2.5129e-04, 2.3723e-05, 6.6519e-05, 2.3723e-05, 2.3723e-05,
          6.6519e-05, 2.3723e-05, 4.5443e-04, 8.1837e-05, 2.3723e-05, 2.3723e-05,
          2.3723e-05, 6.6519e-05, 1.2836e-03, 4.7064e-04, 2.3723e-05, 2.3723e-05,
          6.6996e-04, 8.3208e-04, 1.2457e-04, 2.3723e-05, 2.3723e-05, 1.2529e-04,
          2.3723e-05, 1.0595e-03, 1.2529e-04, 2.3723e-05, 2.3723e-05, 4.5443e-04,
          6.6519e-05, 2.3723e-05, 4.7064e-04, 4.5443e-04, 2.3723e-05, 1.2529e-04,
          2.3723e-05, 2.3723e-05, 1.0176e-03, 1.7762e-04, 2.3723e-05, 1.1721e-03,
          1.0452e-03, 8.1837e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 1.2457e-04,
          8.1837e-05, 2.1279e-04, 2.3723e-05, 2.3723e-05, 1.2529e-04, 2.3723e-05,
          9.5510e-04, 2.7108e-04, 2.3723e-05, 2.3723e-05, 3.9673e-04, 2.3723e-05,
          1.8489e-04, 9.7179e-04, 8.1837e-05, 6.6519e-05, 2.3723e-05, 2.3723e-05,
          8.1837e-05, 2.3723e-05, 2.3723e-05, 2.3723e-05, 1.2457e-04, 4.1151e-04,
          9.5510e-04, 1.2469e-04, 6.6519e-05, 6.6519e-05, 4.1151e-04, 2.9221e-03,
          3.8218e-04, 4.1151e-04, 7.8487e-04, 2.2769e-04, 2.3723e-05, 3.3855e-04,
          9.8705e-04, 1.0576e-03, 2.3723e-05, 6.1560e-04, 4.1151e-04, 2.3723e-05,
          2.3723e-05, 2.3723e-05, 6.6519e-05, 2.3723e-05, 6.6519e-05, 2.3723e-05,
          2.3723e-05, 6.6519e-05, 2.3723e-05, 3.2806e-04, 2.3723e-05, 6.6519e-05,
          2.3723e-05, 2.3723e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.1188e-03, 6.7932e-02, 1.7853e-02, 1.6296e-02, 6.0883e-02,
          5.9967e-02, 1.0681e-02, 4.6806e-03, 3.5572e-03, 9.1476e-03, 1.0399e-02,
          5.0468e-03, 7.5722e-03, 2.2449e-03, 2.0752e-02, 4.0474e-03, 8.0032e-03,
          3.7937e-03, 2.2240e-03, 8.2970e-04, 8.9979e-04, 5.4789e-04, 4.6806e-03,
          1.1230e-02, 1.0429e-02, 8.8806e-03, 8.8272e-03, 1.5659e-03, 8.8043e-03,
          2.7084e-03, 9.4604e-03, 2.9697e-03, 2.1500e-02, 7.9041e-03, 8.2703e-03,
          1.9180e-02, 3.8357e-03, 6.6910e-03, 1.0818e-02, 1.4381e-03, 4.5357e-03,
          1.2264e-03, 1.1911e-03, 9.9564e-04, 1.2884e-03, 6.7711e-03, 1.4534e-02,
          9.1124e-04, 7.8344e-04, 5.3101e-03, 1.1665e-02, 8.7280e-03, 5.8289e-03,
          2.8286e-03, 5.6648e-03, 3.2349e-03, 7.9346e-03, 9.2621e-03, 1.0452e-02,
          7.6637e-03, 4.3869e-03, 3.7746e-03, 5.3291e-03, 2.0084e-03, 3.4790e-02,
          1.9112e-03, 1.7624e-03, 2.7294e-03, 8.5115e-04, 6.9952e-04, 9.4414e-04,
          9.8515e-04, 1.5823e-02, 3.6182e-03, 9.5978e-03, 2.3098e-03, 2.9640e-03,
          6.3248e-03, 6.6414e-03, 4.2419e-03, 2.2827e-02, 4.7264e-03, 7.1793e-03,
          4.7951e-03, 9.0485e-03, 2.6073e-03, 5.2452e-03, 1.2598e-03, 1.3332e-03,
          8.5907e-03, 7.2594e-03, 1.2321e-03, 5.1804e-03, 1.6670e-03, 1.1930e-03,
          2.7618e-03, 2.0657e-03, 9.3460e-03, 1.4084e-02, 5.0163e-03, 7.9422e-03,
          1.7441e-02, 3.4180e-03, 4.1466e-03, 2.8519e-02, 8.5632e-02, 1.1039e-04,
          4.9949e-05, 1.7667e-04, 1.6272e-05, 1.7667e-04, 4.9949e-05, 4.9949e-05,
          1.6272e-05, 1.6272e-05, 1.7667e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.1039e-04, 1.6272e-05, 1.6272e-05, 7.6771e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 4.9949e-05, 4.9949e-05,
          1.8084e-04, 4.9949e-05, 1.6272e-05, 1.2034e-04, 4.9949e-05, 1.7667e-04,
          4.9949e-05, 7.1192e-04, 4.9949e-05, 1.6272e-05, 1.6272e-05, 1.1039e-04,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 4.9949e-05, 1.6272e-05, 1.6272e-05,
          4.9949e-05, 1.7667e-04, 1.1039e-04, 1.7667e-04, 1.6272e-05, 2.7084e-04,
          1.6272e-05, 1.6272e-05, 1.8084e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.1039e-04, 4.9949e-05, 1.6272e-05, 1.6272e-05, 1.1039e-04, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.1039e-04, 4.9949e-05, 1.6272e-05,
          4.9949e-05, 2.1040e-04, 1.6272e-05, 1.6272e-05, 7.6771e-05, 1.6272e-05,
          1.8084e-04, 1.6272e-05, 1.6272e-05, 1.8084e-04, 1.1039e-04, 1.6272e-05,
          1.6272e-05, 4.9949e-05, 1.6272e-05, 4.9949e-05, 4.9949e-05, 1.6272e-05,
          4.9949e-05, 1.6272e-05, 1.6272e-05, 7.6771e-05, 4.9949e-05, 1.8084e-04,
          1.1039e-04, 1.6272e-05, 1.6272e-05, 1.1039e-04, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.1039e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.1039e-04,
          1.6272e-05, 1.7667e-04, 1.6272e-05, 4.9949e-05, 1.6272e-05, 1.6272e-05,
          3.3569e-04, 2.1040e-04, 1.7667e-04, 1.6272e-05, 1.6272e-05, 2.7084e-04,
          1.6272e-05, 1.6272e-05, 7.6771e-05, 1.6272e-05, 1.6272e-05, 2.1780e-04,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.1039e-04, 1.8084e-04,
          1.7667e-04, 4.9949e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          2.1040e-04, 1.7667e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 2.3174e-04,
          1.6272e-05, 7.6771e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 7.6771e-05, 1.1039e-04, 1.8084e-04, 1.6272e-05,
          1.8084e-04, 1.1039e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 8.3303e-04, 4.1938e-04, 1.6272e-05, 1.8275e-04,
          1.1039e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 4.9949e-05, 1.6272e-05,
          2.1780e-04, 1.6272e-05, 1.2034e-04, 2.9564e-04, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.7667e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 4.2033e-04, 2.9564e-04, 1.2034e-04, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 3.0208e-04, 1.6272e-05, 4.9949e-05, 1.6272e-05, 1.6272e-05,
          4.9949e-05, 1.6272e-05, 4.6253e-04, 7.6771e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 4.9949e-05, 7.8726e-04, 4.9114e-04, 1.7667e-04, 1.6272e-05,
          6.0558e-04, 7.6246e-04, 1.2034e-04, 1.6272e-05, 1.6272e-05, 3.2353e-04,
          1.6272e-05, 2.9206e-04, 5.0497e-04, 7.6771e-05, 1.6272e-05, 3.0208e-04,
          7.6771e-05, 1.6272e-05, 4.7731e-04, 4.6253e-04, 4.9949e-05, 3.2353e-04,
          4.9949e-05, 1.6272e-05, 1.8084e-04, 2.9564e-04, 4.5586e-04, 8.3303e-04,
          1.0805e-03, 3.6263e-04, 1.6272e-05, 1.6272e-05, 2.9564e-04, 1.1039e-04,
          2.5845e-04, 2.3520e-04, 1.6272e-05, 1.6272e-05, 5.0497e-04, 1.6272e-05,
          8.6117e-04, 6.7902e-04, 2.3174e-04, 4.9949e-05, 2.9564e-04, 1.6272e-05,
          1.6272e-05, 8.8692e-04, 1.6272e-05, 4.9949e-05, 1.6272e-05, 1.6272e-05,
          3.6263e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 7.6103e-04, 3.0208e-04,
          8.6117e-04, 3.2353e-04, 1.1039e-04, 4.9949e-05, 3.0208e-04, 1.0881e-03,
          4.7731e-04, 5.7745e-04, 9.2888e-04, 1.6475e-04, 1.6272e-05, 5.3501e-04,
          5.7745e-04, 6.5804e-04, 2.3520e-04, 5.7745e-04, 5.7745e-04, 1.6272e-05,
          4.9949e-05, 4.9949e-05, 2.9063e-04, 1.6272e-05, 4.9949e-05, 7.6771e-05,
          1.6272e-05, 1.1039e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 7.5150e-04,
          4.9949e-05, 4.9949e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1139e-02, 5.9692e-02, 2.5848e-02, 2.2049e-02, 8.7646e-02,
          4.3823e-02, 1.7807e-02, 2.0580e-03, 2.1458e-03, 3.2635e-03, 3.7155e-03,
          1.9741e-03, 8.1024e-03, 1.3115e-02, 4.4365e-03, 2.0580e-03, 2.1744e-02,
          1.0651e-02, 1.1702e-03, 8.1787e-03, 5.9471e-03, 5.8441e-03, 4.8828e-03,
          3.8300e-03, 3.7498e-03, 1.3641e-02, 7.4539e-03, 9.5673e-03, 1.4595e-02,
          1.3885e-02, 3.4084e-03, 9.7733e-03, 7.5798e-03, 2.4109e-03, 4.1504e-03,
          7.6485e-03, 4.1542e-03, 5.0659e-03, 4.1809e-03, 1.0872e-02, 6.3362e-03,
          4.4479e-03, 3.0842e-03, 3.0689e-03, 2.6112e-03, 9.1171e-03, 8.4610e-03,
          5.9776e-03, 7.5569e-03, 1.4198e-02, 6.0730e-03, 3.4313e-03, 2.2564e-03,
          2.8191e-03, 2.9736e-03, 4.2458e-03, 6.6261e-03, 4.2038e-03, 5.5542e-03,
          4.9095e-03, 6.0791e-02, 6.7902e-03, 8.7967e-03, 3.6812e-03, 1.0536e-02,
          3.0003e-03, 1.3275e-03, 1.9970e-03, 3.1242e-03, 2.9926e-03, 2.7199e-03,
          3.5343e-03, 5.3902e-03, 4.0588e-03, 4.2229e-03, 2.3441e-03, 1.9703e-03,
          3.5839e-03, 7.2212e-03, 4.1733e-03, 4.7760e-03, 3.0499e-03, 4.2343e-03,
          5.3978e-03, 8.4152e-03, 5.6458e-03, 2.3098e-03, 1.4915e-03, 2.3193e-03,
          3.3817e-03, 3.9177e-03, 2.5654e-03, 2.6054e-03, 4.1199e-03, 4.2572e-03,
          1.5732e-02, 2.7466e-03, 3.6392e-03, 1.1314e-02, 6.3019e-03, 5.3787e-03,
          4.1275e-03, 2.3041e-03, 1.7471e-03, 6.7291e-03, 7.7019e-03, 1.1279e-01,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0860e-04, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 7.1824e-05, 1.6093e-05,
          1.0860e-04, 1.0860e-04, 5.2869e-05, 7.1824e-05, 1.0860e-04, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 5.2869e-05,
          1.6093e-05, 5.2869e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          7.1824e-05, 1.6093e-05, 2.2137e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.0860e-04, 7.1824e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 5.2869e-05, 5.2869e-05, 1.0860e-04, 1.6093e-05, 1.6093e-05,
          1.0860e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2929e-05, 1.0860e-04,
          1.0860e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0860e-04, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.8454e-04, 1.0860e-04, 5.2869e-05,
          5.2929e-05, 7.1824e-05, 1.6093e-05, 5.2869e-05, 1.0860e-04, 1.6093e-05,
          5.2869e-05, 1.6093e-05, 1.6093e-05, 1.0866e-04, 5.2869e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 5.2869e-05, 1.0860e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0860e-04, 2.7132e-04, 1.0860e-04,
          1.0866e-04, 5.2869e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 7.1824e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.0860e-04, 1.6093e-05, 2.1553e-04, 7.1824e-05, 1.0860e-04,
          2.7132e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.0860e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 1.6093e-05,
          1.0860e-04, 5.2929e-05, 2.7132e-04, 1.6093e-05, 1.6093e-05, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0920e-04,
          1.0866e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          5.2869e-05, 1.6093e-05, 1.6093e-05, 7.1824e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 2.2137e-04, 1.6093e-05, 1.6093e-05, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 5.2869e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0860e-04, 1.6093e-05, 1.0860e-04,
          1.0860e-04, 2.2137e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 5.2869e-05, 1.6093e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05,
          5.2869e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05, 3.4738e-04, 1.6093e-05,
          1.6093e-05, 5.2869e-05, 1.6093e-05, 5.2869e-05, 5.2869e-05, 1.6093e-05,
          1.6093e-05, 7.1824e-05, 1.6093e-05, 1.6093e-05, 2.2137e-04, 5.2869e-05,
          5.2869e-05, 5.2869e-05, 5.2869e-05, 3.8409e-04, 1.6093e-05, 2.2137e-04,
          2.2137e-04, 2.2137e-04, 7.1824e-05, 5.2869e-05, 2.2137e-04, 5.2869e-05,
          3.8409e-04, 1.6093e-05, 5.2869e-05, 3.8409e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 5.2869e-05, 1.0860e-04, 1.6093e-05,
          1.0860e-04, 5.2869e-05, 2.2137e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          2.2137e-04, 1.6093e-05, 5.2869e-05, 5.2869e-05, 3.8409e-04, 3.4738e-04,
          1.6093e-05, 3.4738e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.8454e-04, 1.8454e-04, 1.8454e-04, 1.8454e-04, 1.8454e-04, 7.1824e-05,
          1.6093e-05, 2.2137e-04, 1.8454e-04, 3.5238e-04, 7.1824e-05, 1.6093e-05,
          1.8454e-04, 7.1824e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3673e-03, 5.4108e-02, 1.8860e-02, 1.6907e-02, 5.2368e-02,
          4.6173e-02, 1.2215e-02, 1.1192e-02, 3.5343e-03, 3.5000e-03, 5.0163e-03,
          2.1267e-03, 2.5749e-03, 2.1687e-03, 3.5191e-03, 1.1009e-02, 4.4861e-03,
          3.0212e-03, 2.8687e-03, 1.2188e-03, 1.4172e-03, 1.5659e-03, 3.4313e-03,
          6.0120e-03, 4.0665e-03, 1.0857e-02, 6.1035e-03, 1.7653e-03, 4.4060e-03,
          2.4109e-03, 2.3849e-02, 1.9741e-03, 4.5662e-03, 2.5444e-03, 9.5978e-03,
          4.8103e-03, 2.3346e-03, 4.9095e-03, 4.0131e-03, 1.7357e-03, 6.1150e-03,
          3.1834e-03, 2.4109e-03, 2.3479e-03, 1.7767e-03, 3.7537e-03, 5.3215e-03,
          1.5793e-03, 1.6317e-03, 6.6185e-03, 1.6159e-02, 2.6062e-02, 4.5929e-03,
          6.4583e-03, 1.6144e-02, 5.8937e-03, 2.8629e-03, 8.4534e-03, 3.8738e-03,
          3.3150e-03, 5.8289e-03, 2.9892e-02, 2.8046e-02, 4.2419e-03, 9.7885e-03,
          3.6449e-03, 1.7519e-03, 2.1095e-03, 2.0962e-03, 1.9417e-03, 2.3994e-03,
          2.3823e-03, 6.8016e-03, 9.5291e-03, 3.3722e-02, 7.4768e-03, 7.4654e-03,
          1.3374e-02, 2.7084e-03, 2.4662e-03, 4.1847e-03, 2.6855e-03, 3.7270e-03,
          2.8515e-03, 9.5520e-02, 1.9379e-02, 3.3607e-03, 1.8339e-03, 2.1191e-03,
          5.4779e-03, 5.1422e-03, 2.4185e-03, 3.6469e-03, 3.4466e-03, 2.6779e-03,
          3.3035e-03, 4.0550e-03, 1.6037e-02, 5.7068e-03, 2.6150e-03, 2.9354e-03,
          4.6387e-03, 1.4277e-03, 1.6756e-03, 6.0654e-03, 6.9466e-03, 7.1602e-03,
          8.6487e-02, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05,
          1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 5.9247e-05, 5.9247e-05, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          1.6963e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 5.9247e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9545e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 2.1029e-04, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 5.9247e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.6963e-04,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9545e-05, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 5.9247e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 5.9247e-05,
          5.9247e-05, 1.8597e-05, 2.1040e-04, 5.9247e-05, 3.1972e-04, 5.9247e-05,
          5.9247e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.5793e-03,
          1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8895e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 7.3242e-04, 5.9247e-05, 1.8597e-05,
          5.9247e-05, 5.9545e-05, 3.1972e-04, 1.8597e-05, 2.1040e-04, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 5.9545e-05,
          5.9247e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          5.9247e-05, 1.8597e-05, 1.6963e-04, 1.6963e-04, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 2.1040e-04, 5.9247e-05, 1.8597e-05, 1.6963e-04,
          1.6963e-04, 2.1029e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 2.1052e-04,
          2.1040e-04, 7.3242e-04, 5.9545e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          1.6963e-04, 1.8597e-05, 1.8597e-05, 2.1040e-04, 2.1040e-04, 2.1029e-04,
          1.8597e-05, 5.9247e-05, 5.9247e-05, 5.9247e-05, 1.8597e-05, 2.1040e-04,
          2.1040e-04, 4.0197e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.8597e-05, 2.5105e-04, 1.8597e-05, 5.9545e-05, 1.6963e-04,
          1.8597e-05, 1.8597e-05, 1.8597e-05, 5.9247e-05, 1.8597e-05, 1.8597e-05,
          2.1029e-04, 2.5105e-04, 1.8597e-05, 2.1029e-04, 1.8597e-05, 1.8597e-05,
          2.1040e-04, 1.8597e-05, 1.8597e-05, 1.8597e-05, 1.8597e-05, 2.1040e-04,
          1.8597e-05, 5.9247e-05, 6.4468e-04, 1.8597e-05, 1.8597e-05, 2.6107e-04,
          1.8597e-05, 2.1040e-04, 4.0197e-04, 1.8597e-05, 2.5105e-04, 3.6144e-04,
          2.5105e-04, 1.8597e-05, 1.8597e-05, 2.1040e-04, 2.1040e-04, 1.8597e-05,
          1.8597e-05, 1.6963e-04, 1.6963e-04, 2.6536e-04, 4.5633e-04, 5.9247e-05,
          1.8597e-05, 4.9686e-04, 1.8597e-05, 3.8958e-04, 1.8597e-05, 1.8597e-05,
          1.8597e-05, 1.6963e-04], dtype=torch.float16),
  tensor([0.0000e+00, 9.3918e-03, 5.9937e-02, 2.2919e-02, 2.0920e-02, 9.1431e-02,
          4.4769e-02, 2.0706e-02, 1.4591e-03, 1.5402e-03, 4.1771e-03, 6.7291e-03,
          3.1319e-03, 1.8559e-03, 2.5349e-03, 2.8095e-03, 1.5697e-03, 2.8191e-03,
          1.7014e-03, 1.3571e-03, 2.3499e-03, 3.3855e-03, 2.5959e-03, 2.1706e-03,
          8.7738e-03, 6.5842e-03, 1.5419e-02, 5.1079e-03, 3.3226e-03, 2.4624e-03,
          2.5902e-03, 2.1019e-03, 1.4944e-03, 3.4275e-03, 2.0428e-03, 1.8330e-03,
          7.2136e-03, 5.7487e-03, 6.7329e-03, 2.6093e-03, 2.9125e-03, 2.1317e-02,
          3.0945e-02, 1.5732e-02, 1.7014e-02, 8.0566e-03, 2.2106e-03, 6.3782e-03,
          1.9503e-03, 3.0537e-03, 1.5160e-02, 3.2940e-03, 2.0370e-03, 1.6870e-03,
          2.1114e-03, 1.6346e-03, 2.4719e-03, 8.0643e-03, 4.3983e-03, 7.3242e-03,
          5.4207e-03, 5.0049e-03, 8.0185e-03, 8.8425e-03, 2.6474e-02, 1.6678e-02,
          1.3824e-02, 5.5771e-03, 4.4250e-03, 1.5671e-02, 1.5793e-02, 9.5520e-03,
          7.9880e-03, 6.8817e-03, 3.5248e-03, 2.4681e-03, 1.7204e-03, 1.2712e-03,
          2.2564e-03, 1.1093e-02, 1.0910e-02, 5.8823e-03, 7.6981e-03, 4.1466e-03,
          5.0964e-03, 7.2441e-03, 5.8670e-03, 5.8212e-03, 6.2256e-03, 9.6664e-03,
          9.1324e-03, 1.0963e-02, 1.0750e-02, 6.6109e-03, 1.1871e-02, 9.6664e-03,
          3.3092e-03, 2.5635e-03, 1.8444e-03, 5.4703e-03, 1.4656e-02, 7.7515e-03,
          6.2370e-03, 3.8853e-03, 2.9888e-03, 7.4959e-03, 5.3177e-03, 7.1526e-03,
          7.2823e-03, 6.1615e-02, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          7.3910e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.6844e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.1194e-04, 1.2398e-05, 7.3910e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 7.3910e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 2.0635e-04, 7.3910e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 7.3910e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 2.0635e-04, 7.3910e-05, 7.3910e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.6844e-04, 1.2398e-05, 1.2398e-05, 2.0635e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.0687e-04, 1.2398e-05, 2.0635e-04,
          1.0687e-04, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 7.3910e-05, 1.2398e-05, 1.1194e-04,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 2.0635e-04, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 7.3910e-05, 1.0687e-04, 2.0635e-04, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.1194e-04, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.7345e-04, 1.2398e-05, 1.0687e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.1194e-04, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 2.0635e-04, 7.9012e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.7345e-04, 1.0687e-04, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.1194e-04, 1.2398e-05, 7.3910e-05, 7.3910e-05, 1.2398e-05,
          1.1194e-04, 1.2398e-05, 1.2398e-05, 1.1194e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05, 1.1194e-04, 1.1194e-04, 1.2398e-05, 1.2398e-05,
          6.9571e-04, 1.1194e-04, 1.1194e-04, 1.2398e-05, 1.2398e-05, 2.0635e-04,
          1.2398e-05, 1.2398e-05, 2.0635e-04, 1.2398e-05, 7.3910e-05, 7.3910e-05,
          1.1194e-04, 1.2398e-05, 2.6798e-04, 1.2398e-05, 1.2398e-05, 2.0635e-04,
          7.3910e-05, 1.2398e-05, 2.0635e-04, 1.2398e-05, 1.2398e-05, 1.1194e-04,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.0687e-04, 1.0687e-04,
          1.0687e-04, 1.7345e-04, 1.2398e-05, 7.3910e-05, 2.0635e-04, 1.2398e-05,
          2.0635e-04, 2.0635e-04, 1.2398e-05, 1.2398e-05, 1.2398e-05, 1.2398e-05,
          1.0687e-04, 1.1194e-04, 1.2398e-05, 1.1194e-04, 1.2398e-05, 7.3910e-05,
          1.2398e-05, 1.2398e-05, 1.2398e-05, 7.3910e-05, 1.6844e-04, 1.2398e-05,
          2.0635e-04, 2.0635e-04, 1.2398e-05, 1.7345e-04, 1.2398e-05, 1.1194e-04,
          2.0635e-04, 1.2398e-05, 2.0635e-04, 7.3910e-05, 1.2398e-05, 1.1194e-04,
          1.1194e-04, 1.1194e-04, 7.3910e-05, 1.1194e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.1194e-04, 1.1194e-04, 1.2398e-05, 1.1194e-04, 1.1194e-04,
          1.2398e-05, 1.1194e-04, 1.2398e-05, 3.7932e-04, 1.2398e-05, 1.2398e-05,
          1.2398e-05, 1.2398e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.5373e-03, 6.4270e-02, 2.0386e-02, 3.0640e-02, 5.6030e-02,
          4.9622e-02, 1.8311e-02, 2.2869e-03, 2.1629e-03, 1.0773e-02, 1.6632e-02,
          7.6675e-03, 3.0136e-03, 1.6003e-03, 4.7379e-03, 2.4109e-03, 4.7760e-03,
          3.1071e-03, 1.7290e-03, 8.4782e-04, 1.5583e-03, 1.2341e-03, 4.0092e-03,
          2.6245e-02, 7.0343e-03, 1.4946e-02, 8.4152e-03, 1.2140e-03, 5.0354e-03,
          2.3422e-03, 3.9024e-03, 1.7338e-03, 6.7215e-03, 6.5727e-03, 2.9297e-03,
          8.1329e-03, 3.0842e-03, 1.2398e-02, 5.9090e-03, 1.1578e-03, 2.0157e-02,
          7.7667e-03, 5.0278e-03, 4.5128e-03, 4.1924e-03, 4.9553e-03, 1.1047e-02,
          1.1148e-03, 1.1282e-03, 6.4316e-03, 4.5319e-03, 3.1662e-03, 2.7370e-03,
          1.2150e-03, 1.9665e-03, 1.1005e-03, 6.6414e-03, 4.1428e-03, 7.2746e-03,
          3.0556e-03, 3.2787e-03, 3.6297e-03, 6.8436e-03, 1.3184e-02, 2.5269e-02,
          9.6130e-03, 6.8321e-03, 6.5079e-03, 3.7098e-03, 2.9888e-03, 2.7332e-03,
          2.3785e-03, 2.0111e-02, 1.7776e-03, 3.3989e-03, 8.8549e-04, 1.1806e-03,
          1.7929e-03, 7.6981e-03, 8.6441e-03, 1.2978e-02, 7.2174e-03, 2.5635e-03,
          1.8978e-03, 8.5754e-03, 2.8172e-03, 1.4839e-02, 7.2365e-03, 6.0692e-03,
          2.9144e-02, 2.7603e-02, 3.6545e-03, 1.6739e-02, 2.7599e-03, 2.2793e-03,
          1.5755e-03, 8.6164e-04, 3.5381e-03, 7.7820e-03, 7.4921e-03, 1.0651e-02,
          2.6276e-02, 3.0174e-03, 5.0430e-03, 1.1223e-02, 8.4229e-03, 4.6616e-03,
          7.5226e-03, 1.5099e-02, 5.2521e-02, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05, 6.3002e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          7.2896e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.2219e-04,
          1.3709e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 6.3002e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05,
          6.3002e-05, 1.3709e-05, 1.3709e-05, 4.5514e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 2.4319e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05, 1.3709e-05, 2.9254e-04,
          6.3002e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 2.9254e-04,
          6.3002e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 7.2896e-05, 1.2219e-04, 2.9254e-04,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 6.3002e-05, 6.3002e-05, 1.2219e-04, 1.3709e-05,
          6.3002e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05, 7.2896e-05, 1.3709e-05,
          1.3709e-05, 6.3002e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 2.9254e-04, 7.2896e-05,
          6.3002e-05, 1.3709e-05, 1.2219e-04, 6.3002e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 7.2896e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 7.2896e-05, 6.3002e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05,
          7.2896e-05, 2.4319e-04, 2.4319e-04, 7.2896e-05, 1.3709e-05, 1.3709e-05,
          1.2219e-04, 1.3709e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 2.4319e-04, 1.2219e-04, 1.3709e-05, 1.3709e-05,
          7.2896e-05, 6.4039e-04, 7.2896e-05, 1.3709e-05, 1.3709e-05, 1.2219e-04,
          1.3709e-05, 6.4039e-04, 2.9325e-04, 6.3002e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.2219e-04, 7.2896e-05, 1.3709e-05, 2.9254e-04,
          1.3709e-05, 1.3709e-05, 2.9254e-04, 1.3709e-05, 7.2050e-04, 7.2896e-05,
          7.2896e-05, 6.3002e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05, 6.3002e-05,
          7.2050e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05,
          1.2219e-04, 1.2219e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          6.3002e-05, 2.9254e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 6.3002e-05, 1.3709e-05,
          8.1062e-04, 6.3002e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 3.3760e-04,
          6.3002e-05, 1.3709e-05, 1.2219e-04, 1.3709e-05, 1.3709e-05, 7.2896e-05,
          1.0853e-03, 1.0853e-03, 1.3709e-05, 2.4319e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 2.3556e-04, 1.3709e-05,
          1.3709e-05, 7.2896e-05, 1.3709e-05, 4.6659e-04, 7.2896e-05, 7.2896e-05,
          1.3709e-05, 1.3709e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.1526e-03, 6.1432e-02, 2.3102e-02, 1.5511e-02, 7.4219e-02,
          4.3823e-02, 1.5289e-02, 1.9569e-03, 1.8454e-03, 3.7003e-03, 8.3008e-03,
          2.8839e-03, 1.8482e-03, 2.9087e-03, 2.6550e-03, 2.0828e-03, 2.3403e-03,
          1.7128e-03, 1.8301e-03, 1.4505e-03, 1.7290e-03, 1.9464e-03, 2.2621e-03,
          1.0910e-02, 6.7787e-03, 1.1665e-02, 4.0703e-03, 1.9207e-03, 2.1362e-03,
          3.2711e-03, 2.7332e-03, 1.1473e-03, 2.6512e-03, 1.7071e-03, 2.0866e-03,
          3.7823e-03, 3.5534e-03, 3.4637e-03, 2.1973e-03, 2.0485e-03, 1.1055e-02,
          1.5701e-02, 1.0124e-02, 1.2527e-02, 4.3144e-03, 2.3251e-03, 4.2114e-03,
          2.0409e-03, 2.2697e-03, 1.0803e-02, 2.8744e-03, 2.3727e-03, 1.9779e-03,
          2.2831e-03, 1.8759e-03, 2.2335e-03, 3.9253e-03, 6.2332e-03, 5.6305e-03,
          5.3329e-03, 3.7632e-03, 7.2556e-03, 6.5842e-03, 2.5375e-02, 1.1024e-02,
          5.3253e-02, 1.0384e-02, 4.2381e-03, 8.7814e-03, 8.7738e-03, 8.0948e-03,
          4.3602e-03, 5.5313e-03, 2.6531e-03, 3.0994e-03, 2.4853e-03, 1.6747e-03,
          2.5463e-03, 4.8409e-03, 1.6129e-02, 4.5967e-03, 1.0490e-02, 3.7785e-03,
          3.6793e-03, 8.6975e-03, 5.5237e-03, 7.1983e-03, 1.7853e-02, 3.2959e-02,
          1.2276e-02, 9.1858e-03, 1.5671e-02, 8.3923e-03, 5.9891e-03, 4.7264e-03,
          2.3575e-03, 1.8034e-03, 2.1172e-03, 3.9787e-03, 6.4926e-03, 4.8065e-03,
          6.7177e-03, 1.0147e-02, 4.6539e-03, 5.2490e-03, 4.3068e-03, 5.3215e-03,
          8.7738e-03, 3.1708e-02, 1.4229e-02, 7.1777e-02, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.1230e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.8120e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.1230e-04, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          8.4937e-05, 1.6093e-05, 1.6093e-05, 1.8120e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.6093e-05, 1.8120e-04,
          8.4937e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 8.4937e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.8120e-04, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.1230e-04, 1.6093e-05, 1.8120e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.8120e-04, 7.7868e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 8.4937e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.1230e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.1230e-04, 1.1230e-04, 1.1230e-04, 1.1230e-04, 1.6093e-05, 1.6093e-05,
          7.7868e-04, 1.6093e-05, 1.1230e-04, 1.6093e-05, 1.6093e-05, 1.8120e-04,
          1.6093e-05, 1.6332e-05, 1.8346e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.1230e-04, 1.6093e-05, 1.8120e-04, 1.6093e-05, 1.6093e-05, 1.8144e-04,
          1.6093e-05, 1.6093e-05, 1.8144e-04, 1.6093e-05, 1.6093e-05, 1.1230e-04,
          1.6093e-05, 1.1230e-04, 1.6093e-05, 1.6093e-05, 8.4937e-05, 8.4937e-05,
          8.4937e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.8144e-04, 1.1230e-04,
          1.8120e-04, 1.8144e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.1230e-04, 8.4937e-05, 7.1001e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 8.4937e-05, 1.8120e-04, 1.6093e-05,
          1.8120e-04, 1.8120e-04, 1.1230e-04, 1.1230e-04, 1.6093e-05, 1.6093e-05,
          1.8120e-04, 1.6093e-05, 1.8120e-04, 8.4937e-05, 1.6093e-05, 1.1230e-04,
          1.1230e-04, 1.8120e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.1230e-04, 1.1230e-04, 4.1485e-04, 1.1033e-04, 4.1485e-04, 1.1230e-04,
          1.6093e-05, 4.1485e-04, 1.1230e-04, 5.1689e-04, 1.6093e-05, 1.1230e-04,
          1.6093e-05, 1.6093e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.3318e-03, 6.7505e-02, 2.1439e-02, 2.7954e-02, 6.4697e-02,
          5.1636e-02, 1.9653e-02, 1.7309e-03, 1.4153e-03, 4.9324e-03, 1.3718e-02,
          3.7327e-03, 1.7309e-03, 2.3556e-03, 2.0561e-03, 1.8387e-03, 2.1229e-03,
          1.7328e-03, 1.4763e-03, 6.6471e-04, 7.9870e-04, 9.2411e-04, 2.2068e-03,
          1.7548e-02, 4.6844e-03, 1.5717e-02, 6.7635e-03, 1.0529e-03, 2.2202e-03,
          3.3398e-03, 2.7866e-03, 9.8038e-04, 2.1725e-03, 2.1076e-03, 1.3485e-03,
          2.6264e-03, 2.2869e-03, 3.2673e-03, 1.8587e-03, 1.0424e-03, 7.3013e-03,
          6.0616e-03, 4.2419e-03, 5.2643e-03, 2.0561e-03, 2.6436e-03, 3.3703e-03,
          1.3142e-03, 1.1005e-03, 7.3700e-03, 2.0065e-03, 2.1935e-03, 1.4801e-03,
          1.1797e-03, 1.4582e-03, 1.0872e-03, 2.4452e-03, 4.2114e-03, 4.2763e-03,
          3.2673e-03, 2.1553e-03, 3.3131e-03, 2.7122e-03, 1.3687e-02, 7.1144e-03,
          7.6782e-02, 1.5480e-02, 4.2267e-03, 2.8000e-03, 2.6741e-03, 3.7174e-03,
          1.9016e-03, 8.4152e-03, 1.5135e-03, 2.7199e-03, 1.2112e-03, 9.5367e-04,
          1.5087e-03, 2.9736e-03, 2.4506e-02, 5.4321e-03, 2.1820e-02, 2.1629e-03,
          1.8215e-03, 5.8861e-03, 2.6207e-03, 9.4223e-03, 2.8793e-02, 3.6041e-02,
          2.0020e-02, 9.6283e-03, 6.7024e-03, 1.0956e-02, 1.9569e-03, 1.8091e-03,
          1.3008e-03, 9.1028e-04, 1.7862e-03, 3.2215e-03, 4.2152e-03, 4.7569e-03,
          1.3512e-02, 1.7578e-02, 1.0803e-02, 3.2558e-03, 3.2730e-03, 2.5501e-03,
          4.8332e-03, 9.3460e-03, 1.6800e-02, 1.7624e-02, 7.0251e-02, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 5.6148e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 5.6148e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.1148e-04, 7.1704e-05, 7.1704e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.1148e-04,
          2.3127e-05, 2.3127e-05, 3.7622e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 5.6148e-05, 5.6148e-05, 1.0473e-04, 5.6148e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 5.6148e-05, 5.6148e-05, 5.6148e-05, 2.1076e-04, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05,
          5.6148e-05, 2.3127e-05, 2.3127e-05, 2.1076e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 2.3127e-05, 4.8280e-04,
          5.6148e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 2.3127e-05,
          2.3127e-05, 1.0473e-04, 2.3127e-05, 5.6148e-05, 1.0473e-04, 1.0473e-04,
          5.6148e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 1.0473e-04, 5.6148e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 7.1704e-05, 5.6148e-05, 7.1704e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.7846e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 5.6148e-05, 7.1704e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 5.6148e-05, 1.2529e-04, 1.0473e-04, 1.7846e-04,
          2.3127e-05, 2.3127e-05, 1.0473e-04, 2.3127e-05, 1.7846e-04, 2.3127e-05,
          2.3127e-05, 2.1076e-04, 2.3127e-05, 1.0473e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 1.7846e-04, 2.3127e-05,
          5.6148e-05, 2.3127e-05, 3.1757e-04, 3.7622e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.1148e-04, 1.2982e-04, 2.3127e-05,
          2.3127e-05, 1.0473e-04, 5.6148e-05, 2.1148e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.8443e-04, 7.1704e-05, 7.1704e-05, 2.3127e-05, 2.3127e-05,
          3.1757e-04, 2.3127e-05, 1.2982e-04, 7.6652e-05, 7.1704e-05, 1.7846e-04,
          1.7846e-04, 2.3127e-05, 2.8443e-04, 3.1757e-04, 1.2982e-04, 2.8443e-04,
          5.1022e-04, 2.1148e-04, 2.8443e-04, 2.3127e-05, 2.3127e-05, 2.1076e-04,
          2.3127e-05, 3.1757e-04, 3.1781e-04, 7.6652e-05, 2.3127e-05, 1.7774e-04,
          2.8443e-04, 5.6148e-05, 2.1148e-04, 1.7846e-04, 2.3127e-05, 3.1757e-04,
          2.3127e-05, 2.3127e-05, 4.2391e-04, 5.6148e-05, 1.6284e-04, 2.8443e-04,
          1.2982e-04, 2.1148e-04, 2.3127e-05, 2.3127e-05, 1.6260e-04, 1.6260e-04,
          7.6652e-05, 1.7774e-04, 2.3127e-05, 2.3127e-05, 3.1757e-04, 1.7846e-04,
          1.0242e-03, 2.6941e-04, 1.2982e-04, 2.3127e-05, 5.6148e-05, 2.3127e-05,
          5.6148e-05, 2.8443e-04, 5.6148e-05, 4.5657e-04, 2.3127e-05, 2.3127e-05,
          1.2982e-04, 2.3127e-05, 2.3127e-05, 5.6148e-05, 2.6941e-04, 2.3127e-05,
          2.3198e-04, 5.9223e-04, 2.3127e-05, 1.7774e-04, 1.2982e-04, 5.5933e-04,
          1.6284e-04, 1.2982e-04, 6.9904e-04, 5.6148e-05, 2.3127e-05, 4.5657e-04,
          6.6614e-04, 2.1148e-04, 1.2922e-04, 2.8443e-04, 4.0817e-04, 2.3127e-05,
          7.1704e-05, 7.1704e-05, 7.1704e-05, 4.5609e-04, 2.3711e-04, 1.0473e-04,
          2.3127e-05, 2.7013e-04, 7.1704e-05, 5.4693e-04, 2.3711e-04, 2.3711e-04,
          7.1704e-05, 7.1704e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0229e-03, 6.5308e-02, 2.0386e-02, 2.7893e-02, 5.0873e-02,
          4.6326e-02, 1.8311e-02, 1.7157e-03, 1.5564e-03, 6.5918e-03, 1.3725e-02,
          4.2114e-03, 1.9226e-03, 2.0790e-03, 2.1782e-03, 1.8120e-03, 3.1166e-03,
          2.0256e-03, 1.5802e-03, 6.4850e-04, 9.6512e-04, 1.0166e-03, 2.9736e-03,
          1.7975e-02, 4.7836e-03, 1.4595e-02, 8.6899e-03, 1.0347e-03, 2.9449e-03,
          3.4237e-03, 2.8877e-03, 1.1339e-03, 3.1681e-03, 3.1281e-03, 2.1763e-03,
          3.7212e-03, 2.1210e-03, 6.6681e-03, 2.7180e-03, 1.0509e-03, 1.3435e-02,
          6.7444e-03, 4.8027e-03, 5.9624e-03, 2.8439e-03, 3.3112e-03, 5.5466e-03,
          1.4009e-03, 1.1940e-03, 5.8250e-03, 2.8458e-03, 2.4166e-03, 2.1820e-03,
          1.0891e-03, 1.5526e-03, 9.9373e-04, 3.9177e-03, 3.3035e-03, 4.8256e-03,
          2.8210e-03, 2.2850e-03, 2.9736e-03, 3.6278e-03, 1.5274e-02, 1.0933e-02,
          3.8971e-02, 1.6281e-02, 6.5575e-03, 3.2730e-03, 3.1090e-03, 3.7632e-03,
          2.2373e-03, 1.5793e-02, 1.3885e-03, 2.9888e-03, 9.6607e-04, 1.0462e-03,
          1.5364e-03, 5.0888e-03, 2.4963e-02, 8.2626e-03, 1.9089e-02, 1.9722e-03,
          1.6766e-03, 6.1035e-03, 2.3270e-03, 1.2413e-02, 2.4124e-02, 2.2537e-02,
          2.6779e-02, 1.9745e-02, 4.3221e-03, 1.5732e-02, 1.9703e-03, 1.9741e-03,
          1.2779e-03, 7.9584e-04, 3.0136e-03, 6.1150e-03, 6.3820e-03, 7.6180e-03,
          1.6266e-02, 9.9564e-03, 1.0269e-02, 5.6152e-03, 4.0016e-03, 2.8229e-03,
          5.3329e-03, 1.1299e-02, 2.9556e-02, 9.7885e-03, 3.0228e-02, 4.0924e-02,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 4.5180e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 4.5180e-05, 4.5180e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05,
          9.0837e-05, 5.7936e-05, 5.7936e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 9.0837e-05,
          1.2279e-05, 1.2279e-05, 1.9312e-04, 1.2279e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 4.5180e-05, 1.2279e-05, 9.0837e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05,
          1.2279e-05, 4.5180e-05, 4.5180e-05, 4.5180e-05, 1.9312e-04, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05,
          4.5180e-05, 1.2279e-05, 1.2279e-05, 1.9312e-04, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 1.2279e-05, 1.9312e-04,
          4.5180e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 1.2279e-05,
          1.2279e-05, 9.0837e-05, 1.2279e-05, 4.5180e-05, 9.0837e-05, 9.0837e-05,
          4.5180e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 9.0837e-05, 4.5180e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 5.7936e-05, 4.5180e-05, 1.6022e-04,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 5.7936e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 9.0837e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 4.5180e-05, 9.0837e-05, 9.0837e-05, 5.7936e-05,
          1.2279e-05, 1.2279e-05, 9.0837e-05, 1.2279e-05, 5.7936e-05, 1.2279e-05,
          1.2279e-05, 9.0837e-05, 1.2279e-05, 9.0837e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 1.2279e-05, 3.5977e-04, 5.7936e-05,
          4.5180e-05, 1.2279e-05, 1.9312e-04, 1.9312e-04, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.2279e-05, 1.2279e-05, 9.0837e-05, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 9.0837e-05, 4.5180e-05, 4.6206e-04, 1.2279e-05, 1.2279e-05,
          1.2279e-05, 1.6022e-04, 5.7936e-05, 9.0837e-05, 1.2279e-05, 1.2279e-05,
          4.6206e-04, 1.2279e-05, 1.2279e-05, 6.1691e-05, 5.7936e-05, 5.7936e-05,
          5.7936e-05, 1.2279e-05, 4.2915e-04, 4.6206e-04, 1.2279e-05, 1.2279e-05,
          1.9312e-04, 3.5977e-04, 1.6022e-04, 1.2279e-05, 1.2279e-05, 1.9312e-04,
          1.2279e-05, 4.6206e-04, 1.9348e-04, 6.1691e-05, 1.2279e-05, 1.6022e-04,
          1.6022e-04, 4.5180e-05, 9.0837e-05, 3.2687e-04, 1.2279e-05, 1.9312e-04,
          1.2279e-05, 1.2279e-05, 4.6206e-04, 4.5180e-05, 6.6137e-04, 4.2915e-04,
          1.2279e-05, 3.5977e-04, 1.2279e-05, 1.2279e-05, 6.1691e-05, 1.7726e-04,
          3.4690e-04, 1.6022e-04, 1.2279e-05, 1.2279e-05, 1.9312e-04, 5.7936e-05,
          4.9210e-04, 1.7726e-04, 3.1424e-04, 1.2279e-05, 4.5180e-05, 1.2279e-05,
          4.5180e-05, 1.6022e-04, 4.5180e-05, 1.6022e-04, 1.2279e-05, 1.2279e-05,
          5.7936e-05, 1.2279e-05, 1.2279e-05, 4.5180e-05, 6.1691e-05, 1.2279e-05,
          3.5977e-04, 5.2023e-04, 1.2279e-05, 1.6022e-04, 1.2279e-05, 4.8733e-04,
          4.5180e-05, 2.8133e-04, 7.8917e-04, 4.5180e-05, 1.2279e-05, 1.6022e-04,
          7.5626e-04, 3.5977e-04, 1.1456e-04, 4.2915e-04, 4.2915e-04, 1.2279e-05,
          5.7936e-05, 5.7936e-05, 1.2279e-05, 3.6240e-04, 2.2006e-04, 9.0837e-05,
          1.2279e-05, 2.2006e-04, 5.7936e-05, 4.9114e-04, 2.2006e-04, 2.2006e-04,
          1.2279e-05, 5.7936e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.4125e-03, 6.9031e-02, 2.0386e-02, 1.8494e-02, 6.1584e-02,
          5.9631e-02, 1.5068e-02, 3.0155e-03, 2.4872e-03, 8.3542e-03, 1.3931e-02,
          5.0774e-03, 5.2872e-03, 1.4057e-03, 1.7181e-02, 3.3588e-03, 4.6234e-03,
          2.7771e-03, 2.8572e-03, 7.8201e-04, 1.0071e-03, 1.2159e-03, 4.5166e-03,
          2.0248e-02, 2.6581e-02, 1.2329e-02, 6.6032e-03, 9.8324e-04, 4.6043e-03,
          1.8969e-03, 5.0125e-03, 1.3790e-03, 9.9869e-03, 6.3095e-03, 2.9144e-03,
          6.8855e-03, 2.1515e-03, 6.2714e-03, 6.2637e-03, 1.0891e-03, 9.9945e-03,
          4.3221e-03, 3.8033e-03, 3.3436e-03, 3.7117e-03, 6.5231e-03, 1.5762e-02,
          1.3227e-03, 1.3361e-03, 7.7324e-03, 2.8934e-03, 2.8038e-03, 3.0632e-03,
          1.0490e-03, 1.7395e-03, 9.5224e-04, 2.9621e-03, 3.1319e-03, 3.6812e-03,
          3.6316e-03, 2.5024e-03, 3.8471e-03, 5.3101e-03, 7.7934e-03, 4.4403e-02,
          5.4817e-03, 6.2561e-03, 5.1689e-03, 2.5139e-03, 2.1820e-03, 2.5902e-03,
          2.2316e-03, 1.5411e-02, 1.2321e-03, 3.6564e-03, 1.0719e-03, 1.0757e-03,
          1.6880e-03, 2.9831e-03, 3.5210e-03, 8.3771e-03, 3.1891e-03, 3.1223e-03,
          2.1973e-03, 8.6975e-03, 2.6436e-03, 1.2947e-02, 3.6144e-03, 4.1389e-03,
          1.9409e-02, 1.5221e-02, 4.9706e-03, 1.2253e-02, 3.9825e-03, 2.9106e-03,
          1.3208e-03, 7.6342e-04, 4.0169e-03, 6.7558e-03, 3.4885e-03, 4.7379e-03,
          1.3672e-02, 5.1727e-03, 2.7981e-03, 6.9923e-03, 2.3773e-02, 3.2349e-03,
          8.5754e-03, 4.8027e-03, 1.8204e-02, 5.6343e-03, 5.8250e-03, 5.8441e-03,
          9.2407e-02, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 8.7559e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          9.5427e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          9.5427e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.5427e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04,
          1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.5427e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.6439e-04, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 8.7559e-05, 1.8656e-05, 1.6439e-04,
          8.7559e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 8.7559e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 8.7559e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.6439e-04,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.6439e-04, 1.8656e-05,
          1.6439e-04, 8.7559e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05, 1.9729e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.5427e-05, 1.8656e-05,
          8.7559e-05, 1.8656e-05, 1.6439e-04, 1.6439e-04, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.6439e-04, 9.5427e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 7.8297e-04, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05,
          9.5427e-05, 9.5427e-05, 1.8656e-05, 8.7559e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 9.5427e-05, 9.5427e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05,
          9.5427e-05, 1.6439e-04, 9.5427e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04,
          1.8656e-05, 1.6439e-04, 1.6546e-04, 8.7559e-05, 1.8656e-05, 1.8656e-05,
          1.6439e-04, 1.8656e-05, 8.7559e-05, 9.5427e-05, 1.8656e-05, 1.6439e-04,
          1.8656e-05, 1.8656e-05, 1.6439e-04, 8.7559e-05, 8.7559e-05, 9.5427e-05,
          9.5427e-05, 1.6439e-04, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.6439e-04,
          8.7559e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05,
          8.5163e-04, 8.5163e-04, 1.8656e-05, 1.8656e-05, 1.6439e-04, 1.8656e-05,
          8.7559e-05, 1.6439e-04, 1.8656e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05,
          9.5427e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 4.8494e-04, 1.8656e-05,
          8.5163e-04, 1.6439e-04, 1.6439e-04, 9.5427e-05, 1.8656e-05, 9.5427e-05,
          1.6439e-04, 1.8656e-05, 1.6439e-04, 1.8656e-05, 1.8656e-05, 9.5427e-05,
          9.5427e-05, 7.8297e-04, 1.8656e-05, 9.5427e-05, 1.8656e-05, 1.8656e-05,
          1.8656e-05, 1.8656e-05, 1.8656e-05, 1.8656e-05, 9.5427e-05, 1.8656e-05,
          1.8656e-05, 8.7881e-04, 9.5427e-05, 6.4611e-04, 1.8656e-05, 9.5427e-05,
          9.5427e-05, 1.8656e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.1711e-03, 6.1737e-02, 2.4734e-02, 1.9669e-02, 7.6843e-02,
          4.3518e-02, 1.8250e-02, 1.6947e-03, 1.5469e-03, 3.9787e-03, 5.3291e-03,
          2.3212e-03, 2.5043e-03, 2.4681e-03, 4.0855e-03, 1.8921e-03, 4.9782e-03,
          2.4738e-03, 1.4496e-03, 2.9640e-03, 4.2419e-03, 3.9902e-03, 2.8534e-03,
          7.6408e-03, 7.3662e-03, 1.3603e-02, 5.0621e-03, 3.0460e-03, 3.6488e-03,
          2.6855e-03, 2.7180e-03, 1.9255e-03, 6.0387e-03, 2.9335e-03, 2.2354e-03,
          1.0414e-02, 4.4556e-03, 7.5111e-03, 3.8490e-03, 3.0098e-03, 2.0737e-02,
          1.8997e-02, 1.2306e-02, 1.2009e-02, 9.4070e-03, 3.8795e-03, 1.2100e-02,
          2.7390e-03, 4.2648e-03, 1.2482e-02, 3.3455e-03, 2.3823e-03, 1.7519e-03,
          1.5974e-03, 1.8129e-03, 2.3251e-03, 7.6599e-03, 3.4180e-03, 5.9547e-03,
          4.7112e-03, 5.4855e-03, 5.1880e-03, 8.9722e-03, 1.2817e-02, 2.4536e-02,
          6.9809e-03, 4.3411e-03, 5.3482e-03, 1.7258e-02, 1.4572e-02, 9.1095e-03,
          9.6130e-03, 6.6071e-03, 2.6035e-03, 2.6417e-03, 1.2817e-03, 1.1272e-03,
          2.1400e-03, 6.6261e-03, 4.4632e-03, 5.2528e-03, 3.2940e-03, 3.7346e-03,
          4.0817e-03, 9.2926e-03, 4.1504e-03, 4.9019e-03, 3.0422e-03, 4.6768e-03,
          1.0239e-02, 1.1871e-02, 8.0719e-03, 8.0490e-03, 1.7654e-02, 1.1948e-02,
          3.1605e-03, 1.8644e-03, 2.4052e-03, 7.2212e-03, 6.4812e-03, 4.8943e-03,
          4.8523e-03, 2.1496e-03, 1.9646e-03, 7.3509e-03, 7.6141e-03, 6.8321e-03,
          8.9874e-03, 2.3758e-02, 1.3893e-02, 8.1787e-03, 4.9820e-03, 5.6419e-03,
          9.1324e-03, 5.2612e-02, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 6.2227e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          6.2227e-05, 1.5557e-05, 1.5557e-05, 1.2141e-04, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.2141e-04, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          1.2141e-04, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          6.2227e-05, 1.5557e-05, 1.5557e-05, 1.6809e-04, 1.6809e-04, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          6.2227e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.2141e-04, 1.6809e-04,
          6.2227e-05, 1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05,
          6.2227e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.2141e-04,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5855e-05,
          1.5557e-05, 1.5557e-05, 6.2227e-05, 6.2227e-05, 6.2227e-05, 1.5557e-05,
          6.2227e-05, 6.2644e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05, 6.2585e-05,
          6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          6.2227e-05, 1.5557e-05, 1.6809e-04, 1.6809e-04, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05,
          1.5557e-05, 1.5557e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          1.5557e-05, 5.4550e-04, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          1.6809e-04, 6.2227e-05, 1.6809e-04, 1.5557e-05, 1.5557e-05, 1.6809e-04,
          1.5557e-05, 6.2346e-05, 1.6844e-04, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          6.2227e-05, 1.5557e-05, 6.2227e-05, 1.5557e-05, 1.2141e-04, 1.6809e-04,
          1.5557e-05, 1.5557e-05, 1.6809e-04, 1.2141e-04, 5.9175e-04, 1.5557e-05,
          6.2227e-05, 1.6809e-04, 1.5557e-05, 1.5557e-05, 6.2227e-05, 6.2227e-05,
          5.9175e-04, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.6809e-04, 1.5557e-05,
          9.9087e-04, 1.6820e-04, 1.5557e-05, 6.2227e-05, 6.2227e-05, 1.5557e-05,
          6.2227e-05, 1.6809e-04, 6.2227e-05, 1.2141e-04, 1.5557e-05, 1.5557e-05,
          1.6809e-04, 1.5557e-05, 1.5557e-05, 6.2227e-05, 8.8501e-04, 1.5557e-05,
          1.6809e-04, 1.6809e-04, 5.9223e-04, 1.5557e-05, 1.5557e-05, 1.2141e-04,
          1.6809e-04, 1.5557e-05, 1.6809e-04, 6.2227e-05, 1.5557e-05, 1.5557e-05,
          6.2227e-05, 2.9469e-03, 1.5557e-05, 1.5557e-05, 1.5557e-05, 1.5557e-05,
          1.2141e-04, 1.2141e-04, 5.4550e-04, 1.2422e-04, 1.2141e-04, 1.5557e-05,
          1.5557e-05, 1.6809e-04, 1.2141e-04, 3.9029e-04, 1.5557e-05, 1.2141e-04,
          1.5557e-05, 1.5557e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1208e-02, 5.9387e-02, 3.2166e-02, 2.5299e-02, 8.7646e-02,
          4.2603e-02, 2.3911e-02, 1.7300e-03, 1.4887e-03, 2.3556e-03, 3.3398e-03,
          1.4420e-03, 5.7678e-03, 8.3389e-03, 3.1452e-03, 1.9436e-03, 1.6373e-02,
          8.7357e-03, 1.3666e-03, 6.6948e-03, 6.5689e-03, 9.2697e-03, 6.4850e-03,
          4.0131e-03, 4.3373e-03, 1.7822e-02, 7.3929e-03, 7.9498e-03, 9.9792e-03,
          1.0155e-02, 2.4357e-03, 6.1188e-03, 4.4670e-03, 1.7586e-03, 2.3365e-03,
          3.6125e-03, 2.1133e-03, 4.4899e-03, 2.6798e-03, 7.4387e-03, 7.1335e-03,
          6.4011e-03, 4.4518e-03, 4.7569e-03, 3.5973e-03, 9.7198e-03, 8.2245e-03,
          7.1526e-03, 1.1971e-02, 1.7548e-02, 3.7003e-03, 1.9913e-03, 1.6880e-03,
          1.9531e-03, 1.7576e-03, 2.8095e-03, 2.5711e-03, 2.2507e-03, 2.5501e-03,
          2.8038e-03, 3.1403e-02, 4.6806e-03, 6.6643e-03, 4.0970e-03, 1.0399e-02,
          3.4904e-03, 1.7405e-03, 2.7103e-03, 5.2605e-03, 5.3940e-03, 4.5929e-03,
          6.8741e-03, 4.5357e-03, 3.0918e-03, 2.5940e-03, 1.6890e-03, 1.2083e-03,
          2.2049e-03, 2.8954e-03, 2.2411e-03, 2.3098e-03, 1.7090e-03, 2.1591e-03,
          2.8648e-03, 6.4621e-03, 3.9597e-03, 2.5597e-03, 1.8511e-03, 2.5654e-03,
          3.7594e-03, 4.3716e-03, 3.9406e-03, 3.1948e-03, 7.4005e-03, 9.3765e-03,
          1.1314e-02, 2.2602e-03, 2.6665e-03, 7.7248e-03, 3.0975e-03, 2.4109e-03,
          2.7237e-03, 1.5450e-03, 1.1234e-03, 3.3245e-03, 4.5891e-03, 5.0018e-02,
          6.2714e-03, 8.7967e-03, 5.6915e-03, 4.5547e-03, 3.1109e-03, 2.8114e-03,
          3.9673e-03, 1.3016e-02, 1.0175e-01, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8787e-04, 1.8787e-04, 1.8418e-05, 1.3590e-04, 1.8787e-04, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.3590e-04, 1.8418e-05, 1.8787e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.3590e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8787e-04, 2.0142e-03, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05,
          1.8787e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05,
          7.0333e-05, 1.8418e-05, 1.8418e-05, 1.3590e-04, 1.8787e-04, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.3590e-04, 1.8787e-04, 1.8418e-05,
          1.8418e-05, 1.3590e-04, 1.8418e-05, 7.0333e-05, 1.8787e-04, 1.3590e-04,
          7.0333e-05, 1.8418e-05, 1.8418e-05, 1.8787e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 6.5804e-04, 1.8787e-04, 1.8418e-05,
          1.8787e-04, 7.0333e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.3590e-04, 7.0333e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8787e-04, 1.8418e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05,
          7.0333e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8787e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05,
          6.5804e-04, 7.0333e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8787e-04,
          1.8787e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          7.0333e-05, 1.8418e-05, 1.8418e-05, 1.8787e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05, 1.3590e-04,
          7.0333e-05, 5.6362e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05, 6.0606e-04, 1.8418e-05,
          1.8418e-05, 7.0333e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05, 1.8418e-05,
          1.8418e-05, 1.3590e-04, 1.8418e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05,
          7.0333e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05, 1.3590e-04,
          7.0333e-05, 7.0333e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 1.8418e-05,
          1.8787e-04, 1.8418e-05, 7.0333e-05, 1.3590e-04, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05, 1.8418e-05,
          7.0333e-05, 7.0333e-05, 5.6362e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          7.0333e-05, 1.8418e-05, 7.0333e-05, 7.0333e-05, 1.8787e-04, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.3590e-04, 5.1165e-04, 5.1165e-04, 2.4700e-04, 1.3590e-04, 1.8418e-05,
          1.8418e-05, 7.0333e-05, 1.3590e-04, 5.5885e-04, 1.8418e-05, 1.8418e-05,
          1.3590e-04, 1.8418e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9319e-03, 6.1890e-02, 2.2827e-02, 2.4063e-02, 5.8594e-02,
          5.3284e-02, 1.5854e-02, 2.7103e-03, 2.3174e-03, 7.1983e-03, 8.4839e-03,
          3.6678e-03, 6.6452e-03, 1.8740e-03, 1.4435e-02, 3.4809e-03, 9.6436e-03,
          5.3940e-03, 2.1210e-03, 1.0214e-03, 1.5936e-03, 2.2316e-03, 6.0081e-03,
          1.1772e-02, 1.1513e-02, 1.3458e-02, 1.0216e-02, 1.2884e-03, 9.0179e-03,
          2.4376e-03, 5.3635e-03, 2.4986e-03, 1.6113e-02, 6.3515e-03, 4.6196e-03,
          1.0422e-02, 1.6012e-03, 9.0256e-03, 9.0637e-03, 1.2922e-03, 1.2573e-02,
          3.5400e-03, 3.2997e-03, 2.6970e-03, 3.3760e-03, 1.0750e-02, 2.6520e-02,
          1.9550e-03, 1.8139e-03, 7.0381e-03, 4.2725e-03, 3.1738e-03, 3.3607e-03,
          1.0529e-03, 1.8177e-03, 1.1740e-03, 2.7809e-03, 2.8610e-03, 2.7714e-03,
          2.3632e-03, 4.3526e-03, 4.1351e-03, 7.4806e-03, 5.4893e-03, 5.1300e-02,
          3.1586e-03, 3.0308e-03, 5.3482e-03, 3.0518e-03, 2.1667e-03, 2.5120e-03,
          2.3556e-03, 1.5068e-02, 1.4257e-03, 3.4904e-03, 9.5320e-04, 9.5558e-04,
          1.7204e-03, 2.1324e-03, 1.9093e-03, 7.8659e-03, 1.8082e-03, 2.2011e-03,
          1.8272e-03, 1.1734e-02, 2.5444e-03, 6.8588e-03, 1.8663e-03, 1.9569e-03,
          1.3206e-02, 1.3626e-02, 2.9888e-03, 9.1705e-03, 4.6158e-03, 3.3493e-03,
          1.8358e-03, 8.9788e-04, 4.7531e-03, 8.8425e-03, 1.7300e-03, 2.7714e-03,
          8.8959e-03, 1.5221e-03, 1.7252e-03, 5.4359e-03, 2.1652e-02, 4.6349e-03,
          1.0887e-02, 4.6539e-03, 1.6296e-02, 3.0060e-03, 2.6875e-03, 3.5419e-03,
          9.1095e-03, 1.0490e-02, 4.8065e-03, 8.5632e-02, 1.6272e-05, 1.6272e-05,
          5.8830e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05, 1.6272e-05,
          1.4377e-04, 1.6272e-05, 1.6272e-05, 1.0121e-04, 1.0121e-04, 1.6272e-05,
          1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.0121e-04, 1.6272e-05, 1.4377e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 3.4332e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.4377e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.4377e-04, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.4377e-04, 1.6272e-05, 1.6272e-05,
          1.4377e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.9307e-05, 1.6272e-05,
          5.8830e-05, 1.6272e-05, 1.6272e-05, 3.4332e-04, 1.4377e-04, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.0121e-04, 1.4377e-04, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05, 1.4377e-04, 3.4332e-04,
          5.8830e-05, 1.6272e-05, 1.6272e-05, 1.4424e-04, 1.6272e-05, 1.6749e-05,
          1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          2.9898e-04, 5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05,
          1.6272e-05, 1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05, 2.0087e-04,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05, 5.8830e-05,
          1.6272e-05, 1.0121e-04, 1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 4.9162e-04,
          1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6749e-05,
          1.6272e-05, 1.6272e-05, 5.8830e-05, 5.8830e-05, 1.4377e-04, 1.6272e-05,
          1.4377e-04, 5.9545e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 2.0707e-04, 1.4377e-04, 1.6272e-05, 5.9307e-05,
          5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          2.0087e-04, 1.6272e-05, 1.0121e-04, 3.4332e-04, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 5.8830e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 3.4332e-04, 1.0121e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.0121e-04, 2.4962e-04, 3.4332e-04, 1.6272e-05, 1.6272e-05, 3.4380e-04,
          1.6272e-05, 2.0707e-04, 3.4380e-04, 5.8830e-05, 1.6272e-05, 2.0707e-04,
          2.4962e-04, 1.6272e-05, 2.4962e-04, 2.0707e-04, 4.9162e-04, 4.9829e-04,
          1.6272e-05, 1.6272e-05, 1.4377e-04, 1.4377e-04, 4.5156e-04, 2.0707e-04,
          2.4962e-04, 5.3406e-04, 1.6272e-05, 1.6272e-05, 2.8586e-04, 2.0087e-04,
          2.6083e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 3.4380e-04, 1.6272e-05,
          6.3181e-04, 4.4084e-04, 2.0707e-04, 5.8830e-05, 5.8830e-05, 1.6272e-05,
          1.6272e-05, 6.8903e-04, 5.8830e-05, 1.4377e-04, 1.6272e-05, 1.6272e-05,
          4.9162e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05, 3.5596e-04, 4.0650e-04,
          4.7660e-04, 7.8440e-04, 7.8440e-04, 1.6272e-05, 2.0707e-04, 9.3269e-04,
          1.4377e-04, 4.0650e-04, 1.1301e-03, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          9.3269e-04, 2.9182e-04, 1.6272e-05, 2.0707e-04, 2.0707e-04, 1.6272e-05,
          1.0121e-04, 1.6272e-05, 4.4942e-04, 1.0121e-04, 4.4942e-04, 5.8830e-05,
          1.6272e-05, 1.4377e-04, 1.0121e-04, 4.8971e-04, 4.8971e-04, 9.2268e-04,
          1.0121e-04, 1.0121e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.9575e-03, 5.2765e-02, 3.3478e-02, 2.5589e-02, 9.5276e-02,
          3.3752e-02, 2.9938e-02, 1.2894e-03, 8.5402e-04, 2.2030e-03, 4.3182e-03,
          1.4639e-03, 1.4191e-03, 2.0657e-03, 1.8692e-03, 1.3475e-03, 2.6932e-03,
          1.7691e-03, 9.3317e-04, 2.5330e-03, 2.7752e-03, 4.6120e-03, 2.2030e-03,
          5.4703e-03, 4.6577e-03, 2.1790e-02, 4.0932e-03, 2.6321e-03, 1.8606e-03,
          2.0332e-03, 1.7767e-03, 1.1759e-03, 2.2850e-03, 1.0815e-03, 1.2941e-03,
          3.4332e-03, 3.6411e-03, 2.5406e-03, 1.6375e-03, 2.3022e-03, 8.8654e-03,
          1.3779e-02, 7.6981e-03, 1.1345e-02, 6.2599e-03, 2.4662e-03, 5.1079e-03,
          3.4466e-03, 7.7362e-03, 1.6678e-02, 2.2068e-03, 1.5240e-03, 1.1349e-03,
          2.1172e-03, 1.3733e-03, 3.7384e-03, 3.5686e-03, 4.6806e-03, 4.2381e-03,
          5.3520e-03, 3.6697e-03, 4.7226e-03, 5.1003e-03, 5.9624e-03, 8.9417e-03,
          4.4250e-03, 2.1439e-03, 2.4319e-03, 1.4740e-02, 1.8097e-02, 1.5732e-02,
          1.9974e-02, 3.9177e-03, 3.9482e-03, 1.8835e-03, 1.7776e-03, 1.1187e-03,
          2.2888e-03, 3.3302e-03, 2.4967e-03, 2.3098e-03, 2.1877e-03, 4.4060e-03,
          7.1564e-03, 5.9280e-03, 3.9062e-03, 2.6321e-03, 1.8024e-03, 3.3779e-03,
          5.2032e-03, 6.4697e-03, 7.4921e-03, 5.1689e-03, 4.6143e-02, 4.5746e-02,
          3.8795e-03, 2.8954e-03, 1.7376e-03, 4.2953e-03, 4.0321e-03, 2.6550e-03,
          2.9278e-03, 1.3208e-03, 1.0939e-03, 3.6983e-03, 3.6125e-03, 4.3182e-03,
          5.7983e-03, 1.5625e-02, 6.9122e-03, 5.5885e-03, 3.8452e-03, 3.0994e-03,
          4.3297e-03, 2.3148e-02, 1.4374e-02, 9.4910e-03, 9.0637e-02, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04,
          2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 1.0169e-04,
          1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 1.0169e-04, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 1.0169e-04, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.9802e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04,
          2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 1.0169e-04,
          2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 1.0169e-04,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05,
          3.7909e-04, 1.0169e-04, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.0169e-04, 2.0504e-05,
          2.0504e-05, 1.0169e-04, 2.9802e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          1.0169e-04, 2.0504e-05, 1.0169e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.9802e-04, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.9802e-04, 1.3752e-03, 7.3957e-04, 1.2732e-04, 3.7193e-04, 2.0504e-05,
          2.0504e-05, 2.9802e-04, 2.9802e-04, 4.5800e-04, 9.4414e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.1276e-02, 5.7465e-02, 3.1738e-02, 3.0273e-02, 1.0516e-01,
          5.1422e-02, 2.7832e-02, 2.6684e-03, 2.2678e-03, 2.3785e-03, 3.1700e-03,
          1.5688e-03, 5.8022e-03, 5.2223e-03, 2.8381e-03, 1.9197e-03, 1.1757e-02,
          5.9280e-03, 1.1692e-03, 4.0894e-03, 4.1389e-03, 3.6087e-03, 4.5776e-03,
          3.2234e-03, 3.0346e-03, 2.0020e-02, 9.4223e-03, 1.3329e-02, 9.8724e-03,
          5.8403e-03, 3.1319e-03, 6.3362e-03, 4.8866e-03, 1.6441e-03, 3.0994e-03,
          4.4060e-03, 2.7409e-03, 3.2139e-03, 2.6932e-03, 4.7989e-03, 2.8877e-03,
          2.5578e-03, 1.8988e-03, 2.0180e-03, 1.4687e-03, 6.0539e-03, 4.7302e-03,
          2.8000e-03, 4.3449e-03, 1.6678e-02, 7.1297e-03, 3.2578e-03, 2.3766e-03,
          3.5210e-03, 2.3041e-03, 4.0588e-03, 3.9177e-03, 2.8152e-03, 2.9049e-03,
          2.9068e-03, 1.5640e-02, 4.2000e-03, 4.7531e-03, 1.8845e-03, 4.3449e-03,
          2.2621e-03, 9.4604e-04, 1.2407e-03, 1.9531e-03, 1.8358e-03, 1.9722e-03,
          2.6245e-03, 4.1313e-03, 6.7062e-03, 3.7823e-03, 2.2469e-03, 1.6241e-03,
          2.7866e-03, 5.5885e-03, 2.5826e-03, 2.3670e-03, 1.7376e-03, 2.1267e-03,
          3.0441e-03, 4.8828e-03, 3.3245e-03, 1.5106e-03, 1.0309e-03, 1.3151e-03,
          2.2202e-03, 1.9121e-03, 1.3237e-03, 1.7576e-03, 2.2831e-03, 2.9926e-03,
          1.9760e-02, 3.9024e-03, 2.7409e-03, 1.0513e-02, 6.5575e-03, 3.8013e-03,
          2.9373e-03, 1.4067e-03, 1.2321e-03, 3.3627e-03, 3.5267e-03, 3.2410e-02,
          4.0855e-03, 3.5629e-03, 2.8687e-03, 2.1381e-03, 1.9426e-03, 1.5030e-03,
          1.6441e-03, 4.2076e-03, 1.4809e-02, 4.9629e-03, 3.9444e-03, 1.6370e-01,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 8.9645e-05, 5.6386e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 5.6386e-05, 2.3425e-05,
          5.6386e-05, 5.6386e-05, 2.3425e-05, 5.6386e-05, 5.6386e-05, 1.6081e-04,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 8.9645e-05,
          5.6386e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          5.6386e-05, 2.3425e-05, 2.1327e-04, 1.6081e-04, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 5.6386e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          1.2255e-04, 5.6386e-05, 8.9645e-05, 5.6386e-05, 5.6386e-05, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 8.9645e-05, 1.2255e-04, 2.3425e-05, 2.3425e-05,
          1.2255e-04, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 8.9645e-05,
          1.4687e-04, 2.3425e-05, 2.3425e-05, 5.6386e-05, 2.1315e-04, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 5.6386e-05, 5.6386e-05, 2.3425e-05,
          5.6386e-05, 1.9383e-04, 1.6081e-04, 2.3425e-05, 1.2255e-04, 5.6386e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 1.2255e-04, 2.3425e-05, 5.6386e-05,
          5.6386e-05, 2.3425e-05, 2.3425e-05, 5.6386e-05, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 1.6081e-04, 6.5708e-04, 5.6386e-05, 1.2255e-04, 8.9645e-05,
          5.6386e-05, 2.3425e-05, 6.5708e-04, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 5.6386e-05, 8.9645e-05, 2.3425e-05, 2.3425e-05, 8.9645e-05,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 5.6386e-05,
          1.6081e-04, 5.6386e-05, 2.3425e-05, 1.2255e-04, 5.6386e-05, 5.6386e-05,
          1.9383e-04, 1.6081e-04, 2.3425e-05, 2.3425e-05, 2.3425e-05, 5.6386e-05,
          2.3425e-05, 1.2255e-04, 2.3425e-05, 5.6386e-05, 2.3425e-05, 2.3425e-05,
          2.3425e-05, 2.3425e-05, 5.6386e-05, 2.3425e-05, 1.2255e-04, 1.6081e-04,
          5.6386e-05, 2.3425e-05, 1.2255e-04, 6.5708e-04, 5.6386e-05, 2.3425e-05,
          5.1975e-04, 5.6386e-05, 2.3425e-05, 1.2255e-04, 2.3425e-05, 1.2255e-04,
          8.9645e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05,
          8.9645e-05, 2.3425e-05, 5.6386e-05, 2.3746e-04, 2.3425e-05, 2.3425e-05,
          1.6081e-04, 1.9383e-04, 1.6081e-04, 2.3425e-05, 6.8998e-04, 1.6081e-04,
          1.6081e-04, 2.3425e-05, 2.1315e-04, 2.3425e-05, 2.3425e-05, 1.2255e-04,
          2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 1.2255e-04,
          5.6386e-05, 5.6386e-05, 2.3425e-05, 8.9645e-05, 2.3425e-05, 5.6386e-05,
          7.8726e-04, 3.0375e-04, 5.6386e-05, 2.3425e-05, 1.6081e-04, 2.3425e-05,
          1.4710e-04, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 1.2255e-04,
          5.6386e-05, 5.6386e-05, 1.2255e-04, 8.9645e-05, 1.4710e-04, 5.6386e-05,
          1.2255e-04, 2.3425e-05, 1.2255e-04, 2.3425e-05, 2.5425e-03, 1.2255e-04,
          1.4710e-04, 8.9645e-05, 5.6386e-05, 4.4107e-04, 2.3425e-05, 5.6386e-05,
          1.9383e-04, 5.6386e-05, 2.3425e-05, 2.3425e-05, 2.1315e-04, 1.2255e-04,
          1.2255e-04, 4.7231e-04, 3.0375e-04, 4.7231e-04, 2.1315e-04, 2.3746e-04,
          3.0375e-04, 2.1315e-04, 1.4710e-04, 3.0375e-04, 2.3746e-04, 2.3425e-05,
          4.4107e-04, 5.6386e-05, 8.9645e-05, 4.7231e-04, 2.3425e-05, 1.1414e-04,
          5.6386e-05, 2.3425e-05, 2.3425e-05, 2.3425e-05, 2.1315e-04, 2.3425e-05,
          1.4687e-04, 2.1315e-04, 3.0375e-04, 5.6386e-05, 5.6386e-05, 4.7231e-04,
          5.3835e-04, 1.9383e-04, 5.3835e-04, 2.0456e-04, 3.7479e-04, 3.7479e-04,
          4.7231e-04, 2.3746e-04, 2.8443e-04, 2.8443e-04, 2.5153e-04, 2.3425e-05,
          5.6386e-05, 5.6386e-05, 1.4687e-04, 5.6386e-05, 5.6386e-05, 5.6386e-05,
          2.3425e-05, 2.3425e-05, 5.6386e-05, 5.6386e-05, 5.6386e-05, 5.6386e-05,
          5.6386e-05, 2.3425e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.0841e-02, 6.0150e-02, 2.6840e-02, 2.7283e-02, 7.9895e-02,
          4.8950e-02, 2.0065e-02, 3.0041e-03, 2.4948e-03, 3.5763e-03, 4.3335e-03,
          2.0790e-03, 7.9803e-03, 1.2451e-02, 3.5686e-03, 2.1381e-03, 1.8906e-02,
          9.9869e-03, 1.1978e-03, 3.5419e-03, 2.8896e-03, 3.1242e-03, 6.0616e-03,
          4.1504e-03, 2.9030e-03, 1.5472e-02, 9.0256e-03, 9.0942e-03, 1.3779e-02,
          1.6068e-02, 3.9940e-03, 8.0795e-03, 6.1951e-03, 2.5005e-03, 3.6068e-03,
          4.7035e-03, 2.2392e-03, 3.7327e-03, 3.3207e-03, 5.7259e-03, 3.1910e-03,
          2.1038e-03, 1.5707e-03, 1.6584e-03, 1.2722e-03, 8.5068e-03, 5.4359e-03,
          5.1498e-03, 5.1117e-03, 1.1047e-02, 7.0038e-03, 4.0436e-03, 2.5501e-03,
          3.0117e-03, 2.7275e-03, 3.6144e-03, 3.5419e-03, 3.0727e-03, 2.9621e-03,
          2.9125e-03, 1.8494e-02, 3.5419e-03, 4.2381e-03, 2.0752e-03, 5.5923e-03,
          2.6817e-03, 1.1139e-03, 1.6413e-03, 1.3542e-03, 1.3332e-03, 1.7128e-03,
          2.1019e-03, 5.8708e-03, 5.3368e-03, 5.0316e-03, 2.1648e-03, 1.7042e-03,
          3.4046e-03, 4.7874e-03, 3.1586e-03, 3.3379e-03, 2.4643e-03, 2.0142e-03,
          2.4128e-03, 5.2643e-03, 3.5686e-03, 2.0256e-03, 1.3294e-03, 1.8368e-03,
          3.0289e-03, 2.5158e-03, 1.5116e-03, 2.1667e-03, 1.7376e-03, 2.1191e-03,
          1.6525e-02, 3.0823e-03, 3.6297e-03, 1.0918e-02, 5.7945e-03, 4.3831e-03,
          4.1656e-03, 1.6232e-03, 1.5926e-03, 3.4561e-03, 4.5433e-03, 2.8152e-02,
          4.4327e-03, 2.9850e-03, 3.8147e-03, 1.8368e-03, 2.1896e-03, 1.8234e-03,
          1.7843e-03, 2.9602e-03, 1.2077e-02, 5.6725e-03, 2.9869e-03, 5.9937e-02,
          1.1139e-01, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 1.6630e-05, 1.6630e-05,
          9.1493e-05, 9.1493e-05, 4.9889e-05, 5.8174e-05, 9.1493e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05,
          1.6630e-05, 4.9889e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          5.8174e-05, 1.6630e-05, 2.9898e-04, 1.6630e-05, 1.6761e-04, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 5.8174e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          9.1493e-05, 5.8174e-05, 4.9889e-05, 4.9889e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 4.9889e-05, 4.9889e-05, 9.1493e-05, 1.6630e-05, 1.6630e-05,
          9.1493e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05,
          9.1493e-05, 1.6630e-05, 4.9889e-05, 5.8174e-05, 1.9670e-04, 4.9889e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6332e-04, 5.8174e-05, 4.9889e-05,
          5.8174e-05, 5.8174e-05, 2.7585e-04, 4.9889e-05, 9.1493e-05, 5.8174e-05,
          4.9889e-05, 1.6630e-05, 1.6630e-05, 9.1493e-05, 4.9889e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05, 4.9889e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 2.0051e-04, 7.8249e-04, 2.0921e-04, 4.9889e-05,
          5.8174e-05, 1.6761e-04, 7.4577e-04, 1.6761e-04, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 5.8174e-05, 4.9889e-05, 4.9889e-05, 4.9889e-05, 4.9889e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          2.0051e-04, 5.8174e-05, 1.6630e-05, 1.6761e-04, 5.8174e-05, 2.0921e-04,
          1.7595e-04, 2.7585e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05, 5.8174e-05,
          1.6630e-05, 9.1493e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 5.8174e-05,
          1.0595e-03, 9.1493e-05, 2.8443e-04, 7.8678e-04, 5.8174e-05, 1.6761e-04,
          2.0957e-04, 9.1493e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 6.8045e-04,
          9.1493e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          4.9889e-05, 1.6630e-05, 5.8174e-05, 2.9898e-04, 1.6630e-05, 1.3435e-04,
          2.0051e-04, 2.4211e-04, 2.0051e-04, 4.9889e-05, 7.8678e-04, 2.0051e-04,
          2.0051e-04, 1.6630e-05, 1.9670e-04, 1.6630e-05, 1.6630e-05, 9.1493e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05,
          1.6630e-05, 4.9889e-05, 1.6630e-05, 4.9889e-05, 5.8174e-05, 2.0921e-04,
          8.9836e-04, 2.9898e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          5.8174e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.9889e-05, 1.6630e-05,
          4.9889e-05, 1.6630e-05, 4.9889e-05, 1.6630e-05, 1.0386e-03, 1.6630e-05,
          4.9889e-05, 1.6761e-04, 5.8174e-05, 1.5223e-04, 4.9889e-05, 1.6630e-05,
          1.6630e-05, 5.8174e-05, 1.6630e-05, 1.6630e-05, 9.1493e-05, 9.1493e-05,
          4.9889e-05, 1.6630e-05, 1.9383e-04, 1.9383e-04, 1.6630e-05, 4.9889e-05,
          1.9670e-04, 1.9670e-04, 1.1891e-04, 1.6630e-05, 2.9898e-04, 4.9889e-05,
          1.9383e-04, 1.6630e-05, 4.9889e-05, 1.9383e-04, 1.6630e-05, 1.6630e-05,
          5.8174e-05, 1.6630e-05, 1.6630e-05, 4.9889e-05, 9.1493e-05, 1.6630e-05,
          6.7997e-04, 4.9889e-05, 2.9898e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          8.8787e-04, 1.6630e-05, 1.6630e-05, 4.9889e-05, 4.8280e-04, 4.4966e-04,
          1.6630e-05, 2.6560e-04, 1.6630e-05, 5.8174e-05, 1.6630e-05, 1.6630e-05,
          4.2510e-04, 1.6332e-04, 1.6332e-04, 1.6332e-04, 1.6332e-04, 5.8174e-05,
          1.6630e-05, 9.1493e-05, 1.6332e-04, 2.9135e-04, 1.6332e-04, 5.8174e-05,
          1.6332e-04, 5.8174e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.5602e-03, 6.2378e-02, 1.8127e-02, 2.8000e-02, 5.7312e-02,
          5.4657e-02, 1.5625e-02, 2.6951e-03, 3.2959e-03, 1.3969e-02, 1.2009e-02,
          6.0959e-03, 3.6278e-03, 1.4486e-03, 6.5956e-03, 2.0790e-03, 5.4169e-03,
          2.5692e-03, 1.5955e-03, 7.8583e-04, 1.1711e-03, 7.3004e-04, 3.1509e-03,
          1.1414e-02, 6.8245e-03, 1.2764e-02, 8.8043e-03, 1.8005e-03, 5.9814e-03,
          2.1114e-03, 4.3259e-03, 2.1076e-03, 1.2924e-02, 7.1793e-03, 3.9291e-03,
          1.3260e-02, 3.5000e-03, 1.1398e-02, 7.8278e-03, 1.0405e-03, 1.3779e-02,
          3.2539e-03, 2.9564e-03, 1.8072e-03, 3.5191e-03, 4.7035e-03, 1.2466e-02,
          8.3637e-04, 8.8930e-04, 6.3362e-03, 6.2332e-03, 4.1122e-03, 3.8853e-03,
          1.6718e-03, 2.3460e-03, 1.8272e-03, 8.4534e-03, 3.6964e-03, 7.3586e-03,
          4.0932e-03, 2.7523e-03, 2.4014e-03, 4.2610e-03, 5.3864e-03, 1.9577e-02,
          3.0384e-03, 2.2297e-03, 4.4060e-03, 1.6632e-03, 1.2636e-03, 1.2093e-03,
          1.2360e-03, 2.0981e-02, 2.5730e-03, 4.6234e-03, 1.1568e-03, 1.3313e-03,
          2.3518e-03, 1.0765e-02, 6.5422e-03, 1.8967e-02, 4.9973e-03, 3.3875e-03,
          2.4014e-03, 5.5428e-03, 1.7681e-03, 5.6267e-03, 2.0485e-03, 1.8940e-03,
          9.0485e-03, 1.1238e-02, 1.4896e-03, 5.7678e-03, 1.6804e-03, 1.4639e-03,
          2.0847e-03, 1.3685e-03, 4.4861e-03, 1.9455e-02, 1.4114e-02, 1.9089e-02,
          1.7838e-02, 2.7161e-03, 3.7842e-03, 1.7960e-02, 1.5869e-02, 4.1389e-03,
          4.6234e-03, 4.9629e-03, 1.3405e-02, 1.7986e-03, 2.2945e-03, 3.4370e-03,
          4.7417e-03, 6.5918e-03, 1.3170e-03, 1.6281e-02, 1.7490e-03, 6.1836e-03,
          8.7128e-03, 7.1106e-02, 1.6093e-05, 7.4506e-05, 7.4506e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          7.4506e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.0276e-04, 6.1226e-04, 2.5058e-04, 1.6093e-05, 1.6093e-05, 7.4506e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 6.5148e-05, 1.6093e-05, 6.1226e-04,
          1.6093e-05, 1.6093e-05, 1.6117e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6117e-04, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 4.1819e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05,
          7.4506e-05, 1.6093e-05, 1.6093e-05, 7.3481e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05, 9.6989e-04,
          7.4506e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6117e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 4.7660e-04,
          1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 3.2854e-04,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0276e-04,
          1.6093e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 7.4506e-05, 7.4506e-05, 1.6117e-04, 1.6093e-05,
          7.4506e-05, 7.4506e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 4.6730e-04, 3.7766e-04, 1.6093e-05, 7.4506e-05,
          7.4506e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          7.4506e-05, 1.6093e-05, 1.6117e-04, 1.6117e-04, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 7.4506e-05, 7.4506e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05, 7.4506e-05,
          3.9577e-04, 1.6093e-05, 1.8346e-04, 7.4506e-05, 1.6093e-05, 1.6093e-05,
          1.6093e-05, 1.6093e-05, 5.0497e-04, 5.6314e-04, 1.6093e-05, 1.6093e-05,
          5.6314e-04, 4.7660e-04, 1.0276e-04, 1.6093e-05, 1.6093e-05, 1.6117e-04,
          1.6093e-05, 7.4685e-05, 1.6117e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05,
          7.4506e-05, 1.6093e-05, 3.0899e-04, 2.7013e-04, 1.6093e-05, 3.7766e-04,
          1.6093e-05, 1.6093e-05, 3.2854e-04, 1.6093e-05, 3.0947e-04, 5.0497e-04,
          1.6975e-03, 1.6117e-04, 1.6093e-05, 1.6093e-05, 1.6117e-04, 7.4506e-05,
          3.0947e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.6117e-04, 1.6093e-05,
          1.6117e-04, 1.6117e-04, 1.6093e-05, 7.4506e-05, 7.4506e-05, 1.6093e-05,
          7.4506e-05, 6.1226e-04, 7.4506e-05, 1.0276e-04, 1.6093e-05, 1.6093e-05,
          1.6117e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 7.4506e-05, 1.6093e-05,
          1.6117e-04, 1.6117e-04, 7.4506e-05, 1.6093e-05, 1.6093e-05, 5.0497e-04,
          7.4506e-05, 1.8346e-04, 1.8826e-03, 1.6093e-05, 1.6093e-05, 2.7013e-04,
          1.6479e-03, 4.2987e-04, 1.6093e-05, 2.7013e-04, 1.8346e-04, 1.6093e-05,
          1.0276e-04, 1.6093e-05, 1.6093e-05, 1.6093e-05, 1.0276e-04, 1.6093e-05,
          1.6093e-05, 1.0276e-04, 1.0276e-04, 3.8981e-04, 1.0276e-04, 1.1053e-03,
          5.3167e-04, 1.6093e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.9792e-03, 5.6793e-02, 2.0767e-02, 2.1271e-02, 8.0933e-02,
          4.2816e-02, 1.9363e-02, 2.1381e-03, 2.9125e-03, 6.0654e-03, 1.0742e-02,
          4.4250e-03, 1.9779e-03, 2.3060e-03, 2.4662e-03, 1.3151e-03, 2.3975e-03,
          1.3523e-03, 1.0815e-03, 1.1740e-03, 1.2188e-03, 9.2793e-04, 1.7395e-03,
          9.8648e-03, 3.9520e-03, 1.3222e-02, 4.0779e-03, 3.2558e-03, 2.4624e-03,
          3.1452e-03, 2.8572e-03, 1.3723e-03, 3.5019e-03, 2.1725e-03, 1.7595e-03,
          6.6299e-03, 6.0806e-03, 2.8553e-03, 2.0771e-03, 1.6012e-03, 4.8561e-03,
          6.3782e-03, 4.2038e-03, 4.3106e-03, 2.2831e-03, 1.8511e-03, 3.2024e-03,
          1.2131e-03, 1.3580e-03, 8.5907e-03, 4.7264e-03, 3.3436e-03, 2.3632e-03,
          3.1605e-03, 2.1725e-03, 2.7504e-03, 1.4984e-02, 5.3940e-03, 1.3489e-02,
          7.0724e-03, 2.7962e-03, 2.7905e-03, 2.1744e-03, 9.1629e-03, 6.4240e-03,
          8.6823e-03, 3.8166e-03, 2.0123e-03, 3.1662e-03, 3.2406e-03, 3.7823e-03,
          2.3918e-03, 7.8583e-03, 5.1727e-03, 3.8052e-03, 1.9665e-03, 1.4935e-03,
          2.7122e-03, 2.9510e-02, 3.1052e-02, 8.4839e-03, 1.2131e-02, 4.8790e-03,
          5.1918e-03, 3.4599e-03, 2.2678e-03, 4.2381e-03, 4.4289e-03, 6.5460e-03,
          6.9160e-03, 5.0964e-03, 4.1695e-03, 4.9629e-03, 3.0117e-03, 2.3479e-03,
          4.2114e-03, 2.8934e-03, 1.7014e-03, 7.1907e-03, 6.3049e-02, 2.2324e-02,
          1.5060e-02, 6.5269e-03, 6.2561e-03, 1.0262e-02, 5.8708e-03, 3.7746e-03,
          3.1033e-03, 6.1951e-03, 7.2823e-03, 5.2452e-03, 8.7433e-03, 4.6272e-03,
          3.5610e-03, 4.3297e-03, 1.7738e-03, 5.6229e-03, 3.1281e-03, 1.2169e-02,
          1.1147e-02, 1.9348e-02, 5.9296e-02, 7.1466e-05, 7.1466e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.5044e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.5891e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05,
          1.8120e-05, 1.8120e-05, 1.5044e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05,
          7.1466e-05, 1.8120e-05, 1.8120e-05, 1.5044e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05, 1.5044e-04,
          7.1466e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 7.1466e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 1.5044e-04, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 3.5119e-04, 1.5044e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.5044e-04, 1.5044e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 9.7036e-05, 7.1466e-05, 7.1466e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.5044e-04, 9.7036e-05, 1.8120e-05, 7.1466e-05, 1.8120e-05, 9.7036e-05,
          1.8120e-05, 1.8120e-05, 9.7036e-05, 1.5044e-04, 1.8120e-05, 1.8120e-05,
          1.5044e-04, 7.1466e-05, 1.5044e-04, 1.8120e-05, 1.8120e-05, 1.5044e-04,
          1.8120e-05, 7.1466e-05, 1.5044e-04, 7.1466e-05, 1.8120e-05, 1.8120e-05,
          9.7036e-05, 1.8120e-05, 1.2274e-03, 4.3011e-04, 1.8120e-05, 1.5044e-04,
          1.8120e-05, 1.8120e-05, 2.7561e-04, 1.8120e-05, 7.1466e-05, 1.5860e-03,
          1.8120e-05, 7.1466e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 7.1466e-05,
          7.1466e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.5044e-04, 9.7036e-05,
          1.5044e-04, 1.5044e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          7.1466e-05, 9.7036e-05, 7.1466e-05, 9.7036e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 7.1466e-05, 7.1466e-05, 1.8120e-05,
          1.5044e-04, 1.5044e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 9.7036e-05,
          1.5044e-04, 4.3011e-04, 1.5044e-04, 1.8120e-05, 1.8120e-05, 3.4523e-04,
          9.7036e-05, 2.1191e-03, 1.8120e-05, 4.3011e-04, 9.7036e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 9.7036e-05, 1.8120e-05,
          1.8120e-05, 9.7036e-05, 9.7036e-05, 2.9802e-04, 9.7036e-05, 9.7036e-05,
          1.8120e-05, 1.8120e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0545e-03, 6.7932e-02, 1.5175e-02, 1.6327e-02, 4.9927e-02,
          5.0140e-02, 1.0414e-02, 3.6865e-02, 8.7814e-03, 4.7760e-03, 8.1329e-03,
          3.0327e-03, 5.7945e-03, 2.1019e-03, 5.0049e-03, 2.6276e-02, 6.1531e-03,
          4.2305e-03, 8.4991e-03, 5.0688e-04, 4.9734e-04, 5.4073e-04, 4.0779e-03,
          8.6594e-03, 3.0079e-03, 9.3765e-03, 7.6141e-03, 1.0757e-03, 5.8594e-03,
          3.0117e-03, 5.1300e-02, 1.7185e-03, 4.1885e-03, 3.9749e-03, 8.7128e-03,
          3.2425e-03, 1.4334e-03, 1.9741e-03, 3.5877e-03, 8.2159e-04, 1.2150e-03,
          7.5006e-04, 6.3610e-04, 7.1955e-04, 5.0402e-04, 6.4125e-03, 2.4967e-03,
          7.8106e-04, 6.0081e-04, 5.4474e-03, 1.4465e-02, 3.2043e-02, 9.6664e-03,
          6.9046e-03, 1.7181e-02, 4.8218e-03, 1.7014e-03, 9.7198e-03, 2.5063e-03,
          2.9030e-03, 2.4261e-03, 5.8632e-03, 4.1695e-03, 1.1425e-03, 4.0779e-03,
          2.3994e-03, 1.4915e-03, 1.4372e-03, 3.9053e-04, 3.8362e-04, 6.7997e-04,
          6.2513e-04, 8.4915e-03, 6.8893e-03, 4.0344e-02, 8.3618e-03, 9.3689e-03,
          1.3840e-02, 1.7042e-03, 2.2984e-03, 4.4250e-03, 2.8572e-03, 1.8978e-03,
          1.3819e-03, 3.1708e-02, 3.8280e-03, 4.6539e-03, 1.2980e-03, 1.3371e-03,
          7.1259e-03, 2.5635e-03, 7.9536e-04, 3.5172e-03, 7.8249e-04, 5.8889e-04,
          2.3918e-03, 3.0499e-03, 1.6830e-02, 4.1924e-03, 1.6899e-03, 2.3136e-03,
          1.0376e-02, 3.7422e-03, 2.9869e-03, 3.2673e-03, 8.4152e-03, 2.3727e-03,
          2.2781e-02, 1.0281e-03, 3.2215e-03, 1.4458e-03, 2.1915e-03, 1.7338e-03,
          2.7122e-03, 1.0996e-03, 1.1396e-03, 5.0201e-03, 7.4720e-04, 3.4142e-03,
          4.5280e-03, 5.1842e-03, 1.4658e-03, 8.9417e-02, 6.1631e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 6.1631e-05, 1.7822e-05, 1.7822e-05,
          1.1986e-04, 6.1631e-05, 1.7822e-05, 7.6115e-05, 1.7822e-05, 2.3592e-04,
          7.6115e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.1986e-04,
          1.7822e-05, 1.7822e-05, 6.1631e-05, 1.7822e-05, 6.1631e-05, 6.1631e-05,
          1.7822e-05, 6.1631e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 2.3592e-04,
          7.6115e-05, 2.3592e-04, 1.1986e-04, 2.7966e-04, 6.1631e-05, 6.1631e-05,
          1.7822e-05, 1.7822e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.1986e-04, 7.6115e-05, 6.1631e-05, 6.1631e-05, 1.1986e-04, 6.1631e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 3.7527e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 6.1631e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.1986e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.1631e-05, 6.1631e-05,
          1.7822e-05, 1.7822e-05, 6.1631e-05, 7.6115e-05, 1.1986e-04, 6.1631e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6115e-05, 7.6115e-05, 1.7822e-05,
          1.7822e-05, 7.6115e-05, 2.3592e-04, 1.7822e-05, 1.1986e-04, 7.6115e-05,
          1.7822e-05, 1.5724e-04, 1.7822e-05, 1.1986e-04, 6.1631e-05, 6.1631e-05,
          1.7822e-05, 1.1986e-04, 1.7822e-05, 1.1986e-04, 1.7822e-05, 1.1986e-04,
          6.1631e-05, 3.7527e-04, 3.7527e-04, 7.6115e-05, 1.1986e-04, 1.7822e-05,
          7.6115e-05, 7.6115e-05, 2.3592e-04, 1.7822e-05, 1.7822e-05, 6.1631e-05,
          1.5724e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.1631e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 6.1631e-05, 1.1986e-04,
          3.7527e-04, 7.6115e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6115e-05,
          7.6115e-05, 2.3592e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 2.1553e-04,
          1.7822e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 1.7822e-05, 1.1986e-04, 1.1986e-04, 1.7822e-05,
          1.1986e-04, 6.1631e-05, 1.1986e-04, 1.7822e-05, 7.6115e-05, 6.1631e-05,
          1.7822e-05, 1.1986e-04, 2.3592e-04, 1.1986e-04, 1.5724e-04, 1.1986e-04,
          1.1986e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.1986e-04, 1.7822e-05,
          6.1631e-05, 1.7822e-05, 7.6115e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05,
          2.3592e-04, 1.7822e-05, 3.7527e-04, 1.1986e-04, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 1.7822e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          1.7822e-05, 2.1553e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 1.7822e-05,
          7.6115e-05, 7.6115e-05, 4.3368e-04, 6.1631e-05, 1.7822e-05, 7.6115e-05,
          7.6115e-05, 7.6115e-05, 7.6115e-05, 7.6115e-05, 1.7822e-05, 1.7822e-05,
          7.6115e-05, 7.6115e-05, 7.6115e-05, 1.5724e-04, 1.7822e-05, 1.1986e-04,
          1.7822e-05, 1.1986e-04, 1.1986e-04, 6.1631e-05, 1.7822e-05, 7.6115e-05,
          7.6115e-05, 1.7822e-05, 7.6115e-05, 4.3368e-04, 2.1553e-04, 1.1986e-04,
          1.7822e-05, 6.1631e-05, 7.6115e-05, 6.1631e-05, 1.7822e-05, 4.3368e-04,
          2.1553e-04, 2.5940e-04, 1.5724e-04, 1.7822e-05, 7.6115e-05, 1.7822e-05,
          1.7822e-05, 7.6115e-05, 6.1631e-05, 7.6115e-05, 1.1986e-04, 7.6115e-05,
          7.6115e-05, 7.6115e-05, 1.7822e-05, 6.1631e-05, 1.7822e-05, 1.7822e-05,
          1.1986e-04, 2.1553e-04, 1.7822e-05, 1.1986e-04, 1.7822e-05, 1.7822e-05,
          2.1553e-04, 1.7822e-05, 1.7822e-05, 1.7822e-05, 7.6115e-05, 1.5724e-04,
          7.6115e-05, 1.1986e-04, 1.1986e-04, 7.6115e-05, 2.1553e-04, 1.1339e-03,
          7.6115e-05, 4.3368e-04, 2.5940e-04, 1.7822e-05, 2.5940e-04, 4.7731e-04,
          2.5940e-04, 7.6115e-05, 7.6115e-05, 2.1553e-04, 2.1553e-04, 1.7822e-05,
          7.6115e-05, 7.6115e-05, 7.6115e-05, 7.6115e-05, 7.6115e-05, 1.1986e-04,
          1.7822e-05, 1.1986e-04, 7.6115e-05, 3.9625e-04, 7.6115e-05, 3.9625e-04,
          7.6115e-05, 7.6115e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.5804e-03, 6.2988e-02, 2.0554e-02, 2.3758e-02, 5.9967e-02,
          5.8868e-02, 1.5579e-02, 3.7575e-03, 2.8629e-03, 1.1032e-02, 1.2596e-02,
          5.3062e-03, 4.8065e-03, 1.6031e-03, 1.2321e-02, 3.6087e-03, 5.2338e-03,
          3.3588e-03, 2.0809e-03, 5.3024e-04, 7.2479e-04, 5.2929e-04, 5.2757e-03,
          1.3603e-02, 9.0408e-03, 1.2718e-02, 1.0849e-02, 1.1415e-03, 7.2250e-03,
          2.2945e-03, 7.8964e-03, 2.2430e-03, 1.5587e-02, 1.1536e-02, 4.6120e-03,
          1.1024e-02, 2.1019e-03, 7.5989e-03, 1.0201e-02, 8.0585e-04, 4.4327e-03,
          1.2484e-03, 1.0986e-03, 9.3222e-04, 1.0815e-03, 6.4583e-03, 1.1581e-02,
          7.8535e-04, 6.8140e-04, 5.8632e-03, 5.6839e-03, 5.5199e-03, 4.6806e-03,
          1.5774e-03, 2.9640e-03, 1.3857e-03, 3.5877e-03, 3.9749e-03, 3.8891e-03,
          2.9125e-03, 2.8076e-03, 2.3346e-03, 3.3569e-03, 1.9855e-03, 2.8702e-02,
          2.2888e-03, 1.8311e-03, 3.6812e-03, 7.0238e-04, 5.8985e-04, 7.2527e-04,
          7.6151e-04, 2.2568e-02, 2.0638e-03, 5.7335e-03, 1.1740e-03, 1.3924e-03,
          2.4281e-03, 3.3302e-03, 3.3302e-03, 1.6861e-02, 3.4046e-03, 2.7733e-03,
          2.0409e-03, 7.6218e-03, 1.5802e-03, 6.8207e-03, 1.5068e-03, 1.4105e-03,
          1.0963e-02, 8.2016e-03, 9.9182e-04, 6.2752e-03, 9.5224e-04, 9.2030e-04,
          1.9073e-03, 1.1778e-03, 5.9204e-03, 9.1095e-03, 3.4847e-03, 6.3515e-03,
          1.8204e-02, 2.5272e-03, 3.1986e-03, 1.1108e-02, 3.1464e-02, 3.3226e-03,
          6.3744e-03, 1.7939e-03, 1.1642e-02, 1.2112e-03, 2.1458e-03, 2.8458e-03,
          5.2109e-03, 2.2240e-03, 1.0452e-03, 2.4826e-02, 1.2407e-03, 3.1071e-03,
          5.2567e-03, 1.8051e-02, 1.9464e-03, 2.9392e-03, 8.4412e-02, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.6189e-04, 1.8716e-05,
          5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 2.0242e-04,
          1.8716e-05, 3.3188e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 5.9366e-05, 2.0242e-04, 1.8716e-05, 3.4571e-04,
          1.8716e-05, 1.8716e-05, 1.6189e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.6189e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.6189e-04, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.6189e-04, 1.8716e-05, 1.8716e-05,
          1.6189e-04, 2.9135e-04, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05,
          1.6189e-04, 1.8716e-05, 1.8716e-05, 1.6189e-04, 5.9366e-05, 1.8716e-05,
          1.8716e-05, 1.4818e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.6189e-04, 1.6189e-04,
          5.9366e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.6189e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05,
          5.9366e-05, 2.0242e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.1635e-04, 3.4070e-04, 2.0242e-04, 5.9366e-05, 1.8716e-05, 2.4307e-04,
          1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8668e-04,
          1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 3.3188e-04,
          2.0242e-04, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05,
          1.1635e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 9.2649e-04,
          1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05,
          1.8716e-05, 1.8716e-05, 5.9366e-05, 5.9366e-05, 1.6189e-04, 1.8716e-05,
          1.5700e-04, 5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 5.9366e-05, 2.9135e-04, 2.8777e-04, 1.8716e-05, 5.9366e-05,
          5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 5.9366e-05, 1.8716e-05,
          1.8668e-04, 1.8716e-05, 1.6189e-04, 1.6189e-04, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 2.0242e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05,
          1.8716e-05, 1.8716e-05, 1.6189e-04, 1.0753e-04, 1.8716e-05, 5.9366e-05,
          1.8716e-05, 1.0753e-04, 1.8716e-05, 1.0753e-04, 1.8716e-05, 5.9366e-05,
          2.1005e-04, 1.8716e-05, 2.9135e-04, 5.9366e-05, 1.8716e-05, 1.8716e-05,
          1.0753e-04, 1.6189e-04, 3.9387e-04, 3.0494e-04, 1.8716e-05, 1.8716e-05,
          3.0494e-04, 2.9135e-04, 1.2124e-04, 1.8716e-05, 1.8716e-05, 2.5964e-04,
          1.8716e-05, 1.4818e-04, 3.4833e-04, 5.9366e-05, 1.8716e-05, 1.0753e-04,
          1.4818e-04, 1.8716e-05, 1.4818e-04, 8.3113e-04, 1.8716e-05, 5.3215e-04,
          1.8716e-05, 1.8716e-05, 1.6189e-04, 1.4818e-04, 7.1526e-04, 3.9387e-04,
          4.9162e-04, 9.6703e-04, 1.8716e-05, 1.8716e-05, 4.7565e-04, 3.4833e-04,
          7.1526e-04, 1.2124e-04, 1.8716e-05, 1.8716e-05, 3.4833e-04, 1.8716e-05,
          1.0910e-03, 9.1314e-04, 1.0753e-04, 5.9366e-05, 1.6189e-04, 1.8716e-05,
          1.8716e-05, 6.0081e-04, 1.8716e-05, 2.5058e-04, 1.8716e-05, 1.8716e-05,
          1.0753e-04, 1.8716e-05, 1.8716e-05, 1.8716e-05, 4.7565e-04, 1.0753e-04,
          2.0370e-03, 3.4833e-04, 1.6189e-04, 1.8716e-05, 1.0753e-04, 1.7052e-03,
          1.0910e-03, 7.2908e-04, 6.5804e-04, 1.8716e-05, 1.8716e-05, 3.0494e-04,
          3.9387e-04, 1.7719e-03, 1.8716e-05, 2.9135e-04, 2.9135e-04, 1.8716e-05,
          1.6189e-04, 1.8716e-05, 1.2124e-04, 1.8716e-05, 1.2124e-04, 5.9366e-05,
          1.8716e-05, 5.9366e-05, 1.8716e-05, 1.8716e-05, 1.8716e-05, 1.2124e-04,
          1.2124e-04, 1.8716e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.8049e-03, 6.1615e-02, 1.5808e-02, 1.9440e-02, 5.5939e-02,
          4.2664e-02, 1.4389e-02, 1.9569e-03, 2.2316e-03, 5.7106e-03, 1.1597e-02,
          4.3640e-03, 1.8730e-03, 2.3727e-03, 2.3994e-03, 1.3409e-03, 2.2545e-03,
          1.4095e-03, 1.1473e-03, 7.1478e-04, 6.4802e-04, 6.0368e-04, 1.8301e-03,
          1.1604e-02, 3.6030e-03, 1.1063e-02, 4.6692e-03, 1.3800e-03, 2.4357e-03,
          3.8052e-03, 3.0308e-03, 1.1473e-03, 3.2234e-03, 2.3937e-03, 2.2831e-03,
          5.5618e-03, 3.5667e-03, 2.6283e-03, 2.0103e-03, 1.2178e-03, 4.3068e-03,
          3.9864e-03, 2.8763e-03, 3.6488e-03, 1.5345e-03, 1.9722e-03, 2.7809e-03,
          1.1225e-03, 8.8358e-04, 5.2528e-03, 4.1847e-03, 3.5496e-03, 2.4757e-03,
          2.4624e-03, 2.4757e-03, 1.8654e-03, 7.9269e-03, 6.7482e-03, 9.6436e-03,
          5.2681e-03, 2.3842e-03, 2.2240e-03, 1.8387e-03, 8.6899e-03, 6.4507e-03,
          2.0676e-02, 7.5111e-03, 2.8191e-03, 2.2697e-03, 2.0294e-03, 2.6875e-03,
          1.4524e-03, 9.8724e-03, 2.7714e-03, 4.3297e-03, 2.0142e-03, 2.1191e-03,
          2.7714e-03, 1.3687e-02, 3.8269e-02, 1.3504e-02, 2.5070e-02, 4.1428e-03,
          2.9392e-03, 4.1695e-03, 1.8816e-03, 6.1111e-03, 1.3519e-02, 1.7548e-02,
          1.2383e-02, 8.9417e-03, 4.2877e-03, 7.7629e-03, 1.3695e-03, 1.2703e-03,
          1.8845e-03, 1.5650e-03, 2.8458e-03, 6.4011e-03, 2.5543e-02, 2.1683e-02,
          2.3087e-02, 1.1902e-02, 1.5114e-02, 1.1696e-02, 6.3744e-03, 3.1662e-03,
          3.5839e-03, 6.1188e-03, 1.1787e-02, 7.8125e-03, 1.8799e-02, 1.2650e-02,
          3.6125e-03, 3.5191e-03, 1.0138e-03, 5.2223e-03, 1.7309e-03, 4.4174e-03,
          6.6109e-03, 1.9745e-02, 1.8570e-02, 1.7061e-03, 7.7629e-03, 4.3396e-02,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 6.5386e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.0246e-04, 5.1618e-05, 5.1618e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 6.5386e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.0246e-04,
          1.4484e-05, 1.4484e-05, 2.0683e-04, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.0246e-04, 1.4484e-05,
          5.1618e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 5.1618e-05, 6.5386e-05, 1.0246e-04, 5.1618e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 6.5386e-05,
          6.5386e-05, 1.4484e-05, 1.4484e-05, 2.0683e-04, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 5.1618e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 6.5386e-05, 1.4484e-05, 2.0683e-04,
          1.0246e-04, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.0246e-04, 1.4484e-05, 1.4484e-05, 1.0246e-04, 1.0246e-04,
          6.5386e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.0246e-04, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05, 6.5386e-05, 5.1618e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.0246e-04, 1.0246e-04, 2.9850e-04,
          1.4484e-05, 1.4484e-05, 5.1618e-05, 1.4484e-05, 2.6131e-04, 1.4484e-05,
          1.4484e-05, 2.0683e-04, 1.5850e-03, 1.0246e-04, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 1.4484e-05, 2.9850e-04, 5.1618e-05,
          6.5386e-05, 5.1618e-05, 2.0683e-04, 2.0683e-04, 1.4484e-05, 5.1618e-05,
          1.4484e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05, 1.4484e-05, 1.4484e-05,
          1.4484e-05, 5.1618e-05, 1.0246e-04, 2.9850e-04, 1.4484e-05, 1.4484e-05,
          5.1618e-05, 4.0293e-04, 5.1618e-05, 5.1618e-05, 1.4484e-05, 1.4484e-05,
          4.0293e-04, 1.4484e-05, 2.6131e-04, 3.4928e-04, 5.1618e-05, 2.9850e-04,
          5.1618e-05, 1.4484e-05, 5.3978e-04, 4.5371e-04, 1.4484e-05, 5.1618e-05,
          3.4356e-04, 2.9850e-04, 2.9278e-04, 1.4484e-05, 1.4484e-05, 2.0683e-04,
          1.4484e-05, 2.9850e-04, 1.0246e-04, 6.5386e-05, 1.4484e-05, 5.1618e-05,
          1.5593e-04, 6.5386e-05, 2.0683e-04, 2.9850e-04, 1.4484e-05, 2.0683e-04,
          1.4484e-05, 1.4484e-05, 6.6137e-04, 1.4484e-05, 7.0000e-04, 1.9722e-03,
          1.4484e-05, 1.0246e-04, 5.1618e-05, 1.4484e-05, 6.5386e-05, 6.5386e-05,
          4.1580e-04, 1.5593e-04, 1.4484e-05, 1.4484e-05, 1.0246e-04, 5.1618e-05,
          4.5371e-04, 2.0683e-04, 2.6131e-04, 1.4484e-05, 5.1618e-05, 1.4484e-05,
          6.5386e-05, 5.1618e-05, 6.5386e-05, 2.9278e-04, 1.4484e-05, 1.4484e-05,
          5.1618e-05, 1.5593e-04, 1.5593e-04, 6.5386e-05, 6.5386e-05, 1.4484e-05,
          4.5371e-04, 1.0246e-04, 1.4484e-05, 2.9278e-04, 1.4484e-05, 2.9278e-04,
          1.0246e-04, 7.8106e-04, 4.5371e-04, 1.4484e-05, 1.4484e-05, 1.8859e-04,
          4.0293e-04, 5.3978e-04, 1.4484e-05, 9.2268e-04, 2.9278e-04, 1.4484e-05,
          5.1618e-05, 1.4484e-05, 1.4484e-05, 5.1618e-05, 4.8327e-04, 1.0246e-04,
          1.4484e-05, 2.4188e-04, 2.4188e-04, 5.2643e-04, 2.4188e-04, 2.4188e-04,
          5.1618e-05, 5.1618e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.3771e-03, 5.5359e-02, 1.9897e-02, 1.6754e-02, 8.3923e-02,
          4.0039e-02, 1.7273e-02, 2.1000e-03, 2.7008e-03, 5.0735e-03, 6.5880e-03,
          3.3703e-03, 2.0828e-03, 1.9627e-03, 3.8929e-03, 1.2074e-03, 2.2888e-03,
          1.1559e-03, 8.5258e-04, 1.6441e-03, 1.7958e-03, 9.9087e-04, 1.5249e-03,
          6.3362e-03, 4.7874e-03, 1.1856e-02, 4.4594e-03, 2.8782e-03, 2.1858e-03,
          2.3994e-03, 2.5063e-03, 1.3876e-03, 4.8943e-03, 2.3022e-03, 2.5749e-03,
          1.1436e-02, 8.9035e-03, 3.7231e-03, 2.6817e-03, 1.8835e-03, 6.0577e-03,
          5.0583e-03, 3.9482e-03, 3.2806e-03, 2.8744e-03, 1.5383e-03, 5.3864e-03,
          1.0300e-03, 1.4124e-03, 9.0714e-03, 4.8637e-03, 3.3188e-03, 2.0905e-03,
          3.4790e-03, 2.7866e-03, 4.0817e-03, 2.1912e-02, 7.3357e-03, 2.0370e-02,
          9.7275e-03, 3.5191e-03, 2.7542e-03, 2.7962e-03, 5.6381e-03, 8.4991e-03,
          3.4466e-03, 1.5278e-03, 1.6994e-03, 3.7556e-03, 4.2152e-03, 2.9202e-03,
          2.7142e-03, 6.8817e-03, 4.7340e-03, 3.5381e-03, 2.2354e-03, 2.0771e-03,
          3.6201e-03, 3.2562e-02, 1.4900e-02, 9.3231e-03, 7.8583e-03, 8.1482e-03,
          7.2670e-03, 3.3493e-03, 2.4185e-03, 2.7905e-03, 1.7309e-03, 2.9602e-03,
          4.2114e-03, 4.8485e-03, 3.1643e-03, 3.2730e-03, 3.2406e-03, 2.6646e-03,
          3.5400e-03, 2.9602e-03, 2.2030e-03, 8.8577e-03, 4.9530e-02, 2.1362e-02,
          1.1009e-02, 4.1962e-03, 4.0016e-03, 1.7609e-02, 1.2238e-02, 5.1041e-03,
          3.2253e-03, 5.4436e-03, 6.0349e-03, 3.3340e-03, 3.6736e-03, 2.4567e-03,
          4.3907e-03, 6.6566e-03, 1.9417e-03, 8.3084e-03, 3.4275e-03, 1.0063e-02,
          8.2092e-03, 1.6785e-02, 2.1622e-02, 1.4553e-03, 8.4610e-03, 1.1909e-02,
          4.8615e-02, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05,
          4.8935e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05,
          4.8935e-05, 4.8935e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05, 1.2434e-04,
          1.3709e-05, 1.3709e-05, 5.4359e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          4.8935e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.5342e-04, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 4.8935e-05, 1.3709e-05,
          4.8935e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.5342e-04, 1.3709e-05, 1.3709e-05, 6.4802e-04, 4.8935e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 6.4802e-04,
          1.5342e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.5342e-04, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.5342e-04,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05,
          1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.5342e-04, 1.5342e-04, 1.3709e-05,
          4.8935e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 4.8935e-05, 5.9128e-04, 1.5342e-04, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05,
          1.3709e-05, 1.3709e-05, 4.8935e-05, 2.0051e-04, 1.3709e-05, 4.8935e-05,
          1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 4.8935e-05, 1.5342e-04, 4.8935e-05, 1.3709e-05, 1.3709e-05,
          1.3709e-05, 4.8935e-05, 1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05,
          4.8935e-05, 4.8935e-05, 1.3709e-05, 1.1814e-04, 4.8935e-05, 1.3709e-05,
          4.8935e-05, 2.0051e-04, 4.8935e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05,
          4.8935e-05, 4.8935e-05, 5.8889e-04, 1.3709e-05, 1.3709e-05, 1.5342e-04,
          1.3709e-05, 4.8935e-05, 1.5342e-04, 1.3709e-05, 1.3709e-05, 4.8935e-05,
          4.8935e-05, 1.3709e-05, 1.5342e-04, 4.8935e-05, 4.8935e-05, 1.5342e-04,
          1.3709e-05, 1.3709e-05, 1.5342e-04, 2.4581e-04, 4.8935e-05, 1.6346e-03,
          1.6346e-03, 1.5342e-04, 4.8935e-05, 1.3709e-05, 1.5342e-04, 1.5342e-04,
          1.3709e-05, 4.8935e-05, 1.3709e-05, 1.3709e-05, 3.0518e-04, 4.8935e-05,
          8.4448e-04, 3.0494e-04, 4.8935e-05, 4.8935e-05, 4.8935e-05, 1.3709e-05,
          4.8935e-05, 1.5342e-04, 1.3709e-05, 3.9744e-04, 1.3709e-05, 1.3709e-05,
          1.5342e-04, 4.8935e-05, 4.8935e-05, 1.3709e-05, 1.5342e-04, 4.8935e-05,
          2.0051e-04, 3.0494e-04, 2.0051e-04, 4.8935e-05, 4.8935e-05, 2.4581e-04,
          3.0494e-04, 4.8935e-05, 8.4448e-04, 4.8935e-05, 4.8935e-05, 2.4581e-04,
          4.8935e-05, 3.9744e-04, 1.3709e-05, 4.8935e-05, 2.4581e-04, 1.3709e-05,
          4.8935e-05, 4.8935e-05, 4.8935e-05, 4.8935e-05, 4.8935e-05, 4.8935e-05,
          4.8935e-05, 2.0051e-04, 2.0051e-04, 5.1498e-04, 2.0051e-04, 5.4359e-04,
          4.8935e-05, 4.8935e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.1068e-03, 5.8075e-02, 1.6876e-02, 1.7593e-02, 6.2927e-02,
          4.2999e-02, 1.3535e-02, 1.6460e-03, 1.7023e-03, 5.4703e-03, 1.0117e-02,
          3.7727e-03, 1.5831e-03, 2.3746e-03, 2.2411e-03, 1.2770e-03, 2.0866e-03,
          1.4000e-03, 1.1368e-03, 8.2970e-04, 7.7486e-04, 6.2132e-04, 1.7567e-03,
          1.0529e-02, 3.3169e-03, 1.0681e-02, 4.8180e-03, 1.5850e-03, 2.2774e-03,
          3.5057e-03, 2.7752e-03, 1.2770e-03, 2.9984e-03, 2.1191e-03, 2.1687e-03,
          5.7678e-03, 4.7188e-03, 2.6112e-03, 1.9569e-03, 1.4200e-03, 4.6349e-03,
          3.8795e-03, 3.3016e-03, 3.7479e-03, 1.5678e-03, 1.7357e-03, 2.6226e-03,
          1.2007e-03, 8.8120e-04, 5.8670e-03, 4.1237e-03, 3.4866e-03, 1.7481e-03,
          2.7428e-03, 2.7351e-03, 2.3670e-03, 9.8190e-03, 8.9493e-03, 1.3672e-02,
          7.0343e-03, 3.1872e-03, 2.9755e-03, 2.3766e-03, 7.0801e-03, 5.3101e-03,
          2.0416e-02, 5.9929e-03, 2.4281e-03, 2.1439e-03, 2.1191e-03, 3.6316e-03,
          1.7443e-03, 8.3694e-03, 3.2673e-03, 4.1656e-03, 2.4605e-03, 2.1801e-03,
          3.3073e-03, 1.3702e-02, 3.2196e-02, 1.1673e-02, 3.2227e-02, 6.0654e-03,
          4.2915e-03, 4.4823e-03, 2.4834e-03, 4.9782e-03, 8.5526e-03, 1.3054e-02,
          1.1383e-02, 8.7051e-03, 3.5248e-03, 7.4234e-03, 1.6575e-03, 1.4610e-03,
          2.4586e-03, 1.9760e-03, 2.3022e-03, 6.2485e-03, 2.1255e-02, 1.5656e-02,
          1.6342e-02, 1.0941e-02, 1.5549e-02, 1.3268e-02, 6.2828e-03, 3.9825e-03,
          4.3983e-03, 5.6572e-03, 1.0605e-02, 7.5455e-03, 1.7288e-02, 9.3613e-03,
          3.3760e-03, 3.4218e-03, 1.2827e-03, 4.6806e-03, 2.0409e-03, 5.0812e-03,
          6.5460e-03, 1.2062e-02, 1.2939e-02, 1.7481e-03, 6.7139e-03, 1.8524e-02,
          1.4069e-02, 4.8096e-02, 1.5438e-05, 6.1691e-05, 6.1691e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          9.5248e-05, 4.8995e-05, 4.8995e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05,
          1.5438e-05, 1.5438e-05, 5.2786e-04, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 6.1691e-05, 1.5438e-05, 9.5248e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 9.5248e-05, 6.1691e-05, 9.5248e-05, 4.8995e-05, 1.5438e-05,
          1.5438e-05, 4.8995e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05,
          6.1691e-05, 1.5438e-05, 1.5438e-05, 5.2786e-04, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05, 1.5438e-05, 5.2786e-04,
          9.5248e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05, 1.5438e-05,
          1.5438e-05, 9.5248e-05, 1.5438e-05, 4.8995e-05, 9.5248e-05, 9.5248e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05, 6.1691e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 4.8995e-05, 9.5248e-05, 4.9055e-05,
          1.5438e-05, 1.5438e-05, 4.8995e-05, 4.8995e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05, 4.8995e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 6.1691e-05, 9.5248e-05, 9.5248e-05, 4.8995e-05,
          1.5438e-05, 1.5438e-05, 9.5248e-05, 1.5438e-05, 4.8995e-05, 1.5438e-05,
          1.5438e-05, 4.8995e-05, 1.3103e-03, 9.5248e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 3.2210e-04, 4.8995e-05,
          6.1691e-05, 4.8995e-05, 4.0603e-04, 5.2786e-04, 1.5438e-05, 4.8995e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 9.5248e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 9.5248e-05, 9.5248e-05, 5.1117e-04, 1.5438e-05, 1.5438e-05,
          4.8995e-05, 4.0460e-04, 4.8995e-05, 4.8995e-05, 1.5438e-05, 4.8995e-05,
          4.5085e-04, 1.5438e-05, 1.5438e-05, 6.1691e-05, 4.8995e-05, 4.8995e-05,
          4.8995e-05, 4.8995e-05, 1.1168e-03, 4.5085e-04, 1.5438e-05, 4.8995e-05,
          9.3555e-04, 3.2210e-04, 2.9922e-04, 1.5438e-05, 1.5438e-05, 9.5308e-05,
          1.5438e-05, 6.0940e-04, 9.5308e-05, 6.1691e-05, 1.5438e-05, 4.8995e-05,
          4.8995e-05, 1.5438e-05, 9.5248e-05, 2.7585e-04, 1.5438e-05, 9.5248e-05,
          1.5438e-05, 1.5438e-05, 5.7220e-04, 4.8995e-05, 3.2210e-04, 9.9468e-04,
          8.6641e-04, 9.5248e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05, 6.1691e-05,
          6.1691e-05, 2.9922e-04, 1.5438e-05, 1.5438e-05, 2.2388e-04, 4.8995e-05,
          3.2210e-04, 9.5248e-05, 2.4235e-04, 4.8995e-05, 4.8995e-05, 1.5438e-05,
          9.5248e-05, 2.7585e-04, 6.1691e-05, 2.9922e-04, 1.5438e-05, 1.5438e-05,
          4.8995e-05, 1.5438e-05, 1.5438e-05, 6.1691e-05, 9.5248e-05, 1.5438e-05,
          3.2210e-04, 5.1117e-04, 4.8995e-05, 2.9922e-04, 4.8995e-05, 5.8651e-04,
          9.5248e-05, 2.7585e-04, 7.3814e-04, 1.5438e-05, 1.5438e-05, 1.7071e-04,
          6.9189e-04, 5.2595e-04, 1.5438e-05, 5.2595e-04, 5.2595e-04, 1.5438e-05,
          4.8995e-05, 4.8995e-05, 4.8995e-05, 3.2449e-04, 2.3127e-04, 4.8995e-05,
          1.5438e-05, 4.8995e-05, 4.8995e-05, 4.5323e-04, 4.8995e-05, 2.3127e-04,
          1.5438e-05, 4.8995e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.8817e-03, 5.4138e-02, 1.6373e-02, 1.5961e-02, 5.8624e-02,
          4.3335e-02, 1.1589e-02, 1.9131e-03, 1.9932e-03, 5.9128e-03, 1.0147e-02,
          3.8605e-03, 1.9131e-03, 2.2869e-03, 2.5272e-03, 1.4868e-03, 2.3918e-03,
          1.5211e-03, 1.3390e-03, 7.2289e-04, 6.6710e-04, 5.8413e-04, 2.1858e-03,
          1.0582e-02, 3.7613e-03, 9.1248e-03, 4.9324e-03, 1.3733e-03, 2.5558e-03,
          3.8128e-03, 3.2578e-03, 1.2331e-03, 3.3245e-03, 2.3975e-03, 2.1935e-03,
          5.6725e-03, 3.9444e-03, 2.6970e-03, 2.1362e-03, 1.2150e-03, 4.3602e-03,
          3.4294e-03, 2.8839e-03, 3.4428e-03, 1.5860e-03, 1.9989e-03, 2.7676e-03,
          1.0967e-03, 7.9632e-04, 5.1384e-03, 4.0398e-03, 3.5439e-03, 2.0103e-03,
          2.4662e-03, 2.7390e-03, 2.2316e-03, 8.0948e-03, 7.4158e-03, 1.0078e-02,
          6.2027e-03, 2.8839e-03, 2.7657e-03, 2.2583e-03, 7.3853e-03, 5.6992e-03,
          2.4338e-02, 7.8506e-03, 2.8610e-03, 2.0943e-03, 2.0351e-03, 3.4752e-03,
          1.6298e-03, 8.8120e-03, 2.7370e-03, 4.2992e-03, 2.1820e-03, 2.1057e-03,
          3.0804e-03, 1.1711e-02, 2.8244e-02, 1.1703e-02, 2.5635e-02, 4.4937e-03,
          3.3245e-03, 4.4212e-03, 2.3365e-03, 5.3215e-03, 1.1482e-02, 1.4000e-02,
          1.1459e-02, 8.7662e-03, 3.7785e-03, 7.3166e-03, 1.4153e-03, 1.2159e-03,
          2.1992e-03, 1.6661e-03, 2.4548e-03, 6.9466e-03, 1.6174e-02, 1.4847e-02,
          1.6342e-02, 1.2100e-02, 1.5022e-02, 1.0605e-02, 6.4774e-03, 3.8147e-03,
          4.3983e-03, 5.2719e-03, 1.1002e-02, 8.1406e-03, 1.9333e-02, 1.1513e-02,
          3.2349e-03, 3.2959e-03, 1.1158e-03, 4.8981e-03, 1.6747e-03, 4.1618e-03,
          6.0463e-03, 1.3893e-02, 1.0689e-02, 1.8969e-03, 6.8970e-03, 1.7944e-02,
          1.1444e-02, 3.1555e-02, 4.9652e-02, 1.7524e-05, 6.2048e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          9.1136e-05, 4.6611e-05, 4.6611e-05, 1.7524e-05, 1.7524e-05, 9.1136e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 2.1434e-04,
          1.7524e-05, 1.7524e-05, 3.5763e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 9.1136e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 9.1136e-05, 6.2048e-05, 9.1136e-05, 1.2672e-04, 1.7524e-05,
          1.7524e-05, 4.6611e-05, 4.6611e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05,
          6.2048e-05, 1.7524e-05, 1.7524e-05, 3.5763e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05, 1.7524e-05, 3.5763e-04,
          9.1136e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05, 1.7524e-05,
          1.7524e-05, 9.1136e-05, 1.7524e-05, 6.2048e-05, 6.2048e-05, 9.1136e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 9.1136e-05, 6.2048e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 2.1434e-04,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 6.2048e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 4.6611e-05, 9.1136e-05, 1.2672e-04,
          1.7524e-05, 1.7524e-05, 4.6611e-05, 1.9634e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 9.1136e-05, 1.9634e-04,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 6.2048e-05, 1.7130e-04, 9.1136e-05, 4.6611e-05,
          1.7524e-05, 1.7524e-05, 9.1136e-05, 4.6611e-05, 4.6611e-05, 1.7524e-05,
          1.7524e-05, 4.6611e-05, 1.1826e-03, 1.7130e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 2.4092e-04, 4.6611e-05,
          6.2048e-05, 1.7524e-05, 3.5763e-04, 3.5763e-04, 1.7524e-05, 4.6611e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 9.1136e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 9.1136e-05, 9.1136e-05, 3.2091e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 2.7657e-04, 4.6611e-05, 9.1136e-05, 1.7524e-05, 4.6611e-05,
          3.9887e-04, 1.7524e-05, 1.9634e-04, 6.2048e-05, 4.6611e-05, 4.6611e-05,
          4.6611e-05, 4.6611e-05, 8.2397e-04, 3.9887e-04, 4.6611e-05, 4.6611e-05,
          7.1907e-04, 8.3399e-04, 3.5429e-04, 1.7524e-05, 1.7524e-05, 1.7130e-04,
          1.7524e-05, 3.2091e-04, 3.2115e-04, 6.2048e-05, 1.7524e-05, 1.2672e-04,
          4.6611e-05, 1.7524e-05, 9.1136e-05, 2.7657e-04, 1.7524e-05, 3.2091e-04,
          1.7524e-05, 1.7524e-05, 5.2214e-04, 4.6611e-05, 2.4092e-04, 1.4906e-03,
          6.6614e-04, 2.4092e-04, 1.7524e-05, 1.7524e-05, 6.2048e-05, 6.2048e-05,
          9.1136e-05, 3.1304e-04, 1.7524e-05, 1.7524e-05, 4.2939e-04, 4.6611e-05,
          3.1877e-04, 9.1136e-05, 1.6725e-04, 4.6611e-05, 1.2445e-04, 1.7524e-05,
          9.1136e-05, 2.7657e-04, 9.1136e-05, 3.1304e-04, 1.7524e-05, 1.7524e-05,
          1.9634e-04, 1.7524e-05, 1.7524e-05, 6.2048e-05, 2.4092e-04, 1.9634e-04,
          2.4092e-04, 5.4836e-04, 4.6611e-05, 2.0468e-04, 1.9634e-04, 4.7302e-04,
          2.4092e-04, 2.7657e-04, 5.0735e-04, 1.2445e-04, 1.7524e-05, 1.2445e-04,
          6.2275e-04, 3.5429e-04, 2.0468e-04, 3.5429e-04, 3.5429e-04, 1.7524e-05,
          4.6611e-05, 1.7524e-05, 4.6611e-05, 4.6611e-05, 1.5509e-04, 4.6611e-05,
          1.7524e-05, 4.6611e-05, 1.5509e-04, 4.2009e-04, 1.5509e-04, 1.5509e-04,
          1.7524e-05, 4.6611e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4741e-03, 6.0089e-02, 1.6052e-02, 1.8173e-02, 6.1096e-02,
          5.0873e-02, 1.0666e-02, 2.9202e-03, 2.4776e-03, 7.3280e-03, 7.3891e-03,
          3.8586e-03, 4.7913e-03, 1.8787e-03, 1.2772e-02, 2.5120e-03, 7.1220e-03,
          3.6964e-03, 1.5020e-03, 7.2145e-04, 9.2745e-04, 4.8923e-04, 4.4823e-03,
          8.3084e-03, 7.2327e-03, 9.0790e-03, 8.2016e-03, 1.3523e-03, 7.7248e-03,
          2.5978e-03, 6.4735e-03, 2.5330e-03, 1.7502e-02, 6.9427e-03, 7.3586e-03,
          1.5282e-02, 2.7943e-03, 6.5384e-03, 9.4299e-03, 1.1730e-03, 4.8523e-03,
          1.3475e-03, 1.1415e-03, 9.6512e-04, 1.1806e-03, 5.4436e-03, 1.3397e-02,
          8.4209e-04, 6.7711e-04, 4.5815e-03, 9.8190e-03, 6.6566e-03, 5.0850e-03,
          1.9159e-03, 4.0016e-03, 2.2564e-03, 6.7329e-03, 5.6458e-03, 6.4125e-03,
          4.5052e-03, 4.0932e-03, 2.5864e-03, 4.5967e-03, 2.0847e-03, 2.6566e-02,
          1.7176e-03, 1.3304e-03, 2.4414e-03, 9.2554e-04, 6.7949e-04, 8.5831e-04,
          9.7132e-04, 1.3145e-02, 3.0689e-03, 7.2746e-03, 1.4648e-03, 1.9083e-03,
          3.9482e-03, 6.6566e-03, 3.8891e-03, 1.6541e-02, 3.5744e-03, 4.0436e-03,
          3.0460e-03, 8.1100e-03, 2.0657e-03, 3.8204e-03, 1.0786e-03, 1.1435e-03,
          6.7825e-03, 5.9204e-03, 1.0872e-03, 3.9940e-03, 1.5774e-03, 1.2093e-03,
          2.5635e-03, 1.7748e-03, 9.1553e-03, 1.2512e-02, 5.2223e-03, 7.8659e-03,
          1.2985e-02, 1.7614e-03, 3.1261e-03, 1.5465e-02, 3.4698e-02, 4.8447e-03,
          8.1406e-03, 2.6798e-03, 7.8583e-03, 1.2913e-03, 1.4572e-03, 1.6975e-03,
          4.2229e-03, 3.5915e-03, 1.1826e-03, 2.8793e-02, 1.6222e-03, 3.0441e-03,
          5.6038e-03, 1.6388e-02, 2.2945e-03, 2.4548e-03, 2.4506e-02, 2.3689e-03,
          5.3635e-03, 3.4866e-03, 5.2414e-03, 6.6833e-02, 5.9903e-05, 1.4305e-05,
          1.2255e-04, 5.9903e-05, 1.4305e-05, 1.7083e-04, 5.9903e-05, 2.1017e-04,
          1.4305e-05, 8.7214e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.2255e-04,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 7.6890e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          5.9903e-05, 2.1017e-04, 1.2255e-04, 2.1017e-04, 1.4305e-05, 3.1853e-04,
          1.4305e-05, 1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.2255e-04, 5.9903e-05, 1.4305e-05, 1.4305e-05, 1.2255e-04, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05,
          1.2255e-04, 2.1017e-04, 1.4305e-05, 1.4305e-05, 7.6890e-05, 1.4305e-05,
          1.2255e-04, 1.4305e-05, 1.4305e-05, 2.3341e-04, 1.2255e-04, 1.4305e-05,
          1.4305e-05, 1.5557e-04, 1.4305e-05, 5.9903e-05, 5.9903e-05, 1.4305e-05,
          5.9903e-05, 5.9903e-05, 1.4305e-05, 7.6890e-05, 5.9903e-05, 2.3341e-04,
          1.2255e-04, 1.4305e-05, 1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 7.6890e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 5.9903e-05, 1.4305e-05, 1.4305e-05,
          1.8108e-04, 2.7275e-04, 2.1017e-04, 1.4305e-05, 1.4305e-05, 2.7275e-04,
          1.4305e-05, 1.4305e-05, 7.6890e-05, 1.4305e-05, 1.4305e-05, 2.6488e-04,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 7.6890e-05, 4.5967e-04,
          1.4305e-05, 5.9903e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 3.1209e-04,
          1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 7.6890e-05, 1.2255e-04, 1.4305e-05,
          2.3341e-04, 1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 9.0504e-04, 5.2357e-04, 1.4305e-05, 1.2255e-04,
          1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          2.6488e-04, 1.4305e-05, 5.9903e-05, 3.7074e-04, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 1.2255e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.5557e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 1.4305e-05, 3.1209e-04, 7.6890e-05, 1.4305e-05, 1.4305e-05,
          1.4305e-05, 5.9903e-05, 8.9216e-04, 5.9903e-05, 1.4305e-05, 1.4305e-05,
          5.5504e-04, 8.4639e-04, 2.0111e-04, 1.4305e-05, 1.4305e-05, 2.3341e-04,
          1.4305e-05, 1.5557e-04, 3.7456e-04, 7.6890e-05, 1.4305e-05, 1.5557e-04,
          2.1815e-04, 1.4305e-05, 2.1815e-04, 5.0783e-04, 3.3855e-04, 8.8930e-04,
          1.9741e-04, 1.4305e-05, 2.3341e-04, 3.7074e-04, 5.4073e-04, 1.0033e-03,
          1.0614e-03, 3.7456e-04, 1.4305e-05, 1.4305e-05, 1.2255e-04, 1.2255e-04,
          7.6890e-05, 1.7083e-04, 1.4305e-05, 1.4305e-05, 3.7456e-04, 1.4305e-05,
          6.8378e-04, 5.7316e-04, 1.5557e-04, 1.4305e-05, 2.6011e-04, 1.4305e-05,
          1.4305e-05, 9.4748e-04, 1.4305e-05, 5.9903e-05, 1.4305e-05, 1.4305e-05,
          3.7456e-04, 1.4305e-05, 1.4305e-05, 1.4305e-05, 4.5180e-04, 2.6631e-04,
          2.3804e-03, 3.7456e-04, 1.2255e-04, 1.4305e-05, 1.5557e-04, 9.4843e-04,
          4.0126e-04, 5.0783e-04, 1.0262e-03, 1.4305e-05, 1.4305e-05, 1.4305e-05,
          6.4516e-04, 4.4918e-04, 2.6250e-04, 5.0783e-04, 6.4516e-04, 1.4305e-05,
          2.8324e-04, 1.4305e-05, 2.8324e-04, 5.9903e-05, 5.9903e-05, 7.6890e-05,
          1.4305e-05, 1.2255e-04, 5.9903e-05, 3.9601e-04, 5.9903e-05, 7.7581e-04,
          5.9903e-05, 1.4305e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3215e-03, 6.6040e-02, 1.8005e-02, 2.0355e-02, 5.7678e-02,
          5.9967e-02, 1.3069e-02, 9.4986e-03, 5.6763e-03, 6.5231e-03, 7.3738e-03,
          3.2253e-03, 8.5754e-03, 2.8782e-03, 6.5117e-03, 1.2932e-02, 1.6174e-02,
          8.1558e-03, 4.9744e-03, 1.1196e-03, 1.2131e-03, 1.0471e-03, 6.7635e-03,
          8.3160e-03, 4.8141e-03, 1.1192e-02, 1.0315e-02, 1.5965e-03, 1.2833e-02,
          3.7937e-03, 2.1729e-02, 3.6716e-03, 9.4681e-03, 5.9128e-03, 1.0956e-02,
          5.2681e-03, 1.5182e-03, 7.2060e-03, 7.9422e-03, 1.7233e-03, 3.4447e-03,
          1.1253e-03, 9.8515e-04, 9.4175e-04, 9.9468e-04, 1.4603e-02, 7.4310e-03,
          1.4563e-03, 1.1120e-03, 6.5155e-03, 1.4015e-02, 1.4183e-02, 9.5062e-03,
          2.4014e-03, 7.6256e-03, 2.3918e-03, 1.8768e-03, 4.1885e-03, 2.4624e-03,
          2.0905e-03, 6.2599e-03, 5.6610e-03, 9.4147e-03, 1.7853e-03, 9.2316e-03,
          2.2755e-03, 1.7500e-03, 1.9932e-03, 6.7186e-04, 5.7936e-04, 8.9121e-04,
          9.2793e-04, 1.5007e-02, 3.2120e-03, 1.5465e-02, 2.0084e-03, 2.6321e-03,
          5.8136e-03, 1.8139e-03, 1.9054e-03, 5.2567e-03, 2.0466e-03, 1.9293e-03,
          1.4963e-03, 2.5055e-02, 3.5534e-03, 5.3406e-03, 1.5116e-03, 1.3542e-03,
          7.4272e-03, 4.9744e-03, 8.4352e-04, 4.3945e-03, 9.4128e-04, 8.9073e-04,
          2.7637e-03, 1.6966e-03, 1.9272e-02, 9.3231e-03, 1.5392e-03, 2.4662e-03,
          7.9498e-03, 1.8559e-03, 1.9779e-03, 4.2229e-03, 9.1934e-03, 7.4043e-03,
          1.8951e-02, 1.7471e-03, 5.9357e-03, 1.3332e-03, 2.1000e-03, 2.1973e-03,
          3.7937e-03, 2.2545e-03, 2.2469e-03, 1.0338e-02, 1.1358e-03, 4.3869e-03,
          7.7362e-03, 7.1259e-03, 1.0300e-03, 8.2397e-03, 9.1324e-03, 1.0567e-03,
          1.2617e-03, 1.5278e-03, 2.2697e-03, 1.2444e-02, 5.6030e-02, 1.2994e-05,
          1.9240e-04, 5.1856e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05, 4.2176e-04,
          6.0856e-05, 9.9719e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05, 9.9719e-05,
          5.1856e-05, 1.2994e-05, 5.1856e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05,
          1.2994e-05, 5.1856e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05,
          5.1856e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05, 5.1856e-05, 9.9719e-05,
          5.1856e-05, 1.2994e-05, 2.1148e-04, 1.2994e-05, 1.2994e-05, 1.2994e-05,
          9.9719e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05, 9.9719e-05, 1.2994e-05,
          6.0856e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05,
          1.2994e-05, 5.1856e-05, 5.1856e-05, 1.2994e-05, 6.0856e-05, 1.2994e-05,
          9.9719e-05, 5.1856e-05, 6.0856e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05,
          1.2994e-05, 1.2994e-05, 5.1856e-05, 2.5034e-04, 9.9719e-05, 5.1856e-05,
          1.2994e-05, 1.2994e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05, 1.2994e-05,
          5.1856e-05, 1.2994e-05, 1.2994e-05, 1.2994e-05, 9.9719e-05, 2.5034e-04,
          1.2994e-05, 1.2994e-05, 1.2994e-05, 5.1856e-05, 5.1856e-05, 1.4269e-04,
          1.2994e-05, 9.9719e-05, 1.2994e-05, 9.9719e-05, 5.1856e-05, 3.5501e-04,
          1.2994e-05, 1.0377e-04, 1.0377e-04, 5.1856e-05, 2.5249e-04, 5.1856e-05,
          1.1557e-04, 1.1557e-04, 1.5459e-03, 1.9240e-04, 1.2994e-05, 5.1856e-05,
          1.2994e-05, 1.2994e-05, 1.9240e-04, 1.2994e-05, 1.2994e-05, 5.5361e-04,
          1.2994e-05, 1.2994e-05, 1.2994e-05, 6.0856e-05, 9.9719e-05, 1.9050e-04,
          1.0377e-04, 5.1856e-05, 1.0377e-04, 1.9050e-04, 1.2994e-05, 6.0856e-05,
          2.0647e-04, 4.2176e-04, 1.2994e-05, 1.2994e-05, 1.2994e-05, 3.0231e-04,
          1.2994e-05, 1.9240e-04, 1.2994e-05, 1.2994e-05, 1.2994e-05, 1.4269e-04,
          1.2994e-05, 1.2994e-05, 6.0856e-05, 9.9719e-05, 9.9719e-05, 1.2994e-05,
          1.9240e-04, 5.1856e-05, 4.0293e-04, 1.2274e-03, 1.5163e-04, 5.1856e-05,
          1.5354e-04, 2.4021e-04, 5.6028e-04, 1.9050e-04, 1.2994e-05, 1.9240e-04,
          5.1856e-05, 1.0377e-04, 1.0377e-04, 1.2994e-05, 9.9719e-05, 1.2994e-05,
          6.0129e-04, 1.2994e-05, 2.1148e-04, 7.2479e-04, 1.2994e-05, 1.2994e-05,
          1.0377e-04, 1.0377e-04, 1.0377e-04, 9.9719e-05, 1.2994e-05, 1.0377e-04,
          1.2994e-05, 1.2994e-05, 1.2994e-05, 9.9719e-05, 1.2994e-05, 5.1856e-05,
          1.2994e-05, 3.0231e-04, 6.0856e-05, 1.2994e-05, 1.2994e-05, 9.9719e-05,
          9.9719e-05, 9.9719e-05, 1.5163e-04, 1.2994e-05, 6.0856e-05, 2.5034e-04,
          2.4021e-04, 9.9719e-05, 3.0231e-04, 9.9719e-05, 1.0377e-04, 1.0377e-04,
          4.6659e-04, 9.9719e-05, 1.5163e-04, 1.0377e-04, 1.2994e-05, 1.9050e-04,
          1.2994e-05, 9.9719e-05, 3.4118e-04, 5.1856e-05, 1.2994e-05, 1.5163e-04,
          2.0134e-04, 1.2994e-05, 1.2994e-05, 3.0231e-04, 1.5163e-04, 3.4118e-04,
          5.1856e-05, 5.1856e-05, 3.4118e-04, 5.1856e-05, 1.5354e-04, 1.5163e-04,
          5.6028e-04, 3.4118e-04, 1.0377e-04, 1.0377e-04, 1.2994e-05, 1.2994e-05,
          1.2994e-05, 6.0856e-05, 5.1856e-05, 1.2994e-05, 1.9050e-04, 2.0134e-04,
          1.2994e-05, 1.2994e-05, 1.2994e-05, 1.9050e-04, 1.2994e-05, 1.2994e-05,
          9.9719e-05, 1.9050e-04, 1.2994e-05, 6.0701e-04, 1.2994e-05, 1.2994e-05,
          3.0231e-04, 1.0377e-04, 1.0377e-04, 1.2994e-05, 6.0856e-05, 3.0231e-04,
          6.0856e-05, 2.1000e-03, 6.3562e-04, 6.0856e-05, 1.5163e-04, 1.0538e-03,
          6.0856e-05, 7.1096e-04, 2.2507e-03, 1.2994e-05, 1.9050e-04, 6.5899e-04,
          8.3780e-04, 1.2994e-05, 2.2888e-04, 3.0231e-04, 5.1832e-04, 1.2994e-05,
          6.0856e-05, 6.0856e-05, 3.5858e-04, 6.0856e-05, 6.0856e-05, 6.0856e-05,
          1.2994e-05, 9.9719e-05, 6.0856e-05, 3.3379e-04, 2.8586e-04, 1.1282e-03,
          6.0856e-05, 1.2994e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.9248e-03, 5.3802e-02, 1.7944e-02, 1.5572e-02, 4.9652e-02,
          4.6814e-02, 1.0719e-02, 9.5978e-03, 3.0403e-03, 2.9106e-03, 4.4250e-03,
          1.8616e-03, 2.3708e-03, 2.0218e-03, 2.7676e-03, 9.0790e-03, 4.1428e-03,
          2.9125e-03, 2.3937e-03, 1.0004e-03, 1.0500e-03, 1.2951e-03, 3.1815e-03,
          5.5885e-03, 3.2787e-03, 9.7122e-03, 5.9700e-03, 1.4639e-03, 4.1580e-03,
          2.3804e-03, 1.9989e-02, 1.7977e-03, 3.5648e-03, 2.2278e-03, 7.8278e-03,
          3.4218e-03, 1.7023e-03, 3.4180e-03, 3.0251e-03, 1.4791e-03, 4.0321e-03,
          2.3155e-03, 1.6918e-03, 1.7872e-03, 1.2436e-03, 3.4904e-03, 4.0016e-03,
          1.4172e-03, 1.3514e-03, 6.0806e-03, 1.3008e-02, 2.2095e-02, 3.7956e-03,
          5.2757e-03, 1.3557e-02, 4.3640e-03, 2.0142e-03, 7.1564e-03, 2.8954e-03,
          2.5616e-03, 5.1804e-03, 2.6047e-02, 1.8585e-02, 3.1776e-03, 7.1297e-03,
          3.4809e-03, 1.5297e-03, 1.5917e-03, 1.5144e-03, 1.3542e-03, 1.7633e-03,
          1.7538e-03, 5.7373e-03, 7.4310e-03, 2.8290e-02, 5.8594e-03, 5.9967e-03,
          1.0086e-02, 1.9321e-03, 2.1706e-03, 3.2139e-03, 2.4719e-03, 2.7332e-03,
          2.1362e-03, 7.4280e-02, 1.6205e-02, 2.7657e-03, 1.6575e-03, 1.9293e-03,
          5.0049e-03, 3.8567e-03, 1.9283e-03, 3.1528e-03, 2.4185e-03, 2.0409e-03,
          2.7542e-03, 3.1490e-03, 1.2939e-02, 4.3526e-03, 2.1076e-03, 2.3136e-03,
          4.1199e-03, 1.3695e-03, 1.5917e-03, 4.1924e-03, 5.3062e-03, 5.9471e-03,
          6.1127e-02, 5.7220e-03, 5.5199e-03, 3.7766e-03, 2.5501e-03, 2.1458e-03,
          2.6550e-03, 4.7722e-03, 2.8534e-03, 7.0877e-03, 2.8801e-03, 3.5706e-03,
          5.3520e-03, 3.5763e-03, 1.2083e-03, 4.9934e-03, 5.0430e-03, 1.2646e-03,
          1.7719e-03, 2.1877e-03, 2.5272e-03, 6.8893e-03, 1.1452e-02, 7.1533e-02,
          5.6982e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05,
          1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05,
          1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          5.6982e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05, 5.6982e-05, 5.6982e-05,
          5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          5.6982e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.8585e-04, 1.9133e-05,
          1.9133e-05, 5.6982e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          5.6982e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 5.7161e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 2.2936e-04, 2.2936e-04, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05, 1.9133e-05,
          5.6982e-05, 1.8585e-04, 1.9133e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 5.7161e-05, 5.6982e-05, 5.6982e-05,
          1.9133e-05, 5.6982e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05, 5.6982e-05,
          5.6982e-05, 1.9133e-05, 1.8585e-04, 2.2364e-04, 1.8716e-04, 5.6982e-05,
          5.6982e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 5.6982e-05,
          1.8585e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.1292e-03,
          1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 6.9904e-04, 5.6982e-05, 1.9133e-05,
          5.6982e-05, 5.7161e-05, 1.8716e-04, 1.9133e-05, 1.8585e-04, 5.6982e-05,
          5.6982e-05, 5.6982e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05, 5.7161e-05,
          5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05,
          5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9157e-04, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05,
          1.9133e-05, 1.9133e-05, 1.8585e-04, 5.6982e-05, 1.9133e-05, 1.9157e-04,
          1.9157e-04, 2.2936e-04, 1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 2.2936e-04,
          1.8585e-04, 6.9904e-04, 5.7161e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05,
          1.9157e-04, 1.9133e-05, 1.9133e-05, 1.8585e-04, 1.8585e-04, 2.2936e-04,
          1.9133e-05, 5.6982e-05, 2.2364e-04, 5.6982e-05, 1.9133e-05, 1.8585e-04,
          1.8585e-04, 2.2936e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05, 2.2364e-04, 1.9133e-05, 1.1730e-03, 1.9157e-04,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 5.6982e-05, 1.9133e-05, 1.9133e-05,
          2.2936e-04, 2.2364e-04, 1.9133e-05, 2.2936e-04, 1.9133e-05, 1.9133e-05,
          1.8585e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.8585e-04,
          1.9133e-05, 5.6982e-05, 6.6757e-04, 1.9133e-05, 1.9133e-05, 2.9087e-04,
          1.9133e-05, 1.8585e-04, 1.3390e-03, 1.9133e-05, 2.2364e-04, 2.2936e-04,
          2.2364e-04, 1.9133e-05, 1.9133e-05, 1.8585e-04, 1.8585e-04, 1.9133e-05,
          1.9133e-05, 1.9157e-04, 1.9157e-04, 2.8729e-04, 4.9877e-04, 5.6982e-05,
          1.9133e-05, 2.2936e-04, 1.9133e-05, 3.1710e-04, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0915e-03, 6.3782e-02, 1.9196e-02, 2.1683e-02, 5.9387e-02,
          4.9164e-02, 1.5076e-02, 1.9131e-03, 1.5430e-03, 5.3558e-03, 1.5228e-02,
          3.6945e-03, 1.9159e-03, 2.0123e-03, 2.2087e-03, 2.0943e-03, 2.5330e-03,
          1.7891e-03, 1.5774e-03, 6.6805e-04, 8.5115e-04, 9.5892e-04, 2.3479e-03,
          1.7273e-02, 4.8828e-03, 1.2337e-02, 6.7863e-03, 1.1425e-03, 2.5368e-03,
          2.9106e-03, 3.1662e-03, 1.1005e-03, 2.3117e-03, 2.0428e-03, 1.5802e-03,
          2.6436e-03, 2.1191e-03, 3.0594e-03, 1.9169e-03, 1.0386e-03, 6.1035e-03,
          4.8332e-03, 3.6793e-03, 4.1962e-03, 1.6813e-03, 2.7657e-03, 3.0174e-03,
          1.2512e-03, 9.8991e-04, 6.0959e-03, 2.3270e-03, 2.4090e-03, 1.5993e-03,
          1.3151e-03, 1.5593e-03, 1.1816e-03, 2.3727e-03, 4.1389e-03, 4.0665e-03,
          3.2330e-03, 2.5768e-03, 4.0741e-03, 3.6316e-03, 9.7122e-03, 6.2256e-03,
          4.1687e-02, 1.0857e-02, 3.4657e-03, 2.1172e-03, 1.9417e-03, 2.7924e-03,
          1.6365e-03, 9.2621e-03, 1.5240e-03, 2.9488e-03, 1.3294e-03, 1.0662e-03,
          1.6022e-03, 3.0022e-03, 1.7212e-02, 4.6959e-03, 1.4748e-02, 2.0638e-03,
          1.6880e-03, 6.2828e-03, 2.9583e-03, 9.0866e-03, 1.4992e-02, 1.6876e-02,
          1.8311e-02, 9.0637e-03, 4.5433e-03, 9.7351e-03, 1.6260e-03, 1.3876e-03,
          1.4563e-03, 9.9468e-04, 1.8988e-03, 3.4122e-03, 4.0016e-03, 4.5853e-03,
          1.3466e-02, 1.3420e-02, 8.0414e-03, 3.3474e-03, 3.3665e-03, 3.2158e-03,
          5.6534e-03, 8.3923e-03, 1.5778e-02, 1.1871e-02, 3.3752e-02, 1.5579e-02,
          4.0245e-03, 3.5324e-03, 2.1152e-03, 5.5771e-03, 1.9894e-03, 2.5463e-03,
          3.9215e-03, 5.2872e-03, 3.8300e-03, 1.4620e-03, 4.6806e-03, 6.1111e-03,
          2.0695e-03, 9.8801e-03, 1.9669e-02, 4.0741e-03, 3.5744e-03, 6.0234e-03,
          7.5806e-02, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.1617e-04, 6.8009e-05, 6.8009e-05, 1.9968e-05, 1.9968e-05, 6.8009e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.1617e-04,
          1.9968e-05, 1.9968e-05, 2.2709e-04, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05, 6.8009e-05, 1.9968e-05,
          6.8009e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 6.8009e-05, 6.8128e-05, 6.8128e-05, 6.8009e-05, 1.9968e-05,
          6.8009e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05,
          6.8128e-05, 1.9968e-05, 1.9968e-05, 2.2709e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05, 1.9968e-05, 2.2709e-04,
          6.8128e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05, 1.9968e-05,
          1.9968e-05, 1.1617e-04, 1.9968e-05, 6.8128e-05, 6.8128e-05, 1.1617e-04,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.1617e-04, 6.8128e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8009e-05, 6.8128e-05, 1.7893e-04,
          1.9968e-05, 1.9968e-05, 6.8009e-05, 1.7893e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8128e-05, 6.8009e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.1617e-04, 2.2709e-04, 6.8009e-05,
          1.9968e-05, 1.9968e-05, 2.2709e-04, 1.9968e-05, 6.8009e-05, 1.9968e-05,
          1.9968e-05, 1.1617e-04, 1.9968e-05, 2.2709e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 6.8009e-05, 1.9968e-05,
          6.8128e-05, 6.8009e-05, 2.2709e-04, 2.2709e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.9968e-05, 1.9968e-05, 1.1617e-04, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.1617e-04, 6.8128e-05, 6.8009e-05, 1.9968e-05, 1.9968e-05,
          1.9968e-05, 1.7893e-04, 6.8009e-05, 6.8009e-05, 1.9968e-05, 6.8009e-05,
          2.2709e-04, 1.9968e-05, 1.9968e-05, 1.1617e-04, 6.8009e-05, 6.8009e-05,
          6.8009e-05, 6.8009e-05, 4.1366e-04, 2.2709e-04, 1.9968e-05, 1.9968e-05,
          2.2709e-04, 3.5095e-04, 1.7893e-04, 1.9968e-05, 1.9968e-05, 2.2709e-04,
          1.9968e-05, 4.6182e-04, 2.2781e-04, 6.8128e-05, 1.9968e-05, 1.9968e-05,
          1.7893e-04, 1.9968e-05, 1.1617e-04, 1.7893e-04, 1.9968e-05, 2.2709e-04,
          1.9968e-05, 1.9968e-05, 5.5552e-04, 3.0303e-04, 3.0303e-04, 4.1366e-04,
          3.0279e-04, 1.1617e-04, 1.9968e-05, 1.9968e-05, 6.8128e-05, 6.8128e-05,
          6.8128e-05, 1.7893e-04, 2.5487e-04, 1.9968e-05, 1.7893e-04, 3.0279e-04,
          4.4441e-04, 2.0969e-04, 2.5487e-04, 6.8009e-05, 1.9968e-05, 1.9968e-05,
          1.1617e-04, 1.7893e-04, 1.9968e-05, 4.1366e-04, 1.9968e-05, 1.9968e-05,
          6.8009e-05, 1.9968e-05, 1.9968e-05, 1.9968e-05, 1.1617e-04, 1.3089e-04,
          3.5095e-04, 4.7350e-04, 6.8009e-05, 1.7893e-04, 1.9968e-05, 4.2534e-04,
          3.5095e-04, 1.7893e-04, 7.0858e-04, 1.9968e-05, 1.9968e-05, 6.8009e-05,
          6.6042e-04, 3.0279e-04, 1.7893e-04, 4.1366e-04, 3.6573e-04, 1.9968e-05,
          6.8009e-05, 2.3377e-04, 2.3377e-04, 6.8009e-05, 3.4475e-04, 2.3377e-04,
          1.9968e-05, 2.8205e-04, 6.8009e-05, 5.6458e-04, 2.3377e-04, 6.8009e-05,
          6.8009e-05, 2.3377e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.6637e-03, 6.6956e-02, 2.0554e-02, 3.0380e-02, 5.2673e-02,
          5.0781e-02, 1.7303e-02, 1.6823e-03, 1.3685e-03, 6.8169e-03, 1.6342e-02,
          4.7569e-03, 1.9932e-03, 1.7138e-03, 2.0027e-03, 1.8253e-03, 2.7466e-03,
          2.1954e-03, 1.4143e-03, 4.9543e-04, 6.9141e-04, 8.2636e-04, 3.0003e-03,
          2.1194e-02, 3.7575e-03, 1.4259e-02, 8.1024e-03, 7.9060e-04, 3.0575e-03,
          2.5139e-03, 3.0270e-03, 1.1454e-03, 2.2945e-03, 3.3836e-03, 1.2798e-03,
          2.4204e-03, 1.5602e-03, 3.9864e-03, 2.0638e-03, 7.5483e-04, 7.9117e-03,
          3.8414e-03, 2.7332e-03, 2.9202e-03, 1.7014e-03, 3.6850e-03, 3.6602e-03,
          1.1539e-03, 9.2173e-04, 5.5008e-03, 1.7118e-03, 2.4319e-03, 1.3475e-03,
          8.1873e-04, 1.4143e-03, 7.1621e-04, 2.0161e-03, 3.5763e-03, 3.8834e-03,
          2.3117e-03, 1.8187e-03, 2.1572e-03, 2.2583e-03, 9.3536e-03, 7.4768e-03,
          3.0685e-02, 1.4542e-02, 4.8637e-03, 1.7757e-03, 1.6384e-03, 2.2163e-03,
          1.3866e-03, 1.2177e-02, 1.0662e-03, 2.7122e-03, 8.2350e-04, 7.8011e-04,
          1.2007e-03, 2.2774e-03, 1.6403e-02, 6.2447e-03, 2.1759e-02, 1.6527e-03,
          1.2903e-03, 5.4092e-03, 1.7424e-03, 1.2543e-02, 1.8066e-02, 1.4748e-02,
          3.2288e-02, 1.2871e-02, 2.8152e-03, 1.5228e-02, 1.3027e-03, 1.2970e-03,
          9.8801e-04, 5.7316e-04, 1.8730e-03, 2.8152e-03, 2.9221e-03, 4.2076e-03,
          1.9608e-02, 9.0256e-03, 1.4908e-02, 3.9711e-03, 3.3092e-03, 1.9608e-03,
          4.2686e-03, 5.8937e-03, 2.1118e-02, 5.8746e-03, 2.4109e-02, 1.7166e-02,
          3.1948e-03, 3.6163e-03, 1.6403e-03, 5.8365e-03, 1.7443e-03, 1.5354e-03,
          2.5234e-03, 5.7487e-03, 2.2831e-03, 1.2064e-03, 5.2910e-03, 4.3869e-03,
          1.4782e-03, 8.1940e-03, 1.1154e-02, 3.5038e-03, 3.3054e-03, 4.7836e-03,
          2.5452e-02, 3.9948e-02, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          9.5606e-05, 6.4373e-05, 6.4373e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05, 1.8919e-04,
          2.1517e-05, 2.1517e-05, 1.9586e-04, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 7.6056e-05, 5.2750e-05, 9.5606e-05, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 5.2750e-05, 5.2750e-05, 5.2750e-05, 9.5606e-05, 2.1517e-05,
          2.1517e-05, 6.4373e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 7.6056e-05,
          5.2750e-05, 2.1517e-05, 5.2750e-05, 1.9586e-04, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 1.5819e-04, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05, 2.1517e-05, 3.8528e-04,
          7.6056e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 7.6056e-05, 2.1517e-05,
          2.1517e-05, 9.5606e-05, 2.1517e-05, 5.2750e-05, 1.1891e-04, 2.1255e-04,
          5.2750e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 1.1891e-04, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 7.6056e-05, 5.2750e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 6.4373e-05, 1.1891e-04, 1.6463e-04,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.8944e-04, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 9.5606e-05, 6.4373e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          6.4373e-05, 2.1517e-05, 5.2750e-05, 1.1891e-04, 9.5606e-05, 1.5795e-04,
          2.1517e-05, 2.1517e-05, 9.5606e-05, 2.1517e-05, 1.5795e-04, 2.1517e-05,
          2.1517e-05, 9.5606e-05, 1.1516e-04, 9.5606e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1255e-04, 6.4373e-05,
          7.6056e-05, 6.4373e-05, 1.9586e-04, 1.9586e-04, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 1.8919e-04, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 9.5606e-05, 7.6056e-05, 3.8218e-04, 2.1517e-05, 2.1517e-05,
          9.5606e-05, 4.5109e-04, 6.4373e-05, 1.1891e-04, 2.1517e-05, 2.1517e-05,
          4.8232e-04, 2.1517e-05, 1.1516e-04, 7.6056e-05, 6.4373e-05, 6.4373e-05,
          2.5821e-04, 2.1517e-05, 2.5821e-04, 2.8944e-04, 1.1516e-04, 2.5821e-04,
          2.8944e-04, 1.8919e-04, 6.4373e-05, 2.1517e-05, 4.4823e-05, 3.8886e-04,
          2.1517e-05, 5.0545e-04, 1.9598e-04, 7.6056e-05, 2.1517e-05, 2.5821e-04,
          2.5821e-04, 5.2750e-05, 2.1255e-04, 2.5821e-04, 2.1517e-05, 1.9586e-04,
          2.1517e-05, 5.2750e-05, 2.8944e-04, 5.2750e-05, 1.4639e-04, 2.5821e-04,
          1.5795e-04, 9.5606e-05, 2.1517e-05, 2.1517e-05, 7.6056e-05, 1.8060e-04,
          4.0555e-04, 1.6463e-04, 2.1517e-05, 2.1517e-05, 3.8910e-04, 1.5795e-04,
          1.1492e-03, 3.1710e-04, 1.4639e-04, 2.1517e-05, 5.2750e-05, 5.2750e-05,
          5.2750e-05, 2.5821e-04, 5.2750e-05, 1.5795e-04, 2.1517e-05, 2.1517e-05,
          6.4373e-05, 2.1517e-05, 6.4373e-05, 5.2750e-05, 4.6706e-04, 2.1517e-05,
          2.1255e-04, 3.8886e-04, 2.1517e-05, 1.6463e-04, 1.1516e-04, 3.5763e-04,
          1.6963e-04, 2.5821e-04, 4.8232e-04, 5.2750e-05, 2.1517e-05, 2.5821e-04,
          4.5109e-04, 3.8218e-04, 1.2165e-04, 2.5821e-04, 2.5821e-04, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 9.4891e-05, 2.5392e-04, 9.5606e-05,
          2.1517e-05, 9.5606e-05, 6.4373e-05, 2.4378e-04, 2.1517e-05, 6.4373e-05,
          2.1517e-05, 2.1517e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9242e-03, 6.0669e-02, 1.8631e-02, 2.0187e-02, 5.7587e-02,
          6.3904e-02, 1.5762e-02, 1.0521e-02, 3.6926e-03, 1.0559e-02, 2.3972e-02,
          5.8937e-03, 7.9117e-03, 1.7672e-03, 1.4587e-02, 1.0727e-02, 5.3596e-03,
          4.6997e-03, 6.2294e-03, 4.5061e-04, 5.8365e-04, 7.5722e-04, 8.7585e-03,
          2.4231e-02, 8.5907e-03, 1.2619e-02, 7.9422e-03, 8.4114e-04, 6.2714e-03,
          2.3098e-03, 1.4877e-02, 1.5497e-03, 6.8359e-03, 1.5701e-02, 2.7676e-03,
          2.8763e-03, 8.7929e-04, 5.6267e-03, 6.8245e-03, 6.7091e-04, 3.8834e-03,
          1.3170e-03, 1.0948e-03, 1.0891e-03, 9.7084e-04, 1.0033e-02, 6.9847e-03,
          1.2121e-03, 8.8310e-04, 6.6032e-03, 2.4834e-03, 5.2986e-03, 3.6545e-03,
          1.1396e-03, 2.3861e-03, 6.4135e-04, 8.9169e-04, 2.7256e-03, 1.3266e-03,
          1.3580e-03, 1.7366e-03, 1.5945e-03, 2.2717e-03, 2.0943e-03, 1.3596e-02,
          3.5400e-03, 4.2038e-03, 4.2534e-03, 7.1239e-04, 6.7854e-04, 9.2316e-04,
          9.0170e-04, 1.5976e-02, 1.1082e-03, 6.4316e-03, 1.1292e-03, 1.2045e-03,
          1.8511e-03, 7.9823e-04, 1.7157e-03, 5.5008e-03, 2.2926e-03, 1.2856e-03,
          1.0042e-03, 9.3994e-03, 1.2646e-03, 1.7044e-02, 2.8687e-03, 2.5558e-03,
          2.5040e-02, 7.2556e-03, 1.2550e-03, 1.0635e-02, 1.0109e-03, 9.7513e-04,
          1.2331e-03, 6.0749e-04, 5.9547e-03, 3.4218e-03, 9.0027e-04, 1.3437e-03,
          1.5854e-02, 3.5839e-03, 3.0460e-03, 2.4261e-03, 1.2970e-02, 1.6012e-03,
          6.9313e-03, 1.3647e-03, 9.4070e-03, 1.5755e-03, 3.9673e-03, 4.4861e-03,
          9.3765e-03, 1.8721e-03, 1.2283e-03, 1.3962e-02, 1.2579e-03, 1.4706e-03,
          2.5139e-03, 4.4327e-03, 7.3576e-04, 1.0231e-02, 2.5543e-02, 8.9693e-04,
          6.6280e-04, 1.0204e-03, 1.4858e-03, 9.2010e-03, 1.1436e-02, 9.6207e-03,
          4.2953e-03, 4.9858e-03, 7.6782e-02, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          7.6830e-05, 1.3816e-04, 7.6830e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 8.0407e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04,
          1.9073e-05, 1.9073e-05, 2.9421e-04, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 7.6830e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 8.0407e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 2.9421e-04, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.3816e-04, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 2.9421e-04,
          8.0407e-05, 1.9073e-05, 1.9073e-05, 8.0407e-05, 1.9073e-05, 8.0407e-05,
          1.9073e-05, 2.9421e-04, 1.9073e-05, 1.9073e-05, 7.6830e-05, 1.3816e-04,
          8.0407e-05, 1.2970e-04, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.2970e-04, 1.9073e-05, 1.9073e-05, 2.6035e-04, 1.9073e-05, 8.0407e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 3.6311e-04,
          1.9073e-05, 1.9073e-05, 7.6830e-05, 7.6830e-05, 1.3816e-04, 2.9421e-04,
          1.7202e-04, 1.9073e-05, 1.7202e-04, 2.9111e-04, 1.9073e-05, 1.9073e-05,
          1.2970e-04, 1.9073e-05, 1.9073e-05, 1.9073e-05, 7.6830e-05, 1.7202e-04,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04, 1.3816e-04, 1.9073e-05,
          8.0407e-05, 8.0407e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05,
          1.7202e-04, 2.6035e-04, 2.8253e-04, 1.3816e-04, 1.7202e-04, 8.0407e-05,
          8.0407e-05, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04, 7.6830e-05,
          6.0415e-04, 7.6830e-05, 2.9421e-04, 2.9421e-04, 1.2970e-04, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.7202e-04, 7.6830e-05, 1.7202e-04, 1.9073e-05,
          1.9073e-05, 1.9073e-05, 1.3816e-04, 1.3816e-04, 1.9073e-05, 1.9073e-05,
          7.6830e-05, 2.9421e-04, 7.6830e-05, 7.6830e-05, 1.9073e-05, 7.6830e-05,
          3.8576e-04, 7.6830e-05, 3.8576e-04, 8.0407e-05, 7.6830e-05, 1.9073e-05,
          3.8576e-04, 1.9073e-05, 3.8576e-04, 3.8576e-04, 1.9073e-05, 1.7202e-04,
          2.3282e-04, 2.9111e-04, 2.3282e-04, 1.7202e-04, 1.9073e-05, 2.9421e-04,
          1.9073e-05, 2.9111e-04, 2.9421e-04, 8.0407e-05, 1.9073e-05, 3.8576e-04,
          6.2704e-04, 1.9073e-05, 1.9073e-05, 4.9639e-04, 1.9073e-05, 4.0483e-04,
          1.9073e-05, 1.9073e-05, 4.4703e-04, 8.0407e-05, 5.3215e-04, 3.8576e-04,
          3.4046e-04, 2.3329e-04, 1.7202e-04, 1.7202e-04, 1.3816e-04, 8.0407e-05,
          3.7932e-04, 1.9073e-05, 1.9073e-05, 1.9073e-05, 1.3816e-04, 4.7088e-04,
          2.9111e-04, 1.3816e-04, 4.1318e-04, 1.9073e-05, 7.6830e-05, 1.9073e-05,
          7.6830e-05, 4.0483e-04, 1.9073e-05, 8.1491e-04, 1.9073e-05, 1.9073e-05,
          1.7202e-04, 1.7202e-04, 3.8576e-04, 1.9073e-05, 2.9111e-04, 1.7202e-04,
          7.2956e-04, 4.2081e-04, 7.6830e-05, 7.6830e-05, 1.7202e-04, 9.4986e-04,
          5.3215e-04, 4.9639e-04, 4.0483e-04, 1.9073e-05, 1.9073e-05, 4.7088e-04,
          1.1024e-03, 3.8576e-04, 1.9073e-05, 3.8576e-04, 3.8576e-04, 1.9073e-05,
          7.6830e-05, 7.6830e-05, 7.6830e-05, 7.6830e-05, 5.8556e-04, 1.3816e-04,
          1.9073e-05, 7.6830e-05, 7.6830e-05, 1.9073e-05, 1.9073e-05, 7.6830e-05,
          7.6830e-05, 1.9073e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9052e-03, 6.4880e-02, 1.8250e-02, 2.0370e-02, 4.5410e-02,
          4.5868e-02, 1.5373e-02, 1.6184e-03, 1.3552e-03, 4.9400e-03, 1.1345e-02,
          3.8757e-03, 1.6251e-03, 1.5011e-03, 1.9302e-03, 1.8034e-03, 2.1687e-03,
          1.7424e-03, 1.7176e-03, 4.8590e-04, 6.7091e-04, 8.2111e-04, 2.7332e-03,
          1.7624e-02, 4.5166e-03, 1.2367e-02, 6.4697e-03, 7.2289e-04, 2.3155e-03,
          2.3212e-03, 3.0251e-03, 9.2745e-04, 2.1820e-03, 3.0556e-03, 1.5154e-03,
          2.2488e-03, 1.4820e-03, 4.1389e-03, 1.9855e-03, 7.5912e-04, 9.6130e-03,
          5.8937e-03, 3.5763e-03, 4.9171e-03, 2.1172e-03, 3.0556e-03, 3.6602e-03,
          1.1053e-03, 8.9836e-04, 4.9896e-03, 1.8587e-03, 2.4986e-03, 1.6985e-03,
          8.3590e-04, 1.4887e-03, 6.9094e-04, 1.8454e-03, 3.3340e-03, 2.9297e-03,
          2.0504e-03, 1.7624e-03, 2.7180e-03, 2.7599e-03, 1.3710e-02, 1.0048e-02,
          3.2562e-02, 1.4359e-02, 6.1340e-03, 2.9469e-03, 2.7618e-03, 4.1466e-03,
          1.8845e-03, 1.0391e-02, 1.0405e-03, 3.0479e-03, 8.9359e-04, 9.0647e-04,
          1.2426e-03, 2.2182e-03, 1.1452e-02, 5.1422e-03, 1.2634e-02, 1.5736e-03,
          1.2331e-03, 7.1411e-03, 2.0790e-03, 1.2192e-02, 2.2202e-02, 1.9440e-02,
          2.9114e-02, 1.5213e-02, 9.3918e-03, 1.4412e-02, 2.7390e-03, 1.6317e-03,
          8.9693e-04, 5.7459e-04, 2.8286e-03, 2.8973e-03, 2.7370e-03, 3.7041e-03,
          1.5671e-02, 6.6910e-03, 1.0674e-02, 3.5992e-03, 3.1929e-03, 2.0390e-03,
          5.7755e-03, 9.9716e-03, 2.3193e-02, 1.0445e-02, 2.0737e-02, 2.0737e-02,
          5.0583e-03, 7.1716e-03, 1.7910e-03, 6.9885e-03, 2.3422e-03, 1.4400e-03,
          2.2736e-03, 5.6381e-03, 2.3251e-03, 1.2684e-03, 5.3444e-03, 6.5765e-03,
          1.6594e-03, 8.7128e-03, 1.1795e-02, 3.5973e-03, 3.8185e-03, 6.4926e-03,
          2.1469e-02, 2.1561e-02, 2.4185e-03, 3.1677e-02, 1.2934e-05, 1.2934e-05,
          9.5367e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 5.8532e-05, 9.5367e-05,
          1.2934e-05, 1.2934e-05, 9.5367e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05, 1.2934e-05,
          5.8532e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 9.5367e-05, 9.5367e-05, 4.9710e-05, 4.9710e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05,
          4.9710e-05, 1.2934e-05, 1.2934e-05, 3.0541e-04, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 5.8532e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05, 1.2934e-05, 9.5367e-05,
          4.9710e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          5.8532e-05, 9.5367e-05, 1.2934e-05, 9.5367e-05, 9.5367e-05, 9.5367e-05,
          4.9710e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05, 9.5367e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 4.9710e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 5.8532e-05, 9.5367e-05, 5.8532e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 5.8532e-05, 1.2934e-05, 5.8532e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          5.8532e-05, 1.2934e-05, 4.9710e-05, 9.5367e-05, 9.5367e-05, 5.8532e-05,
          1.2934e-05, 1.2934e-05, 3.0541e-04, 1.2934e-05, 9.2316e-04, 1.2934e-05,
          1.2934e-05, 9.5367e-05, 1.2934e-05, 9.5367e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05, 2.6870e-04,
          4.9710e-05, 5.8532e-05, 9.5367e-05, 9.5367e-05, 1.2934e-05, 5.8532e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 3.0541e-04, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 9.5367e-05, 4.9710e-05, 3.5214e-04, 1.2934e-05, 1.2934e-05,
          5.8532e-05, 5.2547e-04, 2.6870e-04, 3.5214e-04, 1.2934e-05, 1.2934e-05,
          3.5214e-04, 5.8532e-05, 1.2934e-05, 9.5367e-05, 5.8532e-05, 2.6870e-04,
          2.6870e-04, 1.2934e-05, 1.1797e-03, 9.5367e-05, 1.2934e-05, 2.2304e-04,
          3.5214e-04, 9.5367e-05, 5.8532e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05,
          1.2934e-05, 3.5214e-04, 9.6262e-05, 1.6785e-04, 1.2934e-05, 3.1543e-04,
          9.2316e-04, 9.5367e-05, 9.5367e-05, 5.8532e-05, 1.2934e-05, 9.5367e-05,
          1.2934e-05, 4.9710e-05, 3.0541e-04, 4.9710e-05, 1.2064e-03, 5.8532e-05,
          1.2934e-05, 4.9710e-05, 1.2934e-05, 1.2934e-05, 1.2231e-04, 4.9710e-05,
          5.0640e-04, 4.9710e-05, 1.2934e-05, 1.2934e-05, 9.5367e-05, 5.8532e-05,
          2.4414e-04, 1.9860e-04, 7.0429e-04, 1.2934e-05, 4.9710e-05, 5.8532e-05,
          9.5367e-05, 9.5367e-05, 4.9710e-05, 5.8532e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 2.6870e-04, 4.9710e-05, 1.2231e-04, 1.2934e-05,
          9.5367e-05, 3.5214e-04, 1.2934e-05, 3.1543e-04, 1.2934e-05, 3.1543e-04,
          4.9710e-05, 1.2934e-05, 3.5214e-04, 4.9710e-05, 1.2934e-05, 1.1797e-03,
          3.1543e-04, 7.4959e-04, 1.2934e-05, 2.6870e-04, 2.2304e-04, 1.2934e-05,
          5.8532e-05, 5.8532e-05, 1.2934e-05, 9.1255e-05, 2.9445e-04, 4.9710e-05,
          1.2934e-05, 2.9445e-04, 2.9445e-04, 6.6900e-04, 2.9445e-04, 2.9445e-04,
          5.8532e-05, 1.2934e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.2365e-03, 6.2042e-02, 1.8845e-02, 2.5024e-02, 4.6722e-02,
          4.8065e-02, 1.6266e-02, 1.7986e-03, 1.6251e-03, 7.8964e-03, 1.3542e-02,
          5.4893e-03, 2.1381e-03, 1.3084e-03, 3.5076e-03, 1.9684e-03, 3.1891e-03,
          2.3251e-03, 1.4114e-03, 5.4693e-04, 9.8991e-04, 9.7513e-04, 3.4904e-03,
          2.1347e-02, 6.1264e-03, 1.3359e-02, 7.1678e-03, 7.5769e-04, 3.4504e-03,
          1.8854e-03, 3.0651e-03, 1.1673e-03, 4.3793e-03, 5.1651e-03, 2.0924e-03,
          4.3869e-03, 1.7538e-03, 7.9193e-03, 3.8509e-03, 7.7009e-04, 1.3641e-02,
          5.6152e-03, 3.7479e-03, 3.7289e-03, 2.8267e-03, 3.7098e-03, 7.8659e-03,
          9.8610e-04, 9.3985e-04, 5.1193e-03, 2.6321e-03, 2.4223e-03, 1.9646e-03,
          8.0109e-04, 1.5135e-03, 6.9237e-04, 3.9787e-03, 4.0741e-03, 4.7874e-03,
          2.2373e-03, 2.1515e-03, 2.4509e-03, 4.5738e-03, 1.0803e-02, 1.8967e-02,
          1.1131e-02, 8.4457e-03, 5.5504e-03, 3.0079e-03, 2.6016e-03, 2.6131e-03,
          1.9913e-03, 1.4709e-02, 1.1044e-03, 2.6684e-03, 7.1192e-04, 8.2970e-04,
          1.2970e-03, 4.5891e-03, 9.2087e-03, 9.0485e-03, 9.3842e-03, 1.9102e-03,
          1.3828e-03, 6.9923e-03, 2.0142e-03, 1.3710e-02, 7.7820e-03, 6.9237e-03,
          3.2745e-02, 2.0920e-02, 3.4904e-03, 1.7731e-02, 2.4738e-03, 1.9646e-03,
          1.0233e-03, 5.7220e-04, 2.7618e-03, 5.3444e-03, 4.4518e-03, 6.2981e-03,
          2.0630e-02, 2.9049e-03, 7.2517e-03, 8.0109e-03, 6.4430e-03, 2.6646e-03,
          6.2561e-03, 1.0628e-02, 3.0472e-02, 5.1193e-03, 6.8474e-03, 1.0529e-02,
          5.0850e-03, 9.6130e-03, 2.2202e-03, 1.3336e-02, 2.9697e-03, 1.5316e-03,
          2.5387e-03, 8.9111e-03, 1.9283e-03, 1.0700e-03, 8.9417e-03, 3.7022e-03,
          2.8172e-03, 3.9635e-03, 4.9477e-03, 6.2752e-03, 4.2152e-03, 6.5041e-03,
          1.0147e-02, 9.2850e-03, 2.0962e-03, 5.7297e-03, 3.2288e-02, 1.4901e-05,
          9.7275e-05, 9.7275e-05, 6.2585e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 6.2585e-05, 9.7275e-05,
          1.4901e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 4.9531e-05, 4.9531e-05, 9.7275e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 4.9531e-05, 4.9531e-05, 4.9531e-05, 4.9531e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05,
          4.9531e-05, 1.4901e-05, 1.4901e-05, 2.7585e-04, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 2.4152e-04, 6.2585e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05, 1.4901e-05, 9.7275e-05,
          1.1683e-04, 1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05, 1.4901e-05,
          1.4901e-05, 9.7275e-05, 1.4901e-05, 4.9531e-05, 9.7275e-05, 9.7275e-05,
          4.9531e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.6451e-04, 4.9531e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 2.4676e-04,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.1683e-04,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 6.2585e-05, 9.7275e-05, 2.7585e-04,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 4.9531e-05, 1.6451e-04, 9.7275e-05, 6.2585e-05,
          1.4901e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05, 6.2585e-05, 1.4901e-05,
          1.4901e-05, 4.9531e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 1.4901e-05, 2.9445e-04, 2.4128e-04,
          1.1683e-04, 1.4901e-05, 9.7275e-05, 9.7275e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 9.7275e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 9.7275e-05, 4.9531e-05, 3.2020e-04, 1.4901e-05, 1.4901e-05,
          6.2585e-05, 6.6137e-04, 2.4128e-04, 3.2020e-04, 1.4901e-05, 1.4901e-05,
          9.7275e-05, 6.2585e-05, 1.4901e-05, 1.1683e-04, 6.2585e-05, 1.4901e-05,
          6.2585e-05, 1.4901e-05, 2.4128e-04, 9.7275e-05, 1.4901e-05, 1.4901e-05,
          2.7585e-04, 2.9445e-04, 6.2585e-05, 1.4901e-05, 1.4901e-05, 9.7275e-05,
          1.4901e-05, 5.8460e-04, 2.7609e-04, 1.1683e-04, 1.4901e-05, 6.2585e-05,
          2.9445e-04, 4.9531e-05, 3.8743e-04, 2.5988e-04, 1.4901e-05, 9.7275e-05,
          1.4901e-05, 4.9531e-05, 2.7585e-04, 4.9531e-05, 6.7425e-04, 2.5988e-04,
          2.1207e-04, 4.9531e-05, 1.4901e-05, 1.4901e-05, 1.1683e-04, 1.1683e-04,
          8.9693e-04, 1.4901e-05, 1.4901e-05, 1.4901e-05, 9.7513e-05, 6.2585e-05,
          5.8460e-04, 3.8743e-04, 6.7425e-04, 1.4901e-05, 4.9531e-05, 1.4901e-05,
          4.9531e-05, 2.7585e-04, 4.9531e-05, 6.2585e-05, 1.4901e-05, 1.4901e-05,
          1.4901e-05, 1.4901e-05, 1.4901e-05, 4.9531e-05, 1.1683e-04, 1.4901e-05,
          7.6342e-04, 3.2020e-04, 1.4901e-05, 6.2585e-05, 2.1207e-04, 2.8563e-04,
          4.9531e-05, 2.1207e-04, 3.2020e-04, 4.9531e-05, 1.4901e-05, 3.7909e-04,
          6.6137e-04, 2.4319e-03, 1.4901e-05, 2.4128e-04, 1.9360e-04, 1.4901e-05,
          6.2585e-05, 1.4901e-05, 1.4901e-05, 2.8586e-04, 6.2585e-05, 4.9531e-05,
          1.4901e-05, 3.2806e-04, 6.2585e-05, 3.3355e-04, 6.2585e-05, 6.2585e-05,
          1.4901e-05, 1.4901e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.2021e-03, 6.0303e-02, 2.3056e-02, 1.6602e-02, 7.2327e-02,
          4.2084e-02, 1.8539e-02, 1.1244e-03, 9.4509e-04, 2.7103e-03, 5.7030e-03,
          1.9169e-03, 1.3075e-03, 2.3842e-03, 1.8892e-03, 1.2503e-03, 2.0294e-03,
          1.3704e-03, 1.0061e-03, 1.9064e-03, 2.0390e-03, 2.4166e-03, 1.8864e-03,
          7.6866e-03, 4.3793e-03, 1.3435e-02, 3.2501e-03, 1.9217e-03, 1.6260e-03,
          2.4624e-03, 1.6050e-03, 1.0052e-03, 2.0924e-03, 1.5697e-03, 1.2188e-03,
          3.4580e-03, 3.0193e-03, 2.8629e-03, 1.6222e-03, 2.1343e-03, 1.0902e-02,
          1.9424e-02, 1.0048e-02, 1.4313e-02, 5.2719e-03, 1.8654e-03, 4.0131e-03,
          2.0638e-03, 3.1738e-03, 1.0094e-02, 1.7138e-03, 1.3933e-03, 1.1034e-03,
          1.3504e-03, 1.2207e-03, 1.9093e-03, 3.7117e-03, 3.1986e-03, 4.4518e-03,
          4.0054e-03, 3.3607e-03, 4.3640e-03, 4.2305e-03, 1.5884e-02, 9.4223e-03,
          1.1124e-02, 4.2076e-03, 3.1376e-03, 1.4824e-02, 1.7090e-02, 1.0483e-02,
          7.4234e-03, 4.4708e-03, 1.9178e-03, 1.7319e-03, 1.3313e-03, 9.5892e-04,
          1.7862e-03, 4.2305e-03, 5.6305e-03, 3.2272e-03, 4.5395e-03, 3.1643e-03,
          3.2425e-03, 5.7182e-03, 4.0054e-03, 4.6577e-03, 4.4899e-03, 9.1705e-03,
          8.8577e-03, 8.0872e-03, 2.9205e-02, 6.4926e-03, 1.3756e-02, 9.2850e-03,
          2.0809e-03, 1.4935e-03, 1.3676e-03, 3.0594e-03, 5.7144e-03, 3.6392e-03,
          4.3106e-03, 2.5291e-03, 2.0809e-03, 4.1504e-03, 3.3092e-03, 4.4136e-03,
          5.9395e-03, 2.8152e-02, 9.9335e-03, 1.4870e-02, 7.7209e-03, 6.1302e-03,
          5.4054e-03, 2.3956e-02, 6.5117e-03, 7.4425e-03, 1.6266e-02, 5.4703e-03,
          4.5280e-03, 3.5534e-03, 3.7518e-03, 8.9645e-04, 3.5820e-03, 2.9316e-03,
          4.7455e-03, 4.5395e-03, 4.9591e-03, 3.5248e-03, 2.1915e-03, 6.1264e-03,
          8.6670e-03, 5.0163e-03, 1.3447e-03, 5.1346e-03, 9.1019e-03, 5.7953e-02,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05, 8.2493e-05,
          1.5378e-05, 1.5378e-05, 2.0337e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.3626e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.5378e-05,
          1.3626e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05,
          8.2493e-05, 1.5378e-05, 1.5378e-05, 2.0337e-04, 1.3626e-04, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 1.3626e-04, 2.0337e-04,
          8.2493e-05, 1.5378e-05, 1.5378e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 8.2493e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5676e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 8.2493e-05, 1.5378e-05,
          1.3626e-04, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.5676e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 2.0337e-04, 2.0337e-04, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.3626e-04, 2.0337e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2552e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 6.9809e-04, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.5378e-05,
          2.0337e-04, 8.2493e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05,
          1.5378e-05, 1.5438e-05, 8.2493e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 8.2493e-05, 1.5378e-05, 1.3626e-04, 8.2493e-05,
          1.5378e-05, 1.5378e-05, 2.0337e-04, 1.5378e-05, 8.2493e-05, 1.5378e-05,
          1.5378e-05, 2.0337e-04, 1.5378e-05, 1.5378e-05, 2.0337e-04, 8.2493e-05,
          8.2493e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.2493e-05, 1.5378e-05,
          2.0337e-04, 2.0349e-04, 1.5378e-05, 1.5378e-05, 2.0337e-04, 1.5378e-05,
          8.2493e-05, 1.5378e-05, 8.2493e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05,
          1.3626e-04, 1.5378e-05, 1.5378e-05, 8.2493e-05, 8.2493e-05, 1.5378e-05,
          1.3626e-04, 8.2493e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          2.0337e-04, 1.5378e-05, 8.2493e-05, 8.2493e-05, 1.3626e-04, 1.3626e-04,
          1.5378e-05, 1.3626e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.3626e-04, 1.3626e-04, 1.3626e-04, 1.1188e-04, 1.3626e-04, 1.3626e-04,
          1.5378e-05, 1.3626e-04, 1.3626e-04, 3.6573e-04, 1.5378e-05, 1.3626e-04,
          1.5378e-05, 1.3626e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.8659e-03, 6.2927e-02, 2.6108e-02, 2.6535e-02, 5.0446e-02,
          4.9866e-02, 1.5305e-02, 3.1338e-03, 2.5768e-03, 6.1150e-03, 7.2403e-03,
          3.1834e-03, 1.4870e-02, 4.5853e-03, 5.3673e-03, 3.5133e-03, 3.0670e-02,
          1.4587e-02, 1.9817e-03, 1.9341e-03, 2.2812e-03, 3.6755e-03, 1.1681e-02,
          9.0408e-03, 4.2801e-03, 1.2306e-02, 1.3672e-02, 2.9240e-03, 2.9648e-02,
          6.9962e-03, 6.1722e-03, 8.7814e-03, 1.1292e-02, 5.5923e-03, 4.3526e-03,
          3.9864e-03, 1.0424e-03, 8.6441e-03, 6.8970e-03, 2.3975e-03, 7.1449e-03,
          1.8272e-03, 1.4706e-03, 1.3733e-03, 1.7729e-03, 3.0029e-02, 1.4313e-02,
          3.6430e-03, 2.9011e-03, 6.2790e-03, 4.6501e-03, 4.0703e-03, 3.2635e-03,
          9.3508e-04, 1.9369e-03, 9.6607e-04, 1.4849e-03, 1.6632e-03, 1.5192e-03,
          1.3456e-03, 8.0185e-03, 2.1324e-03, 4.7493e-03, 2.4815e-03, 1.4282e-02,
          1.8053e-03, 1.8864e-03, 3.3073e-03, 1.0185e-03, 9.0170e-04, 1.1244e-03,
          1.3723e-03, 1.5701e-02, 1.3046e-03, 4.6959e-03, 7.8678e-04, 8.8024e-04,
          1.7767e-03, 1.3876e-03, 1.2894e-03, 3.7785e-03, 1.2760e-03, 1.0157e-03,
          9.4509e-04, 7.5951e-03, 1.6375e-03, 5.2338e-03, 1.2074e-03, 1.1415e-03,
          9.2926e-03, 8.8959e-03, 1.0595e-03, 5.8479e-03, 1.2846e-03, 1.4486e-03,
          3.7899e-03, 7.1621e-04, 5.6725e-03, 8.3466e-03, 1.1311e-03, 1.8034e-03,
          6.6109e-03, 1.1730e-03, 1.2531e-03, 2.3708e-03, 6.7101e-03, 6.8932e-03,
          5.6458e-03, 2.1477e-03, 9.2697e-03, 1.1501e-03, 1.4372e-03, 2.1229e-03,
          2.8934e-03, 3.9177e-03, 4.8676e-03, 1.5373e-02, 1.8787e-03, 4.0245e-03,
          7.1907e-03, 5.6076e-03, 7.0715e-04, 1.8082e-03, 8.3542e-03, 6.1655e-04,
          8.8692e-04, 8.5354e-04, 1.1702e-03, 7.3166e-03, 8.1177e-03, 6.1111e-03,
          2.6608e-03, 1.6479e-03, 1.8940e-03, 1.2798e-03, 4.8523e-03, 1.1902e-03,
          9.3567e-02, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.1778e-04, 6.2108e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.5771e-04, 1.3947e-05, 1.1724e-04, 6.2108e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.5771e-04,
          1.1724e-04, 1.4467e-03, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.5771e-04, 1.3947e-05, 6.2108e-05, 1.1724e-04, 1.3947e-05, 1.3947e-05,
          1.1724e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05, 2.0635e-04, 6.2108e-05,
          6.9141e-05, 1.3037e-03, 1.3947e-05, 1.3947e-05, 2.0576e-04, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.5771e-04, 2.0576e-04, 6.9141e-05,
          2.0576e-04, 1.5771e-04, 1.3947e-05, 6.2108e-05, 1.1724e-04, 1.3947e-05,
          6.2108e-05, 1.3947e-05, 1.3947e-05, 1.1778e-04, 1.3947e-05, 2.0576e-04,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 6.9141e-05, 1.3947e-05, 6.9141e-05,
          1.3947e-05, 1.3947e-05, 6.9141e-05, 2.0576e-04, 6.2108e-05, 1.3947e-05,
          2.5535e-04, 2.7275e-04, 1.3947e-05, 2.4772e-04, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.5771e-04, 6.2644e-05, 1.3947e-05, 1.3947e-05, 2.0957e-04,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 2.5535e-04, 1.3947e-05, 6.2108e-05, 2.5535e-04, 6.9141e-05,
          4.5967e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.6804e-03, 1.3947e-05, 1.3947e-05, 1.3947e-05, 6.2644e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.1724e-04, 1.3947e-05,
          4.6015e-04, 2.0635e-04, 6.2108e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.1724e-04, 1.3947e-05, 5.5838e-04,
          1.1724e-04, 2.1815e-04, 2.1815e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          3.9530e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 3.6192e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05, 2.1815e-04,
          1.3947e-05, 1.3947e-05, 1.1724e-04, 1.3947e-05, 3.6192e-04, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 6.2108e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.1724e-04, 1.3947e-05, 1.5771e-04,
          4.1008e-04, 4.5228e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.3947e-05, 1.1724e-04,
          1.3947e-05, 1.3947e-05, 2.0576e-04, 1.3947e-05, 1.3947e-05, 1.3947e-05,
          5.9938e-04, 1.3947e-05, 6.2108e-05, 1.3947e-05, 1.5771e-04, 2.0576e-04,
          1.5771e-04, 1.3947e-05, 1.3947e-05, 2.0576e-04, 2.4772e-04, 1.3947e-05,
          3.6192e-04, 1.1724e-04, 1.3947e-05, 1.3947e-05, 2.0576e-04, 1.1724e-04,
          2.4772e-04, 1.5771e-04, 1.1724e-04, 1.5771e-04, 2.0576e-04, 3.4332e-04,
          4.1008e-04, 2.0576e-04, 4.5896e-04, 1.1724e-04, 6.9141e-05, 1.3947e-05,
          2.0576e-04, 1.5771e-04, 1.3947e-05, 4.5037e-04, 1.3947e-05, 1.3947e-05,
          6.9141e-05, 2.1815e-04, 1.3947e-05, 1.3947e-05, 3.2139e-04, 1.3947e-05,
          1.6747e-03, 2.0576e-04, 4.5228e-04, 1.3947e-05, 1.3947e-05, 1.5771e-04,
          5.9557e-04, 3.0661e-04, 2.0576e-04, 1.5771e-04, 1.5771e-04, 6.9141e-05,
          3.6192e-04, 2.1210e-03, 1.3947e-05, 1.3947e-05, 1.0246e-04, 1.3947e-05,
          6.9141e-05, 4.0412e-04, 3.1567e-04, 1.3947e-05, 6.9141e-05, 1.3947e-05,
          1.3947e-05, 1.1724e-04, 3.1567e-04, 4.7684e-04, 4.7684e-04, 4.7684e-04,
          6.9141e-05, 6.9141e-05], dtype=torch.float16),
  tensor([0.0000e+00, 9.5367e-03, 6.1615e-02, 2.7390e-02, 2.8946e-02, 6.6040e-02,
          4.2725e-02, 2.2507e-02, 1.9169e-03, 1.4019e-03, 2.2964e-03, 3.5877e-03,
          1.4286e-03, 7.7019e-03, 1.8463e-02, 2.3632e-03, 1.9512e-03, 2.9968e-02,
          1.3969e-02, 1.2035e-03, 4.1122e-03, 3.0956e-03, 5.5885e-03, 6.9656e-03,
          4.3182e-03, 3.0804e-03, 1.7853e-02, 7.6942e-03, 5.5847e-03, 1.1986e-02,
          3.2349e-02, 2.8915e-03, 6.4583e-03, 3.5191e-03, 1.6994e-03, 2.0618e-03,
          2.3975e-03, 1.2884e-03, 3.6430e-03, 2.2984e-03, 6.6147e-03, 5.2109e-03,
          3.2921e-03, 2.4719e-03, 2.4128e-03, 1.9798e-03, 1.2413e-02, 6.2637e-03,
          9.4604e-03, 7.7934e-03, 1.1108e-02, 2.9430e-03, 2.1915e-03, 1.4973e-03,
          1.3151e-03, 1.5898e-03, 1.7033e-03, 1.2999e-03, 2.0294e-03, 1.6308e-03,
          1.8425e-03, 2.1912e-02, 3.1967e-03, 3.9978e-03, 3.0365e-03, 7.3395e-03,
          3.6449e-03, 1.5287e-03, 2.2526e-03, 2.0618e-03, 1.8978e-03, 2.2850e-03,
          2.7752e-03, 4.5395e-03, 1.9064e-03, 2.8191e-03, 1.2627e-03, 9.6273e-04,
          2.1458e-03, 1.4658e-03, 1.7853e-03, 1.6184e-03, 1.7176e-03, 1.3046e-03,
          1.4687e-03, 5.7487e-03, 3.1624e-03, 2.1515e-03, 1.7748e-03, 2.8362e-03,
          3.5973e-03, 3.3493e-03, 2.2411e-03, 2.7847e-03, 2.5940e-03, 3.1223e-03,
          9.4833e-03, 1.2474e-03, 2.4471e-03, 4.6120e-03, 1.5097e-03, 1.4715e-03,
          2.4700e-03, 1.1883e-03, 9.9564e-04, 1.7595e-03, 2.7657e-03, 2.3880e-02,
          4.7035e-03, 3.6659e-03, 4.7340e-03, 2.4567e-03, 2.6836e-03, 2.3594e-03,
          2.2659e-03, 5.3101e-03, 2.5360e-02, 7.0839e-03, 4.7226e-03, 1.0147e-02,
          3.7048e-02, 2.5444e-03, 1.0805e-03, 1.1463e-03, 3.2082e-03, 8.9025e-04,
          1.0872e-03, 1.5574e-03, 1.7405e-03, 3.3150e-03, 3.7098e-03, 5.1842e-03,
          3.8738e-03, 1.9932e-03, 1.1063e-03, 1.3924e-03, 2.8305e-03, 2.9221e-03,
          2.6352e-02, 7.6477e-02, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 5.5254e-05,
          1.3292e-05, 5.5254e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          6.4731e-05, 1.3292e-05, 1.0669e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 6.4731e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.0669e-04, 6.4731e-05, 5.5254e-05, 5.5254e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          6.4731e-05, 1.0669e-04, 1.0669e-04, 1.0669e-04, 1.3292e-05, 1.3292e-05,
          1.0669e-04, 1.3292e-05, 1.3292e-05, 6.4731e-05, 5.5254e-05, 5.5254e-05,
          5.5254e-05, 1.3292e-05, 1.3292e-05, 6.4731e-05, 1.0669e-04, 5.5254e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 4.6253e-04, 4.6253e-04, 6.4731e-05,
          6.4731e-05, 3.0184e-04, 1.3292e-05, 5.5254e-05, 1.0669e-04, 6.4731e-05,
          5.5254e-05, 1.3292e-05, 1.3292e-05, 1.0669e-04, 5.5254e-05, 6.4731e-05,
          1.3292e-05, 1.3292e-05, 5.5254e-05, 5.5254e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 6.4731e-05, 3.4380e-04, 1.0669e-04, 1.0669e-04,
          4.6253e-04, 5.5254e-05, 6.4731e-05, 5.5254e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 3.0184e-04, 5.5254e-05, 5.5254e-05, 1.3292e-05, 5.5254e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 3.0184e-04, 1.3292e-05, 5.5254e-05, 6.4731e-05, 1.0669e-04,
          6.4731e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 6.4731e-05,
          6.4731e-05, 1.0669e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          3.4380e-04, 1.0669e-04, 1.0669e-04, 6.4731e-05, 6.4731e-05, 5.5254e-05,
          1.3292e-05, 5.5254e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.0669e-04,
          1.0669e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          2.1255e-04, 1.3292e-05, 6.4731e-05, 3.6621e-04, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 6.4731e-05, 1.3292e-05, 5.5254e-05, 6.4731e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 5.5254e-05, 1.3292e-05, 6.4731e-05, 1.0669e-04,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 5.5254e-05, 5.5254e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 1.0669e-04, 6.4731e-05, 3.4380e-04,
          1.0669e-04, 2.6751e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.3292e-05, 1.3292e-05, 1.3292e-05, 6.4731e-05, 1.3292e-05, 1.3292e-05,
          6.4731e-05, 6.4731e-05, 1.3292e-05, 1.0669e-04, 5.5254e-05, 1.3292e-05,
          5.5254e-05, 5.5254e-05, 5.5254e-05, 1.3292e-05, 4.6253e-04, 1.3292e-05,
          1.0669e-04, 5.5254e-05, 6.4731e-05, 5.5254e-05, 5.5254e-05, 1.3292e-05,
          1.3292e-05, 6.4731e-05, 6.4731e-05, 1.3292e-05, 5.5254e-05, 5.5254e-05,
          5.5254e-05, 1.3292e-05, 5.5254e-05, 1.0669e-04, 1.3292e-05, 1.0669e-04,
          1.0669e-04, 1.0669e-04, 1.3292e-05, 1.3292e-05, 1.0669e-04, 1.0669e-04,
          1.0669e-04, 1.3292e-05, 5.5254e-05, 1.0669e-04, 1.3292e-05, 1.3292e-05,
          6.4731e-05, 1.3292e-05, 1.3292e-05, 5.5254e-05, 5.5254e-05, 1.3292e-05,
          5.5254e-05, 1.3292e-05, 2.6751e-04, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          1.0669e-04, 1.3292e-05, 1.3292e-05, 5.5254e-05, 3.4380e-04, 6.4731e-05,
          1.3292e-05, 6.4731e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05, 1.3292e-05,
          7.2193e-04, 2.2566e-04, 2.2566e-04, 3.2330e-04, 2.2566e-04, 6.4731e-05,
          1.3292e-05, 1.0669e-04, 2.2566e-04, 5.7888e-04, 2.2566e-04, 6.4731e-05,
          2.2566e-04, 6.4731e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.8474e-03, 6.8359e-02, 2.5085e-02, 2.4246e-02, 5.0751e-02,
          5.3894e-02, 1.9196e-02, 6.4850e-03, 3.3951e-03, 5.7259e-03, 1.0712e-02,
          3.1090e-03, 1.7090e-02, 4.4785e-03, 6.7482e-03, 7.3433e-03, 2.1194e-02,
          1.2115e-02, 4.9973e-03, 1.0757e-03, 1.2999e-03, 2.1515e-03, 1.3649e-02,
          1.2482e-02, 6.2637e-03, 1.5701e-02, 9.1324e-03, 1.7929e-03, 1.5114e-02,
          6.7978e-03, 1.1009e-02, 3.6621e-03, 5.7640e-03, 4.5204e-03, 4.0512e-03,
          2.0256e-03, 8.4734e-04, 3.6697e-03, 3.9864e-03, 1.5774e-03, 2.8858e-03,
          1.4887e-03, 1.2131e-03, 1.3351e-03, 1.2484e-03, 2.0721e-02, 7.2098e-03,
          2.9163e-03, 2.8248e-03, 6.7940e-03, 3.2082e-03, 4.6272e-03, 4.0855e-03,
          1.0099e-03, 2.1305e-03, 9.7227e-04, 8.2874e-04, 1.7252e-03, 1.0767e-03,
          1.1692e-03, 4.9706e-03, 2.6093e-03, 3.0861e-03, 1.7462e-03, 8.4991e-03,
          2.4834e-03, 2.4281e-03, 2.6150e-03, 9.0313e-04, 8.2588e-04, 1.4153e-03,
          1.6241e-03, 1.2962e-02, 1.0862e-03, 6.2828e-03, 9.7847e-04, 9.8610e-04,
          1.8063e-03, 7.7438e-04, 1.1349e-03, 2.0790e-03, 1.1444e-03, 8.6498e-04,
          8.2827e-04, 8.6975e-03, 1.6880e-03, 7.6218e-03, 1.5821e-03, 1.4725e-03,
          9.4986e-03, 5.3978e-03, 1.4524e-03, 6.1684e-03, 1.5192e-03, 1.8539e-03,
          2.3556e-03, 6.7902e-04, 7.5150e-03, 5.3825e-03, 7.8917e-04, 1.0815e-03,
          6.3705e-03, 2.5654e-03, 1.1072e-03, 1.1978e-03, 5.7220e-03, 4.9744e-03,
          7.0992e-03, 1.3237e-03, 5.1765e-03, 1.5173e-03, 2.1172e-03, 2.3098e-03,
          6.0501e-03, 1.9932e-03, 5.7335e-03, 1.1391e-02, 2.8858e-03, 3.5763e-03,
          6.6490e-03, 2.7485e-03, 6.1846e-04, 4.2229e-03, 6.7863e-03, 6.0368e-04,
          6.4325e-04, 7.4530e-04, 1.0290e-03, 5.0201e-03, 1.1108e-02, 8.8272e-03,
          3.6030e-03, 1.9188e-03, 5.6725e-03, 1.7853e-03, 3.4256e-03, 1.1301e-03,
          4.3579e-02, 1.1734e-02, 1.0675e-01, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.6344e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.6344e-04, 1.5855e-05,
          6.0201e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.6344e-04, 1.1903e-04, 6.0201e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05,
          6.0201e-05, 1.6344e-04, 6.0201e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          6.0201e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.6344e-04, 1.5855e-05, 6.0201e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05,
          6.0201e-05, 1.5855e-05, 1.5855e-05, 1.6344e-04, 3.4714e-04, 6.0201e-05,
          6.0201e-05, 1.5855e-05, 1.5855e-05, 4.5061e-04, 6.0201e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 1.6344e-04, 1.6344e-04, 6.0201e-05,
          1.6344e-04, 1.6344e-04, 1.5855e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05,
          1.5855e-05, 1.5855e-05, 1.1903e-04, 2.4390e-04, 1.5855e-05, 1.6344e-04,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 1.6344e-04, 1.6344e-04, 6.0201e-05, 6.0201e-05,
          1.6344e-04, 1.6344e-04, 1.6344e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          6.0201e-05, 1.6344e-04, 1.6344e-04, 1.5855e-05, 1.5855e-05, 1.9944e-04,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.6344e-04, 1.5855e-05, 1.5855e-05, 1.6344e-04, 1.6344e-04,
          1.6344e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 3.4714e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.9944e-04, 1.5855e-05,
          3.4714e-04, 1.6344e-04, 1.6344e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05, 3.4809e-04,
          3.4714e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05,
          1.9944e-04, 1.5855e-05, 6.0201e-05, 4.5061e-04, 1.5855e-05, 1.5855e-05,
          1.1903e-04, 1.6344e-04, 1.6344e-04, 1.5855e-05, 6.0201e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05, 1.6344e-04, 6.0201e-05,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05, 1.6344e-04,
          1.6344e-04, 1.6344e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.5855e-05, 1.5855e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05, 6.0201e-05,
          6.0201e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05,
          6.0201e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.6344e-04, 6.0201e-05,
          6.0201e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05, 3.3641e-04, 1.5855e-05,
          1.5855e-05, 1.6344e-04, 6.0201e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05,
          1.5855e-05, 1.5855e-05, 2.4390e-04, 1.5855e-05, 6.0201e-05, 6.0201e-05,
          6.0201e-05, 1.6344e-04, 6.0201e-05, 1.5855e-05, 6.0201e-05, 6.0201e-05,
          1.6344e-04, 1.5855e-05, 1.5855e-05, 3.8099e-04, 1.5855e-05, 1.5855e-05,
          1.6344e-04, 1.1903e-04, 1.1903e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          6.0201e-05, 2.4390e-04, 3.4714e-04, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          1.6344e-04, 1.5855e-05, 2.4390e-04, 1.5855e-05, 1.6344e-04, 6.0201e-05,
          6.0201e-05, 6.0201e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05, 1.5855e-05,
          4.5061e-04, 3.8815e-04, 3.8815e-04, 3.8815e-04, 3.8815e-04, 6.0201e-05,
          1.5855e-05, 6.0201e-05, 3.8815e-04, 8.6451e-04, 6.0201e-05, 3.8815e-04,
          3.8815e-04, 6.0201e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.8245e-03, 6.0944e-02, 2.5803e-02, 5.2582e-02, 4.9561e-02,
          6.4392e-02, 4.0161e-02, 1.3000e-02, 4.5357e-03, 8.0261e-03, 9.9411e-03,
          4.1504e-03, 5.8594e-03, 2.5883e-03, 6.2103e-03, 1.1871e-02, 7.4387e-03,
          5.3520e-03, 6.4125e-03, 7.2002e-04, 1.0052e-03, 9.6846e-04, 1.1826e-02,
          1.0147e-02, 4.7760e-03, 3.4912e-02, 1.6724e-02, 1.4086e-03, 6.5765e-03,
          3.5210e-03, 1.7975e-02, 1.9836e-03, 6.2332e-03, 8.2855e-03, 9.3079e-03,
          2.5997e-03, 1.0614e-03, 5.1308e-03, 5.3596e-03, 9.5844e-04, 2.1915e-03,
          1.0948e-03, 8.2874e-04, 9.3269e-04, 7.7677e-04, 6.8626e-03, 5.9509e-03,
          1.5011e-03, 1.5402e-03, 7.5188e-03, 7.7095e-03, 9.7656e-03, 8.2855e-03,
          1.4095e-03, 3.7365e-03, 1.2884e-03, 1.0214e-03, 2.4052e-03, 1.3371e-03,
          1.2531e-03, 2.5406e-03, 2.1038e-03, 3.4771e-03, 1.1501e-03, 6.2981e-03,
          2.0370e-03, 1.7357e-03, 2.2659e-03, 6.3086e-04, 5.8270e-04, 9.0504e-04,
          1.0719e-03, 1.7685e-02, 2.0695e-03, 1.1452e-02, 1.5650e-03, 1.6708e-03,
          3.1548e-03, 9.1410e-04, 1.2712e-03, 3.1776e-03, 1.5831e-03, 1.1816e-03,
          1.1139e-03, 1.1826e-02, 1.4648e-03, 6.3210e-03, 1.5097e-03, 1.4133e-03,
          8.5983e-03, 4.0016e-03, 8.1348e-04, 5.0163e-03, 8.1062e-04, 1.0338e-03,
          2.0561e-03, 1.0405e-03, 1.6144e-02, 5.7831e-03, 8.7929e-04, 1.2102e-03,
          7.9193e-03, 2.2335e-03, 1.8339e-03, 2.0828e-03, 6.0234e-03, 2.0370e-03,
          7.1640e-03, 1.0128e-03, 4.2305e-03, 8.8215e-04, 1.7033e-03, 1.9588e-03,
          3.0212e-03, 1.2474e-03, 1.6241e-03, 7.4120e-03, 1.3437e-03, 2.5005e-03,
          3.5820e-03, 3.1338e-03, 7.1716e-04, 5.9319e-03, 7.2937e-03, 6.8712e-04,
          7.0190e-04, 9.1410e-04, 1.1206e-03, 5.6343e-03, 1.5373e-02, 8.6746e-03,
          2.3041e-03, 1.7662e-03, 5.6915e-03, 1.5507e-03, 3.0632e-03, 7.4768e-04,
          1.8021e-02, 3.4828e-03, 1.7578e-02, 7.6660e-02, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.5676e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 3.1376e-04, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 6.0701e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.5676e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 3.1209e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 3.1209e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 3.1209e-04, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 3.1209e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.9312e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          3.1209e-04, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 3.1209e-04, 1.3828e-05, 1.3828e-05, 4.0030e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 4.0030e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.6041e-03, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 9.9373e-04,
          1.3828e-05, 2.1973e-03, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 6.0701e-04, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 4.4727e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.7177e-03, 6.1401e-02, 2.0477e-02, 2.7740e-02, 4.5868e-02,
          6.5247e-02, 1.7258e-02, 4.5815e-03, 2.7027e-03, 8.3389e-03, 7.0229e-03,
          3.6430e-03, 6.1188e-03, 1.9760e-03, 9.7427e-03, 4.7798e-03, 8.4076e-03,
          5.5733e-03, 2.1667e-03, 8.4257e-04, 1.2856e-03, 9.7752e-04, 1.1276e-02,
          7.7820e-03, 6.3057e-03, 1.4885e-02, 2.0096e-02, 2.0809e-03, 1.1078e-02,
          3.0384e-03, 8.6823e-03, 3.2902e-03, 1.5564e-02, 9.9945e-03, 9.1171e-03,
          4.8599e-03, 1.2579e-03, 1.2306e-02, 1.1650e-02, 1.0910e-03, 3.5992e-03,
          1.2026e-03, 1.0509e-03, 9.5654e-04, 1.0214e-03, 9.6970e-03, 1.6464e-02,
          1.3533e-03, 1.5364e-03, 7.7858e-03, 8.7891e-03, 6.4735e-03, 5.7716e-03,
          1.5516e-03, 3.0270e-03, 1.6966e-03, 1.3304e-03, 2.5368e-03, 1.4992e-03,
          1.5211e-03, 4.2191e-03, 2.9202e-03, 6.4659e-03, 1.4238e-03, 1.3924e-02,
          1.6890e-03, 1.5173e-03, 2.8896e-03, 7.7009e-04, 7.5054e-04, 8.4162e-04,
          1.1292e-03, 2.1835e-02, 2.5997e-03, 7.3013e-03, 1.4277e-03, 1.4362e-03,
          2.3632e-03, 1.4582e-03, 1.4372e-03, 4.6692e-03, 1.5278e-03, 1.1339e-03,
          1.0881e-03, 9.8953e-03, 2.0008e-03, 4.8485e-03, 1.3447e-03, 1.2245e-03,
          6.6109e-03, 5.4855e-03, 7.2861e-04, 4.6158e-03, 8.6880e-04, 1.0357e-03,
          3.3779e-03, 1.3447e-03, 1.5350e-02, 9.6893e-03, 1.4620e-03, 2.0084e-03,
          6.7368e-03, 1.5821e-03, 1.5669e-03, 3.3436e-03, 1.0002e-02, 3.9406e-03,
          6.7978e-03, 1.4954e-03, 6.0501e-03, 1.0605e-03, 1.6928e-03, 2.4223e-03,
          3.6221e-03, 2.2564e-03, 2.6379e-03, 1.6403e-02, 1.3275e-03, 5.5656e-03,
          7.2174e-03, 7.8430e-03, 1.0891e-03, 2.5463e-03, 1.3382e-02, 9.2363e-04,
          1.1358e-03, 1.0643e-03, 1.4801e-03, 1.0681e-02, 1.1597e-02, 7.6180e-03,
          2.4090e-03, 1.5192e-03, 3.0460e-03, 1.2522e-03, 3.7117e-03, 8.6546e-04,
          3.2806e-02, 4.3449e-03, 1.0841e-02, 1.9211e-02, 5.6946e-02, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.6403e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.6403e-04, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.6403e-04, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05,
          1.6403e-04, 1.1921e-05, 1.1921e-05, 1.6403e-04, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05,
          1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.6403e-04,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.0782e-04,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05,
          1.0782e-04, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.0782e-04,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05,
          2.5988e-04, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1021e-04,
          1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.0782e-04, 1.1921e-05, 1.6403e-04, 1.6403e-04, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.6403e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.0782e-04,
          1.1921e-05, 1.1921e-05, 1.6403e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 3.0041e-05, 2.5988e-04, 1.0782e-04,
          1.1921e-05, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 6.8569e-04,
          1.6403e-04, 2.5988e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.0782e-04, 1.6403e-04, 1.1921e-05, 6.8569e-04, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05, 2.5988e-04, 1.1921e-05,
          1.6403e-04, 1.0782e-04, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          2.5988e-04, 1.1921e-05, 1.0782e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.7700e-03, 1.1921e-05, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.6403e-04, 1.6403e-04, 1.6403e-04, 1.1921e-05, 1.6403e-04, 1.1921e-05,
          1.1921e-05, 1.6403e-04, 1.6403e-04, 1.1921e-05, 1.1921e-05, 1.1921e-05,
          1.1921e-05, 1.1921e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.2092e-03, 6.0486e-02, 2.1759e-02, 2.8000e-02, 6.4880e-02,
          4.7424e-02, 1.7502e-02, 2.8057e-03, 2.0428e-03, 3.4733e-03, 4.5662e-03,
          2.0523e-03, 9.1705e-03, 1.4084e-02, 3.1166e-03, 2.1362e-03, 3.0533e-02,
          1.4435e-02, 1.0662e-03, 2.6417e-03, 1.8301e-03, 2.6760e-03, 6.9695e-03,
          4.6501e-03, 2.3193e-03, 1.4198e-02, 1.0155e-02, 5.3825e-03, 1.7303e-02,
          2.2293e-02, 4.7150e-03, 8.6517e-03, 5.7564e-03, 2.5978e-03, 3.4161e-03,
          3.9101e-03, 1.5631e-03, 3.4676e-03, 3.1471e-03, 4.6082e-03, 2.5234e-03,
          1.3123e-03, 1.0042e-03, 1.0529e-03, 8.8072e-04, 1.3176e-02, 4.8141e-03,
          6.2218e-03, 4.3602e-03, 7.5684e-03, 5.5389e-03, 4.2419e-03, 2.2717e-03,
          1.9121e-03, 2.6302e-03, 2.3460e-03, 2.3041e-03, 2.5921e-03, 2.2907e-03,
          2.2278e-03, 1.4175e-02, 2.3022e-03, 2.8267e-03, 1.5144e-03, 4.7836e-03,
          2.2297e-03, 1.0452e-03, 1.4830e-03, 7.1239e-04, 6.5851e-04, 1.0014e-03,
          1.1816e-03, 6.8245e-03, 3.1147e-03, 5.3520e-03, 1.5459e-03, 1.3723e-03,
          3.0499e-03, 2.8973e-03, 2.5864e-03, 3.3417e-03, 2.3079e-03, 1.5163e-03,
          1.5831e-03, 4.9973e-03, 2.2621e-03, 2.0657e-03, 1.1969e-03, 1.5469e-03,
          3.4523e-03, 2.5291e-03, 1.0080e-03, 2.2812e-03, 9.4461e-04, 1.1806e-03,
          8.9645e-03, 1.8044e-03, 3.9520e-03, 7.9498e-03, 3.2082e-03, 3.4542e-03,
          4.7531e-03, 1.3742e-03, 1.5631e-03, 2.7885e-03, 3.9253e-03, 1.6235e-02,
          3.9635e-03, 1.7462e-03, 3.5343e-03, 1.1034e-03, 1.6909e-03, 1.5507e-03,
          1.3409e-03, 1.6851e-03, 5.6648e-03, 5.0659e-03, 1.7204e-03, 1.7929e-02,
          4.9774e-02, 6.7024e-03, 1.8673e-03, 1.4763e-03, 4.9667e-03, 1.4744e-03,
          2.2202e-03, 1.8148e-03, 2.4719e-03, 6.2027e-03, 4.5929e-03, 4.5280e-03,
          2.5978e-03, 1.4219e-03, 9.5320e-04, 7.6437e-04, 1.9550e-03, 1.1463e-03,
          1.9516e-02, 2.0554e-02, 8.9417e-03, 6.4240e-03, 6.4430e-03, 6.8481e-02,
          1.2219e-05, 4.7743e-05, 1.2219e-05, 1.2219e-05, 4.7743e-05, 7.5281e-05,
          1.2219e-05, 4.7743e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          1.1063e-04, 1.2219e-05, 2.1052e-04, 1.2219e-05, 1.3733e-04, 1.2219e-05,
          1.2219e-05, 3.9816e-05, 3.9816e-05, 1.2219e-05, 1.2219e-05, 3.9816e-05,
          7.5281e-05, 3.1686e-04, 7.5281e-05, 4.7743e-05, 1.2219e-05, 1.2219e-05,
          3.9816e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          3.9816e-05, 7.5281e-05, 1.6499e-04, 1.3959e-04, 1.2219e-05, 3.9816e-05,
          1.3959e-04, 1.2219e-05, 1.2219e-05, 1.2219e-05, 7.5281e-05, 7.5281e-05,
          7.5281e-05, 1.2219e-05, 4.7743e-05, 3.9816e-05, 1.4615e-04, 4.7743e-05,
          1.2219e-05, 1.2219e-05, 1.2219e-05, 4.2534e-04, 2.8348e-04, 7.5281e-05,
          2.7132e-04, 1.7500e-04, 2.8086e-04, 7.5281e-05, 7.5281e-05, 3.9816e-05,
          7.5281e-05, 1.2219e-05, 3.9816e-05, 7.5281e-05, 7.5281e-05, 3.9816e-05,
          3.9816e-05, 1.2219e-05, 4.7743e-05, 7.5281e-05, 1.2219e-05, 1.2219e-05,
          1.2219e-05, 1.2219e-05, 7.8058e-04, 5.5075e-04, 1.6499e-04, 7.5281e-05,
          4.2534e-04, 1.6499e-04, 7.9203e-04, 1.3733e-04, 1.2219e-05, 1.2219e-05,
          1.2219e-05, 2.1899e-04, 7.5281e-05, 4.7743e-05, 1.5414e-04, 1.2994e-04,
          1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          2.1732e-04, 4.2534e-04, 1.2219e-05, 6.2227e-04, 1.8167e-04, 1.6499e-04,
          2.7132e-04, 2.8086e-04, 2.8086e-04, 1.2219e-05, 1.2219e-05, 3.9816e-05,
          3.9816e-05, 1.8370e-04, 1.2219e-05, 1.5414e-04, 1.2219e-05, 4.7743e-05,
          1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 3.9816e-05,
          5.5075e-04, 2.1720e-04, 2.2852e-04, 6.3848e-04, 3.9816e-05, 1.6499e-04,
          5.8699e-04, 4.7743e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 3.2544e-04,
          2.1720e-04, 1.5414e-04, 1.5414e-04, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          1.2994e-04, 1.2219e-05, 3.9816e-05, 1.7500e-04, 1.2219e-05, 1.3733e-04,
          2.1732e-04, 3.8671e-04, 3.8671e-04, 7.5281e-05, 7.8058e-04, 2.8086e-04,
          2.1732e-04, 3.9816e-05, 3.1900e-04, 1.2219e-05, 2.9016e-04, 2.1720e-04,
          1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05, 4.7743e-05, 4.7743e-05,
          1.2219e-05, 7.5281e-05, 1.2219e-05, 7.5281e-05, 3.9816e-05, 2.7323e-04,
          8.7357e-04, 2.1052e-04, 1.2219e-05, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          3.9816e-05, 1.2219e-05, 1.2219e-05, 3.9816e-05, 1.2219e-05, 1.2219e-05,
          1.8167e-04, 3.9816e-05, 1.2219e-05, 7.5281e-05, 4.7743e-05, 1.2219e-05,
          4.7743e-05, 7.5281e-05, 7.5281e-05, 1.2219e-05, 4.2534e-04, 1.2219e-05,
          1.3959e-04, 1.3733e-04, 3.9816e-05, 2.0170e-04, 4.7743e-05, 1.2219e-05,
          1.2219e-05, 3.9816e-05, 3.9816e-05, 1.2219e-05, 1.4615e-04, 7.5281e-05,
          1.8954e-04, 1.2219e-05, 2.8157e-04, 2.4819e-04, 1.2219e-05, 2.1720e-04,
          5.9080e-04, 4.8208e-04, 2.4605e-04, 1.2219e-05, 3.1900e-04, 7.5281e-05,
          1.3959e-04, 1.2219e-05, 4.7743e-05, 1.3959e-04, 1.2219e-05, 7.6592e-05,
          3.9816e-05, 2.9016e-04, 1.2219e-05, 1.2219e-05, 3.4690e-04, 1.2219e-05,
          4.8208e-04, 1.2219e-05, 2.1052e-04, 1.2219e-05, 1.2219e-05, 1.2219e-05,
          4.6086e-04, 1.2219e-05, 1.2219e-05, 4.7743e-05, 5.2404e-04, 2.4819e-04,
          1.2219e-05, 5.5504e-04, 7.6592e-05, 1.4818e-04, 1.2219e-05, 1.2219e-05,
          2.8348e-04, 2.8348e-04, 1.1063e-04, 1.1063e-04, 1.1063e-04, 3.9816e-05,
          3.9816e-05, 1.4615e-04, 1.1063e-04, 3.4070e-04, 1.1063e-04, 1.1063e-04,
          1.1063e-04, 1.1063e-04], dtype=torch.float16),
  tensor([0.0000e+00, 8.0109e-03, 5.7770e-02, 1.8234e-02, 2.0447e-02, 6.5063e-02,
          4.3610e-02, 1.6327e-02, 1.5640e-03, 2.1648e-03, 5.2605e-03, 8.7204e-03,
          3.5515e-03, 1.5259e-03, 1.8511e-03, 2.0275e-03, 1.0509e-03, 1.9379e-03,
          1.1215e-03, 8.9598e-04, 7.9155e-04, 7.7105e-04, 6.6090e-04, 1.4114e-03,
          8.3008e-03, 3.1643e-03, 1.1574e-02, 3.7670e-03, 2.0466e-03, 2.0237e-03,
          2.7313e-03, 2.3098e-03, 1.0176e-03, 3.0365e-03, 1.9951e-03, 1.4277e-03,
          5.3253e-03, 4.6310e-03, 2.8019e-03, 1.9627e-03, 1.2655e-03, 4.6158e-03,
          4.7035e-03, 3.4561e-03, 3.1586e-03, 1.6775e-03, 1.5211e-03, 2.6989e-03,
          9.3508e-04, 9.1648e-04, 6.5804e-03, 3.9902e-03, 2.7428e-03, 1.9245e-03,
          2.0771e-03, 1.7452e-03, 1.8110e-03, 1.0620e-02, 5.3902e-03, 1.2329e-02,
          6.0883e-03, 2.3174e-03, 2.2430e-03, 1.8501e-03, 8.8272e-03, 5.2071e-03,
          9.6664e-03, 3.8071e-03, 2.0046e-03, 2.1648e-03, 2.2182e-03, 2.6093e-03,
          1.5173e-03, 7.2136e-03, 3.3131e-03, 3.3245e-03, 1.4629e-03, 1.1845e-03,
          2.3403e-03, 1.9135e-02, 2.4567e-02, 7.8430e-03, 1.3161e-02, 3.7918e-03,
          3.7251e-03, 3.1452e-03, 1.8463e-03, 3.5496e-03, 4.9438e-03, 7.1068e-03,
          6.1455e-03, 4.9973e-03, 3.1986e-03, 4.4403e-03, 1.8187e-03, 1.4963e-03,
          2.4071e-03, 1.7958e-03, 1.5965e-03, 6.9275e-03, 4.1626e-02, 1.9333e-02,
          1.2611e-02, 5.6038e-03, 6.4316e-03, 9.6283e-03, 4.8637e-03, 3.1281e-03,
          2.8172e-03, 5.6343e-03, 7.3318e-03, 4.7607e-03, 8.8577e-03, 5.1537e-03,
          3.1013e-03, 3.1471e-03, 1.1778e-03, 4.4060e-03, 2.0657e-03, 7.1831e-03,
          8.0719e-03, 1.8967e-02, 2.9861e-02, 1.1120e-03, 5.4970e-03, 1.3344e-02,
          1.6891e-02, 1.5518e-02, 1.4862e-02, 5.2567e-03, 1.8797e-03, 3.5038e-03,
          8.7280e-03, 5.3291e-03, 9.5129e-04, 2.1458e-03, 5.0354e-03, 3.8929e-03,
          2.6989e-03, 2.6245e-03, 1.6041e-03, 2.8248e-03, 2.5749e-03, 5.7373e-03,
          5.3284e-02, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.3447e-04, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.3447e-04,
          1.8299e-05, 1.8299e-05, 1.4877e-04, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 9.5367e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05,
          7.1704e-05, 1.8299e-05, 1.8299e-05, 1.4877e-04, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.8299e-05, 1.4877e-04,
          7.1704e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 7.1704e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.4877e-04, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 8.6498e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          7.1704e-05, 1.8299e-05, 1.4877e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 7.1704e-05, 7.1704e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          9.5367e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 6.1083e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05,
          6.6423e-04, 7.1704e-05, 9.5367e-05, 1.8299e-05, 1.8299e-05, 1.4877e-04,
          1.8299e-05, 7.1704e-05, 1.4877e-04, 7.1704e-05, 1.8299e-05, 1.8299e-05,
          9.5367e-05, 1.8299e-05, 5.8699e-04, 1.8299e-05, 1.8299e-05, 1.4877e-04,
          1.8299e-05, 1.8299e-05, 2.9135e-04, 1.8299e-05, 7.1704e-05, 9.4223e-04,
          5.3358e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05, 7.1704e-05, 7.1704e-05,
          7.1704e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.4877e-04, 1.8299e-05,
          1.4877e-04, 7.1704e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05,
          7.1704e-05, 9.5367e-05, 7.1704e-05, 9.5367e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 7.1704e-05, 7.1704e-05, 1.8299e-05,
          1.4877e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05, 1.8299e-05, 9.5367e-05,
          7.1704e-05, 3.4976e-04, 1.4877e-04, 1.8299e-05, 1.8299e-05, 9.5367e-05,
          9.5367e-05, 9.5367e-05, 1.8299e-05, 9.5367e-05, 1.8299e-05, 1.8299e-05,
          1.8299e-05, 1.8299e-05, 1.8299e-05, 1.8299e-05, 9.5367e-05, 1.8299e-05,
          1.8299e-05, 9.5367e-05, 1.8299e-05, 2.8014e-04, 9.5367e-05, 9.5367e-05,
          1.8299e-05, 1.8299e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6572e-03, 5.7831e-02, 1.6769e-02, 1.7334e-02, 6.1859e-02,
          5.6610e-02, 1.5015e-02, 4.8256e-03, 2.9240e-03, 9.7427e-03, 2.3941e-02,
          6.1836e-03, 6.2218e-03, 1.0929e-03, 1.3580e-02, 4.7073e-03, 3.5744e-03,
          1.8559e-03, 3.9787e-03, 3.6621e-04, 5.0163e-04, 5.0688e-04, 3.9597e-03,
          2.5116e-02, 1.1589e-02, 1.1459e-02, 5.3520e-03, 7.4816e-04, 3.7270e-03,
          1.4477e-03, 7.0152e-03, 1.0748e-03, 6.3248e-03, 7.0229e-03, 2.5444e-03,
          3.7327e-03, 1.4629e-03, 3.1433e-03, 4.5738e-03, 5.3978e-04, 2.7866e-03,
          1.1396e-03, 1.0681e-03, 9.3460e-04, 1.2608e-03, 4.9057e-03, 5.3177e-03,
          5.6362e-04, 5.6171e-04, 5.4665e-03, 2.4166e-03, 3.5133e-03, 3.7079e-03,
          9.5701e-04, 1.8969e-03, 8.6498e-04, 1.7853e-03, 2.6340e-03, 2.4738e-03,
          2.3918e-03, 1.3208e-03, 1.3151e-03, 1.5697e-03, 2.6760e-03, 1.1452e-02,
          3.4809e-03, 3.3875e-03, 2.2678e-03, 7.3481e-04, 6.4135e-04, 9.3985e-04,
          7.4196e-04, 1.7746e-02, 1.1034e-03, 4.0779e-03, 9.0933e-04, 8.6498e-04,
          1.6909e-03, 1.9913e-03, 3.0136e-03, 6.8550e-03, 2.8191e-03, 2.0885e-03,
          1.4744e-03, 4.3259e-03, 9.6941e-04, 1.5381e-02, 2.3327e-03, 1.8501e-03,
          1.6113e-02, 6.5689e-03, 1.2236e-03, 7.9041e-03, 9.5987e-04, 6.7282e-04,
          1.1263e-03, 6.7472e-04, 3.9558e-03, 4.4823e-03, 2.3594e-03, 3.6335e-03,
          2.6810e-02, 9.2926e-03, 3.7231e-03, 3.3131e-03, 1.3756e-02, 1.7052e-03,
          3.6621e-03, 1.1988e-03, 8.1024e-03, 1.6403e-03, 3.8147e-03, 2.8629e-03,
          1.4061e-02, 1.6794e-03, 7.3433e-04, 1.2527e-02, 8.5258e-04, 1.8787e-03,
          2.3537e-03, 1.0406e-02, 2.2411e-03, 5.9319e-03, 2.1271e-02, 1.9932e-03,
          1.5621e-03, 1.5707e-03, 2.7008e-03, 1.1154e-02, 9.7809e-03, 4.2267e-03,
          6.6223e-03, 3.3817e-03, 1.9547e-02, 2.1954e-03, 5.0545e-03, 1.1168e-03,
          8.7967e-03, 1.4753e-03, 1.7426e-02, 1.5747e-02, 9.5596e-03, 3.1719e-03,
          3.7689e-03, 8.3008e-02, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05,
          2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05,
          7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 3.1781e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 3.1781e-04,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.0313e-04,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.0313e-04, 2.0313e-04, 2.3901e-05,
          7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 4.0722e-04, 2.0313e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05,
          1.4794e-04, 2.3901e-05, 7.9155e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 3.1781e-04, 7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          7.9155e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 7.9155e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05,
          7.9155e-05, 7.9155e-05, 3.1781e-04, 2.3901e-05, 2.3901e-05, 2.0313e-04,
          2.3901e-05, 7.9155e-05, 4.4179e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          7.9155e-05, 2.3901e-05, 2.3901e-05, 3.1781e-04, 2.3901e-05, 2.0313e-04,
          2.3901e-05, 2.3901e-05, 3.1781e-04, 2.3901e-05, 1.4794e-04, 7.9155e-05,
          3.1781e-04, 2.0313e-04, 2.3901e-05, 2.3901e-05, 2.0313e-04, 1.4794e-04,
          1.4794e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.0313e-04, 7.9155e-05,
          4.4179e-04, 4.4179e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 7.9155e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05, 2.3901e-05,
          7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 1.4794e-04, 2.3901e-05,
          1.4210e-03, 2.0313e-04, 2.3901e-05, 7.9155e-05, 2.3901e-05, 7.9155e-05,
          2.0313e-04, 7.0143e-04, 2.0313e-04, 2.3901e-05, 2.3901e-05, 7.9155e-05,
          1.0586e-03, 1.2970e-03, 2.3901e-05, 3.1781e-04, 2.3901e-05, 2.3901e-05,
          7.9155e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 7.9155e-05, 2.3901e-05,
          2.3901e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05, 7.9155e-05,
          7.9155e-05, 2.3901e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.4545e-03, 5.6458e-02, 1.4763e-02, 1.7212e-02, 5.4993e-02,
          4.4464e-02, 1.2192e-02, 1.9140e-03, 2.0218e-03, 7.9269e-03, 1.1917e-02,
          4.5013e-03, 2.0809e-03, 1.4820e-03, 5.5847e-03, 1.5202e-03, 2.0847e-03,
          1.3294e-03, 1.1034e-03, 5.3596e-04, 5.6696e-04, 4.8256e-04, 2.0123e-03,
          1.2238e-02, 7.5760e-03, 9.3079e-03, 5.2605e-03, 1.1597e-03, 2.6608e-03,
          2.4815e-03, 3.3474e-03, 1.1072e-03, 6.6147e-03, 4.2191e-03, 2.5883e-03,
          6.9504e-03, 2.1820e-03, 4.5891e-03, 4.1466e-03, 8.9645e-04, 5.6152e-03,
          2.9335e-03, 3.2463e-03, 2.1648e-03, 2.0294e-03, 2.2564e-03, 6.8779e-03,
          8.0919e-04, 6.7282e-04, 5.2414e-03, 3.8605e-03, 3.3073e-03, 2.7657e-03,
          1.4639e-03, 1.9798e-03, 1.3304e-03, 4.3564e-03, 3.6964e-03, 4.5052e-03,
          3.1166e-03, 1.9970e-03, 1.7443e-03, 1.9722e-03, 6.7978e-03, 2.1011e-02,
          7.9117e-03, 4.0703e-03, 6.0005e-03, 2.2011e-03, 1.6069e-03, 1.6823e-03,
          1.1473e-03, 1.4626e-02, 2.0161e-03, 3.8395e-03, 1.1492e-03, 1.2341e-03,
          2.1248e-03, 6.8817e-03, 1.0437e-02, 1.3985e-02, 8.6975e-03, 2.6207e-03,
          1.8024e-03, 4.0474e-03, 1.3294e-03, 7.4806e-03, 4.8714e-03, 5.6343e-03,
          1.2543e-02, 9.4147e-03, 4.5128e-03, 8.2703e-03, 1.6870e-03, 1.2655e-03,
          1.5965e-03, 1.1063e-03, 3.3913e-03, 5.8975e-03, 1.3542e-02, 1.3229e-02,
          2.0615e-02, 5.4588e-03, 6.3438e-03, 1.0246e-02, 1.2299e-02, 2.6970e-03,
          3.3226e-03, 3.4275e-03, 1.0933e-02, 3.7060e-03, 6.7940e-03, 6.1264e-03,
          1.2558e-02, 4.1504e-03, 8.8072e-04, 1.5358e-02, 1.6527e-03, 3.3398e-03,
          5.7640e-03, 2.3361e-02, 8.3237e-03, 1.4229e-03, 2.0142e-02, 9.4223e-03,
          6.7482e-03, 6.2141e-03, 8.9645e-03, 1.3992e-02, 3.7174e-03, 3.9215e-03,
          8.5297e-03, 5.7182e-03, 1.9627e-03, 3.2501e-03, 8.2016e-03, 3.3283e-03,
          5.3177e-03, 1.8959e-03, 2.4242e-03, 3.9024e-03, 5.2376e-03, 6.4201e-03,
          1.8509e-02, 4.0817e-03, 3.4576e-02, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 4.5776e-04,
          1.7405e-05, 1.7405e-05, 4.8399e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.4436e-04, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05, 8.3625e-05, 1.7405e-05,
          1.7405e-05, 3.3069e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05,
          7.8142e-05, 1.7405e-05, 1.7405e-05, 4.8399e-04, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05, 1.7405e-05, 7.9679e-04,
          7.8142e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.4436e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 3.3069e-04,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 7.8142e-05, 8.3625e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.2207e-03,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.4436e-04, 1.4436e-04, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 8.3625e-05, 8.1205e-04, 1.4436e-04, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 8.3625e-05, 1.7405e-05,
          7.8142e-05, 1.7405e-05, 1.4436e-04, 4.8399e-04, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 7.8142e-05, 8.3625e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 8.3625e-05, 1.7405e-05, 7.8142e-05,
          3.9697e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 7.3624e-04, 4.5776e-04, 1.7405e-05, 1.7405e-05,
          7.9679e-04, 3.3069e-04, 8.3625e-05, 1.7405e-05, 1.7405e-05, 1.4436e-04,
          1.7405e-05, 1.4436e-04, 1.4436e-04, 7.8142e-05, 1.7405e-05, 1.7405e-05,
          1.4436e-04, 1.7405e-05, 4.5776e-04, 1.7405e-05, 1.7405e-05, 1.4436e-04,
          1.7405e-05, 1.7405e-05, 4.8399e-04, 1.7405e-05, 3.7789e-04, 8.7833e-04,
          3.9697e-04, 1.2808e-03, 1.7405e-05, 1.7405e-05, 7.8142e-05, 7.8142e-05,
          3.7789e-04, 8.3625e-05, 1.7405e-05, 1.7405e-05, 1.4436e-04, 1.7405e-05,
          1.4436e-04, 1.4436e-04, 1.7405e-05, 1.7405e-05, 7.8142e-05, 1.7405e-05,
          7.8142e-05, 1.4436e-04, 7.8142e-05, 8.3625e-05, 1.7405e-05, 1.7405e-05,
          1.4436e-04, 1.7405e-05, 1.7405e-05, 7.8142e-05, 7.8142e-05, 1.7405e-05,
          1.4436e-04, 1.4436e-04, 7.8142e-05, 8.3625e-05, 1.7405e-05, 8.3625e-05,
          1.4436e-04, 1.7405e-05, 2.9874e-04, 1.7405e-05, 1.7405e-05, 8.3625e-05,
          1.2207e-03, 1.2207e-03, 1.7405e-05, 8.3625e-05, 8.3625e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 8.3625e-05, 1.7405e-05,
          1.7405e-05, 8.3625e-05, 8.3625e-05, 8.3625e-05, 8.3625e-05, 8.3625e-05,
          1.7405e-05, 1.7405e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.5912e-03, 5.5084e-02, 1.7059e-02, 1.8478e-02, 5.2521e-02,
          4.0924e-02, 1.3062e-02, 1.7586e-03, 2.3003e-03, 7.3357e-03, 1.0635e-02,
          5.2986e-03, 2.2526e-03, 1.6365e-03, 3.2616e-03, 1.2884e-03, 3.1986e-03,
          1.5650e-03, 1.0319e-03, 7.1812e-04, 8.3733e-04, 5.8937e-04, 1.7929e-03,
          1.1726e-02, 4.3945e-03, 9.7046e-03, 5.2299e-03, 1.5554e-03, 3.0804e-03,
          2.5845e-03, 2.9469e-03, 1.3409e-03, 5.6915e-03, 2.8458e-03, 2.5406e-03,
          9.6283e-03, 4.3449e-03, 3.9864e-03, 3.0403e-03, 1.1520e-03, 7.1487e-03,
          3.3607e-03, 2.7618e-03, 2.3632e-03, 2.4681e-03, 2.2697e-03, 5.3101e-03,
          8.7357e-04, 8.1587e-04, 4.9782e-03, 4.6730e-03, 3.4256e-03, 2.3499e-03,
          1.7462e-03, 2.2278e-03, 1.6823e-03, 9.6817e-03, 6.4926e-03, 1.3168e-02,
          6.1302e-03, 2.4414e-03, 1.9274e-03, 2.3556e-03, 7.1945e-03, 9.0485e-03,
          3.8853e-03, 3.4981e-03, 2.5444e-03, 1.9913e-03, 1.8053e-03, 1.8330e-03,
          1.6479e-03, 1.1131e-02, 2.7390e-03, 3.8681e-03, 1.2560e-03, 1.4277e-03,
          2.4414e-03, 1.2856e-02, 1.2024e-02, 1.4328e-02, 1.0216e-02, 5.2338e-03,
          3.2940e-03, 4.0283e-03, 1.6069e-03, 5.0545e-03, 2.5578e-03, 2.8954e-03,
          9.3155e-03, 1.0849e-02, 1.9894e-03, 6.7978e-03, 1.8473e-03, 1.5802e-03,
          2.0123e-03, 1.4715e-03, 2.9411e-03, 1.3283e-02, 2.1057e-02, 2.3224e-02,
          2.1286e-02, 3.7537e-03, 6.4507e-03, 2.1500e-02, 1.0712e-02, 3.2768e-03,
          3.7231e-03, 4.6272e-03, 1.1681e-02, 2.3518e-03, 3.1910e-03, 4.2534e-03,
          3.5439e-03, 4.8637e-03, 1.0986e-03, 8.5373e-03, 1.9932e-03, 4.5700e-03,
          5.9357e-03, 2.7115e-02, 1.0918e-02, 1.3752e-03, 1.1902e-02, 9.1248e-03,
          1.1009e-02, 7.6714e-03, 9.3307e-03, 1.0719e-02, 2.8839e-03, 4.0741e-03,
          4.5815e-03, 3.3340e-03, 9.3174e-04, 1.9836e-03, 8.0185e-03, 2.8954e-03,
          5.0163e-03, 2.7466e-03, 1.6022e-03, 2.8114e-03, 3.9215e-03, 6.2675e-03,
          1.9913e-02, 3.0041e-03, 1.0674e-02, 2.5787e-02, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5579e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 3.9721e-04,
          1.7107e-05, 1.7107e-05, 1.2624e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.2624e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.2624e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 2.3520e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          7.5579e-05, 1.7107e-05, 1.7107e-05, 5.5170e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.7770e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5579e-05, 1.7107e-05, 7.7009e-04,
          7.5579e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.2624e-04, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.2624e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5579e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 2.9373e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5579e-05, 1.2624e-04,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 6.7770e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.2624e-04, 1.2624e-04, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 7.4196e-04, 1.2624e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.2624e-04, 1.7107e-05,
          7.5579e-05, 1.7107e-05, 1.2624e-04, 1.2624e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.7770e-05, 7.5579e-05, 1.2624e-04, 1.7107e-05, 1.7107e-05,
          1.7107e-05, 6.7770e-05, 6.7770e-05, 6.7770e-05, 1.7107e-05, 1.7107e-05,
          2.8586e-04, 6.7770e-05, 1.7107e-05, 7.5579e-05, 6.7770e-05, 1.7107e-05,
          1.7107e-05, 1.7107e-05, 2.8586e-04, 3.4428e-04, 1.7107e-05, 1.7107e-05,
          2.8586e-04, 2.8586e-04, 4.9305e-04, 1.7107e-05, 1.7107e-05, 1.2624e-04,
          1.7107e-05, 1.2624e-04, 1.2660e-04, 7.5579e-05, 1.7107e-05, 6.7770e-05,
          6.7770e-05, 1.7107e-05, 3.4428e-04, 6.7770e-05, 1.7107e-05, 1.2624e-04,
          1.7107e-05, 1.7107e-05, 1.2624e-04, 1.7107e-05, 1.7107e-05, 6.8903e-04,
          6.8903e-04, 1.2624e-04, 1.7107e-05, 1.7107e-05, 7.5579e-05, 7.5579e-05,
          7.5579e-05, 6.7770e-05, 1.7107e-05, 1.7107e-05, 1.2660e-04, 1.7107e-05,
          3.4428e-04, 1.2624e-04, 1.7107e-05, 1.7107e-05, 6.7770e-05, 1.7107e-05,
          1.7107e-05, 5.2977e-04, 1.7107e-05, 6.7770e-05, 1.7107e-05, 1.7107e-05,
          6.7770e-05, 1.7107e-05, 1.7107e-05, 1.7107e-05, 7.5579e-05, 1.7107e-05,
          6.7770e-05, 1.2624e-04, 1.7107e-05, 6.7770e-05, 1.7107e-05, 2.8586e-04,
          1.2624e-04, 6.8903e-04, 1.2624e-04, 1.7107e-05, 1.7107e-05, 2.8586e-04,
          2.8586e-04, 2.8586e-04, 1.7107e-05, 6.7770e-05, 6.7770e-05, 1.7107e-05,
          6.7770e-05, 6.7770e-05, 1.7107e-05, 6.7770e-05, 3.4475e-04, 6.7770e-05,
          1.7107e-05, 3.4475e-04, 6.7770e-05, 3.3092e-04, 6.7770e-05, 3.4475e-04,
          6.7770e-05, 6.7770e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.7673e-03, 6.1707e-02, 1.6708e-02, 2.1255e-02, 5.7739e-02,
          4.5135e-02, 1.4305e-02, 1.5841e-03, 1.4982e-03, 4.7722e-03, 1.1131e-02,
          3.5343e-03, 1.4009e-03, 2.2526e-03, 1.8492e-03, 1.1978e-03, 1.6870e-03,
          1.2417e-03, 9.0408e-04, 4.7636e-04, 4.6802e-04, 4.5133e-04, 1.6136e-03,
          1.1948e-02, 2.9125e-03, 1.1246e-02, 5.4436e-03, 1.1244e-03, 1.9951e-03,
          3.6945e-03, 2.5482e-03, 8.8739e-04, 2.3003e-03, 1.9417e-03, 1.4381e-03,
          2.9049e-03, 2.0885e-03, 2.0409e-03, 1.6146e-03, 8.9788e-04, 2.9163e-03,
          2.9202e-03, 1.9913e-03, 3.3550e-03, 9.7275e-04, 1.6317e-03, 2.0981e-03,
          9.3794e-04, 6.8998e-04, 5.5237e-03, 2.5082e-03, 2.5635e-03, 1.6832e-03,
          1.5812e-03, 1.7376e-03, 1.3580e-03, 3.3360e-03, 4.3449e-03, 4.4136e-03,
          3.2196e-03, 1.8158e-03, 1.8272e-03, 1.3695e-03, 6.2637e-03, 4.3564e-03,
          2.7771e-02, 7.3280e-03, 2.4071e-03, 1.5345e-03, 1.4982e-03, 2.2907e-03,
          1.1425e-03, 8.3008e-03, 1.9588e-03, 3.1948e-03, 1.4982e-03, 1.2617e-03,
          1.9407e-03, 5.3978e-03, 2.3468e-02, 6.6490e-03, 1.9684e-02, 2.0885e-03,
          1.6556e-03, 3.3569e-03, 1.5860e-03, 5.5389e-03, 1.7654e-02, 2.3697e-02,
          1.1841e-02, 6.0501e-03, 3.4199e-03, 6.6185e-03, 9.6655e-04, 9.6989e-04,
          1.4429e-03, 1.1234e-03, 1.8282e-03, 3.8414e-03, 9.7122e-03, 9.0027e-03,
          1.5213e-02, 1.0239e-02, 1.0246e-02, 4.2648e-03, 3.5381e-03, 2.1534e-03,
          2.8687e-03, 3.8986e-03, 9.8267e-03, 7.7591e-03, 2.1698e-02, 1.0147e-02,
          2.7466e-03, 1.7958e-03, 8.0013e-04, 3.6545e-03, 1.3189e-03, 3.2921e-03,
          4.8332e-03, 9.6817e-03, 8.0261e-03, 1.3371e-03, 5.4207e-03, 1.3306e-02,
          4.3526e-03, 1.3893e-02, 1.7807e-02, 4.6806e-03, 2.4109e-03, 3.4237e-03,
          1.8997e-02, 1.4015e-02, 1.1854e-03, 5.9891e-03, 9.2239e-03, 3.2749e-03,
          3.1967e-03, 1.9493e-03, 1.7462e-03, 3.4752e-03, 3.1357e-03, 5.5122e-03,
          1.7197e-02, 1.7977e-03, 8.2474e-03, 8.3771e-03, 4.7668e-02, 5.8055e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.0884e-04,
          2.2054e-05, 2.2054e-05, 1.2350e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 5.8055e-05, 2.2054e-05, 5.8055e-05, 5.8055e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 5.8055e-05, 5.8055e-05, 5.8055e-05, 1.2350e-04, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05,
          5.8055e-05, 2.2054e-05, 2.2054e-05, 5.2452e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05, 2.2054e-05, 4.4703e-04,
          5.8055e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05, 2.2054e-05,
          2.2054e-05, 5.8055e-05, 2.2054e-05, 5.8055e-05, 5.8055e-05, 5.8055e-05,
          5.8055e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.2350e-04, 5.8055e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 1.4329e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.8055e-05, 1.4329e-04,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 5.8055e-05, 5.8055e-05, 5.8055e-05, 2.0874e-04,
          2.2054e-05, 2.2054e-05, 1.2350e-04, 2.2054e-05, 1.4329e-04, 2.2054e-05,
          2.2054e-05, 1.2350e-04, 1.3170e-03, 1.2350e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.7929e-04, 2.2054e-05,
          5.8055e-05, 2.2054e-05, 1.2350e-04, 6.9571e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 1.7929e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 1.2350e-04, 5.8055e-05, 2.4462e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 1.4329e-04, 2.2054e-05, 1.4329e-04, 2.2054e-05, 2.2054e-05,
          2.4462e-04, 2.2054e-05, 1.4329e-04, 1.7929e-04, 2.2054e-05, 2.0874e-04,
          2.0874e-04, 2.2054e-05, 1.2941e-03, 2.4462e-04, 1.4329e-04, 1.4329e-04,
          1.2598e-03, 1.7929e-04, 2.0874e-04, 2.2054e-05, 2.2054e-05, 1.2350e-04,
          2.2054e-05, 1.7929e-04, 5.8055e-05, 5.8055e-05, 2.2054e-05, 1.4329e-04,
          2.0874e-04, 5.8055e-05, 2.4462e-04, 1.4329e-04, 2.2054e-05, 1.2350e-04,
          2.2054e-05, 2.2054e-05, 5.6839e-04, 2.2054e-05, 1.7929e-04, 1.5039e-03,
          1.4329e-04, 1.7929e-04, 2.2054e-05, 2.2054e-05, 1.3745e-04, 7.1621e-04,
          1.8895e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.2350e-04, 1.4329e-04,
          9.0265e-04, 3.2425e-04, 1.4329e-04, 2.2054e-05, 5.8055e-05, 2.2054e-05,
          5.8055e-05, 1.4329e-04, 5.8055e-05, 3.3712e-04, 2.2054e-05, 2.2054e-05,
          1.4329e-04, 2.2054e-05, 2.2054e-05, 5.8055e-05, 8.3733e-04, 1.4329e-04,
          1.7929e-04, 1.2350e-04, 2.2054e-05, 2.1589e-04, 1.4329e-04, 5.5647e-04,
          1.7929e-04, 1.4329e-04, 2.4462e-04, 5.8055e-05, 2.2054e-05, 3.3712e-04,
          2.0874e-04, 3.7313e-04, 2.2054e-05, 2.7156e-04, 2.7156e-04, 2.2054e-05,
          8.7559e-05, 2.2054e-05, 8.7559e-05, 2.2054e-05, 2.5868e-04, 1.2350e-04,
          2.2054e-05, 2.5868e-04, 8.7559e-05, 4.8089e-04, 2.5868e-04, 2.5868e-04,
          2.2054e-05, 8.7559e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.7411e-03, 5.9113e-02, 1.5686e-02, 1.6876e-02, 6.3782e-02,
          5.4779e-02, 1.0674e-02, 2.5902e-03, 2.4891e-03, 6.7520e-03, 8.6060e-03,
          3.8490e-03, 3.0174e-03, 1.0824e-03, 9.5901e-03, 2.0103e-03, 3.0270e-03,
          1.7605e-03, 1.3609e-03, 4.6754e-04, 5.9032e-04, 4.0388e-04, 2.6569e-03,
          9.3079e-03, 1.1307e-02, 8.7357e-03, 6.9313e-03, 1.0033e-03, 3.6736e-03,
          1.5774e-03, 4.4861e-03, 1.3018e-03, 1.0658e-02, 5.0888e-03, 3.5877e-03,
          1.1177e-02, 2.5234e-03, 4.3259e-03, 6.0081e-03, 7.0381e-04, 3.4714e-03,
          1.2522e-03, 1.1120e-03, 9.7513e-04, 1.0271e-03, 3.0403e-03, 9.1705e-03,
          6.0892e-04, 5.7507e-04, 4.8943e-03, 5.3253e-03, 3.9520e-03, 3.4809e-03,
          1.6994e-03, 2.5749e-03, 1.8559e-03, 5.3215e-03, 4.2801e-03, 5.0774e-03,
          3.8586e-03, 1.9970e-03, 1.8740e-03, 2.7084e-03, 2.0027e-03, 2.1393e-02,
          2.0409e-03, 1.3790e-03, 2.0447e-03, 1.0633e-03, 7.5054e-04, 8.1730e-04,
          8.7738e-04, 1.1597e-02, 2.0580e-03, 4.2686e-03, 1.2197e-03, 1.4467e-03,
          2.4738e-03, 6.4545e-03, 3.8910e-03, 1.3008e-02, 3.5152e-03, 3.5114e-03,
          2.5539e-03, 4.6844e-03, 1.5459e-03, 4.1122e-03, 1.2388e-03, 1.2960e-03,
          6.9275e-03, 4.9553e-03, 1.2522e-03, 3.8223e-03, 1.8892e-03, 1.2980e-03,
          1.6489e-03, 1.2836e-03, 4.0665e-03, 9.1553e-03, 7.0152e-03, 9.0179e-03,
          1.3588e-02, 2.0809e-03, 2.9888e-03, 1.2093e-02, 2.1927e-02, 2.6264e-03,
          4.5357e-03, 2.0237e-03, 7.0038e-03, 1.4639e-03, 1.8358e-03, 1.8711e-03,
          8.0719e-03, 3.5915e-03, 9.0694e-04, 2.1500e-02, 1.9894e-03, 3.0708e-03,
          3.9215e-03, 1.8250e-02, 3.1929e-03, 1.9302e-03, 2.3300e-02, 3.0460e-03,
          6.3362e-03, 2.9449e-03, 4.4823e-03, 3.5706e-02, 5.1689e-03, 4.9171e-03,
          3.0880e-03, 1.5984e-03, 2.1458e-03, 1.0529e-03, 3.9940e-03, 1.2455e-03,
          8.6975e-03, 1.4544e-03, 2.4109e-03, 4.9057e-03, 6.2065e-03, 4.8599e-03,
          5.3749e-03, 2.7771e-03, 7.0343e-03, 7.0305e-03, 3.8300e-03, 7.0190e-02,
          2.1100e-05, 2.1100e-05, 2.1100e-05, 4.8518e-05, 2.1100e-05, 2.1100e-05,
          4.8518e-05, 2.3067e-04, 1.7822e-04, 2.3067e-04, 2.1100e-05, 5.8365e-04,
          2.1100e-05, 2.1100e-05, 1.1599e-04, 2.1100e-05, 2.1100e-05, 4.8518e-05,
          1.9968e-04, 9.6130e-04, 2.1100e-05, 2.1100e-05, 1.1599e-04, 2.1100e-05,
          2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05,
          2.1100e-05, 2.1100e-05, 2.1100e-05, 1.1599e-04, 4.8518e-05, 2.1100e-05,
          1.1075e-04, 5.5647e-04, 4.8518e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05,
          1.1599e-04, 2.1100e-05, 2.1100e-05, 3.5429e-04, 1.9968e-04, 2.1100e-05,
          2.1100e-05, 1.2863e-04, 2.1100e-05, 4.8518e-05, 4.8518e-05, 2.1100e-05,
          4.8518e-05, 4.8518e-05, 2.1100e-05, 4.8518e-05, 4.8518e-05, 3.0684e-04,
          1.7822e-04, 2.1100e-05, 2.1100e-05, 1.1599e-04, 2.1100e-05, 2.1100e-05,
          2.1100e-05, 1.1599e-04, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.2352e-04,
          2.1100e-05, 2.1100e-05, 4.8518e-05, 4.8518e-05, 2.1100e-05, 2.1100e-05,
          4.8518e-05, 4.8518e-05, 2.3067e-04, 2.1100e-05, 2.1100e-05, 6.2418e-04,
          2.1100e-05, 4.8518e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.6369e-04,
          2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 4.8518e-05, 4.3297e-04,
          2.1100e-05, 4.8518e-05, 2.1100e-05, 8.8632e-05, 2.1100e-05, 2.1100e-05,
          4.8518e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 1.2455e-03,
          2.1100e-05, 4.8518e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05,
          2.1100e-05, 2.1100e-05, 4.8518e-05, 2.2352e-04, 1.1599e-04, 2.1100e-05,
          1.1599e-04, 1.1599e-04, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05,
          2.1100e-05, 2.1100e-05, 6.2370e-04, 6.1703e-04, 2.1100e-05, 1.1599e-04,
          1.1599e-04, 2.1100e-05, 2.1100e-05, 2.1100e-05, 1.5593e-04, 2.1100e-05,
          2.6369e-04, 2.1100e-05, 1.1599e-04, 3.5429e-04, 2.1100e-05, 4.8518e-05,
          2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05, 2.1100e-05,
          2.1100e-05, 2.3937e-04, 2.6202e-04, 1.5593e-04, 2.1100e-05, 4.8518e-05,
          2.1100e-05, 1.5593e-04, 2.1100e-05, 1.5593e-04, 2.1100e-05, 2.1100e-05,
          1.5593e-04, 2.1100e-05, 2.4843e-04, 2.1100e-05, 4.8518e-05, 2.1100e-05,
          2.1100e-05, 2.8682e-04, 7.1096e-04, 5.5647e-04, 2.1100e-05, 2.1100e-05,
          3.0184e-04, 5.5647e-04, 3.1042e-04, 2.1100e-05, 2.1100e-05, 4.9543e-04,
          2.1100e-05, 1.5593e-04, 3.1590e-04, 8.8632e-05, 2.1100e-05, 1.5593e-04,
          1.5593e-04, 2.1100e-05, 4.1437e-04, 6.1798e-04, 3.1042e-04, 8.0252e-04,
          1.1075e-04, 2.1100e-05, 2.2352e-04, 3.5429e-04, 3.9530e-04, 7.1096e-04,
          9.5797e-04, 1.3123e-03, 4.8518e-05, 2.1100e-05, 1.9968e-04, 1.1599e-04,
          2.1100e-05, 2.8682e-04, 2.1100e-05, 8.3387e-05, 8.0156e-04, 2.1100e-05,
          6.2180e-04, 4.2200e-04, 1.5593e-04, 1.7822e-04, 1.9443e-04, 2.1100e-05,
          2.1100e-05, 9.3174e-04, 4.8518e-05, 1.1075e-04, 2.1100e-05, 8.3387e-05,
          4.6182e-04, 2.1100e-05, 4.8518e-05, 2.1100e-05, 2.2352e-04, 3.1042e-04,
          1.3752e-03, 6.4135e-04, 3.5429e-04, 1.1075e-04, 1.5593e-04, 9.7418e-04,
          3.6955e-04, 6.8045e-04, 1.1282e-03, 1.1075e-04, 1.1075e-04, 3.0184e-04,
          9.7418e-04, 5.8508e-04, 2.0301e-04, 6.1893e-04, 5.2023e-04, 2.1100e-05,
          1.3220e-04, 4.8518e-05, 1.3220e-04, 4.8518e-05, 1.3220e-04, 8.8632e-05,
          4.8518e-05, 4.8518e-05, 1.3220e-04, 3.0637e-04, 1.3220e-04, 5.0974e-04,
          4.8518e-05, 4.8518e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.6261e-03, 6.1493e-02, 1.6068e-02, 1.8661e-02, 5.0507e-02,
          5.3345e-02, 1.1208e-02, 2.4052e-03, 2.0561e-03, 1.2184e-02, 1.2001e-02,
          6.1264e-03, 2.5864e-03, 1.1806e-03, 4.9210e-03, 1.8215e-03, 3.0212e-03,
          1.6937e-03, 1.2264e-03, 5.0116e-04, 6.5470e-04, 4.3225e-04, 2.4986e-03,
          1.4084e-02, 5.3215e-03, 9.0790e-03, 6.9122e-03, 1.0242e-03, 3.4981e-03,
          1.7920e-03, 4.1237e-03, 1.3342e-03, 8.0109e-03, 6.6414e-03, 2.9011e-03,
          9.7504e-03, 2.6855e-03, 5.8289e-03, 5.2834e-03, 7.9060e-04, 7.7133e-03,
          1.6232e-03, 1.4563e-03, 1.2608e-03, 1.7595e-03, 2.7580e-03, 6.4888e-03,
          6.3229e-04, 5.3310e-04, 4.5509e-03, 4.4174e-03, 4.2877e-03, 2.6703e-03,
          1.5535e-03, 2.5940e-03, 1.5049e-03, 6.0806e-03, 6.2256e-03, 7.3204e-03,
          4.2801e-03, 2.0771e-03, 1.7004e-03, 2.4223e-03, 3.4561e-03, 1.2680e-02,
          2.6226e-03, 2.3975e-03, 3.7403e-03, 9.1887e-04, 7.9107e-04, 9.2793e-04,
          9.3985e-04, 1.6068e-02, 2.1553e-03, 4.6959e-03, 1.2188e-03, 1.5612e-03,
          2.6417e-03, 6.5422e-03, 6.0310e-03, 2.0737e-02, 7.1945e-03, 3.7289e-03,
          2.3060e-03, 5.0583e-03, 1.3914e-03, 6.5765e-03, 1.8997e-03, 1.7529e-03,
          1.2054e-02, 1.1719e-02, 1.1854e-03, 7.6332e-03, 1.1368e-03, 1.1082e-03,
          1.6699e-03, 1.1215e-03, 4.2267e-03, 8.6594e-03, 6.3705e-03, 1.0193e-02,
          1.9867e-02, 2.6169e-03, 5.9052e-03, 1.5182e-02, 1.3611e-02, 2.4757e-03,
          4.4212e-03, 2.4796e-03, 1.1124e-02, 1.6499e-03, 1.9379e-03, 3.2558e-03,
          3.4542e-03, 2.7504e-03, 7.0810e-04, 1.0803e-02, 1.3094e-03, 2.4681e-03,
          3.8586e-03, 2.1454e-02, 2.8629e-03, 1.9913e-03, 1.6937e-02, 3.5400e-03,
          4.6310e-03, 4.0283e-03, 5.5923e-03, 1.4793e-02, 4.2992e-03, 4.8943e-03,
          3.4676e-03, 2.3174e-03, 1.4048e-03, 1.7672e-03, 8.0109e-03, 1.3075e-03,
          6.1951e-03, 1.3847e-03, 1.7490e-03, 3.8662e-03, 4.9629e-03, 5.1689e-03,
          5.4054e-03, 2.2240e-03, 6.3286e-03, 9.3918e-03, 5.0888e-03, 1.8433e-02,
          5.6000e-02, 2.9147e-05, 2.9147e-05, 8.6427e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 6.1703e-04,
          2.9147e-05, 2.9147e-05, 1.3328e-04, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04, 7.5936e-05, 2.9147e-05,
          2.9147e-05, 5.2834e-04, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          1.3328e-04, 2.9147e-05, 2.9147e-05, 3.2067e-04, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 7.5936e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 8.6427e-05, 2.9147e-05, 3.8052e-04,
          1.3328e-04, 2.9147e-05, 2.9147e-05, 8.6427e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 1.3328e-04, 2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 8.6427e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 8.2588e-04,
          2.9147e-05, 2.9147e-05, 8.6427e-05, 2.9147e-05, 2.9147e-05, 8.6427e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04, 3.3903e-04,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.3425e-04, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 8.6427e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 8.6427e-05, 1.3328e-04, 1.3328e-04, 2.9147e-05,
          1.3328e-04, 8.6427e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 7.6866e-04, 1.6427e-04, 2.9147e-05, 8.6427e-05,
          8.6427e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04, 2.9147e-05,
          8.6427e-05, 2.9147e-05, 1.3328e-04, 1.3328e-04, 2.9147e-05, 7.5936e-05,
          2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 7.5936e-05, 1.3328e-04, 1.3328e-04, 2.9147e-05, 2.9147e-05,
          2.9147e-05, 4.7231e-04, 2.9147e-05, 1.3328e-04, 2.9147e-05, 2.9147e-05,
          3.8052e-04, 2.9147e-05, 6.7759e-04, 8.6427e-05, 7.5936e-05, 2.9147e-05,
          2.9147e-05, 2.9147e-05, 1.0605e-03, 5.8556e-04, 2.3425e-04, 2.9147e-05,
          5.8556e-04, 7.9489e-04, 7.5936e-05, 2.9147e-05, 2.9147e-05, 1.3387e-04,
          2.9147e-05, 3.4285e-04, 1.3387e-04, 2.9147e-05, 2.9147e-05, 4.7231e-04,
          2.9147e-05, 2.9147e-05, 3.8052e-04, 6.7759e-04, 2.9147e-05, 1.3328e-04,
          2.9147e-05, 2.9147e-05, 3.2067e-04, 2.9147e-05, 4.2653e-04, 7.3767e-04,
          7.6866e-04, 1.3328e-04, 2.9147e-05, 2.9147e-05, 1.3328e-04, 1.3328e-04,
          2.7490e-04, 7.5936e-05, 2.9147e-05, 2.9147e-05, 1.3387e-04, 2.9147e-05,
          8.8835e-04, 2.6941e-04, 7.5936e-05, 2.9147e-05, 2.6941e-04, 2.9147e-05,
          8.6427e-05, 5.7840e-04, 8.6427e-05, 7.5936e-05, 2.9147e-05, 2.9147e-05,
          3.4237e-04, 2.9147e-05, 2.9147e-05, 2.9147e-05, 1.3328e-04, 2.3818e-04,
          8.8835e-04, 1.3328e-04, 2.9147e-05, 7.5936e-05, 2.8491e-04, 1.2865e-03,
          1.3328e-04, 6.7759e-04, 5.4741e-04, 2.9147e-05, 2.9147e-05, 4.1747e-04,
          1.0605e-03, 1.2836e-03, 2.9147e-05, 8.1396e-04, 6.2656e-04, 2.9147e-05,
          7.5936e-05, 2.9147e-05, 2.9147e-05, 2.9147e-05, 7.5936e-05, 2.9147e-05,
          2.9147e-05, 7.5936e-05, 7.5936e-05, 3.7050e-04, 7.5936e-05, 4.8614e-04,
          7.5936e-05, 2.9147e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.9629e-03, 5.5695e-02, 1.4557e-02, 1.4557e-02, 5.0781e-02,
          4.2755e-02, 9.0942e-03, 1.5469e-03, 1.6699e-03, 4.9133e-03, 5.3215e-03,
          2.8839e-03, 2.4910e-03, 1.0920e-03, 6.6605e-03, 1.4038e-03, 3.1910e-03,
          1.6146e-03, 9.6512e-04, 5.2357e-04, 6.5374e-04, 3.5405e-04, 2.6264e-03,
          6.2904e-03, 5.2261e-03, 7.6752e-03, 5.2147e-03, 9.4938e-04, 3.6545e-03,
          1.5860e-03, 3.6011e-03, 1.3599e-03, 1.0735e-02, 4.5242e-03, 3.4142e-03,
          1.4061e-02, 2.8820e-03, 4.2610e-03, 5.8365e-03, 8.0967e-04, 4.4785e-03,
          1.1473e-03, 1.0090e-03, 8.8882e-04, 1.0605e-03, 2.3727e-03, 9.3002e-03,
          5.2929e-04, 5.0449e-04, 3.4294e-03, 6.0501e-03, 3.8643e-03, 2.6436e-03,
          1.5173e-03, 2.7542e-03, 1.9035e-03, 7.3471e-03, 5.5046e-03, 7.8125e-03,
          4.7646e-03, 2.8133e-03, 1.9951e-03, 3.1261e-03, 2.0618e-03, 2.0859e-02,
          1.4229e-03, 1.0490e-03, 1.9293e-03, 8.6212e-04, 6.1178e-04, 7.1430e-04,
          8.2254e-04, 7.9880e-03, 2.4223e-03, 4.4289e-03, 1.1559e-03, 1.5173e-03,
          3.1204e-03, 7.8201e-03, 4.0054e-03, 1.5587e-02, 3.6850e-03, 5.3024e-03,
          3.3836e-03, 5.3558e-03, 1.6451e-03, 2.5787e-03, 9.2268e-04, 1.0185e-03,
          5.0774e-03, 5.2147e-03, 1.0633e-03, 3.1471e-03, 1.5678e-03, 1.2255e-03,
          1.6918e-03, 1.3990e-03, 4.2458e-03, 1.0620e-02, 7.3509e-03, 8.3008e-03,
          9.8267e-03, 1.4381e-03, 2.8248e-03, 2.0844e-02, 2.7039e-02, 3.4065e-03,
          5.5847e-03, 2.7027e-03, 7.0610e-03, 1.2350e-03, 1.2598e-03, 1.7490e-03,
          3.3703e-03, 3.7575e-03, 8.2016e-04, 2.0981e-02, 1.7166e-03, 2.2659e-03,
          3.8357e-03, 1.5503e-02, 2.4700e-03, 1.4029e-03, 1.9974e-02, 2.4643e-03,
          7.3547e-03, 3.8376e-03, 5.1613e-03, 4.3671e-02, 4.6310e-03, 6.2027e-03,
          2.2831e-03, 1.1263e-03, 1.0815e-03, 8.3351e-04, 4.0588e-03, 1.2693e-03,
          8.5907e-03, 1.5421e-03, 1.4353e-03, 3.6049e-03, 5.3101e-03, 5.1003e-03,
          3.8109e-03, 1.5869e-03, 5.3940e-03, 7.2937e-03, 2.9984e-03, 3.9642e-02,
          3.1769e-02, 5.6549e-02, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          5.9128e-05, 2.6870e-04, 5.9128e-05, 2.6870e-04, 2.0444e-05, 6.6710e-04,
          2.0444e-05, 2.0444e-05, 2.5630e-04, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          5.9128e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 1.3494e-04, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 1.3494e-04, 5.9128e-05, 2.0444e-05,
          5.9128e-05, 7.3862e-04, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          1.3494e-04, 2.0444e-05, 2.0444e-05, 4.5204e-04, 5.9128e-05, 2.0444e-05,
          2.0444e-05, 1.6785e-04, 2.0444e-05, 5.9128e-05, 5.9128e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 9.6202e-05, 5.9128e-05, 6.3229e-04,
          1.3494e-04, 2.0444e-05, 2.0444e-05, 9.6202e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 1.3494e-04, 2.0444e-05, 2.0444e-05, 2.0444e-05, 1.3494e-04,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          5.9128e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 5.9128e-04,
          2.0444e-05, 2.0444e-05, 9.6202e-05, 2.0444e-05, 2.0444e-05, 2.5988e-04,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 3.7432e-04,
          2.0444e-05, 5.9128e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.3384e-03,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 1.3494e-04, 1.3494e-04, 2.0444e-05,
          1.3494e-04, 9.6202e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 7.7486e-04, 3.8314e-04, 2.0444e-05, 1.3494e-04,
          9.6202e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.5988e-04, 2.0444e-05, 5.9128e-05, 4.5204e-04, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 3.0470e-04, 1.3494e-04, 5.9128e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.9850e-04, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          5.9128e-05, 2.0444e-05, 2.9850e-04, 9.6202e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 5.9128e-05, 6.8665e-04, 3.4332e-04, 2.0444e-05, 2.0444e-05,
          3.4332e-04, 7.0000e-04, 2.9850e-04, 2.0444e-05, 2.0444e-05, 2.2686e-04,
          2.0444e-05, 2.0659e-04, 3.7456e-04, 9.6202e-05, 2.0444e-05, 2.9850e-04,
          2.0444e-05, 2.0444e-05, 2.8229e-04, 2.9850e-04, 5.9128e-05, 3.7432e-04,
          2.0444e-05, 2.0444e-05, 3.7432e-04, 2.3866e-04, 4.4775e-04, 9.3460e-04,
          8.6689e-04, 4.9543e-04, 2.0444e-05, 2.0444e-05, 1.3494e-04, 1.3494e-04,
          2.0444e-05, 1.5116e-04, 2.0444e-05, 2.0444e-05, 3.7432e-04, 2.0444e-05,
          5.9938e-04, 1.3494e-04, 1.6785e-04, 5.9128e-05, 1.6284e-04, 2.0444e-05,
          2.0444e-05, 6.2227e-04, 2.0444e-05, 5.9128e-05, 2.0444e-05, 2.0444e-05,
          3.7432e-04, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.8229e-04, 2.9850e-04,
          5.0735e-04, 6.7282e-04, 5.9128e-05, 5.9128e-05, 1.6785e-04, 1.8768e-03,
          3.8600e-04, 6.5041e-04, 1.2894e-03, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          9.4891e-04, 5.2357e-04, 2.1601e-04, 4.0221e-04, 6.5041e-04, 2.0444e-05,
          5.9128e-05, 2.0444e-05, 1.8048e-04, 5.9128e-05, 1.8048e-04, 9.6202e-05,
          2.0444e-05, 1.3494e-04, 5.9128e-05, 3.5763e-04, 3.9625e-04, 6.0940e-04,
          5.9128e-05, 5.9128e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.0616e-03, 5.3253e-02, 1.8555e-02, 1.3504e-02, 7.2021e-02,
          3.8971e-02, 1.2978e-02, 1.4029e-03, 1.5726e-03, 4.0855e-03, 5.4817e-03,
          2.4490e-03, 1.7138e-03, 1.3027e-03, 3.1147e-03, 1.0929e-03, 2.0027e-03,
          9.6607e-04, 7.9918e-04, 1.1578e-03, 1.0996e-03, 7.0381e-04, 1.2941e-03,
          5.7755e-03, 4.8218e-03, 1.0056e-02, 3.0479e-03, 1.5049e-03, 1.7872e-03,
          1.3733e-03, 2.3041e-03, 9.8610e-04, 4.4403e-03, 2.2144e-03, 2.0256e-03,
          1.0376e-02, 6.7253e-03, 2.7370e-03, 2.4776e-03, 1.4038e-03, 4.4136e-03,
          2.7046e-03, 2.1114e-03, 2.1572e-03, 1.9312e-03, 1.3685e-03, 4.5700e-03,
          7.5722e-04, 9.9659e-04, 6.9160e-03, 3.7651e-03, 2.6073e-03, 1.5974e-03,
          2.1706e-03, 2.3937e-03, 3.3340e-03, 1.6312e-02, 6.5231e-03, 2.0355e-02,
          1.0757e-02, 3.4428e-03, 2.7637e-03, 2.7905e-03, 3.2482e-03, 7.3280e-03,
          2.3785e-03, 1.1911e-03, 1.4906e-03, 2.2354e-03, 2.1439e-03, 2.4223e-03,
          2.6455e-03, 5.0507e-03, 3.6411e-03, 3.0956e-03, 1.8005e-03, 1.6346e-03,
          3.4027e-03, 1.7044e-02, 7.5455e-03, 9.2010e-03, 6.6605e-03, 1.1909e-02,
          9.0408e-03, 3.9177e-03, 2.6226e-03, 2.2087e-03, 1.1702e-03, 1.8167e-03,
          3.8795e-03, 4.7188e-03, 2.0638e-03, 2.9812e-03, 4.2839e-03, 3.7537e-03,
          2.4147e-03, 2.4128e-03, 1.8129e-03, 7.2556e-03, 2.0370e-02, 1.2253e-02,
          6.9542e-03, 2.0199e-03, 2.8782e-03, 1.8616e-02, 9.2850e-03, 5.0011e-03,
          4.0588e-03, 4.3831e-03, 5.1727e-03, 2.3575e-03, 2.3804e-03, 1.8482e-03,
          2.9411e-03, 5.2299e-03, 1.8034e-03, 7.6714e-03, 4.8828e-03, 4.7836e-03,
          4.6387e-03, 1.1383e-02, 6.6414e-03, 1.2226e-03, 6.1188e-03, 4.5738e-03,
          1.9302e-02, 8.3313e-03, 9.7275e-03, 1.3123e-02, 2.2354e-03, 4.6921e-03,
          3.6030e-03, 1.5583e-03, 6.4898e-04, 1.1206e-03, 3.0460e-03, 4.1618e-03,
          3.3627e-03, 3.3016e-03, 1.6127e-03, 2.6932e-03, 3.0212e-03, 4.5319e-03,
          7.9346e-03, 1.8511e-03, 6.0310e-03, 1.0239e-02, 4.9934e-03, 2.0233e-02,
          1.9150e-02, 1.9516e-02, 7.1472e-02, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04,
          2.3007e-05, 2.3007e-05, 1.7309e-04, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 7.3195e-05, 7.3195e-05, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          3.3689e-04, 2.3007e-05, 2.3007e-05, 1.7309e-04, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04,
          7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 1.2290e-04,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 7.3195e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04, 1.7309e-04, 2.3007e-05,
          7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 7.3195e-05, 1.7309e-04, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 7.3195e-05, 2.3007e-05,
          1.2290e-04, 2.3007e-05, 1.7309e-04, 1.7309e-04, 2.3007e-05, 7.3195e-05,
          2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 3.3689e-04, 7.3195e-05, 2.3007e-05, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05,
          7.3195e-05, 2.3007e-05, 2.3007e-05, 1.2290e-04, 7.3195e-05, 2.3007e-05,
          7.3195e-05, 7.3195e-05, 7.3195e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05,
          7.3195e-05, 7.3195e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05, 1.7309e-04,
          2.3007e-05, 7.3195e-05, 1.7357e-04, 2.3007e-05, 2.3007e-05, 7.3195e-05,
          7.3195e-05, 2.3007e-05, 7.3195e-05, 7.3195e-05, 2.3007e-05, 1.7309e-04,
          2.3007e-05, 2.3007e-05, 1.7309e-04, 7.3195e-05, 2.3007e-05, 7.3195e-05,
          7.3195e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05, 3.3689e-04, 1.7309e-04,
          2.3007e-05, 2.3687e-04, 2.3007e-05, 2.3007e-05, 3.3689e-04, 2.3007e-05,
          5.2309e-04, 5.2309e-04, 7.3195e-05, 7.3195e-05, 4.2319e-04, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 2.3007e-05, 7.3195e-05, 2.3007e-05, 2.3007e-05,
          7.3195e-05, 2.3007e-05, 2.3007e-05, 2.3007e-05, 3.3689e-04, 2.3007e-05,
          2.3687e-04, 3.3689e-04, 7.3195e-05, 7.3195e-05, 7.3195e-05, 4.2319e-04,
          2.3687e-04, 7.3195e-05, 3.3689e-04, 7.3195e-05, 2.3007e-05, 7.3195e-05,
          2.4586e-03, 4.2319e-04, 2.3007e-05, 7.3195e-05, 7.3195e-05, 2.3007e-05,
          7.3195e-05, 7.3195e-05, 7.3195e-05, 3.4189e-04, 7.3195e-05, 2.3007e-05,
          2.3007e-05, 7.3195e-05, 7.3195e-05, 2.5368e-04, 2.5368e-04, 4.6754e-04,
          2.3007e-05, 2.3007e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.7831e-03, 5.5908e-02, 1.9135e-02, 1.7212e-02, 5.5542e-02,
          4.2572e-02, 1.5282e-02, 3.9597e-03, 2.2926e-03, 2.5177e-03, 2.9907e-03,
          1.4048e-03, 2.0599e-03, 1.8978e-03, 2.6417e-03, 5.1918e-03, 5.6801e-03,
          2.7351e-03, 1.5392e-03, 1.8282e-03, 2.2717e-03, 1.8234e-03, 2.9354e-03,
          3.3054e-03, 2.7847e-03, 1.1978e-02, 6.3858e-03, 2.3136e-03, 4.7798e-03,
          1.9274e-03, 9.3460e-03, 2.7542e-03, 4.9629e-03, 2.0218e-03, 8.0795e-03,
          5.6686e-03, 2.4872e-03, 6.1150e-03, 4.2915e-03, 2.3403e-03, 6.5155e-03,
          2.8629e-03, 2.1229e-03, 1.7672e-03, 1.8463e-03, 3.0193e-03, 5.8594e-03,
          1.5249e-03, 1.7843e-03, 7.9269e-03, 1.6846e-02, 1.1124e-02, 3.3073e-03,
          3.4657e-03, 6.7215e-03, 5.3978e-03, 3.3131e-03, 4.6883e-03, 3.5324e-03,
          3.0537e-03, 9.5825e-03, 2.3712e-02, 2.7786e-02, 2.9793e-03, 6.7863e-03,
          1.8673e-03, 9.7275e-04, 1.5993e-03, 1.7004e-03, 1.4410e-03, 1.7672e-03,
          2.1133e-03, 4.3716e-03, 7.3738e-03, 1.3260e-02, 2.9755e-03, 2.9621e-03,
          6.7863e-03, 3.1757e-03, 1.9855e-03, 2.9793e-03, 1.8005e-03, 2.7046e-03,
          3.0499e-03, 4.3945e-02, 1.5945e-02, 1.6317e-03, 9.6607e-04, 1.1606e-03,
          2.8896e-03, 3.2196e-03, 1.1444e-03, 2.0847e-03, 2.0313e-03, 2.2125e-03,
          3.9635e-03, 3.3073e-03, 7.8812e-03, 8.0490e-03, 2.5444e-03, 2.5635e-03,
          2.6894e-03, 9.6893e-04, 1.0462e-03, 4.5357e-03, 4.3640e-03, 9.5596e-03,
          2.8854e-02, 6.2714e-03, 4.6120e-03, 2.0885e-03, 1.4639e-03, 1.4553e-03,
          1.5259e-03, 5.9013e-03, 3.9005e-03, 6.3019e-03, 2.7866e-03, 4.9400e-03,
          5.5885e-03, 3.8376e-03, 1.2436e-03, 2.3174e-03, 3.5896e-03, 9.6130e-04,
          2.0580e-03, 1.7300e-03, 2.0351e-03, 5.9395e-03, 1.0063e-02, 2.8290e-02,
          2.8267e-03, 1.1606e-03, 6.8092e-04, 9.6512e-04, 2.4261e-03, 2.6550e-03,
          8.3542e-03, 4.6997e-03, 2.6875e-03, 4.2953e-03, 5.2071e-03, 4.6806e-03,
          1.6832e-03, 7.7438e-04, 1.6041e-03, 2.1610e-03, 1.4915e-03, 5.3329e-03,
          5.8441e-03, 5.8479e-03, 5.0163e-03, 8.0261e-02, 1.5199e-05, 1.5199e-05,
          1.4758e-04, 1.5199e-05, 1.4758e-04, 6.1035e-05, 1.5199e-05, 6.1035e-05,
          1.5199e-05, 1.5199e-05, 1.0169e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          1.4758e-04, 1.5199e-05, 6.1035e-05, 6.1035e-05, 1.0169e-04, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          6.1035e-05, 6.1035e-05, 6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          1.4758e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 6.1035e-05,
          1.5199e-05, 1.5199e-05, 6.1035e-05, 1.4758e-04, 1.4758e-04, 6.1035e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 6.1035e-05, 6.1035e-05,
          6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 1.4758e-04,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1750e-05, 6.1035e-05, 6.1035e-05,
          1.5199e-05, 1.0169e-04, 1.5199e-05, 1.4758e-04, 1.5199e-05, 1.4758e-04,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 1.4758e-04, 6.1035e-05,
          6.1035e-05, 6.1035e-05, 1.5199e-05, 6.1035e-05, 1.5199e-05, 6.1035e-05,
          1.5199e-05, 1.5199e-05, 6.1035e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.4758e-04, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.0169e-04,
          1.5199e-05, 6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 1.4758e-04, 1.4758e-04, 1.5199e-05,
          6.1035e-05, 6.1095e-05, 1.4758e-04, 1.5199e-05, 1.5199e-05, 6.1035e-05,
          1.5199e-05, 1.0169e-04, 1.5199e-05, 1.4758e-04, 1.5199e-05, 6.1035e-05,
          6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.0169e-04, 1.5199e-05,
          6.1035e-05, 1.5199e-05, 1.0169e-04, 1.0169e-04, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05,
          1.5199e-05, 1.0169e-04, 1.5199e-05, 1.0169e-04, 1.5199e-05, 6.1035e-05,
          1.0169e-04, 6.1035e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.4758e-04,
          6.1035e-05, 1.4758e-04, 1.0169e-04, 1.0169e-04, 1.5199e-05, 1.5199e-05,
          1.0169e-04, 1.4758e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.4758e-04,
          1.5199e-05, 1.4758e-04, 1.4758e-04, 6.1035e-05, 6.1035e-05, 1.5199e-05,
          1.0169e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05, 6.1035e-05, 1.4758e-04,
          6.1035e-05, 6.1035e-05, 1.4758e-04, 6.1035e-05, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.0169e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.0169e-04, 6.1035e-05, 1.4758e-04, 1.4758e-04, 1.0169e-04,
          1.0169e-04, 1.0169e-04, 1.5199e-05, 6.1035e-05, 1.5199e-05, 6.1035e-05,
          1.4758e-04, 1.0169e-04, 1.5199e-05, 1.4758e-04, 1.5199e-05, 1.5199e-05,
          1.0169e-04, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05, 1.5199e-05,
          1.0169e-04, 1.4758e-04, 1.4758e-04, 1.5199e-05, 1.5199e-05, 1.4758e-04,
          1.5199e-05, 1.5199e-05, 1.4758e-04, 1.5199e-05, 6.1035e-05, 1.0169e-04,
          1.0169e-04, 1.0169e-04, 1.5199e-05, 1.0169e-04, 1.5199e-05, 1.5199e-05,
          1.5199e-05, 1.5199e-05, 1.0169e-04, 1.5199e-05, 1.0169e-04, 1.5199e-05,
          1.5199e-05, 1.0169e-04, 1.5199e-05, 3.9220e-04, 3.9220e-04, 1.0169e-04,
          1.5199e-05, 1.5199e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.7487e-03, 4.4952e-02, 2.1759e-02, 1.8143e-02, 7.2998e-02,
          3.3051e-02, 1.7426e-02, 4.3030e-03, 1.6804e-03, 1.2426e-03, 2.0714e-03,
          8.8453e-04, 1.1072e-03, 1.6003e-03, 1.6642e-03, 4.0207e-03, 1.7910e-03,
          1.2836e-03, 1.1425e-03, 1.5736e-03, 1.4896e-03, 1.4772e-03, 1.4133e-03,
          2.3994e-03, 2.1420e-03, 1.4351e-02, 3.9558e-03, 2.0161e-03, 1.6966e-03,
          1.5821e-03, 6.5994e-03, 1.0757e-03, 1.7881e-03, 8.5115e-04, 4.6234e-03,
          2.7580e-03, 2.7065e-03, 1.8358e-03, 1.5364e-03, 2.0332e-03, 2.4567e-03,
          2.5558e-03, 1.7014e-03, 1.9836e-03, 1.0128e-03, 1.3962e-03, 2.1591e-03,
          1.1282e-03, 1.8349e-03, 1.0551e-02, 1.0109e-02, 8.0414e-03, 2.2755e-03,
          6.8512e-03, 6.4240e-03, 9.0485e-03, 2.0428e-03, 8.9188e-03, 2.9488e-03,
          4.0054e-03, 4.7874e-03, 4.8798e-02, 1.4359e-02, 1.9646e-03, 3.0079e-03,
          2.0523e-03, 7.2145e-04, 9.4175e-04, 2.0771e-03, 2.0199e-03, 2.3441e-03,
          2.5692e-03, 2.2507e-03, 1.2047e-02, 1.1497e-02, 1.0345e-02, 4.8218e-03,
          8.6365e-03, 1.7853e-03, 1.4610e-03, 1.6184e-03, 1.6193e-03, 2.8477e-03,
          3.5248e-03, 4.5532e-02, 2.8656e-02, 1.0204e-03, 8.0633e-04, 1.2178e-03,
          1.7500e-03, 1.3790e-03, 1.2836e-03, 1.3371e-03, 2.4738e-03, 2.5406e-03,
          3.0479e-03, 5.6190e-03, 4.8256e-03, 2.5501e-03, 1.9627e-03, 1.4544e-03,
          1.6813e-03, 9.1791e-04, 8.8930e-04, 2.9602e-03, 2.6684e-03, 5.4893e-03,
          3.3447e-02, 5.7182e-03, 2.0981e-03, 2.9907e-03, 1.7014e-03, 1.1444e-03,
          1.3332e-03, 3.7060e-03, 2.9278e-03, 3.3550e-03, 3.4790e-03, 5.5771e-03,
          4.5090e-03, 2.0504e-03, 1.4524e-03, 2.8572e-03, 2.2430e-03, 9.1267e-04,
          2.0447e-03, 1.8663e-03, 2.1172e-03, 3.6411e-03, 4.1046e-03, 3.3752e-02,
          3.3665e-03, 1.0490e-03, 5.9557e-04, 9.8801e-04, 1.4257e-03, 4.5471e-03,
          2.7084e-03, 3.6983e-03, 2.1267e-03, 3.5114e-03, 2.5806e-03, 3.4962e-03,
          1.7653e-03, 8.0299e-04, 1.2283e-03, 1.4210e-03, 1.5221e-03, 3.6716e-03,
          3.2463e-03, 3.9558e-03, 5.1575e-03, 3.4332e-02, 1.1298e-01, 2.6464e-05,
          8.5175e-05, 2.6464e-05, 1.5438e-04, 8.5175e-05, 2.6464e-05, 8.5175e-05,
          2.6464e-05, 2.6464e-05, 9.5725e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          1.5438e-04, 2.6464e-05, 8.5175e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 8.5175e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          1.5438e-04, 2.6464e-05, 2.6464e-05, 2.6464e-05, 8.5175e-05, 8.5175e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 4.2987e-04, 1.5438e-04, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          8.5175e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 1.5438e-04, 1.5438e-04,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 8.5235e-05, 8.5175e-05, 8.5175e-05,
          2.6464e-05, 9.5665e-05, 2.6464e-05, 1.5438e-04, 2.6464e-05, 1.5438e-04,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 1.5438e-04, 2.6464e-05,
          8.5175e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 8.5175e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 8.5175e-05, 2.6464e-05,
          2.6464e-05, 9.5665e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 8.5175e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 1.5438e-04, 8.5175e-05, 2.6464e-05,
          2.6464e-05, 8.5235e-05, 1.5438e-04, 2.6464e-05, 2.6464e-05, 8.5175e-05,
          2.6464e-05, 9.5665e-05, 2.6464e-05, 1.5438e-04, 2.6464e-05, 8.5175e-05,
          8.5175e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 9.5665e-05, 9.5665e-05,
          8.5175e-05, 2.6464e-05, 9.5725e-05, 3.7098e-04, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 9.5665e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05, 1.5438e-04,
          2.6464e-05, 9.5665e-05, 9.5665e-05, 9.5665e-05, 2.6464e-05, 1.5438e-04,
          9.5665e-05, 9.5665e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05, 9.5665e-05,
          9.5665e-05, 1.5438e-04, 9.5665e-05, 9.5665e-05, 2.6464e-05, 2.6464e-05,
          9.5665e-05, 9.5665e-05, 9.5665e-05, 2.6464e-05, 2.6524e-05, 1.5438e-04,
          2.6464e-05, 1.5438e-04, 1.5545e-04, 2.6464e-05, 2.6464e-05, 9.5665e-05,
          9.5665e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 9.5665e-05, 1.5438e-04,
          2.6464e-05, 8.5175e-05, 1.5438e-04, 1.5438e-04, 2.6464e-05, 9.5665e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 8.5175e-05, 9.5665e-05, 1.5438e-04, 9.5665e-05,
          9.5665e-05, 9.5665e-05, 2.6464e-05, 8.5175e-05, 2.6464e-05, 2.6464e-05,
          1.5438e-04, 9.5725e-05, 2.6464e-05, 3.7098e-04, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05,
          9.5665e-05, 1.5438e-04, 4.2987e-04, 9.5665e-05, 9.5665e-05, 9.5665e-05,
          9.5665e-05, 2.6464e-05, 1.5438e-04, 8.5175e-05, 1.5438e-04, 9.5665e-05,
          9.5665e-05, 9.5665e-05, 2.6464e-05, 9.5665e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 9.5665e-05, 9.5665e-05, 1.1760e-04, 9.5665e-05, 9.5665e-05,
          2.6464e-05, 9.5665e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05, 2.6464e-05,
          2.6464e-05, 2.6464e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.9504e-03, 5.4413e-02, 1.9485e-02, 1.8372e-02, 6.7566e-02,
          4.1656e-02, 1.6830e-02, 9.8419e-04, 1.1711e-03, 4.2686e-03, 4.9171e-03,
          2.4033e-03, 1.6603e-03, 1.3533e-03, 2.8439e-03, 1.1082e-03, 2.4109e-03,
          1.4286e-03, 8.7690e-04, 1.5001e-03, 2.3403e-03, 1.6165e-03, 1.8425e-03,
          6.6605e-03, 4.7073e-03, 1.2474e-02, 5.1231e-03, 2.0237e-03, 2.2984e-03,
          1.6699e-03, 1.5926e-03, 1.2255e-03, 4.8141e-03, 2.5578e-03, 1.5621e-03,
          9.0866e-03, 3.2387e-03, 9.5367e-03, 3.5381e-03, 1.7471e-03, 1.8967e-02,
          9.6359e-03, 7.2899e-03, 5.2261e-03, 4.8141e-03, 1.9035e-03, 8.5144e-03,
          1.1845e-03, 1.7824e-03, 8.6365e-03, 3.0270e-03, 1.5345e-03, 1.3247e-03,
          1.0910e-03, 1.1578e-03, 1.4219e-03, 6.5193e-03, 2.7409e-03, 4.8790e-03,
          3.2387e-03, 3.7136e-03, 4.0741e-03, 6.5918e-03, 1.2131e-02, 1.9318e-02,
          5.6343e-03, 3.1071e-03, 4.8256e-03, 5.9166e-03, 5.2605e-03, 3.0479e-03,
          3.3073e-03, 7.3128e-03, 2.1114e-03, 1.9064e-03, 8.8263e-04, 8.4543e-04,
          1.4648e-03, 7.9880e-03, 4.6844e-03, 6.7101e-03, 3.8872e-03, 2.4796e-03,
          2.6855e-03, 5.6038e-03, 3.1376e-03, 4.2152e-03, 2.8496e-03, 3.9368e-03,
          8.1329e-03, 1.0040e-02, 4.3488e-03, 5.5695e-03, 4.4212e-03, 3.5439e-03,
          1.9169e-03, 1.3638e-03, 1.5583e-03, 5.6572e-03, 9.9716e-03, 6.7635e-03,
          5.5199e-03, 1.8425e-03, 1.9779e-03, 7.4196e-03, 6.5269e-03, 4.9133e-03,
          5.3215e-03, 2.0020e-02, 1.5121e-02, 5.8556e-03, 3.4904e-03, 4.2725e-03,
          4.9362e-03, 1.8204e-02, 4.3640e-03, 1.1971e-02, 5.5389e-03, 3.8261e-03,
          3.7613e-03, 9.6130e-03, 3.2883e-03, 7.0715e-04, 6.3171e-03, 2.1877e-03,
          5.4588e-03, 3.0804e-03, 3.2463e-03, 7.4883e-03, 2.1076e-03, 5.4512e-03,
          4.7073e-03, 3.3760e-03, 1.1187e-03, 3.1986e-03, 8.8043e-03, 9.9487e-03,
          6.5842e-03, 4.3945e-03, 2.0218e-03, 2.8744e-03, 3.2921e-03, 3.1910e-03,
          4.5280e-03, 1.6489e-03, 4.5547e-03, 4.2839e-03, 3.3360e-03, 7.2708e-03,
          1.0399e-02, 7.4806e-03, 5.8327e-03, 1.1566e-02, 6.3782e-03, 5.4535e-02,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.0222e-04,
          1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.0222e-04, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.0222e-04, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 2.1017e-04,
          1.1986e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05, 2.1017e-04,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1986e-04, 2.1017e-04, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.0222e-04, 1.1981e-05,
          1.1986e-04, 1.1981e-05, 1.0222e-04, 2.1017e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1986e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.0222e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.0222e-04, 1.0222e-04, 1.0222e-04, 1.1981e-05, 1.1981e-05,
          1.0222e-04, 1.0222e-04, 1.0222e-04, 1.1981e-05, 1.1981e-05, 2.1017e-04,
          1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.0222e-04, 1.1981e-05, 1.1986e-04, 1.1981e-05, 1.1981e-05, 2.1017e-04,
          1.1981e-05, 1.1981e-05, 2.1017e-04, 1.0222e-04, 1.1981e-05, 1.0222e-04,
          1.0222e-04, 1.1986e-04, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1986e-04,
          1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05,
          2.1017e-04, 2.1017e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 2.1017e-04, 1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.1981e-05, 1.1981e-05, 2.1017e-04, 1.1981e-05,
          1.0222e-04, 2.1017e-04, 1.0222e-04, 1.0222e-04, 1.1981e-05, 1.0222e-04,
          2.1017e-04, 1.1981e-05, 2.1017e-04, 1.1981e-05, 1.1981e-05, 1.1981e-05,
          1.0222e-04, 1.0222e-04, 1.1981e-05, 1.0222e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05, 1.0222e-04, 1.1981e-05, 1.0222e-04, 1.1981e-05,
          1.1981e-05, 1.0222e-04, 1.1981e-05, 4.2534e-04, 1.1981e-05, 1.1981e-05,
          1.1981e-05, 1.1981e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.1646e-03, 5.8228e-02, 1.6708e-02, 2.1652e-02, 4.1962e-02,
          3.9978e-02, 1.4252e-02, 1.5144e-03, 1.2894e-03, 6.5231e-03, 1.2047e-02,
          4.5967e-03, 1.9741e-03, 1.1969e-03, 3.1738e-03, 1.5984e-03, 3.1700e-03,
          2.0561e-03, 1.1969e-03, 4.9925e-04, 8.9264e-04, 7.6771e-04, 2.5501e-03,
          1.7532e-02, 5.3558e-03, 1.1681e-02, 5.9814e-03, 6.9571e-04, 3.4599e-03,
          1.8616e-03, 2.7199e-03, 1.0805e-03, 4.3640e-03, 4.2038e-03, 1.6966e-03,
          4.5509e-03, 1.5707e-03, 8.0719e-03, 3.7022e-03, 7.3814e-04, 1.0681e-02,
          4.1351e-03, 2.8419e-03, 2.8191e-03, 2.2678e-03, 3.7212e-03, 7.7591e-03,
          8.5258e-04, 7.2718e-04, 3.9787e-03, 2.3518e-03, 2.0924e-03, 1.7176e-03,
          7.1669e-04, 1.2608e-03, 6.5708e-04, 3.2501e-03, 2.8591e-03, 3.9253e-03,
          2.0161e-03, 1.9760e-03, 2.0313e-03, 3.6774e-03, 8.3771e-03, 1.7685e-02,
          7.5531e-03, 6.8207e-03, 4.8065e-03, 2.1000e-03, 1.6956e-03, 2.0676e-03,
          1.5755e-03, 1.4778e-02, 1.0357e-03, 2.4910e-03, 6.3753e-04, 8.0109e-04,
          1.2445e-03, 4.3182e-03, 7.5836e-03, 7.9041e-03, 6.9160e-03, 1.6994e-03,
          1.2312e-03, 5.8403e-03, 1.6136e-03, 1.2222e-02, 5.3062e-03, 4.8790e-03,
          2.5238e-02, 1.8127e-02, 2.4071e-03, 1.3466e-02, 2.0084e-03, 1.5459e-03,
          9.1457e-04, 5.3215e-04, 2.6150e-03, 4.6272e-03, 4.5052e-03, 6.0463e-03,
          1.9775e-02, 2.7180e-03, 5.2910e-03, 7.1030e-03, 6.1569e-03, 2.3479e-03,
          5.0468e-03, 7.3471e-03, 2.8473e-02, 3.2902e-03, 4.7073e-03, 8.8882e-03,
          4.3945e-03, 8.1100e-03, 1.4076e-03, 1.2726e-02, 2.2354e-03, 1.3065e-03,
          2.2335e-03, 9.1705e-03, 1.8387e-03, 9.8324e-04, 9.5749e-03, 3.2310e-03,
          2.5845e-03, 3.0594e-03, 3.9101e-03, 6.3171e-03, 3.4256e-03, 5.3711e-03,
          6.9008e-03, 6.9237e-03, 1.8301e-03, 4.5052e-03, 2.1759e-02, 3.5000e-03,
          9.7961e-03, 3.0041e-03, 2.5368e-03, 4.8141e-03, 4.8294e-03, 2.7981e-03,
          2.8992e-03, 1.7118e-03, 3.2406e-03, 5.4932e-03, 4.1046e-03, 7.5188e-03,
          1.7197e-02, 6.0196e-03, 3.0651e-03, 8.1711e-03, 3.3360e-03, 2.0950e-02,
          3.4912e-02, 1.4246e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 1.0180e-04,
          1.4246e-05, 1.4246e-05, 2.0194e-04, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 6.3181e-05, 1.0180e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 6.3181e-05,
          6.3181e-05, 1.4246e-05, 1.4246e-05, 2.0194e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.5295e-04, 5.2869e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 6.3181e-05, 1.4246e-05, 2.0194e-04,
          6.3181e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 5.2869e-05, 1.4246e-05, 1.4246e-05, 1.0180e-04, 2.0194e-04,
          6.3181e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.0180e-04, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.0180e-04,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 6.3181e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.5295e-04, 1.0180e-04, 1.5295e-04,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          5.2869e-05, 1.4246e-05, 1.4246e-05, 1.0180e-04, 1.0180e-04, 5.2869e-05,
          1.4246e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05,
          1.4246e-05, 6.3181e-05, 1.4246e-05, 1.0180e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 2.0194e-04, 5.2869e-05,
          6.3181e-05, 5.2869e-05, 2.0194e-04, 1.0180e-04, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 5.2869e-05, 6.3181e-05, 4.2725e-04, 1.4246e-05, 1.4246e-05,
          5.2869e-05, 3.7813e-04, 1.5295e-04, 1.0180e-04, 1.4246e-05, 1.4246e-05,
          1.5295e-04, 5.2869e-05, 1.4246e-05, 1.0180e-04, 5.2869e-05, 5.2869e-05,
          5.2869e-05, 1.4246e-05, 1.5295e-04, 2.0194e-04, 1.4246e-05, 5.2869e-05,
          1.5295e-04, 1.0180e-04, 1.5295e-04, 1.4246e-05, 1.4246e-05, 2.0194e-04,
          1.4246e-05, 3.2711e-04, 3.5501e-04, 6.3181e-05, 1.4246e-05, 5.2869e-05,
          5.2869e-05, 6.3181e-05, 2.0194e-04, 1.5295e-04, 1.4246e-05, 2.0194e-04,
          1.4246e-05, 1.4246e-05, 2.0194e-04, 1.4246e-05, 4.7374e-04, 1.5295e-04,
          5.2869e-05, 6.3181e-05, 1.4246e-05, 1.4246e-05, 6.3181e-05, 6.3181e-05,
          4.7374e-04, 5.2869e-05, 1.4246e-05, 1.4246e-05, 2.0194e-04, 5.2869e-05,
          3.2711e-04, 3.2711e-04, 1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05,
          6.3181e-05, 1.5295e-04, 6.3181e-05, 5.2869e-05, 1.4246e-05, 1.4246e-05,
          1.4246e-05, 1.4246e-05, 1.4246e-05, 1.4246e-05, 6.3181e-05, 1.4246e-05,
          4.2725e-04, 3.2711e-04, 1.4246e-05, 5.2869e-05, 5.2869e-05, 3.7813e-04,
          6.3181e-05, 5.2869e-05, 4.2725e-04, 1.4246e-05, 1.4246e-05, 1.5295e-04,
          3.7813e-04, 1.8291e-03, 1.4246e-05, 1.5295e-04, 1.5295e-04, 1.4246e-05,
          5.2869e-05, 5.2869e-05, 1.4246e-05, 1.4246e-05, 5.2869e-05, 5.2869e-05,
          1.4246e-05, 5.2869e-05, 5.2869e-05, 4.0460e-04, 5.2869e-05, 5.2869e-05,
          1.4246e-05, 1.4246e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9547e-03, 5.7617e-02, 1.7731e-02, 1.6129e-02, 4.9286e-02,
          3.6499e-02, 1.3672e-02, 1.1044e-03, 1.0958e-03, 3.7670e-03, 5.7793e-03,
          2.4624e-03, 1.5020e-03, 1.4544e-03, 2.2717e-03, 1.1826e-03, 2.3232e-03,
          1.4877e-03, 1.0099e-03, 1.0166e-03, 1.6432e-03, 1.4505e-03, 1.9875e-03,
          7.3853e-03, 5.0964e-03, 1.0513e-02, 5.0392e-03, 1.2054e-03, 2.1896e-03,
          2.0485e-03, 1.7662e-03, 1.0157e-03, 3.5400e-03, 2.2373e-03, 1.5993e-03,
          5.1994e-03, 1.9894e-03, 7.3395e-03, 2.9449e-03, 1.3828e-03, 1.2794e-02,
          6.6605e-03, 5.5580e-03, 4.5662e-03, 3.7575e-03, 2.3842e-03, 6.9504e-03,
          1.4067e-03, 1.6918e-03, 5.8289e-03, 2.3994e-03, 1.6050e-03, 1.4963e-03,
          9.6035e-04, 1.1578e-03, 1.0366e-03, 4.2458e-03, 2.6035e-03, 4.1313e-03,
          2.5654e-03, 2.7332e-03, 2.8133e-03, 4.7455e-03, 1.0857e-02, 1.6113e-02,
          9.1934e-03, 5.4550e-03, 5.3940e-03, 4.3297e-03, 3.5954e-03, 3.4122e-03,
          2.5139e-03, 9.0561e-03, 1.4515e-03, 1.9722e-03, 9.0885e-04, 1.0281e-03,
          1.6069e-03, 5.8136e-03, 5.8136e-03, 6.2599e-03, 4.9400e-03, 2.1152e-03,
          1.9550e-03, 5.7487e-03, 2.3060e-03, 6.1455e-03, 5.2032e-03, 7.2670e-03,
          1.1742e-02, 1.3969e-02, 4.3221e-03, 8.4457e-03, 4.2305e-03, 2.9850e-03,
          1.4000e-03, 8.9884e-04, 2.0771e-03, 6.1684e-03, 7.6942e-03, 6.6414e-03,
          7.1678e-03, 3.0422e-03, 2.6894e-03, 8.0643e-03, 5.3291e-03, 3.1319e-03,
          5.6725e-03, 1.2184e-02, 1.8677e-02, 7.4234e-03, 5.8746e-03, 9.0027e-03,
          6.3667e-03, 1.2428e-02, 2.9602e-03, 1.1467e-02, 4.7607e-03, 2.0828e-03,
          2.6932e-03, 9.2773e-03, 2.6188e-03, 7.9155e-04, 6.4888e-03, 3.9330e-03,
          4.8866e-03, 4.4479e-03, 5.0278e-03, 5.8479e-03, 2.4700e-03, 5.9166e-03,
          6.7062e-03, 5.2910e-03, 1.3924e-03, 5.3444e-03, 1.3420e-02, 7.2327e-03,
          7.5798e-03, 4.6005e-03, 3.3836e-03, 3.7861e-03, 4.0398e-03, 2.9507e-03,
          4.2953e-03, 2.0847e-03, 5.8823e-03, 5.7755e-03, 5.1727e-03, 7.1373e-03,
          1.3847e-02, 6.1302e-03, 4.8180e-03, 1.1246e-02, 5.7907e-03, 3.0655e-02,
          2.1210e-02, 3.7170e-02, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05,
          1.3053e-05, 1.3053e-05, 3.9244e-04, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.7083e-04, 6.3598e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.7083e-04, 1.3053e-05, 1.3053e-05, 3.9244e-04, 6.3598e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.2028e-04, 1.3053e-05, 3.9244e-04,
          1.2028e-04, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 6.3598e-05, 6.3598e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 1.7083e-04, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 6.3598e-05, 1.7083e-04, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 6.3598e-05,
          1.2028e-04, 1.3053e-05, 3.9244e-04, 4.9973e-04, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 1.3053e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.7083e-04, 6.3598e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05, 6.3598e-05,
          6.3598e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 3.9244e-04, 6.3598e-05, 1.3053e-05, 1.3053e-05,
          3.9244e-04, 6.3598e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.7083e-04,
          1.3053e-05, 6.3598e-05, 1.7083e-04, 1.3053e-05, 1.3053e-05, 6.3598e-05,
          6.3598e-05, 1.3053e-05, 1.7083e-04, 6.3598e-05, 1.3053e-05, 1.7083e-04,
          1.3053e-05, 1.3053e-05, 3.9244e-04, 6.3598e-05, 5.2500e-04, 6.3598e-05,
          6.3598e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.7083e-04, 1.7083e-04,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05, 1.7083e-04, 6.3598e-05,
          1.7083e-04, 1.7083e-04, 1.3053e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05,
          1.3053e-05, 6.3598e-05, 1.3053e-05, 6.3598e-05, 1.3053e-05, 1.3053e-05,
          6.3598e-05, 1.3053e-05, 1.3053e-05, 1.3053e-05, 1.7083e-04, 1.3053e-05,
          3.9244e-04, 1.7083e-04, 6.3598e-05, 6.3598e-05, 1.3053e-05, 6.3598e-05,
          1.7083e-04, 1.3053e-05, 1.7083e-04, 1.3053e-05, 1.3053e-05, 6.3598e-05,
          6.3598e-05, 3.9244e-04, 1.3053e-05, 6.3598e-05, 6.3598e-05, 1.3053e-05,
          6.3598e-05, 6.3598e-05, 6.3598e-05, 1.3053e-05, 6.3598e-05, 6.3598e-05,
          1.3053e-05, 6.3598e-05, 6.3598e-05, 4.9114e-04, 1.3053e-05, 6.3598e-05,
          1.3053e-05, 1.3053e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3825e-03, 6.0028e-02, 1.6937e-02, 1.8417e-02, 5.5664e-02,
          5.3314e-02, 1.4992e-02, 8.1253e-03, 2.9545e-03, 9.5825e-03, 2.8137e-02,
          6.5308e-03, 5.2643e-03, 1.4067e-03, 5.0087e-03, 9.5978e-03, 3.8910e-03,
          3.2864e-03, 4.8141e-03, 3.3307e-04, 4.8494e-04, 5.8126e-04, 4.7607e-03,
          3.2654e-02, 5.3406e-03, 1.1581e-02, 5.5428e-03, 6.2418e-04, 4.4632e-03,
          1.8759e-03, 1.3397e-02, 1.2178e-03, 3.5076e-03, 8.7051e-03, 2.0561e-03,
          1.9722e-03, 8.1062e-04, 3.0079e-03, 3.6182e-03, 5.1260e-04, 2.9240e-03,
          1.4496e-03, 1.0023e-03, 1.1415e-03, 8.6689e-04, 7.4959e-03, 3.5133e-03,
          8.2827e-04, 6.2656e-04, 6.3133e-03, 2.1248e-03, 5.3329e-03, 2.8172e-03,
          8.6832e-04, 2.2259e-03, 5.6076e-04, 9.8038e-04, 3.0098e-03, 1.5888e-03,
          1.2970e-03, 1.3218e-03, 1.6136e-03, 1.8702e-03, 2.4834e-03, 6.9427e-03,
          5.3291e-03, 5.1804e-03, 2.7218e-03, 6.8760e-04, 6.6280e-04, 1.0519e-03,
          8.2445e-04, 1.4870e-02, 8.5926e-04, 5.3215e-03, 9.0408e-04, 8.8787e-04,
          1.4296e-03, 1.0099e-03, 2.7523e-03, 3.7003e-03, 3.1624e-03, 1.0138e-03,
          7.3099e-04, 8.7738e-03, 1.1940e-03, 2.3422e-02, 3.1033e-03, 2.4147e-03,
          3.2501e-02, 6.7787e-03, 1.1463e-03, 1.2939e-02, 8.3637e-04, 7.0095e-04,
          7.9393e-04, 4.5300e-04, 4.6234e-03, 1.9245e-03, 9.3317e-04, 1.6775e-03,
          2.7176e-02, 4.0054e-03, 4.3335e-03, 1.4296e-03, 4.5471e-03, 1.3237e-03,
          6.9084e-03, 1.4925e-03, 1.0078e-02, 1.7939e-03, 4.5547e-03, 3.6125e-03,
          5.1003e-03, 1.3371e-03, 9.4652e-04, 7.0915e-03, 9.3889e-04, 1.2960e-03,
          2.2545e-03, 3.3417e-03, 8.1730e-04, 7.4005e-03, 8.4915e-03, 9.3746e-04,
          6.1512e-04, 1.0757e-03, 1.6909e-03, 4.7455e-03, 9.7504e-03, 8.6288e-03,
          6.4850e-03, 5.6190e-03, 1.1284e-02, 3.5019e-03, 8.5754e-03, 1.1082e-03,
          9.0637e-03, 2.2297e-03, 1.1726e-02, 1.2840e-02, 7.6599e-03, 3.2463e-03,
          1.2236e-03, 8.9111e-03, 1.4944e-03, 1.4324e-03, 2.6970e-03, 4.4708e-03,
          6.6872e-03, 3.4370e-03, 1.1177e-03, 2.8801e-03, 2.7390e-03, 2.3842e-03,
          7.1411e-03, 2.0218e-03, 6.0242e-02, 2.3484e-05, 1.0091e-04, 1.0091e-04,
          2.3484e-05, 2.3484e-05, 1.8787e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 6.1452e-05, 7.7677e-04, 1.0091e-04, 6.1452e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 6.1452e-05, 6.1452e-05, 1.0091e-04, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 6.1452e-05, 6.1452e-05,
          6.1452e-05, 2.3484e-05, 2.3484e-05, 1.8787e-04, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 1.8787e-04, 1.0091e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 6.1452e-05, 2.3484e-05, 1.8787e-04,
          6.1452e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 6.1452e-05, 2.3484e-05,
          2.3484e-05, 1.8787e-04, 2.3484e-05, 6.1452e-05, 1.0091e-04, 1.8787e-04,
          1.0091e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.0091e-04, 6.1452e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.0091e-04,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 6.1452e-05, 6.1452e-05, 1.9097e-04,
          2.3484e-05, 2.3484e-05, 6.2943e-05, 1.4997e-04, 1.8787e-04, 1.8787e-04,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 1.0091e-04, 2.3484e-05, 6.2943e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.0091e-04, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          6.2943e-05, 2.3484e-05, 6.1452e-05, 2.3031e-04, 1.0091e-04, 2.3484e-05,
          6.1452e-05, 6.1452e-05, 1.0091e-04, 2.3484e-05, 2.7800e-04, 2.3484e-05,
          2.3484e-05, 6.1452e-05, 6.2943e-05, 1.0091e-04, 2.3484e-05, 6.1452e-05,
          6.1452e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 3.1590e-04, 1.4997e-04,
          1.9097e-04, 6.2943e-05, 1.8787e-04, 1.0091e-04, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 1.8787e-04, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 6.1452e-05, 6.1452e-05, 4.4131e-04, 2.3484e-05, 2.3484e-05,
          1.4997e-04, 4.9019e-04, 2.7800e-04, 1.0091e-04, 2.3484e-05, 2.3484e-05,
          1.8787e-04, 2.7800e-04, 2.7800e-04, 2.3031e-04, 6.2943e-05, 6.2943e-05,
          6.2943e-05, 2.3484e-05, 2.7800e-04, 1.8787e-04, 1.5163e-04, 2.7800e-04,
          1.0091e-04, 2.2900e-04, 1.4997e-04, 2.3484e-05, 2.3484e-05, 5.2834e-04,
          2.3484e-05, 6.9904e-04, 1.8787e-04, 2.3031e-04, 2.3484e-05, 1.9109e-04,
          3.1590e-04, 6.1452e-05, 6.1452e-05, 2.7800e-04, 2.3484e-05, 1.8787e-04,
          2.3484e-05, 6.1452e-05, 3.1590e-04, 6.1452e-05, 7.6628e-04, 2.7800e-04,
          2.7800e-04, 1.0091e-04, 2.3484e-05, 2.3484e-05, 1.9097e-04, 1.9097e-04,
          1.4429e-03, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.8787e-04, 6.2943e-05,
          3.5858e-04, 2.3031e-04, 1.5163e-04, 2.3484e-05, 1.0091e-04, 1.0091e-04,
          1.0091e-04, 1.8787e-04, 6.1452e-05, 1.0091e-04, 2.3484e-05, 2.3484e-05,
          1.0091e-04, 2.3484e-05, 1.4997e-04, 6.1452e-05, 3.1900e-04, 2.3484e-05,
          1.1616e-03, 6.5756e-04, 2.3484e-05, 6.2943e-05, 2.7800e-04, 4.9019e-04,
          6.1452e-05, 2.7800e-04, 5.2834e-04, 6.1452e-05, 2.3484e-05, 2.7800e-04,
          6.1846e-04, 1.0319e-03, 2.3484e-05, 2.7800e-04, 2.7800e-04, 2.3484e-05,
          1.0091e-04, 6.2943e-05, 2.3484e-05, 1.5604e-04, 2.8729e-04, 1.0091e-04,
          2.3484e-05, 1.0091e-04, 2.8729e-04, 5.8985e-04, 2.8729e-04, 2.8729e-04,
          6.2943e-05, 6.2943e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6343e-03, 5.8502e-02, 1.9577e-02, 1.6449e-02, 5.5206e-02,
          4.1473e-02, 1.6037e-02, 9.5654e-04, 8.2159e-04, 2.5120e-03, 6.2294e-03,
          1.8950e-03, 9.9277e-04, 2.5978e-03, 1.3857e-03, 1.0786e-03, 1.3885e-03,
          1.1530e-03, 9.0647e-04, 7.6675e-04, 8.0633e-04, 1.1158e-03, 1.4963e-03,
          8.6060e-03, 3.3264e-03, 1.2123e-02, 2.9030e-03, 1.0109e-03, 1.2121e-03,
          3.1471e-03, 1.4381e-03, 6.2084e-04, 1.3990e-03, 1.1787e-03, 8.5735e-04,
          1.7891e-03, 1.7242e-03, 1.9360e-03, 1.1663e-03, 1.2369e-03, 6.7558e-03,
          9.9487e-03, 5.3596e-03, 8.0261e-03, 2.2907e-03, 1.4286e-03, 2.5444e-03,
          1.4811e-03, 1.4944e-03, 6.7520e-03, 1.2369e-03, 1.2016e-03, 9.2363e-04,
          1.0700e-03, 9.3365e-04, 1.1339e-03, 1.8511e-03, 3.2063e-03, 3.0823e-03,
          2.7695e-03, 2.2106e-03, 2.8725e-03, 2.3804e-03, 1.6144e-02, 6.9389e-03,
          2.6047e-02, 6.3629e-03, 2.6169e-03, 4.5815e-03, 5.0964e-03, 5.1003e-03,
          2.9812e-03, 4.1046e-03, 1.3533e-03, 1.5726e-03, 1.1683e-03, 7.7677e-04,
          1.4067e-03, 2.4223e-03, 9.9945e-03, 2.6894e-03, 9.0179e-03, 1.9484e-03,
          1.8473e-03, 4.6120e-03, 2.7084e-03, 4.4746e-03, 1.1261e-02, 2.4475e-02,
          9.7122e-03, 6.4926e-03, 1.1627e-02, 6.2294e-03, 3.9062e-03, 3.0212e-03,
          1.3371e-03, 9.5272e-04, 1.1969e-03, 2.1152e-03, 3.4103e-03, 2.6131e-03,
          4.8065e-03, 3.6602e-03, 3.1681e-03, 2.7466e-03, 2.3346e-03, 2.3899e-03,
          4.4594e-03, 1.6068e-02, 9.7656e-03, 1.8158e-02, 1.7685e-02, 9.8114e-03,
          3.7632e-03, 8.3313e-03, 2.6455e-03, 5.2948e-03, 4.9820e-03, 3.3169e-03,
          3.2120e-03, 2.6588e-03, 3.1242e-03, 8.8167e-04, 2.9335e-03, 4.6844e-03,
          2.2907e-03, 8.0795e-03, 9.1095e-03, 2.5578e-03, 1.6918e-03, 4.9553e-03,
          1.6159e-02, 9.8648e-03, 1.1520e-03, 7.1945e-03, 1.0750e-02, 1.8158e-02,
          3.4332e-03, 5.6572e-03, 2.3823e-03, 3.5095e-03, 2.4948e-03, 3.3741e-03,
          5.0201e-03, 1.1444e-03, 3.1452e-03, 2.7695e-03, 1.4252e-02, 3.2196e-03,
          5.2185e-03, 2.7580e-03, 3.1757e-03, 5.0125e-03, 5.6305e-03, 2.1072e-02,
          9.5596e-03, 1.1971e-02, 2.4643e-03, 6.5857e-02, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 6.3837e-05, 6.3837e-05, 1.1057e-04, 1.5497e-05,
          1.1057e-04, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          6.3837e-05, 1.5497e-05, 1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05, 1.5497e-05, 1.5891e-04,
          6.3837e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 6.3837e-05, 1.5497e-05, 1.1057e-04, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5891e-04, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05, 6.3837e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.1057e-04, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.1057e-04, 1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          6.3837e-05, 1.5497e-05, 1.5891e-04, 1.5891e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.1057e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.1057e-04, 1.5891e-04, 1.1057e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 1.1057e-04, 1.5497e-05, 1.5497e-05,
          1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05, 1.1057e-04, 1.1057e-04,
          1.1057e-04, 1.1057e-04, 1.1057e-04, 1.5891e-04, 1.5497e-05, 1.5497e-05,
          1.5891e-04, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5736e-05, 6.3837e-05, 1.5497e-05, 1.5497e-05,
          1.1057e-04, 1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05, 6.3837e-05,
          1.5497e-05, 1.5497e-05, 1.5891e-04, 6.3837e-05, 1.5891e-04, 1.1057e-04,
          1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05, 1.0484e-04, 6.3837e-05,
          6.3837e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5891e-04, 1.1057e-04,
          1.9991e-04, 1.9991e-04, 1.5497e-05, 1.5497e-05, 1.5891e-04, 1.5497e-05,
          6.3837e-05, 1.5497e-05, 6.3837e-05, 1.1057e-04, 1.5497e-05, 1.5497e-05,
          6.3837e-05, 1.5497e-05, 1.5497e-05, 6.3837e-05, 1.0484e-04, 1.5497e-05,
          1.5891e-04, 1.5891e-04, 1.1057e-04, 1.1057e-04, 1.5497e-05, 1.5497e-05,
          1.5891e-04, 1.5497e-05, 1.5891e-04, 6.3837e-05, 1.5497e-05, 5.3787e-04,
          1.5497e-05, 1.5891e-04, 1.5497e-05, 1.5497e-05, 1.5497e-05, 1.5497e-05,
          1.1057e-04, 1.1057e-04, 1.1057e-04, 4.7994e-04, 1.1057e-04, 1.1057e-04,
          1.5497e-05, 5.5695e-04, 5.5695e-04, 4.9639e-04, 1.1057e-04, 1.1057e-04,
          1.5497e-05, 1.1057e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.1613e-03, 5.6702e-02, 1.7258e-02, 1.6846e-02, 4.9988e-02,
          4.3793e-02, 1.2390e-02, 1.3199e-03, 1.0872e-03, 4.0512e-03, 4.9438e-03,
          2.0771e-03, 2.5959e-03, 1.1702e-03, 5.9929e-03, 1.5717e-03, 4.1809e-03,
          2.2182e-03, 1.0500e-03, 8.7595e-04, 1.4420e-03, 1.3704e-03, 2.9202e-03,
          6.7329e-03, 7.4730e-03, 1.0223e-02, 5.5275e-03, 1.0538e-03, 3.8376e-03,
          1.5631e-03, 2.6283e-03, 1.3666e-03, 9.3994e-03, 3.4008e-03, 2.3441e-03,
          9.9335e-03, 1.6994e-03, 7.6485e-03, 5.4321e-03, 1.0433e-03, 1.1497e-02,
          3.9177e-03, 3.3016e-03, 2.6531e-03, 2.7218e-03, 4.1275e-03, 1.7044e-02,
          1.1940e-03, 1.3075e-03, 5.5008e-03, 2.9888e-03, 1.9646e-03, 1.7529e-03,
          7.1144e-04, 1.2093e-03, 9.8610e-04, 3.4218e-03, 2.1229e-03, 2.9049e-03,
          2.2144e-03, 3.2310e-03, 3.1738e-03, 6.0081e-03, 5.9929e-03, 3.3264e-02,
          3.0174e-03, 2.1992e-03, 4.2114e-03, 2.4834e-03, 1.9932e-03, 1.8501e-03,
          1.9093e-03, 7.8964e-03, 1.0939e-03, 2.2831e-03, 6.2656e-04, 6.6090e-04,
          1.2970e-03, 3.0422e-03, 2.1877e-03, 5.6496e-03, 1.9283e-03, 2.0180e-03,
          1.8063e-03, 7.5150e-03, 2.2850e-03, 3.9062e-03, 1.7195e-03, 2.0618e-03,
          7.7820e-03, 8.5678e-03, 2.4071e-03, 5.3177e-03, 3.3455e-03, 2.5940e-03,
          1.3514e-03, 7.5531e-04, 2.7695e-03, 6.6261e-03, 2.7657e-03, 3.1834e-03,
          4.9515e-03, 1.2026e-03, 1.4324e-03, 6.4507e-03, 1.1536e-02, 4.0321e-03,
          7.0839e-03, 6.5346e-03, 1.1581e-02, 3.0479e-03, 2.3842e-03, 3.2444e-03,
          7.1449e-03, 1.2444e-02, 3.7003e-03, 2.8915e-02, 3.8166e-03, 2.0123e-03,
          2.9812e-03, 8.1635e-03, 1.1892e-03, 9.2840e-04, 1.1955e-02, 9.9468e-04,
          2.3785e-03, 1.5707e-03, 2.0847e-03, 1.7426e-02, 4.0207e-03, 7.3509e-03,
          4.0054e-03, 2.3003e-03, 1.4772e-03, 1.7567e-03, 7.2403e-03, 3.8509e-03,
          1.5343e-02, 5.2834e-03, 3.3035e-03, 4.6806e-03, 5.4321e-03, 3.3894e-03,
          1.7471e-03, 1.4200e-03, 2.9411e-03, 2.5425e-03, 1.7567e-03, 1.7532e-02,
          9.9716e-03, 1.3176e-02, 4.8866e-03, 1.1971e-02, 5.5313e-03, 2.3422e-02,
          9.2545e-03, 1.0132e-02, 1.7271e-03, 4.1313e-03, 5.4138e-02, 1.3447e-04,
          1.7405e-05, 1.7405e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          5.4550e-04, 8.6308e-05, 1.7405e-05, 1.7405e-05, 1.3447e-04, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05, 1.7405e-05, 1.7405e-05,
          1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05, 1.7405e-05,
          1.3447e-04, 1.7405e-05, 1.7405e-05, 1.3447e-04, 1.3447e-04, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 8.6308e-05, 1.3447e-04, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05, 1.3447e-04, 1.3447e-04,
          6.5506e-05, 1.7405e-05, 1.7405e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05, 1.7405e-05,
          1.3447e-04, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05, 6.5506e-05,
          1.7405e-05, 8.6308e-05, 1.7405e-05, 6.5506e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 8.7070e-04,
          1.7405e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7643e-05,
          1.7405e-05, 1.7405e-05, 6.5506e-05, 6.5506e-05, 1.3447e-04, 1.7405e-05,
          1.3447e-04, 6.5804e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.3447e-04, 1.7405e-05, 1.3471e-04,
          1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          3.2043e-04, 1.7405e-05, 1.3447e-04, 3.7193e-04, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 6.5506e-05,
          1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          1.7405e-05, 7.8297e-04, 8.6308e-05, 8.6308e-05, 1.7405e-05, 1.7405e-05,
          8.6308e-05, 6.5506e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05, 1.3447e-04,
          1.7405e-05, 1.7405e-05, 1.3471e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          6.5506e-05, 1.7405e-05, 6.5506e-05, 1.7405e-05, 8.6308e-05, 1.3447e-04,
          1.7405e-05, 1.7405e-05, 1.3447e-04, 1.3447e-04, 4.3607e-04, 8.6308e-05,
          9.1887e-04, 9.1887e-04, 1.7405e-05, 1.7405e-05, 1.3447e-04, 6.5506e-05,
          4.8423e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.3471e-04, 1.7405e-05,
          1.3447e-04, 1.3447e-04, 4.3607e-04, 6.5506e-05, 6.5506e-05, 1.7405e-05,
          6.5506e-05, 1.3447e-04, 6.5506e-05, 1.3447e-04, 1.7405e-05, 1.7405e-05,
          9.1887e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 3.2043e-04, 1.7405e-05,
          1.3447e-04, 1.3447e-04, 7.8297e-04, 1.7405e-05, 1.7405e-05, 8.6308e-05,
          1.3447e-04, 1.7405e-05, 9.1887e-04, 6.5506e-05, 1.7405e-05, 1.7405e-05,
          8.7070e-04, 8.7070e-04, 1.7405e-05, 1.7405e-05, 1.7405e-05, 1.7405e-05,
          8.6308e-05, 1.7405e-05, 3.2377e-04, 1.2267e-04, 8.6308e-05, 1.7405e-05,
          1.7405e-05, 1.3447e-04, 8.6308e-05, 4.5085e-04, 1.7405e-05, 3.2377e-04,
          8.6308e-05, 1.7405e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.3847e-03, 5.8655e-02, 2.5177e-02, 2.4872e-02, 5.1849e-02,
          4.2084e-02, 1.6830e-02, 2.0618e-03, 1.5001e-03, 2.9736e-03, 4.2534e-03,
          1.7376e-03, 7.8583e-03, 7.4959e-03, 2.7084e-03, 2.1553e-03, 2.4292e-02,
          1.3390e-02, 1.1616e-03, 2.5921e-03, 2.4185e-03, 4.3030e-03, 7.2021e-03,
          5.1651e-03, 2.7962e-03, 1.3657e-02, 8.8577e-03, 3.8757e-03, 1.7014e-02,
          1.1276e-02, 3.6793e-03, 7.1526e-03, 5.0507e-03, 2.2297e-03, 2.3613e-03,
          2.3651e-03, 1.0567e-03, 4.3030e-03, 2.9106e-03, 3.7804e-03, 5.0774e-03,
          2.2221e-03, 1.6594e-03, 1.7300e-03, 1.6308e-03, 1.5579e-02, 7.5417e-03,
          5.8136e-03, 5.0659e-03, 7.9117e-03, 3.0441e-03, 2.8820e-03, 1.6899e-03,
          1.0576e-03, 1.6804e-03, 1.3037e-03, 1.2283e-03, 1.6851e-03, 1.5430e-03,
          1.5497e-03, 1.4328e-02, 2.2659e-03, 3.3932e-03, 2.4853e-03, 8.4305e-03,
          2.7657e-03, 1.5020e-03, 2.2697e-03, 1.3762e-03, 1.3170e-03, 1.5697e-03,
          1.9875e-03, 6.3705e-03, 1.4658e-03, 3.8509e-03, 9.6846e-04, 8.7118e-04,
          1.7757e-03, 1.3485e-03, 1.5888e-03, 2.0256e-03, 1.4734e-03, 1.1187e-03,
          1.2426e-03, 5.9967e-03, 1.9970e-03, 2.6531e-03, 1.4467e-03, 1.7109e-03,
          4.8943e-03, 4.6654e-03, 1.4887e-03, 3.3627e-03, 1.7834e-03, 2.3060e-03,
          6.1264e-03, 9.3317e-04, 3.5019e-03, 5.0545e-03, 1.2903e-03, 1.5545e-03,
          3.3970e-03, 1.1921e-03, 1.0567e-03, 1.8387e-03, 3.1834e-03, 1.7548e-02,
          4.7798e-03, 2.6836e-03, 5.5962e-03, 1.7328e-03, 2.0161e-03, 2.0294e-03,
          2.0542e-03, 3.8128e-03, 1.1253e-02, 7.6294e-03, 3.1128e-03, 9.3918e-03,
          2.3560e-02, 2.8934e-03, 8.6737e-04, 1.2417e-03, 3.6087e-03, 8.2207e-04,
          9.7370e-04, 1.0872e-03, 1.4200e-03, 3.7479e-03, 4.5815e-03, 5.3291e-03,
          3.0880e-03, 1.8072e-03, 1.0147e-03, 1.1234e-03, 3.1681e-03, 1.7872e-03,
          2.9480e-02, 2.6840e-02, 1.0094e-02, 6.7444e-03, 5.4932e-03, 2.6596e-02,
          1.2608e-03, 8.0156e-04, 1.2064e-03, 1.3189e-03, 1.8301e-03, 2.8858e-03,
          4.1962e-03, 2.6894e-03, 1.5965e-03, 6.9656e-03, 4.0855e-03, 6.1913e-03,
          3.9673e-03, 3.5534e-03, 1.4362e-03, 2.3727e-03, 9.4833e-03, 6.9519e-02,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.2022e-04, 1.7290e-03, 5.2929e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          8.3029e-05, 1.5736e-05, 1.2022e-04, 1.2022e-04, 1.5736e-05, 1.5736e-05,
          1.2022e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 1.2022e-04,
          5.2929e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 8.3029e-05, 1.2022e-04, 8.3029e-05,
          1.2022e-04, 8.3029e-05, 1.5736e-05, 5.2929e-05, 1.2022e-04, 1.5736e-05,
          5.2929e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 1.5736e-05, 8.3029e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 4.6349e-04, 1.2022e-04, 1.2022e-04,
          1.2022e-04, 1.2022e-04, 1.5736e-05, 5.2929e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 8.3029e-05, 5.2929e-05, 5.2929e-05, 1.5736e-05, 1.5724e-04,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.2022e-04, 1.5736e-05, 5.2929e-05, 1.2022e-04, 1.2022e-04,
          1.2022e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.2022e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 5.2929e-05, 1.5736e-05,
          5.6791e-04, 1.2022e-04, 1.2022e-04, 1.5736e-05, 1.5736e-05, 5.2929e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 2.2495e-04,
          1.2022e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5724e-04, 1.5736e-05, 1.5736e-05, 8.3029e-05, 1.5736e-05, 1.1998e-04,
          1.5736e-05, 8.3029e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.2022e-04, 1.5736e-05, 8.3029e-05, 5.2929e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 5.2929e-05, 5.2929e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 1.5736e-05, 1.2022e-04,
          1.2022e-04, 3.4881e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.5736e-05, 1.5736e-05, 1.5736e-05, 5.2929e-05, 1.5736e-05, 1.5736e-05,
          5.2929e-05, 5.2929e-05, 5.2929e-05, 1.5736e-05, 8.3029e-05, 5.2929e-05,
          1.5736e-05, 5.2929e-05, 1.5736e-05, 5.2929e-05, 1.0800e-04, 1.5736e-05,
          1.5736e-05, 8.3029e-05, 1.5736e-05, 1.5736e-05, 1.2022e-04, 5.2929e-05,
          5.2929e-05, 1.5736e-05, 5.2929e-05, 1.2022e-04, 1.5736e-05, 8.3029e-05,
          3.3355e-04, 1.2022e-04, 1.5736e-05, 5.2929e-05, 1.2022e-04, 1.5736e-05,
          1.2022e-04, 1.5736e-05, 5.2929e-05, 2.7966e-04, 1.5736e-05, 1.5736e-05,
          8.3029e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5724e-04, 1.5736e-05,
          1.2022e-04, 5.2929e-05, 3.4881e-04, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          1.2022e-04, 1.5736e-05, 5.2929e-05, 5.2929e-05, 1.2022e-04, 8.3029e-05,
          1.5736e-05, 2.2163e-03, 1.5736e-05, 1.5736e-05, 1.5736e-05, 1.5736e-05,
          3.1161e-04, 6.5470e-04, 3.1161e-04, 8.3029e-05, 8.3029e-05, 1.5736e-05,
          1.5736e-05, 1.2022e-04, 3.1161e-04, 4.2772e-04, 8.3029e-05, 8.3029e-05,
          8.3029e-05, 8.3029e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.3738e-03, 6.1646e-02, 1.8936e-02, 2.6489e-02, 4.6417e-02,
          4.5593e-02, 1.5778e-02, 1.5306e-03, 1.6108e-03, 5.1994e-03, 5.7106e-03,
          2.5845e-03, 3.8128e-03, 1.1168e-03, 8.6060e-03, 1.8282e-03, 6.2103e-03,
          3.2005e-03, 1.3199e-03, 7.0667e-04, 1.1444e-03, 1.0061e-03, 4.3488e-03,
          8.5678e-03, 6.5880e-03, 1.2856e-02, 1.0605e-02, 9.4700e-04, 6.7673e-03,
          1.6432e-03, 3.1929e-03, 1.7729e-03, 1.2756e-02, 4.9019e-03, 3.4866e-03,
          7.7171e-03, 1.2236e-03, 8.6823e-03, 7.0915e-03, 8.5640e-04, 8.3847e-03,
          2.2964e-03, 1.9302e-03, 1.6470e-03, 1.9779e-03, 7.4577e-03, 2.0477e-02,
          1.0309e-03, 1.1244e-03, 5.3101e-03, 3.0994e-03, 1.9341e-03, 2.7924e-03,
          5.6314e-04, 1.0967e-03, 7.9727e-04, 2.0390e-03, 1.4763e-03, 1.8387e-03,
          1.5917e-03, 2.6627e-03, 2.0752e-03, 4.0779e-03, 3.3588e-03, 2.8748e-02,
          1.7576e-03, 2.0142e-03, 3.5934e-03, 1.3313e-03, 1.2388e-03, 1.2302e-03,
          1.3885e-03, 1.4259e-02, 9.4032e-04, 2.2659e-03, 5.1498e-04, 5.6601e-04,
          1.0920e-03, 1.6670e-03, 1.3075e-03, 4.9477e-03, 1.1292e-03, 1.2064e-03,
          1.1253e-03, 5.5771e-03, 1.4572e-03, 4.9095e-03, 1.2007e-03, 1.1864e-03,
          9.4757e-03, 9.1934e-03, 1.2417e-03, 6.4659e-03, 1.6251e-03, 1.7424e-03,
          1.2436e-03, 5.9795e-04, 4.5700e-03, 6.3515e-03, 1.3256e-03, 2.1954e-03,
          6.5460e-03, 1.2150e-03, 1.1654e-03, 3.4180e-03, 1.2459e-02, 3.1128e-03,
          4.7073e-03, 2.6569e-03, 1.2154e-02, 1.3466e-03, 1.5974e-03, 2.6913e-03,
          5.4626e-03, 5.2872e-03, 2.4147e-03, 3.4790e-02, 2.2697e-03, 1.7052e-03,
          2.8019e-03, 7.7438e-03, 7.6866e-04, 9.7466e-04, 1.5472e-02, 6.6805e-04,
          1.2369e-03, 8.4496e-04, 1.2426e-03, 1.6190e-02, 5.4283e-03, 4.9667e-03,
          2.5692e-03, 1.7271e-03, 1.9836e-03, 1.3514e-03, 6.6299e-03, 1.4057e-03,
          2.9114e-02, 3.8567e-03, 5.1155e-03, 7.8049e-03, 1.0612e-02, 3.7212e-03,
          1.1263e-03, 1.8244e-03, 1.9703e-03, 1.7185e-03, 1.2074e-03, 1.0910e-02,
          9.3536e-03, 1.0162e-02, 2.4338e-03, 7.3280e-03, 3.2082e-03, 9.7733e-03,
          8.1863e-03, 5.6839e-03, 2.4548e-03, 1.5526e-03, 2.2491e-02, 8.9111e-03,
          4.7760e-02, 1.3590e-05, 1.3041e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3041e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3041e-04, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.4365e-05, 1.3590e-05,
          9.8228e-05, 1.3590e-05, 1.3590e-05, 9.5558e-04, 1.3041e-04, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05, 1.3041e-04,
          9.8228e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3041e-04,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 2.1517e-04, 1.3590e-05,
          9.8228e-05, 9.8228e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 2.1517e-04, 1.3590e-05, 9.9003e-05,
          9.8228e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          9.8228e-05, 1.3590e-05, 1.3041e-04, 2.1517e-04, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3041e-04, 1.3041e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3041e-04, 1.3590e-05, 1.3041e-04, 1.3590e-05, 1.3590e-05, 2.1517e-04,
          1.3590e-05, 1.3590e-05, 2.1517e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05, 1.3590e-05, 2.1517e-04,
          1.3590e-05, 1.3590e-05, 1.3041e-04, 1.3590e-05, 4.4441e-04, 1.3590e-05,
          1.1425e-03, 2.1517e-04, 1.3590e-05, 1.3590e-05, 2.1517e-04, 9.8228e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 2.1517e-04, 1.3590e-05,
          2.1517e-04, 2.1517e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 2.1517e-04, 1.3590e-05, 1.3041e-04, 1.3590e-05, 1.3590e-05,
          1.3041e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.8228e-05, 1.3590e-05,
          1.3041e-04, 2.1517e-04, 1.3041e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          2.1517e-04, 1.3590e-05, 2.1517e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.2598e-03, 1.2598e-03, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3041e-04, 1.3590e-05, 1.3041e-04, 1.3590e-05, 1.3041e-04, 1.3590e-05,
          1.3590e-05, 1.3041e-04, 1.3041e-04, 4.5609e-04, 1.3590e-05, 1.3132e-03,
          1.3041e-04, 1.3041e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3902e-03, 6.2469e-02, 1.7502e-02, 2.1423e-02, 4.0894e-02,
          4.5715e-02, 1.4191e-02, 1.8101e-03, 1.3151e-03, 7.9422e-03, 8.3923e-03,
          3.1528e-03, 2.6703e-03, 1.0786e-03, 4.4556e-03, 2.0828e-03, 4.3068e-03,
          2.3594e-03, 1.1902e-03, 7.3528e-04, 1.2722e-03, 1.6880e-03, 3.5591e-03,
          1.0162e-02, 5.2719e-03, 1.2016e-02, 1.0101e-02, 9.6321e-04, 4.1428e-03,
          1.5545e-03, 3.3989e-03, 1.3208e-03, 8.0185e-03, 6.6452e-03, 2.9507e-03,
          6.0654e-03, 1.3514e-03, 1.2863e-02, 6.8359e-03, 9.3699e-04, 1.1742e-02,
          2.6340e-03, 2.6112e-03, 2.1896e-03, 2.7657e-03, 4.8332e-03, 1.4473e-02,
          1.2999e-03, 1.5726e-03, 5.9242e-03, 3.3646e-03, 2.2449e-03, 2.5921e-03,
          7.3099e-04, 1.2493e-03, 9.2077e-04, 2.3842e-03, 1.8482e-03, 2.1610e-03,
          1.6680e-03, 2.4776e-03, 2.8095e-03, 5.2757e-03, 3.9062e-03, 1.8234e-02,
          2.8305e-03, 3.2158e-03, 6.3438e-03, 1.8797e-03, 1.4820e-03, 2.2030e-03,
          2.0847e-03, 1.9958e-02, 1.0929e-03, 2.6112e-03, 7.2193e-04, 7.8630e-04,
          1.3256e-03, 2.4376e-03, 1.9798e-03, 6.3286e-03, 1.9503e-03, 1.3752e-03,
          1.1988e-03, 7.0648e-03, 1.7204e-03, 5.9586e-03, 1.8158e-03, 1.7586e-03,
          1.2482e-02, 1.4618e-02, 1.6556e-03, 9.0408e-03, 2.2717e-03, 2.5597e-03,
          1.2741e-03, 6.7568e-04, 4.1122e-03, 6.2790e-03, 2.0027e-03, 3.4981e-03,
          8.6136e-03, 1.5955e-03, 1.6031e-03, 4.3030e-03, 7.1068e-03, 2.4891e-03,
          6.1035e-03, 3.1490e-03, 1.5549e-02, 1.6928e-03, 2.1629e-03, 4.5700e-03,
          4.8065e-03, 5.4703e-03, 2.7752e-03, 1.6083e-02, 3.8891e-03, 1.6870e-03,
          2.2945e-03, 1.0788e-02, 8.8310e-04, 1.1358e-03, 1.0948e-02, 1.0433e-03,
          1.7443e-03, 1.3371e-03, 1.9341e-03, 7.7400e-03, 4.1504e-03, 6.6757e-03,
          3.5706e-03, 2.4071e-03, 1.5745e-03, 1.9512e-03, 8.5297e-03, 1.9379e-03,
          1.5305e-02, 4.8981e-03, 4.1084e-03, 6.9809e-03, 9.0942e-03, 2.8076e-03,
          1.3733e-03, 1.8206e-03, 2.5673e-03, 2.9449e-03, 1.6346e-03, 6.7863e-03,
          1.4366e-02, 6.7596e-03, 2.5120e-03, 8.8425e-03, 4.6997e-03, 1.1909e-02,
          1.3351e-02, 1.0605e-02, 2.1801e-03, 2.3689e-03, 1.6861e-02, 7.7667e-03,
          1.7609e-02, 4.5776e-02, 1.3959e-04, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          7.0095e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05, 7.0095e-05,
          1.3959e-04, 1.5140e-05, 1.5140e-05, 6.8474e-04, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05, 1.5140e-05, 1.3959e-04,
          7.0095e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.3959e-04, 1.5140e-05, 8.4639e-05, 1.5140e-05, 1.3959e-04,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05, 1.5140e-05,
          1.5140e-05, 7.0095e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05,
          1.5140e-05, 1.5140e-05, 7.0095e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.3959e-04,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 8.4639e-05,
          1.5140e-05, 7.0095e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 7.0095e-05, 1.3959e-04, 1.3959e-04, 1.5140e-05,
          7.0095e-05, 7.0095e-05, 7.0095e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05, 7.0095e-05,
          7.0095e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 8.4639e-05, 1.5140e-05,
          7.0095e-05, 1.5140e-05, 1.3959e-04, 1.3959e-04, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 1.5140e-05, 1.3959e-04, 8.4639e-05, 1.5140e-05, 1.5140e-05,
          1.5140e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 7.0095e-05,
          1.3959e-04, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          7.0095e-05, 8.4639e-05, 8.4639e-05, 1.3959e-04, 1.5140e-05, 1.5140e-05,
          1.3959e-04, 7.0095e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04,
          1.5140e-05, 1.3959e-04, 6.8474e-04, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          1.3959e-04, 1.5140e-05, 7.0095e-05, 8.4639e-05, 1.5140e-05, 1.3959e-04,
          1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05, 2.7108e-04, 1.5140e-05,
          8.4639e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.3959e-04,
          2.7108e-04, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 8.4639e-05,
          1.3959e-04, 1.3959e-04, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05,
          7.0095e-05, 1.3959e-04, 7.0095e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05,
          8.4639e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.3959e-04, 1.5140e-05,
          1.3959e-04, 1.3959e-04, 1.3959e-04, 8.4639e-05, 1.5140e-05, 8.4639e-05,
          1.3959e-04, 1.5140e-05, 1.3959e-04, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          8.4639e-05, 8.4639e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05, 1.5140e-05,
          8.4639e-05, 8.4639e-05, 8.4639e-05, 1.1611e-04, 8.4639e-05, 8.4639e-05,
          1.5140e-05, 1.0929e-03, 8.4639e-05, 6.4945e-04, 4.7874e-04, 1.5020e-03,
          8.4639e-05, 8.4639e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0992e-03, 5.9479e-02, 2.5940e-02, 2.7100e-02, 4.7638e-02,
          4.7119e-02, 1.5167e-02, 3.1891e-03, 2.1648e-03, 4.7569e-03, 6.6414e-03,
          2.5349e-03, 1.2405e-02, 4.8790e-03, 3.3932e-03, 3.5172e-03, 3.0426e-02,
          1.7883e-02, 1.7815e-03, 1.4591e-03, 1.3380e-03, 2.7962e-03, 1.0307e-02,
          7.8011e-03, 2.4223e-03, 1.2459e-02, 1.2169e-02, 2.4166e-03, 3.4485e-02,
          6.9923e-03, 6.3782e-03, 1.3535e-02, 6.3477e-03, 4.7188e-03, 2.6703e-03,
          1.8969e-03, 6.6757e-04, 4.2114e-03, 4.0779e-03, 1.9588e-03, 2.3289e-03,
          9.2745e-04, 7.0095e-04, 8.5831e-04, 7.2336e-04, 3.5431e-02, 7.0724e-03,
          3.1872e-03, 2.1667e-03, 6.2218e-03, 2.9812e-03, 4.2839e-03, 2.4834e-03,
          8.7309e-04, 1.9331e-03, 7.6389e-04, 7.9870e-04, 1.6708e-03, 1.0262e-03,
          9.7847e-04, 6.6147e-03, 1.3437e-03, 2.2373e-03, 1.0910e-03, 6.8550e-03,
          1.6556e-03, 1.4334e-03, 2.0676e-03, 5.2214e-04, 5.0831e-04, 7.4100e-04,
          9.2697e-04, 1.2535e-02, 1.1683e-03, 5.0964e-03, 8.9312e-04, 8.2493e-04,
          1.6356e-03, 8.3828e-04, 1.0853e-03, 2.4605e-03, 1.3514e-03, 7.0143e-04,
          6.3467e-04, 6.2485e-03, 1.0405e-03, 4.6921e-03, 1.0529e-03, 9.0599e-04,
          8.2932e-03, 4.5471e-03, 5.8174e-04, 4.2953e-03, 7.0572e-04, 8.6641e-04,
          3.3169e-03, 6.3658e-04, 4.7264e-03, 4.8256e-03, 7.4768e-04, 1.0519e-03,
          6.3248e-03, 1.0128e-03, 1.4639e-03, 1.3676e-03, 4.0321e-03, 4.2992e-03,
          4.5433e-03, 1.0262e-03, 4.3411e-03, 6.7234e-04, 1.2703e-03, 1.6508e-03,
          1.5841e-03, 1.3876e-03, 2.5463e-03, 7.7095e-03, 1.1587e-03, 3.0975e-03,
          6.1989e-03, 3.3569e-03, 5.3215e-04, 2.4185e-03, 5.9738e-03, 5.0259e-04,
          5.7220e-04, 7.4768e-04, 1.0204e-03, 4.3068e-03, 7.8125e-03, 5.5428e-03,
          2.3804e-03, 1.5383e-03, 1.6928e-03, 8.8453e-04, 2.8534e-03, 6.4516e-04,
          3.7598e-02, 5.9738e-03, 8.3237e-03, 8.7585e-03, 8.5449e-03, 1.0216e-02,
          7.9489e-04, 9.0694e-04, 9.5177e-04, 9.4128e-04, 1.2608e-03, 3.2883e-03,
          5.6992e-03, 2.5482e-03, 9.6321e-04, 4.5204e-03, 2.5673e-03, 2.4223e-03,
          3.2692e-03, 2.0981e-03, 3.3760e-03, 9.7084e-04, 5.0125e-03, 1.3283e-02,
          7.5951e-03, 6.3629e-03, 7.7515e-02, 1.8001e-05, 1.8001e-05, 1.0085e-04,
          7.2479e-05, 9.9468e-04, 7.2479e-05, 1.3018e-04, 1.8001e-05, 4.3154e-05,
          4.7326e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 4.3154e-05, 1.8001e-05,
          1.0085e-04, 7.2479e-05, 7.2479e-05, 1.3018e-04, 1.8001e-05, 1.8001e-05,
          7.2479e-05, 1.8001e-05, 1.8001e-05, 1.0085e-04, 7.2777e-05, 7.2479e-05,
          4.3154e-05, 1.0085e-04, 4.3154e-05, 1.8001e-05, 7.2479e-05, 7.2479e-05,
          1.8001e-05, 1.8001e-05, 7.2479e-05, 4.3154e-05, 7.2479e-05, 1.3018e-04,
          7.2479e-05, 4.3154e-05, 1.8001e-05, 7.2479e-05, 7.2479e-05, 1.8001e-05,
          7.2479e-05, 1.8001e-05, 1.0085e-04, 7.2777e-05, 7.2479e-05, 7.2479e-05,
          1.0085e-04, 1.8001e-05, 1.8001e-05, 1.3018e-04, 4.3154e-05, 7.2479e-05,
          1.8001e-05, 1.3006e-04, 4.3154e-05, 2.6894e-04, 1.0544e-04, 7.2479e-05,
          2.0969e-04, 2.9659e-04, 1.8001e-05, 1.5867e-04, 4.3154e-05, 4.7326e-05,
          1.8001e-05, 1.0085e-04, 2.9635e-04, 7.2479e-05, 4.3154e-05, 8.0287e-05,
          5.0902e-05, 1.8001e-05, 1.8001e-05, 4.3154e-05, 4.7326e-05, 4.3154e-05,
          1.8001e-05, 2.0969e-04, 1.8442e-04, 1.9240e-04, 2.9659e-04, 1.3018e-04,
          6.1703e-04, 1.0490e-04, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          4.3154e-05, 1.1683e-03, 9.1124e-04, 4.3154e-05, 1.8001e-05, 1.0544e-04,
          1.8001e-05, 1.8001e-05, 7.2479e-05, 4.7326e-05, 7.2479e-05, 7.2479e-05,
          6.5041e-04, 7.2777e-05, 7.2479e-05, 4.3154e-05, 1.0085e-04, 1.3018e-04,
          1.3423e-04, 1.9169e-04, 1.8001e-05, 7.2479e-05, 1.8001e-05, 5.1880e-04,
          1.0544e-04, 1.8775e-04, 1.6272e-04, 1.8001e-05, 1.8001e-05, 4.3154e-05,
          1.9169e-04, 4.3154e-05, 1.8001e-05, 1.8001e-05, 9.7513e-05, 1.5938e-04,
          1.3006e-04, 1.8775e-04, 1.3006e-04, 7.2479e-05, 4.3154e-05, 1.0490e-04,
          1.8001e-05, 4.3154e-05, 3.9887e-04, 1.8001e-05, 1.8775e-04, 7.2479e-05,
          4.3154e-05, 1.8001e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05, 7.2479e-05,
          1.8001e-05, 7.2479e-05, 1.8001e-05, 1.0544e-04, 4.3154e-05, 1.3018e-04,
          5.1308e-04, 2.2972e-04, 1.8001e-05, 1.8001e-05, 1.0490e-04, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 4.3154e-05, 4.3154e-05, 7.2479e-05,
          4.3154e-05, 4.7326e-05, 7.2479e-05, 4.7326e-05, 4.3154e-05, 1.0085e-04,
          4.7278e-04, 7.2479e-05, 7.2479e-05, 1.8001e-05, 1.0085e-04, 7.2479e-05,
          1.5271e-04, 7.2479e-05, 1.8001e-05, 1.2445e-04, 2.7084e-04, 1.8001e-05,
          4.3154e-05, 7.2479e-05, 4.3154e-05, 5.0902e-05, 5.1022e-04, 2.4164e-04,
          4.3988e-04, 4.3154e-05, 1.0544e-04, 1.8215e-04, 4.3154e-05, 6.8188e-04,
          7.7772e-04, 4.1056e-04, 5.2023e-04, 7.2479e-05, 2.5225e-04, 4.3154e-05,
          3.1471e-04, 1.0085e-04, 7.2479e-05, 4.1342e-04, 4.3154e-05, 1.5271e-04,
          4.3154e-05, 1.6308e-04, 4.3154e-05, 1.8001e-05, 2.9922e-04, 1.8001e-05,
          7.7772e-04, 7.2479e-05, 1.3018e-04, 1.8001e-05, 1.8001e-05, 1.0085e-04,
          4.7994e-04, 4.3154e-05, 2.4164e-04, 1.5271e-04, 1.3018e-04, 2.6417e-04,
          2.9922e-04, 7.4816e-04, 1.8001e-05, 1.8001e-05, 1.0085e-04, 1.8001e-05,
          1.4257e-04, 2.5225e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.7326e-05,
          1.8001e-05, 1.7190e-04, 2.0039e-04, 3.0589e-04, 4.3154e-05, 1.4257e-04,
          4.3154e-05, 1.4257e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.3934e-03, 6.7993e-02, 1.9867e-02, 2.2385e-02, 4.4434e-02,
          6.5063e-02, 1.5823e-02, 3.2578e-03, 2.2717e-03, 8.7509e-03, 1.0025e-02,
          3.5076e-03, 5.8556e-03, 1.2293e-03, 1.4565e-02, 3.2959e-03, 5.9776e-03,
          3.4943e-03, 2.2221e-03, 5.2214e-04, 8.7166e-04, 6.5470e-04, 8.5373e-03,
          9.6054e-03, 6.1150e-03, 1.2932e-02, 1.1017e-02, 1.4257e-03, 7.4654e-03,
          1.9741e-03, 6.5079e-03, 2.1133e-03, 1.2405e-02, 1.1940e-02, 2.7428e-03,
          3.8280e-03, 9.0981e-04, 7.6408e-03, 9.8495e-03, 6.7282e-04, 2.8095e-03,
          9.2506e-04, 8.5688e-04, 7.4768e-04, 7.9060e-04, 8.6975e-03, 1.1108e-02,
          7.7057e-04, 1.0738e-03, 6.6299e-03, 3.7193e-03, 3.4027e-03, 3.6678e-03,
          8.6021e-04, 1.7424e-03, 1.0958e-03, 9.5272e-04, 1.5221e-03, 1.0595e-03,
          1.1501e-03, 2.2049e-03, 1.6832e-03, 3.5591e-03, 1.0767e-03, 1.4069e-02,
          1.3428e-03, 1.3962e-03, 2.7866e-03, 5.8031e-04, 6.0940e-04, 6.4230e-04,
          8.1921e-04, 2.0950e-02, 1.2636e-03, 4.3106e-03, 7.5769e-04, 8.0729e-04,
          1.5574e-03, 9.4175e-04, 1.0204e-03, 3.9291e-03, 1.0414e-03, 8.1682e-04,
          7.8773e-04, 4.8218e-03, 1.1034e-03, 6.6566e-03, 1.0681e-03, 9.3555e-04,
          7.6981e-03, 5.7220e-03, 6.1035e-04, 4.5166e-03, 6.8712e-04, 7.3481e-04,
          1.6336e-03, 8.1873e-04, 4.7112e-03, 5.8556e-03, 1.0109e-03, 1.3943e-03,
          7.5302e-03, 1.8282e-03, 1.1520e-03, 2.3212e-03, 1.3496e-02, 2.1362e-03,
          3.6297e-03, 9.4032e-04, 5.3711e-03, 8.7500e-04, 1.3504e-03, 2.0084e-03,
          4.8943e-03, 1.3523e-03, 1.6098e-03, 1.2779e-02, 9.6083e-04, 3.6392e-03,
          3.4542e-03, 6.0120e-03, 8.3017e-04, 2.5730e-03, 2.3697e-02, 6.9618e-04,
          8.2064e-04, 7.3385e-04, 1.0748e-03, 9.4681e-03, 8.8272e-03, 4.4594e-03,
          2.0084e-03, 1.2579e-03, 7.3051e-03, 1.0567e-03, 3.0880e-03, 6.3992e-04,
          2.1591e-02, 2.1648e-03, 9.8724e-03, 1.5320e-02, 2.0508e-02, 4.4098e-03,
          1.2112e-03, 4.4174e-03, 1.8539e-03, 1.2369e-03, 1.2674e-03, 7.5989e-03,
          8.1787e-03, 4.8752e-03, 1.1482e-03, 4.1008e-03, 2.6379e-03, 2.1629e-03,
          4.3793e-03, 2.0294e-03, 3.5648e-03, 7.9107e-04, 5.3139e-03, 3.8948e-03,
          1.2466e-02, 1.1124e-02, 1.6998e-02, 6.6467e-02, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7726e-04, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.2827e-04, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7583e-04, 1.7703e-05, 1.7583e-04,
          1.7703e-05, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 2.8658e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7583e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7583e-04, 1.7703e-05, 1.7703e-05, 1.7583e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7583e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04, 2.8658e-04, 1.7703e-05,
          1.7583e-04, 1.7583e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05, 1.7726e-04,
          1.7583e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7583e-04, 1.7703e-05, 1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04,
          1.7703e-05, 1.7703e-05, 2.8658e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05, 2.8658e-04,
          1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05,
          1.7703e-05, 1.2827e-04, 1.7703e-05, 1.7703e-05, 2.8658e-04, 6.8569e-04,
          1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.2827e-04, 1.7703e-05, 6.8569e-04, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05,
          1.2827e-04, 2.8658e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.2827e-04,
          1.2827e-04, 1.7703e-05, 2.8658e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05, 1.2827e-04, 1.7703e-05, 1.2827e-04, 1.7703e-05,
          1.7703e-05, 1.2827e-04, 1.2827e-04, 1.2827e-04, 1.7703e-05, 1.7703e-05,
          1.7703e-05, 1.7703e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.8959e-03, 5.7129e-02, 2.5375e-02, 2.7573e-02, 8.2092e-02,
          4.6967e-02, 2.3285e-02, 1.4429e-03, 1.6508e-03, 4.4136e-03, 5.1575e-03,
          2.4261e-03, 1.8644e-03, 1.4305e-03, 2.3441e-03, 9.7752e-04, 3.0327e-03,
          1.4648e-03, 7.4100e-04, 1.4334e-03, 1.7567e-03, 1.2178e-03, 2.1400e-03,
          4.6310e-03, 3.3970e-03, 1.6663e-02, 7.1144e-03, 3.9062e-03, 3.0174e-03,
          1.8597e-03, 1.9226e-03, 1.6899e-03, 5.1041e-03, 2.4643e-03, 1.7939e-03,
          7.4768e-03, 4.5242e-03, 5.3978e-03, 3.1166e-03, 1.6031e-03, 6.8130e-03,
          3.4657e-03, 2.9106e-03, 1.8454e-03, 2.2335e-03, 1.9817e-03, 5.8403e-03,
          9.1028e-04, 1.5965e-03, 1.0292e-02, 4.7531e-03, 2.1229e-03, 2.0466e-03,
          1.5678e-03, 1.4439e-03, 1.9512e-03, 9.3765e-03, 2.5177e-03, 6.1684e-03,
          3.7136e-03, 2.9736e-03, 2.0180e-03, 3.1013e-03, 3.3417e-03, 6.4583e-03,
          1.9531e-03, 1.0538e-03, 1.9302e-03, 1.8473e-03, 1.5793e-03, 1.4353e-03,
          1.7462e-03, 7.4387e-03, 3.5954e-03, 2.4414e-03, 1.1110e-03, 9.8324e-04,
          1.8358e-03, 1.4114e-02, 4.6234e-03, 5.5046e-03, 3.0117e-03, 2.4128e-03,
          3.0460e-03, 2.5425e-03, 1.4925e-03, 2.3060e-03, 1.1654e-03, 1.3475e-03,
          3.3054e-03, 3.5057e-03, 1.2150e-03, 2.4033e-03, 1.5898e-03, 1.8425e-03,
          4.0741e-03, 2.2373e-03, 1.7214e-03, 1.3596e-02, 2.0432e-02, 1.1589e-02,
          6.4163e-03, 1.9665e-03, 2.0313e-03, 7.6866e-03, 5.1079e-03, 3.9749e-03,
          2.3022e-03, 4.2953e-03, 5.2452e-03, 1.5383e-03, 1.8578e-03, 1.7328e-03,
          2.5063e-03, 5.1918e-03, 2.2297e-03, 5.6496e-03, 2.2717e-03, 1.1299e-02,
          9.9640e-03, 1.9470e-02, 9.6436e-03, 9.4271e-04, 4.9820e-03, 2.7580e-03,
          1.2299e-02, 3.2291e-03, 3.4943e-03, 6.4278e-03, 1.8187e-03, 2.7142e-03,
          2.4319e-03, 1.4849e-03, 7.9536e-04, 8.6594e-04, 2.3117e-03, 2.0447e-03,
          4.7531e-03, 2.5997e-03, 1.6241e-03, 3.6316e-03, 4.0054e-03, 5.4817e-03,
          1.3344e-02, 2.1820e-03, 5.4398e-03, 6.7253e-03, 3.8929e-03, 9.6741e-03,
          1.1322e-02, 6.6490e-03, 8.6899e-03, 5.2414e-03, 3.2635e-03, 1.0475e-02,
          3.1910e-03, 4.8370e-03, 8.6546e-04, 1.9331e-03, 8.5068e-03, 2.7790e-03,
          4.6577e-03, 5.5313e-03, 2.9697e-03, 4.1847e-03, 6.0791e-02, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.3793e-04, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          1.1778e-04, 2.3782e-05, 2.3782e-05, 2.3186e-04, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05, 2.3186e-04,
          1.1778e-04, 2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 1.3793e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.3793e-04,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3186e-04, 2.3782e-05,
          2.3782e-05, 1.1778e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3186e-04, 2.3782e-05, 2.3782e-05,
          1.1778e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          1.1778e-04, 2.3782e-05, 1.3793e-04, 1.3793e-04, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04,
          5.6171e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 5.6171e-04, 4.4751e-04, 2.3782e-05, 2.3782e-05,
          5.6171e-04, 3.4595e-04, 1.3793e-04, 2.3782e-05, 2.3782e-05, 2.3186e-04,
          2.3782e-05, 2.3782e-05, 2.3186e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          1.3793e-04, 2.3782e-05, 5.4121e-04, 2.3782e-05, 2.3782e-05, 2.3186e-04,
          2.3782e-05, 2.3782e-05, 1.3793e-04, 4.4751e-04, 2.3782e-05, 7.6962e-04,
          8.8358e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04, 1.1778e-04,
          1.1778e-04, 1.3793e-04, 2.3782e-05, 2.3782e-05, 2.3186e-04, 2.3782e-05,
          2.3186e-04, 2.3186e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          1.1778e-04, 4.6015e-04, 1.1778e-04, 1.3793e-04, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.1778e-04, 2.3782e-05,
          2.3186e-04, 2.3186e-04, 1.1778e-04, 2.3782e-05, 2.3782e-05, 8.8358e-04,
          2.3186e-04, 2.3782e-05, 9.7752e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          8.8358e-04, 1.3793e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05,
          2.3782e-05, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.3793e-04, 2.3782e-05,
          2.3782e-05, 1.3793e-04, 2.3782e-05, 2.3782e-05, 2.3782e-05, 1.3793e-04,
          2.3782e-05, 2.3782e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0343e-03, 6.2805e-02, 1.7288e-02, 2.4139e-02, 5.0903e-02,
          4.6417e-02, 1.5930e-02, 1.4915e-03, 1.7376e-03, 5.2490e-03, 9.1400e-03,
          3.4447e-03, 1.7176e-03, 1.3599e-03, 1.8988e-03, 1.0986e-03, 2.3613e-03,
          1.2798e-03, 9.0361e-04, 5.3692e-04, 6.6233e-04, 5.5027e-04, 1.6212e-03,
          8.9340e-03, 3.0746e-03, 1.2093e-02, 6.0501e-03, 1.2550e-03, 2.2869e-03,
          2.3518e-03, 2.4109e-03, 9.3651e-04, 3.7308e-03, 2.2011e-03, 1.9245e-03,
          4.7646e-03, 2.4967e-03, 4.5738e-03, 2.5520e-03, 8.4400e-04, 5.2681e-03,
          3.0289e-03, 2.5692e-03, 1.9855e-03, 1.5507e-03, 1.9073e-03, 3.7403e-03,
          8.4019e-04, 8.0347e-04, 4.9133e-03, 3.5362e-03, 2.4471e-03, 2.2736e-03,
          1.3065e-03, 1.5211e-03, 1.2465e-03, 4.9591e-03, 3.9139e-03, 6.8817e-03,
          3.4218e-03, 1.7633e-03, 1.7395e-03, 2.3212e-03, 6.6109e-03, 5.7373e-03,
          5.7220e-03, 3.7098e-03, 2.7580e-03, 1.6146e-03, 1.4391e-03, 1.7796e-03,
          1.2159e-03, 1.0223e-02, 2.1935e-03, 2.9564e-03, 1.1024e-03, 1.1692e-03,
          1.8578e-03, 7.4654e-03, 9.6817e-03, 8.3847e-03, 7.9880e-03, 2.5806e-03,
          1.9588e-03, 3.5038e-03, 1.4009e-03, 4.0512e-03, 3.6297e-03, 3.9139e-03,
          7.4730e-03, 6.9542e-03, 1.6890e-03, 4.9896e-03, 1.1339e-03, 1.1387e-03,
          1.5144e-03, 1.2331e-03, 2.7180e-03, 8.5220e-03, 1.2108e-02, 1.2283e-02,
          1.4183e-02, 3.9597e-03, 5.0240e-03, 1.0612e-02, 4.8485e-03, 2.3270e-03,
          3.1128e-03, 5.1804e-03, 9.6741e-03, 2.6112e-03, 4.2992e-03, 6.1035e-03,
          2.2793e-03, 3.1796e-03, 1.0242e-03, 5.0583e-03, 1.5430e-03, 4.1008e-03,
          5.2643e-03, 1.8280e-02, 8.6441e-03, 1.1625e-03, 6.5269e-03, 7.6103e-03,
          6.2332e-03, 7.3357e-03, 8.8882e-03, 5.5885e-03, 2.3937e-03, 3.5915e-03,
          4.9286e-03, 3.9635e-03, 9.0742e-04, 2.4624e-03, 6.3133e-03, 2.1534e-03,
          4.4670e-03, 2.0294e-03, 1.7757e-03, 3.4027e-03, 4.0283e-03, 4.8141e-03,
          1.8356e-02, 2.8152e-03, 8.7585e-03, 1.2131e-02, 1.0025e-02, 9.3002e-03,
          2.6123e-02, 6.9695e-03, 7.1144e-03, 6.2981e-03, 2.8477e-03, 9.9716e-03,
          9.6664e-03, 1.0139e-02, 1.5345e-03, 3.8948e-03, 6.3400e-03, 2.2278e-03,
          4.3945e-03, 7.6218e-03, 3.1052e-03, 3.8624e-03, 1.8127e-02, 3.3936e-02,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 8.7142e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          8.7142e-05, 1.3590e-05, 1.3590e-05, 6.0844e-04, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 8.7142e-05, 1.3590e-05, 6.0844e-04,
          8.7142e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.6856e-04, 1.6856e-04, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.6856e-04, 1.6856e-04, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 8.7142e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          9.5069e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 9.5069e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05,
          9.5069e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05, 1.6856e-04,
          1.3590e-05, 9.5069e-05, 1.6856e-04, 8.7142e-05, 1.3590e-05, 1.3590e-05,
          9.5069e-05, 1.3590e-05, 8.7142e-05, 1.3590e-05, 1.3590e-05, 1.6856e-04,
          1.3590e-05, 1.3590e-05, 1.6856e-04, 1.3590e-05, 1.3590e-05, 9.5069e-05,
          1.3590e-05, 8.7142e-05, 1.3590e-05, 1.3590e-05, 8.7142e-05, 8.7142e-05,
          8.7142e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.6856e-04, 9.5069e-05,
          1.6856e-04, 1.6856e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 9.5069e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 8.7142e-05, 1.3590e-05,
          1.6856e-04, 1.6856e-04, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05,
          8.7142e-05, 1.3590e-05, 1.6856e-04, 1.3590e-05, 1.3590e-05, 9.5069e-05,
          9.5069e-05, 9.5069e-05, 1.3590e-05, 9.5069e-05, 1.3590e-05, 1.3590e-05,
          1.3590e-05, 1.3590e-05, 1.3590e-05, 1.3590e-05, 9.5069e-05, 9.5069e-05,
          1.3590e-05, 9.5069e-05, 9.5069e-05, 9.5069e-05, 9.5069e-05, 9.5069e-05,
          1.3590e-05, 9.5069e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.2092e-03, 6.0120e-02, 1.7151e-02, 3.4943e-02, 5.2216e-02,
          4.7882e-02, 1.7075e-02, 2.4185e-03, 2.1572e-03, 9.8267e-03, 2.3041e-02,
          7.3586e-03, 2.4853e-03, 1.0815e-03, 3.1128e-03, 2.0885e-03, 2.9640e-03,
          1.9875e-03, 1.2455e-03, 2.9945e-04, 4.5228e-04, 3.8147e-04, 3.0804e-03,
          2.6154e-02, 3.2730e-03, 1.3672e-02, 8.5297e-03, 7.7009e-04, 3.6793e-03,
          1.6766e-03, 4.3945e-03, 1.2331e-03, 4.7302e-03, 6.5880e-03, 1.8187e-03,
          3.9673e-03, 1.6794e-03, 4.0665e-03, 3.5725e-03, 4.5514e-04, 3.8624e-03,
          1.2817e-03, 9.9754e-04, 9.6273e-04, 1.0328e-03, 3.4103e-03, 3.8662e-03,
          5.0068e-04, 4.2272e-04, 4.6196e-03, 2.8419e-03, 3.4904e-03, 2.3956e-03,
          8.8167e-04, 1.7910e-03, 7.3957e-04, 3.1281e-03, 3.0689e-03, 4.1313e-03,
          2.2964e-03, 1.2808e-03, 1.0805e-03, 1.4582e-03, 2.7294e-03, 6.4201e-03,
          3.5820e-03, 4.3716e-03, 2.6760e-03, 7.0524e-04, 6.3848e-04, 8.1062e-04,
          7.1526e-04, 1.8509e-02, 1.2445e-03, 3.5400e-03, 6.4754e-04, 7.6437e-04,
          1.2903e-03, 3.6087e-03, 6.2408e-03, 7.3242e-03, 5.2910e-03, 1.6851e-03,
          1.1101e-03, 3.9024e-03, 8.3351e-04, 1.1749e-02, 2.7943e-03, 1.8120e-03,
          2.3651e-02, 9.8953e-03, 7.4053e-04, 1.2909e-02, 6.2037e-04, 6.6757e-04,
          9.8801e-04, 5.9557e-04, 2.7065e-03, 5.4855e-03, 4.4022e-03, 5.6305e-03,
          4.0100e-02, 3.0537e-03, 5.3864e-03, 3.7689e-03, 4.7874e-03, 1.4124e-03,
          2.8591e-03, 1.9064e-03, 1.3351e-02, 8.9598e-04, 2.6112e-03, 3.5915e-03,
          2.1610e-03, 1.5049e-03, 5.3692e-04, 5.6114e-03, 8.2350e-04, 2.2297e-03,
          3.2063e-03, 1.2505e-02, 3.0594e-03, 1.5774e-03, 8.6365e-03, 2.6302e-03,
          1.9007e-03, 2.2297e-03, 3.4199e-03, 5.6801e-03, 4.5013e-03, 3.1815e-03,
          4.8943e-03, 4.0131e-03, 1.6775e-03, 1.5898e-03, 8.7128e-03, 8.5592e-04,
          6.7902e-03, 1.0061e-03, 2.1782e-03, 4.8790e-03, 5.5008e-03, 3.8528e-03,
          5.5084e-03, 2.2011e-03, 3.6869e-03, 4.2839e-03, 5.2109e-03, 7.2975e-03,
          1.6037e-02, 4.8485e-03, 2.2659e-03, 2.8648e-03, 1.6165e-03, 4.1199e-03,
          8.2016e-03, 3.9978e-03, 5.5962e-03, 1.4038e-03, 3.8414e-03, 1.8082e-03,
          5.3215e-03, 5.6953e-03, 6.0730e-03, 6.1150e-03, 6.8436e-03, 6.4888e-03,
          4.8767e-02, 2.2054e-05, 7.2837e-05, 2.2054e-05, 5.9009e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 1.6248e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 7.2837e-05, 1.0985e-04, 1.0985e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 7.2837e-05,
          7.2837e-05, 2.2054e-05, 2.2054e-05, 1.8418e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 1.3340e-04, 5.9009e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 1.6248e-04, 7.2837e-05, 2.2054e-05, 1.8418e-04,
          7.2837e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          5.9009e-05, 5.9009e-05, 2.2054e-05, 2.2054e-05, 1.0985e-04, 1.8418e-04,
          2.1327e-04, 1.6248e-04, 2.2054e-05, 2.2054e-05, 1.9908e-04, 2.2054e-05,
          2.2054e-05, 1.6248e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 3.0875e-04,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 7.2837e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 1.3340e-04, 1.0985e-04, 2.7370e-04,
          1.6248e-04, 2.2054e-05, 2.2054e-05, 2.2900e-04, 2.2054e-05, 5.9009e-05,
          2.2054e-05, 1.6248e-04, 1.6248e-04, 1.6248e-04, 5.9009e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          5.9009e-05, 2.2054e-05, 2.2054e-05, 1.0985e-04, 1.0985e-04, 1.9944e-04,
          5.9009e-05, 2.2054e-05, 1.0985e-04, 1.6248e-04, 3.6931e-04, 2.2054e-05,
          2.2054e-05, 7.2837e-05, 2.1386e-04, 1.0985e-04, 2.2054e-05, 2.2054e-05,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 2.7990e-04, 1.3340e-04,
          7.2837e-05, 5.9009e-05, 1.0985e-04, 1.0985e-04, 1.6248e-04, 2.2054e-05,
          1.6248e-04, 2.2054e-05, 1.6248e-04, 1.3340e-04, 2.2054e-05, 1.6248e-04,
          1.6248e-04, 5.9009e-05, 7.2837e-05, 3.3903e-04, 2.2054e-05, 2.2054e-05,
          1.3340e-04, 3.6240e-04, 2.2900e-04, 4.1342e-04, 2.2054e-05, 2.2054e-05,
          1.3340e-04, 2.2900e-04, 3.6931e-04, 1.0985e-04, 2.2054e-05, 5.9009e-05,
          2.2054e-05, 2.2054e-05, 1.1387e-03, 1.9944e-04, 1.6248e-04, 1.6248e-04,
          1.3340e-04, 2.0540e-04, 5.9009e-05, 2.2054e-05, 2.2054e-05, 1.0985e-04,
          2.2054e-05, 4.1342e-04, 1.8418e-04, 1.0985e-04, 2.2054e-05, 1.9205e-04,
          1.5461e-04, 7.2837e-05, 1.8418e-04, 3.6931e-04, 2.2054e-05, 1.8418e-04,
          2.2054e-05, 2.2054e-05, 4.2009e-04, 7.2837e-05, 1.6844e-04, 1.1387e-03,
          1.1387e-03, 7.2837e-05, 2.2054e-05, 2.2054e-05, 7.2837e-05, 7.2837e-05,
          1.3781e-03, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.0985e-04, 1.5461e-04,
          4.1342e-04, 2.4343e-04, 1.1760e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05,
          1.0985e-04, 3.6931e-04, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.6248e-04,
          2.2054e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 1.6844e-04, 1.6248e-04,
          4.1342e-04, 2.4343e-04, 2.2054e-05, 2.2054e-05, 3.6931e-04, 4.0722e-04,
          7.2837e-05, 3.6931e-04, 2.7990e-04, 2.2054e-05, 2.2054e-05, 4.6563e-04,
          5.0306e-04, 1.4982e-03, 2.2054e-05, 4.6563e-04, 4.6563e-04, 2.2054e-05,
          5.9009e-05, 2.2054e-05, 2.2054e-05, 2.2054e-05, 5.9009e-05, 1.0985e-04,
          2.2054e-05, 1.0985e-04, 5.9009e-05, 4.6396e-04, 5.9009e-05, 1.9670e-04,
          2.2054e-05, 2.2054e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.6605e-03, 6.2683e-02, 1.4175e-02, 2.3499e-02, 4.3335e-02,
          4.4495e-02, 1.3412e-02, 1.4639e-03, 1.4153e-03, 6.3057e-03, 1.0788e-02,
          4.4975e-03, 1.6193e-03, 1.1244e-03, 2.0161e-03, 1.1177e-03, 2.4624e-03,
          1.3666e-03, 8.0442e-04, 4.0150e-04, 5.1880e-04, 4.2582e-04, 1.6136e-03,
          1.1566e-02, 2.8019e-03, 1.0719e-02, 6.3438e-03, 8.0395e-04, 2.4948e-03,
          2.1877e-03, 2.5692e-03, 9.1839e-04, 4.0092e-03, 2.8133e-03, 1.8225e-03,
          5.7907e-03, 2.2564e-03, 4.8866e-03, 2.8801e-03, 6.5660e-04, 5.5618e-03,
          2.1496e-03, 1.7443e-03, 1.4524e-03, 1.4124e-03, 2.0924e-03, 3.8357e-03,
          6.4898e-04, 5.4502e-04, 3.8147e-03, 3.1586e-03, 2.6398e-03, 2.0180e-03,
          1.0443e-03, 1.5850e-03, 9.5415e-04, 4.5395e-03, 4.1008e-03, 6.4468e-03,
          3.2406e-03, 1.5745e-03, 1.3723e-03, 2.0046e-03, 5.1231e-03, 6.3477e-03,
          4.2114e-03, 4.6921e-03, 3.0003e-03, 1.1168e-03, 1.0281e-03, 1.2531e-03,
          1.0262e-03, 1.2917e-02, 1.6546e-03, 3.1796e-03, 8.6498e-04, 1.0653e-03,
          1.7138e-03, 6.2828e-03, 8.5678e-03, 1.0643e-02, 8.1711e-03, 2.4300e-03,
          1.6441e-03, 3.6011e-03, 1.0881e-03, 5.7602e-03, 3.0098e-03, 2.7733e-03,
          1.1879e-02, 9.4604e-03, 1.2207e-03, 7.7362e-03, 9.2316e-04, 1.0004e-03,
          1.1845e-03, 8.3828e-04, 3.1567e-03, 7.1182e-03, 7.9956e-03, 1.0643e-02,
          2.2217e-02, 3.9024e-03, 6.0272e-03, 1.1528e-02, 5.5733e-03, 1.9255e-03,
          3.1776e-03, 3.6697e-03, 1.2047e-02, 1.5278e-03, 3.0212e-03, 5.5733e-03,
          2.2011e-03, 2.7618e-03, 6.8140e-04, 5.3825e-03, 1.2236e-03, 2.1915e-03,
          3.4389e-03, 1.7288e-02, 4.7836e-03, 1.2026e-03, 8.0719e-03, 5.3329e-03,
          4.2648e-03, 4.9820e-03, 6.4354e-03, 6.7749e-03, 2.5578e-03, 3.5057e-03,
          3.9825e-03, 4.0436e-03, 9.1076e-04, 2.3689e-03, 9.0256e-03, 1.6022e-03,
          4.9934e-03, 1.4362e-03, 1.4505e-03, 3.3112e-03, 3.9330e-03, 3.9444e-03,
          1.0231e-02, 3.0937e-03, 6.5460e-03, 1.0902e-02, 7.4615e-03, 1.0025e-02,
          2.9755e-02, 7.9269e-03, 5.2185e-03, 5.4016e-03, 2.3594e-03, 8.6288e-03,
          1.2474e-02, 8.9035e-03, 2.2335e-03, 3.1548e-03, 5.8937e-03, 1.9932e-03,
          4.4937e-03, 8.2245e-03, 3.6068e-03, 3.6526e-03, 9.8114e-03, 1.9241e-02,
          2.3285e-02, 2.8580e-02, 1.3769e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05,
          4.8161e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.1301e-04, 4.8161e-05, 1.3769e-05,
          1.3769e-05, 2.7537e-04, 1.3769e-05, 1.3769e-05, 1.3769e-05, 7.8678e-05,
          7.8678e-05, 1.3769e-05, 1.3769e-05, 2.5582e-04, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 4.8161e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 7.8678e-05, 1.3769e-05, 5.1737e-04,
          7.8678e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 1.1301e-04,
          7.8678e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.1301e-04, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 3.7456e-04,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 4.8161e-05, 4.8161e-05, 1.1301e-04, 4.8161e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          4.8161e-05, 1.3769e-05, 1.3769e-05, 1.1301e-04, 1.1301e-04, 4.8161e-05,
          1.3769e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05,
          1.3769e-05, 4.8161e-05, 6.5327e-04, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 4.8161e-05,
          1.3769e-05, 4.8161e-05, 2.5582e-04, 1.1301e-04, 1.3769e-05, 4.8161e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 4.8161e-05, 7.8678e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          4.8161e-05, 1.9097e-04, 4.8161e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          4.5252e-04, 4.8161e-05, 3.9482e-04, 7.8678e-05, 4.8161e-05, 4.8161e-05,
          1.3769e-05, 1.3769e-05, 4.5252e-04, 3.7456e-04, 1.3769e-05, 4.8161e-05,
          4.5252e-04, 4.8161e-05, 1.9097e-04, 1.3769e-05, 1.3769e-05, 1.1301e-04,
          1.3769e-05, 1.1307e-04, 1.1301e-04, 7.8678e-05, 1.3769e-05, 4.8161e-05,
          4.8161e-05, 1.3769e-05, 1.1301e-04, 3.9482e-04, 1.3769e-05, 1.1301e-04,
          1.3769e-05, 1.3769e-05, 2.8706e-04, 1.3769e-05, 4.8161e-05, 6.5613e-04,
          3.0971e-04, 1.1301e-04, 1.3769e-05, 1.3769e-05, 7.8678e-05, 7.8678e-05,
          7.8678e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 4.8161e-05,
          1.1301e-04, 1.1301e-04, 1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05,
          1.3769e-05, 4.8161e-05, 1.3769e-05, 4.8161e-05, 1.3769e-05, 1.3769e-05,
          4.8161e-05, 4.8161e-05, 4.8161e-05, 1.3769e-05, 7.8678e-05, 1.3769e-05,
          2.5582e-04, 1.1301e-04, 1.3769e-05, 4.8161e-05, 1.3769e-05, 4.5252e-04,
          7.8678e-05, 6.5613e-04, 1.1301e-04, 1.3769e-05, 1.3769e-05, 4.8161e-05,
          4.5252e-04, 1.9097e-04, 1.3769e-05, 1.9097e-04, 1.9097e-04, 1.3769e-05,
          1.3769e-05, 1.3769e-05, 1.3769e-05, 1.3769e-05, 4.8161e-05, 4.8161e-05,
          1.3769e-05, 4.8161e-05, 4.8161e-05, 4.6229e-04, 4.8161e-05, 4.8161e-05,
          1.3769e-05, 1.3769e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6458e-03, 5.8960e-02, 1.4183e-02, 1.6891e-02, 4.6143e-02,
          4.8218e-02, 1.1162e-02, 1.7729e-03, 1.6804e-03, 6.2828e-03, 7.5645e-03,
          3.0880e-03, 2.3327e-03, 9.0694e-04, 7.0763e-03, 1.5163e-03, 2.4681e-03,
          1.4973e-03, 1.0948e-03, 3.6812e-04, 4.4465e-04, 3.6454e-04, 2.2850e-03,
          8.3084e-03, 7.2823e-03, 9.0408e-03, 6.0501e-03, 7.8630e-04, 3.5419e-03,
          1.4639e-03, 3.7880e-03, 1.0777e-03, 1.0521e-02, 5.8098e-03, 2.3403e-03,
          8.0261e-03, 1.4305e-03, 5.1613e-03, 6.4964e-03, 5.4121e-04, 3.1166e-03,
          9.9087e-04, 9.8324e-04, 7.9727e-04, 8.2207e-04, 3.2616e-03, 1.0002e-02,
          5.2452e-04, 4.4775e-04, 3.8261e-03, 3.1967e-03, 2.9850e-03, 2.7008e-03,
          9.2125e-04, 1.6222e-03, 9.9564e-04, 2.3746e-03, 2.5520e-03, 2.6512e-03,
          2.2125e-03, 1.4896e-03, 1.4095e-03, 1.9178e-03, 1.7433e-03, 2.5726e-02,
          2.3594e-03, 1.4124e-03, 2.6531e-03, 7.4482e-04, 5.2881e-04, 6.0892e-04,
          6.0034e-04, 1.2512e-02, 1.2827e-03, 3.4657e-03, 7.2098e-04, 7.5722e-04,
          1.6279e-03, 3.0289e-03, 3.0823e-03, 9.4299e-03, 2.6665e-03, 1.9531e-03,
          1.3475e-03, 3.9711e-03, 9.9182e-04, 3.9597e-03, 1.4200e-03, 1.3885e-03,
          7.0343e-03, 4.6997e-03, 9.3412e-04, 3.9330e-03, 1.0862e-03, 7.8821e-04,
          1.1654e-03, 7.4816e-04, 3.3722e-03, 5.9090e-03, 4.0016e-03, 5.3902e-03,
          1.2001e-02, 1.9684e-03, 2.3823e-03, 7.9193e-03, 1.6205e-02, 1.9474e-03,
          3.3188e-03, 1.4362e-03, 6.3438e-03, 1.1387e-03, 2.2926e-03, 2.2240e-03,
          9.2163e-03, 3.1700e-03, 6.0844e-04, 2.1423e-02, 1.1301e-03, 2.2678e-03,
          3.8509e-03, 1.8600e-02, 2.3098e-03, 1.3485e-03, 2.5208e-02, 2.0523e-03,
          2.8915e-03, 1.8053e-03, 2.8419e-03, 2.0538e-02, 4.7417e-03, 4.1161e-03,
          3.9291e-03, 1.9817e-03, 2.0523e-03, 1.0138e-03, 3.8376e-03, 9.4509e-04,
          8.0948e-03, 1.2636e-03, 2.1057e-03, 4.4403e-03, 6.0768e-03, 4.7760e-03,
          4.5280e-03, 3.1948e-03, 6.9008e-03, 4.3182e-03, 3.4714e-03, 3.6133e-02,
          1.6190e-02, 1.7975e-02, 4.2305e-03, 3.4332e-03, 2.2964e-03, 4.7760e-03,
          4.8180e-03, 3.1204e-03, 1.8482e-03, 1.3523e-03, 1.5686e-02, 2.4128e-03,
          1.1894e-02, 6.5765e-03, 5.3558e-03, 8.2779e-03, 7.8087e-03, 5.4703e-03,
          1.0750e-02, 6.3324e-03, 3.4637e-02, 1.6809e-05, 1.3554e-04, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.3554e-04, 7.4267e-05, 1.6809e-05,
          1.3554e-04, 6.0415e-04, 7.4267e-05, 1.6809e-05, 7.8022e-05, 1.6809e-05,
          1.3554e-04, 1.6809e-05, 1.6809e-05, 3.6168e-04, 1.3554e-04, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 7.8022e-05, 1.3554e-04, 6.7854e-04,
          1.3554e-04, 1.6809e-05, 1.6809e-05, 7.8022e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.3554e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.3554e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 6.0844e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 7.8022e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 7.8022e-05, 3.4857e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 7.8022e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 3.1424e-04,
          1.6809e-05, 7.8022e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.3554e-04, 1.3554e-04, 1.6809e-05,
          1.3554e-04, 7.8022e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 9.1696e-04, 5.1117e-04, 1.6809e-05, 7.8022e-05,
          7.8022e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          2.3913e-04, 1.6809e-05, 3.6168e-04, 7.0810e-04, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.3554e-04, 7.4267e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 7.4267e-05, 1.6809e-05, 7.8022e-05,
          3.9124e-04, 1.6809e-05, 2.2984e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.3554e-04, 1.0700e-03, 6.0415e-04, 1.6809e-05, 1.6809e-05,
          8.3065e-04, 9.0551e-04, 3.0041e-04, 1.6809e-05, 1.6809e-05, 1.3554e-04,
          1.6809e-05, 3.7551e-04, 1.3554e-04, 7.8022e-05, 1.6809e-05, 2.5678e-04,
          3.7551e-04, 1.6809e-05, 1.3554e-04, 5.2738e-04, 1.6809e-05, 5.1117e-04,
          1.6809e-05, 1.6809e-05, 5.7459e-04, 1.3554e-04, 4.8733e-04, 1.0700e-03,
          8.4448e-04, 3.7551e-04, 1.6809e-05, 1.6809e-05, 1.3554e-04, 1.3554e-04,
          5.4836e-04, 7.4267e-05, 1.6809e-05, 1.6809e-05, 3.7551e-04, 1.6809e-05,
          6.0129e-04, 1.3554e-04, 2.5678e-04, 7.8022e-05, 1.3554e-04, 1.6809e-05,
          1.6809e-05, 6.6137e-04, 1.6809e-05, 1.3554e-04, 1.6809e-05, 1.6809e-05,
          3.7551e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 3.7551e-04, 2.5678e-04,
          6.0129e-04, 1.3554e-04, 1.3554e-04, 7.4267e-05, 2.5678e-04, 9.3794e-04,
          1.3554e-04, 5.2738e-04, 7.5150e-04, 1.6809e-05, 1.6809e-05, 2.8729e-04,
          1.0700e-03, 1.1539e-03, 1.6809e-05, 5.1355e-04, 5.2738e-04, 1.6809e-05,
          7.4267e-05, 1.6809e-05, 7.4267e-05, 1.6809e-05, 7.4267e-05, 7.8022e-05,
          1.6809e-05, 7.4267e-05, 7.4267e-05, 1.6809e-05, 1.6809e-05, 6.4707e-04,
          7.4267e-05, 1.6809e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9357e-03, 6.4331e-02, 1.4847e-02, 1.4900e-02, 5.1422e-02,
          5.1819e-02, 1.0796e-02, 3.0804e-03, 2.3785e-03, 6.5880e-03, 1.0773e-02,
          3.5305e-03, 6.1264e-03, 1.1282e-03, 1.9211e-02, 2.6760e-03, 4.5853e-03,
          2.8706e-03, 2.0428e-03, 2.9802e-04, 3.9458e-04, 3.0518e-04, 3.7804e-03,
          1.0773e-02, 9.1629e-03, 8.3618e-03, 5.5237e-03, 6.2656e-04, 5.1689e-03,
          2.0618e-03, 5.9128e-03, 1.0300e-03, 9.7580e-03, 7.0610e-03, 2.3575e-03,
          4.9591e-03, 1.0052e-03, 3.4943e-03, 6.5117e-03, 4.5204e-04, 1.6537e-03,
          6.0558e-04, 5.8985e-04, 5.1498e-04, 6.1989e-04, 6.5498e-03, 8.0338e-03,
          5.0068e-04, 4.6897e-04, 4.0932e-03, 2.5692e-03, 3.1929e-03, 3.2330e-03,
          8.4496e-04, 1.8835e-03, 8.9550e-04, 1.3771e-03, 2.0981e-03, 1.5688e-03,
          1.7672e-03, 1.2379e-03, 1.2169e-03, 1.6394e-03, 9.7227e-04, 1.7166e-02,
          1.3390e-03, 1.2093e-03, 1.6546e-03, 4.6754e-04, 3.5024e-04, 4.4084e-04,
          4.5013e-04, 1.2474e-02, 9.7227e-04, 4.0321e-03, 7.4768e-04, 7.7581e-04,
          1.9159e-03, 1.4038e-03, 1.4400e-03, 5.5618e-03, 1.2436e-03, 1.1292e-03,
          9.3603e-04, 3.9139e-03, 8.3447e-04, 5.2299e-03, 8.8024e-04, 8.5497e-04,
          7.1716e-03, 3.3646e-03, 6.3419e-04, 3.4561e-03, 6.6853e-04, 4.8685e-04,
          9.0694e-04, 6.0129e-04, 3.2272e-03, 4.1924e-03, 1.6136e-03, 2.5482e-03,
          1.4297e-02, 2.4452e-03, 1.5574e-03, 3.0689e-03, 1.9302e-02, 1.5230e-03,
          3.6068e-03, 7.6342e-04, 4.2877e-03, 7.9679e-04, 1.4429e-03, 1.5640e-03,
          8.6441e-03, 1.5306e-03, 6.5899e-04, 2.0508e-02, 6.5517e-04, 1.5249e-03,
          2.4681e-03, 8.7204e-03, 1.1950e-03, 2.9488e-03, 2.4048e-02, 1.1597e-03,
          1.2722e-03, 8.8787e-04, 1.6155e-03, 1.9043e-02, 7.4348e-03, 4.5586e-03,
          2.3594e-03, 1.1873e-03, 6.5117e-03, 8.6689e-04, 2.4395e-03, 5.6314e-04,
          1.5686e-02, 1.6394e-03, 9.3307e-03, 8.4534e-03, 9.9106e-03, 3.9978e-03,
          2.0695e-03, 9.2239e-03, 3.1643e-03, 1.9913e-03, 1.8435e-03, 1.9211e-02,
          9.6817e-03, 1.1520e-02, 2.3670e-03, 2.4090e-03, 1.8864e-03, 1.9722e-03,
          3.1567e-03, 1.5526e-03, 4.5776e-03, 8.0729e-04, 7.0953e-03, 3.1643e-03,
          1.2405e-02, 5.5046e-03, 1.2222e-02, 1.9394e-02, 3.0193e-03, 2.5349e-03,
          1.2497e-02, 3.3627e-03, 1.3542e-02, 5.4077e-02, 1.7762e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 1.8132e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 6.2823e-05, 2.1696e-05, 2.1696e-05,
          6.2823e-05, 5.3692e-04, 6.2823e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.6630e-04, 2.1696e-05, 2.1696e-05, 5.7077e-04, 6.2823e-05, 2.1696e-05,
          2.1696e-05, 6.2823e-05, 2.1696e-05, 6.2823e-05, 6.2823e-05, 2.1696e-05,
          6.2823e-05, 6.2823e-05, 1.8132e-04, 2.1696e-05, 6.2823e-05, 1.7762e-04,
          1.6630e-04, 2.1696e-05, 2.1696e-05, 1.6630e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 1.7762e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.6630e-04,
          2.1696e-05, 1.8132e-04, 1.8132e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          6.2823e-05, 2.2244e-04, 1.8132e-04, 2.1696e-05, 2.1696e-05, 5.3692e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.2517e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 6.2823e-05, 2.2244e-04,
          1.8132e-04, 6.2823e-05, 1.8132e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.2244e-04, 1.8132e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 4.2653e-04,
          2.1696e-05, 6.2823e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 6.2823e-05, 1.6630e-04, 2.1696e-05,
          6.2823e-05, 1.6630e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 5.3692e-04, 1.6630e-04, 2.1696e-05, 6.2823e-05,
          1.6630e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 6.2823e-05, 2.1696e-05,
          1.2517e-04, 2.1696e-05, 1.7762e-04, 2.8110e-04, 1.8132e-04, 6.2823e-05,
          1.8132e-04, 1.8132e-04, 1.8132e-04, 2.1696e-05, 2.1696e-05, 1.8132e-04,
          2.1696e-05, 2.1696e-05, 1.6630e-04, 6.2823e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 1.7762e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          3.7718e-04, 6.2823e-05, 4.7135e-04, 2.1696e-05, 6.2823e-05, 2.1696e-05,
          2.1696e-05, 6.2823e-05, 6.5184e-04, 5.3692e-04, 1.8132e-04, 2.1696e-05,
          3.7718e-04, 7.8583e-04, 4.2653e-04, 2.1696e-05, 2.1696e-05, 2.8110e-04,
          2.1696e-05, 3.1185e-04, 2.8110e-04, 1.2517e-04, 2.1696e-05, 2.7061e-04,
          6.2823e-05, 2.1696e-05, 3.3617e-04, 5.8603e-04, 1.7762e-04, 4.4060e-04,
          2.1696e-05, 2.1696e-05, 1.7762e-04, 6.2823e-05, 2.1696e-05, 9.0075e-04,
          9.0075e-04, 5.3024e-04, 6.2823e-05, 2.1696e-05, 1.6630e-04, 1.6630e-04,
          2.1696e-05, 6.2823e-05, 2.1696e-05, 2.1696e-05, 2.8110e-04, 2.1696e-05,
          4.9829e-04, 4.9829e-04, 6.2823e-05, 6.2823e-05, 6.2823e-05, 2.1696e-05,
          2.1696e-05, 5.8603e-04, 2.1696e-05, 6.2823e-05, 2.1696e-05, 2.1696e-05,
          4.2653e-04, 2.1696e-05, 6.2823e-05, 2.1696e-05, 3.8362e-04, 4.2653e-04,
          4.2653e-04, 1.6630e-04, 6.2823e-05, 6.2823e-05, 2.7061e-04, 1.2083e-03,
          1.7762e-04, 5.8603e-04, 6.8951e-04, 6.2823e-05, 6.2823e-05, 2.1696e-05,
          9.0075e-04, 4.2653e-04, 2.1696e-05, 5.8603e-04, 5.8603e-04, 2.1696e-05,
          6.2823e-05, 6.2823e-05, 3.5238e-04, 6.2823e-05, 4.6706e-04, 1.6630e-04,
          6.2823e-05, 4.6706e-04, 3.5238e-04, 6.2823e-05, 6.2823e-05, 4.6706e-04,
          3.5238e-04, 6.2823e-05], dtype=torch.float16),
  tensor([0.0000e+00, 8.8425e-03, 5.8258e-02, 2.3010e-02, 2.8961e-02, 5.0812e-02,
          4.1687e-02, 1.4542e-02, 3.0155e-03, 2.1076e-03, 4.7073e-03, 6.5689e-03,
          2.6379e-03, 1.0887e-02, 1.0506e-02, 3.3951e-03, 2.8210e-03, 2.7969e-02,
          1.9119e-02, 1.4772e-03, 1.8387e-03, 1.1911e-03, 3.2234e-03, 7.8964e-03,
          6.8817e-03, 2.1801e-03, 1.2070e-02, 1.0834e-02, 3.3970e-03, 2.2797e-02,
          1.8021e-02, 6.6948e-03, 8.1024e-03, 6.4354e-03, 3.8681e-03, 2.5978e-03,
          2.7866e-03, 9.6703e-04, 4.0359e-03, 3.9005e-03, 3.8872e-03, 2.0771e-03,
          9.2411e-04, 6.8808e-04, 8.1301e-04, 6.2895e-04, 2.2964e-02, 5.1422e-03,
          8.7280e-03, 3.1872e-03, 5.8441e-03, 3.1395e-03, 4.5280e-03, 2.3289e-03,
          1.0233e-03, 2.1019e-03, 1.2350e-03, 1.2026e-03, 1.7786e-03, 1.3828e-03,
          1.3895e-03, 9.0561e-03, 1.4067e-03, 1.8415e-03, 1.1311e-03, 5.6953e-03,
          1.9436e-03, 1.2035e-03, 1.5650e-03, 4.6563e-04, 4.4751e-04, 6.5517e-04,
          7.9060e-04, 1.1154e-02, 1.4553e-03, 5.4703e-03, 9.5749e-04, 9.4175e-04,
          2.0103e-03, 1.3647e-03, 1.6584e-03, 2.9697e-03, 1.6317e-03, 8.8024e-04,
          8.9788e-04, 5.4550e-03, 1.2388e-03, 3.3531e-03, 1.2369e-03, 1.1816e-03,
          5.9166e-03, 3.5973e-03, 5.6696e-04, 3.1376e-03, 5.8794e-04, 7.9393e-04,
          3.4351e-03, 8.1301e-04, 4.3793e-03, 4.6120e-03, 1.3466e-03, 1.9016e-03,
          6.6910e-03, 1.2627e-03, 1.5144e-03, 1.7996e-03, 3.9558e-03, 6.9084e-03,
          3.5610e-03, 1.1024e-03, 3.9864e-03, 6.7568e-04, 1.4458e-03, 1.5726e-03,
          1.2445e-03, 1.0815e-03, 2.3422e-03, 5.5962e-03, 1.0118e-03, 4.2343e-03,
          1.1360e-02, 4.7188e-03, 8.7976e-04, 2.1763e-03, 5.8479e-03, 8.9073e-04,
          9.7561e-04, 1.0662e-03, 1.5144e-03, 5.0049e-03, 5.6190e-03, 4.1161e-03,
          2.1973e-03, 1.6479e-03, 1.1959e-03, 7.9250e-04, 2.2850e-03, 6.3944e-04,
          2.8244e-02, 7.9269e-03, 6.4087e-03, 6.8283e-03, 7.0114e-03, 2.0615e-02,
          1.3838e-03, 9.2411e-04, 1.5297e-03, 1.4038e-03, 1.9760e-03, 3.9482e-03,
          5.5923e-03, 3.1128e-03, 1.3399e-03, 3.8204e-03, 2.5215e-03, 2.2869e-03,
          2.9297e-03, 1.7443e-03, 1.6365e-03, 1.0376e-03, 3.6221e-03, 1.5144e-02,
          6.3057e-03, 4.5013e-03, 3.4088e-02, 5.8975e-03, 3.0651e-03, 2.0008e-03,
          5.1270e-03, 2.3594e-03, 3.4580e-03, 2.7523e-03, 3.7262e-02, 1.2934e-05,
          5.1439e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 6.3241e-05, 1.2934e-05,
          6.3241e-05, 1.0180e-04, 1.0180e-04, 1.0180e-04, 1.2934e-05, 1.2934e-05,
          1.0180e-04, 1.2934e-05, 1.2934e-05, 6.3241e-05, 1.0180e-04, 1.0180e-04,
          6.3241e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.0180e-04, 5.1439e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.8072e-04, 6.3241e-05, 1.0180e-04,
          1.0180e-04, 6.3241e-05, 1.2934e-05, 5.1439e-05, 1.0180e-04, 1.2934e-05,
          5.1439e-05, 1.2934e-05, 1.2934e-05, 1.0180e-04, 1.0180e-04, 1.2934e-05,
          6.3241e-05, 1.2934e-05, 1.2934e-05, 1.0180e-04, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 5.2547e-04, 8.0776e-04, 2.0278e-04, 1.0180e-04,
          1.8072e-04, 2.0278e-04, 4.7493e-04, 1.5247e-04, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 6.3241e-05, 5.1439e-05, 5.1439e-05, 1.2934e-05, 6.6459e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 3.0541e-04, 1.2934e-05, 1.5247e-04, 3.9911e-04, 1.0180e-04,
          3.8338e-04, 8.8215e-05, 8.8215e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.0180e-04, 1.2934e-05, 1.2934e-05, 1.2934e-05, 5.1439e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 6.3241e-05,
          7.4291e-04, 1.0180e-04, 2.0278e-04, 1.0633e-03, 6.3241e-05, 1.0180e-04,
          1.1384e-04, 5.1439e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 2.9731e-04,
          1.0180e-04, 1.9348e-04, 1.9348e-04, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          6.6459e-05, 1.2934e-05, 1.2934e-05, 6.3241e-05, 1.2934e-05, 1.2934e-05,
          4.7493e-04, 2.4378e-04, 6.5565e-04, 5.1439e-05, 6.3241e-05, 1.9348e-04,
          1.2934e-05, 1.2934e-05, 2.1935e-04, 1.2934e-05, 2.4378e-04, 5.1439e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05, 5.1439e-05, 5.1439e-05,
          1.2934e-05, 5.1439e-05, 1.2934e-05, 1.0180e-04, 1.2934e-05, 1.0180e-04,
          1.3332e-03, 2.1935e-04, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.2934e-05, 1.2934e-05, 1.9348e-04, 1.2934e-05, 1.2934e-05,
          6.3241e-05, 5.1439e-05, 1.2934e-05, 1.0180e-04, 1.2934e-05, 1.2934e-05,
          2.3198e-04, 5.1439e-05, 5.1439e-05, 1.2934e-05, 7.6914e-04, 1.2934e-05,
          6.3241e-05, 5.1439e-05, 1.2934e-05, 5.1439e-05, 3.9005e-04, 1.2934e-05,
          1.2934e-05, 6.3241e-05, 1.9348e-04, 1.9348e-04, 1.0180e-04, 1.0180e-04,
          2.3198e-04, 1.2934e-05, 2.8229e-04, 5.1439e-05, 1.2934e-05, 1.0180e-04,
          3.9983e-04, 2.8229e-04, 2.4378e-04, 1.2934e-05, 5.8842e-04, 5.1439e-05,
          1.0180e-04, 1.2934e-05, 5.1439e-05, 4.4036e-04, 1.2934e-05, 1.2934e-05,
          1.2934e-05, 1.9348e-04, 1.2934e-05, 1.2934e-05, 2.8229e-04, 1.2934e-05,
          1.8330e-03, 1.2934e-05, 2.1935e-04, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          8.0776e-04, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.0180e-04, 2.6703e-04,
          1.2934e-05, 2.3193e-03, 1.2934e-05, 1.2934e-05, 1.2934e-05, 1.2934e-05,
          4.2295e-04, 1.8072e-04, 1.8072e-04, 1.8072e-04, 1.8072e-04, 1.0180e-04,
          1.2934e-05, 2.1935e-04, 1.8072e-04, 4.6921e-04, 1.8072e-04, 1.8072e-04,
          1.8072e-04, 1.8072e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3062e-03, 5.8350e-02, 1.4908e-02, 1.8036e-02, 4.3732e-02,
          4.4800e-02, 9.6893e-03, 1.4849e-03, 1.3304e-03, 4.4441e-03, 4.1351e-03,
          2.1954e-03, 2.6550e-03, 9.7036e-04, 5.4016e-03, 1.3266e-03, 5.4817e-03,
          2.1877e-03, 7.7343e-04, 6.6280e-04, 8.1873e-04, 4.7159e-04, 2.7161e-03,
          4.4899e-03, 3.8204e-03, 8.0109e-03, 8.0643e-03, 1.0138e-03, 6.6719e-03,
          1.3599e-03, 3.5782e-03, 2.1210e-03, 1.2299e-02, 4.7569e-03, 3.0289e-03,
          1.0590e-02, 1.6794e-03, 8.3847e-03, 7.2098e-03, 7.9060e-04, 5.7106e-03,
          9.9945e-04, 8.6737e-04, 6.4278e-04, 1.1978e-03, 4.9820e-03, 1.5236e-02,
          5.4121e-04, 5.6696e-04, 3.7594e-03, 4.9820e-03, 3.2673e-03, 2.5024e-03,
          9.8228e-04, 2.0752e-03, 1.3762e-03, 4.0207e-03, 2.4986e-03, 3.4122e-03,
          2.4548e-03, 2.8381e-03, 1.6985e-03, 3.7098e-03, 1.5411e-03, 1.9333e-02,
          1.0252e-03, 7.9203e-04, 2.0790e-03, 6.2513e-04, 4.4560e-04, 5.1355e-04,
          6.6137e-04, 1.0315e-02, 1.5068e-03, 3.7022e-03, 8.0252e-04, 9.9564e-04,
          2.3327e-03, 3.8948e-03, 2.1744e-03, 8.7433e-03, 2.1286e-03, 1.8854e-03,
          1.4944e-03, 4.6921e-03, 1.1129e-03, 2.2221e-03, 7.0429e-04, 7.4387e-04,
          4.4327e-03, 5.6648e-03, 5.2929e-04, 2.8076e-03, 9.3317e-04, 8.2302e-04,
          1.6251e-03, 8.7309e-04, 4.7913e-03, 8.3313e-03, 3.3112e-03, 5.5084e-03,
          7.1220e-03, 1.2112e-03, 1.8244e-03, 8.7433e-03, 1.2253e-02, 3.3531e-03,
          4.1695e-03, 2.5997e-03, 7.3814e-03, 7.2432e-04, 9.2125e-04, 1.8129e-03,
          2.1801e-03, 3.8357e-03, 1.1234e-03, 1.8219e-02, 9.6321e-04, 2.4147e-03,
          4.0321e-03, 1.3985e-02, 1.3561e-03, 1.1187e-03, 1.3535e-02, 1.4009e-03,
          3.2768e-03, 1.6813e-03, 2.5558e-03, 2.1790e-02, 4.7569e-03, 4.6349e-03,
          1.5526e-03, 8.6117e-04, 9.5463e-04, 6.4325e-04, 3.2482e-03, 7.5150e-04,
          1.7349e-02, 1.8387e-03, 1.8959e-03, 4.2076e-03, 7.8087e-03, 4.9744e-03,
          2.4033e-03, 1.5879e-03, 3.3016e-03, 4.3640e-03, 1.7633e-03, 1.5976e-02,
          1.9791e-02, 1.8448e-02, 6.2332e-03, 9.4757e-03, 3.6201e-03, 1.0124e-02,
          6.0081e-03, 4.9934e-03, 9.7513e-04, 9.3555e-04, 1.3702e-02, 4.3869e-03,
          1.6663e-02, 1.2192e-02, 8.6212e-03, 7.4768e-03, 7.2670e-03, 4.8790e-03,
          5.4741e-03, 6.5155e-03, 1.0284e-02, 6.1150e-03, 6.0196e-03, 4.2511e-02,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.4150e-04, 1.2338e-05, 1.2338e-05,
          5.8591e-05, 3.1137e-04, 1.2338e-05, 1.2338e-05, 9.5308e-05, 1.2338e-05,
          1.4150e-04, 1.2338e-05, 1.2338e-05, 3.1662e-04, 5.8591e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 5.8591e-05, 5.8591e-05, 1.2338e-05,
          5.8591e-05, 5.8591e-05, 1.2338e-05, 9.5308e-05, 5.8591e-05, 3.1662e-04,
          1.4150e-04, 1.2338e-05, 1.2338e-05, 1.4150e-04, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 5.8591e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.4150e-04,
          1.2338e-05, 2.6512e-04, 1.2338e-05, 5.8591e-05, 1.2338e-05, 1.2338e-05,
          1.0246e-04, 5.8591e-05, 2.6512e-04, 1.2338e-05, 1.2338e-05, 3.9434e-04,
          1.2338e-05, 1.2338e-05, 9.5308e-05, 1.2338e-05, 1.2338e-05, 9.5308e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.4150e-04,
          1.2338e-05, 5.8591e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          5.8591e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 3.0923e-04,
          1.2338e-05, 1.4150e-04, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.4150e-04, 1.4150e-04, 1.2338e-05,
          1.4150e-04, 1.4150e-04, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 8.7452e-04, 1.4150e-04, 1.2338e-05, 1.4150e-04,
          1.4150e-04, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          9.5308e-05, 1.2338e-05, 5.8591e-05, 3.1662e-04, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.4150e-04, 5.8591e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          5.8591e-05, 1.2338e-05, 5.8591e-05, 1.2338e-05, 1.2338e-05, 1.2338e-05,
          1.2338e-05, 5.8591e-05, 8.7452e-04, 5.8591e-05, 1.2338e-05, 1.2338e-05,
          6.2180e-04, 1.0786e-03, 2.3365e-04, 1.2338e-05, 1.2338e-05, 1.4150e-04,
          1.2338e-05, 2.6298e-04, 5.6744e-04, 1.2338e-05, 1.2338e-05, 2.6298e-04,
          1.2338e-05, 1.2338e-05, 9.5308e-05, 5.6219e-04, 5.8591e-05, 1.4150e-04,
          1.2338e-05, 1.2338e-05, 2.3365e-04, 3.2187e-04, 3.5858e-04, 1.3008e-03,
          1.3008e-03, 5.6744e-04, 1.2338e-05, 1.2338e-05, 1.4150e-04, 1.4150e-04,
          3.5858e-04, 5.8591e-05, 1.2338e-05, 1.2338e-05, 5.6744e-04, 1.2338e-05,
          5.6744e-04, 3.1662e-04, 1.2338e-05, 5.8591e-05, 5.8591e-05, 1.2338e-05,
          1.2338e-05, 8.2016e-04, 1.2338e-05, 5.8591e-05, 1.2338e-05, 1.2338e-05,
          4.8423e-04, 1.2338e-05, 1.2338e-05, 1.2338e-05, 3.9220e-04, 2.6298e-04,
          3.9220e-04, 1.4150e-04, 5.8591e-05, 5.8591e-05, 2.6298e-04, 7.4911e-04,
          3.9220e-04, 7.3719e-04, 9.7132e-04, 1.2338e-05, 5.8591e-05, 1.2338e-05,
          7.3719e-04, 5.7268e-04, 1.2338e-05, 5.8591e-05, 7.3719e-04, 1.2338e-05,
          3.2616e-04, 5.8591e-05, 3.2616e-04, 5.8591e-05, 5.8591e-05, 1.2338e-05,
          1.2338e-05, 1.4150e-04, 5.8591e-05, 3.9506e-04, 5.8591e-05, 8.3780e-04,
          3.2616e-04, 5.8591e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6616e-03, 5.4138e-02, 1.5556e-02, 1.0033e-02, 5.1453e-02,
          4.9072e-02, 7.4654e-03, 2.3460e-03, 1.7042e-03, 3.6240e-03, 6.0005e-03,
          2.5806e-03, 1.8072e-03, 8.1873e-04, 4.1618e-03, 2.0905e-03, 1.3695e-03,
          8.6498e-04, 9.8038e-04, 4.5133e-04, 4.5156e-04, 4.3201e-04, 1.2169e-03,
          7.0152e-03, 5.9166e-03, 6.3248e-03, 2.6798e-03, 7.6532e-04, 1.7271e-03,
          9.7942e-04, 4.0855e-03, 7.4244e-04, 4.0970e-03, 1.9569e-03, 3.1929e-03,
          5.9052e-03, 2.9526e-03, 1.6909e-03, 2.2621e-03, 6.4611e-04, 2.2793e-03,
          1.1244e-03, 9.5940e-04, 9.8324e-04, 8.7357e-04, 1.3523e-03, 3.4866e-03,
          4.9686e-04, 5.0592e-04, 4.6501e-03, 4.8065e-03, 3.6087e-03, 2.0237e-03,
          2.0695e-03, 2.7161e-03, 2.3499e-03, 3.8967e-03, 8.7357e-03, 7.3128e-03,
          6.7673e-03, 1.9817e-03, 3.0575e-03, 3.1548e-03, 1.8263e-03, 8.4305e-03,
          1.5364e-03, 1.0452e-03, 1.0557e-03, 7.2956e-04, 6.2704e-04, 9.8038e-04,
          9.2983e-04, 4.7989e-03, 2.3956e-03, 4.6043e-03, 2.0008e-03, 1.7023e-03,
          3.0365e-03, 3.6926e-03, 2.6913e-03, 7.0572e-03, 3.0994e-03, 8.8806e-03,
          5.4550e-03, 6.3972e-03, 2.3861e-03, 3.1204e-03, 7.7248e-04, 1.0796e-03,
          4.8866e-03, 3.4676e-03, 1.4448e-03, 2.9621e-03, 2.7237e-03, 1.8854e-03,
          1.2674e-03, 1.4706e-03, 3.4351e-03, 4.5624e-03, 3.5458e-03, 4.1656e-03,
          7.4272e-03, 1.5297e-03, 2.0027e-03, 1.2558e-02, 1.3237e-02, 2.6741e-03,
          7.2365e-03, 1.7872e-03, 3.9444e-03, 1.6985e-03, 1.5526e-03, 1.1597e-03,
          5.7793e-03, 2.3823e-03, 9.2888e-04, 8.5449e-03, 2.3975e-03, 2.2850e-03,
          2.9640e-03, 7.0076e-03, 1.8139e-03, 2.0332e-03, 1.0963e-02, 1.7176e-03,
          3.9749e-03, 3.3417e-03, 4.7379e-03, 1.6296e-02, 3.7212e-03, 8.3771e-03,
          3.0098e-03, 9.6750e-04, 1.6975e-03, 7.8297e-04, 2.6283e-03, 1.4143e-03,
          3.2501e-03, 1.7157e-03, 1.9207e-03, 2.9259e-03, 2.8019e-03, 3.5305e-03,
          2.7752e-03, 2.7657e-03, 3.8872e-03, 4.3755e-03, 2.4738e-03, 2.7008e-02,
          1.4870e-02, 1.9638e-02, 2.1896e-02, 6.7978e-03, 8.1863e-03, 3.9711e-03,
          3.2349e-03, 2.9831e-03, 1.6232e-03, 1.9970e-03, 8.3160e-03, 2.1019e-03,
          4.1962e-03, 2.8877e-03, 2.2106e-03, 3.4161e-03, 4.4746e-03, 2.9659e-03,
          4.9057e-03, 3.3340e-03, 8.9722e-03, 4.9362e-03, 2.6627e-03, 8.7357e-03,
          1.3440e-01, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 7.2718e-05, 7.2718e-05, 7.2718e-05, 7.2718e-05, 3.4213e-05,
          3.2043e-04, 3.4213e-05, 3.4213e-05, 1.9825e-04, 1.9503e-04, 3.4213e-05,
          3.4213e-05, 7.2718e-05, 3.4213e-05, 7.2718e-05, 7.2718e-05, 3.4213e-05,
          7.2718e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05, 1.9825e-04,
          1.8787e-04, 3.4213e-05, 7.2718e-05, 7.2718e-05, 7.2718e-05, 7.2718e-05,
          3.4213e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 1.9825e-04,
          3.4213e-05, 3.4213e-05, 7.2718e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05,
          7.2718e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 1.5974e-04,
          3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05, 7.2718e-05,
          3.4213e-05, 7.2718e-05, 3.4213e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 2.1439e-03,
          3.4213e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05,
          3.4213e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05, 1.9825e-04, 3.4213e-05,
          7.2718e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05,
          3.4213e-05, 3.4213e-05, 2.2354e-03, 1.9825e-04, 3.4213e-05, 7.2718e-05,
          7.2718e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 7.2718e-05, 3.4213e-05,
          1.5974e-04, 3.4213e-05, 1.9825e-04, 3.2043e-04, 3.4213e-05, 3.4213e-05,
          3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.4213e-05,
          3.4213e-05, 3.4213e-05, 4.3583e-04, 7.2718e-05, 3.4213e-05, 7.2718e-05,
          3.4213e-05, 7.2718e-05, 3.4213e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 3.4213e-05, 3.7503e-04, 3.4213e-05, 3.4213e-05, 7.2718e-05,
          3.4213e-05, 1.9503e-04, 3.7503e-04, 7.2718e-05, 3.4213e-05, 3.4213e-05,
          7.2718e-05, 7.2718e-05, 7.2718e-05, 3.4213e-05, 3.4213e-05, 1.9932e-04,
          7.2718e-05, 3.7551e-04, 4.2439e-04, 3.4213e-05, 3.4213e-05, 7.2718e-05,
          3.4213e-05, 3.4213e-05, 7.2718e-05, 3.7503e-04, 7.2718e-05, 1.9860e-04,
          7.2718e-05, 3.4213e-05, 5.0068e-04, 3.1352e-04, 3.4213e-05, 3.7503e-04,
          1.0777e-03, 2.3689e-03, 3.4213e-05, 3.4213e-05, 3.2043e-04, 1.9825e-04,
          3.4213e-05, 1.8787e-04, 3.4213e-05, 1.8787e-04, 3.2163e-04, 3.4213e-05,
          4.3583e-04, 6.6042e-04, 7.2718e-05, 7.2718e-05, 3.1018e-04, 3.4213e-05,
          7.2718e-05, 6.0034e-04, 3.4213e-05, 7.2718e-05, 3.4213e-05, 1.4937e-04,
          2.9778e-04, 3.4213e-05, 3.4213e-05, 3.4213e-05, 3.2043e-04, 6.0034e-04,
          6.1274e-04, 6.1131e-04, 1.9503e-04, 3.4213e-05, 7.2718e-05, 2.1648e-03,
          3.1018e-04, 7.1526e-04, 7.9155e-04, 1.8787e-04, 1.8787e-04, 1.8787e-04,
          1.0061e-03, 8.3733e-04, 1.4937e-04, 3.7503e-04, 4.9019e-04, 3.4213e-05,
          5.3501e-04, 5.3501e-04, 1.9503e-04, 3.4213e-05, 1.9503e-04, 3.4213e-05,
          3.4213e-05, 7.2718e-05, 7.2718e-05, 3.5334e-04, 3.1471e-04, 4.7565e-04,
          3.4213e-05, 7.2718e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1308e-03, 5.4993e-02, 1.6785e-02, 1.1650e-02, 6.5918e-02,
          4.4189e-02, 1.0498e-02, 2.1954e-03, 1.9703e-03, 2.5501e-03, 5.6686e-03,
          2.2011e-03, 1.3542e-03, 1.1234e-03, 2.5730e-03, 1.8291e-03, 1.4181e-03,
          8.7738e-04, 9.5892e-04, 6.5851e-04, 5.9652e-04, 5.2166e-04, 9.6989e-04,
          5.8784e-03, 3.9902e-03, 8.5068e-03, 2.3232e-03, 1.1396e-03, 1.4248e-03,
          1.1797e-03, 3.5496e-03, 6.7902e-04, 2.2984e-03, 1.0490e-03, 2.5444e-03,
          4.7073e-03, 4.0321e-03, 1.1921e-03, 1.3447e-03, 8.8978e-04, 2.0046e-03,
          1.6155e-03, 1.2789e-03, 1.4725e-03, 8.5163e-04, 1.1292e-03, 1.9894e-03,
          5.7030e-04, 7.3338e-04, 6.5842e-03, 4.9477e-03, 3.3283e-03, 2.0103e-03,
          3.2787e-03, 2.8439e-03, 3.7365e-03, 4.4289e-03, 1.1459e-02, 8.7967e-03,
          8.5831e-03, 2.3403e-03, 5.6343e-03, 3.3226e-03, 2.0599e-03, 4.0855e-03,
          2.2240e-03, 1.0157e-03, 7.6246e-04, 1.0242e-03, 9.8038e-04, 1.6308e-03,
          1.4524e-03, 3.0994e-03, 3.7537e-03, 4.5242e-03, 3.2463e-03, 2.0504e-03,
          3.8948e-03, 4.4899e-03, 3.4924e-03, 4.8828e-03, 4.1695e-03, 1.0109e-02,
          7.5035e-03, 6.7368e-03, 4.1466e-03, 2.0370e-03, 1.0157e-03, 1.5736e-03,
          3.8052e-03, 2.1915e-03, 1.6708e-03, 2.4052e-03, 2.3727e-03, 2.0599e-03,
          1.6117e-03, 2.3003e-03, 2.1057e-03, 3.6926e-03, 6.1684e-03, 4.8485e-03,
          5.4321e-03, 1.4391e-03, 1.7233e-03, 9.6664e-03, 7.0724e-03, 3.4580e-03,
          7.3204e-03, 2.8496e-03, 2.8820e-03, 2.8458e-03, 2.2087e-03, 1.2207e-03,
          3.1452e-03, 2.1782e-03, 1.2903e-03, 4.4479e-03, 2.6035e-03, 4.2305e-03,
          4.3411e-03, 4.7684e-03, 3.2139e-03, 1.8625e-03, 5.1613e-03, 2.4433e-03,
          5.3024e-03, 4.9400e-03, 5.9242e-03, 8.6823e-03, 2.7733e-03, 8.0643e-03,
          3.9406e-03, 1.1101e-03, 8.4066e-04, 8.5926e-04, 2.0313e-03, 2.2411e-03,
          2.1267e-03, 2.4090e-03, 1.9188e-03, 2.3975e-03, 2.1019e-03, 3.7651e-03,
          4.4975e-03, 1.8158e-03, 3.8204e-03, 4.9248e-03, 3.5381e-03, 1.5030e-02,
          9.8038e-03, 1.1292e-02, 2.2934e-02, 7.4463e-03, 1.1650e-02, 5.0888e-03,
          2.2087e-03, 2.8915e-03, 1.1463e-03, 3.4180e-03, 5.8250e-03, 2.3270e-03,
          2.5921e-03, 1.6565e-03, 1.6155e-03, 2.3689e-03, 6.1989e-03, 3.4752e-03,
          3.4599e-03, 2.7523e-03, 5.2948e-03, 2.7599e-03, 2.3556e-03, 4.8561e-03,
          5.7526e-02, 1.1963e-01, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05,
          8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05,
          2.4605e-04, 2.7835e-05, 2.7835e-05, 2.6202e-04, 2.4605e-04, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 8.8334e-05, 2.7835e-05,
          2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.6202e-04,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 8.8334e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 2.0146e-04,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 8.8334e-05, 2.7835e-05,
          8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.0146e-04,
          2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.0146e-04, 8.8334e-05,
          2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 2.0146e-04, 2.6202e-04, 2.7835e-05,
          8.8334e-05, 2.7895e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 8.8334e-05,
          8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.6202e-04, 4.1962e-04, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 1.4181e-03, 2.4605e-04, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 8.8334e-05,
          2.7835e-05, 2.4605e-04, 8.8334e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05,
          8.8334e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05, 2.6202e-04,
          2.7835e-05, 2.7895e-05, 8.9109e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 8.8334e-05, 8.8334e-05, 8.8334e-05, 8.8394e-05,
          2.7835e-05, 2.7835e-05, 2.6202e-04, 8.8334e-05, 2.7835e-05, 8.8334e-05,
          8.8334e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.4605e-04, 2.6202e-04,
          2.7835e-05, 2.7835e-05, 8.8334e-05, 2.7013e-04, 8.8394e-05, 8.8334e-05,
          4.2796e-04, 6.0177e-04, 8.8334e-05, 2.7835e-05, 4.2796e-04, 2.7835e-05,
          2.7013e-04, 8.8394e-05, 2.7835e-05, 8.8334e-05, 2.7835e-05, 2.7835e-05,
          2.7835e-05, 2.7835e-05, 2.7835e-05, 2.7835e-05, 2.4605e-04, 2.7835e-05,
          2.4605e-04, 2.6202e-04, 2.4605e-04, 2.7835e-05, 2.7835e-05, 2.7013e-04,
          4.2796e-04, 1.9197e-03, 2.6202e-04, 2.7835e-05, 2.7013e-04, 1.6012e-03,
          8.8334e-05, 2.2583e-03, 2.0969e-04, 2.7835e-05, 2.7835e-05, 2.7835e-05,
          2.4605e-04, 4.2796e-04, 2.4605e-04, 4.1342e-04, 2.4605e-04, 8.8334e-05,
          2.7835e-05, 2.4605e-04, 2.7835e-05, 4.7398e-04, 3.2210e-04, 5.4026e-04,
          2.7835e-05, 8.8334e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4817e-03, 5.0079e-02, 1.9592e-02, 1.6632e-02, 9.2285e-02,
          3.2043e-02, 2.0798e-02, 8.4829e-04, 8.1301e-04, 1.9169e-03, 2.6894e-03,
          1.1044e-03, 9.8991e-04, 1.1349e-03, 1.7796e-03, 6.6042e-04, 1.4658e-03,
          7.2432e-04, 4.5133e-04, 1.8578e-03, 1.3361e-03, 7.8917e-04, 8.8930e-04,
          2.7504e-03, 2.9869e-03, 1.4496e-02, 3.4847e-03, 1.7624e-03, 1.1768e-03,
          1.0576e-03, 1.3313e-03, 7.7391e-04, 2.4605e-03, 8.5402e-04, 1.5106e-03,
          5.7297e-03, 6.2637e-03, 1.9627e-03, 1.5497e-03, 1.7023e-03, 2.8763e-03,
          2.2278e-03, 1.7853e-03, 1.9817e-03, 1.5020e-03, 8.8263e-04, 2.7218e-03,
          6.1798e-04, 1.1387e-03, 1.0300e-02, 2.7885e-03, 1.5974e-03, 1.0681e-03,
          1.6584e-03, 1.4324e-03, 4.3755e-03, 1.0178e-02, 4.9782e-03, 1.3069e-02,
          1.0193e-02, 3.4275e-03, 2.9583e-03, 2.3041e-03, 1.7776e-03, 3.6983e-03,
          1.4496e-03, 5.9891e-04, 9.2840e-04, 1.8282e-03, 1.9331e-03, 1.9760e-03,
          2.4395e-03, 2.8400e-03, 3.3169e-03, 2.0027e-03, 1.5497e-03, 1.1864e-03,
          2.7046e-03, 9.6817e-03, 3.4103e-03, 4.7684e-03, 3.7479e-03, 9.8648e-03,
          1.4061e-02, 2.6474e-03, 2.4490e-03, 1.0242e-03, 6.4278e-04, 1.0872e-03,
          1.8454e-03, 2.4014e-03, 1.4324e-03, 1.5526e-03, 4.3869e-03, 4.6272e-03,
          2.3212e-03, 2.4738e-03, 1.4362e-03, 5.2261e-03, 1.1505e-02, 7.3853e-03,
          3.0270e-03, 1.2922e-03, 1.4153e-03, 1.2047e-02, 4.9820e-03, 4.6997e-03,
          2.7847e-03, 3.3588e-03, 2.9221e-03, 1.6870e-03, 1.4687e-03, 1.0424e-03,
          1.8034e-03, 3.9520e-03, 1.6050e-03, 4.2801e-03, 5.6610e-03, 5.2643e-03,
          3.5534e-03, 5.6458e-03, 3.9368e-03, 7.9632e-04, 3.2406e-03, 2.0714e-03,
          1.0948e-02, 5.5618e-03, 5.2528e-03, 6.6147e-03, 1.3237e-03, 3.1948e-03,
          2.0409e-03, 8.8215e-04, 4.0460e-04, 6.8474e-04, 1.7004e-03, 3.1643e-03,
          2.1076e-03, 1.5144e-03, 1.0757e-03, 2.6970e-03, 2.2507e-03, 2.9202e-03,
          4.3526e-03, 1.0614e-03, 2.8706e-03, 5.6267e-03, 2.4700e-03, 9.5901e-03,
          1.2131e-02, 1.1505e-02, 4.1138e-02, 5.4131e-03, 6.7902e-03, 6.0539e-03,
          2.1324e-03, 3.5858e-03, 4.9639e-04, 3.0231e-03, 5.8250e-03, 1.9054e-03,
          3.3207e-03, 2.3899e-03, 1.4610e-03, 2.0580e-03, 1.1497e-02, 4.0245e-03,
          2.1534e-03, 3.2692e-03, 3.0727e-03, 1.8120e-03, 1.8902e-03, 7.6027e-03,
          2.2766e-02, 2.4124e-02, 1.0065e-01, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.9491e-04, 3.5286e-05, 3.5286e-05,
          1.0645e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5095e-04, 3.5286e-05, 3.5286e-05, 1.9491e-04, 1.0645e-04, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.0645e-04, 1.0645e-04, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 1.0645e-04, 1.9491e-04,
          1.9491e-04, 3.5286e-05, 3.5286e-05, 1.9491e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 1.2374e-04, 3.5286e-05, 1.0645e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04, 3.5286e-05,
          3.5286e-05, 1.0645e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04, 1.2374e-04, 3.5286e-05,
          1.9491e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04, 3.5286e-05, 3.5286e-05,
          1.2374e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          1.2374e-04, 3.5286e-05, 1.9491e-04, 1.9491e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 1.0645e-04, 3.5095e-04, 1.0645e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.0645e-04, 3.5286e-05, 3.5286e-05,
          1.0645e-04, 3.5286e-05, 3.5286e-05, 1.2374e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 2.6250e-04, 1.0645e-04, 1.9491e-04, 3.5286e-05, 3.5286e-05,
          3.5167e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04,
          3.5286e-05, 1.0645e-04, 1.2374e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 1.0645e-04, 3.5286e-05, 3.5286e-05, 1.2374e-04,
          3.5286e-05, 3.5286e-05, 1.9491e-04, 2.6321e-04, 3.5286e-05, 1.0645e-04,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5286e-05, 3.5095e-04, 1.9491e-04,
          3.5286e-05, 1.0645e-04, 3.5286e-05, 2.6321e-04, 1.9491e-04, 1.0645e-04,
          5.0783e-04, 5.0783e-04, 1.0645e-04, 3.5286e-05, 4.1938e-04, 3.5286e-05,
          3.5167e-04, 1.0645e-04, 3.5286e-05, 1.0645e-04, 3.5286e-05, 3.5286e-05,
          3.5286e-05, 3.5286e-05, 3.5286e-05, 1.2374e-04, 3.5095e-04, 3.5286e-05,
          2.6250e-04, 1.2374e-04, 3.5095e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          4.1938e-04, 3.5286e-05, 1.2374e-04, 3.5286e-05, 1.0645e-04, 2.6321e-04,
          3.5286e-05, 4.1938e-04, 1.9217e-04, 3.5286e-05, 3.5286e-05, 3.5286e-05,
          2.6250e-04, 1.0645e-04, 2.6250e-04, 3.1114e-04, 2.2936e-04, 3.5286e-05,
          3.5286e-05, 1.0645e-04, 3.5286e-05, 3.0041e-04, 2.2936e-04, 4.5657e-04,
          3.5286e-05, 3.5286e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0000e-03, 5.3253e-02, 1.9913e-02, 2.2690e-02, 9.0210e-02,
          3.7659e-02, 2.2232e-02, 7.8440e-04, 7.5960e-04, 2.7790e-03, 3.6621e-03,
          1.5736e-03, 9.0933e-04, 1.2903e-03, 1.5173e-03, 8.1062e-04, 1.6766e-03,
          1.0757e-03, 5.7888e-04, 2.1877e-03, 2.6455e-03, 1.9350e-03, 1.3571e-03,
          4.5700e-03, 2.9335e-03, 1.6632e-02, 4.8904e-03, 2.3766e-03, 1.5507e-03,
          1.3494e-03, 1.1292e-03, 1.0309e-03, 2.2583e-03, 1.2932e-03, 1.0729e-03,
          3.9482e-03, 3.5038e-03, 4.3297e-03, 1.9016e-03, 2.0466e-03, 1.1665e-02,
          1.6998e-02, 7.1564e-03, 7.7133e-03, 3.7746e-03, 1.1969e-03, 3.6945e-03,
          1.0319e-03, 1.9798e-03, 1.3664e-02, 2.0523e-03, 1.1883e-03, 8.8930e-04,
          1.1358e-03, 9.6798e-04, 1.9341e-03, 4.7684e-03, 2.8534e-03, 3.9902e-03,
          3.5038e-03, 3.8223e-03, 5.5847e-03, 5.4321e-03, 1.0544e-02, 6.0806e-03,
          4.6997e-03, 1.9951e-03, 2.3460e-03, 6.6147e-03, 7.3051e-03, 4.4441e-03,
          4.4746e-03, 4.3564e-03, 2.5063e-03, 1.3828e-03, 9.8038e-04, 7.1430e-04,
          1.4601e-03, 5.6267e-03, 3.5934e-03, 2.7122e-03, 2.9564e-03, 2.3289e-03,
          3.3970e-03, 4.4060e-03, 3.9024e-03, 2.3479e-03, 1.9608e-03, 3.3627e-03,
          4.9210e-03, 5.2528e-03, 3.6278e-03, 3.6583e-03, 4.6501e-03, 4.8332e-03,
          2.0618e-03, 1.5793e-03, 1.0395e-03, 2.9564e-03, 5.8708e-03, 3.1815e-03,
          3.1376e-03, 1.3981e-03, 1.2932e-03, 3.4046e-03, 2.4872e-03, 4.8447e-03,
          4.2877e-03, 2.7649e-02, 7.5912e-03, 6.4545e-03, 2.7199e-03, 2.4033e-03,
          2.1954e-03, 1.3222e-02, 3.8319e-03, 4.6692e-03, 7.0801e-03, 5.2147e-03,
          3.1891e-03, 3.4447e-03, 2.8381e-03, 5.4407e-04, 2.2335e-03, 1.4277e-03,
          3.9330e-03, 2.5997e-03, 2.4223e-03, 3.1242e-03, 1.2598e-03, 4.6463e-03,
          3.3398e-03, 2.2240e-03, 6.4087e-04, 2.2831e-03, 4.9744e-03, 1.1002e-02,
          3.0174e-03, 3.2043e-03, 1.5421e-03, 3.0766e-03, 2.4033e-03, 2.3918e-03,
          3.2024e-03, 8.7738e-04, 1.9703e-03, 2.2945e-03, 2.5520e-03, 3.1357e-03,
          5.1842e-03, 3.4447e-03, 4.4479e-03, 9.0790e-03, 8.4686e-03, 2.6993e-02,
          4.6806e-03, 8.3847e-03, 9.9087e-04, 9.3613e-03, 1.0231e-02, 3.9101e-03,
          3.9215e-03, 3.7041e-03, 1.7958e-03, 2.4738e-03, 6.5193e-03, 2.2717e-03,
          2.2850e-03, 1.9503e-03, 2.0008e-03, 1.1749e-03, 1.6947e-03, 3.6964e-03,
          3.4904e-03, 4.1618e-03, 9.6207e-03, 6.8970e-02, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 1.3351e-04, 1.3351e-04, 1.9765e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04,
          1.9765e-04, 2.1696e-05, 2.1696e-05, 3.0947e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 3.0947e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04, 1.3351e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.9765e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 3.0947e-04, 3.0947e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05, 1.3351e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.9765e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04,
          2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 1.9765e-04, 2.1696e-05, 2.1696e-05, 1.3351e-04,
          2.1696e-05, 1.3351e-04, 3.0947e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 1.3351e-04, 2.1696e-05, 2.1696e-05, 3.0947e-04, 1.3351e-04,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.3351e-04, 2.1696e-05,
          3.0947e-04, 3.0947e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.3351e-04, 2.1696e-05, 2.1696e-05, 1.9765e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.9765e-04, 2.1696e-05,
          5.3406e-04, 1.3351e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.9765e-04, 2.1696e-05, 1.3351e-04, 2.1696e-05, 2.1696e-05, 1.9765e-04,
          2.1696e-05, 5.3406e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 1.9765e-04, 2.1696e-05, 2.4915e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.4915e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1918e-03, 6.0638e-02, 1.5114e-02, 1.5045e-02, 4.1748e-02,
          4.9835e-02, 1.1414e-02, 1.2836e-03, 1.2636e-03, 6.0005e-03, 6.1684e-03,
          2.9526e-03, 1.4591e-03, 7.7820e-04, 4.4289e-03, 1.4963e-03, 1.7204e-03,
          1.1034e-03, 1.0443e-03, 5.1785e-04, 8.5974e-04, 6.0844e-04, 1.7424e-03,
          8.1863e-03, 6.6261e-03, 9.2316e-03, 4.9782e-03, 7.9060e-04, 2.0771e-03,
          9.6655e-04, 2.3155e-03, 7.9584e-04, 6.0234e-03, 4.1161e-03, 1.8854e-03,
          8.6060e-03, 1.8482e-03, 1.0605e-02, 5.0812e-03, 6.6042e-04, 1.2146e-02,
          2.8591e-03, 2.4204e-03, 1.8549e-03, 2.2011e-03, 1.9855e-03, 8.8348e-03,
          5.2977e-04, 6.6710e-04, 4.2305e-03, 2.7580e-03, 1.7624e-03, 1.6069e-03,
          6.5422e-04, 1.0681e-03, 8.3780e-04, 3.3302e-03, 1.9302e-03, 3.5400e-03,
          2.2068e-03, 1.8053e-03, 2.5101e-03, 4.4479e-03, 5.7640e-03, 2.1637e-02,
          2.8343e-03, 2.3422e-03, 4.3678e-03, 1.4362e-03, 1.2560e-03, 1.1234e-03,
          1.0948e-03, 1.1902e-02, 1.0948e-03, 2.1801e-03, 5.9223e-04, 6.5613e-04,
          1.1091e-03, 3.5686e-03, 2.4033e-03, 9.8877e-03, 2.5215e-03, 2.0008e-03,
          1.4391e-03, 5.6496e-03, 1.7929e-03, 5.5542e-03, 2.2068e-03, 1.9608e-03,
          9.6283e-03, 1.0170e-02, 1.4944e-03, 5.3368e-03, 1.6413e-03, 1.1883e-03,
          8.4543e-04, 6.3896e-04, 2.5139e-03, 4.8256e-03, 3.1471e-03, 4.7188e-03,
          7.2136e-03, 1.4591e-03, 1.9274e-03, 9.9411e-03, 1.0162e-02, 2.7885e-03,
          4.6005e-03, 5.4588e-03, 1.5259e-02, 1.8711e-03, 2.0580e-03, 3.5000e-03,
          4.2610e-03, 6.4964e-03, 1.4086e-03, 1.4526e-02, 1.7185e-03, 1.7319e-03,
          2.4700e-03, 1.3840e-02, 1.2846e-03, 9.3412e-04, 1.4053e-02, 1.1835e-03,
          2.2888e-03, 1.4076e-03, 1.9512e-03, 8.7662e-03, 2.7485e-03, 4.7417e-03,
          3.2520e-03, 2.5215e-03, 1.6050e-03, 1.9932e-03, 8.2321e-03, 2.4433e-03,
          7.7171e-03, 2.3994e-03, 1.6575e-03, 3.4504e-03, 4.2572e-03, 2.4624e-03,
          2.1534e-03, 1.7567e-03, 3.0422e-03, 3.3398e-03, 1.9512e-03, 9.5367e-03,
          1.4488e-02, 8.7051e-03, 3.2196e-03, 7.2632e-03, 3.2196e-03, 2.0691e-02,
          1.2291e-02, 1.0162e-02, 1.6699e-03, 3.2101e-03, 1.7578e-02, 3.9635e-03,
          1.0880e-02, 1.0452e-02, 3.3207e-03, 5.2681e-03, 6.3400e-03, 3.7327e-03,
          5.5923e-03, 4.3564e-03, 7.0839e-03, 3.3112e-03, 3.0384e-03, 1.0185e-02,
          4.4098e-03, 2.3327e-03, 3.8090e-03, 1.0124e-02, 4.6173e-02, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05, 1.6928e-05,
          1.5414e-04, 1.6928e-05, 1.6928e-05, 1.1139e-03, 7.2360e-05, 1.6928e-05,
          1.6928e-05, 9.8705e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05, 1.6928e-05, 4.6325e-04,
          7.2360e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.5414e-04, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05, 1.6928e-05,
          7.2360e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04,
          1.6928e-05, 1.6928e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 7.2360e-05, 1.5414e-04,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 7.2360e-05, 1.5414e-04, 1.5414e-04, 1.6928e-05,
          7.2360e-05, 7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 9.8705e-05, 1.5414e-04, 1.6928e-05, 7.2360e-05,
          7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04, 1.6928e-05,
          7.2360e-05, 1.6928e-05, 1.5414e-04, 1.5414e-04, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 7.2360e-05, 7.2360e-05, 1.5414e-04, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 9.8705e-05, 1.6928e-05, 9.8705e-05, 1.6928e-05, 7.2360e-05,
          1.5414e-04, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 9.8705e-05, 1.5414e-04, 1.6928e-05, 1.6928e-05,
          1.5414e-04, 1.5414e-04, 1.5414e-04, 1.6928e-05, 1.6928e-05, 1.5414e-04,
          1.6928e-05, 1.5414e-04, 1.5414e-04, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          7.2360e-05, 1.6928e-05, 1.5414e-04, 9.8705e-05, 1.6928e-05, 1.5414e-04,
          1.6928e-05, 1.6928e-05, 1.5414e-04, 7.2360e-05, 6.4754e-04, 9.8705e-05,
          9.1934e-04, 7.2360e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04, 1.5414e-04,
          7.0286e-04, 9.8705e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04, 1.6928e-05,
          9.1934e-04, 1.5414e-04, 1.6928e-05, 7.2360e-05, 7.2360e-05, 1.6928e-05,
          7.2360e-05, 1.5414e-04, 7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05,
          7.2360e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 1.5414e-04, 1.6928e-05,
          9.1934e-04, 1.5414e-04, 7.2360e-05, 9.8705e-05, 9.8705e-05, 9.8705e-05,
          1.5414e-04, 1.6928e-05, 1.5414e-04, 1.6928e-05, 1.6928e-05, 9.8705e-05,
          9.1934e-04, 1.1730e-03, 1.6928e-05, 9.8705e-05, 1.6928e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 1.6928e-05, 9.8705e-05, 1.6928e-05,
          1.6928e-05, 1.6928e-05, 1.6928e-05, 4.1819e-04, 1.6928e-05, 9.8705e-05,
          1.6928e-05, 1.6928e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4092e-03, 6.4758e-02, 1.7197e-02, 1.8799e-02, 4.7974e-02,
          6.0425e-02, 1.1986e-02, 9.8877e-03, 6.3591e-03, 6.8512e-03, 1.1063e-02,
          3.5801e-03, 7.7896e-03, 1.5945e-03, 5.8479e-03, 1.3405e-02, 6.2256e-03,
          5.1765e-03, 6.1607e-03, 4.4274e-04, 6.2180e-04, 8.3590e-04, 8.8577e-03,
          1.2848e-02, 3.5095e-03, 1.0162e-02, 7.8735e-03, 7.9966e-04, 6.7177e-03,
          2.0370e-03, 1.5060e-02, 1.6870e-03, 5.2795e-03, 8.4915e-03, 2.5597e-03,
          2.4033e-03, 7.7009e-04, 5.0201e-03, 5.5237e-03, 6.4468e-04, 3.1853e-03,
          1.1196e-03, 9.1600e-04, 9.6178e-04, 8.3828e-04, 1.0521e-02, 5.4970e-03,
          1.0023e-03, 8.9645e-04, 5.9280e-03, 3.1662e-03, 6.2866e-03, 3.5419e-03,
          1.1177e-03, 2.7695e-03, 7.0238e-04, 7.6008e-04, 2.1152e-03, 1.1854e-03,
          1.0376e-03, 2.2488e-03, 2.0142e-03, 3.2883e-03, 1.5078e-03, 8.0795e-03,
          1.9255e-03, 2.1515e-03, 2.8477e-03, 5.4407e-04, 5.5408e-04, 7.5531e-04,
          8.1444e-04, 1.2215e-02, 1.0672e-03, 6.2294e-03, 1.0443e-03, 1.2226e-03,
          2.4090e-03, 7.1812e-04, 1.0805e-03, 3.2825e-03, 1.5144e-03, 8.8024e-04,
          6.7472e-04, 8.9951e-03, 1.3151e-03, 8.6212e-03, 1.3304e-03, 1.1873e-03,
          1.3199e-02, 5.0392e-03, 6.8808e-04, 6.1302e-03, 7.3767e-04, 7.6056e-04,
          1.1806e-03, 5.8651e-04, 4.7150e-03, 2.8896e-03, 7.2575e-04, 1.0347e-03,
          8.4381e-03, 1.4963e-03, 1.6499e-03, 1.7233e-03, 5.7678e-03, 1.8330e-03,
          6.5842e-03, 1.2379e-03, 6.1073e-03, 9.4748e-04, 1.5612e-03, 1.9360e-03,
          3.3569e-03, 1.4000e-03, 1.4610e-03, 8.1482e-03, 1.0271e-03, 1.5821e-03,
          2.9507e-03, 3.2177e-03, 5.2595e-04, 1.1909e-02, 8.9645e-03, 5.2595e-04,
          5.4264e-04, 7.3290e-04, 1.0033e-03, 5.1765e-03, 1.1177e-02, 8.3542e-03,
          2.4090e-03, 2.0065e-03, 7.5684e-03, 1.4896e-03, 4.0970e-03, 7.4100e-04,
          1.3420e-02, 2.8210e-03, 9.3307e-03, 1.1513e-02, 8.4610e-03, 4.0627e-03,
          8.0776e-04, 2.8400e-03, 1.1091e-03, 9.0981e-04, 1.2102e-03, 3.9043e-03,
          5.6877e-03, 3.4313e-03, 1.0538e-03, 5.3406e-03, 3.6507e-03, 2.8419e-03,
          4.1351e-03, 1.9293e-03, 6.1302e-03, 1.2779e-03, 5.2567e-03, 4.3297e-03,
          6.4659e-03, 5.7144e-03, 1.6708e-02, 1.2848e-02, 1.6460e-03, 1.3161e-03,
          6.3591e-03, 1.6727e-03, 3.0518e-03, 4.8447e-03, 6.5384e-03, 3.7670e-03,
          2.5234e-03, 1.8969e-03, 1.5135e-03, 2.5311e-03, 6.2447e-03, 7.3792e-02,
          2.0862e-05, 2.0862e-05, 8.3506e-05, 1.4234e-04, 2.0862e-05, 2.0862e-05,
          8.3506e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05, 8.3506e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 1.4234e-04, 8.3506e-05, 2.0862e-05,
          2.0862e-05, 1.4234e-04, 2.0862e-05, 8.3506e-05, 8.3506e-05, 2.0862e-05,
          8.3506e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 1.4234e-04,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05, 2.0862e-05, 8.3506e-05,
          2.0862e-05, 1.4234e-04, 2.0862e-05, 2.0492e-04, 2.0862e-05, 2.0492e-04,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.3293e-04, 8.3506e-05, 2.0862e-05,
          1.9383e-04, 1.9383e-04, 2.0862e-05, 2.3293e-04, 2.0862e-05, 8.3506e-05,
          2.0862e-05, 2.0862e-05, 8.3506e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0492e-04, 1.4234e-04,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05, 2.0862e-05, 2.0862e-05,
          1.9383e-04, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.3293e-04, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0492e-04, 2.0492e-04, 2.0862e-05,
          1.9383e-04, 8.3506e-05, 8.3506e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          1.7023e-04, 2.3293e-04, 2.0862e-05, 2.0492e-04, 2.0862e-05, 8.3506e-05,
          8.3506e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 1.4234e-04, 2.0862e-05,
          7.0667e-04, 2.0862e-05, 1.4234e-04, 1.4234e-04, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 8.3506e-05,
          2.0862e-05, 1.4234e-04, 2.0862e-05, 1.4234e-04, 2.0862e-05, 8.3506e-05,
          1.4234e-04, 8.3506e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          2.3293e-04, 2.0862e-05, 1.4234e-04, 1.4234e-04, 2.0862e-05, 2.0862e-05,
          1.4234e-04, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0492e-04,
          2.0862e-05, 2.0492e-04, 2.0492e-04, 8.3506e-05, 2.0862e-05, 2.0862e-05,
          3.5429e-04, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0492e-04,
          2.0862e-05, 8.3506e-05, 1.4234e-04, 8.3506e-05, 1.7023e-04, 2.0862e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 1.4234e-04, 2.0862e-05,
          1.7023e-04, 1.4234e-04, 2.3293e-04, 2.0862e-05, 2.0492e-04, 3.5429e-04,
          1.4234e-04, 1.4234e-04, 1.7023e-04, 2.0862e-05, 1.4234e-04, 2.0862e-05,
          2.0492e-04, 1.4234e-04, 2.0862e-05, 3.5429e-04, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05, 1.4234e-04, 2.0862e-05,
          2.9182e-04, 2.0492e-04, 2.0492e-04, 2.0862e-05, 2.0862e-05, 1.4234e-04,
          3.5429e-04, 1.0023e-03, 2.0492e-04, 2.0862e-05, 2.3293e-04, 3.5429e-04,
          1.1244e-03, 1.1244e-03, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          1.4234e-04, 1.4234e-04, 2.0862e-05, 2.0862e-05, 1.4234e-04, 2.0862e-05,
          2.0862e-05, 1.4234e-04, 2.0862e-05, 2.0862e-05, 2.0862e-05, 2.0862e-05,
          2.0862e-05, 2.0862e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0812e-03, 6.3416e-02, 1.5617e-02, 1.8433e-02, 3.9398e-02,
          4.6509e-02, 1.2169e-02, 1.5030e-03, 9.9945e-04, 4.3564e-03, 6.3515e-03,
          2.3117e-03, 1.6613e-03, 8.7881e-04, 5.1956e-03, 1.5650e-03, 1.6594e-03,
          1.1778e-03, 1.0929e-03, 4.0817e-04, 6.0749e-04, 5.6887e-04, 2.1515e-03,
          8.1635e-03, 6.2637e-03, 9.5596e-03, 7.2441e-03, 6.0177e-04, 1.9836e-03,
          1.3533e-03, 2.7466e-03, 7.0381e-04, 5.7869e-03, 3.6259e-03, 1.9217e-03,
          5.5275e-03, 1.0433e-03, 5.2757e-03, 4.2648e-03, 5.9938e-04, 5.4169e-03,
          1.9112e-03, 1.8425e-03, 1.4038e-03, 1.3199e-03, 2.4090e-03, 8.5373e-03,
          6.9904e-04, 7.3957e-04, 4.0779e-03, 1.8444e-03, 1.8520e-03, 2.1114e-03,
          5.2547e-04, 9.5987e-04, 6.0272e-04, 1.5507e-03, 2.0981e-03, 1.9646e-03,
          1.5802e-03, 1.5364e-03, 1.5965e-03, 2.5826e-03, 3.4580e-03, 2.6215e-02,
          3.2978e-03, 2.7218e-03, 4.5280e-03, 1.1349e-03, 9.8419e-04, 9.9277e-04,
          9.0933e-04, 1.1864e-02, 7.1955e-04, 2.1667e-03, 5.7793e-04, 5.8937e-04,
          9.9277e-04, 1.5526e-03, 2.3594e-03, 7.6141e-03, 2.5845e-03, 1.5612e-03,
          1.0996e-03, 4.3793e-03, 1.2083e-03, 5.4131e-03, 2.3766e-03, 2.4719e-03,
          9.8724e-03, 7.1526e-03, 1.4324e-03, 5.9967e-03, 1.1959e-03, 1.0357e-03,
          8.2588e-04, 4.7088e-04, 3.7575e-03, 3.0956e-03, 1.5459e-03, 2.1210e-03,
          8.0643e-03, 1.9093e-03, 2.0027e-03, 8.1253e-03, 1.4130e-02, 1.9522e-03,
          3.5934e-03, 2.6264e-03, 9.4833e-03, 2.1954e-03, 3.2692e-03, 5.0774e-03,
          6.7368e-03, 3.3665e-03, 1.2569e-03, 1.8768e-02, 1.4820e-03, 1.1320e-03,
          1.8711e-03, 6.8283e-03, 9.6941e-04, 1.0805e-03, 3.1281e-02, 1.3800e-03,
          1.3161e-03, 1.5631e-03, 2.7084e-03, 1.1787e-02, 3.3817e-03, 3.9940e-03,
          4.7226e-03, 3.7632e-03, 2.9221e-03, 2.7409e-03, 6.8970e-03, 1.3933e-03,
          8.4839e-03, 2.4109e-03, 2.8687e-03, 5.5695e-03, 6.9580e-03, 2.4586e-03,
          1.6794e-03, 2.7542e-03, 4.1428e-03, 1.9341e-03, 2.4376e-03, 2.0935e-02,
          1.3260e-02, 1.0971e-02, 2.6798e-03, 4.3678e-03, 2.5768e-03, 9.0332e-03,
          1.0048e-02, 7.4806e-03, 2.7180e-03, 2.3651e-03, 1.7273e-02, 3.5915e-03,
          1.7944e-02, 1.1688e-02, 4.6654e-03, 9.5825e-03, 2.8172e-03, 3.4008e-03,
          6.9008e-03, 3.7060e-03, 1.4801e-02, 6.5346e-03, 3.3951e-03, 1.0986e-02,
          4.2572e-03, 2.4452e-03, 2.8172e-03, 4.9324e-03, 1.5991e-02, 6.2866e-03,
          2.8214e-02, 1.4186e-05, 1.4186e-05, 1.4448e-04, 7.1406e-05, 1.4186e-05,
          7.1406e-05, 7.1406e-05, 7.1406e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4448e-04, 1.4186e-05, 1.4186e-05, 5.8126e-04, 7.1406e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 8.7261e-05, 1.4186e-05, 5.8126e-04,
          1.4448e-04, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 2.3293e-04, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4448e-04,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 8.7261e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 7.1406e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 6.1846e-04,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4448e-04, 1.4448e-04, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 7.1406e-05, 1.4448e-04, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 7.1406e-05, 1.4186e-05,
          8.7261e-05, 1.4186e-05, 3.0589e-04, 5.8126e-04, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 7.1406e-05, 1.4448e-04, 7.1406e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 2.3293e-04, 1.4186e-05, 7.1406e-05, 1.4186e-05, 1.4186e-05,
          2.3293e-04, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          1.4186e-05, 7.1406e-05, 7.8011e-04, 2.3293e-04, 1.4186e-05, 1.4186e-05,
          2.3293e-04, 6.1846e-04, 2.3293e-04, 1.4186e-05, 1.4186e-05, 3.0589e-04,
          1.4186e-05, 6.1846e-04, 3.0684e-04, 1.4186e-05, 1.4186e-05, 7.1406e-05,
          7.1406e-05, 1.4186e-05, 1.4448e-04, 7.1406e-05, 1.4186e-05, 1.4448e-04,
          1.4186e-05, 1.4186e-05, 3.0589e-04, 1.4448e-04, 6.8426e-04, 7.8011e-04,
          6.1846e-04, 6.1846e-04, 1.4186e-05, 1.4186e-05, 1.4448e-04, 1.4448e-04,
          8.7261e-05, 7.1406e-05, 1.4186e-05, 1.4186e-05, 3.0589e-04, 1.4186e-05,
          6.9141e-04, 1.4448e-04, 1.4186e-05, 1.4186e-05, 1.4448e-04, 1.4186e-05,
          1.4186e-05, 2.3293e-04, 1.4186e-05, 7.1406e-05, 1.4186e-05, 1.4186e-05,
          6.1846e-04, 1.4186e-05, 1.4186e-05, 1.4186e-05, 1.4448e-04, 1.4186e-05,
          8.5306e-04, 1.4448e-04, 7.1406e-05, 2.3293e-04, 1.4186e-05, 2.3293e-04,
          1.4448e-04, 1.4186e-05, 3.0589e-04, 1.4186e-05, 1.4186e-05, 1.4186e-05,
          7.8011e-04, 8.5306e-04, 1.4186e-05, 2.3293e-04, 2.3293e-04, 1.4186e-05,
          7.1406e-05, 1.4186e-05, 7.1406e-05, 9.7513e-05, 7.1406e-05, 7.1406e-05,
          1.4186e-05, 7.1406e-05, 7.1406e-05, 4.8208e-04, 1.4186e-05, 7.1406e-05,
          1.4186e-05, 1.4186e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0392e-03, 5.4016e-02, 1.9180e-02, 1.6342e-02, 4.8431e-02,
          4.4830e-02, 1.3290e-02, 2.0065e-03, 1.3609e-03, 4.3831e-03, 6.2943e-03,
          2.3098e-03, 2.2411e-03, 7.7295e-04, 8.1253e-03, 2.7351e-03, 2.3289e-03,
          1.7242e-03, 1.6279e-03, 3.5501e-04, 5.3835e-04, 5.4741e-04, 2.8763e-03,
          8.4000e-03, 6.8665e-03, 1.0605e-02, 4.5586e-03, 5.9223e-04, 2.9259e-03,
          9.5940e-04, 4.0398e-03, 8.6451e-04, 8.6899e-03, 5.1804e-03, 2.5368e-03,
          5.3368e-03, 9.5320e-04, 6.0158e-03, 6.9199e-03, 4.7469e-04, 5.1918e-03,
          1.5678e-03, 1.2360e-03, 1.1501e-03, 1.1187e-03, 3.5572e-03, 9.8648e-03,
          5.8794e-04, 6.5136e-04, 4.5700e-03, 2.2717e-03, 2.1420e-03, 2.5902e-03,
          6.3992e-04, 1.2302e-03, 6.3801e-04, 1.1234e-03, 1.8511e-03, 1.5430e-03,
          1.4105e-03, 1.6088e-03, 2.2335e-03, 4.2152e-03, 2.8744e-03, 2.2034e-02,
          2.2202e-03, 1.7557e-03, 3.2520e-03, 7.6771e-04, 6.7282e-04, 8.7261e-04,
          8.9502e-04, 9.7198e-03, 7.8487e-04, 2.4567e-03, 5.2977e-04, 5.6171e-04,
          1.0185e-03, 1.0080e-03, 1.2445e-03, 5.0812e-03, 1.1959e-03, 1.5507e-03,
          1.0347e-03, 7.4425e-03, 1.5745e-03, 5.4283e-03, 1.4505e-03, 1.3275e-03,
          8.7357e-03, 7.1220e-03, 1.4095e-03, 5.0278e-03, 1.3542e-03, 9.8515e-04,
          7.8440e-04, 4.8923e-04, 3.4904e-03, 3.2024e-03, 9.5940e-04, 1.4725e-03,
          6.0501e-03, 1.0910e-03, 1.0424e-03, 6.5384e-03, 1.6663e-02, 2.3766e-03,
          6.3820e-03, 2.3308e-03, 9.2926e-03, 1.6232e-03, 1.8711e-03, 2.3174e-03,
          7.1068e-03, 3.5400e-03, 1.5535e-03, 1.9012e-02, 1.4353e-03, 1.5736e-03,
          2.4662e-03, 5.4474e-03, 6.5327e-04, 1.7881e-03, 2.1683e-02, 6.0511e-04,
          7.4434e-04, 7.1859e-04, 1.0748e-03, 1.5488e-02, 6.5727e-03, 6.8626e-03,
          3.2082e-03, 1.8177e-03, 3.8090e-03, 1.6251e-03, 5.7259e-03, 1.5526e-03,
          1.2726e-02, 2.7256e-03, 4.4823e-03, 6.1531e-03, 6.3171e-03, 2.9697e-03,
          1.0738e-03, 2.4357e-03, 1.9083e-03, 1.2960e-03, 1.3676e-03, 1.2451e-02,
          7.6180e-03, 9.3689e-03, 1.7691e-03, 5.9776e-03, 3.2425e-03, 7.6027e-03,
          7.9727e-03, 5.0659e-03, 2.9202e-03, 1.9188e-03, 1.6678e-02, 5.7373e-03,
          1.2810e-02, 9.1858e-03, 6.6643e-03, 1.0262e-02, 2.2068e-03, 1.9264e-03,
          4.4823e-03, 2.2774e-03, 8.7738e-03, 8.7509e-03, 4.2496e-03, 9.5596e-03,
          6.1340e-03, 2.9526e-03, 2.2373e-03, 3.8090e-03, 1.6983e-02, 1.0223e-02,
          7.2556e-03, 4.6356e-02, 2.3305e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05,
          1.4544e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.4544e-04, 2.3305e-05,
          1.4544e-04, 2.3305e-05, 2.3305e-05, 8.4686e-04, 1.4544e-04, 2.3305e-05,
          2.3305e-05, 8.4877e-05, 2.3305e-05, 8.4877e-05, 8.4877e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 8.3864e-05, 1.4544e-04, 1.4544e-04,
          8.3864e-05, 2.3305e-05, 2.3305e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.4544e-04,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          1.4544e-04, 8.4877e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 8.3864e-05,
          2.3305e-05, 2.3305e-05, 8.3864e-05, 2.3305e-05, 2.3305e-05, 2.8253e-04,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.4544e-04, 1.4544e-04,
          2.3305e-05, 8.4877e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          8.4877e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.5306e-03,
          2.3305e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 8.4877e-05, 1.4544e-04, 1.4544e-04, 2.3305e-05,
          1.4544e-04, 8.3864e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 8.4877e-05, 1.4544e-04, 2.3305e-05, 4.5013e-04,
          1.4544e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.8253e-04, 2.3305e-05, 1.4544e-04, 8.4686e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 1.4544e-04, 8.4877e-05, 2.3305e-05, 8.3864e-05,
          2.3305e-05, 8.4877e-05, 2.3305e-05, 8.4877e-05, 2.3305e-05, 8.3864e-05,
          8.4877e-05, 8.4877e-05, 8.4877e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 8.4877e-05, 8.4877e-05, 8.4877e-05, 2.3305e-05, 2.3305e-05,
          8.4877e-05, 8.4877e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05, 1.4544e-04,
          2.3305e-05, 8.4877e-05, 1.4544e-04, 2.3305e-05, 2.3305e-05, 4.4012e-04,
          8.4877e-05, 2.3305e-05, 8.3864e-05, 8.4877e-05, 8.4877e-05, 1.4544e-04,
          2.3305e-05, 2.3305e-05, 4.4894e-04, 1.4544e-04, 6.0272e-04, 7.4387e-04,
          1.2617e-03, 1.5907e-03, 2.3305e-05, 2.3305e-05, 3.4404e-04, 1.4544e-04,
          6.6328e-04, 8.4877e-05, 2.3305e-05, 2.3305e-05, 1.4544e-04, 2.3305e-05,
          5.0068e-04, 3.4404e-04, 2.3305e-05, 2.3305e-05, 1.4544e-04, 2.3305e-05,
          2.3305e-05, 5.0068e-04, 2.3305e-05, 8.4877e-05, 2.3305e-05, 2.3305e-05,
          8.0442e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 3.4404e-04, 3.8838e-04,
          2.0924e-03, 1.4544e-04, 1.4544e-04, 8.4877e-05, 4.4012e-04, 3.8838e-04,
          4.4894e-04, 7.4387e-04, 5.0068e-04, 2.3305e-05, 2.3305e-05, 8.4877e-05,
          7.4387e-04, 2.4033e-03, 2.3305e-05, 3.8838e-04, 4.4012e-04, 2.3305e-05,
          8.4877e-05, 8.4877e-05, 4.8232e-04, 2.3305e-05, 8.4877e-05, 8.3864e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 4.2987e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3291e-03, 5.6244e-02, 1.8372e-02, 2.0966e-02, 4.2603e-02,
          4.4250e-02, 1.4137e-02, 1.9016e-03, 1.1244e-03, 5.4398e-03, 6.7596e-03,
          2.4586e-03, 2.4109e-03, 8.3017e-04, 5.9471e-03, 2.2602e-03, 2.9984e-03,
          2.0657e-03, 1.2436e-03, 3.7622e-04, 6.4182e-04, 6.2275e-04, 3.5458e-03,
          8.6594e-03, 7.2441e-03, 1.1215e-02, 7.4959e-03, 6.2084e-04, 3.4676e-03,
          1.1826e-03, 3.6983e-03, 9.4557e-04, 8.6060e-03, 6.2408e-03, 1.8358e-03,
          6.5689e-03, 8.8501e-04, 6.7215e-03, 6.5880e-03, 4.8137e-04, 5.7144e-03,
          1.6193e-03, 1.3485e-03, 1.1444e-03, 1.1301e-03, 4.5128e-03, 1.0384e-02,
          6.6805e-04, 7.0000e-04, 4.6120e-03, 1.9321e-03, 2.2755e-03, 1.8339e-03,
          5.7268e-04, 1.1349e-03, 5.9223e-04, 1.1501e-03, 1.6575e-03, 1.5306e-03,
          1.2980e-03, 1.5574e-03, 1.8044e-03, 3.1757e-03, 2.6722e-03, 2.7969e-02,
          2.1286e-03, 1.7672e-03, 3.8700e-03, 7.8392e-04, 6.9523e-04, 8.0729e-04,
          8.7070e-04, 1.2848e-02, 7.1955e-04, 2.6398e-03, 5.4264e-04, 5.1641e-04,
          1.0281e-03, 8.9598e-04, 1.1015e-03, 5.7755e-03, 1.3657e-03, 1.2436e-03,
          9.1839e-04, 6.0806e-03, 1.1902e-03, 4.8218e-03, 1.4172e-03, 1.2512e-03,
          9.2621e-03, 7.3700e-03, 1.0433e-03, 5.3825e-03, 1.1406e-03, 1.0509e-03,
          8.2445e-04, 4.6396e-04, 3.0842e-03, 3.5458e-03, 9.4843e-04, 1.5526e-03,
          6.3095e-03, 9.1648e-04, 1.1587e-03, 4.8447e-03, 1.3817e-02, 1.7700e-03,
          4.5280e-03, 2.0180e-03, 9.7504e-03, 1.2312e-03, 1.6890e-03, 2.3956e-03,
          3.8052e-03, 3.0231e-03, 1.3027e-03, 2.2507e-02, 1.5316e-03, 1.2064e-03,
          2.0676e-03, 6.2752e-03, 5.8842e-04, 1.1702e-03, 3.1403e-02, 5.1022e-04,
          7.0286e-04, 7.3004e-04, 1.0281e-03, 1.3573e-02, 4.5395e-03, 4.8332e-03,
          2.8019e-03, 1.9627e-03, 2.6665e-03, 1.4467e-03, 6.1340e-03, 1.1702e-03,
          1.2009e-02, 2.6913e-03, 2.5291e-03, 5.3673e-03, 6.9199e-03, 2.5997e-03,
          9.3174e-04, 1.5965e-03, 1.7242e-03, 1.4124e-03, 1.2093e-03, 1.0490e-02,
          1.0452e-02, 1.0094e-02, 1.6851e-03, 4.9210e-03, 2.5940e-03, 7.2174e-03,
          8.0414e-03, 4.8637e-03, 2.2526e-03, 1.6632e-03, 1.5022e-02, 5.0888e-03,
          1.4587e-02, 9.9411e-03, 7.3433e-03, 8.4457e-03, 2.3136e-03, 2.2411e-03,
          4.6463e-03, 2.6379e-03, 8.9798e-03, 5.4855e-03, 4.2496e-03, 1.0262e-02,
          3.2482e-03, 1.8654e-03, 2.1286e-03, 4.1199e-03, 1.6098e-02, 7.9575e-03,
          8.3160e-03, 1.0521e-02, 4.8462e-02, 1.3888e-04, 1.6809e-05, 1.6809e-05,
          1.3888e-04, 2.6989e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.3888e-04, 1.6809e-05, 1.6809e-05, 4.1866e-04, 1.3888e-04, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05, 1.3888e-04, 4.1866e-04,
          6.6340e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.3888e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.3888e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          6.6340e-05, 1.3888e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05,
          1.6809e-05, 1.6809e-05, 6.6340e-05, 1.6809e-05, 1.6809e-05, 2.2542e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05, 1.3888e-04,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 2.0065e-03,
          1.6809e-05, 1.3888e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.3888e-04, 1.3888e-04, 1.6809e-05,
          6.6340e-05, 6.6340e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 6.6340e-05, 2.6989e-04, 1.3888e-04, 1.6809e-05, 6.6340e-05,
          6.6340e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          2.2542e-04, 1.6809e-05, 1.3888e-04, 4.1866e-04, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 1.3888e-04, 8.9407e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 8.9407e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 6.6340e-05,
          2.6989e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          1.6809e-05, 1.3888e-04, 5.4979e-04, 8.9407e-05, 1.6809e-05, 1.6809e-05,
          3.6931e-04, 3.1924e-04, 3.6931e-04, 1.6809e-05, 1.6809e-05, 1.3888e-04,
          1.6809e-05, 2.6989e-04, 1.3947e-04, 6.6340e-05, 1.6809e-05, 1.9717e-04,
          3.1924e-04, 1.6809e-05, 2.4676e-04, 2.6989e-04, 1.6809e-05, 1.3888e-04,
          1.6809e-05, 1.6809e-05, 4.1866e-04, 1.3888e-04, 5.5885e-04, 5.4979e-04,
          5.9891e-04, 3.1924e-04, 1.6809e-05, 1.6809e-05, 2.9802e-04, 1.3888e-04,
          5.5885e-04, 8.9407e-05, 1.6809e-05, 1.6809e-05, 1.3888e-04, 1.6809e-05,
          4.7851e-04, 4.7851e-04, 6.8951e-04, 6.6340e-05, 1.3888e-04, 1.6809e-05,
          1.6809e-05, 3.1924e-04, 1.6809e-05, 6.3133e-04, 1.6809e-05, 1.6809e-05,
          2.6989e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05, 4.7851e-04, 1.9717e-04,
          7.5817e-04, 1.3888e-04, 1.3888e-04, 8.9407e-05, 1.9717e-04, 3.6931e-04,
          4.1866e-04, 1.9717e-04, 3.1924e-04, 1.6809e-05, 1.6809e-05, 1.6809e-05,
          5.4979e-04, 9.5987e-04, 1.6809e-05, 8.9407e-05, 8.9407e-05, 1.6809e-05,
          1.6809e-05, 1.6809e-05, 8.9407e-05, 8.9407e-05, 8.9407e-05, 6.6340e-05,
          1.6809e-05, 1.3888e-04, 8.9407e-05, 4.0793e-04, 1.6809e-05, 8.9407e-05,
          1.6809e-05, 1.6809e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9101e-03, 6.2683e-02, 1.5976e-02, 1.6022e-02, 4.2908e-02,
          4.6783e-02, 1.0979e-02, 1.8433e-02, 5.0774e-03, 3.2692e-03, 5.4359e-03,
          1.9760e-03, 3.1910e-03, 1.1864e-03, 2.7905e-03, 1.9501e-02, 3.8567e-03,
          2.9202e-03, 5.3101e-03, 3.6144e-04, 4.3964e-04, 5.4026e-04, 3.4256e-03,
          7.4196e-03, 2.8286e-03, 9.5291e-03, 6.9695e-03, 5.9748e-04, 3.9062e-03,
          1.5965e-03, 3.2196e-02, 1.1101e-03, 3.2120e-03, 2.6321e-03, 8.5449e-03,
          1.8177e-03, 7.5245e-04, 2.4624e-03, 3.0746e-03, 4.8923e-04, 1.6956e-03,
          9.0122e-04, 7.1144e-04, 7.9155e-04, 5.6648e-04, 4.3755e-03, 2.8934e-03,
          6.7472e-04, 5.5408e-04, 4.3221e-03, 7.9193e-03, 1.9760e-02, 6.8741e-03,
          2.1591e-03, 9.3002e-03, 1.7195e-03, 7.1716e-04, 2.4891e-03, 9.6893e-04,
          1.0824e-03, 1.6088e-03, 5.1155e-03, 4.2992e-03, 1.2741e-03, 4.2419e-03,
          2.0237e-03, 1.3723e-03, 1.3466e-03, 4.6849e-04, 4.3559e-04, 6.7902e-04,
          6.7759e-04, 7.0724e-03, 2.7847e-03, 2.2293e-02, 1.9932e-03, 2.8400e-03,
          5.6305e-03, 6.5136e-04, 9.5892e-04, 1.9426e-03, 1.1301e-03, 8.1301e-04,
          7.2956e-04, 3.5797e-02, 3.2730e-03, 3.7270e-03, 1.1759e-03, 1.0700e-03,
          6.4125e-03, 2.8248e-03, 6.8378e-04, 3.2597e-03, 7.2289e-04, 6.7616e-04,
          9.8991e-04, 1.1129e-03, 1.5869e-02, 2.6169e-03, 6.8426e-04, 9.4080e-04,
          5.0430e-03, 1.2302e-03, 1.0862e-03, 1.2102e-03, 3.4122e-03, 1.6127e-03,
          2.2736e-02, 1.1692e-03, 3.6697e-03, 1.1978e-03, 1.6804e-03, 1.4591e-03,
          2.2964e-03, 1.0853e-03, 1.0824e-03, 4.6043e-03, 9.7132e-04, 1.5917e-03,
          2.2526e-03, 2.4071e-03, 5.1975e-04, 1.4282e-02, 4.2877e-03, 5.0497e-04,
          4.7731e-04, 7.0763e-04, 1.0576e-03, 4.1389e-03, 1.7334e-02, 2.7802e-02,
          3.0689e-03, 1.3962e-03, 2.0294e-03, 1.0738e-03, 2.4395e-03, 6.9332e-04,
          6.5422e-03, 2.2888e-03, 6.6757e-03, 9.5596e-03, 4.9820e-03, 3.0155e-03,
          7.7295e-04, 1.5593e-03, 1.0014e-03, 9.0027e-04, 1.2569e-03, 3.8223e-03,
          4.4975e-03, 2.9774e-03, 1.1110e-03, 1.0033e-02, 8.8806e-03, 2.2430e-03,
          2.6970e-03, 1.9741e-03, 3.7365e-03, 1.2321e-03, 4.0245e-03, 3.5458e-03,
          3.6488e-03, 3.7060e-03, 7.8583e-03, 5.1918e-03, 1.3494e-03, 1.3771e-03,
          3.9749e-03, 1.5259e-03, 2.5349e-03, 2.9888e-03, 4.4899e-03, 2.7790e-03,
          4.4861e-03, 4.0321e-03, 1.5030e-03, 2.1992e-03, 3.6869e-03, 1.1398e-02,
          2.1229e-03, 4.4250e-03, 4.6234e-03, 6.8298e-02, 2.1219e-05, 2.1219e-05,
          1.6212e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05, 7.4565e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 7.1621e-04, 1.6212e-04, 7.4565e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.6212e-04, 1.0884e-04,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05, 7.4565e-05, 7.4565e-05,
          2.1219e-05, 1.0884e-04, 2.1219e-05, 1.6212e-04, 2.1219e-05, 1.6212e-04,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05, 1.6212e-04, 7.4565e-05,
          7.4565e-05, 7.4565e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 1.0884e-04, 1.6212e-04, 1.6212e-04,
          2.4414e-04, 2.1219e-05, 2.1219e-05, 1.0884e-04, 2.1219e-05, 2.1219e-05,
          7.4565e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.0719e-03,
          2.1219e-05, 7.4565e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 1.6212e-04, 1.6212e-04, 2.1219e-05,
          7.4565e-05, 7.4565e-05, 1.6212e-04, 2.1219e-05, 2.1219e-05, 7.4565e-05,
          2.1219e-05, 7.4565e-05, 2.1219e-05, 1.6212e-04, 2.1219e-05, 7.4565e-05,
          7.4565e-05, 2.9087e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          7.4565e-05, 2.1219e-05, 1.0884e-04, 7.1621e-04, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 1.0884e-04, 2.1219e-05, 7.4565e-05,
          2.1219e-05, 1.0884e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 1.6212e-04,
          1.0884e-04, 1.0884e-04, 3.3188e-04, 7.4565e-05, 1.0884e-04, 1.0884e-04,
          1.0884e-04, 1.6212e-04, 1.0884e-04, 1.0884e-04, 2.1219e-05, 2.1219e-05,
          5.0640e-04, 1.0884e-04, 1.0884e-04, 2.1219e-05, 2.1219e-05, 1.6212e-04,
          1.0884e-04, 1.6212e-04, 1.6212e-04, 7.4565e-05, 2.1219e-05, 2.1219e-05,
          1.0884e-04, 2.1219e-05, 2.1219e-05, 3.3188e-04, 2.4414e-04, 1.6212e-04,
          2.1219e-05, 7.4565e-05, 1.0884e-04, 7.4565e-05, 2.1219e-05, 3.3188e-04,
          2.4414e-04, 3.8505e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05,
          2.1219e-05, 2.1219e-05, 7.4565e-05, 2.1219e-05, 1.6212e-04, 1.0884e-04,
          2.1219e-05, 2.1219e-05, 2.1219e-05, 7.4565e-05, 2.1219e-05, 2.1219e-05,
          1.6212e-04, 3.3188e-04, 2.1219e-05, 8.1015e-04, 2.1219e-05, 2.1219e-05,
          9.3937e-04, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.1219e-05, 2.4414e-04,
          1.0884e-04, 1.2197e-03, 4.7946e-04, 1.0884e-04, 2.4414e-04, 1.3895e-03,
          2.1219e-05, 2.4414e-04, 1.4420e-03, 2.1219e-05, 3.8505e-04, 8.1015e-04,
          1.0719e-03, 2.1219e-05, 2.1219e-05, 2.4414e-04, 2.4414e-04, 2.1219e-05,
          1.6212e-04, 1.0884e-04, 5.0545e-04, 1.0884e-04, 1.0884e-04, 1.6212e-04,
          2.1219e-05, 1.6212e-04, 2.1219e-05, 4.0245e-04, 3.1471e-04, 4.0245e-04,
          1.0884e-04, 2.1219e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.8823e-03, 6.3599e-02, 2.1179e-02, 2.0432e-02, 4.8096e-02,
          5.1880e-02, 1.5244e-02, 1.2161e-02, 4.1122e-03, 3.7060e-03, 7.6904e-03,
          2.2869e-03, 6.2294e-03, 1.6603e-03, 3.2082e-03, 1.2115e-02, 6.9618e-03,
          5.1956e-03, 5.7411e-03, 5.3215e-04, 7.1049e-04, 1.1072e-03, 7.1716e-03,
          9.3994e-03, 3.0155e-03, 1.2688e-02, 6.7940e-03, 8.9264e-04, 6.9122e-03,
          2.4300e-03, 1.7426e-02, 2.1019e-03, 2.9316e-03, 3.8357e-03, 2.4624e-03,
          1.4763e-03, 6.3848e-04, 2.3384e-03, 2.6703e-03, 7.4673e-04, 1.8673e-03,
          9.3269e-04, 6.9761e-04, 8.1968e-04, 7.2432e-04, 9.4681e-03, 3.2692e-03,
          1.2321e-03, 1.0777e-03, 5.3215e-03, 2.8286e-03, 8.6365e-03, 3.5324e-03,
          1.1826e-03, 3.6869e-03, 7.9823e-04, 6.0606e-04, 1.5612e-03, 8.9025e-04,
          9.6750e-04, 2.1038e-03, 2.0733e-03, 2.1210e-03, 1.1072e-03, 4.8828e-03,
          1.9455e-03, 1.7691e-03, 1.7309e-03, 4.8685e-04, 4.8375e-04, 7.6628e-04,
          8.8024e-04, 7.3166e-03, 1.0214e-03, 9.1019e-03, 1.2693e-03, 1.3494e-03,
          2.2068e-03, 5.6696e-04, 9.3222e-04, 1.8473e-03, 1.1501e-03, 7.6771e-04,
          6.4230e-04, 1.1902e-02, 1.4057e-03, 5.3787e-03, 1.2159e-03, 1.1330e-03,
          8.4381e-03, 3.4580e-03, 7.4959e-04, 4.5891e-03, 7.8964e-04, 8.3160e-04,
          1.3905e-03, 5.8126e-04, 5.6267e-03, 3.0346e-03, 5.6124e-04, 8.0204e-04,
          5.9509e-03, 2.0103e-03, 1.2465e-03, 1.1225e-03, 4.0703e-03, 1.8282e-03,
          7.7286e-03, 8.7547e-04, 3.9558e-03, 8.9788e-04, 1.5574e-03, 1.5955e-03,
          3.3073e-03, 1.1358e-03, 1.6346e-03, 5.7831e-03, 1.1730e-03, 1.7729e-03,
          2.2259e-03, 2.1133e-03, 4.4703e-04, 2.0447e-02, 5.2223e-03, 4.5848e-04,
          4.2915e-04, 5.9080e-04, 8.2731e-04, 3.7975e-03, 1.0139e-02, 9.7046e-03,
          2.3956e-03, 1.8997e-03, 5.3177e-03, 1.3371e-03, 2.6588e-03, 7.0620e-04,
          1.1406e-02, 2.9125e-03, 1.9653e-02, 1.4725e-02, 5.6610e-03, 3.4256e-03,
          6.1607e-04, 4.5395e-03, 8.4448e-04, 7.5531e-04, 1.1034e-03, 3.2730e-03,
          4.3182e-03, 2.5940e-03, 8.9502e-04, 3.8452e-03, 3.4752e-03, 1.7281e-03,
          2.5749e-03, 1.4381e-03, 5.8098e-03, 1.1311e-03, 3.9787e-03, 4.6234e-03,
          4.9438e-03, 3.9673e-03, 1.5381e-02, 1.0872e-02, 1.2350e-03, 1.2636e-03,
          4.1313e-03, 1.4620e-03, 2.2621e-03, 4.4556e-03, 5.1765e-03, 2.6627e-03,
          2.4986e-03, 1.6394e-03, 1.3504e-03, 1.7138e-03, 3.1185e-03, 3.5461e-02,
          2.2926e-03, 4.5586e-03, 5.0926e-03, 3.2104e-02, 7.3059e-02, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.9574e-04, 3.4904e-04, 9.6738e-05,
          8.7619e-05, 1.5378e-05, 1.5378e-05, 8.7619e-05, 1.6892e-04, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 2.6774e-04, 8.7619e-05, 1.5378e-05,
          3.4904e-04, 8.7619e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.7619e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 9.6738e-05, 1.5378e-05, 1.6892e-04,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.6892e-04, 1.5378e-05, 1.6892e-04,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 3.4904e-04, 9.6738e-05, 1.5378e-05,
          3.4904e-04, 3.4904e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 9.6738e-05,
          1.5378e-05, 2.6774e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 9.6738e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 9.6738e-05, 1.5378e-05,
          1.5378e-05, 2.6774e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          3.4904e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 3.4904e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 9.6738e-05, 1.6892e-04, 1.5378e-05,
          3.4904e-04, 3.4904e-04, 9.6738e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.6892e-04, 1.5378e-05, 3.4904e-04,
          1.6892e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05,
          9.6738e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.9574e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05, 2.6774e-04, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05, 2.6774e-04,
          1.5378e-05, 8.7619e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05, 1.5378e-05, 1.6892e-04,
          1.5378e-05, 9.6738e-05, 1.6892e-04, 8.7619e-05, 1.5378e-05, 1.5378e-05,
          1.6892e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 2.6774e-04, 1.6892e-04,
          1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          2.6774e-04, 8.7619e-05, 1.5378e-05, 1.5378e-05, 2.6774e-04, 8.7619e-05,
          1.5378e-05, 1.5378e-05, 9.6738e-05, 1.5378e-05, 1.6892e-04, 8.7619e-05,
          2.6774e-04, 2.6774e-04, 2.6774e-04, 9.6738e-05, 8.7619e-05, 1.5378e-05,
          8.7619e-05, 8.7619e-05, 1.5378e-05, 8.5354e-04, 1.5378e-05, 1.5378e-05,
          2.6774e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 8.7619e-05, 1.5378e-05,
          2.6774e-04, 1.6892e-04, 1.6892e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          2.6774e-04, 2.6774e-04, 1.6892e-04, 1.5378e-05, 1.5378e-05, 8.4400e-04,
          8.7619e-05, 2.6774e-04, 1.5378e-05, 1.5378e-05, 1.5378e-05, 1.5378e-05,
          5.4789e-04, 8.7619e-05, 1.5378e-05, 8.7619e-05, 8.7619e-05, 1.5378e-05,
          1.5378e-05, 8.7619e-05, 6.7949e-04, 8.7619e-05, 8.7619e-05, 8.7619e-05,
          8.7619e-05, 8.7619e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0774e-03, 6.0883e-02, 2.0218e-02, 1.6769e-02, 3.9337e-02,
          4.5990e-02, 1.4000e-02, 2.5101e-03, 1.4896e-03, 4.5586e-03, 8.3466e-03,
          2.3746e-03, 8.5068e-03, 1.4362e-03, 1.1353e-02, 2.7905e-03, 7.9422e-03,
          3.9177e-03, 2.1610e-03, 4.4060e-04, 6.7139e-04, 7.8201e-04, 6.1417e-03,
          9.0332e-03, 5.2261e-03, 1.1185e-02, 5.0201e-03, 7.3671e-04, 7.4310e-03,
          2.9449e-03, 4.7569e-03, 1.6356e-03, 6.7558e-03, 5.2223e-03, 2.1267e-03,
          2.6951e-03, 6.2990e-04, 3.4828e-03, 4.5547e-03, 6.4945e-04, 2.0332e-03,
          8.3065e-04, 7.5531e-04, 7.6628e-04, 7.5436e-04, 1.2428e-02, 8.1024e-03,
          1.0967e-03, 1.0777e-03, 4.3907e-03, 1.9073e-03, 2.1248e-03, 2.5101e-03,
          4.9877e-04, 1.1415e-03, 6.8426e-04, 6.5517e-04, 1.1158e-03, 8.4877e-04,
          9.6178e-04, 1.9331e-03, 1.6451e-03, 2.3460e-03, 1.0643e-03, 1.1360e-02,
          1.4429e-03, 1.4524e-03, 2.2240e-03, 5.3787e-04, 5.2500e-04, 8.0156e-04,
          9.4557e-04, 1.1078e-02, 6.7854e-04, 2.8172e-03, 5.0926e-04, 5.1355e-04,
          1.0214e-03, 5.6982e-04, 7.7391e-04, 2.5482e-03, 7.6628e-04, 7.5197e-04,
          6.9427e-04, 4.3716e-03, 9.2125e-04, 6.0768e-03, 9.6130e-04, 8.9550e-04,
          7.5798e-03, 4.1046e-03, 8.2207e-04, 4.4250e-03, 9.1410e-04, 1.1969e-03,
          1.0929e-03, 4.4966e-04, 3.2673e-03, 3.8776e-03, 5.6171e-04, 8.3923e-04,
          6.1569e-03, 1.5688e-03, 8.2350e-04, 1.5240e-03, 9.6283e-03, 1.8330e-03,
          3.8300e-03, 7.8917e-04, 3.9406e-03, 9.8705e-04, 1.3828e-03, 1.6422e-03,
          6.0577e-03, 1.4906e-03, 1.8492e-03, 1.4687e-02, 1.9007e-03, 1.4496e-03,
          2.3499e-03, 3.0880e-03, 4.5562e-04, 2.0885e-03, 1.3565e-02, 4.3964e-04,
          5.0068e-04, 5.1594e-04, 7.6962e-04, 8.1940e-03, 7.3166e-03, 4.8370e-03,
          2.4548e-03, 1.2417e-03, 6.9160e-03, 1.1320e-03, 2.5940e-03, 7.6342e-04,
          2.5665e-02, 3.1261e-03, 2.2400e-02, 1.3519e-02, 1.0078e-02, 3.8166e-03,
          6.8521e-04, 5.5428e-03, 1.1950e-03, 8.0156e-04, 1.0128e-03, 5.7411e-03,
          5.1804e-03, 4.4136e-03, 9.6846e-04, 2.8496e-03, 2.5501e-03, 1.9083e-03,
          3.1185e-03, 1.5659e-03, 4.6616e-03, 8.9407e-04, 7.2098e-03, 5.4436e-03,
          1.3199e-02, 7.7133e-03, 2.0126e-02, 2.6260e-02, 1.2779e-03, 1.2217e-03,
          4.2000e-03, 1.6327e-03, 4.0665e-03, 1.6556e-02, 6.8817e-03, 6.7673e-03,
          2.5063e-03, 1.3924e-03, 1.5488e-03, 1.3161e-03, 4.6577e-03, 1.3847e-02,
          4.7264e-03, 7.1716e-03, 1.1856e-02, 9.5139e-03, 1.1024e-02, 6.5308e-02,
          6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 6.8843e-05, 1.6272e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05,
          1.6272e-05, 6.8128e-05, 1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 6.8843e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 6.8128e-05, 1.6272e-05, 6.8128e-05, 1.6272e-05, 6.8128e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05,
          1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05, 3.0255e-04, 1.6272e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 6.8128e-05, 3.0255e-04, 1.6272e-05, 6.8843e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 1.6272e-05,
          2.5058e-04, 1.6272e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05,
          6.8128e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 6.8128e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05,
          1.6272e-05, 6.8128e-05, 5.9080e-04, 1.6272e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05, 5.9080e-04,
          6.8128e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05, 6.8128e-05, 6.8128e-05,
          1.2064e-03, 6.8128e-05, 1.6272e-05, 1.6272e-05, 3.5644e-04, 6.8128e-05,
          1.6272e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 5.9080e-04, 6.8128e-05,
          3.5644e-04, 3.5644e-04, 6.8128e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05,
          6.8128e-05, 3.5644e-04, 1.6272e-05, 6.8128e-05, 1.6272e-05, 1.6272e-05,
          6.8128e-05, 1.6272e-05, 1.6272e-05, 1.6272e-05, 6.8128e-05, 6.8128e-05,
          1.2064e-03, 5.9080e-04, 6.8128e-05, 6.8128e-05, 6.8128e-05, 3.5644e-04,
          6.8128e-05, 6.8128e-05, 1.4410e-03, 6.8128e-05, 6.8128e-05, 6.8128e-05,
          1.2064e-03, 3.6736e-03, 1.6272e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05,
          6.8128e-05, 6.8128e-05, 6.8128e-05, 1.6272e-05, 6.8128e-05, 1.6272e-05,
          1.6272e-05, 6.8128e-05, 6.8128e-05, 4.2677e-04, 6.8128e-05, 3.5644e-04,
          6.8128e-05, 6.8128e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.4697e-03, 5.4108e-02, 1.7349e-02, 2.4063e-02, 4.7546e-02,
          4.7241e-02, 1.6129e-02, 4.5357e-03, 2.0256e-03, 1.0025e-02, 2.4704e-02,
          7.0457e-03, 3.4180e-03, 1.1921e-03, 3.9043e-03, 4.7226e-03, 3.2520e-03,
          2.5616e-03, 2.6817e-03, 3.0661e-04, 4.7398e-04, 5.5122e-04, 4.1466e-03,
          2.9526e-02, 3.8948e-03, 1.2947e-02, 6.7215e-03, 6.1226e-04, 3.6449e-03,
          1.6546e-03, 7.3357e-03, 1.1225e-03, 4.0665e-03, 7.6752e-03, 1.9283e-03,
          2.2717e-03, 9.3985e-04, 3.5477e-03, 4.0245e-03, 4.3011e-04, 3.1052e-03,
          1.3046e-03, 9.3842e-04, 1.0986e-03, 8.6117e-04, 5.1117e-03, 3.8147e-03,
          7.0190e-04, 6.3038e-04, 5.1727e-03, 2.0733e-03, 3.7842e-03, 2.2545e-03,
          7.6628e-04, 1.6737e-03, 5.4884e-04, 1.1501e-03, 3.2749e-03, 1.9703e-03,
          1.4668e-03, 1.2321e-03, 1.1721e-03, 1.6069e-03, 2.0409e-03, 6.7940e-03,
          3.6392e-03, 3.4599e-03, 2.7485e-03, 6.0177e-04, 5.8746e-04, 1.0471e-03,
          8.5974e-04, 1.6434e-02, 7.5769e-04, 4.0283e-03, 6.5660e-04, 6.8808e-04,
          1.1415e-03, 1.1082e-03, 2.4433e-03, 4.1237e-03, 2.6321e-03, 1.0920e-03,
          7.3004e-04, 5.4207e-03, 8.2207e-04, 1.6693e-02, 2.4700e-03, 1.7395e-03,
          2.3972e-02, 7.2403e-03, 9.3699e-04, 1.1375e-02, 8.2684e-04, 7.7248e-04,
          7.3814e-04, 4.0555e-04, 4.0169e-03, 2.2526e-03, 9.6655e-04, 1.8272e-03,
          2.3315e-02, 2.8496e-03, 3.0708e-03, 1.7805e-03, 4.4479e-03, 1.1730e-03,
          4.1237e-03, 1.0567e-03, 1.0735e-02, 1.1482e-03, 2.9240e-03, 2.9964e-03,
          3.3474e-03, 1.1501e-03, 8.9121e-04, 6.2790e-03, 9.9277e-04, 1.1511e-03,
          1.9512e-03, 3.9978e-03, 8.0156e-04, 3.1719e-03, 8.1253e-03, 8.8882e-04,
          6.1798e-04, 9.7513e-04, 1.4162e-03, 4.2038e-03, 6.8283e-03, 4.8409e-03,
          5.2757e-03, 3.6583e-03, 4.7531e-03, 1.7109e-03, 6.6185e-03, 8.4400e-04,
          7.7858e-03, 1.8406e-03, 5.9853e-03, 9.7504e-03, 6.5689e-03, 2.6760e-03,
          1.1368e-03, 3.5686e-03, 1.4906e-03, 1.4591e-03, 2.6436e-03, 3.8853e-03,
          7.8964e-03, 3.0746e-03, 1.0672e-03, 2.3346e-03, 1.7090e-03, 2.0828e-03,
          5.9242e-03, 1.8959e-03, 1.6037e-02, 1.3170e-03, 3.8834e-03, 2.8572e-03,
          5.2757e-03, 4.9210e-03, 9.8419e-03, 7.8812e-03, 1.2922e-03, 1.5783e-03,
          1.8234e-02, 2.3918e-03, 3.5172e-03, 4.3182e-03, 4.6082e-03, 3.5324e-03,
          2.6531e-03, 1.8959e-03, 1.2321e-03, 1.8902e-03, 4.7035e-03, 1.3786e-02,
          3.2272e-03, 4.7607e-03, 7.5569e-03, 1.5190e-02, 5.0240e-03, 4.7913e-03,
          3.9734e-02, 2.3901e-05, 2.3901e-05, 2.3901e-05, 6.5982e-05, 6.5982e-05,
          6.5982e-05, 2.3901e-05, 2.3901e-05, 3.1757e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 1.9872e-04, 1.5664e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 6.5982e-05, 2.3901e-05, 1.9872e-04,
          1.9085e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05, 6.5982e-05, 2.3901e-05,
          2.3901e-05, 6.5982e-05, 2.3901e-05, 6.5982e-05, 6.5982e-05, 1.9872e-04,
          6.5982e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 1.9872e-04, 6.5982e-05,
          2.3901e-05, 6.5982e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3878e-04,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 6.5982e-05, 6.5982e-05, 1.9085e-04,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 1.5664e-04, 1.9872e-04, 1.9872e-04,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 1.9872e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 6.5982e-05, 1.9085e-04, 1.1671e-04, 2.3901e-05,
          1.1671e-04, 6.5982e-05, 1.9872e-04, 2.3901e-05, 1.5664e-04, 2.3901e-05,
          2.3901e-05, 3.1447e-04, 2.3901e-05, 1.1671e-04, 2.3901e-05, 6.5982e-05,
          6.5982e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 1.9872e-04, 1.0598e-04,
          4.3941e-04, 2.3901e-05, 1.9872e-04, 1.8466e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 1.9872e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 6.5982e-05, 6.5982e-05, 1.1671e-04, 2.3901e-05, 2.3901e-05,
          7.4625e-05, 1.5664e-04, 1.5664e-04, 1.1671e-04, 2.3901e-05, 2.3901e-05,
          1.1671e-04, 6.5982e-05, 3.2949e-04, 1.1671e-04, 2.3901e-05, 7.4625e-05,
          2.3901e-05, 2.3901e-05, 1.5664e-04, 1.9872e-04, 2.3901e-05, 2.3901e-05,
          1.1671e-04, 2.8944e-04, 7.4625e-05, 2.3901e-05, 2.3901e-05, 1.1671e-04,
          2.3901e-05, 7.1144e-04, 1.9872e-04, 1.1671e-04, 2.3901e-05, 2.3901e-05,
          6.5982e-05, 6.5982e-05, 1.9872e-04, 3.2949e-04, 2.3901e-05, 1.9872e-04,
          2.3901e-05, 2.3901e-05, 1.9872e-04, 6.5982e-05, 4.8733e-04, 1.9670e-04,
          3.2949e-04, 6.5982e-05, 2.3901e-05, 2.3901e-05, 1.9085e-04, 1.9085e-04,
          1.3447e-03, 2.3901e-05, 2.3901e-05, 2.3901e-05, 1.1671e-04, 3.2330e-04,
          4.9639e-04, 3.2377e-04, 4.4537e-04, 2.3901e-05, 6.5982e-05, 2.3901e-05,
          6.5982e-05, 1.9872e-04, 6.5982e-05, 3.1447e-04, 2.3901e-05, 2.3901e-05,
          2.3901e-05, 2.3901e-05, 2.3901e-05, 2.3901e-05, 3.6383e-04, 2.3901e-05,
          1.3533e-03, 2.4164e-04, 2.3901e-05, 2.3901e-05, 3.2949e-04, 7.4625e-05,
          3.1447e-04, 3.2949e-04, 1.9872e-04, 6.5982e-05, 2.3901e-05, 1.5664e-04,
          5.4455e-04, 1.4439e-03, 2.3901e-05, 3.2949e-04, 2.7871e-04, 2.3901e-05,
          1.1671e-04, 2.3901e-05, 2.3901e-05, 2.3901e-05, 1.9336e-04, 1.1671e-04,
          2.3901e-05, 1.1671e-04, 7.4625e-05, 5.1069e-04, 3.5524e-04, 3.5524e-04,
          2.3901e-05, 2.3901e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.6942e-03, 6.3599e-02, 2.5574e-02, 2.4933e-02, 5.3864e-02,
          4.4678e-02, 2.1454e-02, 2.1286e-03, 1.1921e-03, 1.7042e-03, 3.2043e-03,
          1.1005e-03, 6.0730e-03, 9.9335e-03, 1.5764e-03, 1.9932e-03, 1.7441e-02,
          9.7275e-03, 1.2550e-03, 2.1114e-03, 1.8673e-03, 3.6335e-03, 6.5231e-03,
          3.9978e-03, 2.4319e-03, 1.6815e-02, 7.8278e-03, 3.4924e-03, 8.7051e-03,
          1.7654e-02, 2.9907e-03, 3.7270e-03, 2.0638e-03, 1.1463e-03, 1.9321e-03,
          1.2293e-03, 7.9107e-04, 2.1133e-03, 1.3733e-03, 3.5763e-03, 2.3918e-03,
          1.6441e-03, 1.3304e-03, 1.4477e-03, 1.1311e-03, 1.0864e-02, 3.7651e-03,
          6.1417e-03, 5.9280e-03, 8.9645e-03, 2.1305e-03, 1.8387e-03, 1.5020e-03,
          9.5749e-04, 1.1349e-03, 1.0080e-03, 7.6103e-04, 1.4296e-03, 1.0014e-03,
          1.1110e-03, 9.0179e-03, 1.9217e-03, 2.0561e-03, 1.5097e-03, 3.7613e-03,
          2.6283e-03, 1.3609e-03, 1.4143e-03, 1.1187e-03, 1.0138e-03, 1.5717e-03,
          1.9684e-03, 3.8834e-03, 1.1988e-03, 2.2278e-03, 9.2125e-04, 7.5674e-04,
          1.3876e-03, 7.5197e-04, 1.0309e-03, 9.3889e-04, 9.2220e-04, 7.5531e-04,
          8.2779e-04, 4.1542e-03, 1.7357e-03, 1.9855e-03, 1.3838e-03, 1.6193e-03,
          3.0785e-03, 2.2850e-03, 1.3895e-03, 2.4738e-03, 1.5135e-03, 2.1267e-03,
          4.7722e-03, 7.6008e-04, 2.9430e-03, 3.0518e-03, 7.3814e-04, 7.7915e-04,
          1.9684e-03, 1.2455e-03, 6.7663e-04, 9.8801e-04, 1.7662e-03, 1.1131e-02,
          3.1910e-03, 1.5430e-03, 2.7332e-03, 1.5841e-03, 1.9178e-03, 1.7214e-03,
          1.9531e-03, 2.1038e-03, 9.8343e-03, 4.3373e-03, 3.2501e-03, 7.9041e-03,
          2.1912e-02, 1.4477e-03, 5.5122e-04, 1.3075e-03, 2.1896e-03, 5.2929e-04,
          5.6171e-04, 7.2765e-04, 9.1267e-04, 1.8635e-03, 3.7689e-03, 3.6201e-03,
          2.7924e-03, 1.4734e-03, 1.0605e-03, 9.7036e-04, 1.8797e-03, 1.2703e-03,
          1.5823e-02, 2.8473e-02, 1.6449e-02, 8.9111e-03, 5.2643e-03, 2.5925e-02,
          7.3528e-04, 8.8930e-04, 7.4387e-04, 7.1478e-04, 1.2064e-03, 1.7385e-03,
          2.0618e-03, 1.4448e-03, 9.4652e-04, 4.0016e-03, 3.1605e-03, 2.8591e-03,
          2.2736e-03, 2.3155e-03, 1.5602e-03, 1.9007e-03, 5.7449e-03, 4.0100e-02,
          4.2915e-03, 3.7651e-03, 1.7578e-02, 3.9940e-03, 1.5030e-03, 1.0471e-03,
          1.6212e-03, 1.0166e-03, 1.2159e-03, 1.4505e-03, 1.0117e-02, 1.8950e-03,
          1.4114e-03, 1.3971e-03, 1.7071e-03, 2.8419e-03, 3.0231e-03, 5.0774e-03,
          1.8797e-03, 2.5730e-03, 3.3035e-03, 4.8943e-03, 4.4975e-03, 4.3755e-03,
          3.1796e-03, 7.5623e-02, 1.8358e-05, 1.8358e-05, 9.2208e-05, 1.4126e-04,
          6.7413e-05, 1.8358e-05, 1.8358e-05, 9.2208e-05, 1.4126e-04, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 9.2208e-05, 9.2208e-05, 9.2208e-05,
          9.2208e-05, 9.2208e-05, 1.8358e-05, 6.7413e-05, 1.4126e-04, 9.2208e-05,
          6.7413e-05, 1.8358e-05, 1.8358e-05, 1.4126e-04, 9.2208e-05, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 6.7413e-05, 6.7413e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 7.7772e-04, 1.4126e-04, 1.4126e-04,
          9.2208e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          9.2208e-05, 9.2208e-05, 1.8358e-05, 6.7413e-05, 1.8358e-05, 6.7413e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 9.2208e-05, 1.8358e-05, 6.7413e-05, 9.2208e-05, 9.2208e-05,
          9.2208e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 1.4126e-04, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 6.7413e-05, 1.8358e-05, 1.8358e-05, 6.7413e-05, 1.8358e-05,
          8.2684e-04, 9.2208e-05, 1.4126e-04, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.4126e-04,
          1.4126e-04, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          6.7413e-05, 1.8358e-05, 9.2208e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05, 9.2208e-05, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 6.7413e-05, 1.8358e-05, 9.2208e-05, 6.7413e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 6.7413e-05, 6.7413e-05,
          1.8358e-05, 1.8358e-05, 1.8358e-05, 1.4126e-04, 9.2208e-05, 9.2208e-05,
          1.4126e-04, 1.4126e-04, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 1.8358e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          1.8358e-05, 9.2208e-05, 1.8358e-05, 1.4126e-04, 1.8358e-05, 1.8358e-05,
          6.7413e-05, 6.7413e-05, 6.7413e-05, 1.8358e-05, 9.2208e-05, 1.8358e-05,
          1.8358e-05, 6.7413e-05, 9.2208e-05, 6.7413e-05, 2.7871e-04, 1.8358e-05,
          1.8358e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05, 1.4126e-04, 1.4126e-04,
          2.7871e-04, 1.8358e-05, 6.7413e-05, 6.7413e-05, 1.8358e-05, 3.5262e-04,
          1.4126e-04, 1.4126e-04, 2.2972e-04, 1.8358e-05, 1.4126e-04, 1.4126e-04,
          1.4126e-04, 1.8358e-05, 6.7413e-05, 3.5262e-04, 1.8358e-05, 1.8358e-05,
          9.2208e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 6.7413e-05, 1.8358e-05,
          1.4126e-04, 6.7413e-05, 1.4126e-04, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          3.5262e-04, 1.8358e-05, 6.7413e-05, 6.7413e-05, 1.4126e-04, 9.2208e-05,
          1.8358e-05, 9.2208e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05, 1.8358e-05,
          4.7731e-04, 4.7731e-04, 9.2208e-05, 5.6696e-04, 4.7731e-04, 9.2208e-05,
          1.8358e-05, 1.4126e-04, 9.2208e-05, 9.1219e-04, 9.2208e-05, 1.8358e-05,
          9.2208e-05, 9.2208e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.2098e-03, 6.8726e-02, 2.3087e-02, 2.9633e-02, 4.5288e-02,
          5.8746e-02, 2.5665e-02, 2.9716e-03, 1.5736e-03, 1.1002e-02, 1.8707e-02,
          5.0354e-03, 2.6283e-03, 1.7252e-03, 5.1155e-03, 2.7580e-03, 2.7046e-03,
          2.4986e-03, 1.9798e-03, 5.0306e-04, 7.0810e-04, 7.9393e-04, 6.2103e-03,
          1.6998e-02, 4.4479e-03, 1.8860e-02, 1.0498e-02, 1.3103e-03, 3.4599e-03,
          3.2139e-03, 4.4937e-03, 1.5154e-03, 3.7441e-03, 7.3395e-03, 1.5879e-03,
          1.8930e-03, 8.1873e-04, 3.5706e-03, 3.7346e-03, 6.9284e-04, 1.7910e-03,
          8.2684e-04, 6.6137e-04, 6.7663e-04, 5.9605e-04, 3.6354e-03, 3.4752e-03,
          1.2245e-03, 1.1950e-03, 7.2021e-03, 2.3899e-03, 2.6417e-03, 2.4223e-03,
          9.6512e-04, 1.4544e-03, 8.2016e-04, 8.1348e-04, 1.7805e-03, 1.1091e-03,
          9.9850e-04, 2.4872e-03, 1.5516e-03, 1.8167e-03, 1.0147e-03, 4.5547e-03,
          2.1095e-03, 1.7757e-03, 2.2488e-03, 4.4584e-04, 4.3845e-04, 6.5517e-04,
          8.3256e-04, 2.1591e-02, 1.2407e-03, 3.2692e-03, 9.1648e-04, 8.7547e-04,
          1.3685e-03, 7.4291e-04, 1.1606e-03, 2.8992e-03, 1.4267e-03, 8.1348e-04,
          7.1239e-04, 3.4809e-03, 1.2445e-03, 9.3765e-03, 1.5898e-03, 1.2169e-03,
          1.0895e-02, 4.5166e-03, 6.0320e-04, 5.6610e-03, 5.1165e-04, 6.7282e-04,
          1.7338e-03, 6.8998e-04, 2.5024e-03, 2.9240e-03, 8.0156e-04, 1.0977e-03,
          9.5291e-03, 2.7962e-03, 1.5078e-03, 1.5574e-03, 4.1389e-03, 2.0161e-03,
          2.8610e-03, 8.0633e-04, 4.5509e-03, 8.6117e-04, 1.7509e-03, 2.1477e-03,
          3.0975e-03, 7.3719e-04, 1.5688e-03, 4.2686e-03, 8.9645e-04, 3.7479e-03,
          3.9749e-03, 3.0651e-03, 6.5327e-04, 2.1667e-03, 8.5983e-03, 6.5422e-04,
          6.1417e-04, 7.7343e-04, 1.0214e-03, 3.4237e-03, 5.2109e-03, 3.5400e-03,
          3.0155e-03, 2.0504e-03, 7.1716e-03, 1.0967e-03, 2.9259e-03, 5.8270e-04,
          6.9923e-03, 3.1433e-03, 1.0063e-02, 1.9470e-02, 1.2596e-02, 5.1041e-03,
          8.9693e-04, 5.0507e-03, 1.2970e-03, 1.0395e-03, 1.6603e-03, 3.1013e-03,
          6.9427e-03, 2.2812e-03, 8.4543e-04, 2.5597e-03, 2.4147e-03, 1.3924e-03,
          3.5744e-03, 1.6127e-03, 5.0087e-03, 8.3017e-04, 2.6207e-03, 2.9144e-03,
          5.6686e-03, 8.5297e-03, 7.5684e-03, 2.5314e-02, 1.7948e-03, 2.0065e-03,
          6.4888e-03, 2.6684e-03, 2.8248e-03, 5.1041e-03, 4.7836e-03, 3.6755e-03,
          1.6775e-03, 1.3590e-03, 1.2503e-03, 1.4000e-03, 3.4428e-03, 1.2527e-02,
          3.5458e-03, 3.2578e-03, 6.9427e-03, 6.6261e-03, 8.0643e-03, 1.0071e-02,
          1.1993e-02, 5.6076e-03, 7.6050e-02, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.0359e-04, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.0359e-04, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.0359e-04, 1.0359e-04,
          1.0359e-04, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 7.6246e-04, 1.0359e-04, 1.7524e-05,
          1.0359e-04, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          7.6246e-04, 1.0359e-04, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.0359e-04, 1.0359e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.0359e-04,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.0359e-04,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.0359e-04, 1.0359e-04, 1.0359e-04, 1.7524e-05, 1.0359e-04, 1.7524e-05,
          1.0359e-04, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05,
          1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04, 1.7524e-05,
          1.0359e-04, 1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.0359e-04,
          1.7524e-05, 1.0359e-04, 1.7524e-05, 1.7524e-05, 1.7524e-05, 1.7524e-05,
          1.0359e-04, 1.0359e-04, 1.0359e-04, 1.0359e-04, 1.0359e-04, 1.7524e-05,
          1.7524e-05, 1.0359e-04, 1.0359e-04, 1.0359e-04, 1.0359e-04, 1.7524e-05,
          1.0359e-04, 1.7524e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.9199e-03, 5.0568e-02, 1.8066e-02, 2.1317e-02, 6.9641e-02,
          3.6163e-02, 1.7960e-02, 1.2541e-03, 1.2865e-03, 3.5686e-03, 6.9504e-03,
          2.3308e-03, 1.2016e-03, 1.3599e-03, 1.6403e-03, 8.1396e-04, 1.5793e-03,
          8.5545e-04, 6.3896e-04, 8.4829e-04, 8.5449e-04, 6.5136e-04, 1.1044e-03,
          6.0501e-03, 2.3785e-03, 1.2711e-02, 3.6545e-03, 2.5234e-03, 1.6594e-03,
          1.8749e-03, 1.7328e-03, 9.1696e-04, 2.5501e-03, 1.4925e-03, 9.9182e-04,
          3.9864e-03, 3.7079e-03, 2.5578e-03, 1.5993e-03, 1.1826e-03, 4.0054e-03,
          3.3970e-03, 2.7504e-03, 2.0733e-03, 1.5335e-03, 1.2121e-03, 2.6760e-03,
          7.3385e-04, 9.3174e-04, 6.7978e-03, 3.0556e-03, 1.8511e-03, 1.2722e-03,
          1.4133e-03, 1.1768e-03, 1.6060e-03, 7.3700e-03, 2.7542e-03, 7.1144e-03,
          4.1389e-03, 1.9007e-03, 1.5316e-03, 1.4038e-03, 5.3596e-03, 4.4785e-03,
          3.7956e-03, 1.6699e-03, 1.6317e-03, 1.7519e-03, 1.6270e-03, 1.8768e-03,
          1.3409e-03, 5.6496e-03, 2.8381e-03, 2.2526e-03, 9.9850e-04, 7.9775e-04,
          1.7471e-03, 1.3596e-02, 9.3918e-03, 4.7684e-03, 5.6190e-03, 2.6302e-03,
          3.2177e-03, 2.0638e-03, 1.2741e-03, 2.5673e-03, 1.8682e-03, 2.5463e-03,
          4.1885e-03, 3.3855e-03, 1.6832e-03, 3.0308e-03, 1.4629e-03, 1.4477e-03,
          2.5520e-03, 1.6613e-03, 1.1435e-03, 5.1880e-03, 3.0823e-02, 1.2207e-02,
          8.9111e-03, 3.0479e-03, 3.3512e-03, 6.2714e-03, 3.9139e-03, 2.4357e-03,
          1.8024e-03, 3.6564e-03, 5.0163e-03, 2.3003e-03, 3.7804e-03, 2.3880e-03,
          2.5215e-03, 3.0823e-03, 9.7752e-04, 4.0550e-03, 1.9855e-03, 6.5269e-03,
          6.5956e-03, 1.2413e-02, 1.6525e-02, 8.2302e-04, 4.3793e-03, 5.6496e-03,
          1.4351e-02, 6.8665e-03, 7.0190e-03, 4.2000e-03, 1.2846e-03, 2.1248e-03,
          4.2458e-03, 2.6569e-03, 6.9714e-04, 9.9754e-04, 2.8839e-03, 2.4567e-03,
          2.2717e-03, 1.6193e-03, 1.2226e-03, 2.5711e-03, 2.1362e-03, 4.1771e-03,
          2.4750e-02, 1.6127e-03, 7.3700e-03, 8.0719e-03, 8.2092e-03, 9.4528e-03,
          9.4452e-03, 5.0659e-03, 9.1629e-03, 2.7676e-03, 2.6531e-03, 6.2561e-03,
          3.1357e-03, 3.6507e-03, 8.4305e-04, 2.5043e-03, 5.4626e-03, 1.5621e-03,
          2.7733e-03, 2.6283e-03, 1.7004e-03, 2.2182e-03, 2.8214e-02, 8.7814e-03,
          7.9422e-03, 5.7106e-03, 5.7068e-03, 1.9188e-03, 2.2030e-03, 4.1771e-03,
          4.2992e-03, 5.5046e-03, 1.0139e-02, 5.7526e-03, 4.0436e-03, 2.0580e-03,
          1.9836e-03, 1.7080e-03, 2.7657e-03, 1.4200e-03, 1.2798e-03, 1.1244e-03,
          2.1248e-03, 1.5450e-03, 1.8196e-03, 5.5328e-02, 2.0266e-05, 9.7036e-05,
          9.7036e-05, 2.0266e-05, 2.0266e-05, 1.9634e-04, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 9.7036e-05, 2.0266e-05, 2.2449e-03,
          9.7036e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 9.7036e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 9.7036e-05, 9.7036e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 7.1144e-04, 9.7036e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          9.7036e-05, 2.0266e-05, 1.9634e-04, 1.9634e-04, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 9.7036e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          5.0545e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 5.0545e-04, 5.8222e-04, 2.0266e-05, 2.0266e-05,
          5.8222e-04, 2.0266e-05, 1.1957e-04, 2.0266e-05, 2.0266e-05, 1.9634e-04,
          2.0266e-05, 2.0266e-05, 1.9634e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          1.1957e-04, 2.0266e-05, 4.8280e-04, 2.0266e-05, 2.0266e-05, 1.9634e-04,
          2.0266e-05, 2.0266e-05, 2.5201e-04, 2.0266e-05, 2.0266e-05, 8.1110e-04,
          7.1144e-04, 9.7036e-05, 2.0266e-05, 2.0266e-05, 9.7036e-05, 9.7036e-05,
          9.7036e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 1.9634e-04, 2.0266e-05,
          1.9634e-04, 9.7036e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          9.7036e-05, 4.2510e-04, 9.7036e-05, 1.1957e-04, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 9.7036e-05, 2.0266e-05,
          1.9634e-04, 1.9634e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05, 5.0545e-04,
          9.7036e-05, 2.0266e-05, 1.9634e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          5.0545e-04, 1.1957e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05, 1.1957e-04, 2.0266e-05,
          2.0266e-05, 1.1957e-04, 2.0266e-05, 2.0266e-05, 2.0266e-05, 2.0266e-05,
          2.0266e-05, 2.0266e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9967e-03, 5.3009e-02, 1.5297e-02, 1.9562e-02, 4.6112e-02,
          3.7079e-02, 1.2924e-02, 1.2302e-03, 1.1349e-03, 3.3741e-03, 8.5678e-03,
          2.6970e-03, 1.0157e-03, 1.9894e-03, 1.2712e-03, 9.1887e-04, 1.2064e-03,
          1.0099e-03, 7.5102e-04, 3.3355e-04, 3.3498e-04, 3.4499e-04, 1.2922e-03,
          9.0866e-03, 2.1400e-03, 1.0338e-02, 4.1275e-03, 8.7023e-04, 1.5211e-03,
          3.1319e-03, 2.0390e-03, 6.9094e-04, 1.6146e-03, 1.4343e-03, 9.3603e-04,
          2.0256e-03, 1.4906e-03, 1.5345e-03, 1.1797e-03, 6.8235e-04, 1.8806e-03,
          1.8616e-03, 1.3351e-03, 1.9989e-03, 6.1703e-04, 1.3514e-03, 1.4172e-03,
          7.6962e-04, 4.9448e-04, 3.8700e-03, 1.7910e-03, 1.9512e-03, 1.2302e-03,
          1.1749e-03, 1.2760e-03, 9.8705e-04, 1.9989e-03, 3.8548e-03, 3.3932e-03,
          2.4204e-03, 1.4801e-03, 1.3981e-03, 1.0071e-03, 4.2610e-03, 2.8744e-03,
          2.2003e-02, 5.2147e-03, 1.5583e-03, 8.5926e-04, 8.6212e-04, 1.6451e-03,
          7.5197e-04, 5.9128e-03, 1.5125e-03, 2.5883e-03, 1.2035e-03, 9.2983e-04,
          1.4772e-03, 3.4561e-03, 1.7685e-02, 4.5013e-03, 1.5556e-02, 1.5364e-03,
          1.1969e-03, 2.7504e-03, 1.2693e-03, 4.0512e-03, 1.4168e-02, 1.9394e-02,
          8.7204e-03, 3.8280e-03, 2.2678e-03, 4.7913e-03, 6.0797e-04, 6.5470e-04,
          1.0614e-03, 8.4257e-04, 1.3504e-03, 2.7256e-03, 7.6370e-03, 6.4049e-03,
          1.1665e-02, 9.2392e-03, 8.8196e-03, 3.0270e-03, 2.4624e-03, 1.6594e-03,
          2.3079e-03, 2.5024e-03, 6.4316e-03, 5.3520e-03, 2.1301e-02, 8.0872e-03,
          2.1286e-03, 1.0300e-03, 5.1641e-04, 2.5063e-03, 8.8549e-04, 2.4509e-03,
          4.2763e-03, 7.2517e-03, 6.9618e-03, 1.0242e-03, 3.6316e-03, 1.0788e-02,
          2.6398e-03, 1.2634e-02, 1.5320e-02, 2.8114e-03, 1.6823e-03, 2.7294e-03,
          1.8585e-02, 1.2138e-02, 8.5545e-04, 4.2191e-03, 6.7177e-03, 1.9035e-03,
          2.0657e-03, 1.3666e-03, 1.3351e-03, 2.9507e-03, 2.1439e-03, 4.6425e-03,
          1.6006e-02, 1.3199e-03, 5.6000e-03, 5.5313e-03, 3.6194e-02, 4.8409e-03,
          9.2621e-03, 3.0746e-03, 2.7351e-03, 1.8167e-03, 2.1095e-03, 3.0575e-03,
          5.6572e-03, 3.9902e-03, 1.9131e-03, 9.3842e-03, 2.5368e-03, 1.2703e-03,
          1.9312e-03, 2.1648e-03, 1.9932e-03, 2.1133e-03, 6.2065e-03, 8.2932e-03,
          1.0284e-02, 6.3591e-03, 4.7188e-03, 1.8559e-03, 3.1166e-03, 3.2139e-03,
          2.7008e-03, 3.2787e-03, 2.6112e-03, 2.8477e-03, 3.3112e-03, 2.7752e-03,
          2.2888e-03, 1.8692e-03, 3.0098e-03, 2.2240e-03, 1.3304e-03, 1.0281e-03,
          4.1733e-03, 1.5907e-03, 2.2717e-03, 1.4458e-02, 5.9174e-02, 6.0141e-05,
          6.0141e-05, 1.9193e-05, 1.9193e-05, 1.7941e-04, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 1.9193e-05, 1.7941e-04,
          6.0141e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 1.9193e-05,
          1.9193e-05, 6.0141e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 6.0141e-05,
          6.0141e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.7941e-04, 6.0141e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 2.1768e-04,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 6.0141e-05, 1.3852e-04,
          1.9193e-05, 1.9193e-05, 1.3852e-04, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.7941e-04, 1.9193e-05, 6.0141e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 2.5868e-04, 1.9193e-05,
          6.0141e-05, 1.9193e-05, 1.7941e-04, 1.7941e-04, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.7941e-04, 6.0141e-05, 2.5868e-04, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 2.1768e-04, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          2.5868e-04, 1.9193e-05, 2.1768e-04, 2.5868e-04, 1.9193e-05, 1.3852e-04,
          1.9193e-05, 1.9193e-05, 3.3689e-04, 2.5868e-04, 1.9193e-05, 1.9193e-05,
          1.7941e-04, 2.1768e-04, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05,
          1.9193e-05, 2.5868e-04, 6.0141e-05, 6.0141e-05, 1.9193e-05, 1.9193e-05,
          2.1768e-04, 6.0141e-05, 1.7941e-04, 2.1768e-04, 1.9193e-05, 6.0141e-05,
          1.9193e-05, 1.9193e-05, 4.1652e-04, 1.9193e-05, 2.5868e-04, 2.1768e-04,
          1.9193e-05, 2.5868e-04, 1.9193e-05, 1.9193e-05, 7.3731e-05, 7.3731e-05,
          2.7227e-04, 1.9193e-05, 1.9193e-05, 1.9193e-05, 6.0141e-05, 1.9193e-05,
          3.9148e-04, 1.2839e-04, 2.1768e-04, 1.9193e-05, 6.0141e-05, 1.9193e-05,
          6.0141e-05, 1.9193e-05, 6.0141e-05, 1.3852e-04, 1.9193e-05, 1.9193e-05,
          2.1768e-04, 1.9193e-05, 1.9193e-05, 6.0141e-05, 7.3731e-05, 1.9193e-05,
          3.7789e-04, 6.0141e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05,
          2.5868e-04, 2.1768e-04, 6.0141e-05, 6.0141e-05, 1.9193e-05, 3.4809e-04,
          2.1768e-04, 3.7789e-04, 1.9193e-05, 2.1768e-04, 1.9193e-05, 1.9193e-05,
          1.9193e-05, 1.9193e-05, 1.9193e-05, 1.9193e-05, 1.3852e-04, 6.0141e-05,
          1.9193e-05, 1.3852e-04, 1.9193e-05, 1.3852e-04, 3.7146e-04, 1.3852e-04,
          1.9193e-05, 1.9193e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.7106e-03, 5.5359e-02, 1.6953e-02, 2.1042e-02, 4.8279e-02,
          4.2816e-02, 1.3405e-02, 2.1362e-03, 1.3399e-03, 5.5580e-03, 6.9008e-03,
          2.8095e-03, 2.6073e-03, 8.1491e-04, 6.9733e-03, 2.1133e-03, 3.0003e-03,
          2.1133e-03, 1.0900e-03, 2.9278e-04, 4.4417e-04, 3.1328e-04, 3.2387e-03,
          7.6942e-03, 5.3329e-03, 1.0704e-02, 6.9923e-03, 6.4278e-04, 4.5013e-03,
          1.2369e-03, 5.0354e-03, 1.2312e-03, 9.5749e-03, 7.0610e-03, 2.0256e-03,
          5.5542e-03, 8.8120e-04, 4.9057e-03, 7.0190e-03, 4.0722e-04, 2.1152e-03,
          6.9141e-04, 5.6839e-04, 5.3644e-04, 5.3358e-04, 4.6387e-03, 7.4234e-03,
          4.4632e-04, 4.0436e-04, 4.0512e-03, 2.4700e-03, 3.1891e-03, 2.4242e-03,
          6.8140e-04, 1.5974e-03, 7.1573e-04, 1.3199e-03, 1.7443e-03, 1.5430e-03,
          1.3342e-03, 1.3762e-03, 1.1158e-03, 1.6060e-03, 9.6416e-04, 1.9577e-02,
          1.1797e-03, 9.1982e-04, 2.0103e-03, 3.9053e-04, 3.1662e-04, 4.3535e-04,
          4.9019e-04, 1.3229e-02, 1.0223e-03, 3.4599e-03, 5.8079e-04, 6.1703e-04,
          1.3609e-03, 1.2035e-03, 1.2970e-03, 7.0190e-03, 1.3981e-03, 1.0910e-03,
          8.6260e-04, 4.3755e-03, 7.7057e-04, 3.6278e-03, 7.5769e-04, 6.7854e-04,
          6.6414e-03, 4.2419e-03, 4.8208e-04, 3.5706e-03, 6.1655e-04, 5.4836e-04,
          1.0014e-03, 6.0749e-04, 3.4542e-03, 4.0321e-03, 1.3800e-03, 2.4147e-03,
          1.0063e-02, 1.1320e-03, 1.5020e-03, 4.7340e-03, 1.7136e-02, 1.4763e-03,
          3.4351e-03, 8.7595e-04, 5.6953e-03, 5.5122e-04, 1.0939e-03, 1.5011e-03,
          3.0689e-03, 1.3800e-03, 5.3835e-04, 1.7868e-02, 7.5245e-04, 1.6813e-03,
          2.5768e-03, 8.6823e-03, 8.7929e-04, 1.5669e-03, 3.8879e-02, 7.9966e-04,
          1.0386e-03, 8.2397e-04, 1.2569e-03, 1.6129e-02, 5.8136e-03, 4.0970e-03,
          1.7958e-03, 1.1187e-03, 2.4681e-03, 7.4196e-04, 3.2368e-03, 5.2977e-04,
          1.2123e-02, 1.0786e-03, 2.4414e-03, 5.5428e-03, 7.5836e-03, 3.7136e-03,
          1.4906e-03, 2.3289e-03, 2.5902e-03, 2.1152e-03, 1.5001e-03, 1.3641e-02,
          1.3756e-02, 1.0727e-02, 1.8587e-03, 3.1414e-03, 1.9093e-03, 2.7714e-03,
          4.5929e-03, 2.4872e-03, 2.2678e-03, 7.1812e-04, 7.1793e-03, 2.2793e-03,
          1.1612e-02, 6.5231e-03, 8.5831e-03, 1.0002e-02, 3.2597e-03, 3.1948e-03,
          8.3389e-03, 4.1771e-03, 1.1169e-02, 9.8114e-03, 5.7602e-03, 1.2650e-02,
          3.6831e-03, 1.9722e-03, 2.2411e-03, 1.8406e-03, 7.4539e-03, 7.5455e-03,
          5.6229e-03, 7.8354e-03, 2.4307e-02, 5.9319e-03, 2.5864e-03, 3.8128e-03,
          5.9319e-03, 1.2684e-03, 4.1809e-03, 2.8152e-03, 1.9350e-03, 4.0741e-02,
          1.3745e-04, 1.7762e-05, 1.7762e-05, 3.3855e-04, 1.3745e-04, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 7.2479e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 7.2479e-05, 3.3855e-04,
          1.3745e-04, 1.7762e-05, 1.7762e-05, 8.2672e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 7.2479e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.3745e-04,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          7.2479e-05, 7.2479e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 8.2672e-05,
          1.7762e-05, 1.7762e-05, 8.2672e-05, 1.7762e-05, 1.7762e-05, 1.7345e-04,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 8.2672e-05, 1.3745e-04,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.3485e-03,
          1.7762e-05, 1.3745e-04, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.3745e-04, 1.3745e-04, 1.7762e-05,
          1.3745e-04, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 8.2672e-05, 5.6934e-04, 1.3745e-04, 1.7762e-05, 1.3745e-04,
          8.2672e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 7.2479e-05, 1.7762e-05,
          1.7345e-04, 1.7762e-05, 7.2479e-05, 3.3855e-04, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 1.7762e-05, 1.3745e-04, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 4.1771e-04, 1.7762e-05, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          7.2479e-05, 1.7762e-05, 2.1672e-04, 8.2672e-05, 1.7762e-05, 1.7762e-05,
          1.7762e-05, 7.2479e-05, 6.2609e-04, 7.2479e-05, 1.7762e-05, 1.7762e-05,
          7.2479e-05, 2.1672e-04, 2.7347e-04, 1.7762e-05, 1.7762e-05, 1.3745e-04,
          1.7762e-05, 2.8157e-04, 1.3745e-04, 8.2672e-05, 1.7762e-05, 2.1672e-04,
          2.1672e-04, 1.7762e-05, 2.2686e-04, 7.7057e-04, 7.2479e-05, 2.8157e-04,
          1.7762e-05, 1.7762e-05, 3.3855e-04, 1.3745e-04, 6.0368e-04, 7.7057e-04,
          2.1420e-03, 1.4133e-03, 1.7762e-05, 1.7762e-05, 1.3745e-04, 1.3745e-04,
          6.6853e-04, 7.2479e-05, 1.7762e-05, 1.7762e-05, 2.8157e-04, 1.7762e-05,
          4.3750e-04, 2.9325e-04, 1.6201e-04, 1.7762e-05, 7.2479e-05, 1.7762e-05,
          1.7762e-05, 8.3542e-04, 1.7762e-05, 7.2479e-05, 1.7762e-05, 1.7762e-05,
          2.1672e-04, 1.7762e-05, 7.2479e-05, 1.7762e-05, 4.3750e-04, 2.1672e-04,
          2.8157e-04, 3.9101e-04, 1.3745e-04, 7.2479e-05, 1.6201e-04, 6.7139e-04,
          1.3745e-04, 7.7057e-04, 7.3624e-04, 1.7762e-05, 1.7762e-05, 1.7762e-05,
          1.0242e-03, 2.1672e-04, 1.7762e-05, 2.7347e-04, 4.1771e-04, 1.7762e-05,
          1.3745e-04, 1.7762e-05, 7.2479e-05, 7.2479e-05, 7.2479e-05, 8.2672e-05,
          1.7762e-05, 1.3745e-04, 7.2479e-05, 3.4499e-04, 7.2479e-05, 5.7840e-04,
          7.2479e-05, 7.2479e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5547e-03, 5.7892e-02, 1.3222e-02, 1.6083e-02, 3.8513e-02,
          4.4434e-02, 9.2239e-03, 1.4862e-02, 4.7989e-03, 4.2038e-03, 5.8670e-03,
          2.2678e-03, 2.9526e-03, 1.1492e-03, 3.0594e-03, 1.5656e-02, 3.1338e-03,
          2.4529e-03, 3.9177e-03, 2.9898e-04, 3.1710e-04, 3.2806e-04, 3.3016e-03,
          6.1073e-03, 2.0084e-03, 8.2016e-03, 6.5002e-03, 6.3229e-04, 3.8910e-03,
          1.5974e-03, 2.9953e-02, 1.2341e-03, 3.8948e-03, 4.1275e-03, 7.8583e-03,
          2.1343e-03, 7.2193e-04, 2.2964e-03, 3.3798e-03, 4.6635e-04, 1.1282e-03,
          5.1689e-04, 4.0436e-04, 4.2605e-04, 3.2687e-04, 3.5801e-03, 2.2678e-03,
          5.0211e-04, 3.6335e-04, 3.0060e-03, 8.8120e-03, 2.1591e-02, 6.2523e-03,
          2.1362e-03, 1.0963e-02, 1.7290e-03, 9.7227e-04, 2.7618e-03, 1.1473e-03,
          1.1177e-03, 1.4572e-03, 2.9926e-03, 2.5883e-03, 7.6962e-04, 3.1662e-03,
          1.3523e-03, 8.4972e-04, 1.0586e-03, 2.5344e-04, 2.4009e-04, 3.7766e-04,
          3.8481e-04, 7.7171e-03, 3.1757e-03, 2.6199e-02, 2.1095e-03, 2.8515e-03,
          6.7902e-03, 1.0204e-03, 1.2913e-03, 2.9259e-03, 1.4896e-03, 8.3351e-04,
          6.7520e-04, 3.2776e-02, 2.2564e-03, 3.0746e-03, 8.3637e-04, 7.5865e-04,
          5.0545e-03, 2.3556e-03, 3.6144e-04, 2.3689e-03, 3.9434e-04, 3.6907e-04,
          1.2722e-03, 1.1787e-03, 1.5823e-02, 2.8687e-03, 9.5415e-04, 1.3733e-03,
          7.1831e-03, 1.4439e-03, 1.6365e-03, 1.5554e-03, 4.2839e-03, 1.2589e-03,
          1.7181e-02, 7.4339e-04, 2.8076e-03, 6.0987e-04, 1.1425e-03, 1.0433e-03,
          1.4172e-03, 6.0844e-04, 5.0020e-04, 3.5172e-03, 5.2071e-04, 1.6193e-03,
          2.6608e-03, 3.5191e-03, 7.2432e-04, 1.1620e-02, 5.3024e-03, 7.9536e-04,
          6.6853e-04, 9.0790e-04, 1.4429e-03, 5.9509e-03, 1.1620e-02, 1.8982e-02,
          1.9989e-03, 1.0433e-03, 1.8034e-03, 6.3086e-04, 1.7061e-03, 4.2200e-04,
          5.0964e-03, 1.1177e-03, 4.2763e-03, 8.2550e-03, 4.9706e-03, 4.1618e-03,
          1.2169e-03, 1.5297e-03, 1.4896e-03, 1.3256e-03, 1.5717e-03, 5.5161e-03,
          6.3591e-03, 4.2267e-03, 1.3638e-03, 6.7406e-03, 6.1531e-03, 1.2913e-03,
          2.0752e-03, 1.3142e-03, 2.8477e-03, 7.2575e-04, 2.3155e-03, 1.9569e-03,
          2.8343e-03, 2.5692e-03, 6.5765e-03, 4.8790e-03, 1.6823e-03, 1.9512e-03,
          6.0539e-03, 2.2278e-03, 3.7899e-03, 3.7346e-03, 4.6844e-03, 4.4708e-03,
          4.1847e-03, 3.1643e-03, 1.6584e-03, 1.5478e-03, 2.3308e-03, 1.0582e-02,
          1.6594e-03, 2.7351e-03, 3.3112e-03, 3.8971e-02, 6.5994e-03, 2.0752e-03,
          5.7030e-03, 2.3232e-03, 3.9253e-03, 1.8435e-03, 2.2297e-03, 6.8321e-03,
          6.0852e-02, 2.8968e-05, 6.6102e-05, 2.9039e-04, 1.3256e-04, 6.6102e-05,
          2.8968e-05, 2.8968e-05, 6.6102e-05, 6.6102e-05, 6.6102e-05, 2.8968e-05,
          6.6102e-05, 9.0420e-05, 3.0351e-04, 2.8968e-05, 1.3256e-04, 9.5427e-05,
          2.8968e-05, 2.8968e-05, 2.8968e-05, 6.6102e-05, 6.6102e-05, 1.2755e-04,
          2.8968e-05, 1.3256e-04, 2.8968e-05, 1.3256e-04, 1.3256e-04, 1.3256e-04,
          6.6102e-05, 3.6478e-04, 3.6478e-04, 1.2755e-04, 6.7759e-04, 1.3256e-04,
          6.6102e-05, 2.2316e-04, 3.6478e-04, 2.5272e-04, 2.8968e-05, 6.6102e-05,
          2.8968e-05, 2.8968e-05, 2.5272e-04, 2.8968e-05, 2.8968e-05, 2.5272e-04,
          2.8968e-05, 2.8968e-05, 1.5688e-04, 9.5427e-05, 6.6102e-05, 1.2755e-04,
          3.6478e-04, 9.5427e-05, 2.8968e-05, 1.3256e-04, 2.8968e-05, 2.9039e-04,
          2.7990e-04, 3.0351e-04, 2.8968e-05, 2.8968e-05, 2.8968e-05, 5.0592e-04,
          2.8968e-05, 2.5272e-04, 2.8968e-05, 2.8968e-05, 2.8968e-05, 3.5167e-04,
          2.8968e-05, 2.8968e-05, 2.8968e-05, 1.3256e-04, 4.8137e-04, 2.9039e-04,
          1.2755e-04, 6.6102e-05, 4.5156e-04, 3.6478e-04, 1.5688e-04, 1.9395e-04,
          6.6102e-05, 2.5272e-04, 4.2176e-04, 1.3256e-04, 9.0420e-05, 1.2755e-04,
          6.6102e-05, 2.8968e-05, 9.0420e-05, 2.8968e-05, 1.3256e-04, 2.8968e-05,
          4.7851e-04, 9.5427e-05, 9.5427e-05, 2.9039e-04, 3.6478e-04, 2.8968e-05,
          3.6478e-04, 3.6478e-04, 3.6478e-04, 1.3256e-04, 9.0420e-05, 3.0351e-04,
          3.0351e-04, 2.8968e-05, 2.8968e-05, 6.6102e-05, 2.8968e-05, 6.6102e-05,
          6.6102e-05, 1.5688e-04, 9.5427e-05, 2.8968e-05, 2.8968e-05, 6.6102e-05,
          1.3256e-04, 1.3256e-04, 3.6478e-04, 6.6102e-05, 9.5427e-05, 3.8886e-04,
          5.3835e-04, 1.3256e-04, 1.5688e-04, 1.5688e-04, 3.6478e-04, 9.0420e-05,
          3.5334e-04, 6.6102e-05, 9.5427e-05, 9.0420e-05, 9.0420e-05, 9.3126e-04,
          9.0420e-05, 1.3256e-04, 5.4264e-04, 1.3256e-04, 2.8968e-05, 9.0420e-05,
          5.1403e-04, 2.8968e-05, 2.8968e-05, 3.6478e-04, 3.5167e-04, 5.4264e-04,
          2.8968e-05, 6.6102e-05, 1.5688e-04, 6.6102e-05, 2.1553e-04, 4.3130e-04,
          3.6478e-04, 3.2735e-04, 9.0420e-05, 9.0420e-05, 2.8968e-05, 2.8968e-05,
          2.8968e-05, 2.8968e-05, 3.5334e-04, 9.0420e-05, 9.9277e-04, 5.1403e-04,
          2.8968e-05, 2.8968e-05, 2.8968e-05, 6.6102e-05, 2.8968e-05, 2.8968e-05,
          1.3256e-04, 1.9395e-04, 2.8968e-05, 7.4005e-04, 2.8968e-05, 2.8968e-05,
          2.9039e-04, 2.8563e-04, 3.5167e-04, 2.8968e-05, 2.8968e-05, 9.0420e-05,
          2.8968e-05, 9.9277e-04, 5.8270e-04, 9.5427e-05, 9.0420e-05, 9.9277e-04,
          2.8968e-05, 3.6478e-04, 9.9277e-04, 2.8968e-05, 1.2755e-04, 5.7554e-04,
          9.5606e-04, 2.8968e-05, 2.8968e-05, 9.0420e-05, 3.6478e-04, 2.8968e-05,
          1.3256e-04, 9.5427e-05, 1.3256e-04, 9.5427e-05, 5.1355e-04, 6.6102e-05,
          2.8968e-05, 1.3256e-04, 3.1853e-04, 4.0412e-04, 9.5427e-05, 9.5427e-05,
          9.5427e-05, 9.5427e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.7978e-03, 5.0446e-02, 1.6388e-02, 1.4969e-02, 6.8665e-02,
          3.2257e-02, 1.5182e-02, 9.9754e-04, 1.1959e-03, 3.3607e-03, 5.7144e-03,
          2.2621e-03, 1.0347e-03, 1.1101e-03, 2.3689e-03, 6.5851e-04, 1.1787e-03,
          6.4087e-04, 5.2786e-04, 7.6056e-04, 8.4496e-04, 4.7755e-04, 9.1553e-04,
          5.3253e-03, 2.7180e-03, 1.0292e-02, 2.8553e-03, 1.5268e-03, 1.2760e-03,
          1.4858e-03, 1.3733e-03, 7.2289e-04, 2.7657e-03, 1.4629e-03, 1.0195e-03,
          4.6234e-03, 3.5629e-03, 2.1191e-03, 1.6232e-03, 9.4366e-04, 3.8643e-03,
          3.1929e-03, 2.3632e-03, 2.2526e-03, 1.6403e-03, 9.1410e-04, 3.1757e-03,
          5.7745e-04, 7.6771e-04, 6.3477e-03, 2.2354e-03, 1.5268e-03, 1.0958e-03,
          1.3800e-03, 1.2436e-03, 1.7538e-03, 8.2474e-03, 2.7180e-03, 7.9651e-03,
          4.3106e-03, 1.5688e-03, 1.3580e-03, 1.2255e-03, 4.4060e-03, 5.3902e-03,
          2.5940e-03, 1.5469e-03, 1.3294e-03, 1.9684e-03, 2.2488e-03, 1.6699e-03,
          1.4200e-03, 4.8256e-03, 2.1935e-03, 1.8759e-03, 9.8515e-04, 8.4829e-04,
          1.8711e-03, 1.5305e-02, 9.4528e-03, 5.7144e-03, 6.1569e-03, 3.0918e-03,
          3.1147e-03, 1.7500e-03, 1.2016e-03, 2.1534e-03, 1.5268e-03, 2.5711e-03,
          3.6163e-03, 3.6564e-03, 1.8272e-03, 2.7981e-03, 1.6985e-03, 1.5230e-03,
          1.6279e-03, 1.4009e-03, 9.7466e-04, 4.2915e-03, 3.0746e-02, 1.2634e-02,
          8.6823e-03, 2.4033e-03, 3.1071e-03, 8.9417e-03, 6.1150e-03, 2.1687e-03,
          1.6212e-03, 3.1548e-03, 4.9248e-03, 2.0123e-03, 2.6588e-03, 2.1229e-03,
          2.8458e-03, 3.5801e-03, 9.9373e-04, 5.2948e-03, 2.1648e-03, 5.2414e-03,
          4.7073e-03, 1.0834e-02, 1.4191e-02, 7.1478e-04, 6.0425e-03, 7.1297e-03,
          2.0996e-02, 6.6795e-03, 7.0763e-03, 5.2452e-03, 1.1244e-03, 1.8511e-03,
          3.1013e-03, 1.9503e-03, 6.5088e-04, 1.0118e-03, 3.0689e-03, 3.0727e-03,
          2.1114e-03, 1.7662e-03, 1.0099e-03, 1.9875e-03, 2.4738e-03, 3.6259e-03,
          1.8143e-02, 1.7643e-03, 7.8583e-03, 1.0475e-02, 8.3542e-03, 1.1681e-02,
          1.0895e-02, 6.8665e-03, 1.1932e-02, 2.6627e-03, 2.8038e-03, 5.9967e-03,
          3.3245e-03, 4.0550e-03, 7.0143e-04, 3.4332e-03, 5.7716e-03, 1.5812e-03,
          3.9139e-03, 2.6264e-03, 1.4734e-03, 2.2545e-03, 1.8936e-02, 8.1177e-03,
          7.6599e-03, 6.6795e-03, 6.0921e-03, 2.2202e-03, 2.1935e-03, 6.8321e-03,
          4.7455e-03, 6.3095e-03, 1.1635e-02, 5.7373e-03, 4.2648e-03, 1.6756e-03,
          2.1324e-03, 1.7729e-03, 2.9526e-03, 1.2312e-03, 1.0271e-03, 1.1215e-03,
          1.9417e-03, 1.8444e-03, 1.8272e-03, 3.2379e-02, 1.0735e-02, 3.6354e-03,
          1.9331e-03, 2.9083e-02, 1.6868e-05, 1.7428e-04, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.0467e-04, 1.6868e-05, 1.7428e-04,
          1.0467e-04, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 8.6486e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.7428e-04,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.0467e-04, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.7428e-04, 1.7428e-04, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 4.0674e-04, 1.7428e-04, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.0467e-04, 1.6868e-05, 1.7428e-04, 1.7428e-04, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 8.6486e-05, 1.0467e-04, 8.6486e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 8.6486e-05, 1.6868e-05, 1.6868e-05,
          8.6486e-05, 8.6486e-05, 1.6868e-05, 1.0467e-04, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 4.7636e-04, 4.7636e-04, 1.6868e-05, 1.6868e-05,
          4.7636e-04, 8.6486e-05, 8.6486e-05, 1.6868e-05, 1.6868e-05, 1.7428e-04,
          1.6868e-05, 8.6486e-05, 1.7428e-04, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          8.6486e-05, 1.6868e-05, 5.6410e-04, 1.6868e-05, 1.6868e-05, 1.7428e-04,
          1.6868e-05, 1.6868e-05, 1.7428e-04, 1.6868e-05, 1.6868e-05, 4.7636e-04,
          4.7636e-04, 1.0467e-04, 1.6868e-05, 1.6868e-05, 1.0467e-04, 1.0467e-04,
          1.6868e-05, 8.6486e-05, 1.6868e-05, 1.6868e-05, 1.7428e-04, 8.6486e-05,
          5.6410e-04, 1.7428e-04, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05,
          1.0467e-04, 8.6486e-05, 1.6868e-05, 8.6486e-05, 1.6868e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 1.0467e-04, 1.6868e-05,
          8.6486e-05, 1.7428e-04, 1.6868e-05, 8.6486e-05, 1.6868e-05, 4.7636e-04,
          1.7428e-04, 1.6868e-05, 1.7428e-04, 1.6868e-05, 1.6868e-05, 8.6486e-05,
          8.6486e-05, 4.7636e-04, 1.6868e-05, 8.6486e-05, 8.6486e-05, 1.6868e-05,
          1.6868e-05, 1.6868e-05, 1.6868e-05, 1.6868e-05, 8.6486e-05, 8.6486e-05,
          1.6868e-05, 8.6486e-05, 8.6486e-05, 2.8086e-04, 1.6868e-05, 8.6486e-05,
          1.6868e-05, 8.6486e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.2452e-03, 5.6732e-02, 1.4046e-02, 1.7868e-02, 3.9093e-02,
          4.1779e-02, 1.0147e-02, 1.6088e-03, 1.3065e-03, 4.7684e-03, 4.8103e-03,
          2.3861e-03, 3.5267e-03, 9.4604e-04, 6.5460e-03, 1.4372e-03, 6.1226e-03,
          2.5845e-03, 8.7214e-04, 4.1485e-04, 5.1975e-04, 3.8552e-04, 3.5763e-03,
          5.5885e-03, 3.5400e-03, 8.1863e-03, 7.3395e-03, 7.5102e-04, 7.6599e-03,
          1.5440e-03, 3.8910e-03, 1.6518e-03, 1.1353e-02, 5.8022e-03, 2.8076e-03,
          6.1874e-03, 1.0004e-03, 6.1836e-03, 7.3929e-03, 5.1594e-04, 3.0785e-03,
          6.6376e-04, 6.1417e-04, 5.1832e-04, 7.4053e-04, 7.3128e-03, 1.2207e-02,
          5.0449e-04, 4.3249e-04, 3.0003e-03, 3.3283e-03, 2.9049e-03, 2.6703e-03,
          6.5804e-04, 1.5974e-03, 9.1171e-04, 1.8816e-03, 1.7729e-03, 1.7185e-03,
          1.5621e-03, 1.8644e-03, 1.0929e-03, 1.9703e-03, 1.0014e-03, 1.7181e-02,
          8.9455e-04, 8.2159e-04, 2.0275e-03, 4.6945e-04, 3.4833e-04, 4.1509e-04,
          5.2691e-04, 1.2108e-02, 9.8419e-04, 3.3264e-03, 5.7268e-04, 6.4850e-04,
          1.5020e-03, 2.1706e-03, 1.5440e-03, 6.7825e-03, 1.4019e-03, 1.0710e-03,
          9.3365e-04, 3.8261e-03, 7.5579e-04, 2.9488e-03, 6.2084e-04, 5.9605e-04,
          5.3291e-03, 4.8218e-03, 4.1652e-04, 3.1033e-03, 5.7888e-04, 5.7602e-04,
          1.1530e-03, 5.7793e-04, 4.6005e-03, 5.5351e-03, 1.9608e-03, 3.6030e-03,
          8.8654e-03, 1.1539e-03, 1.5516e-03, 4.5700e-03, 1.2093e-02, 2.2392e-03,
          3.2902e-03, 8.7643e-04, 5.4817e-03, 5.0688e-04, 8.4877e-04, 1.5001e-03,
          2.4605e-03, 1.9045e-03, 7.3290e-04, 1.7609e-02, 7.1192e-04, 1.7481e-03,
          3.3665e-03, 1.1230e-02, 1.1082e-03, 1.3046e-03, 1.4000e-02, 1.0185e-03,
          1.7548e-03, 1.0557e-03, 1.7071e-03, 1.8478e-02, 5.7907e-03, 3.8815e-03,
          1.3857e-03, 8.5497e-04, 1.3571e-03, 6.1703e-04, 2.7351e-03, 4.8113e-04,
          1.9531e-02, 1.3723e-03, 2.5253e-03, 5.2567e-03, 9.3460e-03, 4.8599e-03,
          1.9684e-03, 2.3079e-03, 2.7866e-03, 2.8114e-03, 1.4286e-03, 1.3603e-02,
          1.5160e-02, 1.3275e-02, 2.9125e-03, 4.8828e-03, 2.1534e-03, 3.7556e-03,
          4.5815e-03, 2.6340e-03, 1.4935e-03, 5.8746e-04, 9.2468e-03, 3.2616e-03,
          1.5526e-02, 9.2468e-03, 1.3618e-02, 1.1307e-02, 4.2000e-03, 3.1586e-03,
          7.3586e-03, 4.6959e-03, 1.0796e-02, 8.6899e-03, 6.7596e-03, 2.0554e-02,
          4.7760e-03, 2.3842e-03, 3.3512e-03, 1.8711e-03, 7.9346e-03, 5.7526e-03,
          4.0550e-03, 5.0545e-03, 1.0887e-02, 5.4665e-03, 2.1687e-03, 3.8185e-03,
          4.0436e-03, 1.3180e-03, 2.9507e-03, 3.4218e-03, 2.0409e-03, 1.5297e-02,
          7.1030e-03, 1.8673e-03, 2.8732e-02, 5.9843e-04, 7.6175e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 7.6175e-05, 7.6175e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.0914e-04, 7.6175e-05, 5.0306e-04,
          1.0914e-04, 1.4126e-05, 1.4126e-05, 1.0914e-04, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 7.6175e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.0914e-04,
          1.4126e-05, 1.9503e-04, 1.4126e-05, 7.6175e-05, 1.4126e-05, 1.4126e-05,
          7.6175e-05, 2.5702e-04, 1.9503e-04, 1.4126e-05, 1.4126e-05, 1.9503e-04,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.0914e-04,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.9503e-04, 7.6175e-05, 1.9503e-04, 1.0914e-04, 1.4126e-05, 1.4126e-05,
          2.5702e-04, 1.9503e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05, 5.9509e-04,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.0914e-04, 1.7118e-04, 1.4126e-05,
          1.7118e-04, 1.7118e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 6.1226e-04, 1.7118e-04, 1.4126e-05, 1.7118e-04,
          1.7118e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.0914e-04, 1.4126e-05, 7.6175e-05, 5.9843e-04, 1.4126e-05, 1.4126e-05,
          1.9503e-04, 1.4126e-05, 1.9503e-04, 1.4126e-05, 1.4126e-05, 1.9503e-04,
          1.4126e-05, 1.4126e-05, 1.7118e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 1.9503e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 7.6175e-05, 6.7425e-04, 6.7425e-04, 1.9503e-04, 1.4126e-05,
          6.7425e-04, 6.1226e-04, 7.6175e-05, 1.4126e-05, 1.4126e-05, 1.7118e-04,
          1.4126e-05, 1.4126e-05, 1.7118e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05,
          1.4126e-05, 1.4126e-05, 5.2643e-04, 2.5702e-04, 7.6175e-05, 1.7118e-04,
          1.4126e-05, 1.4126e-05, 7.6175e-05, 7.6175e-05, 4.4465e-04, 6.7425e-04,
          1.1930e-03, 6.8998e-04, 1.4126e-05, 1.4126e-05, 1.7118e-04, 1.7118e-04,
          4.4465e-04, 7.6175e-05, 1.4126e-05, 1.4126e-05, 1.7118e-04, 1.4126e-05,
          1.7118e-04, 1.7118e-04, 1.4126e-05, 1.4126e-05, 7.6175e-05, 1.4126e-05,
          1.4126e-05, 3.5214e-04, 1.4126e-05, 7.6175e-05, 1.4126e-05, 1.4126e-05,
          5.0306e-04, 1.4126e-05, 1.4126e-05, 1.4126e-05, 1.7118e-04, 1.4126e-05,
          6.8998e-04, 1.7118e-04, 7.6175e-05, 7.6175e-05, 1.4126e-05, 6.7425e-04,
          1.7118e-04, 1.2026e-03, 8.7070e-04, 1.4126e-05, 1.4126e-05, 7.6175e-05,
          7.7581e-04, 1.0118e-03, 1.4126e-05, 2.5702e-04, 1.9503e-04, 1.4126e-05,
          3.7861e-04, 1.4126e-05, 3.7861e-04, 7.6175e-05, 7.6175e-05, 1.4126e-05,
          1.4126e-05, 1.7118e-04, 7.6175e-05, 4.1533e-04, 7.6175e-05, 7.1764e-04,
          3.7861e-04, 7.6175e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.1869e-03, 5.7556e-02, 2.2369e-02, 2.7115e-02, 4.8126e-02,
          4.2297e-02, 1.4938e-02, 2.3022e-03, 1.6661e-03, 3.7441e-03, 5.1651e-03,
          2.0504e-03, 9.0179e-03, 4.2686e-03, 2.6627e-03, 2.3079e-03, 2.5986e-02,
          1.4084e-02, 1.1873e-03, 1.5602e-03, 1.1930e-03, 2.9640e-03, 6.7406e-03,
          5.8746e-03, 1.8215e-03, 1.1856e-02, 9.9487e-03, 2.6302e-03, 2.9877e-02,
          6.4011e-03, 4.7340e-03, 1.2177e-02, 5.7487e-03, 3.7422e-03, 2.2869e-03,
          2.1706e-03, 6.5136e-04, 3.3836e-03, 3.5152e-03, 2.4223e-03, 1.7138e-03,
          7.1621e-04, 5.2452e-04, 6.1464e-04, 5.2691e-04, 2.7466e-02, 4.8981e-03,
          2.9945e-03, 2.0142e-03, 4.9210e-03, 2.8324e-03, 3.4275e-03, 1.9779e-03,
          7.3242e-04, 1.6842e-03, 7.4387e-04, 8.3065e-04, 1.3781e-03, 9.4366e-04,
          9.1457e-04, 5.7755e-03, 1.0614e-03, 1.7748e-03, 8.2827e-04, 4.7913e-03,
          1.2093e-03, 8.9645e-04, 1.3628e-03, 3.5024e-04, 3.4976e-04, 4.8923e-04,
          6.3419e-04, 9.0256e-03, 1.0319e-03, 4.0054e-03, 6.5517e-04, 6.6090e-04,
          1.5411e-03, 9.3126e-04, 1.0080e-03, 2.3994e-03, 1.1368e-03, 6.0987e-04,
          5.7888e-04, 4.5280e-03, 8.2684e-04, 2.9564e-03, 7.3481e-04, 6.4373e-04,
          5.4321e-03, 3.2463e-03, 4.0984e-04, 2.8057e-03, 4.7827e-04, 5.9891e-04,
          3.0632e-03, 5.6648e-04, 3.5648e-03, 4.3793e-03, 8.3447e-04, 1.1530e-03,
          5.4893e-03, 8.1825e-04, 1.2474e-03, 1.2941e-03, 3.2845e-03, 3.9902e-03,
          3.1624e-03, 8.0109e-04, 3.3188e-03, 4.8828e-04, 8.9550e-04, 1.1435e-03,
          1.0624e-03, 9.4271e-04, 1.8196e-03, 5.1270e-03, 7.9012e-04, 3.6240e-03,
          6.8207e-03, 3.8643e-03, 5.4550e-04, 1.5507e-03, 5.0888e-03, 4.6802e-04,
          6.2323e-04, 6.4373e-04, 9.0551e-04, 4.0207e-03, 5.2872e-03, 3.8223e-03,
          1.6584e-03, 1.0815e-03, 1.0929e-03, 5.9795e-04, 1.9875e-03, 4.5991e-04,
          2.6382e-02, 4.4098e-03, 5.7068e-03, 5.8289e-03, 5.7564e-03, 1.0056e-02,
          8.2445e-04, 7.1430e-04, 9.7418e-04, 9.0218e-04, 1.1196e-03, 3.2616e-03,
          5.2223e-03, 2.4834e-03, 9.0694e-04, 3.4122e-03, 1.8616e-03, 1.8339e-03,
          2.2545e-03, 1.3752e-03, 1.6575e-03, 6.5756e-04, 3.2444e-03, 9.0790e-03,
          5.4283e-03, 3.9406e-03, 4.3701e-02, 5.4665e-03, 2.2736e-03, 1.3819e-03,
          4.2534e-03, 1.6232e-03, 2.8267e-03, 2.2755e-03, 1.6663e-02, 5.8289e-03,
          1.5841e-03, 1.1129e-03, 1.3447e-03, 1.7385e-03, 3.2654e-03, 8.4229e-03,
          1.8473e-03, 2.7828e-03, 4.5700e-03, 5.5351e-03, 3.3894e-03, 2.5215e-03,
          4.1237e-03, 5.9738e-03, 2.6855e-03, 1.6031e-03, 1.3723e-03, 5.1308e-03,
          6.4850e-03, 6.6185e-04, 5.3520e-03, 6.6589e-02, 8.8930e-05, 8.8930e-05,
          2.0027e-05, 2.0027e-05, 4.9353e-05, 4.9353e-05, 8.8930e-05, 4.9353e-05,
          8.8930e-05, 4.9353e-05, 3.6073e-04, 8.8930e-05, 8.8930e-05, 2.0027e-05,
          1.6069e-04, 2.0027e-05, 1.2112e-04, 8.9526e-05, 8.8930e-05, 8.8930e-05,
          1.2112e-04, 2.0027e-05, 2.0027e-05, 1.6069e-04, 4.9353e-05, 4.9353e-05,
          2.0027e-05, 3.9005e-04, 3.9005e-04, 2.2042e-04, 8.8930e-05, 8.8930e-05,
          1.2112e-04, 5.5504e-04, 3.6073e-04, 2.2542e-04, 2.0027e-05, 5.9664e-05,
          2.0027e-05, 1.2112e-04, 1.4615e-04, 8.8930e-05, 4.9353e-05, 6.7770e-05,
          2.0027e-05, 2.0027e-05, 2.0027e-05, 4.9353e-05, 5.9664e-05, 2.0027e-05,
          3.6073e-04, 1.7023e-04, 3.6073e-04, 3.0947e-04, 1.7023e-04, 1.6069e-04,
          6.2227e-04, 3.6073e-04, 2.0027e-05, 2.0027e-05, 2.0027e-05, 2.0027e-05,
          4.9353e-05, 1.6880e-04, 2.0027e-05, 4.9353e-05, 2.0027e-05, 8.8930e-05,
          4.9353e-05, 2.0027e-05, 4.9353e-05, 5.9664e-05, 8.8930e-05, 4.9353e-05,
          5.8222e-04, 8.8930e-05, 8.8930e-05, 1.3065e-03, 1.2112e-04, 1.2112e-04,
          2.0027e-05, 2.6298e-04, 2.0027e-05, 4.9353e-05, 2.0027e-05, 2.5368e-04,
          8.8930e-05, 2.0027e-05, 2.0027e-05, 2.0027e-05, 2.0027e-05, 2.0027e-05,
          2.3353e-04, 4.9353e-05, 2.0027e-05, 2.0027e-05, 6.9201e-05, 8.8930e-05,
          3.9005e-04, 4.6182e-04, 3.9005e-04, 8.8930e-05, 4.9353e-05, 2.0027e-05,
          2.0027e-05, 4.9353e-05, 2.4486e-04, 2.0027e-05, 1.2112e-04, 2.0027e-05,
          4.9353e-05, 2.0027e-05, 4.9353e-05, 4.9353e-05, 2.0027e-05, 8.8930e-05,
          2.0027e-05, 4.9353e-05, 2.0027e-05, 8.8930e-05, 4.9353e-05, 1.6069e-04,
          1.0767e-03, 1.7297e-04, 2.0027e-05, 2.0027e-05, 3.6073e-04, 2.0027e-05,
          2.0027e-05, 2.0027e-05, 2.0027e-05, 4.9353e-05, 4.9353e-05, 2.0027e-05,
          4.9353e-05, 5.9664e-05, 4.9353e-05, 5.9664e-05, 4.9353e-05, 4.9353e-05,
          4.8351e-04, 8.8930e-05, 4.9353e-05, 2.0027e-05, 1.2112e-04, 8.8930e-05,
          1.2112e-04, 8.8930e-05, 2.0027e-05, 8.8930e-05, 2.5487e-04, 2.0027e-05,
          4.9353e-05, 5.9664e-05, 4.9353e-05, 2.0027e-05, 2.4486e-04, 8.8930e-05,
          5.9223e-04, 4.9353e-05, 8.8930e-05, 1.6069e-04, 4.9353e-05, 4.1080e-04,
          1.0223e-03, 6.2227e-04, 6.5136e-04, 2.0027e-05, 2.6488e-04, 4.9353e-05,
          3.0446e-04, 4.9353e-05, 2.0027e-05, 3.2663e-04, 2.0027e-05, 1.2112e-04,
          2.0027e-05, 2.0027e-05, 4.9353e-05, 2.0027e-05, 6.2227e-04, 2.0027e-05,
          1.0290e-03, 8.8930e-05, 8.8930e-05, 2.0027e-05, 2.0027e-05, 4.9353e-05,
          4.7016e-04, 4.9353e-05, 8.8930e-05, 1.2112e-04, 8.8930e-05, 1.8084e-04,
          4.9353e-05, 8.6308e-04, 2.0027e-05, 2.0027e-05, 2.0027e-05, 2.0027e-05,
          1.3340e-04, 2.0528e-04, 4.9353e-05, 4.9353e-05, 4.9353e-05, 5.9664e-05,
          2.0027e-05, 1.7297e-04, 2.0528e-04, 3.6764e-04, 1.3340e-04, 1.3340e-04,
          4.9353e-05, 1.3340e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.3438e-03, 5.9021e-02, 1.3542e-02, 2.3102e-02, 4.1138e-02,
          4.4403e-02, 1.2512e-02, 2.0695e-03, 1.3742e-03, 8.1482e-03, 1.5717e-02,
          5.6038e-03, 2.0046e-03, 9.8419e-04, 2.5043e-03, 1.9588e-03, 3.0289e-03,
          2.2221e-03, 1.0195e-03, 2.4402e-04, 3.8123e-04, 3.0565e-04, 3.4313e-03,
          1.8387e-02, 2.2793e-03, 1.0170e-02, 7.5951e-03, 5.9843e-04, 3.7251e-03,
          1.6775e-03, 4.1161e-03, 1.1940e-03, 4.1504e-03, 6.5613e-03, 1.4639e-03,
          3.2139e-03, 1.0738e-03, 3.5820e-03, 3.6678e-03, 4.0317e-04, 3.7746e-03,
          8.8549e-04, 7.0000e-04, 7.2718e-04, 8.2636e-04, 3.2921e-03, 3.2444e-03,
          4.7112e-04, 3.5739e-04, 3.2310e-03, 2.0237e-03, 3.1719e-03, 2.0561e-03,
          6.1321e-04, 1.6184e-03, 5.6696e-04, 2.1000e-03, 2.9373e-03, 2.9030e-03,
          1.6365e-03, 1.2388e-03, 7.9966e-04, 1.1911e-03, 2.1420e-03, 5.6229e-03,
          3.0975e-03, 3.7136e-03, 2.8667e-03, 5.8270e-04, 5.0306e-04, 7.9870e-04,
          6.9809e-04, 1.8280e-02, 8.8120e-04, 3.4828e-03, 5.8508e-04, 6.9618e-04,
          1.2264e-03, 2.4185e-03, 4.4556e-03, 7.0267e-03, 4.6806e-03, 1.2770e-03,
          8.1253e-04, 3.4828e-03, 6.3658e-04, 1.0788e-02, 2.8858e-03, 1.6441e-03,
          2.1439e-02, 9.7809e-03, 5.7364e-04, 1.1856e-02, 5.2691e-04, 6.3992e-04,
          8.0013e-04, 4.4751e-04, 2.9030e-03, 3.0537e-03, 1.9360e-03, 3.6278e-03,
          2.7786e-02, 2.4948e-03, 5.2757e-03, 3.5858e-03, 4.4060e-03, 1.1826e-03,
          2.7447e-03, 1.8511e-03, 1.2589e-02, 7.5102e-04, 2.0523e-03, 4.7569e-03,
          1.5831e-03, 1.3428e-03, 4.1437e-04, 4.9934e-03, 8.0013e-04, 1.1797e-03,
          2.1458e-03, 7.0572e-03, 1.2331e-03, 1.5173e-03, 7.3128e-03, 2.0084e-03,
          1.3199e-03, 2.0103e-03, 3.0537e-03, 6.1340e-03, 4.8790e-03, 3.3245e-03,
          3.9101e-03, 3.6411e-03, 1.7080e-03, 1.7805e-03, 9.4070e-03, 7.0667e-04,
          6.1531e-03, 8.4352e-04, 2.0142e-03, 5.3673e-03, 5.8441e-03, 3.3379e-03,
          2.1420e-03, 1.7300e-03, 2.3785e-03, 3.2368e-03, 3.7384e-03, 5.9128e-03,
          1.5717e-02, 5.4054e-03, 2.1229e-03, 2.4643e-03, 1.5173e-03, 5.7793e-03,
          1.0239e-02, 4.4785e-03, 5.6839e-03, 1.3180e-03, 3.3188e-03, 1.6127e-03,
          4.5090e-03, 6.9389e-03, 6.3362e-03, 5.7716e-03, 2.6703e-03, 4.9171e-03,
          2.7283e-02, 6.6681e-03, 4.8943e-03, 3.0231e-03, 4.7646e-03, 7.2746e-03,
          3.3245e-03, 2.1610e-03, 2.1496e-03, 2.2659e-03, 5.4512e-03, 7.7553e-03,
          3.1357e-03, 2.8038e-03, 6.3286e-03, 6.3820e-03, 2.1572e-03, 1.6298e-03,
          1.4595e-02, 1.0786e-03, 3.7861e-03, 2.9449e-03, 4.0588e-03, 6.6795e-03,
          7.8659e-03, 1.5326e-03, 6.0654e-03, 8.4076e-03, 3.2104e-02, 2.0325e-05,
          2.0325e-05, 1.9073e-04, 1.0383e-04, 2.0325e-05, 2.0325e-05, 2.0325e-05,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 6.0320e-05, 2.0325e-05, 3.0637e-04,
          6.0320e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 6.0320e-05, 2.0325e-05,
          2.0325e-05, 1.0383e-04, 2.0325e-05, 6.0320e-05, 6.0320e-05, 1.9073e-04,
          6.0320e-05, 3.1686e-04, 2.0325e-05, 2.0325e-05, 2.3890e-04, 6.0320e-05,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 3.3259e-04,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 6.0320e-05, 6.0320e-05, 1.6320e-04,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 6.3837e-05, 1.0383e-04, 1.9073e-04,
          3.1686e-04, 2.0325e-05, 2.0325e-05, 1.9073e-04, 2.0325e-05, 6.3837e-05,
          2.0325e-05, 3.1686e-04, 2.0325e-05, 2.0325e-05, 1.0383e-04, 2.0325e-05,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05,
          6.0320e-05, 2.0325e-05, 6.0320e-05, 1.0383e-04, 1.0383e-04, 6.3837e-05,
          1.0383e-04, 6.0320e-05, 1.0383e-04, 2.0325e-05, 2.4438e-04, 2.0325e-05,
          2.0325e-05, 6.0320e-05, 3.1686e-04, 1.0383e-04, 2.0325e-05, 6.0320e-05,
          6.0320e-05, 2.0325e-05, 2.0325e-05, 2.0325e-05, 2.8419e-04, 1.5068e-04,
          1.6320e-04, 2.0325e-05, 1.9073e-04, 2.1958e-04, 2.0325e-05, 2.0325e-05,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 1.9073e-04, 2.0325e-05, 2.0325e-05,
          2.0325e-05, 1.0383e-04, 6.0320e-05, 1.9741e-04, 2.0325e-05, 2.0325e-05,
          1.0383e-04, 2.4438e-04, 1.5068e-04, 1.9741e-04, 2.0325e-05, 2.0325e-05,
          1.9073e-04, 6.0320e-05, 4.9734e-04, 1.0383e-04, 2.0325e-05, 6.3837e-05,
          6.3837e-05, 2.0325e-05, 2.4438e-04, 1.9073e-04, 3.1686e-04, 2.0325e-05,
          1.9073e-04, 1.9741e-04, 1.5068e-04, 2.0325e-05, 2.0325e-05, 1.0383e-04,
          2.0325e-05, 4.2462e-04, 1.9073e-04, 6.0320e-05, 2.0325e-05, 1.0723e-04,
          1.5402e-04, 6.0320e-05, 2.8419e-04, 5.4073e-04, 2.0325e-05, 1.9073e-04,
          2.0325e-05, 6.0320e-05, 2.8419e-04, 6.0320e-05, 6.2275e-04, 4.1056e-04,
          4.5395e-04, 6.0320e-05, 2.0325e-05, 2.0325e-05, 1.6320e-04, 1.6320e-04,
          1.4858e-03, 6.3837e-05, 2.0325e-05, 2.0325e-05, 1.0383e-04, 6.3837e-05,
          5.2738e-04, 5.2738e-04, 1.1396e-04, 2.0325e-05, 6.0320e-05, 2.0325e-05,
          1.0383e-04, 2.8419e-04, 6.0320e-05, 6.0320e-05, 2.0325e-05, 2.0325e-05,
          2.0325e-05, 2.0325e-05, 2.0325e-05, 6.0320e-05, 2.5678e-04, 2.0325e-05,
          1.2503e-03, 1.0383e-04, 2.0325e-05, 2.0325e-05, 2.0087e-04, 2.9087e-04,
          1.5402e-04, 5.4073e-04, 1.9073e-04, 6.0320e-05, 2.0325e-05, 2.5630e-04,
          3.8457e-04, 1.2531e-03, 2.0325e-05, 2.4438e-04, 6.0272e-04, 2.0325e-05,
          1.0383e-04, 2.0325e-05, 2.0325e-05, 6.3837e-05, 3.3355e-04, 1.0383e-04,
          2.0325e-05, 1.0383e-04, 6.3837e-05, 2.8920e-04, 6.3837e-05, 2.1768e-04,
          6.3837e-05, 6.3837e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4588e-03, 5.7068e-02, 1.3451e-02, 1.8585e-02, 4.0894e-02,
          4.2786e-02, 9.9716e-03, 1.2646e-03, 1.0567e-03, 4.3793e-03, 4.3945e-03,
          2.2221e-03, 2.1667e-03, 7.9536e-04, 4.5128e-03, 1.1787e-03, 4.4937e-03,
          1.9083e-03, 6.7472e-04, 3.7670e-04, 4.6825e-04, 3.3498e-04, 2.4529e-03,
          5.1994e-03, 3.3169e-03, 8.0719e-03, 7.2174e-03, 6.5660e-04, 5.5466e-03,
          1.1673e-03, 3.2005e-03, 1.5993e-03, 8.4152e-03, 4.8943e-03, 2.0924e-03,
          7.5150e-03, 1.2188e-03, 5.1422e-03, 6.0387e-03, 5.2643e-04, 3.8891e-03,
          7.2479e-04, 6.5565e-04, 5.7268e-04, 7.8726e-04, 4.3678e-03, 1.0094e-02,
          4.3511e-04, 4.0841e-04, 2.8801e-03, 3.0613e-03, 2.7390e-03, 2.1858e-03,
          6.7472e-04, 1.6298e-03, 8.9073e-04, 2.5368e-03, 2.1420e-03, 2.7294e-03,
          1.9855e-03, 1.7891e-03, 1.1740e-03, 2.1305e-03, 1.1816e-03, 1.6510e-02,
          8.5306e-04, 7.4577e-04, 1.8864e-03, 5.0402e-04, 3.7289e-04, 4.4727e-04,
          5.8651e-04, 9.6207e-03, 1.0462e-03, 3.2673e-03, 5.9748e-04, 7.4720e-04,
          1.7681e-03, 2.3422e-03, 1.6966e-03, 8.2779e-03, 1.9932e-03, 1.5144e-03,
          1.1501e-03, 4.0398e-03, 8.2779e-04, 2.3880e-03, 6.3801e-04, 5.9605e-04,
          5.0392e-03, 4.8065e-03, 4.3917e-04, 3.0041e-03, 6.6853e-04, 6.7043e-04,
          1.1177e-03, 6.2418e-04, 4.6997e-03, 5.1041e-03, 1.9722e-03, 3.5496e-03,
          7.9651e-03, 9.8133e-04, 2.0180e-03, 8.0566e-03, 1.1772e-02, 1.9970e-03,
          3.4733e-03, 1.8797e-03, 5.9319e-03, 5.1355e-04, 7.8630e-04, 1.6460e-03,
          1.8826e-03, 2.4776e-03, 5.8508e-04, 1.6159e-02, 8.2207e-04, 1.3924e-03,
          2.5826e-03, 1.0437e-02, 9.4652e-04, 9.8515e-04, 1.3603e-02, 1.0710e-03,
          1.9264e-03, 1.3428e-03, 2.0370e-03, 1.8402e-02, 4.3335e-03, 4.0092e-03,
          1.4009e-03, 8.8167e-04, 9.5654e-04, 5.6934e-04, 3.0231e-03, 5.6124e-04,
          1.3947e-02, 1.1024e-03, 1.4668e-03, 4.2305e-03, 6.8817e-03, 3.9005e-03,
          1.7118e-03, 1.2569e-03, 2.6798e-03, 3.0880e-03, 1.5545e-03, 1.4923e-02,
          1.9119e-02, 1.8753e-02, 4.4594e-03, 6.0844e-03, 2.7008e-03, 7.4806e-03,
          5.2490e-03, 3.6469e-03, 1.0996e-03, 7.2479e-04, 9.4223e-03, 2.8820e-03,
          1.3779e-02, 8.4915e-03, 8.3466e-03, 6.9008e-03, 4.0932e-03, 3.8033e-03,
          6.4278e-03, 5.5275e-03, 9.7046e-03, 5.5618e-03, 5.1994e-03, 2.4521e-02,
          6.4430e-03, 3.2101e-03, 4.8828e-03, 3.3989e-03, 9.9564e-03, 4.2267e-03,
          4.2534e-03, 4.4136e-03, 1.1185e-02, 3.4447e-03, 1.4353e-03, 2.2182e-03,
          3.3131e-03, 9.8419e-04, 1.9588e-03, 3.1872e-03, 1.9159e-03, 1.4427e-02,
          4.9248e-03, 1.8911e-03, 1.6617e-02, 1.0086e-02, 7.8659e-03, 2.5467e-02,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 7.3195e-05, 7.3195e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.0324e-04, 7.3195e-05, 1.6260e-04,
          1.6260e-04, 1.3828e-05, 1.3828e-05, 1.0324e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 7.3195e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.0324e-04,
          1.3828e-05, 2.3806e-04, 1.3828e-05, 7.3195e-05, 1.3828e-05, 1.3828e-05,
          7.3195e-05, 1.3828e-05, 2.3806e-04, 1.3828e-05, 1.3828e-05, 3.2735e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.0324e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.0324e-04, 3.8695e-04,
          1.3828e-05, 7.3195e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 3.4857e-04,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.0324e-04, 1.6260e-04, 1.3828e-05,
          1.6260e-04, 7.3195e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 7.6103e-04, 1.6260e-04, 1.3828e-05, 1.6260e-04,
          1.6260e-04, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.0324e-04, 1.3828e-05, 7.3195e-05, 4.9591e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.6260e-04, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 2.3806e-04, 1.0324e-04, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 7.3195e-05, 8.2016e-04, 8.2016e-04, 1.3828e-05, 1.3828e-05,
          8.2016e-04, 7.6103e-04, 7.3195e-05, 1.3828e-05, 1.3828e-05, 1.6260e-04,
          1.3828e-05, 2.8920e-04, 1.6260e-04, 1.3828e-05, 1.3828e-05, 1.3828e-05,
          1.3828e-05, 1.3828e-05, 1.6260e-04, 5.7268e-04, 7.3195e-05, 3.8695e-04,
          1.3828e-05, 1.3828e-05, 7.3195e-05, 3.6407e-04, 2.6941e-04, 1.0958e-03,
          1.0958e-03, 4.3797e-04, 1.3828e-05, 1.3828e-05, 1.6260e-04, 1.6260e-04,
          2.6941e-04, 7.3195e-05, 1.3828e-05, 1.3828e-05, 1.6260e-04, 1.3828e-05,
          3.6407e-04, 1.6260e-04, 1.3828e-05, 1.3828e-05, 7.3195e-05, 1.3828e-05,
          1.3828e-05, 6.6233e-04, 1.3828e-05, 7.3195e-05, 1.3828e-05, 1.3828e-05,
          3.4857e-04, 1.3828e-05, 1.3828e-05, 1.3828e-05, 1.6260e-04, 1.3828e-05,
          4.3797e-04, 1.6260e-04, 7.3195e-05, 1.3828e-05, 2.8920e-04, 1.2264e-03,
          1.6260e-04, 9.0599e-04, 6.6233e-04, 1.3828e-05, 1.3828e-05, 2.3806e-04,
          1.2999e-03, 5.4979e-04, 1.3828e-05, 2.9731e-04, 5.7268e-04, 1.3828e-05,
          7.3195e-05, 1.3828e-05, 3.1018e-04, 1.3828e-05, 7.3195e-05, 1.3828e-05,
          1.3828e-05, 7.3195e-05, 7.3195e-05, 3.2234e-04, 1.3828e-05, 9.5224e-04,
          7.3195e-05, 7.3195e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3520e-03, 4.9011e-02, 2.0203e-02, 1.4542e-02, 8.0505e-02,
          3.3722e-02, 1.4206e-02, 1.0290e-03, 8.8024e-04, 1.7633e-03, 3.0937e-03,
          1.1759e-03, 9.5892e-04, 1.0900e-03, 1.7872e-03, 7.9727e-04, 1.1425e-03,
          6.5756e-04, 4.9162e-04, 1.1892e-03, 8.9359e-04, 6.5088e-04, 8.1348e-04,
          3.1357e-03, 2.7370e-03, 1.0818e-02, 2.6302e-03, 1.4133e-03, 1.0338e-03,
          1.0490e-03, 1.6584e-03, 6.4373e-04, 1.9798e-03, 7.5102e-04, 1.4439e-03,
          4.2381e-03, 3.8204e-03, 1.3161e-03, 1.1845e-03, 1.1978e-03, 1.8253e-03,
          1.8511e-03, 1.3638e-03, 1.7338e-03, 1.1349e-03, 8.0681e-04, 2.0485e-03,
          5.8174e-04, 9.8228e-04, 7.8506e-03, 2.9716e-03, 1.8072e-03, 1.0490e-03,
          1.9588e-03, 1.6937e-03, 4.3297e-03, 5.7373e-03, 6.3667e-03, 8.6746e-03,
          9.1705e-03, 2.2964e-03, 3.0785e-03, 1.9703e-03, 1.5955e-03, 3.1471e-03,
          1.5564e-03, 6.0511e-04, 7.4625e-04, 1.3342e-03, 1.3714e-03, 1.9684e-03,
          2.0561e-03, 2.5005e-03, 3.5877e-03, 2.5024e-03, 1.8549e-03, 1.3123e-03,
          3.2787e-03, 6.5269e-03, 3.0289e-03, 3.8185e-03, 3.3474e-03, 7.9498e-03,
          8.6212e-03, 3.0136e-03, 2.4796e-03, 1.0262e-03, 6.3276e-04, 1.1520e-03,
          1.7509e-03, 1.8044e-03, 1.2779e-03, 1.4439e-03, 3.2539e-03, 3.2005e-03,
          1.9207e-03, 2.5024e-03, 1.3704e-03, 3.3188e-03, 8.9722e-03, 5.2338e-03,
          2.9640e-03, 1.1539e-03, 1.3752e-03, 9.4223e-03, 4.7951e-03, 3.1929e-03,
          3.0861e-03, 2.4776e-03, 2.3232e-03, 1.6870e-03, 1.5869e-03, 9.1362e-04,
          1.9703e-03, 2.3060e-03, 1.3247e-03, 3.7060e-03, 4.0817e-03, 3.7212e-03,
          3.5934e-03, 4.4594e-03, 3.5038e-03, 8.8692e-04, 3.3741e-03, 1.9131e-03,
          7.9575e-03, 5.0049e-03, 5.0621e-03, 5.8594e-03, 1.3952e-03, 3.4962e-03,
          2.3823e-03, 8.0585e-04, 4.4942e-04, 6.4516e-04, 1.4429e-03, 2.4872e-03,
          1.6689e-03, 1.5602e-03, 1.2589e-03, 2.5234e-03, 1.7958e-03, 2.9068e-03,
          4.3297e-03, 1.2779e-03, 2.6951e-03, 4.8065e-03, 2.8019e-03, 9.7656e-03,
          8.8043e-03, 9.1248e-03, 2.7039e-02, 3.9940e-03, 6.8550e-03, 4.4632e-03,
          1.7042e-03, 2.5349e-03, 5.4169e-04, 2.7370e-03, 4.9171e-03, 1.7185e-03,
          2.4261e-03, 1.5411e-03, 1.2913e-03, 1.8368e-03, 8.7128e-03, 3.1509e-03,
          2.1782e-03, 2.5749e-03, 3.1548e-03, 1.7490e-03, 1.7433e-03, 4.4937e-03,
          3.0060e-02, 4.1107e-02, 5.6946e-02, 5.9204e-03, 3.3455e-03, 1.3905e-03,
          1.3294e-03, 1.6756e-03, 2.0332e-03, 1.6794e-03, 1.0357e-03, 1.3294e-03,
          1.3075e-03, 2.0103e-03, 1.8368e-03, 9.1782e-03, 3.1471e-03, 2.5578e-03,
          1.6851e-03, 7.0801e-03, 2.6894e-03, 2.0790e-03, 1.7071e-03, 3.2864e-03,
          8.7158e-02, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 2.7132e-04,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 1.1349e-04, 1.1349e-04, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 1.1349e-04,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 1.9598e-04,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 1.9598e-04, 1.1349e-04,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 2.7132e-04, 1.1349e-04, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 2.7132e-04, 2.7132e-04, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 2.7132e-04, 1.1349e-04, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05,
          1.1349e-04, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 1.1349e-04, 1.1349e-04, 1.1349e-04, 3.8147e-05, 3.8147e-05,
          1.1349e-04, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05, 1.1349e-04,
          3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05, 1.1349e-04,
          3.8147e-05, 3.8147e-05, 2.7132e-04, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          1.1349e-04, 3.8147e-05, 3.8147e-05, 3.8147e-05, 4.7207e-04, 1.9598e-04,
          3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05, 1.1349e-04, 1.1349e-04,
          6.7663e-04, 4.7207e-04, 3.8147e-05, 3.8147e-05, 1.1349e-04, 3.8147e-05,
          3.8147e-05, 1.1349e-04, 3.8147e-05, 1.1349e-04, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05, 4.7207e-04, 3.8147e-05,
          3.1424e-04, 1.1349e-04, 1.1349e-04, 3.8147e-05, 3.8147e-05, 3.1757e-04,
          1.1349e-04, 3.8147e-05, 1.1349e-04, 6.7091e-04, 3.8147e-05, 3.1757e-04,
          1.1349e-04, 5.1880e-04, 3.8147e-05, 3.8147e-05, 3.8147e-05, 3.8147e-05,
          1.1349e-04, 1.1349e-04, 3.1424e-04, 2.9588e-04, 3.8147e-05, 3.8147e-05,
          3.8147e-05, 3.8147e-05, 3.8147e-05, 2.9683e-04, 2.2161e-04, 3.1424e-04,
          3.8147e-05, 3.8147e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.1068e-03, 4.7760e-02, 1.9653e-02, 1.7151e-02, 8.9233e-02,
          3.2288e-02, 1.6541e-02, 1.0509e-03, 9.8801e-04, 1.0290e-03, 1.6127e-03,
          7.4482e-04, 2.5139e-03, 4.6158e-03, 1.4820e-03, 9.2554e-04, 6.1455e-03,
          3.2024e-03, 5.5933e-04, 4.1771e-03, 2.5520e-03, 2.1248e-03, 1.9150e-03,
          1.6527e-03, 1.5736e-03, 1.2192e-02, 4.0359e-03, 4.6921e-03, 4.4098e-03,
          6.4163e-03, 1.5450e-03, 2.8400e-03, 2.4242e-03, 8.0729e-04, 1.7586e-03,
          2.9602e-03, 1.8473e-03, 1.9951e-03, 1.5669e-03, 4.7493e-03, 1.8511e-03,
          1.5717e-03, 1.1253e-03, 1.3790e-03, 8.1205e-04, 3.2425e-03, 2.7180e-03,
          1.8997e-03, 2.7161e-03, 1.0567e-02, 2.7332e-03, 1.4143e-03, 9.7942e-04,
          1.4496e-03, 1.3638e-03, 2.8496e-03, 2.2812e-03, 1.9197e-03, 2.1515e-03,
          2.5597e-03, 1.6571e-02, 3.1338e-03, 3.4866e-03, 1.1826e-03, 2.8038e-03,
          1.3971e-03, 5.0163e-04, 7.3099e-04, 1.1415e-03, 1.2589e-03, 1.2655e-03,
          1.6661e-03, 1.9312e-03, 2.3937e-03, 1.9588e-03, 1.3485e-03, 1.0138e-03,
          2.1057e-03, 2.7657e-03, 1.4668e-03, 1.4448e-03, 1.2550e-03, 1.6088e-03,
          2.2717e-03, 3.3245e-03, 2.8362e-03, 7.0190e-04, 7.0763e-04, 9.8419e-04,
          1.1692e-03, 9.9945e-04, 1.0881e-03, 9.2459e-04, 1.3781e-03, 1.8673e-03,
          5.9090e-03, 1.6136e-03, 1.4734e-03, 4.5776e-03, 2.7351e-03, 1.9026e-03,
          1.4906e-03, 8.3828e-04, 7.4148e-04, 2.4433e-03, 2.3174e-03, 3.1677e-02,
          3.3016e-03, 2.7866e-03, 1.6003e-03, 1.4029e-03, 1.3638e-03, 8.9693e-04,
          9.3222e-04, 2.6283e-03, 1.0117e-02, 3.2406e-03, 2.5978e-03, 2.4643e-02,
          3.0502e-02, 2.7180e-03, 1.6661e-03, 7.4434e-04, 2.3613e-03, 9.9659e-04,
          2.9488e-03, 1.6785e-03, 2.0218e-03, 3.5458e-03, 2.0523e-03, 4.0398e-03,
          2.1877e-03, 8.1873e-04, 4.9591e-04, 6.1226e-04, 9.2602e-04, 2.0390e-03,
          6.4850e-03, 1.3062e-02, 3.2463e-03, 3.2501e-03, 2.4357e-03, 1.7853e-02,
          2.1496e-03, 6.7139e-04, 1.4133e-03, 1.6432e-03, 1.5621e-03, 3.7289e-03,
          2.8877e-03, 3.4618e-03, 4.2725e-03, 7.1716e-03, 6.5613e-03, 5.0316e-03,
          1.2598e-03, 2.0084e-03, 5.2023e-04, 2.4586e-03, 4.2839e-03, 1.4030e-02,
          3.0174e-03, 1.7891e-03, 5.6648e-03, 2.0676e-03, 6.4468e-03, 1.7824e-03,
          1.3494e-03, 1.3504e-03, 2.0695e-03, 1.4591e-03, 6.6490e-03, 4.7607e-03,
          3.4561e-03, 4.6425e-03, 8.9951e-03, 8.1787e-03, 2.9812e-03, 2.0733e-03,
          1.2102e-03, 1.7653e-03, 2.1477e-03, 1.8568e-03, 1.5011e-03, 1.4563e-03,
          1.0414e-03, 1.0147e-02, 2.2812e-03, 4.6501e-03, 2.0943e-03, 2.4948e-03,
          1.7977e-03, 3.2787e-03, 2.7180e-03, 8.0185e-03, 1.0653e-03, 2.6741e-03,
          7.4348e-03, 1.2561e-01, 2.0742e-05, 5.0640e-04, 1.9884e-04, 2.0742e-05,
          2.0742e-05, 1.9884e-04, 2.0742e-05, 1.9884e-04, 3.2592e-04, 8.1062e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0826e-04, 1.4782e-04, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 1.4782e-04, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 8.1062e-05, 8.1062e-05, 1.4782e-04, 1.4782e-04,
          8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 1.9884e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          8.1062e-05, 8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 1.4782e-04, 2.0742e-05, 2.0742e-05,
          8.1062e-05, 8.1062e-05, 1.4782e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 8.1062e-05,
          1.4782e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.4782e-04, 2.0742e-05, 8.1062e-05, 1.9884e-04, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 8.1062e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0826e-04, 2.0742e-05, 8.1062e-05, 8.1062e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0826e-04,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 8.1062e-05, 8.1062e-05, 8.1062e-05,
          8.1062e-05, 8.1778e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 6.9046e-04, 2.0742e-05,
          2.0742e-05, 1.4782e-04, 2.0742e-05, 2.0826e-04, 2.0742e-05, 2.0742e-05,
          2.0742e-05, 8.1062e-05, 8.1062e-05, 2.0742e-05, 2.0826e-04, 1.4782e-04,
          2.0742e-05, 2.0742e-05, 2.0826e-04, 1.9884e-04, 2.0742e-05, 8.1062e-05,
          3.2592e-04, 2.0826e-04, 2.0742e-05, 2.0742e-05, 8.1062e-05, 2.0742e-05,
          2.0826e-04, 2.0742e-05, 2.0742e-05, 6.9046e-04, 2.0742e-05, 2.0742e-05,
          8.1062e-05, 8.1062e-05, 8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          8.1062e-05, 2.0742e-05, 3.2592e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.9884e-04, 2.0742e-05, 2.0742e-05, 2.0742e-05, 5.1022e-04, 8.1062e-05,
          2.0742e-05, 8.1062e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05, 2.0742e-05,
          1.9884e-04, 8.1062e-05, 1.9884e-04, 1.9884e-04, 8.1062e-05, 8.1062e-05,
          2.0742e-05, 8.1062e-05, 1.9884e-04, 1.9884e-04, 1.9884e-04, 1.9884e-04,
          1.9884e-04, 2.0742e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6730e-03, 5.7709e-02, 1.4870e-02, 1.4809e-02, 4.8218e-02,
          4.8279e-02, 1.0849e-02, 1.0958e-03, 1.0490e-03, 3.7498e-03, 3.8452e-03,
          1.6994e-03, 1.7376e-03, 6.7854e-04, 4.4098e-03, 1.3313e-03, 1.9026e-03,
          1.1644e-03, 8.9693e-04, 4.5300e-04, 7.8678e-04, 5.9795e-04, 2.1191e-03,
          5.0354e-03, 5.6725e-03, 8.7433e-03, 4.5395e-03, 7.0858e-04, 2.2793e-03,
          8.7023e-04, 2.0847e-03, 7.9727e-04, 9.2239e-03, 3.8204e-03, 2.0714e-03,
          9.6359e-03, 1.2217e-03, 9.6283e-03, 6.9199e-03, 5.4407e-04, 8.1253e-03,
          1.9779e-03, 1.5392e-03, 1.2131e-03, 1.3313e-03, 2.2087e-03, 1.0277e-02,
          5.0306e-04, 6.3944e-04, 4.4212e-03, 2.5387e-03, 1.4544e-03, 1.6232e-03,
          5.8603e-04, 9.2697e-04, 7.1669e-04, 1.7834e-03, 1.6527e-03, 1.8396e-03,
          1.4944e-03, 1.9264e-03, 2.7122e-03, 5.6496e-03, 3.6736e-03, 1.7517e-02,
          1.8139e-03, 1.2102e-03, 2.8820e-03, 9.3508e-04, 8.1205e-04, 7.5340e-04,
          8.8358e-04, 7.6027e-03, 8.1730e-04, 1.6775e-03, 4.7612e-04, 4.9686e-04,
          8.8215e-04, 1.6975e-03, 1.3494e-03, 4.9248e-03, 1.2293e-03, 1.6603e-03,
          1.3762e-03, 5.2032e-03, 1.9741e-03, 2.8229e-03, 1.1797e-03, 1.1654e-03,
          5.2490e-03, 5.7526e-03, 1.1501e-03, 3.0251e-03, 1.3218e-03, 1.0176e-03,
          8.3685e-04, 5.1737e-04, 2.1610e-03, 4.8866e-03, 1.6279e-03, 2.1305e-03,
          3.5648e-03, 7.5054e-04, 8.8120e-04, 8.0872e-03, 1.0010e-02, 2.9163e-03,
          4.5395e-03, 4.5319e-03, 9.3994e-03, 1.5802e-03, 1.4277e-03, 2.0981e-03,
          3.7594e-03, 5.2643e-03, 1.7519e-03, 1.2383e-02, 1.4868e-03, 1.6146e-03,
          2.2411e-03, 8.1711e-03, 8.2445e-04, 7.7295e-04, 9.5367e-03, 6.7616e-04,
          1.2932e-03, 9.2411e-04, 1.2417e-03, 1.2733e-02, 3.2749e-03, 4.6577e-03,
          2.5120e-03, 1.3494e-03, 1.2646e-03, 1.2007e-03, 4.6768e-03, 1.8368e-03,
          9.1324e-03, 2.6150e-03, 1.8272e-03, 3.5534e-03, 4.4518e-03, 2.3479e-03,
          1.3590e-03, 1.1253e-03, 1.7328e-03, 1.7281e-03, 1.2197e-03, 1.1360e-02,
          7.3929e-03, 9.4833e-03, 2.7924e-03, 8.9493e-03, 3.4885e-03, 1.5823e-02,
          7.4043e-03, 6.8321e-03, 1.2112e-03, 2.0905e-03, 1.8539e-02, 5.2872e-03,
          9.2545e-03, 8.4076e-03, 3.7975e-03, 5.1689e-03, 4.3068e-03, 2.3003e-03,
          2.8267e-03, 2.3766e-03, 6.7024e-03, 3.4122e-03, 3.1662e-03, 1.1360e-02,
          5.5389e-03, 2.7256e-03, 4.0550e-03, 1.0033e-02, 2.4643e-02, 4.2496e-03,
          4.9591e-03, 1.2161e-02, 1.6571e-02, 3.2177e-03, 1.9779e-03, 2.5406e-03,
          2.8915e-03, 1.9426e-03, 2.1782e-03, 2.3460e-03, 1.3752e-03, 7.9346e-03,
          2.0027e-03, 1.2474e-03, 6.0196e-03, 4.6425e-03, 2.8572e-03, 6.7978e-03,
          3.1700e-03, 4.0855e-03, 5.2765e-02, 1.0639e-04, 1.0639e-04, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05, 1.0639e-04, 1.7965e-04,
          9.5248e-05, 2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 1.0639e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.7965e-04,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          1.7965e-04, 1.7965e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05,
          2.1994e-05, 2.1994e-05, 9.5248e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05, 1.7965e-04,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.0639e-04,
          2.1994e-05, 1.7965e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05, 1.7965e-04, 2.1994e-05,
          1.7965e-04, 9.5248e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.0639e-04, 1.7965e-04, 2.1994e-05, 1.7965e-04,
          1.7965e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          9.5248e-05, 2.1994e-05, 1.7965e-04, 4.7588e-04, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 9.5248e-05,
          1.0639e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 4.0269e-04, 1.0639e-04, 1.0639e-04, 2.1994e-05, 2.1994e-05,
          1.0639e-04, 1.0639e-04, 1.0639e-04, 2.1994e-05, 2.1994e-05, 1.7965e-04,
          2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 9.5248e-05, 1.0639e-04, 1.0639e-04, 1.7965e-04,
          2.1994e-05, 2.1994e-05, 1.7965e-04, 1.0639e-04, 2.1994e-05, 1.0639e-04,
          8.9455e-04, 9.6798e-04, 2.1994e-05, 2.1994e-05, 1.7965e-04, 1.7965e-04,
          2.1994e-05, 1.0639e-04, 2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05,
          1.7965e-04, 1.7965e-04, 2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05,
          2.1994e-05, 1.7965e-04, 9.5248e-05, 1.0639e-04, 2.1994e-05, 2.1994e-05,
          9.6798e-04, 2.1994e-05, 2.1994e-05, 2.1994e-05, 1.7965e-04, 2.1994e-05,
          9.6798e-04, 1.7965e-04, 4.7588e-04, 1.0639e-04, 2.1994e-05, 1.0639e-04,
          1.7965e-04, 2.1994e-05, 1.7965e-04, 2.1994e-05, 1.0639e-04, 2.1994e-05,
          8.9455e-04, 1.1911e-03, 2.1994e-05, 1.0639e-04, 2.1994e-05, 2.1994e-05,
          1.0639e-04, 2.1994e-05, 4.0269e-04, 2.1994e-05, 1.0639e-04, 2.1994e-05,
          2.1994e-05, 2.1994e-05, 1.0639e-04, 4.2820e-04, 2.1994e-05, 4.0269e-04,
          1.0639e-04, 2.1994e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6038e-03, 5.3650e-02, 1.5190e-02, 1.7273e-02, 4.2969e-02,
          3.7170e-02, 1.2100e-02, 8.6260e-04, 8.2254e-04, 4.1046e-03, 8.1558e-03,
          2.9507e-03, 1.0328e-03, 1.0481e-03, 1.3676e-03, 9.3365e-04, 1.6022e-03,
          1.1635e-03, 8.6355e-04, 4.3964e-04, 6.3705e-04, 6.4564e-04, 1.8187e-03,
          1.1795e-02, 3.1834e-03, 9.5444e-03, 4.3640e-03, 6.5136e-04, 1.7729e-03,
          1.4915e-03, 1.5135e-03, 7.2479e-04, 1.6336e-03, 2.0695e-03, 8.1205e-04,
          2.2163e-03, 1.3905e-03, 3.8357e-03, 1.4372e-03, 6.6376e-04, 7.0648e-03,
          3.5992e-03, 2.5101e-03, 2.9869e-03, 1.5173e-03, 1.9207e-03, 2.7523e-03,
          7.6342e-04, 6.9237e-04, 3.7766e-03, 1.2608e-03, 1.3361e-03, 8.7452e-04,
          6.0081e-04, 8.8978e-04, 5.6601e-04, 1.9312e-03, 2.4357e-03, 3.2349e-03,
          1.8988e-03, 1.3742e-03, 1.7815e-03, 1.9083e-03, 8.7509e-03, 6.6872e-03,
          1.8768e-02, 8.0109e-03, 3.1204e-03, 1.8473e-03, 1.8625e-03, 2.0866e-03,
          1.3113e-03, 7.2823e-03, 8.2970e-04, 1.5984e-03, 6.2037e-04, 6.0844e-04,
          9.6083e-04, 2.4757e-03, 1.1139e-02, 4.0817e-03, 1.3817e-02, 1.4114e-03,
          1.0166e-03, 3.6068e-03, 1.4706e-03, 8.0490e-03, 1.0094e-02, 1.0323e-02,
          1.8097e-02, 9.4376e-03, 2.5425e-03, 9.0179e-03, 1.2722e-03, 1.1063e-03,
          7.1239e-04, 4.7398e-04, 1.2550e-03, 2.2125e-03, 3.2654e-03, 3.2940e-03,
          1.0773e-02, 4.4823e-03, 8.0795e-03, 3.9139e-03, 2.5158e-03, 1.6384e-03,
          2.9793e-03, 7.0801e-03, 1.5114e-02, 5.4474e-03, 1.1208e-02, 1.0315e-02,
          2.4948e-03, 4.0817e-03, 1.0633e-03, 4.6692e-03, 1.6098e-03, 1.1978e-03,
          1.8873e-03, 4.0741e-03, 1.9236e-03, 6.1941e-04, 3.3169e-03, 3.8891e-03,
          1.7042e-03, 5.7182e-03, 6.6872e-03, 2.5368e-03, 1.5688e-03, 3.3264e-03,
          1.3443e-02, 1.3412e-02, 1.0805e-03, 5.7602e-03, 1.5350e-02, 3.8395e-03,
          3.4943e-03, 2.2774e-03, 1.4658e-03, 3.0651e-03, 2.2240e-03, 2.0065e-03,
          3.3703e-03, 1.0719e-03, 2.2564e-03, 3.0422e-03, 8.2779e-03, 3.9062e-03,
          8.3923e-03, 2.9545e-03, 2.1935e-03, 4.0703e-03, 2.6150e-03, 1.2421e-02,
          1.4816e-02, 8.8882e-03, 2.6646e-03, 1.1116e-02, 8.0948e-03, 2.9964e-03,
          3.4428e-03, 4.9934e-03, 2.5749e-03, 2.2068e-03, 2.3880e-03, 3.5820e-03,
          8.4152e-03, 3.9482e-03, 2.7294e-03, 1.0462e-03, 2.0828e-03, 2.7943e-03,
          2.0962e-03, 1.9217e-03, 2.3842e-03, 1.0612e-02, 1.2383e-02, 3.1815e-03,
          3.6850e-03, 2.5406e-03, 5.6305e-03, 2.2507e-03, 1.5717e-03, 1.1854e-03,
          5.1422e-03, 2.1362e-03, 1.7033e-03, 3.3455e-03, 6.7368e-03, 2.3384e-03,
          1.6546e-03, 1.9779e-03, 1.7948e-03, 3.4580e-03, 6.3820e-03, 2.3289e-03,
          2.3003e-03, 2.6035e-03, 1.0109e-02, 3.7933e-02, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05, 1.6212e-05, 2.5511e-04,
          6.3539e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05, 1.6212e-05,
          1.6212e-05, 1.1313e-04, 1.6212e-05, 1.1313e-04, 1.1313e-04, 2.5511e-04,
          6.3539e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.1313e-04, 6.3539e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.1313e-04,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 6.5804e-05, 1.1313e-04, 2.0790e-04,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 6.5804e-05, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 6.3539e-05, 1.1313e-04, 1.1313e-04, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.1313e-04, 1.6212e-05, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 6.3539e-05, 6.5804e-05, 1.1313e-04, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.6212e-05, 7.0953e-04, 6.5804e-05,
          6.3539e-05, 1.6212e-05, 1.1313e-04, 1.1313e-04, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 1.1313e-04, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.1313e-04, 6.3539e-05, 1.1313e-04, 1.6212e-05, 1.6212e-05,
          6.5804e-05, 8.0442e-04, 6.5804e-05, 1.1313e-04, 1.6212e-05, 1.6212e-05,
          2.5511e-04, 6.5804e-05, 1.6212e-05, 6.3539e-05, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 2.0790e-04, 2.5511e-04, 1.6212e-05, 1.6212e-05,
          2.5511e-04, 1.1313e-04, 2.0790e-04, 1.6212e-05, 1.6212e-05, 1.1313e-04,
          1.6212e-05, 9.5510e-04, 2.5558e-04, 6.3539e-05, 1.6212e-05, 6.5804e-05,
          6.5804e-05, 1.6212e-05, 1.1313e-04, 6.5804e-05, 1.6212e-05, 2.5511e-04,
          1.6212e-05, 1.6212e-05, 2.5511e-04, 6.3539e-05, 6.3539e-05, 2.0790e-04,
          6.5804e-05, 6.3539e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05, 6.3539e-05,
          1.1313e-04, 6.5804e-05, 1.6212e-05, 1.6212e-05, 2.5558e-04, 6.5804e-05,
          7.0953e-04, 1.3757e-04, 1.6212e-05, 1.6212e-05, 6.3539e-05, 1.6212e-05,
          6.3539e-05, 2.0790e-04, 6.3539e-05, 6.5804e-05, 1.6212e-05, 1.6212e-05,
          1.6212e-05, 1.6212e-05, 1.6212e-05, 6.3539e-05, 1.3757e-04, 1.6212e-05,
          8.5163e-04, 3.5906e-04, 1.6212e-05, 6.5804e-05, 6.5804e-05, 4.5371e-04,
          1.1313e-04, 1.6212e-05, 2.5511e-04, 6.3539e-05, 1.6212e-05, 6.5804e-05,
          4.5371e-04, 1.0977e-03, 1.6212e-05, 2.0790e-04, 1.5819e-04, 1.6212e-05,
          1.6212e-05, 6.5804e-05, 1.6212e-05, 9.8169e-05, 6.5804e-05, 6.5804e-05,
          1.6212e-05, 6.5804e-05, 6.5804e-05, 3.0828e-04, 1.6212e-05, 6.5804e-05,
          1.6212e-05, 1.6212e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1308e-03, 6.4087e-02, 1.4442e-02, 1.8158e-02, 3.6316e-02,
          4.8981e-02, 1.1772e-02, 1.4458e-03, 8.9216e-04, 5.2071e-03, 1.2535e-02,
          3.3302e-03, 1.2016e-03, 8.5449e-04, 1.4954e-03, 1.4191e-03, 1.5802e-03,
          1.3943e-03, 9.9468e-04, 2.9159e-04, 4.0579e-04, 4.4847e-04, 2.5330e-03,
          1.5587e-02, 2.7542e-03, 9.5520e-03, 7.0496e-03, 4.7207e-04, 1.8129e-03,
          1.3037e-03, 2.4452e-03, 6.9189e-04, 1.8444e-03, 3.3398e-03, 9.7942e-04,
          1.7986e-03, 1.0004e-03, 3.6392e-03, 1.7643e-03, 4.1747e-04, 4.4022e-03,
          1.7977e-03, 1.3962e-03, 1.4744e-03, 9.5749e-04, 2.3460e-03, 2.2755e-03,
          6.2847e-04, 4.8566e-04, 3.3226e-03, 1.2074e-03, 1.7633e-03, 1.1501e-03,
          5.6267e-04, 1.0109e-03, 4.5085e-04, 1.3494e-03, 3.3894e-03, 2.3613e-03,
          1.4477e-03, 1.0853e-03, 1.5192e-03, 1.8406e-03, 4.3259e-03, 4.6272e-03,
          1.1497e-02, 9.1629e-03, 3.3722e-03, 8.5258e-04, 8.4066e-04, 1.8053e-03,
          1.0109e-03, 1.2230e-02, 6.7043e-04, 2.2182e-03, 6.2275e-04, 5.9605e-04,
          9.1600e-04, 1.4467e-03, 8.4991e-03, 4.1542e-03, 1.2108e-02, 9.9468e-04,
          6.7949e-04, 3.9940e-03, 1.0471e-03, 1.0757e-02, 8.2703e-03, 5.4970e-03,
          2.4231e-02, 9.3613e-03, 1.3905e-03, 1.1246e-02, 7.4625e-04, 7.7105e-04,
          5.6601e-04, 3.5143e-04, 1.8091e-03, 1.9188e-03, 1.4467e-03, 2.3041e-03,
          1.4091e-02, 4.5319e-03, 1.0612e-02, 2.4815e-03, 2.1038e-03, 1.0929e-03,
          3.1643e-03, 2.2907e-03, 1.3283e-02, 2.0924e-03, 7.4997e-03, 1.0048e-02,
          2.2507e-03, 1.7366e-03, 7.0572e-04, 3.5458e-03, 9.8991e-04, 8.4972e-04,
          1.5087e-03, 4.4212e-03, 1.0738e-03, 9.3555e-04, 3.5172e-03, 3.0499e-03,
          8.7404e-04, 5.0926e-03, 6.7673e-03, 2.2793e-03, 2.3460e-03, 3.6259e-03,
          1.0406e-02, 1.1391e-02, 1.4505e-03, 5.4893e-03, 1.4702e-02, 1.3275e-03,
          3.9368e-03, 1.7853e-03, 2.0027e-03, 4.6806e-03, 3.8605e-03, 1.8330e-03,
          1.9140e-03, 1.5106e-03, 1.8015e-03, 2.3899e-03, 6.6452e-03, 3.0975e-03,
          1.0590e-02, 2.4986e-03, 1.3638e-03, 3.2902e-03, 2.0866e-03, 6.0349e-03,
          1.5411e-02, 6.6490e-03, 4.5319e-03, 4.4899e-03, 5.7182e-03, 2.2144e-03,
          3.4218e-03, 7.7515e-03, 3.5744e-03, 3.5877e-03, 1.5440e-03, 4.6082e-03,
          1.1864e-02, 5.7716e-03, 2.6970e-03, 1.2970e-03, 2.6188e-03, 2.7809e-03,
          2.1076e-03, 1.7481e-03, 1.4038e-03, 4.3411e-03, 9.8267e-03, 5.5122e-03,
          3.7346e-03, 2.7027e-03, 5.7526e-03, 3.9864e-03, 2.1935e-03, 1.4534e-03,
          9.2850e-03, 1.7700e-03, 3.5553e-03, 1.8978e-03, 6.1150e-03, 2.9469e-03,
          2.9659e-03, 1.0424e-03, 2.4261e-03, 4.5166e-03, 1.3435e-02, 2.6073e-03,
          1.4400e-03, 1.5078e-03, 1.0483e-02, 2.0370e-02, 3.5431e-02, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05, 1.7047e-05, 3.6669e-04,
          6.1810e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05, 1.7047e-05,
          1.7047e-05, 2.1827e-04, 1.7047e-05, 6.1810e-05, 1.1605e-04, 2.1827e-04,
          6.1810e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.1605e-04, 6.1810e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.1605e-04,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7357e-04, 2.1827e-04, 1.7357e-04,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 7.1287e-05, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 6.1810e-05, 2.1827e-04, 2.1827e-04, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.1605e-04, 1.7047e-05, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 6.1810e-05, 7.1287e-05, 2.1827e-04, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.7047e-05, 2.1827e-04, 7.1287e-05,
          6.1810e-05, 1.7047e-05, 2.1827e-04, 1.1605e-04, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 1.1605e-04, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 6.1810e-05, 6.1810e-05, 2.1827e-04, 1.7047e-05, 1.7047e-05,
          7.1287e-05, 1.7357e-04, 1.7357e-04, 1.1605e-04, 1.7047e-05, 1.7047e-05,
          2.1827e-04, 1.7047e-05, 7.1287e-05, 6.1810e-05, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7357e-04, 2.1827e-04, 1.7047e-05, 1.7047e-05,
          3.6669e-04, 1.1605e-04, 1.7357e-04, 1.7047e-05, 1.7047e-05, 3.6669e-04,
          1.7047e-05, 1.0996e-03, 3.6669e-04, 6.1810e-05, 1.7047e-05, 7.1287e-05,
          7.1287e-05, 6.1810e-05, 1.1605e-04, 1.7357e-04, 1.7047e-05, 2.1827e-04,
          1.7047e-05, 1.7047e-05, 2.1827e-04, 1.7047e-05, 5.0688e-04, 1.7357e-04,
          1.7357e-04, 6.1810e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05, 6.1810e-05,
          5.0688e-04, 1.7357e-04, 1.7047e-05, 1.7047e-05, 2.1827e-04, 7.1287e-05,
          2.1827e-04, 1.1605e-04, 4.0793e-04, 1.7047e-05, 1.7047e-05, 1.7047e-05,
          6.1810e-05, 1.7357e-04, 6.1810e-05, 7.1287e-05, 1.7047e-05, 1.7047e-05,
          1.7047e-05, 1.7047e-05, 1.7047e-05, 6.1810e-05, 6.1810e-05, 1.7047e-05,
          1.0996e-03, 2.1827e-04, 1.7047e-05, 7.1287e-05, 7.1287e-05, 4.6968e-04,
          6.1810e-05, 1.7357e-04, 2.1827e-04, 1.7047e-05, 1.7047e-05, 7.1287e-05,
          4.6968e-04, 1.0996e-03, 1.7047e-05, 1.7357e-04, 1.7357e-04, 1.7047e-05,
          7.1287e-05, 7.1287e-05, 1.7047e-05, 1.7047e-05, 4.3464e-04, 7.1287e-05,
          1.7047e-05, 1.1605e-04, 7.1287e-05, 4.0364e-04, 7.1287e-05, 7.1287e-05,
          1.7047e-05, 7.1287e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.9973e-03, 6.1462e-02, 1.5335e-02, 1.8509e-02, 3.8055e-02,
          4.5776e-02, 1.1932e-02, 1.0214e-03, 8.0395e-04, 3.9520e-03, 5.5580e-03,
          1.9817e-03, 1.3523e-03, 6.4325e-04, 3.2997e-03, 1.1415e-03, 1.5974e-03,
          1.0328e-03, 8.5068e-04, 3.3116e-04, 5.6648e-04, 4.6444e-04, 1.9398e-03,
          7.2861e-03, 4.8027e-03, 9.5367e-03, 7.2784e-03, 4.9543e-04, 1.8139e-03,
          9.9087e-04, 2.0447e-03, 5.9891e-04, 5.1422e-03, 3.4580e-03, 1.5020e-03,
          6.4354e-03, 8.8120e-04, 6.3820e-03, 4.1695e-03, 4.4489e-04, 6.6261e-03,
          1.7672e-03, 1.4715e-03, 1.1091e-03, 1.2178e-03, 2.0428e-03, 7.8430e-03,
          5.1165e-04, 5.7507e-04, 3.4332e-03, 1.9016e-03, 1.4868e-03, 1.5879e-03,
          4.3869e-04, 8.4639e-04, 4.6706e-04, 1.7672e-03, 1.6441e-03, 1.7281e-03,
          1.1873e-03, 1.2236e-03, 1.4677e-03, 2.7409e-03, 3.0918e-03, 2.1912e-02,
          2.2526e-03, 1.8463e-03, 3.5973e-03, 8.7166e-04, 7.1526e-04, 7.7200e-04,
          7.9918e-04, 1.1192e-02, 6.3229e-04, 1.7529e-03, 4.2915e-04, 4.7278e-04,
          8.1301e-04, 1.4906e-03, 1.6327e-03, 6.7329e-03, 1.6155e-03, 1.2093e-03,
          8.7881e-04, 4.0703e-03, 1.0862e-03, 4.4632e-03, 1.6279e-03, 1.4420e-03,
          8.8425e-03, 7.6714e-03, 1.1759e-03, 5.0049e-03, 1.1435e-03, 9.7036e-04,
          6.4754e-04, 3.8838e-04, 2.7714e-03, 3.0403e-03, 1.5316e-03, 2.2278e-03,
          6.0387e-03, 1.0309e-03, 1.2274e-03, 6.8550e-03, 9.0027e-03, 1.5450e-03,
          3.4866e-03, 3.0994e-03, 1.1185e-02, 1.3466e-03, 1.7958e-03, 3.5133e-03,
          4.3640e-03, 3.5496e-03, 1.1311e-03, 1.3916e-02, 1.3456e-03, 9.9850e-04,
          1.4896e-03, 6.7215e-03, 6.6376e-04, 6.7139e-04, 1.7532e-02, 8.0299e-04,
          1.1530e-03, 1.0271e-03, 1.3885e-03, 8.8959e-03, 2.4586e-03, 3.8910e-03,
          2.8267e-03, 2.1420e-03, 1.8005e-03, 1.6766e-03, 6.4011e-03, 1.3189e-03,
          7.2784e-03, 2.1133e-03, 1.8597e-03, 3.6907e-03, 4.7493e-03, 1.7614e-03,
          1.0691e-03, 1.5078e-03, 2.2526e-03, 1.8978e-03, 1.4277e-03, 1.1360e-02,
          1.1269e-02, 1.0033e-02, 2.2469e-03, 4.7226e-03, 2.2278e-03, 1.1421e-02,
          1.0170e-02, 7.9651e-03, 1.8654e-03, 1.8930e-03, 1.4290e-02, 3.7518e-03,
          1.3458e-02, 1.1765e-02, 3.5629e-03, 5.7983e-03, 2.8992e-03, 3.2005e-03,
          5.1880e-03, 3.5038e-03, 8.7280e-03, 3.6793e-03, 2.4776e-03, 1.1154e-02,
          3.1605e-03, 1.6737e-03, 2.2640e-03, 5.7945e-03, 1.8387e-02, 4.4441e-03,
          1.1185e-02, 6.9084e-03, 2.4185e-02, 3.0212e-03, 2.0008e-03, 2.7943e-03,
          4.5319e-03, 1.8034e-03, 2.6951e-03, 1.8492e-03, 1.4505e-03, 1.2230e-02,
          2.1038e-03, 9.2840e-04, 6.8359e-03, 4.4289e-03, 6.8283e-03, 7.8506e-03,
          1.8635e-03, 2.3727e-03, 1.9211e-02, 7.3547e-03, 5.9357e-03, 2.4246e-02,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 6.7520e-04,
          1.0484e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.0264e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.9181e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.0484e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.0264e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 7.2289e-04,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.0484e-04, 1.9181e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.0264e-04, 1.9181e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.0484e-04, 1.5676e-05, 1.0264e-04, 6.7520e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.9181e-04, 1.0264e-04, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.0264e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.0264e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.5676e-05, 1.0264e-04, 5.8603e-04, 1.0264e-04, 1.5676e-05, 1.5676e-05,
          1.0264e-04, 1.0264e-04, 5.8603e-04, 1.5676e-05, 1.5676e-05, 1.9181e-04,
          1.5676e-05, 6.3562e-04, 1.9181e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          1.0264e-04, 1.5676e-05, 1.0484e-04, 1.0264e-04, 1.5676e-05, 1.9181e-04,
          1.5676e-05, 1.5676e-05, 6.7520e-04, 1.5676e-05, 1.5676e-05, 5.8603e-04,
          7.2289e-04, 7.2289e-04, 1.5676e-05, 1.5676e-05, 1.9181e-04, 1.0484e-04,
          1.0484e-04, 1.0264e-04, 1.5676e-05, 1.5676e-05, 1.9181e-04, 1.5676e-05,
          8.1205e-04, 1.9181e-04, 1.5676e-05, 1.5676e-05, 1.0264e-04, 1.5676e-05,
          1.5676e-05, 1.9181e-04, 1.5676e-05, 1.0264e-04, 1.5676e-05, 1.5676e-05,
          7.2289e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05, 1.0484e-04, 1.5676e-05,
          8.1205e-04, 1.9181e-04, 1.0264e-04, 1.0264e-04, 1.5676e-05, 1.0264e-04,
          1.9181e-04, 1.0264e-04, 1.9181e-04, 1.5676e-05, 1.5676e-05, 1.5676e-05,
          7.2289e-04, 7.2289e-04, 1.5676e-05, 1.0264e-04, 1.0264e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 1.0264e-04, 1.0264e-04, 1.5676e-05,
          1.5676e-05, 1.5676e-05, 1.5676e-05, 4.9162e-04, 1.5676e-05, 1.0264e-04,
          1.0264e-04, 1.5676e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1003e-03, 4.9133e-02, 1.6006e-02, 1.4923e-02, 5.0934e-02,
          3.5889e-02, 1.3168e-02, 7.7200e-04, 6.7282e-04, 2.3518e-03, 7.2174e-03,
          2.0676e-03, 8.3303e-04, 1.7414e-03, 1.0481e-03, 8.3876e-04, 1.0862e-03,
          9.3699e-04, 7.1478e-04, 4.6992e-04, 5.5742e-04, 6.7282e-04, 1.3027e-03,
          1.0155e-02, 2.7905e-03, 1.0048e-02, 2.5997e-03, 7.5817e-04, 1.1320e-03,
          2.5063e-03, 1.2207e-03, 5.0783e-04, 1.0700e-03, 1.0290e-03, 5.9319e-04,
          1.4305e-03, 1.4439e-03, 1.6251e-03, 9.1887e-04, 7.6389e-04, 4.8256e-03,
          4.7722e-03, 2.9106e-03, 4.0283e-03, 1.3409e-03, 1.3409e-03, 1.9760e-03,
          9.5892e-04, 8.5163e-04, 5.3406e-03, 9.7847e-04, 1.0433e-03, 6.4850e-04,
          8.8549e-04, 7.8773e-04, 7.8440e-04, 1.3018e-03, 2.6016e-03, 2.7580e-03,
          2.1496e-03, 1.3790e-03, 1.9817e-03, 1.5659e-03, 9.0027e-03, 5.3749e-03,
          2.2598e-02, 6.4812e-03, 2.0466e-03, 2.1420e-03, 2.4014e-03, 2.6321e-03,
          1.6375e-03, 3.8090e-03, 1.0757e-03, 1.3418e-03, 8.5592e-04, 5.6219e-04,
          1.0128e-03, 1.5583e-03, 8.9340e-03, 2.3499e-03, 1.4397e-02, 1.5583e-03,
          1.2665e-03, 3.0708e-03, 1.7576e-03, 4.9706e-03, 1.0757e-02, 1.7914e-02,
          1.1772e-02, 5.2147e-03, 4.3716e-03, 6.3477e-03, 1.6775e-03, 1.4544e-03,
          9.9945e-04, 6.9714e-04, 8.2064e-04, 1.4515e-03, 2.4147e-03, 2.0256e-03,
          6.3286e-03, 3.5896e-03, 4.2114e-03, 2.3174e-03, 1.8616e-03, 1.5259e-03,
          2.7370e-03, 9.2010e-03, 9.1019e-03, 8.6060e-03, 1.5991e-02, 9.1324e-03,
          2.4357e-03, 3.9635e-03, 1.4343e-03, 3.9139e-03, 2.2182e-03, 2.3556e-03,
          2.4338e-03, 2.1458e-03, 2.3746e-03, 6.2227e-04, 2.4948e-03, 3.1376e-03,
          1.3227e-03, 6.5689e-03, 6.9733e-03, 1.8158e-03, 1.1330e-03, 3.2101e-03,
          1.3916e-02, 1.2505e-02, 1.0242e-03, 6.1188e-03, 9.6359e-03, 6.0425e-03,
          2.4548e-03, 3.0308e-03, 1.4400e-03, 2.3613e-03, 1.6146e-03, 2.3479e-03,
          3.8700e-03, 8.8930e-04, 1.9302e-03, 1.9970e-03, 1.1703e-02, 2.5196e-03,
          4.1733e-03, 1.9312e-03, 1.9217e-03, 2.9583e-03, 3.3531e-03, 1.2115e-02,
          7.5035e-03, 6.4621e-03, 2.3060e-03, 2.4963e-02, 6.3667e-03, 2.8839e-03,
          2.7924e-03, 2.6436e-03, 1.8826e-03, 1.6594e-03, 2.0523e-03, 2.4109e-03,
          4.9400e-03, 2.1534e-03, 1.9379e-03, 1.0328e-03, 1.6222e-03, 1.8654e-03,
          2.3212e-03, 2.8629e-03, 2.5291e-03, 1.1711e-02, 8.4305e-03, 2.6798e-03,
          2.8076e-03, 2.9984e-03, 4.8714e-03, 2.0790e-03, 1.5249e-03, 1.1816e-03,
          4.1199e-03, 2.8057e-03, 2.0313e-03, 3.3913e-03, 1.0681e-02, 2.0657e-03,
          1.2655e-03, 1.9455e-03, 1.3666e-03, 2.4166e-03, 3.1090e-03, 1.7118e-03,
          2.9163e-03, 2.8896e-03, 9.6741e-03, 2.3087e-02, 7.3090e-03, 4.5967e-03,
          6.6345e-02, 2.0504e-05, 2.0504e-05, 6.9439e-05, 2.0504e-05, 1.4663e-04,
          6.9439e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 1.4663e-04, 2.0504e-05, 2.0504e-05, 6.9439e-05, 1.4663e-04,
          6.9439e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.4663e-04, 6.9439e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 6.9439e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 6.9439e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 9.7632e-05, 6.9439e-05, 9.7811e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 6.9439e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 6.9439e-05, 1.4663e-04, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 9.7632e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 6.9439e-05, 2.0504e-05, 1.4663e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 7.0524e-04, 9.7632e-05,
          6.9439e-05, 9.7632e-05, 1.4663e-04, 1.4663e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 1.4663e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 1.4663e-04, 6.9439e-05, 1.4663e-04, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 9.7632e-05, 9.7632e-05, 9.7632e-05, 2.0504e-05, 2.0504e-05,
          1.4663e-04, 2.0504e-05, 2.0504e-05, 6.9439e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 1.2083e-03, 8.3160e-04, 2.0504e-05, 2.0504e-05,
          7.9441e-04, 6.9439e-05, 9.7632e-05, 2.0504e-05, 2.0504e-05, 1.4675e-04,
          2.0504e-05, 8.3160e-04, 1.4675e-04, 1.2553e-04, 2.0504e-05, 2.0504e-05,
          9.7632e-05, 2.0504e-05, 6.9439e-05, 2.0504e-05, 2.0504e-05, 1.4675e-04,
          2.0504e-05, 2.0504e-05, 8.3160e-04, 6.9439e-05, 6.9439e-05, 5.2357e-04,
          2.0504e-05, 6.9439e-05, 2.0504e-05, 2.0504e-05, 1.2553e-04, 6.9439e-05,
          6.9439e-05, 2.0504e-05, 2.0504e-05, 2.0504e-05, 1.4675e-04, 9.7632e-05,
          1.2553e-04, 1.2553e-04, 2.0504e-05, 2.0504e-05, 6.9439e-05, 2.0504e-05,
          6.9439e-05, 9.7811e-05, 6.9439e-05, 9.7632e-05, 2.0504e-05, 2.0504e-05,
          2.0504e-05, 2.0504e-05, 2.0504e-05, 6.9439e-05, 1.2553e-04, 2.0504e-05,
          1.4663e-04, 1.4663e-04, 2.0504e-05, 9.7632e-05, 2.0504e-05, 5.4646e-04,
          6.9439e-05, 2.0504e-05, 1.4663e-04, 6.9439e-05, 2.0504e-05, 3.1900e-04,
          9.7632e-05, 8.3160e-04, 2.0504e-05, 9.7632e-05, 5.2357e-04, 2.0504e-05,
          9.7632e-05, 9.7632e-05, 9.7632e-05, 3.9744e-04, 9.7632e-05, 9.7632e-05,
          2.0504e-05, 9.7632e-05, 9.7632e-05, 3.9935e-04, 2.0504e-05, 9.7632e-05,
          2.0504e-05, 2.0504e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.8365e-03, 5.0629e-02, 1.4359e-02, 1.9653e-02, 4.3579e-02,
          3.7323e-02, 1.3321e-02, 9.4557e-04, 7.7772e-04, 4.5624e-03, 7.3929e-03,
          3.1853e-03, 9.9564e-04, 7.9060e-04, 1.9321e-03, 1.0405e-03, 1.4162e-03,
          1.0214e-03, 7.3004e-04, 4.4227e-04, 7.0572e-04, 6.2704e-04, 1.5335e-03,
          1.1597e-02, 4.0894e-03, 1.0506e-02, 4.4479e-03, 6.1178e-04, 1.5697e-03,
          1.0386e-03, 1.6289e-03, 6.5470e-04, 2.3670e-03, 2.5539e-03, 1.0433e-03,
          3.2864e-03, 1.4915e-03, 4.9095e-03, 2.2411e-03, 5.9366e-04, 9.5749e-03,
          4.1618e-03, 2.8534e-03, 2.9850e-03, 2.2621e-03, 1.6994e-03, 4.2725e-03,
          6.3753e-04, 7.1096e-04, 4.1580e-03, 1.5392e-03, 1.3542e-03, 9.1648e-04,
          5.5885e-04, 9.0456e-04, 5.9891e-04, 2.7332e-03, 2.3041e-03, 3.7975e-03,
          1.8625e-03, 1.2674e-03, 1.5144e-03, 2.3594e-03, 7.5073e-03, 1.0124e-02,
          7.8201e-03, 5.2414e-03, 2.9774e-03, 2.4395e-03, 2.4605e-03, 1.8635e-03,
          1.5373e-03, 8.1100e-03, 8.5688e-04, 1.5650e-03, 4.9353e-04, 5.4264e-04,
          9.9277e-04, 3.1662e-03, 5.1308e-03, 5.6610e-03, 6.8130e-03, 1.4744e-03,
          1.0824e-03, 4.0627e-03, 1.3514e-03, 7.2670e-03, 4.5509e-03, 4.4136e-03,
          1.5732e-02, 1.1185e-02, 2.0790e-03, 8.7585e-03, 1.9951e-03, 1.5278e-03,
          7.4625e-04, 4.9639e-04, 1.5230e-03, 3.2101e-03, 3.4389e-03, 4.0474e-03,
          1.0178e-02, 1.9131e-03, 4.7112e-03, 5.9128e-03, 3.7632e-03, 1.6079e-03,
          3.2921e-03, 7.0915e-03, 1.6464e-02, 3.4752e-03, 4.5357e-03, 5.8899e-03,
          4.0054e-03, 6.6948e-03, 1.2321e-03, 7.4768e-03, 2.3594e-03, 1.2035e-03,
          1.7672e-03, 6.1836e-03, 1.3914e-03, 6.1131e-04, 4.5280e-03, 1.9255e-03,
          2.1095e-03, 2.8667e-03, 3.3474e-03, 3.4752e-03, 1.5316e-03, 3.6201e-03,
          5.5542e-03, 5.8517e-03, 1.1711e-03, 3.0594e-03, 1.4046e-02, 4.1885e-03,
          4.1924e-03, 2.2678e-03, 1.3170e-03, 2.9526e-03, 2.6093e-03, 1.7672e-03,
          2.2793e-03, 1.1091e-03, 2.3041e-03, 3.7212e-03, 3.6507e-03, 5.6839e-03,
          1.2947e-02, 3.9597e-03, 2.5978e-03, 4.2992e-03, 2.4357e-03, 1.4969e-02,
          1.2856e-02, 8.8959e-03, 2.2717e-03, 6.5727e-03, 1.1070e-02, 3.1624e-03,
          4.8752e-03, 5.4703e-03, 2.5387e-03, 2.6932e-03, 3.0327e-03, 3.2921e-03,
          7.6675e-03, 3.9749e-03, 3.1414e-03, 1.3161e-03, 1.8530e-03, 3.9978e-03,
          2.7771e-03, 1.9093e-03, 2.5635e-03, 1.1208e-02, 1.4160e-02, 3.1586e-03,
          3.5076e-03, 3.5458e-03, 7.5722e-03, 2.5406e-03, 1.4038e-03, 1.3285e-03,
          5.3864e-03, 1.8406e-03, 1.6737e-03, 2.8343e-03, 2.9297e-03, 2.8191e-03,
          1.7090e-03, 1.8606e-03, 2.3518e-03, 3.0766e-03, 5.8861e-03, 3.0518e-03,
          2.3880e-03, 2.1820e-03, 1.2230e-02, 1.6846e-02, 7.0686e-03, 5.4550e-03,
          1.1765e-02, 3.7506e-02, 1.6510e-05, 7.3671e-05, 1.6510e-05, 1.4174e-04,
          7.3671e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05, 7.3671e-05, 1.4174e-04,
          7.3671e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 7.3671e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 7.3671e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 7.3671e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 8.4579e-05, 1.4174e-04, 1.4198e-04,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 7.3671e-05, 1.4174e-04, 1.4174e-04, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 7.3671e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.4174e-04, 8.4579e-05,
          7.3671e-05, 1.6510e-05, 1.4174e-04, 7.3671e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 7.3671e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 3.9911e-04, 8.4579e-05, 8.4579e-05, 1.6510e-05, 1.6510e-05,
          1.4174e-04, 1.6510e-05, 1.6510e-05, 7.3671e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 3.9911e-04, 1.4174e-04, 1.6510e-05, 1.6510e-05,
          1.4174e-04, 7.3671e-05, 8.4579e-05, 1.6510e-05, 1.6510e-05, 1.4198e-04,
          1.6510e-05, 1.4174e-04, 4.5657e-04, 7.3671e-05, 1.6510e-05, 1.6510e-05,
          8.4579e-05, 1.6510e-05, 1.4174e-04, 8.4579e-05, 1.6510e-05, 1.4174e-04,
          1.6510e-05, 1.6510e-05, 1.4174e-04, 1.6510e-05, 4.6802e-04, 8.4579e-05,
          8.4579e-05, 7.3671e-05, 1.6510e-05, 1.6510e-05, 7.3671e-05, 7.3671e-05,
          4.6802e-04, 8.4579e-05, 1.6510e-05, 1.6510e-05, 1.4198e-04, 1.6510e-05,
          1.4174e-04, 1.4174e-04, 4.1103e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          7.3671e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 7.3671e-05, 1.6510e-05,
          4.5633e-04, 1.4174e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 8.4579e-05,
          7.3671e-05, 1.6510e-05, 1.4174e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          8.4579e-05, 1.1959e-03, 1.6510e-05, 8.4579e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 3.8528e-04, 2.5940e-04, 1.6510e-05,
          1.6510e-05, 8.4579e-05, 8.4579e-05, 3.7289e-04, 1.6510e-05, 8.4579e-05,
          1.6510e-05, 1.6510e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.8479e-03, 4.4281e-02, 1.8875e-02, 1.4832e-02, 6.0577e-02,
          3.3020e-02, 1.5076e-02, 7.7534e-04, 7.4005e-04, 2.1248e-03, 3.6526e-03,
          1.4343e-03, 9.9850e-04, 9.3699e-04, 1.4086e-03, 8.4209e-04, 1.3351e-03,
          8.4448e-04, 6.2180e-04, 1.3266e-03, 1.5211e-03, 1.4200e-03, 1.2999e-03,
          5.5237e-03, 3.8834e-03, 1.0719e-02, 2.4338e-03, 1.0920e-03, 1.1921e-03,
          9.3365e-04, 1.0662e-03, 5.8842e-04, 1.9236e-03, 1.3142e-03, 7.1764e-04,
          2.9984e-03, 1.7862e-03, 2.8019e-03, 1.5831e-03, 1.0290e-03, 8.1482e-03,
          9.3231e-03, 5.6725e-03, 7.0457e-03, 3.8643e-03, 1.3685e-03, 4.6806e-03,
          9.2793e-04, 1.8167e-03, 8.3466e-03, 1.0738e-03, 8.4877e-04, 7.2527e-04,
          6.9857e-04, 7.5245e-04, 1.1053e-03, 2.6588e-03, 1.5383e-03, 2.8572e-03,
          2.5387e-03, 1.6022e-03, 2.1324e-03, 2.5139e-03, 7.2784e-03, 7.8278e-03,
          4.6921e-03, 2.3823e-03, 2.6379e-03, 8.2169e-03, 9.4910e-03, 5.4474e-03,
          5.3673e-03, 3.7556e-03, 1.0710e-03, 1.1597e-03, 6.3658e-04, 4.6062e-04,
          9.0218e-04, 2.5806e-03, 2.3937e-03, 2.6321e-03, 2.3861e-03, 1.9274e-03,
          2.1915e-03, 2.8095e-03, 1.9045e-03, 2.9488e-03, 1.9369e-03, 3.1242e-03,
          5.5351e-03, 5.3635e-03, 4.3602e-03, 3.9864e-03, 9.1171e-03, 6.9275e-03,
          1.1358e-03, 8.2350e-04, 8.5020e-04, 2.0313e-03, 2.7943e-03, 2.3403e-03,
          2.8629e-03, 1.4334e-03, 1.3933e-03, 2.9125e-03, 2.4700e-03, 2.4509e-03,
          2.5940e-03, 1.4450e-02, 6.9618e-03, 5.0812e-03, 3.6888e-03, 3.0632e-03,
          3.9253e-03, 1.7670e-02, 4.3564e-03, 6.9580e-03, 1.1467e-02, 3.3417e-03,
          2.4319e-03, 3.8033e-03, 2.0084e-03, 5.3120e-04, 2.6703e-03, 1.1263e-03,
          2.7218e-03, 1.9836e-03, 2.6112e-03, 2.9068e-03, 1.3227e-03, 2.8915e-03,
          4.6463e-03, 2.7008e-03, 9.5510e-04, 1.8797e-03, 5.3482e-03, 1.3657e-02,
          3.5496e-03, 4.0321e-03, 2.0370e-03, 2.9793e-03, 1.8940e-03, 1.9073e-03,
          2.6608e-03, 1.4496e-03, 2.2831e-03, 1.8673e-03, 2.3193e-03, 3.9558e-03,
          4.3221e-03, 3.7766e-03, 4.5471e-03, 4.3106e-03, 4.0359e-03, 2.0493e-02,
          4.8866e-03, 6.4316e-03, 1.5039e-03, 8.4915e-03, 1.3672e-02, 4.7646e-03,
          5.2795e-03, 4.1885e-03, 2.2182e-03, 2.2316e-03, 4.0207e-03, 1.7443e-03,
          2.4281e-03, 1.4753e-03, 2.5921e-03, 1.6088e-03, 1.3752e-03, 3.6106e-03,
          4.2686e-03, 3.3360e-03, 6.4163e-03, 2.4353e-02, 9.3842e-03, 2.3842e-03,
          2.9030e-03, 2.9850e-03, 4.6768e-03, 1.9064e-03, 1.5898e-03, 2.7847e-03,
          2.7714e-03, 3.4218e-03, 2.4967e-03, 4.3259e-03, 2.1782e-03, 2.1973e-03,
          1.1072e-03, 3.2024e-03, 2.1343e-03, 2.4796e-03, 2.0866e-03, 2.7142e-03,
          5.2643e-03, 4.9973e-03, 1.0521e-02, 7.8964e-03, 3.2158e-03, 4.2343e-03,
          1.0918e-02, 1.4565e-02, 7.0862e-02, 1.0562e-04, 2.2292e-05, 3.7193e-04,
          1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04, 1.0562e-04, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 3.7193e-04, 3.7193e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04,
          2.2292e-05, 2.2292e-05, 3.7193e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05, 1.0562e-04,
          2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 1.0562e-04, 2.2292e-05, 2.2292e-05, 1.0562e-04, 1.0562e-04,
          1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05, 3.7193e-04, 2.2292e-05,
          1.0562e-04, 1.0562e-04, 2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05,
          1.0562e-04, 1.0562e-04, 2.2292e-05, 2.8849e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.0562e-04, 2.2292e-05,
          3.7193e-04, 1.0562e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          1.0562e-04, 2.2292e-05, 3.7193e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.8849e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 1.1110e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 3.9601e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4823e-03, 6.4148e-02, 1.4435e-02, 1.7166e-02, 4.7089e-02,
          5.0079e-02, 1.2962e-02, 1.1688e-02, 3.2024e-03, 4.0741e-03, 7.6408e-03,
          2.6875e-03, 2.3212e-03, 8.8644e-04, 2.8458e-03, 1.0994e-02, 2.1057e-03,
          1.7204e-03, 4.7073e-03, 2.8872e-04, 3.5191e-04, 4.1127e-04, 3.1281e-03,
          8.3771e-03, 2.8419e-03, 1.0727e-02, 5.8212e-03, 4.6062e-04, 2.2202e-03,
          1.0939e-03, 1.7044e-02, 6.7759e-04, 2.3499e-03, 4.3335e-03, 1.9588e-03,
          1.3494e-03, 5.7936e-04, 2.3041e-03, 2.7733e-03, 3.9172e-04, 1.4668e-03,
          6.6710e-04, 5.4169e-04, 6.1035e-04, 4.5085e-04, 2.9221e-03, 2.3117e-03,
          5.3596e-04, 5.3549e-04, 4.4365e-03, 2.4147e-03, 8.3466e-03, 3.2692e-03,
          1.3142e-03, 4.1122e-03, 7.8487e-04, 5.2643e-04, 2.0142e-03, 8.6117e-04,
          9.1028e-04, 1.0414e-03, 1.6251e-03, 1.6804e-03, 8.9121e-04, 3.7403e-03,
          1.6842e-03, 1.3256e-03, 1.4067e-03, 3.6073e-04, 3.6931e-04, 5.7888e-04,
          5.7364e-04, 7.3967e-03, 1.1139e-03, 1.1078e-02, 1.6317e-03, 1.6947e-03,
          2.8000e-03, 4.7231e-04, 8.8024e-04, 2.0809e-03, 1.2064e-03, 7.4148e-04,
          6.0511e-04, 1.4214e-02, 1.2693e-03, 4.7188e-03, 9.9754e-04, 9.6798e-04,
          7.4768e-03, 2.6741e-03, 5.6982e-04, 3.7384e-03, 5.6791e-04, 5.4598e-04,
          6.7329e-04, 6.1274e-04, 5.2452e-03, 1.7729e-03, 5.0974e-04, 8.2159e-04,
          6.4774e-03, 1.8511e-03, 1.3990e-03, 1.1749e-03, 3.6659e-03, 8.4114e-04,
          1.0422e-02, 6.8331e-04, 3.6526e-03, 7.6771e-04, 1.4238e-03, 1.3666e-03,
          3.1910e-03, 8.2636e-04, 7.1096e-04, 3.9444e-03, 6.9666e-04, 9.1839e-04,
          1.2951e-03, 2.3766e-03, 4.0150e-04, 2.1698e-02, 5.2948e-03, 4.2963e-04,
          3.7384e-04, 6.0558e-04, 8.8167e-04, 3.5954e-03, 1.0674e-02, 1.4313e-02,
          2.2621e-03, 1.5144e-03, 5.8632e-03, 1.0939e-03, 2.4910e-03, 5.6505e-04,
          3.8872e-03, 1.2836e-03, 7.6904e-03, 1.0956e-02, 4.5853e-03, 1.8568e-03,
          5.9366e-04, 4.5738e-03, 8.9931e-04, 7.1669e-04, 1.1091e-03, 3.3073e-03,
          4.4746e-03, 2.7466e-03, 8.4257e-04, 3.1738e-03, 3.1166e-03, 1.3399e-03,
          2.5558e-03, 1.3666e-03, 6.1264e-03, 9.8038e-04, 2.5349e-03, 1.7710e-03,
          3.4161e-03, 3.0422e-03, 5.0240e-03, 7.0686e-03, 9.7752e-04, 1.2255e-03,
          4.8561e-03, 1.6241e-03, 2.3460e-03, 4.0627e-03, 2.5635e-03, 2.4109e-03,
          3.2558e-03, 2.1305e-03, 1.1749e-03, 1.2493e-03, 3.0479e-03, 2.1988e-02,
          2.1191e-03, 3.6201e-03, 4.6272e-03, 3.5492e-02, 2.3926e-02, 5.8556e-03,
          1.1848e-02, 2.9621e-03, 7.7667e-03, 8.5020e-04, 1.2445e-03, 5.3635e-03,
          3.8269e-02, 4.5967e-04, 2.6722e-03, 5.2643e-03, 5.7793e-03, 2.7714e-03,
          1.3571e-03, 1.0653e-03, 4.2648e-03, 1.6975e-03, 3.6983e-03, 2.7523e-03,
          1.3275e-03, 2.4395e-03, 9.2268e-04, 6.6895e-02, 9.6619e-05, 7.8821e-04,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 9.6619e-05, 9.6619e-05, 9.6619e-05,
          2.3127e-05, 1.6105e-04, 2.3127e-05, 9.6619e-05, 2.3127e-05, 1.6105e-04,
          8.7500e-05, 2.3127e-05, 2.3127e-05, 9.4354e-05, 1.6105e-04, 2.3127e-05,
          2.3127e-05, 1.5879e-04, 9.4354e-05, 9.4354e-05, 2.3127e-05, 2.3222e-04,
          2.3127e-05, 9.4354e-05, 9.4354e-05, 9.4354e-05, 2.3127e-05, 1.6785e-04,
          2.3127e-05, 2.3127e-05, 1.5879e-04, 8.7500e-05, 1.6105e-04, 1.6105e-04,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 8.7500e-05, 2.3127e-05, 2.3127e-05,
          9.4354e-05, 2.3127e-05, 2.3127e-05, 9.4354e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 1.6785e-04, 9.4354e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.6105e-04, 1.6105e-04, 2.3127e-05,
          1.6785e-04, 9.6619e-05, 1.6105e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          9.4354e-05, 9.4354e-05, 8.7500e-05, 1.6105e-04, 2.3127e-05, 1.6785e-04,
          9.6619e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 1.6105e-04, 8.7500e-05,
          1.6785e-04, 2.3127e-05, 8.7500e-05, 8.7500e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.6105e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 8.7500e-05, 2.3127e-05, 1.6785e-04,
          8.7500e-05, 7.8821e-04, 8.7500e-05, 8.7500e-05, 2.3127e-05, 2.3127e-05,
          8.7500e-05, 2.3222e-04, 7.8297e-04, 2.3127e-05, 2.3127e-05, 8.7500e-05,
          9.4354e-05, 9.6619e-05, 8.7500e-05, 8.7500e-05, 2.3127e-05, 2.3127e-05,
          8.7500e-05, 1.6105e-04, 8.7500e-05, 2.3127e-05, 2.3127e-05, 1.6105e-04,
          9.4354e-05, 2.3222e-04, 1.6105e-04, 9.6619e-05, 2.3127e-05, 2.3127e-05,
          1.5879e-04, 2.3127e-05, 8.7500e-05, 7.8297e-04, 9.4354e-05, 1.6105e-04,
          9.4354e-05, 2.3127e-05, 8.7500e-05, 1.6785e-04, 9.4354e-05, 8.7500e-05,
          7.8297e-04, 8.7500e-05, 2.3127e-05, 2.3127e-05, 8.7500e-05, 2.3127e-05,
          1.5879e-04, 8.7500e-05, 1.6785e-04, 2.3127e-05, 1.6105e-04, 1.5879e-04,
          8.7500e-05, 8.7500e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 8.7500e-05, 2.3127e-05, 1.6785e-04, 2.3127e-05, 2.3127e-05,
          8.7500e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 8.7500e-05, 7.1859e-04,
          8.7500e-05, 1.6105e-04, 9.6619e-05, 8.7500e-05, 8.7500e-05, 8.7500e-05,
          8.7500e-05, 7.8297e-04, 1.6105e-04, 2.3127e-05, 1.6785e-04, 2.3222e-04,
          7.8297e-04, 8.7500e-05, 2.3127e-05, 1.5879e-04, 1.5879e-04, 2.3127e-05,
          8.7500e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 8.7500e-05, 2.3127e-05,
          2.3127e-05, 8.7500e-05, 2.3127e-05, 8.7500e-05, 2.3127e-05, 1.5879e-04,
          2.3127e-05, 2.3127e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.3128e-03, 5.8380e-02, 1.7059e-02, 3.3051e-02, 4.5624e-02,
          4.3610e-02, 1.7746e-02, 1.6174e-03, 1.0195e-03, 7.0076e-03, 1.5541e-02,
          5.1956e-03, 1.6603e-03, 8.9169e-04, 2.1782e-03, 1.6718e-03, 2.4242e-03,
          1.9140e-03, 1.0033e-03, 2.9826e-04, 5.0735e-04, 5.8651e-04, 3.1471e-03,
          2.1179e-02, 2.8172e-03, 1.4320e-02, 8.0795e-03, 5.5790e-04, 2.7599e-03,
          1.3485e-03, 3.1033e-03, 9.6083e-04, 3.3798e-03, 4.8141e-03, 1.3094e-03,
          2.2812e-03, 9.3508e-04, 4.3488e-03, 3.1776e-03, 4.1199e-04, 5.2834e-03,
          1.9159e-03, 1.3771e-03, 1.4153e-03, 1.2178e-03, 3.2082e-03, 4.2267e-03,
          6.2847e-04, 6.0606e-04, 4.8714e-03, 1.6623e-03, 2.1286e-03, 1.4038e-03,
          5.0926e-04, 1.0853e-03, 4.5800e-04, 1.8711e-03, 2.0695e-03, 2.5978e-03,
          1.3180e-03, 1.2522e-03, 1.1616e-03, 1.9083e-03, 3.2558e-03, 7.0496e-03,
          3.9368e-03, 3.4904e-03, 3.1395e-03, 8.3303e-04, 7.6294e-04, 9.9277e-04,
          9.4509e-04, 1.4252e-02, 7.1812e-04, 2.3346e-03, 4.7421e-04, 4.7684e-04,
          8.4591e-04, 1.7509e-03, 3.8013e-03, 4.1237e-03, 3.3016e-03, 1.0366e-03,
          7.3147e-04, 4.1542e-03, 8.4686e-04, 1.0162e-02, 2.8534e-03, 1.9970e-03,
          2.2324e-02, 9.6664e-03, 9.4414e-04, 1.2947e-02, 7.3099e-04, 8.6880e-04,
          7.0953e-04, 3.7050e-04, 2.0103e-03, 2.5444e-03, 1.4915e-03, 2.5997e-03,
          1.9791e-02, 1.6098e-03, 3.0937e-03, 2.3155e-03, 3.1414e-03, 1.2159e-03,
          3.1052e-03, 2.0695e-03, 1.4725e-02, 1.1673e-03, 2.7142e-03, 3.8452e-03,
          2.0123e-03, 2.0981e-03, 9.8419e-04, 5.8365e-03, 1.1444e-03, 1.0366e-03,
          1.6594e-03, 4.0398e-03, 8.6880e-04, 8.9836e-04, 5.1804e-03, 1.0643e-03,
          7.8678e-04, 1.5278e-03, 1.9121e-03, 3.3913e-03, 3.4599e-03, 3.5267e-03,
          4.7493e-03, 3.9139e-03, 1.4448e-03, 1.7977e-03, 9.4376e-03, 1.0881e-03,
          7.1716e-03, 2.0351e-03, 2.0428e-03, 4.9210e-03, 4.3945e-03, 2.0981e-03,
          1.2560e-03, 9.0408e-04, 1.3094e-03, 2.0084e-03, 2.5883e-03, 3.6144e-03,
          9.3460e-03, 2.8629e-03, 1.3628e-03, 3.2940e-03, 1.7490e-03, 5.2261e-03,
          9.0179e-03, 4.2458e-03, 3.8910e-03, 2.0199e-03, 6.1150e-03, 3.2196e-03,
          5.8060e-03, 6.6528e-03, 5.4779e-03, 4.3678e-03, 2.1954e-03, 2.1877e-03,
          1.6571e-02, 2.8229e-03, 2.6588e-03, 1.6546e-03, 3.4084e-03, 4.0550e-03,
          2.0218e-03, 1.5821e-03, 1.4696e-03, 3.9864e-03, 7.1831e-03, 5.7755e-03,
          3.4389e-03, 3.6240e-03, 6.9008e-03, 4.5929e-03, 1.8568e-03, 1.6651e-03,
          1.2733e-02, 2.1229e-03, 3.3703e-03, 1.7138e-03, 2.2163e-03, 3.8815e-03,
          3.5839e-03, 7.9489e-04, 3.1033e-03, 6.4697e-03, 1.1009e-02, 3.3817e-03,
          1.3456e-03, 1.2493e-03, 7.7629e-03, 7.3051e-03, 6.4926e-03, 4.9629e-03,
          4.4746e-03, 1.1673e-02, 2.7142e-03, 2.0332e-03, 3.8574e-02, 2.3890e-04,
          8.0526e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 7.2062e-05, 2.4080e-05, 2.4080e-05, 7.2062e-05, 2.3890e-04,
          8.0526e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.2851e-04, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 8.0526e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 8.0526e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 1.8251e-04, 2.3890e-04, 2.3890e-04,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.3890e-04, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          7.2062e-05, 2.4080e-05, 8.0526e-05, 1.2851e-04, 1.2851e-04, 7.2062e-05,
          8.0526e-05, 2.4080e-05, 2.3890e-04, 2.4080e-05, 1.8251e-04, 2.4080e-05,
          2.4080e-05, 8.0526e-05, 4.0889e-04, 1.2851e-04, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.3890e-04, 1.8251e-04,
          8.0526e-05, 2.4080e-05, 1.8251e-04, 8.0526e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 1.8251e-04, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 7.2062e-05, 8.0526e-05, 7.2062e-05, 2.4080e-05, 2.4080e-05,
          7.2062e-05, 1.8251e-04, 1.8251e-04, 7.2062e-05, 2.4080e-05, 2.4080e-05,
          7.2062e-05, 7.2062e-05, 2.4080e-05, 1.2851e-04, 2.4080e-05, 7.2062e-05,
          7.2062e-05, 2.4080e-05, 1.8251e-04, 7.2062e-05, 2.4080e-05, 2.4080e-05,
          7.2062e-05, 4.7350e-04, 7.2062e-05, 2.4080e-05, 2.4080e-05, 1.2851e-04,
          2.4080e-05, 6.5851e-04, 2.3890e-04, 8.0526e-05, 2.4080e-05, 2.4080e-05,
          7.2062e-05, 2.4080e-05, 1.2851e-04, 1.8251e-04, 2.4080e-05, 2.3890e-04,
          2.4080e-05, 2.4080e-05, 2.3890e-04, 8.0526e-05, 7.6056e-04, 2.4080e-05,
          7.2062e-05, 8.0526e-05, 2.4080e-05, 2.4080e-05, 1.8024e-04, 1.8024e-04,
          1.4610e-03, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.2851e-04, 7.2062e-05,
          7.5817e-04, 2.2829e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          8.0526e-05, 1.8251e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 2.4080e-05, 1.8024e-04, 2.4080e-05,
          7.6866e-04, 1.2851e-04, 2.4080e-05, 2.4080e-05, 2.4080e-05, 2.0897e-04,
          8.0526e-05, 2.4080e-05, 1.2851e-04, 2.4080e-05, 2.4080e-05, 1.8251e-04,
          3.6740e-04, 1.7481e-03, 2.4080e-05, 1.8251e-04, 2.4080e-05, 2.4080e-05,
          2.4080e-05, 2.4080e-05, 2.4080e-05, 8.6427e-05, 1.8525e-04, 8.0526e-05,
          2.4080e-05, 1.2851e-04, 7.2062e-05, 3.8266e-04, 3.2043e-04, 7.2062e-05,
          2.4080e-05, 2.4080e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.8512e-03, 5.3009e-02, 2.2583e-02, 2.3270e-02, 4.0924e-02,
          3.2715e-02, 1.3641e-02, 1.4677e-03, 1.0166e-03, 2.4128e-03, 3.3989e-03,
          1.3933e-03, 6.9122e-03, 3.4561e-03, 2.0809e-03, 1.7309e-03, 2.4994e-02,
          1.2794e-02, 8.9025e-04, 1.2684e-03, 1.1768e-03, 3.2196e-03, 6.9962e-03,
          4.4365e-03, 1.7157e-03, 1.0559e-02, 8.7128e-03, 2.1286e-03, 2.7206e-02,
          6.9618e-03, 3.5305e-03, 1.0742e-02, 4.5395e-03, 2.6150e-03, 2.0771e-03,
          1.3962e-03, 4.7803e-04, 3.7098e-03, 2.9716e-03, 1.9436e-03, 2.1954e-03,
          7.8201e-04, 5.9175e-04, 6.8426e-04, 6.3658e-04, 2.4780e-02, 7.3357e-03,
          3.3913e-03, 2.0905e-03, 4.1084e-03, 2.1973e-03, 2.6073e-03, 1.5364e-03,
          4.5156e-04, 1.2579e-03, 5.1689e-04, 5.3692e-04, 9.8419e-04, 7.1955e-04,
          7.0763e-04, 6.1607e-03, 1.0586e-03, 1.9703e-03, 9.3031e-04, 6.2294e-03,
          1.0815e-03, 8.6308e-04, 1.5268e-03, 4.2796e-04, 4.0364e-04, 5.9748e-04,
          7.8678e-04, 8.1024e-03, 7.8392e-04, 3.3016e-03, 4.8423e-04, 4.6301e-04,
          1.1263e-03, 5.7125e-04, 6.6042e-04, 1.4610e-03, 7.4816e-04, 4.7636e-04,
          4.5896e-04, 4.3068e-03, 8.1825e-04, 2.3079e-03, 6.8808e-04, 6.0320e-04,
          4.6768e-03, 3.7251e-03, 4.6253e-04, 2.7676e-03, 6.2895e-04, 8.0109e-04,
          2.9659e-03, 4.3511e-04, 3.6564e-03, 4.1199e-03, 5.2786e-04, 7.8249e-04,
          3.3875e-03, 5.8651e-04, 8.3447e-04, 1.0672e-03, 2.8553e-03, 4.2038e-03,
          3.1166e-03, 8.8167e-04, 3.6201e-03, 4.6921e-04, 8.3256e-04, 1.1482e-03,
          1.1101e-03, 1.4973e-03, 2.6531e-03, 6.9962e-03, 1.0967e-03, 2.5635e-03,
          5.1041e-03, 2.5349e-03, 3.4690e-04, 9.8228e-04, 4.1122e-03, 3.2258e-04,
          4.0627e-04, 4.6349e-04, 6.3848e-04, 3.1872e-03, 4.8943e-03, 3.7766e-03,
          1.5268e-03, 9.8419e-04, 9.0075e-04, 6.1846e-04, 2.1667e-03, 5.5885e-04,
          3.8910e-02, 6.2408e-03, 5.3673e-03, 6.0997e-03, 5.9547e-03, 8.1787e-03,
          5.1308e-04, 4.9162e-04, 6.1369e-04, 6.8378e-04, 7.6437e-04, 2.2392e-03,
          3.9825e-03, 1.8530e-03, 7.3242e-04, 4.4937e-03, 2.0065e-03, 2.7313e-03,
          2.7008e-03, 1.8520e-03, 1.3094e-03, 7.4053e-04, 5.6686e-03, 1.3168e-02,
          9.3460e-03, 6.0883e-03, 4.6051e-02, 5.1308e-03, 1.5211e-03, 1.0471e-03,
          2.5921e-03, 1.3170e-03, 1.8167e-03, 1.6308e-03, 1.3657e-02, 6.2714e-03,
          1.2283e-03, 8.8263e-04, 1.1282e-03, 2.0790e-03, 4.2572e-03, 7.3280e-03,
          2.2774e-03, 2.9449e-03, 5.9280e-03, 4.8828e-03, 3.3970e-03, 2.8782e-03,
          3.4637e-03, 6.6223e-03, 2.4548e-03, 9.5606e-04, 7.9060e-04, 3.8624e-03,
          4.4518e-03, 4.5323e-04, 5.4893e-03, 5.4504e-02, 2.8782e-03, 3.5439e-03,
          1.1482e-03, 3.1147e-03, 6.0043e-03, 1.9550e-03, 2.3670e-03, 3.0098e-03,
          1.1091e-03, 2.8744e-03, 1.2321e-03, 1.4505e-03, 5.5771e-03, 3.5736e-02,
          8.7798e-05, 1.5974e-05, 5.0068e-05, 8.7798e-05, 8.7798e-05, 8.7798e-05,
          5.0068e-05, 1.5974e-05, 1.5974e-05, 1.7178e-04, 1.5974e-05, 5.0068e-05,
          1.5974e-05, 1.5974e-05, 5.0068e-05, 1.7178e-04, 8.7798e-05, 8.7798e-05,
          1.7178e-04, 4.5061e-04, 1.5974e-05, 1.9681e-04, 1.5974e-05, 1.5974e-05,
          5.0068e-05, 1.3399e-04, 1.2529e-04, 8.7798e-05, 5.0068e-05, 9.1195e-05,
          1.5974e-05, 1.5974e-05, 1.5974e-05, 5.0068e-05, 1.5974e-05, 5.0068e-05,
          1.5974e-05, 8.7798e-05, 1.5974e-05, 8.7798e-05, 8.7798e-05, 1.7178e-04,
          4.0293e-04, 1.5974e-05, 1.5974e-05, 1.5974e-05, 1.5974e-05, 1.5974e-05,
          5.0068e-05, 2.0921e-04, 1.5974e-05, 5.0068e-05, 1.5974e-05, 5.3704e-05,
          1.5974e-05, 1.5974e-05, 8.7798e-05, 5.3704e-05, 8.7798e-05, 8.7798e-05,
          5.9795e-04, 8.7798e-05, 8.7798e-05, 5.0068e-05, 5.0068e-05, 8.7798e-05,
          5.3704e-05, 2.3437e-04, 1.5974e-05, 5.0068e-05, 1.5974e-05, 2.0981e-04,
          8.7798e-05, 2.4724e-04, 1.5974e-05, 1.5974e-05, 1.5974e-05, 5.0068e-05,
          2.3437e-04, 5.0068e-05, 1.5974e-05, 1.5974e-05, 1.5974e-05, 8.7798e-05,
          1.5974e-05, 1.3399e-04, 5.0068e-05, 8.7798e-05, 5.0068e-05, 1.5974e-05,
          1.5974e-05, 5.0068e-05, 2.8181e-04, 1.5974e-05, 1.3399e-04, 5.3704e-05,
          5.0068e-05, 1.5974e-05, 5.0068e-05, 1.5974e-05, 1.5974e-05, 8.7798e-05,
          1.5974e-05, 8.7798e-05, 1.5974e-05, 8.7798e-05, 5.0068e-05, 1.7178e-04,
          4.0293e-04, 2.8181e-04, 1.5974e-05, 1.5974e-05, 1.5974e-05, 1.5974e-05,
          1.5974e-05, 1.5974e-05, 1.5974e-05, 5.0068e-05, 1.5974e-05, 1.5974e-05,
          5.0068e-05, 5.3704e-05, 5.0068e-05, 5.3704e-05, 5.0068e-05, 5.0068e-05,
          4.9973e-04, 5.0068e-05, 8.7798e-05, 1.5974e-05, 1.3399e-04, 8.7798e-05,
          1.3399e-04, 8.7798e-05, 1.5974e-05, 8.7798e-05, 2.3103e-04, 1.5974e-05,
          5.0068e-05, 5.3704e-05, 1.5974e-05, 1.5974e-05, 5.7840e-04, 8.7798e-05,
          4.5586e-04, 5.0068e-05, 8.7798e-05, 1.7178e-04, 5.0068e-05, 7.2145e-04,
          1.0509e-03, 4.6849e-04, 5.0831e-04, 8.7798e-05, 2.4402e-04, 5.0068e-05,
          3.6597e-04, 5.0068e-05, 5.3704e-05, 2.6846e-04, 1.5974e-05, 1.3399e-04,
          1.5974e-05, 1.5974e-05, 5.0068e-05, 1.5974e-05, 3.8457e-04, 1.5974e-05,
          9.1505e-04, 8.7798e-05, 8.7798e-05, 1.5974e-05, 1.5974e-05, 5.0068e-05,
          4.2486e-04, 5.0068e-05, 3.8457e-04, 1.3399e-04, 8.7798e-05, 3.9887e-04,
          5.0068e-05, 8.5592e-04, 1.5974e-05, 1.5974e-05, 1.5974e-05, 1.5974e-05,
          1.6022e-04, 2.4402e-04, 5.0068e-05, 5.0068e-05, 5.0068e-05, 1.5974e-05,
          1.5974e-05, 1.9789e-04, 1.6022e-04, 3.9792e-04, 5.0068e-05, 1.6022e-04,
          5.0068e-05, 1.6022e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.7384e-03, 6.1737e-02, 1.7746e-02, 1.9913e-02, 3.8300e-02,
          4.4708e-02, 1.6556e-02, 1.6830e-02, 2.9812e-03, 2.2297e-03, 3.9787e-03,
          1.4248e-03, 2.0771e-03, 1.0738e-03, 1.9932e-03, 1.4107e-02, 2.4529e-03,
          1.8272e-03, 3.5801e-03, 3.9077e-04, 4.3821e-04, 6.2609e-04, 2.9373e-03,
          4.9858e-03, 2.1572e-03, 1.3885e-02, 6.8359e-03, 6.8569e-04, 2.3346e-03,
          1.3428e-03, 2.7237e-02, 7.9727e-04, 1.8072e-03, 1.9207e-03, 3.6716e-03,
          1.1148e-03, 6.3801e-04, 1.4791e-03, 1.7118e-03, 5.0306e-04, 1.0128e-03,
          6.8378e-04, 5.4741e-04, 6.5660e-04, 4.1866e-04, 2.6970e-03, 1.9855e-03,
          6.8855e-04, 6.6519e-04, 4.3716e-03, 5.3139e-03, 1.9775e-02, 3.7498e-03,
          2.2888e-03, 8.0414e-03, 1.8272e-03, 5.3501e-04, 2.1954e-03, 8.2159e-04,
          9.3317e-04, 1.2894e-03, 4.4403e-03, 2.3861e-03, 7.5865e-04, 2.6398e-03,
          1.5850e-03, 9.6369e-04, 9.8038e-04, 3.8981e-04, 3.7694e-04, 7.1239e-04,
          7.3624e-04, 4.2801e-03, 3.0231e-03, 2.3468e-02, 2.7695e-03, 3.1776e-03,
          6.0272e-03, 4.2796e-04, 6.8092e-04, 1.2417e-03, 9.2602e-04, 7.1573e-04,
          6.9618e-04, 3.3905e-02, 2.9888e-03, 2.3861e-03, 8.2064e-04, 8.0490e-04,
          4.3144e-03, 1.7967e-03, 5.3978e-04, 2.3785e-03, 6.6900e-04, 7.6628e-04,
          9.0933e-04, 1.2484e-03, 8.6823e-03, 1.6499e-03, 4.5466e-04, 5.8222e-04,
          3.3131e-03, 1.2884e-03, 8.7452e-04, 8.9979e-04, 2.3136e-03, 9.2936e-04,
          1.7410e-02, 7.4339e-04, 2.1839e-03, 8.2159e-04, 1.3189e-03, 1.0509e-03,
          1.8797e-03, 7.3814e-04, 1.0099e-03, 3.3379e-03, 1.2627e-03, 1.3905e-03,
          1.5221e-03, 1.3247e-03, 3.4523e-04, 1.7960e-02, 2.8572e-03, 3.4761e-04,
          3.2640e-04, 5.4598e-04, 7.5483e-04, 2.4567e-03, 8.8806e-03, 2.1744e-02,
          2.1763e-03, 1.0128e-03, 1.9321e-03, 8.0347e-04, 1.5717e-03, 5.2595e-04,
          3.5992e-03, 1.8768e-03, 7.3166e-03, 1.0612e-02, 3.6526e-03, 1.9093e-03,
          4.8518e-04, 1.8921e-03, 6.1750e-04, 5.8174e-04, 8.8024e-04, 2.2583e-03,
          2.8515e-03, 1.9188e-03, 8.5449e-04, 6.0196e-03, 8.9645e-03, 1.2140e-03,
          1.5326e-03, 1.1158e-03, 3.1338e-03, 8.7881e-04, 2.3556e-03, 2.1954e-03,
          2.4586e-03, 2.4509e-03, 4.8103e-03, 4.2763e-03, 8.0204e-04, 9.4986e-04,
          2.4242e-03, 1.0509e-03, 1.3990e-03, 2.3136e-03, 2.2392e-03, 1.6232e-03,
          2.7199e-03, 2.6035e-03, 1.2302e-03, 1.4610e-03, 1.7691e-03, 8.8196e-03,
          1.3494e-03, 2.3117e-03, 2.7218e-03, 4.0924e-02, 1.2436e-02, 3.6221e-03,
          5.6534e-03, 4.2038e-03, 6.9809e-03, 6.8855e-04, 1.0386e-03, 3.0270e-03,
          3.5065e-02, 3.6073e-04, 1.5907e-03, 4.8370e-03, 2.7905e-03, 1.7595e-03,
          1.6708e-03, 1.3123e-03, 3.0880e-03, 1.2197e-03, 2.0485e-03, 1.7052e-03,
          1.1911e-03, 1.6451e-03, 9.5177e-04, 1.6357e-02, 4.8141e-03, 2.2869e-03,
          6.5796e-02, 2.5570e-05, 2.5570e-05, 9.1136e-05, 9.1136e-05, 9.1136e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 2.5570e-05, 1.9312e-04,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 9.1136e-05,
          2.5570e-05, 1.2743e-04, 2.5570e-05, 3.3998e-04, 2.5570e-05, 9.1136e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          1.2743e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 1.9312e-04, 2.5570e-05,
          9.1136e-05, 9.1136e-05, 9.1136e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 3.3998e-04, 2.5570e-05, 1.9312e-04, 2.5570e-05, 1.9312e-04,
          9.1136e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 1.2743e-04, 2.5570e-05,
          4.0555e-04, 2.5570e-05, 1.2743e-04, 1.2743e-04, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 1.2743e-04, 2.5570e-05, 2.5570e-05, 9.1136e-05,
          1.2743e-04, 1.2743e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 9.1136e-05, 1.2743e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          1.2743e-04, 2.5570e-05, 1.2743e-04, 2.5570e-05, 2.5570e-05, 1.9312e-04,
          2.5570e-05, 9.1136e-05, 1.9312e-04, 9.1136e-05, 2.5570e-05, 2.5570e-05,
          4.4203e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 1.2743e-04, 1.9312e-04,
          2.5570e-05, 9.1136e-05, 1.2743e-04, 9.1136e-05, 3.3998e-04, 2.5570e-05,
          1.2743e-04, 1.9312e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 2.5570e-05, 9.1136e-05, 2.5570e-05, 1.9312e-04, 4.4203e-04,
          2.5570e-05, 2.5570e-05, 2.5570e-05, 9.1136e-05, 2.5570e-05, 2.5570e-05,
          1.9312e-04, 1.2743e-04, 2.5570e-05, 5.0735e-04, 2.5570e-05, 2.5570e-05,
          1.2743e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          1.2743e-04, 1.9312e-04, 1.9312e-04, 2.5570e-05, 1.2743e-04, 1.2743e-04,
          1.2743e-04, 2.5570e-05, 1.9312e-04, 2.5570e-05, 9.1136e-05, 5.0735e-04,
          1.9588e-03, 1.2743e-04, 2.5570e-05, 2.5570e-05, 2.5570e-05, 2.5570e-05,
          2.5570e-05, 1.2743e-04, 2.5570e-05, 2.5570e-05, 1.2743e-04, 1.2743e-04,
          2.5570e-05, 1.2743e-04, 2.5570e-05, 1.2743e-04, 2.5570e-05, 3.9959e-04,
          1.2743e-04, 1.2743e-04], dtype=torch.float16),
  tensor([0.0000e+00, 8.5220e-03, 5.3436e-02, 2.3102e-02, 2.6520e-02, 7.5256e-02,
          4.1962e-02, 2.0981e-02, 1.0099e-03, 1.1234e-03, 3.1109e-03, 3.7498e-03,
          1.7014e-03, 1.8015e-03, 1.2512e-03, 2.0390e-03, 7.5102e-04, 3.4866e-03,
          1.6165e-03, 5.3215e-04, 1.3266e-03, 1.6127e-03, 1.1425e-03, 2.2621e-03,
          3.5229e-03, 2.7866e-03, 1.5656e-02, 7.2479e-03, 3.6354e-03, 3.5172e-03,
          1.6890e-03, 1.4687e-03, 1.8415e-03, 4.7684e-03, 1.8902e-03, 1.4801e-03,
          5.4512e-03, 2.2850e-03, 4.8218e-03, 2.8610e-03, 1.4610e-03, 4.8027e-03,
          2.1687e-03, 1.9531e-03, 1.2484e-03, 1.5411e-03, 2.2526e-03, 5.4932e-03,
          8.6164e-04, 1.5278e-03, 9.4757e-03, 4.3297e-03, 1.6069e-03, 1.5135e-03,
          1.0834e-03, 1.0595e-03, 1.5001e-03, 5.4092e-03, 1.6785e-03, 3.0479e-03,
          2.1591e-03, 2.9278e-03, 1.6251e-03, 2.9125e-03, 2.0218e-03, 5.4855e-03,
          1.5087e-03, 8.1015e-04, 1.5554e-03, 1.3237e-03, 1.0920e-03, 9.7847e-04,
          1.2999e-03, 6.0310e-03, 2.6894e-03, 1.8301e-03, 7.6008e-04, 7.2527e-04,
          1.3800e-03, 7.8506e-03, 2.9793e-03, 3.9482e-03, 1.9913e-03, 1.4315e-03,
          1.8911e-03, 2.0790e-03, 1.2121e-03, 1.6479e-03, 9.4080e-04, 1.0529e-03,
          2.6016e-03, 2.7275e-03, 8.1730e-04, 1.9531e-03, 1.2045e-03, 1.5488e-03,
          3.8395e-03, 1.6518e-03, 1.5831e-03, 1.1009e-02, 9.4833e-03, 6.8703e-03,
          4.5471e-03, 1.3771e-03, 1.3456e-03, 4.6959e-03, 3.9597e-03, 3.6221e-03,
          1.7557e-03, 2.9202e-03, 4.1962e-03, 1.1024e-03, 1.3838e-03, 1.3342e-03,
          1.8883e-03, 4.2229e-03, 2.2659e-03, 4.8180e-03, 1.9493e-03, 1.0429e-02,
          9.3231e-03, 1.2810e-02, 4.7798e-03, 5.7554e-04, 3.9043e-03, 1.6479e-03,
          6.2943e-03, 1.9531e-03, 2.3041e-03, 5.4817e-03, 1.5993e-03, 2.1648e-03,
          1.8034e-03, 1.1101e-03, 6.3467e-04, 6.4135e-04, 1.8997e-03, 1.3227e-03,
          5.5199e-03, 2.4052e-03, 1.4524e-03, 3.3417e-03, 3.7766e-03, 5.3062e-03,
          5.8441e-03, 1.3762e-03, 3.4084e-03, 4.3221e-03, 2.5654e-03, 7.0381e-03,
          9.1171e-03, 5.1537e-03, 4.2725e-03, 4.4136e-03, 2.5692e-03, 8.1253e-03,
          2.5806e-03, 3.6106e-03, 6.1893e-04, 1.3542e-03, 7.2670e-03, 3.2272e-03,
          4.5395e-03, 4.9057e-03, 3.4409e-03, 3.8834e-03, 3.4424e-02, 6.1340e-03,
          4.1733e-03, 4.2763e-03, 4.4518e-03, 1.8845e-03, 3.5419e-03, 8.0032e-03,
          2.5368e-03, 2.5330e-03, 7.1220e-03, 6.8016e-03, 5.5962e-03, 2.1439e-03,
          2.0485e-03, 2.0256e-03, 3.4103e-03, 1.2589e-03, 1.3161e-03, 1.5583e-03,
          1.5612e-03, 2.3518e-03, 2.3441e-03, 1.7136e-02, 2.9678e-03, 3.3455e-03,
          1.4715e-03, 5.7335e-03, 3.6659e-03, 5.1041e-03, 2.6302e-03, 4.2839e-03,
          4.9362e-03, 6.5117e-03, 5.5809e-03, 1.9703e-03, 1.9932e-03, 2.3022e-03,
          1.7633e-03, 2.6894e-03, 2.6741e-03, 1.0700e-03, 3.4256e-03, 1.9817e-03,
          1.0624e-03, 4.6844e-02, 2.7061e-05, 1.7190e-04, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 1.7190e-04, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 3.6860e-04, 2.7061e-05,
          1.7190e-04, 1.7190e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.2376e-04, 2.7061e-05, 2.7061e-05,
          1.7190e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.2376e-04, 2.2376e-04, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 1.7190e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 1.7190e-04,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.2376e-04,
          2.7061e-05, 2.7061e-05, 2.2376e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 3.6860e-04,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 3.4094e-04,
          3.4094e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05, 3.6860e-04, 1.7190e-04,
          2.7061e-05, 2.2376e-04, 2.7061e-05, 2.7061e-05, 3.6860e-04, 2.7061e-05,
          3.6860e-04, 3.6860e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          1.7190e-04, 5.3787e-04, 2.7061e-05, 2.2376e-04, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 1.7190e-04, 2.7061e-05,
          2.2376e-04, 3.6860e-04, 3.6860e-04, 2.7061e-05, 2.7061e-05, 5.3787e-04,
          2.2376e-04, 3.4094e-04, 6.8235e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          5.3787e-04, 2.2376e-04, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          1.8060e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05, 2.7061e-05,
          2.7061e-05, 2.7061e-05, 2.2376e-04, 2.8515e-04, 2.7061e-05, 2.2376e-04,
          2.7061e-05, 2.7061e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.0534e-03, 4.7668e-02, 2.3087e-02, 2.2095e-02, 8.6426e-02,
          3.4332e-02, 2.3331e-02, 1.1406e-03, 1.0567e-03, 2.6436e-03, 6.5002e-03,
          1.8778e-03, 1.1034e-03, 1.3132e-03, 1.2484e-03, 7.0381e-04, 1.3695e-03,
          8.0347e-04, 5.2929e-04, 9.1457e-04, 8.3971e-04, 7.2002e-04, 1.0128e-03,
          5.7602e-03, 2.0561e-03, 1.5511e-02, 3.8891e-03, 3.8948e-03, 1.4191e-03,
          1.6251e-03, 1.4267e-03, 7.9775e-04, 1.7624e-03, 9.8133e-04, 6.9952e-04,
          2.8782e-03, 3.9177e-03, 1.6956e-03, 1.0967e-03, 1.0939e-03, 2.4052e-03,
          3.5019e-03, 2.2449e-03, 2.2392e-03, 1.1063e-03, 1.0662e-03, 1.7605e-03,
          6.7329e-04, 1.0729e-03, 9.0332e-03, 2.2221e-03, 1.3876e-03, 9.2840e-04,
          1.2980e-03, 9.0361e-04, 1.7052e-03, 6.5308e-03, 2.1839e-03, 5.6953e-03,
          3.8700e-03, 1.5650e-03, 1.6499e-03, 1.0614e-03, 3.6354e-03, 2.7466e-03,
          3.0422e-03, 1.2455e-03, 1.1053e-03, 1.4944e-03, 1.5287e-03, 1.9312e-03,
          1.5945e-03, 3.9902e-03, 3.3875e-03, 1.6432e-03, 9.1314e-04, 6.1607e-04,
          1.3866e-03, 1.2222e-02, 7.3090e-03, 2.8896e-03, 4.4975e-03, 2.2888e-03,
          3.0575e-03, 1.5306e-03, 1.1673e-03, 1.9798e-03, 1.3962e-03, 2.0466e-03,
          3.1948e-03, 2.0599e-03, 1.2751e-03, 2.2602e-03, 1.3256e-03, 1.5392e-03,
          3.1128e-03, 1.9989e-03, 7.8058e-04, 3.3970e-03, 2.3956e-02, 7.8888e-03,
          6.8893e-03, 2.0103e-03, 2.3632e-03, 3.7289e-03, 2.5864e-03, 1.9102e-03,
          1.3065e-03, 3.2120e-03, 3.2749e-03, 1.8930e-03, 2.7275e-03, 1.7529e-03,
          1.6136e-03, 1.8826e-03, 9.9182e-04, 2.6150e-03, 1.9970e-03, 8.8501e-03,
          6.0425e-03, 6.3782e-03, 1.4313e-02, 6.2895e-04, 3.1681e-03, 4.3068e-03,
          1.0994e-02, 5.3635e-03, 5.5847e-03, 2.5864e-03, 8.5402e-04, 1.5268e-03,
          3.2616e-03, 1.9760e-03, 5.9414e-04, 9.0933e-04, 1.8654e-03, 2.0657e-03,
          1.6155e-03, 1.4372e-03, 1.0118e-03, 2.1839e-03, 1.6556e-03, 3.2043e-03,
          1.5656e-02, 1.0891e-03, 4.2076e-03, 5.2338e-03, 6.4430e-03, 5.7449e-03,
          6.0005e-03, 2.8095e-03, 6.0120e-03, 1.7967e-03, 2.6836e-03, 3.6297e-03,
          1.8806e-03, 2.0313e-03, 6.3658e-04, 2.2011e-03, 2.8591e-03, 1.2455e-03,
          1.7958e-03, 1.4563e-03, 1.3857e-03, 1.8206e-03, 2.0889e-02, 5.2948e-03,
          5.6953e-03, 3.2997e-03, 3.4618e-03, 1.3313e-03, 1.5678e-03, 2.4681e-03,
          2.8763e-03, 4.9133e-03, 8.3237e-03, 4.5471e-03, 2.3251e-03, 1.5078e-03,
          1.4858e-03, 1.3199e-03, 1.9855e-03, 1.0891e-03, 9.8419e-04, 9.5367e-04,
          1.6890e-03, 1.4067e-03, 1.8082e-03, 3.3813e-02, 7.8735e-03, 2.3632e-03,
          1.2674e-03, 1.1612e-02, 1.6985e-03, 2.0866e-03, 2.4014e-03, 1.9760e-03,
          8.1482e-03, 4.6692e-03, 2.3556e-03, 2.2564e-03, 2.3022e-03, 1.3523e-03,
          2.9831e-03, 2.0809e-03, 3.0174e-03, 8.7929e-04, 3.4599e-03, 7.9060e-04,
          1.0529e-03, 1.8661e-02, 7.2876e-02, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.3113e-04, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 1.3113e-04, 2.1338e-04, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 1.0550e-04, 2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          1.3113e-04, 2.3305e-05, 2.1338e-04, 4.4107e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 1.0550e-04, 1.3113e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          1.0550e-04, 1.0550e-04, 2.3305e-05, 1.3113e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 6.9284e-04, 1.0550e-04, 2.3305e-05, 2.3305e-05,
          8.0109e-04, 2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05, 2.1338e-04,
          2.3305e-05, 2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          1.0550e-04, 2.3305e-05, 8.0109e-04, 2.3305e-05, 2.3305e-05, 2.1338e-04,
          2.3305e-05, 2.3305e-05, 2.9302e-04, 6.1083e-04, 2.3305e-05, 1.0900e-03,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.3113e-04, 1.3113e-04,
          1.3113e-04, 1.0550e-04, 2.3305e-05, 2.3305e-05, 1.0550e-04, 1.0550e-04,
          2.1338e-04, 2.1338e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          1.3113e-04, 1.0550e-04, 2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05, 1.3113e-04, 2.3305e-05,
          1.0550e-04, 2.1338e-04, 2.3305e-05, 1.0550e-04, 2.3305e-05, 6.9284e-04,
          2.1338e-04, 2.3305e-05, 2.1338e-04, 2.3305e-05, 2.3305e-05, 6.9284e-04,
          1.0550e-04, 1.0550e-04, 2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 1.0550e-04, 1.0550e-04, 2.3305e-05, 1.0550e-04, 1.0550e-04,
          2.3305e-05, 1.0550e-04, 2.3305e-05, 2.3305e-05, 2.3305e-05, 2.3305e-05,
          2.3305e-05, 2.3305e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.6490e-03, 5.6610e-02, 1.5457e-02, 2.4017e-02, 4.3976e-02,
          4.3060e-02, 1.2703e-02, 1.6489e-03, 1.2169e-03, 5.4016e-03, 1.3809e-02,
          4.0855e-03, 1.6041e-03, 9.8991e-04, 1.5125e-03, 1.2722e-03, 1.9484e-03,
          1.4219e-03, 8.1873e-04, 2.4366e-04, 3.1495e-04, 2.8729e-04, 2.1725e-03,
          1.5701e-02, 2.1324e-03, 1.0597e-02, 5.5733e-03, 5.8699e-04, 2.3632e-03,
          1.4601e-03, 2.7905e-03, 8.8215e-04, 1.9474e-03, 2.4605e-03, 9.3508e-04,
          2.0103e-03, 1.3247e-03, 1.6994e-03, 1.4009e-03, 3.8028e-04, 2.1820e-03,
          1.1368e-03, 7.9250e-04, 9.3555e-04, 6.0129e-04, 2.3651e-03, 1.7262e-03,
          4.7493e-04, 3.4451e-04, 3.2139e-03, 1.7452e-03, 2.4204e-03, 1.1244e-03,
          7.7343e-04, 1.3008e-03, 5.5361e-04, 2.1915e-03, 3.5667e-03, 3.5954e-03,
          1.9836e-03, 9.8419e-04, 9.5749e-04, 9.1696e-04, 2.7885e-03, 3.5343e-03,
          1.2093e-02, 8.0566e-03, 1.6327e-03, 4.8494e-04, 4.7374e-04, 8.0872e-04,
          5.1880e-04, 8.7280e-03, 1.1034e-03, 2.6321e-03, 6.2799e-04, 6.5851e-04,
          1.0204e-03, 2.7580e-03, 9.5901e-03, 4.7417e-03, 1.1620e-02, 1.2980e-03,
          8.6260e-04, 2.8591e-03, 7.6342e-04, 6.4964e-03, 6.6986e-03, 4.6272e-03,
          1.7288e-02, 5.2795e-03, 8.5926e-04, 8.1711e-03, 4.0674e-04, 4.1246e-04,
          7.5245e-04, 5.0020e-04, 1.2875e-03, 2.7332e-03, 4.3297e-03, 4.7607e-03,
          2.0569e-02, 6.6452e-03, 1.2810e-02, 3.0289e-03, 2.9697e-03, 9.3746e-04,
          2.0714e-03, 1.3018e-03, 7.6637e-03, 1.4801e-03, 8.4534e-03, 4.4746e-03,
          1.3437e-03, 7.8964e-04, 3.3402e-04, 3.0270e-03, 5.4359e-04, 1.8387e-03,
          2.6340e-03, 6.5384e-03, 3.1471e-03, 1.0710e-03, 4.4708e-03, 3.6983e-03,
          1.6203e-03, 4.5662e-03, 6.7787e-03, 2.9564e-03, 1.9455e-03, 2.4338e-03,
          1.1414e-02, 1.1253e-02, 9.9182e-04, 2.5368e-03, 7.4005e-03, 8.0395e-04,
          3.0270e-03, 7.5006e-04, 1.2836e-03, 2.9297e-03, 2.5158e-03, 2.9144e-03,
          6.1455e-03, 1.2894e-03, 2.8286e-03, 3.9978e-03, 1.1093e-02, 4.2992e-03,
          1.0818e-02, 2.8343e-03, 1.9646e-03, 1.7014e-03, 1.4791e-03, 2.2564e-03,
          5.1804e-03, 2.5024e-03, 2.8019e-03, 2.3289e-03, 2.2411e-03, 1.0986e-03,
          2.2068e-03, 2.1935e-03, 3.3417e-03, 2.3003e-03, 4.9667e-03, 5.4512e-03,
          2.1973e-02, 5.2910e-03, 3.6240e-03, 1.7347e-03, 2.8820e-03, 2.9831e-03,
          2.0809e-03, 1.9159e-03, 1.7118e-03, 2.3060e-03, 3.1757e-03, 4.2076e-03,
          1.8902e-03, 1.7719e-03, 3.3150e-03, 2.8419e-03, 1.5850e-03, 8.7547e-04,
          6.7940e-03, 1.1492e-03, 2.0676e-03, 7.8354e-03, 1.4351e-02, 3.7994e-03,
          3.9864e-03, 2.1133e-03, 2.5005e-03, 5.2261e-03, 9.6970e-03, 2.7180e-03,
          1.6994e-03, 1.3170e-03, 2.8820e-03, 6.0883e-03, 6.3362e-03, 1.8816e-03,
          4.1199e-03, 5.8365e-03, 9.4795e-04, 1.6918e-03, 1.3260e-02, 1.3323e-03,
          1.9989e-03, 6.5804e-03, 7.8430e-03, 3.8727e-02, 4.9496e-04, 3.1114e-05,
          3.1114e-05, 1.6880e-04, 3.1114e-05, 7.7009e-05, 1.6880e-04, 2.4223e-04,
          2.3139e-04, 1.8549e-04, 3.1114e-05, 3.1114e-05, 5.8222e-04, 7.7009e-05,
          3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 2.1756e-04,
          3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 1.3041e-04,
          3.1114e-05, 3.1114e-05, 6.9618e-05, 1.2302e-04, 1.6880e-04, 1.9646e-04,
          1.8549e-04, 3.1114e-05, 1.9646e-04, 1.9646e-04, 3.1114e-05, 3.1114e-05,
          3.1114e-05, 1.8549e-04, 1.8549e-04, 1.8549e-04, 1.1545e-04, 3.1114e-05,
          3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05,
          8.4519e-05, 3.1114e-05, 7.7009e-05, 1.6880e-04, 1.1545e-04, 2.7728e-04,
          3.1114e-05, 3.1114e-05, 2.1756e-04, 1.8549e-04, 2.7728e-04, 3.1114e-05,
          3.1114e-05, 7.7009e-05, 1.0872e-03, 1.1545e-04, 3.1114e-05, 3.1114e-05,
          3.1114e-05, 3.1114e-05, 3.1114e-05, 3.1114e-05, 2.4223e-04, 1.9646e-04,
          7.7009e-05, 3.1114e-05, 1.8895e-04, 1.1545e-04, 1.8549e-04, 6.9618e-05,
          1.8549e-04, 3.1114e-05, 1.8549e-04, 1.6880e-04, 3.1114e-05, 3.1114e-05,
          1.8549e-04, 1.1545e-04, 7.7009e-05, 1.6880e-04, 3.1114e-05, 3.1114e-05,
          1.2302e-04, 3.5238e-04, 1.9646e-04, 1.6880e-04, 3.1114e-05, 3.1114e-05,
          3.3665e-04, 1.2302e-04, 3.5071e-04, 1.3041e-04, 6.9618e-05, 1.2302e-04,
          6.9618e-05, 3.1114e-05, 1.1206e-03, 3.9673e-04, 2.3890e-04, 3.5071e-04,
          1.0605e-03, 1.6880e-04, 1.9646e-04, 3.1114e-05, 3.1114e-05, 3.9816e-04,
          3.1114e-05, 3.9816e-04, 6.6090e-04, 1.1545e-04, 3.1114e-05, 1.9646e-04,
          1.9646e-04, 7.7009e-05, 1.6880e-04, 3.5071e-04, 3.1114e-05, 2.4223e-04,
          3.1114e-05, 3.1114e-05, 2.9111e-04, 7.7009e-05, 3.2330e-04, 1.1206e-03,
          3.5071e-04, 7.7009e-05, 3.1114e-05, 3.1114e-05, 1.3041e-04, 1.3041e-04,
          7.4339e-04, 2.3746e-04, 3.1114e-05, 3.1114e-05, 3.9816e-04, 1.2302e-04,
          6.2323e-04, 5.4979e-04, 2.8467e-04, 6.9618e-05, 6.9618e-05, 3.1114e-05,
          2.0993e-04, 3.5071e-04, 7.7009e-05, 1.2302e-04, 3.1114e-05, 1.8549e-04,
          3.1114e-05, 1.9646e-04, 1.4305e-04, 7.7009e-05, 2.6512e-04, 3.1233e-04,
          9.1124e-04, 3.4499e-04, 3.1114e-05, 6.9618e-05, 3.5071e-04, 5.4789e-04,
          1.3041e-04, 4.4513e-04, 9.1124e-04, 3.1114e-05, 3.1114e-05, 4.4513e-04,
          6.0129e-04, 1.0653e-03, 3.5357e-04, 4.4513e-04, 4.4513e-04, 3.1114e-05,
          6.9618e-05, 3.1114e-05, 3.1114e-05, 6.9618e-05, 2.2352e-04, 5.3358e-04,
          3.1114e-05, 7.7009e-05, 6.9618e-05, 2.2352e-04, 2.2352e-04, 2.2352e-04,
          6.9618e-05, 6.9618e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4817e-03, 5.3833e-02, 1.3817e-02, 1.7441e-02, 4.4922e-02,
          4.2236e-02, 1.0567e-02, 1.5030e-03, 1.1969e-03, 4.3068e-03, 5.8136e-03,
          2.3422e-03, 2.3804e-03, 7.4387e-04, 5.3596e-03, 1.2655e-03, 2.9182e-03,
          1.5879e-03, 8.3017e-04, 2.8491e-04, 3.5024e-04, 3.0494e-04, 2.2717e-03,
          6.4888e-03, 5.2223e-03, 8.5220e-03, 5.5580e-03, 7.1287e-04, 3.8815e-03,
          1.1787e-03, 3.2940e-03, 1.0662e-03, 8.0338e-03, 4.1351e-03, 1.9217e-03,
          4.8218e-03, 9.5367e-04, 3.8223e-03, 4.6310e-03, 4.3893e-04, 2.2068e-03,
          7.8773e-04, 7.2241e-04, 5.8365e-04, 5.7697e-04, 3.7270e-03, 8.0490e-03,
          4.1604e-04, 3.8815e-04, 3.3569e-03, 2.6817e-03, 2.4738e-03, 1.9817e-03,
          6.6185e-04, 1.2360e-03, 7.2050e-04, 1.9054e-03, 1.7748e-03, 1.7662e-03,
          1.4162e-03, 1.3533e-03, 1.1120e-03, 1.3952e-03, 1.1454e-03, 1.7456e-02,
          1.7548e-03, 9.3031e-04, 1.6832e-03, 7.4530e-04, 4.3035e-04, 4.8542e-04,
          5.1212e-04, 9.4299e-03, 1.0586e-03, 2.8954e-03, 5.1641e-04, 5.5027e-04,
          1.1578e-03, 2.3289e-03, 2.1458e-03, 5.5199e-03, 1.7300e-03, 1.1139e-03,
          8.9836e-04, 3.0251e-03, 6.9952e-04, 2.8305e-03, 9.6798e-04, 9.0361e-04,
          5.2757e-03, 3.3169e-03, 5.9605e-04, 2.9087e-03, 8.0681e-04, 6.2180e-04,
          1.0281e-03, 5.8842e-04, 2.8534e-03, 4.9706e-03, 2.9526e-03, 3.9139e-03,
          9.1476e-03, 1.2722e-03, 1.6232e-03, 4.1237e-03, 1.0132e-02, 1.5850e-03,
          2.4509e-03, 1.0204e-03, 4.7493e-03, 7.1335e-04, 1.6441e-03, 1.4076e-03,
          6.3286e-03, 2.5959e-03, 4.9400e-04, 1.8402e-02, 8.6498e-04, 2.2583e-03,
          3.4962e-03, 1.1337e-02, 1.6356e-03, 1.0462e-03, 1.4503e-02, 1.1921e-03,
          2.0962e-03, 1.1530e-03, 1.6937e-03, 1.5358e-02, 3.6316e-03, 2.9335e-03,
          2.5616e-03, 1.3895e-03, 1.2732e-03, 6.5088e-04, 2.6321e-03, 6.7568e-04,
          8.3389e-03, 1.0138e-03, 1.9369e-03, 3.5248e-03, 5.0392e-03, 4.1313e-03,
          2.8286e-03, 1.6241e-03, 3.7422e-03, 2.7905e-03, 2.2297e-03, 2.2705e-02,
          1.0902e-02, 1.0994e-02, 2.8706e-03, 2.5654e-03, 1.6708e-03, 3.0956e-03,
          3.1872e-03, 2.0943e-03, 1.3361e-03, 8.8882e-04, 1.1124e-02, 2.6627e-03,
          1.0139e-02, 4.5052e-03, 5.9090e-03, 6.1798e-03, 5.9090e-03, 3.5076e-03,
          7.4615e-03, 3.8223e-03, 1.4854e-02, 8.5297e-03, 4.3335e-03, 1.0605e-02,
          3.8109e-03, 1.9608e-03, 2.7504e-03, 1.9569e-03, 5.3940e-03, 4.2114e-03,
          3.7632e-03, 4.2992e-03, 9.9030e-03, 3.5343e-03, 1.5478e-03, 2.7676e-03,
          3.6182e-03, 1.3132e-03, 2.3232e-03, 5.7297e-03, 2.8229e-03, 1.5511e-02,
          4.5891e-03, 2.1687e-03, 1.0246e-02, 8.0338e-03, 5.5008e-03, 9.1476e-03,
          2.3232e-03, 2.1515e-03, 7.4005e-03, 1.8806e-03, 2.2526e-03, 4.0169e-03,
          1.2779e-03, 2.8019e-03, 1.2560e-03, 1.4639e-03, 5.8708e-03, 2.3403e-03,
          1.5621e-03, 8.5449e-03, 4.4861e-03, 4.3030e-03, 3.8116e-02, 2.0087e-05,
          2.0087e-05, 7.6056e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.6618e-04,
          2.0087e-05, 2.1398e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.6989e-04, 2.1398e-04, 2.0087e-05, 2.0087e-05, 2.1398e-04,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.1027e-04,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.6989e-04,
          2.0087e-05, 7.6056e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.5631e-03,
          2.0087e-05, 7.6056e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 1.6618e-04, 1.6618e-04, 2.0087e-05,
          1.6618e-04, 1.1027e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 9.1219e-04, 3.6001e-04, 2.0087e-05, 1.6618e-04,
          1.1027e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          1.1027e-04, 2.0087e-05, 7.6056e-05, 7.8058e-04, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.1398e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 1.6618e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 7.6056e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.1027e-04,
          7.6056e-05, 2.0087e-05, 2.6989e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 7.6056e-05, 9.6798e-04, 2.6989e-04, 2.0087e-05, 2.0087e-05,
          2.6989e-04, 2.6989e-04, 7.6056e-05, 2.0087e-05, 2.0087e-05, 1.6618e-04,
          2.0087e-05, 3.2377e-04, 1.6618e-04, 1.1027e-04, 2.0087e-05, 3.2377e-04,
          3.2377e-04, 2.0087e-05, 1.1027e-04, 5.1785e-04, 7.6056e-05, 3.6001e-04,
          2.0087e-05, 2.0087e-05, 7.6056e-05, 7.6056e-05, 4.1056e-04, 1.5659e-03,
          1.2159e-03, 1.6527e-03, 7.6056e-05, 2.0087e-05, 1.6618e-04, 1.6618e-04,
          4.1056e-04, 7.6056e-05, 2.0087e-05, 2.0087e-05, 4.1389e-04, 2.0087e-05,
          3.5930e-04, 3.5930e-04, 2.0087e-05, 2.0087e-05, 7.6056e-05, 2.0087e-05,
          1.1027e-04, 6.8045e-04, 2.0087e-05, 7.6056e-05, 2.0087e-05, 2.0087e-05,
          4.1389e-04, 7.6056e-05, 7.6056e-05, 2.0087e-05, 1.1027e-04, 3.2377e-04,
          4.1389e-04, 1.6618e-04, 1.6618e-04, 7.6056e-05, 2.6774e-04, 1.3199e-03,
          1.6618e-04, 5.1785e-04, 1.0834e-03, 2.0087e-05, 2.0087e-05, 2.6989e-04,
          8.6737e-04, 3.2377e-04, 2.0087e-05, 2.6989e-04, 7.5293e-04, 2.0087e-05,
          7.6056e-05, 2.0087e-05, 7.6056e-05, 7.6056e-05, 7.6056e-05, 2.0087e-05,
          2.0087e-05, 1.6618e-04, 3.4022e-04, 3.4833e-04, 7.6056e-05, 6.8998e-04,
          7.6056e-05, 7.6056e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9357e-03, 5.5298e-02, 1.4374e-02, 1.4526e-02, 4.5837e-02,
          3.6713e-02, 1.1253e-02, 9.3460e-04, 8.7881e-04, 2.6436e-03, 7.4883e-03,
          2.1229e-03, 8.5068e-04, 1.6422e-03, 1.1702e-03, 6.7568e-04, 1.0414e-03,
          7.0810e-04, 5.5838e-04, 3.7527e-04, 3.4809e-04, 3.6693e-04, 1.1406e-03,
          7.5569e-03, 1.9112e-03, 8.4305e-03, 2.6722e-03, 8.3685e-04, 1.1139e-03,
          2.6970e-03, 1.4572e-03, 5.2500e-04, 1.3046e-03, 1.1101e-03, 7.3814e-04,
          1.8520e-03, 1.5030e-03, 1.1444e-03, 8.4972e-04, 7.0953e-04, 1.6270e-03,
          2.1629e-03, 1.3123e-03, 1.8349e-03, 6.8283e-04, 9.6416e-04, 1.2484e-03,
          6.9189e-04, 5.5695e-04, 3.7212e-03, 1.4334e-03, 1.4200e-03, 8.8406e-04,
          1.0395e-03, 9.9850e-04, 9.8610e-04, 2.5387e-03, 2.9793e-03, 3.9902e-03,
          2.6379e-03, 1.2531e-03, 1.0605e-03, 7.2002e-04, 4.2229e-03, 2.5883e-03,
          9.1553e-03, 2.7466e-03, 1.1244e-03, 9.9468e-04, 1.2846e-03, 1.4248e-03,
          9.1505e-04, 4.2534e-03, 1.3828e-03, 1.9255e-03, 9.3412e-04, 7.8726e-04,
          1.4019e-03, 4.7455e-03, 1.4214e-02, 3.8433e-03, 1.1116e-02, 1.7452e-03,
          1.3838e-03, 1.9093e-03, 1.0052e-03, 3.0308e-03, 6.1150e-03, 1.1620e-02,
          5.7945e-03, 3.0708e-03, 3.3607e-03, 3.6755e-03, 1.0042e-03, 8.6546e-04,
          1.0576e-03, 8.2827e-04, 1.0519e-03, 2.3975e-03, 1.0956e-02, 6.9695e-03,
          9.9335e-03, 5.1384e-03, 5.0735e-03, 3.6850e-03, 2.6455e-03, 1.3237e-03,
          1.6584e-03, 2.4166e-03, 4.4022e-03, 3.8071e-03, 1.0040e-02, 4.5547e-03,
          2.3327e-03, 1.2207e-03, 5.8413e-04, 2.4166e-03, 1.2302e-03, 2.2774e-03,
          3.2024e-03, 6.3286e-03, 8.7585e-03, 8.0013e-04, 3.3646e-03, 9.8267e-03,
          5.2071e-03, 1.1528e-02, 1.1963e-02, 2.7485e-03, 1.1244e-03, 2.0199e-03,
          1.0353e-02, 4.9934e-03, 6.3133e-04, 2.1210e-03, 4.3411e-03, 3.1147e-03,
          1.7624e-03, 1.5135e-03, 1.1768e-03, 2.1687e-03, 1.7176e-03, 3.5839e-03,
          1.6205e-02, 1.0881e-03, 6.5193e-03, 6.8207e-03, 2.7298e-02, 5.9509e-03,
          8.4381e-03, 3.3569e-03, 3.6144e-03, 1.6365e-03, 1.9665e-03, 2.8629e-03,
          3.8338e-03, 3.4809e-03, 1.0900e-03, 1.2650e-02, 2.6836e-03, 1.3008e-03,
          1.8826e-03, 1.6108e-03, 1.4982e-03, 1.6680e-03, 6.2675e-03, 7.0343e-03,
          8.2245e-03, 5.3406e-03, 4.5815e-03, 1.6384e-03, 2.1133e-03, 3.2406e-03,
          3.2253e-03, 3.9978e-03, 3.4924e-03, 2.5654e-03, 2.5826e-03, 1.7443e-03,
          1.7176e-03, 1.4305e-03, 2.2297e-03, 1.4410e-03, 1.0738e-03, 9.9087e-04,
          2.8191e-03, 1.6785e-03, 1.8787e-03, 1.5823e-02, 3.0365e-02, 2.8076e-03,
          1.9169e-03, 8.1177e-03, 2.2869e-03, 2.5387e-03, 4.3106e-03, 2.7237e-03,
          4.3526e-03, 2.6932e-03, 2.5272e-03, 5.1422e-03, 3.2730e-03, 1.5860e-03,
          8.4076e-03, 4.2419e-03, 1.8473e-03, 1.0366e-03, 5.0774e-03, 7.9584e-04,
          1.1826e-03, 6.5460e-03, 1.5587e-02, 1.3870e-02, 7.2098e-03, 5.1819e-02,
          2.0981e-05, 1.1051e-04, 2.0981e-05, 5.7817e-05, 2.0981e-05, 2.0981e-05,
          5.7817e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 1.3292e-04, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 5.7817e-05, 2.0981e-05, 7.3612e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 5.7817e-05, 1.1712e-04, 5.7817e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 5.7817e-05, 5.7817e-05, 7.3612e-05, 1.3292e-04, 8.0228e-05,
          8.0228e-05, 2.0981e-05, 8.0228e-05, 2.0981e-05, 8.0228e-05, 2.0981e-05,
          2.0981e-05, 1.1712e-04, 1.2541e-03, 1.3292e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          7.3612e-05, 2.0981e-05, 1.6975e-04, 1.6975e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 8.0228e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 1.1712e-04, 1.3292e-04, 1.1712e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 5.7817e-05, 5.7817e-05, 2.0981e-05,
          1.1712e-04, 8.0228e-05, 5.7817e-05, 7.3612e-05, 8.0228e-05, 8.0228e-05,
          8.0228e-05, 8.0228e-05, 8.5545e-04, 1.6975e-04, 2.0981e-05, 2.0981e-05,
          9.0837e-04, 2.0981e-05, 1.1712e-04, 2.0981e-05, 2.0981e-05, 1.3292e-04,
          2.0981e-05, 2.0981e-05, 1.3292e-04, 7.3612e-05, 5.7817e-05, 1.1712e-04,
          8.0228e-05, 5.7817e-05, 1.6975e-04, 2.0981e-05, 2.0981e-05, 1.3292e-04,
          5.7817e-05, 2.0981e-05, 1.3292e-04, 5.7817e-05, 1.3292e-04, 8.5545e-04,
          5.7817e-05, 7.3612e-05, 2.0981e-05, 2.0981e-05, 7.3612e-05, 7.3612e-05,
          1.1051e-04, 5.7817e-05, 5.7817e-05, 5.7817e-05, 1.3292e-04, 8.0228e-05,
          1.3292e-04, 1.3292e-04, 2.0981e-05, 5.7817e-05, 7.3612e-05, 2.0981e-05,
          1.3292e-04, 8.0228e-05, 7.3612e-05, 1.1712e-04, 2.0981e-05, 5.7817e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 1.1051e-04, 1.1051e-04, 5.7817e-05,
          1.3292e-04, 1.3292e-04, 8.0228e-05, 1.1712e-04, 2.0981e-05, 1.1712e-04,
          1.3292e-04, 8.0228e-05, 1.3292e-04, 1.1051e-04, 2.0981e-05, 3.9053e-04,
          1.1712e-04, 1.3292e-04, 5.7817e-05, 8.0228e-05, 1.1712e-04, 2.0981e-05,
          8.0228e-05, 8.0228e-05, 8.0228e-05, 8.0228e-05, 3.5572e-04, 8.0228e-05,
          2.0981e-05, 3.5572e-04, 3.5572e-04, 5.9319e-04, 8.0228e-05, 3.5572e-04,
          1.1712e-04, 8.0228e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.8784e-03, 5.9174e-02, 1.4900e-02, 1.7731e-02, 4.5227e-02,
          4.5135e-02, 1.0803e-02, 1.1768e-03, 1.1454e-03, 3.1376e-03, 3.2654e-03,
          1.6222e-03, 1.6403e-03, 8.7547e-04, 3.2272e-03, 9.5606e-04, 3.2349e-03,
          1.4553e-03, 6.1750e-04, 4.9114e-04, 7.3862e-04, 3.7742e-04, 1.8940e-03,
          3.3188e-03, 2.9297e-03, 8.5526e-03, 5.6725e-03, 9.4461e-04, 3.2864e-03,
          1.3380e-03, 2.4490e-03, 1.1587e-03, 7.3357e-03, 2.6035e-03, 1.9417e-03,
          7.2098e-03, 1.3981e-03, 5.3635e-03, 4.5280e-03, 6.0129e-04, 3.8033e-03,
          1.0309e-03, 8.8024e-04, 6.3467e-04, 9.5272e-04, 2.4071e-03, 7.2021e-03,
          4.5252e-04, 5.2023e-04, 4.0855e-03, 3.9253e-03, 2.6684e-03, 1.8969e-03,
          9.4366e-04, 1.6212e-03, 1.3294e-03, 3.1681e-03, 2.0409e-03, 2.8629e-03,
          2.0809e-03, 1.7815e-03, 1.2636e-03, 2.7943e-03, 1.4219e-03, 8.7357e-03,
          1.0061e-03, 6.4707e-04, 1.4315e-03, 7.0381e-04, 5.6314e-04, 5.6744e-04,
          7.4291e-04, 6.4316e-03, 1.5697e-03, 3.3016e-03, 6.8474e-04, 8.6784e-04,
          1.9312e-03, 3.9482e-03, 2.1286e-03, 5.8708e-03, 1.7138e-03, 1.6346e-03,
          1.3571e-03, 3.4351e-03, 9.3079e-04, 1.6603e-03, 7.7391e-04, 8.2970e-04,
          2.6722e-03, 3.2463e-03, 5.5408e-04, 1.8072e-03, 9.8991e-04, 9.5177e-04,
          1.3027e-03, 8.8358e-04, 2.8763e-03, 1.3756e-02, 4.3411e-03, 5.4970e-03,
          5.2261e-03, 1.1930e-03, 1.2827e-03, 8.4686e-03, 7.2632e-03, 2.1648e-03,
          2.9259e-03, 2.0657e-03, 4.2839e-03, 8.2016e-04, 9.2936e-04, 1.6623e-03,
          2.0142e-03, 4.3335e-03, 8.7976e-04, 8.9340e-03, 1.1101e-03, 2.3689e-03,
          3.6087e-03, 1.7624e-02, 1.8282e-03, 8.7929e-04, 6.5765e-03, 1.4610e-03,
          3.5248e-03, 1.4410e-03, 2.1687e-03, 1.2138e-02, 2.8572e-03, 3.2063e-03,
          1.4791e-03, 8.4591e-04, 6.5994e-04, 5.7077e-04, 2.0905e-03, 8.4496e-04,
          6.7215e-03, 1.4267e-03, 1.4429e-03, 3.3913e-03, 4.7607e-03, 3.6488e-03,
          3.3283e-03, 1.6870e-03, 3.6430e-03, 4.3411e-03, 2.0142e-03, 1.3641e-02,
          1.5961e-02, 1.2085e-02, 5.9395e-03, 7.4425e-03, 2.6283e-03, 7.2441e-03,
          3.6831e-03, 4.0817e-03, 7.3290e-04, 1.0900e-03, 1.2047e-02, 2.6875e-03,
          6.9618e-03, 7.0877e-03, 3.9940e-03, 5.0507e-03, 1.0490e-02, 6.1264e-03,
          4.6577e-03, 6.5842e-03, 8.7738e-03, 4.8180e-03, 3.9101e-03, 1.6357e-02,
          6.2218e-03, 3.4771e-03, 5.2414e-03, 3.5706e-03, 8.4152e-03, 2.6665e-03,
          2.7313e-03, 3.1929e-03, 5.9166e-03, 2.0962e-03, 1.4896e-03, 1.9894e-03,
          1.9531e-03, 1.2894e-03, 1.7309e-03, 6.6605e-03, 2.6093e-03, 7.3357e-03,
          3.0670e-03, 3.3779e-03, 9.0408e-03, 5.2109e-03, 4.0054e-03, 9.5062e-03,
          4.6768e-03, 3.7537e-03, 9.7656e-03, 2.8095e-03, 1.8215e-03, 3.6068e-03,
          1.3981e-03, 5.3329e-03, 1.9855e-03, 1.1282e-03, 3.7766e-03, 1.7128e-03,
          1.2884e-03, 1.2947e-02, 4.8065e-03, 2.8229e-03, 1.5373e-02, 3.5362e-03,
          2.9739e-02, 6.7413e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 2.0874e-04,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          6.7413e-05, 6.7413e-05, 2.3329e-04, 1.6630e-05, 1.6630e-05, 4.6539e-04,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 6.7413e-05, 4.6539e-04,
          1.6630e-05, 6.7413e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 6.7413e-05,
          1.6630e-05, 6.7413e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 2.4867e-04, 2.0874e-04, 1.6630e-05,
          2.0874e-04, 1.5795e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 4.6539e-04, 4.2534e-04, 1.6630e-05, 6.7413e-05,
          2.0874e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 6.7413e-05, 6.7413e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 1.9777e-04, 2.0874e-04, 6.7413e-05, 1.6630e-05, 1.6630e-05,
          1.6630e-05, 6.7413e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          2.4867e-04, 1.6630e-05, 2.8396e-04, 1.6630e-05, 6.7413e-05, 1.6630e-05,
          1.6630e-05, 6.7413e-05, 4.6539e-04, 4.6539e-04, 1.6630e-05, 1.6630e-05,
          2.4867e-04, 4.6539e-04, 4.9400e-04, 1.6630e-05, 1.6630e-05, 2.0874e-04,
          1.6630e-05, 6.7413e-05, 2.0874e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05,
          6.7413e-05, 1.6630e-05, 1.9777e-04, 2.8396e-04, 6.7413e-05, 4.2534e-04,
          1.6630e-05, 1.6630e-05, 6.7413e-05, 2.4867e-04, 1.6630e-05, 4.6539e-04,
          4.6539e-04, 2.0771e-03, 1.6630e-05, 1.6630e-05, 2.0874e-04, 2.0874e-04,
          1.6630e-05, 2.4867e-04, 1.6630e-05, 1.6630e-05, 2.0874e-04, 1.6630e-05,
          3.9005e-04, 2.0874e-04, 1.6630e-05, 6.7413e-05, 2.4867e-04, 1.6630e-05,
          1.6630e-05, 4.6539e-04, 1.6630e-05, 6.7413e-05, 1.6630e-05, 1.6630e-05,
          4.9400e-04, 1.6630e-05, 1.6630e-05, 1.6630e-05, 2.0874e-04, 1.6630e-05,
          6.7413e-05, 2.0874e-04, 2.0874e-04, 6.7413e-05, 1.6630e-05, 4.6539e-04,
          6.7413e-05, 2.8396e-04, 6.0654e-04, 1.6630e-05, 1.6630e-05, 1.9777e-04,
          1.9083e-03, 2.4867e-04, 1.6630e-05, 6.7413e-05, 6.7413e-05, 1.6630e-05,
          6.7413e-05, 6.7413e-05, 3.1662e-04, 6.7413e-05, 6.7413e-05, 1.6630e-05,
          6.7413e-05, 6.7413e-05, 6.7413e-05, 6.7413e-05, 6.7413e-05, 7.4291e-04,
          6.7413e-05, 6.7413e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.7406e-03, 5.1941e-02, 1.9302e-02, 1.9897e-02, 4.6814e-02,
          3.5339e-02, 1.1879e-02, 1.8015e-03, 1.2331e-03, 2.5940e-03, 3.5458e-03,
          1.5306e-03, 6.5613e-03, 5.8060e-03, 2.5730e-03, 1.6298e-03, 2.0660e-02,
          9.7809e-03, 7.9775e-04, 1.6108e-03, 9.9850e-04, 2.3384e-03, 4.8637e-03,
          3.7899e-03, 1.8520e-03, 9.4604e-03, 7.0190e-03, 2.8057e-03, 1.5488e-02,
          9.7885e-03, 3.9139e-03, 6.3400e-03, 5.1003e-03, 2.1572e-03, 2.2831e-03,
          2.8744e-03, 8.8215e-04, 3.2272e-03, 2.8744e-03, 3.2997e-03, 1.9855e-03,
          7.1669e-04, 5.5313e-04, 5.8508e-04, 5.7936e-04, 1.4229e-02, 5.1308e-03,
          5.2719e-03, 2.5597e-03, 4.6043e-03, 2.9411e-03, 2.9736e-03, 1.5926e-03,
          7.9536e-04, 1.6251e-03, 1.1129e-03, 1.2579e-03, 1.4133e-03, 1.3227e-03,
          1.2474e-03, 9.5749e-03, 1.2617e-03, 1.8654e-03, 9.4175e-04, 4.7035e-03,
          1.2398e-03, 6.7139e-04, 1.0519e-03, 3.7789e-04, 3.3140e-04, 4.9829e-04,
          5.9223e-04, 6.0616e-03, 1.2007e-03, 3.7651e-03, 7.3528e-04, 6.8760e-04,
          1.6775e-03, 1.4324e-03, 1.2302e-03, 2.4490e-03, 1.1501e-03, 8.4209e-04,
          8.6260e-04, 3.8986e-03, 1.0786e-03, 1.7443e-03, 6.8951e-04, 7.6056e-04,
          2.9144e-03, 2.5063e-03, 4.7278e-04, 1.8196e-03, 5.1260e-04, 6.2418e-04,
          3.1776e-03, 6.9475e-04, 3.3722e-03, 5.0507e-03, 1.3523e-03, 1.9169e-03,
          4.0474e-03, 8.2970e-04, 9.7322e-04, 2.0294e-03, 3.5114e-03, 1.0490e-02,
          2.7637e-03, 9.0551e-04, 3.0537e-03, 5.3310e-04, 9.5034e-04, 9.4080e-04,
          1.2016e-03, 1.1740e-03, 3.2387e-03, 5.4283e-03, 8.4448e-04, 4.7684e-03,
          1.3641e-02, 4.4899e-03, 7.2718e-04, 1.0929e-03, 4.2725e-03, 6.6280e-04,
          1.1196e-03, 8.6355e-04, 1.2274e-03, 4.9934e-03, 4.2267e-03, 3.2959e-03,
          1.6174e-03, 9.0218e-04, 6.8140e-04, 4.7517e-04, 1.5240e-03, 5.7125e-04,
          2.5421e-02, 8.3694e-03, 6.7482e-03, 5.2223e-03, 4.8714e-03, 2.0493e-02,
          1.1215e-03, 6.7377e-04, 1.3313e-03, 1.4133e-03, 1.3313e-03, 4.4632e-03,
          4.9133e-03, 3.2578e-03, 1.6813e-03, 4.0245e-03, 2.5616e-03, 2.3861e-03,
          2.1629e-03, 1.5135e-03, 8.6784e-04, 7.3147e-04, 4.2648e-03, 1.8631e-02,
          5.5237e-03, 3.9635e-03, 2.0248e-02, 4.0131e-03, 3.1757e-03, 1.5755e-03,
          3.0384e-03, 1.9159e-03, 3.2120e-03, 2.1687e-03, 1.8250e-02, 7.1373e-03,
          2.1210e-03, 1.6747e-03, 2.6379e-03, 2.0638e-03, 3.6469e-03, 4.4937e-03,
          1.6375e-03, 2.1839e-03, 3.8567e-03, 3.7041e-03, 2.1629e-03, 2.2602e-03,
          2.1362e-03, 1.0979e-02, 3.0251e-03, 2.1687e-03, 1.6823e-03, 4.4975e-03,
          4.2839e-03, 1.1749e-03, 5.0850e-03, 2.6947e-02, 2.7142e-03, 3.8605e-03,
          2.3766e-03, 7.9956e-03, 4.6082e-03, 1.4544e-03, 1.8044e-03, 2.1038e-03,
          1.0099e-03, 2.1057e-03, 1.0099e-03, 1.1702e-03, 3.4332e-03, 9.0408e-03,
          1.8816e-03, 5.6076e-03, 2.0103e-03, 1.8702e-03, 8.2321e-03, 1.7128e-03,
          4.4708e-03, 4.0771e-02, 1.8120e-05, 1.3793e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 8.2254e-05, 9.7227e-04, 1.3793e-04, 7.3791e-05,
          5.4979e-04, 2.6250e-04, 1.8120e-05, 1.4257e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 8.2254e-05, 7.3791e-05, 7.3791e-05, 1.8120e-05, 9.4593e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 3.9649e-04, 1.8120e-05, 1.9825e-04, 4.1294e-04, 1.3793e-04,
          5.9319e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.3793e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.0605e-03, 1.3793e-04, 1.3793e-04, 1.1864e-03, 8.2254e-05, 8.2254e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 4.6849e-04,
          4.6849e-04, 3.4881e-04, 3.4881e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          9.4593e-05, 1.8120e-05, 1.8120e-05, 8.2254e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 8.2254e-05, 1.8120e-05, 1.8120e-05, 8.2254e-05, 3.4881e-04,
          1.8120e-05, 1.8120e-05, 1.3793e-04, 1.8120e-05, 4.1294e-04, 7.3791e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 7.3791e-05, 7.3791e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.3793e-04, 1.8120e-05, 1.3793e-04,
          1.5726e-03, 2.9135e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.3793e-04, 1.8120e-05, 1.8120e-05,
          7.3791e-05, 7.3791e-05, 7.3791e-05, 1.8120e-05, 2.7204e-04, 1.8120e-05,
          8.2254e-05, 7.3791e-05, 1.8120e-05, 7.3791e-05, 3.0255e-04, 1.8120e-05,
          1.8120e-05, 8.2254e-05, 1.8120e-05, 1.8120e-05, 1.3793e-04, 1.3793e-04,
          7.3791e-05, 1.8120e-05, 1.3793e-04, 1.3793e-04, 1.8120e-05, 3.6645e-04,
          4.6849e-04, 1.3793e-04, 8.2254e-05, 1.8120e-05, 8.2254e-05, 7.3791e-05,
          1.3793e-04, 1.8120e-05, 7.3791e-05, 3.6645e-04, 1.8120e-05, 1.8120e-05,
          8.2254e-05, 3.4881e-04, 1.8120e-05, 1.8120e-05, 1.3793e-04, 1.8120e-05,
          4.1294e-04, 1.8120e-05, 1.3793e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          6.2180e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.3793e-04, 8.2254e-05,
          1.8120e-05, 6.0272e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          5.4979e-04, 2.3580e-04, 2.3580e-04, 2.3580e-04, 2.3580e-04, 8.2254e-05,
          1.8120e-05, 1.3793e-04, 2.3580e-04, 5.3835e-04, 2.3580e-04, 8.2254e-05,
          2.3580e-04, 2.3580e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.5695e-03, 5.9875e-02, 1.5976e-02, 1.9989e-02, 3.2928e-02,
          5.5054e-02, 1.1810e-02, 2.1458e-03, 1.1692e-03, 9.6893e-03, 1.7700e-02,
          4.3602e-03, 2.2984e-03, 9.3079e-04, 4.1008e-03, 2.0180e-03, 2.5005e-03,
          1.8435e-03, 1.4057e-03, 2.3544e-04, 3.4285e-04, 3.8195e-04, 4.0436e-03,
          1.6281e-02, 2.7790e-03, 9.9487e-03, 7.1831e-03, 5.5027e-04, 2.9202e-03,
          1.4009e-03, 4.7684e-03, 9.0981e-04, 4.3602e-03, 1.5915e-02, 1.1034e-03,
          1.9932e-03, 6.0654e-04, 7.2098e-03, 6.1646e-03, 3.5024e-04, 2.1477e-03,
          6.2895e-04, 5.0020e-04, 5.5647e-04, 4.7803e-04, 3.4676e-03, 3.4294e-03,
          5.3740e-04, 4.6802e-04, 3.2024e-03, 1.7366e-03, 3.0918e-03, 1.9197e-03,
          5.4932e-04, 1.5173e-03, 4.9353e-04, 6.1941e-04, 1.7214e-03, 9.3126e-04,
          8.6784e-04, 1.0023e-03, 7.8249e-04, 1.3237e-03, 8.7357e-04, 4.8714e-03,
          2.1687e-03, 2.2125e-03, 2.5349e-03, 3.0756e-04, 3.1257e-04, 4.6253e-04,
          4.8184e-04, 1.8600e-02, 7.6818e-04, 3.6888e-03, 5.4455e-04, 5.8365e-04,
          1.2007e-03, 5.6839e-04, 1.1997e-03, 3.5248e-03, 1.6317e-03, 6.0177e-04,
          5.0926e-04, 3.7575e-03, 5.3453e-04, 1.1024e-02, 1.7204e-03, 1.1063e-03,
          1.7517e-02, 6.1340e-03, 4.6420e-04, 7.0724e-03, 3.6383e-04, 4.3583e-04,
          8.0967e-04, 4.2653e-04, 2.3708e-03, 1.9569e-03, 6.3276e-04, 1.0290e-03,
          1.1566e-02, 2.1744e-03, 2.2793e-03, 1.3237e-03, 4.0245e-03, 7.9536e-04,
          2.8973e-03, 5.5027e-04, 6.1569e-03, 5.8889e-04, 1.6203e-03, 2.7084e-03,
          2.1343e-03, 6.1750e-04, 4.8566e-04, 4.2725e-03, 6.2370e-04, 1.0757e-03,
          1.7262e-03, 3.4599e-03, 5.2023e-04, 1.7996e-03, 8.3618e-03, 5.5838e-04,
          4.3511e-04, 6.8569e-04, 9.2888e-04, 4.1161e-03, 5.5580e-03, 4.1046e-03,
          2.7370e-03, 2.4872e-03, 5.9700e-03, 1.3704e-03, 4.2267e-03, 4.7541e-04,
          5.2834e-03, 9.6607e-04, 3.8223e-03, 8.4229e-03, 8.1863e-03, 2.4033e-03,
          7.3528e-04, 3.3169e-03, 1.0567e-03, 8.5831e-04, 1.5574e-03, 3.2711e-03,
          7.3776e-03, 3.0651e-03, 6.9094e-04, 1.6870e-03, 1.2693e-03, 1.2093e-03,
          4.5204e-03, 1.5707e-03, 4.6005e-03, 6.9952e-04, 2.1915e-03, 1.3084e-03,
          4.0741e-03, 6.4812e-03, 6.0081e-03, 1.5625e-02, 1.2417e-03, 1.6394e-03,
          8.5907e-03, 2.5253e-03, 2.9984e-03, 4.2458e-03, 3.0460e-03, 3.8338e-03,
          1.5593e-03, 1.0386e-03, 9.9277e-04, 9.9850e-04, 3.6926e-03, 1.2543e-02,
          2.8286e-03, 3.7079e-03, 6.6795e-03, 6.7749e-03, 4.0665e-03, 5.3978e-03,
          1.2520e-02, 1.3962e-03, 1.4267e-02, 1.1215e-03, 1.5059e-03, 8.3847e-03,
          7.7400e-03, 5.1069e-04, 4.9553e-03, 6.5536e-03, 1.4107e-02, 4.2229e-03,
          9.8991e-04, 8.3733e-04, 4.7989e-03, 2.2163e-03, 1.0918e-02, 3.1052e-03,
          1.0557e-03, 3.0956e-03, 7.2098e-04, 4.2038e-03, 1.1436e-02, 2.3766e-03,
          3.0193e-03, 1.9512e-03, 1.0710e-03, 3.8948e-03, 5.7983e-03, 1.0920e-03,
          2.3174e-03, 2.6722e-03, 7.5623e-02, 1.9133e-05, 1.9133e-05, 9.6381e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 9.6381e-05, 9.6381e-05, 1.9133e-05,
          9.6381e-05, 9.6381e-05, 2.3210e-04, 1.1361e-04, 1.9133e-05, 4.6110e-04,
          1.9133e-05, 1.9133e-05, 9.6381e-05, 1.1361e-04, 1.9133e-05, 9.6381e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 9.6381e-05, 3.9959e-04,
          1.9133e-05, 1.9133e-05, 1.7095e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          2.4819e-04, 2.3210e-04, 2.3210e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 9.6381e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 9.6381e-05,
          1.9133e-05, 1.9133e-05, 9.6381e-05, 9.6381e-05, 9.6381e-05, 1.9133e-05,
          9.6381e-05, 9.6381e-05, 9.6381e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          1.1361e-04, 1.1361e-04, 3.8385e-04, 9.6381e-05, 1.9133e-05, 9.6381e-05,
          9.6381e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 2.4819e-04, 1.9133e-05,
          6.6471e-04, 1.9133e-05, 9.6381e-05, 1.7059e-04, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.9133e-05, 2.3210e-04,
          1.9133e-05, 1.9133e-05, 3.4237e-04, 9.6381e-05, 1.9133e-05, 1.9133e-05,
          1.9133e-05, 3.9959e-04, 1.9133e-05, 9.6381e-05, 1.9133e-05, 1.9133e-05,
          9.6381e-05, 1.1361e-04, 3.8385e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05,
          1.1361e-04, 1.9133e-05, 5.3549e-04, 6.1274e-04, 2.3210e-04, 1.9133e-05,
          3.2234e-04, 4.6110e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 2.4772e-04,
          1.9133e-05, 3.4261e-04, 2.4772e-04, 1.9133e-05, 1.9133e-05, 1.7095e-04,
          1.9085e-04, 1.9133e-05, 1.1361e-04, 3.8385e-04, 1.9133e-05, 3.9959e-04,
          1.9133e-05, 1.9133e-05, 3.9959e-04, 1.9133e-05, 1.9085e-04, 3.8385e-04,
          6.1274e-04, 2.4819e-04, 1.9133e-05, 1.9133e-05, 3.4237e-04, 1.9085e-04,
          1.9085e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 2.4772e-04, 1.1361e-04,
          3.4237e-04, 3.4237e-04, 1.9085e-04, 1.9133e-05, 1.1361e-04, 1.9133e-05,
          9.6381e-05, 6.1274e-04, 1.9133e-05, 1.9085e-04, 1.9133e-05, 1.9133e-05,
          1.7095e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 3.4237e-04, 1.7095e-04,
          4.9400e-04, 9.6381e-05, 1.9133e-05, 1.9133e-05, 2.4819e-04, 5.3549e-04,
          4.9400e-04, 3.8385e-04, 3.9959e-04, 1.9133e-05, 1.1361e-04, 4.7803e-04,
          6.1274e-04, 4.1676e-04, 1.9133e-05, 4.7851e-04, 4.7851e-04, 1.9133e-05,
          1.7059e-04, 1.9133e-05, 1.9133e-05, 1.9133e-05, 1.7059e-04, 1.9133e-05,
          1.9133e-05, 1.7059e-04, 1.9133e-05, 3.3474e-04, 1.9133e-05, 2.6512e-04,
          1.9133e-05, 1.9133e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5586e-03, 5.8350e-02, 1.4885e-02, 1.2955e-02, 4.0558e-02,
          5.0171e-02, 9.7885e-03, 2.2564e-03, 1.4591e-03, 4.7722e-03, 9.0027e-03,
          2.8534e-03, 4.8714e-03, 9.5940e-04, 1.3199e-02, 2.1667e-03, 3.6716e-03,
          1.8911e-03, 1.6575e-03, 2.4188e-04, 3.2735e-04, 3.0065e-04, 3.1052e-03,
          9.3384e-03, 5.0812e-03, 7.7400e-03, 4.0970e-03, 5.1451e-04, 3.4142e-03,
          1.3399e-03, 4.8332e-03, 8.7976e-04, 5.5122e-03, 5.9052e-03, 1.6317e-03,
          2.8133e-03, 6.3372e-04, 2.3880e-03, 4.3869e-03, 4.2510e-04, 1.2217e-03,
          4.4179e-04, 3.9983e-04, 4.0197e-04, 4.0936e-04, 4.2000e-03, 4.6539e-03,
          4.5681e-04, 4.3654e-04, 3.2597e-03, 1.6241e-03, 2.4796e-03, 2.1229e-03,
          5.2595e-04, 1.5869e-03, 6.3372e-04, 8.9455e-04, 1.6403e-03, 1.1578e-03,
          1.2159e-03, 1.0090e-03, 7.9870e-04, 1.0538e-03, 6.6996e-04, 8.9798e-03,
          9.7847e-04, 7.9918e-04, 1.1044e-03, 2.9135e-04, 2.4343e-04, 3.4714e-04,
          4.0460e-04, 8.7814e-03, 7.1526e-04, 3.5133e-03, 5.2166e-04, 5.5838e-04,
          1.8282e-03, 7.3528e-04, 8.3447e-04, 3.5801e-03, 8.8263e-04, 9.3555e-04,
          7.3290e-04, 3.6621e-03, 5.5456e-04, 4.2000e-03, 5.9223e-04, 5.4836e-04,
          6.1150e-03, 2.5654e-03, 4.3654e-04, 3.0975e-03, 4.1533e-04, 3.6907e-04,
          6.5899e-04, 4.4465e-04, 2.3479e-03, 2.6665e-03, 6.8998e-04, 1.1034e-03,
          8.5297e-03, 1.5526e-03, 1.1969e-03, 2.2659e-03, 1.2634e-02, 9.9087e-04,
          3.5057e-03, 4.9829e-04, 2.9964e-03, 5.1498e-04, 9.3508e-04, 8.7500e-04,
          5.5695e-03, 7.6199e-04, 4.9782e-04, 1.0880e-02, 5.0926e-04, 1.0710e-03,
          1.4696e-03, 3.6411e-03, 5.1308e-04, 2.4929e-03, 1.5450e-02, 4.9829e-04,
          6.1178e-04, 5.6839e-04, 9.0790e-04, 1.1215e-02, 5.6190e-03, 5.2147e-03,
          1.7042e-03, 7.9298e-04, 7.2403e-03, 5.7316e-04, 1.7376e-03, 4.3106e-04,
          1.0368e-02, 1.1749e-03, 9.0637e-03, 9.2316e-03, 6.8703e-03, 2.4204e-03,
          7.6914e-04, 7.8888e-03, 1.3189e-03, 9.5129e-04, 1.0357e-03, 7.9117e-03,
          6.7329e-03, 6.9351e-03, 1.4935e-03, 1.5965e-03, 1.4515e-03, 1.1606e-03,
          2.1324e-03, 9.6321e-04, 3.7155e-03, 5.4264e-04, 3.3817e-03, 2.4395e-03,
          7.3891e-03, 3.9940e-03, 7.5531e-03, 1.5411e-02, 1.3256e-03, 1.1654e-03,
          6.3171e-03, 1.6260e-03, 4.9133e-03, 2.1088e-02, 4.4746e-03, 6.2675e-03,
          4.6158e-03, 2.3708e-03, 1.9150e-03, 8.5306e-04, 3.3302e-03, 9.9792e-03,
          2.6855e-03, 5.3711e-03, 7.8964e-03, 7.0152e-03, 5.2910e-03, 2.1759e-02,
          8.3694e-03, 2.0847e-03, 1.0918e-02, 1.1148e-03, 1.2732e-03, 1.4404e-02,
          9.1934e-03, 6.3705e-04, 8.1024e-03, 9.4757e-03, 7.0114e-03, 7.2021e-03,
          2.1877e-03, 1.1663e-03, 6.1646e-03, 1.0004e-03, 2.1076e-03, 2.9297e-03,
          7.4434e-04, 1.4553e-03, 6.5422e-04, 5.1613e-03, 6.2141e-03, 2.6932e-03,
          4.3831e-03, 2.1439e-03, 9.0933e-04, 2.7027e-03, 1.1215e-02, 1.2960e-03,
          3.8090e-03, 5.2376e-03, 1.5144e-02, 4.7516e-02, 2.5094e-05, 8.1420e-05,
          2.5094e-05, 3.6764e-04, 2.5094e-05, 8.1420e-05, 2.5094e-05, 2.5094e-05,
          2.0838e-04, 2.0838e-04, 4.2415e-04, 2.5094e-05, 2.5094e-05, 8.1420e-05,
          2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 1.5211e-04,
          2.5094e-05, 2.5094e-05, 8.1420e-05, 2.5094e-05, 8.1420e-05, 2.0838e-04,
          3.6764e-04, 8.1420e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05,
          8.1420e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 4.6659e-04,
          2.5094e-05, 8.1420e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05,
          2.5094e-05, 2.5094e-05, 2.5094e-05, 8.1420e-05, 2.0850e-04, 2.5094e-05,
          3.3545e-04, 3.3545e-04, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05,
          2.5094e-05, 2.5094e-05, 4.2415e-04, 8.1420e-05, 2.5094e-05, 3.3545e-04,
          2.0850e-04, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.0838e-04, 2.5094e-05,
          1.5211e-04, 2.5094e-05, 2.0838e-04, 2.0838e-04, 2.5094e-05, 2.5094e-05,
          2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05, 2.5094e-05,
          2.5094e-05, 8.1420e-05, 2.0850e-04, 8.1420e-05, 2.5094e-05, 2.5094e-05,
          2.5094e-05, 2.0838e-04, 2.5094e-05, 8.1420e-05, 2.5094e-05, 2.5094e-05,
          8.1420e-05, 2.5094e-05, 4.6659e-04, 2.5094e-05, 8.1420e-05, 2.5094e-05,
          2.5094e-05, 8.1420e-05, 2.0838e-04, 8.1420e-05, 2.5094e-05, 2.5094e-05,
          2.0838e-04, 3.3951e-04, 2.0838e-04, 2.5094e-05, 2.5094e-05, 3.3545e-04,
          2.5094e-05, 4.6659e-04, 2.0838e-04, 8.1420e-05, 2.5094e-05, 2.8324e-04,
          2.5094e-05, 2.5094e-05, 8.1420e-05, 8.0919e-04, 8.1420e-05, 3.3545e-04,
          8.1420e-05, 2.5094e-05, 2.0838e-04, 8.1420e-05, 2.5094e-05, 8.0919e-04,
          8.0919e-04, 4.6659e-04, 2.5094e-05, 2.5094e-05, 2.0850e-04, 2.0850e-04,
          2.5094e-05, 8.1420e-05, 2.5094e-05, 2.5094e-05, 3.3545e-04, 2.5094e-05,
          3.3545e-04, 2.0850e-04, 2.5094e-05, 2.5094e-05, 8.1420e-05, 2.5094e-05,
          2.5094e-05, 2.0838e-04, 2.5094e-05, 8.1420e-05, 2.5094e-05, 2.5094e-05,
          4.6659e-04, 2.5094e-05, 8.1420e-05, 2.5094e-05, 2.0850e-04, 4.6659e-04,
          4.6659e-04, 2.0850e-04, 8.1420e-05, 8.1420e-05, 4.6659e-04, 2.0838e-04,
          2.0850e-04, 8.0919e-04, 5.9366e-04, 2.5094e-05, 2.5094e-05, 8.1420e-05,
          4.6659e-04, 3.3951e-04, 2.5094e-05, 4.6659e-04, 2.0838e-04, 2.5094e-05,
          8.1420e-05, 2.5094e-05, 3.6192e-04, 8.1420e-05, 3.6192e-04, 8.1420e-05,
          8.1420e-05, 8.1420e-05, 8.1420e-05, 8.1420e-05, 8.1420e-05, 3.6192e-04,
          8.1420e-05, 8.1420e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.7607e-03, 5.2765e-02, 1.5190e-02, 1.7838e-02, 3.8055e-02,
          3.9429e-02, 1.0338e-02, 1.6012e-03, 1.0767e-03, 3.7327e-03, 3.7460e-03,
          1.7738e-03, 3.5896e-03, 9.7036e-04, 6.0844e-03, 1.5764e-03, 7.1754e-03,
          2.9716e-03, 7.6914e-04, 4.7898e-04, 5.9414e-04, 4.6492e-04, 3.2578e-03,
          4.1885e-03, 3.0766e-03, 8.5144e-03, 6.9962e-03, 8.2541e-04, 8.0948e-03,
          1.5249e-03, 4.0321e-03, 2.0180e-03, 1.1642e-02, 4.7607e-03, 2.8305e-03,
          4.6654e-03, 9.2888e-04, 5.8517e-03, 7.3318e-03, 6.4516e-04, 2.7294e-03,
          6.0558e-04, 5.1975e-04, 4.8184e-04, 6.2990e-04, 7.7705e-03, 1.3657e-02,
          5.9223e-04, 5.2261e-04, 3.3321e-03, 3.5915e-03, 2.9163e-03, 2.1229e-03,
          5.9366e-04, 1.5574e-03, 1.0500e-03, 1.3962e-03, 1.5268e-03, 1.5182e-03,
          1.4210e-03, 2.4986e-03, 1.3399e-03, 2.8381e-03, 8.3208e-04, 1.2924e-02,
          7.8630e-04, 6.1893e-04, 1.5554e-03, 3.7718e-04, 3.0875e-04, 3.7551e-04,
          5.0735e-04, 9.4681e-03, 9.6703e-04, 3.3836e-03, 5.1308e-04, 5.8126e-04,
          1.5736e-03, 1.4467e-03, 1.0424e-03, 4.4098e-03, 1.0967e-03, 9.3174e-04,
          8.4496e-04, 4.3945e-03, 8.6880e-04, 2.2011e-03, 5.1403e-04, 4.6563e-04,
          3.9444e-03, 3.9482e-03, 3.4809e-04, 2.3479e-03, 5.4026e-04, 5.9271e-04,
          1.2674e-03, 5.8222e-04, 4.2381e-03, 4.9057e-03, 1.2035e-03, 2.1172e-03,
          5.2414e-03, 7.2241e-04, 1.1044e-03, 3.5496e-03, 8.4000e-03, 2.7447e-03,
          3.3665e-03, 8.2779e-04, 4.2801e-03, 4.4346e-04, 6.6757e-04, 1.0023e-03,
          2.0123e-03, 1.8892e-03, 9.1553e-04, 1.4862e-02, 7.4339e-04, 1.9188e-03,
          3.5801e-03, 6.7024e-03, 6.5041e-04, 1.1311e-03, 9.5825e-03, 5.7507e-04,
          1.0977e-03, 7.2002e-04, 1.1358e-03, 1.3756e-02, 5.2834e-03, 4.0359e-03,
          1.1663e-03, 6.5517e-04, 1.0185e-03, 4.3559e-04, 2.0790e-03, 4.4537e-04,
          2.2079e-02, 1.6804e-03, 2.8114e-03, 4.6501e-03, 8.0261e-03, 4.8447e-03,
          1.0614e-03, 1.1330e-03, 1.5287e-03, 1.6832e-03, 9.9850e-04, 9.1171e-03,
          1.0345e-02, 8.5907e-03, 2.2659e-03, 5.2605e-03, 2.4128e-03, 3.3970e-03,
          3.1033e-03, 1.9817e-03, 1.0147e-03, 5.1308e-04, 8.9035e-03, 4.5509e-03,
          1.2024e-02, 7.4310e-03, 1.3626e-02, 9.0256e-03, 2.8267e-03, 1.7519e-03,
          3.6449e-03, 2.5063e-03, 5.6534e-03, 4.7798e-03, 6.4011e-03, 1.4694e-02,
          3.8147e-03, 1.9798e-03, 3.1509e-03, 1.9779e-03, 5.9776e-03, 5.0583e-03,
          2.9812e-03, 4.2038e-03, 8.4381e-03, 4.5929e-03, 1.9665e-03, 3.7270e-03,
          2.9354e-03, 1.6823e-03, 2.4796e-03, 1.9121e-03, 1.2169e-03, 9.6130e-03,
          5.6152e-03, 1.0748e-03, 1.2047e-02, 1.5945e-02, 4.6082e-03, 9.9182e-03,
          2.8954e-03, 3.2597e-03, 8.1711e-03, 1.4305e-03, 1.7824e-03, 3.3741e-03,
          8.2111e-04, 2.5520e-03, 9.6989e-04, 1.3523e-03, 4.5280e-03, 4.5891e-03,
          1.5564e-03, 4.6425e-03, 1.5163e-03, 2.0332e-03, 1.4320e-02, 1.4153e-03,
          7.7095e-03, 1.0170e-02, 5.8327e-03, 3.6678e-03, 3.9917e-02, 9.8467e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.2636e-04, 1.6510e-05, 1.6510e-05,
          2.0826e-04, 9.8467e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 9.8467e-05, 1.6510e-05, 1.6510e-05, 9.8467e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 9.8467e-05,
          1.6510e-05, 1.2636e-04, 1.6510e-05, 9.8467e-05, 1.6510e-05, 1.6510e-05,
          1.2636e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.2636e-04,
          1.6510e-05, 9.8467e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05,
          2.0826e-04, 9.8467e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05, 2.0826e-04,
          2.0826e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          9.8467e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 9.8467e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 2.0826e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04,
          1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 9.8467e-05, 1.6510e-05, 1.2636e-04, 2.0826e-04,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 3.4499e-04, 1.6510e-05,
          1.1816e-03, 8.9836e-04, 1.6510e-05, 1.6510e-05, 2.0826e-04, 2.0826e-04,
          3.4499e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04, 3.4499e-04,
          2.0826e-04, 2.0826e-04, 3.4499e-04, 9.8467e-05, 2.0826e-04, 1.6510e-05,
          9.8467e-05, 2.0826e-04, 1.6510e-05, 4.2701e-04, 1.6510e-05, 1.6510e-05,
          1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05, 2.0826e-04, 1.6510e-05,
          9.8038e-04, 2.0826e-04, 2.0826e-04, 1.6510e-05, 1.6510e-05, 1.2636e-04,
          5.3692e-04, 1.6510e-05, 1.1406e-03, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.2913e-03, 8.9836e-04, 1.6510e-05, 1.6510e-05, 1.6510e-05, 1.6510e-05,
          1.2636e-04, 1.2636e-04, 3.7527e-04, 1.6510e-05, 1.2636e-04, 1.6510e-05,
          1.6510e-05, 2.0826e-04, 1.2636e-04, 1.6510e-05, 1.2636e-04, 3.7527e-04,
          1.6510e-05, 1.6510e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3635e-03, 5.5725e-02, 1.5686e-02, 1.4351e-02, 4.0863e-02,
          4.3457e-02, 1.0078e-02, 2.3556e-03, 1.4725e-03, 2.0161e-03, 3.1338e-03,
          1.3857e-03, 4.6616e-03, 4.7188e-03, 2.1534e-03, 2.0771e-03, 9.5215e-03,
          6.3629e-03, 9.4748e-04, 1.0204e-03, 8.2541e-04, 1.0853e-03, 4.2839e-03,
          3.4618e-03, 1.7509e-03, 8.5144e-03, 6.4888e-03, 1.9493e-03, 8.8120e-03,
          8.4763e-03, 4.2992e-03, 3.1662e-03, 2.9964e-03, 1.5078e-03, 2.7542e-03,
          1.9064e-03, 8.7595e-04, 1.7576e-03, 1.8206e-03, 1.6956e-03, 1.2312e-03,
          7.6437e-04, 5.6744e-04, 7.4720e-04, 4.6134e-04, 7.3318e-03, 2.5387e-03,
          2.4109e-03, 1.9245e-03, 4.7722e-03, 3.2349e-03, 3.5934e-03, 1.5926e-03,
          1.4448e-03, 2.3155e-03, 1.6165e-03, 1.0490e-03, 2.3270e-03, 1.4315e-03,
          1.6642e-03, 6.7139e-03, 2.1038e-03, 2.1000e-03, 8.2541e-04, 3.0231e-03,
          1.5612e-03, 6.9427e-04, 7.5245e-04, 4.0770e-04, 4.2963e-04, 6.5851e-04,
          7.2384e-04, 4.0321e-03, 1.8959e-03, 5.2032e-03, 1.4029e-03, 1.1988e-03,
          2.4471e-03, 1.1663e-03, 1.3361e-03, 1.6975e-03, 1.3733e-03, 1.0080e-03,
          1.1187e-03, 4.9782e-03, 1.7319e-03, 1.4362e-03, 7.0858e-04, 8.9931e-04,
          2.4414e-03, 1.3866e-03, 7.1192e-04, 1.5593e-03, 6.5947e-04, 7.5912e-04,
          3.8834e-03, 1.0624e-03, 3.5019e-03, 3.5915e-03, 1.1396e-03, 1.1997e-03,
          2.8610e-03, 1.0023e-03, 1.0128e-03, 1.6117e-03, 2.6093e-03, 1.0025e-02,
          4.2458e-03, 9.9182e-04, 1.8234e-03, 8.7690e-04, 1.3304e-03, 9.5987e-04,
          1.0653e-03, 1.0042e-03, 3.2177e-03, 3.5400e-03, 1.1015e-03, 1.0483e-02,
          2.7130e-02, 2.3537e-03, 7.4291e-04, 1.4286e-03, 3.0651e-03, 8.4066e-04,
          1.0252e-03, 1.1873e-03, 1.7614e-03, 4.0779e-03, 3.9978e-03, 4.9286e-03,
          2.2850e-03, 8.8167e-04, 6.2084e-04, 5.2547e-04, 1.1578e-03, 7.5817e-04,
          1.1238e-02, 1.0262e-02, 8.3771e-03, 4.8370e-03, 4.6768e-03, 3.1036e-02,
          1.2779e-03, 6.4182e-04, 1.2131e-03, 1.1501e-03, 1.7624e-03, 3.8967e-03,
          3.7861e-03, 3.4981e-03, 1.8377e-03, 4.8141e-03, 4.7722e-03, 1.8425e-03,
          1.3428e-03, 1.2913e-03, 9.4032e-04, 1.2312e-03, 2.9182e-03, 2.0096e-02,
          2.6665e-03, 1.8034e-03, 1.2154e-02, 2.7180e-03, 2.3994e-03, 1.4572e-03,
          2.0638e-03, 1.3628e-03, 2.3708e-03, 1.6260e-03, 1.0414e-02, 3.6125e-03,
          3.4218e-03, 2.9354e-03, 3.1796e-03, 2.3594e-03, 2.4147e-03, 3.6926e-03,
          1.1940e-03, 1.9588e-03, 2.4586e-03, 3.9787e-03, 2.0676e-03, 1.8396e-03,
          2.1000e-03, 2.9877e-02, 2.4433e-03, 2.0866e-03, 2.5005e-03, 2.9545e-03,
          4.0703e-03, 1.0643e-03, 2.7027e-03, 1.3901e-02, 1.9798e-03, 2.6131e-03,
          2.9182e-03, 1.1322e-02, 3.8891e-03, 1.4524e-03, 1.3266e-03, 1.2627e-03,
          1.4868e-03, 1.5554e-03, 1.2236e-03, 1.3227e-03, 2.6112e-03, 4.3983e-03,
          2.3308e-03, 4.6310e-03, 2.3556e-03, 1.9093e-03, 5.4054e-03, 2.4986e-03,
          3.3741e-03, 2.0645e-02, 2.6474e-03, 1.4868e-03, 5.6992e-03, 9.8389e-02,
          1.8418e-05, 1.8418e-05, 6.5863e-05, 4.2439e-04, 1.2118e-04, 7.3731e-05,
          6.5863e-05, 1.2118e-04, 6.5863e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 2.8706e-04, 6.5863e-05, 1.8418e-05, 1.8418e-05, 1.8811e-04,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 2.8706e-04, 1.8418e-05, 1.8418e-05, 6.5863e-05, 1.2118e-04,
          6.5863e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 6.5863e-05,
          1.8418e-05, 1.2118e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 6.5863e-05,
          3.4237e-04, 6.5863e-05, 1.2118e-04, 6.5863e-05, 6.5863e-05, 1.2118e-04,
          1.8418e-05, 1.2118e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.2118e-04,
          1.2118e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          1.8811e-04, 1.8418e-05, 6.5863e-05, 1.9681e-04, 1.9467e-04, 1.8418e-05,
          1.8418e-05, 6.5863e-05, 1.8418e-05, 1.8418e-05, 6.5863e-05, 6.5863e-05,
          1.8418e-05, 6.5863e-05, 2.5225e-04, 1.8418e-05, 6.5863e-05, 1.2118e-04,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 6.5863e-05, 1.2118e-04,
          1.8418e-05, 1.8418e-05, 1.8418e-05, 1.2118e-04, 6.5863e-05, 6.5863e-05,
          1.2118e-04, 6.1083e-04, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          6.5863e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05, 1.8418e-05,
          6.5863e-05, 1.8418e-05, 1.8418e-05, 1.2118e-04, 1.8418e-05, 1.8418e-05,
          1.2118e-04, 7.3731e-05, 1.2118e-04, 1.8418e-05, 4.2439e-04, 7.3731e-05,
          6.5863e-05, 7.3731e-05, 6.5863e-05, 1.2118e-04, 7.3731e-05, 1.8418e-05,
          1.8418e-05, 1.9681e-04, 6.5863e-05, 1.8418e-05, 1.2118e-04, 1.2118e-04,
          7.3731e-05, 1.8418e-05, 1.2118e-04, 2.5845e-04, 1.8418e-05, 6.5863e-05,
          1.2118e-04, 1.2118e-04, 1.2684e-03, 7.3731e-05, 2.0325e-04, 7.3731e-05,
          1.2118e-04, 1.8418e-05, 7.3731e-05, 1.2118e-04, 1.8418e-05, 1.8418e-05,
          6.5863e-05, 2.8706e-04, 1.8418e-05, 1.8418e-05, 1.2118e-04, 1.8418e-05,
          6.5863e-05, 7.3731e-05, 3.8958e-04, 6.5863e-05, 1.8418e-05, 1.8418e-05,
          3.8958e-04, 1.8418e-05, 7.3731e-05, 7.3731e-05, 3.3426e-04, 2.0325e-04,
          1.8418e-05, 2.0325e-04, 1.8418e-05, 6.5863e-05, 1.8418e-05, 1.8418e-05,
          5.5552e-04, 4.1819e-04, 1.9681e-04, 1.9681e-04, 1.9681e-04, 1.9681e-04,
          6.5863e-05, 2.5225e-04, 1.9681e-04, 3.2449e-04, 6.5863e-05, 6.5863e-05,
          1.9681e-04, 1.9681e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.4893e-03, 6.1676e-02, 1.6708e-02, 1.5594e-02, 4.5685e-02,
          4.7485e-02, 1.0529e-02, 1.2445e-03, 9.1791e-04, 2.3842e-03, 2.5406e-03,
          1.0738e-03, 3.2940e-03, 1.8444e-03, 2.1629e-03, 1.3819e-03, 1.2787e-02,
          3.9005e-03, 7.6151e-04, 1.6890e-03, 2.3193e-03, 2.5520e-03, 2.9716e-03,
          2.6779e-03, 2.2831e-03, 8.3389e-03, 5.8823e-03, 2.1858e-03, 8.7128e-03,
          2.2449e-03, 2.8419e-03, 3.9597e-03, 5.9624e-03, 2.3994e-03, 2.4033e-03,
          3.4027e-03, 1.0347e-03, 1.0490e-02, 4.9286e-03, 1.9045e-03, 5.3062e-03,
          1.2312e-03, 1.0729e-03, 7.8583e-04, 1.2398e-03, 6.9962e-03, 7.3280e-03,
          1.5192e-03, 1.5678e-03, 5.4131e-03, 4.0550e-03, 2.3994e-03, 1.5297e-03,
          7.9679e-04, 1.4515e-03, 1.1806e-03, 1.5030e-03, 1.4191e-03, 1.3857e-03,
          1.2674e-03, 7.5760e-03, 2.0752e-03, 4.4289e-03, 1.4458e-03, 5.4741e-03,
          1.1435e-03, 6.9141e-04, 1.4086e-03, 7.3195e-04, 5.6267e-04, 6.7997e-04,
          9.1314e-04, 5.1880e-03, 1.1950e-03, 2.7313e-03, 6.6185e-04, 7.2145e-04,
          1.4400e-03, 1.5726e-03, 1.1110e-03, 2.3956e-03, 9.8991e-04, 9.1887e-04,
          1.0214e-03, 5.9280e-03, 1.3456e-03, 1.4086e-03, 6.4135e-04, 7.1478e-04,
          2.3422e-03, 3.8300e-03, 5.3787e-04, 1.6193e-03, 8.9598e-04, 9.4175e-04,
          3.0098e-03, 7.8154e-04, 3.0956e-03, 1.0033e-02, 1.5049e-03, 2.2202e-03,
          2.3594e-03, 8.0919e-04, 7.0858e-04, 2.4090e-03, 3.2310e-03, 7.0305e-03,
          4.4403e-03, 1.8234e-03, 4.5853e-03, 8.5592e-04, 9.3842e-04, 1.2379e-03,
          1.5068e-03, 2.8400e-03, 2.8934e-03, 5.4550e-03, 1.2112e-03, 4.8409e-03,
          5.9624e-03, 5.7640e-03, 7.1669e-04, 8.2874e-04, 3.3302e-03, 5.7983e-04,
          1.2360e-03, 8.5640e-04, 1.1759e-03, 4.7035e-03, 4.6921e-03, 5.1041e-03,
          1.6680e-03, 7.7677e-04, 5.9128e-04, 5.6982e-04, 1.9588e-03, 8.5926e-04,
          1.3313e-02, 4.0741e-03, 3.7060e-03, 4.3335e-03, 5.2490e-03, 5.9090e-03,
          1.1597e-03, 8.4591e-04, 1.3046e-03, 1.8272e-03, 1.0090e-03, 4.1313e-03,
          5.4245e-03, 3.8567e-03, 2.0638e-03, 1.0292e-02, 3.4618e-03, 6.0654e-03,
          3.3550e-03, 3.3588e-03, 6.5184e-04, 8.9312e-04, 6.7902e-03, 7.7477e-03,
          4.5929e-03, 6.8398e-03, 9.7885e-03, 5.2147e-03, 4.2534e-03, 2.5425e-03,
          1.9255e-03, 2.4853e-03, 2.9869e-03, 1.6832e-03, 5.9166e-03, 7.3128e-03,
          2.8782e-03, 1.9426e-03, 3.5343e-03, 4.1656e-03, 6.1646e-03, 3.4180e-03,
          1.8339e-03, 3.0746e-03, 4.0703e-03, 2.9945e-03, 2.2297e-03, 2.4757e-03,
          1.4982e-03, 3.9787e-03, 2.3441e-03, 2.1858e-03, 1.1921e-03, 3.4409e-03,
          2.7161e-03, 1.1234e-03, 4.0169e-03, 1.1940e-02, 2.0847e-03, 3.8700e-03,
          2.9316e-03, 8.3237e-03, 9.0256e-03, 1.9684e-03, 1.8930e-03, 2.5501e-03,
          1.2140e-03, 3.3836e-03, 1.9293e-03, 1.1606e-03, 2.7561e-03, 4.4594e-03,
          1.6794e-03, 7.0534e-03, 1.9026e-03, 1.2388e-03, 5.4817e-03, 1.2684e-03,
          6.5346e-03, 1.0818e-02, 3.5057e-03, 1.3456e-03, 1.0788e-02, 7.3318e-03,
          9.5398e-02, 2.1040e-05, 2.1040e-05, 8.7738e-05, 8.7738e-05, 2.1040e-05,
          8.7738e-05, 8.7738e-05, 2.1040e-05, 2.7800e-04, 2.1040e-05, 8.7738e-05,
          2.1040e-05, 2.1040e-05, 8.7738e-05, 8.7738e-05, 2.1040e-05, 8.7738e-05,
          8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 8.7738e-05, 2.1040e-05,
          2.1040e-05, 8.7738e-05, 2.1040e-05, 8.7738e-05, 2.1040e-05, 2.1040e-05,
          8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 8.7738e-05,
          2.1040e-05, 2.1040e-05, 8.7738e-05, 8.7738e-05, 2.2006e-04, 2.1040e-05,
          2.2006e-04, 8.7738e-05, 8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.2006e-04, 2.1040e-05, 8.7738e-05,
          8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          6.9046e-04, 2.1040e-05, 2.1040e-05, 2.2006e-04, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          2.1040e-05, 2.1040e-05, 2.2006e-04, 2.1040e-05, 2.1040e-05, 8.7738e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 8.7738e-05,
          2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          3.4475e-04, 2.2006e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          1.5342e-04, 8.7738e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.2006e-04,
          2.1040e-05, 8.7738e-05, 2.2006e-04, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          4.7708e-04, 2.1040e-05, 8.7738e-05, 2.1040e-05, 1.5342e-04, 2.2006e-04,
          2.1040e-05, 8.7738e-05, 8.7738e-05, 2.2006e-04, 2.7800e-04, 2.1040e-05,
          2.1040e-05, 2.2006e-04, 2.1040e-05, 2.1040e-05, 2.2006e-04, 2.2006e-04,
          3.4475e-04, 2.2006e-04, 8.7738e-05, 2.1040e-05, 2.2006e-04, 4.7708e-04,
          2.2006e-04, 2.2006e-04, 2.7800e-04, 8.7738e-05, 8.7738e-05, 2.1040e-05,
          2.2006e-04, 2.2006e-04, 8.7738e-05, 4.7708e-04, 2.1040e-05, 2.1040e-05,
          1.5342e-04, 2.1040e-05, 2.1040e-05, 8.7738e-05, 4.5061e-04, 2.1040e-05,
          1.5764e-03, 2.2006e-04, 2.2006e-04, 2.1040e-05, 2.1040e-05, 1.5342e-04,
          4.7708e-04, 2.1040e-05, 1.5764e-03, 8.7738e-05, 2.2006e-04, 1.5342e-04,
          1.5342e-04, 1.5764e-03, 2.1040e-05, 2.1040e-05, 2.1040e-05, 2.1040e-05,
          1.5342e-04, 1.5342e-04, 6.2609e-04, 2.1040e-05, 1.5342e-04, 2.1040e-05,
          2.1040e-05, 2.2006e-04, 1.5342e-04, 2.1040e-05, 2.1040e-05, 1.5342e-04,
          1.5342e-04, 2.1040e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.7297e-03, 4.6967e-02, 1.9653e-02, 1.6968e-02, 8.4961e-02,
          3.5370e-02, 1.7609e-02, 6.3324e-04, 6.5613e-04, 1.3723e-03, 3.2787e-03,
          1.0834e-03, 7.0333e-04, 1.0405e-03, 1.0824e-03, 6.3658e-04, 1.0777e-03,
          7.2861e-04, 4.9496e-04, 1.5383e-03, 1.3771e-03, 1.2474e-03, 9.0408e-04,
          4.1771e-03, 2.8076e-03, 1.2848e-02, 2.9640e-03, 1.7920e-03, 9.9277e-04,
          9.5844e-04, 8.7404e-04, 5.5075e-04, 1.3866e-03, 7.2241e-04, 6.6805e-04,
          2.6493e-03, 3.1281e-03, 2.0962e-03, 1.1549e-03, 1.2884e-03, 6.6681e-03,
          8.1635e-03, 3.9940e-03, 4.4556e-03, 1.7862e-03, 8.8167e-04, 2.2373e-03,
          6.7377e-04, 1.3161e-03, 1.0971e-02, 1.3142e-03, 7.9823e-04, 6.0606e-04,
          8.1825e-04, 6.5470e-04, 1.4553e-03, 2.9449e-03, 3.1071e-03, 5.7449e-03,
          4.9591e-03, 1.7748e-03, 3.6411e-03, 2.9011e-03, 7.0000e-03, 4.0703e-03,
          3.6049e-03, 1.4181e-03, 1.4114e-03, 2.8343e-03, 3.4733e-03, 2.4681e-03,
          2.4548e-03, 2.4929e-03, 1.8368e-03, 9.7752e-04, 7.8249e-04, 4.9877e-04,
          1.1129e-03, 2.6779e-03, 2.5482e-03, 1.8816e-03, 2.5463e-03, 3.1414e-03,
          3.7651e-03, 2.8591e-03, 2.6340e-03, 1.8215e-03, 1.5039e-03, 2.4147e-03,
          3.6736e-03, 2.8019e-03, 2.4757e-03, 2.2469e-03, 2.8477e-03, 2.6321e-03,
          1.3094e-03, 1.1501e-03, 6.7997e-04, 1.6851e-03, 3.7632e-03, 1.9722e-03,
          2.2049e-03, 9.5510e-04, 9.1934e-04, 3.7251e-03, 1.6861e-03, 2.6512e-03,
          2.4090e-03, 2.0630e-02, 5.1155e-03, 4.5280e-03, 2.4986e-03, 1.9455e-03,
          1.9474e-03, 6.4545e-03, 2.1229e-03, 3.1300e-03, 3.6030e-03, 2.5787e-03,
          2.1610e-03, 2.4548e-03, 2.4357e-03, 3.9983e-04, 1.6851e-03, 1.0891e-03,
          2.4567e-03, 2.2125e-03, 2.1343e-03, 2.1057e-03, 8.5735e-04, 2.6169e-03,
          3.1815e-03, 1.9665e-03, 5.7888e-04, 1.5440e-03, 3.3283e-03, 6.2485e-03,
          1.9741e-03, 2.0618e-03, 1.0757e-03, 2.2316e-03, 1.3275e-03, 1.5888e-03,
          2.8896e-03, 7.6389e-04, 1.5783e-03, 1.5583e-03, 1.9951e-03, 2.6302e-03,
          3.1490e-03, 2.5463e-03, 4.4975e-03, 4.7226e-03, 5.6419e-03, 2.2141e-02,
          3.0918e-03, 4.8141e-03, 7.5817e-04, 7.0343e-03, 7.1068e-03, 2.2793e-03,
          2.4700e-03, 1.8330e-03, 1.3456e-03, 1.4896e-03, 3.9139e-03, 1.5917e-03,
          1.7958e-03, 1.2302e-03, 1.6518e-03, 9.0885e-04, 1.1854e-03, 2.1725e-03,
          4.0283e-03, 5.1689e-03, 7.2098e-03, 4.4250e-02, 7.3547e-03, 1.6375e-03,
          1.9321e-03, 2.1992e-03, 3.0060e-03, 1.2035e-03, 1.0309e-03, 1.0462e-03,
          1.3638e-03, 1.8291e-03, 1.5869e-03, 4.7455e-03, 2.0256e-03, 1.5335e-03,
          7.9393e-04, 2.4891e-03, 1.2665e-03, 1.7824e-03, 1.2321e-03, 1.7061e-03,
          7.5760e-03, 6.2675e-03, 8.5220e-03, 6.9542e-03, 2.0466e-03, 2.7199e-03,
          1.2749e-02, 1.0445e-02, 2.1500e-02, 7.7772e-04, 3.2768e-03, 1.0052e-03,
          1.2083e-03, 5.4016e-03, 1.0384e-02, 1.2274e-03, 2.3556e-03, 2.7657e-03,
          2.1648e-03, 1.8702e-03, 1.3084e-03, 7.1764e-04, 2.0370e-03, 1.8377e-03,
          4.9019e-03, 9.4971e-02, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 1.7071e-04, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.8099e-04, 3.8099e-04, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 3.3319e-05, 3.3319e-05,
          3.8099e-04, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04,
          3.3319e-05, 3.3319e-05, 3.8099e-04, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 1.7071e-04,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 3.3319e-05,
          3.8099e-04, 3.8099e-04, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05, 1.7071e-04, 3.3319e-05,
          2.4378e-04, 1.7071e-04, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          2.4378e-04, 3.3319e-05, 1.7071e-04, 3.3319e-05, 3.3319e-05, 2.4378e-04,
          3.3319e-05, 2.4378e-04, 3.3319e-05, 3.3319e-05, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 2.4378e-04, 1.1855e-04, 2.4378e-04, 3.3319e-05,
          3.3319e-05, 3.3319e-05, 3.3319e-05, 2.9397e-04, 3.3319e-05, 3.3319e-05,
          3.3319e-05, 3.3319e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.8523e-03, 5.5054e-02, 1.3672e-02, 1.4221e-02, 4.2847e-02,
          4.2816e-02, 1.0796e-02, 7.0000e-04, 7.4816e-04, 3.4084e-03, 5.6496e-03,
          2.5215e-03, 7.6771e-04, 5.9271e-04, 1.9722e-03, 7.8344e-04, 9.0790e-04,
          6.1369e-04, 5.7650e-04, 4.3249e-04, 6.3276e-04, 5.4455e-04, 1.1253e-03,
          8.4610e-03, 4.9171e-03, 8.4152e-03, 3.1891e-03, 6.6900e-04, 1.0071e-03,
          7.5579e-04, 1.2331e-03, 4.5276e-04, 2.1400e-03, 1.7786e-03, 8.6737e-04,
          3.9139e-03, 1.8625e-03, 4.4365e-03, 1.8568e-03, 5.5265e-04, 9.9792e-03,
          3.6373e-03, 2.6588e-03, 2.2926e-03, 1.8435e-03, 1.0090e-03, 4.0245e-03,
          4.5586e-04, 6.1274e-04, 4.0474e-03, 1.3380e-03, 9.8801e-04, 7.7629e-04,
          5.5027e-04, 6.8378e-04, 6.5804e-04, 2.7142e-03, 2.0447e-03, 5.9357e-03,
          2.5120e-03, 1.0920e-03, 1.8129e-03, 2.2659e-03, 7.0877e-03, 1.0628e-02,
          3.9825e-03, 2.7351e-03, 2.4815e-03, 1.7939e-03, 1.8806e-03, 1.3475e-03,
          1.2417e-03, 6.2485e-03, 9.0265e-04, 1.2178e-03, 5.0926e-04, 4.5967e-04,
          8.1205e-04, 2.8019e-03, 3.1948e-03, 6.2561e-03, 4.3221e-03, 2.6855e-03,
          1.6518e-03, 3.2291e-03, 1.4410e-03, 4.6616e-03, 2.4128e-03, 2.7466e-03,
          8.4839e-03, 7.5874e-03, 1.8473e-03, 4.8485e-03, 1.7405e-03, 1.3542e-03,
          6.7425e-04, 5.4312e-04, 1.0309e-03, 2.3308e-03, 3.3188e-03, 3.6774e-03,
          6.5918e-03, 1.5745e-03, 2.0885e-03, 9.7733e-03, 4.4937e-03, 1.6441e-03,
          2.6283e-03, 8.1406e-03, 1.2260e-02, 2.6169e-03, 2.5959e-03, 3.7518e-03,
          3.8738e-03, 5.9547e-03, 1.0815e-03, 7.5378e-03, 2.2583e-03, 1.3008e-03,
          1.6441e-03, 6.3362e-03, 1.5688e-03, 5.2404e-04, 5.7869e-03, 1.4229e-03,
          1.9951e-03, 1.9836e-03, 2.4643e-03, 3.7994e-03, 1.1330e-03, 2.8992e-03,
          3.2501e-03, 2.9144e-03, 8.6641e-04, 2.1572e-03, 7.9041e-03, 3.7937e-03,
          3.0956e-03, 1.6546e-03, 1.0595e-03, 2.2926e-03, 2.2068e-03, 1.5154e-03,
          2.5425e-03, 1.1330e-03, 2.6016e-03, 3.1052e-03, 2.3460e-03, 6.0997e-03,
          1.0315e-02, 4.8561e-03, 3.3970e-03, 3.8681e-03, 2.7161e-03, 1.7303e-02,
          8.6517e-03, 7.6218e-03, 1.3275e-03, 5.0888e-03, 1.0681e-02, 2.4357e-03,
          4.7302e-03, 4.4518e-03, 1.5306e-03, 2.5673e-03, 3.6850e-03, 2.7695e-03,
          4.7226e-03, 3.0689e-03, 3.6411e-03, 1.4315e-03, 1.4219e-03, 3.8452e-03,
          3.4027e-03, 2.3060e-03, 3.6907e-03, 1.3382e-02, 1.8784e-02, 2.2411e-03,
          3.7365e-03, 3.5973e-03, 8.6823e-03, 1.6155e-03, 1.1683e-03, 1.2665e-03,
          3.1891e-03, 1.3676e-03, 1.9083e-03, 3.6850e-03, 2.1114e-03, 3.0689e-03,
          1.1158e-03, 2.1210e-03, 2.3918e-03, 2.0313e-03, 3.4218e-03, 3.3550e-03,
          3.3531e-03, 2.8439e-03, 1.2787e-02, 1.3252e-02, 4.5662e-03, 5.6572e-03,
          8.6060e-03, 2.2171e-02, 8.5526e-03, 9.9945e-04, 8.1177e-03, 1.1902e-03,
          1.1368e-03, 4.2191e-03, 3.2291e-03, 2.8782e-03, 4.2152e-03, 2.3842e-03,
          3.8605e-03, 2.2736e-03, 2.5921e-03, 1.0185e-03, 3.2940e-03, 1.7796e-03,
          7.0229e-03, 1.5114e-02, 4.7546e-02, 2.0981e-05, 9.4950e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05, 1.7214e-04,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 9.4950e-05, 9.4950e-05, 1.7214e-04, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 1.7214e-04, 2.0981e-05,
          9.4950e-05, 2.0981e-05, 1.7214e-04, 1.7214e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 1.7214e-04, 9.4950e-05, 1.7214e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 9.8228e-05, 2.0981e-05, 9.8228e-05, 2.0981e-05, 2.0981e-05,
          1.7214e-04, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 9.8228e-05, 1.7214e-04, 2.0981e-05, 2.0981e-05,
          1.7214e-04, 1.7214e-04, 9.8228e-05, 2.0981e-05, 2.0981e-05, 1.7214e-04,
          2.0981e-05, 1.7214e-04, 1.7285e-04, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 1.7214e-04, 9.8228e-05, 2.0981e-05, 1.7214e-04,
          2.0981e-05, 2.0981e-05, 1.7214e-04, 2.0981e-05, 2.0981e-05, 9.8228e-05,
          9.8228e-05, 9.4950e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05, 9.4950e-05,
          9.4950e-05, 9.8228e-05, 2.0981e-05, 2.0981e-05, 1.7214e-04, 2.0981e-05,
          1.7214e-04, 1.7214e-04, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 1.7214e-04, 2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 2.0981e-05, 9.4950e-05, 2.0981e-05,
          1.7214e-04, 1.7214e-04, 2.0981e-05, 9.8228e-05, 2.0981e-05, 9.8228e-05,
          1.7214e-04, 2.0981e-05, 1.7214e-04, 2.0981e-05, 2.0981e-05, 9.8228e-05,
          9.8228e-05, 9.8228e-05, 2.0981e-05, 9.8228e-05, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 3.3426e-04, 9.8228e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05, 2.0981e-05, 3.3426e-04, 2.0981e-05, 2.0981e-05,
          2.0981e-05, 2.0981e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6921e-03, 5.4138e-02, 1.3039e-02, 1.7853e-02, 3.4760e-02,
          3.7628e-02, 1.1421e-02, 9.7036e-04, 6.6137e-04, 2.9335e-03, 8.0109e-03,
          2.0638e-03, 8.3733e-04, 8.8882e-04, 9.3937e-04, 9.8515e-04, 1.1826e-03,
          9.3985e-04, 8.2445e-04, 2.6226e-04, 3.4285e-04, 4.1318e-04, 1.6727e-03,
          9.9335e-03, 2.4891e-03, 9.2697e-03, 6.3210e-03, 4.2725e-04, 1.1997e-03,
          1.6251e-03, 1.7519e-03, 4.5753e-04, 1.1835e-03, 1.5774e-03, 7.6151e-04,
          1.3447e-03, 8.0633e-04, 2.2793e-03, 1.1168e-03, 4.1485e-04, 3.2940e-03,
          1.8730e-03, 1.5326e-03, 1.8883e-03, 7.7677e-04, 1.5354e-03, 1.7776e-03,
          6.3801e-04, 4.8876e-04, 3.2024e-03, 9.6989e-04, 1.2102e-03, 8.9931e-04,
          4.6587e-04, 7.0763e-04, 4.0317e-04, 8.8358e-04, 1.9064e-03, 1.6937e-03,
          1.2951e-03, 9.7513e-04, 1.3294e-03, 1.3523e-03, 4.3335e-03, 3.7079e-03,
          2.7481e-02, 7.7934e-03, 2.6932e-03, 9.4509e-04, 9.1791e-04, 1.7099e-03,
          7.9966e-04, 7.6866e-03, 5.6410e-04, 1.5659e-03, 5.3406e-04, 4.8232e-04,
          7.1335e-04, 1.0881e-03, 1.0071e-02, 2.8229e-03, 1.0559e-02, 7.9107e-04,
          6.0606e-04, 2.9087e-03, 9.8228e-04, 6.3286e-03, 1.2794e-02, 1.0399e-02,
          1.3908e-02, 5.9776e-03, 1.5831e-03, 7.0610e-03, 6.6233e-04, 7.0095e-04,
          4.8876e-04, 3.2544e-04, 1.3695e-03, 1.6632e-03, 1.5068e-03, 1.8559e-03,
          8.5907e-03, 8.4839e-03, 6.1378e-03, 1.8415e-03, 1.5841e-03, 9.3555e-04,
          2.4147e-03, 2.6588e-03, 9.3536e-03, 4.4403e-03, 1.7899e-02, 1.2688e-02,
          2.2354e-03, 1.5926e-03, 7.4720e-04, 2.9278e-03, 1.0233e-03, 7.9060e-04,
          1.2770e-03, 2.8992e-03, 1.3571e-03, 7.2289e-04, 2.4242e-03, 3.0804e-03,
          7.6389e-04, 6.3133e-03, 1.0330e-02, 1.7738e-03, 1.7786e-03, 2.8076e-03,
          2.0416e-02, 1.5228e-02, 1.1511e-03, 5.4855e-03, 9.6741e-03, 1.4954e-03,
          3.1185e-03, 1.8368e-03, 1.9350e-03, 3.5686e-03, 3.0136e-03, 1.6956e-03,
          2.6112e-03, 1.1406e-03, 1.9627e-03, 1.8492e-03, 1.1284e-02, 2.2907e-03,
          6.2408e-03, 2.0142e-03, 1.2007e-03, 2.1362e-03, 1.8692e-03, 4.9324e-03,
          9.5673e-03, 5.5428e-03, 2.8706e-03, 5.8937e-03, 4.6196e-03, 1.8501e-03,
          3.2444e-03, 5.5771e-03, 2.5368e-03, 2.7695e-03, 1.5326e-03, 4.0512e-03,
          8.0109e-03, 3.8013e-03, 2.1210e-03, 1.0958e-03, 2.0542e-03, 2.4319e-03,
          1.6050e-03, 1.3790e-03, 1.1921e-03, 3.4847e-03, 6.7329e-03, 3.8815e-03,
          4.1695e-03, 2.2240e-03, 4.7302e-03, 2.9469e-03, 1.8988e-03, 1.3981e-03,
          5.5389e-03, 2.1076e-03, 2.7447e-03, 2.1019e-03, 9.1934e-03, 2.2087e-03,
          1.8950e-03, 9.2745e-04, 1.8740e-03, 3.0975e-03, 6.5994e-03, 2.1305e-03,
          1.3123e-03, 1.3580e-03, 6.3896e-03, 1.5503e-02, 1.5457e-02, 5.3215e-03,
          1.2749e-02, 1.3443e-02, 2.6627e-03, 1.5554e-03, 1.3123e-02, 1.7271e-03,
          2.0447e-03, 1.9455e-03, 2.1095e-03, 7.8964e-03, 2.6932e-03, 4.2076e-03,
          2.2945e-03, 2.1343e-03, 5.7411e-03, 1.0023e-03, 2.7046e-03, 1.9588e-03,
          5.1537e-03, 4.0474e-03, 1.2321e-02, 3.0930e-02, 9.0480e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 9.0480e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 7.5758e-05, 1.4830e-04, 2.1851e-04,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 7.5758e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.4830e-04, 1.4830e-04, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 9.0480e-05, 7.5758e-05, 7.5758e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 7.5758e-05, 7.5758e-05,
          9.0480e-05, 1.8001e-05, 2.9087e-04, 1.4830e-04, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 9.0480e-05, 9.0480e-05, 7.5758e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 2.1851e-04, 7.5758e-05, 7.5758e-05, 1.8001e-05, 1.8001e-05,
          2.9087e-04, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 1.8001e-05, 2.1851e-04, 2.9087e-04, 1.8001e-05, 1.8001e-05,
          2.9087e-04, 7.5758e-05, 2.1851e-04, 1.8001e-05, 1.8001e-05, 2.9087e-04,
          1.8001e-05, 1.4830e-04, 2.9087e-04, 9.0480e-05, 1.8001e-05, 7.5758e-05,
          7.5758e-05, 1.8001e-05, 1.4830e-04, 2.1851e-04, 1.8001e-05, 2.9087e-04,
          1.8001e-05, 1.8001e-05, 2.9087e-04, 1.8001e-05, 3.9339e-04, 2.1851e-04,
          7.5758e-05, 7.5758e-05, 1.8001e-05, 1.8001e-05, 9.0480e-05, 9.0480e-05,
          3.3545e-04, 2.1851e-04, 1.8001e-05, 1.8001e-05, 2.1851e-04, 7.5758e-05,
          1.4830e-04, 1.4830e-04, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05,
          1.8001e-05, 2.1851e-04, 1.8001e-05, 2.5439e-04, 1.8001e-05, 1.8001e-05,
          7.5758e-05, 1.8001e-05, 1.8001e-05, 1.8001e-05, 1.4830e-04, 7.5758e-05,
          2.9087e-04, 2.9087e-04, 7.5758e-05, 7.5758e-05, 7.5758e-05, 2.1851e-04,
          1.4830e-04, 7.5758e-05, 2.9087e-04, 1.8001e-05, 1.8001e-05, 7.5758e-05,
          2.1851e-04, 2.9087e-04, 1.8001e-05, 2.1851e-04, 2.1851e-04, 1.8001e-05,
          7.5758e-05, 7.5758e-05, 7.5758e-05, 9.9838e-05, 4.5967e-04, 7.5758e-05,
          1.8001e-05, 7.5758e-05, 1.8001e-05, 3.9649e-04, 7.5758e-05, 7.5758e-05,
          1.8001e-05, 7.5758e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.9667e-03, 5.4962e-02, 1.6205e-02, 2.1469e-02, 4.3060e-02,
          4.1473e-02, 1.4015e-02, 1.3981e-03, 9.8991e-04, 4.0207e-03, 5.4970e-03,
          1.9894e-03, 1.8473e-03, 6.7139e-04, 4.2076e-03, 1.6632e-03, 2.3003e-03,
          1.6336e-03, 1.0214e-03, 2.8872e-04, 4.8590e-04, 4.3297e-04, 2.8381e-03,
          7.2098e-03, 5.5237e-03, 1.1154e-02, 6.6605e-03, 4.7517e-04, 2.7676e-03,
          9.1505e-04, 2.7351e-03, 7.5150e-04, 5.4665e-03, 4.1771e-03, 1.4162e-03,
          5.5733e-03, 7.3051e-04, 4.5280e-03, 4.5090e-03, 3.4857e-04, 4.4365e-03,
          1.3170e-03, 1.0004e-03, 9.1457e-04, 7.9060e-04, 3.2368e-03, 6.3896e-03,
          4.7779e-04, 5.1403e-04, 4.1656e-03, 1.4076e-03, 1.5869e-03, 1.3943e-03,
          4.5061e-04, 8.9264e-04, 4.6468e-04, 9.1743e-04, 1.3533e-03, 1.3371e-03,
          1.1101e-03, 1.0853e-03, 1.2856e-03, 2.3098e-03, 1.8921e-03, 1.6220e-02,
          1.6241e-03, 1.1988e-03, 2.4548e-03, 5.6839e-04, 5.1641e-04, 6.0511e-04,
          6.6137e-04, 8.9340e-03, 5.4836e-04, 1.7834e-03, 3.8362e-04, 3.9196e-04,
          7.8011e-04, 7.0667e-04, 9.1648e-04, 4.7607e-03, 1.1539e-03, 1.0948e-03,
          7.7868e-04, 4.5815e-03, 9.3079e-04, 3.5267e-03, 9.7418e-04, 9.0313e-04,
          6.8054e-03, 4.7531e-03, 7.9107e-04, 3.7212e-03, 8.3923e-04, 7.9107e-04,
          5.7459e-04, 3.4046e-04, 2.1210e-03, 2.5253e-03, 7.4005e-04, 1.1387e-03,
          4.7989e-03, 6.9761e-04, 8.4162e-04, 5.1155e-03, 1.0345e-02, 1.2131e-03,
          3.6793e-03, 1.5306e-03, 7.0343e-03, 9.1076e-04, 1.2503e-03, 1.6289e-03,
          2.9793e-03, 2.0390e-03, 9.7322e-04, 1.3184e-02, 1.2217e-03, 9.0885e-04,
          1.3628e-03, 3.8338e-03, 4.3607e-04, 8.9407e-04, 2.1484e-02, 3.7408e-04,
          5.2595e-04, 5.7650e-04, 7.5626e-04, 9.5291e-03, 3.3379e-03, 3.8280e-03,
          2.1477e-03, 1.3475e-03, 1.8892e-03, 9.9277e-04, 4.0092e-03, 8.6021e-04,
          8.5983e-03, 1.7118e-03, 2.0523e-03, 4.7226e-03, 5.2605e-03, 1.6928e-03,
          6.6328e-04, 1.0395e-03, 1.1225e-03, 9.9850e-04, 9.4843e-04, 7.6752e-03,
          7.0496e-03, 7.0648e-03, 1.2875e-03, 3.2749e-03, 1.9102e-03, 4.7798e-03,
          5.0163e-03, 3.2043e-03, 1.7366e-03, 1.2159e-03, 8.3160e-03, 3.0804e-03,
          8.4915e-03, 6.0196e-03, 5.1880e-03, 6.6338e-03, 1.6356e-03, 1.4696e-03,
          3.2864e-03, 1.7328e-03, 4.7379e-03, 3.5858e-03, 3.2120e-03, 6.0730e-03,
          2.4071e-03, 1.4811e-03, 1.7138e-03, 2.9774e-03, 1.0872e-02, 5.9662e-03,
          5.7220e-03, 7.7553e-03, 2.2415e-02, 3.8147e-03, 2.0161e-03, 2.3994e-03,
          4.3831e-03, 1.6155e-03, 3.3188e-03, 1.1034e-03, 9.4223e-04, 1.3580e-02,
          2.9907e-03, 5.0640e-04, 4.7340e-03, 5.3329e-03, 3.4065e-03, 5.1727e-03,
          1.5669e-03, 1.3552e-03, 1.4465e-02, 3.6278e-03, 3.2787e-03, 8.1482e-03,
          2.2011e-03, 7.0152e-03, 2.2945e-03, 1.7433e-03, 7.7744e-03, 2.6894e-03,
          1.8005e-03, 2.3270e-03, 1.1625e-03, 1.5411e-03, 7.9956e-03, 8.8263e-04,
          3.0327e-03, 3.1643e-03, 5.8365e-03, 2.6817e-03, 6.7787e-03, 1.8339e-03,
          6.7062e-03, 2.6054e-03, 8.1253e-03, 2.6245e-03, 7.2327e-02, 2.3484e-05,
          7.9811e-05, 3.0899e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.4603e-04,
          2.3484e-05, 7.9811e-05, 1.4603e-04, 2.3484e-05, 2.3484e-05, 1.7464e-04,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 8.9765e-05, 1.4603e-04,
          2.3484e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          7.9811e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 1.0195e-03,
          2.3484e-05, 3.7527e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 7.9811e-05, 1.4603e-04, 1.4603e-04, 2.3484e-05,
          3.7527e-04, 7.9811e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 8.9765e-05, 7.9811e-05, 1.4603e-04, 2.3484e-05, 3.7575e-04,
          1.4603e-04, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          1.7464e-04, 2.3484e-05, 7.9811e-05, 1.4603e-04, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 2.3484e-05, 1.4603e-04, 7.9811e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          7.9811e-05, 2.3484e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05, 2.3484e-05,
          2.3484e-05, 4.8089e-04, 7.9811e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05,
          7.9811e-05, 7.9811e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05, 1.4603e-04,
          2.3484e-05, 2.5821e-04, 1.4651e-04, 8.9765e-05, 2.3484e-05, 2.0146e-04,
          2.5773e-04, 2.3484e-05, 2.6774e-04, 7.9811e-05, 7.9811e-05, 2.5129e-04,
          7.9811e-05, 2.3484e-05, 1.4603e-04, 3.7527e-04, 6.2132e-04, 4.8685e-04,
          4.8685e-04, 1.0853e-03, 2.3484e-05, 2.3484e-05, 3.7527e-04, 1.4603e-04,
          8.9765e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05, 1.4651e-04, 2.3484e-05,
          4.2915e-04, 3.3617e-04, 2.5773e-04, 2.3484e-05, 7.9811e-05, 2.3484e-05,
          2.3484e-05, 4.2915e-04, 2.3484e-05, 7.9811e-05, 2.3484e-05, 2.3484e-05,
          2.5773e-04, 2.3484e-05, 7.9811e-05, 2.3484e-05, 4.0889e-04, 3.0899e-04,
          1.4868e-03, 1.4603e-04, 1.4603e-04, 7.9811e-05, 2.0146e-04, 5.6076e-04,
          7.2479e-04, 4.8685e-04, 4.2915e-04, 7.9811e-05, 7.9811e-05, 7.9811e-05,
          7.3910e-04, 1.6842e-03, 2.3484e-05, 7.9811e-05, 2.5773e-04, 2.3484e-05,
          7.9811e-05, 2.3484e-05, 7.9811e-05, 7.9811e-05, 2.5177e-04, 8.9765e-05,
          2.3484e-05, 8.9765e-05, 7.9811e-05, 3.8648e-04, 2.3484e-05, 2.5177e-04,
          7.9811e-05, 7.9811e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4212e-03, 5.4535e-02, 1.4114e-02, 1.6785e-02, 3.5278e-02,
          4.4891e-02, 1.1299e-02, 1.5173e-03, 9.0218e-04, 4.0932e-03, 5.0430e-03,
          1.9312e-03, 1.9798e-03, 5.7316e-04, 5.9471e-03, 1.8463e-03, 2.3632e-03,
          1.6832e-03, 9.6321e-04, 2.2781e-04, 3.3712e-04, 3.2163e-04, 2.8343e-03,
          6.3286e-03, 3.6983e-03, 8.9722e-03, 6.0387e-03, 4.5967e-04, 3.3493e-03,
          8.4448e-04, 3.4161e-03, 8.0395e-04, 7.9498e-03, 7.2784e-03, 1.6499e-03,
          4.1466e-03, 6.1798e-04, 5.7526e-03, 6.8321e-03, 3.3736e-04, 2.5806e-03,
          7.6008e-04, 5.8603e-04, 6.2466e-04, 5.1737e-04, 4.0054e-03, 7.3357e-03,
          3.9387e-04, 4.1652e-04, 2.7885e-03, 1.6394e-03, 1.8549e-03, 1.9569e-03,
          4.0078e-04, 9.6273e-04, 4.6515e-04, 6.7854e-04, 1.1473e-03, 9.0837e-04,
          9.4080e-04, 1.0414e-03, 1.1044e-03, 1.9627e-03, 9.6941e-04, 1.6296e-02,
          9.2745e-04, 8.3351e-04, 2.0294e-03, 3.1924e-04, 3.1209e-04, 4.0388e-04,
          4.7159e-04, 1.0887e-02, 5.9700e-04, 2.3193e-03, 3.6907e-04, 3.9244e-04,
          7.9012e-04, 5.7077e-04, 6.7854e-04, 3.9635e-03, 7.8249e-04, 7.2527e-04,
          6.0225e-04, 4.4365e-03, 7.4768e-04, 3.6907e-03, 6.6614e-04, 5.5027e-04,
          6.1989e-03, 3.8986e-03, 4.6253e-04, 3.1090e-03, 6.1941e-04, 5.3263e-04,
          5.9414e-04, 3.5286e-04, 3.4485e-03, 2.1935e-03, 6.1417e-04, 9.7561e-04,
          5.2223e-03, 6.1750e-04, 8.1062e-04, 3.2978e-03, 1.2978e-02, 1.2054e-03,
          3.4485e-03, 7.6914e-04, 5.1003e-03, 5.0640e-04, 8.0872e-04, 1.3561e-03,
          3.2616e-03, 1.1673e-03, 6.5899e-04, 1.2291e-02, 7.8201e-04, 1.0290e-03,
          1.7242e-03, 4.2839e-03, 4.4703e-04, 1.3103e-03, 2.1378e-02, 3.6407e-04,
          4.5180e-04, 4.3464e-04, 6.5517e-04, 9.6054e-03, 5.3864e-03, 4.1351e-03,
          1.3943e-03, 9.1457e-04, 2.6951e-03, 7.0143e-04, 2.9259e-03, 4.3941e-04,
          1.1665e-02, 1.2178e-03, 2.1076e-03, 5.3444e-03, 7.8278e-03, 2.3518e-03,
          6.6853e-04, 1.3285e-03, 1.1482e-03, 8.0633e-04, 7.6056e-04, 8.2626e-03,
          6.3591e-03, 5.8479e-03, 9.3365e-04, 2.9831e-03, 1.6098e-03, 2.8172e-03,
          4.5166e-03, 2.1534e-03, 1.9913e-03, 5.7697e-04, 5.9853e-03, 2.4548e-03,
          9.0332e-03, 7.2594e-03, 8.7051e-03, 1.3161e-02, 1.4000e-03, 1.3180e-03,
          3.7556e-03, 1.7195e-03, 5.6992e-03, 5.5351e-03, 3.8071e-03, 7.2403e-03,
          2.4796e-03, 1.3123e-03, 1.3676e-03, 1.5478e-03, 8.5678e-03, 8.8577e-03,
          4.8752e-03, 9.9716e-03, 1.7944e-02, 6.6261e-03, 2.3041e-03, 4.0092e-03,
          5.5542e-03, 1.2598e-03, 3.9978e-03, 1.1511e-03, 9.1887e-04, 1.5160e-02,
          5.8098e-03, 4.9877e-04, 9.0790e-03, 8.7204e-03, 4.8256e-03, 6.7444e-03,
          1.3046e-03, 1.2932e-03, 1.2169e-02, 2.1114e-03, 2.9316e-03, 6.0654e-03,
          1.0986e-03, 3.7403e-03, 1.0757e-03, 2.2011e-03, 6.7444e-03, 3.3054e-03,
          1.8272e-03, 2.3537e-03, 9.9659e-04, 1.7862e-03, 1.0620e-02, 8.2016e-04,
          3.5057e-03, 4.0779e-03, 1.1215e-02, 4.4937e-03, 1.1986e-02, 2.4414e-03,
          8.5754e-03, 1.1196e-03, 4.1351e-03, 1.9112e-03, 3.6438e-02, 2.6230e-02,
          1.8430e-04, 9.1553e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.8430e-04,
          1.5438e-05, 1.5438e-05, 1.0812e-04, 1.5438e-05, 1.5438e-05, 1.0812e-04,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.0812e-04, 1.8430e-04,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          9.1553e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5125e-03,
          1.5438e-05, 1.8430e-04, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.8430e-04, 1.8430e-04, 1.5438e-05,
          1.8430e-04, 1.0812e-04, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 3.6287e-04, 4.1842e-04, 1.8430e-04, 1.5438e-05, 1.8430e-04,
          1.0812e-04, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          3.1519e-04, 1.5438e-05, 9.1553e-05, 5.5265e-04, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.8430e-04, 9.1553e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          9.1553e-05, 1.5438e-05, 9.1553e-05, 1.5438e-05, 1.5438e-05, 1.5438e-05,
          1.5438e-05, 9.1553e-05, 9.1553e-05, 9.1553e-05, 1.5438e-05, 1.5438e-05,
          9.1553e-05, 9.1553e-05, 9.1553e-05, 1.5438e-05, 1.5438e-05, 1.8430e-04,
          1.5438e-05, 4.7684e-04, 1.8430e-04, 1.5438e-05, 1.5438e-05, 3.0804e-04,
          1.5438e-05, 1.5438e-05, 1.0812e-04, 7.8583e-04, 1.5438e-05, 1.8430e-04,
          1.5438e-05, 1.5438e-05, 1.8430e-04, 1.8430e-04, 3.6287e-04, 1.1129e-03,
          1.1129e-03, 4.7684e-04, 1.5438e-05, 1.5438e-05, 1.8430e-04, 1.8430e-04,
          4.5562e-04, 9.1553e-05, 1.5438e-05, 1.5438e-05, 1.8430e-04, 3.6287e-04,
          1.8430e-04, 1.8430e-04, 3.6287e-04, 1.5438e-05, 9.1553e-05, 1.5438e-05,
          1.5438e-05, 4.7684e-04, 1.5438e-05, 4.3917e-04, 1.5438e-05, 1.5438e-05,
          3.8433e-04, 1.5438e-05, 1.5438e-05, 1.5438e-05, 1.8430e-04, 9.1553e-05,
          2.0065e-03, 1.8430e-04, 9.1553e-05, 9.1553e-05, 3.0804e-04, 4.9305e-04,
          5.3167e-04, 7.8583e-04, 4.7684e-04, 1.5438e-05, 1.5438e-05, 9.1553e-05,
          1.1129e-03, 1.8806e-03, 1.5438e-05, 4.9305e-04, 9.1553e-05, 1.5438e-05,
          9.1553e-05, 1.5438e-05, 9.1553e-05, 1.5438e-05, 9.1553e-05, 1.5438e-05,
          1.5438e-05, 1.5438e-05, 1.5438e-05, 3.6860e-04, 1.5438e-05, 9.1553e-05,
          9.1553e-05, 1.5438e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.6305e-03, 5.4047e-02, 1.5381e-02, 2.2202e-02, 4.2938e-02,
          3.8086e-02, 1.3924e-02, 9.9659e-04, 6.3181e-04, 3.3436e-03, 8.8196e-03,
          2.3251e-03, 9.9087e-04, 1.0328e-03, 9.9087e-04, 1.0214e-03, 1.4849e-03,
          1.2827e-03, 7.5293e-04, 3.4499e-04, 4.4370e-04, 5.6553e-04, 1.8091e-03,
          1.1627e-02, 2.0962e-03, 1.1070e-02, 5.9814e-03, 5.4646e-04, 1.5535e-03,
          1.3847e-03, 1.7080e-03, 6.3419e-04, 1.0176e-03, 1.5497e-03, 5.8079e-04,
          1.1263e-03, 9.0790e-04, 1.8568e-03, 9.5320e-04, 4.8780e-04, 4.0627e-03,
          2.4490e-03, 1.7195e-03, 2.3041e-03, 9.8896e-04, 1.9369e-03, 1.7986e-03,
          7.5388e-04, 6.6042e-04, 4.2458e-03, 8.3494e-04, 1.3638e-03, 6.0606e-04,
          5.4646e-04, 8.2207e-04, 4.9400e-04, 9.3174e-04, 1.9817e-03, 2.0332e-03,
          1.4124e-03, 1.0004e-03, 1.2150e-03, 1.0786e-03, 4.8561e-03, 3.8471e-03,
          1.8570e-02, 8.0185e-03, 2.4509e-03, 1.2970e-03, 1.2970e-03, 1.8587e-03,
          1.0986e-03, 5.9128e-03, 6.6566e-04, 1.5736e-03, 5.9271e-04, 4.7493e-04,
          7.8726e-04, 1.0166e-03, 8.3542e-03, 2.4033e-03, 1.3771e-02, 9.0790e-04,
          7.1526e-04, 3.0613e-03, 1.0309e-03, 6.5842e-03, 8.7357e-03, 8.4610e-03,
          1.7334e-02, 5.9700e-03, 1.6499e-03, 8.3618e-03, 8.5974e-04, 9.6416e-04,
          6.2418e-04, 3.9411e-04, 9.7084e-04, 1.2541e-03, 1.4267e-03, 1.6079e-03,
          8.4000e-03, 4.5700e-03, 8.3313e-03, 1.9894e-03, 1.5364e-03, 9.4175e-04,
          2.3975e-03, 3.2635e-03, 8.7357e-03, 3.5667e-03, 1.2604e-02, 8.4686e-03,
          1.7729e-03, 1.9875e-03, 8.8072e-04, 3.0098e-03, 1.3428e-03, 1.0042e-03,
          1.4582e-03, 2.0962e-03, 1.1101e-03, 7.1764e-04, 2.3880e-03, 2.1267e-03,
          7.6818e-04, 6.4926e-03, 7.3204e-03, 1.4400e-03, 1.2808e-03, 2.8419e-03,
          1.3039e-02, 1.7242e-02, 1.0500e-03, 4.7264e-03, 1.0620e-02, 2.1534e-03,
          2.8038e-03, 1.9608e-03, 1.4143e-03, 3.3722e-03, 1.9217e-03, 1.5650e-03,
          1.8663e-03, 7.7820e-04, 1.2026e-03, 1.5192e-03, 7.6866e-03, 2.0218e-03,
          6.0158e-03, 1.5640e-03, 1.2083e-03, 1.9312e-03, 1.8139e-03, 5.4474e-03,
          7.7477e-03, 4.7379e-03, 2.7695e-03, 6.5041e-03, 4.0894e-03, 2.1572e-03,
          2.3651e-03, 3.2806e-03, 2.7313e-03, 1.8215e-03, 1.2569e-03, 2.1076e-03,
          6.4240e-03, 2.3575e-03, 1.5011e-03, 7.7915e-04, 1.6823e-03, 1.4458e-03,
          1.3208e-03, 1.3618e-03, 1.4553e-03, 4.7836e-03, 5.6229e-03, 3.8471e-03,
          2.4433e-03, 2.0065e-03, 4.1008e-03, 2.4815e-03, 1.6422e-03, 9.5701e-04,
          5.3291e-03, 2.0199e-03, 2.0599e-03, 1.7157e-03, 6.1188e-03, 1.7433e-03,
          1.8215e-03, 9.1505e-04, 1.1168e-03, 3.1757e-03, 5.0201e-03, 1.4639e-03,
          1.4563e-03, 1.3237e-03, 4.9706e-03, 1.6373e-02, 1.0330e-02, 3.7212e-03,
          1.4671e-02, 1.6678e-02, 3.9368e-03, 1.4219e-03, 1.1787e-02, 1.4744e-03,
          1.8158e-03, 1.6489e-03, 2.0199e-03, 8.7204e-03, 2.1133e-03, 3.0842e-03,
          1.4887e-03, 1.5984e-03, 4.2648e-03, 7.5722e-04, 1.6718e-03, 1.8349e-03,
          4.1695e-03, 4.8599e-03, 1.0681e-02, 1.1528e-02, 8.1863e-03, 2.8362e-03,
          4.4647e-02, 2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05, 7.0691e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 7.0691e-05, 1.0550e-04, 7.0691e-05,
          2.7001e-05, 2.7001e-05, 7.7486e-05, 7.7486e-05, 1.5593e-04, 2.1029e-04,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 1.2112e-04, 2.7001e-05, 2.7001e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05, 7.0691e-05, 2.7001e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05,
          1.2112e-04, 2.7001e-05, 7.0691e-05, 1.5593e-04, 1.2112e-04, 7.7486e-05,
          2.7001e-05, 2.7001e-05, 1.5593e-04, 2.7001e-05, 2.7001e-05, 7.0691e-05,
          2.7001e-05, 1.2112e-04, 2.7001e-05, 1.2112e-04, 2.7001e-05, 2.7001e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 2.7001e-05, 3.4022e-04, 7.7486e-05,
          7.0691e-05, 7.7486e-05, 2.5392e-04, 1.2112e-04, 2.7001e-05, 7.7486e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 1.5593e-04, 2.7001e-05, 2.7001e-05,
          2.7001e-05, 1.2112e-04, 7.0691e-05, 3.0541e-04, 2.7001e-05, 2.7001e-05,
          1.2112e-04, 3.9458e-04, 7.7486e-05, 1.2112e-04, 2.7001e-05, 2.7001e-05,
          4.3821e-04, 2.7001e-05, 2.7001e-05, 1.0550e-04, 7.7486e-05, 7.0691e-05,
          7.7486e-05, 2.7001e-05, 3.9458e-04, 4.3821e-04, 2.7001e-05, 7.7486e-05,
          2.5392e-04, 3.0541e-04, 2.1029e-04, 2.7001e-05, 2.7001e-05, 5.1880e-04,
          2.7001e-05, 6.0511e-04, 2.1381e-03, 1.0550e-04, 2.7001e-05, 2.1029e-04,
          3.9458e-04, 7.0691e-05, 1.2112e-04, 2.6178e-04, 2.7001e-05, 2.5392e-04,
          2.7001e-05, 7.0691e-05, 4.3821e-04, 7.0691e-05, 3.0541e-04, 2.1029e-04,
          2.6178e-04, 1.2112e-04, 2.7001e-05, 2.7001e-05, 1.0550e-04, 1.0550e-04,
          1.5593e-04, 2.1029e-04, 2.7001e-05, 2.7001e-05, 5.1880e-04, 7.7486e-05,
          1.0185e-03, 1.9872e-04, 2.5487e-04, 7.7486e-05, 7.0691e-05, 7.0691e-05,
          7.0691e-05, 4.3821e-04, 7.0691e-05, 2.1029e-04, 2.7001e-05, 2.7001e-05,
          7.7486e-05, 7.7486e-05, 7.7486e-05, 7.0691e-05, 2.8968e-04, 2.7001e-05,
          4.7302e-04, 5.1880e-04, 2.7001e-05, 7.7486e-05, 2.7001e-05, 4.7493e-04,
          7.0691e-05, 2.7001e-05, 7.0286e-04, 7.0691e-05, 2.7001e-05, 7.7486e-05,
          6.5947e-04, 3.0541e-04, 2.7001e-05, 2.1029e-04, 2.1029e-04, 2.7001e-05,
          2.7001e-05, 2.7001e-05, 2.7001e-05, 3.2496e-04, 2.8443e-04, 7.7486e-05,
          2.7001e-05, 3.2806e-04, 7.7486e-05, 2.9755e-04, 7.7486e-05, 7.7486e-05,
          2.7001e-05, 2.7001e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.1351e-03, 4.9286e-02, 1.5236e-02, 1.3649e-02, 4.8798e-02,
          3.5583e-02, 1.2787e-02, 9.4461e-04, 6.0558e-04, 1.2770e-03, 3.7937e-03,
          1.1435e-03, 7.3814e-04, 1.1320e-03, 9.2506e-04, 9.8705e-04, 8.9788e-04,
          7.1383e-04, 7.4673e-04, 4.7278e-04, 4.7207e-04, 6.4659e-04, 9.3317e-04,
          5.1689e-03, 2.2945e-03, 1.0193e-02, 2.6665e-03, 6.3896e-04, 7.7152e-04,
          1.3771e-03, 1.3447e-03, 3.7289e-04, 8.2922e-04, 6.2990e-04, 5.5981e-04,
          1.0290e-03, 1.1835e-03, 9.7561e-04, 6.8188e-04, 6.3896e-04, 2.1629e-03,
          2.9049e-03, 2.1858e-03, 3.1910e-03, 9.7847e-04, 9.9564e-04, 1.4000e-03,
          7.7343e-04, 9.1028e-04, 5.2643e-03, 9.6464e-04, 9.8228e-04, 6.3086e-04,
          8.4782e-04, 7.3671e-04, 8.9407e-04, 9.1505e-04, 3.5839e-03, 2.3613e-03,
          2.5043e-03, 1.1311e-03, 2.3632e-03, 1.3485e-03, 5.0888e-03, 2.8954e-03,
          2.9297e-02, 4.1428e-03, 1.3094e-03, 1.8854e-03, 2.0084e-03, 4.5624e-03,
          1.8911e-03, 2.4147e-03, 9.1982e-04, 1.3084e-03, 1.0643e-03, 6.0415e-04,
          1.0071e-03, 1.0424e-03, 5.4359e-03, 1.4925e-03, 6.8703e-03, 1.5879e-03,
          1.3447e-03, 3.3016e-03, 1.9779e-03, 2.7142e-03, 8.2855e-03, 1.4900e-02,
          5.5199e-03, 2.4853e-03, 4.1542e-03, 3.2787e-03, 2.2221e-03, 1.8578e-03,
          8.0490e-04, 6.1035e-04, 8.2159e-04, 1.2379e-03, 1.4820e-03, 1.3714e-03,
          3.3875e-03, 4.9210e-03, 2.8076e-03, 1.6003e-03, 1.3609e-03, 1.1997e-03,
          3.1033e-03, 4.6616e-03, 3.8300e-03, 1.4084e-02, 2.4460e-02, 6.3477e-03,
          2.8400e-03, 2.7771e-03, 1.4496e-03, 2.9087e-03, 2.6608e-03, 1.4210e-03,
          1.7052e-03, 1.5240e-03, 1.7118e-03, 7.9203e-04, 1.7595e-03, 2.0771e-03,
          9.3508e-04, 7.5645e-03, 1.2566e-02, 1.5774e-03, 1.2436e-03, 3.5858e-03,
          2.4200e-02, 7.8201e-03, 8.7214e-04, 4.1046e-03, 4.5280e-03, 4.9782e-03,
          1.7233e-03, 2.9335e-03, 1.8625e-03, 2.6207e-03, 1.5717e-03, 1.7529e-03,
          2.8267e-03, 1.1959e-03, 1.4915e-03, 1.3189e-03, 9.5139e-03, 2.1992e-03,
          2.9163e-03, 1.6909e-03, 2.0447e-03, 2.2507e-03, 3.7766e-03, 5.4665e-03,
          3.6678e-03, 3.7804e-03, 1.9569e-03, 1.7227e-02, 4.7531e-03, 2.5196e-03,
          2.1439e-03, 2.0752e-03, 1.6317e-03, 1.4963e-03, 1.4620e-03, 1.9588e-03,
          2.7809e-03, 1.4544e-03, 1.5783e-03, 1.0090e-03, 1.2407e-03, 1.4267e-03,
          3.3646e-03, 5.0850e-03, 2.5520e-03, 5.2757e-03, 3.4771e-03, 2.3632e-03,
          2.0790e-03, 2.1648e-03, 2.9526e-03, 2.7542e-03, 1.6212e-03, 1.3866e-03,
          3.0537e-03, 3.2463e-03, 2.0390e-03, 2.4643e-03, 8.8882e-03, 1.6365e-03,
          1.4257e-03, 1.2550e-03, 1.1406e-03, 1.8768e-03, 2.0752e-03, 1.3733e-03,
          3.7899e-03, 2.3842e-03, 4.7493e-03, 9.7961e-03, 4.4823e-03, 2.4986e-03,
          2.1164e-02, 9.1934e-03, 7.5035e-03, 1.5373e-03, 5.5580e-03, 1.1368e-03,
          2.4357e-03, 2.2240e-03, 3.1719e-03, 4.9477e-03, 2.2278e-03, 5.2261e-03,
          1.5945e-03, 1.6584e-03, 2.3346e-03, 1.0195e-03, 1.7805e-03, 2.6035e-03,
          3.6411e-03, 1.0757e-02, 8.5983e-03, 1.0811e-02, 6.5765e-03, 2.4166e-03,
          1.9989e-02, 7.7637e-02, 2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05, 9.6619e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 1.3459e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.0897e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.0897e-04, 7.6818e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 9.6619e-05, 2.0897e-04, 1.3459e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          9.6619e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.3459e-04,
          2.2292e-05, 1.3459e-04, 1.3459e-04, 2.0897e-04, 2.2292e-05, 2.2292e-05,
          2.0897e-04, 9.6619e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.0897e-04,
          2.2292e-05, 2.0897e-04, 1.3459e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 9.6619e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05,
          2.2292e-05, 2.2292e-05, 2.0897e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05, 9.6619e-05,
          9.6619e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 1.3459e-04, 2.2292e-05,
          2.0897e-04, 2.0897e-04, 2.2292e-05, 2.2292e-05, 1.3459e-04, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 1.3459e-04, 2.2292e-05, 2.2292e-05,
          2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05, 9.6619e-05, 2.2292e-05,
          9.6619e-05, 2.0897e-04, 1.3459e-04, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          9.6619e-05, 2.2292e-05, 2.0897e-04, 2.2292e-05, 2.2292e-05, 1.3459e-04,
          1.3459e-04, 1.9417e-03, 2.2292e-05, 2.2292e-05, 2.2292e-05, 2.2292e-05,
          1.3459e-04, 1.3459e-04, 3.5954e-04, 1.0133e-04, 1.3459e-04, 1.3459e-04,
          2.2292e-05, 2.2292e-05, 1.3459e-04, 7.4005e-04, 3.5954e-04, 3.5954e-04,
          2.2292e-05, 2.2292e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6387e-03, 4.5685e-02, 1.5625e-02, 1.2810e-02, 5.5115e-02,
          3.2196e-02, 1.2543e-02, 8.2731e-04, 7.7581e-04, 1.7843e-03, 3.9291e-03,
          1.3924e-03, 8.6832e-04, 7.4148e-04, 1.4467e-03, 9.4366e-04, 1.0757e-03,
          6.8855e-04, 8.1491e-04, 5.7888e-04, 6.6996e-04, 7.3004e-04, 1.3008e-03,
          6.4087e-03, 5.1384e-03, 9.2697e-03, 2.1152e-03, 5.5790e-04, 9.5987e-04,
          8.7643e-04, 1.1797e-03, 4.0293e-04, 1.3695e-03, 1.0262e-03, 6.7377e-04,
          1.6069e-03, 1.3208e-03, 1.5707e-03, 1.1101e-03, 6.1703e-04, 4.9858e-03,
          4.8637e-03, 3.2463e-03, 3.9978e-03, 2.2488e-03, 1.2197e-03, 3.4447e-03,
          6.7234e-04, 9.9945e-04, 6.7482e-03, 9.1314e-04, 8.3542e-04, 6.6853e-04,
          5.1975e-04, 6.3133e-04, 7.1096e-04, 1.5411e-03, 1.6184e-03, 2.7027e-03,
          2.4796e-03, 1.0729e-03, 1.6203e-03, 1.6212e-03, 6.2065e-03, 6.2752e-03,
          5.3215e-03, 2.9087e-03, 1.8225e-03, 3.7537e-03, 4.2000e-03, 3.9406e-03,
          2.7657e-03, 3.2806e-03, 7.4863e-04, 1.0729e-03, 5.2118e-04, 4.0674e-04,
          7.7152e-04, 1.6003e-03, 2.9182e-03, 1.9779e-03, 3.7098e-03, 1.7977e-03,
          1.5421e-03, 2.5234e-03, 1.4439e-03, 3.1643e-03, 2.3479e-03, 3.9711e-03,
          6.2981e-03, 4.6272e-03, 3.9635e-03, 4.3945e-03, 6.8398e-03, 4.7073e-03,
          6.9809e-04, 5.4359e-04, 8.6260e-04, 1.7490e-03, 1.8740e-03, 1.6680e-03,
          3.5248e-03, 1.4877e-03, 1.6794e-03, 2.0676e-03, 2.2011e-03, 1.5097e-03,
          2.3670e-03, 6.4125e-03, 6.0768e-03, 4.7760e-03, 4.5662e-03, 4.1161e-03,
          4.4594e-03, 7.5378e-03, 2.1400e-03, 6.4240e-03, 7.5378e-03, 1.3905e-03,
          1.4153e-03, 2.4529e-03, 1.6117e-03, 5.8603e-04, 2.6875e-03, 1.1864e-03,
          1.5488e-03, 2.4204e-03, 3.6430e-03, 2.2812e-03, 1.3742e-03, 2.7466e-03,
          4.6196e-03, 3.1490e-03, 1.1349e-03, 2.1420e-03, 5.6534e-03, 7.4196e-03,
          2.9583e-03, 2.7523e-03, 2.2755e-03, 2.9202e-03, 2.0313e-03, 1.6413e-03,
          2.1820e-03, 1.6537e-03, 1.9531e-03, 1.6298e-03, 2.9335e-03, 3.4542e-03,
          3.7155e-03, 2.5692e-03, 4.2419e-03, 3.0060e-03, 2.9030e-03, 1.0612e-02,
          4.8180e-03, 4.7493e-03, 2.0275e-03, 8.0490e-03, 1.0651e-02, 3.4256e-03,
          4.3602e-03, 3.1586e-03, 2.0924e-03, 2.0008e-03, 2.1057e-03, 1.6346e-03,
          3.1815e-03, 1.5650e-03, 2.0657e-03, 1.3790e-03, 1.2112e-03, 2.8763e-03,
          4.7569e-03, 3.8414e-03, 4.8180e-03, 1.0651e-02, 8.1558e-03, 2.3098e-03,
          2.6932e-03, 3.1605e-03, 4.6387e-03, 2.2736e-03, 1.7700e-03, 2.2316e-03,
          3.6182e-03, 3.0823e-03, 1.9398e-03, 2.5291e-03, 2.7714e-03, 2.1381e-03,
          1.2827e-03, 1.7281e-03, 1.7920e-03, 2.4433e-03, 2.6550e-03, 2.2182e-03,
          5.1918e-03, 2.8839e-03, 8.8959e-03, 8.1635e-03, 3.4237e-03, 3.7918e-03,
          1.2619e-02, 1.3840e-02, 3.2898e-02, 1.4524e-03, 7.7171e-03, 1.5192e-03,
          1.7414e-03, 2.8038e-03, 3.1185e-03, 2.1381e-03, 2.9984e-03, 3.5591e-03,
          2.2831e-03, 2.0752e-03, 2.4719e-03, 1.1787e-03, 2.7142e-03, 2.4357e-03,
          5.2834e-03, 1.5961e-02, 1.5388e-02, 4.5395e-03, 1.1383e-02, 2.8896e-03,
          9.5978e-03, 1.4030e-02, 6.8726e-02, 2.6286e-05, 2.6286e-05, 1.1510e-04,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 9.5189e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 1.8406e-04, 1.8406e-04, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 1.8406e-04, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 9.5189e-05, 1.8406e-04, 1.8406e-04, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 2.6286e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 9.5189e-05, 1.8406e-04, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          2.6286e-05, 9.5189e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          9.5189e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 9.5189e-05, 9.5189e-05,
          9.5189e-05, 9.5189e-05, 9.5189e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          1.8406e-04, 9.5189e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05, 1.8406e-04,
          2.6286e-05, 9.5189e-05, 9.5785e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05,
          9.5189e-05, 2.6286e-05, 1.8406e-04, 9.5189e-05, 2.6286e-05, 1.8406e-04,
          2.6286e-05, 2.6286e-05, 1.8406e-04, 2.6286e-05, 9.5189e-05, 9.5189e-05,
          9.5189e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05, 1.8406e-04, 1.1510e-04,
          2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05, 1.8454e-04, 9.5189e-05,
          1.8406e-04, 1.8406e-04, 2.6286e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05,
          9.5189e-05, 9.5189e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          9.5189e-05, 2.6286e-05, 2.6286e-05, 2.6286e-05, 1.8406e-04, 2.6286e-05,
          9.5189e-05, 1.8406e-04, 9.5189e-05, 9.5189e-05, 2.6286e-05, 9.5189e-05,
          1.8406e-04, 9.5189e-05, 1.8406e-04, 2.6286e-05, 2.6286e-05, 9.5189e-05,
          9.5189e-05, 9.5189e-05, 2.6286e-05, 9.5189e-05, 2.6286e-05, 2.6286e-05,
          9.5189e-05, 9.5189e-05, 9.5189e-05, 4.2844e-04, 9.5189e-05, 2.6286e-05,
          2.6286e-05, 9.5189e-05, 9.5189e-05, 8.7214e-04, 4.2319e-04, 8.7738e-04,
          9.5189e-05, 2.6286e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.0588e-03, 5.6183e-02, 1.5175e-02, 1.3771e-02, 4.6631e-02,
          4.8553e-02, 1.1253e-02, 1.4591e-03, 9.2888e-04, 1.7824e-03, 3.7422e-03,
          1.3905e-03, 1.1454e-03, 6.6423e-04, 2.3441e-03, 1.5583e-03, 1.1148e-03,
          7.3385e-04, 9.3937e-04, 4.2081e-04, 4.7016e-04, 5.5790e-04, 1.1683e-03,
          6.0539e-03, 7.0992e-03, 8.8730e-03, 2.8648e-03, 5.0068e-04, 9.5558e-04,
          7.3624e-04, 2.0809e-03, 3.7384e-04, 2.2411e-03, 9.6130e-04, 1.6184e-03,
          2.4090e-03, 1.7014e-03, 1.4677e-03, 1.4753e-03, 4.5490e-04, 2.7046e-03,
          2.0771e-03, 1.7033e-03, 1.8396e-03, 1.4038e-03, 1.2817e-03, 3.7975e-03,
          5.0735e-04, 7.2908e-04, 5.3673e-03, 1.8110e-03, 1.2951e-03, 1.1272e-03,
          7.1573e-04, 8.9598e-04, 8.6069e-04, 1.4620e-03, 3.4065e-03, 3.2082e-03,
          4.1924e-03, 1.1177e-03, 2.2163e-03, 2.0924e-03, 2.9583e-03, 7.7209e-03,
          2.4986e-03, 1.6747e-03, 1.1969e-03, 1.2264e-03, 1.2274e-03, 1.9684e-03,
          1.7118e-03, 3.4332e-03, 9.0170e-04, 1.8368e-03, 7.9012e-04, 5.4598e-04,
          9.6893e-04, 1.1320e-03, 1.3218e-03, 2.4319e-03, 1.5039e-03, 3.6373e-03,
          2.4643e-03, 3.9902e-03, 1.5087e-03, 2.4376e-03, 1.1244e-03, 1.5726e-03,
          4.6844e-03, 3.1185e-03, 2.1954e-03, 3.2501e-03, 4.6310e-03, 4.2458e-03,
          6.9523e-04, 5.9843e-04, 1.9932e-03, 2.3193e-03, 1.3342e-03, 1.4801e-03,
          3.5496e-03, 1.1234e-03, 9.2363e-04, 3.2082e-03, 4.6997e-03, 1.6403e-03,
          3.6430e-03, 2.0924e-03, 3.9711e-03, 2.3727e-03, 2.6741e-03, 1.9321e-03,
          8.1711e-03, 3.1452e-03, 1.6050e-03, 7.8278e-03, 5.3444e-03, 1.2350e-03,
          1.6003e-03, 3.3817e-03, 9.5654e-04, 1.0633e-03, 4.7607e-03, 5.8508e-04,
          9.5272e-04, 1.1587e-03, 1.9970e-03, 4.3869e-03, 2.2297e-03, 4.0512e-03,
          3.6583e-03, 1.3790e-03, 1.5373e-03, 1.1330e-03, 3.2959e-03, 1.8053e-03,
          3.2024e-03, 2.5005e-03, 3.1967e-03, 3.5267e-03, 2.1992e-03, 1.4982e-03,
          1.3247e-03, 2.1610e-03, 1.8082e-03, 1.3723e-03, 1.4057e-03, 1.1246e-02,
          4.6692e-03, 4.1733e-03, 5.5008e-03, 3.1681e-03, 3.4065e-03, 4.1580e-03,
          2.7313e-03, 2.2469e-03, 1.9236e-03, 2.4090e-03, 9.7275e-03, 3.0689e-03,
          4.9019e-03, 3.1586e-03, 2.0409e-03, 2.9297e-03, 1.9341e-03, 1.2569e-03,
          2.8610e-03, 1.2465e-03, 4.5586e-03, 2.9163e-03, 1.3275e-03, 2.7714e-03,
          3.0502e-02, 2.1515e-02, 9.6130e-03, 3.3455e-03, 6.0577e-03, 2.9087e-03,
          2.8419e-03, 3.9787e-03, 5.9929e-03, 3.9940e-03, 2.4014e-03, 2.8172e-03,
          4.1695e-03, 3.2139e-03, 2.5368e-03, 1.8415e-03, 1.5574e-03, 3.0270e-03,
          2.1744e-03, 9.5987e-04, 2.2392e-03, 2.3899e-03, 2.4567e-03, 2.3613e-03,
          1.7258e-02, 2.3251e-03, 9.3002e-03, 3.3474e-03, 2.2583e-03, 3.7880e-03,
          3.7804e-03, 6.1150e-03, 6.1989e-03, 2.1782e-03, 7.8430e-03, 1.5631e-03,
          3.1223e-03, 2.7294e-03, 1.8864e-03, 1.8578e-03, 5.5237e-03, 1.8568e-03,
          2.7008e-03, 2.6302e-03, 2.5406e-03, 2.2507e-03, 3.2883e-03, 2.3823e-03,
          5.7793e-03, 6.2447e-03, 7.9727e-03, 3.0823e-03, 1.3351e-02, 3.6068e-03,
          3.1319e-03, 8.2703e-03, 9.5520e-03, 9.6741e-02, 2.5451e-05, 1.2898e-04,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 1.2898e-04, 2.5451e-05,
          2.5451e-05, 1.6868e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 1.2898e-04, 1.2898e-04, 2.5451e-05,
          1.6868e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 1.2898e-04, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.7204e-04, 2.7204e-04, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.7204e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 1.6868e-04,
          2.5451e-05, 1.6868e-04, 2.5451e-05, 1.2898e-04, 2.5451e-05, 2.5451e-05,
          2.7204e-04, 2.5451e-05, 1.6868e-04, 2.5451e-05, 2.5451e-05, 1.2898e-04,
          2.5451e-05, 2.5451e-05, 1.2898e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 1.6868e-04, 1.6868e-04, 1.2898e-04,
          2.5451e-05, 2.5451e-05, 2.7204e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          1.6868e-04, 1.6868e-04, 2.5451e-05, 2.5451e-05, 2.7204e-04, 2.7204e-04,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.7204e-04, 2.5451e-05,
          2.7204e-04, 2.7204e-04, 1.6868e-04, 2.5451e-05, 1.6868e-04, 2.5451e-05,
          1.6868e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.7204e-04, 2.5451e-05,
          1.6868e-04, 1.2898e-04, 1.6868e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          1.6868e-04, 1.6868e-04, 2.7204e-04, 2.5451e-05, 1.6868e-04, 1.6868e-04,
          1.6868e-04, 1.6868e-04, 2.5451e-05, 2.5451e-05, 2.5451e-05, 2.5451e-05,
          1.1625e-03, 6.6900e-04, 1.6868e-04, 2.5010e-04, 1.6868e-04, 2.5451e-05,
          2.5451e-05, 2.5451e-05, 1.6868e-04, 1.6575e-03, 3.9887e-04, 1.0424e-03,
          2.5451e-05, 1.6868e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.2648e-03, 5.2856e-02, 1.3298e-02, 1.4435e-02, 4.2480e-02,
          4.9408e-02, 1.0620e-02, 1.0796e-03, 8.1921e-04, 2.7599e-03, 3.1719e-03,
          1.4277e-03, 1.3180e-03, 5.6744e-04, 4.0741e-03, 1.2503e-03, 1.5364e-03,
          9.2697e-04, 7.4434e-04, 4.1485e-04, 6.5136e-04, 5.3358e-04, 2.0447e-03,
          4.6577e-03, 8.4381e-03, 8.6746e-03, 4.9782e-03, 5.6648e-04, 1.7109e-03,
          6.6805e-04, 1.9054e-03, 6.3038e-04, 6.4392e-03, 2.5959e-03, 2.1591e-03,
          4.8141e-03, 1.2836e-03, 4.7607e-03, 4.5967e-03, 4.7970e-04, 5.4893e-03,
          1.6756e-03, 1.4267e-03, 1.1225e-03, 1.3103e-03, 1.7033e-03, 1.0170e-02,
          4.7302e-04, 6.7949e-04, 5.4626e-03, 2.5444e-03, 1.2808e-03, 1.3180e-03,
          5.0354e-04, 8.1921e-04, 7.2527e-04, 1.4200e-03, 1.5182e-03, 1.9331e-03,
          1.6193e-03, 1.5268e-03, 2.0370e-03, 3.8147e-03, 2.1572e-03, 1.3481e-02,
          1.4238e-03, 1.0910e-03, 2.2278e-03, 1.0052e-03, 8.9264e-04, 9.1553e-04,
          9.7847e-04, 6.2180e-03, 8.2588e-04, 1.5688e-03, 4.2200e-04, 4.7278e-04,
          8.5115e-04, 1.2865e-03, 7.9298e-04, 2.9335e-03, 8.4496e-04, 1.5268e-03,
          1.2894e-03, 4.0359e-03, 1.3990e-03, 2.3327e-03, 8.4400e-04, 8.9884e-04,
          4.2229e-03, 3.6011e-03, 9.6893e-04, 2.7046e-03, 1.8959e-03, 1.3533e-03,
          7.1096e-04, 5.4836e-04, 1.8864e-03, 4.4746e-03, 1.3781e-03, 1.6470e-03,
          2.9831e-03, 7.4053e-04, 6.6185e-04, 3.6087e-03, 7.5951e-03, 1.9779e-03,
          3.4161e-03, 2.3308e-03, 5.2261e-03, 1.2665e-03, 1.2407e-03, 1.3857e-03,
          5.2567e-03, 4.3335e-03, 1.6327e-03, 1.4236e-02, 1.8559e-03, 1.4200e-03,
          2.0199e-03, 6.3438e-03, 6.1846e-04, 7.0333e-04, 6.6833e-03, 4.1604e-04,
          9.6416e-04, 6.1750e-04, 9.1124e-04, 8.1711e-03, 2.6932e-03, 3.7804e-03,
          1.8482e-03, 9.8515e-04, 1.2493e-03, 8.9264e-04, 3.0365e-03, 1.1263e-03,
          8.2016e-03, 2.3136e-03, 2.4529e-03, 3.3646e-03, 4.2305e-03, 2.0428e-03,
          9.3699e-04, 1.5945e-03, 1.6327e-03, 1.3914e-03, 8.6021e-04, 9.3536e-03,
          5.1155e-03, 6.3019e-03, 2.4109e-03, 5.7373e-03, 2.8114e-03, 7.1259e-03,
          3.4580e-03, 3.3875e-03, 1.2493e-03, 1.3685e-03, 1.5656e-02, 4.5891e-03,
          1.0895e-02, 6.3934e-03, 3.2997e-03, 5.2757e-03, 3.7880e-03, 1.6165e-03,
          2.2526e-03, 1.6870e-03, 5.3902e-03, 4.0131e-03, 2.4414e-03, 7.6790e-03,
          5.9395e-03, 2.5959e-03, 3.8376e-03, 5.0163e-03, 1.3260e-02, 3.6411e-03,
          3.7289e-03, 5.7106e-03, 1.0422e-02, 3.1147e-03, 1.9608e-03, 3.8433e-03,
          3.1929e-03, 2.6817e-03, 2.3232e-03, 1.8463e-03, 9.5844e-04, 5.5656e-03,
          2.0962e-03, 9.6893e-04, 5.2528e-03, 4.1008e-03, 2.3117e-03, 5.1422e-03,
          3.5305e-03, 2.7370e-03, 1.6190e-02, 2.8324e-03, 2.1019e-03, 5.6763e-03,
          2.1114e-03, 5.9814e-03, 4.3602e-03, 1.6222e-03, 6.1760e-03, 3.1376e-03,
          1.9121e-03, 4.7684e-03, 1.8549e-03, 1.1158e-03, 8.9188e-03, 9.3889e-04,
          6.3286e-03, 4.8790e-03, 3.4904e-03, 2.2755e-03, 1.0391e-02, 2.6855e-03,
          1.1169e-02, 4.0894e-03, 1.0071e-02, 2.3956e-03, 1.6556e-02, 6.2027e-03,
          2.0390e-03, 2.2545e-03, 4.1084e-03, 1.1360e-02, 5.2063e-02, 1.1098e-04,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 1.1098e-04, 1.4293e-04,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 1.4293e-04,
          2.2948e-05, 1.1098e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.1098e-04, 2.3091e-04, 2.3091e-04, 2.2948e-05,
          1.1098e-04, 1.1098e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.3091e-04, 2.2948e-05, 1.1098e-04,
          1.1098e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          1.1098e-04, 2.2948e-05, 2.3091e-04, 2.3091e-04, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.3091e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 1.1098e-04,
          1.4293e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.3091e-04, 1.4293e-04, 1.4293e-04, 2.2948e-05, 2.2948e-05,
          2.3091e-04, 1.1098e-04, 1.4293e-04, 2.2948e-05, 2.2948e-05, 2.3150e-04,
          2.2948e-05, 2.3091e-04, 2.3150e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.1098e-04, 2.2948e-05, 2.2948e-05, 2.3091e-04,
          2.2948e-05, 2.2948e-05, 2.3091e-04, 2.2948e-05, 5.9557e-04, 1.4293e-04,
          1.4293e-04, 1.4293e-04, 2.2948e-05, 2.2948e-05, 2.3091e-04, 2.3091e-04,
          5.9557e-04, 1.4293e-04, 2.2948e-05, 2.2948e-05, 2.3150e-04, 2.2948e-05,
          2.3091e-04, 2.3091e-04, 2.2948e-05, 1.1098e-04, 1.1098e-04, 2.2948e-05,
          1.1098e-04, 2.3091e-04, 1.1098e-04, 1.4293e-04, 2.2948e-05, 2.2948e-05,
          1.4293e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.3091e-04, 2.2948e-05,
          2.3091e-04, 2.3091e-04, 2.3091e-04, 2.2948e-05, 2.2948e-05, 1.4293e-04,
          2.3091e-04, 2.2948e-05, 2.3091e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.1286e-03, 1.4293e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.4293e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 1.1098e-04, 1.4293e-04, 4.2200e-04, 2.2948e-05, 5.9271e-04,
          2.2948e-05, 2.2948e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1613e-03, 5.6244e-02, 2.1194e-02, 1.5900e-02, 4.2664e-02,
          3.9978e-02, 1.2604e-02, 2.0199e-03, 1.0347e-03, 1.4477e-03, 2.7256e-03,
          9.7036e-04, 4.9019e-03, 4.3945e-03, 1.4458e-03, 2.1000e-03, 1.2421e-02,
          6.7711e-03, 1.0529e-03, 1.2112e-03, 1.1063e-03, 2.9907e-03, 4.6463e-03,
          3.4027e-03, 2.0161e-03, 1.0376e-02, 5.8441e-03, 2.1210e-03, 8.1329e-03,
          6.8741e-03, 3.4523e-03, 2.9030e-03, 1.9016e-03, 9.9468e-04, 1.8330e-03,
          1.0834e-03, 5.5838e-04, 1.5650e-03, 1.1835e-03, 1.6623e-03, 1.5888e-03,
          1.1568e-03, 9.5701e-04, 1.0738e-03, 7.7486e-04, 8.9493e-03, 2.8439e-03,
          3.3455e-03, 3.0785e-03, 6.1378e-03, 2.2163e-03, 2.2430e-03, 1.2808e-03,
          7.6818e-04, 1.2665e-03, 8.0061e-04, 5.0545e-04, 1.1778e-03, 7.9012e-04,
          9.5749e-04, 5.4665e-03, 1.7662e-03, 1.9321e-03, 1.1234e-03, 3.6354e-03,
          2.3136e-03, 1.1034e-03, 1.0710e-03, 8.0442e-04, 6.9427e-04, 1.2035e-03,
          1.3847e-03, 3.0937e-03, 1.0328e-03, 3.0861e-03, 7.3099e-04, 5.8460e-04,
          1.3599e-03, 5.0545e-04, 7.7343e-04, 8.0061e-04, 7.0810e-04, 6.3467e-04,
          6.4945e-04, 4.9896e-03, 1.3132e-03, 1.4744e-03, 1.0376e-03, 1.1330e-03,
          2.6302e-03, 1.6537e-03, 1.1616e-03, 1.9760e-03, 1.3161e-03, 1.7977e-03,
          3.5744e-03, 6.2513e-04, 2.8515e-03, 2.2354e-03, 5.1498e-04, 5.7030e-04,
          1.7748e-03, 8.1158e-04, 5.6458e-04, 8.1587e-04, 1.6890e-03, 7.4310e-03,
          3.9330e-03, 1.1082e-03, 1.9970e-03, 1.3132e-03, 1.7405e-03, 1.3266e-03,
          1.5879e-03, 1.6508e-03, 7.5150e-03, 4.3793e-03, 2.6264e-03, 8.4381e-03,
          1.6678e-02, 1.1520e-03, 4.0483e-04, 1.0881e-03, 2.1915e-03, 3.8147e-04,
          3.9101e-04, 5.1832e-04, 7.4816e-04, 2.0256e-03, 3.4161e-03, 4.5547e-03,
          3.0689e-03, 1.2007e-03, 7.8249e-04, 8.0633e-04, 1.3704e-03, 9.8515e-04,
          1.3733e-02, 2.0767e-02, 1.1780e-02, 6.1378e-03, 3.5248e-03, 1.6846e-02,
          5.8126e-04, 6.1321e-04, 5.5552e-04, 5.4932e-04, 9.7036e-04, 1.7204e-03,
          1.9569e-03, 1.5125e-03, 7.9012e-04, 3.5954e-03, 3.0537e-03, 2.0428e-03,
          1.4763e-03, 1.5478e-03, 1.2188e-03, 1.3800e-03, 4.7874e-03, 3.7170e-02,
          3.2177e-03, 2.6226e-03, 1.3641e-02, 2.5444e-03, 1.1559e-03, 8.0252e-04,
          1.4582e-03, 7.8154e-04, 1.0881e-03, 1.2474e-03, 6.5117e-03, 1.5297e-03,
          1.4648e-03, 1.3037e-03, 1.4057e-03, 2.0142e-03, 2.2526e-03, 3.5496e-03,
          1.3103e-03, 2.3880e-03, 2.8152e-03, 4.6272e-03, 2.9297e-03, 3.0956e-03,
          2.5902e-03, 3.8391e-02, 2.8572e-03, 8.3923e-04, 1.0166e-03, 1.8597e-03,
          2.6913e-03, 3.9554e-04, 1.3618e-03, 1.3649e-02, 1.3752e-03, 1.4210e-03,
          1.4973e-03, 5.2452e-03, 3.9330e-03, 1.5841e-03, 1.6756e-03, 1.7576e-03,
          1.7366e-03, 1.9855e-03, 2.3136e-03, 1.6537e-03, 3.7155e-03, 7.2784e-03,
          3.5286e-03, 2.1172e-03, 1.0433e-03, 9.8419e-04, 2.2125e-03, 7.9346e-04,
          1.2827e-03, 9.0714e-03, 2.5539e-03, 1.3275e-03, 2.7046e-03, 2.8152e-02,
          6.9885e-03, 1.7548e-03, 2.2144e-03, 2.2011e-03, 6.3286e-03, 2.6112e-03,
          2.1992e-03, 2.5845e-03, 2.0256e-03, 4.5395e-03, 3.6354e-03, 7.8674e-02,
          2.2113e-05, 1.2565e-04, 7.8619e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 1.2565e-04, 2.2113e-05, 7.8619e-05, 1.2565e-04, 1.2565e-04,
          1.2565e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 1.2565e-04,
          2.2113e-05, 1.8215e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05, 2.2113e-05,
          1.8215e-04, 1.2565e-04, 1.8215e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05, 2.2113e-05, 1.8311e-04,
          1.8215e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          3.3402e-04, 2.2113e-05, 1.2565e-04, 1.8215e-04, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 1.2565e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 2.2113e-05, 7.8619e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05, 7.8619e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 1.2565e-04, 1.2565e-04, 1.2565e-04,
          1.8215e-04, 4.2629e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          2.2113e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05, 7.8619e-05, 2.2113e-05,
          1.2565e-04, 7.8619e-05, 7.8619e-05, 2.2113e-05, 3.5019e-03, 7.8619e-05,
          2.2113e-05, 7.8619e-05, 2.2113e-05, 7.8619e-05, 1.2195e-04, 2.2113e-05,
          2.2113e-05, 1.8215e-04, 1.2565e-04, 2.2113e-05, 7.8619e-05, 7.8619e-05,
          2.2113e-05, 2.2113e-05, 7.8619e-05, 7.8619e-05, 7.8619e-05, 1.2565e-04,
          1.8215e-04, 1.8215e-04, 2.2113e-05, 7.8619e-05, 1.8215e-04, 7.8619e-05,
          1.8215e-04, 2.2113e-05, 7.8619e-05, 2.2554e-04, 2.2113e-05, 2.2113e-05,
          1.2565e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 7.8619e-05, 2.2113e-05,
          7.8619e-05, 7.8619e-05, 1.8215e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          1.8215e-04, 2.2113e-05, 7.8619e-05, 7.8619e-05, 1.8215e-04, 1.2565e-04,
          2.2113e-05, 1.2565e-04, 2.2113e-05, 2.2113e-05, 2.2113e-05, 2.2113e-05,
          3.6979e-04, 9.2745e-04, 3.6979e-04, 2.1040e-04, 3.6979e-04, 7.8619e-05,
          2.2113e-05, 4.2629e-04, 3.6979e-04, 8.8835e-04, 1.2565e-04, 1.2565e-04,
          3.6979e-04, 3.6979e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.9008e-03, 5.6519e-02, 2.5970e-02, 2.6550e-02, 9.6497e-02,
          4.6783e-02, 2.6855e-02, 8.4019e-04, 7.2336e-04, 2.0523e-03, 4.8676e-03,
          1.3905e-03, 1.0471e-03, 1.2913e-03, 9.6512e-04, 5.4979e-04, 1.4448e-03,
          8.8453e-04, 3.5691e-04, 1.3390e-03, 1.1778e-03, 9.5320e-04, 1.1244e-03,
          4.4975e-03, 1.6470e-03, 1.8494e-02, 5.8098e-03, 5.6763e-03, 1.4181e-03,
          1.5926e-03, 1.0109e-03, 9.5463e-04, 1.5507e-03, 7.5340e-04, 6.5899e-04,
          2.4910e-03, 3.5038e-03, 1.4696e-03, 9.9087e-04, 1.1873e-03, 2.0084e-03,
          2.4490e-03, 1.5850e-03, 1.8415e-03, 9.4461e-04, 1.0662e-03, 1.5249e-03,
          6.6376e-04, 1.4496e-03, 1.2169e-02, 2.1210e-03, 1.0624e-03, 7.3195e-04,
          1.1292e-03, 7.3099e-04, 1.8663e-03, 5.0163e-03, 1.8654e-03, 3.4847e-03,
          2.9526e-03, 1.5078e-03, 1.7767e-03, 1.0700e-03, 1.7290e-03, 1.8835e-03,
          1.6279e-03, 7.3528e-04, 7.7772e-04, 1.1263e-03, 1.3533e-03, 1.4515e-03,
          1.5783e-03, 3.1452e-03, 3.6907e-03, 1.2531e-03, 8.2636e-04, 5.2452e-04,
          1.1539e-03, 8.5144e-03, 3.4180e-03, 1.9436e-03, 2.2373e-03, 1.5268e-03,
          2.6779e-03, 1.2550e-03, 1.2388e-03, 1.2980e-03, 8.1158e-04, 1.0996e-03,
          2.3308e-03, 1.4677e-03, 1.0157e-03, 1.6775e-03, 1.1234e-03, 1.6794e-03,
          4.1656e-03, 2.0313e-03, 7.3147e-04, 3.0289e-03, 1.3321e-02, 4.6120e-03,
          4.2381e-03, 1.1034e-03, 1.2703e-03, 2.3880e-03, 1.5078e-03, 1.7309e-03,
          1.0366e-03, 1.9550e-03, 2.2259e-03, 1.0366e-03, 1.3990e-03, 9.6178e-04,
          9.7322e-04, 1.4315e-03, 1.0328e-03, 1.8444e-03, 1.9855e-03, 1.0216e-02,
          5.9280e-03, 3.9711e-03, 6.6643e-03, 4.9162e-04, 1.8635e-03, 1.5831e-03,
          6.1569e-03, 2.2125e-03, 2.2087e-03, 1.6747e-03, 6.8045e-04, 1.1625e-03,
          1.6460e-03, 1.0786e-03, 4.7517e-04, 5.2023e-04, 1.2112e-03, 1.3838e-03,
          1.7242e-03, 1.2083e-03, 9.3889e-04, 2.4204e-03, 1.6508e-03, 2.8362e-03,
          6.7482e-03, 7.9584e-04, 2.0905e-03, 3.0365e-03, 2.9449e-03, 2.5978e-03,
          4.6120e-03, 1.6956e-03, 3.2616e-03, 1.5326e-03, 2.4338e-03, 2.3384e-03,
          1.2398e-03, 1.3781e-03, 5.1403e-04, 1.3094e-03, 1.8797e-03, 1.2283e-03,
          1.4601e-03, 1.2627e-03, 1.5526e-03, 1.6232e-03, 1.5854e-02, 3.0880e-03,
          3.6354e-03, 2.1420e-03, 2.1420e-03, 9.2125e-04, 1.4448e-03, 1.9522e-03,
          1.4706e-03, 2.3842e-03, 6.0883e-03, 4.6463e-03, 1.5879e-03, 1.2293e-03,
          9.3889e-04, 8.8406e-04, 1.3771e-03, 7.6103e-04, 7.9536e-04, 8.2922e-04,
          1.2789e-03, 1.4057e-03, 1.8263e-03, 1.7563e-02, 3.4657e-03, 1.5697e-03,
          9.3031e-04, 6.8016e-03, 1.2131e-03, 2.2049e-03, 1.6546e-03, 1.4944e-03,
          5.2109e-03, 3.8166e-03, 1.6918e-03, 1.3790e-03, 1.0614e-03, 9.3365e-04,
          1.8110e-03, 1.4267e-03, 2.1210e-03, 7.2289e-04, 2.5654e-03, 8.8215e-04,
          9.2697e-04, 1.5617e-02, 4.4312e-02, 3.1414e-03, 3.2978e-03, 4.3602e-03,
          2.3842e-03, 2.1362e-03, 1.4505e-03, 6.2943e-04, 1.9464e-03, 1.8177e-03,
          3.1071e-03, 5.7526e-03, 1.8311e-03, 1.3132e-03, 2.1687e-03, 1.3876e-03,
          1.3170e-03, 1.7681e-03, 1.9588e-03, 1.6174e-03, 1.4763e-03, 1.2856e-03,
          1.1481e-01, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 1.8561e-04, 3.8505e-05, 1.8561e-04, 3.8505e-05, 1.8561e-04,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 1.5998e-04, 1.5998e-04, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 1.5998e-04, 3.8505e-05, 3.8505e-05, 1.8561e-04,
          1.8561e-04, 1.8561e-04, 1.9765e-04, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          1.5998e-04, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.0446e-04, 5.4836e-04, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 1.8561e-04, 3.4475e-04, 3.8505e-05,
          3.8505e-05, 1.8561e-04, 1.5998e-04, 3.8505e-05, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 1.5998e-04,
          1.8561e-04, 1.8299e-04, 3.8505e-05, 1.5998e-04, 3.8505e-05, 3.8505e-05,
          1.8561e-04, 3.8505e-05, 3.4475e-04, 3.4475e-04, 3.8505e-05, 3.8505e-05,
          4.6349e-04, 1.9765e-04, 3.8505e-05, 3.8505e-05, 3.8505e-05, 1.5998e-04,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 1.8561e-04,
          3.8505e-05, 3.8505e-05, 3.4475e-04, 3.8505e-05, 3.4475e-04, 3.8505e-05,
          3.4475e-04, 3.0708e-04, 5.4979e-04, 3.4475e-04, 1.9765e-04, 3.4475e-04,
          1.9765e-04, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.0446e-04, 1.5998e-04,
          1.5998e-04, 1.8561e-04, 3.8505e-05, 1.8561e-04, 3.8505e-05, 3.8505e-05,
          5.4836e-04, 3.0446e-04, 3.4475e-04, 1.8561e-04, 3.8505e-05, 3.8505e-05,
          4.6611e-04, 1.9765e-04, 3.8505e-05, 3.4213e-04, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 3.8505e-05, 3.0446e-04, 3.8505e-05,
          4.2677e-04, 1.5998e-04, 4.5133e-04, 3.8505e-05, 3.8505e-05, 3.4475e-04,
          8.5449e-04, 3.4475e-04, 3.1900e-04, 1.8561e-04, 3.4475e-04, 4.8923e-04,
          3.4475e-04, 9.7466e-04, 3.4475e-04, 1.9765e-04, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 1.8299e-04, 1.8299e-04, 3.8505e-05, 1.8299e-04, 3.8505e-05,
          3.8505e-05, 3.8505e-05, 3.8505e-05, 1.9920e-04, 3.8505e-05, 3.8505e-05,
          3.8505e-05, 3.8505e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.5695e-03, 5.2826e-02, 1.5099e-02, 2.1423e-02, 4.0558e-02,
          3.3936e-02, 1.2756e-02, 9.7084e-04, 8.9979e-04, 3.6011e-03, 8.9111e-03,
          2.4529e-03, 9.9373e-04, 7.9584e-04, 1.0710e-03, 7.3576e-04, 1.3123e-03,
          7.6914e-04, 5.4312e-04, 2.8729e-04, 3.2687e-04, 3.0494e-04, 1.1845e-03,
          8.8425e-03, 2.0332e-03, 1.0010e-02, 4.8714e-03, 7.6532e-04, 1.4391e-03,
          1.4935e-03, 1.6632e-03, 5.7268e-04, 1.8959e-03, 1.4181e-03, 8.9788e-04,
          2.3193e-03, 1.4677e-03, 2.0752e-03, 1.3199e-03, 4.7851e-04, 2.6588e-03,
          1.4906e-03, 1.3676e-03, 1.1148e-03, 8.0299e-04, 1.2083e-03, 1.8673e-03,
          4.8685e-04, 4.2343e-04, 3.2558e-03, 1.9188e-03, 1.5154e-03, 1.1921e-03,
          6.7997e-04, 8.5878e-04, 6.6233e-04, 2.2621e-03, 2.3518e-03, 3.4790e-03,
          1.9140e-03, 9.4604e-04, 9.1982e-04, 1.0147e-03, 4.5128e-03, 3.3722e-03,
          7.9041e-03, 3.9978e-03, 2.0542e-03, 8.3399e-04, 7.4100e-04, 1.2331e-03,
          7.0477e-04, 7.1449e-03, 1.1959e-03, 1.8921e-03, 5.6028e-04, 6.1607e-04,
          1.0328e-03, 3.3264e-03, 8.5373e-03, 4.8180e-03, 8.0872e-03, 1.2999e-03,
          1.0080e-03, 2.0580e-03, 7.6914e-04, 4.0855e-03, 5.0507e-03, 4.3640e-03,
          8.8348e-03, 4.6501e-03, 9.6607e-04, 4.8218e-03, 5.6934e-04, 5.9509e-04,
          8.4782e-04, 6.3372e-04, 1.4505e-03, 3.4504e-03, 5.6305e-03, 5.5428e-03,
          1.2589e-02, 5.4626e-03, 5.6763e-03, 4.6120e-03, 2.6302e-03, 1.1253e-03,
          1.6193e-03, 2.2926e-03, 6.3820e-03, 1.9369e-03, 6.4507e-03, 5.7831e-03,
          1.6823e-03, 1.3666e-03, 4.2701e-04, 2.7046e-03, 8.2159e-04, 1.9932e-03,
          2.6913e-03, 8.7204e-03, 4.1847e-03, 7.1764e-04, 3.4847e-03, 4.7684e-03,
          2.6855e-03, 7.0038e-03, 9.0866e-03, 3.0956e-03, 1.4563e-03, 1.9045e-03,
          7.4196e-03, 5.6114e-03, 6.2037e-04, 2.0618e-03, 4.9820e-03, 1.1206e-03,
          2.4624e-03, 9.9945e-04, 1.1530e-03, 2.4815e-03, 2.6932e-03, 2.7866e-03,
          8.7738e-03, 1.9646e-03, 4.8676e-03, 5.6572e-03, 1.0216e-02, 5.8327e-03,
          1.2947e-02, 3.7289e-03, 3.4256e-03, 2.4910e-03, 1.5383e-03, 4.3907e-03,
          5.5771e-03, 4.6501e-03, 1.2140e-03, 3.8242e-03, 3.5629e-03, 1.1549e-03,
          2.3251e-03, 4.2992e-03, 1.9855e-03, 2.5158e-03, 7.5455e-03, 1.1879e-02,
          1.1597e-02, 9.7733e-03, 4.2381e-03, 1.6499e-03, 2.5520e-03, 4.3755e-03,
          2.4891e-03, 2.4605e-03, 2.9068e-03, 2.7504e-03, 4.2992e-03, 2.3212e-03,
          2.6665e-03, 1.3628e-03, 3.1528e-03, 1.8883e-03, 1.2074e-03, 9.8038e-04,
          3.2578e-03, 1.1168e-03, 1.9665e-03, 1.3214e-02, 1.2207e-02, 3.0766e-03,
          2.2984e-03, 3.5801e-03, 2.8534e-03, 3.3092e-03, 7.4615e-03, 3.1738e-03,
          3.1967e-03, 1.8673e-03, 3.3131e-03, 6.8283e-03, 6.0158e-03, 2.6779e-03,
          4.5891e-03, 5.6190e-03, 1.5259e-03, 1.0653e-03, 7.0953e-03, 1.1396e-03,
          1.3866e-03, 9.3155e-03, 1.0445e-02, 1.2550e-02, 6.5842e-03, 8.8882e-03,
          5.5809e-03, 3.5725e-03, 3.9101e-03, 1.0357e-03, 4.1962e-03, 2.5272e-03,
          5.9891e-03, 2.1172e-03, 5.0049e-03, 7.8964e-03, 4.7112e-03, 2.6646e-03,
          6.5231e-03, 2.5082e-03, 1.5621e-03, 2.1820e-03, 2.2507e-03, 1.0519e-03,
          1.1116e-02, 2.4567e-02, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 8.2493e-05, 1.5235e-04, 8.2493e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 8.7976e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.5235e-04, 1.5235e-04, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 4.5204e-04, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 8.2493e-05, 1.8120e-05,
          8.7976e-05, 1.8120e-05, 1.5235e-04, 1.5235e-04, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 8.7976e-05, 8.7976e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 8.2493e-05, 8.2493e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.5235e-04, 1.8120e-05, 1.8120e-05, 8.7976e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 7.2241e-04, 1.5235e-04, 1.8120e-05, 1.8120e-05,
          7.9250e-04, 8.2493e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05, 1.5235e-04,
          1.8120e-05, 1.5235e-04, 1.5235e-04, 8.7976e-05, 1.8120e-05, 1.8120e-05,
          8.2493e-05, 1.8120e-05, 7.9250e-04, 8.2493e-05, 1.8120e-05, 1.5235e-04,
          1.8120e-05, 1.8120e-05, 1.5235e-04, 1.8120e-05, 8.7976e-05, 1.1568e-03,
          7.2241e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 8.7976e-05, 8.7976e-05,
          8.7976e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05, 1.5235e-04, 8.2493e-05,
          1.5235e-04, 1.5235e-04, 1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05,
          8.7976e-05, 8.2493e-05, 1.8120e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 8.7976e-05, 1.8120e-05,
          1.5235e-04, 1.5235e-04, 1.8120e-05, 8.2493e-05, 1.8120e-05, 7.2241e-04,
          1.5235e-04, 5.1641e-04, 1.5235e-04, 1.8120e-05, 1.8120e-05, 8.2493e-05,
          8.2493e-05, 8.2493e-05, 1.8120e-05, 8.2493e-05, 1.8120e-05, 1.8120e-05,
          1.8120e-05, 1.8120e-05, 1.8120e-05, 1.8120e-05, 8.2493e-05, 8.2493e-05,
          1.8120e-05, 8.2493e-05, 8.2493e-05, 8.2493e-05, 1.8120e-05, 8.2493e-05,
          1.8120e-05, 8.2493e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4245e-03, 4.8401e-02, 1.7395e-02, 1.6632e-02, 4.8615e-02,
          3.5645e-02, 1.2802e-02, 1.1969e-03, 9.8419e-04, 1.8444e-03, 6.0883e-03,
          1.6623e-03, 8.6260e-04, 1.2445e-03, 9.7322e-04, 8.4877e-04, 8.7643e-04,
          6.7854e-04, 6.6805e-04, 3.3116e-04, 2.7800e-04, 2.9802e-04, 8.7309e-04,
          6.2981e-03, 1.6317e-03, 9.6664e-03, 2.6855e-03, 8.4782e-04, 1.0014e-03,
          2.0885e-03, 1.8930e-03, 4.4322e-04, 1.0138e-03, 7.7677e-04, 6.5994e-04,
          1.3161e-03, 1.3485e-03, 7.5579e-04, 6.7425e-04, 4.9877e-04, 9.0933e-04,
          1.1415e-03, 8.4829e-04, 1.2493e-03, 4.0126e-04, 9.7609e-04, 8.9693e-04,
          4.9543e-04, 3.9911e-04, 3.7212e-03, 1.4153e-03, 1.6689e-03, 9.2125e-04,
          1.2026e-03, 1.0805e-03, 1.0462e-03, 1.3084e-03, 3.5839e-03, 2.9011e-03,
          2.8229e-03, 9.1410e-04, 1.4124e-03, 7.5197e-04, 2.3136e-03, 1.8539e-03,
          1.8661e-02, 2.9621e-03, 7.6008e-04, 5.5456e-04, 5.6124e-04, 1.8167e-03,
          7.3433e-04, 3.0689e-03, 1.5898e-03, 2.2831e-03, 1.2541e-03, 7.8011e-04,
          1.2693e-03, 1.8873e-03, 8.2092e-03, 2.1572e-03, 8.1635e-03, 1.3828e-03,
          1.2112e-03, 2.2659e-03, 1.1816e-03, 2.7103e-03, 6.8855e-03, 1.0063e-02,
          5.1804e-03, 1.7147e-03, 1.2236e-03, 2.5330e-03, 5.0735e-04, 5.8365e-04,
          9.8610e-04, 8.1539e-04, 9.2411e-04, 1.9569e-03, 4.1962e-03, 3.1414e-03,
          6.9962e-03, 8.4991e-03, 5.0926e-03, 1.9226e-03, 1.7595e-03, 9.9754e-04,
          1.9217e-03, 1.5106e-03, 2.9125e-03, 4.7264e-03, 2.0645e-02, 4.2458e-03,
          1.5259e-03, 7.7057e-04, 4.6420e-04, 1.9522e-03, 7.5626e-04, 2.5463e-03,
          3.4294e-03, 3.7327e-03, 4.9820e-03, 9.3126e-04, 2.4643e-03, 4.9782e-03,
          1.5936e-03, 8.2321e-03, 1.5068e-02, 1.8835e-03, 1.1244e-03, 2.2697e-03,
          2.3056e-02, 7.0114e-03, 5.7220e-04, 2.1191e-03, 2.9984e-03, 9.9087e-04,
          1.2589e-03, 1.0843e-03, 1.1139e-03, 1.9054e-03, 1.4486e-03, 3.0518e-03,
          1.0406e-02, 1.6861e-03, 2.8114e-03, 2.9411e-03, 1.8677e-02, 3.1815e-03,
          4.2686e-03, 1.8978e-03, 2.5005e-03, 1.2703e-03, 2.0695e-03, 1.5497e-03,
          2.4185e-03, 1.7052e-03, 1.2980e-03, 4.3182e-03, 1.7509e-03, 8.9502e-04,
          1.3638e-03, 1.0757e-03, 1.4172e-03, 1.3475e-03, 4.1656e-03, 4.6768e-03,
          7.1373e-03, 3.0441e-03, 2.7695e-03, 1.4257e-03, 1.8225e-03, 1.6975e-03,
          2.3594e-03, 4.0016e-03, 2.2030e-03, 1.7138e-03, 1.6699e-03, 1.9665e-03,
          1.3514e-03, 1.2703e-03, 1.8816e-03, 1.8482e-03, 1.1225e-03, 8.0872e-04,
          2.9030e-03, 1.4105e-03, 1.8387e-03, 1.0536e-02, 2.7420e-02, 2.2831e-03,
          2.1877e-03, 2.7828e-03, 1.4591e-03, 2.3651e-03, 3.2063e-03, 1.6451e-03,
          3.0231e-03, 2.0485e-03, 2.2049e-03, 4.4518e-03, 3.1147e-03, 1.2007e-03,
          6.6605e-03, 3.2101e-03, 1.1940e-03, 1.1177e-03, 4.6768e-03, 7.2813e-04,
          1.6022e-03, 4.8485e-03, 1.3077e-02, 1.6998e-02, 4.5242e-03, 1.3512e-02,
          2.5978e-03, 2.4147e-03, 2.7275e-03, 1.0452e-03, 2.0046e-03, 3.1719e-03,
          2.9278e-03, 2.6989e-03, 2.5768e-03, 6.1226e-03, 3.5610e-03, 2.1992e-03,
          8.9417e-03, 1.4992e-02, 1.7338e-03, 3.4542e-03, 1.5297e-03, 1.6241e-03,
          1.0010e-02, 9.9335e-03, 8.1726e-02, 2.9325e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 1.3947e-04,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 9.2328e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 1.3947e-04, 2.9325e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.0242e-04, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 2.9325e-05,
          9.2328e-05, 2.9325e-05, 2.0242e-04, 2.0242e-04, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 9.2328e-05, 2.0242e-04, 9.2328e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05,
          2.0242e-04, 2.9325e-05, 2.9325e-05, 9.2328e-05, 2.9325e-05, 1.3947e-04,
          2.9325e-05, 2.9325e-05, 5.0116e-04, 9.2328e-05, 2.9325e-05, 2.9325e-05,
          2.0242e-04, 4.5395e-04, 2.9325e-05, 2.9325e-05, 2.9325e-05, 2.0242e-04,
          2.9325e-05, 5.6410e-04, 2.0242e-04, 9.2328e-05, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 9.2328e-05, 1.3947e-04, 2.9325e-05, 2.0242e-04,
          2.9325e-05, 2.9325e-05, 6.5184e-04, 2.9325e-05, 3.9077e-04, 1.3947e-04,
          2.9325e-05, 1.3947e-04, 2.9325e-05, 2.9325e-05, 9.2328e-05, 9.2328e-05,
          9.2328e-05, 1.3947e-04, 2.9325e-05, 2.9325e-05, 1.3947e-04, 2.9325e-05,
          2.0242e-04, 2.0242e-04, 3.9077e-04, 2.9325e-05, 1.3947e-04, 2.9325e-05,
          2.9325e-05, 1.3947e-04, 2.9325e-05, 1.3947e-04, 2.9325e-05, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 2.9325e-05, 2.9325e-05, 9.2328e-05, 2.9325e-05,
          5.6410e-04, 2.0242e-04, 2.9325e-05, 2.9325e-05, 2.9325e-05, 1.3947e-04,
          5.6410e-04, 1.3947e-04, 2.0242e-04, 2.9325e-05, 2.9325e-05, 1.3947e-04,
          5.0116e-04, 5.6410e-04, 2.9325e-05, 3.9077e-04, 3.9077e-04, 2.9325e-05,
          1.3947e-04, 2.9325e-05, 1.3947e-04, 1.3947e-04, 1.3947e-04, 2.9325e-05,
          2.9325e-05, 2.9325e-05, 1.3947e-04, 4.6015e-04, 1.3947e-04, 1.3947e-04,
          2.9325e-05, 1.3947e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.6692e-03, 5.7678e-02, 1.4847e-02, 1.0918e-02, 3.9246e-02,
          4.8828e-02, 8.2092e-03, 1.6651e-03, 1.4601e-03, 3.4199e-03, 5.6725e-03,
          2.0599e-03, 1.4420e-03, 5.0879e-04, 5.0926e-03, 1.4429e-03, 1.1368e-03,
          6.4898e-04, 1.0290e-03, 2.2972e-04, 2.7037e-04, 2.3794e-04, 1.2922e-03,
          6.2218e-03, 6.2637e-03, 6.6605e-03, 2.9716e-03, 5.1880e-04, 1.4143e-03,
          6.6090e-04, 3.2330e-03, 4.7135e-04, 4.7607e-03, 2.7657e-03, 1.8721e-03,
          3.1319e-03, 8.6403e-04, 2.5845e-03, 3.6526e-03, 2.8706e-04, 1.5936e-03,
          6.4373e-04, 5.6314e-04, 5.1212e-04, 4.8923e-04, 1.3275e-03, 3.6964e-03,
          2.9182e-04, 3.2759e-04, 2.7714e-03, 2.3556e-03, 2.1935e-03, 2.3098e-03,
          7.3051e-04, 1.3819e-03, 8.6260e-04, 1.1482e-03, 2.6646e-03, 1.6594e-03,
          1.6518e-03, 7.0238e-04, 1.0118e-03, 1.5802e-03, 1.1702e-03, 7.7858e-03,
          1.3866e-03, 8.0633e-04, 1.0881e-03, 4.6110e-04, 3.7813e-04, 5.1641e-04,
          4.7541e-04, 6.8398e-03, 9.8610e-04, 3.0251e-03, 6.0654e-04, 6.2418e-04,
          1.4315e-03, 1.3933e-03, 1.3485e-03, 4.1313e-03, 1.3218e-03, 1.8244e-03,
          1.2035e-03, 3.2349e-03, 8.2159e-04, 2.8553e-03, 8.0919e-04, 7.9250e-04,
          4.3907e-03, 3.0289e-03, 6.2799e-04, 2.4490e-03, 7.8630e-04, 5.9032e-04,
          7.0858e-04, 5.5313e-04, 2.4052e-03, 3.6297e-03, 1.5440e-03, 2.1877e-03,
          7.2441e-03, 1.7242e-03, 1.1377e-03, 4.6806e-03, 9.4833e-03, 8.5545e-04,
          2.9526e-03, 7.8392e-04, 3.7327e-03, 9.6321e-04, 1.4372e-03, 1.1702e-03,
          8.2855e-03, 1.4172e-03, 4.1151e-04, 7.9575e-03, 8.0776e-04, 1.5669e-03,
          1.5392e-03, 7.7209e-03, 1.2760e-03, 1.6203e-03, 1.0376e-02, 9.0885e-04,
          1.1854e-03, 9.2649e-04, 1.4372e-03, 9.9030e-03, 3.7365e-03, 3.5896e-03,
          2.4052e-03, 9.0504e-04, 2.6150e-03, 6.1178e-04, 2.2964e-03, 5.6744e-04,
          3.2310e-03, 6.2656e-04, 3.3665e-03, 4.5319e-03, 4.5967e-03, 1.7099e-03,
          2.1515e-03, 8.8196e-03, 2.5330e-03, 2.1038e-03, 1.5526e-03, 1.6373e-02,
          9.2697e-03, 8.4000e-03, 2.8095e-03, 2.0332e-03, 1.8148e-03, 1.8358e-03,
          3.2177e-03, 1.9684e-03, 1.9798e-03, 7.0095e-04, 5.5199e-03, 1.1854e-03,
          5.1422e-03, 4.5929e-03, 2.2469e-03, 8.2169e-03, 3.1586e-03, 2.9354e-03,
          7.0457e-03, 3.0613e-03, 9.1858e-03, 1.1436e-02, 1.8902e-03, 5.5695e-03,
          1.0727e-02, 4.3373e-03, 3.0155e-03, 1.1463e-03, 5.2452e-03, 4.2686e-03,
          2.7714e-03, 5.4665e-03, 6.2714e-03, 4.4403e-03, 3.0270e-03, 5.5885e-03,
          4.7684e-03, 1.2569e-03, 5.2223e-03, 3.0842e-03, 2.0695e-03, 9.2850e-03,
          4.9782e-03, 1.1988e-03, 6.8054e-03, 3.4695e-03, 5.8174e-03, 5.2986e-03,
          3.1834e-03, 1.2980e-03, 8.0261e-03, 1.2541e-03, 2.2774e-03, 3.1300e-03,
          1.0080e-03, 2.0885e-03, 1.0023e-03, 3.0651e-03, 5.2261e-03, 1.2951e-03,
          2.9984e-03, 4.0512e-03, 2.1992e-03, 3.4752e-03, 1.5335e-02, 2.1286e-03,
          6.7787e-03, 3.1071e-03, 7.4844e-03, 8.0643e-03, 9.1019e-03, 2.6741e-03,
          7.5264e-03, 1.0700e-03, 2.3804e-03, 1.7786e-03, 1.4816e-02, 7.1373e-03,
          1.1787e-03, 1.6365e-03, 1.1578e-03, 8.3084e-03, 6.7825e-03, 1.4963e-03,
          2.9736e-03, 3.5763e-03, 3.3684e-03, 7.0801e-02, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 8.8751e-05, 3.6240e-04,
          2.1458e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 1.1654e-03,
          2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 8.8751e-05, 8.8751e-05, 3.2091e-04, 2.1458e-05,
          8.8751e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 1.0128e-03, 8.8751e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05,
          2.1458e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05, 2.1458e-05,
          8.8751e-05, 8.8751e-05, 3.6240e-04, 2.1458e-05, 8.8751e-05, 2.1458e-05,
          2.1458e-05, 8.8751e-05, 1.0128e-03, 8.8751e-05, 2.1458e-05, 2.1458e-05,
          7.3910e-04, 8.8751e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05, 8.8751e-05,
          2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05, 8.8751e-05,
          8.8751e-05, 2.1458e-05, 6.7186e-04, 3.6240e-04, 8.8751e-05, 8.8751e-05,
          2.1458e-05, 2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05, 1.0128e-03,
          1.0128e-03, 1.3971e-03, 2.1458e-05, 2.1458e-05, 8.8751e-05, 3.2091e-04,
          2.1458e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05, 3.2091e-04, 8.8751e-05,
          3.2091e-04, 3.2091e-04, 2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05,
          8.8751e-05, 3.6240e-04, 2.1458e-05, 8.8751e-05, 2.1458e-05, 2.1458e-05,
          1.0595e-03, 2.1458e-05, 2.1458e-05, 2.1458e-05, 3.2091e-04, 8.8751e-05,
          8.8751e-05, 3.2091e-04, 8.8751e-05, 8.8751e-05, 8.8751e-05, 1.0128e-03,
          8.8751e-05, 3.6240e-04, 5.9462e-04, 2.1458e-05, 8.8751e-05, 8.8751e-05,
          1.4391e-03, 8.8751e-05, 2.1458e-05, 8.8751e-05, 8.8751e-05, 2.1458e-05,
          8.8751e-05, 8.8751e-05, 5.8270e-04, 8.8751e-05, 8.8751e-05, 2.1458e-05,
          2.1458e-05, 8.8751e-05, 8.8751e-05, 8.8751e-05, 8.8751e-05, 5.8270e-04,
          8.8751e-05, 8.8751e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1765e-03, 4.9957e-02, 1.4076e-02, 1.3031e-02, 4.8065e-02,
          3.6804e-02, 9.8419e-03, 1.0605e-03, 1.1339e-03, 3.0842e-03, 4.5547e-03,
          1.7529e-03, 1.4362e-03, 6.1607e-04, 3.5553e-03, 7.4387e-04, 1.5125e-03,
          7.9250e-04, 5.2452e-04, 3.6359e-04, 3.8791e-04, 3.6764e-04, 1.2388e-03,
          4.3106e-03, 6.2408e-03, 7.4310e-03, 3.1891e-03, 8.2445e-04, 1.8482e-03,
          8.5974e-04, 1.8377e-03, 6.3181e-04, 4.3106e-03, 1.9016e-03, 1.2217e-03,
          4.4212e-03, 1.3847e-03, 2.1992e-03, 2.3327e-03, 4.8113e-04, 2.4605e-03,
          1.2627e-03, 1.2531e-03, 9.4843e-04, 1.0452e-03, 1.6346e-03, 4.6692e-03,
          3.7360e-04, 4.3964e-04, 4.1466e-03, 2.5749e-03, 1.8406e-03, 1.2159e-03,
          8.4591e-04, 1.1654e-03, 1.1196e-03, 2.6875e-03, 1.7595e-03, 2.9392e-03,
          2.3785e-03, 1.1072e-03, 9.9087e-04, 1.3628e-03, 1.6890e-03, 8.8196e-03,
          1.3227e-03, 6.8569e-04, 1.2989e-03, 1.0109e-03, 8.1110e-04, 8.5545e-04,
          7.4100e-04, 4.7989e-03, 1.3065e-03, 2.4109e-03, 6.0081e-04, 5.6505e-04,
          1.3838e-03, 4.1580e-03, 2.5806e-03, 4.3869e-03, 2.0084e-03, 1.7490e-03,
          1.3227e-03, 2.0676e-03, 8.0013e-04, 1.6661e-03, 6.4898e-04, 9.3365e-04,
          2.7504e-03, 2.2278e-03, 9.4509e-04, 1.8578e-03, 1.9064e-03, 1.0023e-03,
          1.0710e-03, 7.7486e-04, 1.2445e-03, 4.2152e-03, 7.5150e-03, 5.9090e-03,
          6.0806e-03, 1.1959e-03, 1.4706e-03, 5.0926e-03, 6.3477e-03, 1.5364e-03,
          1.7672e-03, 1.2007e-03, 3.0174e-03, 9.4795e-04, 1.2550e-03, 9.7609e-04,
          7.1869e-03, 4.3106e-03, 6.6185e-04, 8.7357e-03, 1.3762e-03, 2.7924e-03,
          3.6163e-03, 1.3054e-02, 3.8815e-03, 7.0143e-04, 7.0763e-03, 1.8835e-03,
          4.1924e-03, 1.7986e-03, 2.5692e-03, 7.1220e-03, 1.5526e-03, 2.0828e-03,
          1.7481e-03, 8.6308e-04, 9.0647e-04, 4.7112e-04, 1.6775e-03, 1.0691e-03,
          3.7498e-03, 1.1129e-03, 1.1654e-03, 1.8110e-03, 2.4376e-03, 2.9850e-03,
          6.7329e-03, 1.4105e-03, 6.0997e-03, 4.4365e-03, 2.8419e-03, 1.9791e-02,
          8.9035e-03, 7.7820e-03, 5.8479e-03, 1.9226e-03, 1.6623e-03, 3.8528e-03,
          1.9588e-03, 1.9255e-03, 6.8188e-04, 1.1129e-03, 9.8572e-03, 1.7567e-03,
          4.4365e-03, 2.6474e-03, 2.4509e-03, 2.9793e-03, 1.1101e-02, 3.6907e-03,
          5.3520e-03, 3.2864e-03, 1.0246e-02, 4.2648e-03, 2.2907e-03, 5.9662e-03,
          6.2370e-03, 3.8567e-03, 4.6387e-03, 2.2125e-03, 4.7379e-03, 2.0428e-03,
          2.3289e-03, 2.5730e-03, 4.4975e-03, 1.3828e-03, 1.0128e-03, 1.5717e-03,
          1.8950e-03, 1.2064e-03, 1.4124e-03, 1.2978e-02, 4.1199e-03, 4.7874e-03,
          1.9426e-03, 5.3673e-03, 4.6616e-03, 3.4695e-03, 2.9011e-03, 4.8256e-03,
          4.8447e-03, 3.4351e-03, 5.6534e-03, 1.4849e-03, 1.1845e-03, 2.1839e-03,
          1.5182e-03, 2.2602e-03, 2.4071e-03, 8.5831e-04, 3.3188e-03, 1.0500e-03,
          8.7404e-04, 1.1780e-02, 1.1017e-02, 3.8300e-03, 1.5915e-02, 6.4201e-03,
          7.9880e-03, 4.7836e-03, 2.4185e-03, 1.8044e-03, 6.0883e-03, 3.2845e-03,
          6.6528e-03, 2.2411e-03, 3.3703e-03, 1.2026e-03, 8.1635e-03, 3.2806e-03,
          1.1940e-03, 1.3943e-03, 2.4929e-03, 6.9656e-03, 6.2294e-03, 1.6823e-03,
          1.0078e-02, 4.9553e-03, 5.1956e-03, 6.0158e-03, 7.4280e-02, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.0991e-04, 6.2323e-04,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.2088e-04,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 4.0483e-04, 2.0063e-04, 3.0279e-05,
          1.0991e-04, 1.0991e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 4.9543e-04, 4.6134e-04, 3.0279e-05, 3.0279e-05,
          1.0991e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.0991e-04, 3.0279e-05, 2.0063e-04, 5.5456e-04, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 2.3460e-04, 1.0991e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.0991e-04,
          4.0483e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 2.0063e-04, 5.4359e-04, 4.0483e-04, 3.0279e-05, 3.0279e-05,
          4.0483e-04, 6.2323e-04, 1.2088e-04, 3.0279e-05, 3.0279e-05, 2.0063e-04,
          3.0279e-05, 1.0991e-04, 2.0063e-04, 1.0991e-04, 3.0279e-05, 3.0279e-05,
          2.0063e-04, 3.0279e-05, 3.1424e-04, 3.0279e-05, 3.0279e-05, 4.6134e-04,
          3.0279e-05, 3.0279e-05, 2.4283e-04, 1.0991e-04, 3.0279e-05, 5.4359e-04,
          5.4359e-04, 2.0063e-04, 3.0279e-05, 3.0279e-05, 2.0063e-04, 1.0991e-04,
          3.0279e-05, 3.2520e-04, 1.0991e-04, 3.0279e-05, 2.0063e-04, 3.0279e-05,
          4.0483e-04, 2.0063e-04, 3.0279e-05, 1.0991e-04, 3.0279e-05, 3.0279e-05,
          1.0991e-04, 6.6566e-04, 3.0279e-05, 1.2088e-04, 3.0279e-05, 3.0279e-05,
          1.2088e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.0991e-04, 3.0279e-05,
          1.2088e-04, 2.0063e-04, 2.0063e-04, 3.0279e-05, 3.0279e-05, 5.4359e-04,
          1.2088e-04, 4.5300e-04, 8.5402e-04, 3.0279e-05, 3.0279e-05, 2.3460e-04,
          5.4359e-04, 3.2520e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 1.2088e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 2.0063e-04, 1.2088e-04, 3.9816e-04, 3.0279e-05, 4.7493e-04,
          3.0279e-05, 3.0279e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.7798e-03, 5.2460e-02, 1.3466e-02, 1.3718e-02, 4.9469e-02,
          4.4464e-02, 8.7128e-03, 1.4582e-03, 1.4057e-03, 3.0746e-03, 6.0501e-03,
          2.4185e-03, 1.3351e-03, 7.8058e-04, 2.1248e-03, 1.0653e-03, 1.4420e-03,
          8.6546e-04, 6.3467e-04, 3.7146e-04, 3.9315e-04, 3.7575e-04, 1.0281e-03,
          6.4278e-03, 4.3831e-03, 7.1335e-03, 3.7327e-03, 7.7772e-04, 1.6842e-03,
          1.0023e-03, 2.3899e-03, 6.7234e-04, 2.5253e-03, 1.5459e-03, 1.6880e-03,
          3.5763e-03, 1.7090e-03, 1.5306e-03, 1.6232e-03, 4.9496e-04, 1.8129e-03,
          1.3056e-03, 1.0452e-03, 1.1597e-03, 8.2016e-04, 1.2503e-03, 2.2392e-03,
          4.3249e-04, 4.3702e-04, 4.1237e-03, 2.8019e-03, 1.9341e-03, 1.4257e-03,
          1.2026e-03, 1.3638e-03, 1.3075e-03, 2.6207e-03, 3.6411e-03, 3.9635e-03,
          3.7537e-03, 1.1444e-03, 1.3151e-03, 1.4744e-03, 2.6474e-03, 3.5419e-03,
          2.0294e-03, 1.4172e-03, 9.0265e-04, 6.8760e-04, 7.1859e-04, 1.3933e-03,
          8.6355e-04, 4.3449e-03, 1.4133e-03, 2.5253e-03, 7.7629e-04, 7.0953e-04,
          1.4248e-03, 3.9215e-03, 2.8706e-03, 3.4771e-03, 2.1591e-03, 2.5692e-03,
          1.8425e-03, 2.7943e-03, 1.0729e-03, 2.3117e-03, 9.6035e-04, 1.1654e-03,
          4.5929e-03, 2.2316e-03, 9.8228e-04, 2.6627e-03, 1.3018e-03, 9.3031e-04,
          9.3842e-04, 7.5197e-04, 1.5774e-03, 3.2959e-03, 5.3596e-03, 4.0779e-03,
          7.0686e-03, 1.3123e-03, 1.5125e-03, 3.5229e-03, 3.4466e-03, 1.6184e-03,
          2.5120e-03, 1.4248e-03, 3.1452e-03, 1.1911e-03, 1.7033e-03, 1.0424e-03,
          3.2043e-03, 1.5678e-03, 5.9319e-04, 3.5000e-03, 1.1702e-03, 2.7370e-03,
          3.2444e-03, 7.4501e-03, 3.1986e-03, 9.2840e-04, 3.4084e-03, 1.6718e-03,
          2.9678e-03, 1.7910e-03, 2.7008e-03, 4.1847e-03, 1.8291e-03, 2.7466e-03,
          2.5978e-03, 1.0033e-03, 7.1287e-04, 5.5456e-04, 2.0905e-03, 8.5211e-04,
          2.1477e-03, 1.0529e-03, 1.3237e-03, 1.9464e-03, 1.9932e-03, 2.6379e-03,
          5.1155e-03, 1.6689e-03, 3.1319e-03, 2.9202e-03, 3.1033e-03, 8.5449e-03,
          6.1340e-03, 4.6539e-03, 4.2343e-03, 2.3193e-03, 2.0447e-03, 2.3556e-03,
          1.9512e-03, 1.5268e-03, 1.0929e-03, 1.1959e-03, 3.9482e-03, 1.3857e-03,
          2.4948e-03, 1.6556e-03, 1.6613e-03, 1.9178e-03, 5.5618e-03, 2.7771e-03,
          5.8861e-03, 2.2507e-03, 5.3596e-03, 2.2602e-03, 2.0142e-03, 3.9825e-03,
          7.4348e-03, 6.0463e-03, 4.0207e-03, 2.2507e-03, 3.1605e-03, 1.7385e-03,
          1.4286e-03, 2.0008e-03, 2.4757e-03, 1.9474e-03, 9.2077e-04, 1.1349e-03,
          2.7409e-03, 1.4133e-03, 1.3533e-03, 7.4501e-03, 4.6539e-03, 3.0289e-03,
          1.9894e-03, 4.0588e-03, 2.7695e-03, 2.6932e-03, 3.0441e-03, 2.9373e-03,
          5.2834e-03, 3.3569e-03, 4.1847e-03, 1.8263e-03, 1.3828e-03, 1.6108e-03,
          1.8082e-03, 2.1973e-03, 1.7300e-03, 1.0061e-03, 4.4708e-03, 8.9264e-04,
          1.2217e-03, 6.6948e-03, 7.7515e-03, 4.3564e-03, 7.5836e-03, 6.0043e-03,
          5.0430e-03, 3.8433e-03, 2.4624e-03, 1.5202e-03, 4.1580e-03, 3.5763e-03,
          5.5428e-03, 2.9564e-03, 2.9163e-03, 1.3943e-03, 5.0850e-03, 2.9545e-03,
          1.5936e-03, 2.6932e-03, 2.6302e-03, 8.5678e-03, 4.1008e-03, 1.7862e-03,
          8.6594e-03, 4.1847e-03, 7.3547e-03, 6.0158e-03, 2.5635e-02, 1.5552e-01,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 1.1277e-04, 2.9504e-05,
          2.9504e-05, 1.1575e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          1.1575e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 1.1575e-04,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 1.1277e-04, 1.9908e-04, 2.9504e-05,
          1.1575e-04, 1.1575e-04, 1.9908e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 4.6062e-04, 1.1575e-04, 2.9504e-05, 1.1575e-04,
          1.1575e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 1.9908e-04, 3.1614e-04, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 1.1277e-04, 3.5572e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05,
          1.1277e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 1.1575e-04,
          2.9504e-05, 3.1614e-04, 2.9504e-05, 1.1277e-04, 2.9504e-05, 2.9504e-05,
          1.1277e-04, 2.9504e-05, 1.1575e-04, 2.9504e-05, 2.9504e-05, 1.9908e-04,
          2.9504e-05, 1.1277e-04, 1.1575e-04, 1.1277e-04, 2.9504e-05, 2.9504e-05,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 2.9504e-05, 2.7227e-04, 1.9908e-04,
          2.9504e-05, 2.9504e-05, 1.9908e-04, 1.8609e-04, 2.9504e-05, 2.9504e-05,
          2.1324e-03, 1.9665e-03, 2.9504e-05, 2.9504e-05, 3.9959e-04, 1.9908e-04,
          2.9504e-05, 2.9504e-05, 2.9504e-05, 1.8609e-04, 1.1575e-04, 2.9504e-05,
          5.5599e-04, 5.5599e-04, 2.9504e-05, 2.9504e-05, 2.7227e-04, 2.9504e-05,
          2.6941e-04, 5.4693e-04, 2.9504e-05, 1.1575e-04, 2.9504e-05, 1.8609e-04,
          1.1575e-04, 2.9504e-05, 2.9504e-05, 2.9504e-05, 1.9908e-04, 2.9504e-05,
          1.1575e-04, 1.9908e-04, 4.7278e-04, 2.9504e-05, 2.9504e-05, 2.7227e-04,
          4.7278e-04, 2.7227e-04, 1.9908e-04, 1.8609e-04, 2.7227e-04, 2.7227e-04,
          7.0333e-04, 4.7278e-04, 1.8609e-04, 1.8609e-04, 2.7227e-04, 2.9504e-05,
          4.7278e-04, 3.1614e-04, 3.1614e-04, 1.1575e-04, 1.1575e-04, 2.9504e-05,
          2.9504e-05, 3.9959e-04, 3.1614e-04, 7.1573e-04, 3.1614e-04, 4.7278e-04,
          3.1614e-04, 3.1614e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.4169e-03, 4.8798e-02, 1.7303e-02, 1.4565e-02, 6.4697e-02,
          3.7537e-02, 1.3214e-02, 9.3842e-04, 1.0710e-03, 2.2774e-03, 5.9090e-03,
          1.9407e-03, 8.4782e-04, 9.8610e-04, 1.2016e-03, 6.1417e-04, 8.8644e-04,
          5.6124e-04, 4.1652e-04, 5.5552e-04, 4.5705e-04, 4.7350e-04, 6.4802e-04,
          5.2528e-03, 2.1152e-03, 9.7504e-03, 1.9875e-03, 1.2350e-03, 9.0313e-04,
          1.0109e-03, 1.3275e-03, 5.1308e-04, 1.3103e-03, 7.4816e-04, 8.0013e-04,
          2.4147e-03, 2.7599e-03, 7.5436e-04, 7.6580e-04, 6.6519e-04, 1.1101e-03,
          1.9255e-03, 1.1368e-03, 1.9588e-03, 8.1062e-04, 7.1001e-04, 1.0357e-03,
          4.3249e-04, 5.3215e-04, 5.6343e-03, 1.9121e-03, 1.3514e-03, 8.3828e-04,
          1.4133e-03, 1.0939e-03, 2.3975e-03, 3.3474e-03, 3.3817e-03, 5.0278e-03,
          5.0163e-03, 1.1358e-03, 1.6146e-03, 8.4114e-04, 2.0771e-03, 1.6680e-03,
          2.0905e-03, 1.0414e-03, 5.5647e-04, 7.7868e-04, 9.1267e-04, 2.3136e-03,
          1.4973e-03, 2.6016e-03, 2.1687e-03, 1.7958e-03, 1.0300e-03, 7.7200e-04,
          1.8024e-03, 4.7646e-03, 3.6221e-03, 2.1935e-03, 2.9049e-03, 2.5196e-03,
          2.7180e-03, 1.7157e-03, 1.3523e-03, 1.6603e-03, 8.8167e-04, 1.3762e-03,
          2.8954e-03, 1.3018e-03, 1.0681e-03, 1.8148e-03, 1.1511e-03, 1.2760e-03,
          1.0910e-03, 1.2732e-03, 7.4625e-04, 1.8845e-03, 7.9117e-03, 3.7708e-03,
          4.6425e-03, 1.2407e-03, 1.3342e-03, 2.8381e-03, 1.9197e-03, 1.4057e-03,
          1.6117e-03, 1.4410e-03, 1.8892e-03, 1.2999e-03, 1.9016e-03, 8.5449e-04,
          1.2894e-03, 1.0939e-03, 6.0034e-04, 1.7519e-03, 1.5755e-03, 3.0632e-03,
          2.8801e-03, 3.9940e-03, 5.1422e-03, 6.8808e-04, 1.9989e-03, 2.3117e-03,
          4.7760e-03, 3.3188e-03, 3.9787e-03, 2.2678e-03, 8.1730e-04, 1.7557e-03,
          2.3117e-03, 9.1410e-04, 3.6049e-04, 4.8590e-04, 1.3094e-03, 1.1063e-03,
          9.7275e-04, 9.4318e-04, 7.1001e-04, 1.5659e-03, 1.2045e-03, 2.2011e-03,
          7.0419e-03, 1.1091e-03, 2.2926e-03, 3.2501e-03, 4.1504e-03, 4.1504e-03,
          4.2152e-03, 2.8305e-03, 5.4283e-03, 1.4200e-03, 2.5692e-03, 1.7614e-03,
          1.1883e-03, 1.1177e-03, 6.0177e-04, 1.6794e-03, 1.8177e-03, 8.3876e-04,
          1.3237e-03, 8.2588e-04, 8.7357e-04, 1.2598e-03, 7.0457e-03, 2.8057e-03,
          3.9825e-03, 1.9531e-03, 2.4281e-03, 1.2865e-03, 1.3809e-03, 2.2697e-03,
          4.7951e-03, 8.4381e-03, 8.1253e-03, 2.5673e-03, 1.5831e-03, 9.3317e-04,
          9.0837e-04, 1.0099e-03, 1.2522e-03, 1.0386e-03, 6.2943e-04, 6.3992e-04,
          1.7910e-03, 9.8038e-04, 1.3704e-03, 1.4923e-02, 5.6686e-03, 1.7786e-03,
          1.2369e-03, 9.9411e-03, 1.7185e-03, 1.5945e-03, 1.8444e-03, 1.8406e-03,
          1.0735e-02, 4.2305e-03, 2.1648e-03, 1.4439e-03, 1.0386e-03, 8.7738e-04,
          2.4261e-03, 1.5316e-03, 2.0008e-03, 6.5231e-04, 2.6989e-03, 5.6934e-04,
          9.4175e-04, 6.9962e-03, 2.5574e-02, 3.3855e-03, 4.0741e-03, 9.2621e-03,
          3.3550e-03, 2.5120e-03, 1.3285e-03, 9.1982e-04, 2.3842e-03, 2.7027e-03,
          2.9488e-03, 5.6953e-03, 2.3060e-03, 1.1692e-03, 2.4433e-03, 1.5631e-03,
          1.3666e-03, 3.6106e-03, 3.7212e-03, 5.3787e-03, 1.8644e-03, 1.1425e-03,
          2.5009e-02, 5.1994e-03, 1.0933e-02, 3.0937e-03, 1.7899e-02, 6.9092e-02,
          1.1823e-01, 3.8266e-05, 3.8266e-05, 3.8266e-05, 1.5128e-04, 3.8266e-05,
          3.8266e-05, 1.0628e-04, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 1.5128e-04, 3.8266e-05, 3.8266e-05,
          1.0628e-04, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 1.0628e-04, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 2.1923e-04, 3.4189e-04, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.4189e-04, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.8266e-05, 1.0628e-04,
          3.8266e-05, 2.2900e-04, 1.0628e-04, 2.1923e-04, 3.8266e-05, 3.8266e-05,
          2.1923e-04, 3.8266e-05, 2.2900e-04, 3.8266e-05, 3.8266e-05, 1.0628e-04,
          3.8266e-05, 3.8266e-05, 1.0628e-04, 1.5128e-04, 3.8266e-05, 3.8266e-05,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 1.0628e-04, 1.0628e-04, 1.0628e-04,
          3.8266e-05, 3.8266e-05, 3.4189e-04, 3.8266e-05, 3.8266e-05, 1.0628e-04,
          1.0628e-04, 1.0628e-04, 3.8266e-05, 3.8266e-05, 3.4189e-04, 2.7394e-04,
          3.8266e-05, 3.8266e-05, 3.8266e-05, 2.3925e-04, 1.0628e-04, 3.8266e-05,
          5.4312e-04, 3.4189e-04, 3.8266e-05, 3.8266e-05, 4.3011e-04, 3.8266e-05,
          3.8266e-05, 1.0628e-04, 3.8266e-05, 1.0628e-04, 3.8266e-05, 3.8266e-05,
          1.0628e-04, 3.8266e-05, 3.8266e-05, 3.8266e-05, 3.4189e-04, 3.8266e-05,
          2.2900e-04, 1.0628e-04, 2.2900e-04, 1.0628e-04, 3.8266e-05, 4.3011e-04,
          4.3011e-04, 1.0628e-04, 1.0628e-04, 3.8266e-05, 2.2900e-04, 4.3011e-04,
          1.0628e-04, 4.3011e-04, 3.8266e-05, 1.0628e-04, 1.0628e-04, 3.8266e-05,
          2.2900e-04, 2.2900e-04, 2.2900e-04, 2.9874e-04, 1.0628e-04, 1.0628e-04,
          3.8266e-05, 1.0628e-04, 2.2900e-04, 4.2152e-04, 2.2900e-04, 2.2900e-04,
          1.0628e-04, 2.2900e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.1836e-03, 4.6478e-02, 1.8143e-02, 1.6632e-02, 8.7036e-02,
          2.9953e-02, 1.7868e-02, 8.1491e-04, 8.7261e-04, 1.6232e-03, 3.5343e-03,
          1.2121e-03, 7.4768e-04, 8.1444e-04, 1.2541e-03, 5.5361e-04, 8.5735e-04,
          4.9305e-04, 4.0913e-04, 7.5483e-04, 6.7806e-04, 4.6086e-04, 6.0987e-04,
          3.1586e-03, 2.2278e-03, 1.2001e-02, 2.5635e-03, 1.4553e-03, 7.8011e-04,
          8.7261e-04, 1.1616e-03, 4.6158e-04, 1.4734e-03, 5.8222e-04, 8.6498e-04,
          2.6035e-03, 2.8419e-03, 9.6607e-04, 8.8692e-04, 7.1049e-04, 1.0662e-03,
          1.7605e-03, 1.1168e-03, 1.5993e-03, 7.1764e-04, 6.0320e-04, 1.2217e-03,
          3.9959e-04, 6.4611e-04, 6.9885e-03, 1.7328e-03, 1.0328e-03, 7.9966e-04,
          1.2131e-03, 9.2268e-04, 2.7046e-03, 3.4370e-03, 2.7599e-03, 3.6793e-03,
          4.8790e-03, 1.1883e-03, 1.7452e-03, 1.0090e-03, 1.4830e-03, 1.6575e-03,
          1.4954e-03, 5.1308e-04, 5.3644e-04, 7.9155e-04, 9.2745e-04, 1.7481e-03,
          1.7252e-03, 2.1114e-03, 1.9932e-03, 1.3161e-03, 9.6750e-04, 6.5708e-04,
          1.5707e-03, 4.7073e-03, 2.2221e-03, 1.7071e-03, 1.7309e-03, 2.1610e-03,
          3.7308e-03, 1.5640e-03, 1.2941e-03, 8.6737e-04, 5.8413e-04, 9.4748e-04,
          1.4715e-03, 9.1934e-04, 1.0405e-03, 1.0233e-03, 1.0843e-03, 1.4496e-03,
          1.2522e-03, 1.2817e-03, 7.5150e-04, 2.0123e-03, 6.9427e-03, 3.1319e-03,
          2.6569e-03, 8.2016e-04, 8.3351e-04, 2.6703e-03, 1.9093e-03, 1.5755e-03,
          1.5078e-03, 1.5097e-03, 1.5001e-03, 1.0958e-03, 1.3609e-03, 6.6566e-04,
          1.0233e-03, 1.2007e-03, 6.9523e-04, 1.9341e-03, 1.8673e-03, 3.4256e-03,
          2.8496e-03, 3.5973e-03, 3.8280e-03, 6.1035e-04, 1.9512e-03, 1.4515e-03,
          4.9973e-03, 2.2182e-03, 2.7199e-03, 2.3346e-03, 7.4196e-04, 1.5869e-03,
          1.8148e-03, 6.0558e-04, 3.2687e-04, 4.0197e-04, 8.7690e-04, 1.0481e-03,
          1.1139e-03, 8.6594e-04, 7.4434e-04, 1.7395e-03, 1.4286e-03, 2.1172e-03,
          4.7646e-03, 8.9025e-04, 1.8969e-03, 2.4967e-03, 2.5368e-03, 4.0092e-03,
          3.8242e-03, 2.7676e-03, 5.1613e-03, 1.5879e-03, 2.7657e-03, 1.8950e-03,
          9.1839e-04, 1.0624e-03, 3.7074e-04, 1.3037e-03, 1.9369e-03, 8.8644e-04,
          1.8473e-03, 9.9373e-04, 9.0981e-04, 1.2226e-03, 8.8959e-03, 2.4090e-03,
          2.5539e-03, 1.6317e-03, 2.2316e-03, 1.2398e-03, 1.3771e-03, 3.4161e-03,
          3.9043e-03, 7.6103e-03, 1.0612e-02, 3.1357e-03, 1.5402e-03, 8.2684e-04,
          9.1887e-04, 9.2602e-04, 1.3323e-03, 9.9468e-04, 5.9366e-04, 6.3324e-04,
          1.0557e-03, 1.0099e-03, 1.3208e-03, 1.3290e-02, 3.3970e-03, 1.7691e-03,
          1.1978e-03, 9.7580e-03, 1.7824e-03, 1.6394e-03, 1.1988e-03, 1.9875e-03,
          1.1658e-02, 5.6305e-03, 2.1992e-03, 1.0710e-03, 9.7847e-04, 9.7370e-04,
          1.7214e-03, 1.1587e-03, 1.7910e-03, 5.5647e-04, 1.8559e-03, 5.8317e-04,
          9.3699e-04, 8.1787e-03, 2.5497e-02, 1.9522e-03, 3.8948e-03, 5.7755e-03,
          3.4008e-03, 2.6817e-03, 1.1654e-03, 8.9741e-04, 2.8934e-03, 2.8801e-03,
          3.4542e-03, 7.5722e-03, 1.9569e-03, 9.9373e-04, 2.5349e-03, 1.4315e-03,
          9.2459e-04, 2.5692e-03, 2.2678e-03, 4.0131e-03, 1.9226e-03, 1.1158e-03,
          2.9617e-02, 3.9101e-03, 7.2899e-03, 2.7809e-03, 1.5106e-02, 4.7638e-02,
          8.1787e-02, 8.1848e-02, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 1.3638e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          1.3638e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 2.9588e-04, 4.7946e-04, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.7946e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 1.3638e-04,
          4.0889e-05, 3.1972e-04, 1.3638e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          1.3638e-04, 4.0889e-05, 1.3638e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 1.3638e-04, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 2.9588e-04, 4.0889e-05, 4.0889e-05, 1.3638e-04,
          4.0889e-05, 1.3638e-04, 4.0889e-05, 4.0889e-05, 4.7946e-04, 2.0051e-04,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 5.0449e-04, 1.3638e-04, 1.3638e-04,
          4.7946e-04, 4.7946e-04, 4.0889e-05, 4.0889e-05, 1.3638e-04, 4.0889e-05,
          1.3638e-04, 4.0889e-05, 4.0889e-05, 1.3638e-04, 4.0889e-05, 4.0889e-05,
          4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05, 2.9588e-04, 4.0889e-05,
          1.3638e-04, 4.0889e-05, 3.1972e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          3.1972e-04, 1.3638e-04, 1.3638e-04, 4.0889e-05, 3.1972e-04, 6.0034e-04,
          1.3638e-04, 3.1972e-04, 4.0889e-05, 4.0889e-05, 4.0889e-05, 4.0889e-05,
          3.1972e-04, 1.3638e-04, 3.1972e-04, 2.0885e-04, 1.3638e-04, 4.0889e-05,
          4.0889e-05, 1.3638e-04, 1.3638e-04, 4.9686e-04, 1.3638e-04, 3.1972e-04,
          1.3638e-04, 1.3638e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.9820e-03, 4.1870e-02, 1.3290e-02, 1.3115e-02, 5.8289e-02,
          2.4445e-02, 1.2390e-02, 1.0033e-03, 9.5797e-04, 2.1076e-03, 3.8929e-03,
          1.3647e-03, 9.9468e-04, 8.8930e-04, 1.4868e-03, 8.2827e-04, 1.4400e-03,
          6.7472e-04, 5.7602e-04, 6.3515e-04, 6.8140e-04, 5.4932e-04, 8.7595e-04,
          3.6812e-03, 2.3441e-03, 8.7585e-03, 2.7752e-03, 1.3599e-03, 1.4772e-03,
          1.1625e-03, 1.8520e-03, 7.9012e-04, 2.4109e-03, 1.0033e-03, 1.5020e-03,
          3.5038e-03, 2.0809e-03, 2.0485e-03, 1.4858e-03, 9.2268e-04, 2.6627e-03,
          1.6041e-03, 1.3523e-03, 1.0595e-03, 8.2684e-04, 9.2268e-04, 2.4509e-03,
          5.2214e-04, 7.2098e-04, 4.8523e-03, 2.9297e-03, 1.5898e-03, 1.1892e-03,
          1.1148e-03, 1.3151e-03, 1.9293e-03, 3.7079e-03, 3.1128e-03, 3.4199e-03,
          3.0231e-03, 1.7357e-03, 1.7605e-03, 1.9112e-03, 2.0504e-03, 2.7294e-03,
          2.4910e-03, 8.5783e-04, 1.0500e-03, 8.1491e-04, 7.7200e-04, 1.2684e-03,
          1.0195e-03, 3.3798e-03, 1.6356e-03, 2.0084e-03, 8.6308e-04, 8.8263e-04,
          1.7338e-03, 5.2071e-03, 3.0613e-03, 2.6722e-03, 2.3956e-03, 1.8740e-03,
          2.5425e-03, 2.5082e-03, 1.3227e-03, 1.4448e-03, 1.4915e-03, 1.6556e-03,
          2.4090e-03, 2.4242e-03, 1.4591e-03, 1.6136e-03, 1.0500e-03, 9.8801e-04,
          1.4420e-03, 1.0309e-03, 1.4677e-03, 4.1046e-03, 6.6071e-03, 5.1651e-03,
          4.0894e-03, 1.9798e-03, 1.7529e-03, 4.1847e-03, 2.8725e-03, 2.3594e-03,
          2.3575e-03, 2.1076e-03, 3.4466e-03, 1.5955e-03, 2.7676e-03, 1.4820e-03,
          1.6022e-03, 1.9951e-03, 8.6594e-04, 2.7027e-03, 1.3189e-03, 3.2063e-03,
          3.4599e-03, 6.9046e-03, 3.0556e-03, 8.3828e-04, 2.4929e-03, 2.4395e-03,
          5.7907e-03, 3.4218e-03, 3.9978e-03, 3.8605e-03, 1.4534e-03, 2.5806e-03,
          3.1757e-03, 1.1435e-03, 3.9005e-04, 7.6389e-04, 1.9150e-03, 1.2598e-03,
          1.9083e-03, 1.1730e-03, 1.3905e-03, 2.0447e-03, 1.8005e-03, 3.0403e-03,
          5.1041e-03, 1.5745e-03, 2.7046e-03, 4.3182e-03, 4.1580e-03, 5.6000e-03,
          7.6180e-03, 4.4174e-03, 5.1651e-03, 3.4046e-03, 2.9106e-03, 4.2343e-03,
          2.4319e-03, 3.0212e-03, 5.2929e-04, 1.7071e-03, 4.1771e-03, 1.4849e-03,
          2.4166e-03, 2.5673e-03, 1.5297e-03, 1.8835e-03, 1.1322e-02, 5.5084e-03,
          4.1733e-03, 4.0474e-03, 3.3741e-03, 1.4772e-03, 2.1248e-03, 5.3329e-03,
          4.9820e-03, 5.9433e-03, 8.0414e-03, 3.8319e-03, 3.4351e-03, 1.0672e-03,
          1.3685e-03, 1.4181e-03, 2.0409e-03, 2.0504e-03, 1.0900e-03, 1.0347e-03,
          1.3762e-03, 1.4381e-03, 1.6203e-03, 9.4147e-03, 4.7073e-03, 2.4586e-03,
          2.0294e-03, 6.0310e-03, 2.7466e-03, 2.7981e-03, 2.2697e-03, 3.0136e-03,
          8.4991e-03, 6.4201e-03, 4.1504e-03, 2.2049e-03, 1.9665e-03, 1.7672e-03,
          2.0275e-03, 2.8362e-03, 1.8520e-03, 7.7581e-04, 2.8687e-03, 1.0309e-03,
          1.2951e-03, 1.0170e-02, 1.0925e-02, 2.9659e-03, 5.7526e-03, 6.0158e-03,
          6.8207e-03, 4.7264e-03, 1.9741e-03, 1.2102e-03, 4.9019e-03, 4.2839e-03,
          7.3509e-03, 4.3221e-03, 3.6755e-03, 2.0218e-03, 3.5725e-03, 2.1152e-03,
          1.6661e-03, 3.1414e-03, 2.0752e-03, 4.7760e-03, 2.9945e-03, 1.7395e-03,
          1.4275e-02, 5.7297e-03, 6.9389e-03, 3.4275e-03, 1.8753e-02, 5.2216e-02,
          3.7109e-02, 2.9968e-02, 4.1321e-02, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.0538e-04,
          1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 2.7514e-04, 1.0538e-04, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 2.7514e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.9372e-05, 1.0538e-04,
          1.0538e-04, 3.9124e-04, 1.0538e-04, 1.0538e-04, 1.9372e-05, 1.9372e-05,
          1.0538e-04, 1.9372e-05, 1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.0538e-04, 1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 1.0538e-04, 1.9372e-05, 1.0538e-04, 1.9372e-05,
          1.9372e-05, 1.9372e-05, 2.7514e-04, 1.0538e-04, 1.9372e-05, 1.0538e-04,
          1.9372e-05, 1.0538e-04, 1.9372e-05, 1.9372e-05, 2.7514e-04, 2.7514e-04,
          1.9372e-05, 1.0538e-04, 1.0538e-04, 1.0538e-04, 1.0538e-04, 1.0538e-04,
          2.7514e-04, 2.7514e-04, 1.0538e-04, 1.9372e-05, 1.0538e-04, 1.9372e-05,
          1.0538e-04, 1.0538e-04, 1.9372e-05, 1.0538e-04, 1.9372e-05, 1.9372e-05,
          1.0538e-04, 1.9372e-05, 1.9372e-05, 1.9372e-05, 2.7514e-04, 1.9372e-05,
          1.0538e-04, 1.0538e-04, 3.9124e-04, 1.0538e-04, 1.9372e-05, 1.0538e-04,
          1.0538e-04, 1.0538e-04, 2.7514e-04, 1.9372e-05, 1.0538e-04, 1.0538e-04,
          1.0538e-04, 1.0538e-04, 1.9372e-05, 1.0538e-04, 1.9372e-05, 1.9372e-05,
          3.9124e-04, 3.9124e-04, 3.9124e-04, 1.0538e-04, 1.0538e-04, 1.0538e-04,
          1.9372e-05, 1.0538e-04, 1.0538e-04, 3.9124e-04, 1.9372e-05, 3.9124e-04,
          1.0538e-04, 1.0538e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.4855e-03, 5.2948e-02, 1.6006e-02, 1.4473e-02, 5.3375e-02,
          3.5431e-02, 1.0612e-02, 1.0109e-03, 9.1648e-04, 1.9093e-03, 2.1439e-03,
          1.0166e-03, 2.0905e-03, 1.0109e-03, 2.5921e-03, 9.8896e-04, 4.0092e-03,
          1.5707e-03, 5.7030e-04, 9.2220e-04, 1.1816e-03, 7.3957e-04, 2.2411e-03,
          2.2240e-03, 2.2049e-03, 8.2626e-03, 5.1460e-03, 1.6441e-03, 3.9253e-03,
          1.3227e-03, 2.6550e-03, 2.1439e-03, 5.9052e-03, 1.9131e-03, 2.1629e-03,
          5.4817e-03, 1.3561e-03, 4.4556e-03, 3.5992e-03, 1.2360e-03, 2.8267e-03,
          8.3160e-04, 7.2622e-04, 5.7507e-04, 7.6532e-04, 2.4529e-03, 7.3929e-03,
          6.8951e-04, 9.5320e-04, 5.1651e-03, 4.3335e-03, 2.7103e-03, 1.4200e-03,
          9.5367e-04, 1.7424e-03, 1.5869e-03, 2.9678e-03, 1.8444e-03, 2.4490e-03,
          1.9855e-03, 4.6005e-03, 1.7004e-03, 3.2120e-03, 9.3746e-04, 5.5733e-03,
          7.9966e-04, 4.9305e-04, 9.8324e-04, 5.6744e-04, 4.3941e-04, 5.1737e-04,
          7.0667e-04, 4.1580e-03, 1.6184e-03, 3.2558e-03, 7.2718e-04, 9.1457e-04,
          1.9474e-03, 3.2215e-03, 1.5192e-03, 3.4428e-03, 1.2836e-03, 1.4467e-03,
          1.4372e-03, 4.2191e-03, 1.1387e-03, 1.0920e-03, 5.3406e-04, 5.6219e-04,
          1.6851e-03, 2.1324e-03, 4.9353e-04, 1.2465e-03, 8.2397e-04, 9.5463e-04,
          2.1477e-03, 9.4891e-04, 3.0518e-03, 1.1452e-02, 2.8706e-03, 3.5877e-03,
          2.8210e-03, 8.7595e-04, 8.8692e-04, 4.8409e-03, 5.0545e-03, 4.6501e-03,
          3.1548e-03, 1.8330e-03, 3.2368e-03, 6.5470e-04, 7.2718e-04, 1.1683e-03,
          1.3142e-03, 2.8305e-03, 1.9722e-03, 6.4430e-03, 1.0719e-03, 3.8776e-03,
          5.0545e-03, 7.8430e-03, 1.0691e-03, 7.7581e-04, 4.6768e-03, 9.2745e-04,
          2.7618e-03, 1.2522e-03, 1.6766e-03, 8.4381e-03, 2.9430e-03, 3.6697e-03,
          1.3285e-03, 6.2084e-04, 4.1699e-04, 4.6301e-04, 1.4486e-03, 7.6723e-04,
          8.6365e-03, 2.4605e-03, 1.9531e-03, 3.1891e-03, 4.6310e-03, 4.8523e-03,
          1.6460e-03, 7.7343e-04, 1.7805e-03, 3.2673e-03, 1.3437e-03, 7.6294e-03,
          1.0750e-02, 7.1831e-03, 4.2191e-03, 7.7858e-03, 3.4313e-03, 5.5771e-03,
          2.5711e-03, 3.6411e-03, 4.8184e-04, 7.5388e-04, 7.1831e-03, 4.6730e-03,
          5.2414e-03, 5.4398e-03, 4.3488e-03, 4.3488e-03, 6.9389e-03, 3.9215e-03,
          2.4166e-03, 3.7727e-03, 3.9444e-03, 2.0447e-03, 4.3945e-03, 1.2749e-02,
          4.0131e-03, 2.8725e-03, 5.9891e-03, 4.1656e-03, 4.9896e-03, 2.2545e-03,
          1.7967e-03, 2.2488e-03, 3.8242e-03, 1.9817e-03, 1.6785e-03, 1.9274e-03,
          1.3533e-03, 2.4490e-03, 2.1057e-03, 3.6068e-03, 1.4410e-03, 4.5471e-03,
          2.9736e-03, 2.2087e-03, 5.4512e-03, 6.3400e-03, 2.3956e-03, 6.5384e-03,
          5.0278e-03, 7.2365e-03, 6.4163e-03, 1.4505e-03, 1.5860e-03, 2.0885e-03,
          9.1505e-04, 2.9049e-03, 1.4858e-03, 9.6941e-04, 2.4014e-03, 2.7809e-03,
          1.2379e-03, 8.7509e-03, 3.6716e-03, 1.4734e-03, 7.7515e-03, 2.0046e-03,
          1.2047e-02, 1.0391e-02, 2.8286e-03, 2.1343e-03, 1.5434e-02, 7.6103e-03,
          2.8564e-02, 2.4071e-03, 3.4313e-03, 1.4153e-03, 6.1607e-03, 3.3550e-03,
          9.0647e-04, 8.5974e-04, 9.7752e-04, 2.0332e-03, 4.3983e-03, 2.3327e-03,
          5.0507e-03, 2.7447e-03, 1.5831e-03, 2.6321e-03, 1.0864e-02, 1.0666e-02,
          5.6648e-03, 5.5084e-03, 6.4163e-03, 3.8086e-02, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 1.2362e-04, 2.0146e-05, 2.0146e-05, 1.2362e-04, 2.0146e-05,
          1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 1.2362e-04,
          2.0146e-05, 1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 1.6534e-04, 2.0146e-05,
          6.5327e-04, 1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 1.6534e-04, 2.0146e-05, 2.6894e-04,
          2.6894e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          1.6534e-04, 2.0146e-05, 1.2362e-04, 1.2362e-04, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.6894e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.6894e-04,
          2.0146e-05, 2.0146e-05, 1.6534e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 1.2362e-04, 2.6894e-04,
          2.0146e-05, 2.0146e-05, 1.2362e-04, 1.2362e-04, 2.0146e-05, 2.0146e-05,
          2.0146e-05, 6.5327e-04, 2.0146e-05, 2.0146e-05, 2.6894e-04, 2.6894e-04,
          2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.6894e-04, 2.0146e-05,
          2.6894e-04, 2.6894e-04, 2.0146e-05, 2.0146e-05, 1.2362e-04, 2.0146e-05,
          1.6534e-04, 1.2362e-04, 2.0146e-05, 1.2362e-04, 2.0146e-05, 2.0146e-05,
          1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.6894e-04, 2.0146e-05,
          1.2362e-04, 2.6894e-04, 1.2362e-04, 2.0146e-05, 2.0146e-05, 1.2362e-04,
          1.2362e-04, 2.0146e-05, 2.6894e-04, 2.0146e-05, 1.2362e-04, 2.0146e-05,
          1.2362e-04, 1.2362e-04, 2.0146e-05, 2.0146e-05, 2.0146e-05, 2.0146e-05,
          4.1246e-04, 1.2362e-04, 4.1246e-04, 2.0146e-05, 1.2362e-04, 2.0146e-05,
          2.0146e-05, 1.2362e-04, 1.2362e-04, 1.2362e-04, 2.0146e-05, 4.1246e-04,
          1.2362e-04, 1.2362e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.8975e-03, 5.5054e-02, 1.7075e-02, 1.5221e-02, 4.2633e-02,
          3.8452e-02, 1.0574e-02, 1.7023e-03, 1.2798e-03, 1.9836e-03, 2.9793e-03,
          1.2369e-03, 4.1542e-03, 2.8572e-03, 2.3594e-03, 1.5955e-03, 8.3466e-03,
          4.8981e-03, 7.3576e-04, 9.6416e-04, 8.5735e-04, 1.0538e-03, 3.8452e-03,
          3.2654e-03, 1.8578e-03, 8.5831e-03, 5.7335e-03, 1.7509e-03, 8.5144e-03,
          4.2419e-03, 3.4599e-03, 2.9774e-03, 3.9043e-03, 1.5335e-03, 2.4834e-03,
          2.0771e-03, 8.2588e-04, 2.1439e-03, 2.1019e-03, 1.3762e-03, 1.4334e-03,
          8.0395e-04, 6.0844e-04, 7.3242e-04, 5.6362e-04, 6.7406e-03, 3.3474e-03,
          1.5850e-03, 1.5774e-03, 4.4327e-03, 2.9507e-03, 2.6417e-03, 1.7004e-03,
          8.1444e-04, 1.5182e-03, 1.1597e-03, 9.4652e-04, 1.3857e-03, 1.1797e-03,
          1.3094e-03, 5.0507e-03, 1.5297e-03, 2.1172e-03, 8.5020e-04, 3.7899e-03,
          1.1702e-03, 5.7507e-04, 7.7295e-04, 4.1556e-04, 3.9721e-04, 5.4169e-04,
          6.6185e-04, 4.0588e-03, 1.2045e-03, 3.5267e-03, 7.6580e-04, 6.6376e-04,
          1.4849e-03, 1.0681e-03, 1.0271e-03, 1.5917e-03, 9.2649e-04, 8.4114e-04,
          9.1410e-04, 4.1389e-03, 1.1730e-03, 1.3514e-03, 5.5933e-04, 6.4182e-04,
          2.2964e-03, 1.6403e-03, 5.5122e-04, 1.4982e-03, 6.1846e-04, 7.5293e-04,
          2.8133e-03, 7.4720e-04, 3.9062e-03, 4.3221e-03, 1.1034e-03, 1.2426e-03,
          2.5959e-03, 6.7186e-04, 6.9952e-04, 1.5335e-03, 2.9125e-03, 8.0566e-03,
          3.3741e-03, 9.0408e-04, 2.1019e-03, 6.9475e-04, 1.0080e-03, 8.2731e-04,
          1.0834e-03, 1.1654e-03, 3.0804e-03, 4.3449e-03, 1.0471e-03, 1.1848e-02,
          2.0737e-02, 2.9964e-03, 6.6471e-04, 9.8515e-04, 3.4752e-03, 5.8460e-04,
          9.1076e-04, 7.2384e-04, 1.0958e-03, 4.1199e-03, 3.8185e-03, 3.8948e-03,
          1.8930e-03, 7.0429e-04, 5.6410e-04, 4.0913e-04, 1.1415e-03, 5.9509e-04,
          1.2337e-02, 7.0953e-03, 5.6953e-03, 4.5776e-03, 4.0970e-03, 1.8143e-02,
          1.0519e-03, 5.3692e-04, 1.0414e-03, 1.0319e-03, 1.2293e-03, 3.5534e-03,
          3.3760e-03, 2.9335e-03, 1.3971e-03, 4.1084e-03, 2.9564e-03, 1.9817e-03,
          1.4181e-03, 1.2283e-03, 6.7377e-04, 8.4209e-04, 3.5458e-03, 1.6968e-02,
          3.3951e-03, 2.1229e-03, 1.0567e-02, 3.0518e-03, 2.9984e-03, 1.2541e-03,
          2.0180e-03, 1.2140e-03, 2.5272e-03, 1.8253e-03, 8.4381e-03, 3.8776e-03,
          2.1515e-03, 1.7977e-03, 2.6417e-03, 2.1667e-03, 2.9087e-03, 3.2253e-03,
          1.3542e-03, 2.2049e-03, 2.8744e-03, 3.0556e-03, 1.4868e-03, 1.8368e-03,
          1.8730e-03, 1.2627e-02, 1.9588e-03, 2.0065e-03, 1.5335e-03, 3.0632e-03,
          2.9354e-03, 9.1887e-04, 2.8858e-03, 1.2115e-02, 1.6489e-03, 2.5463e-03,
          2.3918e-03, 9.9335e-03, 4.6196e-03, 1.2770e-03, 1.1234e-03, 1.5116e-03,
          1.0977e-03, 1.6003e-03, 1.2169e-03, 1.0071e-03, 2.6569e-03, 4.4937e-03,
          1.6298e-03, 5.4245e-03, 2.2659e-03, 1.3800e-03, 5.6725e-03, 1.5745e-03,
          3.2940e-03, 1.6373e-02, 2.2259e-03, 1.4782e-03, 6.3782e-03, 3.1311e-02,
          1.0712e-02, 2.0027e-03, 2.2697e-03, 1.2197e-03, 5.8174e-03, 3.0041e-03,
          1.0900e-03, 1.1969e-03, 1.0204e-03, 2.2316e-03, 2.7790e-03, 1.5839e-02,
          3.6335e-03, 1.6975e-03, 1.9312e-03, 2.0962e-03, 5.8975e-03, 9.5978e-03,
          4.5776e-03, 3.9330e-03, 3.3112e-03, 7.2594e-03, 7.9651e-02, 2.2650e-05,
          2.2650e-05, 1.4210e-04, 2.2650e-05, 1.5080e-04, 2.2650e-05, 2.2650e-05,
          1.4210e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 1.4210e-04,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 1.0395e-04, 2.2650e-05,
          1.4210e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 1.0395e-04, 2.2650e-05, 1.4210e-04,
          2.2340e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          1.0395e-04, 2.2650e-05, 2.2650e-05, 1.4210e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 1.4210e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 1.0395e-04,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          1.4210e-04, 4.4465e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 1.0395e-04,
          2.2650e-05, 2.2650e-05, 1.0395e-04, 1.0395e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 1.4210e-04, 1.0395e-04,
          2.2650e-05, 1.0395e-04, 2.2650e-05, 1.4210e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2340e-04, 2.2650e-05, 2.2650e-05, 2.2340e-04, 1.0395e-04,
          2.2650e-05, 2.2650e-05, 1.0395e-04, 2.2650e-05, 1.0395e-04, 1.4210e-04,
          4.6515e-04, 2.2340e-04, 2.2650e-05, 1.0395e-04, 1.4210e-04, 2.2650e-05,
          1.0395e-04, 2.2650e-05, 1.0395e-04, 2.2340e-04, 2.2650e-05, 2.2650e-05,
          1.4210e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 3.4571e-04, 2.2650e-05,
          1.4210e-04, 1.0395e-04, 4.4465e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          2.2340e-04, 2.2650e-05, 1.0395e-04, 2.2650e-05, 1.4210e-04, 1.4210e-04,
          2.2650e-05, 3.8385e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          3.6335e-04, 3.6335e-04, 3.6335e-04, 3.6335e-04, 3.6335e-04, 1.4210e-04,
          2.2650e-05, 4.4465e-04, 3.6335e-04, 1.4210e-04, 2.2650e-05, 1.4210e-04,
          3.6335e-04, 3.6335e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.7896e-03, 4.4922e-02, 1.6983e-02, 2.0340e-02, 7.0984e-02,
          2.7969e-02, 1.5282e-02, 1.0462e-03, 8.2350e-04, 1.3409e-03, 1.9970e-03,
          8.3780e-04, 3.3474e-03, 7.1106e-03, 1.4391e-03, 9.1648e-04, 8.3389e-03,
          4.7073e-03, 5.3501e-04, 2.0638e-03, 1.4067e-03, 1.6956e-03, 2.7428e-03,
          2.0561e-03, 1.3466e-03, 1.1871e-02, 4.8103e-03, 3.1452e-03, 5.6305e-03,
          1.3580e-02, 1.7271e-03, 2.6932e-03, 2.2736e-03, 1.0166e-03, 1.3552e-03,
          1.8559e-03, 1.0252e-03, 1.8969e-03, 1.4715e-03, 3.6755e-03, 1.6651e-03,
          1.1339e-03, 8.2970e-04, 9.3889e-04, 6.0415e-04, 5.2681e-03, 2.4548e-03,
          2.9354e-03, 2.8629e-03, 8.1711e-03, 1.8816e-03, 1.2627e-03, 8.4352e-04,
          8.0824e-04, 9.2936e-04, 1.3437e-03, 1.1587e-03, 1.2350e-03, 1.2159e-03,
          1.3809e-03, 1.1414e-02, 1.7166e-03, 1.8063e-03, 1.0185e-03, 2.6379e-03,
          1.4257e-03, 4.9162e-04, 8.4734e-04, 6.1607e-04, 5.7793e-04, 7.9346e-04,
          9.6369e-04, 2.4872e-03, 1.3256e-03, 1.6756e-03, 7.3576e-04, 5.5361e-04,
          1.5421e-03, 1.3285e-03, 1.2579e-03, 1.0967e-03, 1.2236e-03, 8.9121e-04,
          1.1520e-03, 2.3384e-03, 1.5678e-03, 8.7023e-04, 6.5327e-04, 1.1539e-03,
          1.4343e-03, 1.0147e-03, 7.0047e-04, 1.0023e-03, 6.6566e-04, 9.9277e-04,
          3.4752e-03, 9.3460e-04, 1.3437e-03, 2.2869e-03, 1.4048e-03, 1.0967e-03,
          1.6527e-03, 6.4611e-04, 6.4898e-04, 1.3714e-03, 1.6623e-03, 1.4343e-02,
          1.9569e-03, 1.5059e-03, 1.5745e-03, 8.1062e-04, 1.0586e-03, 8.0109e-04,
          7.7677e-04, 1.3695e-03, 6.2294e-03, 2.6302e-03, 1.4286e-03, 8.2321e-03,
          2.2446e-02, 1.7233e-03, 8.4972e-04, 5.9557e-04, 1.9255e-03, 6.8140e-04,
          1.1816e-03, 1.0786e-03, 1.2236e-03, 2.2240e-03, 1.8053e-03, 2.1687e-03,
          1.4744e-03, 7.7868e-04, 4.5156e-04, 4.7755e-04, 8.8453e-04, 1.0109e-03,
          8.5983e-03, 1.5221e-02, 5.3215e-03, 3.3226e-03, 2.4452e-03, 2.2125e-02,
          1.2159e-03, 3.9959e-04, 9.2125e-04, 8.9741e-04, 1.4620e-03, 2.1973e-03,
          2.1324e-03, 1.8339e-03, 1.5841e-03, 3.4180e-03, 3.0956e-03, 2.9202e-03,
          1.0862e-03, 1.2722e-03, 5.1832e-04, 1.1797e-03, 3.0041e-03, 1.4999e-02,
          2.4700e-03, 1.6241e-03, 7.5912e-03, 1.9140e-03, 2.7981e-03, 9.7322e-04,
          1.3456e-03, 8.9216e-04, 1.4534e-03, 1.0214e-03, 8.0414e-03, 2.6894e-03,
          1.4305e-03, 1.8206e-03, 3.7270e-03, 4.1389e-03, 2.3308e-03, 2.2392e-03,
          1.1644e-03, 1.4038e-03, 2.1038e-03, 1.7138e-03, 1.1721e-03, 1.2751e-03,
          1.0738e-03, 1.0902e-02, 1.7071e-03, 2.1591e-03, 1.9321e-03, 1.8864e-03,
          1.4629e-03, 1.1406e-03, 1.8320e-03, 1.0033e-02, 1.0395e-03, 1.7738e-03,
          3.2864e-03, 2.3865e-02, 3.8509e-03, 1.4400e-03, 1.3332e-03, 1.4133e-03,
          1.5306e-03, 1.6546e-03, 2.2182e-03, 6.9046e-04, 1.7824e-03, 3.0403e-03,
          1.2045e-03, 4.7264e-03, 3.1967e-03, 1.0118e-03, 3.0899e-03, 1.5459e-03,
          1.9608e-03, 1.5190e-02, 1.6413e-03, 9.2363e-04, 3.8223e-03, 1.9455e-02,
          9.6588e-03, 3.8910e-03, 2.4185e-03, 1.1482e-03, 3.9902e-03, 2.0847e-03,
          1.2569e-03, 1.3151e-03, 1.4009e-03, 1.6432e-03, 2.0733e-03, 9.9106e-03,
          5.3062e-03, 1.3294e-03, 1.4725e-03, 1.1473e-03, 3.6736e-03, 5.7564e-03,
          4.1084e-03, 5.0163e-03, 2.8095e-03, 5.6992e-03, 3.4180e-02, 7.7698e-02,
          1.8477e-05, 1.2124e-04, 1.8477e-05, 9.3699e-05, 1.8477e-05, 9.3699e-05,
          1.2124e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.2124e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          6.5184e-04, 1.2124e-04, 9.3699e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.2124e-04,
          1.9646e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          9.3699e-05, 1.8477e-05, 1.2124e-04, 3.7098e-04, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 9.3699e-05, 1.2124e-04, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.9646e-04, 1.8477e-05, 1.2124e-04, 9.3699e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 9.3699e-05, 9.3699e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05, 1.9646e-04,
          1.2124e-04, 3.7098e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05, 2.9588e-04, 1.8477e-05,
          1.8477e-05, 9.3699e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.2124e-04, 1.8477e-05, 1.8477e-05, 1.9646e-04, 9.3699e-05,
          1.8477e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          1.9646e-04, 1.9646e-04, 1.8477e-05, 1.8477e-05, 1.2124e-04, 9.3699e-05,
          9.3699e-05, 1.8477e-05, 9.3699e-05, 1.9646e-04, 1.8477e-05, 1.8477e-05,
          1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05, 9.3699e-05, 1.8477e-05,
          1.2124e-04, 1.8477e-05, 3.7098e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          3.7098e-04, 1.8477e-05, 1.8477e-05, 9.3699e-05, 1.9646e-04, 1.2124e-04,
          1.8477e-05, 1.2124e-04, 1.8477e-05, 1.8477e-05, 1.8477e-05, 1.8477e-05,
          2.9588e-04, 2.9588e-04, 2.9588e-04, 2.9588e-04, 2.9588e-04, 1.2124e-04,
          1.8477e-05, 1.9646e-04, 2.9588e-04, 3.2163e-04, 1.2124e-04, 1.2124e-04,
          2.9588e-04, 1.2124e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.8245e-03, 4.7913e-02, 1.8738e-02, 2.1164e-02, 8.1665e-02,
          2.9907e-02, 2.0309e-02, 4.5371e-04, 4.7803e-04, 1.5545e-03, 2.4185e-03,
          9.4938e-04, 6.8951e-04, 8.0967e-04, 1.0338e-03, 4.9400e-04, 1.4496e-03,
          7.9966e-04, 3.4356e-04, 1.4582e-03, 1.7099e-03, 1.2007e-03, 8.7357e-04,
          3.0918e-03, 2.1992e-03, 1.4343e-02, 3.8033e-03, 1.6470e-03, 1.2865e-03,
          8.5354e-04, 7.1812e-04, 6.7520e-04, 1.8549e-03, 8.8978e-04, 6.6423e-04,
          3.5076e-03, 2.4109e-03, 3.8548e-03, 1.6594e-03, 1.2693e-03, 9.1324e-03,
          8.5144e-03, 4.2686e-03, 3.7804e-03, 2.6913e-03, 9.5224e-04, 3.3989e-03,
          6.3705e-04, 1.3351e-03, 9.8801e-03, 1.2989e-03, 6.6757e-04, 5.4884e-04,
          5.5218e-04, 5.4407e-04, 1.1225e-03, 3.4409e-03, 1.7071e-03, 3.5458e-03,
          2.9755e-03, 2.0485e-03, 2.4357e-03, 3.2558e-03, 5.2834e-03, 4.7379e-03,
          2.1877e-03, 1.0529e-03, 1.7080e-03, 3.7575e-03, 3.5038e-03, 1.9588e-03,
          2.6417e-03, 3.0365e-03, 1.4029e-03, 7.8106e-04, 4.8161e-04, 3.5715e-04,
          8.6927e-04, 3.3226e-03, 1.7290e-03, 2.1038e-03, 1.6632e-03, 1.8234e-03,
          2.6493e-03, 2.8076e-03, 1.8444e-03, 1.3828e-03, 1.0109e-03, 1.5059e-03,
          2.9240e-03, 3.4599e-03, 1.4772e-03, 2.0924e-03, 2.2945e-03, 2.8477e-03,
          1.1816e-03, 9.2888e-04, 6.8092e-04, 2.1286e-03, 3.7193e-03, 2.0790e-03,
          2.0428e-03, 7.8583e-04, 7.5436e-04, 3.3913e-03, 1.8845e-03, 2.7962e-03,
          2.1706e-03, 1.3008e-02, 5.4359e-03, 2.3365e-03, 1.4219e-03, 1.4381e-03,
          1.5373e-03, 9.3765e-03, 2.2316e-03, 3.5477e-03, 3.8700e-03, 2.3212e-03,
          1.8482e-03, 3.1567e-03, 1.6565e-03, 2.8253e-04, 1.7757e-03, 7.3338e-04,
          2.6360e-03, 1.3599e-03, 1.3332e-03, 2.3155e-03, 7.8630e-04, 2.2793e-03,
          1.7614e-03, 1.2455e-03, 4.4012e-04, 1.0090e-03, 3.0937e-03, 4.6730e-03,
          2.6455e-03, 1.7424e-03, 9.1934e-04, 2.1019e-03, 1.5697e-03, 1.4772e-03,
          1.9493e-03, 6.4278e-04, 1.3628e-03, 1.4877e-03, 1.3132e-03, 2.4223e-03,
          3.8567e-03, 2.6417e-03, 3.4409e-03, 5.4855e-03, 3.9902e-03, 2.1729e-02,
          3.4256e-03, 5.6152e-03, 5.2786e-04, 3.6716e-03, 7.8049e-03, 2.6035e-03,
          3.1357e-03, 2.8400e-03, 1.5574e-03, 1.6565e-03, 6.0959e-03, 1.4343e-03,
          1.7090e-03, 1.2398e-03, 1.5984e-03, 7.8440e-04, 1.2817e-03, 3.8548e-03,
          2.4166e-03, 2.5806e-03, 6.6643e-03, 3.2623e-02, 8.2855e-03, 1.3361e-03,
          1.8301e-03, 1.7109e-03, 3.0499e-03, 8.7261e-04, 7.9918e-04, 9.4891e-04,
          1.1101e-03, 1.5574e-03, 1.3714e-03, 4.1351e-03, 1.2226e-03, 1.5450e-03,
          7.0381e-04, 2.1458e-03, 1.5526e-03, 1.9817e-03, 1.2016e-03, 2.1687e-03,
          5.0316e-03, 5.5771e-03, 7.6408e-03, 4.7073e-03, 1.7815e-03, 2.5978e-03,
          4.2458e-03, 8.1940e-03, 1.3893e-02, 5.9891e-04, 3.1013e-03, 1.0729e-03,
          8.5354e-04, 7.3624e-03, 6.8436e-03, 9.6750e-04, 2.3346e-03, 1.7242e-03,
          2.5349e-03, 2.1420e-03, 1.3113e-03, 5.8270e-04, 2.6169e-03, 1.6794e-03,
          5.8975e-03, 2.6062e-02, 1.3367e-02, 1.8902e-03, 5.0316e-03, 1.7672e-03,
          2.8591e-03, 2.9507e-03, 6.2599e-03, 2.7637e-03, 3.6507e-03, 1.4553e-03,
          9.4604e-03, 1.6813e-03, 1.6155e-03, 1.2083e-03, 5.0316e-03, 5.4054e-03,
          3.9368e-03, 4.9629e-03, 3.4466e-03, 4.2648e-03, 3.8033e-03, 7.7286e-03,
          4.7699e-02, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 1.4257e-04, 1.4257e-04, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 1.4257e-04, 2.4319e-05, 2.4319e-05,
          1.4257e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          1.4257e-04, 2.4319e-05, 3.6430e-04, 3.6430e-04, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 1.4257e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 1.4257e-04, 2.4319e-05, 2.4319e-05,
          3.6430e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 1.4257e-04,
          2.4319e-05, 2.4319e-05, 1.4257e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 1.4257e-04,
          2.4319e-05, 2.4319e-05, 3.6430e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 3.6430e-04, 1.4257e-04,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 3.6430e-04, 2.4319e-05,
          3.6430e-04, 3.6430e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          1.4257e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05, 3.6430e-04, 2.4319e-05,
          2.4605e-04, 1.4257e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          3.6430e-04, 2.4319e-05, 3.6430e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4605e-04, 2.4319e-05, 2.4319e-05, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 1.2058e-04, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05, 2.4319e-05, 3.2258e-04, 2.4319e-05, 2.4319e-05,
          2.4319e-05, 2.4319e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0926e-03, 5.1605e-02, 1.4427e-02, 1.5656e-02, 3.9062e-02,
          3.9093e-02, 1.0910e-02, 9.1219e-04, 7.3051e-04, 2.6016e-03, 7.3662e-03,
          2.1458e-03, 9.0265e-04, 7.2241e-04, 9.4414e-04, 9.5606e-04, 1.2341e-03,
          9.1839e-04, 6.9809e-04, 2.8324e-04, 3.4928e-04, 4.4942e-04, 1.4439e-03,
          1.0635e-02, 2.1362e-03, 8.6746e-03, 3.9558e-03, 4.6635e-04, 1.3981e-03,
          1.0338e-03, 1.6575e-03, 5.8985e-04, 1.0777e-03, 1.2789e-03, 5.8031e-04,
          1.3504e-03, 9.6369e-04, 1.6918e-03, 8.9693e-04, 4.0698e-04, 3.5744e-03,
          1.9484e-03, 1.4601e-03, 1.6537e-03, 7.9346e-04, 1.7347e-03, 1.5726e-03,
          4.9829e-04, 4.3821e-04, 3.5591e-03, 9.4843e-04, 1.2856e-03, 6.2895e-04,
          5.1451e-04, 8.0347e-04, 4.2486e-04, 9.9850e-04, 2.1400e-03, 2.3632e-03,
          1.5516e-03, 8.8310e-04, 1.3933e-03, 1.2732e-03, 4.5433e-03, 3.7594e-03,
          1.8738e-02, 6.6833e-03, 1.7653e-03, 8.6403e-04, 8.0538e-04, 1.2264e-03,
          7.5436e-04, 4.7951e-03, 6.2180e-04, 1.6184e-03, 5.4836e-04, 5.0879e-04,
          7.3528e-04, 1.0529e-03, 5.6915e-03, 2.4414e-03, 9.2621e-03, 1.0796e-03,
          6.8903e-04, 3.0918e-03, 1.0176e-03, 5.4436e-03, 8.0032e-03, 7.4959e-03,
          1.3695e-02, 4.4975e-03, 1.4524e-03, 6.3324e-03, 6.4754e-04, 6.5041e-04,
          5.2595e-04, 3.3450e-04, 9.9468e-04, 1.1768e-03, 1.3723e-03, 1.5755e-03,
          7.7667e-03, 5.9319e-03, 6.3438e-03, 2.1725e-03, 1.7433e-03, 9.1314e-04,
          2.4223e-03, 3.2139e-03, 8.1177e-03, 4.3831e-03, 1.3298e-02, 6.9962e-03,
          1.7586e-03, 1.6441e-03, 6.7616e-04, 2.9144e-03, 8.9788e-04, 8.4543e-04,
          1.2417e-03, 2.0809e-03, 1.0920e-03, 6.6423e-04, 2.3308e-03, 1.8854e-03,
          6.5660e-04, 4.3106e-03, 6.7368e-03, 1.7090e-03, 1.2608e-03, 2.7981e-03,
          1.8036e-02, 1.1971e-02, 7.8058e-04, 3.9368e-03, 7.6447e-03, 1.8091e-03,
          2.2640e-03, 1.5345e-03, 1.2836e-03, 2.4471e-03, 1.5459e-03, 1.3475e-03,
          1.8482e-03, 9.1553e-04, 1.1864e-03, 1.4868e-03, 7.0801e-03, 2.1763e-03,
          4.5128e-03, 2.0504e-03, 1.3609e-03, 2.0428e-03, 1.9274e-03, 5.5809e-03,
          5.8937e-03, 4.0588e-03, 2.4872e-03, 6.6414e-03, 4.0741e-03, 1.7900e-03,
          2.1362e-03, 2.6112e-03, 2.1286e-03, 1.4486e-03, 1.2064e-03, 1.9226e-03,
          6.4468e-03, 1.9741e-03, 1.7023e-03, 8.6403e-04, 1.4267e-03, 1.5116e-03,
          1.8978e-03, 1.5583e-03, 1.2941e-03, 4.6005e-03, 5.6648e-03, 2.8172e-03,
          2.1858e-03, 1.9302e-03, 3.7613e-03, 2.3785e-03, 1.3676e-03, 9.1219e-04,
          4.6806e-03, 1.6556e-03, 1.7004e-03, 1.6327e-03, 5.9204e-03, 1.7681e-03,
          1.7328e-03, 7.8869e-04, 1.1692e-03, 2.7752e-03, 4.0359e-03, 1.5554e-03,
          1.4210e-03, 1.3752e-03, 5.0507e-03, 1.3771e-02, 7.4463e-03, 3.0708e-03,
          1.3428e-02, 1.3298e-02, 3.0651e-03, 1.2016e-03, 1.0826e-02, 1.3132e-03,
          1.7357e-03, 1.5907e-03, 1.7662e-03, 8.3847e-03, 2.3022e-03, 2.9030e-03,
          1.5163e-03, 1.5850e-03, 3.3550e-03, 7.5912e-04, 1.7118e-03, 1.9455e-03,
          3.5400e-03, 6.5422e-03, 1.1925e-02, 1.2138e-02, 7.5493e-03, 2.5387e-03,
          2.4902e-02, 8.7051e-03, 2.9736e-03, 3.9253e-03, 2.8114e-03, 1.8559e-03,
          2.0599e-03, 3.2997e-03, 7.9193e-03, 1.4257e-03, 2.4376e-03, 6.3629e-03,
          2.9678e-03, 1.8177e-03, 1.5154e-03, 1.3285e-03, 2.8229e-03, 4.0092e-03,
          6.4354e-03, 4.6875e-02, 2.4915e-05, 8.8573e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 9.4771e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 1.5843e-04, 1.5843e-04, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 9.4771e-05, 8.8573e-05, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 1.5843e-04, 8.8573e-05,
          9.4771e-05, 2.4915e-05, 1.5843e-04, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 1.5843e-04, 9.4771e-05, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          8.8573e-05, 3.1662e-04, 8.8573e-05, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          1.5843e-04, 2.4915e-05, 8.8573e-05, 9.4771e-05, 8.8573e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 3.1662e-04, 1.5843e-04, 2.4915e-05, 2.4915e-05,
          1.5843e-04, 1.5843e-04, 3.1662e-04, 2.4915e-05, 2.4915e-05, 1.5843e-04,
          2.4915e-05, 6.5947e-04, 3.8719e-04, 9.4771e-05, 2.4915e-05, 8.8573e-05,
          8.8573e-05, 2.4915e-05, 1.5843e-04, 3.1662e-04, 2.4915e-05, 1.5843e-04,
          2.4915e-05, 2.4915e-05, 3.8648e-04, 2.4915e-05, 5.9557e-04, 8.8573e-05,
          5.8937e-04, 2.4915e-05, 2.4915e-05, 2.4915e-05, 9.4771e-05, 9.4771e-05,
          9.4771e-05, 8.8573e-05, 2.4915e-05, 2.4915e-05, 1.5843e-04, 2.4915e-05,
          6.5947e-04, 1.5843e-04, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          9.4771e-05, 3.1662e-04, 2.4915e-05, 8.8573e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 1.5843e-04, 8.8573e-05,
          2.3556e-03, 1.5843e-04, 2.4915e-05, 8.8573e-05, 8.8573e-05, 7.6485e-04,
          1.5843e-04, 8.8573e-05, 3.8648e-04, 2.4915e-05, 2.4915e-05, 8.8573e-05,
          1.2655e-03, 2.3556e-03, 2.4915e-05, 8.8573e-05, 5.8937e-04, 2.4915e-05,
          8.8573e-05, 8.8573e-05, 2.4915e-05, 1.8191e-04, 2.9159e-04, 8.8573e-05,
          2.4915e-05, 1.5843e-04, 2.4915e-05, 4.9162e-04, 8.8573e-05, 2.9159e-04,
          2.4915e-05, 8.8573e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6082e-03, 5.1697e-02, 1.9836e-02, 1.2466e-02, 6.8970e-02,
          4.1321e-02, 1.3870e-02, 8.3733e-04, 7.7438e-04, 1.1721e-03, 4.3945e-03,
          1.1425e-03, 7.2145e-04, 8.0442e-04, 1.0443e-03, 8.2922e-04, 8.6832e-04,
          6.0320e-04, 5.6171e-04, 8.2254e-04, 7.1335e-04, 1.0967e-03, 7.4863e-04,
          5.3635e-03, 2.7332e-03, 1.0086e-02, 1.7443e-03, 1.0681e-03, 7.8440e-04,
          8.8596e-04, 1.1368e-03, 4.0317e-04, 9.6512e-04, 5.5313e-04, 6.7425e-04,
          1.6594e-03, 2.5616e-03, 9.7418e-04, 7.4816e-04, 8.1110e-04, 2.4414e-03,
          2.9888e-03, 1.9236e-03, 2.5368e-03, 9.1314e-04, 8.1444e-04, 1.4477e-03,
          5.3930e-04, 9.1839e-04, 8.7051e-03, 1.1253e-03, 8.9455e-04, 6.3467e-04,
          8.4019e-04, 7.0524e-04, 1.2426e-03, 1.5087e-03, 4.3373e-03, 6.5498e-03,
          1.1032e-02, 1.1206e-03, 3.3684e-03, 1.7729e-03, 3.5725e-03, 3.0594e-03,
          3.2673e-03, 1.1969e-03, 8.6975e-04, 1.3275e-03, 1.5249e-03, 1.6298e-03,
          1.4706e-03, 2.0542e-03, 1.2636e-03, 1.1978e-03, 8.7404e-04, 5.0163e-04,
          1.1539e-03, 1.0052e-03, 1.4610e-03, 1.2302e-03, 1.9236e-03, 6.1569e-03,
          3.1338e-03, 3.0060e-03, 2.3956e-03, 2.1095e-03, 1.2045e-03, 1.8787e-03,
          3.3855e-03, 1.6947e-03, 2.5845e-03, 1.9197e-03, 2.0981e-03, 1.7939e-03,
          8.5068e-04, 7.6771e-04, 6.9761e-04, 1.0872e-03, 1.5144e-03, 1.0214e-03,
          2.4567e-03, 8.5735e-04, 7.4959e-04, 2.4261e-03, 1.6146e-03, 1.5316e-03,
          2.5005e-03, 5.5885e-03, 2.7924e-03, 3.8509e-03, 2.5673e-03, 1.5059e-03,
          2.0618e-03, 2.6283e-03, 1.3943e-03, 2.6016e-03, 2.4242e-03, 1.5249e-03,
          1.5535e-03, 1.3924e-03, 1.1158e-03, 6.1750e-04, 1.8778e-03, 6.7711e-04,
          1.0271e-03, 1.3933e-03, 1.6565e-03, 1.6508e-03, 8.9407e-04, 2.7752e-03,
          3.6011e-03, 1.4973e-03, 6.9332e-04, 1.1482e-03, 2.1763e-03, 3.8185e-03,
          1.4200e-03, 1.6680e-03, 1.2140e-03, 1.9531e-03, 1.0471e-03, 1.3733e-03,
          1.5612e-03, 1.7939e-03, 1.0786e-03, 9.9754e-04, 1.6670e-03, 2.3022e-03,
          2.0313e-03, 1.9407e-03, 2.5444e-03, 2.5940e-03, 4.6272e-03, 7.4196e-03,
          1.8682e-03, 2.2697e-03, 1.1005e-03, 4.5853e-03, 4.5052e-03, 1.7862e-03,
          1.9054e-03, 1.1988e-03, 1.0986e-03, 1.2884e-03, 1.5917e-03, 9.3031e-04,
          1.8358e-03, 7.7248e-04, 1.3876e-03, 9.0075e-04, 9.5129e-04, 1.4954e-03,
          4.5776e-03, 5.9204e-03, 4.3640e-03, 1.1314e-02, 4.7379e-03, 1.4801e-03,
          1.5831e-03, 2.0771e-03, 2.8248e-03, 1.6642e-03, 1.1606e-03, 1.1292e-03,
          1.7471e-03, 1.8129e-03, 1.6804e-03, 2.1248e-03, 1.7586e-03, 1.4248e-03,
          9.2173e-04, 1.2407e-03, 9.7132e-04, 1.5106e-03, 1.1415e-03, 1.2455e-03,
          5.7259e-03, 3.5496e-03, 6.3057e-03, 4.9438e-03, 1.7529e-03, 2.0695e-03,
          1.0727e-02, 6.8016e-03, 1.2047e-02, 9.3126e-04, 3.0499e-03, 8.6832e-04,
          1.5001e-03, 2.3479e-03, 4.1885e-03, 1.1873e-03, 1.9293e-03, 2.1038e-03,
          1.3409e-03, 1.4668e-03, 1.2350e-03, 8.2588e-04, 1.4992e-03, 1.9035e-03,
          3.1414e-03, 4.2084e-02, 1.4809e-02, 2.7428e-03, 6.3400e-03, 1.6870e-03,
          4.4861e-03, 9.2621e-03, 1.2932e-02, 1.0086e-02, 4.1428e-03, 2.3251e-03,
          4.9477e-03, 1.5726e-03, 3.3169e-03, 2.0580e-03, 3.7441e-03, 1.4130e-02,
          7.9346e-03, 5.9967e-03, 3.0308e-03, 2.1210e-03, 3.4199e-03, 7.1602e-03,
          1.8829e-02, 1.1414e-02, 1.1035e-01, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 1.5616e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4869e-05,
          2.1112e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.3307e-04, 3.3307e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.3307e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 6.4087e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.3307e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 1.5628e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.3307e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 2.1112e-04, 3.4034e-05, 3.4034e-05, 3.3307e-04, 1.5616e-04,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.3307e-04, 3.3307e-04, 3.4034e-05, 3.4034e-05, 2.1112e-04, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 2.1112e-04, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 3.4034e-05, 3.4034e-05, 1.5616e-04, 1.5616e-04, 3.4034e-05,
          2.1112e-04, 3.4034e-05, 2.1112e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          2.1112e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          3.4034e-05, 2.1112e-04, 3.4034e-05, 3.4034e-05, 3.4034e-05, 3.4034e-05,
          2.1112e-04, 2.1112e-04, 2.1112e-04, 5.2738e-04, 2.1112e-04, 3.4034e-05,
          3.4034e-05, 2.1112e-04, 3.4034e-05, 4.4060e-04, 3.4034e-05, 2.1112e-04,
          3.4034e-05, 2.1112e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.3602e-03, 5.2002e-02, 1.6800e-02, 1.3145e-02, 5.0751e-02,
          4.2938e-02, 1.1292e-02, 1.3132e-03, 9.4032e-04, 1.4753e-03, 5.2071e-03,
          1.5497e-03, 8.5163e-04, 5.8126e-04, 1.2245e-03, 1.3103e-03, 9.0885e-04,
          6.0225e-04, 8.6308e-04, 3.7622e-04, 4.6730e-04, 5.9271e-04, 1.0252e-03,
          6.9656e-03, 3.6964e-03, 8.7280e-03, 2.1572e-03, 5.4502e-04, 9.0981e-04,
          6.6042e-04, 1.8320e-03, 3.8862e-04, 1.1444e-03, 7.7820e-04, 9.1171e-04,
          1.7223e-03, 2.0809e-03, 1.0347e-03, 8.7690e-04, 4.5371e-04, 2.2163e-03,
          1.8959e-03, 1.3666e-03, 1.5087e-03, 7.3004e-04, 1.0223e-03, 1.6060e-03,
          4.2582e-04, 5.6553e-04, 5.5847e-03, 1.1797e-03, 1.2283e-03, 7.4005e-04,
          6.6853e-04, 8.2874e-04, 7.5579e-04, 1.3866e-03, 5.6725e-03, 8.4915e-03,
          1.4404e-02, 8.5068e-04, 2.6512e-03, 1.7099e-03, 2.8381e-03, 3.5210e-03,
          3.9177e-03, 1.7338e-03, 1.0109e-03, 8.1062e-04, 8.3733e-04, 1.1530e-03,
          9.8801e-04, 2.6741e-03, 8.9264e-04, 1.4687e-03, 7.2432e-04, 4.9019e-04,
          9.1267e-04, 8.3113e-04, 1.4744e-03, 1.8034e-03, 2.4357e-03, 9.2010e-03,
          3.1452e-03, 3.6736e-03, 1.6699e-03, 2.8477e-03, 1.7719e-03, 1.9760e-03,
          5.0049e-03, 2.4376e-03, 1.8625e-03, 2.8381e-03, 1.5669e-03, 1.1301e-03,
          6.1607e-04, 5.2929e-04, 1.0691e-03, 1.3123e-03, 1.0672e-03, 1.0214e-03,
          3.8128e-03, 1.5888e-03, 1.1415e-03, 3.3512e-03, 2.3651e-03, 1.1721e-03,
          3.1261e-03, 2.6894e-03, 3.7632e-03, 3.0022e-03, 3.3932e-03, 1.7967e-03,
          3.7575e-03, 1.9512e-03, 9.9373e-04, 3.2368e-03, 1.6651e-03, 1.2569e-03,
          1.2522e-03, 1.7920e-03, 8.2159e-04, 8.6641e-04, 2.5063e-03, 6.9332e-04,
          7.1335e-04, 1.4544e-03, 1.8139e-03, 2.0847e-03, 1.2655e-03, 3.3779e-03,
          5.2071e-03, 1.9245e-03, 9.6321e-04, 1.3933e-03, 3.1033e-03, 1.7424e-03,
          1.6861e-03, 1.5450e-03, 1.8167e-03, 2.5768e-03, 1.4868e-03, 1.2922e-03,
          1.2913e-03, 3.3245e-03, 1.1806e-03, 1.0509e-03, 1.8911e-03, 3.4752e-03,
          3.3150e-03, 2.4319e-03, 2.0027e-03, 2.5520e-03, 3.7785e-03, 4.1084e-03,
          2.5387e-03, 2.1057e-03, 1.6718e-03, 3.0518e-03, 5.0774e-03, 1.8969e-03,
          2.4147e-03, 1.6613e-03, 1.2999e-03, 1.7824e-03, 1.3466e-03, 1.1168e-03,
          2.8019e-03, 1.0300e-03, 2.4376e-03, 1.1663e-03, 9.9659e-04, 1.9627e-03,
          6.2599e-03, 5.4169e-03, 2.7008e-03, 5.1918e-03, 4.8714e-03, 2.0733e-03,
          2.1152e-03, 2.5196e-03, 3.7060e-03, 2.5101e-03, 1.7023e-03, 1.3628e-03,
          3.1147e-03, 2.1019e-03, 2.5291e-03, 1.6584e-03, 2.0199e-03, 1.7672e-03,
          1.4200e-03, 8.2922e-04, 1.2321e-03, 1.7681e-03, 1.9178e-03, 1.5583e-03,
          4.2305e-03, 2.2087e-03, 7.0267e-03, 5.0468e-03, 3.0727e-03, 2.9526e-03,
          7.3280e-03, 6.1226e-03, 4.4594e-03, 1.5678e-03, 5.4855e-03, 1.0481e-03,
          1.9608e-03, 2.0123e-03, 2.5501e-03, 2.0752e-03, 2.6073e-03, 2.0275e-03,
          1.5707e-03, 1.5879e-03, 2.0752e-03, 1.2817e-03, 1.9779e-03, 1.9417e-03,
          3.5648e-03, 1.6083e-02, 1.1292e-02, 4.4403e-03, 8.1329e-03, 2.4242e-03,
          5.0850e-03, 9.5139e-03, 5.7755e-03, 1.5526e-02, 5.4054e-03, 2.8477e-03,
          3.1891e-03, 1.9531e-03, 3.6354e-03, 4.6272e-03, 4.1237e-03, 2.1393e-02,
          6.7177e-03, 4.6120e-03, 3.1223e-03, 2.0294e-03, 3.6392e-03, 4.4174e-03,
          9.2316e-03, 1.5221e-02, 3.7720e-02, 1.0333e-01, 2.4378e-05, 2.4378e-05,
          1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3518e-04,
          2.4378e-05, 1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 7.5758e-05, 2.4378e-05, 2.4378e-05,
          1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3578e-04,
          1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 1.8656e-04, 1.8656e-04, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 1.3518e-04, 1.8656e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 4.6349e-04, 1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          1.3518e-04, 2.4378e-05, 1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 7.5758e-05, 1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3518e-04, 1.3518e-04,
          2.4378e-05, 2.4378e-05, 1.8656e-04, 2.4378e-05, 2.4378e-05, 1.3518e-04,
          1.3518e-04, 1.3518e-04, 2.4378e-05, 2.4378e-05, 1.8656e-04, 7.5758e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3518e-04, 2.4378e-05,
          1.8656e-04, 1.8656e-04, 2.4378e-05, 2.4378e-05, 1.3518e-04, 2.4378e-05,
          2.4378e-05, 1.3518e-04, 2.4378e-05, 1.3518e-04, 2.4378e-05, 2.4378e-05,
          1.3518e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 7.5758e-05, 2.4378e-05,
          1.3518e-04, 1.3518e-04, 1.3518e-04, 2.4378e-05, 2.4378e-05, 1.3518e-04,
          1.3518e-04, 1.3518e-04, 1.8656e-04, 2.4378e-05, 2.4378e-05, 1.3518e-04,
          1.3518e-04, 1.3518e-04, 2.4378e-05, 1.3518e-04, 2.4378e-05, 2.4378e-05,
          4.6349e-04, 1.3518e-04, 4.6349e-04, 1.0133e-04, 1.3518e-04, 1.3518e-04,
          2.4378e-05, 1.3518e-04, 1.3518e-04, 4.5300e-04, 1.3518e-04, 4.6349e-04,
          1.3518e-04, 1.3518e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.1308e-03, 4.8523e-02, 1.9119e-02, 1.3206e-02, 6.5979e-02,
          4.0833e-02, 1.3229e-02, 9.8515e-04, 7.6199e-04, 1.4486e-03, 4.3335e-03,
          1.3332e-03, 7.2193e-04, 6.8378e-04, 1.4935e-03, 1.0614e-03, 8.0395e-04,
          6.1512e-04, 6.7806e-04, 6.1321e-04, 6.3229e-04, 6.3753e-04, 8.9693e-04,
          6.2332e-03, 3.7022e-03, 9.6664e-03, 2.4052e-03, 8.4162e-04, 8.5115e-04,
          7.1859e-04, 1.3161e-03, 4.1914e-04, 1.1988e-03, 7.7581e-04, 6.9523e-04,
          1.7509e-03, 1.5049e-03, 1.1406e-03, 9.2745e-04, 5.9366e-04, 2.5806e-03,
          2.5349e-03, 1.6613e-03, 2.1000e-03, 8.6451e-04, 8.5545e-04, 1.7843e-03,
          4.9877e-04, 7.8154e-04, 9.6664e-03, 1.0653e-03, 9.0313e-04, 6.6948e-04,
          6.8808e-04, 7.0524e-04, 9.9564e-04, 1.1415e-03, 2.6703e-03, 3.0613e-03,
          3.2482e-03, 1.0061e-03, 2.4853e-03, 1.7242e-03, 3.1643e-03, 3.7212e-03,
          3.0727e-03, 1.3456e-03, 9.6369e-04, 1.1206e-03, 1.3285e-03, 1.3514e-03,
          1.2150e-03, 2.4147e-03, 1.0109e-03, 1.1292e-03, 7.2861e-04, 4.1962e-04,
          9.0551e-04, 9.1791e-04, 1.4086e-03, 1.5383e-03, 1.9064e-03, 2.4185e-03,
          2.0046e-03, 2.6913e-03, 1.8177e-03, 2.0809e-03, 1.0939e-03, 1.6775e-03,
          4.3449e-03, 1.9093e-03, 1.5984e-03, 2.5349e-03, 1.6460e-03, 1.4830e-03,
          8.1062e-04, 6.2227e-04, 7.2050e-04, 1.0767e-03, 1.1301e-03, 8.8739e-04,
          2.7542e-03, 7.7581e-04, 8.4114e-04, 2.0866e-03, 1.7834e-03, 1.4362e-03,
          2.3079e-03, 4.8523e-03, 3.0727e-03, 2.9316e-03, 2.3804e-03, 1.4610e-03,
          2.2793e-03, 2.4128e-03, 1.2426e-03, 3.2997e-03, 2.0733e-03, 1.6747e-03,
          1.4553e-03, 1.4744e-03, 9.0551e-04, 5.9748e-04, 2.1858e-03, 6.3038e-04,
          8.2111e-04, 1.1578e-03, 1.5554e-03, 2.0523e-03, 1.0586e-03, 2.5692e-03,
          3.3035e-03, 1.5278e-03, 1.6356e-03, 1.1225e-03, 2.6302e-03, 2.6512e-03,
          1.6899e-03, 1.5583e-03, 1.3161e-03, 2.1801e-03, 1.3399e-03, 1.3189e-03,
          1.3313e-03, 1.3657e-03, 1.0262e-03, 8.6546e-04, 1.4534e-03, 2.4948e-03,
          2.3193e-03, 2.2240e-03, 1.8339e-03, 2.3098e-03, 3.6201e-03, 6.4507e-03,
          2.0046e-03, 2.0008e-03, 1.5697e-03, 3.4561e-03, 4.6120e-03, 1.8301e-03,
          2.1515e-03, 1.2741e-03, 1.2159e-03, 1.5354e-03, 1.4830e-03, 8.5258e-04,
          2.0695e-03, 7.8154e-04, 1.4658e-03, 1.1692e-03, 1.0071e-03, 1.5841e-03,
          4.3335e-03, 4.1618e-03, 3.1910e-03, 1.0193e-02, 4.5395e-03, 1.9722e-03,
          1.6069e-03, 3.0499e-03, 3.3760e-03, 1.9369e-03, 1.2579e-03, 1.1177e-03,
          2.2697e-03, 1.6232e-03, 1.8740e-03, 1.9493e-03, 1.4944e-03, 1.6432e-03,
          1.1349e-03, 1.0462e-03, 1.0834e-03, 1.5039e-03, 1.3590e-03, 1.4315e-03,
          3.7117e-03, 2.9545e-03, 6.3934e-03, 4.0588e-03, 1.5516e-03, 2.1820e-03,
          6.7482e-03, 6.5422e-03, 9.7885e-03, 1.1806e-03, 4.0665e-03, 9.0742e-04,
          1.4887e-03, 2.4948e-03, 4.6387e-03, 1.4439e-03, 2.0142e-03, 1.6880e-03,
          1.3590e-03, 1.5135e-03, 1.3323e-03, 1.0471e-03, 1.8034e-03, 1.8740e-03,
          2.6894e-03, 2.5513e-02, 1.1246e-02, 2.3136e-03, 8.0795e-03, 1.9951e-03,
          3.7365e-03, 5.5237e-03, 9.5749e-03, 9.0027e-03, 5.0774e-03, 2.1687e-03,
          6.3362e-03, 1.4915e-03, 2.7084e-03, 2.4071e-03, 3.3836e-03, 1.0544e-02,
          6.1569e-03, 4.9744e-03, 2.4605e-03, 1.8854e-03, 3.2806e-03, 5.4779e-03,
          1.5076e-02, 8.8806e-03, 2.7588e-02, 1.3954e-02, 1.3928e-01, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 1.3244e-04, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 1.3244e-04, 2.5010e-04, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 1.3244e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          1.3244e-04, 4.2021e-05, 1.6260e-03, 8.5783e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 2.5010e-04, 1.3244e-04, 2.5010e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 1.5974e-04, 1.5974e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          2.5010e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 1.5974e-04, 2.5010e-04, 2.5010e-04, 4.2021e-05, 4.2021e-05,
          5.5981e-04, 2.5010e-04, 1.5974e-04, 4.2021e-05, 4.2021e-05, 2.5034e-04,
          4.2021e-05, 2.5010e-04, 2.5034e-04, 1.3244e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.5010e-04,
          4.2021e-05, 4.2021e-05, 2.5010e-04, 4.2021e-05, 4.2021e-05, 1.5974e-04,
          4.2021e-05, 2.5010e-04, 4.2021e-05, 4.2021e-05, 1.3244e-04, 1.3244e-04,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.5034e-04, 4.2021e-05,
          1.3244e-04, 1.3244e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 2.5010e-04, 4.2021e-05, 1.5974e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 1.3244e-04, 4.2021e-05,
          1.5974e-04, 2.5010e-04, 1.5974e-04, 1.5974e-04, 4.2021e-05, 4.6921e-04,
          1.5974e-04, 4.2021e-05, 2.5010e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          2.5120e-03, 2.5120e-03, 4.2021e-05, 1.5974e-04, 4.2021e-05, 4.2021e-05,
          1.5974e-04, 4.5824e-04, 4.5824e-04, 3.2353e-04, 4.5824e-04, 1.5974e-04,
          4.2021e-05, 1.5974e-04, 4.2021e-05, 3.7527e-04, 4.2021e-05, 1.5974e-04,
          4.2021e-05, 4.2021e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.8853e-03, 4.9805e-02, 1.2184e-02, 1.3039e-02, 3.8025e-02,
          3.7445e-02, 9.6970e-03, 8.3065e-04, 7.5054e-04, 2.5806e-03, 3.7003e-03,
          1.4191e-03, 1.1349e-03, 4.7779e-04, 3.3379e-03, 1.0004e-03, 1.3456e-03,
          8.6069e-04, 6.4421e-04, 2.7728e-04, 4.3344e-04, 3.7789e-04, 1.5888e-03,
          5.3101e-03, 4.4670e-03, 7.6065e-03, 4.2267e-03, 4.0340e-04, 1.5259e-03,
          6.7520e-04, 1.5650e-03, 4.4298e-04, 4.2076e-03, 2.4338e-03, 1.0614e-03,
          5.4054e-03, 9.1839e-04, 4.2763e-03, 3.5000e-03, 3.4857e-04, 5.1918e-03,
          1.5812e-03, 1.2989e-03, 1.1520e-03, 9.6035e-04, 1.5869e-03, 6.6261e-03,
          3.6073e-04, 4.5538e-04, 3.3932e-03, 1.2541e-03, 9.7942e-04, 9.6512e-04,
          3.4070e-04, 6.2561e-04, 4.8637e-04, 1.2455e-03, 1.3485e-03, 2.0561e-03,
          1.3943e-03, 8.6498e-04, 1.3828e-03, 2.6073e-03, 2.6321e-03, 1.6937e-02,
          1.7004e-03, 1.2121e-03, 2.3155e-03, 7.1955e-04, 6.9332e-04, 6.3944e-04,
          6.6042e-04, 5.4588e-03, 5.1212e-04, 1.2121e-03, 3.0375e-04, 3.3331e-04,
          6.3896e-04, 9.4843e-04, 9.9659e-04, 4.5891e-03, 1.2770e-03, 1.9293e-03,
          9.9182e-04, 3.5591e-03, 1.0014e-03, 2.5215e-03, 1.0805e-03, 1.0681e-03,
          5.2605e-03, 4.4250e-03, 9.7847e-04, 3.1147e-03, 1.0462e-03, 7.9250e-04,
          4.5562e-04, 3.3426e-04, 1.3857e-03, 2.3937e-03, 9.8991e-04, 1.4448e-03,
          3.4370e-03, 5.3406e-04, 8.4162e-04, 1.1330e-02, 9.3536e-03, 1.2045e-03,
          2.8934e-03, 2.6112e-03, 7.0152e-03, 1.1940e-03, 1.2503e-03, 1.7948e-03,
          3.0098e-03, 2.9869e-03, 9.6273e-04, 1.2505e-02, 1.2093e-03, 7.7677e-04,
          1.0872e-03, 4.4746e-03, 5.0735e-04, 6.0368e-04, 1.1520e-02, 4.6921e-04,
          7.3242e-04, 6.8140e-04, 9.2268e-04, 8.7433e-03, 1.7548e-03, 2.9335e-03,
          1.8778e-03, 1.3399e-03, 1.1101e-03, 1.0500e-03, 4.2992e-03, 1.2779e-03,
          5.7945e-03, 1.6165e-03, 1.2302e-03, 2.5043e-03, 2.9202e-03, 1.2026e-03,
          8.6117e-04, 7.6246e-04, 1.3514e-03, 1.2922e-03, 1.0204e-03, 8.0414e-03,
          6.4125e-03, 9.9640e-03, 2.0485e-03, 3.5381e-03, 1.8482e-03, 8.7357e-03,
          5.0774e-03, 4.5815e-03, 1.0080e-03, 1.8873e-03, 1.1978e-02, 2.7428e-03,
          7.7744e-03, 5.2071e-03, 2.6398e-03, 3.4199e-03, 1.9197e-03, 1.5535e-03,
          2.4109e-03, 1.6775e-03, 4.7913e-03, 2.5787e-03, 1.9207e-03, 6.3820e-03,
          3.1624e-03, 1.7519e-03, 2.3632e-03, 4.6349e-03, 1.7044e-02, 3.2425e-03,
          4.8294e-03, 6.2637e-03, 1.8188e-02, 2.1820e-03, 1.3132e-03, 1.6060e-03,
          2.6608e-03, 1.2484e-03, 1.5745e-03, 1.3037e-03, 9.4414e-04, 6.2637e-03,
          1.4515e-03, 6.7711e-04, 4.0207e-03, 2.7542e-03, 2.5063e-03, 5.4550e-03,
          2.0409e-03, 1.6413e-03, 1.8036e-02, 4.9248e-03, 2.6741e-03, 8.4457e-03,
          3.3073e-03, 1.0414e-02, 3.8757e-03, 1.1606e-03, 6.4621e-03, 1.6327e-03,
          1.1387e-03, 2.3956e-03, 1.2255e-03, 1.1702e-03, 6.0310e-03, 9.0885e-04,
          3.9520e-03, 2.4719e-03, 2.9373e-03, 1.7471e-03, 5.8327e-03, 1.3008e-03,
          6.4468e-03, 4.6844e-03, 1.4420e-02, 2.7199e-03, 3.6377e-02, 6.7177e-03,
          3.6621e-03, 1.6756e-03, 2.7180e-03, 4.2000e-03, 7.3700e-03, 1.7977e-03,
          1.7366e-03, 1.3094e-03, 8.7452e-04, 2.7180e-03, 4.6158e-03, 5.3673e-03,
          1.7033e-03, 1.6041e-03, 1.8463e-03, 2.9774e-03, 3.6488e-03, 3.4733e-03,
          7.5760e-03, 6.3515e-03, 3.7537e-03, 4.2419e-03, 7.5684e-03, 4.1718e-02,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 3.0746e-03,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.0492e-04, 2.0492e-04, 2.2948e-05,
          1.1683e-04, 1.1683e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.1104e-04, 2.0492e-04, 2.2948e-05, 2.2948e-05,
          1.1683e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          3.9959e-04, 2.2948e-05, 2.0492e-04, 2.0492e-04, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 2.0492e-04, 1.1104e-04, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 1.1104e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          1.1104e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 1.1104e-04, 1.1104e-04, 2.0492e-04, 2.2948e-05, 2.2948e-05,
          1.1104e-04, 2.0492e-04, 1.1104e-04, 2.2948e-05, 2.2948e-05, 2.0540e-04,
          2.2948e-05, 8.9407e-04, 2.0540e-04, 1.1683e-04, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.1683e-04, 1.1104e-04, 2.2948e-05, 2.0492e-04,
          2.2948e-05, 2.2948e-05, 2.0492e-04, 2.0492e-04, 5.7507e-04, 1.1104e-04,
          7.9966e-04, 8.9359e-04, 2.2948e-05, 2.2948e-05, 2.0492e-04, 1.1683e-04,
          5.7507e-04, 1.1104e-04, 2.2948e-05, 2.2948e-05, 2.0540e-04, 2.2948e-05,
          2.0492e-04, 2.0492e-04, 5.7507e-04, 2.2948e-05, 1.1104e-04, 2.2948e-05,
          2.2948e-05, 2.0492e-04, 2.2948e-05, 1.1104e-04, 2.2948e-05, 2.2948e-05,
          1.1104e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05, 1.1683e-04, 2.2948e-05,
          2.0492e-04, 2.0492e-04, 2.0492e-04, 1.1104e-04, 2.2948e-05, 1.1104e-04,
          2.0492e-04, 1.1104e-04, 2.0492e-04, 2.2948e-05, 2.2948e-05, 2.2948e-05,
          7.9966e-04, 7.9966e-04, 2.2948e-05, 1.1104e-04, 2.2948e-05, 2.2948e-05,
          2.2948e-05, 2.2948e-05, 1.1104e-04, 1.2755e-04, 1.1104e-04, 1.1683e-04,
          2.2948e-05, 2.2948e-05, 2.2948e-05, 3.7003e-04, 2.2948e-05, 1.1104e-04,
          2.2948e-05, 2.2948e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.3449e-03, 4.6967e-02, 1.4427e-02, 1.5205e-02, 4.5563e-02,
          2.9358e-02, 1.3023e-02, 5.5838e-04, 4.5371e-04, 1.2932e-03, 3.8776e-03,
          1.0729e-03, 4.8709e-04, 8.5306e-04, 7.6485e-04, 6.0415e-04, 6.8426e-04,
          5.6458e-04, 5.8031e-04, 3.8457e-04, 4.2558e-04, 5.3358e-04, 8.1825e-04,
          5.6725e-03, 2.0847e-03, 1.0017e-02, 2.0866e-03, 5.5647e-04, 7.1335e-04,
          1.1730e-03, 9.3985e-04, 3.4404e-04, 7.3242e-04, 6.5851e-04, 3.9482e-04,
          9.2697e-04, 8.7166e-04, 1.1501e-03, 6.3801e-04, 5.5218e-04, 2.9430e-03,
          2.8820e-03, 2.0332e-03, 2.7618e-03, 8.9121e-04, 8.4400e-04, 1.3466e-03,
          6.6328e-04, 6.9952e-04, 4.5700e-03, 7.0524e-04, 7.4196e-04, 4.8637e-04,
          5.1212e-04, 5.5647e-04, 5.9891e-04, 7.7868e-04, 1.6003e-03, 1.5469e-03,
          1.4963e-03, 8.9741e-04, 1.3657e-03, 1.0014e-03, 4.9782e-03, 2.9812e-03,
          1.8417e-02, 3.4943e-03, 1.4687e-03, 1.9627e-03, 2.1896e-03, 2.5597e-03,
          1.3504e-03, 2.6836e-03, 6.5327e-04, 1.0405e-03, 5.5981e-04, 3.8099e-04,
          7.5054e-04, 9.5844e-04, 4.9934e-03, 1.2865e-03, 5.5199e-03, 8.7070e-04,
          8.3733e-04, 2.2049e-03, 1.1921e-03, 2.8267e-03, 5.6267e-03, 9.0942e-03,
          6.3820e-03, 3.1033e-03, 2.7447e-03, 3.6945e-03, 1.3304e-03, 1.3170e-03,
          7.0667e-04, 4.6587e-04, 5.6601e-04, 9.3937e-04, 1.7729e-03, 1.2188e-03,
          3.4599e-03, 2.7657e-03, 2.1629e-03, 1.3380e-03, 1.0910e-03, 1.0042e-03,
          2.0370e-03, 5.7983e-03, 5.2834e-03, 6.8741e-03, 1.3718e-02, 6.8932e-03,
          1.8692e-03, 3.5191e-03, 1.0490e-03, 2.6360e-03, 2.0561e-03, 1.2922e-03,
          1.4400e-03, 1.3990e-03, 1.6413e-03, 4.8113e-04, 1.5898e-03, 1.9531e-03,
          9.6560e-04, 7.2823e-03, 1.1848e-02, 1.2789e-03, 8.5497e-04, 2.3079e-03,
          1.3145e-02, 7.2556e-03, 6.3467e-04, 3.0022e-03, 5.7335e-03, 5.8250e-03,
          1.5869e-03, 1.9407e-03, 1.2455e-03, 2.3975e-03, 1.2941e-03, 1.4849e-03,
          2.6741e-03, 6.8617e-04, 1.3275e-03, 1.1883e-03, 7.2556e-03, 1.5850e-03,
          2.8038e-03, 1.4486e-03, 1.5593e-03, 1.8253e-03, 2.2888e-03, 8.0643e-03,
          4.3488e-03, 4.2229e-03, 1.6642e-03, 1.3924e-02, 4.3869e-03, 1.9684e-03,
          1.9989e-03, 2.0752e-03, 1.3046e-03, 1.2674e-03, 1.1959e-03, 1.6346e-03,
          2.9068e-03, 1.3475e-03, 1.3237e-03, 7.3528e-04, 1.0834e-03, 1.4563e-03,
          1.4658e-03, 1.8301e-03, 1.7443e-03, 8.3008e-03, 4.5319e-03, 1.7357e-03,
          2.0466e-03, 1.9093e-03, 3.0708e-03, 1.6241e-03, 1.0576e-03, 1.1272e-03,
          2.6703e-03, 1.8864e-03, 1.7424e-03, 2.3384e-03, 6.5804e-03, 1.4172e-03,
          9.5272e-04, 1.1978e-03, 1.0185e-03, 1.6384e-03, 2.0294e-03, 1.2960e-03,
          2.1229e-03, 1.9102e-03, 4.8447e-03, 1.1475e-02, 4.7531e-03, 2.6417e-03,
          1.7181e-02, 1.3710e-02, 1.4214e-02, 9.5463e-04, 5.9280e-03, 9.4318e-04,
          1.2779e-03, 1.8530e-03, 2.8439e-03, 3.2864e-03, 1.9550e-03, 3.9673e-03,
          1.2798e-03, 1.4114e-03, 2.4776e-03, 6.8617e-04, 1.4715e-03, 1.8253e-03,
          3.2310e-03, 1.0880e-02, 1.1681e-02, 8.4305e-03, 6.7444e-03, 2.0962e-03,
          2.0401e-02, 1.7395e-02, 1.2215e-02, 4.5586e-03, 2.7618e-03, 2.2430e-03,
          2.6741e-03, 2.5330e-03, 7.8583e-03, 1.4648e-03, 2.8191e-03, 5.5733e-03,
          3.9253e-03, 2.7370e-03, 1.5640e-03, 1.4524e-03, 2.5768e-03, 4.6425e-03,
          1.1330e-02, 2.0386e-02, 8.7357e-03, 5.4092e-03, 9.6664e-03, 7.5378e-03,
          4.6844e-02, 2.4915e-05, 2.4915e-05, 2.4915e-05, 8.4698e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 8.4698e-05, 8.4698e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 1.8454e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          8.4698e-05, 2.4915e-05, 4.9877e-04, 4.9877e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 8.4698e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 8.4698e-05, 8.4698e-05, 1.8454e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          1.8454e-04, 2.4915e-05, 2.4915e-05, 8.4698e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 1.2481e-04, 1.8454e-04, 2.4915e-05, 2.4915e-05,
          1.8454e-04, 8.4698e-05, 1.2481e-04, 2.4915e-05, 2.4915e-05, 1.8454e-04,
          2.4915e-05, 1.8454e-04, 1.8454e-04, 8.4698e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 8.4698e-05, 2.4915e-05, 2.4915e-05, 1.8454e-04,
          2.4915e-05, 2.4915e-05, 1.8454e-04, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 8.4698e-05, 5.8222e-04,
          8.4698e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 1.8454e-04, 2.4915e-05,
          5.8222e-04, 8.4698e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          8.4698e-05, 2.4915e-05, 2.4915e-05, 1.2481e-04, 2.4915e-05, 2.4915e-05,
          2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05, 5.8222e-04, 2.4915e-05,
          8.4698e-05, 1.8454e-04, 2.4915e-05, 1.2481e-04, 2.4915e-05, 1.2481e-04,
          8.4698e-05, 2.4915e-05, 1.8454e-04, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          1.2481e-04, 1.0424e-03, 2.4915e-05, 2.4915e-05, 2.4915e-05, 2.4915e-05,
          1.2481e-04, 1.2481e-04, 1.2481e-04, 3.8099e-04, 3.1209e-04, 1.2481e-04,
          2.4915e-05, 1.8454e-04, 2.4915e-05, 4.0841e-04, 1.2481e-04, 1.2481e-04,
          2.4915e-05, 2.4915e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6387e-03, 4.3518e-02, 1.8265e-02, 1.3504e-02, 6.7566e-02,
          2.8992e-02, 1.6342e-02, 5.4884e-04, 4.5347e-04, 9.0456e-04, 1.9150e-03,
          6.4993e-04, 6.1893e-04, 6.3038e-04, 9.8419e-04, 5.9032e-04, 8.4257e-04,
          5.7173e-04, 4.6802e-04, 1.1787e-03, 9.8705e-04, 1.1816e-03, 7.2622e-04,
          2.6588e-03, 2.9354e-03, 1.1383e-02, 1.6708e-03, 7.0763e-04, 7.1907e-04,
          5.7030e-04, 7.9155e-04, 3.9363e-04, 1.1024e-03, 5.9175e-04, 4.5609e-04,
          1.6088e-03, 1.5354e-03, 1.1845e-03, 7.8011e-04, 6.6137e-04, 3.7746e-03,
          4.5929e-03, 2.8782e-03, 3.9902e-03, 2.1381e-03, 8.6832e-04, 2.5826e-03,
          7.0620e-04, 1.5640e-03, 7.7438e-03, 6.5041e-04, 5.5790e-04, 4.7398e-04,
          5.0211e-04, 4.7088e-04, 9.5940e-04, 1.8177e-03, 1.3418e-03, 2.4300e-03,
          2.9697e-03, 1.0519e-03, 1.4515e-03, 1.3723e-03, 2.9049e-03, 4.3640e-03,
          2.1362e-03, 1.1320e-03, 1.2150e-03, 4.2839e-03, 4.6120e-03, 3.5324e-03,
          3.7479e-03, 1.8482e-03, 6.8760e-04, 7.1907e-04, 4.7636e-04, 3.1018e-04,
          8.2302e-04, 1.3323e-03, 1.0347e-03, 1.1206e-03, 1.2226e-03, 1.7481e-03,
          2.2335e-03, 1.9445e-03, 1.2360e-03, 1.2026e-03, 8.6021e-04, 1.4639e-03,
          2.5463e-03, 2.4033e-03, 2.3975e-03, 1.9464e-03, 9.0714e-03, 7.5760e-03,
          7.0572e-04, 5.4312e-04, 5.1641e-04, 1.3800e-03, 1.5373e-03, 1.1320e-03,
          1.3952e-03, 6.5374e-04, 7.1192e-04, 1.9236e-03, 1.7023e-03, 1.4467e-03,
          1.8740e-03, 6.8092e-03, 3.0403e-03, 2.3842e-03, 1.8091e-03, 1.5678e-03,
          2.1400e-03, 1.0872e-02, 3.1357e-03, 4.8904e-03, 1.2489e-02, 1.4687e-03,
          1.2503e-03, 1.8330e-03, 9.5129e-04, 3.8028e-04, 1.8311e-03, 5.1355e-04,
          1.4277e-03, 1.1425e-03, 1.4572e-03, 1.9331e-03, 7.6151e-04, 2.0714e-03,
          2.0065e-03, 1.2608e-03, 5.2118e-04, 9.3651e-04, 2.2831e-03, 7.4272e-03,
          2.1877e-03, 2.5349e-03, 1.4248e-03, 2.0847e-03, 1.0738e-03, 1.1101e-03,
          1.1539e-03, 7.5817e-04, 1.1320e-03, 9.8133e-04, 1.1005e-03, 2.8267e-03,
          2.3575e-03, 2.5024e-03, 5.5275e-03, 2.4261e-03, 2.7561e-03, 1.2909e-02,
          2.2430e-03, 3.4752e-03, 8.5211e-04, 4.0207e-03, 8.9645e-03, 3.2997e-03,
          3.3207e-03, 2.2373e-03, 1.4229e-03, 1.2760e-03, 2.2564e-03, 9.1076e-04,
          1.2178e-03, 7.6151e-04, 1.3075e-03, 9.5797e-04, 8.2111e-04, 2.2640e-03,
          4.1809e-03, 3.5515e-03, 7.8011e-03, 1.4671e-02, 5.6534e-03, 1.2875e-03,
          1.6432e-03, 1.8950e-03, 3.0365e-03, 1.2789e-03, 1.1263e-03, 1.6289e-03,
          1.5078e-03, 2.6054e-03, 1.2255e-03, 1.8778e-03, 1.0567e-03, 1.4696e-03,
          7.1144e-04, 1.2999e-03, 1.2541e-03, 1.5764e-03, 1.0481e-03, 1.6050e-03,
          5.4474e-03, 2.9869e-03, 5.7831e-03, 3.2997e-03, 1.4353e-03, 2.2507e-03,
          4.5433e-03, 7.3280e-03, 4.1046e-02, 8.1730e-04, 3.1357e-03, 1.1854e-03,
          1.2598e-03, 2.8248e-03, 3.1624e-03, 8.5163e-04, 2.0638e-03, 1.3857e-03,
          1.8482e-03, 1.3428e-03, 1.4381e-03, 7.6151e-04, 1.8730e-03, 1.6413e-03,
          3.9711e-03, 1.6174e-02, 8.5144e-03, 1.9627e-03, 6.7787e-03, 1.6527e-03,
          3.7079e-03, 6.2904e-03, 2.6596e-02, 8.1711e-03, 5.7144e-03, 3.0613e-03,
          4.2953e-03, 1.1749e-03, 2.1515e-03, 1.6556e-03, 4.4594e-03, 5.6534e-03,
          4.6158e-03, 4.7112e-03, 1.8244e-03, 2.5787e-03, 3.0899e-03, 5.0888e-03,
          2.1103e-02, 4.6272e-03, 9.3384e-03, 4.3640e-03, 1.1803e-02, 6.9504e-03,
          1.0559e-02, 7.7942e-02, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 1.5986e-04, 2.6131e-04, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 2.6131e-04, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 2.6131e-04, 2.6131e-04, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 2.6131e-04, 1.3995e-04, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          1.3995e-04, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 1.3995e-04, 1.3995e-04, 1.3995e-04, 3.8564e-05, 3.8564e-05,
          1.3995e-04, 3.8564e-05, 1.3995e-04, 3.8564e-05, 3.8564e-05, 1.3995e-04,
          3.8564e-05, 3.8564e-05, 1.3995e-04, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 1.5986e-04, 3.8564e-05, 3.8564e-05, 2.6131e-04,
          3.8564e-05, 3.8564e-05, 2.6131e-04, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 1.3995e-04, 3.8564e-05, 3.8564e-05, 2.6131e-04, 1.5986e-04,
          3.8564e-05, 1.3995e-04, 3.8564e-05, 3.8564e-05, 1.3995e-04, 3.8564e-05,
          2.6131e-04, 2.6131e-04, 3.8564e-05, 3.8564e-05, 2.6131e-04, 3.8564e-05,
          1.5986e-04, 1.3995e-04, 3.8564e-05, 1.3995e-04, 3.8564e-05, 3.8564e-05,
          3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05, 2.6131e-04, 3.8564e-05,
          1.3995e-04, 2.6131e-04, 1.3995e-04, 3.8564e-05, 3.8564e-05, 1.3995e-04,
          1.3995e-04, 3.8564e-05, 2.6131e-04, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          1.3995e-04, 1.3995e-04, 3.8564e-05, 3.8564e-05, 3.8564e-05, 3.8564e-05,
          1.3995e-04, 1.3995e-04, 1.3995e-04, 3.8147e-04, 1.3995e-04, 3.8564e-05,
          3.8564e-05, 1.3995e-04, 1.3995e-04, 3.8147e-04, 3.1066e-04, 1.3995e-04,
          3.8564e-05, 3.8564e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.1237e-03, 4.5013e-02, 1.9974e-02, 1.5572e-02, 7.4890e-02,
          3.1311e-02, 1.9653e-02, 4.9162e-04, 4.9782e-04, 8.5783e-04, 1.9226e-03,
          6.1893e-04, 5.4026e-04, 6.4230e-04, 9.2793e-04, 5.1498e-04, 7.8487e-04,
          5.4026e-04, 4.4012e-04, 1.3161e-03, 1.0433e-03, 1.2407e-03, 8.1873e-04,
          2.4204e-03, 2.4166e-03, 1.3680e-02, 2.0180e-03, 8.5020e-04, 6.8378e-04,
          6.2943e-04, 6.5374e-04, 3.6502e-04, 9.9182e-04, 5.2500e-04, 4.6110e-04,
          1.3771e-03, 1.8215e-03, 1.0862e-03, 7.6866e-04, 6.5804e-04, 2.9087e-03,
          3.8242e-03, 2.4242e-03, 3.7308e-03, 1.6918e-03, 7.6580e-04, 1.9798e-03,
          7.1907e-04, 1.6851e-03, 9.0103e-03, 7.1144e-04, 4.6968e-04, 5.1880e-04,
          5.5695e-04, 4.5562e-04, 1.2970e-03, 1.5450e-03, 1.6174e-03, 2.2488e-03,
          3.5572e-03, 9.8610e-04, 1.8959e-03, 1.2712e-03, 2.2697e-03, 3.1815e-03,
          1.8444e-03, 7.9107e-04, 9.6846e-04, 3.0403e-03, 3.9635e-03, 3.4122e-03,
          4.1008e-03, 1.6346e-03, 8.8882e-04, 6.3276e-04, 5.7650e-04, 3.4046e-04,
          8.2111e-04, 1.3170e-03, 1.0004e-03, 9.3269e-04, 1.1015e-03, 1.6165e-03,
          2.5864e-03, 1.6489e-03, 1.4687e-03, 8.9836e-04, 7.7152e-04, 1.3332e-03,
          1.8616e-03, 1.6203e-03, 1.9665e-03, 1.4296e-03, 6.5613e-03, 7.4387e-03,
          8.0061e-04, 6.7472e-04, 4.8399e-04, 1.1616e-03, 1.6527e-03, 1.0147e-03,
          1.1797e-03, 4.8518e-04, 5.6505e-04, 1.5621e-03, 1.2836e-03, 1.3571e-03,
          1.6813e-03, 5.2376e-03, 2.3346e-03, 1.9417e-03, 1.5841e-03, 1.2074e-03,
          1.5831e-03, 6.2370e-03, 3.3283e-03, 3.4313e-03, 1.1856e-02, 1.3866e-03,
          1.3409e-03, 1.6842e-03, 9.5034e-04, 3.4881e-04, 1.4439e-03, 4.2534e-04,
          1.2388e-03, 1.0376e-03, 1.3046e-03, 1.5087e-03, 7.3051e-04, 1.8177e-03,
          1.6842e-03, 1.0071e-03, 4.3559e-04, 8.3065e-04, 1.6985e-03, 4.4937e-03,
          1.8988e-03, 2.6264e-03, 1.2245e-03, 2.1763e-03, 1.0672e-03, 1.1206e-03,
          1.0719e-03, 5.1832e-04, 8.8310e-04, 8.0109e-04, 1.0300e-03, 1.9417e-03,
          2.0199e-03, 1.8091e-03, 3.5477e-03, 2.1992e-03, 3.0346e-03, 9.1705e-03,
          1.5917e-03, 2.5043e-03, 6.1703e-04, 3.4790e-03, 6.2790e-03, 2.8820e-03,
          2.5425e-03, 1.7462e-03, 1.2484e-03, 1.1730e-03, 2.4986e-03, 9.3508e-04,
          1.0195e-03, 7.0095e-04, 1.1063e-03, 8.0538e-04, 7.5960e-04, 1.9169e-03,
          3.0174e-03, 3.9787e-03, 7.4120e-03, 1.1833e-02, 3.7022e-03, 1.1301e-03,
          1.2989e-03, 1.6708e-03, 2.4471e-03, 1.0567e-03, 9.7847e-04, 1.3237e-03,
          1.2236e-03, 2.6321e-03, 1.2846e-03, 2.0123e-03, 9.7847e-04, 1.2569e-03,
          5.9175e-04, 1.3151e-03, 1.0500e-03, 1.3752e-03, 8.1301e-04, 1.2541e-03,
          6.2370e-03, 3.2902e-03, 4.9210e-03, 2.5826e-03, 1.1234e-03, 1.8082e-03,
          4.5929e-03, 5.1575e-03, 2.8824e-02, 7.2765e-04, 2.3613e-03, 1.0691e-03,
          1.2751e-03, 2.9945e-03, 4.8370e-03, 7.3338e-04, 1.7204e-03, 1.3742e-03,
          1.5221e-03, 1.1930e-03, 1.1072e-03, 6.3038e-04, 1.6193e-03, 1.4296e-03,
          3.3512e-03, 1.5137e-02, 6.5002e-03, 1.5736e-03, 5.7144e-03, 1.4544e-03,
          2.7733e-03, 5.1193e-03, 2.2156e-02, 8.0795e-03, 5.2948e-03, 3.2635e-03,
          7.0000e-03, 1.1606e-03, 1.8072e-03, 1.2808e-03, 3.9520e-03, 5.1384e-03,
          5.9509e-03, 6.2256e-03, 2.0123e-03, 2.0905e-03, 2.6245e-03, 5.0278e-03,
          1.7258e-02, 3.7479e-03, 9.7885e-03, 4.8332e-03, 1.2711e-02, 6.3782e-03,
          8.1787e-03, 5.1178e-02, 9.5825e-02, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 2.0266e-04, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 4.1962e-04, 2.0266e-04, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 2.0266e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          2.5463e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 2.0266e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 2.0266e-04, 2.0266e-04,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          4.1962e-04, 4.1962e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          2.0266e-04, 3.7849e-05, 3.7849e-05, 2.5463e-04, 3.7849e-05, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 2.0266e-04, 3.7849e-05,
          2.5463e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          2.5463e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05, 2.5463e-04,
          3.7849e-05, 2.5463e-04, 3.7849e-05, 3.7849e-05, 3.7849e-05, 3.7849e-05,
          2.5463e-04, 2.5463e-04, 2.5463e-04, 3.3545e-04, 3.1257e-04, 3.7849e-05,
          3.7849e-05, 3.7849e-05, 3.7849e-05, 6.2466e-04, 3.3545e-04, 3.7849e-05,
          3.7849e-05, 3.7849e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.3449e-03, 5.0171e-02, 1.6006e-02, 1.5320e-02, 5.4626e-02,
          3.3478e-02, 1.3863e-02, 5.4455e-04, 5.7745e-04, 1.3533e-03, 2.4776e-03,
          8.2064e-04, 8.3017e-04, 5.3549e-04, 1.4143e-03, 6.8521e-04, 1.4610e-03,
          7.2002e-04, 5.0402e-04, 6.6328e-04, 9.9850e-04, 7.2622e-04, 1.2455e-03,
          3.3340e-03, 4.1466e-03, 1.0841e-02, 3.1948e-03, 7.8917e-04, 1.3008e-03,
          6.8140e-04, 9.4938e-04, 5.7316e-04, 2.1515e-03, 1.0262e-03, 8.4400e-04,
          2.7695e-03, 1.3227e-03, 3.2997e-03, 1.7900e-03, 6.3133e-04, 5.1231e-03,
          2.2984e-03, 1.9341e-03, 1.5602e-03, 1.8063e-03, 1.2550e-03, 5.2109e-03,
          5.9700e-04, 1.0948e-03, 6.4240e-03, 1.3866e-03, 7.1573e-04, 7.1669e-04,
          4.4274e-04, 5.5742e-04, 8.0442e-04, 1.7014e-03, 1.0633e-03, 1.8167e-03,
          1.9083e-03, 1.3809e-03, 1.6403e-03, 2.3632e-03, 2.2907e-03, 6.8207e-03,
          1.9274e-03, 1.0443e-03, 1.7576e-03, 1.9245e-03, 1.6823e-03, 1.4782e-03,
          1.9140e-03, 3.2444e-03, 8.1158e-04, 9.2316e-04, 4.1914e-04, 3.7718e-04,
          7.0858e-04, 1.8063e-03, 1.1272e-03, 1.6956e-03, 9.9945e-04, 1.2989e-03,
          1.7061e-03, 2.4834e-03, 1.1644e-03, 1.5984e-03, 1.0099e-03, 1.2732e-03,
          2.9869e-03, 3.7785e-03, 1.3857e-03, 2.3251e-03, 3.7365e-03, 3.5210e-03,
          8.4639e-04, 6.2275e-04, 1.0424e-03, 4.1542e-03, 2.2659e-03, 1.9264e-03,
          2.3594e-03, 7.4148e-04, 6.2227e-04, 3.2043e-03, 2.7561e-03, 1.8225e-03,
          2.2659e-03, 4.0092e-03, 6.1569e-03, 1.6356e-03, 1.4725e-03, 2.1782e-03,
          3.0136e-03, 8.1406e-03, 2.4605e-03, 7.0992e-03, 4.8943e-03, 1.5383e-03,
          1.5211e-03, 4.8218e-03, 9.9564e-04, 3.6478e-04, 2.8076e-03, 5.5933e-04,
          1.5507e-03, 9.8610e-04, 1.2083e-03, 3.3798e-03, 1.4334e-03, 2.5101e-03,
          2.0332e-03, 1.1902e-03, 5.3692e-04, 1.0080e-03, 3.3207e-03, 2.5330e-03,
          5.0583e-03, 2.6360e-03, 2.1324e-03, 2.5616e-03, 2.4643e-03, 1.4191e-03,
          1.4353e-03, 7.5531e-04, 1.2827e-03, 1.8005e-03, 1.1721e-03, 4.8790e-03,
          5.5237e-03, 4.1389e-03, 3.9368e-03, 3.9597e-03, 2.6474e-03, 1.2589e-02,
          4.6310e-03, 6.8665e-03, 6.8426e-04, 2.3136e-03, 1.4084e-02, 3.7212e-03,
          6.6605e-03, 6.0577e-03, 2.5024e-03, 2.5444e-03, 5.1498e-03, 2.5120e-03,
          2.1896e-03, 2.1172e-03, 2.5921e-03, 1.4334e-03, 1.5059e-03, 5.5733e-03,
          3.2616e-03, 1.9588e-03, 6.7101e-03, 8.4229e-03, 1.0628e-02, 1.7366e-03,
          2.9697e-03, 2.7294e-03, 5.4970e-03, 1.4744e-03, 1.5383e-03, 2.1572e-03,
          1.6432e-03, 2.4910e-03, 1.7004e-03, 3.0098e-03, 1.0262e-03, 2.3670e-03,
          8.6117e-04, 1.4172e-03, 2.5501e-03, 2.8057e-03, 1.8721e-03, 3.2310e-03,
          4.6539e-03, 3.3321e-03, 1.0490e-02, 3.2825e-03, 2.2430e-03, 4.6577e-03,
          2.7809e-03, 7.6790e-03, 1.0201e-02, 8.8310e-04, 4.5586e-03, 2.0943e-03,
          1.3094e-03, 5.3825e-03, 3.9253e-03, 9.7418e-04, 3.8033e-03, 1.2093e-03,
          5.4054e-03, 2.8610e-03, 1.9388e-03, 9.2745e-04, 4.5700e-03, 1.6556e-03,
          1.0178e-02, 6.7406e-03, 1.0445e-02, 2.8114e-03, 9.7122e-03, 3.0308e-03,
          3.0537e-03, 2.9640e-03, 6.4011e-03, 7.1907e-03, 1.1093e-02, 3.1128e-03,
          4.2839e-03, 1.8902e-03, 1.3523e-03, 2.2411e-03, 7.0419e-03, 6.0120e-03,
          3.1586e-03, 3.5763e-03, 3.7575e-03, 6.5041e-03, 4.3907e-03, 5.1079e-03,
          1.3718e-02, 4.3259e-03, 4.9477e-03, 4.3831e-03, 8.0719e-03, 1.2764e-02,
          4.4479e-03, 1.2558e-02, 1.8417e-02, 3.1830e-02, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.2803e-04, 1.2803e-04, 2.3127e-05,
          1.2803e-04, 1.2803e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.2803e-04, 2.3127e-05, 2.3127e-05,
          1.2803e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 3.0041e-04, 3.0041e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 1.2803e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 1.9562e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          1.9562e-04, 2.3127e-05, 1.9562e-04, 2.3127e-05, 2.3127e-05, 1.2803e-04,
          2.3127e-05, 2.3127e-05, 3.0041e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 1.2803e-04, 2.3127e-05, 2.3127e-05, 1.2803e-04,
          2.3127e-05, 2.3127e-05, 3.0041e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 1.9562e-04, 2.3127e-05, 2.3127e-05, 3.0041e-04, 1.2803e-04,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 1.9562e-04, 3.0041e-04, 2.3127e-05,
          3.0041e-04, 3.0041e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          1.2803e-04, 2.3127e-05, 2.3127e-05, 1.9562e-04, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05, 3.0041e-04, 2.3127e-05,
          1.9562e-04, 3.0041e-04, 3.0041e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          1.9562e-04, 2.3127e-05, 3.0041e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          2.3127e-05, 1.9562e-04, 2.3127e-05, 2.3127e-05, 2.3127e-05, 2.3127e-05,
          1.9562e-04, 1.9562e-04, 6.1417e-04, 1.0782e-04, 1.9562e-04, 2.3127e-05,
          2.3127e-05, 2.3127e-05, 2.3127e-05, 5.3215e-04, 2.3127e-05, 4.4751e-04,
          1.9562e-04, 2.3127e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9673e-03, 5.2429e-02, 1.6891e-02, 1.4244e-02, 3.2623e-02,
          4.1840e-02, 9.8877e-03, 2.4147e-03, 1.2350e-03, 3.2120e-03, 5.3520e-03,
          1.7214e-03, 9.2010e-03, 1.8015e-03, 4.4098e-03, 2.8553e-03, 1.0696e-02,
          6.1989e-03, 1.6632e-03, 4.5037e-04, 5.5408e-04, 8.9312e-04, 6.4011e-03,
          6.5308e-03, 2.2449e-03, 8.1787e-03, 5.0201e-03, 7.0667e-04, 1.0620e-02,
          3.0861e-03, 5.3253e-03, 2.1896e-03, 5.3291e-03, 4.1046e-03, 1.5545e-03,
          1.9274e-03, 4.1175e-04, 3.2482e-03, 4.0703e-03, 6.6280e-04, 1.9627e-03,
          6.1989e-04, 4.8804e-04, 5.4836e-04, 5.3644e-04, 1.5038e-02, 5.8098e-03,
          1.2197e-03, 9.4080e-04, 2.8267e-03, 1.5917e-03, 2.3575e-03, 1.6594e-03,
          3.7766e-04, 1.0681e-03, 4.2582e-04, 4.4584e-04, 9.2554e-04, 5.8556e-04,
          6.3658e-04, 2.1858e-03, 1.0977e-03, 1.9455e-03, 7.8106e-04, 7.0572e-03,
          8.7118e-04, 9.2506e-04, 1.7242e-03, 3.3998e-04, 3.1805e-04, 4.8542e-04,
          5.8794e-04, 8.2626e-03, 5.2929e-04, 3.3302e-03, 3.6955e-04, 3.8433e-04,
          9.6321e-04, 3.6478e-04, 4.6730e-04, 1.7567e-03, 5.9080e-04, 4.2367e-04,
          3.9029e-04, 4.3373e-03, 7.0620e-04, 3.9673e-03, 5.6076e-04, 4.8614e-04,
          6.4316e-03, 3.7880e-03, 3.9363e-04, 3.5877e-03, 5.7602e-04, 6.0940e-04,
          1.1349e-03, 3.0565e-04, 3.1719e-03, 2.4853e-03, 3.4451e-04, 5.1785e-04,
          4.2458e-03, 6.0320e-04, 6.6757e-04, 1.0138e-03, 4.3106e-03, 1.7414e-03,
          3.4122e-03, 5.1546e-04, 3.4847e-03, 4.1223e-04, 6.9475e-04, 8.9121e-04,
          1.9951e-03, 1.1082e-03, 1.1845e-03, 9.0714e-03, 8.8263e-04, 1.1320e-03,
          2.4395e-03, 1.7767e-03, 2.4962e-04, 1.5974e-03, 5.5008e-03, 2.1541e-04,
          2.8706e-04, 3.2139e-04, 4.6086e-04, 4.6959e-03, 5.5923e-03, 4.5967e-03,
          1.1959e-03, 7.8964e-04, 2.1687e-03, 6.2704e-04, 2.2449e-03, 4.1056e-04,
          3.2745e-02, 2.7962e-03, 1.1497e-02, 8.8501e-03, 6.6376e-03, 4.2038e-03,
          3.5405e-04, 1.2264e-03, 5.3835e-04, 4.5228e-04, 5.3549e-04, 2.7847e-03,
          3.6755e-03, 2.6283e-03, 6.0272e-04, 2.9335e-03, 1.9608e-03, 1.4029e-03,
          2.3422e-03, 1.1158e-03, 2.3861e-03, 5.1403e-04, 4.3182e-03, 6.7062e-03,
          7.6218e-03, 5.0812e-03, 2.4216e-02, 8.8882e-03, 7.1335e-04, 6.6042e-04,
          2.6608e-03, 9.0885e-04, 1.8015e-03, 4.1275e-03, 6.6376e-03, 3.9253e-03,
          1.4429e-03, 8.0347e-04, 9.0837e-04, 1.0338e-03, 3.3855e-03, 1.1330e-02,
          2.0504e-03, 3.9558e-03, 5.9433e-03, 8.3237e-03, 5.9204e-03, 9.5596e-03,
          6.4392e-03, 3.9062e-03, 4.0512e-03, 5.7364e-04, 5.6458e-04, 5.5122e-03,
          6.5079e-03, 2.9063e-04, 4.5013e-03, 2.1835e-02, 4.3488e-03, 3.2845e-03,
          8.8882e-04, 1.2226e-03, 5.1155e-03, 1.2541e-03, 2.4242e-03, 2.4624e-03,
          7.5245e-04, 2.3003e-03, 8.9455e-04, 2.5902e-03, 6.6872e-03, 9.3231e-03,
          3.1509e-03, 1.3494e-03, 5.6553e-04, 1.1225e-03, 4.9782e-03, 5.1880e-04,
          1.6975e-03, 6.6681e-03, 7.6942e-03, 4.5891e-03, 1.0620e-02, 4.9629e-03,
          8.0338e-03, 6.3848e-04, 1.8253e-03, 1.4677e-03, 1.1414e-02, 6.9351e-03,
          1.2522e-03, 6.8331e-04, 7.5102e-04, 2.0161e-03, 3.9043e-03, 3.8261e-03,
          9.7942e-04, 8.2445e-04, 5.4455e-04, 2.5368e-03, 1.6346e-03, 3.0327e-03,
          1.0118e-03, 8.3447e-04, 5.9319e-04, 2.6951e-03, 8.1482e-03, 2.8553e-03,
          1.1196e-03, 2.0542e-03, 5.7507e-04, 1.0328e-03, 1.0433e-03, 3.9024e-03,
          6.0987e-04, 6.5041e-04, 1.1806e-03, 1.8597e-03, 4.9164e-02, 1.3673e-04,
          2.4378e-05, 2.4891e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4891e-04, 2.4378e-05,
          6.0892e-04, 2.4891e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4891e-04, 2.4378e-05, 2.4939e-04,
          2.4891e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          3.2663e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 1.3673e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4891e-04,
          2.4378e-05, 2.4378e-05, 6.0892e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3673e-04, 2.4891e-04,
          2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05, 3.5310e-04, 2.4378e-05,
          4.9639e-04, 1.3673e-04, 2.4378e-05, 2.4378e-05, 2.4891e-04, 1.3673e-04,
          3.5310e-04, 1.3673e-04, 2.4378e-05, 2.4378e-05, 2.4891e-04, 4.6539e-04,
          2.4891e-04, 2.4891e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          1.3673e-04, 4.9639e-04, 2.4378e-05, 5.7793e-04, 2.4378e-05, 2.4378e-05,
          1.3673e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05, 1.3673e-04, 2.4378e-05,
          1.3673e-04, 2.4891e-04, 1.3673e-04, 2.4378e-05, 2.4378e-05, 4.9639e-04,
          2.4891e-04, 2.4378e-05, 6.0892e-04, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          4.9639e-04, 3.9101e-03, 2.4378e-05, 2.4378e-05, 2.4378e-05, 2.4378e-05,
          1.3673e-04, 1.3673e-04, 1.3673e-04, 2.4378e-05, 1.3673e-04, 2.4378e-05,
          2.4378e-05, 1.3673e-04, 1.3673e-04, 4.3464e-04, 2.4378e-05, 5.4693e-04,
          1.3673e-04, 2.4378e-05], dtype=torch.float16),
  tensor([0.0000e+00, 7.4120e-03, 6.7139e-02, 1.9257e-02, 3.6163e-02, 4.9164e-02,
          5.2155e-02, 2.5513e-02, 4.4136e-03, 1.0519e-03, 2.5997e-03, 2.4605e-03,
          1.2197e-03, 1.5535e-03, 9.4843e-04, 1.6327e-03, 3.3932e-03, 2.9125e-03,
          1.5678e-03, 8.1778e-04, 6.4468e-04, 1.0014e-03, 6.1846e-04, 2.8114e-03,
          2.5749e-03, 1.6546e-03, 2.0172e-02, 1.2939e-02, 1.4677e-03, 2.6302e-03,
          1.2665e-03, 1.1047e-02, 1.2293e-03, 3.4866e-03, 2.1725e-03, 1.0201e-02,
          2.8572e-03, 1.0195e-03, 5.0964e-03, 2.9907e-03, 7.7295e-04, 2.3632e-03,
          8.1587e-04, 6.0272e-04, 5.9557e-04, 6.5279e-04, 1.9121e-03, 3.5973e-03,
          7.0953e-04, 1.2417e-03, 7.8659e-03, 1.3840e-02, 9.2850e-03, 4.8790e-03,
          1.2503e-03, 3.9215e-03, 1.9855e-03, 1.0786e-03, 1.3342e-03, 1.1435e-03,
          9.2173e-04, 1.9026e-03, 2.3270e-03, 5.5046e-03, 7.9298e-04, 3.1033e-03,
          8.1444e-04, 5.4216e-04, 1.2045e-03, 4.5657e-04, 4.2152e-04, 5.4073e-04,
          8.5878e-04, 6.3629e-03, 3.1223e-03, 1.1238e-02, 1.0719e-03, 1.3208e-03,
          3.7632e-03, 1.0214e-03, 7.9489e-04, 2.1305e-03, 9.1887e-04, 7.9107e-04,
          8.0729e-04, 1.3405e-02, 1.4582e-03, 1.2732e-03, 5.5742e-04, 5.6553e-04,
          2.1935e-03, 2.2717e-03, 3.1614e-04, 1.5707e-03, 5.3978e-04, 8.5592e-04,
          2.0161e-03, 1.3571e-03, 1.3855e-02, 4.7760e-03, 8.6641e-04, 1.1415e-03,
          2.3022e-03, 6.7949e-04, 6.8569e-04, 2.6760e-03, 2.3060e-03, 1.5144e-03,
          6.0577e-03, 8.7643e-04, 2.6836e-03, 4.4298e-04, 6.6757e-04, 9.1219e-04,
          9.4700e-04, 1.0347e-03, 1.3218e-03, 3.1700e-03, 9.7370e-04, 3.1471e-03,
          2.6531e-03, 2.6302e-03, 5.6314e-04, 1.4200e-03, 2.5425e-03, 4.5085e-04,
          7.0047e-04, 5.9175e-04, 7.1049e-04, 2.7447e-03, 4.3144e-03, 5.4779e-03,
          9.0933e-04, 6.3038e-04, 8.2064e-04, 5.1117e-04, 1.3905e-03, 4.7255e-04,
          8.0032e-03, 1.5974e-03, 2.9564e-03, 1.1795e-02, 9.4299e-03, 2.6112e-03,
          7.3767e-04, 7.0286e-04, 9.1457e-04, 9.3079e-04, 7.9298e-04, 2.1496e-03,
          4.5471e-03, 1.9283e-03, 1.1520e-03, 1.0483e-02, 3.9215e-03, 2.2392e-03,
          2.2469e-03, 1.9093e-03, 6.7043e-04, 5.4789e-04, 2.6875e-03, 3.4542e-03,
          4.2076e-03, 5.3749e-03, 4.2458e-03, 5.6267e-03, 2.3651e-03, 1.5736e-03,
          1.9379e-03, 2.0256e-03, 1.5154e-03, 1.1005e-03, 3.2673e-03, 4.2725e-03,
          1.6098e-03, 1.1625e-03, 1.9722e-03, 2.0370e-03, 3.4389e-03, 3.7670e-03,
          1.6804e-03, 1.4172e-03, 2.8992e-03, 7.1678e-03, 2.0542e-03, 1.7118e-03,
          1.6766e-03, 2.4796e-03, 4.6082e-03, 1.3781e-03, 9.0837e-04, 2.2144e-03,
          7.1411e-03, 6.5517e-04, 2.5864e-03, 4.3869e-03, 2.1248e-03, 2.4300e-03,
          1.6985e-03, 1.6832e-03, 3.1414e-03, 1.1406e-03, 1.6136e-03, 2.0142e-03,
          7.1526e-04, 1.7262e-03, 7.6437e-04, 1.5898e-03, 2.9278e-03, 2.4605e-03,
          2.8858e-03, 4.2458e-03, 1.6527e-03, 1.2836e-03, 2.6817e-03, 8.2970e-04,
          3.0670e-03, 4.3144e-03, 3.6297e-03, 9.6416e-04, 5.3825e-03, 2.4605e-03,
          7.5836e-03, 9.6750e-04, 2.0256e-03, 1.2197e-03, 3.8815e-03, 2.6207e-03,
          8.5163e-04, 6.5708e-04, 5.9700e-04, 1.6918e-03, 2.2087e-03, 1.6184e-03,
          3.7823e-03, 1.5659e-03, 8.1301e-04, 1.5860e-03, 2.6360e-03, 2.6531e-03,
          1.3800e-03, 1.5640e-03, 1.3885e-03, 4.4632e-03, 4.5090e-03, 2.5482e-03,
          2.1267e-03, 1.2379e-03, 6.5422e-04, 8.0967e-04, 9.3794e-04, 2.4261e-03,
          5.9509e-04, 5.6505e-04, 1.0118e-03, 1.5688e-03, 3.8033e-03, 7.8857e-02,
          2.0444e-05, 7.7605e-05, 2.0444e-05, 7.7605e-05, 2.0444e-05, 2.0444e-05,
          7.7605e-05, 2.0444e-05, 2.0444e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05,
          6.7377e-04, 2.0444e-05, 7.7605e-05, 2.0444e-05, 7.7605e-05, 4.8065e-04,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 7.7605e-05, 2.0444e-05, 2.7418e-04,
          7.7605e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.7084e-04, 7.7605e-05,
          2.0444e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.7084e-04, 7.7605e-05, 2.0444e-05,
          2.0444e-05, 7.7605e-05, 2.7084e-04, 7.7605e-05, 2.0444e-05, 2.0444e-05,
          2.7084e-04, 7.7605e-05, 2.7084e-04, 7.7605e-05, 2.0444e-05, 7.7605e-05,
          7.7605e-05, 2.7084e-04, 2.0444e-05, 2.7084e-04, 7.7605e-05, 7.7605e-05,
          4.8065e-04, 2.0444e-05, 7.7605e-05, 2.7084e-04, 2.0444e-05, 2.0444e-05,
          7.7605e-05, 7.7605e-05, 7.7605e-05, 2.0444e-05, 2.0444e-05, 7.7605e-05,
          2.0444e-05, 7.7605e-05, 7.7605e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          7.7605e-05, 2.0444e-05, 7.7605e-05, 7.7605e-05, 2.0444e-05, 7.7605e-05,
          2.0444e-05, 2.0444e-05, 2.7084e-04, 2.0444e-05, 7.7605e-05, 7.7605e-05,
          7.7605e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 7.4148e-04, 2.7084e-04,
          4.8065e-04, 2.0444e-05, 5.4836e-04, 2.0444e-05, 7.7605e-05, 1.1711e-03,
          1.1444e-03, 1.1444e-03, 1.1444e-03, 2.0444e-05, 2.7084e-04, 2.0444e-05,
          7.7605e-05, 7.7605e-05, 2.0444e-05, 7.7605e-05, 2.0444e-05, 2.0444e-05,
          2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05, 1.1444e-03, 2.0444e-05,
          1.8692e-03, 7.7605e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05,
          1.1444e-03, 2.0444e-05, 7.7605e-05, 7.7605e-05, 2.0444e-05, 5.9605e-04,
          7.7605e-05, 1.4696e-03, 2.0444e-05, 2.0444e-05, 2.0444e-05, 2.0444e-05,
          7.7605e-05, 7.7605e-05, 2.0444e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05,
          2.0444e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05, 7.7605e-05,
          7.7605e-05, 7.7605e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.8022e-03, 4.8431e-02, 1.6998e-02, 2.0370e-02, 5.5939e-02,
          3.2806e-02, 1.5350e-02, 7.6246e-04, 7.1383e-04, 2.3689e-03, 4.0588e-03,
          1.4515e-03, 9.2125e-04, 7.1287e-04, 1.6451e-03, 5.7840e-04, 1.4973e-03,
          7.4911e-04, 4.0364e-04, 5.4073e-04, 6.3181e-04, 4.4012e-04, 9.2363e-04,
          3.6888e-03, 2.2736e-03, 1.1665e-02, 3.4180e-03, 1.4706e-03, 1.6632e-03,
          1.0290e-03, 1.2150e-03, 7.1335e-04, 2.7771e-03, 1.2093e-03, 8.2254e-04,
          3.3169e-03, 1.8892e-03, 2.2907e-03, 1.6766e-03, 6.8855e-04, 3.0727e-03,
          1.5879e-03, 1.3657e-03, 1.0433e-03, 1.0347e-03, 1.1749e-03, 3.3932e-03,
          4.1151e-04, 6.2895e-04, 5.2643e-03, 2.2926e-03, 1.1864e-03, 9.2173e-04,
          6.5899e-04, 7.5197e-04, 1.0433e-03, 4.2229e-03, 1.4305e-03, 3.7899e-03,
          2.2507e-03, 1.2445e-03, 1.1015e-03, 1.3151e-03, 1.9140e-03, 4.7264e-03,
          1.2960e-03, 6.7854e-04, 1.2350e-03, 9.3746e-04, 8.8120e-04, 9.9468e-04,
          9.4366e-04, 4.4632e-03, 1.7471e-03, 1.4858e-03, 5.1737e-04, 4.8637e-04,
          1.0853e-03, 5.8632e-03, 2.7409e-03, 3.3569e-03, 2.2602e-03, 1.4400e-03,
          1.5583e-03, 1.5364e-03, 7.9250e-04, 1.5287e-03, 7.5674e-04, 9.3412e-04,
          2.5978e-03, 2.4605e-03, 6.3658e-04, 1.7614e-03, 1.0948e-03, 1.0586e-03,
          1.4410e-03, 1.0748e-03, 1.0242e-03, 6.1226e-03, 9.8114e-03, 6.5727e-03,
          5.3329e-03, 1.3542e-03, 1.5287e-03, 4.7531e-03, 4.8180e-03, 1.4114e-03,
          1.2903e-03, 1.9264e-03, 3.6278e-03, 8.0061e-04, 1.2617e-03, 1.1692e-03,
          1.9646e-03, 2.5940e-03, 6.6710e-04, 4.5853e-03, 1.3695e-03, 3.5229e-03,
          3.9482e-03, 1.1490e-02, 3.9482e-03, 5.3024e-04, 4.8676e-03, 1.6565e-03,
          6.6147e-03, 2.3098e-03, 2.5501e-03, 4.5280e-03, 9.5749e-04, 1.4553e-03,
          1.6527e-03, 1.0967e-03, 4.9639e-04, 5.2214e-04, 1.7519e-03, 1.0471e-03,
          2.8152e-03, 8.9693e-04, 7.1144e-04, 1.8682e-03, 2.0008e-03, 2.6703e-03,
          6.4049e-03, 1.0452e-03, 3.7193e-03, 4.8866e-03, 2.8305e-03, 9.1782e-03,
          9.3918e-03, 5.1155e-03, 5.3368e-03, 2.4452e-03, 1.7929e-03, 5.6763e-03,
          2.3289e-03, 2.9507e-03, 5.1451e-04, 9.8896e-04, 5.3673e-03, 1.1606e-03,
          3.3932e-03, 2.7256e-03, 1.7481e-03, 2.2659e-03, 1.9745e-02, 5.4283e-03,
          4.4746e-03, 4.0245e-03, 4.7302e-03, 1.5860e-03, 1.9493e-03, 4.9782e-03,
          2.4071e-03, 2.1839e-03, 6.1417e-03, 3.8033e-03, 4.2915e-03, 1.4601e-03,
          2.0542e-03, 1.3685e-03, 3.1109e-03, 1.0128e-03, 8.0299e-04, 9.0456e-04,
          1.3580e-03, 8.8120e-04, 1.1539e-03, 1.7715e-02, 3.7231e-03, 3.1090e-03,
          1.2159e-03, 7.1487e-03, 2.6741e-03, 2.5883e-03, 2.5253e-03, 3.6068e-03,
          5.2605e-03, 3.1986e-03, 3.5534e-03, 1.9054e-03, 1.4601e-03, 1.8072e-03,
          1.3971e-03, 2.5425e-03, 2.3327e-03, 7.6628e-04, 3.2539e-03, 9.6560e-04,
          9.0456e-04, 1.8051e-02, 2.0020e-02, 3.5248e-03, 8.3084e-03, 4.4861e-03,
          6.6032e-03, 3.4561e-03, 2.1000e-03, 8.4496e-04, 3.9673e-03, 1.9684e-03,
          5.8594e-03, 2.7008e-03, 3.4428e-03, 1.5097e-03, 4.8218e-03, 2.1744e-03,
          1.3161e-03, 9.7847e-04, 1.3161e-03, 1.8206e-03, 2.7523e-03, 9.6416e-04,
          2.4155e-02, 5.9395e-03, 3.9787e-03, 2.6131e-03, 1.7853e-02, 9.3842e-03,
          6.9466e-03, 7.8812e-03, 5.8670e-03, 6.7558e-03, 5.6801e-03, 4.9438e-03,
          4.9248e-03, 1.8206e-03, 1.7004e-03, 1.3905e-03, 1.7576e-03, 3.3817e-03,
          1.0986e-03, 1.6651e-03, 2.3632e-03, 3.2463e-03, 1.5545e-03, 4.6539e-03,
          2.9633e-02, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 1.3447e-04, 2.4748e-04, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 5.6887e-04, 2.4748e-04, 2.3365e-05, 2.3365e-05,
          1.3447e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.4748e-04, 2.4748e-04, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.6393e-04, 1.3447e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          3.7718e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 3.7718e-04, 3.7718e-04, 2.3365e-05, 2.3365e-05,
          3.7718e-04, 2.6393e-04, 1.3638e-04, 2.3365e-05, 2.3365e-05, 2.4748e-04,
          2.3365e-05, 2.3365e-05, 2.4748e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 3.7503e-04, 2.3365e-05, 2.3365e-05, 2.4748e-04,
          2.3365e-05, 2.3365e-05, 2.4748e-04, 2.6393e-04, 2.3365e-05, 5.6887e-04,
          5.6887e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05, 1.3447e-04, 1.3447e-04,
          2.3365e-05, 3.7718e-04, 2.3365e-05, 2.3365e-05, 2.4748e-04, 2.3365e-05,
          2.4748e-04, 2.4748e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          3.7503e-04, 4.4131e-04, 2.3365e-05, 1.3638e-04, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 1.3447e-04, 2.3365e-05,
          1.3638e-04, 2.4748e-04, 1.3447e-04, 2.3365e-05, 2.3365e-05, 6.8188e-04,
          2.4748e-04, 2.6393e-04, 4.8804e-04, 2.3365e-05, 2.3365e-05, 2.6393e-04,
          6.8188e-04, 3.7718e-04, 2.6393e-04, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 1.3638e-04, 2.3365e-05,
          2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05, 2.3365e-05,
          2.3365e-05, 2.3365e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.8370e-03, 5.0690e-02, 1.4023e-02, 1.9897e-02, 3.4576e-02,
          3.1708e-02, 1.1436e-02, 1.0023e-03, 7.4196e-04, 2.5978e-03, 7.4883e-03,
          1.9045e-03, 7.5960e-04, 9.3174e-04, 8.3494e-04, 7.4530e-04, 1.0366e-03,
          7.3004e-04, 5.8365e-04, 2.0945e-04, 2.0814e-04, 2.3067e-04, 1.2684e-03,
          8.1863e-03, 1.5707e-03, 9.2010e-03, 4.9248e-03, 5.2881e-04, 1.1625e-03,
          1.8282e-03, 1.7347e-03, 3.9959e-04, 1.1005e-03, 1.1616e-03, 7.1907e-04,
          1.1311e-03, 8.2111e-04, 1.1568e-03, 8.0204e-04, 3.7456e-04, 1.3971e-03,
          1.0586e-03, 8.8549e-04, 1.0891e-03, 4.5896e-04, 1.1644e-03, 1.2131e-03,
          4.7874e-04, 3.3593e-04, 2.6760e-03, 1.2026e-03, 1.4763e-03, 1.0881e-03,
          6.0987e-04, 8.2731e-04, 6.0368e-04, 9.5081e-04, 1.7843e-03, 1.6737e-03,
          1.4057e-03, 7.1621e-04, 8.1873e-04, 6.0558e-04, 2.7885e-03, 2.4242e-03,
          2.1484e-02, 5.6877e-03, 1.4629e-03, 5.9605e-04, 5.6744e-04, 1.4305e-03,
          5.7459e-04, 6.7749e-03, 9.2125e-04, 1.9522e-03, 6.7329e-04, 5.9128e-04,
          9.4223e-04, 1.5202e-03, 8.9188e-03, 2.6646e-03, 7.5684e-03, 7.2289e-04,
          6.3610e-04, 1.8749e-03, 7.1812e-04, 3.9864e-03, 9.4376e-03, 7.0915e-03,
          1.0002e-02, 3.6068e-03, 9.0742e-04, 4.9858e-03, 3.8195e-04, 4.5609e-04,
          6.0177e-04, 4.9829e-04, 1.3304e-03, 1.6327e-03, 2.5921e-03, 2.6264e-03,
          1.1070e-02, 8.9722e-03, 5.6114e-03, 1.6594e-03, 1.5631e-03, 7.2241e-04,
          1.4524e-03, 1.4477e-03, 5.5084e-03, 3.1109e-03, 1.4191e-02, 7.3128e-03,
          1.3189e-03, 7.6199e-04, 3.3450e-04, 1.9913e-03, 6.3992e-04, 1.3914e-03,
          1.9703e-03, 3.8319e-03, 2.6512e-03, 8.1253e-04, 2.5215e-03, 3.7441e-03,
          1.0424e-03, 5.1727e-03, 9.3384e-03, 1.8358e-03, 1.3971e-03, 1.7548e-03,
          1.4915e-02, 1.2032e-02, 6.3038e-04, 2.7084e-03, 5.0163e-03, 8.5735e-04,
          2.3365e-03, 8.8739e-04, 1.3075e-03, 2.6932e-03, 2.4319e-03, 2.2449e-03,
          5.7144e-03, 1.1683e-03, 2.5291e-03, 2.5101e-03, 1.2062e-02, 2.5749e-03,
          6.8474e-03, 1.7481e-03, 1.2474e-03, 1.0519e-03, 1.2856e-03, 1.9569e-03,
          4.6844e-03, 2.7256e-03, 1.6928e-03, 3.9711e-03, 1.8892e-03, 8.7595e-04,
          2.0046e-03, 3.0975e-03, 2.0409e-03, 2.2621e-03, 3.1071e-03, 5.1041e-03,
          1.1742e-02, 4.6883e-03, 2.6855e-03, 1.3466e-03, 2.2812e-03, 2.8152e-03,
          1.4229e-03, 1.4858e-03, 1.3762e-03, 1.3933e-03, 2.2297e-03, 2.3861e-03,
          2.2373e-03, 1.1091e-03, 2.5330e-03, 2.0199e-03, 1.2913e-03, 8.8787e-04,
          3.9482e-03, 1.1511e-03, 2.1133e-03, 6.3248e-03, 1.4336e-02, 2.5005e-03,
          2.5692e-03, 1.6613e-03, 2.1286e-03, 2.9697e-03, 7.1411e-03, 2.1839e-03,
          1.6661e-03, 1.2703e-03, 1.8330e-03, 6.1378e-03, 6.4316e-03, 2.0447e-03,
          5.1613e-03, 4.4250e-03, 9.4414e-04, 1.1425e-03, 8.7357e-03, 9.9659e-04,
          1.4982e-03, 3.8738e-03, 5.6000e-03, 1.3916e-02, 4.3678e-03, 7.1793e-03,
          2.8305e-03, 2.3804e-03, 3.9940e-03, 1.0471e-03, 2.9602e-03, 2.2774e-03,
          3.4885e-03, 1.3323e-03, 2.6951e-03, 1.2238e-02, 3.9043e-03, 2.3537e-03,
          1.3931e-02, 4.9095e-03, 1.2865e-03, 1.7967e-03, 1.3704e-03, 1.0004e-03,
          5.2185e-03, 9.7198e-03, 2.3056e-02, 2.2030e-03, 6.3400e-03, 1.0765e-02,
          7.1030e-03, 3.2482e-03, 2.2068e-03, 2.2888e-03, 2.7828e-03, 2.4147e-03,
          1.3781e-03, 1.0429e-02, 1.2436e-03, 1.8034e-03, 1.4658e-03, 2.3785e-03,
          5.2795e-03, 6.6710e-04, 1.1549e-03, 1.4601e-03, 1.9159e-03, 6.7406e-03,
          5.8823e-03, 3.2837e-02, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 8.9586e-05, 1.9586e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 1.2791e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 8.9586e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          8.9586e-05, 2.1696e-05, 1.9586e-04, 1.9586e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 8.9586e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 8.9586e-05, 8.9586e-05, 8.9586e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.9586e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.2791e-04,
          2.1696e-05, 2.1696e-05, 1.2791e-04, 8.9586e-05, 2.1696e-05, 2.1696e-05,
          1.9586e-04, 8.9586e-05, 1.2791e-04, 2.1696e-05, 2.1696e-05, 1.9586e-04,
          2.1696e-05, 8.9586e-05, 1.2791e-04, 8.9586e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 8.9586e-05, 2.1696e-05, 2.1696e-05, 1.2791e-04,
          2.1696e-05, 2.1696e-05, 1.9586e-04, 2.1696e-05, 2.0087e-04, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 8.9586e-05, 8.9586e-05,
          2.0087e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.2791e-04, 2.1696e-05,
          1.9586e-04, 1.9586e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 1.2791e-04, 2.1696e-05, 2.1696e-05,
          2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05, 8.9586e-05, 2.1696e-05,
          1.9586e-04, 1.9586e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 1.2791e-04,
          8.9586e-05, 2.1696e-05, 1.2791e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.2791e-04, 1.9586e-04, 2.1696e-05, 2.1696e-05, 2.1696e-05, 2.1696e-05,
          1.2791e-04, 1.2791e-04, 1.2791e-04, 1.2791e-04, 1.2791e-04, 1.2791e-04,
          2.1696e-05, 1.2791e-04, 1.2791e-04, 1.2791e-04, 1.2791e-04, 1.2791e-04,
          2.1696e-05, 1.2791e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.5090e-03, 5.6793e-02, 1.3489e-02, 1.0864e-02, 3.7781e-02,
          4.5441e-02, 7.5302e-03, 1.0967e-03, 9.1934e-04, 2.1400e-03, 3.2043e-03,
          1.3561e-03, 1.0118e-03, 4.7565e-04, 2.4452e-03, 7.3338e-04, 8.7881e-04,
          4.8804e-04, 4.9782e-04, 2.4796e-04, 2.7180e-04, 2.4164e-04, 8.4066e-04,
          3.1719e-03, 3.4409e-03, 5.9166e-03, 2.1172e-03, 6.1321e-04, 1.0576e-03,
          6.7472e-04, 1.8721e-03, 4.0698e-04, 2.8515e-03, 1.4448e-03, 1.0815e-03,
          3.5820e-03, 1.1511e-03, 1.3685e-03, 1.5821e-03, 3.6049e-04, 1.5116e-03,
          8.6880e-04, 7.0143e-04, 6.3181e-04, 6.2323e-04, 9.3079e-04, 3.1681e-03,
          2.8610e-04, 3.4475e-04, 2.5291e-03, 1.9779e-03, 1.6365e-03, 1.2789e-03,
          6.8283e-04, 1.0395e-03, 9.1076e-04, 2.0752e-03, 2.3899e-03, 2.5177e-03,
          2.2488e-03, 7.1144e-04, 7.8487e-04, 1.0509e-03, 1.2140e-03, 6.3858e-03,
          9.5081e-04, 5.3549e-04, 7.9775e-04, 5.9319e-04, 5.1641e-04, 5.8031e-04,
          5.6839e-04, 3.5210e-03, 1.1263e-03, 2.1343e-03, 5.9319e-04, 6.3801e-04,
          1.2951e-03, 2.8381e-03, 1.7271e-03, 3.8700e-03, 1.5402e-03, 2.2755e-03,
          1.5335e-03, 1.9951e-03, 6.1178e-04, 1.2770e-03, 5.3787e-04, 6.9046e-04,
          2.1381e-03, 1.7853e-03, 8.0633e-04, 1.3943e-03, 1.7843e-03, 1.0061e-03,
          7.9393e-04, 6.9094e-04, 1.8682e-03, 3.5858e-03, 4.1389e-03, 3.9711e-03,
          5.0278e-03, 1.3266e-03, 1.2617e-03, 7.5264e-03, 7.8125e-03, 8.8644e-04,
          1.7471e-03, 9.7322e-04, 2.3727e-03, 8.5640e-04, 9.9373e-04, 8.9884e-04,
          5.1613e-03, 2.5253e-03, 3.8958e-04, 7.2136e-03, 1.4057e-03, 1.5135e-03,
          1.7853e-03, 9.9869e-03, 2.5463e-03, 7.6962e-04, 7.1449e-03, 1.4677e-03,
          3.2082e-03, 1.3695e-03, 2.0733e-03, 7.2632e-03, 1.3638e-03, 2.0542e-03,
          1.3857e-03, 7.2527e-04, 7.0715e-04, 4.3893e-04, 1.3399e-03, 8.1921e-04,
          1.9484e-03, 6.3801e-04, 1.1482e-03, 1.9407e-03, 2.0924e-03, 1.8053e-03,
          4.4518e-03, 2.2240e-03, 3.9101e-03, 3.5133e-03, 2.0523e-03, 3.0853e-02,
          9.1324e-03, 9.9182e-03, 9.0408e-03, 1.8206e-03, 1.6518e-03, 2.9526e-03,
          1.8454e-03, 1.8578e-03, 6.5947e-04, 8.0013e-04, 6.7482e-03, 8.9550e-04,
          3.6964e-03, 2.8477e-03, 1.4162e-03, 2.8572e-03, 6.4812e-03, 3.8223e-03,
          5.1346e-03, 3.7212e-03, 1.0239e-02, 3.9368e-03, 1.3590e-03, 6.1150e-03,
          1.3313e-02, 6.9885e-03, 7.5607e-03, 1.6460e-03, 4.1962e-03, 1.7300e-03,
          2.4014e-03, 1.9989e-03, 3.5591e-03, 1.4896e-03, 1.1387e-03, 1.7147e-03,
          1.7462e-03, 8.4591e-04, 1.3895e-03, 7.2327e-03, 2.8172e-03, 4.8790e-03,
          2.0218e-03, 3.2482e-03, 4.3564e-03, 2.0275e-03, 3.0422e-03, 4.8714e-03,
          8.2703e-03, 1.6670e-03, 4.8027e-03, 1.2245e-03, 1.0796e-03, 2.2621e-03,
          9.9564e-04, 1.8988e-03, 1.6651e-03, 1.0176e-03, 2.7180e-03, 7.0047e-04,
          1.0462e-03, 6.9695e-03, 4.9896e-03, 3.2349e-03, 1.6602e-02, 4.1771e-03,
          8.8882e-03, 2.8400e-03, 2.5349e-03, 1.8806e-03, 5.5084e-03, 2.1801e-03,
          5.1651e-03, 1.5230e-03, 2.9240e-03, 1.1663e-03, 7.1526e-03, 2.9545e-03,
          9.4843e-04, 1.2398e-03, 1.5268e-03, 8.3542e-03, 4.5891e-03, 9.4557e-04,
          5.4474e-03, 4.9400e-03, 3.6812e-03, 9.0332e-03, 3.2471e-02, 2.5909e-02,
          9.3460e-03, 6.8703e-03, 6.3133e-03, 5.4855e-03, 5.5351e-03, 2.1267e-03,
          2.0733e-03, 1.7509e-03, 1.5507e-03, 2.3155e-03, 2.1152e-03, 4.5815e-03,
          8.0872e-04, 1.5221e-03, 1.9550e-03, 2.9774e-03, 1.9798e-03, 4.7531e-03,
          1.0696e-02, 3.5591e-03, 3.4515e-02, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 7.8857e-05, 3.8147e-04, 2.1052e-04, 2.1517e-05,
          7.8857e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 5.2691e-04, 2.1052e-04, 2.1517e-05, 7.8857e-05,
          1.5318e-04, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1052e-04, 5.0259e-04, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 1.9240e-04, 2.1052e-04, 7.8857e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 2.1517e-05, 2.1517e-05, 7.8857e-05, 2.1517e-05, 2.1517e-05,
          2.4986e-04, 7.8857e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05,
          2.1517e-05, 7.8857e-05, 5.2691e-04, 2.4986e-04, 2.1517e-05, 2.1517e-05,
          2.4986e-04, 2.4986e-04, 7.8857e-05, 2.1517e-05, 2.1517e-05, 2.1052e-04,
          2.1517e-05, 7.8857e-05, 2.1052e-04, 2.1517e-05, 2.1517e-05, 2.4986e-04,
          7.8857e-05, 2.1517e-05, 1.9240e-04, 5.2691e-04, 7.8857e-05, 2.1052e-04,
          2.1517e-05, 2.1517e-05, 7.8857e-05, 2.4986e-04, 7.8857e-05, 5.2691e-04,
          5.2691e-04, 7.8857e-05, 2.1517e-05, 2.1517e-05, 2.1052e-04, 2.1052e-04,
          7.8857e-05, 2.4986e-04, 7.8857e-05, 2.1517e-05, 2.1052e-04, 7.8857e-05,
          3.8147e-04, 2.1052e-04, 7.8857e-05, 2.1517e-05, 2.4986e-04, 2.1517e-05,
          2.4986e-04, 5.2691e-04, 2.1517e-05, 7.8857e-05, 2.1517e-05, 2.1517e-05,
          7.8857e-05, 2.1517e-05, 2.1517e-05, 2.1517e-05, 2.1052e-04, 2.1517e-05,
          7.8857e-05, 2.1052e-04, 2.1052e-04, 7.8857e-05, 7.8857e-05, 5.2691e-04,
          2.4986e-04, 5.2691e-04, 6.5851e-04, 2.4986e-04, 7.8857e-05, 2.4986e-04,
          5.2691e-04, 2.4986e-04, 2.1517e-05, 2.4986e-04, 7.8857e-05, 2.1517e-05,
          7.8857e-05, 7.8857e-05, 7.8857e-05, 7.8857e-05, 7.8857e-05, 7.8857e-05,
          7.8857e-05, 7.8857e-05, 5.0259e-04, 3.5858e-04, 2.1517e-05, 7.8857e-05,
          7.8857e-05, 7.8857e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.0398e-03, 5.5420e-02, 1.1925e-02, 1.0094e-02, 3.9154e-02,
          4.8187e-02, 7.4387e-03, 1.1349e-03, 8.0156e-04, 2.1229e-03, 3.2558e-03,
          1.2522e-03, 1.3418e-03, 4.5609e-04, 2.9049e-03, 9.1362e-04, 1.2550e-03,
          6.0129e-04, 5.1737e-04, 2.1589e-04, 2.4235e-04, 2.2554e-04, 7.7152e-04,
          3.5591e-03, 4.2801e-03, 5.8937e-03, 2.4853e-03, 4.4298e-04, 1.3828e-03,
          6.9857e-04, 2.1553e-03, 4.2391e-04, 4.2076e-03, 1.6937e-03, 1.2579e-03,
          3.4256e-03, 7.9775e-04, 1.8625e-03, 2.3766e-03, 2.8372e-04, 1.2360e-03,
          5.9128e-04, 5.2452e-04, 4.8757e-04, 4.9210e-04, 1.3142e-03, 3.8033e-03,
          2.7704e-04, 3.0446e-04, 2.4624e-03, 1.8396e-03, 1.5535e-03, 1.3313e-03,
          5.3215e-04, 9.8896e-04, 7.2718e-04, 1.3094e-03, 1.7796e-03, 1.6174e-03,
          1.6165e-03, 6.9332e-04, 7.6532e-04, 1.0729e-03, 8.4829e-04, 7.1602e-03,
          9.0885e-04, 5.0640e-04, 8.6021e-04, 5.0211e-04, 3.9124e-04, 5.9175e-04,
          5.0640e-04, 4.4518e-03, 7.4196e-04, 2.1210e-03, 4.7731e-04, 4.7326e-04,
          1.0338e-03, 1.6775e-03, 1.1616e-03, 3.7117e-03, 1.0681e-03, 1.3628e-03,
          9.8896e-04, 2.1935e-03, 5.5075e-04, 1.6136e-03, 5.0926e-04, 5.9509e-04,
          2.4567e-03, 1.9102e-03, 5.8794e-04, 1.5774e-03, 1.4725e-03, 7.9012e-04,
          6.1703e-04, 4.6277e-04, 2.0294e-03, 3.7212e-03, 1.9665e-03, 2.7637e-03,
          4.8027e-03, 9.6226e-04, 9.4748e-04, 4.1618e-03, 6.5765e-03, 9.1171e-04,
          1.9970e-03, 6.3229e-04, 2.5940e-03, 6.2466e-04, 9.0027e-04, 7.8058e-04,
          6.2904e-03, 2.0962e-03, 4.1127e-04, 8.4000e-03, 1.0014e-03, 1.1797e-03,
          1.6193e-03, 9.1324e-03, 1.1158e-03, 8.5783e-04, 6.1569e-03, 8.4019e-04,
          1.6775e-03, 8.6975e-04, 1.3924e-03, 1.0048e-02, 2.0981e-03, 2.4033e-03,
          1.3170e-03, 6.2609e-04, 7.9775e-04, 4.1628e-04, 1.5087e-03, 5.5027e-04,
          3.3264e-03, 6.7568e-04, 1.5135e-03, 2.1400e-03, 2.6321e-03, 2.0008e-03,
          1.9798e-03, 2.1553e-03, 3.0956e-03, 2.4147e-03, 1.5202e-03, 1.9318e-02,
          9.4070e-03, 9.9869e-03, 3.8548e-03, 1.7767e-03, 1.5640e-03, 2.1267e-03,
          2.1267e-03, 1.4544e-03, 8.5545e-04, 5.9366e-04, 6.9351e-03, 1.3876e-03,
          4.5891e-03, 2.9774e-03, 2.2812e-03, 3.5954e-03, 3.9978e-03, 2.7294e-03,
          4.2458e-03, 3.0422e-03, 1.0468e-02, 6.0043e-03, 1.9398e-03, 7.1678e-03,
          9.7275e-03, 4.4670e-03, 3.6621e-03, 1.0900e-03, 4.3907e-03, 2.1534e-03,
          2.0599e-03, 2.8591e-03, 4.7112e-03, 2.1648e-03, 1.1511e-03, 2.3804e-03,
          2.3441e-03, 1.1215e-03, 1.6537e-03, 3.5324e-03, 1.9817e-03, 6.9046e-03,
          2.7046e-03, 1.7490e-03, 6.8626e-03, 3.2959e-03, 3.4809e-03, 6.2485e-03,
          4.1428e-03, 1.8845e-03, 6.3667e-03, 1.0662e-03, 1.2188e-03, 2.2812e-03,
          8.2970e-04, 1.7986e-03, 1.4210e-03, 1.0710e-03, 3.3512e-03, 9.7370e-04,
          1.1683e-03, 4.7417e-03, 2.4834e-03, 2.5291e-03, 1.8753e-02, 2.6550e-03,
          9.6893e-03, 3.8338e-03, 3.1796e-03, 2.9144e-03, 8.9951e-03, 2.9335e-03,
          8.9188e-03, 1.0471e-03, 2.3975e-03, 1.1339e-03, 9.0179e-03, 4.2381e-03,
          8.9884e-04, 1.2426e-03, 1.8015e-03, 8.7128e-03, 6.5041e-03, 1.2560e-03,
          3.1967e-03, 3.1643e-03, 2.6608e-03, 1.2993e-02, 2.9266e-02, 3.2013e-02,
          8.8425e-03, 5.8899e-03, 6.3782e-03, 6.3438e-03, 9.5062e-03, 2.4281e-03,
          1.4601e-03, 1.8520e-03, 1.3247e-03, 2.2621e-03, 2.1381e-03, 5.3673e-03,
          7.2098e-04, 1.2751e-03, 1.6346e-03, 2.9774e-03, 3.4790e-03, 5.3368e-03,
          7.2708e-03, 2.9392e-03, 1.7120e-02, 3.2318e-02, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.2829e-05, 2.2829e-05, 4.6659e-04, 2.3365e-04, 2.2829e-05,
          8.6069e-05, 8.6069e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.2829e-05, 5.6267e-04, 4.7755e-04, 2.2829e-05, 8.6069e-05,
          2.3365e-04, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.2829e-05, 8.6069e-05, 5.5504e-04, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.5582e-04, 2.3365e-04, 8.6069e-05, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05,
          3.1900e-04, 2.2829e-05, 3.2997e-04, 2.2829e-05, 2.2829e-05, 2.2829e-05,
          2.2829e-05, 8.6069e-05, 5.6267e-04, 5.6267e-04, 2.2829e-05, 2.2829e-05,
          3.1900e-04, 5.6267e-04, 8.6069e-05, 2.2829e-05, 2.2829e-05, 2.3365e-04,
          2.2829e-05, 8.6069e-05, 2.3365e-04, 2.2829e-05, 2.2829e-05, 8.6069e-05,
          8.6069e-05, 2.2829e-05, 2.5582e-04, 3.2997e-04, 8.6069e-05, 4.7755e-04,
          2.2829e-05, 2.2829e-05, 2.3365e-04, 3.1900e-04, 2.2829e-05, 5.6267e-04,
          1.7023e-03, 1.3733e-03, 2.2829e-05, 2.2829e-05, 2.3365e-04, 2.3365e-04,
          2.2829e-05, 3.1900e-04, 2.2829e-05, 2.2829e-05, 2.3365e-04, 2.2829e-05,
          2.3365e-04, 2.3365e-04, 2.2829e-05, 8.6069e-05, 8.6069e-05, 2.2829e-05,
          2.2829e-05, 5.6267e-04, 2.2829e-05, 8.6069e-05, 2.2829e-05, 2.2829e-05,
          8.6069e-05, 2.2829e-05, 2.2829e-05, 2.2829e-05, 2.3365e-04, 8.6069e-05,
          8.6069e-05, 2.3365e-04, 8.6069e-05, 8.6069e-05, 2.2829e-05, 5.6267e-04,
          8.6069e-05, 3.2997e-04, 7.1049e-04, 2.5582e-04, 8.6069e-05, 3.1900e-04,
          1.7023e-03, 3.1900e-04, 2.2829e-05, 8.6069e-05, 3.2997e-04, 2.2829e-05,
          8.6069e-05, 4.0722e-04, 4.0722e-04, 8.6069e-05, 4.0722e-04, 8.6069e-05,
          8.6069e-05, 8.6069e-05, 4.0722e-04, 3.3712e-04, 2.2829e-05, 4.0722e-04,
          8.6069e-05, 8.6069e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.2452e-03, 4.3549e-02, 1.6922e-02, 1.2428e-02, 6.5186e-02,
          2.8030e-02, 1.2650e-02, 1.0700e-03, 1.1015e-03, 1.5011e-03, 4.0703e-03,
          1.2102e-03, 8.5735e-04, 5.2118e-04, 2.0332e-03, 6.9714e-04, 7.1096e-04,
          3.9554e-04, 4.1771e-04, 4.2439e-04, 4.9782e-04, 3.1996e-04, 4.9877e-04,
          3.5934e-03, 2.5864e-03, 8.8196e-03, 1.5650e-03, 8.7881e-04, 7.7868e-04,
          5.8603e-04, 1.3638e-03, 3.6359e-04, 1.4639e-03, 8.1730e-04, 7.7438e-04,
          2.3365e-03, 1.9150e-03, 7.8821e-04, 8.7881e-04, 4.4250e-04, 9.6369e-04,
          1.2083e-03, 8.8358e-04, 1.0967e-03, 7.0953e-04, 6.8855e-04, 1.6108e-03,
          2.7347e-04, 4.3941e-04, 5.9395e-03, 1.7633e-03, 1.1568e-03, 8.7738e-04,
          1.1148e-03, 9.6321e-04, 1.7319e-03, 2.3041e-03, 2.1725e-03, 3.4828e-03,
          3.9711e-03, 7.1001e-04, 1.1187e-03, 8.4686e-04, 1.1997e-03, 2.4128e-03,
          9.4414e-04, 4.5919e-04, 4.3583e-04, 7.3671e-04, 8.0109e-04, 9.1124e-04,
          9.3794e-04, 2.1420e-03, 1.6470e-03, 1.5087e-03, 8.5735e-04, 6.0701e-04,
          1.4801e-03, 2.9297e-03, 1.7977e-03, 1.7900e-03, 1.6317e-03, 1.9779e-03,
          1.9035e-03, 1.4601e-03, 8.5115e-04, 9.9754e-04, 4.2677e-04, 6.3562e-04,
          1.7757e-03, 9.6226e-04, 1.1444e-03, 1.1568e-03, 2.8324e-03, 1.6203e-03,
          8.6212e-04, 1.0347e-03, 6.4611e-04, 2.0485e-03, 6.5575e-03, 2.9526e-03,
          3.3855e-03, 7.4625e-04, 8.6403e-04, 2.8172e-03, 2.7084e-03, 9.5081e-04,
          1.2732e-03, 1.0633e-03, 1.4372e-03, 8.8978e-04, 9.7036e-04, 5.7316e-04,
          2.0142e-03, 1.6298e-03, 4.6849e-04, 3.0632e-03, 2.3460e-03, 2.3823e-03,
          1.9331e-03, 4.3831e-03, 4.7340e-03, 6.2418e-04, 3.3073e-03, 1.3714e-03,
          4.0398e-03, 1.6613e-03, 2.4147e-03, 2.8820e-03, 7.0763e-04, 1.3456e-03,
          1.3161e-03, 5.8031e-04, 6.0177e-04, 3.0065e-04, 8.7118e-04, 9.7656e-04,
          1.1311e-03, 5.4836e-04, 8.3351e-04, 1.4114e-03, 1.2445e-03, 1.4734e-03,
          5.6839e-03, 3.1376e-03, 2.3994e-03, 2.7771e-03, 2.3079e-03, 6.6795e-03,
          4.0054e-03, 3.4618e-03, 6.6071e-03, 1.0557e-03, 1.8749e-03, 1.5974e-03,
          7.9250e-04, 8.0919e-04, 5.0116e-04, 9.3269e-04, 2.2278e-03, 6.0511e-04,
          2.0237e-03, 8.2207e-04, 9.3508e-04, 1.5202e-03, 6.0616e-03, 2.1381e-03,
          2.9373e-03, 1.5802e-03, 3.3722e-03, 1.9760e-03, 9.8610e-04, 2.8858e-03,
          6.4087e-03, 5.8670e-03, 7.6332e-03, 2.0370e-03, 1.8492e-03, 1.1301e-03,
          8.6927e-04, 1.0548e-03, 1.6241e-03, 1.0223e-03, 6.8474e-04, 1.2245e-03,
          1.4563e-03, 7.7486e-04, 1.6518e-03, 1.1230e-02, 3.2673e-03, 2.3804e-03,
          1.3342e-03, 8.2779e-03, 1.8072e-03, 1.5535e-03, 1.4744e-03, 2.1629e-03,
          8.1253e-03, 2.9278e-03, 2.4033e-03, 1.0462e-03, 6.9666e-04, 9.8896e-04,
          1.3332e-03, 1.2274e-03, 2.1133e-03, 7.5436e-04, 2.0542e-03, 5.3978e-04,
          9.0218e-04, 6.0234e-03, 1.8082e-02, 2.4910e-03, 5.1003e-03, 6.6986e-03,
          3.5191e-03, 1.9140e-03, 1.4877e-03, 1.5125e-03, 2.6875e-03, 1.9646e-03,
          2.6836e-03, 4.2534e-03, 2.0142e-03, 8.0204e-04, 3.8033e-03, 1.5097e-03,
          1.0271e-03, 1.6479e-03, 3.5210e-03, 5.9891e-03, 2.3193e-03, 9.2936e-04,
          1.6785e-02, 3.3035e-03, 7.4654e-03, 4.2305e-03, 2.5955e-02, 4.0558e-02,
          3.8818e-02, 2.6550e-02, 8.2626e-03, 4.1504e-03, 4.1046e-03, 4.2038e-03,
          3.2692e-03, 2.1267e-03, 4.7493e-03, 3.6564e-03, 4.4708e-03, 2.3155e-03,
          1.3781e-03, 3.5763e-03, 4.9171e-03, 2.5215e-03, 1.2712e-03, 2.3861e-03,
          1.1749e-02, 3.2654e-03, 1.0307e-02, 6.9847e-03, 7.8613e-02, 3.6955e-05,
          3.6955e-05, 3.6955e-05, 8.0526e-05, 1.7905e-04, 2.5225e-04, 3.6955e-05,
          1.1009e-04, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 8.0526e-05, 9.8991e-04, 1.1009e-04, 3.6955e-05, 1.1009e-04,
          1.7905e-04, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 3.6955e-05, 2.9588e-04, 6.1846e-04, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 4.0317e-04, 2.5225e-04, 8.0526e-05, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 3.6955e-05, 3.6955e-05, 8.0526e-05, 8.0526e-05, 3.6955e-05,
          1.5366e-04, 3.6955e-05, 8.0526e-05, 3.6955e-05, 3.6955e-05, 3.6955e-05,
          3.6955e-05, 1.1009e-04, 4.7636e-04, 4.7636e-04, 3.6955e-05, 3.6955e-05,
          4.7636e-04, 3.6955e-05, 1.5366e-04, 3.6955e-05, 3.6955e-05, 2.5225e-04,
          3.6955e-05, 8.0526e-05, 2.5225e-04, 3.6955e-05, 8.0526e-05, 8.0526e-05,
          3.6955e-05, 8.0526e-05, 4.0317e-04, 1.1009e-04, 1.1009e-04, 2.5225e-04,
          8.0526e-05, 3.6955e-05, 2.5225e-04, 1.5366e-04, 3.6955e-05, 4.7636e-04,
          9.8991e-04, 1.1009e-04, 3.6955e-05, 3.6955e-05, 5.2547e-04, 2.5225e-04,
          8.0526e-05, 1.5366e-04, 1.5366e-04, 8.0526e-05, 2.5225e-04, 3.6955e-05,
          5.2547e-04, 2.5225e-04, 1.1009e-04, 8.0526e-05, 1.1009e-04, 3.6955e-05,
          1.1009e-04, 1.1009e-04, 3.6955e-05, 3.6955e-05, 3.6955e-05, 8.0526e-05,
          1.1009e-04, 3.6955e-05, 3.6955e-05, 8.0526e-05, 2.9588e-04, 8.0526e-05,
          3.8314e-04, 2.5225e-04, 1.1009e-04, 8.0526e-05, 3.6955e-05, 7.5293e-04,
          3.8314e-04, 4.3273e-04, 2.5225e-04, 8.0526e-05, 1.1009e-04, 7.5293e-04,
          9.8991e-04, 1.0262e-03, 8.0526e-05, 3.6955e-05, 1.1009e-04, 3.6955e-05,
          1.1009e-04, 3.8314e-04, 1.1009e-04, 4.8161e-04, 3.8314e-04, 1.1009e-04,
          8.0526e-05, 1.1009e-04, 1.1009e-04, 4.0007e-04, 3.6955e-05, 3.8314e-04,
          3.6955e-05, 1.1009e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.6921e-03, 4.3610e-02, 1.4931e-02, 1.4114e-02, 4.8523e-02,
          3.1021e-02, 1.1650e-02, 7.9775e-04, 7.8726e-04, 1.9798e-03, 5.2757e-03,
          1.6346e-03, 7.0620e-04, 8.2922e-04, 1.1711e-03, 5.4455e-04, 8.1587e-04,
          5.1451e-04, 4.0483e-04, 4.1890e-04, 3.3593e-04, 3.5954e-04, 6.1369e-04,
          4.9591e-03, 1.9445e-03, 8.8043e-03, 1.8806e-03, 1.0157e-03, 8.8453e-04,
          1.0014e-03, 1.3161e-03, 4.7135e-04, 1.2331e-03, 7.8392e-04, 6.8378e-04,
          1.8501e-03, 1.5202e-03, 7.1907e-04, 7.5626e-04, 5.9223e-04, 1.0128e-03,
          1.4448e-03, 8.8215e-04, 1.3571e-03, 6.1178e-04, 6.6328e-04, 1.1110e-03,
          4.4823e-04, 4.5395e-04, 4.0550e-03, 1.7157e-03, 1.2226e-03, 7.2432e-04,
          9.5844e-04, 9.8038e-04, 1.6928e-03, 2.2697e-03, 2.5711e-03, 3.4752e-03,
          3.3398e-03, 9.4700e-04, 1.1387e-03, 6.7186e-04, 1.9970e-03, 1.7071e-03,
          2.0790e-03, 1.0815e-03, 5.6553e-04, 5.6648e-04, 6.6328e-04, 1.9417e-03,
          9.9754e-04, 2.5959e-03, 1.6403e-03, 1.8826e-03, 8.5926e-04, 6.4516e-04,
          1.6928e-03, 3.5248e-03, 2.9869e-03, 2.0123e-03, 2.4700e-03, 1.8911e-03,
          1.7538e-03, 1.5421e-03, 9.7227e-04, 1.6346e-03, 9.5510e-04, 1.4105e-03,
          3.1204e-03, 1.4467e-03, 7.7152e-04, 1.8339e-03, 8.8596e-04, 9.6941e-04,
          8.9979e-04, 9.5844e-04, 7.6580e-04, 1.7052e-03, 5.0964e-03, 3.1509e-03,
          4.5776e-03, 1.4086e-03, 1.2789e-03, 2.4986e-03, 1.9627e-03, 1.1225e-03,
          1.4830e-03, 1.0233e-03, 1.9875e-03, 1.1110e-03, 2.1782e-03, 8.3590e-04,
          1.3533e-03, 9.2220e-04, 4.7779e-04, 1.8721e-03, 1.2360e-03, 2.1935e-03,
          2.3937e-03, 3.5419e-03, 3.4657e-03, 6.1798e-04, 2.0351e-03, 1.8253e-03,
          3.2558e-03, 2.9449e-03, 3.7842e-03, 2.3289e-03, 7.7438e-04, 1.5965e-03,
          2.6817e-03, 8.8644e-04, 3.7360e-04, 4.2439e-04, 1.3790e-03, 7.9298e-04,
          1.0376e-03, 8.1682e-04, 7.5626e-04, 1.5497e-03, 1.2455e-03, 2.1095e-03,
          5.2681e-03, 1.5049e-03, 1.9627e-03, 2.9182e-03, 4.2648e-03, 4.1962e-03,
          4.4327e-03, 2.8915e-03, 3.8090e-03, 1.1082e-03, 1.9932e-03, 1.4629e-03,
          1.2999e-03, 1.0281e-03, 6.3705e-04, 1.3552e-03, 1.8921e-03, 7.6675e-04,
          1.5459e-03, 8.7023e-04, 8.8739e-04, 1.2550e-03, 5.5504e-03, 2.4242e-03,
          3.6640e-03, 1.8177e-03, 2.4452e-03, 1.1845e-03, 1.3542e-03, 2.6226e-03,
          3.4924e-03, 5.3329e-03, 4.7874e-03, 1.7490e-03, 1.6336e-03, 9.2936e-04,
          1.0004e-03, 9.3603e-04, 1.2875e-03, 9.4986e-04, 6.0511e-04, 7.2813e-04,
          1.8158e-03, 8.7976e-04, 1.0881e-03, 1.0216e-02, 5.5199e-03, 1.8349e-03,
          1.1978e-03, 6.2599e-03, 1.7347e-03, 1.5869e-03, 1.9026e-03, 2.0046e-03,
          6.7253e-03, 2.8553e-03, 2.0275e-03, 1.3161e-03, 1.0252e-03, 9.8228e-04,
          1.8091e-03, 1.4658e-03, 1.4200e-03, 5.8079e-04, 2.8725e-03, 5.5456e-04,
          9.0170e-04, 5.4512e-03, 1.5160e-02, 3.4218e-03, 4.1466e-03, 8.1558e-03,
          2.9240e-03, 2.4109e-03, 1.4467e-03, 9.3460e-04, 2.4662e-03, 2.4185e-03,
          2.6417e-03, 3.1471e-03, 1.8530e-03, 1.1244e-03, 2.5692e-03, 1.5640e-03,
          1.2856e-03, 2.6741e-03, 2.2984e-03, 3.8052e-03, 1.6193e-03, 8.9407e-04,
          1.5427e-02, 4.1618e-03, 9.6893e-03, 2.6016e-03, 1.5900e-02, 7.7332e-02,
          5.7404e-02, 2.7634e-02, 1.0269e-02, 3.8853e-03, 4.4670e-03, 5.3711e-03,
          2.8877e-03, 2.3899e-03, 3.3550e-03, 2.8782e-03, 3.2082e-03, 1.6317e-03,
          1.5640e-03, 2.2430e-03, 2.7294e-03, 1.5917e-03, 1.1387e-03, 2.3499e-03,
          1.0376e-02, 3.8052e-03, 6.2408e-03, 4.5586e-03, 1.5572e-02, 6.7749e-02,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 1.3995e-04, 1.3995e-04, 3.3557e-05,
          1.0425e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0425e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.2234e-04, 3.2234e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.2234e-04, 1.0425e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0425e-04,
          3.3557e-05, 2.1601e-04, 1.0425e-04, 2.1052e-04, 3.3557e-05, 3.3557e-05,
          1.0242e-03, 3.3557e-05, 2.1601e-04, 3.3557e-05, 3.3557e-05, 1.3995e-04,
          3.3557e-05, 1.3995e-04, 1.0425e-04, 1.3995e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0425e-04, 3.3557e-05, 2.1052e-04,
          3.3557e-05, 3.3557e-05, 3.2234e-04, 3.3557e-05, 3.3557e-05, 9.1791e-04,
          1.0425e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.2234e-04, 3.2234e-04,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 1.9324e-04, 1.0425e-04, 3.3557e-05,
          4.8184e-04, 4.8184e-04, 3.3557e-05, 3.3557e-05, 3.7551e-04, 3.3557e-05,
          3.3557e-05, 1.0425e-04, 3.3557e-05, 1.0425e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.2234e-04, 3.3557e-05,
          2.1601e-04, 2.1052e-04, 2.1601e-04, 3.3557e-05, 3.3557e-05, 2.6393e-04,
          3.7551e-04, 1.0425e-04, 2.1052e-04, 3.3557e-05, 3.3557e-05, 1.1902e-03,
          1.0425e-04, 3.7551e-04, 3.3557e-05, 1.0425e-04, 3.3557e-05, 3.3557e-05,
          2.1601e-04, 2.1601e-04, 2.1601e-04, 1.0055e-04, 1.0425e-04, 1.0425e-04,
          3.3557e-05, 1.0425e-04, 2.1601e-04, 3.9458e-04, 2.1601e-04, 2.1601e-04,
          3.3557e-05, 1.0425e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.8166e-03, 5.1697e-02, 1.1681e-02, 1.1482e-02, 4.1992e-02,
          4.4312e-02, 8.0032e-03, 1.2636e-03, 9.6893e-04, 2.3499e-03, 2.9163e-03,
          1.3514e-03, 1.5621e-03, 5.5361e-04, 3.4409e-03, 1.0118e-03, 1.8415e-03,
          8.4400e-04, 5.0879e-04, 2.6822e-04, 3.4237e-04, 2.8634e-04, 1.1635e-03,
          3.3569e-03, 4.0741e-03, 6.4926e-03, 3.1910e-03, 5.7888e-04, 1.9255e-03,
          7.8583e-04, 2.5425e-03, 6.4754e-04, 5.8022e-03, 1.8787e-03, 2.3727e-03,
          4.4479e-03, 9.4652e-04, 2.4509e-03, 3.4904e-03, 3.6025e-04, 1.7281e-03,
          6.2275e-04, 5.3787e-04, 4.9829e-04, 5.5075e-04, 1.4257e-03, 5.3177e-03,
          3.1900e-04, 3.7313e-04, 2.6989e-03, 3.0823e-03, 1.7948e-03, 1.6785e-03,
          5.5742e-04, 1.0948e-03, 9.2936e-04, 1.3771e-03, 1.8358e-03, 1.5411e-03,
          1.6479e-03, 9.5844e-04, 1.1768e-03, 1.9302e-03, 8.6689e-04, 7.2060e-03,
          8.0061e-04, 5.2643e-04, 9.3031e-04, 4.0340e-04, 3.5262e-04, 4.7779e-04,
          5.5504e-04, 4.7035e-03, 8.8453e-04, 2.2564e-03, 4.4942e-04, 4.9877e-04,
          1.0738e-03, 1.6088e-03, 9.5272e-04, 3.5591e-03, 8.9216e-04, 1.2856e-03,
          1.1635e-03, 3.1967e-03, 7.3862e-04, 1.3981e-03, 4.9257e-04, 4.9353e-04,
          2.4090e-03, 1.8530e-03, 5.0306e-04, 1.5287e-03, 9.1076e-04, 7.2098e-04,
          7.4339e-04, 5.2881e-04, 3.7651e-03, 4.2725e-03, 1.5383e-03, 2.2812e-03,
          3.9940e-03, 6.7806e-04, 7.4577e-04, 3.8872e-03, 7.5760e-03, 1.2579e-03,
          2.3880e-03, 7.6914e-04, 2.6569e-03, 5.5742e-04, 7.1716e-04, 6.9141e-04,
          2.6588e-03, 1.7347e-03, 4.7779e-04, 8.6670e-03, 9.1124e-04, 1.4629e-03,
          2.0962e-03, 7.8735e-03, 8.5449e-04, 7.1335e-04, 5.5618e-03, 6.5279e-04,
          1.3075e-03, 7.0000e-04, 1.1368e-03, 1.1131e-02, 2.2488e-03, 2.7218e-03,
          1.0557e-03, 5.0497e-04, 6.9618e-04, 4.0483e-04, 1.4563e-03, 4.8804e-04,
          5.3024e-03, 7.9632e-04, 1.3113e-03, 2.3460e-03, 3.2368e-03, 2.3956e-03,
          1.5459e-03, 1.3332e-03, 2.1210e-03, 1.8253e-03, 1.2026e-03, 1.2199e-02,
          8.1253e-03, 9.3689e-03, 2.9449e-03, 3.8719e-03, 1.8120e-03, 2.6741e-03,
          2.0447e-03, 1.7376e-03, 6.7377e-04, 5.6458e-04, 6.7749e-03, 1.9503e-03,
          5.3940e-03, 3.3627e-03, 2.8915e-03, 3.6736e-03, 4.0283e-03, 2.3136e-03,
          3.2101e-03, 2.4815e-03, 6.6910e-03, 4.1542e-03, 2.5730e-03, 8.3160e-03,
          6.8588e-03, 4.0016e-03, 3.9520e-03, 1.5612e-03, 5.0964e-03, 1.8759e-03,
          1.7967e-03, 3.0460e-03, 4.7379e-03, 2.4643e-03, 1.0366e-03, 1.9798e-03,
          1.9274e-03, 1.2608e-03, 1.3762e-03, 2.8839e-03, 1.6193e-03, 5.6686e-03,
          2.6779e-03, 1.4296e-03, 6.9885e-03, 4.1962e-03, 2.7676e-03, 6.4659e-03,
          4.3983e-03, 2.3861e-03, 7.5417e-03, 1.1539e-03, 1.1263e-03, 2.2297e-03,
          7.8392e-04, 2.0523e-03, 1.1368e-03, 8.7357e-04, 3.2692e-03, 1.4687e-03,
          1.0767e-03, 5.1422e-03, 2.4586e-03, 1.8921e-03, 1.2238e-02, 1.9007e-03,
          7.9880e-03, 4.7417e-03, 2.6855e-03, 2.2907e-03, 1.1047e-02, 3.3188e-03,
          9.0714e-03, 1.2655e-03, 2.6932e-03, 1.0824e-03, 8.3160e-03, 4.4022e-03,
          7.3957e-04, 9.9373e-04, 1.2808e-03, 5.3673e-03, 6.8207e-03, 1.4772e-03,
          3.8204e-03, 2.6779e-03, 2.1629e-03, 6.8283e-03, 1.9180e-02, 3.6072e-02,
          1.0796e-02, 7.9727e-03, 6.6223e-03, 7.0763e-03, 1.0933e-02, 3.6182e-03,
          2.1381e-03, 1.7672e-03, 1.3533e-03, 1.9274e-03, 2.1687e-03, 6.2180e-03,
          6.2609e-04, 1.0071e-03, 1.6136e-03, 3.4695e-03, 3.8128e-03, 8.0032e-03,
          6.7215e-03, 2.4986e-03, 9.7656e-03, 1.4259e-02, 5.6915e-03, 1.3191e-02,
          2.7893e-02, 1.9729e-05, 1.9729e-05, 1.9729e-05, 2.8992e-04, 1.9729e-05,
          1.1033e-04, 1.1033e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 2.7251e-04, 2.8992e-04, 1.9729e-05, 1.1033e-04,
          2.8992e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 1.1033e-04, 5.8126e-04, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 2.8992e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 4.0150e-04, 1.6756e-03, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.4229e-03, 2.7251e-04, 1.1033e-04, 1.9729e-05, 1.9729e-05, 2.8992e-04,
          1.9729e-05, 1.9729e-05, 2.8992e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.9729e-05, 1.9729e-05, 1.9729e-05, 1.1033e-04, 1.1033e-04, 5.4264e-04,
          1.9729e-05, 1.9729e-05, 1.1033e-04, 1.1033e-04, 1.9729e-05, 3.6311e-04,
          3.6311e-04, 1.1033e-04, 1.9729e-05, 1.9729e-05, 2.8992e-04, 2.8992e-04,
          1.9729e-05, 1.1033e-04, 1.9729e-05, 1.9729e-05, 2.8992e-04, 1.9729e-05,
          5.8126e-04, 2.8992e-04, 1.9729e-05, 1.9729e-05, 1.1033e-04, 1.9729e-05,
          1.9729e-05, 3.6311e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          1.1033e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05, 2.8992e-04, 1.9729e-05,
          1.1033e-04, 2.8992e-04, 4.0150e-04, 1.1033e-04, 1.9729e-05, 3.6311e-04,
          1.1033e-04, 3.6311e-04, 5.4264e-04, 1.9729e-05, 1.1033e-04, 1.1033e-04,
          3.6311e-04, 5.0354e-04, 1.9729e-05, 1.9729e-05, 1.9729e-05, 1.9729e-05,
          4.0150e-04, 4.0150e-04, 4.0150e-04, 1.1033e-04, 1.1033e-04, 1.9729e-05,
          1.9729e-05, 1.1033e-04, 1.1033e-04, 1.1033e-04, 1.9729e-05, 7.0095e-04,
          4.0150e-04, 1.1033e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4975e-03, 6.0822e-02, 1.2878e-02, 1.3901e-02, 3.2166e-02,
          5.0354e-02, 9.4376e-03, 1.7548e-03, 8.5497e-04, 6.4926e-03, 9.3994e-03,
          2.8172e-03, 1.9283e-03, 7.2670e-04, 4.6158e-03, 1.5621e-03, 2.1229e-03,
          1.2712e-03, 9.8801e-04, 2.0754e-04, 2.6488e-04, 2.7514e-04, 2.4414e-03,
          8.6441e-03, 2.6169e-03, 7.5912e-03, 5.3902e-03, 4.3941e-04, 2.5921e-03,
          1.0910e-03, 4.2915e-03, 7.6580e-04, 4.8866e-03, 8.6899e-03, 1.1215e-03,
          2.2945e-03, 5.6410e-04, 4.0970e-03, 5.6152e-03, 3.1543e-04, 1.5144e-03,
          4.6873e-04, 4.0984e-04, 4.1485e-04, 3.7003e-04, 2.6875e-03, 3.6526e-03,
          3.6812e-04, 3.3951e-04, 2.5692e-03, 1.7624e-03, 2.7027e-03, 1.7090e-03,
          4.9877e-04, 1.5030e-03, 5.7459e-04, 6.4802e-04, 1.6575e-03, 9.4175e-04,
          9.6273e-04, 8.9788e-04, 8.3017e-04, 1.3323e-03, 6.3229e-04, 6.3553e-03,
          1.5182e-03, 1.1835e-03, 1.7366e-03, 2.4307e-04, 2.2876e-04, 3.4475e-04,
          3.6526e-04, 1.5007e-02, 6.3848e-04, 3.8719e-03, 4.8089e-04, 5.2834e-04,
          1.5469e-03, 6.2704e-04, 9.2173e-04, 3.8147e-03, 1.1463e-03, 6.2943e-04,
          5.4216e-04, 3.2902e-03, 5.3740e-04, 5.2528e-03, 1.0290e-03, 6.8998e-04,
          7.2212e-03, 3.8567e-03, 3.1590e-04, 3.6564e-03, 3.2830e-04, 3.5453e-04,
          6.8903e-04, 3.8695e-04, 2.4719e-03, 1.6260e-03, 5.9891e-04, 9.4318e-04,
          7.9575e-03, 1.6994e-03, 1.4639e-03, 1.8320e-03, 6.0005e-03, 7.6437e-04,
          2.8934e-03, 4.3797e-04, 3.7441e-03, 4.7135e-04, 1.2922e-03, 1.7023e-03,
          2.9125e-03, 6.0558e-04, 4.0722e-04, 5.1727e-03, 4.8137e-04, 9.3079e-04,
          1.3208e-03, 3.3054e-03, 4.3511e-04, 1.5154e-03, 1.1078e-02, 4.7064e-04,
          4.7398e-04, 5.5170e-04, 8.7261e-04, 6.2866e-03, 4.8714e-03, 4.2572e-03,
          2.6722e-03, 1.4534e-03, 4.6501e-03, 6.7902e-04, 2.3346e-03, 3.4690e-04,
          6.0997e-03, 7.5960e-04, 3.5534e-03, 7.2021e-03, 7.6714e-03, 1.9627e-03,
          6.3992e-04, 3.7651e-03, 1.0948e-03, 7.7057e-04, 1.2674e-03, 5.1041e-03,
          8.3237e-03, 5.0430e-03, 9.6941e-04, 1.8768e-03, 1.4172e-03, 1.0662e-03,
          3.1338e-03, 1.1559e-03, 2.9259e-03, 4.4084e-04, 2.5234e-03, 1.2512e-03,
          5.0888e-03, 6.4507e-03, 5.4398e-03, 1.8494e-02, 1.0767e-03, 1.2045e-03,
          5.6229e-03, 1.9760e-03, 4.0169e-03, 6.4621e-03, 2.8305e-03, 5.1155e-03,
          2.3365e-03, 1.3218e-03, 1.4572e-03, 8.1301e-04, 3.2444e-03, 8.6899e-03,
          3.0193e-03, 3.5858e-03, 7.7133e-03, 5.5580e-03, 3.4237e-03, 7.5340e-03,
          7.8735e-03, 1.4095e-03, 1.5610e-02, 9.5558e-04, 1.2093e-03, 8.8577e-03,
          6.8588e-03, 5.1689e-04, 6.6261e-03, 6.4201e-03, 9.9182e-03, 5.4054e-03,
          1.3971e-03, 9.2173e-04, 4.8676e-03, 1.2789e-03, 5.3596e-03, 2.9755e-03,
          6.3133e-04, 1.6613e-03, 5.3167e-04, 2.9678e-03, 7.1754e-03, 2.0962e-03,
          2.2831e-03, 1.6785e-03, 7.7200e-04, 2.5864e-03, 7.5226e-03, 9.4128e-04,
          2.8019e-03, 2.7428e-03, 2.9648e-02, 8.4305e-03, 9.1934e-03, 2.6588e-03,
          5.8250e-03, 5.3501e-04, 1.3742e-03, 3.1223e-03, 1.6266e-02, 9.9792e-03,
          1.6575e-03, 6.2609e-04, 4.8089e-04, 1.3485e-03, 2.8706e-03, 1.1320e-03,
          1.2999e-03, 1.6031e-03, 1.1072e-03, 6.0158e-03, 2.9964e-03, 5.0659e-03,
          1.9102e-03, 1.4496e-03, 1.1139e-03, 2.9106e-03, 3.7937e-03, 1.4734e-03,
          7.8297e-04, 1.8988e-03, 4.9019e-04, 8.0776e-04, 8.4114e-04, 3.3398e-03,
          5.0879e-04, 3.7456e-04, 6.4516e-04, 1.2636e-03, 9.4299e-03, 9.4757e-03,
          1.4229e-03, 2.6188e-03, 2.2430e-03, 3.0594e-03, 8.9359e-04, 3.0003e-03,
          3.9253e-03, 2.8336e-02, 2.1935e-05, 2.2101e-04, 3.2926e-04, 2.1935e-05,
          1.3018e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.6107e-04, 2.1935e-05, 3.2926e-04, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.2101e-04, 2.1935e-05, 1.3018e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.1935e-05, 1.3018e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          1.3018e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 1.3018e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          1.3018e-04, 2.1935e-05, 1.3018e-04, 2.1935e-05, 2.1935e-05, 3.2926e-04,
          2.1935e-05, 2.1935e-05, 3.2926e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05, 3.2926e-04,
          2.1935e-05, 2.1935e-05, 1.3018e-04, 2.1935e-05, 2.6107e-04, 1.3018e-04,
          9.9945e-04, 8.6021e-04, 2.1935e-05, 2.1935e-05, 1.3018e-04, 1.3018e-04,
          2.6107e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05, 3.2926e-04, 2.6107e-04,
          1.3018e-04, 1.3018e-04, 2.6107e-04, 2.1935e-05, 1.3018e-04, 2.1935e-05,
          2.1935e-05, 1.3018e-04, 2.1935e-05, 3.6931e-04, 2.1935e-05, 2.1935e-05,
          6.6090e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05, 3.2926e-04, 2.1935e-05,
          6.6090e-04, 3.2926e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05, 1.3018e-04,
          3.6931e-04, 5.5265e-04, 3.2926e-04, 2.1935e-05, 2.1935e-05, 2.6107e-04,
          6.6090e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05, 2.1935e-05, 2.1935e-05,
          1.3018e-04, 1.3018e-04, 1.3018e-04, 2.1935e-05, 1.3018e-04, 2.1935e-05,
          2.1935e-05, 1.3018e-04, 1.3018e-04, 2.1935e-05, 2.1935e-05, 1.3018e-04,
          2.1935e-05, 1.3018e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.0474e-03, 5.1819e-02, 1.3885e-02, 1.2238e-02, 3.6926e-02,
          3.5461e-02, 8.8425e-03, 2.8973e-03, 1.6804e-03, 3.1376e-03, 6.1493e-03,
          2.0695e-03, 7.2289e-03, 2.2888e-03, 3.5324e-03, 2.7370e-03, 8.5907e-03,
          4.9324e-03, 1.5383e-03, 3.5667e-04, 3.1638e-04, 4.0841e-04, 4.4556e-03,
          6.3591e-03, 2.1000e-03, 7.2250e-03, 4.1351e-03, 7.0477e-04, 7.2136e-03,
          3.7327e-03, 6.9466e-03, 1.6823e-03, 3.4599e-03, 3.2215e-03, 2.2907e-03,
          1.3199e-03, 4.3464e-04, 1.6251e-03, 3.2215e-03, 6.2799e-04, 1.0099e-03,
          4.2796e-04, 3.7146e-04, 4.0078e-04, 3.5834e-04, 9.3994e-03, 2.4872e-03,
          9.0694e-04, 6.5279e-04, 2.7847e-03, 2.5921e-03, 4.7417e-03, 2.4147e-03,
          6.9189e-04, 2.6035e-03, 5.7077e-04, 5.1737e-04, 1.8854e-03, 7.6103e-04,
          8.5735e-04, 1.7586e-03, 9.0742e-04, 1.0929e-03, 6.2370e-04, 3.5610e-03,
          1.1892e-03, 9.3174e-04, 8.4496e-04, 2.3365e-04, 1.9741e-04, 3.4070e-04,
          3.6001e-04, 6.1531e-03, 7.5579e-04, 7.6141e-03, 6.6090e-04, 7.2575e-04,
          2.6627e-03, 4.9829e-04, 8.5974e-04, 1.8349e-03, 9.3317e-04, 4.9210e-04,
          4.0269e-04, 5.8784e-03, 7.2670e-04, 3.1281e-03, 6.9904e-04, 6.0272e-04,
          5.3253e-03, 1.8234e-03, 4.0078e-04, 2.6646e-03, 3.0947e-04, 3.1829e-04,
          1.1225e-03, 4.1151e-04, 4.7073e-03, 2.3632e-03, 4.6873e-04, 7.3957e-04,
          6.4201e-03, 1.4248e-03, 1.0805e-03, 9.3508e-04, 4.3526e-03, 1.7681e-03,
          5.1079e-03, 4.2510e-04, 2.1515e-03, 4.6086e-04, 9.6655e-04, 9.3842e-04,
          2.0828e-03, 5.2023e-04, 7.7009e-04, 4.5319e-03, 4.2725e-04, 1.8215e-03,
          5.6572e-03, 1.9932e-03, 3.5143e-04, 2.9011e-03, 4.7913e-03, 4.4060e-04,
          3.6311e-04, 4.9829e-04, 8.3113e-04, 5.4550e-03, 6.4583e-03, 8.8196e-03,
          1.5907e-03, 9.0313e-04, 2.1591e-03, 6.1989e-04, 1.4067e-03, 3.5167e-04,
          1.2299e-02, 3.1071e-03, 1.5717e-02, 7.5531e-03, 5.3825e-03, 7.7629e-03,
          5.5695e-04, 2.7752e-03, 7.6151e-04, 7.0381e-04, 1.1196e-03, 3.7384e-03,
          4.0398e-03, 3.4752e-03, 8.5688e-04, 1.9207e-03, 1.8072e-03, 8.6069e-04,
          1.5564e-03, 8.0347e-04, 3.1433e-03, 5.5790e-04, 2.1496e-03, 6.9084e-03,
          3.5343e-03, 2.7294e-03, 1.5465e-02, 6.1378e-03, 8.8263e-04, 9.8419e-04,
          5.2452e-03, 1.3161e-03, 2.3975e-03, 4.6654e-03, 6.0387e-03, 3.0518e-03,
          2.9678e-03, 2.2202e-03, 1.0586e-03, 7.3910e-04, 1.6699e-03, 8.4915e-03,
          1.4572e-03, 2.2869e-03, 3.3302e-03, 9.0485e-03, 7.3204e-03, 6.2103e-03,
          7.2060e-03, 9.1476e-03, 4.7760e-03, 7.8487e-04, 1.2064e-03, 5.1537e-03,
          1.2405e-02, 4.1509e-04, 3.8261e-03, 1.6479e-02, 5.7983e-03, 3.1891e-03,
          1.4162e-03, 1.5411e-03, 3.1281e-03, 9.0885e-04, 2.1343e-03, 1.6785e-03,
          6.3658e-04, 1.1902e-03, 4.8923e-04, 5.1270e-03, 5.8212e-03, 4.5815e-03,
          5.5733e-03, 1.4887e-03, 7.6914e-04, 2.7027e-03, 5.5199e-03, 1.0109e-03,
          2.1477e-03, 7.6942e-03, 8.0643e-03, 5.3444e-03, 7.1678e-03, 1.3496e-02,
          6.8245e-03, 5.4312e-04, 1.0366e-03, 1.6537e-03, 6.6681e-03, 4.7646e-03,
          1.1692e-03, 7.5674e-04, 5.6648e-04, 2.1381e-03, 2.5177e-03, 5.3177e-03,
          1.2226e-03, 1.4467e-03, 1.2655e-03, 4.1656e-03, 2.5673e-03, 7.2212e-03,
          2.3766e-03, 1.5783e-03, 1.0681e-03, 2.8343e-03, 2.2095e-02, 5.3482e-03,
          7.3719e-04, 2.0866e-03, 5.8126e-04, 9.5558e-04, 9.3508e-04, 2.0504e-03,
          5.1641e-04, 4.0364e-04, 6.8760e-04, 1.0319e-03, 1.4687e-02, 8.5220e-03,
          1.2636e-03, 2.4490e-03, 1.9522e-03, 2.7256e-03, 9.2602e-04, 4.3640e-03,
          3.5553e-03, 5.9204e-03, 3.5889e-02, 2.1338e-05, 1.2290e-04, 2.1338e-05,
          1.2290e-04, 1.2290e-04, 1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 2.5272e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05, 1.2290e-04,
          4.0984e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05, 1.2290e-04, 2.1338e-05,
          3.0828e-04, 2.1338e-05, 1.2290e-04, 5.4169e-04, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 2.1338e-05, 2.1338e-05, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 2.1338e-05, 1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 2.1338e-05, 2.1338e-05, 1.2290e-04, 2.1338e-05, 1.2290e-04,
          2.1338e-05, 1.2290e-04, 1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          1.2290e-04, 2.1338e-05, 1.2290e-04, 2.1338e-05, 2.1338e-05, 1.2290e-04,
          2.1338e-05, 1.2290e-04, 1.2290e-04, 3.0828e-04, 2.1338e-05, 2.1338e-05,
          2.1338e-05, 2.1338e-05, 2.1338e-05, 1.2290e-04, 1.2290e-04, 1.2290e-04,
          2.5272e-04, 2.1338e-05, 1.2290e-04, 2.1338e-05, 2.5272e-04, 2.1338e-05,
          2.1338e-05, 1.2290e-04, 2.1338e-05, 2.1338e-05, 1.2290e-04, 1.2290e-04,
          2.5272e-04, 2.1338e-05, 5.3978e-04, 2.1338e-05, 1.2290e-04, 2.5272e-04,
          1.2290e-04, 1.2290e-04, 1.2290e-04, 2.1338e-05, 1.2290e-04, 2.1338e-05,
          2.1338e-05, 1.2290e-04, 2.1338e-05, 6.4135e-04, 2.1338e-05, 2.1338e-05,
          1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05, 1.2290e-04, 2.1338e-05,
          1.2290e-04, 3.0828e-04, 1.2290e-04, 2.1338e-05, 2.1338e-05, 2.1338e-05,
          1.2290e-04, 1.2290e-04, 4.0984e-04, 2.1338e-05, 2.5272e-04, 3.5429e-04,
          1.2290e-04, 1.2290e-04, 2.1338e-05, 2.5272e-04, 2.5272e-04, 2.1338e-05,
          1.2290e-04, 1.2290e-04, 1.2290e-04, 1.2290e-04, 1.2290e-04, 1.2290e-04,
          2.1338e-05, 1.2290e-04, 1.2290e-04, 1.2290e-04, 1.2290e-04, 7.2861e-04,
          1.2290e-04, 1.2290e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.6463e-03, 5.4291e-02, 1.7014e-02, 1.7380e-02, 3.6835e-02,
          4.2664e-02, 1.1024e-02, 2.0313e-03, 1.1606e-03, 2.1591e-03, 3.2463e-03,
          1.3218e-03, 4.1199e-03, 2.9106e-03, 2.2125e-03, 1.8473e-03, 7.2289e-03,
          5.9929e-03, 8.1968e-04, 5.5075e-04, 5.3310e-04, 7.2527e-04, 4.5357e-03,
          3.7956e-03, 1.4467e-03, 9.3079e-03, 6.8932e-03, 1.0366e-03, 8.4686e-03,
          5.6915e-03, 3.8204e-03, 2.6855e-03, 3.7899e-03, 1.8396e-03, 2.5997e-03,
          1.5068e-03, 5.5265e-04, 1.9665e-03, 2.1782e-03, 8.4591e-04, 1.1129e-03,
          5.6887e-04, 4.0770e-04, 5.0068e-04, 3.6359e-04, 7.0763e-03, 2.9373e-03,
          1.4811e-03, 1.1282e-03, 3.7289e-03, 2.7447e-03, 2.8286e-03, 1.7347e-03,
          7.3004e-04, 1.4153e-03, 7.7868e-04, 5.6458e-04, 1.3552e-03, 8.4209e-04,
          8.8882e-04, 4.0932e-03, 1.3428e-03, 1.7710e-03, 6.0701e-04, 3.4828e-03,
          1.1587e-03, 5.9414e-04, 7.6914e-04, 2.4676e-04, 2.4438e-04, 3.9053e-04,
          4.6539e-04, 4.2915e-03, 9.7418e-04, 3.5172e-03, 7.1621e-04, 6.4278e-04,
          1.2503e-03, 5.7697e-04, 7.5483e-04, 1.4610e-03, 8.4162e-04, 5.6219e-04,
          5.5885e-04, 4.3793e-03, 1.0386e-03, 1.5593e-03, 5.9414e-04, 6.2561e-04,
          3.1662e-03, 1.5926e-03, 4.1127e-04, 1.7281e-03, 4.0340e-04, 4.2295e-04,
          2.0790e-03, 5.3787e-04, 4.2648e-03, 2.6073e-03, 5.3883e-04, 7.1526e-04,
          3.1853e-03, 6.7759e-04, 7.7248e-04, 1.0290e-03, 2.5578e-03, 5.1918e-03,
          3.4981e-03, 6.5374e-04, 1.9951e-03, 5.0068e-04, 9.1696e-04, 8.2541e-04,
          9.4175e-04, 6.7902e-04, 1.8873e-03, 3.7498e-03, 6.4850e-04, 5.8479e-03,
          1.5457e-02, 1.9741e-03, 3.8505e-04, 1.1654e-03, 3.2806e-03, 3.9458e-04,
          4.4036e-04, 5.5742e-04, 8.1968e-04, 3.3646e-03, 3.8319e-03, 4.2267e-03,
          1.5326e-03, 7.6342e-04, 7.0190e-04, 4.5371e-04, 1.1616e-03, 3.9434e-04,
          1.1047e-02, 6.6376e-03, 5.6763e-03, 5.0735e-03, 4.6463e-03, 1.8234e-02,
          5.8937e-04, 5.1498e-04, 6.4707e-04, 6.1131e-04, 1.0471e-03, 2.4986e-03,
          3.1528e-03, 2.2945e-03, 8.1778e-04, 3.3665e-03, 2.6493e-03, 1.3819e-03,
          1.3371e-03, 9.0170e-04, 1.0052e-03, 6.4421e-04, 2.7351e-03, 1.3359e-02,
          3.0518e-03, 2.1038e-03, 1.2291e-02, 2.9507e-03, 1.1883e-03, 8.7357e-04,
          2.0790e-03, 9.3794e-04, 1.8139e-03, 1.6737e-03, 8.4839e-03, 3.0136e-03,
          1.6050e-03, 1.3933e-03, 1.4448e-03, 1.4896e-03, 2.4509e-03, 4.2038e-03,
          1.2121e-03, 2.0466e-03, 2.9144e-03, 4.1618e-03, 1.8940e-03, 1.6870e-03,
          2.5978e-03, 1.3428e-02, 1.8721e-03, 9.2459e-04, 1.2341e-03, 3.0708e-03,
          4.0054e-03, 4.3440e-04, 2.6093e-03, 1.4076e-02, 2.0485e-03, 2.2621e-03,
          1.4410e-03, 4.0131e-03, 4.3030e-03, 1.1616e-03, 1.2989e-03, 1.4277e-03,
          9.5510e-04, 1.4334e-03, 7.7152e-04, 1.2121e-03, 3.3207e-03, 5.1308e-03,
          2.0962e-03, 2.3842e-03, 1.0443e-03, 1.4496e-03, 4.2267e-03, 1.0042e-03,
          2.0428e-03, 1.1932e-02, 3.0251e-03, 1.5059e-03, 6.0539e-03, 2.3590e-02,
          8.5220e-03, 1.3332e-03, 1.8044e-03, 1.3142e-03, 5.9814e-03, 3.2616e-03,
          1.0958e-03, 9.7609e-04, 7.0286e-04, 1.7805e-03, 2.5558e-03, 8.7280e-03,
          1.7309e-03, 1.1415e-03, 1.2331e-03, 1.7176e-03, 2.8858e-03, 5.8594e-03,
          2.4796e-03, 1.9112e-03, 1.3447e-03, 3.9330e-03, 3.1204e-02, 1.2726e-02,
          1.8911e-03, 2.2030e-03, 1.2264e-03, 1.3838e-03, 1.4582e-03, 2.6569e-03,
          8.5211e-04, 6.3992e-04, 1.1559e-03, 1.8187e-03, 7.3204e-03, 7.4806e-03,
          1.7595e-03, 1.7958e-03, 1.5535e-03, 1.9798e-03, 1.0500e-03, 2.8706e-03,
          3.4790e-03, 1.6956e-03, 6.8550e-03, 8.2581e-02, 1.1301e-04, 2.7299e-05,
          1.1063e-03, 1.6248e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.4867e-04,
          2.4819e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.5129e-04, 2.7299e-05, 2.7299e-05, 1.6248e-04, 1.6153e-04, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 1.6248e-04, 2.7299e-05, 1.6248e-04, 1.1301e-04,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 1.1301e-04,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 1.6248e-04, 2.7299e-05, 1.6248e-04,
          1.6248e-04, 2.4819e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05, 1.6248e-04, 2.7299e-05,
          2.7299e-05, 2.7299e-05, 2.7299e-05, 1.1301e-04, 3.2723e-05, 2.7299e-05,
          2.7299e-05, 1.6248e-04, 2.7299e-05, 2.7299e-05, 2.4819e-04, 2.4819e-04,
          1.1843e-04, 2.7299e-05, 1.1301e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.4819e-04, 2.4819e-04, 2.7299e-05, 2.7299e-05, 1.6248e-04, 2.7299e-05,
          1.1301e-04, 2.7299e-05, 2.7299e-05, 1.1843e-04, 2.7299e-05, 2.7299e-05,
          1.6248e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05, 1.1301e-04, 2.7299e-05,
          2.4819e-04, 1.1301e-04, 2.4819e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          2.4819e-04, 2.7299e-05, 1.1301e-04, 2.7299e-05, 1.6248e-04, 1.6248e-04,
          2.7299e-05, 1.6248e-04, 2.7299e-05, 2.7299e-05, 2.7299e-05, 2.7299e-05,
          5.2547e-04, 5.2547e-04, 1.6248e-04, 1.6248e-04, 5.2547e-04, 1.6248e-04,
          2.7299e-05, 2.4819e-04, 1.6248e-04, 1.6248e-04, 1.6248e-04, 1.6248e-04,
          5.2547e-04, 1.6248e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3024e-03, 5.1727e-02, 1.6586e-02, 3.1677e-02, 4.1962e-02,
          4.0039e-02, 1.7258e-02, 1.5459e-03, 9.2268e-04, 2.2564e-03, 3.7575e-03,
          1.4620e-03, 3.8509e-03, 9.4833e-03, 1.3294e-03, 1.3924e-03, 1.0811e-02,
          8.7128e-03, 8.8024e-04, 8.4448e-04, 6.0511e-04, 1.0481e-03, 5.6229e-03,
          4.2458e-03, 1.2856e-03, 1.4343e-02, 1.0445e-02, 1.4343e-03, 7.2556e-03,
          1.8539e-02, 2.8191e-03, 2.9259e-03, 1.8167e-03, 2.0981e-03, 1.1940e-03,
          1.0185e-03, 5.3692e-04, 1.9197e-03, 1.4715e-03, 1.5583e-03, 1.2407e-03,
          6.5947e-04, 4.8780e-04, 5.9319e-04, 3.8362e-04, 8.3466e-03, 1.9722e-03,
          3.5324e-03, 2.2049e-03, 4.5662e-03, 1.5869e-03, 2.1706e-03, 1.1072e-03,
          7.6485e-04, 1.2150e-03, 6.6996e-04, 5.3263e-04, 1.3905e-03, 7.1907e-04,
          8.2922e-04, 4.5929e-03, 1.0948e-03, 1.0433e-03, 7.7391e-04, 2.3079e-03,
          1.8063e-03, 1.0176e-03, 1.0414e-03, 3.1614e-04, 2.9182e-04, 5.3358e-04,
          6.1655e-04, 4.9133e-03, 9.3269e-04, 2.4624e-03, 7.4434e-04, 6.5994e-04,
          1.5659e-03, 4.8685e-04, 9.8324e-04, 1.1206e-03, 1.2236e-03, 4.6754e-04,
          4.8447e-04, 3.0460e-03, 1.1101e-03, 1.7881e-03, 1.1835e-03, 1.4753e-03,
          3.5019e-03, 1.4839e-03, 5.1308e-04, 2.0142e-03, 3.8171e-04, 5.3263e-04,
          2.3975e-03, 5.1785e-04, 2.1114e-03, 1.7023e-03, 5.1212e-04, 6.2513e-04,
          2.8706e-03, 8.3017e-04, 9.7418e-04, 7.3242e-04, 1.3599e-03, 3.7308e-03,
          2.2640e-03, 6.5374e-04, 2.0103e-03, 5.6458e-04, 1.2560e-03, 1.1806e-03,
          7.7152e-04, 6.1321e-04, 2.1954e-03, 2.2945e-03, 7.8106e-04, 2.3861e-03,
          1.4771e-02, 1.2388e-03, 3.8195e-04, 9.5606e-04, 1.9341e-03, 4.3774e-04,
          3.3498e-04, 7.8154e-04, 9.8419e-04, 1.6956e-03, 2.9793e-03, 2.7752e-03,
          1.7185e-03, 1.2455e-03, 8.4925e-04, 6.6948e-04, 1.3819e-03, 5.0735e-04,
          1.0048e-02, 1.0361e-02, 5.3024e-03, 5.8098e-03, 4.7035e-03, 1.8936e-02,
          5.5122e-04, 4.0674e-04, 5.4026e-04, 5.0354e-04, 1.5182e-03, 1.4400e-03,
          2.1667e-03, 1.3914e-03, 5.7650e-04, 2.0046e-03, 1.8301e-03, 1.2465e-03,
          1.4963e-03, 1.0023e-03, 1.0796e-03, 9.1124e-04, 1.9207e-03, 1.1246e-02,
          2.1400e-03, 2.1725e-03, 1.2711e-02, 2.8191e-03, 9.9850e-04, 8.1873e-04,
          1.9817e-03, 8.9025e-04, 1.0834e-03, 9.5987e-04, 1.0826e-02, 1.7014e-03,
          9.4986e-04, 9.6035e-04, 1.0023e-03, 1.2350e-03, 1.7271e-03, 5.1613e-03,
          1.2875e-03, 1.5497e-03, 2.3174e-03, 2.8992e-03, 2.1076e-03, 1.3647e-03,
          2.5368e-03, 1.2794e-02, 2.4014e-03, 8.0967e-04, 1.8272e-03, 1.9951e-03,
          2.6608e-03, 3.6383e-04, 1.5898e-03, 1.3901e-02, 1.8501e-03, 1.5764e-03,
          1.0185e-03, 2.9984e-03, 2.5616e-03, 1.5888e-03, 1.8368e-03, 1.3142e-03,
          1.4524e-03, 1.6146e-03, 7.3671e-04, 1.3380e-03, 3.3054e-03, 4.0817e-03,
          1.7910e-03, 1.8263e-03, 1.0061e-03, 1.4877e-03, 2.3499e-03, 9.4414e-04,
          1.2255e-03, 7.9346e-03, 3.9787e-03, 1.0357e-03, 3.0193e-03, 1.2283e-02,
          6.4545e-03, 8.1539e-04, 1.3933e-03, 1.8005e-03, 4.1733e-03, 2.8782e-03,
          1.9026e-03, 1.0176e-03, 6.2847e-04, 1.2112e-03, 1.5841e-03, 7.2899e-03,
          1.8120e-03, 1.0662e-03, 1.0519e-03, 9.3699e-04, 1.5049e-03, 2.7237e-03,
          1.6546e-03, 1.3590e-03, 6.7091e-04, 1.8301e-03, 1.2566e-02, 1.4679e-02,
          1.2989e-03, 2.4147e-03, 7.5340e-04, 8.3590e-04, 1.0223e-03, 2.0828e-03,
          9.6273e-04, 5.6410e-04, 1.0614e-03, 1.1234e-03, 5.4169e-03, 6.3591e-03,
          1.3237e-03, 1.7281e-03, 8.5783e-04, 9.8133e-04, 5.6982e-04, 1.7233e-03,
          1.3914e-03, 1.5602e-03, 4.4136e-03, 2.4399e-02, 9.9121e-02, 2.2650e-05,
          7.3433e-04, 1.2577e-04, 8.2970e-05, 2.2650e-05, 2.2650e-05, 2.6536e-04,
          2.0504e-04, 8.2970e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 3.0804e-04,
          1.8609e-04, 2.0504e-04, 2.0504e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          1.6844e-04, 8.2970e-05, 1.2577e-04, 1.8609e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 8.2970e-05, 3.0804e-04, 2.0504e-04,
          2.2650e-05, 2.2650e-05, 1.8609e-04, 2.2650e-05, 2.0504e-04, 8.2970e-05,
          8.2970e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05, 8.2970e-05, 8.2970e-05,
          2.2650e-05, 2.6536e-04, 2.2650e-05, 1.6844e-04, 1.2577e-04, 1.8609e-04,
          3.6836e-04, 1.8609e-04, 2.2650e-05, 2.2650e-05, 2.0504e-04, 2.2650e-05,
          2.2650e-05, 2.2650e-05, 2.2650e-05, 2.0504e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.6536e-04, 2.2650e-05, 8.2970e-05, 2.2650e-05, 2.2650e-05,
          2.6536e-04, 8.2970e-05, 8.2970e-05, 2.2650e-05, 1.2577e-04, 2.2650e-05,
          2.2650e-05, 8.2970e-05, 2.2650e-05, 8.2970e-05, 8.2970e-05, 2.2650e-05,
          2.2650e-05, 1.2577e-04, 2.2650e-05, 2.0504e-04, 3.6836e-04, 1.6844e-04,
          2.6536e-04, 2.2650e-05, 8.2970e-05, 8.2970e-05, 2.2650e-05, 8.2970e-05,
          5.1785e-04, 6.0368e-04, 2.0504e-04, 2.2650e-05, 7.6008e-04, 8.2970e-05,
          8.2970e-05, 2.2650e-05, 2.2650e-05, 3.0613e-04, 2.2650e-05, 2.2650e-05,
          2.2650e-05, 2.0504e-04, 2.2650e-05, 2.2650e-05, 3.5071e-04, 2.2650e-05,
          5.1785e-04, 2.2650e-05, 1.8609e-04, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          3.6836e-04, 2.0504e-04, 2.2650e-05, 2.2650e-05, 8.2970e-05, 1.8609e-04,
          2.2650e-05, 2.5749e-03, 2.2650e-05, 2.2650e-05, 2.2650e-05, 2.2650e-05,
          8.0061e-04, 2.8443e-04, 2.8443e-04, 3.5715e-04, 2.8443e-04, 1.8609e-04,
          2.2650e-05, 1.8609e-04, 2.8443e-04, 3.8648e-04, 1.2577e-04, 2.2650e-05,
          2.8443e-04, 1.2577e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.7708e-03, 5.2399e-02, 1.3084e-02, 1.0918e-02, 3.9368e-02,
          3.9185e-02, 8.9874e-03, 6.9332e-04, 6.0892e-04, 1.9569e-03, 3.2310e-03,
          1.0424e-03, 9.4080e-04, 4.0412e-04, 3.2578e-03, 8.2397e-04, 8.3637e-04,
          5.5456e-04, 5.6505e-04, 2.5940e-04, 4.0865e-04, 3.4380e-04, 1.1778e-03,
          4.0894e-03, 4.3907e-03, 6.9656e-03, 2.5749e-03, 4.2343e-04, 9.6846e-04,
          5.3930e-04, 1.2646e-03, 3.7742e-04, 3.5896e-03, 1.8625e-03, 8.5735e-04,
          4.9324e-03, 8.9121e-04, 3.1128e-03, 2.8229e-03, 3.3236e-04, 4.0855e-03,
          1.1740e-03, 1.0138e-03, 8.6308e-04, 7.9012e-04, 1.0395e-03, 5.2185e-03,
          3.1805e-04, 4.2844e-04, 3.2864e-03, 1.1950e-03, 7.6914e-04, 8.5497e-04,
          3.8481e-04, 5.4741e-04, 5.0783e-04, 9.8133e-04, 1.3304e-03, 1.6499e-03,
          1.4105e-03, 9.0933e-04, 1.8082e-03, 2.7771e-03, 2.0294e-03, 1.5419e-02,
          1.3809e-03, 8.2779e-04, 1.5440e-03, 5.7030e-04, 5.1928e-04, 5.4455e-04,
          5.8365e-04, 3.9139e-03, 5.4121e-04, 9.8515e-04, 3.4761e-04, 3.3832e-04,
          6.1846e-04, 7.6771e-04, 7.7677e-04, 3.0136e-03, 9.2602e-04, 1.8692e-03,
          1.1492e-03, 2.9011e-03, 1.1005e-03, 1.7881e-03, 7.9584e-04, 9.1839e-04,
          3.3016e-03, 2.8915e-03, 8.2159e-04, 2.0103e-03, 9.7275e-04, 7.0715e-04,
          4.6062e-04, 3.6073e-04, 9.8515e-04, 1.8263e-03, 9.0313e-04, 1.1683e-03,
          2.4014e-03, 5.5265e-04, 6.0034e-04, 1.1436e-02, 8.6975e-03, 1.2894e-03,
          2.6627e-03, 2.2259e-03, 4.4556e-03, 1.0910e-03, 1.0777e-03, 1.2655e-03,
          3.9330e-03, 2.7447e-03, 8.8072e-04, 1.0300e-02, 1.0967e-03, 8.3065e-04,
          1.0500e-03, 3.5458e-03, 4.6301e-04, 5.8746e-04, 1.2245e-02, 4.1604e-04,
          6.8808e-04, 5.8937e-04, 7.8487e-04, 7.4425e-03, 1.5078e-03, 2.7618e-03,
          1.7748e-03, 9.3174e-04, 1.3132e-03, 8.1635e-04, 2.6207e-03, 1.1568e-03,
          3.7842e-03, 1.3504e-03, 1.3227e-03, 2.0905e-03, 2.2049e-03, 1.1587e-03,
          7.5674e-04, 7.9966e-04, 1.2569e-03, 1.1215e-03, 1.0843e-03, 7.5607e-03,
          4.3335e-03, 6.8893e-03, 1.8673e-03, 3.3169e-03, 2.4605e-03, 8.3542e-03,
          3.4389e-03, 3.5286e-03, 7.5674e-04, 1.5850e-03, 1.1063e-02, 2.1801e-03,
          5.6229e-03, 4.1618e-03, 1.7033e-03, 3.5267e-03, 1.8158e-03, 1.2131e-03,
          1.7481e-03, 1.2655e-03, 4.3831e-03, 2.7409e-03, 1.3742e-03, 5.1117e-03,
          3.6621e-03, 2.0618e-03, 2.4166e-03, 4.3297e-03, 1.4687e-02, 2.7771e-03,
          3.8471e-03, 7.5188e-03, 1.4557e-02, 1.8415e-03, 1.4257e-03, 2.0485e-03,
          1.9121e-03, 1.1702e-03, 1.9064e-03, 1.2188e-03, 9.7656e-04, 5.5580e-03,
          1.1129e-03, 6.6042e-04, 3.1147e-03, 1.9722e-03, 1.7157e-03, 4.1351e-03,
          2.4223e-03, 1.8682e-03, 2.1103e-02, 3.5076e-03, 1.9360e-03, 5.9929e-03,
          2.4853e-03, 6.9389e-03, 3.1414e-03, 1.1578e-03, 3.9062e-03, 1.1873e-03,
          1.1311e-03, 2.5196e-03, 1.1730e-03, 9.1553e-04, 5.5504e-03, 9.9087e-04,
          3.6259e-03, 2.4185e-03, 2.4776e-03, 2.0142e-03, 5.0850e-03, 1.3342e-03,
          6.2294e-03, 5.7373e-03, 1.3443e-02, 2.2411e-03, 3.5217e-02, 6.1531e-03,
          2.1687e-03, 1.5860e-03, 2.1667e-03, 4.2343e-03, 7.6065e-03, 1.5526e-03,
          1.5955e-03, 1.2407e-03, 9.3603e-04, 3.3073e-03, 5.8212e-03, 6.6948e-03,
          2.0809e-03, 1.9798e-03, 2.3098e-03, 3.7766e-03, 3.9177e-03, 3.6068e-03,
          8.9645e-03, 4.3831e-03, 4.4212e-03, 4.6234e-03, 8.9340e-03, 3.0472e-02,
          2.1305e-03, 2.5635e-03, 4.4594e-03, 6.2904e-03, 3.5324e-03, 4.2305e-03,
          2.5959e-03, 1.1902e-03, 3.4714e-03, 3.5629e-03, 1.4458e-03, 2.4567e-03,
          4.9706e-03, 1.8187e-03, 1.2064e-03, 3.8261e-03, 3.6831e-03, 4.2328e-02,
          1.1462e-04, 1.1462e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 2.7418e-05, 1.1462e-04, 2.1338e-04, 2.7418e-05, 2.1338e-04,
          2.1338e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05,
          1.2624e-04, 2.7418e-05, 2.1338e-04, 6.0225e-04, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 2.7418e-05, 2.1338e-04, 1.1462e-04, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 1.1462e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05, 1.2624e-04,
          1.1462e-04, 2.7418e-05, 1.1462e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 1.1462e-04, 1.1462e-04, 2.1338e-04, 2.7418e-05, 2.7418e-05,
          2.1338e-04, 1.1462e-04, 1.1462e-04, 2.7418e-05, 2.7418e-05, 2.1338e-04,
          2.7418e-05, 1.1462e-04, 2.1338e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05,
          2.7418e-05, 2.7418e-05, 1.2624e-04, 1.1462e-04, 1.1462e-04, 2.1338e-04,
          1.1462e-04, 2.7418e-05, 2.1338e-04, 2.1338e-04, 2.7418e-05, 1.1462e-04,
          1.4868e-03, 2.1338e-04, 2.7418e-05, 2.7418e-05, 2.1338e-04, 2.1338e-04,
          2.7418e-05, 1.1462e-04, 2.7418e-05, 2.7418e-05, 2.1338e-04, 2.7418e-05,
          2.1338e-04, 2.1338e-04, 2.7418e-05, 2.7418e-05, 1.1462e-04, 2.7418e-05,
          2.7418e-05, 2.1338e-04, 2.7418e-05, 1.1462e-04, 2.7418e-05, 2.7418e-05,
          1.1462e-04, 2.7418e-05, 2.7418e-05, 2.7418e-05, 2.1338e-04, 1.1462e-04,
          2.1338e-04, 2.1338e-04, 2.1338e-04, 1.1462e-04, 1.1462e-04, 1.1462e-04,
          2.1338e-04, 1.1462e-04, 2.1338e-04, 2.7418e-05, 2.7418e-05, 1.1462e-04,
          1.4868e-03, 1.5860e-03, 2.7418e-05, 1.1462e-04, 2.7418e-05, 2.7418e-05,
          1.1462e-04, 1.1462e-04, 1.1462e-04, 2.0981e-04, 1.1462e-04, 2.7418e-05,
          2.7418e-05, 2.7418e-05, 1.1462e-04, 4.2391e-04, 2.7418e-05, 1.1462e-04,
          1.1462e-04, 2.7418e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4250e-03, 4.8584e-02, 1.3580e-02, 1.3199e-02, 4.4159e-02,
          3.9612e-02, 1.0223e-02, 7.5674e-04, 5.3740e-04, 1.4210e-03, 5.1270e-03,
          1.1683e-03, 6.5517e-04, 8.9836e-04, 8.1444e-04, 7.7105e-04, 8.2684e-04,
          6.7759e-04, 5.6028e-04, 3.2353e-04, 3.5381e-04, 4.1223e-04, 8.6451e-04,
          6.2561e-03, 1.6747e-03, 8.2932e-03, 2.7103e-03, 5.5981e-04, 8.7738e-04,
          1.2608e-03, 1.2083e-03, 4.1080e-04, 7.9489e-04, 6.0701e-04, 5.1212e-04,
          1.0014e-03, 9.0790e-04, 8.9502e-04, 6.1989e-04, 4.5466e-04, 1.8024e-03,
          1.5669e-03, 1.2503e-03, 1.5583e-03, 5.2977e-04, 9.1362e-04, 1.0386e-03,
          5.2404e-04, 4.6611e-04, 3.9940e-03, 9.5701e-04, 1.0290e-03, 5.2023e-04,
          6.3276e-04, 7.0095e-04, 6.2323e-04, 7.7963e-04, 2.6550e-03, 1.8549e-03,
          1.8845e-03, 9.7656e-04, 1.6060e-03, 1.0653e-03, 3.1643e-03, 2.2526e-03,
          1.8890e-02, 3.2425e-03, 9.9754e-04, 7.8297e-04, 7.5912e-04, 1.2760e-03,
          7.6103e-04, 2.6073e-03, 7.9727e-04, 1.4000e-03, 7.1859e-04, 4.4489e-04,
          7.8106e-04, 8.4829e-04, 4.2953e-03, 1.4982e-03, 7.0305e-03, 1.0738e-03,
          8.3542e-04, 2.4586e-03, 1.2245e-03, 2.6608e-03, 5.5237e-03, 7.8087e-03,
          6.4697e-03, 2.3613e-03, 1.5631e-03, 3.2482e-03, 6.7759e-04, 6.6996e-04,
          7.0381e-04, 5.0735e-04, 6.5613e-04, 1.0548e-03, 1.3809e-03, 1.3247e-03,
          4.1008e-03, 4.5395e-03, 2.9926e-03, 1.4515e-03, 1.2417e-03, 1.0595e-03,
          2.1877e-03, 2.6798e-03, 4.1428e-03, 5.3368e-03, 1.4160e-02, 4.2992e-03,
          1.6079e-03, 1.2760e-03, 6.6805e-04, 2.1057e-03, 9.5844e-04, 1.2436e-03,
          1.6184e-03, 1.5440e-03, 1.2293e-03, 6.2323e-04, 1.6689e-03, 1.6537e-03,
          6.8951e-04, 5.1727e-03, 9.2392e-03, 1.4219e-03, 9.3412e-04, 2.4757e-03,
          2.1347e-02, 7.3776e-03, 5.6362e-04, 2.6455e-03, 4.3373e-03, 1.7605e-03,
          1.2894e-03, 1.4114e-03, 9.7179e-04, 1.7462e-03, 1.1702e-03, 1.6632e-03,
          2.2488e-03, 9.2411e-04, 1.2054e-03, 1.2789e-03, 7.3967e-03, 1.8797e-03,
          2.9030e-03, 1.5888e-03, 1.3237e-03, 1.7614e-03, 2.3518e-03, 3.9597e-03,
          3.1490e-03, 2.6913e-03, 1.2331e-03, 6.8130e-03, 2.7599e-03, 1.3590e-03,
          1.4458e-03, 1.3723e-03, 1.3170e-03, 1.0605e-03, 1.2283e-03, 1.6890e-03,
          3.1223e-03, 1.4076e-03, 1.3647e-03, 7.4005e-04, 1.2026e-03, 1.2884e-03,
          1.9836e-03, 2.3479e-03, 1.4811e-03, 3.6430e-03, 3.1643e-03, 1.7557e-03,
          1.5421e-03, 1.5249e-03, 2.2182e-03, 1.7166e-03, 1.1139e-03, 7.2813e-04,
          2.3689e-03, 1.4324e-03, 1.4505e-03, 1.9350e-03, 6.6605e-03, 1.3390e-03,
          1.1206e-03, 8.9455e-04, 9.5749e-04, 1.7271e-03, 1.9360e-03, 1.2226e-03,
          1.7967e-03, 1.9131e-03, 3.7079e-03, 9.0561e-03, 3.7041e-03, 1.6689e-03,
          1.2619e-02, 8.0719e-03, 2.7523e-03, 1.0014e-03, 5.1460e-03, 8.5068e-04,
          1.4429e-03, 1.8139e-03, 2.4071e-03, 4.5547e-03, 1.9054e-03, 3.7174e-03,
          1.2283e-03, 1.5373e-03, 1.9779e-03, 6.8998e-04, 1.3762e-03, 2.3193e-03,
          2.6646e-03, 8.0566e-03, 7.8201e-03, 7.7820e-03, 4.6577e-03, 1.8024e-03,
          1.4885e-02, 1.2543e-02, 2.9640e-03, 4.2419e-03, 2.1782e-03, 2.0103e-03,
          2.6646e-03, 2.8725e-03, 1.2543e-02, 1.5402e-03, 2.6646e-03, 7.7782e-03,
          4.3869e-03, 2.6989e-03, 1.7481e-03, 1.4277e-03, 3.0022e-03, 4.4518e-03,
          5.5389e-03, 2.5345e-02, 8.9798e-03, 7.4539e-03, 8.4152e-03, 4.7989e-03,
          1.1589e-02, 2.0351e-03, 3.3703e-03, 2.7790e-03, 1.5888e-03, 2.7275e-03,
          1.9207e-03, 6.5231e-03, 1.5621e-03, 1.4725e-03, 1.4038e-03, 4.0207e-03,
          1.7405e-03, 9.6655e-04, 9.4891e-04, 4.2877e-03, 6.7787e-03, 4.9362e-03,
          7.8003e-02, 3.3557e-05, 1.0544e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 1.7679e-04, 3.3557e-05, 1.7679e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0544e-04, 3.3557e-05,
          1.0490e-04, 3.3557e-05, 1.5240e-03, 4.7970e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0490e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 1.0490e-04, 1.0490e-04, 1.7679e-04, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 2.6536e-04, 1.0544e-04, 1.0544e-04, 3.3557e-05, 3.3557e-05,
          3.3665e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 3.3557e-05, 5.3596e-04, 1.7679e-04, 3.3557e-05, 3.3557e-05,
          5.0783e-04, 4.4727e-04, 2.6536e-04, 3.3557e-05, 3.3557e-05, 3.3665e-04,
          3.3557e-05, 4.4727e-04, 3.3665e-04, 1.0490e-04, 3.3557e-05, 3.3557e-05,
          1.0544e-04, 3.3557e-05, 1.0490e-04, 1.0544e-04, 3.3557e-05, 1.7679e-04,
          3.3557e-05, 3.3557e-05, 6.0701e-04, 1.0490e-04, 3.0398e-04, 2.6536e-04,
          3.7575e-04, 1.0544e-04, 3.3557e-05, 3.3557e-05, 1.0490e-04, 1.0490e-04,
          1.0490e-04, 2.6536e-04, 3.3557e-05, 3.3557e-05, 3.3665e-04, 1.0544e-04,
          4.4727e-04, 1.0490e-04, 3.0398e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          3.3557e-05, 5.3596e-04, 3.3557e-05, 1.0544e-04, 3.3557e-05, 3.3557e-05,
          1.0544e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05, 1.0490e-04, 3.3557e-05,
          4.4727e-04, 3.3665e-04, 1.0544e-04, 1.0544e-04, 3.3557e-05, 4.3631e-04,
          1.0490e-04, 3.3557e-05, 6.0701e-04, 3.3557e-05, 3.3557e-05, 3.3557e-05,
          5.3596e-04, 4.4727e-04, 3.3557e-05, 3.7575e-04, 3.0398e-04, 3.3557e-05,
          3.3557e-05, 2.4843e-04, 2.4843e-04, 1.0502e-04, 4.0841e-04, 2.4843e-04,
          3.3557e-05, 1.7679e-04, 3.3557e-05, 3.7289e-04, 2.4843e-04, 1.0544e-04,
          3.3557e-05, 1.0544e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.0545e-03, 5.3345e-02, 1.4091e-02, 1.9058e-02, 4.0070e-02,
          3.7567e-02, 1.1436e-02, 8.8024e-04, 5.5885e-04, 2.9068e-03, 8.4000e-03,
          2.1152e-03, 8.6355e-04, 1.2045e-03, 9.3889e-04, 8.8263e-04, 1.1396e-03,
          1.1759e-03, 6.4421e-04, 2.5272e-04, 3.0446e-04, 3.9124e-04, 1.4620e-03,
          1.0452e-02, 1.7204e-03, 9.1248e-03, 5.1956e-03, 5.4073e-04, 1.3628e-03,
          1.7605e-03, 1.6022e-03, 5.5981e-04, 1.0023e-03, 1.4753e-03, 4.9877e-04,
          1.0300e-03, 7.3624e-04, 1.6127e-03, 9.3889e-04, 4.4417e-04, 2.4719e-03,
          1.5879e-03, 1.1234e-03, 1.6041e-03, 5.7793e-04, 1.6527e-03, 1.5278e-03,
          7.2241e-04, 4.8518e-04, 3.5648e-03, 7.2718e-04, 1.2798e-03, 5.7983e-04,
          4.6301e-04, 7.4816e-04, 4.4537e-04, 7.0095e-04, 1.8225e-03, 1.4801e-03,
          1.1873e-03, 9.6464e-04, 1.0223e-03, 7.5912e-04, 3.1643e-03, 3.1662e-03,
          1.4687e-02, 5.3101e-03, 1.8692e-03, 7.2336e-04, 7.0238e-04, 1.0872e-03,
          6.5231e-04, 5.3101e-03, 6.0368e-04, 1.5507e-03, 5.4741e-04, 4.0936e-04,
          7.7438e-04, 8.0204e-04, 6.8970e-03, 1.9932e-03, 1.5312e-02, 7.2861e-04,
          5.7316e-04, 2.3403e-03, 8.6498e-04, 5.3864e-03, 8.3923e-03, 9.8038e-03,
          1.3474e-02, 4.1046e-03, 1.0099e-03, 6.3667e-03, 4.7421e-04, 5.5504e-04,
          5.5599e-04, 3.5310e-04, 8.4972e-04, 1.0796e-03, 1.2074e-03, 1.2398e-03,
          8.0032e-03, 3.7594e-03, 6.1073e-03, 1.5106e-03, 1.4763e-03, 7.9727e-04,
          1.7776e-03, 1.9665e-03, 6.3705e-03, 2.5635e-03, 9.5901e-03, 6.6032e-03,
          1.4715e-03, 9.9754e-04, 4.8423e-04, 2.4891e-03, 7.6723e-04, 8.4019e-04,
          1.5640e-03, 1.8425e-03, 1.0071e-03, 6.8665e-04, 2.3460e-03, 2.0180e-03,
          5.7554e-04, 4.5738e-03, 5.1041e-03, 1.3933e-03, 1.1253e-03, 2.1420e-03,
          8.6212e-03, 1.3336e-02, 9.2888e-04, 3.6259e-03, 7.1144e-03, 1.2531e-03,
          2.2678e-03, 1.2798e-03, 1.1349e-03, 2.6264e-03, 1.7986e-03, 1.7719e-03,
          1.8377e-03, 7.1383e-04, 1.1997e-03, 1.1730e-03, 9.5444e-03, 1.8644e-03,
          4.5662e-03, 1.3742e-03, 8.3256e-04, 1.2388e-03, 1.4000e-03, 3.0060e-03,
          5.6763e-03, 3.2310e-03, 2.4052e-03, 4.4899e-03, 2.4891e-03, 1.3266e-03,
          1.9360e-03, 2.6150e-03, 2.2774e-03, 1.7262e-03, 1.1015e-03, 1.7281e-03,
          5.8365e-03, 1.8883e-03, 1.4887e-03, 8.0061e-04, 1.7929e-03, 1.4753e-03,
          1.0014e-03, 1.0767e-03, 1.1368e-03, 2.7065e-03, 3.8185e-03, 3.4161e-03,
          2.1000e-03, 1.5354e-03, 3.4351e-03, 1.9312e-03, 1.3523e-03, 7.8487e-04,
          4.6272e-03, 1.2541e-03, 1.6727e-03, 1.6708e-03, 8.1253e-03, 1.8110e-03,
          1.6298e-03, 7.5006e-04, 1.1730e-03, 2.8591e-03, 4.2801e-03, 1.4429e-03,
          1.1911e-03, 1.0586e-03, 3.2482e-03, 1.1024e-02, 7.5226e-03, 2.6340e-03,
          1.2047e-02, 9.7275e-03, 1.5116e-03, 1.2245e-03, 8.4534e-03, 1.1587e-03,
          1.4162e-03, 1.5144e-03, 1.9131e-03, 6.8207e-03, 2.2545e-03, 3.5515e-03,
          1.3466e-03, 1.6842e-03, 3.7365e-03, 6.9046e-04, 1.7776e-03, 1.6947e-03,
          3.4389e-03, 2.7580e-03, 6.5918e-03, 8.4610e-03, 6.0883e-03, 2.6283e-03,
          2.4551e-02, 4.1275e-03, 1.6890e-03, 1.6890e-03, 1.7891e-03, 1.1625e-03,
          2.3365e-03, 3.0041e-03, 4.7531e-03, 1.2369e-03, 1.8616e-03, 3.5610e-03,
          2.2011e-03, 1.4648e-03, 8.7929e-04, 1.0700e-03, 2.1477e-03, 3.3417e-03,
          3.2578e-03, 1.5610e-02, 2.0409e-03, 2.0866e-03, 2.7637e-03, 4.2953e-03,
          5.4169e-03, 9.6560e-04, 1.7576e-03, 1.7481e-03, 2.2640e-03, 4.6768e-03,
          1.6632e-03, 5.7678e-03, 1.2341e-03, 1.0948e-03, 7.0000e-04, 2.3766e-03,
          1.2245e-03, 1.2827e-03, 9.7847e-04, 3.9024e-03, 1.2169e-02, 3.0746e-03,
          1.4992e-02, 4.7211e-02, 8.2552e-05, 3.1531e-05, 3.1531e-05, 8.2552e-05,
          3.1531e-05, 8.2552e-05, 1.6761e-04, 1.2612e-04, 3.1531e-05, 3.1531e-05,
          3.1531e-05, 3.1531e-05, 3.1531e-05, 3.1531e-05, 2.7585e-04, 3.1531e-05,
          1.3971e-04, 8.2552e-05, 1.7715e-04, 1.6308e-03, 3.1531e-05, 3.1531e-05,
          3.1531e-05, 3.1531e-05, 3.1531e-05, 8.2552e-05, 3.1531e-05, 3.1531e-05,
          3.1531e-05, 8.2552e-05, 8.2552e-05, 3.1328e-04, 3.1531e-05, 3.1531e-05,
          8.2552e-05, 2.6226e-04, 3.1531e-05, 1.2612e-04, 3.1531e-05, 8.2552e-05,
          3.1328e-04, 3.1531e-05, 1.6761e-04, 1.3971e-04, 3.1531e-05, 8.2552e-05,
          1.6761e-04, 3.1531e-05, 6.6853e-04, 3.1328e-04, 3.1531e-05, 1.6761e-04,
          5.8365e-04, 1.7715e-04, 1.2612e-04, 3.1531e-05, 3.1531e-05, 1.7715e-04,
          3.1531e-05, 3.1328e-04, 1.7726e-04, 1.3971e-04, 3.1531e-05, 1.2612e-04,
          2.6226e-04, 8.2552e-05, 8.2552e-05, 1.6761e-04, 3.1531e-05, 1.7715e-04,
          3.1531e-05, 8.2552e-05, 3.1328e-04, 8.2552e-05, 2.1863e-04, 6.6853e-04,
          1.6761e-04, 8.2552e-05, 3.1531e-05, 3.1531e-05, 1.3971e-04, 1.3971e-04,
          2.7585e-04, 1.2612e-04, 3.1531e-05, 3.1531e-05, 1.2612e-04, 2.6226e-04,
          1.1253e-03, 1.3971e-04, 2.1863e-04, 3.1531e-05, 8.2552e-05, 8.2552e-05,
          8.2552e-05, 2.6226e-04, 8.2552e-05, 1.2612e-04, 3.1531e-05, 3.1531e-05,
          1.6761e-04, 3.1531e-05, 3.1531e-05, 8.2552e-05, 2.7585e-04, 3.1531e-05,
          3.7050e-04, 1.7715e-04, 8.2552e-05, 1.2612e-04, 3.1531e-05, 1.2612e-04,
          2.1863e-04, 1.6761e-04, 3.1328e-04, 8.2552e-05, 3.1531e-05, 2.6226e-04,
          2.6226e-04, 3.1328e-04, 3.1531e-05, 2.6226e-04, 5.7411e-04, 3.1531e-05,
          3.1531e-05, 3.1531e-05, 3.1531e-05, 1.0592e-04, 1.2612e-04, 1.2612e-04,
          3.1531e-05, 1.7715e-04, 1.2612e-04, 1.2612e-04, 3.1531e-05, 1.2612e-04,
          3.1531e-05, 3.1531e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4403e-03, 5.2643e-02, 1.3870e-02, 1.7044e-02, 3.9734e-02,
          3.6469e-02, 1.2062e-02, 8.8072e-04, 6.9284e-04, 2.7332e-03, 4.3716e-03,
          1.4563e-03, 1.3227e-03, 4.7064e-04, 3.1509e-03, 1.0786e-03, 1.5564e-03,
          1.0157e-03, 6.6948e-04, 2.2721e-04, 3.6287e-04, 3.3927e-04, 1.7872e-03,
          5.6305e-03, 3.9139e-03, 9.2850e-03, 4.3907e-03, 3.8123e-04, 1.7529e-03,
          6.5184e-04, 1.6861e-03, 4.6897e-04, 4.0474e-03, 2.5158e-03, 9.2411e-04,
          5.0316e-03, 6.5947e-04, 3.1071e-03, 3.2272e-03, 2.7966e-04, 3.3913e-03,
          1.0986e-03, 8.2064e-04, 8.1778e-04, 6.1512e-04, 2.0542e-03, 5.4092e-03,
          3.4595e-04, 4.0865e-04, 3.4447e-03, 9.6035e-04, 9.5177e-04, 9.1171e-04,
          3.0923e-04, 5.8651e-04, 3.9577e-04, 7.5436e-04, 1.1644e-03, 1.2789e-03,
          1.0691e-03, 7.3195e-04, 1.0853e-03, 1.7881e-03, 1.6222e-03, 1.3626e-02,
          1.3294e-03, 8.9455e-04, 1.7624e-03, 4.6778e-04, 4.3583e-04, 5.0354e-04,
          5.5361e-04, 6.0501e-03, 4.2200e-04, 1.1778e-03, 2.7633e-04, 2.7037e-04,
          5.6362e-04, 5.7459e-04, 7.3910e-04, 4.0207e-03, 9.5510e-04, 1.2026e-03,
          7.7057e-04, 3.0003e-03, 7.8154e-04, 2.4948e-03, 7.9823e-04, 7.4291e-04,
          5.4207e-03, 3.5763e-03, 6.6471e-04, 2.8744e-03, 7.6675e-04, 6.6757e-04,
          4.2820e-04, 2.7490e-04, 1.2636e-03, 1.7605e-03, 6.4516e-04, 9.6560e-04,
          3.7079e-03, 4.7922e-04, 6.7568e-04, 7.4692e-03, 9.6741e-03, 8.8549e-04,
          2.4586e-03, 1.4753e-03, 5.7220e-03, 8.0824e-04, 1.0643e-03, 1.3151e-03,
          2.3384e-03, 1.7052e-03, 7.5150e-04, 1.0857e-02, 1.0519e-03, 7.3481e-04,
          9.5272e-04, 3.2005e-03, 3.6955e-04, 5.8460e-04, 1.4732e-02, 3.1900e-04,
          4.6158e-04, 4.8494e-04, 6.4468e-04, 8.5068e-03, 1.8673e-03, 2.5597e-03,
          1.6594e-03, 1.0729e-03, 1.1997e-03, 8.1825e-04, 3.2024e-03, 7.5674e-04,
          5.4436e-03, 1.2598e-03, 1.2808e-03, 2.9316e-03, 3.2711e-03, 1.1024e-03,
          5.8126e-04, 7.1049e-04, 9.3460e-04, 8.5735e-04, 8.2016e-04, 7.2174e-03,
          5.8289e-03, 7.9117e-03, 1.3695e-03, 2.3785e-03, 1.4887e-03, 4.5853e-03,
          3.8719e-03, 2.6264e-03, 1.1387e-03, 1.0700e-03, 7.1487e-03, 2.1095e-03,
          6.8054e-03, 4.2076e-03, 3.1776e-03, 3.9825e-03, 1.3580e-03, 1.0700e-03,
          2.5902e-03, 1.2798e-03, 4.2267e-03, 2.5978e-03, 1.8606e-03, 5.0621e-03,
          2.3403e-03, 1.4477e-03, 1.7881e-03, 2.8687e-03, 9.7733e-03, 3.7518e-03,
          4.1199e-03, 5.9128e-03, 1.7883e-02, 2.2411e-03, 1.3676e-03, 1.6069e-03,
          2.8820e-03, 1.1358e-03, 1.9703e-03, 9.7275e-04, 7.9203e-04, 8.6365e-03,
          1.6298e-03, 4.4823e-04, 3.6221e-03, 3.2063e-03, 2.6608e-03, 4.4365e-03,
          1.7004e-03, 1.2074e-03, 1.3351e-02, 3.2425e-03, 2.3251e-03, 6.7139e-03,
          2.2411e-03, 6.5575e-03, 2.2602e-03, 1.1816e-03, 5.7411e-03, 1.5955e-03,
          1.1406e-03, 1.9169e-03, 1.0500e-03, 1.1787e-03, 6.1264e-03, 7.4434e-04,
          2.5673e-03, 2.0657e-03, 3.4332e-03, 1.8492e-03, 5.1842e-03, 1.2560e-03,
          5.0316e-03, 3.1090e-03, 8.3847e-03, 2.0828e-03, 5.0781e-02, 7.2937e-03,
          2.4967e-03, 1.2302e-03, 1.6508e-03, 3.1395e-03, 7.8125e-03, 1.4286e-03,
          1.6232e-03, 1.0452e-03, 8.1635e-04, 2.7046e-03, 3.1490e-03, 4.3945e-03,
          1.5430e-03, 1.4839e-03, 1.3132e-03, 2.0123e-03, 3.0556e-03, 2.4338e-03,
          4.6005e-03, 4.7493e-03, 2.2621e-03, 2.5387e-03, 4.7493e-03, 1.9684e-02,
          1.7910e-03, 1.6680e-03, 2.9469e-03, 3.7479e-03, 4.2458e-03, 6.2561e-03,
          2.0161e-03, 1.1110e-03, 2.2106e-03, 2.3823e-03, 9.7752e-04, 1.7338e-03,
          3.3951e-03, 1.8339e-03, 1.1368e-03, 4.0970e-03, 4.7455e-03, 1.3374e-02,
          3.4161e-03, 2.7161e-03, 4.7363e-02, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          2.5749e-05, 2.5749e-05, 1.0705e-04, 1.9765e-04, 2.5749e-05, 1.0705e-04,
          1.9765e-04, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          1.9920e-04, 2.5749e-05, 1.9765e-04, 1.9765e-04, 2.5749e-05, 2.5749e-05,
          2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          2.5749e-05, 2.5749e-05, 1.9765e-04, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          2.5749e-05, 1.0705e-04, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          1.0705e-04, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05,
          2.5749e-05, 3.1543e-04, 1.0705e-04, 1.0705e-04, 2.5749e-05, 2.5749e-05,
          1.0705e-04, 1.0705e-04, 1.0705e-04, 2.5749e-05, 2.5749e-05, 1.9801e-04,
          2.5749e-05, 4.3774e-04, 1.9801e-04, 2.5749e-05, 2.5749e-05, 3.5620e-04,
          2.5749e-05, 2.5749e-05, 1.1635e-04, 1.0705e-04, 1.0705e-04, 1.9765e-04,
          1.0705e-04, 2.5749e-05, 1.9765e-04, 1.9765e-04, 2.5749e-05, 4.3750e-04,
          4.3750e-04, 5.2786e-04, 2.5749e-05, 2.5749e-05, 1.9765e-04, 1.9765e-04,
          2.5749e-05, 2.5749e-05, 2.5749e-05, 2.5749e-05, 1.9801e-04, 2.5749e-05,
          5.2786e-04, 1.9765e-04, 2.5749e-05, 2.5749e-05, 1.0705e-04, 2.5749e-05,
          2.5749e-05, 1.9765e-04, 2.5749e-05, 1.0705e-04, 2.5749e-05, 2.5749e-05,
          4.3750e-04, 2.5749e-05, 2.5749e-05, 2.5749e-05, 2.8062e-04, 1.0705e-04,
          2.1305e-03, 1.9765e-04, 1.0705e-04, 1.0705e-04, 3.5620e-04, 1.0705e-04,
          4.0603e-04, 4.3750e-04, 1.9765e-04, 2.5749e-05, 2.5749e-05, 1.0705e-04,
          4.3750e-04, 2.0409e-03, 2.5749e-05, 1.0705e-04, 4.3750e-04, 2.5749e-05,
          1.0705e-04, 2.5749e-05, 1.0705e-04, 1.0705e-04, 3.1543e-04, 1.1635e-04,
          2.5749e-05, 2.5749e-05, 1.0705e-04, 3.8290e-04, 2.5749e-05, 3.1543e-04,
          1.0705e-04, 2.5749e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.2223e-03, 4.4464e-02, 1.5823e-02, 1.2878e-02, 6.2164e-02,
          2.9572e-02, 1.2703e-02, 6.6757e-04, 6.1893e-04, 1.4286e-03, 3.1223e-03,
          1.0176e-03, 7.3147e-04, 5.4550e-04, 1.5678e-03, 7.0667e-04, 8.8310e-04,
          6.0225e-04, 4.7207e-04, 6.1083e-04, 7.5245e-04, 6.0606e-04, 1.0309e-03,
          4.2229e-03, 3.2234e-03, 9.2392e-03, 2.2621e-03, 6.1035e-04, 8.8072e-04,
          5.9271e-04, 1.0481e-03, 3.5405e-04, 1.6470e-03, 9.8896e-04, 6.4898e-04,
          2.9678e-03, 1.7214e-03, 1.8387e-03, 1.3361e-03, 5.4455e-04, 4.0588e-03,
          2.8381e-03, 2.1172e-03, 2.1114e-03, 1.2207e-03, 1.0452e-03, 2.9049e-03,
          4.5300e-04, 8.1730e-04, 7.4196e-03, 8.9264e-04, 6.9189e-04, 5.6934e-04,
          4.9591e-04, 5.5122e-04, 7.5769e-04, 1.4801e-03, 2.0332e-03, 3.9711e-03,
          3.2215e-03, 8.5115e-04, 1.6985e-03, 1.7099e-03, 3.1872e-03, 5.7220e-03,
          2.2049e-03, 1.1435e-03, 1.3313e-03, 1.6232e-03, 1.7910e-03, 1.1759e-03,
          1.2074e-03, 2.6417e-03, 7.5531e-04, 8.8120e-04, 5.1689e-04, 3.6335e-04,
          7.6103e-04, 1.0443e-03, 1.4067e-03, 2.0943e-03, 1.8959e-03, 2.8381e-03,
          2.0351e-03, 2.1820e-03, 1.3027e-03, 1.6394e-03, 9.8991e-04, 1.5793e-03,
          3.4809e-03, 2.6836e-03, 1.5535e-03, 2.1973e-03, 1.7490e-03, 1.4439e-03,
          5.7888e-04, 5.5265e-04, 6.2227e-04, 1.2712e-03, 1.3580e-03, 1.2150e-03,
          2.2526e-03, 7.9298e-04, 8.2541e-04, 5.3444e-03, 2.8534e-03, 1.2121e-03,
          1.7767e-03, 5.4893e-03, 4.1695e-03, 2.1248e-03, 1.7014e-03, 1.5144e-03,
          2.5234e-03, 4.2114e-03, 1.3361e-03, 4.4174e-03, 2.1477e-03, 1.2856e-03,
          1.1454e-03, 2.2316e-03, 8.4066e-04, 4.1175e-04, 2.8210e-03, 6.0701e-04,
          1.0490e-03, 1.1225e-03, 1.3208e-03, 2.6608e-03, 8.9598e-04, 1.9646e-03,
          2.0885e-03, 1.3485e-03, 8.5115e-04, 9.6655e-04, 2.9182e-03, 3.2558e-03,
          2.4910e-03, 1.4944e-03, 1.0557e-03, 2.0142e-03, 1.3781e-03, 1.1206e-03,
          1.2636e-03, 9.7847e-04, 1.2102e-03, 1.0881e-03, 1.3723e-03, 3.2387e-03,
          3.3283e-03, 3.1910e-03, 2.5043e-03, 2.3708e-03, 2.6207e-03, 9.9869e-03,
          2.6112e-03, 3.1052e-03, 9.4652e-04, 3.3493e-03, 6.7749e-03, 1.8911e-03,
          3.0212e-03, 1.9341e-03, 1.6222e-03, 1.7796e-03, 2.0256e-03, 9.5034e-04,
          1.8797e-03, 9.0456e-04, 1.9836e-03, 1.2178e-03, 1.1597e-03, 2.1496e-03,
          3.3779e-03, 2.6798e-03, 3.5801e-03, 1.0483e-02, 7.7286e-03, 1.8034e-03,
          2.1114e-03, 2.6550e-03, 4.6997e-03, 1.2999e-03, 1.1721e-03, 1.2236e-03,
          1.7233e-03, 1.5173e-03, 1.5411e-03, 2.0008e-03, 1.3323e-03, 2.0504e-03,
          8.2827e-04, 1.1034e-03, 1.3161e-03, 1.8263e-03, 1.3161e-03, 1.9569e-03,
          3.7365e-03, 2.3804e-03, 8.5449e-03, 4.2534e-03, 1.7424e-03, 3.0518e-03,
          5.7564e-03, 8.8501e-03, 9.0103e-03, 8.1968e-04, 3.9482e-03, 1.0452e-03,
          1.0490e-03, 2.7752e-03, 3.1967e-03, 1.0958e-03, 2.5845e-03, 1.5316e-03,
          1.9083e-03, 1.5774e-03, 1.5926e-03, 1.0023e-03, 2.1992e-03, 1.3075e-03,
          3.4828e-03, 2.2079e-02, 1.4305e-02, 2.0447e-03, 9.4147e-03, 2.3098e-03,
          3.5400e-03, 3.5515e-03, 6.7062e-03, 4.8904e-03, 5.2185e-03, 1.6813e-03,
          4.1008e-03, 1.3351e-03, 1.8978e-03, 2.0466e-03, 4.3831e-03, 7.1030e-03,
          3.9864e-03, 3.8471e-03, 2.3937e-03, 2.4147e-03, 2.8419e-03, 5.0812e-03,
          1.7914e-02, 5.9776e-03, 1.4793e-02, 8.3313e-03, 2.1164e-02, 1.3100e-02,
          5.5656e-03, 8.0338e-03, 1.2093e-02, 6.2523e-03, 2.0370e-03, 3.2043e-03,
          2.9411e-03, 1.2760e-03, 2.7142e-03, 2.2793e-03, 2.5578e-03, 3.4904e-03,
          2.5616e-03, 1.1091e-03, 9.1839e-04, 2.7828e-03, 3.6716e-03, 1.3176e-02,
          8.1177e-03, 4.0779e-03, 1.4046e-02, 5.1208e-02, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 1.4997e-04, 3.2425e-05, 3.2425e-05,
          1.4997e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          1.4997e-04, 3.2425e-05, 3.4785e-04, 3.4785e-04, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.4785e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 2.3043e-04, 3.4785e-04, 3.2425e-05, 3.2425e-05,
          2.3043e-04, 3.2425e-05, 2.3043e-04, 3.2425e-05, 3.2425e-05, 1.4997e-04,
          3.2425e-05, 3.2425e-05, 3.4785e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.4997e-04,
          3.2425e-05, 3.2425e-05, 3.4785e-04, 3.2425e-05, 3.2425e-05, 2.3043e-04,
          3.2425e-05, 1.4997e-04, 3.2425e-05, 3.2425e-05, 3.4785e-04, 1.4997e-04,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.4997e-04, 3.2425e-05,
          3.4785e-04, 3.4785e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.4785e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.4997e-04, 3.2425e-05,
          2.3043e-04, 1.4997e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.4785e-04, 3.2425e-05, 3.4785e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          2.3043e-04, 2.3043e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 2.3043e-04, 4.7803e-04, 2.3043e-04, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.9577e-04, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.8109e-03, 5.5389e-02, 1.4839e-02, 1.3931e-02, 3.4912e-02,
          3.9124e-02, 9.9030e-03, 5.6124e-04, 4.3821e-04, 1.6642e-03, 2.4986e-03,
          8.6021e-04, 1.3371e-03, 5.6267e-04, 1.9951e-03, 6.4945e-04, 2.9449e-03,
          1.1778e-03, 4.6325e-04, 4.3297e-04, 7.6103e-04, 5.2357e-04, 1.3809e-03,
          3.1548e-03, 2.6779e-03, 7.7705e-03, 4.1008e-03, 5.9891e-04, 2.8458e-03,
          8.5306e-04, 1.1749e-03, 8.3303e-04, 4.3945e-03, 1.4563e-03, 1.0872e-03,
          4.1122e-03, 6.3038e-04, 6.2790e-03, 2.9812e-03, 4.9305e-04, 6.3286e-03,
          1.4172e-03, 1.0853e-03, 8.5592e-04, 1.1120e-03, 2.4261e-03, 7.5531e-03,
          4.5252e-04, 6.3229e-04, 3.0804e-03, 1.8206e-03, 8.5878e-04, 8.7118e-04,
          2.8586e-04, 5.3501e-04, 4.6110e-04, 1.3247e-03, 8.0490e-04, 1.0910e-03,
          8.8596e-04, 1.3456e-03, 1.0538e-03, 2.4700e-03, 1.9426e-03, 1.0536e-02,
          1.0853e-03, 7.6485e-04, 1.8072e-03, 8.3447e-04, 6.6328e-04, 6.1560e-04,
          7.8487e-04, 4.6883e-03, 5.6314e-04, 1.1063e-03, 2.6989e-04, 3.0923e-04,
          5.9843e-04, 1.5612e-03, 8.6403e-04, 2.0676e-03, 6.7854e-04, 6.9714e-04,
          7.1287e-04, 2.8725e-03, 7.4434e-04, 1.5135e-03, 7.1383e-04, 6.8760e-04,
          3.5610e-03, 4.1428e-03, 5.5552e-04, 2.1763e-03, 9.5367e-04, 1.0319e-03,
          6.1083e-04, 3.6907e-04, 1.5383e-03, 5.2528e-03, 1.6727e-03, 2.1496e-03,
          2.5444e-03, 4.2915e-04, 4.7112e-04, 3.2234e-03, 4.3449e-03, 1.4067e-03,
          2.4319e-03, 2.7294e-03, 6.3820e-03, 7.0858e-04, 7.7438e-04, 1.5020e-03,
          1.8959e-03, 4.7798e-03, 1.2560e-03, 1.0399e-02, 1.4715e-03, 8.7547e-04,
          1.1892e-03, 5.7831e-03, 4.6110e-04, 3.5048e-04, 4.2839e-03, 4.2462e-04,
          1.1444e-03, 5.3120e-04, 6.7139e-04, 7.4730e-03, 1.6251e-03, 2.6703e-03,
          1.2436e-03, 8.0490e-04, 5.3692e-04, 6.3610e-04, 3.0117e-03, 9.7370e-04,
          9.2926e-03, 1.8187e-03, 1.5278e-03, 2.3098e-03, 3.3340e-03, 1.4286e-03,
          6.9809e-04, 5.1832e-04, 9.2745e-04, 1.8845e-03, 7.0143e-04, 5.8746e-03,
          5.5542e-03, 5.7983e-03, 1.7004e-03, 4.7531e-03, 1.7891e-03, 1.1696e-02,
          5.2414e-03, 5.3749e-03, 5.7602e-04, 1.0109e-03, 1.4847e-02, 4.1695e-03,
          8.9645e-03, 9.1705e-03, 4.3106e-03, 3.0422e-03, 3.6564e-03, 2.5177e-03,
          1.9178e-03, 2.3632e-03, 3.0556e-03, 1.9836e-03, 1.9836e-03, 1.0643e-02,
          1.9083e-03, 9.9182e-04, 2.2259e-03, 5.0507e-03, 1.1459e-02, 2.1076e-03,
          3.1719e-03, 4.2877e-03, 8.7509e-03, 1.4811e-03, 1.1396e-03, 2.6169e-03,
          1.5707e-03, 1.5154e-03, 1.1225e-03, 1.3704e-03, 6.3372e-04, 4.0321e-03,
          9.7179e-04, 6.8665e-04, 4.6577e-03, 4.5624e-03, 2.2564e-03, 4.9095e-03,
          1.7462e-03, 2.0504e-03, 1.3840e-02, 2.8534e-03, 2.1954e-03, 6.0768e-03,
          1.3332e-03, 7.1640e-03, 3.2101e-03, 7.4148e-04, 4.3564e-03, 2.1973e-03,
          1.0128e-03, 3.6030e-03, 1.2722e-03, 7.2336e-04, 5.6610e-03, 6.6519e-04,
          6.7024e-03, 3.2444e-03, 2.5978e-03, 1.2302e-03, 8.5220e-03, 1.4763e-03,
          1.6113e-02, 2.3308e-03, 8.5144e-03, 1.9207e-03, 1.5045e-02, 4.7989e-03,
          1.9112e-03, 9.3174e-04, 1.6203e-03, 2.3041e-03, 5.9929e-03, 2.1095e-03,
          1.8473e-03, 1.2426e-03, 5.9795e-04, 1.8797e-03, 4.0588e-03, 3.7098e-03,
          1.2827e-03, 1.3504e-03, 2.1725e-03, 7.4120e-03, 5.7373e-03, 3.3035e-03,
          9.3842e-03, 2.8038e-03, 1.2960e-03, 1.7500e-03, 2.2087e-03, 1.3885e-02,
          1.6327e-03, 2.6722e-03, 4.2458e-03, 7.0190e-03, 5.3902e-03, 7.2021e-03,
          3.3970e-03, 1.2627e-03, 1.8597e-03, 2.8305e-03, 8.0490e-04, 1.5039e-03,
          3.6774e-03, 1.7071e-03, 1.0061e-03, 5.1193e-03, 4.0131e-03, 1.0498e-02,
          2.2240e-03, 1.6375e-03, 1.4893e-02, 5.8479e-03, 4.0131e-02, 2.4259e-05,
          2.4259e-05, 2.4259e-05, 2.4259e-05, 2.7323e-04, 2.4259e-05, 2.4259e-05,
          2.7323e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          1.6725e-04, 2.4259e-05, 1.3030e-04, 2.7323e-04, 2.4259e-05, 2.4259e-05,
          2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          2.4259e-05, 2.4259e-05, 2.7323e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          1.3030e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          2.4259e-05, 1.3030e-04, 1.3030e-04, 1.3030e-04, 2.4259e-05, 2.4259e-05,
          1.3030e-04, 2.4259e-05, 1.3030e-04, 2.4259e-05, 2.4259e-05, 2.7323e-04,
          2.4259e-05, 2.4259e-05, 2.7323e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 1.3030e-04, 2.7323e-04,
          2.4259e-05, 2.4259e-05, 1.3030e-04, 1.3030e-04, 2.4259e-05, 1.3030e-04,
          1.3030e-04, 2.1629e-03, 2.4259e-05, 2.4259e-05, 2.7323e-04, 2.7323e-04,
          2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.7323e-04, 2.4259e-05,
          2.7323e-04, 2.7323e-04, 2.4259e-05, 2.4259e-05, 1.3030e-04, 2.4259e-05,
          2.4259e-05, 1.3030e-04, 2.4259e-05, 1.3030e-04, 2.4259e-05, 2.4259e-05,
          1.3030e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.7323e-04, 2.4259e-05,
          1.3030e-04, 2.7323e-04, 1.3030e-04, 2.4259e-05, 2.4259e-05, 1.3030e-04,
          1.3030e-04, 2.4259e-05, 2.7323e-04, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          1.3030e-04, 2.8877e-03, 2.4259e-05, 2.4259e-05, 2.4259e-05, 2.4259e-05,
          1.3030e-04, 1.3030e-04, 4.3964e-04, 2.1946e-04, 1.3030e-04, 2.4259e-05,
          2.4259e-05, 1.3030e-04, 1.3030e-04, 3.3569e-04, 2.4259e-05, 1.3030e-04,
          1.3030e-04, 2.4259e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.4676e-03, 4.6570e-02, 1.1971e-02, 1.2680e-02, 3.6774e-02,
          3.5309e-02, 9.6283e-03, 8.5306e-04, 6.4087e-04, 2.0180e-03, 3.4122e-03,
          1.1663e-03, 1.3008e-03, 4.7874e-04, 3.4332e-03, 9.1982e-04, 1.3885e-03,
          8.1015e-04, 6.1798e-04, 2.5392e-04, 3.5524e-04, 3.6812e-04, 1.2293e-03,
          4.6806e-03, 6.8741e-03, 7.4272e-03, 3.2959e-03, 3.6001e-04, 1.4076e-03,
          6.2799e-04, 1.6413e-03, 4.0722e-04, 3.3913e-03, 1.6584e-03, 1.0509e-03,
          2.8305e-03, 6.7759e-04, 2.4471e-03, 2.4147e-03, 3.2616e-04, 3.5515e-03,
          1.2474e-03, 1.1034e-03, 9.6035e-04, 8.5354e-04, 1.6813e-03, 6.2485e-03,
          3.7956e-04, 4.5609e-04, 3.2730e-03, 1.1559e-03, 1.0147e-03, 8.8072e-04,
          3.1996e-04, 6.0129e-04, 4.4155e-04, 8.8120e-04, 1.1368e-03, 1.2941e-03,
          1.2388e-03, 8.2111e-04, 1.0977e-03, 1.7309e-03, 1.8530e-03, 1.3062e-02,
          1.5907e-03, 9.4175e-04, 1.6756e-03, 7.7581e-04, 6.1512e-04, 6.9284e-04,
          6.8569e-04, 4.0474e-03, 4.4131e-04, 1.3399e-03, 3.1471e-04, 2.8348e-04,
          5.6839e-04, 7.9823e-04, 9.2888e-04, 2.2411e-03, 9.0790e-04, 1.1587e-03,
          7.9870e-04, 2.7103e-03, 7.3385e-04, 2.1229e-03, 8.0013e-04, 8.9550e-04,
          4.1847e-03, 3.0079e-03, 9.5177e-04, 2.5120e-03, 2.5330e-03, 1.1988e-03,
          4.1556e-04, 2.9755e-04, 1.4410e-03, 2.2564e-03, 9.9182e-04, 1.1511e-03,
          3.0003e-03, 6.1226e-04, 6.4325e-04, 2.6035e-03, 6.7139e-03, 1.0586e-03,
          2.2888e-03, 1.4877e-03, 4.4823e-03, 1.1320e-03, 1.3227e-03, 1.2560e-03,
          7.6294e-03, 3.2673e-03, 9.2793e-04, 1.5137e-02, 1.7719e-03, 6.8521e-04,
          9.4414e-04, 3.3379e-03, 5.1451e-04, 5.4789e-04, 6.9466e-03, 3.6049e-04,
          7.2956e-04, 6.0606e-04, 8.5258e-04, 7.4921e-03, 1.6441e-03, 2.5330e-03,
          2.0180e-03, 1.1063e-03, 1.0080e-03, 7.1859e-04, 2.9373e-03, 1.0567e-03,
          5.1346e-03, 1.4734e-03, 1.7347e-03, 2.4090e-03, 2.5959e-03, 1.1702e-03,
          7.0667e-04, 1.0061e-03, 1.4887e-03, 9.7942e-04, 1.0223e-03, 1.5320e-02,
          4.4746e-03, 7.3929e-03, 3.0746e-03, 2.2869e-03, 1.6222e-03, 4.7417e-03,
          2.8248e-03, 2.1534e-03, 1.0433e-03, 1.2465e-03, 1.2947e-02, 2.7142e-03,
          7.2670e-03, 3.6488e-03, 2.6398e-03, 3.3970e-03, 1.6470e-03, 9.3651e-04,
          1.9760e-03, 9.8705e-04, 5.4398e-03, 3.8986e-03, 1.5459e-03, 4.1618e-03,
          4.2000e-03, 1.9464e-03, 2.5120e-03, 2.6855e-03, 7.8659e-03, 2.5749e-03,
          3.4466e-03, 4.3297e-03, 9.5596e-03, 2.1038e-03, 1.1272e-03, 2.5063e-03,
          2.6684e-03, 1.5898e-03, 1.7023e-03, 1.1501e-03, 9.0170e-04, 4.4899e-03,
          1.3599e-03, 6.2370e-04, 3.4351e-03, 2.8801e-03, 1.8797e-03, 3.9368e-03,
          2.4357e-03, 1.3580e-03, 1.0994e-02, 2.7752e-03, 1.7014e-03, 5.0125e-03,
          1.7519e-03, 6.0234e-03, 3.9520e-03, 1.0138e-03, 5.6381e-03, 1.5154e-03,
          1.2102e-03, 2.0008e-03, 1.2226e-03, 1.0576e-03, 9.1476e-03, 9.2745e-04,
          3.0174e-03, 2.6455e-03, 2.5806e-03, 2.0390e-03, 5.1460e-03, 1.5640e-03,
          6.4163e-03, 2.5654e-03, 7.0076e-03, 2.0180e-03, 1.9699e-02, 4.6768e-03,
          2.4338e-03, 1.7319e-03, 3.2101e-03, 6.9313e-03, 8.7891e-03, 2.0466e-03,
          1.6127e-03, 9.5463e-04, 1.0576e-03, 4.1885e-03, 8.4457e-03, 8.3542e-03,
          2.0180e-03, 1.9503e-03, 1.7176e-03, 2.7847e-03, 4.3793e-03, 2.6932e-03,
          4.2419e-03, 4.3716e-03, 2.5578e-03, 3.1967e-03, 4.8943e-03, 1.6586e-02,
          1.9932e-03, 3.8242e-03, 5.2681e-03, 6.2675e-03, 4.5738e-03, 5.2261e-03,
          2.2068e-03, 1.1520e-03, 3.9253e-03, 5.5389e-03, 1.3876e-03, 2.4147e-03,
          4.5242e-03, 1.6127e-03, 1.4877e-03, 4.2076e-03, 4.5586e-03, 1.1688e-02,
          3.8013e-03, 2.2087e-03, 2.0660e-02, 7.5951e-03, 1.3512e-02, 4.9286e-02,
          2.9564e-05, 2.9564e-05, 2.9564e-05, 1.2732e-04, 2.9564e-05, 2.9564e-05,
          1.2732e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          1.2732e-04, 2.9564e-05, 3.2663e-04, 3.2663e-04, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 1.2732e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.2900e-04, 2.2900e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.2900e-04, 2.9564e-05, 2.2900e-04, 2.9564e-05, 2.9564e-05, 3.2663e-04,
          2.9564e-05, 2.9564e-05, 3.2711e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 1.2732e-04, 2.9564e-05, 2.9564e-05, 1.2732e-04,
          2.9564e-05, 2.9564e-05, 3.2663e-04, 1.2732e-04, 5.5838e-04, 2.9564e-05,
          1.1749e-03, 1.4725e-03, 2.9564e-05, 2.9564e-05, 1.2732e-04, 1.2732e-04,
          1.2732e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05, 3.2663e-04, 2.9564e-05,
          3.2663e-04, 3.2663e-04, 2.9564e-05, 1.2732e-04, 1.2732e-04, 2.9564e-05,
          2.9564e-05, 1.2732e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.2900e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05, 1.2732e-04, 2.9564e-05,
          3.2663e-04, 1.2732e-04, 3.2663e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          1.2732e-04, 2.9564e-05, 3.2663e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          1.3742e-03, 2.2900e-04, 2.9564e-05, 2.9564e-05, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 2.2900e-04, 1.2177e-04, 2.2900e-04, 2.9564e-05,
          2.9564e-05, 2.9564e-05, 2.9564e-05, 4.9496e-04, 2.9564e-05, 2.9564e-05,
          2.9564e-05, 2.9564e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.0474e-03, 4.9103e-02, 1.4252e-02, 1.1948e-02, 4.4952e-02,
          3.8177e-02, 1.0147e-02, 8.5688e-04, 6.2609e-04, 1.2569e-03, 2.9297e-03,
          9.6655e-04, 7.9155e-04, 4.9877e-04, 1.6336e-03, 1.0185e-03, 9.2077e-04,
          5.8556e-04, 5.7888e-04, 3.6001e-04, 4.3654e-04, 4.8733e-04, 8.9598e-04,
          4.2305e-03, 4.9438e-03, 7.7477e-03, 2.2640e-03, 4.1485e-04, 8.5354e-04,
          5.4646e-04, 1.3838e-03, 3.3855e-04, 1.4791e-03, 7.9679e-04, 7.7248e-04,
          1.5821e-03, 9.8228e-04, 1.3876e-03, 1.0958e-03, 4.0102e-04, 2.9869e-03,
          1.8415e-03, 1.4544e-03, 1.4944e-03, 1.1120e-03, 1.0395e-03, 3.3989e-03,
          4.2772e-04, 6.3372e-04, 4.6310e-03, 1.0328e-03, 9.0647e-04, 6.6662e-04,
          4.3511e-04, 6.3372e-04, 6.3562e-04, 8.9264e-04, 1.4925e-03, 1.9722e-03,
          2.4490e-03, 8.6737e-04, 1.3094e-03, 1.5354e-03, 2.3632e-03, 6.8665e-03,
          1.6594e-03, 1.0605e-03, 1.0748e-03, 1.1044e-03, 1.0328e-03, 1.1883e-03,
          1.1158e-03, 2.2659e-03, 6.1607e-04, 1.2932e-03, 4.4823e-04, 3.3975e-04,
          7.1526e-04, 7.2527e-04, 8.4209e-04, 1.4772e-03, 1.0481e-03, 1.7815e-03,
          1.3371e-03, 2.7599e-03, 9.5987e-04, 1.8330e-03, 7.4387e-04, 1.0920e-03,
          3.7136e-03, 2.5520e-03, 1.4954e-03, 2.4147e-03, 4.3907e-03, 2.8114e-03,
          5.2738e-04, 4.3154e-04, 9.7847e-04, 1.6346e-03, 9.4032e-04, 9.8038e-04,
          2.3403e-03, 5.7507e-04, 6.1703e-04, 1.7719e-03, 2.7637e-03, 1.2121e-03,
          2.3422e-03, 2.0943e-03, 3.4466e-03, 1.5841e-03, 1.5163e-03, 1.2150e-03,
          5.5580e-03, 3.4790e-03, 1.2131e-03, 6.8741e-03, 4.1771e-03, 9.6416e-04,
          1.0443e-03, 2.1629e-03, 6.1560e-04, 5.2071e-04, 2.8362e-03, 3.5310e-04,
          6.9284e-04, 7.2241e-04, 1.2093e-03, 2.6817e-03, 1.2035e-03, 2.6054e-03,
          2.0142e-03, 1.0138e-03, 1.3981e-03, 7.3576e-04, 2.6302e-03, 1.8768e-03,
          2.6150e-03, 1.5249e-03, 1.7071e-03, 2.2411e-03, 1.4801e-03, 1.0767e-03,
          8.4782e-04, 1.2007e-03, 1.1234e-03, 8.6355e-04, 9.0361e-04, 5.9052e-03,
          2.9087e-03, 2.9926e-03, 5.4359e-03, 2.1343e-03, 2.0466e-03, 4.7226e-03,
          2.1095e-03, 2.0370e-03, 1.2560e-03, 1.9283e-03, 8.8120e-03, 2.2717e-03,
          3.6373e-03, 2.1648e-03, 1.4906e-03, 1.8969e-03, 1.3790e-03, 7.4387e-04,
          1.5764e-03, 7.1430e-04, 2.5826e-03, 1.6727e-03, 9.5606e-04, 2.0523e-03,
          1.0956e-02, 4.3945e-03, 4.5204e-03, 3.6926e-03, 5.4512e-03, 2.0199e-03,
          1.7557e-03, 2.7752e-03, 4.4212e-03, 1.9436e-03, 1.1921e-03, 1.6155e-03,
          2.5406e-03, 1.6823e-03, 1.3380e-03, 1.2512e-03, 8.9550e-04, 1.8654e-03,
          1.2608e-03, 6.8235e-04, 1.4420e-03, 1.6222e-03, 1.3494e-03, 1.8187e-03,
          5.4703e-03, 1.7204e-03, 7.3395e-03, 2.7771e-03, 1.3676e-03, 2.6493e-03,
          2.7599e-03, 5.7220e-03, 6.5117e-03, 1.2760e-03, 4.8752e-03, 9.9754e-04,
          1.3399e-03, 1.9083e-03, 1.4162e-03, 1.0281e-03, 3.2310e-03, 1.1377e-03,
          1.8463e-03, 1.7691e-03, 1.5888e-03, 1.1702e-03, 2.3003e-03, 1.5450e-03,
          4.4594e-03, 4.9515e-03, 7.1640e-03, 1.7834e-03, 9.7046e-03, 2.2984e-03,
          2.7084e-03, 4.3755e-03, 1.0544e-02, 2.3148e-02, 9.4604e-03, 2.1801e-03,
          1.9569e-03, 9.6846e-04, 1.4439e-03, 3.2120e-03, 6.1760e-03, 1.4389e-02,
          3.8090e-03, 2.9087e-03, 2.0332e-03, 1.9894e-03, 3.5152e-03, 2.7695e-03,
          5.9052e-03, 5.6114e-03, 5.6114e-03, 7.0877e-03, 1.0094e-02, 1.0292e-02,
          3.8033e-03, 8.8730e-03, 1.3252e-02, 7.5836e-03, 2.6379e-03, 3.7022e-03,
          1.9741e-03, 1.0376e-03, 4.2229e-03, 4.8065e-03, 3.0460e-03, 3.8052e-03,
          3.4542e-03, 9.8896e-04, 1.4172e-03, 3.1624e-03, 3.4599e-03, 8.9645e-03,
          6.5002e-03, 2.2583e-03, 1.2520e-02, 9.8038e-03, 7.1869e-03, 2.5742e-02,
          7.8369e-02, 3.1590e-05, 3.1590e-05, 3.4857e-04, 3.1590e-05, 3.1590e-05,
          1.3506e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          1.3506e-04, 3.1590e-05, 3.4857e-04, 1.3506e-04, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 1.3506e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 2.4533e-04, 3.1590e-05, 3.1590e-05, 3.4857e-04,
          3.1590e-05, 3.1590e-05, 3.4928e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 1.3506e-04, 3.1590e-05, 3.1590e-05, 3.4857e-04,
          3.1590e-05, 3.1590e-05, 1.3506e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 2.4533e-04, 3.1590e-05, 3.1590e-05, 3.4857e-04, 1.3506e-04,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.4857e-04, 3.1590e-05,
          3.4857e-04, 3.4857e-04, 3.1590e-05, 3.1590e-05, 1.3506e-04, 3.1590e-05,
          1.3506e-04, 2.4533e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05, 1.3506e-04, 3.1590e-05,
          2.4533e-04, 3.4857e-04, 1.3506e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          1.3506e-04, 3.1590e-05, 3.4857e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          2.4533e-04, 2.4533e-04, 3.1590e-05, 3.1590e-05, 3.1590e-05, 3.1590e-05,
          2.4533e-04, 7.6056e-04, 2.4533e-04, 4.4942e-04, 2.4533e-04, 3.1590e-05,
          3.1590e-05, 1.3506e-04, 3.1590e-05, 9.0647e-04, 3.6621e-04, 1.0948e-03,
          3.1590e-05, 3.1590e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.3030e-03, 5.7922e-02, 1.4458e-02, 1.3115e-02, 4.1962e-02,
          4.4647e-02, 1.1185e-02, 1.0319e-03, 5.5885e-04, 1.4153e-03, 2.7695e-03,
          1.0138e-03, 8.9169e-04, 4.4632e-04, 1.8234e-03, 1.0929e-03, 8.3971e-04,
          5.0402e-04, 6.8951e-04, 2.5940e-04, 3.2163e-04, 3.5667e-04, 9.3985e-04,
          4.7455e-03, 4.7150e-03, 8.5449e-03, 2.4014e-03, 3.1662e-04, 8.3542e-04,
          4.7994e-04, 1.5955e-03, 3.3021e-04, 1.8034e-03, 1.0595e-03, 8.0729e-04,
          1.8311e-03, 8.0967e-04, 1.3885e-03, 1.3609e-03, 2.8777e-04, 2.1667e-03,
          1.0242e-03, 8.5735e-04, 8.7786e-04, 8.5640e-04, 1.1387e-03, 3.2330e-03,
          3.2425e-04, 4.3082e-04, 3.6564e-03, 1.1425e-03, 8.8835e-04, 7.4148e-04,
          4.0078e-04, 5.6553e-04, 4.1938e-04, 7.2622e-04, 1.5688e-03, 1.4639e-03,
          1.7405e-03, 7.1144e-04, 1.3208e-03, 1.5316e-03, 1.5097e-03, 7.6294e-03,
          1.1330e-03, 9.3317e-04, 9.4891e-04, 6.4754e-04, 5.2547e-04, 9.3174e-04,
          8.7261e-04, 3.0193e-03, 4.7755e-04, 1.2493e-03, 4.1342e-04, 3.0541e-04,
          5.8746e-04, 6.0272e-04, 6.5899e-04, 1.6880e-03, 7.0572e-04, 1.6670e-03,
          1.0815e-03, 2.8000e-03, 8.4877e-04, 1.9388e-03, 6.0606e-04, 6.8760e-04,
          3.8185e-03, 2.4662e-03, 1.0128e-03, 2.4433e-03, 5.1575e-03, 3.0842e-03,
          4.7398e-04, 2.9397e-04, 1.1473e-03, 1.5583e-03, 7.7391e-04, 8.8596e-04,
          2.8534e-03, 7.3814e-04, 5.7936e-04, 2.1915e-03, 3.8986e-03, 9.1934e-04,
          2.5272e-03, 9.5940e-04, 3.8071e-03, 8.6784e-04, 1.1797e-03, 9.8419e-04,
          8.0719e-03, 2.3136e-03, 7.8487e-04, 7.7553e-03, 3.9177e-03, 6.7759e-04,
          8.7976e-04, 2.6817e-03, 4.4560e-04, 7.1335e-04, 5.3444e-03, 2.8038e-04,
          5.4502e-04, 4.5753e-04, 7.0190e-04, 3.4161e-03, 1.7252e-03, 2.9716e-03,
          1.7595e-03, 8.3017e-04, 1.2102e-03, 6.2895e-04, 2.3689e-03, 9.6512e-04,
          2.4090e-03, 1.0576e-03, 2.8286e-03, 3.0785e-03, 2.1458e-03, 9.8991e-04,
          5.9366e-04, 1.8539e-03, 1.1740e-03, 8.0442e-04, 6.7139e-04, 9.7504e-03,
          3.3817e-03, 3.4332e-03, 2.5082e-03, 1.8320e-03, 2.0828e-03, 2.5063e-03,
          1.9646e-03, 1.3800e-03, 1.5135e-03, 8.9455e-04, 6.8207e-03, 1.6527e-03,
          4.3335e-03, 2.5463e-03, 1.6842e-03, 3.2387e-03, 1.0805e-03, 7.7963e-04,
          2.3079e-03, 8.5545e-04, 5.1346e-03, 3.1853e-03, 9.4795e-04, 2.1343e-03,
          9.5825e-03, 3.2864e-03, 3.3627e-03, 1.7977e-03, 4.6692e-03, 2.4853e-03,
          2.2259e-03, 3.4428e-03, 5.7220e-03, 2.8954e-03, 2.4090e-03, 3.0994e-03,
          4.2915e-03, 1.7042e-03, 2.5253e-03, 9.4604e-04, 7.3671e-04, 3.4866e-03,
          1.8158e-03, 5.3215e-04, 2.1915e-03, 2.0752e-03, 2.1057e-03, 2.2907e-03,
          4.2419e-03, 1.1168e-03, 6.8321e-03, 1.7681e-03, 1.5049e-03, 2.9640e-03,
          1.3285e-03, 3.5706e-03, 2.6836e-03, 1.8854e-03, 7.1182e-03, 1.3466e-03,
          2.4834e-03, 1.5202e-03, 8.6689e-04, 1.3599e-03, 5.7259e-03, 8.2588e-04,
          2.1591e-03, 1.7090e-03, 2.9888e-03, 2.4624e-03, 3.2406e-03, 1.4601e-03,
          4.4403e-03, 1.9703e-03, 4.3411e-03, 1.7433e-03, 1.2939e-02, 3.9024e-03,
          1.5240e-03, 2.5520e-03, 3.6221e-03, 2.8091e-02, 1.0094e-02, 1.9026e-03,
          1.1683e-03, 9.3794e-04, 1.1368e-03, 7.1411e-03, 7.6027e-03, 1.3588e-02,
          2.9907e-03, 2.1744e-03, 1.5965e-03, 1.8940e-03, 3.1281e-03, 1.8377e-03,
          2.4261e-03, 4.2610e-03, 2.7542e-03, 5.5313e-03, 4.9896e-03, 8.2626e-03,
          1.4286e-03, 3.4122e-03, 5.3482e-03, 4.9934e-03, 4.1237e-03, 5.0659e-03,
          1.6813e-03, 1.0691e-03, 5.5351e-03, 7.3280e-03, 2.1687e-03, 3.1776e-03,
          4.6616e-03, 2.0008e-03, 2.2926e-03, 4.2191e-03, 3.9520e-03, 7.3776e-03,
          4.2000e-03, 1.6432e-03, 1.1475e-02, 4.6082e-03, 5.9242e-03, 2.2995e-02,
          3.6407e-02, 6.0547e-02, 3.4928e-05, 2.7966e-04, 3.4928e-05, 6.0654e-04,
          2.7966e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 1.4746e-04, 2.7966e-04, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 2.7966e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 1.4746e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 6.0654e-04, 3.4928e-05, 3.4928e-05, 2.7966e-04,
          3.4928e-05, 1.4746e-04, 2.7966e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          3.4928e-05, 3.4928e-05, 3.4928e-05, 1.4746e-04, 1.4746e-04, 2.7966e-04,
          3.4928e-05, 3.4928e-05, 1.4746e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          1.4746e-04, 1.4746e-04, 3.4928e-05, 3.4928e-05, 7.3862e-04, 7.3862e-04,
          3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05, 2.7966e-04, 1.4746e-04,
          7.3862e-04, 7.3862e-04, 1.4746e-04, 3.4928e-05, 1.4746e-04, 3.4928e-05,
          1.4746e-04, 1.4746e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          1.4746e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05, 2.7966e-04, 3.4928e-05,
          6.0654e-04, 1.6713e-04, 1.4746e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          1.4746e-04, 6.0654e-04, 2.7966e-04, 3.4928e-05, 3.4928e-05, 1.4746e-04,
          1.4746e-04, 6.0654e-04, 3.4928e-05, 3.4928e-05, 3.4928e-05, 3.4928e-05,
          6.7139e-04, 1.2074e-03, 1.4746e-04, 1.4746e-04, 1.4746e-04, 3.4928e-05,
          3.4928e-05, 1.4746e-04, 1.4746e-04, 7.4911e-04, 3.4928e-05, 1.1292e-03,
          1.4746e-04, 1.4746e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.3662e-03, 4.5807e-02, 2.4445e-02, 2.1286e-02, 6.9702e-02,
          2.7832e-02, 2.2308e-02, 7.7200e-04, 4.7994e-04, 8.4448e-04, 1.5659e-03,
          5.7602e-04, 1.7967e-03, 2.4853e-03, 9.8705e-04, 6.9284e-04, 4.4327e-03,
          2.7809e-03, 4.7183e-04, 2.2678e-03, 2.2717e-03, 3.5038e-03, 2.2049e-03,
          1.8463e-03, 1.4477e-03, 1.5533e-02, 4.1046e-03, 2.9354e-03, 2.9011e-03,
          2.7122e-03, 1.1234e-03, 1.6584e-03, 1.2274e-03, 6.8998e-04, 7.0238e-04,
          1.0185e-03, 6.8140e-04, 1.6432e-03, 9.4080e-04, 2.9297e-03, 1.9760e-03,
          1.8234e-03, 1.3046e-03, 1.4715e-03, 9.8038e-04, 2.9488e-03, 2.2984e-03,
          2.5787e-03, 5.0735e-03, 1.1139e-02, 1.0824e-03, 7.1430e-04, 5.3692e-04,
          5.4979e-04, 5.5408e-04, 8.7070e-04, 6.7997e-04, 7.2050e-04, 7.4244e-04,
          9.0075e-04, 6.7787e-03, 1.3275e-03, 1.5936e-03, 1.0748e-03, 2.2297e-03,
          1.3142e-03, 5.5313e-04, 8.7309e-04, 1.3809e-03, 1.4477e-03, 1.3943e-03,
          2.1877e-03, 1.7309e-03, 8.8596e-04, 9.2840e-04, 4.8733e-04, 3.5810e-04,
          8.1062e-04, 7.4053e-04, 6.6328e-04, 5.4550e-04, 5.6887e-04, 6.6328e-04,
          8.9169e-04, 1.8435e-03, 1.1988e-03, 7.8106e-04, 5.8126e-04, 8.3399e-04,
          1.2817e-03, 1.0157e-03, 9.6369e-04, 1.0242e-03, 1.5497e-03, 2.7065e-03,
          2.6550e-03, 7.2002e-04, 8.9312e-04, 1.6823e-03, 8.1444e-04, 5.9795e-04,
          1.0376e-03, 4.6039e-04, 4.0197e-04, 8.6498e-04, 1.1139e-03, 1.1116e-02,
          1.5459e-03, 2.0332e-03, 1.4324e-03, 1.1463e-03, 1.0405e-03, 8.2588e-04,
          9.6893e-04, 2.6951e-03, 1.7593e-02, 2.5120e-03, 3.7689e-03, 6.9199e-03,
          9.2010e-03, 9.2030e-04, 5.7173e-04, 4.2009e-04, 1.4496e-03, 3.7384e-04,
          6.5947e-04, 5.1117e-04, 6.0844e-04, 1.1597e-03, 1.0252e-03, 1.6928e-03,
          1.3609e-03, 7.5674e-04, 5.7077e-04, 5.0163e-04, 8.9455e-04, 1.6022e-03,
          6.6757e-03, 1.6998e-02, 4.4594e-03, 3.2616e-03, 1.8702e-03, 6.6719e-03,
          7.2098e-04, 4.6635e-04, 5.5742e-04, 5.4312e-04, 8.0156e-04, 1.1015e-03,
          1.2178e-03, 9.8228e-04, 9.2173e-04, 2.5711e-03, 2.2106e-03, 3.5057e-03,
          9.7942e-04, 1.5240e-03, 5.1403e-04, 1.2283e-03, 3.4122e-03, 1.9196e-02,
          2.2125e-03, 1.8644e-03, 5.2719e-03, 1.7462e-03, 2.1381e-03, 6.8045e-04,
          9.2125e-04, 5.5552e-04, 6.4707e-04, 7.4959e-04, 3.5381e-03, 1.3905e-03,
          9.9754e-04, 1.3084e-03, 2.4261e-03, 4.4594e-03, 1.8740e-03, 1.6890e-03,
          9.1648e-04, 1.1969e-03, 1.7500e-03, 1.2398e-03, 1.2579e-03, 1.8396e-03,
          1.0910e-03, 1.3908e-02, 2.3117e-03, 1.3847e-03, 8.3780e-04, 1.0872e-03,
          8.7166e-04, 7.0190e-04, 9.2697e-04, 5.3253e-03, 7.2050e-04, 1.0414e-03,
          2.1439e-03, 1.1879e-02, 2.7561e-03, 1.2045e-03, 9.9468e-04, 1.1654e-03,
          1.4496e-03, 1.7548e-03, 4.4365e-03, 8.8930e-04, 1.8396e-03, 3.2005e-03,
          1.2751e-03, 3.5248e-03, 2.4776e-03, 6.0368e-04, 1.2112e-03, 8.6212e-04,
          1.0462e-03, 6.5804e-03, 1.2560e-03, 7.8440e-04, 1.9016e-03, 7.9269e-03,
          6.7444e-03, 3.3112e-03, 2.1305e-03, 1.0757e-03, 3.3531e-03, 1.3485e-03,
          1.3714e-03, 1.5001e-03, 2.4128e-03, 2.8248e-03, 2.2354e-03, 1.3206e-02,
          4.4899e-03, 8.6355e-04, 1.1206e-03, 8.3685e-04, 1.8120e-03, 3.1090e-03,
          2.1229e-03, 2.8400e-03, 1.5640e-03, 3.5210e-03, 1.4076e-02, 2.6749e-02,
          6.0234e-03, 1.6069e-03, 2.4395e-03, 1.6642e-03, 3.4027e-03, 2.4033e-03,
          1.8015e-03, 4.8485e-03, 7.4539e-03, 3.8795e-03, 3.4637e-03, 3.3207e-03,
          2.0580e-03, 8.6403e-04, 1.1568e-03, 1.0185e-03, 1.0462e-03, 1.9760e-03,
          1.3294e-03, 8.7404e-04, 3.6182e-03, 1.3260e-02, 1.1726e-02, 2.3289e-03,
          2.8000e-03, 1.4601e-03, 3.0975e-03, 3.8910e-03, 4.2458e-03, 3.4847e-03,
          4.1771e-03, 1.5278e-03, 9.2712e-02, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.4237e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          1.3483e-04, 3.1769e-05, 3.1769e-05, 1.3483e-04, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 1.3483e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 1.3483e-04,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 2.3925e-04,
          2.3925e-04, 3.4237e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 1.3483e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 1.3483e-04, 3.1769e-05, 1.3483e-04, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05, 1.3483e-04, 1.3483e-04,
          3.1769e-05, 3.1769e-05, 1.3483e-04, 3.1769e-05, 3.1769e-05, 2.3925e-04,
          3.4237e-04, 3.4237e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          1.3483e-04, 3.1769e-05, 1.3483e-04, 3.4237e-04, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05, 3.1769e-05, 1.3483e-04, 1.3483e-04, 3.1769e-05,
          2.3925e-04, 1.3483e-04, 1.3483e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          3.4237e-04, 3.1769e-05, 1.3483e-04, 1.3483e-04, 1.3483e-04, 3.1769e-05,
          3.1769e-05, 2.3925e-04, 3.1769e-05, 3.1769e-05, 3.1769e-05, 3.1769e-05,
          2.3925e-04, 2.3925e-04, 2.3925e-04, 3.3808e-04, 2.3925e-04, 3.1769e-05,
          3.1769e-05, 7.5102e-04, 2.3925e-04, 7.1812e-04, 3.1769e-05, 3.1769e-05,
          3.1769e-05, 3.1769e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.2948e-03, 4.7791e-02, 1.8463e-02, 2.7069e-02, 4.0009e-02,
          3.1921e-02, 1.6708e-02, 1.1168e-03, 5.4979e-04, 1.2550e-03, 2.2507e-03,
          8.6594e-04, 2.5139e-03, 8.0719e-03, 9.7275e-04, 1.0643e-03, 7.7362e-03,
          5.8212e-03, 5.8889e-04, 9.8515e-04, 6.9094e-04, 1.4496e-03, 3.5591e-03,
          2.7428e-03, 1.0996e-03, 1.3710e-02, 7.8506e-03, 1.3256e-03, 4.8943e-03,
          1.6891e-02, 2.0504e-03, 2.0561e-03, 1.2674e-03, 1.0319e-03, 8.9312e-04,
          7.7248e-04, 4.7207e-04, 1.4505e-03, 1.0290e-03, 1.7509e-03, 1.2102e-03,
          8.1396e-04, 5.8985e-04, 7.6628e-04, 4.3225e-04, 5.9929e-03, 2.0714e-03,
          3.7651e-03, 2.4567e-03, 4.7493e-03, 1.2112e-03, 1.4677e-03, 7.1096e-04,
          5.5599e-04, 8.7500e-04, 6.2609e-04, 3.8576e-04, 1.1177e-03, 6.2704e-04,
          7.6771e-04, 4.8904e-03, 1.1997e-03, 1.0719e-03, 8.0347e-04, 2.2697e-03,
          1.6842e-03, 7.4291e-04, 9.1887e-04, 4.1556e-04, 4.0269e-04, 7.0667e-04,
          8.3160e-04, 2.9049e-03, 8.1539e-04, 1.8492e-03, 6.2323e-04, 4.5753e-04,
          1.3142e-03, 3.8743e-04, 7.0906e-04, 7.1096e-04, 8.6927e-04, 4.5609e-04,
          5.1641e-04, 2.8706e-03, 1.0986e-03, 1.1339e-03, 9.0742e-04, 1.4400e-03,
          2.2755e-03, 1.1272e-03, 5.1975e-04, 1.5402e-03, 5.5504e-04, 8.6546e-04,
          2.0447e-03, 4.8733e-04, 1.5507e-03, 1.1435e-03, 3.9387e-04, 4.2534e-04,
          1.6623e-03, 5.7793e-04, 6.1274e-04, 6.0463e-04, 1.0414e-03, 4.6425e-03,
          2.2202e-03, 7.6246e-04, 1.6050e-03, 6.4039e-04, 1.1415e-03, 1.0271e-03,
          6.7568e-04, 8.4734e-04, 3.2444e-03, 2.5444e-03, 1.3609e-03, 2.4452e-03,
          1.3847e-02, 7.6103e-04, 2.9540e-04, 6.9714e-04, 1.4381e-03, 3.2377e-04,
          2.6965e-04, 6.1750e-04, 7.3433e-04, 1.2960e-03, 2.0866e-03, 2.5940e-03,
          1.4858e-03, 1.0357e-03, 5.4121e-04, 5.8079e-04, 1.1368e-03, 5.9175e-04,
          9.9716e-03, 1.3649e-02, 5.5733e-03, 4.4746e-03, 3.1681e-03, 1.6388e-02,
          4.4250e-04, 2.7394e-04, 3.7551e-04, 3.5834e-04, 1.1206e-03, 1.1444e-03,
          1.4477e-03, 1.1330e-03, 5.3644e-04, 2.1744e-03, 2.1458e-03, 1.4801e-03,
          1.1873e-03, 9.8038e-04, 7.3576e-04, 1.1740e-03, 2.5921e-03, 1.6983e-02,
          2.3518e-03, 1.8473e-03, 1.0506e-02, 1.8406e-03, 7.3051e-04, 5.1689e-04,
          1.2236e-03, 5.7125e-04, 7.7391e-04, 6.9475e-04, 7.9498e-03, 1.2369e-03,
          8.9407e-04, 9.1457e-04, 1.0223e-03, 1.5564e-03, 1.6451e-03, 3.2196e-03,
          1.1368e-03, 1.4114e-03, 2.1973e-03, 2.3861e-03, 1.4896e-03, 1.2836e-03,
          1.7500e-03, 1.6113e-02, 1.7681e-03, 6.4039e-04, 1.3580e-03, 1.4277e-03,
          1.7099e-03, 2.9397e-04, 1.1892e-03, 1.1002e-02, 1.1396e-03, 1.1950e-03,
          1.0786e-03, 4.0703e-03, 2.7809e-03, 1.5755e-03, 1.3866e-03, 1.2817e-03,
          1.6661e-03, 1.7738e-03, 1.2989e-03, 9.4891e-04, 2.8057e-03, 4.3678e-03,
          1.6670e-03, 1.3752e-03, 8.5306e-04, 9.1743e-04, 1.6947e-03, 7.8440e-04,
          8.2254e-04, 6.7749e-03, 2.4395e-03, 7.7343e-04, 2.2373e-03, 1.6647e-02,
          5.0049e-03, 1.0681e-03, 1.5888e-03, 1.5507e-03, 4.6196e-03, 2.2106e-03,
          1.8625e-03, 1.2093e-03, 9.8896e-04, 1.6022e-03, 1.7223e-03, 1.2932e-02,
          1.5182e-03, 7.1287e-04, 7.3624e-04, 7.1335e-04, 1.0509e-03, 2.3079e-03,
          1.4219e-03, 1.2903e-03, 5.5552e-04, 1.3685e-03, 1.5556e-02, 1.8738e-02,
          1.8539e-03, 2.5558e-03, 1.0109e-03, 1.1492e-03, 1.6146e-03, 2.5616e-03,
          1.3771e-03, 1.1425e-03, 2.0981e-03, 1.6527e-03, 5.4893e-03, 5.0812e-03,
          9.3985e-04, 1.0948e-03, 6.3896e-04, 7.2908e-04, 4.7088e-04, 1.4992e-03,
          1.0300e-03, 1.0252e-03, 5.1079e-03, 3.7903e-02, 5.7373e-02, 2.4796e-03,
          2.8934e-03, 3.3970e-03, 3.7689e-03, 1.8682e-03, 3.8872e-03, 3.6945e-03,
          3.1071e-03, 1.3304e-03, 1.3603e-02, 5.7739e-02, 2.2233e-05, 1.3351e-04,
          1.9598e-04, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05,
          1.4246e-04, 2.2233e-05, 1.3351e-04, 1.3351e-04, 2.2233e-05, 2.2233e-05,
          2.2233e-05, 2.2233e-05, 2.2233e-05, 8.4698e-05, 5.1641e-04, 2.2233e-05,
          2.2233e-05, 2.2233e-05, 8.4698e-05, 2.2233e-05, 1.3351e-04, 8.4698e-05,
          8.4698e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05, 8.4698e-05, 8.4698e-05,
          2.2233e-05, 8.4698e-05, 2.2233e-05, 1.4246e-04, 1.3351e-04, 1.9598e-04,
          5.7888e-04, 1.9598e-04, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05,
          2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05,
          2.2233e-05, 2.2233e-05, 2.2233e-05, 1.9598e-04, 2.2233e-05, 2.2233e-05,
          8.4698e-05, 8.4698e-05, 8.4698e-05, 2.2233e-05, 6.2752e-04, 2.2233e-05,
          8.4698e-05, 8.4698e-05, 1.3351e-04, 8.4698e-05, 8.4698e-05, 2.2233e-05,
          2.2233e-05, 1.3351e-04, 2.2233e-05, 2.2233e-05, 8.4698e-05, 8.4698e-05,
          8.4698e-05, 2.2233e-05, 8.4698e-05, 8.4698e-05, 2.2233e-05, 8.4698e-05,
          8.1873e-04, 8.1873e-04, 5.1641e-04, 2.2233e-05, 1.3351e-04, 8.4698e-05,
          8.4698e-05, 2.2233e-05, 8.4698e-05, 3.0875e-04, 2.2233e-05, 2.2233e-05,
          2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05, 8.4698e-05, 2.2233e-05,
          7.5626e-04, 2.2233e-05, 1.9598e-04, 2.2233e-05, 2.2233e-05, 2.2233e-05,
          6.8998e-04, 2.2233e-05, 2.2233e-05, 2.2233e-05, 8.4698e-05, 1.9598e-04,
          2.2233e-05, 3.0403e-03, 2.2233e-05, 2.2233e-05, 2.2233e-05, 2.2233e-05,
          7.3385e-04, 7.3385e-04, 3.2449e-04, 3.9458e-04, 3.2449e-04, 1.3351e-04,
          2.2233e-05, 1.9598e-04, 1.3351e-04, 4.3845e-04, 1.3351e-04, 2.2233e-05,
          3.2449e-04, 1.3351e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.3967e-03, 4.7424e-02, 2.2949e-02, 2.6398e-02, 9.2896e-02,
          3.0121e-02, 2.7267e-02, 7.8106e-04, 5.6982e-04, 1.8415e-03, 4.7913e-03,
          1.2484e-03, 9.2268e-04, 8.5258e-04, 1.0052e-03, 5.3453e-04, 1.2846e-03,
          6.8045e-04, 3.4499e-04, 9.1648e-04, 1.1501e-03, 7.7343e-04, 1.0805e-03,
          4.0665e-03, 1.6394e-03, 1.7914e-02, 5.2719e-03, 2.7618e-03, 1.3084e-03,
          1.0166e-03, 1.0328e-03, 7.8297e-04, 1.5745e-03, 7.5722e-04, 6.1607e-04,
          2.3403e-03, 2.1076e-03, 1.8101e-03, 1.0424e-03, 8.5020e-04, 2.4052e-03,
          1.8253e-03, 1.3924e-03, 1.1034e-03, 9.7227e-04, 9.4175e-04, 1.9255e-03,
          5.1832e-04, 1.2045e-03, 1.0101e-02, 1.9112e-03, 8.8453e-04, 7.3910e-04,
          6.9904e-04, 5.7697e-04, 1.1721e-03, 3.4466e-03, 1.3304e-03, 2.2316e-03,
          1.7519e-03, 1.2779e-03, 1.1835e-03, 1.1387e-03, 1.2169e-03, 2.0905e-03,
          9.6226e-04, 4.7827e-04, 7.6675e-04, 9.5558e-04, 9.7942e-04, 9.3174e-04,
          1.2665e-03, 3.3417e-03, 2.1801e-03, 9.9468e-04, 4.9162e-04, 4.0364e-04,
          8.3971e-04, 5.0774e-03, 1.8234e-03, 1.5879e-03, 1.2827e-03, 1.1568e-03,
          1.9264e-03, 1.0557e-03, 8.1825e-04, 1.3084e-03, 5.4693e-04, 6.7520e-04,
          2.0065e-03, 1.5860e-03, 5.9366e-04, 1.4315e-03, 8.8406e-04, 1.3132e-03,
          2.2545e-03, 1.4305e-03, 7.2098e-04, 3.8891e-03, 7.4768e-03, 3.4561e-03,
          3.8185e-03, 8.8644e-04, 7.8487e-04, 2.2736e-03, 1.8425e-03, 1.5898e-03,
          8.4734e-04, 1.7986e-03, 2.5043e-03, 7.1239e-04, 9.6798e-04, 7.8726e-04,
          1.0490e-03, 1.7719e-03, 1.0633e-03, 2.0866e-03, 1.6947e-03, 6.8474e-03,
          4.3945e-03, 4.2496e-03, 3.4332e-03, 4.0030e-04, 2.0962e-03, 9.5415e-04,
          4.4594e-03, 1.2550e-03, 1.3981e-03, 1.9236e-03, 6.7377e-04, 9.5844e-04,
          1.3905e-03, 7.4673e-04, 4.5657e-04, 3.8743e-04, 1.1845e-03, 9.7847e-04,
          1.9026e-03, 1.1187e-03, 1.0538e-03, 2.5158e-03, 1.9875e-03, 2.2984e-03,
          3.7689e-03, 9.8896e-04, 1.5831e-03, 2.4452e-03, 1.8024e-03, 2.9182e-03,
          4.3221e-03, 1.8501e-03, 2.4261e-03, 1.5135e-03, 1.7424e-03, 3.2959e-03,
          1.3762e-03, 1.6232e-03, 3.9458e-04, 8.3017e-04, 2.4738e-03, 1.1959e-03,
          1.8482e-03, 1.8759e-03, 1.5087e-03, 2.0504e-03, 1.8494e-02, 3.0079e-03,
          3.4351e-03, 2.1687e-03, 1.9217e-03, 9.7418e-04, 1.4734e-03, 2.7790e-03,
          1.8215e-03, 2.0828e-03, 5.5008e-03, 4.6005e-03, 2.0142e-03, 1.2474e-03,
          9.7656e-04, 8.9788e-04, 1.5059e-03, 7.7152e-04, 8.9884e-04, 9.7036e-04,
          1.1187e-03, 1.2455e-03, 2.2068e-03, 1.2291e-02, 2.0332e-03, 1.6413e-03,
          8.7595e-04, 4.8790e-03, 1.3256e-03, 2.1744e-03, 1.7138e-03, 1.6499e-03,
          4.2763e-03, 3.5992e-03, 2.2926e-03, 1.1244e-03, 1.3781e-03, 1.1053e-03,
          1.1234e-03, 1.4191e-03, 1.8787e-03, 7.7677e-04, 2.5291e-03, 8.5211e-04,
          8.2207e-04, 1.9806e-02, 2.4216e-02, 2.0161e-03, 3.4924e-03, 2.6608e-03,
          2.9411e-03, 2.3937e-03, 1.6584e-03, 7.1049e-04, 2.3785e-03, 1.6365e-03,
          4.0741e-03, 3.3245e-03, 1.9550e-03, 1.1206e-03, 2.4052e-03, 1.3390e-03,
          9.4509e-04, 1.0481e-03, 1.1578e-03, 1.8892e-03, 1.5774e-03, 1.0691e-03,
          4.8523e-02, 3.6278e-03, 2.7122e-03, 1.4076e-03, 6.5918e-03, 5.9204e-03,
          5.8823e-03, 8.5602e-03, 5.0888e-03, 4.7874e-03, 3.9978e-03, 5.1994e-03,
          4.7798e-03, 1.2960e-03, 1.8234e-03, 1.5364e-03, 1.9665e-03, 1.6031e-03,
          8.7500e-04, 1.5240e-03, 2.7351e-03, 2.3174e-03, 1.3466e-03, 6.5231e-03,
          1.6647e-02, 2.1572e-03, 3.9940e-03, 2.2564e-03, 4.1237e-03, 5.8479e-03,
          2.6512e-03, 9.9277e-04, 1.0700e-03, 2.5063e-03, 3.1834e-03, 1.5974e-03,
          2.4395e-03, 1.5688e-03, 2.0313e-03, 2.2850e-03, 3.2177e-03, 2.2106e-03,
          1.8911e-03, 1.1311e-03, 3.6755e-03, 1.6851e-03, 3.6316e-02, 4.7624e-05,
          2.9683e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 2.1768e-04, 2.1768e-04, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 2.9683e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          2.1768e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          2.1768e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 2.1768e-04,
          4.7624e-05, 4.7624e-05, 2.1768e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 2.1768e-04,
          4.7624e-05, 4.7624e-05, 2.1768e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 2.9683e-04, 2.9683e-04,
          4.7624e-05, 2.1768e-04, 4.7624e-05, 4.7624e-05, 2.1768e-04, 2.1768e-04,
          4.6682e-04, 4.6682e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          2.9683e-04, 2.1768e-04, 4.7624e-05, 2.1768e-04, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 2.9683e-04, 4.7624e-05,
          2.1768e-04, 2.1768e-04, 2.1768e-04, 4.7624e-05, 4.7624e-05, 2.9144e-03,
          2.1768e-04, 4.7624e-05, 2.1768e-04, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          2.1768e-04, 7.2289e-03, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05, 4.7624e-05,
          4.7624e-05, 4.7624e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1270e-03, 4.8065e-02, 1.3474e-02, 1.9150e-02, 4.1138e-02,
          3.3386e-02, 1.2230e-02, 7.6532e-04, 5.9509e-04, 2.0447e-03, 6.6566e-03,
          1.6155e-03, 6.9189e-04, 1.0509e-03, 7.6342e-04, 5.6791e-04, 9.1219e-04,
          6.5660e-04, 4.6945e-04, 2.3878e-04, 2.3103e-04, 2.4509e-04, 8.9979e-04,
          6.5117e-03, 1.2169e-03, 9.6054e-03, 3.3264e-03, 6.5422e-04, 1.0233e-03,
          1.7881e-03, 1.3218e-03, 4.4417e-04, 9.4318e-04, 7.8154e-04, 4.8923e-04,
          1.0633e-03, 9.7036e-04, 9.4795e-04, 6.7472e-04, 4.1556e-04, 1.0405e-03,
          9.3603e-04, 7.2145e-04, 8.8930e-04, 3.3855e-04, 9.6083e-04, 9.1076e-04,
          4.5872e-04, 3.3784e-04, 2.8687e-03, 1.1864e-03, 1.2150e-03, 6.4945e-04,
          6.6662e-04, 7.4196e-04, 6.7711e-04, 1.1168e-03, 1.8177e-03, 2.0905e-03,
          1.6460e-03, 8.5735e-04, 9.8896e-04, 5.9128e-04, 1.8988e-03, 1.7233e-03,
          1.2718e-02, 3.0079e-03, 9.4891e-04, 4.3607e-04, 4.5300e-04, 9.0313e-04,
          5.1689e-04, 3.9330e-03, 1.1072e-03, 1.5907e-03, 6.9904e-04, 5.2023e-04,
          9.1362e-04, 1.6918e-03, 7.5722e-03, 2.2335e-03, 7.7057e-03, 8.2636e-04,
          7.4196e-04, 1.5926e-03, 8.0919e-04, 2.6970e-03, 5.2032e-03, 6.2294e-03,
          6.3019e-03, 2.4414e-03, 8.3828e-04, 3.2005e-03, 3.2473e-04, 4.2892e-04,
          7.9012e-04, 6.1846e-04, 7.1287e-04, 1.6689e-03, 3.6469e-03, 3.0937e-03,
          8.3160e-03, 5.3978e-03, 4.9210e-03, 1.8549e-03, 1.4076e-03, 8.1062e-04,
          1.3561e-03, 1.3018e-03, 4.0169e-03, 2.3060e-03, 1.1658e-02, 4.8256e-03,
          9.9754e-04, 5.8937e-04, 3.1686e-04, 1.5764e-03, 6.3229e-04, 1.7767e-03,
          2.3956e-03, 3.4866e-03, 3.0727e-03, 5.8222e-04, 2.1057e-03, 3.8147e-03,
          1.5039e-03, 9.4223e-03, 1.6525e-02, 1.5240e-03, 8.3542e-04, 1.5841e-03,
          1.3710e-02, 7.1793e-03, 4.4155e-04, 2.3956e-03, 3.7861e-03, 8.1825e-04,
          1.3638e-03, 7.6437e-04, 7.5340e-04, 1.8539e-03, 1.3313e-03, 2.3098e-03,
          6.3629e-03, 6.9141e-04, 2.1210e-03, 2.8229e-03, 1.2070e-02, 2.4090e-03,
          5.2948e-03, 1.7281e-03, 1.8988e-03, 1.0099e-03, 1.4505e-03, 1.6937e-03,
          3.0575e-03, 2.0847e-03, 9.0551e-04, 4.8065e-03, 1.4362e-03, 7.1001e-04,
          1.2321e-03, 1.4515e-03, 1.3971e-03, 1.2255e-03, 3.6583e-03, 4.1771e-03,
          7.7972e-03, 3.2558e-03, 2.0046e-03, 9.0265e-04, 1.5812e-03, 1.6909e-03,
          1.2369e-03, 1.6088e-03, 1.6537e-03, 1.6994e-03, 1.7223e-03, 1.5669e-03,
          1.3218e-03, 9.1028e-04, 1.6594e-03, 1.1778e-03, 8.4066e-04, 5.7411e-04,
          2.2888e-03, 8.3208e-04, 1.3390e-03, 7.4081e-03, 1.4763e-02, 1.8206e-03,
          1.3971e-03, 2.2984e-03, 1.2627e-03, 2.1458e-03, 3.7479e-03, 1.5745e-03,
          1.8711e-03, 1.5078e-03, 1.5602e-03, 4.8637e-03, 3.6049e-03, 1.1587e-03,
          4.9629e-03, 3.7937e-03, 8.8120e-04, 7.7963e-04, 5.3596e-03, 6.3324e-04,
          1.0128e-03, 4.5128e-03, 9.1476e-03, 1.1513e-02, 3.4771e-03, 8.4076e-03,
          2.1973e-03, 1.9255e-03, 2.5959e-03, 6.6328e-04, 1.7147e-03, 2.0866e-03,
          2.6817e-03, 1.7147e-03, 2.2831e-03, 5.8479e-03, 2.9278e-03, 1.6899e-03,
          9.5062e-03, 4.4365e-03, 1.0614e-03, 1.3208e-03, 1.0996e-03, 8.1396e-04,
          8.6746e-03, 8.1711e-03, 2.0859e-02, 1.5574e-03, 6.1646e-03, 8.6594e-03,
          6.9656e-03, 4.3144e-03, 2.5997e-03, 1.9722e-03, 2.7180e-03, 2.9240e-03,
          1.7405e-03, 7.5836e-03, 1.4458e-03, 1.3590e-03, 1.5726e-03, 1.6842e-03,
          6.2180e-03, 6.4373e-04, 1.2341e-03, 1.3514e-03, 1.2550e-03, 3.3283e-03,
          6.8169e-03, 1.4130e-02, 2.8553e-03, 1.8206e-03, 2.3499e-03, 6.6566e-03,
          2.1572e-03, 1.1578e-03, 9.9182e-04, 3.4561e-03, 7.0839e-03, 1.3237e-03,
          1.5038e-02, 1.1902e-02, 2.2850e-03, 1.6069e-03, 2.8419e-03, 2.0199e-03,
          1.7929e-03, 1.0338e-03, 8.9788e-04, 2.7523e-03, 5.9967e-03, 5.0903e-02,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          1.0419e-04, 2.9862e-05, 1.9896e-04, 1.9896e-04, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 1.0419e-04, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 1.0419e-04, 1.0419e-04, 1.0419e-04, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          4.9305e-04, 2.9862e-05, 2.9862e-05, 1.0419e-04, 2.9862e-05, 1.2469e-04,
          2.9862e-05, 2.9862e-05, 4.1890e-04, 4.9305e-04, 2.9862e-05, 2.9862e-05,
          1.9896e-04, 3.9840e-04, 1.2469e-04, 2.9862e-05, 2.9862e-05, 1.9896e-04,
          2.9862e-05, 4.9305e-04, 1.2469e-04, 1.0419e-04, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 1.0419e-04, 3.2401e-04, 2.9862e-05, 1.2469e-04,
          2.9862e-05, 2.9862e-05, 5.6458e-04, 2.9862e-05, 3.2401e-04, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.0419e-04, 1.0419e-04,
          1.0419e-04, 1.2469e-04, 2.9862e-05, 2.9862e-05, 1.2469e-04, 2.9862e-05,
          1.9896e-04, 1.0419e-04, 3.2401e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          1.0419e-04, 2.9862e-05, 2.9862e-05, 1.2469e-04, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.0419e-04, 2.9862e-05,
          4.9305e-04, 1.9896e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.2469e-04,
          1.0419e-04, 2.9862e-05, 4.1890e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          4.1890e-04, 4.1890e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 1.2469e-04, 1.2469e-04, 1.2469e-04, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 1.2469e-04, 1.2469e-04, 1.2469e-04,
          2.9862e-05, 1.2469e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3749e-03, 5.9082e-02, 1.3794e-02, 2.0706e-02, 3.8025e-02,
          4.3579e-02, 1.0651e-02, 1.0529e-03, 6.9189e-04, 3.1567e-03, 9.6130e-03,
          2.3079e-03, 9.7609e-04, 9.4795e-04, 8.2636e-04, 8.2445e-04, 1.2674e-03,
          1.2140e-03, 5.2023e-04, 2.0182e-04, 2.3723e-04, 2.6298e-04, 1.4601e-03,
          1.0674e-02, 1.4172e-03, 8.9569e-03, 5.6534e-03, 5.6410e-04, 1.6832e-03,
          1.4858e-03, 1.8816e-03, 6.9666e-04, 9.9564e-04, 1.3943e-03, 5.3596e-04,
          1.0223e-03, 7.9870e-04, 1.0729e-03, 7.7438e-04, 3.4451e-04, 1.3723e-03,
          8.5592e-04, 6.4039e-04, 8.9979e-04, 3.9339e-04, 1.5974e-03, 9.9373e-04,
          5.1403e-04, 3.2496e-04, 2.9297e-03, 9.9182e-04, 1.6079e-03, 6.8998e-04,
          5.6648e-04, 8.9645e-04, 5.3215e-04, 9.5558e-04, 1.8892e-03, 1.8034e-03,
          1.2817e-03, 8.3303e-04, 7.6675e-04, 5.6982e-04, 1.7691e-03, 1.8425e-03,
          1.4305e-02, 5.7602e-03, 1.2159e-03, 4.2105e-04, 3.9029e-04, 7.5674e-04,
          4.6039e-04, 5.6953e-03, 7.9823e-04, 1.9255e-03, 5.8317e-04, 4.8494e-04,
          7.8106e-04, 1.2426e-03, 6.7062e-03, 2.1076e-03, 8.7433e-03, 6.8569e-04,
          5.5742e-04, 1.9054e-03, 6.6328e-04, 4.1885e-03, 6.5422e-03, 5.4626e-03,
          1.1810e-02, 3.2520e-03, 6.5613e-04, 5.5161e-03, 2.6155e-04, 3.6311e-04,
          6.6662e-04, 3.9911e-04, 8.0729e-04, 1.3237e-03, 1.8234e-03, 2.1000e-03,
          1.0857e-02, 6.1150e-03, 7.3967e-03, 1.1797e-03, 1.2703e-03, 6.7186e-04,
          1.4553e-03, 9.4128e-04, 4.9019e-03, 1.4772e-03, 9.8495e-03, 4.5242e-03,
          8.9979e-04, 5.0402e-04, 2.7823e-04, 1.6727e-03, 4.9686e-04, 1.3885e-03,
          1.9035e-03, 2.6703e-03, 1.4648e-03, 6.8045e-04, 2.1400e-03, 2.2144e-03,
          7.3385e-04, 4.1008e-03, 6.1798e-03, 1.4343e-03, 1.1692e-03, 1.7490e-03,
          1.2024e-02, 1.0513e-02, 6.6376e-04, 2.8191e-03, 5.2643e-03, 5.9080e-04,
          1.7385e-03, 6.1989e-04, 9.8801e-04, 2.3918e-03, 1.8873e-03, 2.0657e-03,
          2.8839e-03, 7.3814e-04, 1.3790e-03, 1.8129e-03, 8.3542e-03, 1.8187e-03,
          4.7226e-03, 1.4286e-03, 1.0481e-03, 9.0837e-04, 1.1568e-03, 1.3332e-03,
          3.4466e-03, 1.7233e-03, 1.7309e-03, 2.1114e-03, 1.2665e-03, 7.5150e-04,
          1.2560e-03, 1.4849e-03, 2.2125e-03, 1.5717e-03, 1.9722e-03, 2.9297e-03,
          9.6741e-03, 2.8419e-03, 1.5984e-03, 9.1219e-04, 2.0409e-03, 1.4124e-03,
          1.1292e-03, 1.0223e-03, 1.0834e-03, 1.4458e-03, 1.7395e-03, 2.7866e-03,
          1.2255e-03, 1.0176e-03, 1.8110e-03, 1.7071e-03, 1.0481e-03, 5.8508e-04,
          3.9368e-03, 8.1778e-04, 1.6279e-03, 3.1166e-03, 9.2010e-03, 2.0485e-03,
          2.5272e-03, 9.8610e-04, 1.2112e-03, 3.3569e-03, 5.4245e-03, 1.4744e-03,
          1.1024e-03, 9.3269e-04, 1.5335e-03, 5.1956e-03, 5.4054e-03, 1.2503e-03,
          3.9253e-03, 4.1504e-03, 6.3324e-04, 1.1244e-03, 7.7324e-03, 9.9277e-04,
          1.3132e-03, 2.8172e-03, 3.4962e-03, 1.4282e-02, 3.1071e-03, 4.1695e-03,
          1.6794e-03, 2.0370e-03, 3.7098e-03, 7.7009e-04, 1.8616e-03, 2.0237e-03,
          2.2888e-03, 1.2207e-03, 2.0046e-03, 6.9046e-03, 3.3112e-03, 2.0351e-03,
          1.4175e-02, 2.6550e-03, 7.7677e-04, 1.0977e-03, 1.0300e-03, 7.7152e-04,
          4.4708e-03, 5.3024e-03, 1.1345e-02, 1.2808e-03, 3.2387e-03, 6.4964e-03,
          3.9215e-03, 2.0370e-03, 1.2207e-03, 1.3828e-03, 2.2812e-03, 1.9417e-03,
          1.2531e-03, 1.0483e-02, 9.7084e-04, 1.2207e-03, 1.1911e-03, 1.7481e-03,
          3.1700e-03, 4.3344e-04, 8.1968e-04, 9.1743e-04, 1.6966e-03, 3.6716e-03,
          3.5667e-03, 1.1864e-02, 1.7948e-03, 1.3704e-03, 1.1597e-03, 3.6545e-03,
          1.6804e-03, 1.2817e-03, 1.1120e-03, 4.3488e-03, 9.1400e-03, 1.1549e-03,
          1.1383e-02, 1.4183e-02, 2.3727e-03, 1.2197e-03, 2.4052e-03, 1.6184e-03,
          1.3943e-03, 1.0862e-03, 6.1560e-04, 2.5158e-03, 3.1624e-03, 2.3438e-02,
          4.5898e-02, 4.3452e-05, 4.3452e-05, 4.3452e-05, 2.4033e-04, 4.3452e-05,
          1.0085e-04, 4.3452e-05, 1.6475e-04, 1.6475e-04, 1.1903e-04, 1.1903e-04,
          4.3452e-05, 4.3452e-05, 5.0116e-04, 2.0313e-04, 1.1903e-04, 1.1903e-04,
          5.0116e-04, 1.3912e-04, 1.0085e-04, 2.4033e-04, 4.3452e-05, 4.3452e-05,
          1.0085e-04, 3.3903e-04, 1.8299e-04, 1.6475e-04, 4.3452e-05, 4.3452e-05,
          3.9649e-04, 4.3452e-05, 5.0116e-04, 1.0085e-04, 1.0741e-04, 1.8299e-04,
          1.8299e-04, 4.3452e-05, 4.6396e-04, 2.7871e-04, 5.0116e-04, 5.6553e-04,
          4.8423e-04, 2.7871e-04, 1.8299e-04, 4.3452e-05, 4.3452e-05, 4.8327e-04,
          4.3452e-05, 2.4033e-04, 2.4033e-04, 1.0085e-04, 4.3452e-05, 2.6345e-04,
          3.3903e-04, 1.0085e-04, 1.0085e-04, 7.2145e-04, 4.3452e-05, 3.3903e-04,
          4.3452e-05, 1.0085e-04, 4.3464e-04, 1.7643e-04, 1.7643e-04, 7.2145e-04,
          1.8299e-04, 1.0741e-04, 4.3452e-05, 4.3452e-05, 1.0085e-04, 1.0264e-04,
          9.0981e-04, 3.8838e-04, 4.3452e-05, 4.3452e-05, 1.8299e-04, 3.0780e-04,
          3.2783e-04, 3.2783e-04, 1.7643e-04, 4.3452e-05, 4.3452e-05, 4.3452e-05,
          2.8968e-04, 3.3903e-04, 1.0085e-04, 4.6396e-04, 4.3452e-05, 4.3452e-05,
          1.0741e-04, 4.3452e-05, 1.1903e-04, 4.3452e-05, 9.1171e-04, 1.1903e-04,
          1.0986e-03, 4.8327e-04, 4.3452e-05, 1.0741e-04, 1.1903e-04, 7.0715e-04,
          2.4033e-04, 7.2145e-04, 7.1478e-04, 4.3452e-05, 4.3452e-05, 8.4639e-04,
          7.8249e-04, 1.2550e-03, 1.6820e-04, 8.4639e-04, 8.4639e-04, 4.3452e-05,
          1.0741e-04, 4.3452e-05, 4.3452e-05, 1.0741e-04, 2.9707e-04, 1.0741e-04,
          4.3452e-05, 1.6475e-04, 1.0741e-04, 1.0741e-04, 2.9707e-04, 2.9707e-04,
          4.3452e-05, 1.0741e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.7722e-03, 3.8605e-02, 1.6357e-02, 1.3855e-02, 6.9519e-02,
          2.2903e-02, 1.3725e-02, 7.8440e-04, 6.7854e-04, 1.4114e-03, 4.2305e-03,
          1.1606e-03, 8.2922e-04, 7.0715e-04, 1.2093e-03, 5.4121e-04, 8.2445e-04,
          4.5371e-04, 3.8910e-04, 5.8556e-04, 4.4942e-04, 4.1938e-04, 5.4789e-04,
          3.7498e-03, 1.8492e-03, 9.1858e-03, 1.7805e-03, 1.7862e-03, 8.0156e-04,
          8.2970e-04, 1.0824e-03, 4.5061e-04, 1.0967e-03, 6.4182e-04, 5.4407e-04,
          1.5354e-03, 1.5516e-03, 8.1635e-04, 6.8760e-04, 6.4182e-04, 1.0519e-03,
          2.0008e-03, 1.2484e-03, 1.6117e-03, 7.4720e-04, 6.9141e-04, 1.1864e-03,
          4.0054e-04, 5.9795e-04, 5.6648e-03, 1.2465e-03, 9.1839e-04, 6.4945e-04,
          7.3433e-04, 6.8855e-04, 1.3962e-03, 2.1286e-03, 1.5726e-03, 2.4815e-03,
          2.6131e-03, 8.6641e-04, 1.0691e-03, 6.0987e-04, 1.9083e-03, 1.9073e-03,
          1.5221e-03, 6.7711e-04, 6.1131e-04, 9.7418e-04, 1.0233e-03, 1.6336e-03,
          1.1692e-03, 2.1152e-03, 1.6966e-03, 1.1911e-03, 6.8378e-04, 4.4370e-04,
          1.1749e-03, 3.3188e-03, 2.3556e-03, 1.5039e-03, 1.8702e-03, 1.3828e-03,
          1.7605e-03, 1.1959e-03, 7.7677e-04, 1.0672e-03, 6.1941e-04, 1.0338e-03,
          1.9484e-03, 1.0920e-03, 9.2220e-04, 1.3199e-03, 1.3351e-03, 1.2884e-03,
          1.2712e-03, 1.0614e-03, 5.8508e-04, 1.7014e-03, 7.2861e-03, 2.9659e-03,
          3.3855e-03, 1.0471e-03, 1.0386e-03, 2.0161e-03, 1.6756e-03, 1.0262e-03,
          1.0986e-03, 1.2865e-03, 1.5831e-03, 1.1034e-03, 1.5898e-03, 8.5878e-04,
          1.4029e-03, 1.4267e-03, 5.2452e-04, 2.1629e-03, 1.8663e-03, 2.9621e-03,
          2.3651e-03, 3.4904e-03, 5.8479e-03, 5.1546e-04, 2.0332e-03, 1.6584e-03,
          3.9177e-03, 2.0905e-03, 2.6150e-03, 1.7090e-03, 6.2418e-04, 1.1616e-03,
          1.7805e-03, 8.1015e-04, 4.3702e-04, 4.3583e-04, 1.0300e-03, 1.1024e-03,
          1.1272e-03, 7.3004e-04, 9.2888e-04, 1.5345e-03, 1.2169e-03, 1.7452e-03,
          7.9956e-03, 1.8082e-03, 2.4891e-03, 2.8667e-03, 3.3073e-03, 3.1757e-03,
          3.3932e-03, 2.0485e-03, 3.8204e-03, 9.4080e-04, 1.7099e-03, 1.6308e-03,
          9.5701e-04, 9.5272e-04, 4.7421e-04, 1.1301e-03, 2.1439e-03, 7.4673e-04,
          1.7233e-03, 8.3160e-04, 1.0233e-03, 1.1873e-03, 7.0229e-03, 2.5177e-03,
          3.4466e-03, 1.7242e-03, 2.1038e-03, 1.1864e-03, 1.1778e-03, 2.4261e-03,
          2.0695e-03, 2.3880e-03, 4.4899e-03, 2.0905e-03, 1.5678e-03, 8.9359e-04,
          9.9564e-04, 8.0109e-04, 1.3008e-03, 8.6117e-04, 6.1989e-04, 1.0166e-03,
          1.3475e-03, 8.1635e-04, 1.1787e-03, 1.6434e-02, 5.1727e-03, 1.7595e-03,
          1.0614e-03, 7.7095e-03, 1.4324e-03, 1.7586e-03, 1.5535e-03, 1.7700e-03,
          5.0964e-03, 2.7027e-03, 1.7405e-03, 1.0891e-03, 1.0042e-03, 9.5177e-04,
          1.4992e-03, 1.2970e-03, 1.9093e-03, 6.1703e-04, 2.4395e-03, 6.2370e-04,
          8.1730e-04, 6.7177e-03, 2.8793e-02, 3.3531e-03, 3.8128e-03, 7.5493e-03,
          2.2755e-03, 2.1935e-03, 1.4277e-03, 1.0328e-03, 2.0618e-03, 1.7929e-03,
          2.8591e-03, 3.1662e-03, 1.8110e-03, 9.7942e-04, 2.5139e-03, 1.3657e-03,
          1.1253e-03, 1.5202e-03, 2.1420e-03, 2.9106e-03, 1.4210e-03, 9.3365e-04,
          2.6001e-02, 4.1161e-03, 7.6752e-03, 2.3022e-03, 1.5884e-02, 3.0441e-02,
          2.3193e-02, 1.9272e-02, 7.4959e-03, 3.0441e-03, 3.8605e-03, 4.8065e-03,
          2.8458e-03, 1.8187e-03, 2.6722e-03, 2.0180e-03, 2.6779e-03, 1.5478e-03,
          1.3876e-03, 2.4242e-03, 3.2120e-03, 1.5326e-03, 1.2245e-03, 3.1357e-03,
          1.4687e-02, 3.6526e-03, 6.1188e-03, 4.1428e-03, 1.6281e-02, 2.5558e-02,
          4.3297e-03, 1.0033e-03, 1.4229e-03, 2.8114e-03, 2.6188e-03, 1.5039e-03,
          3.7861e-03, 2.1553e-03, 2.2163e-03, 2.2984e-03, 2.0695e-03, 2.7561e-03,
          3.7308e-03, 1.6031e-03, 2.3937e-03, 1.3742e-03, 1.5053e-02, 1.3077e-02,
          4.4479e-03, 5.5725e-02, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.3951e-04, 3.3951e-04, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 1.8835e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 1.8942e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          1.8942e-04, 3.8385e-05, 1.8942e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.3951e-04, 3.8385e-05, 3.8385e-05, 1.8942e-04,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 1.8835e-04, 1.8835e-04,
          1.8835e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.3951e-04, 3.3951e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 3.8385e-05, 3.8385e-05, 3.8385e-05, 1.8835e-04, 3.8385e-05,
          1.8942e-04, 1.8835e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05, 1.8942e-04,
          1.8942e-04, 3.8385e-05, 1.8835e-04, 3.8385e-05, 3.8385e-05, 3.8385e-05,
          1.8942e-04, 5.7125e-04, 3.8385e-05, 1.8942e-04, 3.8385e-05, 3.8385e-05,
          3.8385e-05, 1.8942e-04, 1.8942e-04, 1.8942e-04, 1.8942e-04, 3.8385e-05,
          3.8385e-05, 1.8942e-04, 3.8385e-05, 1.8942e-04, 3.8385e-05, 1.8942e-04,
          3.8385e-05, 1.8942e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.2948e-03, 4.1351e-02, 1.5701e-02, 1.5495e-02, 7.5256e-02,
          2.2537e-02, 1.5762e-02, 7.4196e-04, 6.4850e-04, 1.3809e-03, 3.8795e-03,
          1.1387e-03, 8.5211e-04, 7.8583e-04, 1.2550e-03, 5.1880e-04, 8.3923e-04,
          4.4274e-04, 3.7503e-04, 6.3992e-04, 4.9877e-04, 3.7837e-04, 5.2404e-04,
          3.3760e-03, 1.5001e-03, 1.0460e-02, 1.9417e-03, 1.3924e-03, 7.9918e-04,
          9.2554e-04, 1.0033e-03, 4.4489e-04, 1.1883e-03, 6.2227e-04, 5.4741e-04,
          1.7071e-03, 1.8721e-03, 8.3876e-04, 7.3528e-04, 6.4945e-04, 1.2302e-03,
          2.0351e-03, 1.3485e-03, 1.7233e-03, 8.4019e-04, 7.1287e-04, 1.2693e-03,
          3.7885e-04, 6.0987e-04, 5.7983e-03, 1.1787e-03, 8.2970e-04, 5.6314e-04,
          7.5483e-04, 7.1764e-04, 1.4915e-03, 2.7409e-03, 1.5106e-03, 2.8019e-03,
          2.8992e-03, 8.9025e-04, 9.9754e-04, 6.1798e-04, 1.5850e-03, 1.8921e-03,
          1.5297e-03, 6.5899e-04, 6.2132e-04, 1.1930e-03, 1.2941e-03, 1.5640e-03,
          1.3285e-03, 2.0828e-03, 1.4696e-03, 1.0767e-03, 6.7806e-04, 4.7326e-04,
          1.3828e-03, 4.2839e-03, 2.6093e-03, 1.7176e-03, 2.2640e-03, 1.3847e-03,
          1.8578e-03, 1.0967e-03, 7.8535e-04, 9.8419e-04, 6.5899e-04, 1.1272e-03,
          1.7519e-03, 1.2560e-03, 8.7500e-04, 1.2903e-03, 1.3437e-03, 1.3657e-03,
          1.1253e-03, 1.0138e-03, 5.6982e-04, 2.0409e-03, 8.4686e-03, 3.7251e-03,
          3.5477e-03, 1.0748e-03, 1.3256e-03, 2.5902e-03, 1.9588e-03, 1.1177e-03,
          1.0214e-03, 1.3485e-03, 1.8072e-03, 9.8610e-04, 1.5783e-03, 9.6703e-04,
          1.2236e-03, 1.6336e-03, 5.5838e-04, 2.1801e-03, 2.1954e-03, 2.9354e-03,
          2.4128e-03, 3.8242e-03, 4.7188e-03, 4.7255e-04, 2.5330e-03, 1.9131e-03,
          5.3825e-03, 2.6913e-03, 3.2082e-03, 1.8530e-03, 5.7268e-04, 1.0691e-03,
          1.6756e-03, 9.4843e-04, 3.6192e-04, 4.6825e-04, 1.1234e-03, 1.3113e-03,
          1.2980e-03, 7.6246e-04, 7.7677e-04, 1.5564e-03, 1.1635e-03, 1.8940e-03,
          6.3858e-03, 1.0738e-03, 2.6245e-03, 3.5725e-03, 3.4199e-03, 3.5820e-03,
          4.8828e-03, 2.3422e-03, 5.0545e-03, 1.0405e-03, 1.7157e-03, 2.0027e-03,
          1.2007e-03, 1.3847e-03, 3.9530e-04, 1.3189e-03, 2.2774e-03, 7.8392e-04,
          1.8044e-03, 1.0738e-03, 1.1215e-03, 1.2226e-03, 8.1863e-03, 2.9106e-03,
          3.5515e-03, 2.2297e-03, 2.1553e-03, 1.1139e-03, 1.2665e-03, 3.2501e-03,
          1.8482e-03, 2.5291e-03, 5.6610e-03, 2.5444e-03, 1.6069e-03, 8.7929e-04,
          9.6846e-04, 7.1955e-04, 1.3647e-03, 7.6342e-04, 5.6696e-04, 8.0824e-04,
          1.1625e-03, 8.5592e-04, 1.2770e-03, 1.3145e-02, 4.8561e-03, 1.8463e-03,
          1.0328e-03, 8.2092e-03, 1.4896e-03, 1.9131e-03, 1.6642e-03, 1.8768e-03,
          6.0997e-03, 3.1872e-03, 1.9550e-03, 1.1578e-03, 1.1206e-03, 9.1076e-04,
          1.5659e-03, 1.5020e-03, 2.0733e-03, 5.7602e-04, 2.2907e-03, 6.5136e-04,
          7.3147e-04, 7.4310e-03, 2.2018e-02, 3.0956e-03, 3.8700e-03, 7.0190e-03,
          2.8324e-03, 2.4090e-03, 1.4820e-03, 9.6464e-04, 2.2221e-03, 1.9236e-03,
          3.1796e-03, 3.0060e-03, 1.7300e-03, 9.9754e-04, 2.7180e-03, 1.2064e-03,
          1.2655e-03, 1.3933e-03, 1.9102e-03, 2.1687e-03, 1.3666e-03, 8.9979e-04,
          2.0370e-02, 4.0092e-03, 7.2365e-03, 2.0885e-03, 1.3962e-02, 1.5671e-02,
          1.8188e-02, 1.7090e-02, 6.8359e-03, 4.0627e-03, 4.4403e-03, 5.6076e-03,
          3.2597e-03, 1.6203e-03, 2.2736e-03, 1.4648e-03, 2.1229e-03, 1.5497e-03,
          1.4610e-03, 2.4719e-03, 3.4771e-03, 1.7595e-03, 1.1349e-03, 3.2921e-03,
          1.4297e-02, 3.5133e-03, 5.8479e-03, 3.9368e-03, 1.4778e-02, 1.9119e-02,
          3.7708e-03, 1.0614e-03, 1.1692e-03, 3.0632e-03, 2.8343e-03, 1.4143e-03,
          3.6564e-03, 2.3003e-03, 2.2564e-03, 2.3479e-03, 2.6093e-03, 2.8725e-03,
          2.8934e-03, 1.3609e-03, 2.5959e-03, 1.4868e-03, 1.4954e-02, 1.1703e-02,
          4.7302e-03, 3.4790e-02, 3.6865e-02, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.0730e-04, 2.0730e-04, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.5344e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          2.0730e-04, 3.9399e-05, 2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.0730e-04, 3.9399e-05, 3.9399e-05, 2.0730e-04,
          3.9399e-05, 2.0730e-04, 3.9399e-05, 3.9399e-05, 2.5344e-04, 2.5344e-04,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          4.2129e-04, 4.2129e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 2.5344e-04, 3.9399e-05,
          2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05, 2.0730e-04,
          2.0730e-04, 3.9399e-05, 2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          2.0730e-04, 2.0730e-04, 3.9399e-05, 3.9399e-05, 3.9399e-05, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.0730e-04, 1.1003e-04, 2.0730e-04, 3.9399e-05,
          3.9399e-05, 3.9399e-05, 2.0730e-04, 3.7837e-04, 3.9399e-05, 2.0730e-04,
          3.9399e-05, 3.9399e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.7188e-03, 4.3243e-02, 1.6800e-02, 1.4954e-02, 7.8857e-02,
          2.6825e-02, 1.5121e-02, 7.3719e-04, 6.4898e-04, 1.1473e-03, 3.4447e-03,
          8.0872e-04, 8.0538e-04, 6.0844e-04, 1.2283e-03, 4.9686e-04, 7.4291e-04,
          3.8123e-04, 3.1781e-04, 7.1001e-04, 5.4455e-04, 3.4714e-04, 5.2595e-04,
          2.8419e-03, 1.5373e-03, 1.0818e-02, 2.0428e-03, 1.4858e-03, 6.9475e-04,
          6.5804e-04, 9.9659e-04, 3.3498e-04, 1.1673e-03, 5.3549e-04, 6.3372e-04,
          1.8864e-03, 1.8501e-03, 7.6485e-04, 6.8951e-04, 5.7030e-04, 9.2840e-04,
          1.4257e-03, 9.7084e-04, 1.3132e-03, 7.2813e-04, 6.9189e-04, 1.2541e-03,
          3.0136e-04, 5.8317e-04, 6.5269e-03, 1.3704e-03, 8.7214e-04, 6.4468e-04,
          9.9468e-04, 7.6056e-04, 2.2907e-03, 2.0809e-03, 1.9951e-03, 2.5482e-03,
          3.6716e-03, 7.5960e-04, 1.1711e-03, 7.1144e-04, 9.2745e-04, 1.7281e-03,
          8.3351e-04, 3.5357e-04, 4.1461e-04, 7.8392e-04, 9.3031e-04, 9.9087e-04,
          1.1015e-03, 1.7099e-03, 1.8883e-03, 1.1711e-03, 8.3113e-04, 5.4979e-04,
          1.5650e-03, 2.7771e-03, 1.4448e-03, 1.3847e-03, 1.3504e-03, 1.5631e-03,
          2.0828e-03, 1.1406e-03, 8.7357e-04, 7.1716e-04, 3.9053e-04, 6.1178e-04,
          1.1673e-03, 7.9966e-04, 8.3303e-04, 8.4877e-04, 2.0714e-03, 1.8606e-03,
          1.1253e-03, 1.3132e-03, 5.9271e-04, 1.9417e-03, 5.1575e-03, 2.3079e-03,
          2.4796e-03, 6.4278e-04, 7.2670e-04, 2.2087e-03, 1.8063e-03, 9.1410e-04,
          1.0576e-03, 1.0052e-03, 1.2102e-03, 6.8569e-04, 8.0824e-04, 5.1737e-04,
          1.1578e-03, 1.5545e-03, 4.9114e-04, 2.0771e-03, 2.5234e-03, 2.2106e-03,
          1.7986e-03, 3.3455e-03, 3.0766e-03, 5.0735e-04, 2.1706e-03, 1.0309e-03,
          3.3989e-03, 1.3609e-03, 1.7776e-03, 2.0275e-03, 5.8937e-04, 1.1091e-03,
          9.9277e-04, 4.9734e-04, 3.5238e-04, 2.9612e-04, 7.3957e-04, 9.9659e-04,
          1.0042e-03, 5.8460e-04, 6.4135e-04, 1.4591e-03, 1.0729e-03, 1.4715e-03,
          3.8662e-03, 9.1410e-04, 1.5993e-03, 2.1267e-03, 1.9894e-03, 3.9330e-03,
          3.4924e-03, 2.4090e-03, 5.0507e-03, 9.6273e-04, 1.7967e-03, 1.5669e-03,
          7.1430e-04, 7.4911e-04, 3.2353e-04, 9.5415e-04, 1.9178e-03, 5.6362e-04,
          1.5469e-03, 7.5722e-04, 9.9564e-04, 1.0643e-03, 5.6038e-03, 1.8454e-03,
          2.2202e-03, 1.4839e-03, 2.0771e-03, 1.2321e-03, 9.2936e-04, 2.5711e-03,
          2.9488e-03, 3.5477e-03, 8.1329e-03, 2.1210e-03, 1.3838e-03, 9.3699e-04,
          7.3385e-04, 7.1144e-04, 1.2112e-03, 7.0810e-04, 5.6648e-04, 7.0477e-04,
          9.4891e-04, 7.2956e-04, 1.0338e-03, 9.1019e-03, 2.5692e-03, 1.6842e-03,
          9.6512e-04, 6.7902e-03, 1.4467e-03, 1.5411e-03, 1.0977e-03, 1.7052e-03,
          8.1482e-03, 2.8706e-03, 1.7662e-03, 7.7820e-04, 6.6090e-04, 7.7534e-04,
          1.0853e-03, 1.0023e-03, 1.6632e-03, 5.8222e-04, 1.4906e-03, 4.5681e-04,
          7.1335e-04, 5.5695e-03, 1.6388e-02, 1.7729e-03, 3.6221e-03, 5.1460e-03,
          2.7847e-03, 1.6880e-03, 1.1244e-03, 8.5163e-04, 2.1534e-03, 1.6613e-03,
          2.5425e-03, 3.4218e-03, 1.4086e-03, 6.9427e-04, 2.3594e-03, 1.0643e-03,
          7.4625e-04, 1.1683e-03, 2.3251e-03, 3.1338e-03, 1.4830e-03, 7.9298e-04,
          1.8845e-02, 2.5253e-03, 4.5853e-03, 2.0809e-03, 1.5144e-02, 1.9165e-02,
          2.3132e-02, 2.1545e-02, 6.1455e-03, 3.2501e-03, 3.3913e-03, 4.2229e-03,
          2.8210e-03, 1.2293e-03, 3.2635e-03, 1.7948e-03, 2.6207e-03, 1.4553e-03,
          1.0929e-03, 3.0041e-03, 4.6768e-03, 1.9646e-03, 1.0433e-03, 2.9926e-03,
          1.0979e-02, 2.5368e-03, 6.4507e-03, 4.1580e-03, 3.2501e-02, 2.1423e-02,
          4.1924e-03, 9.5987e-04, 1.1272e-03, 2.4281e-03, 2.1687e-03, 1.4744e-03,
          2.8782e-03, 1.3924e-03, 2.2240e-03, 2.5043e-03, 2.0828e-03, 2.6474e-03,
          3.9253e-03, 1.7138e-03, 2.4509e-03, 1.0805e-03, 1.1139e-02, 7.4234e-03,
          2.7637e-03, 2.7847e-02, 2.6199e-02, 6.1981e-02, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 3.5167e-04, 3.5167e-04, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 2.2340e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 2.2340e-04, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 1.9264e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          1.9264e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 1.9264e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 1.3866e-03, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 3.5167e-04, 6.4313e-05, 6.4313e-05, 1.3866e-03,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 3.5167e-04, 2.2340e-04,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 1.9264e-04, 6.4313e-05,
          3.5167e-04, 3.5167e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 1.9264e-04, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05, 3.5167e-04, 6.4313e-05,
          1.9264e-04, 2.2340e-04, 3.5167e-04, 6.4313e-05, 6.4313e-05, 1.9264e-04,
          1.9264e-04, 6.4313e-05, 3.5167e-04, 6.4313e-05, 6.4313e-05, 1.9264e-04,
          1.9264e-04, 1.9264e-04, 6.4313e-05, 6.4313e-05, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 1.9264e-04, 1.9264e-04, 1.2302e-04, 1.9264e-04, 1.9264e-04,
          6.4313e-05, 1.9264e-04, 6.4313e-05, 3.8123e-04, 6.4313e-05, 6.4313e-05,
          6.4313e-05, 6.4313e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5471e-03, 5.2094e-02, 1.4412e-02, 1.1795e-02, 3.3813e-02,
          3.6987e-02, 8.6975e-03, 1.1148e-03, 8.0204e-04, 1.5144e-03, 2.4776e-03,
          9.3174e-04, 3.2845e-03, 1.2722e-03, 2.2144e-03, 1.0500e-03, 7.0457e-03,
          2.8915e-03, 4.9782e-04, 5.6458e-04, 5.3692e-04, 5.6791e-04, 2.5711e-03,
          2.6035e-03, 1.7490e-03, 7.0229e-03, 3.2921e-03, 1.1082e-03, 6.4392e-03,
          2.0752e-03, 2.6817e-03, 2.1458e-03, 3.9749e-03, 1.1597e-03, 1.5078e-03,
          2.0695e-03, 7.2432e-04, 1.9627e-03, 2.0199e-03, 7.5817e-04, 1.1969e-03,
          4.7159e-04, 3.5453e-04, 4.0030e-04, 3.4785e-04, 6.1836e-03, 3.6430e-03,
          7.7772e-04, 7.4291e-04, 2.5253e-03, 2.6779e-03, 2.1572e-03, 1.2274e-03,
          6.1560e-04, 1.2465e-03, 8.8882e-04, 9.7752e-04, 1.2484e-03, 1.1091e-03,
          1.2407e-03, 2.5692e-03, 1.1024e-03, 1.7605e-03, 5.1928e-04, 4.1199e-03,
          7.7629e-04, 4.0340e-04, 5.5885e-04, 2.6822e-04, 2.2101e-04, 3.1662e-04,
          4.0770e-04, 3.4409e-03, 9.8515e-04, 2.9411e-03, 5.3596e-04, 5.5313e-04,
          1.3199e-03, 1.2579e-03, 8.3351e-04, 2.0275e-03, 7.2908e-04, 7.2050e-04,
          7.1764e-04, 2.9125e-03, 7.7486e-04, 1.0109e-03, 3.7694e-04, 4.0436e-04,
          1.8692e-03, 1.4400e-03, 3.0112e-04, 1.1415e-03, 4.6802e-04, 4.5228e-04,
          1.8206e-03, 6.1035e-04, 2.5253e-03, 6.5651e-03, 1.4410e-03, 1.9073e-03,
          2.8782e-03, 5.2118e-04, 6.0987e-04, 2.1534e-03, 4.0283e-03, 4.7264e-03,
          2.3880e-03, 5.6314e-04, 1.9989e-03, 4.0174e-04, 6.5422e-04, 6.0368e-04,
          1.2474e-03, 9.1791e-04, 1.1959e-03, 5.4359e-03, 6.2466e-04, 5.5542e-03,
          8.9340e-03, 5.1956e-03, 6.5756e-04, 7.8964e-04, 3.9101e-03, 5.5933e-04,
          1.0529e-03, 5.8746e-04, 9.2888e-04, 5.4398e-03, 2.9392e-03, 2.8553e-03,
          1.1702e-03, 4.8256e-04, 4.2129e-04, 2.9540e-04, 1.0452e-03, 3.3474e-04,
          1.0338e-02, 2.5368e-03, 3.2310e-03, 3.1319e-03, 3.1528e-03, 7.5035e-03,
          1.0929e-03, 5.6982e-04, 1.1606e-03, 1.5678e-03, 1.0080e-03, 5.3253e-03,
          5.0201e-03, 3.6259e-03, 1.6251e-03, 3.2940e-03, 2.1839e-03, 1.6012e-03,
          1.4992e-03, 1.2159e-03, 4.7135e-04, 4.5705e-04, 3.5591e-03, 6.4926e-03,
          3.4256e-03, 2.1782e-03, 8.4152e-03, 2.6150e-03, 3.9215e-03, 1.8730e-03,
          2.3422e-03, 1.7834e-03, 3.3340e-03, 2.1515e-03, 4.6654e-03, 4.9667e-03,
          2.8305e-03, 1.7939e-03, 2.5120e-03, 1.4057e-03, 2.8000e-03, 2.0084e-03,
          1.1120e-03, 1.8234e-03, 2.7599e-03, 2.1992e-03, 1.3065e-03, 1.7414e-03,
          1.4448e-03, 3.5362e-03, 1.3313e-03, 2.3365e-03, 1.2493e-03, 3.5076e-03,
          2.5558e-03, 1.1759e-03, 3.5572e-03, 1.0284e-02, 1.8511e-03, 3.3569e-03,
          2.4738e-03, 5.4321e-03, 4.1809e-03, 8.2064e-04, 9.1648e-04, 1.3142e-03,
          5.7125e-04, 1.3504e-03, 6.6185e-04, 7.9060e-04, 2.4929e-03, 3.4676e-03,
          1.1768e-03, 5.4932e-03, 2.0199e-03, 1.3914e-03, 7.7133e-03, 1.5440e-03,
          5.1346e-03, 1.1169e-02, 2.0046e-03, 1.4963e-03, 7.6256e-03, 1.2047e-02,
          1.0201e-02, 9.2030e-04, 1.6279e-03, 8.6260e-04, 5.3444e-03, 2.6207e-03,
          6.5422e-04, 6.3658e-04, 6.4039e-04, 2.0084e-03, 3.4466e-03, 6.5384e-03,
          2.9736e-03, 1.8482e-03, 1.5020e-03, 2.9488e-03, 1.0666e-02, 1.2283e-02,
          4.3488e-03, 3.4409e-03, 3.7937e-03, 8.6975e-03, 3.9917e-02, 5.8441e-03,
          1.7071e-03, 1.3056e-03, 7.7915e-04, 1.2207e-03, 1.2283e-03, 2.6398e-03,
          5.0545e-04, 5.6267e-04, 9.7322e-04, 1.9445e-03, 4.8141e-03, 5.2872e-03,
          5.9242e-03, 1.9798e-03, 3.8662e-03, 4.3907e-03, 2.5635e-03, 6.3629e-03,
          6.7024e-03, 1.5945e-03, 4.9973e-03, 3.0930e-02, 7.4883e-03, 2.5616e-03,
          1.9426e-03, 1.0395e-03, 4.0894e-03, 1.3971e-03, 5.4359e-03, 3.9825e-03,
          2.4471e-03, 2.0084e-03, 1.7996e-03, 3.8795e-03, 3.1109e-03, 3.1815e-03,
          2.0046e-03, 2.5520e-03, 2.3289e-03, 3.2768e-03, 5.1758e-02, 2.4438e-05,
          2.0838e-04, 2.4438e-05, 1.0699e-04, 1.0699e-04, 2.4438e-05, 2.4438e-05,
          2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05,
          2.4438e-05, 2.4438e-05, 1.0699e-04, 2.4438e-05, 2.4438e-05, 2.4438e-05,
          2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.0838e-04,
          2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05, 1.0699e-04,
          1.0699e-04, 1.0939e-03, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05,
          1.0699e-04, 2.4438e-05, 1.0699e-04, 2.4438e-05, 2.4438e-05, 1.0699e-04,
          2.4438e-05, 2.4438e-05, 1.0699e-04, 2.4438e-05, 2.4438e-05, 2.4438e-05,
          1.0699e-04, 2.4438e-05, 2.4438e-05, 2.4438e-05, 8.3971e-04, 1.0699e-04,
          2.4438e-05, 2.4438e-05, 2.4438e-05, 3.2616e-04, 1.9240e-04, 1.0699e-04,
          1.0699e-04, 1.0699e-04, 2.4438e-05, 2.4438e-05, 2.9087e-04, 1.0699e-04,
          2.4438e-05, 1.0699e-04, 1.0699e-04, 3.2616e-04, 1.0699e-04, 1.0699e-04,
          2.9087e-04, 2.9087e-04, 1.0699e-04, 1.0699e-04, 1.0699e-04, 2.4438e-05,
          2.9087e-04, 1.0699e-04, 2.4438e-05, 2.7490e-04, 2.4438e-05, 2.4438e-05,
          1.0699e-04, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.9087e-04, 2.4438e-05,
          1.0699e-04, 2.9087e-04, 1.2770e-03, 2.4438e-05, 2.4438e-05, 1.0699e-04,
          3.2616e-04, 1.0699e-04, 2.9087e-04, 2.4438e-05, 8.3971e-04, 3.2616e-04,
          1.0699e-04, 3.2616e-04, 2.4438e-05, 2.4438e-05, 2.4438e-05, 2.4438e-05,
          3.6049e-04, 3.6049e-04, 3.6049e-04, 1.0699e-04, 1.0699e-04, 1.0699e-04,
          1.0699e-04, 1.0699e-04, 3.6049e-04, 1.0699e-04, 1.0699e-04, 3.6049e-04,
          3.6049e-04, 3.6049e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4136e-03, 5.6702e-02, 1.3199e-02, 1.1459e-02, 3.3417e-02,
          4.7424e-02, 7.5569e-03, 9.6655e-04, 7.4911e-04, 1.5745e-03, 2.2221e-03,
          9.3174e-04, 1.9817e-03, 8.2731e-04, 2.2545e-03, 9.0933e-04, 3.8967e-03,
          1.4114e-03, 5.2834e-04, 3.6478e-04, 3.9935e-04, 3.0923e-04, 1.6747e-03,
          2.4643e-03, 1.9255e-03, 6.0844e-03, 3.8891e-03, 6.8235e-04, 3.4618e-03,
          1.2522e-03, 2.5158e-03, 1.1015e-03, 4.3373e-03, 1.6432e-03, 1.5659e-03,
          3.1281e-03, 6.7377e-04, 2.4204e-03, 2.8000e-03, 5.4073e-04, 1.3590e-03,
          4.4036e-04, 3.9744e-04, 3.5763e-04, 3.8123e-04, 3.0899e-03, 3.9291e-03,
          4.6420e-04, 4.2772e-04, 2.0657e-03, 2.8458e-03, 2.0580e-03, 1.3618e-03,
          4.8423e-04, 1.1759e-03, 8.1682e-04, 1.0300e-03, 1.3628e-03, 1.1511e-03,
          1.2474e-03, 1.4629e-03, 1.0185e-03, 1.7033e-03, 6.3419e-04, 4.8141e-03,
          6.1798e-04, 4.5180e-04, 7.4291e-04, 3.6216e-04, 2.6965e-04, 3.2330e-04,
          4.1819e-04, 4.0131e-03, 7.4625e-04, 2.7103e-03, 4.0674e-04, 5.0354e-04,
          1.2999e-03, 1.1921e-03, 7.7629e-04, 2.7466e-03, 6.8808e-04, 7.4768e-04,
          7.3195e-04, 3.2406e-03, 7.4530e-04, 1.1330e-03, 4.1699e-04, 3.9291e-04,
          1.9569e-03, 1.7576e-03, 3.6216e-04, 1.2569e-03, 5.3263e-04, 4.6468e-04,
          9.4652e-04, 4.4680e-04, 2.9850e-03, 1.4458e-02, 1.0977e-03, 1.9169e-03,
          3.1986e-03, 7.3528e-04, 6.0749e-04, 4.2038e-03, 5.2338e-03, 1.8454e-03,
          2.4166e-03, 7.0381e-04, 2.3422e-03, 5.1594e-04, 5.9986e-04, 8.3733e-04,
          1.5974e-03, 1.4420e-03, 5.1498e-04, 5.9967e-03, 6.0701e-04, 1.5755e-03,
          2.5234e-03, 7.2136e-03, 5.9462e-04, 7.1383e-04, 4.1847e-03, 5.0831e-04,
          9.7847e-04, 5.9128e-04, 9.6846e-04, 7.9880e-03, 2.9545e-03, 2.9392e-03,
          1.0290e-03, 4.7946e-04, 5.8889e-04, 4.0412e-04, 1.2426e-03, 4.0388e-04,
          6.9542e-03, 1.1530e-03, 2.3365e-03, 2.8801e-03, 4.4823e-03, 3.1433e-03,
          9.8705e-04, 1.3647e-03, 1.2789e-03, 2.0084e-03, 8.2397e-04, 7.8201e-03,
          7.5150e-03, 6.1035e-03, 2.1572e-03, 4.5395e-03, 1.9741e-03, 2.3384e-03,
          2.0599e-03, 1.8082e-03, 7.3290e-04, 4.4727e-04, 5.4970e-03, 2.9373e-03,
          5.3101e-03, 3.9520e-03, 4.6997e-03, 5.3558e-03, 3.2387e-03, 3.0613e-03,
          3.0708e-03, 2.7809e-03, 4.3144e-03, 3.2291e-03, 3.2673e-03, 8.3237e-03,
          3.7689e-03, 2.1915e-03, 2.7981e-03, 1.2283e-03, 4.0894e-03, 2.2964e-03,
          1.8482e-03, 2.3403e-03, 3.6373e-03, 2.5806e-03, 2.1000e-03, 2.5616e-03,
          1.9531e-03, 1.6155e-03, 2.2907e-03, 1.7767e-03, 1.0586e-03, 4.5662e-03,
          3.2196e-03, 8.4925e-04, 6.6910e-03, 6.3095e-03, 2.8362e-03, 5.2643e-03,
          2.8114e-03, 2.3022e-03, 5.5313e-03, 1.0633e-03, 1.2856e-03, 2.3746e-03,
          5.7936e-04, 2.8687e-03, 7.3671e-04, 1.0910e-03, 3.3474e-03, 2.5520e-03,
          1.3771e-03, 3.8567e-03, 1.4868e-03, 1.4935e-03, 8.8806e-03, 1.2102e-03,
          9.8724e-03, 6.8398e-03, 3.1433e-03, 2.3479e-03, 1.2604e-02, 4.0398e-03,
          1.9547e-02, 8.4114e-04, 2.3899e-03, 1.2722e-03, 6.2943e-03, 4.1504e-03,
          7.2432e-04, 6.8665e-04, 7.0190e-04, 2.4319e-03, 4.3259e-03, 2.3003e-03,
          2.2831e-03, 2.3441e-03, 1.3285e-03, 4.7722e-03, 8.9493e-03, 1.2489e-02,
          4.6425e-03, 3.3875e-03, 3.6430e-03, 9.7198e-03, 1.3008e-02, 2.8515e-03,
          1.6985e-03, 1.4439e-03, 7.8440e-04, 1.2617e-03, 1.1969e-03, 4.1237e-03,
          6.1131e-04, 6.0987e-04, 1.0052e-03, 2.5291e-03, 5.6114e-03, 9.2239e-03,
          4.4441e-03, 2.5406e-03, 5.2528e-03, 7.0343e-03, 2.5806e-03, 6.8207e-03,
          1.0567e-02, 3.4389e-03, 2.9736e-03, 1.2787e-02, 6.4240e-03, 3.7289e-03,
          1.7719e-03, 1.1473e-03, 5.1079e-03, 1.6594e-03, 9.7961e-03, 6.2904e-03,
          2.8839e-03, 3.3760e-03, 9.4414e-04, 2.9030e-03, 2.1973e-03, 2.6188e-03,
          1.6603e-03, 1.8539e-03, 1.5669e-03, 2.7199e-03, 2.2781e-02, 3.0960e-02,
          2.0087e-05, 2.0087e-05, 1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          1.4389e-04, 2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05, 1.4389e-04,
          2.0087e-05, 2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.4389e-04, 1.4389e-04,
          2.0087e-05, 2.0087e-05, 1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05,
          1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05, 1.4389e-04, 1.4389e-04,
          2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05, 1.4389e-04, 1.4389e-04,
          1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05, 1.4389e-04, 2.0087e-05,
          1.4389e-04, 1.4389e-04, 2.0087e-05, 1.4389e-04, 2.0087e-05, 2.0087e-05,
          1.4389e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 1.4389e-04, 2.0087e-05,
          1.4389e-04, 1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05, 1.4389e-04,
          1.4389e-04, 1.4389e-04, 1.4389e-04, 2.0087e-05, 1.4389e-04, 1.4389e-04,
          1.4389e-04, 1.4389e-04, 2.0087e-05, 2.0087e-05, 2.0087e-05, 2.0087e-05,
          1.4389e-04, 1.4389e-04, 6.4659e-04, 1.4389e-04, 1.4389e-04, 1.4389e-04,
          2.0087e-05, 1.4389e-04, 1.4389e-04, 1.4389e-04, 2.0087e-05, 6.4659e-04,
          1.4389e-04, 1.4389e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.9520e-03, 4.8584e-02, 1.1353e-02, 1.1391e-02, 3.5706e-02,
          4.0283e-02, 7.7515e-03, 1.1826e-03, 8.2493e-04, 2.0275e-03, 4.5853e-03,
          1.7338e-03, 9.4557e-04, 5.3787e-04, 1.6870e-03, 8.8549e-04, 9.9754e-04,
          6.0749e-04, 5.1594e-04, 1.8549e-04, 2.0599e-04, 1.9693e-04, 7.7772e-04,
          5.2032e-03, 2.9964e-03, 6.3629e-03, 2.7122e-03, 3.9387e-04, 1.1635e-03,
          7.4053e-04, 2.3537e-03, 4.5133e-04, 1.9388e-03, 1.2722e-03, 1.1597e-03,
          2.0542e-03, 8.9169e-04, 1.0281e-03, 1.3762e-03, 2.8753e-04, 8.7357e-04,
          5.6791e-04, 4.7660e-04, 5.6887e-04, 3.8767e-04, 1.0395e-03, 1.6289e-03,
          2.8276e-04, 2.3913e-04, 2.3766e-03, 1.8177e-03, 1.7376e-03, 1.0471e-03,
          6.9189e-04, 1.2016e-03, 7.4434e-04, 1.2598e-03, 3.3340e-03, 2.3594e-03,
          2.1935e-03, 6.4182e-04, 8.5497e-04, 9.4795e-04, 1.3599e-03, 2.7122e-03,
          1.2712e-03, 1.0738e-03, 5.9414e-04, 2.9612e-04, 3.1829e-04, 6.0177e-04,
          4.5824e-04, 3.3798e-03, 8.2445e-04, 2.4376e-03, 5.9128e-04, 5.9986e-04,
          1.2503e-03, 1.5574e-03, 1.4677e-03, 2.7561e-03, 1.5917e-03, 1.7538e-03,
          1.1415e-03, 2.2869e-03, 7.0524e-04, 1.7948e-03, 6.6328e-04, 7.0190e-04,
          4.3831e-03, 1.6098e-03, 5.6076e-04, 2.3861e-03, 6.7854e-04, 4.6682e-04,
          5.6124e-04, 4.4155e-04, 1.4296e-03, 1.9913e-03, 1.6193e-03, 1.9350e-03,
          6.7177e-03, 1.0929e-03, 1.5430e-03, 3.0651e-03, 3.3932e-03, 7.9966e-04,
          2.1191e-03, 7.0953e-04, 2.4357e-03, 5.9462e-04, 1.1826e-03, 8.2302e-04,
          2.5158e-03, 8.4639e-04, 2.6274e-04, 2.8839e-03, 6.9952e-04, 1.0099e-03,
          1.3828e-03, 4.2953e-03, 1.0204e-03, 9.0218e-04, 3.3379e-03, 9.8324e-04,
          1.2808e-03, 1.1320e-03, 1.8225e-03, 4.0398e-03, 1.6565e-03, 2.4319e-03,
          1.7529e-03, 9.0551e-04, 6.7377e-04, 4.7469e-04, 1.8263e-03, 4.6229e-04,
          1.5335e-03, 4.7278e-04, 1.0557e-03, 1.9293e-03, 1.8282e-03, 1.5593e-03,
          1.8024e-03, 1.7834e-03, 1.9436e-03, 1.8005e-03, 1.9913e-03, 7.9498e-03,
          6.0387e-03, 4.3068e-03, 2.4338e-03, 1.4763e-03, 1.4858e-03, 1.2665e-03,
          1.6289e-03, 9.5558e-04, 1.0281e-03, 7.0524e-04, 2.9869e-03, 7.7343e-04,
          2.0275e-03, 1.3828e-03, 1.4420e-03, 1.8110e-03, 1.8978e-03, 1.6632e-03,
          6.4774e-03, 1.9245e-03, 4.5280e-03, 2.2202e-03, 1.3323e-03, 3.5267e-03,
          6.6719e-03, 4.8103e-03, 2.5787e-03, 1.0281e-03, 2.5158e-03, 1.5774e-03,
          1.2159e-03, 1.6146e-03, 2.1362e-03, 2.0676e-03, 9.3269e-04, 1.0414e-03,
          3.2082e-03, 8.6117e-04, 1.1377e-03, 2.4853e-03, 2.6531e-03, 3.0823e-03,
          2.7294e-03, 1.3952e-03, 2.6817e-03, 2.3575e-03, 4.2610e-03, 2.9984e-03,
          4.0817e-03, 1.3294e-03, 3.0174e-03, 1.3504e-03, 1.2016e-03, 1.3609e-03,
          1.0386e-03, 1.7786e-03, 8.2874e-04, 1.1234e-03, 5.2605e-03, 7.7009e-04,
          1.2836e-03, 2.5425e-03, 2.2297e-03, 4.0550e-03, 6.3972e-03, 3.3531e-03,
          3.5954e-03, 2.3842e-03, 2.6093e-03, 1.7033e-03, 3.9139e-03, 2.4490e-03,
          3.4580e-03, 1.2083e-03, 2.0027e-03, 1.0710e-03, 4.2915e-03, 2.5082e-03,
          1.2980e-03, 1.6184e-03, 1.3342e-03, 7.3776e-03, 3.3245e-03, 9.6989e-04,
          2.9449e-03, 2.6932e-03, 3.9101e-03, 5.2795e-03, 1.3237e-02, 9.0332e-02,
          2.0035e-02, 1.0292e-02, 6.5613e-03, 3.4122e-03, 4.5242e-03, 1.7519e-03,
          1.2035e-03, 2.8706e-03, 1.9722e-03, 3.4161e-03, 2.6722e-03, 3.0594e-03,
          7.9060e-04, 6.9666e-04, 1.2531e-03, 1.5173e-03, 1.8568e-03, 3.9177e-03,
          3.6983e-03, 3.3150e-03, 8.6288e-03, 9.0256e-03, 5.6458e-03, 2.5421e-02,
          9.4223e-03, 1.8063e-03, 2.3155e-03, 4.6463e-03, 4.2953e-03, 2.8801e-03,
          3.6850e-03, 1.5411e-03, 3.7804e-03, 1.8415e-03, 2.9259e-03, 5.8212e-03,
          6.1188e-03, 4.9973e-03, 5.0545e-04, 1.4553e-03, 2.2125e-03, 5.7755e-03,
          3.6259e-03, 4.5662e-03, 2.8210e-03, 4.7989e-03, 7.9498e-03, 4.8485e-03,
          6.3721e-02, 3.2425e-05, 2.8110e-04, 1.6701e-04, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 1.4663e-04, 2.8110e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          1.4663e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.6701e-04,
          3.2425e-05, 1.6701e-04, 3.2425e-05, 1.4663e-04, 3.2425e-05, 3.2425e-05,
          1.4663e-04, 3.2425e-05, 1.6701e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 1.4663e-04, 3.2425e-05, 1.4663e-04, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.6701e-04, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 2.8110e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 1.6701e-04, 3.2425e-05, 3.2425e-05, 2.8110e-04, 2.8110e-04,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 1.6701e-04, 3.2425e-05,
          2.8110e-04, 2.8110e-04, 3.2425e-05, 3.2425e-05, 1.6701e-04, 3.2425e-05,
          1.4663e-04, 3.2425e-05, 3.2425e-05, 1.6701e-04, 3.2425e-05, 3.2425e-05,
          3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05, 2.8110e-04, 3.2425e-05,
          1.6701e-04, 2.8110e-04, 1.6701e-04, 3.2425e-05, 3.2425e-05, 1.6701e-04,
          1.6701e-04, 1.6701e-04, 2.8110e-04, 3.2425e-05, 1.6701e-04, 1.6701e-04,
          1.6701e-04, 6.3705e-04, 3.2425e-05, 3.2425e-05, 3.2425e-05, 3.2425e-05,
          5.5504e-04, 5.5504e-04, 1.6701e-04, 1.6701e-04, 1.6701e-04, 1.6701e-04,
          3.2425e-05, 1.6701e-04, 1.6701e-04, 6.2752e-04, 1.0157e-03, 5.5504e-04,
          5.5504e-04, 1.6701e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.9210e-03, 5.5023e-02, 1.2894e-02, 1.9485e-02, 3.2288e-02,
          4.2145e-02, 1.1230e-02, 1.1683e-03, 7.3576e-04, 4.3564e-03, 4.5128e-03,
          1.8587e-03, 1.6174e-03, 5.1403e-04, 4.1199e-03, 1.1492e-03, 2.4319e-03,
          1.4620e-03, 5.7650e-04, 2.2399e-04, 3.4785e-04, 2.6894e-04, 2.2545e-03,
          4.9515e-03, 2.3441e-03, 8.7891e-03, 8.0185e-03, 5.3787e-04, 3.3035e-03,
          8.2350e-04, 2.5463e-03, 9.1934e-04, 7.1640e-03, 6.0959e-03, 1.5697e-03,
          4.2610e-03, 6.5327e-04, 6.4697e-03, 6.2447e-03, 3.3569e-04, 3.2177e-03,
          5.8031e-04, 5.1451e-04, 4.5037e-04, 5.6124e-04, 3.1757e-03, 8.8272e-03,
          3.3259e-04, 3.8362e-04, 2.8458e-03, 1.7042e-03, 1.6136e-03, 1.7481e-03,
          3.3998e-04, 7.9012e-04, 4.9400e-04, 9.2316e-04, 9.0694e-04, 8.9216e-04,
          8.6832e-04, 1.0891e-03, 9.0265e-04, 1.8463e-03, 7.7105e-04, 1.3107e-02,
          6.4898e-04, 6.4230e-04, 1.9522e-03, 3.0589e-04, 2.6155e-04, 3.0684e-04,
          4.0793e-04, 1.1497e-02, 5.6505e-04, 1.8902e-03, 3.1018e-04, 3.3712e-04,
          7.3147e-04, 8.1444e-04, 7.3528e-04, 3.6373e-03, 8.7929e-04, 5.5027e-04,
          5.0688e-04, 2.9850e-03, 5.5027e-04, 2.5024e-03, 4.8518e-04, 4.0650e-04,
          4.8141e-03, 3.8662e-03, 2.6298e-04, 2.7637e-03, 3.4952e-04, 4.2939e-04,
          6.6900e-04, 3.4666e-04, 2.8400e-03, 2.2888e-03, 7.8535e-04, 1.2846e-03,
          5.1155e-03, 5.3787e-04, 8.0490e-04, 2.7409e-03, 7.9727e-03, 1.0662e-03,
          2.1477e-03, 6.3705e-04, 4.4365e-03, 3.1590e-04, 5.2786e-04, 1.0214e-03,
          1.5059e-03, 1.1454e-03, 4.2868e-04, 1.0345e-02, 5.8699e-04, 1.1845e-03,
          1.7624e-03, 5.2795e-03, 4.6682e-04, 6.9666e-04, 1.4297e-02, 3.6478e-04,
          5.7793e-04, 4.1175e-04, 6.2323e-04, 7.6408e-03, 3.2501e-03, 2.4910e-03,
          8.8978e-04, 6.5136e-04, 1.0166e-03, 4.1389e-04, 2.1763e-03, 3.1471e-04,
          1.0605e-02, 7.3147e-04, 1.2732e-03, 4.0894e-03, 6.9389e-03, 2.0695e-03,
          7.2670e-04, 7.9679e-04, 1.0195e-03, 8.6975e-04, 6.9571e-04, 5.1918e-03,
          7.1259e-03, 5.2299e-03, 9.5654e-04, 2.4700e-03, 1.1988e-03, 2.7332e-03,
          3.6621e-03, 2.0580e-03, 9.3794e-04, 3.8958e-04, 5.4283e-03, 1.6108e-03,
          1.0025e-02, 8.2626e-03, 6.0539e-03, 8.8425e-03, 1.9131e-03, 1.2522e-03,
          3.6430e-03, 1.8749e-03, 3.9444e-03, 2.9125e-03, 3.3283e-03, 9.7580e-03,
          1.3590e-03, 7.4673e-04, 1.4381e-03, 1.3943e-03, 6.4392e-03, 4.6730e-03,
          3.5782e-03, 3.3627e-03, 1.0742e-02, 2.9831e-03, 1.2875e-03, 2.1191e-03,
          3.1700e-03, 7.8440e-04, 2.7542e-03, 1.3447e-03, 7.9107e-04, 8.9035e-03,
          3.4447e-03, 5.5790e-04, 8.3237e-03, 6.3629e-03, 5.0354e-03, 8.2245e-03,
          1.1721e-03, 1.2217e-03, 7.2556e-03, 1.4935e-03, 2.1400e-03, 4.0855e-03,
          6.6280e-04, 2.5959e-03, 7.5388e-04, 9.8133e-04, 5.0507e-03, 2.3632e-03,
          9.4318e-04, 3.4313e-03, 1.0262e-03, 1.5249e-03, 8.0185e-03, 7.2908e-04,
          4.0283e-03, 3.4981e-03, 7.0724e-03, 2.2278e-03, 1.3115e-02, 1.9627e-03,
          7.0496e-03, 7.8583e-04, 2.7142e-03, 1.2560e-03, 2.0340e-02, 8.6746e-03,
          9.5224e-04, 4.1604e-04, 5.0735e-04, 9.5701e-04, 3.6087e-03, 7.4482e-04,
          1.8024e-03, 1.3056e-03, 6.4945e-04, 2.0103e-03, 3.9520e-03, 4.0016e-03,
          1.4334e-03, 1.3294e-03, 1.2054e-03, 4.6196e-03, 4.4708e-03, 2.1343e-03,
          1.8559e-03, 1.4782e-03, 4.8232e-04, 6.7568e-04, 8.1062e-04, 5.8136e-03,
          4.6229e-04, 4.6134e-04, 8.0204e-04, 1.7471e-03, 5.7716e-03, 1.0658e-02,
          2.3251e-03, 1.3027e-03, 1.8301e-03, 2.3785e-03, 7.8773e-04, 1.9522e-03,
          4.4174e-03, 5.1460e-03, 1.5717e-03, 6.8016e-03, 5.4131e-03, 3.6316e-03,
          1.1473e-03, 1.3266e-03, 1.0521e-02, 1.2503e-03, 6.4011e-03, 4.3793e-03,
          1.5135e-03, 1.2598e-03, 7.0858e-04, 1.9274e-03, 1.6689e-03, 1.2798e-03,
          1.4095e-03, 8.5878e-04, 6.7806e-04, 1.1187e-03, 3.7022e-03, 3.6049e-03,
          2.1915e-03, 3.8574e-02, 2.5392e-05, 2.4414e-04, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 1.8489e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.4414e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 1.8489e-04,
          2.5392e-05, 2.5392e-05, 1.8489e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.6083e-04, 2.5392e-05, 1.8489e-04, 2.5392e-05, 2.5392e-05, 1.8489e-04,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.6083e-04, 2.5392e-05,
          3.7599e-04, 2.4414e-04, 2.5392e-05, 2.5392e-05, 1.8489e-04, 1.8489e-04,
          2.6083e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05, 1.8489e-04, 2.6083e-04,
          4.0364e-04, 4.0364e-04, 2.6083e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 2.6083e-04, 2.5392e-05, 2.5392e-05,
          2.4414e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05, 4.2033e-04, 2.5392e-05,
          4.2033e-04, 1.8489e-04, 2.4414e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          4.2033e-04, 2.5392e-05, 4.0364e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.4414e-04, 2.4414e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.4414e-04, 2.5392e-05, 2.4414e-04, 2.5392e-05, 2.5392e-05, 2.5392e-05,
          2.5392e-05, 2.5392e-05, 2.5392e-05, 3.4666e-04, 2.5392e-05, 2.4414e-04,
          2.4414e-04, 2.5392e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1003e-03, 4.8065e-02, 1.7532e-02, 1.7258e-02, 3.5950e-02,
          3.3661e-02, 1.0330e-02, 1.7262e-03, 1.0128e-03, 2.2163e-03, 3.4599e-03,
          1.2913e-03, 7.4883e-03, 2.2297e-03, 1.9302e-03, 1.8654e-03, 1.8616e-02,
          8.5526e-03, 8.3780e-04, 6.7377e-04, 6.0558e-04, 1.8368e-03, 5.4588e-03,
          4.0092e-03, 1.0881e-03, 8.0795e-03, 6.9771e-03, 1.2369e-03, 1.9714e-02,
          3.8853e-03, 4.0245e-03, 5.4703e-03, 3.5763e-03, 2.4986e-03, 1.2770e-03,
          1.1673e-03, 3.8457e-04, 2.2335e-03, 2.4128e-03, 1.0643e-03, 1.0796e-03,
          4.4012e-04, 3.0732e-04, 3.9864e-04, 3.2020e-04, 2.1042e-02, 3.6392e-03,
          1.7271e-03, 9.8515e-04, 3.0117e-03, 1.6785e-03, 2.5921e-03, 1.2693e-03,
          3.8409e-04, 1.1120e-03, 4.3774e-04, 4.3392e-04, 8.4829e-04, 5.4789e-04,
          5.6934e-04, 4.3221e-03, 6.8474e-04, 1.1578e-03, 4.8876e-04, 3.6373e-03,
          6.2943e-04, 5.4407e-04, 9.4318e-04, 1.9991e-04, 1.9276e-04, 3.1400e-04,
          4.1008e-04, 6.8130e-03, 6.3562e-04, 3.4142e-03, 3.8719e-04, 3.8290e-04,
          1.1511e-03, 4.4656e-04, 4.8780e-04, 1.3800e-03, 6.1512e-04, 3.4809e-04,
          3.3450e-04, 3.3760e-03, 4.8971e-04, 2.0638e-03, 3.9959e-04, 3.2234e-04,
          3.9978e-03, 2.5253e-03, 2.1648e-04, 2.0943e-03, 3.2473e-04, 3.7789e-04,
          2.0447e-03, 3.3927e-04, 2.7828e-03, 2.3422e-03, 3.9649e-04, 6.0272e-04,
          3.6221e-03, 4.5490e-04, 7.2670e-04, 8.2064e-04, 2.4300e-03, 2.6989e-03,
          2.3766e-03, 4.3035e-04, 2.2621e-03, 2.2662e-04, 4.4537e-04, 6.2799e-04,
          6.9094e-04, 6.0701e-04, 1.0071e-03, 4.3259e-03, 5.2357e-04, 1.7958e-03,
          3.3493e-03, 1.9512e-03, 2.5892e-04, 1.1969e-03, 3.6068e-03, 2.1827e-04,
          3.0327e-04, 3.3259e-04, 4.7874e-04, 2.6264e-03, 3.8433e-03, 3.1567e-03,
          8.6498e-04, 5.8126e-04, 7.3433e-04, 3.5000e-04, 1.3943e-03, 2.6584e-04,
          2.9266e-02, 2.1839e-03, 4.0588e-03, 4.3602e-03, 4.5929e-03, 4.9744e-03,
          3.6716e-04, 4.3106e-04, 4.9543e-04, 4.7708e-04, 5.4693e-04, 1.7824e-03,
          3.4084e-03, 1.5554e-03, 5.2023e-04, 2.2659e-03, 1.5097e-03, 1.1015e-03,
          1.5593e-03, 8.3113e-04, 1.1015e-03, 3.4881e-04, 2.1782e-03, 5.5199e-03,
          4.7493e-03, 3.1853e-03, 3.4546e-02, 4.0855e-03, 1.0643e-03, 7.0286e-04,
          2.4776e-03, 9.7847e-04, 1.4248e-03, 1.5078e-03, 9.2850e-03, 3.7937e-03,
          9.0122e-04, 6.0844e-04, 8.1491e-04, 1.0500e-03, 2.2297e-03, 6.8283e-03,
          1.3218e-03, 1.7767e-03, 3.7441e-03, 4.9515e-03, 2.9678e-03, 1.8806e-03,
          3.1204e-03, 2.7313e-03, 1.9197e-03, 6.8617e-04, 5.8174e-04, 3.3894e-03,
          5.0735e-03, 3.0875e-04, 3.5744e-03, 3.4180e-02, 3.1338e-03, 2.6016e-03,
          7.9966e-04, 1.7033e-03, 2.9202e-03, 9.4032e-04, 1.6575e-03, 1.5364e-03,
          5.1498e-04, 1.3342e-03, 4.9686e-04, 1.1921e-03, 3.9368e-03, 1.4954e-02,
          1.7471e-03, 2.0504e-03, 7.8440e-04, 1.2245e-03, 3.8757e-03, 5.3358e-04,
          1.7576e-03, 8.8577e-03, 4.0054e-03, 1.3218e-03, 7.3891e-03, 5.7755e-03,
          7.7133e-03, 5.7411e-04, 1.2197e-03, 9.6560e-04, 6.1302e-03, 3.5152e-03,
          9.1362e-04, 3.7622e-04, 3.8290e-04, 8.8596e-04, 1.7929e-03, 2.5349e-03,
          1.5802e-03, 9.4271e-04, 5.4455e-04, 1.1778e-03, 1.8320e-03, 2.9087e-03,
          1.1148e-03, 9.3746e-04, 6.5708e-04, 2.8896e-03, 9.1248e-03, 3.4676e-03,
          1.0862e-03, 1.4057e-03, 4.1533e-04, 6.2752e-04, 6.6853e-04, 2.0866e-03,
          3.7408e-04, 3.3712e-04, 6.5184e-04, 1.0595e-03, 1.1505e-02, 6.9504e-03,
          1.3657e-03, 1.1482e-03, 9.6464e-04, 1.2150e-03, 5.1641e-04, 1.2474e-03,
          2.0123e-03, 1.8749e-03, 3.4733e-03, 1.9043e-02, 1.3832e-02, 1.5106e-03,
          1.1263e-03, 1.3390e-03, 3.3684e-03, 7.9346e-04, 3.3512e-03, 2.1381e-03,
          1.2093e-03, 1.3609e-03, 1.5974e-03, 5.9433e-03, 1.2865e-03, 1.1568e-03,
          1.5879e-03, 6.6948e-04, 5.1022e-04, 7.0333e-04, 5.0049e-03, 2.5883e-03,
          1.9255e-03, 8.7814e-03, 5.2368e-02, 2.7120e-05, 2.7120e-05, 9.5367e-05,
          6.6376e-04, 9.5367e-05, 9.5367e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05,
          2.7120e-05, 2.7120e-05, 9.5367e-05, 2.7120e-05, 6.3467e-04, 1.2934e-04,
          2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05, 1.2934e-04,
          2.7120e-05, 9.5367e-05, 2.7120e-05, 1.9765e-04, 2.7120e-05, 9.5367e-05,
          7.3671e-04, 1.9765e-04, 2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05,
          2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05,
          2.7120e-05, 2.7120e-05, 9.5367e-05, 1.2934e-04, 9.5367e-05, 9.5367e-05,
          9.9373e-04, 2.7120e-05, 2.7120e-05, 2.7120e-05, 9.5367e-05, 1.9765e-04,
          9.5367e-05, 1.2934e-04, 2.7120e-05, 9.5367e-05, 2.0313e-04, 2.7120e-05,
          9.5367e-05, 1.2934e-04, 2.7120e-05, 2.7120e-05, 9.3460e-04, 9.5367e-05,
          5.2214e-04, 9.5367e-05, 1.9765e-04, 3.3307e-04, 9.5367e-05, 2.7132e-04,
          7.5436e-04, 5.1689e-04, 5.7268e-04, 9.5367e-05, 9.5367e-05, 2.7120e-05,
          5.7650e-04, 9.5367e-05, 2.7120e-05, 3.7360e-04, 2.7120e-05, 2.7120e-05,
          2.7120e-05, 3.2854e-04, 2.7120e-05, 2.7120e-05, 4.1461e-04, 2.7120e-05,
          8.9169e-04, 1.9765e-04, 4.3535e-04, 2.7120e-05, 2.7120e-05, 9.5367e-05,
          3.7360e-04, 9.5367e-05, 5.1689e-04, 9.5367e-05, 1.9765e-04, 2.7132e-04,
          9.5367e-05, 1.0948e-03, 2.7120e-05, 2.7120e-05, 2.7120e-05, 2.7120e-05,
          2.7537e-04, 2.7537e-04, 9.5367e-05, 9.5367e-05, 9.5367e-05, 2.7120e-05,
          2.7120e-05, 3.7766e-04, 2.7537e-04, 5.7220e-04, 9.5367e-05, 7.5245e-04,
          9.5367e-05, 9.5367e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.5618e-03, 5.6732e-02, 1.7990e-02, 1.8173e-02, 3.9948e-02,
          4.0161e-02, 1.2268e-02, 2.3308e-03, 1.4601e-03, 2.6302e-03, 4.4670e-03,
          1.5430e-03, 9.9716e-03, 3.9711e-03, 1.7948e-03, 2.3079e-03, 2.0798e-02,
          1.0307e-02, 1.3742e-03, 8.4877e-04, 6.8855e-04, 1.6537e-03, 5.2185e-03,
          5.3177e-03, 1.3056e-03, 9.8877e-03, 6.2408e-03, 1.3456e-03, 1.7471e-02,
          5.7526e-03, 4.4975e-03, 5.6839e-03, 2.9869e-03, 2.9659e-03, 1.3895e-03,
          1.1082e-03, 3.7909e-04, 1.9779e-03, 2.3842e-03, 1.1253e-03, 1.1959e-03,
          4.3392e-04, 3.2425e-04, 3.7956e-04, 3.8505e-04, 2.0554e-02, 2.7256e-03,
          1.9569e-03, 1.1063e-03, 3.8128e-03, 1.7509e-03, 2.9011e-03, 1.4648e-03,
          4.7684e-04, 1.3790e-03, 4.2605e-04, 3.9029e-04, 9.3889e-04, 5.7220e-04,
          6.2466e-04, 2.5997e-03, 7.3528e-04, 1.0109e-03, 5.5456e-04, 2.9945e-03,
          8.7738e-04, 7.7105e-04, 8.7738e-04, 2.1231e-04, 1.8966e-04, 3.4094e-04,
          4.3392e-04, 6.0730e-03, 6.6185e-04, 3.9635e-03, 4.2200e-04, 4.7278e-04,
          1.3838e-03, 3.6764e-04, 5.5075e-04, 1.2465e-03, 6.6710e-04, 3.8576e-04,
          3.4833e-04, 3.9787e-03, 5.5552e-04, 2.5921e-03, 5.2071e-04, 4.2367e-04,
          4.6425e-03, 2.1992e-03, 2.7418e-04, 2.4166e-03, 3.1877e-04, 3.8481e-04,
          1.7748e-03, 3.2806e-04, 2.5120e-03, 2.0275e-03, 3.4547e-04, 5.0735e-04,
          4.2763e-03, 7.2193e-04, 8.3399e-04, 6.8188e-04, 2.1553e-03, 1.4544e-03,
          2.7752e-03, 4.1676e-04, 2.3079e-03, 3.0255e-04, 6.2847e-04, 7.5293e-04,
          1.0319e-03, 5.2691e-04, 6.6948e-04, 3.3398e-03, 4.9496e-04, 1.2789e-03,
          2.2316e-03, 1.5411e-03, 2.4867e-04, 1.7338e-03, 3.0918e-03, 2.2924e-04,
          2.4629e-04, 3.2711e-04, 4.5991e-04, 2.2602e-03, 3.9101e-03, 4.0474e-03,
          1.0509e-03, 7.9298e-04, 1.1816e-03, 4.7684e-04, 1.3962e-03, 2.8586e-04,
          1.5884e-02, 2.0733e-03, 6.5079e-03, 5.8098e-03, 3.9673e-03, 3.7937e-03,
          3.3879e-04, 1.1292e-03, 4.3106e-04, 4.1914e-04, 6.0606e-04, 1.7700e-03,
          2.7351e-03, 1.4677e-03, 4.9877e-04, 1.9646e-03, 1.3313e-03, 8.7452e-04,
          1.3447e-03, 6.8712e-04, 2.1172e-03, 4.0555e-04, 1.8597e-03, 4.1161e-03,
          2.9430e-03, 2.2831e-03, 2.7298e-02, 4.3716e-03, 7.7248e-04, 6.3133e-04,
          3.5706e-03, 8.0776e-04, 1.3323e-03, 2.0351e-03, 7.6447e-03, 2.0809e-03,
          1.0290e-03, 6.9237e-04, 8.1348e-04, 8.9836e-04, 1.6632e-03, 8.0032e-03,
          1.0977e-03, 1.7557e-03, 2.6741e-03, 5.6915e-03, 5.1880e-03, 3.0804e-03,
          5.1270e-03, 4.2686e-03, 2.5349e-03, 5.7030e-04, 6.2084e-04, 3.1166e-03,
          6.0349e-03, 2.6584e-04, 2.3079e-03, 3.1464e-02, 3.4199e-03, 1.9083e-03,
          8.4877e-04, 1.0757e-03, 2.7714e-03, 9.6893e-04, 1.5907e-03, 1.3103e-03,
          5.4646e-04, 1.2417e-03, 4.8351e-04, 2.3746e-03, 5.1498e-03, 9.6512e-03,
          3.1643e-03, 1.4839e-03, 6.6137e-04, 1.7891e-03, 3.0994e-03, 5.0974e-04,
          1.3123e-03, 5.5580e-03, 5.2757e-03, 2.0485e-03, 4.8370e-03, 4.6005e-03,
          6.1646e-03, 5.1117e-04, 9.8801e-04, 1.1444e-03, 4.8027e-03, 3.3245e-03,
          1.1292e-03, 5.0116e-04, 3.9959e-04, 1.0891e-03, 1.7920e-03, 2.6569e-03,
          1.3018e-03, 8.3065e-04, 6.3372e-04, 1.4372e-03, 1.4277e-03, 2.5578e-03,
          1.0376e-03, 8.7404e-04, 5.0068e-04, 1.8654e-03, 6.3515e-03, 2.3441e-03,
          8.3256e-04, 1.9531e-03, 4.4918e-04, 6.8808e-04, 6.9761e-04, 1.6537e-03,
          4.1103e-04, 3.3736e-04, 7.0572e-04, 8.6164e-04, 1.0262e-02, 6.8016e-03,
          9.4748e-04, 1.1520e-03, 8.8692e-04, 1.2169e-03, 4.7326e-04, 1.2321e-03,
          1.6613e-03, 2.6894e-03, 4.8447e-03, 1.4290e-02, 1.6495e-02, 1.4467e-03,
          1.2751e-03, 1.5192e-03, 2.8305e-03, 7.4530e-04, 2.4929e-03, 2.1095e-03,
          1.2989e-03, 2.2049e-03, 1.3514e-03, 5.9471e-03, 1.0262e-03, 1.1539e-03,
          1.8892e-03, 5.6314e-04, 4.1890e-04, 5.8985e-04, 3.4828e-03, 2.2182e-03,
          2.1057e-03, 5.7030e-03, 2.0447e-02, 4.4128e-02, 3.2902e-05, 8.7023e-05,
          3.2902e-05, 8.7023e-05, 3.2902e-05, 8.7023e-05, 3.2902e-05, 3.2902e-05,
          3.2902e-05, 3.2902e-05, 8.7023e-05, 3.2902e-05, 2.3735e-04, 3.2902e-05,
          8.7023e-05, 2.3735e-04, 8.7023e-05, 8.7023e-05, 3.2902e-05, 1.6332e-04,
          8.7023e-05, 8.7023e-05, 8.7023e-05, 3.6788e-04, 8.7023e-05, 2.3735e-04,
          2.3735e-04, 3.2902e-05, 8.7023e-05, 8.7023e-05, 3.2902e-05, 3.2902e-05,
          3.2902e-05, 3.2902e-05, 8.7023e-05, 3.2902e-05, 3.2902e-05, 8.7023e-05,
          3.2902e-05, 2.1744e-04, 2.3735e-04, 2.1744e-04, 3.2902e-05, 8.7023e-05,
          2.1744e-04, 3.2902e-05, 8.7023e-05, 2.3735e-04, 3.2902e-05, 3.6788e-04,
          8.7023e-05, 3.2902e-05, 3.2902e-05, 8.7023e-05, 7.5293e-04, 3.2902e-05,
          2.3735e-04, 8.7023e-05, 8.7023e-05, 3.2902e-05, 2.3735e-04, 8.7023e-05,
          8.7023e-05, 8.7023e-05, 1.6332e-04, 3.2902e-05, 2.3735e-04, 3.6788e-04,
          2.3735e-04, 2.3735e-04, 2.3735e-04, 8.7023e-05, 2.3735e-04, 8.7023e-05,
          3.6788e-04, 2.3735e-04, 3.2902e-05, 7.5293e-04, 3.2902e-05, 3.2902e-05,
          8.7023e-05, 8.7023e-05, 8.7023e-05, 3.2902e-05, 8.7023e-05, 8.7023e-05,
          9.1219e-04, 2.1744e-04, 2.1744e-04, 3.2902e-05, 8.7023e-05, 8.7023e-05,
          2.1744e-04, 2.3735e-04, 1.0424e-03, 8.7023e-05, 3.2902e-05, 8.7023e-05,
          9.1219e-04, 1.8740e-03, 3.2902e-05, 3.2902e-05, 8.7023e-05, 3.2902e-05,
          8.7023e-05, 8.7023e-05, 3.2902e-05, 3.2902e-05, 8.7023e-05, 3.2902e-05,
          3.2902e-05, 8.7023e-05, 3.2997e-04, 5.0592e-04, 8.7023e-05, 6.5613e-04,
          8.7023e-05, 8.7023e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.7569e-03, 4.6021e-02, 1.7059e-02, 1.5976e-02, 7.7881e-02,
          3.1464e-02, 1.6556e-02, 4.7421e-04, 4.4608e-04, 7.9823e-04, 2.0103e-03,
          6.1560e-04, 5.0259e-04, 6.4611e-04, 8.5402e-04, 4.5967e-04, 6.7711e-04,
          4.3941e-04, 3.5310e-04, 1.2560e-03, 9.1267e-04, 8.5115e-04, 5.6267e-04,
          2.3804e-03, 2.1763e-03, 1.1871e-02, 2.2793e-03, 1.3170e-03, 6.1893e-04,
          6.0034e-04, 6.5517e-04, 3.3236e-04, 9.0790e-04, 4.4870e-04, 4.6778e-04,
          1.6451e-03, 2.2774e-03, 1.1330e-03, 7.4291e-04, 7.9679e-04, 2.5043e-03,
          2.9697e-03, 1.8063e-03, 2.2984e-03, 9.8038e-04, 5.6839e-04, 1.3924e-03,
          4.1580e-04, 9.5320e-04, 8.5373e-03, 9.1648e-04, 6.0511e-04, 4.7660e-04,
          5.3883e-04, 4.8423e-04, 1.3199e-03, 1.5850e-03, 2.4529e-03, 3.3035e-03,
          4.9553e-03, 1.0233e-03, 2.4891e-03, 1.5411e-03, 2.0313e-03, 2.3232e-03,
          1.4038e-03, 5.9843e-04, 7.2813e-04, 1.1711e-03, 1.4124e-03, 1.2503e-03,
          1.3494e-03, 1.4954e-03, 1.3599e-03, 7.7248e-04, 5.1451e-04, 3.5000e-04,
          8.4734e-04, 1.1883e-03, 1.0118e-03, 9.9087e-04, 1.1854e-03, 2.7161e-03,
          3.4008e-03, 1.7004e-03, 1.6346e-03, 7.9536e-04, 6.3038e-04, 9.0265e-04,
          1.4935e-03, 1.1873e-03, 1.0433e-03, 1.0481e-03, 1.6375e-03, 1.9102e-03,
          8.9741e-04, 8.6689e-04, 4.8590e-04, 1.0586e-03, 1.4467e-03, 9.2030e-04,
          1.2102e-03, 4.4084e-04, 5.0735e-04, 2.1496e-03, 1.2112e-03, 1.3056e-03,
          1.4143e-03, 5.5428e-03, 2.0275e-03, 1.4954e-03, 1.0796e-03, 8.1015e-04,
          1.3390e-03, 2.4357e-03, 1.1015e-03, 2.1439e-03, 2.5463e-03, 1.3714e-03,
          1.1635e-03, 1.3685e-03, 8.9645e-04, 3.0780e-04, 1.2693e-03, 4.0674e-04,
          1.0519e-03, 8.7452e-04, 9.4461e-04, 1.4038e-03, 5.5599e-04, 1.5020e-03,
          1.3533e-03, 8.1968e-04, 4.5896e-04, 6.5994e-04, 1.3132e-03, 1.8368e-03,
          1.1988e-03, 9.4891e-04, 6.6519e-04, 1.7023e-03, 8.8787e-04, 9.8991e-04,
          1.0786e-03, 9.4748e-04, 7.3719e-04, 7.5293e-04, 9.1982e-04, 1.8396e-03,
          1.8120e-03, 1.7014e-03, 2.4719e-03, 2.3022e-03, 3.4599e-03, 7.5188e-03,
          1.2770e-03, 1.7281e-03, 4.3607e-04, 1.9875e-03, 3.6812e-03, 1.0881e-03,
          1.5898e-03, 9.6798e-04, 8.5783e-04, 1.0605e-03, 1.9913e-03, 7.7438e-04,
          1.0443e-03, 6.5517e-04, 9.8419e-04, 6.8188e-04, 7.2432e-04, 1.4639e-03,
          2.7294e-03, 3.5343e-03, 6.2752e-03, 1.5060e-02, 3.7384e-03, 1.0262e-03,
          1.1101e-03, 1.2236e-03, 1.8282e-03, 7.3767e-04, 6.4135e-04, 7.9823e-04,
          8.2350e-04, 9.5749e-04, 1.0900e-03, 1.9627e-03, 8.8549e-04, 1.0872e-03,
          5.3930e-04, 1.0357e-03, 8.5306e-04, 1.1215e-03, 7.1049e-04, 1.1816e-03,
          5.8708e-03, 3.7689e-03, 4.3907e-03, 2.4929e-03, 9.6178e-04, 1.3494e-03,
          3.4924e-03, 4.1924e-03, 7.6790e-03, 5.4312e-04, 1.7395e-03, 5.9271e-04,
          8.3876e-04, 2.5311e-03, 4.1809e-03, 6.7139e-04, 1.5182e-03, 1.1177e-03,
          1.2484e-03, 1.1139e-03, 8.4209e-04, 5.5552e-04, 1.3552e-03, 1.0633e-03,
          2.7637e-03, 3.1433e-02, 7.9041e-03, 1.2913e-03, 3.5896e-03, 1.1320e-03,
          1.8120e-03, 2.2717e-03, 6.1913e-03, 3.9101e-03, 3.1891e-03, 1.0080e-03,
          5.7716e-03, 1.0271e-03, 1.3161e-03, 1.4582e-03, 2.8973e-03, 6.6109e-03,
          6.0272e-03, 6.5575e-03, 3.0460e-03, 1.9360e-03, 2.3289e-03, 6.2637e-03,
          2.5681e-02, 3.5782e-03, 2.0828e-02, 8.1100e-03, 1.8753e-02, 5.3558e-03,
          3.4580e-03, 7.6866e-03, 1.2932e-02, 4.5166e-03, 1.0796e-03, 3.2158e-03,
          2.7065e-03, 9.2077e-04, 1.8702e-03, 1.7118e-03, 2.9602e-03, 5.3406e-03,
          1.9464e-03, 7.3719e-04, 6.7377e-04, 1.9455e-03, 1.9035e-03, 6.0310e-03,
          5.6877e-03, 2.2449e-03, 5.2910e-03, 2.1393e-02, 4.2572e-03, 5.0507e-03,
          9.1095e-03, 2.2907e-03, 5.0735e-03, 1.5011e-03, 4.9820e-03, 1.7681e-03,
          1.0929e-03, 3.7079e-03, 3.0556e-03, 4.6883e-03, 1.7767e-03, 1.5211e-03,
          2.8534e-03, 1.9817e-03, 1.0319e-03, 1.2016e-03, 9.0759e-02, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 2.5487e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          3.2806e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 2.5487e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 2.5487e-04, 2.5487e-04,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          5.3501e-04, 2.5487e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 2.5487e-04, 4.7922e-05,
          3.2806e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          3.2806e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05, 3.2806e-04,
          4.7922e-05, 3.2806e-04, 4.7922e-05, 4.7922e-05, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 3.2806e-04, 3.2806e-04, 1.1992e-04, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05, 4.7922e-05, 2.9540e-04, 4.7922e-05, 4.7922e-05,
          4.7922e-05, 4.7922e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6234e-03, 5.0629e-02, 1.5549e-02, 1.0513e-02, 4.9103e-02,
          3.6743e-02, 9.8114e-03, 6.0987e-04, 6.4659e-04, 1.1473e-03, 3.8452e-03,
          1.1368e-03, 5.9843e-04, 5.1260e-04, 9.4748e-04, 6.7377e-04, 8.3923e-04,
          5.0211e-04, 4.8089e-04, 4.2820e-04, 5.2452e-04, 5.6314e-04, 6.8617e-04,
          5.0125e-03, 2.5520e-03, 7.4577e-03, 1.8387e-03, 6.6805e-04, 7.9632e-04,
          6.0558e-04, 9.6655e-04, 3.6216e-04, 9.4891e-04, 5.7983e-04, 5.7459e-04,
          1.4601e-03, 1.8578e-03, 1.0052e-03, 7.0047e-04, 5.0354e-04, 2.3403e-03,
          1.8625e-03, 1.1921e-03, 1.4391e-03, 6.9618e-04, 7.6914e-04, 1.4229e-03,
          3.8791e-04, 5.5027e-04, 5.4398e-03, 1.0281e-03, 8.1444e-04, 5.6124e-04,
          5.1975e-04, 6.4373e-04, 8.1825e-04, 1.2312e-03, 3.0117e-03, 5.8479e-03,
          1.0162e-02, 8.6927e-04, 2.1172e-03, 1.5087e-03, 2.3346e-03, 2.7752e-03,
          2.5730e-03, 1.1559e-03, 7.9298e-04, 7.8440e-04, 8.4829e-04, 9.6703e-04,
          8.7070e-04, 1.9875e-03, 9.3126e-04, 1.1492e-03, 5.4789e-04, 3.8290e-04,
          8.5878e-04, 7.8392e-04, 1.1148e-03, 1.2407e-03, 1.7099e-03, 5.7373e-03,
          2.7256e-03, 2.3155e-03, 1.4133e-03, 1.6680e-03, 1.1187e-03, 1.2970e-03,
          3.4351e-03, 2.0390e-03, 1.0109e-03, 2.0618e-03, 1.1320e-03, 1.0538e-03,
          6.1035e-04, 5.9032e-04, 5.9366e-04, 1.0757e-03, 1.0900e-03, 9.1410e-04,
          2.6665e-03, 7.1001e-04, 8.0824e-04, 2.0409e-03, 1.5411e-03, 1.0910e-03,
          1.9531e-03, 3.7327e-03, 3.2921e-03, 1.7185e-03, 1.7948e-03, 1.2283e-03,
          1.7452e-03, 1.9283e-03, 7.6342e-04, 2.5463e-03, 1.4696e-03, 1.0443e-03,
          1.1091e-03, 1.5478e-03, 7.4291e-04, 4.0221e-04, 1.6737e-03, 4.8208e-04,
          6.7091e-04, 1.1806e-03, 1.3351e-03, 1.5640e-03, 7.4339e-04, 2.0657e-03,
          2.4872e-03, 1.3199e-03, 4.5896e-04, 9.1362e-04, 2.6112e-03, 1.5602e-03,
          1.3304e-03, 1.0271e-03, 8.6880e-04, 1.5402e-03, 9.5463e-04, 1.0424e-03,
          1.1177e-03, 1.4257e-03, 7.7009e-04, 8.7023e-04, 1.3027e-03, 2.0657e-03,
          2.1629e-03, 1.8854e-03, 1.8568e-03, 2.0828e-03, 2.9812e-03, 5.9319e-03,
          2.2373e-03, 2.0313e-03, 8.2636e-04, 2.5845e-03, 3.8433e-03, 1.2026e-03,
          1.8148e-03, 1.1587e-03, 1.0166e-03, 1.1473e-03, 1.5306e-03, 9.0027e-04,
          1.9665e-03, 7.9679e-04, 1.3809e-03, 7.7820e-04, 8.0156e-04, 1.6584e-03,
          3.3531e-03, 3.4523e-03, 3.2120e-03, 8.6136e-03, 4.3869e-03, 1.1444e-03,
          1.2770e-03, 1.5278e-03, 2.4433e-03, 1.1425e-03, 8.2684e-04, 9.4461e-04,
          1.6537e-03, 9.9564e-04, 1.0595e-03, 1.7099e-03, 1.2589e-03, 1.3170e-03,
          7.5340e-04, 8.2016e-04, 9.4652e-04, 1.3361e-03, 1.3657e-03, 1.2903e-03,
          3.8700e-03, 1.9855e-03, 5.2681e-03, 4.0359e-03, 1.6155e-03, 1.8482e-03,
          6.0272e-03, 6.4163e-03, 5.1689e-03, 7.0524e-04, 3.5820e-03, 6.8855e-04,
          9.9277e-04, 1.8644e-03, 2.4719e-03, 1.2178e-03, 1.9255e-03, 1.3733e-03,
          1.3437e-03, 1.2379e-03, 1.2016e-03, 7.2050e-04, 1.5326e-03, 1.1806e-03,
          2.7027e-03, 2.0218e-02, 9.7275e-03, 2.1133e-03, 5.0735e-03, 1.5106e-03,
          3.4370e-03, 3.8605e-03, 5.0697e-03, 5.4703e-03, 3.7498e-03, 1.2932e-03,
          3.0766e-03, 1.3733e-03, 1.8921e-03, 1.8272e-03, 3.0346e-03, 1.2398e-02,
          4.9934e-03, 3.8776e-03, 2.8992e-03, 1.8053e-03, 2.3441e-03, 4.0131e-03,
          1.4381e-02, 8.1711e-03, 2.4750e-02, 2.2446e-02, 1.6556e-02, 6.5994e-03,
          4.9248e-03, 3.8605e-03, 6.2637e-03, 3.9787e-03, 1.4086e-03, 2.7084e-03,
          2.2469e-03, 1.3742e-03, 2.2430e-03, 1.8606e-03, 2.3327e-03, 4.7379e-03,
          2.1267e-03, 7.7295e-04, 7.7248e-04, 2.4471e-03, 2.9240e-03, 7.3891e-03,
          1.2123e-02, 4.2496e-03, 7.3891e-03, 1.6922e-02, 4.4250e-03, 6.0921e-03,
          1.1490e-02, 3.2635e-03, 3.3607e-03, 1.9474e-03, 2.6646e-03, 2.7161e-03,
          2.0390e-03, 2.6302e-03, 1.9875e-03, 2.7447e-03, 2.0924e-03, 1.8959e-03,
          4.9057e-03, 2.4338e-03, 1.2970e-03, 1.4391e-03, 3.5217e-02, 6.1127e-02,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05, 1.1975e-04, 1.1975e-04, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 2.1291e-04, 3.7014e-05, 1.1975e-04, 3.7014e-05, 3.7014e-05,
          2.9564e-04, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 1.1975e-04,
          3.7014e-05, 1.1975e-04, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05, 1.1975e-04, 3.7014e-05, 3.7014e-05, 1.1975e-04,
          3.7014e-05, 3.7014e-05, 2.9564e-04, 3.7014e-05, 3.7014e-05, 2.1291e-04,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 1.1975e-04, 1.1975e-04,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 1.1975e-04, 3.7014e-05,
          2.9564e-04, 1.1975e-04, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          2.1291e-04, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 1.1975e-04, 1.1975e-04, 3.7014e-05,
          2.1291e-04, 3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7014e-05, 2.1291e-04,
          2.9564e-04, 3.7014e-05, 1.1975e-04, 3.7014e-05, 3.7014e-05, 4.4465e-04,
          2.1291e-04, 2.1291e-04, 3.7014e-05, 2.1291e-04, 3.7014e-05, 3.7014e-05,
          2.1291e-04, 2.1291e-04, 2.1291e-04, 1.0967e-04, 2.1291e-04, 2.1291e-04,
          3.7014e-05, 3.7014e-05, 3.7014e-05, 3.7384e-04, 3.7014e-05, 3.7014e-05,
          3.7014e-05, 3.7014e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.9858e-03, 4.5105e-02, 1.5671e-02, 1.4603e-02, 6.9336e-02,
          2.5757e-02, 1.4366e-02, 5.5265e-04, 3.7479e-04, 1.2026e-03, 3.8872e-03,
          9.7132e-04, 6.1131e-04, 6.3181e-04, 1.0052e-03, 5.6887e-04, 9.4223e-04,
          5.4741e-04, 3.6311e-04, 6.6423e-04, 7.9536e-04, 6.5994e-04, 7.3624e-04,
          4.6577e-03, 2.2488e-03, 1.0201e-02, 2.1572e-03, 8.6403e-04, 8.7976e-04,
          6.7425e-04, 8.3399e-04, 3.8457e-04, 1.0262e-03, 6.3992e-04, 4.3464e-04,
          1.8682e-03, 1.6470e-03, 1.5669e-03, 8.1062e-04, 6.7472e-04, 3.7842e-03,
          3.3073e-03, 2.0256e-03, 2.0428e-03, 1.2569e-03, 9.0933e-04, 1.9875e-03,
          4.5872e-04, 8.9931e-04, 6.9046e-03, 7.9393e-04, 6.0558e-04, 4.5466e-04,
          4.6372e-04, 4.7159e-04, 8.0061e-04, 1.6232e-03, 1.3294e-03, 2.7885e-03,
          2.4891e-03, 9.9468e-04, 1.4715e-03, 1.3304e-03, 2.9449e-03, 3.3970e-03,
          1.9283e-03, 1.0757e-03, 1.1196e-03, 1.4610e-03, 1.4992e-03, 1.2751e-03,
          1.3847e-03, 2.4681e-03, 8.1682e-04, 7.4577e-04, 4.3511e-04, 3.2043e-04,
          7.9298e-04, 1.5001e-03, 1.3723e-03, 1.4477e-03, 1.7424e-03, 1.7462e-03,
          1.7099e-03, 1.6479e-03, 1.1711e-03, 1.7300e-03, 9.5558e-04, 1.3409e-03,
          3.4809e-03, 2.5635e-03, 1.3371e-03, 2.1420e-03, 1.3227e-03, 1.5345e-03,
          6.5422e-04, 6.1512e-04, 5.1641e-04, 1.1091e-03, 1.9627e-03, 1.3657e-03,
          2.8782e-03, 8.8930e-04, 8.4877e-04, 2.8591e-03, 1.7099e-03, 1.2465e-03,
          1.4801e-03, 6.0081e-03, 4.0092e-03, 1.7567e-03, 1.5898e-03, 1.3113e-03,
          1.7300e-03, 3.9940e-03, 1.2159e-03, 3.1223e-03, 2.3880e-03, 1.3285e-03,
          1.2846e-03, 1.7281e-03, 1.0357e-03, 3.2902e-04, 1.8730e-03, 6.0463e-04,
          1.3504e-03, 1.0824e-03, 1.2436e-03, 1.6546e-03, 6.8235e-04, 1.6279e-03,
          1.7557e-03, 1.3552e-03, 4.8304e-04, 9.5034e-04, 2.5215e-03, 3.1967e-03,
          2.1763e-03, 1.2560e-03, 1.0738e-03, 1.7719e-03, 1.1005e-03, 1.1177e-03,
          1.3885e-03, 8.1730e-04, 1.1272e-03, 1.2083e-03, 1.4458e-03, 1.8415e-03,
          2.6150e-03, 2.1362e-03, 2.7599e-03, 2.0790e-03, 2.4681e-03, 1.1909e-02,
          2.4815e-03, 3.1128e-03, 7.5197e-04, 2.6798e-03, 5.0125e-03, 1.6575e-03,
          2.3651e-03, 1.8301e-03, 1.4696e-03, 1.3142e-03, 2.3422e-03, 1.0014e-03,
          2.1648e-03, 9.4652e-04, 1.2541e-03, 7.9250e-04, 1.0643e-03, 1.8415e-03,
          1.7767e-03, 1.9913e-03, 4.0550e-03, 1.4511e-02, 5.7144e-03, 1.3371e-03,
          1.5278e-03, 1.4420e-03, 2.8954e-03, 1.0643e-03, 8.5020e-04, 1.0805e-03,
          1.4009e-03, 1.2751e-03, 1.2150e-03, 2.3441e-03, 1.3371e-03, 1.4782e-03,
          7.6866e-04, 1.3857e-03, 1.0977e-03, 1.8435e-03, 1.2703e-03, 1.4629e-03,
          3.9177e-03, 2.9583e-03, 4.9820e-03, 3.8738e-03, 1.8349e-03, 1.9894e-03,
          4.1008e-03, 6.9695e-03, 1.0307e-02, 6.3992e-04, 3.4885e-03, 9.6941e-04,
          1.0109e-03, 2.7084e-03, 3.3112e-03, 1.1587e-03, 1.8663e-03, 1.5879e-03,
          1.4305e-03, 1.5821e-03, 1.5650e-03, 7.0047e-04, 1.6489e-03, 1.2398e-03,
          3.4504e-03, 1.8127e-02, 1.1726e-02, 1.8091e-03, 5.7716e-03, 1.5783e-03,
          2.9144e-03, 2.3460e-03, 5.5733e-03, 2.5311e-03, 2.7790e-03, 1.2503e-03,
          4.1313e-03, 1.2617e-03, 1.6890e-03, 1.0948e-03, 3.2196e-03, 4.6310e-03,
          3.6507e-03, 3.9787e-03, 2.3499e-03, 2.1648e-03, 2.8343e-03, 5.8289e-03,
          2.3621e-02, 4.7836e-03, 8.8654e-03, 3.7174e-03, 8.7738e-03, 6.1340e-03,
          4.0703e-03, 8.1940e-03, 1.0414e-02, 4.7913e-03, 1.6203e-03, 3.4161e-03,
          3.0079e-03, 1.1368e-03, 2.0142e-03, 1.6947e-03, 2.3575e-03, 3.6449e-03,
          1.8101e-03, 9.3985e-04, 8.6021e-04, 2.6512e-03, 2.9984e-03, 5.8556e-03,
          5.4092e-03, 3.4904e-03, 6.7253e-03, 1.8387e-02, 5.3406e-03, 5.5428e-03,
          7.0915e-03, 1.8826e-03, 7.0038e-03, 2.3079e-03, 4.2725e-03, 2.0828e-03,
          1.6737e-03, 3.2520e-03, 3.3016e-03, 3.5725e-03, 1.8425e-03, 1.5106e-03,
          1.8978e-03, 2.7351e-03, 1.5421e-03, 1.6985e-03, 2.4216e-02, 1.4206e-02,
          5.0598e-02, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 2.2054e-04, 2.0981e-04, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          2.0981e-04, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 2.0981e-04, 2.0981e-04, 4.5061e-05, 4.5061e-05,
          2.0981e-04, 4.5061e-05, 2.0981e-04, 4.5061e-05, 4.5061e-05, 3.8528e-04,
          4.5061e-05, 4.5061e-05, 3.8528e-04, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 3.8528e-04,
          4.5061e-05, 4.5061e-05, 3.8528e-04, 4.5061e-05, 4.5061e-05, 2.0981e-04,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 2.2054e-04, 2.2054e-04,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 2.2054e-04, 4.5061e-05,
          3.8528e-04, 3.8528e-04, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 2.0981e-04, 4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.5061e-05, 2.2054e-04, 4.5061e-05,
          2.0981e-04, 2.2054e-04, 4.5061e-05, 4.5061e-05, 4.5061e-05, 2.0981e-04,
          3.8528e-04, 4.5061e-05, 3.8528e-04, 4.5061e-05, 4.5061e-05, 2.0981e-04,
          2.0981e-04, 2.0981e-04, 4.5061e-05, 2.0981e-04, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 4.8733e-04, 2.0981e-04, 4.5061e-05,
          4.5061e-05, 4.5061e-05, 4.5061e-05, 3.9482e-04, 4.5061e-05, 4.5061e-05,
          4.5061e-05, 4.5061e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5242e-03, 4.3121e-02, 1.4191e-02, 1.2451e-02, 6.1401e-02,
          2.4811e-02, 1.2085e-02, 5.7459e-04, 4.2009e-04, 1.2207e-03, 3.9482e-03,
          1.1034e-03, 5.3501e-04, 6.6519e-04, 8.4686e-04, 5.9795e-04, 7.2432e-04,
          4.6968e-04, 4.5466e-04, 5.5885e-04, 6.1178e-04, 5.1355e-04, 6.2132e-04,
          5.1956e-03, 2.0390e-03, 8.4534e-03, 1.8129e-03, 7.1096e-04, 6.7520e-04,
          7.4577e-04, 8.7547e-04, 3.3736e-04, 8.1873e-04, 5.9319e-04, 3.9840e-04,
          1.2941e-03, 1.3237e-03, 1.1148e-03, 7.0000e-04, 5.5218e-04, 2.5597e-03,
          2.7905e-03, 1.8139e-03, 2.1381e-03, 1.0166e-03, 7.3671e-04, 1.4696e-03,
          4.5943e-04, 7.3576e-04, 6.5994e-03, 6.3848e-04, 5.9986e-04, 4.2629e-04,
          4.6110e-04, 4.7064e-04, 6.9666e-04, 1.1768e-03, 1.5640e-03, 2.4986e-03,
          2.1839e-03, 8.2302e-04, 1.3409e-03, 9.6178e-04, 2.7905e-03, 2.6932e-03,
          2.9087e-03, 1.4858e-03, 1.1129e-03, 1.3113e-03, 1.4534e-03, 1.4229e-03,
          1.2159e-03, 2.2774e-03, 6.5660e-04, 7.4673e-04, 5.0116e-04, 3.2043e-04,
          7.7343e-04, 1.1311e-03, 1.9913e-03, 1.3962e-03, 3.1910e-03, 1.5574e-03,
          1.5583e-03, 1.6241e-03, 1.1415e-03, 1.9236e-03, 1.3075e-03, 2.3327e-03,
          4.0321e-03, 2.3518e-03, 1.3571e-03, 2.4700e-03, 1.1787e-03, 1.3313e-03,
          5.5933e-04, 4.7588e-04, 4.7159e-04, 8.7070e-04, 1.5621e-03, 1.1301e-03,
          2.9335e-03, 9.9564e-04, 1.2360e-03, 2.2488e-03, 1.3256e-03, 1.0386e-03,
          1.4448e-03, 4.5929e-03, 3.6602e-03, 2.1095e-03, 2.5101e-03, 1.9855e-03,
          1.5726e-03, 2.6722e-03, 8.9502e-04, 2.4624e-03, 1.9817e-03, 1.2054e-03,
          1.1864e-03, 1.3752e-03, 1.0710e-03, 3.5238e-04, 1.6394e-03, 8.6594e-04,
          1.0433e-03, 1.7653e-03, 2.1057e-03, 1.3046e-03, 6.2609e-04, 1.5678e-03,
          2.2526e-03, 1.9989e-03, 5.7745e-04, 1.3247e-03, 2.9640e-03, 2.9793e-03,
          1.5087e-03, 1.2083e-03, 9.3412e-04, 1.6441e-03, 9.9373e-04, 1.1501e-03,
          1.5793e-03, 1.0729e-03, 1.2045e-03, 1.1740e-03, 2.2736e-03, 1.5917e-03,
          2.5158e-03, 1.6756e-03, 2.4891e-03, 1.5621e-03, 2.2774e-03, 7.5417e-03,
          2.4395e-03, 2.5654e-03, 9.6273e-04, 3.4389e-03, 3.6335e-03, 1.3342e-03,
          1.8501e-03, 1.4324e-03, 1.1835e-03, 1.0529e-03, 1.5221e-03, 1.0061e-03,
          2.4357e-03, 1.0347e-03, 1.1272e-03, 6.6614e-04, 8.9884e-04, 1.6966e-03,
          1.7290e-03, 2.2278e-03, 3.3798e-03, 1.0925e-02, 4.6234e-03, 1.2884e-03,
          1.4877e-03, 1.3161e-03, 2.5654e-03, 1.1549e-03, 8.1205e-04, 8.5926e-04,
          1.6880e-03, 1.2188e-03, 1.1301e-03, 1.9836e-03, 2.5139e-03, 1.3380e-03,
          8.1778e-04, 1.2197e-03, 8.9836e-04, 1.5574e-03, 1.5478e-03, 1.2598e-03,
          3.4924e-03, 2.4891e-03, 4.2686e-03, 4.8065e-03, 2.3937e-03, 1.8883e-03,
          6.6681e-03, 8.1558e-03, 8.7128e-03, 7.1335e-04, 3.9711e-03, 8.3971e-04,
          1.0328e-03, 1.9207e-03, 3.0155e-03, 1.6050e-03, 1.6441e-03, 2.1496e-03,
          1.1673e-03, 1.3065e-03, 1.6012e-03, 6.5374e-04, 1.4009e-03, 1.2207e-03,
          2.8038e-03, 1.7349e-02, 1.1131e-02, 2.4319e-03, 5.4245e-03, 1.4915e-03,
          4.6425e-03, 3.6354e-03, 6.5842e-03, 3.0346e-03, 2.5101e-03, 1.3485e-03,
          3.8166e-03, 1.4238e-03, 2.3975e-03, 1.1683e-03, 2.7161e-03, 5.1346e-03,
          4.3144e-03, 4.1924e-03, 2.0123e-03, 1.5459e-03, 2.5635e-03, 5.7564e-03,
          1.7746e-02, 7.5531e-03, 1.1017e-02, 4.9438e-03, 1.1848e-02, 6.0387e-03,
          6.2065e-03, 7.4310e-03, 9.8648e-03, 3.7804e-03, 1.4429e-03, 3.0670e-03,
          2.3556e-03, 1.5268e-03, 1.8539e-03, 1.6222e-03, 2.2945e-03, 4.2572e-03,
          1.7033e-03, 8.4686e-04, 7.7963e-04, 2.7161e-03, 3.5381e-03, 5.2834e-03,
          9.3994e-03, 7.0686e-03, 7.1182e-03, 1.9424e-02, 4.1428e-03, 5.4779e-03,
          8.2397e-03, 2.2106e-03, 5.6229e-03, 2.5597e-03, 4.1199e-03, 3.1509e-03,
          2.4490e-03, 3.6602e-03, 3.3875e-03, 3.4790e-03, 1.7338e-03, 1.4944e-03,
          2.2678e-03, 2.4567e-03, 1.4210e-03, 1.7271e-03, 2.3499e-02, 1.6296e-02,
          3.9062e-02, 3.6987e-02, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 2.0218e-04, 2.3341e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          2.3341e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 2.3341e-04, 3.9363e-04, 4.2021e-05, 4.2021e-05,
          3.9363e-04, 4.2021e-05, 2.3341e-04, 4.2021e-05, 4.2021e-05, 2.0218e-04,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 2.0218e-04, 4.2021e-05, 4.2021e-05, 2.0218e-04,
          4.2021e-05, 4.2021e-05, 3.9363e-04, 4.2021e-05, 4.2021e-05, 2.3341e-04,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.0218e-04, 2.0218e-04,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.0218e-04, 4.2021e-05,
          3.9363e-04, 2.0218e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.0218e-04, 4.2021e-05,
          2.3341e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05, 2.3341e-04,
          2.0218e-04, 4.2021e-05, 2.0218e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 2.3341e-04, 4.2021e-05, 4.2021e-05, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 3.9625e-04, 2.3341e-04, 4.2021e-05,
          4.2021e-05, 4.2021e-05, 4.2021e-05, 3.9625e-04, 4.2021e-05, 4.2021e-05,
          4.2021e-05, 4.2021e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1079e-03, 4.1473e-02, 1.5915e-02, 1.2321e-02, 6.5552e-02,
          2.1057e-02, 1.5053e-02, 3.9220e-04, 4.7183e-04, 1.0366e-03, 2.4776e-03,
          8.5831e-04, 5.0116e-04, 5.5933e-04, 9.4223e-04, 4.4465e-04, 6.6662e-04,
          4.0960e-04, 3.5787e-04, 9.2316e-04, 7.3910e-04, 5.9319e-04, 5.6934e-04,
          3.2806e-03, 2.0218e-03, 9.9258e-03, 1.8520e-03, 7.3433e-04, 6.1512e-04,
          5.7650e-04, 6.1703e-04, 3.0327e-04, 9.6989e-04, 6.3848e-04, 3.9363e-04,
          1.5383e-03, 1.5545e-03, 1.2846e-03, 9.2459e-04, 6.0272e-04, 2.8858e-03,
          2.8267e-03, 1.8692e-03, 2.1610e-03, 1.0395e-03, 6.5899e-04, 1.7157e-03,
          4.1842e-04, 9.1124e-04, 7.1716e-03, 6.2180e-04, 4.8089e-04, 4.0221e-04,
          3.9220e-04, 4.4727e-04, 8.1062e-04, 1.3638e-03, 1.4915e-03, 2.8343e-03,
          3.0212e-03, 7.9441e-04, 1.3218e-03, 9.8801e-04, 2.6665e-03, 2.9564e-03,
          2.1267e-03, 1.0176e-03, 9.7847e-04, 1.4973e-03, 1.8044e-03, 1.4057e-03,
          1.3409e-03, 2.0790e-03, 6.5327e-04, 6.3753e-04, 4.7064e-04, 3.1567e-04,
          7.5531e-04, 1.2817e-03, 1.5583e-03, 1.4114e-03, 1.8587e-03, 1.8950e-03,
          1.8454e-03, 1.4982e-03, 1.1425e-03, 1.3475e-03, 9.8896e-04, 1.7138e-03,
          2.4948e-03, 2.1915e-03, 1.4772e-03, 1.7176e-03, 1.3790e-03, 1.5173e-03,
          5.9175e-04, 5.2261e-04, 4.6062e-04, 9.2554e-04, 1.5240e-03, 1.2112e-03,
          1.9341e-03, 8.2636e-04, 8.0729e-04, 2.7409e-03, 1.5621e-03, 1.0729e-03,
          1.3704e-03, 5.4131e-03, 3.3646e-03, 2.1229e-03, 1.9236e-03, 1.4706e-03,
          1.7786e-03, 3.5458e-03, 1.0452e-03, 2.7466e-03, 2.1629e-03, 1.0462e-03,
          1.0118e-03, 1.5774e-03, 9.4080e-04, 2.7680e-04, 1.7653e-03, 7.2765e-04,
          1.3294e-03, 1.3857e-03, 1.5697e-03, 1.4935e-03, 5.9557e-04, 1.4915e-03,
          2.0180e-03, 1.3332e-03, 4.4441e-04, 1.1110e-03, 2.3861e-03, 3.1090e-03,
          1.6642e-03, 1.1797e-03, 9.1982e-04, 1.6747e-03, 1.0366e-03, 1.0233e-03,
          1.3380e-03, 7.9107e-04, 1.1806e-03, 1.1492e-03, 1.6470e-03, 1.7424e-03,
          2.9106e-03, 1.8835e-03, 2.3689e-03, 1.6003e-03, 2.2812e-03, 8.4152e-03,
          2.3975e-03, 3.0479e-03, 7.6151e-04, 3.6430e-03, 4.3297e-03, 1.3885e-03,
          2.3117e-03, 1.8339e-03, 1.1625e-03, 1.1787e-03, 1.7424e-03, 1.0433e-03,
          1.7958e-03, 9.9277e-04, 1.2398e-03, 7.2384e-04, 8.5115e-04, 2.2736e-03,
          1.9360e-03, 2.2678e-03, 4.0474e-03, 1.1253e-02, 4.6730e-03, 1.1415e-03,
          1.5097e-03, 1.3151e-03, 2.5406e-03, 9.9373e-04, 7.5769e-04, 9.1171e-04,
          1.2922e-03, 1.3075e-03, 1.2369e-03, 2.0027e-03, 1.6689e-03, 1.4305e-03,
          6.2990e-04, 1.2617e-03, 1.1253e-03, 1.5421e-03, 1.2646e-03, 1.5154e-03,
          3.9406e-03, 2.7618e-03, 4.5967e-03, 3.5305e-03, 1.8072e-03, 2.0618e-03,
          5.9052e-03, 6.6376e-03, 9.0179e-03, 5.8556e-04, 2.9545e-03, 8.3399e-04,
          8.8263e-04, 2.0599e-03, 3.0174e-03, 1.0481e-03, 1.7347e-03, 1.8311e-03,
          1.4782e-03, 1.3103e-03, 1.4029e-03, 6.9380e-04, 1.6518e-03, 1.2140e-03,
          3.0594e-03, 1.8127e-02, 1.0712e-02, 1.9445e-03, 5.5542e-03, 1.4915e-03,
          3.5400e-03, 3.6888e-03, 6.9885e-03, 3.3970e-03, 2.5215e-03, 1.3762e-03,
          4.3983e-03, 1.3371e-03, 2.4033e-03, 1.3847e-03, 3.0556e-03, 5.4398e-03,
          4.4899e-03, 4.6692e-03, 2.6703e-03, 1.9045e-03, 2.5711e-03, 5.5466e-03,
          1.7334e-02, 5.3253e-03, 1.2093e-02, 5.4665e-03, 1.1993e-02, 5.9204e-03,
          6.1378e-03, 7.3738e-03, 9.8877e-03, 4.6425e-03, 1.4534e-03, 3.4599e-03,
          2.3651e-03, 1.2894e-03, 2.1801e-03, 1.9035e-03, 2.2316e-03, 4.5815e-03,
          1.8377e-03, 9.5463e-04, 8.2874e-04, 2.5826e-03, 2.9411e-03, 5.5008e-03,
          8.8043e-03, 4.3068e-03, 6.8130e-03, 2.0386e-02, 4.9362e-03, 5.4970e-03,
          9.1171e-03, 2.3556e-03, 6.0768e-03, 2.3193e-03, 3.9253e-03, 2.4204e-03,
          1.7738e-03, 3.0479e-03, 2.8801e-03, 3.4447e-03, 1.7223e-03, 1.6556e-03,
          2.5215e-03, 2.6722e-03, 1.4019e-03, 1.6956e-03, 2.1072e-02, 1.4809e-02,
          3.1708e-02, 2.5085e-02, 2.7664e-02, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 2.3615e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          2.8753e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 2.3615e-04,
          3.9756e-05, 3.9756e-05, 4.8375e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 2.3615e-04, 2.3615e-04,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 2.3615e-04, 3.9756e-05,
          4.8375e-04, 4.8375e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 2.8753e-04, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 2.3615e-04, 3.9756e-05,
          2.8753e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          2.8753e-04, 3.9756e-05, 2.3615e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 2.8753e-04, 3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05, 3.9756e-05, 3.9649e-04, 2.8753e-04, 3.9756e-05,
          3.9756e-05, 2.8753e-04, 3.9756e-05, 3.2449e-04, 3.9756e-05, 3.9756e-05,
          3.9756e-05, 3.9756e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.6469e-03, 5.2521e-02, 1.2169e-02, 1.3977e-02, 3.1464e-02,
          3.7994e-02, 9.7427e-03, 6.5088e-04, 4.5824e-04, 1.9417e-03, 2.6913e-03,
          9.9564e-04, 1.8730e-03, 4.7541e-04, 3.9940e-03, 8.6260e-04, 2.9297e-03,
          1.4629e-03, 5.4264e-04, 2.5058e-04, 3.7837e-04, 3.3402e-04, 2.1114e-03,
          3.6163e-03, 2.4796e-03, 7.8087e-03, 4.5052e-03, 3.7050e-04, 3.2825e-03,
          7.4911e-04, 1.6499e-03, 6.7377e-04, 6.2370e-03, 2.6875e-03, 1.2207e-03,
          3.0766e-03, 4.2367e-04, 3.7422e-03, 3.9673e-03, 2.9874e-04, 2.7618e-03,
          7.2050e-04, 5.5265e-04, 5.4598e-04, 4.8661e-04, 4.1771e-03, 8.8806e-03,
          3.6693e-04, 4.1318e-04, 2.2717e-03, 1.0099e-03, 8.7166e-04, 1.0328e-03,
          2.0969e-04, 5.0545e-04, 3.1447e-04, 5.5552e-04, 6.7616e-04, 6.3801e-04,
          6.7139e-04, 9.6369e-04, 9.5844e-04, 1.6470e-03, 9.3794e-04, 1.3649e-02,
          6.7949e-04, 5.8889e-04, 1.4486e-03, 3.6764e-04, 3.1996e-04, 3.8266e-04,
          5.0020e-04, 5.7335e-03, 3.4642e-04, 1.1520e-03, 2.0254e-04, 2.1327e-04,
          5.1403e-04, 4.3535e-04, 4.2009e-04, 2.0504e-03, 4.2439e-04, 4.4823e-04,
          4.5085e-04, 2.7485e-03, 5.9271e-04, 1.9484e-03, 4.4322e-04, 3.9387e-04,
          4.0932e-03, 3.1643e-03, 3.7122e-04, 2.2602e-03, 5.5695e-04, 6.0701e-04,
          4.4560e-04, 2.4056e-04, 1.7509e-03, 2.6817e-03, 4.2224e-04, 7.4768e-04,
          2.8267e-03, 3.4189e-04, 4.1294e-04, 1.6899e-03, 7.4501e-03, 1.0195e-03,
          2.4014e-03, 7.8201e-04, 4.5853e-03, 4.3416e-04, 5.7459e-04, 8.9455e-04,
          1.9464e-03, 1.6909e-03, 7.3338e-04, 1.7120e-02, 8.2970e-04, 6.1560e-04,
          1.0424e-03, 3.2806e-03, 2.4772e-04, 4.9067e-04, 8.7128e-03, 2.0123e-04,
          3.3665e-04, 2.9087e-04, 4.2558e-04, 9.8419e-03, 2.6836e-03, 2.7847e-03,
          9.7942e-04, 6.0987e-04, 9.3269e-04, 4.8065e-04, 2.3537e-03, 4.6301e-04,
          1.4114e-02, 1.3676e-03, 2.1076e-03, 3.4657e-03, 5.1575e-03, 1.5240e-03,
          3.5596e-04, 5.9843e-04, 6.5947e-04, 6.0844e-04, 4.5943e-04, 5.2147e-03,
          4.2610e-03, 5.5237e-03, 8.8835e-04, 2.5749e-03, 1.4849e-03, 3.4504e-03,
          3.0117e-03, 1.9951e-03, 9.4080e-04, 5.0688e-04, 8.9111e-03, 3.5362e-03,
          1.3550e-02, 5.7068e-03, 7.7171e-03, 5.8632e-03, 1.1616e-03, 1.0338e-03,
          1.8930e-03, 1.1530e-03, 3.2578e-03, 3.6850e-03, 2.8572e-03, 8.1863e-03,
          1.6146e-03, 7.8440e-04, 1.2779e-03, 1.7834e-03, 6.6109e-03, 3.7918e-03,
          3.0632e-03, 5.6190e-03, 1.0223e-02, 3.1605e-03, 1.3018e-03, 3.5362e-03,
          2.6493e-03, 1.1969e-03, 1.6518e-03, 6.4182e-04, 4.5538e-04, 7.5302e-03,
          2.1725e-03, 2.9778e-04, 6.8245e-03, 7.9727e-03, 2.4605e-03, 5.8670e-03,
          1.1072e-03, 1.2951e-03, 1.0277e-02, 1.6489e-03, 1.6642e-03, 4.4136e-03,
          8.4209e-04, 3.5534e-03, 1.3571e-03, 9.8419e-04, 5.1994e-03, 3.4370e-03,
          1.1663e-03, 1.6718e-03, 6.3705e-04, 7.3147e-04, 7.7286e-03, 4.4680e-04,
          3.2921e-03, 3.8319e-03, 3.7842e-03, 2.6722e-03, 1.1963e-02, 2.3079e-03,
          9.5978e-03, 1.1797e-03, 4.1084e-03, 1.2941e-03, 2.1393e-02, 7.8659e-03,
          1.1196e-03, 6.3610e-04, 9.5749e-04, 1.7281e-03, 5.5847e-03, 1.7300e-03,
          1.1396e-03, 7.0620e-04, 4.4370e-04, 2.3041e-03, 2.4643e-03, 3.3379e-03,
          9.3603e-04, 9.2983e-04, 8.3637e-04, 3.3875e-03, 5.9738e-03, 2.4738e-03,
          2.7905e-03, 2.3670e-03, 8.0347e-04, 1.1854e-03, 1.4763e-03, 8.6212e-03,
          7.1526e-04, 1.0881e-03, 1.9684e-03, 3.1052e-03, 1.0918e-02, 9.6893e-03,
          1.4219e-03, 8.3399e-04, 1.3266e-03, 2.2564e-03, 5.5075e-04, 1.1292e-03,
          3.7651e-03, 2.5177e-03, 2.2945e-03, 8.9417e-03, 4.8981e-03, 6.2065e-03,
          1.5774e-03, 1.2293e-03, 1.3649e-02, 2.6245e-03, 1.3512e-02, 9.7122e-03,
          4.3068e-03, 2.5730e-03, 1.7557e-03, 2.7180e-03, 1.0242e-03, 8.1015e-04,
          7.8964e-04, 5.8937e-04, 4.7898e-04, 7.2527e-04, 5.0163e-03, 4.4708e-03,
          1.9407e-03, 1.3542e-02, 7.8430e-03, 5.0049e-03, 2.2163e-03, 3.0289e-03,
          3.5858e-03, 1.5144e-03, 1.3571e-03, 3.0716e-02, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 1.6069e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 1.6975e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05, 3.0422e-04,
          2.6226e-05, 2.6226e-05, 3.0422e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 1.6069e-04, 2.6226e-05, 1.6975e-04, 1.6069e-04,
          2.6226e-05, 2.6226e-05, 1.6975e-04, 2.6226e-05, 4.7135e-04, 2.6226e-05,
          1.6975e-04, 3.0422e-04, 2.6226e-05, 2.6226e-05, 3.0422e-04, 1.6069e-04,
          4.7135e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05, 3.0422e-04, 2.6226e-05,
          3.0422e-04, 3.0422e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          2.6226e-05, 3.0422e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          1.6975e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05, 1.6069e-04, 2.6226e-05,
          3.0422e-04, 3.0422e-04, 1.6975e-04, 2.6226e-05, 2.6226e-05, 1.6975e-04,
          3.0422e-04, 2.6226e-05, 3.0422e-04, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          1.6975e-04, 2.2640e-03, 2.6226e-05, 2.6226e-05, 2.6226e-05, 2.6226e-05,
          1.6975e-04, 2.6226e-05, 4.0507e-04, 2.6226e-05, 1.6975e-04, 2.6226e-05,
          2.6226e-05, 2.6226e-05, 1.6975e-04, 3.3998e-04, 2.6226e-05, 1.6975e-04,
          1.6975e-04, 2.6226e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9597e-03, 5.2307e-02, 1.3458e-02, 1.2749e-02, 3.2867e-02,
          3.2471e-02, 9.1019e-03, 5.0783e-04, 4.4298e-04, 1.9226e-03, 3.1776e-03,
          1.1683e-03, 9.2030e-04, 3.7813e-04, 1.3409e-03, 5.7602e-04, 1.3962e-03,
          7.2813e-04, 4.0030e-04, 2.6155e-04, 4.5061e-04, 3.6168e-04, 1.0376e-03,
          4.3297e-03, 2.8210e-03, 7.3204e-03, 3.5667e-03, 3.5620e-04, 1.4248e-03,
          6.4039e-04, 1.1358e-03, 4.1580e-04, 2.4185e-03, 1.3380e-03, 6.8617e-04,
          2.9488e-03, 7.4577e-04, 4.4823e-03, 2.1152e-03, 3.1567e-04, 5.4321e-03,
          1.5383e-03, 1.2531e-03, 1.0977e-03, 1.2541e-03, 1.5240e-03, 4.4975e-03,
          3.5214e-04, 4.2057e-04, 2.5883e-03, 1.1320e-03, 8.6069e-04, 6.6519e-04,
          2.7108e-04, 5.3692e-04, 4.2224e-04, 1.2722e-03, 1.0366e-03, 1.5430e-03,
          1.1663e-03, 7.9012e-04, 1.1082e-03, 2.0332e-03, 2.5139e-03, 7.7286e-03,
          1.8759e-03, 1.4038e-03, 1.9741e-03, 1.0185e-03, 8.5306e-04, 8.8120e-04,
          7.9727e-04, 5.1613e-03, 5.6696e-04, 1.1320e-03, 2.6536e-04, 3.1304e-04,
          6.3086e-04, 1.5402e-03, 1.3161e-03, 2.8095e-03, 1.2541e-03, 8.6975e-04,
          7.6294e-04, 2.6875e-03, 7.4387e-04, 2.2659e-03, 1.1501e-03, 1.0328e-03,
          5.6648e-03, 5.7907e-03, 6.6090e-04, 3.5667e-03, 1.2159e-03, 1.0576e-03,
          4.4680e-04, 3.2663e-04, 1.1339e-03, 3.1967e-03, 2.0294e-03, 2.5806e-03,
          3.6926e-03, 6.6042e-04, 9.2173e-04, 3.8490e-03, 3.2883e-03, 8.4877e-04,
          2.3079e-03, 2.7771e-03, 7.8964e-03, 8.0633e-04, 1.0462e-03, 2.2888e-03,
          2.2736e-03, 4.2496e-03, 6.3896e-04, 7.0381e-03, 1.6441e-03, 5.6648e-04,
          8.4591e-04, 6.0501e-03, 5.6648e-04, 3.5143e-04, 3.4924e-03, 6.1989e-04,
          1.3065e-03, 8.8787e-04, 1.1406e-03, 3.9597e-03, 1.1234e-03, 2.5921e-03,
          1.5345e-03, 1.2751e-03, 4.1342e-04, 9.0551e-04, 4.5547e-03, 1.3208e-03,
          4.9095e-03, 1.0023e-03, 9.9182e-04, 1.7748e-03, 2.3365e-03, 9.7227e-04,
          9.4461e-04, 5.2404e-04, 1.2255e-03, 2.6588e-03, 1.0843e-03, 4.2267e-03,
          8.3923e-03, 5.1994e-03, 2.7676e-03, 3.8300e-03, 2.0103e-03, 1.0399e-02,
          7.3204e-03, 7.1487e-03, 7.1764e-04, 1.6632e-03, 1.0292e-02, 1.9646e-03,
          5.7640e-03, 7.8659e-03, 2.5253e-03, 2.1305e-03, 3.0499e-03, 3.4389e-03,
          2.8305e-03, 3.2005e-03, 2.7294e-03, 1.3094e-03, 1.3094e-03, 6.1073e-03,
          2.5520e-03, 1.2636e-03, 2.5024e-03, 5.3864e-03, 1.0239e-02, 1.6193e-03,
          2.8286e-03, 2.1038e-03, 5.7259e-03, 1.3866e-03, 1.0109e-03, 1.1692e-03,
          2.3518e-03, 9.1743e-04, 9.6655e-04, 1.6661e-03, 8.7357e-04, 2.5291e-03,
          9.3079e-04, 1.0290e-03, 3.3340e-03, 2.6455e-03, 3.7231e-03, 3.9520e-03,
          2.4319e-03, 1.2321e-03, 8.7204e-03, 4.1924e-03, 3.2959e-03, 4.5815e-03,
          1.8139e-03, 8.8959e-03, 3.7327e-03, 7.3338e-04, 6.8398e-03, 1.2980e-03,
          9.0837e-04, 2.8992e-03, 1.5945e-03, 1.2522e-03, 3.9291e-03, 9.7227e-04,
          5.2109e-03, 1.9255e-03, 2.4605e-03, 7.8201e-04, 4.6692e-03, 1.1482e-03,
          8.6517e-03, 2.6321e-03, 8.9340e-03, 2.7218e-03, 9.3002e-03, 3.0785e-03,
          3.1815e-03, 1.1587e-03, 2.5997e-03, 3.1013e-03, 5.1956e-03, 1.1625e-03,
          2.0008e-03, 1.7490e-03, 6.7186e-04, 1.4153e-03, 5.7411e-03, 5.2147e-03,
          1.6832e-03, 1.5211e-03, 2.0771e-03, 3.7613e-03, 3.2330e-03, 2.0065e-03,
          7.8964e-03, 4.0855e-03, 1.4915e-03, 2.0790e-03, 2.3079e-03, 1.0315e-02,
          2.0008e-03, 3.2387e-03, 4.7798e-03, 6.4201e-03, 2.7847e-03, 5.3902e-03,
          3.5858e-03, 1.7042e-03, 2.1725e-03, 2.2850e-03, 1.1358e-03, 2.2545e-03,
          2.8114e-03, 1.3790e-03, 7.3957e-04, 2.9945e-03, 3.3092e-03, 7.7362e-03,
          2.8858e-03, 2.6112e-03, 9.2239e-03, 5.6648e-03, 1.7609e-02, 1.2733e-02,
          1.0559e-02, 3.9749e-03, 2.1095e-03, 2.2278e-03, 1.6537e-03, 1.3876e-03,
          1.3704e-03, 9.9087e-04, 1.2388e-03, 1.2999e-03, 3.1452e-03, 2.8820e-03,
          2.6569e-03, 7.0457e-03, 2.2449e-03, 2.2602e-03, 3.6106e-03, 5.6725e-03,
          7.7591e-03, 3.3817e-03, 3.0689e-03, 6.8474e-03, 3.4363e-02, 2.8789e-05,
          2.8789e-05, 2.8789e-05, 2.9182e-04, 1.4532e-04, 2.8789e-05, 2.8789e-05,
          2.8789e-05, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          1.4532e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          2.8789e-05, 2.8789e-05, 1.4532e-04, 1.4532e-04, 2.8789e-05, 2.8789e-05,
          1.4532e-04, 2.8789e-05, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.9182e-04,
          2.8789e-05, 1.4532e-04, 2.9182e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          2.8789e-05, 2.8789e-05, 2.8789e-05, 1.4532e-04, 2.8789e-05, 2.9182e-04,
          2.8789e-05, 2.8789e-05, 2.9182e-04, 2.8789e-05, 2.8789e-05, 1.4532e-04,
          1.4532e-04, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.9182e-04, 2.9182e-04,
          2.8789e-05, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.9182e-04, 2.8789e-05,
          2.9182e-04, 2.9182e-04, 2.8789e-05, 2.8789e-05, 1.4532e-04, 2.8789e-05,
          2.8789e-05, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          1.4532e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.9182e-04, 2.8789e-05,
          1.4532e-04, 2.9182e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 1.4532e-04,
          1.4532e-04, 1.4532e-04, 2.9182e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          1.4532e-04, 1.4532e-04, 2.8789e-05, 2.8789e-05, 2.8789e-05, 2.8789e-05,
          1.4532e-04, 2.8789e-05, 2.8789e-05, 1.0824e-04, 1.4532e-04, 2.8789e-05,
          2.8789e-05, 1.4532e-04, 2.8789e-05, 1.1578e-03, 4.2820e-04, 9.8228e-04,
          1.4532e-04, 2.8789e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.2229e-03, 4.4830e-02, 1.7365e-02, 1.5213e-02, 6.9031e-02,
          2.9678e-02, 1.7410e-02, 6.0081e-04, 3.7193e-04, 8.6641e-04, 2.4929e-03,
          7.7772e-04, 5.3406e-04, 5.9986e-04, 7.7963e-04, 6.5136e-04, 7.5531e-04,
          5.2547e-04, 3.7432e-04, 6.8521e-04, 5.9748e-04, 6.9904e-04, 6.1512e-04,
          3.3169e-03, 2.2907e-03, 1.1986e-02, 1.7366e-03, 6.2084e-04, 7.0047e-04,
          5.9986e-04, 9.7656e-04, 3.5763e-04, 7.3338e-04, 4.9543e-04, 4.3821e-04,
          9.2649e-04, 1.3018e-03, 9.0551e-04, 6.0940e-04, 5.1498e-04, 1.8997e-03,
          2.2507e-03, 1.4534e-03, 2.1133e-03, 1.0052e-03, 8.5115e-04, 1.4124e-03,
          5.5504e-04, 1.1539e-03, 6.5269e-03, 7.4244e-04, 6.4993e-04, 3.9196e-04,
          5.2118e-04, 5.3978e-04, 1.0080e-03, 8.0538e-04, 1.5326e-03, 1.7338e-03,
          2.7313e-03, 7.8821e-04, 1.4963e-03, 8.9216e-04, 1.6098e-03, 2.2354e-03,
          1.5268e-03, 7.3242e-04, 7.1144e-04, 1.5135e-03, 1.6861e-03, 1.5812e-03,
          1.6994e-03, 1.6766e-03, 8.6260e-04, 8.7833e-04, 5.2118e-04, 3.0661e-04,
          8.6117e-04, 7.2861e-04, 7.6580e-04, 7.8583e-04, 1.0815e-03, 1.3580e-03,
          1.7309e-03, 1.7366e-03, 1.1196e-03, 1.1311e-03, 5.8174e-04, 1.0319e-03,
          2.4319e-03, 1.4496e-03, 1.2064e-03, 1.6451e-03, 3.3016e-03, 3.7365e-03,
          7.0620e-04, 5.8985e-04, 5.2881e-04, 8.2827e-04, 9.4843e-04, 7.0333e-04,
          1.6527e-03, 3.7313e-04, 5.3930e-04, 1.1768e-03, 1.0862e-03, 9.5129e-04,
          1.5211e-03, 2.5673e-03, 2.0618e-03, 1.2589e-03, 1.3361e-03, 9.3937e-04,
          1.4248e-03, 2.9373e-03, 1.3561e-03, 2.4147e-03, 6.4240e-03, 1.1492e-03,
          1.0061e-03, 1.0433e-03, 6.2418e-04, 3.5667e-04, 1.3142e-03, 3.2616e-04,
          6.7282e-04, 7.6199e-04, 1.0042e-03, 1.1797e-03, 6.6233e-04, 1.5860e-03,
          1.4000e-03, 8.8358e-04, 5.0879e-04, 6.0320e-04, 1.5516e-03, 2.3785e-03,
          1.4610e-03, 1.4305e-03, 9.1362e-04, 1.8587e-03, 1.0223e-03, 9.6369e-04,
          8.4591e-04, 6.3086e-04, 6.5231e-04, 6.2704e-04, 8.9693e-04, 1.5364e-03,
          1.6184e-03, 1.4439e-03, 2.1000e-03, 1.4515e-03, 2.2354e-03, 4.9057e-03,
          1.2770e-03, 1.4601e-03, 7.2622e-04, 1.9913e-03, 3.5820e-03, 1.7328e-03,
          1.7052e-03, 1.0796e-03, 1.1578e-03, 1.0824e-03, 1.3895e-03, 5.7936e-04,
          1.1759e-03, 5.0211e-04, 8.6308e-04, 6.6471e-04, 7.4244e-04, 1.1272e-03,
          3.0861e-03, 3.1090e-03, 4.6234e-03, 5.4893e-03, 2.8419e-03, 1.0824e-03,
          9.1219e-04, 1.2608e-03, 1.9131e-03, 1.0471e-03, 7.1096e-04, 9.0313e-04,
          1.5154e-03, 1.5335e-03, 1.1120e-03, 1.3323e-03, 9.8419e-04, 1.0872e-03,
          7.7295e-04, 7.5483e-04, 7.7057e-04, 1.4553e-03, 9.5558e-04, 9.9468e-04,
          4.3983e-03, 1.9417e-03, 3.1681e-03, 2.0313e-03, 8.9121e-04, 1.2283e-03,
          3.0231e-03, 4.1733e-03, 1.2848e-02, 6.7949e-04, 2.8210e-03, 8.7452e-04,
          1.0138e-03, 1.8387e-03, 2.4529e-03, 8.0967e-04, 1.3666e-03, 1.1158e-03,
          9.9087e-04, 1.1034e-03, 9.9468e-04, 5.9175e-04, 1.2188e-03, 1.2188e-03,
          2.4986e-03, 7.7820e-03, 5.1155e-03, 1.2169e-03, 4.4632e-03, 1.2598e-03,
          2.2945e-03, 3.4981e-03, 1.3542e-02, 7.5073e-03, 4.2191e-03, 1.9722e-03,
          3.8681e-03, 7.9441e-04, 1.4009e-03, 1.2093e-03, 2.6245e-03, 6.6032e-03,
          4.5586e-03, 4.5815e-03, 1.5869e-03, 1.4315e-03, 2.2316e-03, 3.3245e-03,
          1.0071e-02, 3.3627e-03, 6.9580e-03, 4.2000e-03, 9.4223e-03, 4.2381e-03,
          5.3825e-03, 1.6800e-02, 2.5986e-02, 5.1956e-03, 1.5726e-03, 2.7885e-03,
          1.9741e-03, 8.5735e-04, 1.7233e-03, 1.5669e-03, 3.9482e-03, 4.0588e-03,
          1.7080e-03, 6.4754e-04, 7.7629e-04, 2.3918e-03, 2.9469e-03, 3.8853e-03,
          5.0964e-03, 2.6627e-03, 5.5313e-03, 1.2024e-02, 4.1809e-03, 7.0686e-03,
          2.4323e-02, 4.6692e-03, 9.4376e-03, 2.9278e-03, 4.6654e-03, 1.9722e-03,
          1.1578e-03, 3.2692e-03, 2.8648e-03, 4.8599e-03, 1.7366e-03, 1.3103e-03,
          2.8477e-03, 1.7719e-03, 1.2960e-03, 1.5182e-03, 1.4839e-02, 9.6359e-03,
          1.7334e-02, 1.1780e-02, 9.1782e-03, 3.2482e-03, 5.8327e-03, 8.6914e-02,
          4.6968e-05, 4.6968e-05, 2.3174e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 2.3174e-04,
          4.6968e-05, 4.6968e-05, 2.3174e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 5.2786e-04, 2.3174e-04,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 2.3174e-04, 4.6968e-05,
          5.2786e-04, 5.2786e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05, 2.3174e-04, 4.6968e-05,
          3.4332e-04, 2.3174e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          3.4332e-04, 4.6968e-05, 2.3174e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          4.6968e-05, 3.4332e-04, 4.6968e-05, 4.6968e-05, 4.6968e-05, 4.6968e-05,
          3.4332e-04, 7.7868e-04, 3.4332e-04, 1.1557e-04, 1.1557e-04, 4.6968e-05,
          4.6968e-05, 4.6968e-05, 4.6968e-05, 3.8099e-04, 1.1557e-04, 4.6968e-05,
          4.6968e-05, 4.6968e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.3564e-03, 4.2633e-02, 1.9211e-02, 1.6907e-02, 7.5806e-02,
          2.5665e-02, 2.0782e-02, 4.9734e-04, 3.4380e-04, 6.9189e-04, 1.6794e-03,
          5.3644e-04, 4.8208e-04, 5.8270e-04, 7.4673e-04, 5.4598e-04, 7.0095e-04,
          4.7588e-04, 3.4976e-04, 1.0948e-03, 7.5483e-04, 7.6008e-04, 6.1893e-04,
          2.2354e-03, 2.0847e-03, 1.4046e-02, 1.8444e-03, 7.1287e-04, 6.1178e-04,
          5.6553e-04, 7.5150e-04, 3.1209e-04, 7.7772e-04, 5.0306e-04, 3.8290e-04,
          1.0624e-03, 1.5364e-03, 1.0166e-03, 6.7711e-04, 5.7793e-04, 1.9722e-03,
          2.3060e-03, 1.5030e-03, 2.1362e-03, 1.0195e-03, 7.3671e-04, 1.4725e-03,
          5.4646e-04, 1.3962e-03, 8.1787e-03, 6.5184e-04, 4.9877e-04, 3.7408e-04,
          4.4656e-04, 4.8637e-04, 1.0738e-03, 8.2064e-04, 1.2388e-03, 1.4305e-03,
          2.8172e-03, 7.5245e-04, 1.3676e-03, 8.7881e-04, 1.3857e-03, 2.2411e-03,
          1.2140e-03, 5.3835e-04, 6.6471e-04, 1.7853e-03, 2.1362e-03, 1.6270e-03,
          2.1496e-03, 1.3885e-03, 7.7868e-04, 6.5279e-04, 4.6825e-04, 2.8229e-04,
          7.8154e-04, 7.3528e-04, 6.6090e-04, 6.7854e-04, 7.5912e-04, 1.1911e-03,
          2.0409e-03, 1.4582e-03, 1.0614e-03, 7.8773e-04, 4.9305e-04, 8.0109e-04,
          1.5221e-03, 1.1234e-03, 1.2922e-03, 1.1129e-03, 3.6774e-03, 4.4022e-03,
          6.6471e-04, 6.0749e-04, 4.5753e-04, 8.1682e-04, 9.2125e-04, 6.5041e-04,
          1.0643e-03, 3.1710e-04, 3.8934e-04, 1.1425e-03, 1.0796e-03, 9.8228e-04,
          1.3208e-03, 2.8610e-03, 1.6241e-03, 1.0767e-03, 1.0633e-03, 7.7915e-04,
          1.3123e-03, 3.5324e-03, 1.7176e-03, 2.3041e-03, 7.5226e-03, 1.1320e-03,
          1.0223e-03, 1.1377e-03, 5.7793e-04, 3.1567e-04, 1.3247e-03, 2.8729e-04,
          7.4148e-04, 6.1893e-04, 7.8487e-04, 1.1988e-03, 5.8842e-04, 1.3418e-03,
          1.2064e-03, 6.9094e-04, 4.2892e-04, 4.9877e-04, 1.1282e-03, 2.5177e-03,
          1.4887e-03, 1.5984e-03, 9.5081e-04, 1.9207e-03, 1.0509e-03, 9.6846e-04,
          7.5912e-04, 4.8566e-04, 6.1798e-04, 5.3883e-04, 7.4244e-04, 1.5116e-03,
          1.5545e-03, 1.4191e-03, 1.9722e-03, 1.3571e-03, 2.0504e-03, 5.5466e-03,
          1.0462e-03, 1.4782e-03, 5.6028e-04, 1.7319e-03, 3.6488e-03, 1.7710e-03,
          1.6985e-03, 1.1778e-03, 1.1721e-03, 1.1091e-03, 1.6174e-03, 6.0034e-04,
          8.9502e-04, 4.9925e-04, 8.6689e-04, 6.6233e-04, 7.5436e-04, 1.3103e-03,
          2.1381e-03, 2.3785e-03, 5.7945e-03, 6.3362e-03, 2.5787e-03, 1.0509e-03,
          8.9884e-04, 1.1349e-03, 1.8501e-03, 9.2793e-04, 6.8378e-04, 9.5987e-04,
          1.1139e-03, 1.6603e-03, 1.1969e-03, 1.4343e-03, 7.6580e-04, 1.1015e-03,
          6.3181e-04, 7.6342e-04, 8.5449e-04, 1.4057e-03, 7.5483e-04, 1.0509e-03,
          4.6501e-03, 2.3289e-03, 3.2177e-03, 1.5240e-03, 7.8106e-04, 1.1969e-03,
          2.4338e-03, 3.2082e-03, 1.3039e-02, 5.8794e-04, 1.9913e-03, 8.2731e-04,
          1.0080e-03, 2.1400e-03, 2.9716e-03, 5.8651e-04, 1.4067e-03, 9.4223e-04,
          1.1234e-03, 1.1101e-03, 9.8133e-04, 5.7745e-04, 1.2960e-03, 1.1759e-03,
          2.7924e-03, 8.0490e-03, 4.1885e-03, 1.0223e-03, 4.3106e-03, 1.2169e-03,
          1.8187e-03, 2.6474e-03, 1.0406e-02, 5.4665e-03, 3.7899e-03, 2.0084e-03,
          5.5809e-03, 7.8964e-04, 1.3037e-03, 1.2341e-03, 2.6932e-03, 5.0545e-03,
          4.4861e-03, 5.7869e-03, 1.7500e-03, 1.7452e-03, 2.1362e-03, 3.6430e-03,
          1.1055e-02, 2.5139e-03, 6.4430e-03, 3.2959e-03, 8.5907e-03, 3.8319e-03,
          4.4441e-03, 1.6251e-02, 2.6382e-02, 6.0692e-03, 1.4801e-03, 2.5616e-03,
          2.1915e-03, 7.6818e-04, 1.6718e-03, 1.4629e-03, 3.2444e-03, 3.6640e-03,
          1.6136e-03, 7.2002e-04, 8.4114e-04, 2.1267e-03, 2.8877e-03, 3.5534e-03,
          4.2648e-03, 1.9369e-03, 5.2299e-03, 1.1070e-02, 4.3564e-03, 6.5613e-03,
          1.6617e-02, 3.4466e-03, 1.1421e-02, 2.8343e-03, 4.8637e-03, 1.5707e-03,
          9.2220e-04, 3.0651e-03, 2.8725e-03, 4.5929e-03, 1.6518e-03, 1.3266e-03,
          2.3460e-03, 1.8692e-03, 1.2302e-03, 1.4849e-03, 1.5572e-02, 7.5760e-03,
          1.6205e-02, 1.0468e-02, 9.0866e-03, 3.1223e-03, 5.5656e-03, 5.9540e-02,
          5.1666e-02, 4.7743e-05, 2.3603e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          2.3603e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 2.3603e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 2.3603e-04, 2.3603e-04,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          5.7077e-04, 5.7077e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 2.3603e-04, 2.3603e-04, 4.7743e-05,
          3.8266e-04, 4.7743e-05, 2.3603e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          3.8266e-04, 4.7743e-05, 2.3603e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          4.7743e-05, 3.8266e-04, 4.7743e-05, 4.7743e-05, 4.7743e-05, 4.7743e-05,
          3.8266e-04, 3.8266e-04, 3.8266e-04, 3.5644e-04, 1.1140e-04, 4.7743e-05,
          4.7743e-05, 4.7743e-05, 4.7743e-05, 3.5644e-04, 1.1140e-04, 4.7743e-05,
          4.7743e-05, 4.7743e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5433e-03, 5.3040e-02, 1.6159e-02, 1.4809e-02, 3.6652e-02,
          3.5797e-02, 1.2047e-02, 1.7576e-03, 8.4400e-04, 2.7294e-03, 5.4054e-03,
          1.5287e-03, 5.5389e-03, 1.1425e-03, 3.1929e-03, 1.9550e-03, 5.4550e-03,
          3.4771e-03, 1.3351e-03, 3.3951e-04, 3.8862e-04, 6.7854e-04, 3.9635e-03,
          6.7291e-03, 2.1229e-03, 9.4910e-03, 4.3373e-03, 5.0068e-04, 5.5847e-03,
          1.7176e-03, 3.4046e-03, 1.3542e-03, 3.0212e-03, 2.7599e-03, 9.3555e-04,
          1.3475e-03, 3.3593e-04, 1.9894e-03, 2.4014e-03, 4.6277e-04, 1.2484e-03,
          5.5742e-04, 4.3321e-04, 4.8971e-04, 4.4990e-04, 8.3160e-03, 3.3951e-03,
          8.7023e-04, 7.1812e-04, 3.1013e-03, 1.1911e-03, 1.6136e-03, 1.0691e-03,
          3.2759e-04, 7.6246e-04, 3.5262e-04, 3.0589e-04, 7.7534e-04, 4.7445e-04,
          5.7697e-04, 1.8606e-03, 9.7084e-04, 1.3533e-03, 6.7139e-04, 5.0545e-03,
          1.0214e-03, 9.9277e-04, 1.3409e-03, 2.8849e-04, 2.7418e-04, 4.5872e-04,
          5.4312e-04, 6.0043e-03, 4.0245e-04, 2.0790e-03, 3.2377e-04, 2.8372e-04,
          6.7329e-04, 2.7275e-04, 4.7755e-04, 1.2980e-03, 5.5218e-04, 3.4499e-04,
          3.3140e-04, 3.0937e-03, 6.6948e-04, 3.6125e-03, 6.2084e-04, 5.2166e-04,
          5.7678e-03, 2.4166e-03, 4.3845e-04, 3.2654e-03, 4.7231e-04, 5.6839e-04,
          1.0281e-03, 2.3937e-04, 1.8482e-03, 1.3609e-03, 2.7514e-04, 3.9387e-04,
          3.8548e-03, 7.2956e-04, 6.1560e-04, 7.0953e-04, 2.8801e-03, 1.6336e-03,
          2.3918e-03, 4.6253e-04, 2.7332e-03, 4.2486e-04, 8.1110e-04, 9.0694e-04,
          2.2907e-03, 8.2111e-04, 9.5558e-04, 5.8365e-03, 8.1921e-04, 1.2436e-03,
          2.6093e-03, 1.2455e-03, 2.0611e-04, 1.3275e-03, 4.5929e-03, 1.8823e-04,
          2.1124e-04, 2.7061e-04, 4.0650e-04, 2.9755e-03, 3.7384e-03, 3.1796e-03,
          1.4086e-03, 9.2936e-04, 2.0542e-03, 6.0415e-04, 1.7385e-03, 4.0317e-04,
          1.4923e-02, 2.3403e-03, 1.1955e-02, 6.6605e-03, 4.5853e-03, 3.3989e-03,
          2.9874e-04, 1.5802e-03, 4.9353e-04, 3.3760e-04, 5.9223e-04, 2.1381e-03,
          2.8267e-03, 2.0771e-03, 4.7088e-04, 1.6727e-03, 1.5287e-03, 1.0366e-03,
          1.5554e-03, 7.6532e-04, 3.2558e-03, 5.1165e-04, 3.1319e-03, 5.9052e-03,
          4.5967e-03, 3.2024e-03, 1.3901e-02, 6.6719e-03, 5.6744e-04, 5.3787e-04,
          2.6627e-03, 6.9475e-04, 1.5507e-03, 4.1924e-03, 4.4250e-03, 2.3060e-03,
          1.2836e-03, 7.1621e-04, 7.6771e-04, 8.8739e-04, 2.1648e-03, 7.5798e-03,
          1.6079e-03, 2.7122e-03, 4.5128e-03, 6.6261e-03, 5.2567e-03, 9.9487e-03,
          8.0261e-03, 5.2299e-03, 5.6190e-03, 4.6444e-04, 5.6887e-04, 3.8719e-03,
          4.9019e-03, 2.2817e-04, 2.7943e-03, 1.4702e-02, 3.6469e-03, 2.2774e-03,
          8.7452e-04, 1.0166e-03, 3.4943e-03, 1.0748e-03, 2.0790e-03, 1.7948e-03,
          6.5899e-04, 1.7090e-03, 8.0204e-04, 2.6150e-03, 6.7368e-03, 5.9776e-03,
          3.2749e-03, 1.0786e-03, 4.8733e-04, 1.1845e-03, 3.6564e-03, 4.5633e-04,
          1.1501e-03, 3.9177e-03, 7.0152e-03, 4.6120e-03, 5.3978e-03, 6.0692e-03,
          4.8409e-03, 5.8270e-04, 1.3752e-03, 1.5364e-03, 8.9417e-03, 4.7607e-03,
          1.3828e-03, 7.7438e-04, 7.4148e-04, 1.9875e-03, 3.1700e-03, 4.0321e-03,
          8.3256e-04, 7.0238e-04, 5.5170e-04, 2.4586e-03, 1.4572e-03, 3.1300e-03,
          9.7942e-04, 8.3208e-04, 5.2118e-04, 2.0905e-03, 6.9885e-03, 2.2316e-03,
          8.9931e-04, 2.2659e-03, 5.8937e-04, 1.0309e-03, 1.0262e-03, 2.8572e-03,
          6.4182e-04, 6.7902e-04, 1.4219e-03, 1.6193e-03, 2.3621e-02, 5.7106e-03,
          7.3195e-04, 1.0948e-03, 9.5892e-04, 1.3475e-03, 4.4942e-04, 1.1578e-03,
          1.8253e-03, 4.1313e-03, 7.5912e-03, 1.8280e-02, 1.5930e-02, 2.1000e-03,
          1.5936e-03, 1.6394e-03, 5.3406e-03, 1.1177e-03, 3.4008e-03, 4.2953e-03,
          2.5826e-03, 3.9330e-03, 2.0981e-03, 6.8245e-03, 7.4339e-04, 9.4795e-04,
          1.4181e-03, 5.3215e-04, 3.8552e-04, 5.5170e-04, 3.4428e-03, 2.9583e-03,
          2.4567e-03, 6.9122e-03, 1.6327e-02, 1.6769e-02, 1.0443e-03, 1.8940e-03,
          1.4563e-03, 8.3017e-04, 7.0572e-04, 8.4839e-03, 3.1834e-03, 2.1820e-03,
          1.3323e-03, 4.3121e-02, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          4.0114e-05, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          4.0114e-05, 4.0114e-05, 1.7083e-04, 4.0114e-05, 4.0114e-05, 1.7083e-04,
          4.0114e-05, 4.0114e-05, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 1.7083e-04, 1.7083e-04,
          4.0114e-05, 4.0114e-05, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          1.7083e-04, 1.7083e-04, 4.0114e-05, 4.0114e-05, 1.7083e-04, 1.7083e-04,
          4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05, 1.7083e-04, 1.7083e-04,
          1.7083e-04, 1.7083e-04, 4.0114e-05, 4.0114e-05, 1.7083e-04, 4.0114e-05,
          1.7083e-04, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05, 1.7083e-04, 4.0114e-05,
          1.7083e-04, 4.9162e-04, 1.7083e-04, 4.0114e-05, 4.0114e-05, 1.7083e-04,
          1.7083e-04, 4.0114e-05, 4.9162e-04, 4.0114e-05, 1.7083e-04, 4.0114e-05,
          1.7083e-04, 1.7083e-04, 4.0114e-05, 4.0114e-05, 4.0114e-05, 4.0114e-05,
          1.7083e-04, 1.7083e-04, 1.7083e-04, 1.7083e-04, 1.7083e-04, 4.0114e-05,
          4.0114e-05, 1.7083e-04, 1.7083e-04, 6.3229e-04, 4.0114e-05, 1.7083e-04,
          1.7083e-04, 1.7083e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.3817e-03, 5.5603e-02, 1.3794e-02, 1.6373e-02, 2.9037e-02,
          3.9520e-02, 1.1505e-02, 1.0155e-02, 2.1210e-03, 1.9073e-03, 3.1185e-03,
          1.1797e-03, 1.2741e-03, 7.4816e-04, 1.4772e-03, 1.0078e-02, 1.8024e-03,
          1.3733e-03, 2.7237e-03, 2.4915e-04, 2.6822e-04, 3.7479e-04, 1.9722e-03,
          3.8452e-03, 1.4849e-03, 9.8877e-03, 4.8981e-03, 4.3535e-04, 1.8320e-03,
          9.2506e-04, 2.3178e-02, 6.2513e-04, 1.7262e-03, 2.0161e-03, 5.4169e-03,
          9.3412e-04, 4.2534e-04, 1.4086e-03, 1.7929e-03, 3.4690e-04, 8.5592e-04,
          4.6062e-04, 3.6597e-04, 4.3988e-04, 2.9612e-04, 1.9684e-03, 1.6470e-03,
          5.2261e-04, 4.6444e-04, 2.5368e-03, 6.3324e-03, 1.9760e-02, 4.0436e-03,
          1.3723e-03, 7.7782e-03, 1.2913e-03, 3.9148e-04, 1.7300e-03, 6.2037e-04,
          7.1955e-04, 8.9693e-04, 2.8877e-03, 2.1191e-03, 5.5218e-04, 2.2392e-03,
          1.0366e-03, 6.3753e-04, 7.8821e-04, 2.6011e-04, 2.3758e-04, 4.4751e-04,
          4.9829e-04, 3.9062e-03, 2.1572e-03, 2.4704e-02, 1.6336e-03, 2.1458e-03,
          5.8136e-03, 3.1710e-04, 5.1022e-04, 1.2255e-03, 7.3051e-04, 5.1689e-04,
          4.8494e-04, 3.0762e-02, 2.1095e-03, 1.8167e-03, 5.8746e-04, 5.0926e-04,
          3.6697e-03, 1.6117e-03, 2.8753e-04, 1.9321e-03, 4.0364e-04, 5.1212e-04,
          7.3624e-04, 9.3937e-04, 1.3023e-02, 1.3599e-03, 3.0136e-04, 4.3464e-04,
          2.8896e-03, 7.0381e-04, 7.6675e-04, 8.4782e-04, 2.0065e-03, 5.7411e-04,
          1.7593e-02, 4.9782e-04, 1.7977e-03, 4.3344e-04, 7.4053e-04, 7.3385e-04,
          1.1997e-03, 5.5218e-04, 4.8041e-04, 2.7542e-03, 7.4053e-04, 8.1253e-04,
          9.7942e-04, 1.0920e-03, 2.1660e-04, 8.2474e-03, 2.3212e-03, 2.2519e-04,
          2.1970e-04, 3.6430e-04, 5.0354e-04, 2.3918e-03, 7.7591e-03, 2.2278e-02,
          1.2455e-03, 7.1001e-04, 1.2751e-03, 4.6778e-04, 1.2074e-03, 3.3307e-04,
          3.2654e-03, 9.3174e-04, 3.6507e-03, 8.0032e-03, 3.0499e-03, 1.3332e-03,
          2.9564e-04, 9.9850e-04, 4.4942e-04, 4.1771e-04, 6.2227e-04, 1.9550e-03,
          2.8706e-03, 1.7776e-03, 6.2084e-04, 6.1340e-03, 6.4545e-03, 9.8801e-04,
          1.2751e-03, 9.6941e-04, 1.8120e-03, 4.9543e-04, 1.9321e-03, 1.5659e-03,
          2.1038e-03, 2.1000e-03, 3.5286e-03, 3.0003e-03, 5.2834e-04, 6.6471e-04,
          1.9197e-03, 8.3065e-04, 1.2283e-03, 1.5039e-03, 1.7834e-03, 1.5249e-03,
          2.0485e-03, 1.6756e-03, 9.0218e-04, 9.4986e-04, 1.5860e-03, 6.6566e-03,
          1.0414e-03, 1.7090e-03, 2.2221e-03, 2.6886e-02, 6.0158e-03, 2.1496e-03,
          3.7956e-03, 2.1439e-03, 2.9812e-03, 4.3297e-04, 6.2275e-04, 2.3136e-03,
          2.8580e-02, 2.2602e-04, 1.4458e-03, 3.5286e-03, 2.3155e-03, 1.5631e-03,
          1.1864e-03, 7.4053e-04, 2.2106e-03, 8.0061e-04, 1.4076e-03, 1.1873e-03,
          6.2370e-04, 1.1482e-03, 5.5456e-04, 8.7280e-03, 3.5858e-03, 1.5020e-03,
          2.3926e-02, 8.2111e-04, 5.2309e-04, 1.2150e-03, 2.2945e-03, 4.6396e-04,
          1.1635e-03, 1.8682e-03, 4.5395e-03, 2.0370e-03, 2.9202e-03, 2.4281e-03,
          3.4199e-03, 5.6744e-04, 1.0090e-03, 1.1425e-03, 4.1122e-03, 2.5482e-03,
          1.0519e-03, 9.3985e-04, 6.0415e-04, 2.3212e-03, 1.9913e-03, 1.9283e-03,
          9.2268e-04, 7.9966e-04, 6.9237e-04, 1.9455e-03, 1.1749e-03, 2.3918e-03,
          9.8228e-04, 7.8678e-04, 6.4039e-04, 1.4629e-03, 2.9888e-03, 1.1787e-03,
          8.1253e-04, 1.5793e-03, 6.3467e-04, 9.1171e-04, 1.0395e-03, 2.0218e-03,
          5.8126e-04, 4.9400e-04, 9.5463e-04, 1.1082e-03, 4.6120e-03, 1.3992e-02,
          7.2670e-04, 1.0433e-03, 1.2074e-03, 1.2989e-03, 4.4513e-04, 1.1129e-03,
          1.5526e-03, 2.7027e-03, 4.3030e-03, 6.3972e-03, 6.5918e-03, 1.8139e-03,
          1.5955e-03, 1.3199e-03, 2.7809e-03, 8.4829e-04, 2.2125e-03, 2.4509e-03,
          2.0027e-03, 3.1910e-03, 1.0386e-03, 2.7580e-03, 7.2956e-04, 1.0691e-03,
          1.4658e-03, 5.1117e-04, 3.7313e-04, 5.9080e-04, 2.0695e-03, 2.0027e-03,
          2.6054e-03, 3.4447e-03, 3.8452e-03, 5.7487e-03, 1.0223e-03, 1.6308e-03,
          1.2321e-03, 7.5197e-04, 5.2929e-04, 3.3207e-03, 2.3670e-03, 1.6069e-03,
          9.2220e-04, 5.3711e-03, 5.3101e-02, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 1.5879e-04, 1.5879e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 1.5879e-04,
          2.9874e-04, 3.5048e-05, 1.5879e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 1.5879e-04, 3.5048e-05, 3.5048e-05, 2.9564e-04,
          3.5048e-05, 1.7190e-04, 2.9564e-04, 1.7190e-04, 3.5048e-05, 3.5048e-05,
          4.2248e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 2.9564e-04,
          3.5048e-05, 1.7190e-04, 1.5879e-04, 1.7190e-04, 2.9874e-04, 3.5048e-05,
          3.5048e-05, 2.9564e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 4.3559e-04, 3.5048e-05, 1.5879e-04, 4.2248e-04,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          2.9564e-04, 1.5879e-04, 3.5048e-05, 5.5933e-04, 3.5048e-05, 3.5048e-05,
          1.5879e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 2.9564e-04, 1.7190e-04, 3.5048e-05, 3.5048e-05, 1.5879e-04,
          3.5048e-05, 3.5048e-05, 2.9564e-04, 3.5048e-05, 2.9874e-04, 5.5933e-04,
          1.5059e-03, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 1.5879e-04, 3.5048e-05,
          3.5048e-05, 1.5879e-04, 3.5048e-05, 5.6076e-04, 2.5868e-04, 2.5868e-04,
          3.5048e-05, 3.5048e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3406e-03, 5.1331e-02, 1.8021e-02, 1.7197e-02, 3.7201e-02,
          3.4576e-02, 1.0857e-02, 1.1234e-03, 7.2241e-04, 1.4877e-03, 2.2144e-03,
          8.9550e-04, 3.5706e-03, 2.5158e-03, 1.3256e-03, 1.0910e-03, 1.2161e-02,
          5.4550e-03, 5.1594e-04, 7.7868e-04, 7.1383e-04, 1.2503e-03, 3.6068e-03,
          2.5387e-03, 1.0557e-03, 8.5602e-03, 5.6648e-03, 1.7967e-03, 1.0193e-02,
          4.7226e-03, 2.4643e-03, 3.8757e-03, 3.2654e-03, 1.2941e-03, 1.5516e-03,
          1.4029e-03, 4.4608e-04, 2.2087e-03, 1.8835e-03, 1.1921e-03, 1.1663e-03,
          4.9973e-04, 3.7718e-04, 4.1103e-04, 3.9411e-04, 8.5831e-03, 3.3970e-03,
          1.8539e-03, 1.3885e-03, 3.3054e-03, 2.5063e-03, 1.9264e-03, 1.1473e-03,
          4.7612e-04, 9.4604e-04, 6.4373e-04, 5.3644e-04, 7.0620e-04, 5.8079e-04,
          6.2466e-04, 4.3449e-03, 8.1730e-04, 1.6193e-03, 5.1498e-04, 2.7561e-03,
          7.3624e-04, 4.2224e-04, 6.9427e-04, 2.3448e-04, 2.2423e-04, 3.1328e-04,
          4.2224e-04, 3.8452e-03, 9.6989e-04, 2.5330e-03, 4.5180e-04, 4.2367e-04,
          1.0052e-03, 6.2323e-04, 5.5981e-04, 1.0729e-03, 5.2691e-04, 3.7241e-04,
          3.8385e-04, 2.8191e-03, 7.1287e-04, 1.0090e-03, 3.6979e-04, 3.7527e-04,
          1.9913e-03, 1.4858e-03, 2.4128e-04, 1.2417e-03, 3.2806e-04, 4.2081e-04,
          2.8191e-03, 5.2500e-04, 2.8572e-03, 3.3188e-03, 6.5327e-04, 8.9836e-04,
          2.2697e-03, 4.6182e-04, 5.1403e-04, 8.4734e-04, 1.7815e-03, 5.2528e-03,
          2.0103e-03, 4.9162e-04, 1.8520e-03, 3.2043e-04, 5.5456e-04, 5.7554e-04,
          6.6090e-04, 6.9857e-04, 2.1744e-03, 3.1281e-03, 6.0415e-04, 5.3368e-03,
          9.4070e-03, 2.5501e-03, 4.0150e-04, 6.2561e-04, 2.3060e-03, 2.7966e-04,
          4.4107e-04, 3.5882e-04, 5.4502e-04, 2.3651e-03, 2.5883e-03, 2.3556e-03,
          8.8739e-04, 5.2214e-04, 4.3845e-04, 2.9683e-04, 9.4175e-04, 2.8920e-04,
          1.5991e-02, 5.7945e-03, 5.0125e-03, 3.9711e-03, 3.9902e-03, 1.1116e-02,
          6.1035e-04, 4.0078e-04, 6.0749e-04, 6.3658e-04, 6.2561e-04, 1.6985e-03,
          2.3842e-03, 1.5488e-03, 5.9271e-04, 2.7580e-03, 1.5993e-03, 1.2245e-03,
          1.1864e-03, 8.3590e-04, 5.2881e-04, 3.6144e-04, 2.3651e-03, 1.4122e-02,
          3.7136e-03, 2.7218e-03, 1.4069e-02, 2.8038e-03, 2.2488e-03, 8.7929e-04,
          1.7490e-03, 9.9087e-04, 1.4830e-03, 1.1015e-03, 7.2289e-03, 3.3092e-03,
          8.5592e-04, 5.7983e-04, 1.0328e-03, 1.1158e-03, 2.0218e-03, 2.6932e-03,
          1.0672e-03, 1.4601e-03, 2.2850e-03, 2.3251e-03, 1.5802e-03, 1.4687e-03,
          1.5221e-03, 9.2392e-03, 1.6356e-03, 1.3752e-03, 7.8154e-04, 2.4261e-03,
          2.6665e-03, 4.6062e-04, 2.5444e-03, 1.7761e-02, 1.6375e-03, 2.1286e-03,
          1.0214e-03, 3.3379e-03, 2.6550e-03, 7.7105e-04, 1.1082e-03, 1.3685e-03,
          5.0974e-04, 1.1063e-03, 5.4550e-04, 7.7438e-04, 2.5520e-03, 6.5956e-03,
          1.2026e-03, 4.3411e-03, 1.4973e-03, 1.1425e-03, 3.7327e-03, 6.4325e-04,
          2.0046e-03, 1.1780e-02, 2.3365e-03, 9.6035e-04, 6.1455e-03, 1.1696e-02,
          8.1329e-03, 6.2370e-04, 1.1806e-03, 8.5402e-04, 3.7651e-03, 2.5635e-03,
          6.9237e-04, 4.2844e-04, 4.3964e-04, 9.3365e-04, 1.6747e-03, 6.6261e-03,
          2.9373e-03, 1.1797e-03, 7.7963e-04, 1.1349e-03, 2.8839e-03, 3.4389e-03,
          1.3781e-03, 1.2102e-03, 9.4604e-04, 4.1275e-03, 1.9150e-02, 6.5155e-03,
          1.2140e-03, 1.1120e-03, 4.8637e-04, 7.2241e-04, 7.3195e-04, 1.7262e-03,
          4.3821e-04, 3.7694e-04, 7.5102e-04, 1.1616e-03, 6.8054e-03, 6.4735e-03,
          2.8591e-03, 1.3094e-03, 1.1911e-03, 1.3676e-03, 7.1669e-04, 1.6937e-03,
          2.4509e-03, 1.2684e-03, 4.3716e-03, 4.0405e-02, 2.0996e-02, 1.4620e-03,
          1.2455e-03, 1.0681e-03, 2.4605e-03, 8.8549e-04, 3.3588e-03, 1.9989e-03,
          1.3123e-03, 1.2188e-03, 2.9373e-03, 1.4740e-02, 2.8572e-03, 1.8587e-03,
          1.5888e-03, 1.0509e-03, 8.4925e-04, 1.0595e-03, 8.9722e-03, 2.8362e-03,
          1.9741e-03, 6.2141e-03, 1.5472e-02, 9.2926e-03, 1.0977e-03, 1.6727e-03,
          1.4019e-03, 7.5674e-04, 6.3133e-04, 4.1504e-03, 2.1286e-03, 9.6941e-04,
          8.2493e-04, 3.8013e-03, 2.6684e-03, 4.9561e-02, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.7452e-04,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.0242e-04, 2.0242e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.0242e-04, 2.9862e-05,
          2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.1855e-04, 2.9862e-05,
          2.9862e-05, 2.0242e-04, 2.9862e-05, 2.9862e-05, 3.4714e-04, 1.7452e-04,
          1.1855e-04, 2.9862e-05, 1.7452e-04, 2.9862e-05, 2.9862e-05, 1.1855e-04,
          3.4714e-04, 3.4714e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          3.4714e-04, 2.9862e-05, 2.9862e-05, 2.6321e-04, 2.9862e-05, 2.9862e-05,
          2.0242e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 1.7452e-04, 2.9862e-05,
          2.0242e-04, 1.7452e-04, 3.4714e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          3.4714e-04, 2.9862e-05, 1.7452e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.9862e-05, 2.0242e-04, 2.9862e-05, 2.9862e-05, 2.9862e-05, 2.9862e-05,
          2.0242e-04, 2.0242e-04, 2.0242e-04, 2.0242e-04, 2.0242e-04, 2.9862e-05,
          2.9862e-05, 3.4714e-04, 2.0242e-04, 5.7364e-04, 2.9862e-05, 2.0242e-04,
          2.0242e-04, 2.0242e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3101e-03, 4.5319e-02, 1.9424e-02, 1.6983e-02, 6.8970e-02,
          2.7222e-02, 1.7136e-02, 7.7820e-04, 4.8304e-04, 1.4467e-03, 6.9733e-03,
          1.2922e-03, 6.5422e-04, 8.3637e-04, 8.5831e-04, 5.3692e-04, 7.5865e-04,
          4.4870e-04, 3.7313e-04, 5.6267e-04, 4.6682e-04, 4.3440e-04, 5.3549e-04,
          5.6114e-03, 1.3914e-03, 1.1536e-02, 2.2011e-03, 1.9608e-03, 7.8773e-04,
          1.1072e-03, 1.0281e-03, 4.2820e-04, 9.3317e-04, 5.3692e-04, 4.1294e-04,
          1.1883e-03, 1.4038e-03, 8.5497e-04, 5.9509e-04, 6.3610e-04, 1.1454e-03,
          1.7967e-03, 1.1969e-03, 1.3676e-03, 6.0987e-04, 6.5947e-04, 1.0366e-03,
          4.3917e-04, 6.8188e-04, 6.7444e-03, 1.1911e-03, 8.7118e-04, 5.6028e-04,
          6.4659e-04, 5.5981e-04, 1.0481e-03, 2.0962e-03, 1.2722e-03, 2.2316e-03,
          2.0218e-03, 7.6818e-04, 9.5415e-04, 6.0844e-04, 1.8845e-03, 1.5574e-03,
          2.4300e-03, 8.2970e-04, 6.7616e-04, 8.6832e-04, 1.0147e-03, 1.5631e-03,
          1.1415e-03, 2.3975e-03, 1.7548e-03, 1.0881e-03, 6.2895e-04, 3.7909e-04,
          9.8705e-04, 3.1929e-03, 2.8954e-03, 1.2703e-03, 2.2850e-03, 1.0958e-03,
          1.3466e-03, 1.0653e-03, 8.1015e-04, 1.4601e-03, 9.6035e-04, 1.5125e-03,
          2.5959e-03, 1.2426e-03, 7.6962e-04, 1.5888e-03, 9.0408e-04, 1.1702e-03,
          1.2312e-03, 1.0948e-03, 5.4121e-04, 1.5554e-03, 6.6681e-03, 2.9068e-03,
          4.9744e-03, 1.4076e-03, 1.3981e-03, 1.6489e-03, 1.2894e-03, 8.1682e-04,
          9.3460e-04, 1.5602e-03, 1.9684e-03, 1.4524e-03, 2.3270e-03, 1.2808e-03,
          1.0252e-03, 9.8515e-04, 4.3941e-04, 1.5659e-03, 1.4448e-03, 2.6398e-03,
          2.2049e-03, 2.9011e-03, 4.7760e-03, 4.9353e-04, 1.7757e-03, 1.6317e-03,
          2.9373e-03, 2.5177e-03, 3.1548e-03, 1.2798e-03, 5.2166e-04, 1.0214e-03,
          2.8667e-03, 1.4095e-03, 4.0364e-04, 5.7125e-04, 1.2760e-03, 1.0557e-03,
          9.5892e-04, 7.0715e-04, 7.4720e-04, 1.6336e-03, 9.9182e-04, 1.6365e-03,
          6.6719e-03, 1.1644e-03, 2.1038e-03, 2.4223e-03, 3.7174e-03, 2.2717e-03,
          3.0060e-03, 1.3657e-03, 2.4796e-03, 8.2064e-04, 1.6642e-03, 1.5945e-03,
          1.1692e-03, 1.0481e-03, 4.4131e-04, 1.3580e-03, 1.6489e-03, 6.2466e-04,
          1.2035e-03, 9.3174e-04, 9.2554e-04, 1.1377e-03, 7.3547e-03, 2.7084e-03,
          4.9858e-03, 1.8625e-03, 1.7109e-03, 8.6451e-04, 9.2459e-04, 1.5430e-03,
          1.2360e-03, 1.6623e-03, 3.1452e-03, 2.1172e-03, 1.2875e-03, 9.5749e-04,
          9.6083e-04, 6.9189e-04, 1.2436e-03, 8.7261e-04, 6.7091e-04, 6.7139e-04,
          1.2655e-03, 7.6771e-04, 1.4429e-03, 1.4038e-02, 4.7646e-03, 1.4305e-03,
          9.1076e-04, 4.6921e-03, 1.0023e-03, 1.4629e-03, 1.7366e-03, 1.2388e-03,
          3.2063e-03, 1.9474e-03, 1.3924e-03, 1.4772e-03, 1.7071e-03, 8.4162e-04,
          1.8606e-03, 1.4076e-03, 1.5841e-03, 6.4325e-04, 3.0079e-03, 5.1212e-04,
          8.4734e-04, 7.0305e-03, 2.8137e-02, 3.8738e-03, 3.1891e-03, 5.1918e-03,
          1.7767e-03, 1.3561e-03, 1.5335e-03, 6.2037e-04, 1.4486e-03, 1.2741e-03,
          2.2278e-03, 2.6150e-03, 1.5945e-03, 1.4257e-03, 2.1267e-03, 1.1463e-03,
          1.6298e-03, 1.8072e-03, 1.5478e-03, 1.6985e-03, 1.1482e-03, 7.5531e-04,
          2.9968e-02, 5.0011e-03, 8.0109e-03, 1.4935e-03, 8.7433e-03, 9.5215e-03,
          9.5673e-03, 9.1934e-03, 4.1656e-03, 2.1191e-03, 2.4624e-03, 3.1166e-03,
          2.4319e-03, 2.0676e-03, 2.0752e-03, 1.3418e-03, 2.2278e-03, 1.3008e-03,
          1.8635e-03, 1.8454e-03, 2.8286e-03, 1.4305e-03, 9.4843e-04, 2.8076e-03,
          1.2398e-02, 4.0359e-03, 3.5000e-03, 2.2545e-03, 5.5695e-03, 8.8730e-03,
          2.0561e-03, 8.5688e-04, 8.6260e-04, 1.9016e-03, 2.6417e-03, 1.1072e-03,
          4.5929e-03, 2.8591e-03, 1.8272e-03, 1.7529e-03, 1.8167e-03, 2.1400e-03,
          2.3060e-03, 1.0920e-03, 1.8988e-03, 1.3781e-03, 1.4915e-02, 1.4107e-02,
          6.0577e-03, 1.8494e-02, 1.5945e-02, 1.1147e-02, 2.6760e-03, 1.6737e-03,
          3.8395e-03, 2.4719e-03, 1.1063e-03, 1.2999e-03, 3.9062e-03, 3.5973e-03,
          4.1161e-03, 2.9583e-03, 2.3556e-03, 1.6317e-03, 1.6823e-03, 4.0588e-03,
          3.4561e-03, 9.5749e-04, 1.3533e-03, 3.8776e-03, 6.0303e-02, 4.8637e-05,
          4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          1.4758e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 1.4758e-04, 1.4758e-04, 1.4758e-04, 4.8637e-05, 4.8637e-05,
          3.1948e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 1.4758e-04,
          4.8637e-05, 4.8637e-05, 1.4758e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 4.8637e-05, 1.6708e-03, 4.8637e-05, 4.8637e-05, 1.4758e-04,
          4.8637e-05, 4.8637e-05, 3.1948e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 2.2054e-04, 2.2054e-04,
          4.8637e-05, 1.4758e-04, 4.8637e-05, 4.8637e-05, 1.4758e-04, 4.8637e-05,
          3.1948e-04, 3.1948e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 1.4758e-04, 4.8637e-05, 1.4758e-04, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05, 2.2054e-04, 4.8637e-05,
          1.4758e-04, 3.1948e-04, 1.4758e-04, 4.8637e-05, 4.8637e-05, 1.4758e-04,
          1.4758e-04, 4.8637e-05, 3.1948e-04, 4.6778e-04, 4.8637e-05, 4.8637e-05,
          1.4758e-04, 1.4758e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 1.4758e-04, 1.4758e-04, 4.8637e-05, 1.4758e-04, 4.8637e-05,
          4.8637e-05, 1.4758e-04, 4.8637e-05, 4.8637e-05, 4.8637e-05, 4.8637e-05,
          4.8637e-05, 4.8637e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5433e-03, 4.9469e-02, 1.1253e-02, 1.3100e-02, 3.1555e-02,
          3.5492e-02, 8.0490e-03, 8.5735e-04, 7.1907e-04, 3.7746e-03, 6.5651e-03,
          2.3632e-03, 1.0271e-03, 3.5548e-04, 2.8229e-03, 7.8487e-04, 1.1969e-03,
          6.6423e-04, 4.3225e-04, 1.4234e-04, 1.9109e-04, 1.7011e-04, 1.0672e-03,
          7.8888e-03, 3.2330e-03, 6.4163e-03, 3.5534e-03, 3.4571e-04, 1.5059e-03,
          5.7364e-04, 1.8845e-03, 4.6062e-04, 3.6430e-03, 2.2106e-03, 9.9087e-04,
          2.9716e-03, 7.6151e-04, 2.2221e-03, 2.2850e-03, 2.2161e-04, 2.5482e-03,
          6.4516e-04, 6.1607e-04, 5.0688e-04, 8.8453e-04, 1.4935e-03, 4.9019e-03,
          2.3103e-04, 2.1005e-04, 2.1839e-03, 1.2932e-03, 1.4524e-03, 1.0128e-03,
          3.3545e-04, 7.4530e-04, 3.9816e-04, 1.3008e-03, 1.1921e-03, 1.8969e-03,
          1.1473e-03, 6.0129e-04, 6.1512e-04, 9.3651e-04, 2.1305e-03, 9.8724e-03,
          1.0424e-03, 1.4429e-03, 1.3390e-03, 4.1294e-04, 3.8362e-04, 4.5156e-04,
          3.9148e-04, 8.0414e-03, 5.4741e-04, 1.8740e-03, 3.1257e-04, 3.0875e-04,
          6.4707e-04, 1.6356e-03, 1.7538e-03, 5.0583e-03, 1.8749e-03, 9.7370e-04,
          6.0081e-04, 2.0504e-03, 4.2439e-04, 3.3455e-03, 6.4182e-04, 5.3978e-04,
          6.5346e-03, 5.4283e-03, 3.4380e-04, 4.2953e-03, 4.2248e-04, 3.8338e-04,
          4.6754e-04, 2.9397e-04, 1.4610e-03, 2.0008e-03, 1.8187e-03, 3.1204e-03,
          1.0094e-02, 1.0557e-03, 1.7052e-03, 4.0436e-03, 6.5689e-03, 6.8235e-04,
          1.5554e-03, 6.8188e-04, 5.5313e-03, 3.8290e-04, 8.6927e-04, 1.0710e-03,
          3.0422e-03, 1.2598e-03, 2.3031e-04, 9.4604e-03, 5.3358e-04, 8.9884e-04,
          1.3142e-03, 7.7324e-03, 1.0395e-03, 6.4611e-04, 1.1497e-02, 8.0442e-04,
          1.0262e-03, 7.8297e-04, 1.2207e-03, 6.3019e-03, 1.7529e-03, 1.8778e-03,
          1.3971e-03, 1.0853e-03, 7.2002e-04, 5.0831e-04, 3.2043e-03, 3.9577e-04,
          4.1580e-03, 4.7541e-04, 1.0281e-03, 2.2354e-03, 3.1013e-03, 1.5202e-03,
          2.0370e-03, 1.4381e-03, 2.1420e-03, 2.8076e-03, 1.4038e-03, 9.3155e-03,
          1.0170e-02, 5.1804e-03, 2.4471e-03, 1.5182e-03, 9.4986e-04, 2.4643e-03,
          3.0117e-03, 1.6537e-03, 1.3380e-03, 4.7541e-04, 5.1460e-03, 1.0166e-03,
          5.6229e-03, 4.2496e-03, 2.4872e-03, 3.6240e-03, 2.6150e-03, 2.2182e-03,
          1.0475e-02, 3.2272e-03, 5.5771e-03, 2.7084e-03, 1.7681e-03, 5.3520e-03,
          2.7294e-03, 1.3409e-03, 1.7309e-03, 1.3332e-03, 4.7340e-03, 2.5234e-03,
          2.5520e-03, 1.8158e-03, 6.7253e-03, 2.1992e-03, 8.7690e-04, 1.3313e-03,
          4.1695e-03, 5.6458e-04, 1.6899e-03, 3.3379e-03, 1.9569e-03, 5.6419e-03,
          2.4452e-03, 1.2712e-03, 4.8256e-03, 3.4866e-03, 6.9580e-03, 4.7531e-03,
          1.7433e-03, 8.3160e-04, 4.1161e-03, 1.8282e-03, 2.3365e-03, 2.6245e-03,
          9.0551e-04, 4.0398e-03, 9.6560e-04, 9.4986e-04, 7.6561e-03, 1.1940e-03,
          9.6893e-04, 3.5534e-03, 2.2602e-03, 5.1765e-03, 1.0025e-02, 1.8959e-03,
          4.0550e-03, 2.7523e-03, 3.4599e-03, 1.5707e-03, 6.6872e-03, 1.8225e-03,
          4.3297e-03, 8.0538e-04, 3.1700e-03, 1.7376e-03, 1.4465e-02, 3.8567e-03,
          1.5144e-03, 6.0177e-04, 8.3637e-04, 2.0428e-03, 3.1929e-03, 6.5804e-04,
          2.7637e-03, 3.2272e-03, 2.0008e-03, 3.3798e-03, 1.0391e-02, 1.2665e-02,
          3.5439e-03, 1.8749e-03, 1.9083e-03, 2.9373e-03, 4.3831e-03, 1.6127e-03,
          1.5059e-03, 2.7237e-03, 6.6614e-04, 1.1864e-03, 1.0986e-03, 5.1651e-03,
          5.8556e-04, 6.7759e-04, 9.1457e-04, 1.7500e-03, 2.6627e-03, 5.9547e-03,
          4.8332e-03, 2.9850e-03, 4.4136e-03, 4.5700e-03, 1.9608e-03, 4.5509e-03,
          5.0812e-03, 2.0027e-03, 1.3885e-03, 4.3793e-03, 3.6793e-03, 2.8191e-03,
          2.0504e-03, 2.1458e-03, 9.0790e-03, 1.5650e-03, 5.0392e-03, 5.9738e-03,
          2.9182e-03, 2.6436e-03, 4.7278e-04, 1.3294e-03, 2.3136e-03, 4.5776e-03,
          3.7556e-03, 2.0370e-03, 1.5612e-03, 1.8616e-03, 5.9471e-03, 3.8929e-03,
          7.6065e-03, 1.0078e-02, 3.1910e-03, 2.6264e-03, 1.3113e-03, 2.5654e-03,
          1.9550e-03, 1.2302e-03, 8.5783e-04, 6.6147e-03, 6.5918e-03, 1.3285e-03,
          9.7752e-04, 1.3523e-03, 2.2469e-03, 5.3444e-03, 3.6259e-03, 3.3691e-02,
          3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05,
          5.9938e-04, 3.6001e-05, 2.7800e-04, 1.3280e-04, 3.6001e-05, 3.6001e-05,
          3.6001e-05, 3.6001e-05, 7.4482e-04, 1.3280e-04, 2.7800e-04, 3.6001e-05,
          5.9938e-04, 1.3280e-04, 1.8775e-04, 3.6001e-05, 3.6001e-05, 1.3280e-04,
          3.6001e-05, 1.3280e-04, 2.8467e-04, 1.3280e-04, 3.6001e-05, 3.6001e-05,
          3.6001e-05, 3.6001e-05, 1.3280e-04, 2.7800e-04, 3.6001e-05, 5.2643e-04,
          3.6001e-05, 3.6001e-05, 2.8467e-04, 3.6001e-05, 4.5252e-04, 7.4482e-04,
          7.4482e-04, 1.3280e-04, 3.6001e-05, 3.6001e-05, 1.3280e-04, 1.3280e-04,
          4.5252e-04, 5.0259e-04, 3.6001e-05, 3.6001e-05, 2.8467e-04, 3.6001e-05,
          2.8467e-04, 2.8467e-04, 3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05,
          1.3280e-04, 4.7112e-04, 3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05,
          3.6001e-05, 3.6001e-05, 3.6001e-05, 3.6001e-05, 1.3280e-04, 3.6001e-05,
          2.8467e-04, 1.3280e-04, 3.6001e-05, 3.6001e-05, 3.6001e-05, 7.4482e-04,
          1.3280e-04, 2.7800e-04, 9.9373e-04, 3.6001e-05, 3.6001e-05, 2.7800e-04,
          8.9645e-04, 1.8775e-04, 3.6001e-05, 2.7800e-04, 2.7800e-04, 3.6001e-05,
          1.8775e-04, 3.6001e-05, 3.6001e-05, 3.6001e-05, 1.8775e-04, 3.6001e-05,
          3.6001e-05, 1.8775e-04, 1.8775e-04, 1.8775e-04, 1.8775e-04, 5.3215e-04,
          1.8775e-04, 1.8775e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.5324e-03, 5.3680e-02, 1.0780e-02, 1.1612e-02, 3.0823e-02,
          4.6326e-02, 8.3160e-03, 7.6294e-04, 5.8413e-04, 2.0390e-03, 2.7027e-03,
          1.1015e-03, 1.4133e-03, 3.6263e-04, 3.5458e-03, 7.6675e-04, 1.8587e-03,
          7.7438e-04, 4.6682e-04, 1.8787e-04, 2.5392e-04, 2.1100e-04, 1.1559e-03,
          3.1891e-03, 2.6665e-03, 6.7406e-03, 3.3569e-03, 4.0841e-04, 2.1839e-03,
          5.5742e-04, 1.9684e-03, 5.4359e-04, 5.9204e-03, 2.2869e-03, 1.3151e-03,
          2.9011e-03, 5.2547e-04, 2.7618e-03, 3.7422e-03, 2.7084e-04, 1.2693e-03,
          4.6754e-04, 4.0984e-04, 3.7432e-04, 3.5429e-04, 2.1420e-03, 6.9504e-03,
          2.4462e-04, 2.9612e-04, 2.0027e-03, 1.5574e-03, 1.3084e-03, 1.1969e-03,
          2.8396e-04, 7.2050e-04, 4.8256e-04, 7.7820e-04, 8.9598e-04, 8.1873e-04,
          8.4019e-04, 7.6962e-04, 8.9121e-04, 1.3170e-03, 6.0987e-04, 8.5754e-03,
          5.6076e-04, 4.0364e-04, 8.9836e-04, 3.3832e-04, 2.8062e-04, 3.1710e-04,
          4.2844e-04, 5.4932e-03, 5.0163e-04, 1.8129e-03, 2.6512e-04, 2.7013e-04,
          7.2098e-04, 7.8392e-04, 5.7316e-04, 2.5711e-03, 5.5790e-04, 5.5647e-04,
          5.4073e-04, 2.2564e-03, 4.9448e-04, 1.5755e-03, 3.5501e-04, 3.2735e-04,
          2.8839e-03, 1.8721e-03, 2.8014e-04, 1.6584e-03, 4.6492e-04, 4.2796e-04,
          5.2214e-04, 3.4070e-04, 2.2945e-03, 3.6106e-03, 7.5579e-04, 1.4381e-03,
          4.1275e-03, 4.7302e-04, 5.7650e-04, 2.1839e-03, 6.5765e-03, 9.2506e-04,
          1.8568e-03, 4.9686e-04, 2.8172e-03, 3.6287e-04, 5.1785e-04, 6.5851e-04,
          2.1572e-03, 1.6060e-03, 3.6407e-04, 1.1421e-02, 5.7030e-04, 9.9945e-04,
          1.3046e-03, 5.9509e-03, 4.4584e-04, 6.0368e-04, 6.3324e-03, 3.2210e-04,
          6.9094e-04, 3.8338e-04, 6.3801e-04, 1.0323e-02, 2.5959e-03, 2.2697e-03,
          8.4734e-04, 4.7255e-04, 8.0156e-04, 3.3092e-04, 1.4582e-03, 3.2592e-04,
          8.1863e-03, 6.2990e-04, 1.7204e-03, 2.8954e-03, 4.6005e-03, 1.7529e-03,
          7.6103e-04, 1.0662e-03, 1.1168e-03, 1.1311e-03, 6.5613e-04, 9.7351e-03,
          6.2599e-03, 6.8703e-03, 1.5650e-03, 2.0332e-03, 1.3304e-03, 2.0065e-03,
          1.9321e-03, 1.1139e-03, 8.4639e-04, 3.3259e-04, 7.4921e-03, 1.7214e-03,
          7.9041e-03, 3.9101e-03, 4.8637e-03, 5.8098e-03, 2.0370e-03, 1.4296e-03,
          2.8381e-03, 1.7214e-03, 5.5161e-03, 5.0888e-03, 2.4452e-03, 8.4991e-03,
          2.6875e-03, 1.1158e-03, 1.8225e-03, 1.0414e-03, 4.0054e-03, 2.5482e-03,
          2.0618e-03, 3.5553e-03, 5.3291e-03, 2.5177e-03, 1.0843e-03, 3.0842e-03,
          2.3022e-03, 8.2254e-04, 1.6413e-03, 1.4229e-03, 8.0299e-04, 6.7139e-03,
          2.8324e-03, 6.2180e-04, 8.2245e-03, 6.4468e-03, 3.3379e-03, 6.6147e-03,
          1.5688e-03, 1.4238e-03, 6.5727e-03, 9.0265e-04, 1.2960e-03, 2.4090e-03,
          5.3692e-04, 1.6518e-03, 7.7820e-04, 9.9564e-04, 3.9253e-03, 2.1324e-03,
          1.0548e-03, 2.9697e-03, 1.0319e-03, 1.3285e-03, 1.4717e-02, 8.3017e-04,
          5.5084e-03, 4.2877e-03, 3.7956e-03, 2.9545e-03, 1.3870e-02, 2.8763e-03,
          1.5388e-02, 6.9714e-04, 1.9608e-03, 1.0061e-03, 9.8343e-03, 5.6572e-03,
          6.6185e-04, 4.9543e-04, 6.2466e-04, 1.7643e-03, 4.5624e-03, 1.0376e-03,
          1.7538e-03, 1.4286e-03, 8.7166e-04, 5.1804e-03, 7.8812e-03, 7.7019e-03,
          2.0180e-03, 1.6565e-03, 1.6346e-03, 4.7302e-03, 8.0795e-03, 1.9207e-03,
          1.3018e-03, 1.3742e-03, 5.6648e-04, 8.5068e-04, 9.6560e-04, 4.7035e-03,
          4.7588e-04, 5.7507e-04, 9.2077e-04, 2.0618e-03, 6.9923e-03, 8.7280e-03,
          3.2005e-03, 1.4744e-03, 3.4752e-03, 6.5002e-03, 1.3456e-03, 2.9411e-03,
          8.6594e-03, 3.0613e-03, 2.6340e-03, 8.1711e-03, 3.8548e-03, 3.3646e-03,
          1.2779e-03, 1.0366e-03, 6.7825e-03, 1.2627e-03, 9.1629e-03, 7.6180e-03,
          2.6741e-03, 2.5806e-03, 6.7711e-04, 1.6212e-03, 1.7042e-03, 1.6251e-03,
          1.2960e-03, 1.1873e-03, 9.5177e-04, 1.5707e-03, 1.0704e-02, 8.1177e-03,
          4.4746e-03, 1.2642e-02, 5.7373e-03, 4.1695e-03, 1.3819e-03, 2.0428e-03,
          1.7185e-03, 9.3746e-04, 7.6103e-04, 1.5083e-02, 6.4926e-03, 1.3256e-03,
          1.0366e-03, 2.7828e-03, 2.9030e-03, 8.0948e-03, 1.6575e-03, 1.0780e-02,
          2.8717e-02, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          2.6643e-05, 1.4877e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          1.4877e-04, 2.8682e-04, 1.4877e-04, 2.6643e-05, 2.6643e-05, 3.4118e-04,
          2.6643e-05, 2.6643e-05, 3.4118e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05, 1.4877e-04, 3.4118e-04,
          2.6643e-05, 2.6643e-05, 1.4877e-04, 1.4877e-04, 2.4819e-04, 4.0913e-04,
          4.0913e-04, 3.4118e-04, 2.6643e-05, 2.6643e-05, 3.4118e-04, 3.4118e-04,
          2.4819e-04, 1.4877e-04, 2.6643e-05, 2.6643e-05, 3.4118e-04, 2.6643e-05,
          3.4118e-04, 3.4118e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          2.6643e-05, 4.0913e-04, 2.6643e-05, 1.4877e-04, 2.6643e-05, 2.6643e-05,
          1.4877e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05, 3.4118e-04, 2.6643e-05,
          1.4877e-04, 3.4118e-04, 1.4877e-04, 2.6643e-05, 2.6643e-05, 4.0913e-04,
          1.4877e-04, 2.6643e-05, 6.0129e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          4.0913e-04, 1.4877e-04, 2.6643e-05, 2.6643e-05, 2.6643e-05, 2.6643e-05,
          1.4877e-04, 2.6643e-05, 4.7970e-04, 2.6643e-05, 1.4877e-04, 2.6643e-05,
          2.6643e-05, 2.6643e-05, 1.4877e-04, 2.7704e-04, 2.6643e-05, 1.4877e-04,
          1.4877e-04, 1.4877e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.4926e-03, 4.4769e-02, 2.0142e-02, 1.6068e-02, 8.4473e-02,
          2.4734e-02, 1.5305e-02, 1.0262e-03, 7.4053e-04, 6.9666e-04, 1.3227e-03,
          5.3740e-04, 1.7939e-03, 1.6050e-03, 1.2283e-03, 8.4972e-04, 3.3817e-03,
          1.5287e-03, 5.3787e-04, 1.5173e-03, 1.1702e-03, 1.2131e-03, 1.0948e-03,
          1.2712e-03, 1.1311e-03, 1.0864e-02, 3.2253e-03, 3.7308e-03, 2.9278e-03,
          1.7738e-03, 1.6632e-03, 1.5736e-03, 1.5631e-03, 6.1464e-04, 1.1368e-03,
          1.2836e-03, 8.0299e-04, 1.2484e-03, 8.9169e-04, 2.0847e-03, 7.0858e-04,
          6.8521e-04, 5.0735e-04, 5.9938e-04, 3.5739e-04, 2.2335e-03, 1.4887e-03,
          9.3317e-04, 1.7910e-03, 8.2397e-03, 1.9798e-03, 1.1635e-03, 7.7295e-04,
          7.8440e-04, 8.5354e-04, 1.6861e-03, 8.8024e-04, 1.0424e-03, 8.8024e-04,
          1.2293e-03, 4.1542e-03, 1.2150e-03, 1.2150e-03, 4.4394e-04, 1.4315e-03,
          7.7534e-04, 2.3973e-04, 3.7360e-04, 4.3631e-04, 4.6062e-04, 5.3263e-04,
          6.9189e-04, 1.3113e-03, 1.6890e-03, 1.4467e-03, 6.6614e-04, 4.9543e-04,
          1.5011e-03, 1.1034e-03, 6.7568e-04, 6.0034e-04, 5.4598e-04, 6.3992e-04,
          1.0166e-03, 1.9989e-03, 9.8991e-04, 4.6563e-04, 2.9254e-04, 4.4584e-04,
          6.9809e-04, 4.2987e-04, 5.2977e-04, 5.0259e-04, 6.5231e-04, 7.5626e-04,
          3.0327e-03, 1.2589e-03, 1.2808e-03, 2.4204e-03, 1.4210e-03, 7.9107e-04,
          1.0262e-03, 4.2892e-04, 4.0436e-04, 1.0242e-03, 1.3399e-03, 7.9651e-03,
          1.4648e-03, 7.7915e-04, 7.0477e-04, 5.7650e-04, 6.3562e-04, 4.0889e-04,
          6.8521e-04, 1.1082e-03, 2.9297e-03, 1.8110e-03, 1.1616e-03, 9.6436e-03,
          8.9722e-03, 1.4706e-03, 1.0090e-03, 6.0177e-04, 1.6880e-03, 4.3344e-04,
          1.2045e-03, 6.0654e-04, 7.9060e-04, 1.6985e-03, 1.2665e-03, 1.6117e-03,
          1.0643e-03, 3.8433e-04, 4.2939e-04, 2.5058e-04, 4.0436e-04, 6.3562e-04,
          4.2191e-03, 3.4695e-03, 3.1757e-03, 2.4948e-03, 1.8806e-03, 6.4201e-03,
          1.2627e-03, 8.1205e-04, 7.8201e-04, 6.9809e-04, 8.0585e-04, 1.9503e-03,
          1.3723e-03, 1.4620e-03, 1.3752e-03, 2.0256e-03, 2.4128e-03, 1.4229e-03,
          4.7660e-04, 6.4945e-04, 3.0255e-04, 5.3883e-04, 1.8520e-03, 4.3411e-03,
          1.6451e-03, 9.2363e-04, 3.8929e-03, 1.9112e-03, 3.7842e-03, 9.0027e-04,
          1.0090e-03, 6.6185e-04, 1.1578e-03, 1.0433e-03, 3.6621e-03, 2.4014e-03,
          1.3237e-03, 1.4648e-03, 3.0804e-03, 2.1591e-03, 1.1330e-03, 1.2140e-03,
          6.2799e-04, 8.6927e-04, 1.2026e-03, 1.4954e-03, 9.7466e-04, 1.3199e-03,
          7.4387e-04, 3.5248e-03, 2.0142e-03, 2.9316e-03, 1.1244e-03, 1.6813e-03,
          1.5612e-03, 1.4877e-03, 1.4238e-03, 5.6343e-03, 6.5804e-04, 1.4124e-03,
          3.0098e-03, 1.9455e-02, 2.3346e-03, 5.6171e-04, 6.1655e-04, 6.8283e-04,
          7.3481e-04, 5.9986e-04, 9.7561e-04, 6.6948e-04, 1.0090e-03, 1.6069e-03,
          9.7513e-04, 6.4659e-03, 4.2953e-03, 7.6056e-04, 2.6665e-03, 1.2960e-03,
          2.0676e-03, 9.3842e-03, 1.1940e-03, 1.4076e-03, 3.1929e-03, 7.5989e-03,
          8.2397e-03, 1.9875e-03, 9.6798e-04, 5.7602e-04, 2.4185e-03, 1.2617e-03,
          5.7507e-04, 8.5831e-04, 9.1219e-04, 1.6117e-03, 1.4172e-03, 4.1504e-03,
          7.1449e-03, 1.2579e-03, 1.5526e-03, 1.7519e-03, 5.0507e-03, 7.0648e-03,
          4.3983e-03, 6.0501e-03, 3.7422e-03, 7.4959e-03, 2.1545e-02, 2.2583e-02,
          2.8191e-03, 9.0551e-04, 1.7309e-03, 1.1158e-03, 2.1896e-03, 1.2321e-03,
          7.3957e-04, 1.3065e-03, 2.2354e-03, 1.7443e-03, 2.5291e-03, 3.7861e-03,
          4.9019e-03, 1.2131e-03, 2.2449e-03, 2.2602e-03, 3.1223e-03, 4.7951e-03,
          3.2444e-03, 1.1320e-03, 3.5877e-03, 1.1917e-02, 5.9586e-03, 1.4133e-03,
          2.2316e-03, 9.8419e-04, 1.9932e-03, 1.9236e-03, 2.3422e-03, 2.1477e-03,
          1.9703e-03, 1.1272e-03, 9.1095e-03, 4.1885e-03, 6.8321e-03, 2.7046e-03,
          1.4277e-03, 5.2948e-03, 5.4588e-03, 5.9662e-03, 1.0422e-02, 3.2005e-03,
          3.2043e-03, 2.8439e-03, 4.0703e-03, 4.2114e-03, 4.2534e-03, 3.6106e-03,
          3.8967e-03, 2.2125e-03, 2.0199e-03, 2.5158e-03, 1.3924e-03, 2.8820e-03,
          2.8801e-03, 2.0256e-03, 1.7338e-03, 1.6769e-02, 8.8120e-03, 2.7084e-03,
          3.8261e-03, 7.5439e-02, 5.4240e-05, 5.4240e-05, 5.4240e-05, 2.1017e-04,
          5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05, 1.7524e-04,
          1.7524e-04, 5.1355e-04, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05,
          1.7524e-04, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05,
          5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05,
          1.7524e-04, 5.4240e-05, 2.1017e-04, 5.4240e-05, 1.7524e-04, 5.4240e-05,
          1.7524e-04, 2.1017e-04, 5.4240e-05, 1.7524e-04, 5.4240e-05, 5.4240e-05,
          5.4240e-05, 1.7524e-04, 5.4240e-05, 5.4240e-05, 3.3116e-04, 3.3116e-04,
          5.4240e-05, 5.4240e-05, 1.7524e-04, 1.7524e-04, 5.4240e-05, 1.7524e-04,
          3.3116e-04, 3.3116e-04, 1.7524e-04, 5.4240e-05, 1.7524e-04, 5.4240e-05,
          3.3116e-04, 5.4240e-05, 2.1017e-04, 1.7524e-04, 5.4240e-05, 5.4240e-05,
          1.7524e-04, 5.4240e-05, 5.4240e-05, 5.4240e-05, 3.3116e-04, 5.4240e-05,
          1.7524e-04, 2.1017e-04, 6.6948e-04, 5.4240e-05, 5.4240e-05, 1.7524e-04,
          1.7524e-04, 5.4240e-05, 3.3116e-04, 5.4240e-05, 1.7524e-04, 1.7524e-04,
          1.7524e-04, 1.7524e-04, 5.4240e-05, 5.4240e-05, 5.4240e-05, 5.4240e-05,
          1.7524e-04, 1.7524e-04, 5.1355e-04, 1.7524e-04, 1.7524e-04, 5.4240e-05,
          5.4240e-05, 1.7524e-04, 5.1355e-04, 1.7524e-04, 5.4240e-05, 5.4240e-05,
          5.4240e-05, 1.7524e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.2992e-03, 5.6549e-02, 1.2955e-02, 1.3458e-02, 3.6072e-02,
          4.0619e-02, 8.7128e-03, 6.2990e-04, 5.1498e-04, 1.6594e-03, 2.0523e-03,
          8.1587e-04, 1.1368e-03, 4.8971e-04, 1.7595e-03, 6.1274e-04, 2.3785e-03,
          9.8228e-04, 3.3545e-04, 3.4475e-04, 4.9162e-04, 3.8242e-04, 1.2712e-03,
          2.1782e-03, 1.5860e-03, 6.9275e-03, 4.0321e-03, 6.9571e-04, 2.6302e-03,
          7.4196e-04, 1.6203e-03, 9.0599e-04, 4.6997e-03, 1.4887e-03, 1.1139e-03,
          3.5057e-03, 7.4482e-04, 3.9558e-03, 3.2043e-03, 4.6825e-04, 2.2907e-03,
          6.2799e-04, 5.0688e-04, 4.1509e-04, 5.2643e-04, 2.0027e-03, 5.1498e-03,
          3.5262e-04, 4.6134e-04, 2.8019e-03, 2.4643e-03, 1.5688e-03, 9.6083e-04,
          4.4012e-04, 1.0023e-03, 8.1062e-04, 1.6012e-03, 1.1196e-03, 1.4887e-03,
          1.1473e-03, 1.3580e-03, 1.0729e-03, 2.0123e-03, 6.8998e-04, 5.3635e-03,
          5.7650e-04, 3.5787e-04, 8.5640e-04, 3.7074e-04, 2.9945e-04, 3.5310e-04,
          4.9591e-04, 4.2801e-03, 8.6308e-04, 2.1458e-03, 3.7360e-04, 4.4894e-04,
          1.1950e-03, 1.8854e-03, 8.6498e-04, 2.5520e-03, 7.9918e-04, 7.5102e-04,
          7.4863e-04, 2.8172e-03, 6.6423e-04, 9.8038e-04, 4.1747e-04, 4.1151e-04,
          1.8358e-03, 1.9855e-03, 3.2997e-04, 1.2093e-03, 5.4121e-04, 5.4550e-04,
          8.5306e-04, 5.1689e-04, 1.9608e-03, 5.6686e-03, 1.7605e-03, 2.4700e-03,
          2.5730e-03, 4.8423e-04, 5.5361e-04, 4.5547e-03, 4.0398e-03, 1.4687e-03,
          2.2278e-03, 1.2331e-03, 2.8381e-03, 4.0984e-04, 5.0640e-04, 8.4877e-04,
          1.0481e-03, 2.3556e-03, 5.8699e-04, 5.6648e-03, 7.6914e-04, 1.4687e-03,
          2.1667e-03, 7.5607e-03, 6.1464e-04, 4.5061e-04, 4.2953e-03, 5.1594e-04,
          1.5879e-03, 6.2990e-04, 8.9264e-04, 6.4392e-03, 1.8253e-03, 2.4643e-03,
          8.5163e-04, 4.6778e-04, 3.4714e-04, 3.1614e-04, 1.2836e-03, 4.5991e-04,
          6.0005e-03, 8.4400e-04, 1.0300e-03, 2.0943e-03, 3.8338e-03, 2.2087e-03,
          1.0948e-03, 6.4850e-04, 1.2341e-03, 2.0752e-03, 8.7404e-04, 6.1646e-03,
          7.7553e-03, 5.5199e-03, 2.3918e-03, 4.5509e-03, 1.9178e-03, 4.5547e-03,
          2.4109e-03, 2.6531e-03, 3.8981e-04, 4.5800e-04, 7.2441e-03, 1.8406e-03,
          5.0049e-03, 6.2256e-03, 3.5133e-03, 3.7594e-03, 4.3373e-03, 2.9640e-03,
          2.1973e-03, 3.0479e-03, 3.4332e-03, 1.8120e-03, 2.4529e-03, 1.0056e-02,
          2.8152e-03, 1.4286e-03, 2.9240e-03, 2.2774e-03, 4.6806e-03, 1.6994e-03,
          1.8234e-03, 2.1477e-03, 3.9673e-03, 1.4019e-03, 9.8801e-04, 1.4381e-03,
          1.2035e-03, 8.6641e-04, 1.3294e-03, 2.2736e-03, 1.0052e-03, 4.0855e-03,
          1.8330e-03, 1.1806e-03, 5.1956e-03, 4.4594e-03, 2.4490e-03, 5.7907e-03,
          2.4223e-03, 2.3651e-03, 6.4583e-03, 1.2321e-03, 1.3580e-03, 2.3479e-03,
          6.1846e-04, 3.1776e-03, 1.0099e-03, 6.5756e-04, 2.5768e-03, 1.7738e-03,
          7.8392e-04, 5.5351e-03, 1.8902e-03, 1.1091e-03, 6.9695e-03, 1.1797e-03,
          1.0445e-02, 4.5319e-03, 2.4700e-03, 1.3256e-03, 1.0864e-02, 2.8210e-03,
          1.4168e-02, 1.2255e-03, 3.0632e-03, 1.1730e-03, 6.4888e-03, 3.2425e-03,
          7.4625e-04, 5.4741e-04, 6.5756e-04, 1.5335e-03, 3.4370e-03, 1.1072e-03,
          2.8782e-03, 2.0580e-03, 1.0424e-03, 2.5654e-03, 9.8267e-03, 8.3542e-03,
          3.0842e-03, 2.7332e-03, 3.5095e-03, 1.0399e-02, 9.7504e-03, 3.0041e-03,
          3.8147e-03, 1.0929e-03, 8.2588e-04, 1.0376e-03, 1.0490e-03, 4.1542e-03,
          6.7377e-04, 7.0190e-04, 1.0929e-03, 3.1891e-03, 3.3703e-03, 7.3929e-03,
          5.6496e-03, 2.1152e-03, 4.1618e-03, 4.2915e-03, 2.0180e-03, 4.8180e-03,
          6.5155e-03, 2.0866e-03, 1.2760e-03, 6.5155e-03, 3.4580e-03, 3.6030e-03,
          1.4219e-03, 1.1320e-03, 5.1117e-03, 1.9646e-03, 1.1482e-02, 4.4327e-03,
          2.3918e-03, 1.8816e-03, 1.1482e-03, 1.6222e-03, 2.8496e-03, 2.1114e-03,
          1.3838e-03, 1.7376e-03, 1.8244e-03, 2.5444e-03, 9.7122e-03, 8.7967e-03,
          4.3373e-03, 1.1642e-02, 4.5815e-03, 2.9049e-03, 1.9073e-03, 2.7828e-03,
          2.8133e-03, 1.2426e-03, 9.9659e-04, 7.3814e-03, 9.5367e-03, 1.3847e-03,
          1.0977e-03, 1.7986e-03, 1.9121e-03, 9.8419e-03, 2.6131e-03, 7.3395e-03,
          9.9030e-03, 5.2834e-03, 2.8214e-02, 2.9087e-05, 2.9087e-05, 2.9087e-05,
          2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05,
          2.9087e-05, 4.0603e-04, 2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05,
          1.3280e-04, 2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05, 1.3280e-04,
          2.9087e-05, 2.9087e-05, 1.3280e-04, 2.9087e-05, 2.9087e-05, 2.9087e-05,
          2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05, 1.3280e-04, 1.3280e-04,
          2.9087e-05, 2.9087e-05, 1.3280e-04, 1.3280e-04, 2.9087e-05, 4.9639e-04,
          4.9639e-04, 1.3280e-04, 2.9087e-05, 2.9087e-05, 3.1090e-04, 3.1090e-04,
          2.9087e-05, 1.3280e-04, 1.3280e-04, 1.3280e-04, 3.1090e-04, 2.9087e-05,
          3.1090e-04, 3.1090e-04, 2.9087e-05, 2.9087e-05, 1.3280e-04, 2.9087e-05,
          2.9087e-05, 4.9639e-04, 2.9087e-05, 1.3280e-04, 2.9087e-05, 2.9087e-05,
          1.3280e-04, 2.9087e-05, 2.9087e-05, 2.9087e-05, 3.1090e-04, 2.9087e-05,
          1.3280e-04, 3.1090e-04, 3.1090e-04, 2.9087e-05, 2.9087e-05, 1.3280e-04,
          1.3280e-04, 2.9087e-05, 3.1090e-04, 2.9087e-05, 1.3280e-04, 1.3280e-04,
          4.9639e-04, 1.3280e-04, 2.9087e-05, 2.9087e-05, 2.9087e-05, 2.9087e-05,
          4.0603e-04, 4.0603e-04, 4.0603e-04, 1.3280e-04, 1.3280e-04, 2.9087e-05,
          2.9087e-05, 1.3280e-04, 4.0603e-04, 1.3280e-04, 1.3280e-04, 4.0603e-04,
          1.3280e-04, 1.3280e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4708e-03, 5.3772e-02, 1.6068e-02, 1.5701e-02, 3.0853e-02,
          3.4576e-02, 9.0027e-03, 1.6270e-03, 8.6164e-04, 1.1902e-03, 2.2774e-03,
          8.0252e-04, 2.6455e-03, 2.3994e-03, 1.2188e-03, 1.6060e-03, 6.6605e-03,
          3.1834e-03, 7.2575e-04, 4.0436e-04, 3.6669e-04, 5.8079e-04, 3.2063e-03,
          2.5635e-03, 9.6273e-04, 7.3433e-03, 5.6725e-03, 8.5258e-04, 5.5733e-03,
          5.7373e-03, 4.1733e-03, 1.7061e-03, 2.1000e-03, 1.2674e-03, 1.6413e-03,
          1.0891e-03, 3.8052e-04, 1.2999e-03, 1.7462e-03, 6.5374e-04, 5.8794e-04,
          3.1328e-04, 2.3651e-04, 2.9397e-04, 2.2519e-04, 6.8665e-03, 1.6146e-03,
          1.3180e-03, 8.2874e-04, 2.3098e-03, 3.3951e-03, 3.3150e-03, 1.4238e-03,
          6.3419e-04, 1.9283e-03, 7.9441e-04, 3.7694e-04, 1.0939e-03, 5.1594e-04,
          6.4421e-04, 2.0962e-03, 1.0099e-03, 1.6413e-03, 3.2377e-04, 1.7672e-03,
          5.9175e-04, 3.6955e-04, 5.1546e-04, 1.5736e-04, 1.3590e-04, 2.4939e-04,
          3.1471e-04, 3.2101e-03, 9.7132e-04, 5.7335e-03, 5.8174e-04, 6.6328e-04,
          2.5024e-03, 3.7026e-04, 4.2200e-04, 7.9775e-04, 4.4870e-04, 3.3402e-04,
          3.1257e-04, 4.1389e-03, 7.6485e-04, 9.7942e-04, 3.1471e-04, 3.0613e-04,
          1.9779e-03, 1.0796e-03, 2.1064e-04, 1.1415e-03, 3.0184e-04, 2.8181e-04,
          2.3022e-03, 5.0354e-04, 3.4466e-03, 2.2392e-03, 3.7551e-04, 4.9353e-04,
          2.2411e-03, 6.0749e-04, 4.7636e-04, 7.1526e-04, 1.6413e-03, 1.9970e-03,
          3.3932e-03, 3.1924e-04, 1.2684e-03, 2.7561e-04, 5.0831e-04, 5.0974e-04,
          6.7663e-04, 4.2224e-04, 8.8644e-04, 2.2507e-03, 4.0126e-04, 2.7523e-03,
          7.2784e-03, 1.3533e-03, 2.6560e-04, 1.0948e-03, 2.1629e-03, 2.3818e-04,
          2.8872e-04, 3.1137e-04, 4.9210e-04, 2.3861e-03, 3.4122e-03, 4.9019e-03,
          8.7929e-04, 4.2248e-04, 5.4884e-04, 3.1400e-04, 7.5483e-04, 2.3794e-04,
          9.8267e-03, 5.0201e-03, 6.0959e-03, 4.7379e-03, 5.1422e-03, 9.8495e-03,
          4.0317e-04, 6.7663e-04, 4.5991e-04, 4.4942e-04, 5.4646e-04, 1.8015e-03,
          2.0943e-03, 1.7099e-03, 5.9080e-04, 2.6360e-03, 2.1324e-03, 7.0906e-04,
          9.3842e-04, 5.8222e-04, 7.5865e-04, 3.1304e-04, 1.4534e-03, 6.9466e-03,
          2.3441e-03, 2.1858e-03, 1.3695e-02, 3.4676e-03, 1.0128e-03, 7.5531e-04,
          1.8415e-03, 9.6893e-04, 1.2484e-03, 1.3800e-03, 5.6267e-03, 2.7313e-03,
          1.5488e-03, 1.1463e-03, 8.7929e-04, 6.2990e-04, 1.1463e-03, 3.1796e-03,
          9.9564e-04, 1.3494e-03, 1.8511e-03, 4.0207e-03, 2.9564e-03, 2.1687e-03,
          2.0466e-03, 9.3231e-03, 2.2888e-03, 6.9904e-04, 6.9571e-04, 2.6970e-03,
          5.1308e-03, 3.1209e-04, 2.1706e-03, 1.6922e-02, 1.9951e-03, 2.0409e-03,
          1.1282e-03, 1.8778e-03, 2.0123e-03, 5.8317e-04, 1.3409e-03, 1.1511e-03,
          3.9029e-04, 7.7915e-04, 3.5357e-04, 1.3590e-03, 2.7409e-03, 4.2953e-03,
          2.3365e-03, 2.1324e-03, 7.0906e-04, 1.2121e-03, 3.0727e-03, 6.1846e-04,
          1.9255e-03, 7.4806e-03, 3.2005e-03, 1.6136e-03, 5.4665e-03, 1.4267e-02,
          6.8703e-03, 3.9744e-04, 7.6675e-04, 9.7752e-04, 3.5839e-03, 2.6340e-03,
          6.2990e-04, 4.4703e-04, 3.3712e-04, 1.0538e-03, 1.5135e-03, 6.0120e-03,
          1.2379e-03, 1.1425e-03, 8.9741e-04, 1.8263e-03, 2.3422e-03, 5.1537e-03,
          2.0103e-03, 1.4238e-03, 9.7752e-04, 3.8548e-03, 1.9791e-02, 5.5580e-03,
          6.8188e-04, 1.1692e-03, 4.0746e-04, 7.5054e-04, 6.5231e-04, 1.3342e-03,
          3.7193e-04, 2.6822e-04, 5.1785e-04, 8.3303e-04, 6.8626e-03, 8.4457e-03,
          1.4582e-03, 1.6451e-03, 1.5059e-03, 1.7958e-03, 9.4891e-04, 3.0327e-03,
          2.8973e-03, 2.2526e-03, 7.5951e-03, 3.9490e-02, 3.5492e-02, 1.2722e-03,
          1.3409e-03, 9.6273e-04, 2.2869e-03, 6.5327e-04, 2.4643e-03, 1.7519e-03,
          1.2083e-03, 2.2297e-03, 1.7347e-03, 2.1164e-02, 1.3227e-03, 1.6766e-03,
          1.6737e-03, 9.8705e-04, 7.1049e-04, 1.0757e-03, 1.3214e-02, 5.1537e-03,
          3.6144e-03, 6.1111e-03, 1.2413e-02, 1.5099e-02, 8.2111e-04, 1.6041e-03,
          9.3699e-04, 5.5838e-04, 4.3559e-04, 3.5076e-03, 2.0828e-03, 8.2684e-04,
          6.0892e-04, 6.4964e-03, 5.2338e-03, 2.8961e-02, 1.1559e-03, 4.0207e-03,
          4.3869e-03, 3.5725e-03, 5.0583e-03, 2.9785e-02, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.8334e-04,
          1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 1.8334e-04, 3.0279e-05, 3.0279e-05, 1.8334e-04,
          3.0279e-05, 3.0279e-05, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.8334e-04, 1.8334e-04,
          3.0279e-05, 3.0279e-05, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 1.8334e-04, 1.8334e-04,
          3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 1.8334e-04, 1.8334e-04,
          1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 1.8334e-04, 3.0279e-05,
          1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.8334e-04, 1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.8334e-04, 3.0279e-05, 1.8334e-04, 3.0279e-05, 1.8334e-04, 3.0279e-05,
          1.8334e-04, 1.8334e-04, 3.0279e-05, 3.0279e-05, 3.0279e-05, 3.0279e-05,
          1.8334e-04, 6.8760e-04, 1.8334e-04, 1.8334e-04, 6.8760e-04, 3.0279e-05,
          3.0279e-05, 1.8334e-04, 1.8334e-04, 1.8334e-04, 1.8334e-04, 1.2226e-03,
          6.8760e-04, 1.8334e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.2419e-03, 5.4626e-02, 1.3870e-02, 1.5160e-02, 4.4495e-02,
          4.8615e-02, 1.2527e-02, 2.0199e-03, 8.1158e-04, 4.3983e-03, 7.4196e-03,
          2.2564e-03, 1.5154e-03, 4.8804e-04, 4.2038e-03, 2.0275e-03, 1.1511e-03,
          7.0667e-04, 1.2636e-03, 1.6320e-04, 2.1052e-04, 2.0194e-04, 2.0599e-03,
          7.9269e-03, 3.2539e-03, 9.9106e-03, 3.6888e-03, 3.0780e-04, 1.2312e-03,
          5.8508e-04, 3.4485e-03, 3.7622e-04, 2.5482e-03, 5.9357e-03, 7.7105e-04,
          1.5240e-03, 4.7493e-04, 1.7204e-03, 2.5482e-03, 2.2185e-04, 1.1339e-03,
          4.2892e-04, 3.7289e-04, 3.6573e-04, 3.7766e-04, 1.7195e-03, 2.1057e-03,
          2.4891e-04, 2.8968e-04, 2.9583e-03, 9.0694e-04, 1.6232e-03, 1.1044e-03,
          3.7217e-04, 8.7690e-04, 3.7408e-04, 4.7088e-04, 1.0452e-03, 7.4053e-04,
          7.9489e-04, 4.8494e-04, 6.2418e-04, 6.9761e-04, 6.0320e-04, 4.7188e-03,
          8.2302e-04, 7.9250e-04, 1.0624e-03, 2.1768e-04, 2.0623e-04, 3.2473e-04,
          3.5310e-04, 6.9695e-03, 4.0650e-04, 2.4548e-03, 3.5882e-04, 3.4022e-04,
          9.1028e-04, 4.3535e-04, 6.0463e-04, 2.6131e-03, 7.6866e-04, 6.4945e-04,
          5.1451e-04, 2.9259e-03, 4.2725e-04, 4.1733e-03, 5.4979e-04, 4.6158e-04,
          5.9052e-03, 2.0561e-03, 3.2449e-04, 2.7351e-03, 3.1424e-04, 3.3116e-04,
          4.0317e-04, 2.7895e-04, 1.4153e-03, 1.2741e-03, 4.3511e-04, 6.7234e-04,
          6.5231e-03, 1.3790e-03, 1.0242e-03, 1.5125e-03, 5.6915e-03, 4.3249e-04,
          2.5063e-03, 3.7432e-04, 2.3899e-03, 3.7885e-04, 6.8808e-04, 7.3385e-04,
          3.1395e-03, 5.1117e-04, 2.8205e-04, 4.4098e-03, 4.5562e-04, 6.1321e-04,
          6.9714e-04, 2.3689e-03, 3.2640e-04, 2.0161e-03, 1.3748e-02, 3.0303e-04,
          3.4952e-04, 3.6883e-04, 5.5838e-04, 4.3259e-03, 2.9144e-03, 3.9444e-03,
          1.1168e-03, 8.8501e-04, 1.2543e-02, 6.2513e-04, 1.4658e-03, 3.3832e-04,
          2.8248e-03, 5.0211e-04, 4.0207e-03, 7.0801e-03, 4.2343e-03, 9.3222e-04,
          4.7398e-04, 7.3318e-03, 7.5340e-04, 5.6839e-04, 6.7377e-04, 3.7060e-03,
          5.2605e-03, 3.2177e-03, 7.1144e-04, 1.0433e-03, 1.1415e-03, 8.9836e-04,
          1.4296e-03, 6.9046e-04, 3.8013e-03, 3.9887e-04, 1.9779e-03, 7.1764e-04,
          3.1662e-03, 2.9697e-03, 3.1567e-03, 1.2039e-02, 8.5068e-04, 8.2541e-04,
          4.4327e-03, 1.1263e-03, 2.6112e-03, 6.1302e-03, 1.2741e-03, 2.4338e-03,
          2.0313e-03, 1.2045e-03, 1.2121e-03, 6.6376e-04, 2.8286e-03, 1.1909e-02,
          2.2736e-03, 2.9869e-03, 7.8125e-03, 5.7983e-03, 5.7335e-03, 7.1297e-03,
          8.6288e-03, 1.1148e-03, 1.1833e-02, 7.3290e-04, 7.4816e-04, 7.5569e-03,
          5.7564e-03, 3.5810e-04, 3.5419e-03, 3.3722e-03, 5.9891e-03, 2.9697e-03,
          1.3447e-03, 6.0225e-04, 3.3226e-03, 8.5163e-04, 2.2068e-03, 2.0390e-03,
          5.3072e-04, 1.2245e-03, 4.6611e-04, 7.2021e-03, 5.4054e-03, 1.0710e-03,
          3.2330e-03, 1.2598e-03, 6.4993e-04, 1.9817e-03, 4.0817e-03, 6.4278e-04,
          1.7614e-03, 1.3056e-03, 1.3885e-02, 8.8882e-03, 4.7188e-03, 1.2102e-03,
          2.7027e-03, 4.9496e-04, 1.4296e-03, 1.4143e-03, 1.8494e-02, 7.0648e-03,
          1.1225e-03, 6.9427e-04, 5.5265e-04, 2.5921e-03, 2.7676e-03, 8.0681e-04,
          1.0195e-03, 1.0824e-03, 8.0872e-04, 5.7983e-03, 2.0809e-03, 3.7899e-03,
          1.4429e-03, 1.2016e-03, 8.2445e-04, 1.4343e-03, 1.7929e-03, 9.0408e-04,
          6.9618e-04, 1.5745e-03, 5.8889e-04, 9.6083e-04, 1.0929e-03, 3.6869e-03,
          3.9220e-04, 4.1699e-04, 7.6103e-04, 9.8801e-04, 6.1836e-03, 6.9199e-03,
          1.0109e-03, 1.3676e-03, 2.1172e-03, 2.3861e-03, 7.3290e-04, 2.0847e-03,
          2.8057e-03, 1.0681e-02, 4.3831e-03, 3.6259e-03, 4.8370e-03, 2.4319e-03,
          1.3437e-03, 1.4887e-03, 8.0261e-03, 1.0395e-03, 2.1687e-03, 3.4466e-03,
          1.9703e-03, 3.9406e-03, 5.4789e-04, 1.3924e-03, 9.4938e-04, 1.1864e-03,
          1.7090e-03, 7.8821e-04, 5.7030e-04, 8.9025e-04, 2.0332e-03, 3.5706e-03,
          3.4447e-03, 8.3389e-03, 3.6488e-03, 5.8670e-03, 9.6178e-04, 1.3905e-03,
          1.2598e-03, 7.4434e-04, 6.2943e-04, 5.6038e-03, 2.2144e-03, 1.4400e-03,
          9.2554e-04, 4.8599e-03, 6.6795e-03, 2.8725e-03, 8.5878e-04, 5.8670e-03,
          5.2414e-03, 1.0786e-03, 3.1757e-03, 2.6760e-03, 6.3843e-02, 4.6611e-05,
          1.5855e-04, 1.5855e-04, 4.6611e-05, 4.6611e-05, 4.6611e-05, 4.6611e-05,
          4.6611e-05, 4.6611e-05, 1.5855e-04, 1.5855e-04, 4.6611e-05, 4.6611e-05,
          1.5855e-04, 1.5855e-04, 1.5855e-04, 4.6611e-05, 4.6611e-05, 4.0722e-04,
          4.6611e-05, 1.5855e-04, 4.0722e-04, 4.6611e-05, 4.6611e-05, 4.6611e-05,
          1.5855e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05, 4.0722e-04,
          4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04,
          1.5855e-04, 4.0722e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04, 1.5855e-04,
          4.6611e-05, 4.6611e-05, 4.6611e-05, 4.6611e-05, 4.0722e-04, 4.6611e-05,
          1.5855e-04, 1.5855e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05,
          4.6611e-05, 1.5855e-04, 4.6611e-05, 4.6611e-05, 4.6611e-05, 4.6611e-05,
          4.6611e-05, 4.6611e-05, 4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05,
          1.5855e-04, 4.0722e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05,
          1.5855e-04, 4.6611e-05, 4.0722e-04, 4.6611e-05, 4.6611e-05, 1.5855e-04,
          1.5855e-04, 1.5855e-04, 4.6611e-05, 1.5855e-04, 4.6611e-05, 4.6611e-05,
          1.5855e-04, 4.6611e-05, 4.6611e-05, 4.6611e-05, 1.5855e-04, 4.6611e-05,
          4.6611e-05, 1.5855e-04, 1.5855e-04, 1.5855e-04, 1.5855e-04, 1.5855e-04,
          1.5855e-04, 4.6611e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4823e-03, 4.6448e-02, 1.3168e-02, 1.2802e-02, 4.2847e-02,
          4.4434e-02, 1.0971e-02, 2.4624e-03, 1.2035e-03, 5.5466e-03, 1.7120e-02,
          3.6278e-03, 2.6798e-03, 5.8031e-04, 3.3035e-03, 2.5425e-03, 1.6928e-03,
          1.1415e-03, 1.7338e-03, 1.5759e-04, 2.1100e-04, 2.3282e-04, 2.0771e-03,
          1.7212e-02, 2.3365e-03, 8.1863e-03, 3.1204e-03, 3.0899e-04, 1.9684e-03,
          7.2193e-04, 4.4441e-03, 5.6791e-04, 2.0866e-03, 4.1695e-03, 7.2384e-04,
          1.2751e-03, 5.2023e-04, 1.1978e-03, 2.1248e-03, 2.2519e-04, 9.2745e-04,
          4.2200e-04, 3.5357e-04, 3.9005e-04, 3.7169e-04, 2.7523e-03, 1.5297e-03,
          2.8205e-04, 2.7680e-04, 3.7346e-03, 1.0147e-03, 2.3098e-03, 1.1406e-03,
          3.7122e-04, 1.1663e-03, 3.7408e-04, 5.3978e-04, 1.4181e-03, 8.9836e-04,
          9.6607e-04, 5.4550e-04, 5.8413e-04, 6.6376e-04, 7.3528e-04, 3.0727e-03,
          1.5440e-03, 1.2655e-03, 9.7656e-04, 2.2340e-04, 2.0766e-04, 3.4499e-04,
          3.3975e-04, 8.0185e-03, 4.9400e-04, 3.4676e-03, 3.8981e-04, 3.7265e-04,
          9.6416e-04, 5.0402e-04, 8.5545e-04, 2.3842e-03, 1.1902e-03, 5.5027e-04,
          4.3035e-04, 2.8934e-03, 4.2772e-04, 7.8735e-03, 8.3590e-04, 6.4707e-04,
          1.0841e-02, 2.2793e-03, 3.5310e-04, 4.4250e-03, 2.8515e-04, 2.7299e-04,
          4.5562e-04, 2.8992e-04, 1.4582e-03, 9.3651e-04, 4.7779e-04, 8.3447e-04,
          1.1650e-02, 1.7195e-03, 1.5726e-03, 9.9945e-04, 3.2234e-03, 4.7827e-04,
          2.2812e-03, 3.4904e-04, 2.8114e-03, 4.6182e-04, 1.2445e-03, 8.6164e-04,
          2.2144e-03, 4.6396e-04, 2.6226e-04, 3.2940e-03, 3.9577e-04, 6.5088e-04,
          8.5068e-04, 2.2068e-03, 3.9601e-04, 2.4643e-03, 5.4436e-03, 3.7289e-04,
          3.2544e-04, 4.5037e-04, 7.5769e-04, 3.2635e-03, 3.5839e-03, 3.2997e-03,
          2.2717e-03, 1.3809e-03, 7.1983e-03, 5.9319e-04, 1.7843e-03, 3.4761e-04,
          2.9716e-03, 5.4646e-04, 5.3558e-03, 5.6572e-03, 3.5858e-03, 1.1606e-03,
          5.7459e-04, 9.3231e-03, 7.2050e-04, 6.3753e-04, 1.0138e-03, 2.7180e-03,
          4.7112e-03, 3.0117e-03, 6.7282e-04, 8.8024e-04, 9.5749e-04, 6.6519e-04,
          1.3676e-03, 5.1212e-04, 1.5076e-02, 5.1737e-04, 1.4715e-03, 8.6784e-04,
          2.4338e-03, 1.6909e-03, 4.6005e-03, 5.8250e-03, 6.2609e-04, 6.6137e-04,
          9.9030e-03, 1.0357e-03, 2.1343e-03, 5.2147e-03, 1.6956e-03, 2.1915e-03,
          2.3117e-03, 1.1892e-03, 9.0933e-04, 6.2418e-04, 1.7204e-03, 9.4604e-03,
          1.4238e-03, 2.3193e-03, 3.2997e-03, 8.2169e-03, 4.4746e-03, 4.9820e-03,
          2.2095e-02, 1.3113e-03, 9.0408e-03, 7.9727e-04, 1.0567e-03, 4.8904e-03,
          8.2397e-03, 3.7003e-04, 2.8381e-03, 5.0278e-03, 8.7509e-03, 2.9678e-03,
          1.0204e-03, 5.8985e-04, 2.5406e-03, 9.9754e-04, 2.6703e-03, 1.3924e-03,
          7.4482e-04, 1.3609e-03, 4.9686e-04, 5.5428e-03, 9.6054e-03, 1.3781e-03,
          4.0855e-03, 9.8515e-04, 6.6042e-04, 3.2616e-03, 4.2610e-03, 8.2159e-04,
          1.3714e-03, 1.4820e-03, 1.1963e-02, 7.0534e-03, 3.7212e-03, 1.9932e-03,
          2.1458e-03, 5.0211e-04, 1.0929e-03, 1.5287e-03, 6.8321e-03, 4.8790e-03,
          1.8864e-03, 1.1425e-03, 5.9986e-04, 2.3136e-03, 2.1286e-03, 1.0567e-03,
          1.0843e-03, 1.0977e-03, 1.2341e-03, 4.4060e-03, 2.0866e-03, 5.4321e-03,
          1.9684e-03, 1.2445e-03, 6.9857e-04, 1.1511e-03, 2.3727e-03, 9.5034e-04,
          5.8937e-04, 2.5368e-03, 6.0987e-04, 1.1501e-03, 1.5717e-03, 2.2526e-03,
          5.5361e-04, 3.5381e-04, 5.6696e-04, 7.6246e-04, 7.1678e-03, 5.0659e-03,
          9.4604e-04, 1.7004e-03, 1.5049e-03, 1.9665e-03, 7.7105e-04, 2.5234e-03,
          2.5349e-03, 7.2174e-03, 5.0049e-03, 5.1231e-03, 6.5727e-03, 1.5411e-03,
          2.4948e-03, 2.3689e-03, 3.7460e-03, 8.4782e-04, 1.3714e-03, 2.5368e-03,
          1.7948e-03, 3.2082e-03, 4.7755e-04, 1.7195e-03, 7.6151e-04, 1.7157e-03,
          3.8452e-03, 9.0599e-04, 5.3024e-04, 8.5402e-04, 2.0332e-03, 2.3041e-03,
          4.6310e-03, 6.4087e-03, 4.6158e-03, 8.2321e-03, 8.7357e-04, 1.5211e-03,
          9.8133e-04, 7.1144e-04, 5.2071e-04, 3.9005e-03, 1.7595e-03, 9.5034e-04,
          6.5184e-04, 5.5504e-03, 8.4076e-03, 2.9240e-03, 9.9659e-04, 6.2370e-03,
          4.4441e-03, 8.7881e-04, 2.2202e-03, 2.4433e-03, 1.0643e-02, 4.5135e-02,
          4.7207e-05, 4.7207e-05, 4.7207e-05, 1.7166e-04, 4.7207e-05, 4.7207e-05,
          4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05,
          4.7207e-05, 4.7207e-05, 2.3663e-04, 4.7207e-05, 4.7207e-05, 1.7166e-04,
          4.7207e-05, 1.7166e-04, 3.6097e-04, 1.7166e-04, 4.7207e-05, 4.7207e-05,
          4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05, 3.6097e-04,
          4.7207e-05, 4.7207e-05, 3.6097e-04, 4.7207e-05, 4.0841e-04, 4.7207e-05,
          5.6124e-04, 3.6097e-04, 4.7207e-05, 4.7207e-05, 3.6097e-04, 1.7166e-04,
          4.0841e-04, 4.7207e-05, 4.7207e-05, 4.7207e-05, 3.6097e-04, 4.0841e-04,
          3.6097e-04, 3.6097e-04, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05,
          4.7207e-05, 2.3663e-04, 4.7207e-05, 4.0841e-04, 4.7207e-05, 4.7207e-05,
          4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05, 1.7166e-04, 4.7207e-05,
          3.6097e-04, 1.7166e-04, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05,
          1.7166e-04, 1.0853e-03, 3.6097e-04, 4.7207e-05, 4.7207e-05, 4.0841e-04,
          1.2751e-03, 1.9083e-03, 4.7207e-05, 4.7207e-05, 4.7207e-05, 4.7207e-05,
          2.3663e-04, 4.7207e-05, 4.7207e-05, 4.7207e-05, 2.3663e-04, 1.7166e-04,
          4.7207e-05, 2.3663e-04, 4.7207e-05, 3.4499e-04, 2.3663e-04, 5.3453e-04,
          2.3663e-04, 2.3663e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.3945e-03, 4.8523e-02, 1.5686e-02, 1.4091e-02, 3.0472e-02,
          3.0350e-02, 8.6288e-03, 9.5892e-04, 5.7268e-04, 1.2131e-03, 2.1458e-03,
          7.8297e-04, 2.9678e-03, 1.7385e-03, 1.2722e-03, 1.0452e-03, 7.2594e-03,
          3.9482e-03, 5.1069e-04, 4.2701e-04, 3.7599e-04, 6.7282e-04, 2.7714e-03,
          2.4948e-03, 9.0981e-04, 7.0000e-03, 5.2528e-03, 7.8297e-04, 7.2899e-03,
          3.6201e-03, 2.5749e-03, 2.2354e-03, 2.5024e-03, 1.2293e-03, 1.0843e-03,
          1.0214e-03, 3.5548e-04, 1.4563e-03, 1.6136e-03, 7.1096e-04, 7.3719e-04,
          3.4213e-04, 2.5034e-04, 3.1805e-04, 2.5225e-04, 7.1640e-03, 2.2736e-03,
          1.1749e-03, 7.5102e-04, 2.4052e-03, 1.4973e-03, 1.8234e-03, 9.7084e-04,
          3.4595e-04, 1.0328e-03, 5.1689e-04, 3.7146e-04, 7.4482e-04, 5.1403e-04,
          6.1655e-04, 3.5610e-03, 6.9523e-04, 1.0204e-03, 3.8028e-04, 2.3746e-03,
          6.0511e-04, 3.5095e-04, 5.7364e-04, 1.6725e-04, 1.5831e-04, 2.6131e-04,
          3.3331e-04, 3.5629e-03, 5.8126e-04, 2.9507e-03, 3.8862e-04, 3.5810e-04,
          1.1587e-03, 3.8385e-04, 4.4370e-04, 8.9884e-04, 4.3440e-04, 3.3736e-04,
          3.5143e-04, 2.9259e-03, 5.5122e-04, 1.0176e-03, 3.1042e-04, 3.1471e-04,
          2.0809e-03, 1.4725e-03, 2.1267e-04, 1.1959e-03, 2.6131e-04, 3.1257e-04,
          1.3962e-03, 3.3426e-04, 2.2755e-03, 2.0065e-03, 3.5524e-04, 5.1928e-04,
          2.3193e-03, 4.5800e-04, 4.5729e-04, 7.0906e-04, 1.7939e-03, 3.7880e-03,
          2.2755e-03, 3.4356e-04, 1.6699e-03, 2.6226e-04, 4.5967e-04, 5.0449e-04,
          6.6662e-04, 4.8113e-04, 1.2894e-03, 2.8954e-03, 4.8566e-04, 1.9569e-03,
          6.4125e-03, 1.5097e-03, 2.4486e-04, 6.9189e-04, 2.3136e-03, 2.1434e-04,
          3.1114e-04, 3.1781e-04, 4.6873e-04, 2.5654e-03, 2.9221e-03, 3.0270e-03,
          7.9393e-04, 4.2391e-04, 4.5753e-04, 2.9564e-04, 9.0361e-04, 2.5129e-04,
          1.5671e-02, 4.7684e-03, 5.8746e-03, 4.2076e-03, 3.9825e-03, 9.4376e-03,
          3.6407e-04, 4.3917e-04, 4.4560e-04, 4.3726e-04, 5.2786e-04, 1.7729e-03,
          2.3479e-03, 1.7691e-03, 5.9509e-04, 2.1172e-03, 1.5898e-03, 9.1267e-04,
          1.1902e-03, 6.9475e-04, 6.3133e-04, 3.2568e-04, 2.0065e-03, 1.1887e-02,
          3.5992e-03, 2.4719e-03, 1.5884e-02, 3.0231e-03, 8.9979e-04, 6.1226e-04,
          1.7128e-03, 8.4782e-04, 1.3523e-03, 1.2817e-03, 7.3013e-03, 3.0327e-03,
          1.2131e-03, 7.9298e-04, 1.0824e-03, 8.0347e-04, 1.8682e-03, 3.0537e-03,
          1.0319e-03, 1.3819e-03, 2.3575e-03, 3.1433e-03, 1.9646e-03, 1.8253e-03,
          1.8749e-03, 7.5264e-03, 1.7605e-03, 6.2323e-04, 5.9891e-04, 2.6512e-03,
          3.5000e-03, 3.1257e-04, 2.8229e-03, 1.9684e-02, 1.9140e-03, 2.1839e-03,
          1.1244e-03, 2.7771e-03, 2.5082e-03, 7.3195e-04, 1.2636e-03, 1.3399e-03,
          4.2892e-04, 1.1721e-03, 4.5633e-04, 9.2411e-04, 2.7943e-03, 6.6299e-03,
          1.4887e-03, 1.8034e-03, 6.2943e-04, 9.5606e-04, 3.6030e-03, 5.4979e-04,
          1.6985e-03, 1.1414e-02, 2.6951e-03, 1.3323e-03, 7.0534e-03, 1.2947e-02,
          7.7133e-03, 4.9639e-04, 1.2074e-03, 8.9121e-04, 4.4250e-03, 2.7084e-03,
          6.9714e-04, 3.9291e-04, 4.1556e-04, 9.6416e-04, 1.7223e-03, 5.7487e-03,
          1.1311e-03, 8.4019e-04, 5.8556e-04, 1.2865e-03, 2.2964e-03, 4.1962e-03,
          1.5526e-03, 1.2894e-03, 8.7118e-04, 3.6659e-03, 2.1606e-02, 8.0719e-03,
          9.8324e-04, 1.1530e-03, 4.6396e-04, 7.7629e-04, 6.7616e-04, 1.8034e-03,
          3.8052e-04, 3.5357e-04, 6.4898e-04, 1.2169e-03, 8.5602e-03, 7.0839e-03,
          1.3466e-03, 1.1930e-03, 1.1168e-03, 1.5230e-03, 7.0477e-04, 2.3232e-03,
          2.9221e-03, 1.7033e-03, 6.2180e-03, 4.7791e-02, 2.2583e-02, 1.6022e-03,
          1.1892e-03, 1.0309e-03, 2.8553e-03, 9.1887e-04, 3.1738e-03, 2.1667e-03,
          1.3533e-03, 1.5926e-03, 2.2907e-03, 1.6876e-02, 1.1330e-03, 1.2217e-03,
          1.2836e-03, 8.5497e-04, 6.5422e-04, 9.4080e-04, 9.8648e-03, 3.6240e-03,
          2.5387e-03, 6.9771e-03, 2.0157e-02, 1.2016e-02, 9.9850e-04, 1.6966e-03,
          1.3762e-03, 7.4673e-04, 5.9223e-04, 5.7869e-03, 2.6913e-03, 1.0691e-03,
          8.2302e-04, 4.9744e-03, 3.8128e-03, 3.6316e-02, 1.0090e-03, 4.9438e-03,
          4.6844e-03, 3.4218e-03, 4.3030e-03, 9.6359e-03, 1.5697e-03, 2.1076e-03,
          2.7084e-02, 3.2187e-05, 3.2187e-05, 1.5903e-04, 3.2187e-05, 1.5903e-04,
          1.5903e-04, 1.5903e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05,
          3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05,
          3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05,
          3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05, 1.5903e-04, 3.2187e-05,
          1.5903e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05, 9.5606e-05, 3.2187e-05,
          3.2187e-05, 1.5903e-04, 3.2187e-05, 3.2187e-05, 3.3116e-04, 3.3116e-04,
          9.5606e-05, 3.2187e-05, 2.0421e-04, 3.2187e-05, 3.2187e-05, 2.2244e-04,
          3.3116e-04, 3.3116e-04, 1.5903e-04, 3.2187e-05, 1.5903e-04, 3.2187e-05,
          1.5903e-04, 3.2187e-05, 3.2187e-05, 2.2244e-04, 3.2187e-05, 3.2187e-05,
          1.5903e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.3116e-04, 3.2187e-05,
          1.5903e-04, 3.2187e-05, 3.3116e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05,
          1.5903e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05, 1.5903e-04, 1.5903e-04,
          3.2187e-05, 1.5903e-04, 3.2187e-05, 3.2187e-05, 3.2187e-05, 3.2187e-05,
          4.7469e-04, 4.7469e-04, 4.7469e-04, 1.5903e-04, 1.5903e-04, 3.2187e-05,
          3.2187e-05, 3.3116e-04, 1.5903e-04, 5.3596e-04, 1.5903e-04, 8.5163e-04,
          4.7469e-04, 1.5903e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4060e-03, 5.1361e-02, 1.2634e-02, 9.0561e-03, 3.4058e-02,
          4.9652e-02, 7.0648e-03, 2.0943e-03, 6.0463e-04, 1.1950e-03, 2.1820e-03,
          8.4305e-04, 8.3876e-04, 3.9887e-04, 2.0580e-03, 1.9445e-03, 7.6818e-04,
          4.0984e-04, 4.3797e-04, 2.4378e-04, 2.7227e-04, 2.5487e-04, 6.6423e-04,
          2.4796e-03, 2.3766e-03, 5.6839e-03, 2.6646e-03, 4.7326e-04, 9.7275e-04,
          5.3549e-04, 5.3101e-03, 4.2129e-04, 2.0885e-03, 8.1301e-04, 1.6422e-03,
          2.5616e-03, 1.0033e-03, 8.5068e-04, 1.1911e-03, 3.6144e-04, 8.6546e-04,
          6.2704e-04, 5.1785e-04, 5.6410e-04, 3.6001e-04, 7.7343e-04, 2.3041e-03,
          2.8205e-04, 3.7599e-04, 2.5501e-03, 2.1610e-03, 2.8782e-03, 1.0357e-03,
          6.3086e-04, 1.5316e-03, 1.0271e-03, 1.3351e-03, 2.3251e-03, 2.4872e-03,
          2.0390e-03, 1.0109e-03, 1.2169e-03, 1.3180e-03, 1.0109e-03, 4.1199e-03,
          9.2459e-04, 5.3692e-04, 5.7316e-04, 4.9162e-04, 3.8767e-04, 6.1321e-04,
          5.5742e-04, 2.2240e-03, 9.3508e-04, 6.0997e-03, 7.3338e-04, 5.9462e-04,
          1.4992e-03, 1.2646e-03, 1.2493e-03, 2.1057e-03, 1.2684e-03, 2.7828e-03,
          1.6212e-03, 6.6261e-03, 8.9550e-04, 1.0462e-03, 5.2309e-04, 5.2977e-04,
          1.8625e-03, 1.2789e-03, 4.6396e-04, 1.3056e-03, 1.2302e-03, 1.2312e-03,
          6.0177e-04, 6.0034e-04, 3.7155e-03, 1.9341e-03, 1.2960e-03, 1.5450e-03,
          2.4452e-03, 8.3971e-04, 6.8903e-04, 6.4125e-03, 5.6305e-03, 1.1883e-03,
          3.6488e-03, 8.1062e-04, 1.5936e-03, 8.1730e-04, 8.7500e-04, 6.6519e-04,
          2.3975e-03, 1.0681e-03, 4.0889e-04, 5.2452e-03, 1.3247e-03, 1.0767e-03,
          1.2207e-03, 2.9678e-03, 6.7282e-04, 1.1072e-03, 4.6692e-03, 6.7997e-04,
          1.1864e-03, 1.0910e-03, 1.7023e-03, 5.5618e-03, 1.4467e-03, 3.9005e-03,
          1.5926e-03, 6.2227e-04, 7.3910e-04, 3.9768e-04, 1.0977e-03, 5.1928e-04,
          1.8559e-03, 6.1083e-04, 1.2131e-03, 2.1973e-03, 2.0752e-03, 1.3742e-03,
          1.0452e-03, 1.2894e-03, 1.4591e-03, 1.7614e-03, 1.0824e-03, 1.0452e-02,
          4.6387e-03, 6.8817e-03, 9.3613e-03, 2.3022e-03, 2.6016e-03, 1.5507e-03,
          1.2875e-03, 1.0748e-03, 6.0225e-04, 6.1703e-04, 3.9902e-03, 9.7322e-04,
          2.7714e-03, 1.9493e-03, 1.3647e-03, 2.4033e-03, 1.8148e-03, 1.6642e-03,
          1.9217e-03, 1.6146e-03, 3.8872e-03, 2.4357e-03, 1.2112e-03, 4.4670e-03,
          2.2186e-02, 7.7629e-03, 7.8659e-03, 1.2503e-03, 2.8210e-03, 1.6689e-03,
          1.6680e-03, 1.6651e-03, 2.6836e-03, 5.0888e-03, 1.1711e-03, 1.4057e-03,
          1.5049e-03, 9.3555e-04, 1.5411e-03, 1.5783e-03, 1.2131e-03, 2.9278e-03,
          4.6768e-03, 9.7227e-04, 2.8992e-03, 2.0065e-03, 1.8511e-03, 3.3016e-03,
          9.1171e-03, 2.3460e-03, 3.4618e-03, 9.5797e-04, 9.0790e-04, 1.6050e-03,
          8.0013e-04, 1.2321e-03, 1.0176e-03, 1.2712e-03, 2.2812e-03, 9.1887e-04,
          2.4662e-03, 2.0714e-03, 1.5020e-03, 1.2436e-03, 6.1340e-03, 1.4706e-03,
          3.7365e-03, 2.5501e-03, 1.7633e-03, 2.0027e-03, 4.7684e-03, 2.6112e-03,
          3.9711e-03, 1.3199e-03, 1.9417e-03, 1.0271e-03, 5.7564e-03, 2.2831e-03,
          7.5722e-04, 1.4324e-03, 1.3123e-03, 1.0803e-02, 3.9825e-03, 1.5774e-03,
          1.7977e-03, 1.8549e-03, 2.1744e-03, 5.6267e-03, 8.1177e-03, 1.6052e-02,
          6.3095e-03, 4.9324e-03, 3.7098e-03, 4.8103e-03, 5.3024e-03, 2.4776e-03,
          1.6088e-03, 1.8921e-03, 1.6708e-03, 2.7351e-03, 2.9316e-03, 3.5419e-03,
          8.6260e-04, 1.1530e-03, 1.6603e-03, 2.1896e-03, 2.1267e-03, 6.3858e-03,
          3.2463e-03, 2.1744e-03, 8.8806e-03, 7.2098e-03, 3.1109e-03, 6.8474e-03,
          6.7673e-03, 1.8501e-03, 1.7233e-03, 5.7716e-03, 2.8362e-03, 3.6736e-03,
          3.7670e-03, 1.1473e-03, 4.5929e-03, 2.0313e-03, 3.6526e-03, 7.7438e-03,
          8.4763e-03, 6.8779e-03, 9.5987e-04, 1.3523e-03, 1.6327e-03, 3.0766e-03,
          1.7452e-03, 2.2469e-03, 1.7166e-03, 2.9507e-03, 7.3204e-03, 6.3133e-03,
          1.0086e-02, 5.0964e-03, 3.1681e-03, 2.1648e-03, 2.6798e-03, 4.6921e-03,
          2.4605e-03, 1.4305e-03, 1.1826e-03, 3.8185e-03, 4.2992e-03, 2.7580e-03,
          1.6088e-03, 1.7099e-03, 4.9973e-03, 5.3368e-03, 2.4586e-03, 8.2245e-03,
          6.2141e-03, 3.5439e-03, 6.9847e-03, 2.9449e-03, 2.8343e-03, 2.5482e-03,
          3.4714e-03, 4.2114e-02, 3.1352e-05, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          1.5378e-04, 3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          3.1352e-05, 3.1352e-05, 3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05,
          3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 1.5378e-04, 3.8075e-04,
          3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05,
          1.5378e-04, 3.8075e-04, 3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05,
          3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05, 1.5378e-04, 3.1352e-05,
          1.5378e-04, 1.5378e-04, 1.5378e-04, 3.1352e-05, 3.1352e-05, 1.5378e-04,
          1.5378e-04, 3.1352e-05, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          1.5378e-04, 1.5378e-04, 3.1352e-05, 3.1352e-05, 3.1352e-05, 3.1352e-05,
          4.1008e-04, 1.5378e-04, 4.1008e-04, 1.5378e-04, 4.1008e-04, 3.1352e-05,
          3.1352e-05, 1.5378e-04, 1.5378e-04, 1.5378e-04, 1.5378e-04, 4.1008e-04,
          1.5378e-04, 1.5378e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.0686e-03, 4.3884e-02, 1.7868e-02, 1.8814e-02, 7.7637e-02,
          2.5726e-02, 1.6266e-02, 5.9891e-04, 5.9700e-04, 7.8821e-04, 1.2798e-03,
          5.1451e-04, 1.9550e-03, 2.4529e-03, 1.0157e-03, 6.0797e-04, 4.8790e-03,
          2.4662e-03, 4.5156e-04, 1.7796e-03, 1.7567e-03, 1.5974e-03, 1.8520e-03,
          1.3199e-03, 9.9850e-04, 1.2260e-02, 4.3831e-03, 2.6550e-03, 3.4409e-03,
          3.0975e-03, 1.0080e-03, 1.8063e-03, 1.8454e-03, 7.9012e-04, 9.4318e-04,
          1.4906e-03, 8.3065e-04, 2.0027e-03, 1.4763e-03, 2.5711e-03, 1.4277e-03,
          9.5034e-04, 7.3195e-04, 7.3004e-04, 6.0034e-04, 2.7275e-03, 2.1458e-03,
          1.4791e-03, 2.2430e-03, 1.0460e-02, 1.4229e-03, 7.5197e-04, 5.6601e-04,
          5.6267e-04, 6.6233e-04, 1.0796e-03, 1.0071e-03, 8.9836e-04, 8.6403e-04,
          1.1406e-03, 8.3694e-03, 1.3599e-03, 1.8253e-03, 6.6137e-04, 1.7271e-03,
          7.8917e-04, 2.6369e-04, 5.6791e-04, 9.1839e-04, 6.0081e-04, 5.9938e-04,
          8.6451e-04, 1.5030e-03, 9.8801e-04, 9.6273e-04, 4.4990e-04, 3.7408e-04,
          9.2602e-04, 9.9182e-04, 6.7234e-04, 6.7377e-04, 5.4932e-04, 6.7663e-04,
          1.1606e-03, 1.6994e-03, 1.0319e-03, 5.1641e-04, 3.3021e-04, 5.4550e-04,
          8.3494e-04, 6.6280e-04, 4.7398e-04, 5.8651e-04, 6.2132e-04, 9.4700e-04,
          2.5902e-03, 7.6151e-04, 9.1362e-04, 1.8768e-03, 9.7513e-04, 7.0477e-04,
          9.2316e-04, 4.0245e-04, 3.4857e-04, 1.0519e-03, 1.1654e-03, 9.8953e-03,
          1.4009e-03, 1.2255e-03, 1.0080e-03, 6.4898e-04, 6.1512e-04, 4.6706e-04,
          5.2643e-04, 1.3809e-03, 5.0125e-03, 1.8806e-03, 1.2503e-03, 4.4327e-03,
          6.1646e-03, 1.2989e-03, 5.3692e-04, 3.8981e-04, 1.3876e-03, 3.6669e-04,
          9.4557e-04, 5.5218e-04, 6.7759e-04, 1.6165e-03, 1.1282e-03, 1.6260e-03,
          9.5987e-04, 4.4918e-04, 3.6359e-04, 3.2496e-04, 5.4169e-04, 7.4291e-04,
          5.3673e-03, 5.7945e-03, 2.3479e-03, 2.3384e-03, 2.0962e-03, 6.4430e-03,
          7.5579e-04, 3.9506e-04, 7.0095e-04, 5.9223e-04, 6.8426e-04, 1.4353e-03,
          1.4353e-03, 1.4038e-03, 1.2512e-03, 3.0212e-03, 2.3956e-03, 2.6703e-03,
          7.2718e-04, 1.0824e-03, 3.1710e-04, 6.4373e-04, 2.5463e-03, 6.6948e-03,
          1.8692e-03, 1.4334e-03, 4.5013e-03, 1.8625e-03, 2.7065e-03, 6.9857e-04,
          8.0538e-04, 6.1274e-04, 9.3365e-04, 7.0953e-04, 4.4479e-03, 3.0804e-03,
          9.6655e-04, 1.1339e-03, 3.4332e-03, 3.5763e-03, 1.8501e-03, 1.4629e-03,
          7.3004e-04, 9.5177e-04, 1.4801e-03, 9.9182e-04, 8.8739e-04, 9.2697e-04,
          7.1478e-04, 4.4365e-03, 1.5726e-03, 1.4334e-03, 8.2064e-04, 1.3247e-03,
          9.5034e-04, 7.9870e-04, 1.3571e-03, 6.0997e-03, 6.3467e-04, 1.4257e-03,
          2.6054e-03, 1.2772e-02, 3.1013e-03, 8.0204e-04, 1.0090e-03, 8.8453e-04,
          8.4066e-04, 1.0204e-03, 1.6117e-03, 5.8079e-04, 1.0958e-03, 2.1420e-03,
          7.7152e-04, 4.3488e-03, 1.9245e-03, 5.3740e-04, 1.8053e-03, 7.6675e-04,
          1.6394e-03, 9.2163e-03, 1.0567e-03, 6.9046e-04, 3.3417e-03, 5.7449e-03,
          1.0284e-02, 2.5730e-03, 1.5945e-03, 6.6471e-04, 2.5196e-03, 1.2836e-03,
          7.1383e-04, 8.5545e-04, 1.0967e-03, 1.1425e-03, 1.8797e-03, 4.0092e-03,
          3.4122e-03, 8.0395e-04, 8.4066e-04, 8.3017e-04, 2.7981e-03, 3.6049e-03,
          2.5692e-03, 3.7727e-03, 2.5673e-03, 6.7062e-03, 1.2611e-02, 3.0167e-02,
          5.0850e-03, 1.1139e-03, 1.9045e-03, 1.1005e-03, 2.2717e-03, 1.7014e-03,
          8.9169e-04, 1.6985e-03, 2.6855e-03, 2.5368e-03, 2.3613e-03, 3.8948e-03,
          2.6150e-03, 8.1301e-04, 1.1196e-03, 1.2264e-03, 1.2407e-03, 2.5730e-03,
          2.1973e-03, 9.0551e-04, 2.6035e-03, 1.0651e-02, 8.9188e-03, 1.9836e-03,
          2.1935e-03, 1.1234e-03, 2.3384e-03, 2.5749e-03, 3.3932e-03, 2.2087e-03,
          1.9236e-03, 8.8120e-04, 1.7075e-02, 7.0839e-03, 3.4657e-03, 1.5430e-03,
          1.0252e-03, 2.0046e-03, 2.0771e-03, 2.3766e-03, 4.6310e-03, 2.1362e-03,
          1.7061e-03, 3.3169e-03, 4.8866e-03, 4.4594e-03, 5.0735e-03, 3.4904e-03,
          5.8861e-03, 3.2368e-03, 2.8362e-03, 2.9335e-03, 2.1515e-03, 3.4218e-03,
          3.3836e-03, 2.0199e-03, 1.4992e-03, 1.2360e-02, 3.3188e-03, 2.0142e-03,
          2.5387e-03, 1.6068e-02, 4.2152e-03, 2.0714e-03, 1.0557e-03, 8.7643e-04,
          4.3259e-03, 2.5425e-03, 8.1360e-02, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 1.9538e-04, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 1.9538e-04, 4.4048e-05, 4.4048e-05, 2.4724e-04, 2.4724e-04,
          4.4048e-05, 4.4048e-05, 6.7568e-04, 4.4048e-05, 4.4048e-05, 3.8147e-04,
          3.9840e-04, 3.9840e-04, 4.4048e-05, 4.4048e-05, 1.9538e-04, 4.4048e-05,
          2.4724e-04, 4.4048e-05, 4.4048e-05, 5.3263e-04, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          1.9538e-04, 4.4048e-05, 3.9840e-04, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          1.9538e-04, 4.4048e-05, 4.4048e-05, 4.7231e-04, 4.4048e-05, 4.4048e-05,
          4.4048e-05, 1.9538e-04, 4.4048e-05, 4.4048e-05, 4.4048e-05, 4.4048e-05,
          1.9538e-04, 4.3559e-04, 4.3559e-04, 1.9538e-04, 1.9538e-04, 1.9538e-04,
          4.4048e-05, 1.9538e-04, 1.9538e-04, 4.1151e-04, 4.4048e-05, 4.4048e-05,
          1.9538e-04, 4.4048e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4327e-03, 5.6641e-02, 1.3023e-02, 1.3931e-02, 3.5370e-02,
          4.5288e-02, 9.7504e-03, 1.8902e-03, 8.5068e-04, 2.5215e-03, 3.4046e-03,
          1.3580e-03, 9.2411e-04, 7.5006e-04, 1.8406e-03, 1.9426e-03, 1.4439e-03,
          1.1578e-03, 8.0538e-04, 2.8634e-04, 3.3808e-04, 3.6025e-04, 2.0542e-03,
          4.0474e-03, 2.0542e-03, 8.0185e-03, 4.3068e-03, 4.7064e-04, 1.7786e-03,
          8.9788e-04, 3.7556e-03, 7.6485e-04, 3.1147e-03, 2.5826e-03, 1.6661e-03,
          1.7796e-03, 5.2071e-04, 3.3875e-03, 3.6354e-03, 3.9983e-04, 2.0123e-03,
          6.0654e-04, 4.9543e-04, 5.0783e-04, 4.6253e-04, 1.5898e-03, 2.3117e-03,
          6.6185e-04, 5.7316e-04, 3.3607e-03, 2.4757e-03, 2.8458e-03, 1.5793e-03,
          6.5088e-04, 1.6575e-03, 6.4516e-04, 4.6754e-04, 1.5764e-03, 8.2541e-04,
          7.6199e-04, 1.2102e-03, 1.6432e-03, 2.7866e-03, 8.5831e-04, 3.1967e-03,
          1.3199e-03, 9.8133e-04, 1.2245e-03, 3.0160e-04, 2.8443e-04, 4.3201e-04,
          5.0306e-04, 5.6496e-03, 8.1158e-04, 3.7899e-03, 6.7091e-04, 7.8917e-04,
          1.4544e-03, 4.5848e-04, 7.4720e-04, 1.7309e-03, 9.3460e-04, 6.2037e-04,
          5.3692e-04, 8.3160e-03, 2.1591e-03, 1.9798e-03, 9.6703e-04, 7.3242e-04,
          3.5095e-03, 2.7466e-03, 4.2105e-04, 2.0504e-03, 3.9935e-04, 4.7088e-04,
          7.2384e-04, 5.2261e-04, 3.2463e-03, 1.6899e-03, 5.1308e-04, 7.6628e-04,
          3.0079e-03, 8.0585e-04, 7.5769e-04, 1.8682e-03, 2.6073e-03, 1.0567e-03,
          6.8207e-03, 8.6308e-04, 3.4313e-03, 6.3610e-04, 9.4318e-04, 1.4973e-03,
          1.1663e-03, 9.1362e-04, 5.6267e-04, 3.2520e-03, 6.9094e-04, 1.2255e-03,
          1.6956e-03, 2.1229e-03, 3.3164e-04, 1.3981e-03, 3.3302e-03, 3.7169e-04,
          3.4690e-04, 4.8470e-04, 6.9380e-04, 3.6488e-03, 5.9509e-03, 7.6141e-03,
          1.6375e-03, 8.5878e-04, 9.9850e-04, 6.5947e-04, 2.1610e-03, 5.0735e-04,
          3.7289e-03, 1.1663e-03, 2.1038e-03, 4.6806e-03, 4.0169e-03, 1.9627e-03,
          5.0974e-04, 1.0128e-03, 6.7997e-04, 7.2479e-04, 9.0504e-04, 3.0327e-03,
          4.5662e-03, 2.8267e-03, 7.6914e-04, 5.4817e-03, 3.9787e-03, 1.9951e-03,
          3.1757e-03, 2.3937e-03, 1.2360e-03, 8.4257e-04, 3.1509e-03, 2.0065e-03,
          2.9030e-03, 3.4637e-03, 3.0899e-03, 3.8548e-03, 1.0481e-03, 1.3037e-03,
          2.1267e-03, 1.6003e-03, 2.0866e-03, 1.5011e-03, 2.5444e-03, 2.8915e-03,
          2.2163e-03, 1.7128e-03, 1.2674e-03, 1.9646e-03, 4.0169e-03, 3.8147e-03,
          1.6050e-03, 4.0436e-03, 4.0779e-03, 5.8899e-03, 2.0847e-03, 1.6518e-03,
          2.7122e-03, 1.7843e-03, 3.0594e-03, 7.8821e-04, 9.8038e-04, 3.5114e-03,
          4.2152e-03, 3.6454e-04, 2.4376e-03, 3.6945e-03, 2.6646e-03, 2.6112e-03,
          1.4219e-03, 1.3599e-03, 5.5885e-03, 2.0771e-03, 2.6512e-03, 2.2335e-03,
          1.0929e-03, 3.1853e-03, 8.4639e-04, 1.8930e-03, 4.0665e-03, 1.9951e-03,
          2.5539e-03, 1.6661e-03, 7.9012e-04, 1.2980e-03, 3.8319e-03, 7.0143e-04,
          2.4242e-03, 2.8801e-03, 4.9591e-03, 1.6718e-03, 5.1231e-03, 2.9926e-03,
          6.6109e-03, 1.2598e-03, 2.8419e-03, 2.2678e-03, 7.6981e-03, 5.3253e-03,
          1.6251e-03, 1.0281e-03, 7.3719e-04, 2.0046e-03, 3.1815e-03, 1.6508e-03,
          1.3046e-03, 1.5554e-03, 9.3889e-04, 2.9984e-03, 2.7046e-03, 4.8027e-03,
          1.8272e-03, 1.4915e-03, 1.4124e-03, 2.9926e-03, 4.6921e-03, 2.2850e-03,
          2.1229e-03, 2.2087e-03, 1.3676e-03, 1.4105e-03, 2.3537e-03, 4.2152e-03,
          9.1362e-04, 5.9843e-04, 1.0920e-03, 1.6890e-03, 4.6387e-03, 7.6790e-03,
          1.4191e-03, 1.9989e-03, 1.6441e-03, 2.0943e-03, 8.3447e-04, 2.4891e-03,
          3.3016e-03, 2.9335e-03, 2.3899e-03, 7.2594e-03, 7.5035e-03, 3.8738e-03,
          2.4319e-03, 2.3766e-03, 5.9929e-03, 2.2335e-03, 4.1122e-03, 3.9597e-03,
          2.2678e-03, 1.9217e-03, 1.2589e-03, 3.0766e-03, 1.2817e-03, 1.7090e-03,
          1.7166e-03, 8.8882e-04, 6.3276e-04, 9.4366e-04, 4.4746e-03, 3.2978e-03,
          3.0174e-03, 8.2397e-03, 5.0087e-03, 3.3092e-03, 1.8959e-03, 3.0346e-03,
          2.3632e-03, 1.4668e-03, 1.0815e-03, 5.1384e-03, 4.9591e-03, 1.6766e-03,
          1.1740e-03, 2.9392e-03, 5.6572e-03, 6.8588e-03, 1.1139e-03, 4.4212e-03,
          4.6158e-03, 1.8883e-03, 4.8027e-03, 2.9125e-03, 2.1114e-03, 2.5291e-03,
          3.4027e-03, 3.1109e-03, 4.1313e-03, 1.0437e-01, 3.6180e-05, 3.6180e-05,
          3.6180e-05, 3.3212e-04, 1.9670e-04, 3.3212e-04, 3.6180e-05, 3.6180e-05,
          3.3212e-04, 3.6180e-05, 1.9670e-04, 3.6180e-05, 3.6180e-05, 3.3212e-04,
          3.6180e-05, 3.6180e-05, 3.3212e-04, 3.6180e-05, 3.6180e-05, 3.6180e-05,
          3.6180e-05, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.3212e-04,
          3.6180e-05, 3.6180e-05, 3.3212e-04, 1.7142e-04, 3.6180e-05, 1.9670e-04,
          1.2245e-03, 3.3212e-04, 3.6180e-05, 3.6180e-05, 3.3212e-04, 1.7142e-04,
          3.6180e-05, 3.6180e-05, 1.7142e-04, 3.6180e-05, 3.3212e-04, 3.6180e-05,
          3.3212e-04, 1.7142e-04, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.6180e-05,
          1.7142e-04, 1.9670e-04, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.6180e-05,
          1.9670e-04, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.3212e-04, 3.6180e-05,
          3.3212e-04, 3.3212e-04, 3.3212e-04, 3.6180e-05, 3.6180e-05, 1.9670e-04,
          3.3212e-04, 3.6180e-05, 1.7624e-03, 3.6180e-05, 3.6180e-05, 3.6180e-05,
          2.6550e-03, 1.9670e-04, 3.6180e-05, 3.6180e-05, 3.6180e-05, 3.6180e-05,
          1.9670e-04, 3.6180e-05, 1.9670e-04, 3.6180e-05, 1.9670e-04, 3.6180e-05,
          3.6180e-05, 3.6180e-05, 3.6180e-05, 3.9101e-04, 3.6180e-05, 3.6180e-05,
          3.6180e-05, 3.6180e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.6349e-03, 5.2155e-02, 1.3962e-02, 2.1011e-02, 3.2623e-02,
          4.1229e-02, 1.1536e-02, 8.0442e-04, 5.3644e-04, 2.1057e-03, 2.2144e-03,
          1.0004e-03, 2.4166e-03, 1.1301e-03, 1.9531e-03, 9.2316e-04, 7.8201e-03,
          3.2825e-03, 5.2023e-04, 8.4257e-04, 9.7036e-04, 8.6546e-04, 3.9368e-03,
          2.8744e-03, 1.5850e-03, 9.6359e-03, 7.8430e-03, 9.4938e-04, 7.0992e-03,
          2.1515e-03, 1.9026e-03, 2.2278e-03, 5.7602e-03, 2.8801e-03, 1.3533e-03,
          2.3880e-03, 4.6921e-04, 5.7602e-03, 4.4632e-03, 7.7391e-04, 2.6321e-03,
          7.1716e-04, 5.7364e-04, 5.3644e-04, 5.9605e-04, 7.6180e-03, 7.5645e-03,
          9.1457e-04, 9.4652e-04, 3.7880e-03, 2.1496e-03, 1.4229e-03, 1.0767e-03,
          3.1853e-04, 7.0572e-04, 5.0068e-04, 6.4373e-04, 6.9952e-04, 5.8746e-04,
          6.0892e-04, 2.4261e-03, 1.0099e-03, 2.6779e-03, 7.7105e-04, 5.4779e-03,
          6.5184e-04, 5.3978e-04, 1.3819e-03, 3.4142e-04, 3.4022e-04, 4.0197e-04,
          6.1131e-04, 5.8937e-03, 5.6124e-04, 1.6766e-03, 2.8563e-04, 3.0971e-04,
          7.5102e-04, 5.5742e-04, 4.4990e-04, 1.2598e-03, 4.7016e-04, 3.6383e-04,
          4.0174e-04, 3.0193e-03, 7.1859e-04, 1.3123e-03, 4.3678e-04, 3.9935e-04,
          2.7046e-03, 2.2488e-03, 2.6035e-04, 1.7700e-03, 4.0603e-04, 6.5422e-04,
          1.1263e-03, 3.6359e-04, 2.1858e-03, 4.2610e-03, 4.5776e-04, 7.0667e-04,
          1.9951e-03, 3.7336e-04, 4.6659e-04, 1.2207e-03, 2.6283e-03, 1.9989e-03,
          2.2736e-03, 6.5994e-04, 2.7180e-03, 3.1877e-04, 5.2118e-04, 7.3814e-04,
          1.0777e-03, 1.3151e-03, 1.2903e-03, 5.7106e-03, 8.2016e-04, 1.4935e-03,
          2.4261e-03, 2.7370e-03, 2.7180e-04, 4.9114e-04, 3.0708e-03, 2.0838e-04,
          3.6693e-04, 3.0231e-04, 4.3249e-04, 3.2806e-03, 2.9430e-03, 2.7084e-03,
          8.2684e-04, 5.9223e-04, 6.0844e-04, 4.0030e-04, 1.5135e-03, 3.7432e-04,
          1.6556e-02, 2.5234e-03, 2.6016e-03, 4.2801e-03, 7.3166e-03, 3.0727e-03,
          3.8815e-04, 4.4799e-04, 5.1165e-04, 5.2929e-04, 4.7541e-04, 2.1610e-03,
          2.7657e-03, 2.1095e-03, 6.1750e-04, 3.8910e-03, 1.5535e-03, 2.6169e-03,
          1.8406e-03, 1.4505e-03, 6.1512e-04, 4.1389e-04, 4.7493e-03, 5.0621e-03,
          6.4583e-03, 5.3787e-03, 1.3092e-02, 6.3324e-03, 1.5316e-03, 8.7643e-04,
          1.5354e-03, 1.0662e-03, 1.7357e-03, 1.2627e-03, 4.5395e-03, 5.2719e-03,
          9.3269e-04, 5.7936e-04, 1.1873e-03, 1.7328e-03, 4.4098e-03, 3.9597e-03,
          1.9321e-03, 2.5578e-03, 4.5547e-03, 2.3022e-03, 1.5879e-03, 2.6646e-03,
          1.8167e-03, 2.4452e-03, 1.9398e-03, 7.2718e-04, 5.2023e-04, 2.7542e-03,
          2.0676e-03, 3.3665e-04, 3.8033e-03, 1.2878e-02, 1.8387e-03, 3.2349e-03,
          1.0576e-03, 1.8139e-03, 5.9128e-03, 1.2951e-03, 1.4563e-03, 2.5082e-03,
          7.2432e-04, 2.4509e-03, 9.3079e-04, 9.0885e-04, 3.4142e-03, 4.9438e-03,
          1.0338e-03, 2.6646e-03, 7.9441e-04, 7.9012e-04, 3.7518e-03, 4.3368e-04,
          2.7485e-03, 5.8899e-03, 3.5820e-03, 1.0815e-03, 1.1093e-02, 3.0518e-03,
          1.6403e-02, 8.7070e-04, 2.6531e-03, 1.1148e-03, 6.8626e-03, 4.5090e-03,
          1.0538e-03, 4.9877e-04, 6.1893e-04, 1.1148e-03, 3.5038e-03, 2.3041e-03,
          1.4658e-03, 8.0633e-04, 4.2868e-04, 1.1415e-03, 2.0962e-03, 2.2259e-03,
          8.1301e-04, 9.4509e-04, 9.7179e-04, 5.1804e-03, 7.5073e-03, 3.6869e-03,
          2.3327e-03, 1.6289e-03, 5.2404e-04, 7.2813e-04, 8.6546e-04, 4.2076e-03,
          6.0940e-04, 6.6185e-04, 1.4534e-03, 2.1038e-03, 8.3618e-03, 8.9798e-03,
          1.6012e-03, 9.0075e-04, 9.7942e-04, 1.4076e-03, 4.3488e-04, 1.0109e-03,
          2.2526e-03, 2.2259e-03, 2.0180e-03, 9.5673e-03, 7.5912e-03, 3.0727e-03,
          1.2379e-03, 1.3103e-03, 5.2605e-03, 1.5392e-03, 8.8577e-03, 4.3221e-03,
          2.0428e-03, 1.3943e-03, 2.7466e-03, 4.9133e-03, 1.3876e-03, 8.6308e-04,
          8.9216e-04, 6.2084e-04, 5.3024e-04, 7.5626e-04, 4.2267e-03, 3.5610e-03,
          1.3018e-03, 1.2741e-02, 9.6588e-03, 8.2245e-03, 1.4772e-03, 1.8835e-03,
          2.2869e-03, 1.0557e-03, 9.6798e-04, 8.3694e-03, 5.0659e-03, 1.9436e-03,
          1.4048e-03, 3.5648e-03, 2.4109e-03, 1.2016e-02, 9.3269e-04, 3.6526e-03,
          5.1994e-03, 2.0180e-03, 5.4283e-03, 2.4700e-03, 1.3952e-03, 1.3475e-03,
          4.7150e-03, 1.3628e-03, 6.6795e-03, 5.0278e-03, 3.9429e-02, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 1.9121e-04,
          2.9981e-05, 2.9981e-05, 1.9121e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 1.9121e-04,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 3.7670e-04, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 4.2892e-04, 1.9121e-04,
          3.7670e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05, 1.9121e-04, 3.7670e-04,
          4.2892e-04, 4.2892e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          1.9121e-04, 2.9981e-05, 2.9981e-05, 5.3787e-04, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 1.9121e-04, 2.9981e-05,
          2.6774e-04, 1.9121e-04, 1.9121e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          7.7581e-04, 2.9981e-05, 1.9121e-04, 2.9981e-05, 2.9981e-05, 3.7670e-04,
          2.9981e-05, 4.5586e-03, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.6774e-04, 2.6774e-04, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05, 2.9981e-05,
          2.9981e-05, 2.9981e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.6221e-03, 4.7119e-02, 1.0742e-02, 1.1810e-02, 3.4821e-02,
          3.8605e-02, 8.6441e-03, 7.6675e-04, 4.9162e-04, 2.2011e-03, 2.8057e-03,
          1.0395e-03, 9.1028e-04, 3.2306e-04, 3.7556e-03, 9.8324e-04, 9.3555e-04,
          7.4053e-04, 5.1641e-04, 1.8013e-04, 2.7943e-04, 2.7180e-04, 1.2388e-03,
          3.5419e-03, 2.9812e-03, 6.9237e-03, 2.8820e-03, 3.1066e-04, 1.2255e-03,
          4.2629e-04, 1.5268e-03, 3.8195e-04, 3.9787e-03, 2.7256e-03, 1.0529e-03,
          3.0537e-03, 4.9067e-04, 3.9482e-03, 3.7994e-03, 2.3890e-04, 3.1204e-03,
          7.9632e-04, 6.1512e-04, 6.3753e-04, 5.1832e-04, 1.3466e-03, 4.7646e-03,
          2.7490e-04, 3.5357e-04, 2.6054e-03, 1.0586e-03, 7.8201e-04, 9.4604e-04,
          2.2519e-04, 4.7684e-04, 3.5429e-04, 5.8985e-04, 8.3733e-04, 9.3460e-04,
          8.1205e-04, 7.1335e-04, 9.4604e-04, 1.8091e-03, 1.2617e-03, 9.1934e-03,
          8.3876e-04, 6.3133e-04, 1.4706e-03, 3.3498e-04, 3.2949e-04, 3.9911e-04,
          4.5347e-04, 4.7798e-03, 3.7217e-04, 1.0672e-03, 2.0945e-04, 2.2590e-04,
          4.9067e-04, 4.9734e-04, 5.0735e-04, 2.4128e-03, 5.7077e-04, 8.4782e-04,
          5.8985e-04, 2.8782e-03, 7.0620e-04, 1.8415e-03, 5.9319e-04, 5.4932e-04,
          3.5076e-03, 2.7733e-03, 4.2224e-04, 1.8415e-03, 5.0831e-04, 4.9591e-04,
          3.5810e-04, 2.5845e-04, 1.5640e-03, 1.4553e-03, 4.6253e-04, 7.3147e-04,
          2.3479e-03, 3.3236e-04, 4.6897e-04, 5.4169e-03, 6.9695e-03, 8.7976e-04,
          2.4242e-03, 1.0757e-03, 3.8586e-03, 4.8709e-04, 6.4707e-04, 8.8263e-04,
          1.7633e-03, 1.4515e-03, 5.9319e-04, 8.0261e-03, 7.5769e-04, 5.5933e-04,
          8.5831e-04, 2.9545e-03, 2.7657e-04, 5.8842e-04, 7.8888e-03, 2.3186e-04,
          3.6311e-04, 3.3092e-04, 4.8971e-04, 6.8550e-03, 2.3861e-03, 2.6474e-03,
          1.0147e-03, 6.9141e-04, 1.8129e-03, 5.8699e-04, 2.3136e-03, 5.4073e-04,
          5.4970e-03, 1.0595e-03, 1.1091e-03, 2.3193e-03, 2.6512e-03, 1.0300e-03,
          4.2009e-04, 7.3576e-04, 6.7520e-04, 6.0701e-04, 6.1703e-04, 5.5542e-03,
          3.8853e-03, 5.8060e-03, 1.0176e-03, 2.4509e-03, 1.3494e-03, 4.3297e-03,
          3.0708e-03, 2.0790e-03, 8.2254e-04, 7.2384e-04, 6.6109e-03, 2.0180e-03,
          5.0011e-03, 4.0359e-03, 2.5520e-03, 4.1046e-03, 1.1215e-03, 8.3590e-04,
          1.5430e-03, 9.9373e-04, 3.5210e-03, 2.6264e-03, 1.8606e-03, 4.7684e-03,
          2.0733e-03, 1.1816e-03, 1.3514e-03, 2.2888e-03, 1.1993e-02, 4.2191e-03,
          2.7409e-03, 6.9847e-03, 1.0750e-02, 2.4815e-03, 1.1768e-03, 1.7986e-03,
          2.2144e-03, 9.1553e-04, 1.4868e-03, 6.9761e-04, 5.8174e-04, 5.7259e-03,
          1.8854e-03, 3.3355e-04, 3.6068e-03, 2.5787e-03, 1.9102e-03, 3.9177e-03,
          1.3237e-03, 1.1320e-03, 1.3084e-02, 2.3155e-03, 1.5621e-03, 4.0207e-03,
          1.2445e-03, 4.5738e-03, 1.5574e-03, 1.0405e-03, 3.6259e-03, 1.5144e-03,
          8.7547e-04, 1.4801e-03, 7.7391e-04, 7.2479e-04, 5.1270e-03, 5.0735e-04,
          2.8648e-03, 2.3842e-03, 3.1185e-03, 2.0008e-03, 5.8365e-03, 1.1988e-03,
          5.3520e-03, 1.9646e-03, 6.9809e-03, 1.3218e-03, 2.0767e-02, 7.1831e-03,
          1.4133e-03, 8.0252e-04, 1.2302e-03, 2.2602e-03, 6.3286e-03, 1.1806e-03,
          1.2436e-03, 7.9679e-04, 4.8923e-04, 2.5063e-03, 3.4485e-03, 3.8223e-03,
          1.1387e-03, 1.1091e-03, 1.0967e-03, 2.3155e-03, 4.0245e-03, 2.8515e-03,
          3.8338e-03, 2.4624e-03, 1.4839e-03, 1.4420e-03, 4.2953e-03, 1.6708e-02,
          1.0414e-03, 1.1692e-03, 2.2202e-03, 2.8572e-03, 4.3373e-03, 5.2299e-03,
          1.5802e-03, 7.8964e-04, 1.7271e-03, 2.4433e-03, 7.1764e-04, 1.4076e-03,
          3.7289e-03, 2.0599e-03, 1.3065e-03, 4.3564e-03, 3.4676e-03, 1.4702e-02,
          1.8597e-03, 1.7214e-03, 2.0721e-02, 6.1302e-03, 1.0307e-02, 1.1353e-02,
          5.0049e-03, 2.0466e-03, 1.5421e-03, 2.0542e-03, 1.0252e-03, 9.0790e-04,
          8.1825e-04, 6.6328e-04, 5.1069e-04, 8.2350e-04, 3.4370e-03, 2.5616e-03,
          2.2354e-03, 9.2850e-03, 3.4370e-03, 1.9426e-03, 3.5038e-03, 4.6196e-03,
          4.8676e-03, 2.4338e-03, 1.9522e-03, 9.7122e-03, 8.4229e-03, 2.9316e-03,
          2.1362e-03, 1.7986e-03, 2.4796e-03, 3.3817e-03, 1.0118e-03, 4.3106e-03,
          4.8752e-03, 1.1110e-03, 4.1008e-03, 8.1015e-04, 1.9274e-03, 1.4229e-03,
          1.5268e-03, 2.1076e-03, 4.8180e-03, 6.0768e-03, 8.8730e-03, 4.6417e-02,
          3.6299e-05, 1.2922e-04, 1.2922e-04, 2.3651e-04, 3.6299e-05, 3.6299e-05,
          2.3651e-04, 2.3651e-04, 1.2922e-04, 3.6299e-05, 3.6299e-05, 2.3651e-04,
          3.6299e-05, 2.3651e-04, 2.3699e-04, 3.6299e-05, 3.6299e-05, 3.6299e-05,
          3.6299e-05, 3.6299e-05, 2.3651e-04, 1.2922e-04, 3.6299e-05, 2.3651e-04,
          3.6299e-05, 3.6299e-05, 2.3651e-04, 2.3651e-04, 5.2977e-04, 1.2922e-04,
          9.3842e-04, 2.3651e-04, 3.6299e-05, 3.6299e-05, 2.3651e-04, 2.3651e-04,
          6.3705e-04, 1.2922e-04, 3.6299e-05, 3.6299e-05, 2.3651e-04, 5.2977e-04,
          2.3651e-04, 2.3651e-04, 3.6299e-05, 1.4353e-04, 2.3651e-04, 3.6299e-05,
          3.6299e-05, 2.3651e-04, 3.6299e-05, 7.3004e-04, 3.6299e-05, 3.6299e-05,
          2.3651e-04, 3.6299e-05, 3.6299e-05, 3.6299e-05, 5.0545e-04, 1.2922e-04,
          1.6508e-03, 2.3651e-04, 2.3651e-04, 1.2922e-04, 3.6299e-05, 1.2922e-04,
          2.3651e-04, 1.2922e-04, 2.3651e-04, 3.6299e-05, 3.6299e-05, 3.6299e-05,
          9.3842e-04, 2.9488e-03, 3.6299e-05, 1.2922e-04, 3.6299e-05, 3.6299e-05,
          3.6299e-05, 3.6299e-05, 1.2922e-04, 1.4079e-04, 1.2922e-04, 3.6299e-05,
          3.6299e-05, 3.6299e-05, 3.6299e-05, 3.2783e-04, 3.6299e-05, 1.2922e-04,
          3.6299e-05, 3.6299e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.3512e-03, 4.8065e-02, 1.2115e-02, 1.0101e-02, 3.6530e-02,
          3.7537e-02, 8.6594e-03, 1.2236e-03, 6.4325e-04, 1.5507e-03, 5.2719e-03,
          1.3504e-03, 7.3099e-04, 3.5238e-04, 1.4677e-03, 1.4820e-03, 6.5088e-04,
          4.0650e-04, 7.5531e-04, 1.7846e-04, 2.1911e-04, 2.4486e-04, 8.2970e-04,
          7.4463e-03, 3.0422e-03, 6.4812e-03, 2.9430e-03, 2.7966e-04, 6.1226e-04,
          4.3797e-04, 2.0008e-03, 2.2280e-04, 1.0920e-03, 7.7009e-04, 7.5340e-04,
          9.9850e-04, 6.2275e-04, 7.8583e-04, 8.3160e-04, 2.5105e-04, 1.0252e-03,
          7.2098e-04, 5.9032e-04, 6.8188e-04, 4.0340e-04, 8.4162e-04, 1.3924e-03,
          2.5344e-04, 3.2544e-04, 3.3417e-03, 8.0824e-04, 9.2125e-04, 7.6628e-04,
          3.9148e-04, 5.8365e-04, 4.3130e-04, 5.3835e-04, 2.8877e-03, 1.7262e-03,
          1.4668e-03, 5.0926e-04, 1.3781e-03, 1.0872e-03, 1.3704e-03, 3.4714e-03,
          2.8534e-03, 1.3857e-03, 7.9679e-04, 3.3236e-04, 3.4642e-04, 6.9380e-04,
          5.2643e-04, 3.0289e-03, 4.4680e-04, 1.2140e-03, 4.6563e-04, 3.4714e-04,
          6.4230e-04, 4.1795e-04, 9.3317e-04, 1.1692e-03, 1.2512e-03, 1.4811e-03,
          9.0265e-04, 2.9869e-03, 8.7881e-04, 2.5635e-03, 1.2140e-03, 1.1654e-03,
          5.0507e-03, 1.6851e-03, 7.3624e-04, 2.6684e-03, 6.9666e-04, 5.6553e-04,
          3.4618e-04, 2.8586e-04, 1.1044e-03, 7.2098e-04, 4.2796e-04, 5.7840e-04,
          3.5076e-03, 1.2255e-03, 7.2432e-04, 1.4668e-03, 2.3117e-03, 6.2704e-04,
          2.4910e-03, 7.8106e-04, 2.3880e-03, 1.1520e-03, 2.1954e-03, 1.0853e-03,
          3.2787e-03, 7.2861e-04, 4.4131e-04, 3.5706e-03, 7.8487e-04, 5.6505e-04,
          7.1859e-04, 1.1568e-03, 3.7003e-04, 9.1124e-04, 3.1872e-03, 4.0627e-04,
          2.9755e-04, 5.9128e-04, 9.6083e-04, 1.9512e-03, 1.4868e-03, 2.9011e-03,
          3.4771e-03, 1.4839e-03, 1.3294e-03, 7.9346e-04, 2.0847e-03, 6.4754e-04,
          1.8539e-03, 8.1968e-04, 2.3823e-03, 2.8191e-03, 1.7691e-03, 8.2016e-04,
          6.2990e-04, 2.1191e-03, 7.6294e-04, 6.0797e-04, 1.3361e-03, 2.6188e-03,
          2.1839e-03, 1.8301e-03, 9.5272e-04, 1.3714e-03, 2.2182e-03, 1.4400e-03,
          1.5697e-03, 9.0694e-04, 1.9646e-03, 1.2026e-03, 2.9774e-03, 1.1978e-03,
          2.3174e-03, 1.6918e-03, 1.4439e-03, 2.4853e-03, 5.6362e-04, 6.7520e-04,
          2.7657e-03, 8.1062e-04, 1.8568e-03, 1.8778e-03, 9.1410e-04, 1.2589e-03,
          4.5052e-03, 3.7460e-03, 1.3084e-03, 1.2941e-03, 2.6646e-03, 2.5444e-03,
          1.5955e-03, 2.2449e-03, 3.8185e-03, 3.9978e-03, 1.8759e-03, 1.7967e-03,
          4.0321e-03, 1.4105e-03, 3.1319e-03, 6.9904e-04, 1.2970e-03, 2.1229e-03,
          2.3174e-03, 3.2759e-04, 1.3714e-03, 1.6499e-03, 2.1439e-03, 1.3590e-03,
          2.3041e-03, 9.5654e-04, 4.0932e-03, 1.9226e-03, 1.9054e-03, 1.9522e-03,
          2.3823e-03, 2.9373e-03, 1.3418e-03, 1.6508e-03, 6.0387e-03, 8.9216e-04,
          2.2430e-03, 8.0442e-04, 7.2956e-04, 1.5259e-03, 2.4910e-03, 9.5797e-04,
          9.9659e-04, 1.2350e-03, 2.2354e-03, 2.0580e-03, 2.2831e-03, 1.4133e-03,
          1.9913e-03, 2.5387e-03, 4.3411e-03, 2.4834e-03, 9.4681e-03, 2.5787e-03,
          3.4046e-03, 3.7060e-03, 2.4261e-03, 1.3832e-02, 4.3373e-03, 1.7128e-03,
          1.0185e-03, 1.1435e-03, 1.8044e-03, 3.9368e-03, 2.2430e-03, 1.4305e-02,
          5.0011e-03, 2.3956e-03, 1.5554e-03, 1.0338e-03, 2.8095e-03, 1.5879e-03,
          2.0676e-03, 6.9695e-03, 5.9891e-03, 1.1604e-02, 4.1382e-02, 8.4915e-03,
          1.8835e-03, 1.3676e-03, 3.0842e-03, 2.4853e-03, 2.8801e-03, 5.4398e-03,
          9.0265e-04, 1.9913e-03, 2.0828e-03, 2.4929e-03, 1.2999e-03, 6.0844e-03,
          2.7161e-03, 1.6813e-03, 2.2411e-03, 4.6501e-03, 4.2992e-03, 7.0648e-03,
          1.1261e-02, 3.2558e-03, 1.1124e-02, 6.6528e-03, 3.3798e-03, 9.7351e-03,
          1.4000e-02, 9.5367e-03, 1.2875e-03, 1.8587e-03, 7.5436e-04, 2.2945e-03,
          3.2101e-03, 1.0977e-03, 5.8603e-04, 1.0729e-03, 2.3212e-03, 2.3251e-03,
          1.2924e-02, 3.9024e-03, 1.9522e-03, 2.6054e-03, 5.2223e-03, 1.3145e-02,
          4.5509e-03, 3.5152e-03, 2.6245e-03, 4.9171e-03, 5.5161e-03, 7.3662e-03,
          3.4542e-03, 2.7142e-03, 4.3259e-03, 2.5597e-03, 1.2903e-03, 5.4665e-03,
          3.1776e-03, 8.7452e-04, 2.0618e-03, 1.5621e-03, 3.3741e-03, 3.4580e-03,
          1.8473e-03, 6.7024e-03, 3.5496e-03, 5.1155e-03, 5.7945e-03, 8.1863e-03,
          6.1340e-02, 1.5080e-04, 1.5080e-04, 1.5080e-04, 3.0398e-05, 3.0398e-05,
          1.5080e-04, 1.5080e-04, 1.5080e-04, 3.0398e-05, 3.0398e-05, 3.1853e-04,
          3.0398e-05, 1.5080e-04, 1.5128e-04, 3.0398e-05, 3.0398e-05, 3.0398e-05,
          3.0398e-05, 3.0398e-05, 3.0398e-05, 1.5080e-04, 3.0398e-05, 3.1853e-04,
          3.0398e-05, 3.0398e-05, 9.7752e-04, 3.0398e-05, 3.0398e-05, 1.5080e-04,
          1.5080e-04, 1.5080e-04, 3.0398e-05, 3.0398e-05, 3.1853e-04, 1.9801e-04,
          3.0398e-05, 3.0398e-05, 3.0398e-05, 3.0398e-05, 3.1853e-04, 3.0398e-05,
          3.1853e-04, 3.1853e-04, 3.0398e-05, 3.0398e-05, 1.5080e-04, 3.0398e-05,
          3.0398e-05, 1.5080e-04, 3.0398e-05, 3.0398e-05, 3.0398e-05, 3.0398e-05,
          1.5080e-04, 3.0398e-05, 3.0398e-05, 3.0398e-05, 1.9801e-04, 1.5080e-04,
          1.5080e-04, 3.1853e-04, 1.5080e-04, 1.5080e-04, 3.0398e-05, 1.5080e-04,
          1.5080e-04, 1.5080e-04, 3.1853e-04, 3.0398e-05, 3.0398e-05, 1.5080e-04,
          1.5080e-04, 1.5080e-04, 3.0398e-05, 1.5080e-04, 3.0398e-05, 3.0398e-05,
          1.5080e-04, 1.5080e-04, 1.5080e-04, 2.2578e-04, 1.5080e-04, 3.0398e-05,
          3.0398e-05, 3.0398e-05, 3.0398e-05, 3.7360e-04, 3.0398e-05, 1.5080e-04,
          3.0398e-05, 3.0398e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9776e-03, 5.3894e-02, 1.3535e-02, 2.3697e-02, 3.4698e-02,
          3.9337e-02, 1.3702e-02, 1.4811e-03, 7.4053e-04, 5.0697e-03, 1.1627e-02,
          3.7422e-03, 1.4238e-03, 6.7520e-04, 1.5955e-03, 1.5688e-03, 2.2221e-03,
          1.7319e-03, 7.9632e-04, 1.8167e-04, 2.8419e-04, 3.2163e-04, 2.6836e-03,
          1.4626e-02, 1.7004e-03, 1.0849e-02, 6.3553e-03, 3.7766e-04, 2.5692e-03,
          1.0872e-03, 3.0689e-03, 7.8058e-04, 2.4948e-03, 4.4060e-03, 8.8358e-04,
          1.3409e-03, 4.9257e-04, 2.7943e-03, 2.6474e-03, 2.6107e-04, 2.6741e-03,
          8.1730e-04, 5.4741e-04, 6.4087e-04, 5.6076e-04, 2.8152e-03, 2.6684e-03,
          4.2200e-04, 3.5667e-04, 3.1490e-03, 1.1101e-03, 1.9627e-03, 1.0366e-03,
          3.1304e-04, 8.9931e-04, 2.7966e-04, 9.0981e-04, 1.5306e-03, 1.4324e-03,
          8.4352e-04, 7.9012e-04, 5.7077e-04, 8.7738e-04, 1.3504e-03, 4.3526e-03,
          1.9798e-03, 1.9836e-03, 1.9398e-03, 3.6645e-04, 3.2759e-04, 5.1928e-04,
          5.4312e-04, 1.1436e-02, 4.6587e-04, 2.2640e-03, 3.4428e-04, 3.3331e-04,
          6.7472e-04, 7.2241e-04, 1.7395e-03, 2.4738e-03, 1.9875e-03, 6.9427e-04,
          3.8576e-04, 3.0785e-03, 4.4656e-04, 8.6288e-03, 1.5230e-03, 9.0837e-04,
          1.6235e-02, 5.3368e-03, 3.6120e-04, 8.1711e-03, 3.3116e-04, 4.4537e-04,
          4.4584e-04, 2.3830e-04, 1.9035e-03, 1.4715e-03, 5.5981e-04, 1.1845e-03,
          1.4259e-02, 9.9945e-04, 2.2831e-03, 1.2503e-03, 2.2430e-03, 5.8794e-04,
          2.1572e-03, 7.5483e-04, 7.8430e-03, 4.4060e-04, 1.2779e-03, 2.1915e-03,
          1.1368e-03, 7.5436e-04, 3.6669e-04, 3.6697e-03, 5.7459e-04, 5.6124e-04,
          1.0109e-03, 2.2869e-03, 3.8195e-04, 9.3794e-04, 3.9520e-03, 5.0449e-04,
          3.2496e-04, 6.4898e-04, 8.7738e-04, 2.4414e-03, 2.9411e-03, 2.7313e-03,
          2.1935e-03, 2.2221e-03, 1.3342e-03, 9.4891e-04, 5.3787e-03, 4.2987e-04,
          5.0545e-03, 7.9346e-04, 1.4935e-03, 4.3373e-03, 3.6678e-03, 1.3952e-03,
          5.5552e-04, 7.9203e-04, 6.8045e-04, 9.5510e-04, 1.3466e-03, 2.3174e-03,
          5.8784e-03, 1.7853e-03, 6.5231e-04, 1.4906e-03, 9.3842e-04, 2.2964e-03,
          5.4131e-03, 1.8206e-03, 4.2305e-03, 6.9809e-04, 2.6417e-03, 1.3609e-03,
          3.6640e-03, 4.3449e-03, 5.0392e-03, 3.7346e-03, 9.5463e-04, 1.2131e-03,
          1.0826e-02, 1.7900e-03, 1.7080e-03, 1.3027e-03, 2.3003e-03, 2.6417e-03,
          1.1845e-03, 8.7595e-04, 6.9952e-04, 1.3762e-03, 3.3684e-03, 5.5466e-03,
          1.9159e-03, 1.9493e-03, 4.4746e-03, 4.6768e-03, 1.5659e-03, 1.2016e-03,
          1.1566e-02, 9.1839e-04, 2.8725e-03, 7.3290e-04, 1.1377e-03, 2.7847e-03,
          4.1237e-03, 3.3498e-04, 2.4319e-03, 5.0812e-03, 1.0139e-02, 2.3880e-03,
          6.8855e-04, 5.6410e-04, 3.7594e-03, 3.4199e-03, 5.0468e-03, 2.5158e-03,
          1.4820e-03, 5.1956e-03, 8.5258e-04, 1.6193e-03, 1.8082e-02, 1.6308e-03,
          1.6489e-03, 1.1797e-03, 8.9550e-04, 4.4594e-03, 3.1776e-03, 8.2445e-04,
          1.4677e-03, 1.7052e-03, 7.3128e-03, 1.3657e-03, 3.5534e-03, 1.7653e-03,
          3.7994e-03, 8.0585e-04, 3.1261e-03, 2.8515e-03, 7.7400e-03, 4.2953e-03,
          3.7193e-03, 7.2575e-04, 6.5565e-04, 1.0319e-03, 1.5936e-03, 8.8263e-04,
          1.6413e-03, 1.2341e-03, 7.5293e-04, 1.2550e-03, 1.6108e-03, 2.8725e-03,
          1.4582e-03, 8.6164e-04, 5.5265e-04, 9.0599e-04, 2.5463e-03, 1.2350e-03,
          1.3800e-03, 3.8357e-03, 5.8222e-04, 8.7357e-04, 1.0395e-03, 3.4618e-03,
          8.0633e-04, 4.7040e-04, 8.0729e-04, 8.7547e-04, 3.7460e-03, 6.8092e-03,
          1.1368e-03, 1.9932e-03, 8.0490e-04, 9.8991e-04, 4.6110e-04, 1.8911e-03,
          1.4563e-03, 2.6188e-03, 1.5793e-03, 4.6310e-03, 8.2855e-03, 2.0657e-03,
          2.0905e-03, 4.3144e-03, 4.8676e-03, 1.3466e-03, 4.3030e-03, 2.6474e-03,
          1.5707e-03, 1.2684e-03, 7.7629e-04, 2.6855e-03, 8.7881e-04, 1.8559e-03,
          3.6888e-03, 5.5361e-04, 4.1795e-04, 4.9686e-04, 1.6851e-03, 1.2150e-03,
          1.6880e-03, 7.9498e-03, 3.5267e-03, 4.1580e-03, 1.0719e-03, 1.8320e-03,
          2.0542e-03, 1.1482e-03, 7.8869e-04, 4.3106e-03, 5.7640e-03, 1.3294e-03,
          8.3065e-04, 2.2240e-03, 4.1809e-03, 2.7847e-03, 1.0481e-03, 4.2229e-03,
          2.6684e-03, 6.2752e-04, 2.0771e-03, 1.0185e-03, 1.7567e-03, 3.0518e-03,
          1.3237e-03, 9.1743e-04, 1.8511e-03, 4.3411e-03, 8.0185e-03, 5.2414e-03,
          1.3609e-03, 5.0049e-02, 2.4748e-04, 3.3998e-04, 5.2214e-05, 5.2214e-05,
          1.3041e-04, 2.2292e-04, 5.2214e-05, 5.2214e-05, 5.2214e-05, 3.3998e-04,
          5.2214e-05, 8.4305e-04, 3.3998e-04, 1.4472e-04, 5.2214e-05, 5.2214e-05,
          5.2214e-05, 1.4472e-04, 1.4472e-04, 1.6916e-04, 5.2214e-05, 3.3998e-04,
          5.2214e-05, 5.2214e-05, 3.3998e-04, 1.4472e-04, 4.6897e-04, 5.2214e-05,
          2.4748e-04, 1.4472e-04, 5.2214e-05, 5.2214e-05, 2.9325e-04, 2.9325e-04,
          1.4992e-03, 5.2214e-05, 5.2214e-05, 5.2214e-05, 3.3998e-04, 9.0313e-04,
          8.2016e-04, 2.9325e-04, 3.7646e-04, 5.2214e-05, 5.2214e-05, 5.2214e-05,
          1.4472e-04, 3.3998e-04, 5.2214e-05, 5.2214e-05, 5.2214e-05, 5.2214e-05,
          5.2214e-05, 5.2214e-05, 5.2214e-05, 5.2214e-05, 8.2016e-04, 5.2214e-05,
          1.3695e-03, 2.2292e-04, 5.2214e-05, 5.2214e-05, 1.6916e-04, 2.4748e-04,
          4.6897e-04, 5.2214e-05, 3.3998e-04, 5.2214e-05, 5.2214e-05, 2.4748e-04,
          2.4748e-04, 1.2770e-03, 5.2214e-05, 2.4748e-04, 1.6916e-04, 5.2214e-05,
          5.2214e-05, 5.2214e-05, 5.2214e-05, 1.2970e-04, 1.3041e-04, 2.2292e-04,
          5.2214e-05, 2.2292e-04, 1.3041e-04, 5.0163e-04, 1.3041e-04, 1.3041e-04,
          5.2214e-05, 5.2214e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.8218e-03, 5.1178e-02, 1.6632e-02, 1.3351e-02, 3.2379e-02,
          3.0319e-02, 9.3231e-03, 7.6389e-04, 4.5252e-04, 1.0605e-03, 1.7090e-03,
          6.1274e-04, 3.4199e-03, 1.8644e-03, 1.2856e-03, 8.7929e-04, 1.0056e-02,
          4.8637e-03, 4.7302e-04, 7.4053e-04, 6.8045e-04, 1.3704e-03, 3.3932e-03,
          2.2545e-03, 1.1120e-03, 7.3433e-03, 4.3869e-03, 1.2178e-03, 8.1100e-03,
          3.9005e-03, 1.5879e-03, 2.7351e-03, 2.1973e-03, 8.9550e-04, 1.0214e-03,
          9.3126e-04, 3.2187e-04, 1.9407e-03, 1.3409e-03, 1.2598e-03, 1.7433e-03,
          5.3883e-04, 3.9339e-04, 4.4847e-04, 4.3893e-04, 9.2468e-03, 3.8853e-03,
          1.8845e-03, 1.3895e-03, 3.0937e-03, 1.1730e-03, 1.0529e-03, 7.3481e-04,
          2.6917e-04, 5.7316e-04, 3.7527e-04, 3.5310e-04, 5.1308e-04, 4.2295e-04,
          4.8637e-04, 5.0850e-03, 6.8760e-04, 1.1864e-03, 6.3038e-04, 3.8013e-03,
          6.4850e-04, 4.0984e-04, 7.4244e-04, 2.7800e-04, 2.6608e-04, 3.6478e-04,
          5.0545e-04, 3.1338e-03, 4.4322e-04, 1.4429e-03, 3.0065e-04, 2.6941e-04,
          6.8045e-04, 3.4976e-04, 3.5024e-04, 6.5327e-04, 3.4571e-04, 2.9707e-04,
          3.1662e-04, 2.3727e-03, 5.7507e-04, 9.7656e-04, 3.3450e-04, 3.6287e-04,
          2.0466e-03, 1.6727e-03, 2.9230e-04, 1.3084e-03, 3.9721e-04, 5.4741e-04,
          1.5993e-03, 2.7156e-04, 1.8234e-03, 2.3327e-03, 3.0613e-04, 4.2558e-04,
          1.4114e-03, 3.1185e-04, 3.1638e-04, 6.7425e-04, 1.6241e-03, 5.4321e-03,
          1.8492e-03, 6.1035e-04, 2.1400e-03, 3.3522e-04, 4.7779e-04, 5.7125e-04,
          7.6485e-04, 1.0042e-03, 2.8648e-03, 4.3030e-03, 8.2254e-04, 1.8864e-03,
          4.9553e-03, 1.2617e-03, 1.8871e-04, 4.5538e-04, 1.9331e-03, 1.5497e-04,
          2.6011e-04, 2.6464e-04, 3.5572e-04, 1.8463e-03, 1.9913e-03, 2.1114e-03,
          8.1730e-04, 4.7374e-04, 4.2415e-04, 3.4451e-04, 1.1349e-03, 3.8481e-04,
          2.2720e-02, 5.3215e-03, 4.6921e-03, 3.8624e-03, 3.0956e-03, 7.4348e-03,
          2.6846e-04, 2.6989e-04, 3.2735e-04, 3.4738e-04, 3.8958e-04, 1.2789e-03,
          1.7548e-03, 1.2302e-03, 5.0402e-04, 2.4586e-03, 1.4277e-03, 2.1286e-03,
          1.3828e-03, 1.1177e-03, 5.4979e-04, 4.2963e-04, 3.8300e-03, 1.3359e-02,
          5.0926e-03, 3.0575e-03, 1.6373e-02, 2.5158e-03, 8.5258e-04, 4.8208e-04,
          1.0262e-03, 6.3038e-04, 9.0551e-04, 8.8453e-04, 6.7101e-03, 2.6684e-03,
          7.3147e-04, 5.0831e-04, 9.1028e-04, 1.5917e-03, 3.1281e-03, 2.8305e-03,
          1.1911e-03, 1.4944e-03, 3.0899e-03, 2.0447e-03, 1.8730e-03, 1.8187e-03,
          1.3838e-03, 7.0496e-03, 1.8148e-03, 4.8566e-04, 4.0340e-04, 1.6985e-03,
          1.5869e-03, 2.3091e-04, 2.0351e-03, 1.8692e-02, 1.1339e-03, 1.7109e-03,
          9.5034e-04, 3.2253e-03, 3.9215e-03, 1.1101e-03, 1.1597e-03, 1.8530e-03,
          6.1607e-04, 2.3880e-03, 1.0214e-03, 6.8808e-04, 2.8019e-03, 7.9422e-03,
          1.2817e-03, 1.6737e-03, 5.6648e-04, 5.2023e-04, 2.2106e-03, 3.4595e-04,
          1.2188e-03, 1.3054e-02, 1.8549e-03, 8.5735e-04, 5.1460e-03, 7.7438e-03,
          7.7019e-03, 9.4461e-04, 2.2697e-03, 8.9550e-04, 5.6763e-03, 2.2945e-03,
          1.0176e-03, 5.1165e-04, 6.3324e-04, 1.1683e-03, 2.2240e-03, 6.4201e-03,
          1.1044e-03, 5.4789e-04, 3.8075e-04, 7.7248e-04, 1.3409e-03, 2.1973e-03,
          8.6927e-04, 8.8930e-04, 6.4611e-04, 3.2616e-03, 1.4549e-02, 7.7629e-03,
          2.4910e-03, 1.6031e-03, 7.0715e-04, 9.1267e-04, 1.2264e-03, 3.3760e-03,
          6.5613e-04, 7.5579e-04, 1.4639e-03, 2.0275e-03, 8.5678e-03, 5.8365e-03,
          9.5081e-04, 7.0047e-04, 6.1560e-04, 8.1110e-04, 3.7241e-04, 1.0605e-03,
          1.2484e-03, 9.8991e-04, 3.6316e-03, 2.7634e-02, 1.5129e-02, 2.9202e-03,
          1.5030e-03, 1.1644e-03, 4.6844e-03, 2.0466e-03, 5.6458e-03, 3.6793e-03,
          2.5234e-03, 1.3475e-03, 5.4665e-03, 1.1726e-02, 9.9373e-04, 7.1812e-04,
          7.5436e-04, 5.1546e-04, 4.3654e-04, 5.6982e-04, 5.3101e-03, 2.1439e-03,
          1.2293e-03, 5.0468e-03, 1.5556e-02, 1.0033e-02, 1.4896e-03, 2.4242e-03,
          2.9144e-03, 1.2789e-03, 1.0824e-03, 6.3477e-03, 3.5667e-03, 2.5253e-03,
          1.6823e-03, 4.6272e-03, 2.6379e-03, 2.1530e-02, 7.2765e-04, 2.3537e-03,
          2.6970e-03, 2.4891e-03, 2.6855e-03, 3.5858e-03, 9.7752e-04, 9.7609e-04,
          1.0406e-02, 1.4057e-03, 9.0027e-03, 3.7270e-03, 1.3763e-02, 5.6686e-03,
          1.6165e-03, 5.6458e-03, 3.8025e-02, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 3.6538e-05, 1.6296e-04, 3.6538e-05, 1.8358e-04, 3.6538e-05,
          1.8358e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 1.6701e-04, 3.6538e-05,
          3.6538e-05, 1.8358e-04, 3.6538e-05, 3.6538e-05, 3.0994e-04, 1.6296e-04,
          1.6701e-04, 3.6538e-05, 1.6296e-04, 3.6538e-05, 3.6538e-05, 3.1400e-04,
          3.0994e-04, 3.0994e-04, 1.6701e-04, 3.6538e-05, 1.8358e-04, 3.6538e-05,
          3.0994e-04, 3.6538e-05, 3.6538e-05, 4.4060e-04, 3.6538e-05, 3.6538e-05,
          1.8358e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 1.6296e-04, 3.6538e-05,
          1.8358e-04, 1.6296e-04, 3.0994e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          4.4060e-04, 3.6538e-05, 1.6296e-04, 3.6538e-05, 1.8358e-04, 1.8358e-04,
          3.6538e-05, 1.8358e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          5.0926e-04, 5.0926e-04, 1.8358e-04, 2.5034e-04, 1.8358e-04, 3.6538e-05,
          3.6538e-05, 3.0994e-04, 5.0926e-04, 5.8413e-04, 1.8358e-04, 1.8358e-04,
          1.8358e-04, 1.8358e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4174e-03, 4.7974e-02, 1.6022e-02, 1.3474e-02, 3.3966e-02,
          3.1891e-02, 9.4757e-03, 1.1072e-03, 5.5361e-04, 1.0662e-03, 2.1515e-03,
          7.0000e-04, 4.0550e-03, 2.3346e-03, 9.9087e-04, 1.2808e-03, 1.0208e-02,
          5.4550e-03, 6.1274e-04, 5.5742e-04, 5.0783e-04, 1.2417e-03, 3.7861e-03,
          2.7809e-03, 1.0214e-03, 7.6370e-03, 4.1046e-03, 1.0557e-03, 8.4076e-03,
          4.9095e-03, 2.4776e-03, 2.4338e-03, 1.5478e-03, 1.0004e-03, 7.0238e-04,
          7.1621e-04, 3.0041e-04, 1.2836e-03, 1.1568e-03, 9.3460e-04, 1.0777e-03,
          4.9162e-04, 3.6168e-04, 4.4346e-04, 3.4189e-04, 1.1833e-02, 2.1763e-03,
          1.7414e-03, 1.1721e-03, 2.9984e-03, 1.0157e-03, 1.5945e-03, 6.9571e-04,
          2.8682e-04, 8.0442e-04, 3.6526e-04, 2.8563e-04, 6.5851e-04, 4.1795e-04,
          4.9686e-04, 3.8700e-03, 6.8188e-04, 9.4652e-04, 5.9652e-04, 2.6684e-03,
          8.7357e-04, 5.6171e-04, 7.2956e-04, 2.4629e-04, 2.4164e-04, 3.7694e-04,
          4.7350e-04, 2.8343e-03, 4.4298e-04, 2.5177e-03, 3.6645e-04, 2.9016e-04,
          7.9870e-04, 2.7013e-04, 3.8910e-04, 5.8317e-04, 4.2915e-04, 2.9087e-04,
          3.0017e-04, 2.9907e-03, 5.8317e-04, 1.2817e-03, 4.8375e-04, 4.4394e-04,
          2.6703e-03, 1.5574e-03, 3.3450e-04, 1.5221e-03, 3.8338e-04, 4.8184e-04,
          1.5574e-03, 2.5821e-04, 1.5736e-03, 1.4553e-03, 2.6417e-04, 3.5262e-04,
          1.7452e-03, 4.3297e-04, 4.0364e-04, 5.3310e-04, 1.2350e-03, 4.7646e-03,
          2.2278e-03, 4.6015e-04, 1.7862e-03, 3.4022e-04, 6.1035e-04, 6.9809e-04,
          7.3099e-04, 7.4816e-04, 1.7586e-03, 3.0041e-03, 7.2908e-04, 1.8339e-03,
          5.4474e-03, 9.0599e-04, 1.8287e-04, 7.1383e-04, 1.5869e-03, 1.6356e-04,
          2.1029e-04, 2.7680e-04, 3.9196e-04, 1.4420e-03, 2.3403e-03, 2.7981e-03,
          9.8801e-04, 6.5088e-04, 4.6968e-04, 3.8695e-04, 1.1492e-03, 3.5810e-04,
          1.6037e-02, 5.8479e-03, 6.0349e-03, 3.9368e-03, 2.6684e-03, 7.5569e-03,
          2.7180e-04, 3.1400e-04, 2.9874e-04, 3.1328e-04, 5.1832e-04, 1.0977e-03,
          1.6441e-03, 1.0653e-03, 4.2343e-04, 1.8091e-03, 1.4772e-03, 1.2388e-03,
          1.1616e-03, 7.7105e-04, 7.8630e-04, 4.9496e-04, 2.8172e-03, 1.2840e-02,
          2.8210e-03, 1.9197e-03, 1.9043e-02, 2.2411e-03, 5.6219e-04, 4.5180e-04,
          1.2217e-03, 5.4884e-04, 8.4829e-04, 8.8215e-04, 6.1913e-03, 1.4420e-03,
          7.5340e-04, 5.3167e-04, 7.4196e-04, 1.0424e-03, 1.9026e-03, 3.5763e-03,
          1.0090e-03, 1.4477e-03, 2.5539e-03, 3.5896e-03, 2.3384e-03, 1.8282e-03,
          1.9207e-03, 9.6970e-03, 1.5717e-03, 4.2748e-04, 4.9305e-04, 1.5478e-03,
          2.7618e-03, 2.0158e-04, 1.4353e-03, 2.1057e-02, 1.3132e-03, 1.3208e-03,
          7.9536e-04, 2.2163e-03, 2.9087e-03, 1.0958e-03, 1.3657e-03, 1.3542e-03,
          6.7520e-04, 1.9426e-03, 8.5306e-04, 9.5129e-04, 3.1528e-03, 7.4615e-03,
          1.8225e-03, 1.0691e-03, 4.6492e-04, 6.7806e-04, 1.9569e-03, 3.7932e-04,
          9.3651e-04, 9.6512e-03, 2.0943e-03, 8.8406e-04, 3.3474e-03, 1.0666e-02,
          6.9122e-03, 7.1335e-04, 1.5926e-03, 1.2054e-03, 4.6082e-03, 2.3098e-03,
          1.3294e-03, 5.9366e-04, 6.1226e-04, 1.1806e-03, 1.8530e-03, 8.6670e-03,
          8.9121e-04, 5.6696e-04, 4.5204e-04, 8.6784e-04, 1.0958e-03, 2.3212e-03,
          8.9788e-04, 8.1205e-04, 4.6873e-04, 1.9989e-03, 1.4641e-02, 6.2790e-03,
          1.4248e-03, 1.8663e-03, 6.5470e-04, 9.0265e-04, 1.1120e-03, 2.6722e-03,
          7.0000e-04, 6.6710e-04, 1.2875e-03, 1.5173e-03, 9.2316e-03, 4.2114e-03,
          7.3862e-04, 9.1505e-04, 6.0701e-04, 7.8726e-04, 3.7122e-04, 1.0624e-03,
          1.1911e-03, 1.0843e-03, 4.4632e-03, 2.7374e-02, 2.1133e-02, 2.1954e-03,
          1.8120e-03, 1.6813e-03, 3.7060e-03, 1.5278e-03, 4.2076e-03, 3.3627e-03,
          2.4376e-03, 1.4505e-03, 4.4136e-03, 1.4420e-02, 7.4053e-04, 8.6546e-04,
          1.1549e-03, 4.7255e-04, 3.7003e-04, 5.3167e-04, 4.6654e-03, 2.0065e-03,
          1.3695e-03, 3.7384e-03, 1.6861e-02, 1.5114e-02, 1.2131e-03, 2.2678e-03,
          2.0828e-03, 1.0748e-03, 8.3542e-04, 5.0964e-03, 3.1052e-03, 2.3212e-03,
          1.4572e-03, 6.7749e-03, 3.6144e-03, 1.8158e-02, 6.6090e-04, 2.1744e-03,
          2.5463e-03, 1.7271e-03, 2.0561e-03, 4.1656e-03, 1.0433e-03, 1.2455e-03,
          9.3384e-03, 1.5841e-03, 6.2943e-03, 4.4250e-03, 1.1215e-02, 4.7531e-03,
          1.6890e-03, 6.8932e-03, 2.3087e-02, 3.1830e-02, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 1.5640e-04, 3.5048e-05, 3.5048e-05,
          3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05, 1.9026e-04, 3.5048e-05,
          3.5048e-05, 1.5640e-04, 3.5048e-05, 3.5048e-05, 1.7071e-04, 3.5048e-05,
          3.5048e-05, 1.9026e-04, 3.5048e-05, 3.5048e-05, 3.1161e-04, 1.5640e-04,
          1.7071e-04, 3.5048e-05, 1.5640e-04, 3.5048e-05, 3.5048e-05, 3.2592e-04,
          3.1161e-04, 3.1161e-04, 1.7071e-04, 3.5048e-05, 1.9026e-04, 3.5048e-05,
          3.1161e-04, 3.5048e-05, 3.5048e-05, 4.4727e-04, 3.5048e-05, 3.5048e-05,
          1.9026e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05, 1.5640e-04, 3.5048e-05,
          1.9026e-04, 1.5640e-04, 3.1161e-04, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          3.1161e-04, 3.5048e-05, 1.5640e-04, 3.5048e-05, 3.1161e-04, 3.5048e-05,
          3.5048e-05, 2.4376e-03, 3.5048e-05, 3.5048e-05, 3.5048e-05, 3.5048e-05,
          5.1546e-04, 5.1546e-04, 1.9026e-04, 1.9026e-04, 1.9026e-04, 3.5048e-05,
          3.5048e-05, 3.1161e-04, 5.1546e-04, 5.9843e-04, 1.9026e-04, 5.9843e-04,
          1.9026e-04, 1.9026e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.0812e-03, 4.5258e-02, 1.9348e-02, 1.8204e-02, 8.2031e-02,
          2.2736e-02, 2.1896e-02, 5.0831e-04, 2.8086e-04, 5.9366e-04, 1.5564e-03,
          4.6659e-04, 4.3225e-04, 4.9829e-04, 7.2479e-04, 5.4646e-04, 6.6566e-04,
          4.1986e-04, 3.1042e-04, 7.5245e-04, 7.6532e-04, 7.2050e-04, 5.9795e-04,
          1.7996e-03, 1.9894e-03, 1.4473e-02, 2.0809e-03, 6.9380e-04, 5.7316e-04,
          5.0831e-04, 8.7166e-04, 2.8038e-04, 7.1716e-04, 4.2868e-04, 3.9196e-04,
          9.0647e-04, 1.1330e-03, 9.8610e-04, 6.7329e-04, 5.3406e-04, 1.6623e-03,
          2.0123e-03, 1.3332e-03, 1.7872e-03, 9.0265e-04, 6.6328e-04, 1.4715e-03,
          4.7398e-04, 1.4658e-03, 9.3384e-03, 6.8521e-04, 5.2834e-04, 3.9458e-04,
          4.1890e-04, 4.4441e-04, 1.1396e-03, 7.4816e-04, 1.1721e-03, 1.0529e-03,
          2.2945e-03, 7.8773e-04, 1.3914e-03, 9.1839e-04, 1.0338e-03, 1.9875e-03,
          1.0147e-03, 4.5300e-04, 6.5041e-04, 1.5554e-03, 1.7614e-03, 1.3371e-03,
          1.7595e-03, 1.2865e-03, 8.9788e-04, 6.6662e-04, 4.7135e-04, 2.8181e-04,
          8.4209e-04, 6.4182e-04, 4.8637e-04, 5.3167e-04, 5.5456e-04, 8.9550e-04,
          1.8539e-03, 1.5802e-03, 1.0834e-03, 7.5626e-04, 4.1294e-04, 6.8903e-04,
          1.2093e-03, 9.6035e-04, 9.1696e-04, 8.8549e-04, 2.3804e-03, 3.3531e-03,
          6.3562e-04, 8.1730e-04, 5.3835e-04, 7.8297e-04, 8.2350e-04, 5.5599e-04,
          9.2745e-04, 3.3593e-04, 3.1805e-04, 1.0138e-03, 9.8324e-04, 9.1743e-04,
          1.3008e-03, 2.3937e-03, 1.3800e-03, 1.0614e-03, 8.1253e-04, 6.4468e-04,
          9.8991e-04, 3.2349e-03, 1.6470e-03, 2.0866e-03, 5.6572e-03, 1.0529e-03,
          8.9169e-04, 8.8453e-04, 4.8232e-04, 3.6168e-04, 1.2121e-03, 2.5702e-04,
          6.8665e-04, 5.3453e-04, 6.5947e-04, 1.0881e-03, 5.6076e-04, 1.4000e-03,
          1.0357e-03, 5.4884e-04, 3.8528e-04, 4.1723e-04, 8.8072e-04, 1.9512e-03,
          1.4687e-03, 1.3390e-03, 9.4891e-04, 1.9760e-03, 1.0948e-03, 8.0395e-04,
          5.8460e-04, 5.4979e-04, 5.1546e-04, 4.7350e-04, 5.7125e-04, 1.3580e-03,
          1.1511e-03, 1.3161e-03, 1.8206e-03, 1.3971e-03, 2.4414e-03, 4.8790e-03,
          8.9455e-04, 1.3275e-03, 4.3440e-04, 1.2732e-03, 3.6697e-03, 1.5812e-03,
          1.7242e-03, 1.2197e-03, 1.0757e-03, 1.0948e-03, 1.5306e-03, 4.9210e-04,
          8.6927e-04, 4.1127e-04, 6.9714e-04, 5.6696e-04, 6.0749e-04, 1.7967e-03,
          1.9112e-03, 2.3346e-03, 5.6114e-03, 5.9700e-03, 2.3441e-03, 9.0933e-04,
          8.2588e-04, 9.9468e-04, 1.6928e-03, 1.0443e-03, 6.5279e-04, 8.3876e-04,
          8.4686e-04, 1.4181e-03, 1.0157e-03, 1.1406e-03, 5.8317e-04, 9.6893e-04,
          7.9012e-04, 6.3801e-04, 8.3256e-04, 1.2856e-03, 5.9223e-04, 1.1311e-03,
          4.7188e-03, 2.2526e-03, 2.8229e-03, 1.1749e-03, 7.6008e-04, 1.0853e-03,
          1.5850e-03, 2.4338e-03, 8.9264e-03, 6.0320e-04, 1.5678e-03, 6.8665e-04,
          9.5081e-04, 1.8549e-03, 1.9836e-03, 4.5276e-04, 1.1387e-03, 6.7043e-04,
          1.0157e-03, 9.3317e-04, 9.0694e-04, 4.8423e-04, 1.3371e-03, 1.0033e-03,
          2.8191e-03, 6.2637e-03, 3.4218e-03, 8.5068e-04, 3.5992e-03, 1.0357e-03,
          1.1921e-03, 1.9426e-03, 6.2904e-03, 3.6869e-03, 3.1357e-03, 1.5545e-03,
          3.4084e-03, 6.0272e-04, 9.2173e-04, 9.4128e-04, 2.2259e-03, 3.7823e-03,
          3.4027e-03, 4.3755e-03, 1.5078e-03, 1.5478e-03, 2.0218e-03, 3.3264e-03,
          1.0124e-02, 1.5688e-03, 4.5700e-03, 2.2659e-03, 6.3782e-03, 3.4084e-03,
          2.7599e-03, 1.1826e-02, 2.0721e-02, 5.7831e-03, 1.2121e-03, 3.3703e-03,
          2.0256e-03, 5.9652e-04, 1.3781e-03, 1.2922e-03, 3.0384e-03, 2.9812e-03,
          1.5163e-03, 6.0987e-04, 7.2098e-04, 1.7433e-03, 2.1324e-03, 3.2425e-03,
          2.8839e-03, 1.2321e-03, 4.0321e-03, 8.8425e-03, 4.2038e-03, 5.4169e-03,
          1.1841e-02, 2.6207e-03, 1.0674e-02, 2.2240e-03, 3.0727e-03, 1.0986e-03,
          6.7997e-04, 2.2831e-03, 2.0580e-03, 4.0245e-03, 1.3876e-03, 1.1425e-03,
          1.6470e-03, 1.8272e-03, 9.8228e-04, 1.1778e-03, 1.4969e-02, 7.0114e-03,
          1.5930e-02, 9.4299e-03, 7.5569e-03, 2.5921e-03, 5.2185e-03, 4.6295e-02,
          3.4576e-02, 1.7147e-03, 1.5469e-03, 1.6079e-03, 4.7340e-03, 1.3952e-03,
          1.6117e-03, 3.5572e-03, 1.8520e-03, 5.8031e-04, 1.1435e-03, 6.6710e-04,
          8.3208e-04, 2.2774e-03, 1.3046e-02, 1.8892e-03, 3.7251e-03, 4.0207e-03,
          3.6888e-03, 2.6321e-03, 3.9520e-03, 2.5043e-03, 5.9296e-02, 6.7890e-05,
          3.1042e-04, 6.7890e-05, 6.7890e-05, 6.7890e-05, 6.7890e-05, 6.7890e-05,
          6.7890e-05, 6.7890e-05, 6.7890e-05, 6.7890e-05, 1.3793e-04, 1.3793e-04,
          6.7890e-05, 1.3793e-04, 1.3793e-04, 6.7890e-05, 6.7890e-05, 6.7890e-05,
          1.3793e-04, 6.7890e-05, 2.4045e-04, 1.3793e-04, 6.7890e-05, 6.7890e-05,
          6.7890e-05, 6.7890e-05, 6.7890e-05, 6.7890e-05, 2.4045e-04, 2.4045e-04,
          1.3793e-04, 1.3793e-04, 1.3793e-04, 1.3793e-04, 6.7890e-05, 6.7890e-05,
          5.0974e-04, 5.0974e-04, 6.7890e-05, 1.3793e-04, 2.4045e-04, 6.7890e-05,
          1.3793e-04, 6.7890e-05, 6.7890e-05, 1.3793e-04, 6.7890e-05, 1.3793e-04,
          6.7890e-05, 6.7890e-05, 6.7890e-05, 3.1042e-04, 3.1042e-04, 6.7890e-05,
          3.3712e-04, 6.7890e-05, 5.0974e-04, 1.3793e-04, 6.7890e-05, 1.3793e-04,
          4.0722e-04, 6.7890e-05, 2.4045e-04, 3.1042e-04, 6.7890e-05, 6.7890e-05,
          6.7890e-05, 3.3712e-04, 1.3793e-04, 6.7890e-05, 6.7890e-05, 6.7890e-05,
          3.3712e-04, 3.3712e-04, 3.3712e-04, 4.0150e-04, 1.4734e-04, 1.3793e-04,
          6.7890e-05, 6.7890e-05, 6.7890e-05, 4.0150e-04, 6.7890e-05, 4.0722e-04,
          1.3793e-04, 6.7890e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.4354e-03, 4.7699e-02, 2.0264e-02, 2.2552e-02, 8.3557e-02,
          2.2018e-02, 2.3895e-02, 4.5419e-04, 3.1757e-04, 7.6008e-04, 1.8530e-03,
          5.4836e-04, 4.9019e-04, 5.3787e-04, 8.8406e-04, 5.1785e-04, 8.3828e-04,
          4.9734e-04, 3.6526e-04, 9.0694e-04, 9.8515e-04, 6.3515e-04, 7.5531e-04,
          2.1305e-03, 2.1076e-03, 1.6449e-02, 2.7008e-03, 7.2479e-04, 6.8855e-04,
          5.3406e-04, 7.8487e-04, 3.1352e-04, 1.0052e-03, 5.4407e-04, 4.1366e-04,
          1.2331e-03, 1.1444e-03, 1.4772e-03, 8.9836e-04, 5.8031e-04, 2.5101e-03,
          2.2182e-03, 1.4791e-03, 1.6127e-03, 1.0891e-03, 7.2813e-04, 2.1305e-03,
          4.6754e-04, 1.3809e-03, 1.0544e-02, 6.8045e-04, 4.8256e-04, 4.0460e-04,
          3.7432e-04, 4.2725e-04, 9.4080e-04, 8.4877e-04, 9.2840e-04, 1.0004e-03,
          1.8616e-03, 8.2779e-04, 1.2512e-03, 1.0777e-03, 1.1387e-03, 2.4128e-03,
          1.0166e-03, 4.5300e-04, 8.6164e-04, 1.9245e-03, 2.0180e-03, 1.2569e-03,
          1.8940e-03, 1.7643e-03, 7.5245e-04, 5.9414e-04, 3.6788e-04, 2.6655e-04,
          6.4278e-04, 7.4434e-04, 4.9973e-04, 6.2990e-04, 5.3596e-04, 8.3590e-04,
          1.7700e-03, 1.4477e-03, 9.5654e-04, 8.7309e-04, 4.4084e-04, 6.8521e-04,
          1.4639e-03, 1.2207e-03, 8.5115e-04, 1.0481e-03, 1.9312e-03, 2.9068e-03,
          6.0606e-04, 6.5422e-04, 5.1308e-04, 1.0710e-03, 8.6641e-04, 6.2323e-04,
          1.1234e-03, 3.3498e-04, 3.2663e-04, 1.1396e-03, 1.1930e-03, 1.0138e-03,
          1.1377e-03, 2.9182e-03, 1.9131e-03, 9.5510e-04, 7.8773e-04, 6.9618e-04,
          9.9277e-04, 3.9902e-03, 1.7977e-03, 2.5158e-03, 4.5624e-03, 1.0900e-03,
          9.3269e-04, 1.1778e-03, 4.5609e-04, 3.3402e-04, 1.4353e-03, 2.4056e-04,
          7.2432e-04, 4.9353e-04, 5.9700e-04, 1.3599e-03, 6.2561e-04, 1.2655e-03,
          1.0271e-03, 5.6458e-04, 4.1318e-04, 4.2224e-04, 1.0700e-03, 1.8177e-03,
          1.9951e-03, 1.4133e-03, 1.0319e-03, 2.0638e-03, 1.3180e-03, 8.7786e-04,
          5.8651e-04, 5.1355e-04, 5.5647e-04, 5.1689e-04, 5.9748e-04, 1.4687e-03,
          1.4334e-03, 1.5059e-03, 1.6088e-03, 1.5612e-03, 1.9970e-03, 6.0959e-03,
          1.1749e-03, 1.7185e-03, 4.2701e-04, 1.1864e-03, 4.2305e-03, 1.8997e-03,
          2.4071e-03, 1.8892e-03, 1.3075e-03, 1.3132e-03, 2.3003e-03, 5.7077e-04,
          1.0223e-03, 5.1928e-04, 8.2302e-04, 6.2323e-04, 8.2731e-04, 2.3193e-03,
          1.2980e-03, 1.4534e-03, 5.6381e-03, 6.9122e-03, 2.9945e-03, 1.0509e-03,
          1.0147e-03, 1.0757e-03, 2.0771e-03, 9.0837e-04, 7.2813e-04, 9.5367e-04,
          8.6403e-04, 1.4534e-03, 1.0414e-03, 1.3065e-03, 5.6934e-04, 1.1883e-03,
          7.1192e-04, 6.7568e-04, 1.0414e-03, 1.5717e-03, 7.2908e-04, 1.4400e-03,
          4.2152e-03, 2.6836e-03, 3.2883e-03, 1.2846e-03, 9.5844e-04, 1.3723e-03,
          1.4954e-03, 2.6722e-03, 7.4883e-03, 5.9319e-04, 1.9321e-03, 8.4400e-04,
          8.4877e-04, 2.7695e-03, 2.1496e-03, 4.6825e-04, 1.2875e-03, 6.5470e-04,
          1.3533e-03, 1.1988e-03, 1.0538e-03, 5.2452e-04, 1.8454e-03, 9.6607e-04,
          3.5305e-03, 5.0163e-03, 3.3855e-03, 8.5545e-04, 4.0512e-03, 1.2941e-03,
          1.2875e-03, 1.5736e-03, 4.2801e-03, 2.7370e-03, 3.2959e-03, 1.4763e-03,
          3.8414e-03, 6.6710e-04, 9.0456e-04, 8.8596e-04, 2.2449e-03, 3.0403e-03,
          2.7046e-03, 4.1084e-03, 1.7805e-03, 2.1877e-03, 2.1706e-03, 4.1046e-03,
          1.0399e-02, 1.4791e-03, 3.0537e-03, 1.6823e-03, 4.8599e-03, 3.8586e-03,
          2.3270e-03, 8.1406e-03, 1.3588e-02, 7.5569e-03, 1.4076e-03, 3.8948e-03,
          2.6760e-03, 6.1417e-04, 1.2884e-03, 1.2560e-03, 1.7424e-03, 2.4300e-03,
          1.8129e-03, 7.1955e-04, 6.9618e-04, 1.8969e-03, 2.4490e-03, 3.2959e-03,
          2.5272e-03, 1.3227e-03, 4.4594e-03, 7.2784e-03, 5.6496e-03, 5.1880e-03,
          7.4310e-03, 1.9178e-03, 1.0986e-02, 2.2984e-03, 3.5648e-03, 1.0538e-03,
          7.2432e-04, 1.9474e-03, 2.0103e-03, 3.3817e-03, 1.5163e-03, 1.4324e-03,
          1.3638e-03, 2.7294e-03, 1.1559e-03, 1.3199e-03, 1.0887e-02, 5.2795e-03,
          1.3794e-02, 8.0795e-03, 6.9313e-03, 3.2654e-03, 6.0158e-03, 2.4521e-02,
          2.1423e-02, 1.8005e-03, 1.3638e-03, 2.0275e-03, 3.9711e-03, 1.6260e-03,
          1.9331e-03, 3.9253e-03, 2.4643e-03, 6.2990e-04, 1.0509e-03, 7.0190e-04,
          9.0504e-04, 1.9140e-03, 1.5625e-02, 2.0180e-03, 4.8447e-03, 4.6196e-03,
          3.0632e-03, 2.9888e-03, 4.4670e-03, 2.5845e-03, 3.5461e-02, 4.3243e-02,
          2.3067e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 2.3067e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 2.3067e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 2.3067e-04, 2.3067e-04,
          6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          4.6349e-04, 4.6349e-04, 6.2406e-05, 6.2406e-05, 2.3067e-04, 6.2406e-05,
          2.3067e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 6.2406e-05, 2.3067e-04, 2.3067e-04, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 4.6349e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          2.9516e-04, 6.2406e-05, 2.3067e-04, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 4.0092e-03, 6.2406e-05, 6.2406e-05, 6.2406e-05, 6.2406e-05,
          2.9516e-04, 2.9516e-04, 2.9516e-04, 1.4281e-04, 1.4281e-04, 6.2406e-05,
          6.2406e-05, 6.2406e-05, 6.2406e-05, 4.0936e-04, 6.2406e-05, 6.2406e-05,
          6.2406e-05, 6.2406e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.5700e-03, 5.3345e-02, 1.7944e-02, 1.5778e-02, 3.2867e-02,
          3.3813e-02, 1.1139e-02, 1.4830e-03, 6.1512e-04, 1.1435e-03, 2.6379e-03,
          7.4196e-04, 7.9575e-03, 6.0081e-03, 9.1553e-04, 1.4372e-03, 2.3087e-02,
          9.3002e-03, 8.1778e-04, 9.6798e-04, 5.5504e-04, 1.9341e-03, 4.1733e-03,
          3.2902e-03, 9.2125e-04, 8.9264e-03, 4.3945e-03, 1.4429e-03, 1.2398e-02,
          8.2397e-03, 2.9221e-03, 3.9940e-03, 1.4572e-03, 1.0719e-03, 6.4993e-04,
          7.1001e-04, 2.7227e-04, 1.1244e-03, 1.1253e-03, 1.3456e-03, 7.7248e-04,
          4.5323e-04, 3.3903e-04, 4.4155e-04, 3.0875e-04, 1.6922e-02, 1.9341e-03,
          2.6340e-03, 1.4057e-03, 3.1605e-03, 9.8705e-04, 1.7624e-03, 6.7854e-04,
          2.9993e-04, 8.2302e-04, 3.2687e-04, 2.3806e-04, 6.4182e-04, 3.9458e-04,
          4.5443e-04, 3.0384e-03, 6.6948e-04, 8.0967e-04, 4.7231e-04, 2.3212e-03,
          8.5449e-04, 6.5470e-04, 7.4530e-04, 2.2960e-04, 2.3246e-04, 4.2105e-04,
          5.2214e-04, 2.9469e-03, 4.1747e-04, 2.4471e-03, 3.2997e-04, 2.9516e-04,
          7.7248e-04, 2.1684e-04, 3.8362e-04, 6.4945e-04, 4.7183e-04, 2.7680e-04,
          2.6202e-04, 2.9011e-03, 5.2547e-04, 1.6298e-03, 4.7660e-04, 3.9577e-04,
          3.2291e-03, 1.4219e-03, 3.3116e-04, 1.8492e-03, 3.8171e-04, 5.3740e-04,
          2.1019e-03, 2.4295e-04, 1.4486e-03, 1.2093e-03, 2.2125e-04, 2.9445e-04,
          1.9875e-03, 5.7364e-04, 4.4060e-04, 4.8614e-04, 1.2846e-03, 1.9817e-03,
          2.0866e-03, 3.5405e-04, 1.4210e-03, 3.0494e-04, 5.7936e-04, 6.1226e-04,
          9.4843e-04, 5.9652e-04, 1.3151e-03, 2.8954e-03, 8.2350e-04, 1.3123e-03,
          3.0861e-03, 7.0000e-04, 1.6356e-04, 9.9945e-04, 1.6565e-03, 1.4710e-04,
          1.7238e-04, 2.3603e-04, 3.2568e-04, 1.4391e-03, 2.2316e-03, 2.8839e-03,
          9.9373e-04, 6.5279e-04, 6.8045e-04, 4.0722e-04, 9.9373e-04, 3.2616e-04,
          1.6571e-02, 3.6869e-03, 8.2245e-03, 3.9101e-03, 2.3460e-03, 4.4975e-03,
          2.2840e-04, 4.9067e-04, 2.6917e-04, 2.5678e-04, 4.8232e-04, 1.0366e-03,
          1.6441e-03, 1.0729e-03, 3.6073e-04, 1.3027e-03, 1.2589e-03, 7.0572e-04,
          8.8072e-04, 5.3358e-04, 1.2598e-03, 4.1151e-04, 1.9798e-03, 7.9422e-03,
          2.5654e-03, 1.8892e-03, 2.2324e-02, 2.5692e-03, 4.5633e-04, 4.1437e-04,
          1.3533e-03, 4.9400e-04, 6.9904e-04, 1.0767e-03, 5.6267e-03, 1.1702e-03,
          7.2145e-04, 5.0163e-04, 5.9223e-04, 7.5531e-04, 1.2283e-03, 4.7226e-03,
          8.3256e-04, 1.3428e-03, 2.0390e-03, 3.9978e-03, 4.0131e-03, 2.7790e-03,
          3.0727e-03, 8.5373e-03, 2.0466e-03, 3.6764e-04, 4.4489e-04, 1.5278e-03,
          3.0556e-03, 1.7262e-04, 1.1864e-03, 2.4002e-02, 1.5268e-03, 1.1244e-03,
          6.6805e-04, 1.1997e-03, 2.1362e-03, 7.4768e-04, 1.3857e-03, 1.0414e-03,
          5.1117e-04, 1.0366e-03, 6.1941e-04, 1.3065e-03, 3.6106e-03, 8.9645e-03,
          2.4357e-03, 8.4162e-04, 4.3058e-04, 7.5293e-04, 1.7233e-03, 3.4618e-04,
          7.2098e-04, 5.2299e-03, 2.6398e-03, 1.1644e-03, 2.6855e-03, 5.3940e-03,
          4.1695e-03, 4.7326e-04, 8.8453e-04, 1.1292e-03, 3.9368e-03, 2.2297e-03,
          1.0881e-03, 5.7077e-04, 5.0449e-04, 1.3123e-03, 1.8539e-03, 5.7526e-03,
          8.3208e-04, 5.4169e-04, 4.2510e-04, 8.5831e-04, 9.1028e-04, 2.0275e-03,
          7.6437e-04, 6.7759e-04, 3.5906e-04, 1.3523e-03, 7.5951e-03, 2.7428e-03,
          7.4816e-04, 1.6422e-03, 4.3702e-04, 8.0013e-04, 8.1635e-04, 1.7738e-03,
          4.7421e-04, 4.6611e-04, 1.1501e-03, 1.2798e-03, 1.1391e-02, 4.1885e-03,
          5.6362e-04, 8.8310e-04, 5.4073e-04, 7.0477e-04, 3.1543e-04, 8.9979e-04,
          9.7275e-04, 1.4286e-03, 3.7613e-03, 2.0096e-02, 1.6769e-02, 1.4963e-03,
          1.3361e-03, 1.3285e-03, 2.7313e-03, 8.3828e-04, 2.4261e-03, 2.4605e-03,
          1.8549e-03, 2.0657e-03, 3.0003e-03, 8.6670e-03, 6.2609e-04, 7.2718e-04,
          1.2484e-03, 4.1580e-04, 3.0613e-04, 4.4942e-04, 2.7199e-03, 1.6651e-03,
          1.4362e-03, 3.4561e-03, 2.0325e-02, 2.3849e-02, 8.5020e-04, 1.6031e-03,
          1.1454e-03, 6.3658e-04, 5.2738e-04, 4.0932e-03, 1.9474e-03, 1.5726e-03,
          1.0080e-03, 9.8495e-03, 4.8447e-03, 1.2115e-02, 5.4884e-04, 1.9464e-03,
          2.1191e-03, 1.0481e-03, 1.5516e-03, 3.7422e-03, 1.6747e-03, 2.0733e-03,
          7.0000e-03, 1.5078e-03, 3.1681e-03, 3.7975e-03, 9.2239e-03, 3.9368e-03,
          1.9054e-03, 8.4686e-03, 1.3268e-02, 1.6022e-02, 1.1959e-03, 1.2197e-03,
          4.1870e-02, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05,
          3.3975e-05, 3.3975e-05, 3.3975e-05, 1.7297e-04, 3.3975e-05, 3.3975e-05,
          3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05, 9.8133e-04, 1.7297e-04,
          1.7297e-04, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05,
          3.3975e-05, 1.7297e-04, 3.3975e-05, 3.3975e-05, 1.7297e-04, 1.7297e-04,
          3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05, 1.7297e-04, 1.7297e-04,
          1.7297e-04, 1.7297e-04, 3.3975e-05, 3.3975e-05, 1.7297e-04, 3.3975e-05,
          1.7297e-04, 3.3975e-05, 3.3975e-05, 9.3222e-05, 3.3975e-05, 3.3975e-05,
          1.7297e-04, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05,
          1.7297e-04, 1.7297e-04, 5.0545e-04, 3.3975e-05, 3.3975e-05, 3.3975e-05,
          1.7297e-04, 3.3975e-05, 1.7297e-04, 3.3975e-05, 1.7297e-04, 1.7297e-04,
          3.3975e-05, 1.6756e-03, 3.3975e-05, 3.3975e-05, 3.3975e-05, 3.3975e-05,
          1.7297e-04, 1.7297e-04, 1.7297e-04, 1.7297e-04, 1.7297e-04, 3.3975e-05,
          3.3975e-05, 5.9557e-04, 5.9557e-04, 6.3229e-04, 1.7297e-04, 1.7297e-04,
          1.7297e-04, 1.7297e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.2610e-03, 5.5298e-02, 1.9012e-02, 1.4961e-02, 3.3295e-02,
          3.6011e-02, 1.1497e-02, 1.1244e-03, 5.2595e-04, 9.4080e-04, 2.0065e-03,
          6.8951e-04, 2.5520e-03, 1.8301e-03, 8.9455e-04, 1.1797e-03, 5.2032e-03,
          4.0283e-03, 5.7554e-04, 4.6682e-04, 4.8971e-04, 1.1740e-03, 2.8877e-03,
          2.6665e-03, 1.0309e-03, 9.2163e-03, 4.3869e-03, 8.0252e-04, 5.1079e-03,
          3.4351e-03, 2.1877e-03, 1.6947e-03, 1.2150e-03, 7.6962e-04, 7.6580e-04,
          5.6934e-04, 3.0422e-04, 9.2459e-04, 8.3923e-04, 6.7711e-04, 8.4782e-04,
          5.8556e-04, 4.4298e-04, 5.8842e-04, 3.6454e-04, 5.7678e-03, 1.7662e-03,
          1.6966e-03, 1.3161e-03, 3.6869e-03, 1.1101e-03, 1.5306e-03, 6.1798e-04,
          4.4656e-04, 9.1648e-04, 4.1056e-04, 2.5225e-04, 7.8583e-04, 4.4107e-04,
          5.3024e-04, 2.4548e-03, 9.7370e-04, 1.0357e-03, 5.4550e-04, 2.0885e-03,
          9.9087e-04, 5.4502e-04, 5.9557e-04, 3.0875e-04, 2.9182e-04, 5.3644e-04,
          6.3467e-04, 2.0638e-03, 5.7173e-04, 2.3746e-03, 4.9543e-04, 3.7766e-04,
          9.5892e-04, 2.2995e-04, 3.7527e-04, 4.7183e-04, 4.1652e-04, 2.9182e-04,
          3.0184e-04, 3.0918e-03, 7.7677e-04, 1.0262e-03, 4.6229e-04, 4.9734e-04,
          2.2678e-03, 1.0328e-03, 4.2033e-04, 1.4048e-03, 5.3501e-04, 6.7234e-04,
          1.6184e-03, 2.9516e-04, 1.3638e-03, 1.0853e-03, 2.5535e-04, 2.8205e-04,
          1.4563e-03, 4.0865e-04, 3.8004e-04, 4.4799e-04, 1.0071e-03, 2.5158e-03,
          2.5063e-03, 5.0163e-04, 1.3208e-03, 4.8423e-04, 6.7854e-04, 6.0844e-04,
          7.4053e-04, 7.2908e-04, 2.1172e-03, 2.4643e-03, 1.1225e-03, 3.2539e-03,
          9.6512e-03, 6.2943e-04, 1.8346e-04, 6.3801e-04, 1.3933e-03, 1.7345e-04,
          1.9312e-04, 2.7823e-04, 3.9101e-04, 1.1749e-03, 1.7366e-03, 3.2482e-03,
          1.1044e-03, 5.7316e-04, 5.5647e-04, 3.8719e-04, 9.1171e-04, 4.3082e-04,
          8.1863e-03, 6.9351e-03, 5.9013e-03, 3.8395e-03, 2.4261e-03, 8.3847e-03,
          2.6488e-04, 3.2663e-04, 2.5916e-04, 2.5368e-04, 4.9496e-04, 9.2125e-04,
          1.2016e-03, 9.4557e-04, 4.1008e-04, 1.8387e-03, 1.8072e-03, 9.9182e-04,
          8.9216e-04, 6.9046e-04, 8.1348e-04, 6.1464e-04, 2.4014e-03, 1.2444e-02,
          2.0695e-03, 1.5669e-03, 1.0582e-02, 1.7986e-03, 4.8137e-04, 3.4213e-04,
          9.7847e-04, 3.9434e-04, 6.0272e-04, 7.4625e-04, 4.1618e-03, 1.0452e-03,
          9.3794e-04, 6.8331e-04, 6.4230e-04, 9.8228e-04, 1.2388e-03, 3.1872e-03,
          7.6914e-04, 1.2627e-03, 1.8520e-03, 2.6836e-03, 1.9302e-03, 1.6127e-03,
          1.9493e-03, 1.4847e-02, 1.5163e-03, 3.9029e-04, 5.0545e-04, 1.1864e-03,
          1.9360e-03, 1.8609e-04, 9.1934e-04, 1.0559e-02, 1.0300e-03, 9.9659e-04,
          7.0477e-04, 1.9445e-03, 2.3308e-03, 9.5749e-04, 1.1063e-03, 1.0538e-03,
          8.1444e-04, 1.3704e-03, 1.0166e-03, 1.0376e-03, 2.8896e-03, 5.4207e-03,
          2.0161e-03, 9.5987e-04, 4.5896e-04, 6.5422e-04, 1.3332e-03, 3.9148e-04,
          6.8808e-04, 4.9973e-03, 1.9293e-03, 8.5592e-04, 1.9474e-03, 2.1286e-02,
          4.1542e-03, 7.7438e-04, 1.1959e-03, 1.0262e-03, 4.0398e-03, 1.7424e-03,
          1.1873e-03, 8.8692e-04, 8.4162e-04, 1.6527e-03, 1.8482e-03, 1.3130e-02,
          8.2064e-04, 4.5633e-04, 5.2118e-04, 7.5912e-04, 8.8644e-04, 2.2945e-03,
          1.0223e-03, 8.2445e-04, 4.1389e-04, 1.2903e-03, 1.9958e-02, 6.5269e-03,
          1.2074e-03, 2.1763e-03, 8.3208e-04, 1.2531e-03, 1.3332e-03, 2.1553e-03,
          8.6212e-04, 8.4114e-04, 1.8053e-03, 1.7147e-03, 5.9128e-03, 4.0245e-03,
          6.0606e-04, 6.7949e-04, 4.8876e-04, 6.1798e-04, 3.7026e-04, 1.0986e-03,
          8.8167e-04, 9.1600e-04, 5.3864e-03, 4.0283e-02, 2.4338e-02, 1.9312e-03,
          2.1839e-03, 1.4620e-03, 3.2825e-03, 1.3666e-03, 3.1567e-03, 3.1357e-03,
          3.1433e-03, 1.7691e-03, 7.8583e-03, 2.2568e-02, 6.7616e-04, 8.8644e-04,
          9.8515e-04, 4.7731e-04, 3.7622e-04, 4.9686e-04, 5.2261e-03, 1.3866e-03,
          1.4706e-03, 2.4757e-03, 8.3771e-03, 8.0719e-03, 1.2426e-03, 2.5177e-03,
          1.7967e-03, 9.9850e-04, 7.4768e-04, 3.7251e-03, 2.5902e-03, 3.2234e-03,
          1.8301e-03, 6.9313e-03, 3.9673e-03, 1.5427e-02, 6.3324e-04, 1.4515e-03,
          1.7004e-03, 1.9131e-03, 1.4248e-03, 5.5008e-03, 1.0519e-03, 1.3638e-03,
          7.9575e-03, 1.5202e-03, 5.5962e-03, 3.4351e-03, 8.6670e-03, 3.6488e-03,
          2.3365e-03, 5.9242e-03, 1.5686e-02, 1.5434e-02, 2.3365e-03, 2.1019e-03,
          1.3771e-02, 4.2725e-02, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05, 2.4223e-04, 4.3154e-05,
          4.3154e-05, 2.5296e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          4.3154e-05, 2.4223e-04, 4.3154e-05, 4.3154e-05, 2.4223e-04, 2.5296e-04,
          4.3154e-05, 4.3154e-05, 2.5296e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          2.4223e-04, 4.5204e-04, 4.3154e-05, 4.3154e-05, 2.4223e-04, 4.3154e-05,
          4.5204e-04, 4.3154e-05, 4.3154e-05, 2.0051e-04, 4.3154e-05, 4.3154e-05,
          2.4223e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.5204e-04, 4.3154e-05,
          2.4223e-04, 4.3154e-05, 4.5204e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          2.4223e-04, 4.3154e-05, 4.3154e-05, 4.3154e-05, 2.4223e-04, 4.3154e-05,
          4.3154e-05, 2.3060e-03, 4.3154e-05, 4.3154e-05, 4.3154e-05, 4.3154e-05,
          2.4223e-04, 2.4223e-04, 2.4223e-04, 3.1161e-04, 2.4223e-04, 4.3154e-05,
          4.3154e-05, 2.4223e-04, 2.4223e-04, 1.1263e-03, 4.3154e-05, 4.3154e-05,
          2.4223e-04, 2.4223e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.9662e-03, 4.8340e-02, 1.8219e-02, 2.5620e-02, 3.3600e-02,
          3.6102e-02, 1.2459e-02, 1.8339e-03, 7.7200e-04, 1.9140e-03, 3.2425e-03,
          1.1225e-03, 3.9711e-03, 7.4348e-03, 1.0891e-03, 1.5326e-03, 1.6571e-02,
          1.1551e-02, 7.5388e-04, 1.1311e-03, 7.5006e-04, 1.7004e-03, 8.3237e-03,
          3.4752e-03, 9.7656e-04, 1.0811e-02, 9.6207e-03, 2.0752e-03, 1.2169e-02,
          1.5129e-02, 3.4294e-03, 4.8141e-03, 1.9608e-03, 1.9684e-03, 9.3126e-04,
          8.1730e-04, 4.1127e-04, 1.7242e-03, 1.5974e-03, 1.7939e-03, 7.5340e-04,
          5.3787e-04, 3.9387e-04, 6.0892e-04, 3.2520e-04, 1.1131e-02, 1.9398e-03,
          5.1041e-03, 2.3365e-03, 5.1384e-03, 1.4973e-03, 2.9564e-03, 9.4604e-04,
          5.6076e-04, 1.2245e-03, 6.6662e-04, 3.8576e-04, 9.8228e-04, 5.4312e-04,
          6.1846e-04, 4.1695e-03, 9.2316e-04, 9.9850e-04, 4.3535e-04, 1.8501e-03,
          1.4048e-03, 7.4816e-04, 8.4591e-04, 2.7132e-04, 3.6454e-04, 4.6611e-04,
          7.1669e-04, 4.6997e-03, 8.3876e-04, 4.0436e-03, 7.0381e-04, 5.9366e-04,
          1.1778e-03, 3.4714e-04, 5.7077e-04, 7.6056e-04, 8.2350e-04, 3.2496e-04,
          3.4404e-04, 3.3989e-03, 7.4148e-04, 1.5602e-03, 7.5579e-04, 6.4373e-04,
          3.2196e-03, 1.1654e-03, 2.6774e-04, 1.7519e-03, 3.5357e-04, 6.0272e-04,
          2.8896e-03, 4.4966e-04, 1.9093e-03, 1.6069e-03, 3.4547e-04, 3.8409e-04,
          2.4014e-03, 8.1301e-04, 7.6962e-04, 5.6982e-04, 1.1892e-03, 2.1057e-03,
          2.1610e-03, 4.0722e-04, 1.3285e-03, 3.2544e-04, 8.8263e-04, 8.4591e-04,
          6.1083e-04, 4.6325e-04, 1.2856e-03, 1.9684e-03, 8.7690e-04, 2.1667e-03,
          4.5586e-03, 8.9931e-04, 2.5105e-04, 1.0872e-03, 1.7443e-03, 2.4557e-04,
          2.4676e-04, 3.8671e-04, 5.0640e-04, 1.2379e-03, 2.5082e-03, 2.7905e-03,
          1.2426e-03, 9.9277e-04, 6.4850e-04, 4.2963e-04, 9.5367e-04, 3.4976e-04,
          8.5297e-03, 3.7994e-03, 3.5896e-03, 5.6267e-03, 6.8855e-03, 6.3210e-03,
          3.5906e-04, 3.4070e-04, 3.7885e-04, 3.3593e-04, 8.5783e-04, 9.7847e-04,
          1.7605e-03, 8.9741e-04, 3.8743e-04, 1.9217e-03, 1.6031e-03, 7.1001e-04,
          9.2411e-04, 6.3372e-04, 8.5115e-04, 5.1928e-04, 1.2913e-03, 4.6425e-03,
          1.9102e-03, 2.0695e-03, 1.7822e-02, 3.6201e-03, 7.2002e-04, 6.3419e-04,
          1.5535e-03, 7.6437e-04, 8.0729e-04, 7.5579e-04, 8.8272e-03, 1.4334e-03,
          5.4169e-04, 4.9067e-04, 8.6641e-04, 8.5926e-04, 1.1339e-03, 6.6299e-03,
          9.9564e-04, 1.0500e-03, 1.9608e-03, 3.4447e-03, 3.2539e-03, 1.3075e-03,
          2.0905e-03, 4.8065e-03, 2.5368e-03, 5.8460e-04, 8.5068e-04, 1.4877e-03,
          3.0708e-03, 2.6965e-04, 1.3456e-03, 1.7044e-02, 1.7433e-03, 1.2608e-03,
          9.5749e-04, 1.4238e-03, 1.6794e-03, 8.6212e-04, 1.6661e-03, 8.7547e-04,
          6.5899e-04, 9.2173e-04, 5.3310e-04, 1.1492e-03, 2.7237e-03, 5.6763e-03,
          1.8158e-03, 1.4620e-03, 7.2670e-04, 1.0662e-03, 1.7319e-03, 5.2738e-04,
          9.2840e-04, 5.8441e-03, 3.1681e-03, 7.7724e-04, 2.8133e-03, 4.7989e-03,
          5.0468e-03, 5.5027e-04, 7.5150e-04, 1.3838e-03, 3.1128e-03, 2.0828e-03,
          1.3990e-03, 5.3501e-04, 4.1866e-04, 7.7868e-04, 1.3628e-03, 2.7161e-03,
          1.6985e-03, 8.5926e-04, 6.2466e-04, 7.0524e-04, 9.1028e-04, 1.5526e-03,
          8.4877e-04, 9.2363e-04, 4.3130e-04, 1.5659e-03, 4.3182e-03, 3.7708e-03,
          7.7200e-04, 1.3695e-03, 3.7766e-04, 4.8423e-04, 6.6137e-04, 1.3809e-03,
          4.9973e-04, 3.9625e-04, 9.9659e-04, 9.7847e-04, 4.9858e-03, 6.9466e-03,
          8.6546e-04, 1.3828e-03, 5.7983e-04, 6.7902e-04, 3.5214e-04, 9.1982e-04,
          9.4128e-04, 1.4067e-03, 2.4967e-03, 1.0674e-02, 3.7201e-02, 1.0500e-03,
          1.3247e-03, 2.0256e-03, 2.0123e-03, 6.4516e-04, 1.8988e-03, 1.4591e-03,
          8.3303e-04, 7.2861e-04, 2.4815e-03, 1.1932e-02, 1.2074e-03, 1.2836e-03,
          2.2850e-03, 6.1369e-04, 5.3740e-04, 7.4005e-04, 2.2049e-03, 1.4162e-03,
          9.5320e-04, 4.8141e-03, 1.0895e-02, 1.1078e-02, 8.8501e-04, 1.1473e-03,
          9.9850e-04, 6.3372e-04, 5.9652e-04, 2.5005e-03, 1.5306e-03, 9.2077e-04,
          8.7976e-04, 2.7657e-03, 3.1719e-03, 1.1414e-02, 9.5701e-04, 1.8063e-03,
          1.7633e-03, 1.2960e-03, 1.7691e-03, 2.8286e-03, 9.8038e-04, 1.1930e-03,
          4.7760e-03, 1.0033e-03, 3.8586e-03, 3.0727e-03, 1.0086e-02, 2.2907e-03,
          1.0509e-03, 5.4779e-03, 7.4692e-03, 6.8474e-03, 9.6703e-04, 1.1978e-03,
          8.1482e-03, 5.3101e-03, 5.6091e-02, 3.2902e-05, 3.2902e-05, 3.2902e-05,
          3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05,
          1.4567e-04, 3.2902e-05, 3.2902e-05, 3.2902e-05, 1.6570e-04, 3.2902e-05,
          3.2902e-05, 1.4567e-04, 3.2902e-05, 3.2902e-05, 4.2868e-04, 3.2902e-05,
          3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05, 2.7847e-04, 2.7847e-04,
          1.4567e-04, 3.2902e-05, 2.7847e-04, 2.7847e-04, 3.2902e-05, 9.2506e-04,
          2.7847e-04, 2.7847e-04, 2.7847e-04, 3.2902e-05, 1.6570e-04, 3.2902e-05,
          2.7847e-04, 3.2902e-05, 3.2902e-05, 9.2506e-04, 3.2902e-05, 3.2902e-05,
          3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05, 2.7847e-04, 3.2902e-05,
          1.7471e-03, 3.2902e-05, 2.7847e-04, 3.2902e-05, 3.2902e-05, 3.2902e-05,
          2.7847e-04, 3.2902e-05, 3.2902e-05, 3.2902e-05, 2.7847e-04, 9.2506e-04,
          3.2902e-05, 1.6346e-03, 3.2902e-05, 3.2902e-05, 3.2902e-05, 3.2902e-05,
          1.6570e-04, 1.6570e-04, 1.6570e-04, 3.2902e-05, 3.2902e-05, 3.2902e-05,
          3.2902e-05, 1.6570e-04, 1.6570e-04, 1.6570e-04, 1.6570e-04, 1.6570e-04,
          3.2902e-05, 3.2902e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1765e-03, 4.9438e-02, 1.9974e-02, 1.9943e-02, 7.2937e-02,
          3.2135e-02, 2.1347e-02, 5.8126e-04, 4.4632e-04, 1.5745e-03, 5.5542e-03,
          1.1387e-03, 6.5231e-04, 6.2275e-04, 9.4652e-04, 3.8671e-04, 8.6021e-04,
          4.0770e-04, 2.3627e-04, 5.7125e-04, 5.2547e-04, 4.2295e-04, 5.0163e-04,
          4.5662e-03, 1.5516e-03, 1.3695e-02, 2.5177e-03, 1.9627e-03, 8.5402e-04,
          7.5626e-04, 8.0681e-04, 4.1938e-04, 1.1368e-03, 5.3883e-04, 3.7956e-04,
          1.6394e-03, 1.7233e-03, 1.0090e-03, 6.7329e-04, 5.7793e-04, 1.6632e-03,
          1.5421e-03, 1.0900e-03, 1.1549e-03, 7.3624e-04, 6.5327e-04, 1.3256e-03,
          3.4499e-04, 6.5041e-04, 6.6376e-03, 1.2217e-03, 7.1001e-04, 4.8637e-04,
          4.8876e-04, 4.5395e-04, 1.0586e-03, 3.2673e-03, 1.3018e-03, 3.3436e-03,
          2.7313e-03, 6.8474e-04, 8.9598e-04, 5.7793e-04, 1.2674e-03, 1.9159e-03,
          1.1549e-03, 4.7612e-04, 6.2227e-04, 7.5436e-04, 8.3876e-04, 9.8228e-04,
          1.0357e-03, 2.5158e-03, 1.7233e-03, 9.0694e-04, 4.0579e-04, 2.6870e-04,
          7.6866e-04, 3.9043e-03, 1.9379e-03, 1.4601e-03, 1.6747e-03, 1.5011e-03,
          1.6804e-03, 8.3780e-04, 6.5613e-04, 1.1950e-03, 5.1641e-04, 7.6628e-04,
          2.0332e-03, 1.3533e-03, 6.1703e-04, 1.2875e-03, 1.5831e-03, 1.4658e-03,
          1.1683e-03, 9.5367e-04, 4.8780e-04, 1.9073e-03, 7.3700e-03, 4.3144e-03,
          4.3182e-03, 8.1205e-04, 7.9107e-04, 2.0847e-03, 1.6060e-03, 7.1907e-04,
          6.6948e-04, 1.1768e-03, 1.8978e-03, 6.7759e-04, 1.0881e-03, 6.9618e-04,
          1.1835e-03, 1.3561e-03, 4.0579e-04, 1.9817e-03, 1.7471e-03, 2.6455e-03,
          1.9684e-03, 3.8834e-03, 3.2444e-03, 3.2401e-04, 1.9817e-03, 9.7179e-04,
          3.5591e-03, 1.2894e-03, 1.3838e-03, 1.4467e-03, 4.0984e-04, 7.5674e-04,
          1.1826e-03, 7.5006e-04, 2.9063e-04, 3.0231e-04, 9.9182e-04, 7.9918e-04,
          1.0128e-03, 4.9973e-04, 5.6458e-04, 1.3599e-03, 9.8228e-04, 1.2388e-03,
          4.5967e-03, 8.5640e-04, 1.7738e-03, 2.8267e-03, 2.1038e-03, 2.5940e-03,
          3.8052e-03, 1.5526e-03, 2.9125e-03, 7.2098e-04, 1.2760e-03, 1.8778e-03,
          1.0405e-03, 9.8515e-04, 2.9755e-04, 7.1907e-04, 1.9264e-03, 5.2881e-04,
          1.3485e-03, 1.0672e-03, 8.1062e-04, 1.1911e-03, 1.0590e-02, 2.4223e-03,
          3.7518e-03, 1.7653e-03, 1.6890e-03, 8.2731e-04, 7.6723e-04, 1.7118e-03,
          1.6718e-03, 1.7300e-03, 4.4899e-03, 2.0657e-03, 1.5078e-03, 7.7581e-04,
          8.7261e-04, 5.9557e-04, 1.2121e-03, 5.3072e-04, 4.7970e-04, 6.3658e-04,
          9.4748e-04, 5.6362e-04, 1.0929e-03, 1.1009e-02, 2.3842e-03, 1.3361e-03,
          6.3753e-04, 4.2572e-03, 1.0290e-03, 1.2789e-03, 1.6603e-03, 1.2741e-03,
          3.7899e-03, 1.5707e-03, 1.3390e-03, 9.1839e-04, 9.0075e-04, 8.1635e-04,
          9.7466e-04, 1.1902e-03, 1.8091e-03, 4.5013e-04, 2.2659e-03, 4.2963e-04,
          5.6791e-04, 8.0338e-03, 1.9974e-02, 2.0065e-03, 3.4046e-03, 3.0022e-03,
          1.9512e-03, 1.1358e-03, 1.1425e-03, 5.3692e-04, 1.4658e-03, 8.4877e-04,
          2.2182e-03, 1.8349e-03, 1.4467e-03, 7.7820e-04, 2.1400e-03, 9.9182e-04,
          8.5735e-04, 8.3828e-04, 1.2083e-03, 2.2964e-03, 1.1406e-03, 4.9067e-04,
          2.8351e-02, 3.2101e-03, 3.7518e-03, 1.4238e-03, 7.2975e-03, 7.2479e-03,
          6.0730e-03, 7.0877e-03, 3.1700e-03, 2.1133e-03, 2.1648e-03, 2.1267e-03,
          2.2812e-03, 1.0061e-03, 1.1787e-03, 9.3746e-04, 1.4000e-03, 1.2503e-03,
          7.4911e-04, 1.6937e-03, 2.3708e-03, 1.4658e-03, 7.9346e-04, 2.7752e-03,
          1.2939e-02, 2.0313e-03, 3.4027e-03, 2.2221e-03, 3.9253e-03, 5.3596e-03,
          1.8559e-03, 6.6185e-04, 6.2466e-04, 1.4753e-03, 1.7471e-03, 9.6607e-04,
          1.8454e-03, 1.5430e-03, 1.7986e-03, 1.3466e-03, 2.0638e-03, 2.1973e-03,
          3.3550e-03, 1.3113e-03, 1.2913e-03, 8.6212e-04, 1.3855e-02, 5.9624e-03,
          2.8381e-03, 1.0178e-02, 9.9869e-03, 8.1100e-03, 2.1915e-03, 1.5268e-03,
          3.1948e-03, 2.3575e-03, 8.9169e-04, 9.9373e-04, 2.7962e-03, 2.4910e-03,
          3.0193e-03, 1.8253e-03, 1.5144e-03, 1.4143e-03, 1.9150e-03, 3.1853e-03,
          2.5959e-03, 6.4659e-04, 9.1505e-04, 2.9068e-03, 2.7634e-02, 4.5204e-03,
          2.2411e-03, 6.3744e-03, 2.7237e-03, 6.3419e-04, 8.7118e-04, 7.3910e-04,
          6.5470e-04, 1.8702e-03, 7.9193e-03, 1.1911e-03, 2.2717e-03, 1.3685e-03,
          1.3256e-03, 3.1185e-03, 1.1749e-03, 8.2636e-04, 4.9973e-03, 4.0970e-03,
          9.1648e-04, 6.9904e-04, 1.9341e-03, 6.6833e-02, 7.1347e-05, 2.1112e-04,
          7.1347e-05, 7.1347e-05, 2.0576e-04, 7.1347e-05, 2.2554e-04, 2.2554e-04,
          2.2554e-04, 7.1347e-05, 3.6526e-04, 7.1347e-05, 7.1347e-05, 3.4547e-04,
          2.2554e-04, 2.2554e-04, 4.0388e-04, 2.2554e-04, 2.2554e-04, 2.2554e-04,
          2.2554e-04, 7.1347e-05, 7.1347e-05, 7.1347e-05, 3.4547e-04, 2.1112e-04,
          7.1347e-05, 2.2554e-04, 2.2554e-04, 2.2554e-04, 3.4547e-04, 7.1347e-05,
          4.9973e-04, 3.4547e-04, 2.2554e-04, 2.2554e-04, 2.2554e-04, 7.1347e-05,
          3.6526e-04, 2.0576e-04, 7.1347e-05, 7.1347e-05, 7.1347e-05, 2.2554e-04,
          7.1347e-05, 7.1347e-05, 7.1347e-05, 2.2554e-04, 3.4547e-04, 7.1347e-05,
          2.0576e-04, 2.1112e-04, 3.6526e-04, 2.2554e-04, 2.2554e-04, 2.5158e-03,
          4.9973e-04, 2.2554e-04, 4.9973e-04, 7.0429e-04, 2.2554e-04, 2.2554e-04,
          2.2554e-04, 2.5158e-03, 2.2554e-04, 7.1347e-05, 7.1347e-05, 7.1347e-05,
          2.0576e-04, 2.0576e-04, 2.0576e-04, 7.1347e-05, 2.0576e-04, 7.1347e-05,
          7.1347e-05, 7.1347e-05, 7.1347e-05, 7.1347e-05, 7.1347e-05, 7.1347e-05,
          7.1347e-05, 7.1347e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4670e-03, 5.2002e-02, 1.2527e-02, 1.1665e-02, 3.8574e-02,
          4.0283e-02, 8.9111e-03, 6.6519e-04, 5.2691e-04, 1.7481e-03, 3.5706e-03,
          1.1063e-03, 1.0195e-03, 3.9411e-04, 2.1362e-03, 5.4884e-04, 1.0176e-03,
          5.4216e-04, 3.3569e-04, 2.2328e-04, 2.7704e-04, 2.9922e-04, 7.7486e-04,
          3.4561e-03, 3.3207e-03, 6.8626e-03, 2.2564e-03, 5.8651e-04, 1.3943e-03,
          5.7602e-04, 1.4334e-03, 4.9543e-04, 2.5768e-03, 1.2646e-03, 6.3658e-04,
          2.1896e-03, 7.4816e-04, 1.4219e-03, 1.3580e-03, 3.2640e-04, 1.5612e-03,
          6.5184e-04, 5.8699e-04, 4.8256e-04, 6.1846e-04, 1.2169e-03, 3.3646e-03,
          2.9254e-04, 3.4165e-04, 3.0327e-03, 1.5154e-03, 1.1301e-03, 6.6805e-04,
          3.8195e-04, 6.4373e-04, 5.4169e-04, 1.2693e-03, 9.6178e-04, 1.4534e-03,
          1.3008e-03, 7.3767e-04, 6.4278e-04, 9.1362e-04, 8.6021e-04, 5.6496e-03,
          8.6498e-04, 4.5633e-04, 8.1921e-04, 4.2367e-04, 3.3617e-04, 4.3249e-04,
          4.9925e-04, 3.5019e-03, 7.6866e-04, 1.6165e-03, 3.2163e-04, 3.0637e-04,
          6.8760e-04, 1.6508e-03, 1.1292e-03, 2.0733e-03, 1.0080e-03, 9.1314e-04,
          7.4577e-04, 1.4305e-03, 4.5085e-04, 1.2503e-03, 4.2129e-04, 5.2643e-04,
          2.2984e-03, 1.5421e-03, 3.8791e-04, 1.4019e-03, 1.4896e-03, 7.6485e-04,
          6.7091e-04, 4.5609e-04, 8.1348e-04, 2.2602e-03, 2.7084e-03, 2.3479e-03,
          4.0512e-03, 7.8011e-04, 8.0585e-04, 2.3232e-03, 4.0932e-03, 8.2827e-04,
          1.1339e-03, 6.0415e-04, 2.0676e-03, 4.6396e-04, 8.1968e-04, 6.1846e-04,
          4.1656e-03, 1.9894e-03, 3.6597e-04, 7.2327e-03, 1.0853e-03, 1.4677e-03,
          1.7805e-03, 5.8441e-03, 1.3533e-03, 4.2725e-04, 4.7951e-03, 6.5899e-04,
          1.7939e-03, 7.6246e-04, 1.0500e-03, 4.4670e-03, 8.9216e-04, 1.3304e-03,
          1.1358e-03, 5.9748e-04, 6.8712e-04, 2.7704e-04, 1.0681e-03, 4.9353e-04,
          2.5349e-03, 5.2929e-04, 8.7070e-04, 1.4868e-03, 1.6155e-03, 1.6174e-03,
          2.1992e-03, 9.8991e-04, 2.3136e-03, 1.7986e-03, 1.5535e-03, 1.2222e-02,
          4.7073e-03, 4.5815e-03, 4.2000e-03, 1.1244e-03, 1.0366e-03, 2.0332e-03,
          1.1330e-03, 9.9087e-04, 4.9019e-04, 4.7135e-04, 5.1231e-03, 9.0361e-04,
          3.4180e-03, 2.0370e-03, 1.6279e-03, 2.4433e-03, 5.1460e-03, 1.6394e-03,
          3.3474e-03, 1.5154e-03, 5.5275e-03, 2.5330e-03, 1.2951e-03, 3.5324e-03,
          3.5400e-03, 1.5345e-03, 2.8915e-03, 1.2159e-03, 2.7924e-03, 1.4868e-03,
          1.5278e-03, 1.3189e-03, 3.0041e-03, 9.3603e-04, 7.2193e-04, 1.3027e-03,
          1.4067e-03, 6.6996e-04, 1.1320e-03, 5.2452e-03, 1.7948e-03, 3.4618e-03,
          1.1549e-03, 2.2717e-03, 2.7370e-03, 2.5139e-03, 1.9312e-03, 3.1967e-03,
          2.8458e-03, 1.4038e-03, 3.2253e-03, 9.0456e-04, 8.5449e-04, 1.4191e-03,
          6.5994e-04, 1.3514e-03, 1.5907e-03, 7.0477e-04, 2.8534e-03, 7.9393e-04,
          7.0953e-04, 6.6910e-03, 5.2605e-03, 2.0046e-03, 1.1047e-02, 2.4414e-03,
          3.8052e-03, 2.7103e-03, 1.8959e-03, 1.1969e-03, 4.3030e-03, 1.6060e-03,
          4.0970e-03, 9.5463e-04, 1.8606e-03, 8.3876e-04, 5.4665e-03, 2.2087e-03,
          7.9536e-04, 7.3528e-04, 1.1330e-03, 3.8013e-03, 3.5744e-03, 7.4434e-04,
          6.7253e-03, 2.5501e-03, 2.1896e-03, 2.8400e-03, 2.0340e-02, 1.4503e-02,
          4.4441e-03, 3.8853e-03, 2.9297e-03, 3.5496e-03, 5.3673e-03, 2.3136e-03,
          1.5125e-03, 1.3809e-03, 8.7500e-04, 1.1997e-03, 1.2436e-03, 3.0899e-03,
          5.3787e-04, 1.1702e-03, 1.3399e-03, 2.1915e-03, 2.0294e-03, 2.7027e-03,
          9.1782e-03, 1.9627e-03, 7.0839e-03, 6.3515e-03, 3.6583e-03, 5.3558e-03,
          5.0392e-03, 1.1377e-03, 1.0786e-03, 3.2253e-03, 2.5921e-03, 2.3518e-03,
          1.8492e-03, 1.3504e-03, 4.1199e-03, 1.5602e-03, 3.5534e-03, 1.0643e-02,
          6.3362e-03, 4.1046e-03, 7.8726e-04, 1.1854e-03, 5.1842e-03, 4.2458e-03,
          2.4776e-03, 4.4060e-03, 4.1695e-03, 4.7379e-03, 6.1302e-03, 3.3951e-03,
          6.7024e-03, 5.1765e-03, 2.3422e-03, 1.8635e-03, 1.6365e-03, 2.8019e-03,
          2.0885e-03, 1.1454e-03, 8.6594e-04, 3.4142e-03, 3.5000e-03, 2.0542e-03,
          1.3571e-03, 1.2493e-03, 1.2665e-03, 5.9013e-03, 9.1248e-03, 1.0376e-02,
          5.7411e-03, 4.7836e-03, 5.9738e-03, 1.0900e-03, 1.6909e-03, 1.3199e-03,
          1.3800e-03, 3.6411e-03, 5.3215e-03, 2.5520e-03, 4.3297e-03, 3.1509e-03,
          2.0180e-03, 4.4289e-03, 2.2545e-03, 1.5593e-03, 2.0447e-03, 2.1496e-03,
          1.3227e-03, 9.6321e-04, 2.2621e-03, 2.3117e-02, 4.9927e-02, 1.6201e-04,
          4.5896e-05, 4.5896e-05, 3.0041e-04, 4.5896e-05, 4.5896e-05, 3.2997e-04,
          4.5896e-05, 4.5896e-05, 4.4632e-04, 3.2997e-04, 4.5896e-05, 3.0041e-04,
          4.5896e-05, 4.5896e-05, 4.5896e-05, 3.2997e-04, 4.5896e-05, 6.1226e-04,
          6.1226e-04, 4.5896e-05, 4.5896e-05, 4.5896e-05, 3.0041e-04, 3.0041e-04,
          4.5896e-05, 3.2997e-04, 4.5896e-05, 3.2997e-04, 3.0041e-04, 4.5896e-05,
          3.0041e-04, 3.0041e-04, 4.5896e-05, 4.5896e-05, 1.6201e-04, 4.5896e-05,
          4.4632e-04, 4.6635e-04, 4.5896e-05, 4.5896e-05, 4.5896e-05, 4.5896e-05,
          4.5896e-05, 4.5896e-05, 4.5896e-05, 4.5896e-05, 3.0041e-04, 4.5896e-05,
          1.8418e-04, 3.0041e-04, 3.0041e-04, 4.5896e-05, 4.5896e-05, 6.1226e-04,
          3.0041e-04, 6.1226e-04, 8.6641e-04, 4.5896e-05, 4.5896e-05, 6.1226e-04,
          7.5054e-04, 4.6825e-04, 3.2997e-04, 4.5896e-05, 4.5896e-05, 4.5896e-05,
          1.8418e-04, 1.8418e-04, 1.8418e-04, 4.5896e-05, 4.5896e-05, 4.5896e-05,
          4.5896e-05, 1.6201e-04, 1.8418e-04, 1.8418e-04, 4.5896e-05, 1.8418e-04,
          1.8418e-04, 4.5896e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.9814e-03, 4.7943e-02, 1.5854e-02, 3.0548e-02, 4.0955e-02,
          3.1677e-02, 1.5732e-02, 1.0719e-03, 6.1893e-04, 1.3075e-03, 2.1877e-03,
          8.7166e-04, 2.4204e-03, 6.5498e-03, 8.8882e-04, 8.8406e-04, 6.8474e-03,
          5.3329e-03, 4.3988e-04, 7.8011e-04, 4.9591e-04, 8.6117e-04, 2.9087e-03,
          2.2793e-03, 7.3242e-04, 1.2749e-02, 8.1940e-03, 1.5554e-03, 5.4588e-03,
          1.1780e-02, 2.1381e-03, 2.3308e-03, 1.4248e-03, 1.0891e-03, 9.0790e-04,
          8.5878e-04, 4.6515e-04, 1.2417e-03, 1.0147e-03, 1.7738e-03, 7.3862e-04,
          4.6897e-04, 3.2616e-04, 4.4274e-04, 2.4533e-04, 5.0316e-03, 1.3523e-03,
          3.0785e-03, 1.5421e-03, 3.6793e-03, 1.4057e-03, 1.6279e-03, 7.2479e-04,
          5.6124e-04, 9.8801e-04, 7.2527e-04, 4.7994e-04, 9.4128e-04, 6.2752e-04,
          7.5436e-04, 4.3793e-03, 8.1158e-04, 7.3290e-04, 4.9782e-04, 1.4391e-03,
          1.0338e-03, 4.2987e-04, 5.8746e-04, 1.8847e-04, 1.8573e-04, 3.5048e-04,
          4.0913e-04, 2.7065e-03, 8.9836e-04, 2.1286e-03, 5.3501e-04, 4.4155e-04,
          1.3161e-03, 4.7874e-04, 7.6771e-04, 7.5197e-04, 9.0933e-04, 3.8099e-04,
          4.5013e-04, 2.1477e-03, 8.2254e-04, 8.5449e-04, 5.6744e-04, 8.7357e-04,
          1.7471e-03, 7.8297e-04, 2.6274e-04, 1.0242e-03, 2.4045e-04, 3.6144e-04,
          2.4319e-03, 5.2214e-04, 1.5926e-03, 1.2741e-03, 5.1546e-04, 5.3740e-04,
          1.9112e-03, 4.5371e-04, 6.2037e-04, 6.1989e-04, 9.4128e-04, 4.0169e-03,
          1.5078e-03, 4.1366e-04, 1.1683e-03, 2.8515e-04, 6.6423e-04, 5.7173e-04,
          3.8433e-04, 3.7265e-04, 1.4105e-03, 1.5135e-03, 5.4216e-04, 3.5477e-03,
          1.3985e-02, 1.0500e-03, 3.6025e-04, 5.4407e-04, 1.3390e-03, 3.3665e-04,
          3.1400e-04, 5.7554e-04, 6.9141e-04, 1.2569e-03, 1.6279e-03, 1.8129e-03,
          8.3447e-04, 6.4564e-04, 3.8934e-04, 3.1209e-04, 7.2384e-04, 2.9635e-04,
          6.9122e-03, 7.7209e-03, 3.8643e-03, 3.2730e-03, 2.5921e-03, 1.5839e-02,
          5.3358e-04, 2.1362e-04, 4.0746e-04, 4.0174e-04, 1.1721e-03, 1.1292e-03,
          1.4982e-03, 1.0023e-03, 4.7779e-04, 1.4153e-03, 1.3943e-03, 7.3671e-04,
          7.5006e-04, 5.1451e-04, 4.6897e-04, 4.9734e-04, 1.1272e-03, 7.8812e-03,
          1.4772e-03, 1.0805e-03, 7.9956e-03, 1.5192e-03, 1.1425e-03, 5.3740e-04,
          1.3762e-03, 5.7030e-04, 8.2159e-04, 6.2132e-04, 7.8812e-03, 1.3676e-03,
          6.2609e-04, 6.3801e-04, 9.8896e-04, 9.4748e-04, 1.0118e-03, 2.4719e-03,
          6.8426e-04, 8.1968e-04, 1.3180e-03, 1.6117e-03, 9.7036e-04, 7.3242e-04,
          1.1616e-03, 8.0185e-03, 1.2388e-03, 9.2745e-04, 1.3914e-03, 1.4572e-03,
          1.7109e-03, 3.3569e-04, 1.2150e-03, 1.0101e-02, 1.0967e-03, 1.1454e-03,
          9.7847e-04, 3.5191e-03, 1.4706e-03, 8.4114e-04, 8.7786e-04, 6.8760e-04,
          8.0061e-04, 8.8072e-04, 4.1556e-04, 6.2084e-04, 1.8396e-03, 2.3556e-03,
          9.3508e-04, 2.3899e-03, 1.1883e-03, 1.0881e-03, 2.1095e-03, 8.0824e-04,
          9.4604e-04, 7.7705e-03, 1.9913e-03, 5.7173e-04, 2.4586e-03, 1.2939e-02,
          4.4289e-03, 5.4646e-04, 8.2159e-04, 7.3719e-04, 2.3441e-03, 1.6422e-03,
          8.7070e-04, 4.5633e-04, 3.3116e-04, 6.2084e-04, 7.9632e-04, 5.7449e-03,
          2.5349e-03, 7.8201e-04, 7.5579e-04, 6.0892e-04, 1.4582e-03, 2.4166e-03,
          1.5078e-03, 1.3475e-03, 6.3610e-04, 1.7109e-03, 1.7288e-02, 1.2955e-02,
          8.6403e-04, 1.0910e-03, 4.2248e-04, 4.3750e-04, 6.2656e-04, 1.0900e-03,
          5.0116e-04, 3.3140e-04, 6.6662e-04, 6.0034e-04, 3.3474e-03, 3.9253e-03,
          1.5717e-03, 1.0309e-03, 6.9809e-04, 7.6199e-04, 5.1832e-04, 1.5793e-03,
          1.1806e-03, 8.2397e-04, 3.9978e-03, 3.1067e-02, 3.3722e-02, 8.8978e-04,
          1.2293e-03, 1.8740e-03, 1.6050e-03, 6.8283e-04, 1.4181e-03, 1.2741e-03,
          7.7152e-04, 4.9925e-04, 3.3112e-03, 2.4475e-02, 1.9569e-03, 1.8702e-03,
          1.6680e-03, 9.0027e-04, 7.3385e-04, 8.7547e-04, 4.1237e-03, 1.1101e-03,
          1.2417e-03, 3.2368e-03, 5.4588e-03, 5.5084e-03, 1.0395e-03, 1.1921e-03,
          1.2245e-03, 7.6246e-04, 5.2881e-04, 1.8530e-03, 9.5272e-04, 8.8978e-04,
          7.6771e-04, 2.2240e-03, 1.6537e-03, 1.7593e-02, 1.8158e-03, 1.9951e-03,
          2.0084e-03, 4.9210e-03, 2.1172e-03, 3.5076e-03, 6.2847e-04, 8.1778e-04,
          4.6654e-03, 9.2554e-04, 8.5831e-03, 2.8439e-03, 5.8937e-03, 1.8425e-03,
          6.3658e-04, 3.3932e-03, 5.7487e-03, 5.1956e-03, 9.2459e-04, 1.1702e-03,
          4.4098e-03, 8.7967e-03, 1.0811e-02, 3.1185e-03, 3.3016e-03, 6.7749e-02,
          3.5882e-05, 4.6682e-04, 3.5882e-05, 2.2542e-04, 3.5882e-05, 3.5882e-05,
          1.2410e-04, 1.2410e-04, 3.5882e-05, 3.5882e-05, 1.3709e-04, 3.5882e-05,
          3.5882e-05, 1.2410e-04, 3.5882e-05, 1.2410e-04, 3.5882e-05, 3.5882e-05,
          3.5882e-05, 1.3709e-04, 3.5882e-05, 3.5882e-05, 2.2542e-04, 2.2542e-04,
          4.6682e-04, 3.5882e-05, 5.5504e-04, 1.2410e-04, 3.5882e-05, 1.2410e-04,
          6.5660e-04, 6.5660e-04, 5.6839e-04, 3.5882e-05, 6.7091e-04, 2.2542e-04,
          1.2410e-04, 3.5882e-05, 1.2410e-04, 1.2410e-04, 3.5882e-05, 3.5882e-05,
          3.5882e-05, 3.5882e-05, 3.5882e-05, 3.5882e-05, 5.5504e-04, 3.5882e-05,
          5.6839e-04, 3.5882e-05, 2.2542e-04, 3.5882e-05, 3.5882e-05, 3.5882e-05,
          6.5660e-04, 3.5882e-05, 3.5882e-05, 3.5882e-05, 1.2410e-04, 2.2542e-04,
          3.5882e-05, 2.9488e-03, 3.5882e-05, 3.5882e-05, 3.5882e-05, 3.5882e-05,
          6.7091e-04, 5.2547e-04, 2.6274e-04, 2.6274e-04, 2.6274e-04, 1.3709e-04,
          3.5882e-05, 2.2542e-04, 2.6274e-04, 4.8280e-04, 2.6274e-04, 1.3709e-04,
          2.6274e-04, 2.6274e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.8370e-03, 5.6152e-02, 1.1383e-02, 2.3651e-02, 3.4515e-02,
          3.8422e-02, 1.2291e-02, 6.8521e-04, 5.1212e-04, 4.8637e-03, 5.6267e-03,
          2.3708e-03, 8.4591e-04, 5.1308e-04, 1.1215e-03, 6.1941e-04, 1.5163e-03,
          8.2397e-04, 3.7336e-04, 2.5439e-04, 3.6931e-04, 3.0279e-04, 1.1759e-03,
          6.2294e-03, 1.4067e-03, 9.8495e-03, 5.7755e-03, 4.9400e-04, 1.5602e-03,
          8.5974e-04, 1.4238e-03, 5.7983e-04, 2.7485e-03, 2.6207e-03, 7.5817e-04,
          2.9602e-03, 9.9564e-04, 4.7607e-03, 2.3384e-03, 3.7432e-04, 4.8103e-03,
          1.2112e-03, 1.0405e-03, 7.5197e-04, 1.0595e-03, 1.3237e-03, 3.0308e-03,
          3.2449e-04, 3.8171e-04, 3.2005e-03, 1.3828e-03, 1.0777e-03, 9.1553e-04,
          3.6716e-04, 6.2990e-04, 4.6134e-04, 2.2240e-03, 1.1511e-03, 2.4700e-03,
          1.2217e-03, 8.6403e-04, 7.4434e-04, 1.2541e-03, 1.9646e-03, 3.7975e-03,
          1.4524e-03, 1.0777e-03, 2.0161e-03, 6.7568e-04, 5.5504e-04, 6.5470e-04,
          7.2813e-04, 1.1192e-02, 6.4373e-04, 1.3714e-03, 3.0279e-04, 3.5596e-04,
          7.1001e-04, 2.3823e-03, 2.3155e-03, 4.6196e-03, 2.0123e-03, 7.8726e-04,
          6.5899e-04, 1.9503e-03, 5.3453e-04, 2.6054e-03, 1.1005e-03, 9.1791e-04,
          6.0387e-03, 5.8060e-03, 3.3545e-04, 3.8090e-03, 4.1056e-04, 6.3801e-04,
          6.5088e-04, 3.5691e-04, 1.3981e-03, 2.7294e-03, 2.3403e-03, 3.8071e-03,
          7.1564e-03, 1.0233e-03, 1.5297e-03, 3.6297e-03, 2.4185e-03, 8.1730e-04,
          1.4706e-03, 2.5711e-03, 7.5607e-03, 4.5013e-04, 9.5892e-04, 2.1400e-03,
          1.0586e-03, 1.7138e-03, 3.7169e-04, 3.0041e-03, 7.5388e-04, 8.8120e-04,
          1.2856e-03, 7.4310e-03, 9.2363e-04, 4.5037e-04, 3.3302e-03, 1.0538e-03,
          1.5039e-03, 1.5020e-03, 1.6069e-03, 2.9202e-03, 1.2541e-03, 1.6632e-03,
          1.3218e-03, 1.2770e-03, 4.6158e-04, 6.0749e-04, 3.3970e-03, 5.4312e-04,
          3.1834e-03, 6.2323e-04, 8.2207e-04, 2.2621e-03, 2.8915e-03, 1.3742e-03,
          1.5602e-03, 8.1682e-04, 1.5440e-03, 2.8019e-03, 1.6603e-03, 2.8934e-03,
          1.1398e-02, 3.5820e-03, 2.1324e-03, 2.5940e-03, 1.0805e-03, 6.4316e-03,
          4.9973e-03, 4.0474e-03, 7.9584e-04, 7.5769e-04, 3.6526e-03, 1.0328e-03,
          3.3722e-03, 7.5569e-03, 2.2602e-03, 2.5368e-03, 4.0741e-03, 4.1084e-03,
          6.6528e-03, 5.2948e-03, 2.6588e-03, 1.0815e-03, 1.6937e-03, 5.1918e-03,
          1.3561e-03, 8.7976e-04, 2.2068e-03, 2.8896e-03, 5.3673e-03, 1.8749e-03,
          1.8578e-03, 1.3313e-03, 3.4027e-03, 1.3065e-03, 8.4925e-04, 7.1287e-04,
          3.0937e-03, 6.2847e-04, 1.3800e-03, 2.8419e-03, 1.5383e-03, 2.9507e-03,
          1.6527e-03, 1.3304e-03, 3.0117e-03, 2.9793e-03, 7.0305e-03, 4.1122e-03,
          1.6441e-03, 1.1625e-03, 3.8776e-03, 2.8515e-03, 3.8280e-03, 2.6150e-03,
          1.0843e-03, 5.6915e-03, 1.1406e-03, 7.8058e-04, 7.1411e-03, 1.0376e-03,
          7.7105e-04, 4.4708e-03, 2.5520e-03, 2.7771e-03, 4.1962e-03, 1.5402e-03,
          4.4785e-03, 2.1152e-03, 4.3869e-03, 7.5340e-04, 4.4556e-03, 1.3247e-03,
          5.5733e-03, 1.1683e-03, 4.0855e-03, 2.2564e-03, 4.6692e-03, 2.9354e-03,
          1.7023e-03, 5.6982e-04, 7.2050e-04, 9.4557e-04, 2.3155e-03, 5.4789e-04,
          3.7575e-03, 3.0270e-03, 1.0452e-03, 1.2083e-03, 4.9934e-03, 4.2419e-03,
          2.1191e-03, 1.8034e-03, 1.8654e-03, 3.2806e-03, 2.8973e-03, 1.9073e-03,
          3.9177e-03, 1.6193e-03, 5.8317e-04, 7.5865e-04, 7.7868e-04, 3.3302e-03,
          7.7820e-04, 6.1083e-04, 9.5224e-04, 1.9817e-03, 1.8330e-03, 7.1526e-03,
          4.6806e-03, 2.6245e-03, 2.0809e-03, 1.8129e-03, 1.0910e-03, 2.8343e-03,
          2.7103e-03, 2.1687e-03, 9.1362e-04, 2.8915e-03, 3.5267e-03, 2.4471e-03,
          1.4458e-03, 2.2202e-03, 3.3188e-03, 1.6394e-03, 6.4278e-03, 2.4929e-03,
          1.5135e-03, 1.1053e-03, 8.8120e-04, 1.3914e-03, 2.6836e-03, 2.6226e-03,
          2.4929e-03, 1.4467e-03, 1.6184e-03, 1.5135e-03, 2.5120e-03, 2.3518e-03,
          2.1954e-03, 1.0780e-02, 2.2545e-03, 2.2907e-03, 1.4677e-03, 1.9522e-03,
          2.7103e-03, 1.3332e-03, 1.1206e-03, 3.2215e-03, 8.4152e-03, 1.1635e-03,
          9.2602e-04, 1.1129e-03, 1.7653e-03, 3.4332e-03, 2.7180e-03, 7.3013e-03,
          3.8834e-03, 1.7862e-03, 6.3896e-03, 1.0624e-03, 1.5440e-03, 2.0733e-03,
          1.1806e-03, 1.4896e-03, 3.7231e-03, 2.8496e-03, 6.5498e-03, 3.9635e-03,
          1.1406e-03, 1.1711e-02, 2.3518e-03, 1.3847e-03, 1.2188e-03, 1.8940e-03,
          1.2646e-03, 9.3508e-04, 3.0651e-03, 5.1308e-03, 7.0152e-03, 4.3373e-03,
          3.8208e-02, 1.7846e-04, 1.7846e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          4.2677e-05, 4.2677e-05, 4.0293e-04, 4.2677e-05, 4.2677e-05, 1.7846e-04,
          4.2677e-05, 4.2677e-05, 2.6655e-04, 4.2677e-05, 2.6703e-04, 3.5286e-04,
          3.5286e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 1.7846e-04, 1.7846e-04,
          4.0293e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 1.7846e-04, 2.6703e-04,
          4.0245e-04, 4.0245e-04, 2.6703e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          4.2677e-05, 3.5286e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05, 1.7846e-04, 4.2677e-05,
          4.0245e-04, 1.7846e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          4.0293e-04, 3.5286e-04, 4.0245e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          5.7697e-04, 2.6655e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          2.6655e-04, 4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05, 4.2677e-05,
          4.2677e-05, 4.2677e-05, 4.2677e-05, 3.0088e-04, 2.6655e-04, 2.6655e-04,
          2.6655e-04, 4.2677e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.4054e-03, 5.8472e-02, 1.8250e-02, 1.5976e-02, 4.0375e-02,
          3.4882e-02, 1.0567e-02, 9.1648e-04, 5.5885e-04, 1.1415e-03, 2.2869e-03,
          7.6056e-04, 1.7338e-03, 1.1463e-03, 1.1091e-03, 9.7942e-04, 3.6182e-03,
          2.5043e-03, 4.1795e-04, 3.1614e-04, 4.9591e-04, 5.0545e-04, 2.4052e-03,
          2.5082e-03, 1.1530e-03, 8.3771e-03, 3.8166e-03, 5.5742e-04, 4.4975e-03,
          1.7157e-03, 2.0828e-03, 1.6289e-03, 1.8053e-03, 9.7942e-04, 7.8535e-04,
          9.2745e-04, 4.3011e-04, 1.0118e-03, 1.2350e-03, 3.9601e-04, 5.4502e-04,
          3.7956e-04, 2.6846e-04, 3.6693e-04, 2.4891e-04, 4.2915e-03, 1.5278e-03,
          6.7949e-04, 6.0034e-04, 3.1013e-03, 1.2932e-03, 1.4877e-03, 6.6423e-04,
          4.3654e-04, 8.9645e-04, 5.0211e-04, 3.2973e-04, 7.6866e-04, 4.9973e-04,
          6.4802e-04, 1.7328e-03, 7.4100e-04, 9.6035e-04, 3.4189e-04, 2.1687e-03,
          6.6757e-04, 3.2902e-04, 4.9734e-04, 1.8358e-04, 1.7738e-04, 2.9039e-04,
          3.8338e-04, 2.2469e-03, 6.1226e-04, 2.3937e-03, 4.1342e-04, 3.4118e-04,
          9.9564e-04, 3.2759e-04, 4.0054e-04, 7.2765e-04, 4.2605e-04, 3.9530e-04,
          4.2081e-04, 2.7905e-03, 5.7554e-04, 9.1600e-04, 3.1281e-04, 3.1638e-04,
          2.0924e-03, 9.0075e-04, 2.5821e-04, 1.1330e-03, 3.8886e-04, 3.5763e-04,
          1.0881e-03, 3.6192e-04, 1.2636e-03, 1.2932e-03, 3.6860e-04, 4.3368e-04,
          1.9894e-03, 3.5548e-04, 4.0412e-04, 6.7139e-04, 1.5125e-03, 2.3613e-03,
          2.2869e-03, 3.6955e-04, 1.1940e-03, 3.1447e-04, 5.0783e-04, 4.3583e-04,
          5.5504e-04, 5.0831e-04, 9.1839e-04, 2.5158e-03, 5.3835e-04, 2.6398e-03,
          6.2332e-03, 1.2312e-03, 2.5535e-04, 5.5933e-04, 2.2259e-03, 2.1172e-04,
          2.8086e-04, 3.0208e-04, 4.4537e-04, 2.0847e-03, 1.8759e-03, 2.9297e-03,
          9.1553e-04, 4.0984e-04, 5.8222e-04, 2.8372e-04, 7.2336e-04, 2.5678e-04,
          6.4011e-03, 3.4637e-03, 3.4924e-03, 2.8362e-03, 2.7084e-03, 6.1989e-03,
          3.9744e-04, 3.6550e-04, 4.2939e-04, 3.6883e-04, 5.3263e-04, 1.6174e-03,
          1.9178e-03, 1.4400e-03, 5.4741e-04, 1.7862e-03, 1.6193e-03, 7.7629e-04,
          7.3957e-04, 5.1022e-04, 5.8794e-04, 3.8457e-04, 1.7338e-03, 5.5656e-03,
          2.0065e-03, 1.2903e-03, 8.7814e-03, 2.2011e-03, 8.3733e-04, 4.9782e-04,
          1.4296e-03, 5.5599e-04, 1.2007e-03, 1.1425e-03, 4.4022e-03, 2.1610e-03,
          1.0128e-03, 8.8596e-04, 9.1600e-04, 8.2302e-04, 1.1988e-03, 2.6169e-03,
          6.7759e-04, 1.3332e-03, 2.1725e-03, 2.4376e-03, 1.2894e-03, 1.3561e-03,
          1.6747e-03, 5.2376e-03, 1.1950e-03, 7.0286e-04, 6.6757e-04, 2.1820e-03,
          2.7103e-03, 2.9373e-04, 1.6623e-03, 1.0765e-02, 1.3628e-03, 1.5640e-03,
          1.0033e-03, 2.4433e-03, 2.4357e-03, 6.5422e-04, 9.5129e-04, 9.5940e-04,
          5.1832e-04, 8.7929e-04, 5.0783e-04, 7.9489e-04, 2.4948e-03, 3.7575e-03,
          1.3361e-03, 1.6356e-03, 7.7581e-04, 9.5177e-04, 2.9869e-03, 6.3324e-04,
          1.3533e-03, 5.7220e-03, 2.1038e-03, 9.9850e-04, 3.8757e-03, 1.2192e-02,
          4.9057e-03, 6.3705e-04, 9.1696e-04, 7.1907e-04, 4.6387e-03, 2.1782e-03,
          6.9141e-04, 5.3024e-04, 5.5075e-04, 1.0576e-03, 1.6117e-03, 5.7487e-03,
          1.4362e-03, 7.4482e-04, 8.4543e-04, 1.1435e-03, 2.2335e-03, 4.3983e-03,
          1.8282e-03, 1.5278e-03, 9.6750e-04, 2.4719e-03, 2.5909e-02, 5.3291e-03,
          9.8991e-04, 1.2960e-03, 6.2990e-04, 8.6498e-04, 9.5034e-04, 1.6775e-03,
          4.6825e-04, 4.3988e-04, 8.3780e-04, 1.1368e-03, 5.2261e-03, 4.0627e-03,
          1.3456e-03, 1.1110e-03, 9.6989e-04, 1.3180e-03, 7.7057e-04, 2.2678e-03,
          2.5005e-03, 1.1673e-03, 4.9896e-03, 3.8635e-02, 1.5266e-02, 1.4429e-03,
          1.6203e-03, 1.1539e-03, 3.0479e-03, 9.8419e-04, 2.2430e-03, 2.2011e-03,
          1.6270e-03, 1.4496e-03, 2.6932e-03, 1.1147e-02, 1.2455e-03, 1.7500e-03,
          1.6346e-03, 9.3412e-04, 7.3528e-04, 1.0843e-03, 1.2962e-02, 2.5578e-03,
          2.7084e-03, 3.8891e-03, 9.9030e-03, 7.1754e-03, 1.3008e-03, 2.0065e-03,
          1.3723e-03, 7.7200e-04, 6.1893e-04, 4.0054e-03, 1.9245e-03, 1.2941e-03,
          9.9659e-04, 6.2904e-03, 2.9297e-03, 2.0554e-02, 1.4362e-03, 3.0651e-03,
          3.6316e-03, 4.5242e-03, 3.2711e-03, 7.0457e-03, 1.2503e-03, 1.6594e-03,
          8.5297e-03, 2.2106e-03, 5.7564e-03, 4.2839e-03, 6.8321e-03, 3.3169e-03,
          1.7920e-03, 5.7983e-03, 9.1171e-03, 9.9258e-03, 1.2913e-03, 1.5593e-03,
          9.0790e-03, 1.4442e-02, 8.4915e-03, 2.5921e-03, 4.7874e-03, 3.4454e-02,
          4.5052e-03, 3.7323e-02, 5.1081e-05, 5.1081e-05, 1.3506e-04, 1.3506e-04,
          5.1081e-05, 1.3506e-04, 1.3506e-04, 5.1081e-05, 2.3961e-04, 5.1081e-05,
          1.3506e-04, 5.1081e-05, 5.1081e-05, 5.1081e-05, 1.3506e-04, 1.3506e-04,
          5.1081e-05, 2.3961e-04, 5.1081e-05, 5.1081e-05, 2.3961e-04, 5.1081e-05,
          1.3506e-04, 1.3506e-04, 1.3506e-04, 1.3506e-04, 2.3961e-04, 2.3961e-04,
          2.3961e-04, 2.3961e-04, 1.3506e-04, 1.3506e-04, 5.1081e-05, 5.1081e-05,
          5.1081e-05, 2.3961e-04, 5.1081e-05, 9.1374e-05, 5.1081e-05, 1.3506e-04,
          2.3961e-04, 5.1081e-05, 5.1081e-05, 1.3506e-04, 1.3506e-04, 5.1081e-05,
          2.3961e-04, 5.1081e-05, 5.6219e-04, 1.3506e-04, 5.1081e-05, 1.3506e-04,
          3.2353e-04, 5.1081e-05, 2.3961e-04, 1.3506e-04, 2.3961e-04, 5.1081e-05,
          5.1081e-05, 2.3961e-04, 1.3506e-04, 5.1081e-05, 5.1081e-05, 5.1081e-05,
          2.3961e-04, 2.3961e-04, 2.3961e-04, 2.3961e-04, 2.3961e-04, 5.1081e-05,
          5.1081e-05, 2.3961e-04, 2.3961e-04, 2.3961e-04, 5.1081e-05, 2.3961e-04,
          2.3961e-04, 2.3961e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.9128e-03, 4.5013e-02, 1.4763e-02, 2.5192e-02, 4.7058e-02,
          2.7176e-02, 1.5015e-02, 8.5974e-04, 4.9305e-04, 9.6273e-04, 1.7090e-03,
          6.8188e-04, 1.9608e-03, 5.7182e-03, 8.1015e-04, 6.9761e-04, 5.2109e-03,
          3.5400e-03, 3.5119e-04, 9.2173e-04, 5.5885e-04, 7.9393e-04, 1.9007e-03,
          1.7433e-03, 6.6090e-04, 1.1971e-02, 5.9738e-03, 1.7252e-03, 3.8929e-03,
          9.7427e-03, 1.6108e-03, 1.8139e-03, 1.1444e-03, 7.3051e-04, 8.7404e-04,
          8.6546e-04, 5.4884e-04, 9.5701e-04, 7.9775e-04, 1.8959e-03, 6.2752e-04,
          4.8685e-04, 3.4237e-04, 4.7350e-04, 2.3758e-04, 3.6678e-03, 1.1425e-03,
          2.2202e-03, 1.3514e-03, 4.2725e-03, 1.3666e-03, 1.3056e-03, 6.1893e-04,
          6.3896e-04, 8.7547e-04, 8.5831e-04, 5.7316e-04, 1.0376e-03, 7.5865e-04,
          9.1314e-04, 4.2686e-03, 8.7690e-04, 6.7377e-04, 4.6682e-04, 1.2236e-03,
          9.8038e-04, 3.4714e-04, 5.0449e-04, 2.2280e-04, 2.1338e-04, 4.1223e-04,
          4.7755e-04, 1.9741e-03, 1.0891e-03, 1.8044e-03, 6.1798e-04, 4.7469e-04,
          1.4534e-03, 5.9080e-04, 7.9298e-04, 6.8378e-04, 9.7609e-04, 4.9353e-04,
          5.7554e-04, 1.7376e-03, 9.0647e-04, 6.3896e-04, 4.9210e-04, 8.4305e-04,
          1.2493e-03, 5.8937e-04, 2.8443e-04, 8.0347e-04, 2.6512e-04, 4.4680e-04,
          2.3861e-03, 6.5947e-04, 1.2789e-03, 1.1606e-03, 6.2990e-04, 5.5742e-04,
          1.4877e-03, 4.1318e-04, 5.6362e-04, 7.4673e-04, 9.1505e-04, 3.9406e-03,
          1.3056e-03, 4.5991e-04, 8.9121e-04, 2.9802e-04, 6.3896e-04, 4.9686e-04,
          3.6240e-04, 4.0364e-04, 1.4172e-03, 1.3819e-03, 6.6185e-04, 3.6888e-03,
          1.3496e-02, 9.2220e-04, 4.1771e-04, 4.5872e-04, 1.1997e-03, 3.9744e-04,
          4.3988e-04, 6.6853e-04, 7.6914e-04, 1.2274e-03, 1.1358e-03, 1.5240e-03,
          7.8917e-04, 5.5075e-04, 2.9850e-04, 2.7776e-04, 5.7077e-04, 3.6883e-04,
          5.0049e-03, 6.8359e-03, 3.1548e-03, 2.6321e-03, 1.9464e-03, 1.4397e-02,
          6.1464e-04, 2.0015e-04, 4.4632e-04, 4.6206e-04, 1.1969e-03, 1.2112e-03,
          1.4410e-03, 1.0681e-03, 6.5470e-04, 1.3294e-03, 1.6136e-03, 7.8773e-04,
          5.9032e-04, 4.8137e-04, 3.6502e-04, 5.3978e-04, 1.0996e-03, 6.0692e-03,
          1.2970e-03, 8.4066e-04, 5.4245e-03, 1.1692e-03, 1.2808e-03, 5.3167e-04,
          1.0958e-03, 5.3501e-04, 7.7677e-04, 5.4979e-04, 5.4703e-03, 1.3237e-03,
          7.3814e-04, 7.9870e-04, 1.3943e-03, 1.1349e-03, 8.6355e-04, 1.6642e-03,
          5.8746e-04, 6.2799e-04, 1.0958e-03, 1.1330e-03, 6.9809e-04, 6.4421e-04,
          8.8120e-04, 6.5804e-03, 9.4938e-04, 1.0738e-03, 1.4954e-03, 1.2064e-03,
          1.2436e-03, 4.5824e-04, 1.0462e-03, 7.1716e-03, 8.6737e-04, 1.0738e-03,
          1.3599e-03, 5.2528e-03, 1.2541e-03, 7.6246e-04, 7.8440e-04, 5.8174e-04,
          8.0872e-04, 7.5436e-04, 4.6587e-04, 4.8137e-04, 1.3733e-03, 1.7719e-03,
          7.9679e-04, 2.3823e-03, 1.4944e-03, 9.2077e-04, 1.8930e-03, 9.4938e-04,
          9.6893e-04, 7.1945e-03, 1.4668e-03, 5.0211e-04, 2.1000e-03, 1.1803e-02,
          3.9101e-03, 7.1287e-04, 7.8821e-04, 6.5088e-04, 1.9426e-03, 1.1797e-03,
          7.6675e-04, 5.1069e-04, 3.6979e-04, 6.1607e-04, 7.4100e-04, 5.1689e-03,
          2.9106e-03, 8.0013e-04, 8.6260e-04, 5.7745e-04, 1.7567e-03, 2.8076e-03,
          2.0638e-03, 2.1210e-03, 9.4652e-04, 2.1095e-03, 2.0294e-02, 2.0020e-02,
          1.1091e-03, 9.6703e-04, 6.0940e-04, 5.0735e-04, 8.2636e-04, 9.5892e-04,
          5.6219e-04, 4.4894e-04, 8.7976e-04, 6.3324e-04, 2.4910e-03, 3.2940e-03,
          1.7939e-03, 1.0700e-03, 7.7057e-04, 8.4829e-04, 7.6008e-04, 2.2335e-03,
          1.2941e-03, 6.6614e-04, 4.0932e-03, 2.9037e-02, 2.4765e-02, 8.4686e-04,
          1.4181e-03, 1.7004e-03, 1.4200e-03, 7.9918e-04, 1.2274e-03, 1.2369e-03,
          7.9393e-04, 4.5300e-04, 4.6921e-03, 2.1500e-02, 2.2850e-03, 2.0790e-03,
          1.5411e-03, 1.4009e-03, 1.2417e-03, 1.4210e-03, 4.8485e-03, 1.1482e-03,
          1.3151e-03, 2.4033e-03, 4.0665e-03, 4.5547e-03, 1.4782e-03, 1.5202e-03,
          1.6413e-03, 1.0424e-03, 7.6962e-04, 1.5650e-03, 8.7309e-04, 1.2074e-03,
          1.0681e-03, 1.9217e-03, 1.3866e-03, 1.5717e-02, 2.9202e-03, 1.9484e-03,
          1.8835e-03, 1.0033e-02, 2.1782e-03, 3.2406e-03, 5.4550e-04, 7.1669e-04,
          4.4708e-03, 1.1492e-03, 1.5411e-02, 2.3327e-03, 4.4785e-03, 1.5116e-03,
          6.6996e-04, 2.5654e-03, 5.2223e-03, 4.2610e-03, 1.4553e-03, 1.6994e-03,
          3.8223e-03, 7.9269e-03, 8.5449e-03, 5.0354e-03, 3.9673e-03, 5.0964e-02,
          3.1147e-03, 3.7155e-03, 6.3171e-02, 2.1040e-04, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 1.3244e-04, 3.6538e-05, 3.6538e-05, 1.1444e-04, 3.6538e-05,
          3.6538e-05, 1.3244e-04, 3.6538e-05, 1.3244e-04, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 1.1444e-04, 3.6538e-05, 3.6538e-05, 2.1040e-04, 2.1040e-04,
          3.6538e-05, 3.6538e-05, 1.3244e-04, 1.3244e-04, 3.6538e-05, 1.3244e-04,
          7.7915e-04, 7.7915e-04, 6.0558e-04, 3.6538e-05, 3.8314e-04, 1.3244e-04,
          1.3244e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05, 7.0143e-04, 3.6538e-05,
          6.8331e-04, 3.6538e-05, 2.1040e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          8.6164e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 2.1040e-04, 2.0730e-04,
          3.6538e-05, 7.7629e-04, 3.6538e-05, 3.6538e-05, 3.6538e-05, 3.6538e-05,
          3.8314e-04, 1.9670e-04, 1.9670e-04, 2.5725e-04, 1.9670e-04, 1.1444e-04,
          1.1444e-04, 2.1040e-04, 1.9670e-04, 3.4642e-04, 1.9670e-04, 1.1444e-04,
          1.9670e-04, 1.9670e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.1460e-03, 5.1392e-02, 1.2894e-02, 2.3148e-02, 3.4729e-02,
          3.3264e-02, 1.2505e-02, 1.0147e-03, 5.5408e-04, 4.0817e-03, 7.0419e-03,
          1.8787e-03, 9.6655e-04, 6.0225e-04, 1.2970e-03, 9.2173e-04, 1.9522e-03,
          1.0872e-03, 4.3750e-04, 2.8300e-04, 4.3559e-04, 4.3845e-04, 1.7061e-03,
          7.1259e-03, 1.4915e-03, 1.0315e-02, 7.4005e-03, 6.8903e-04, 2.0542e-03,
          9.1696e-04, 2.0943e-03, 9.0742e-04, 3.1052e-03, 3.3894e-03, 1.0967e-03,
          1.9989e-03, 6.6328e-04, 1.3832e-02, 3.8090e-03, 4.2748e-04, 4.0627e-03,
          8.6784e-04, 7.3624e-04, 5.7745e-04, 8.4543e-04, 1.6623e-03, 4.1313e-03,
          4.6873e-04, 5.2929e-04, 3.6602e-03, 1.8845e-03, 1.7242e-03, 1.0405e-03,
          4.1342e-04, 8.6498e-04, 5.2834e-04, 9.3317e-04, 9.6416e-04, 9.7847e-04,
          7.3624e-04, 1.2007e-03, 7.7868e-04, 1.6022e-03, 1.2693e-03, 3.7079e-03,
          2.7313e-03, 1.5173e-03, 2.1877e-03, 5.3835e-04, 3.6550e-04, 5.9748e-04,
          6.3801e-04, 1.0094e-02, 6.7234e-04, 1.8892e-03, 3.5810e-04, 3.7694e-04,
          8.3399e-04, 9.7036e-04, 1.7586e-03, 2.3556e-03, 1.5621e-03, 4.4608e-04,
          4.5919e-04, 2.4967e-03, 5.5456e-04, 3.1223e-03, 1.7815e-03, 1.1501e-03,
          7.8659e-03, 5.9395e-03, 3.5596e-04, 4.5738e-03, 3.6907e-04, 5.4502e-04,
          9.3746e-04, 4.1771e-04, 1.6937e-03, 1.9817e-03, 1.0042e-03, 1.8711e-03,
          6.4011e-03, 1.4000e-03, 1.3542e-03, 1.4524e-03, 1.8721e-03, 9.7466e-04,
          1.7757e-03, 1.1759e-03, 6.9962e-03, 6.8140e-04, 1.7014e-03, 3.4962e-03,
          9.4175e-04, 1.1587e-03, 4.4727e-04, 2.9659e-03, 6.6948e-04, 1.4038e-03,
          1.8015e-03, 3.5210e-03, 5.3406e-04, 5.1689e-04, 3.0537e-03, 7.4482e-04,
          6.6948e-04, 7.6199e-04, 1.0395e-03, 2.2392e-03, 1.8959e-03, 2.1267e-03,
          2.5597e-03, 2.1267e-03, 5.5027e-04, 8.3637e-04, 3.7994e-03, 5.2404e-04,
          3.8986e-03, 7.5912e-04, 1.2455e-03, 3.1815e-03, 4.3488e-03, 1.8702e-03,
          8.0776e-04, 5.1403e-04, 9.5510e-04, 1.3933e-03, 1.5850e-03, 2.0046e-03,
          5.9433e-03, 1.8501e-03, 7.7724e-04, 2.2449e-03, 1.1854e-03, 2.9354e-03,
          4.3602e-03, 2.8954e-03, 7.3051e-04, 7.6723e-04, 2.5864e-03, 1.6212e-03,
          3.6869e-03, 8.2626e-03, 2.9278e-03, 3.4771e-03, 1.8930e-03, 2.4204e-03,
          5.4245e-03, 2.9049e-03, 1.8959e-03, 9.3412e-04, 2.1992e-03, 3.9215e-03,
          8.3542e-04, 6.3848e-04, 1.2398e-03, 1.9102e-03, 3.6526e-03, 2.6093e-03,
          1.8539e-03, 1.4563e-03, 3.7079e-03, 2.0294e-03, 1.0052e-03, 9.8610e-04,
          2.2926e-03, 9.4128e-04, 2.2736e-03, 1.3771e-03, 1.1625e-03, 2.7580e-03,
          2.1992e-03, 5.6219e-04, 2.5425e-03, 3.8567e-03, 5.6343e-03, 2.8248e-03,
          9.3174e-04, 1.0853e-03, 3.4847e-03, 2.6760e-03, 4.6043e-03, 2.6112e-03,
          9.5987e-04, 4.9515e-03, 7.9012e-04, 8.1968e-04, 6.9847e-03, 1.5182e-03,
          9.5367e-04, 2.9354e-03, 1.4210e-03, 2.1343e-03, 3.3722e-03, 8.6689e-04,
          2.4948e-03, 2.5291e-03, 6.3972e-03, 7.5054e-04, 4.7913e-03, 1.8368e-03,
          8.4381e-03, 7.8869e-04, 2.3746e-03, 4.0321e-03, 5.1346e-03, 3.4866e-03,
          2.3556e-03, 7.2336e-04, 5.2404e-04, 7.8249e-04, 1.9093e-03, 8.4019e-04,
          2.4185e-03, 2.3594e-03, 8.9407e-04, 1.0099e-03, 2.5005e-03, 3.0594e-03,
          1.2455e-03, 1.2512e-03, 1.2054e-03, 3.1433e-03, 4.0092e-03, 1.9054e-03,
          1.6956e-03, 1.6689e-03, 4.8208e-04, 5.4884e-04, 6.7234e-04, 2.6054e-03,
          8.3303e-04, 4.4847e-04, 7.5197e-04, 1.6022e-03, 2.5616e-03, 7.6218e-03,
          2.2373e-03, 3.2768e-03, 1.2045e-03, 1.2016e-03, 5.8603e-04, 1.5697e-03,
          1.9503e-03, 2.4433e-03, 1.1301e-03, 5.2872e-03, 4.7684e-03, 1.7853e-03,
          1.5812e-03, 2.6379e-03, 3.3264e-03, 1.1959e-03, 4.8447e-03, 2.1381e-03,
          1.1387e-03, 7.7629e-04, 8.6069e-04, 1.7757e-03, 1.8559e-03, 1.9522e-03,
          3.3569e-03, 9.0313e-04, 8.0299e-04, 9.3031e-04, 2.4948e-03, 1.7090e-03,
          1.8120e-03, 1.1467e-02, 3.4599e-03, 2.2755e-03, 1.0853e-03, 1.5221e-03,
          1.5965e-03, 9.0885e-04, 7.4100e-04, 3.6507e-03, 5.3482e-03, 9.0790e-04,
          7.4148e-04, 1.4324e-03, 2.1172e-03, 5.4131e-03, 1.6642e-03, 4.8141e-03,
          2.9793e-03, 1.5640e-03, 4.4785e-03, 1.3084e-03, 1.2264e-03, 1.2741e-03,
          1.5640e-03, 1.1187e-03, 3.6221e-03, 3.8757e-03, 8.2092e-03, 3.4885e-03,
          8.4829e-04, 9.6741e-03, 2.8419e-03, 1.9226e-03, 9.3889e-04, 1.5049e-03,
          1.6079e-03, 1.5135e-03, 4.4746e-03, 2.7695e-03, 4.2686e-03, 5.2376e-03,
          1.6266e-02, 1.4753e-03, 3.8624e-03, 6.9031e-02, 4.9174e-05, 4.9174e-05,
          2.7895e-04, 4.9174e-05, 1.9979e-04, 4.9174e-05, 4.9174e-05, 3.9363e-04,
          4.9174e-05, 4.9174e-05, 4.9174e-05, 4.9174e-05, 2.7895e-04, 4.3631e-04,
          7.6532e-04, 3.7813e-04, 4.9174e-05, 4.9174e-05, 3.9363e-04, 1.9979e-04,
          4.2939e-04, 4.9174e-05, 4.9174e-05, 4.9174e-05, 3.9363e-04, 2.7895e-04,
          6.2323e-04, 6.2323e-04, 2.7895e-04, 4.9174e-05, 1.9979e-04, 4.9174e-05,
          1.9979e-04, 5.7173e-04, 4.9174e-05, 4.9174e-05, 4.9174e-05, 4.9174e-05,
          4.9174e-05, 4.9174e-05, 4.9174e-05, 4.9174e-05, 6.2323e-04, 4.9174e-05,
          9.5224e-04, 1.9979e-04, 3.9363e-04, 4.9174e-05, 4.9174e-05, 4.9174e-05,
          6.2323e-04, 4.9174e-05, 7.2241e-04, 4.9174e-05, 4.9174e-05, 2.7895e-04,
          5.7173e-04, 5.7173e-04, 4.9174e-05, 4.9174e-05, 4.9174e-05, 4.9174e-05,
          2.4295e-04, 4.9174e-05, 2.4295e-04, 4.9174e-05, 4.9174e-05, 4.9174e-05,
          4.9174e-05, 4.9174e-05, 2.4295e-04, 3.3665e-04, 4.9174e-05, 2.4295e-04,
          4.9174e-05, 4.9174e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.8180e-03, 5.8258e-02, 1.4725e-02, 1.6251e-02, 3.4271e-02,
          4.9835e-02, 1.2794e-02, 2.1477e-03, 7.6866e-04, 6.7902e-03, 2.0020e-02,
          3.8948e-03, 1.9426e-03, 5.9843e-04, 1.8616e-03, 1.9407e-03, 1.6909e-03,
          1.1492e-03, 1.2550e-03, 1.5581e-04, 2.1815e-04, 2.2018e-04, 2.9030e-03,
          1.8417e-02, 1.5078e-03, 9.5596e-03, 5.0621e-03, 3.6049e-04, 1.7509e-03,
          8.6546e-04, 3.8738e-03, 4.9591e-04, 1.5278e-03, 6.4125e-03, 6.0129e-04,
          8.3303e-04, 3.5429e-04, 1.3561e-03, 1.6603e-03, 2.3592e-04, 8.0872e-04,
          3.6287e-04, 2.9135e-04, 3.3689e-04, 2.7990e-04, 2.4643e-03, 1.3208e-03,
          3.2473e-04, 3.6001e-04, 3.1204e-03, 8.4543e-04, 2.0123e-03, 9.4223e-04,
          3.5715e-04, 9.0075e-04, 3.3164e-04, 3.9744e-04, 1.0185e-03, 6.0701e-04,
          5.4216e-04, 5.5313e-04, 4.7278e-04, 5.0449e-04, 5.5504e-04, 2.1172e-03,
          9.5892e-04, 1.1892e-03, 1.1024e-03, 2.0719e-04, 2.0111e-04, 3.1757e-04,
          3.6955e-04, 1.0597e-02, 4.4298e-04, 2.4967e-03, 3.6979e-04, 3.4308e-04,
          7.1287e-04, 3.2902e-04, 5.8985e-04, 1.5478e-03, 9.4986e-04, 3.6097e-04,
          2.9993e-04, 2.3346e-03, 3.1662e-04, 9.0103e-03, 6.7854e-04, 4.8327e-04,
          1.2817e-02, 2.6436e-03, 2.5129e-04, 5.4703e-03, 2.1267e-04, 2.8777e-04,
          4.2701e-04, 2.9254e-04, 1.2589e-03, 9.7847e-04, 3.8218e-04, 5.2881e-04,
          1.0475e-02, 1.2016e-03, 1.3161e-03, 6.6042e-04, 1.7443e-03, 4.1127e-04,
          1.7977e-03, 2.9778e-04, 2.3746e-03, 3.1233e-04, 6.5184e-04, 9.0694e-04,
          1.2550e-03, 3.0303e-04, 3.0708e-04, 2.1210e-03, 3.8934e-04, 7.1526e-04,
          8.0490e-04, 1.2522e-03, 2.6798e-04, 1.4906e-03, 3.6297e-03, 2.6727e-04,
          3.1185e-04, 3.4356e-04, 4.9114e-04, 1.7786e-03, 2.7218e-03, 2.5654e-03,
          1.1759e-03, 1.0948e-03, 4.0283e-03, 5.0640e-04, 1.8730e-03, 2.8038e-04,
          3.0880e-03, 5.4121e-04, 5.0049e-03, 7.7553e-03, 4.5967e-03, 1.1053e-03,
          3.4404e-04, 3.0022e-03, 4.9019e-04, 4.5443e-04, 6.8903e-04, 1.3733e-03,
          3.5133e-03, 1.5059e-03, 3.8004e-04, 6.2180e-04, 7.8440e-04, 5.3358e-04,
          1.4553e-03, 5.7983e-04, 6.1989e-03, 3.6979e-04, 1.1024e-03, 7.3481e-04,
          1.9817e-03, 2.3708e-03, 4.2229e-03, 7.4120e-03, 6.1655e-04, 6.9714e-04,
          5.9280e-03, 9.5510e-04, 1.1406e-03, 2.1324e-03, 1.5259e-03, 1.4477e-03,
          9.3365e-04, 5.6982e-04, 5.7983e-04, 4.4465e-04, 1.1177e-03, 8.1940e-03,
          1.2188e-03, 1.2970e-03, 2.7485e-03, 4.6692e-03, 5.1842e-03, 4.0207e-03,
          1.2924e-02, 1.2627e-03, 1.5854e-02, 5.3883e-04, 6.8569e-04, 3.1796e-03,
          5.7526e-03, 2.6917e-04, 1.8578e-03, 3.9368e-03, 7.3204e-03, 1.7214e-03,
          6.6519e-04, 5.2261e-04, 1.5984e-03, 7.8917e-04, 3.6621e-03, 1.2550e-03,
          4.7469e-04, 1.1024e-03, 3.5024e-04, 4.4479e-03, 7.8735e-03, 1.3456e-03,
          3.1681e-03, 8.8358e-04, 5.2166e-04, 1.7920e-03, 2.1629e-03, 5.3120e-04,
          1.0090e-03, 1.2150e-03, 1.8036e-02, 3.6774e-03, 2.4719e-03, 1.1787e-03,
          1.8177e-03, 2.9421e-04, 7.2432e-04, 1.6441e-03, 4.7722e-03, 3.7918e-03,
          1.2636e-03, 4.8876e-04, 3.2878e-04, 1.0509e-03, 1.2732e-03, 9.0933e-04,
          9.2220e-04, 8.8453e-04, 6.1798e-04, 1.9350e-03, 9.5940e-04, 2.0638e-03,
          8.4448e-04, 7.6389e-04, 4.6992e-04, 8.8596e-04, 1.3466e-03, 8.2636e-04,
          3.7193e-04, 1.3084e-03, 3.0899e-04, 5.4646e-04, 5.3978e-04, 1.4400e-03,
          3.2592e-04, 2.3663e-04, 4.2963e-04, 5.6076e-04, 5.3024e-03, 5.5695e-03,
          7.0858e-04, 1.3828e-03, 8.4496e-04, 8.8501e-04, 3.5453e-04, 1.0672e-03,
          1.0748e-03, 7.5645e-03, 3.9368e-03, 2.7885e-03, 7.0686e-03, 9.6130e-04,
          9.9277e-04, 1.5564e-03, 2.3575e-03, 4.9782e-04, 1.3437e-03, 1.3876e-03,
          8.5068e-04, 2.0618e-03, 6.2084e-04, 1.6441e-03, 7.3338e-04, 9.9945e-04,
          2.1343e-03, 5.4693e-04, 4.2772e-04, 5.0449e-04, 1.1368e-03, 1.6384e-03,
          1.7881e-03, 5.8403e-03, 4.2801e-03, 6.9885e-03, 5.1260e-04, 7.5006e-04,
          6.8998e-04, 4.4465e-04, 3.5381e-04, 2.5215e-03, 1.2350e-03, 5.7554e-04,
          4.3225e-04, 5.6305e-03, 6.7062e-03, 2.4586e-03, 6.4039e-04, 2.8877e-03,
          2.1858e-03, 8.3542e-04, 1.6851e-03, 2.2545e-03, 9.6283e-03, 1.1917e-02,
          2.0218e-03, 1.5936e-03, 1.4467e-03, 4.2038e-03, 5.1231e-03, 2.4261e-03,
          2.2640e-03, 2.0370e-02, 2.0008e-03, 2.0695e-03, 5.2738e-04, 5.4693e-04,
          4.7073e-03, 1.5907e-03, 5.8365e-03, 7.7438e-04, 1.6403e-03, 3.5820e-03,
          5.4398e-03, 1.6241e-03, 2.3251e-03, 7.9803e-03, 8.3557e-02, 6.1989e-05,
          1.8132e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05, 1.8132e-04,
          6.1989e-05, 6.1989e-05, 1.8132e-04, 6.1989e-05, 3.1543e-04, 1.8132e-04,
          1.8132e-04, 1.8132e-04, 6.1989e-05, 6.1989e-05, 1.8132e-04, 1.8132e-04,
          3.1543e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 1.8132e-04, 4.3488e-04,
          1.8132e-04, 1.8132e-04, 4.3488e-04, 6.1989e-05, 1.8132e-04, 6.1989e-05,
          6.1989e-05, 1.8132e-04, 6.1989e-05, 4.3488e-04, 6.1989e-05, 6.1989e-05,
          1.8132e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 1.8132e-04, 6.1989e-05,
          4.3488e-04, 1.8132e-04, 1.8132e-04, 6.1989e-05, 6.1989e-05, 1.8132e-04,
          4.3488e-04, 6.1989e-05, 1.8132e-04, 6.1989e-05, 6.1989e-05, 1.8132e-04,
          1.8132e-04, 1.7929e-03, 6.1989e-05, 1.8132e-04, 6.1989e-05, 6.1989e-05,
          1.8132e-04, 1.8132e-04, 1.8132e-04, 1.8132e-04, 1.8132e-04, 6.1989e-05,
          6.1989e-05, 1.8132e-04, 1.8132e-04, 1.8132e-04, 6.1989e-05, 1.8132e-04,
          6.1989e-05, 6.1989e-05], dtype=torch.float16),
  tensor([0.0000e+00, 6.6757e-03, 5.6213e-02, 1.5854e-02, 1.6861e-02, 4.4373e-02,
          3.7384e-02, 1.2138e-02, 1.6861e-03, 6.2275e-04, 7.4673e-04, 1.4296e-03,
          4.9543e-04, 2.2030e-03, 5.2147e-03, 6.8855e-04, 1.1835e-03, 1.4015e-02,
          2.7637e-03, 5.3501e-04, 5.3940e-03, 2.2259e-03, 1.0262e-02, 2.1534e-03,
          1.5278e-03, 8.7404e-04, 9.2163e-03, 4.9553e-03, 7.0801e-03, 5.5733e-03,
          3.7918e-03, 2.6340e-03, 5.6839e-03, 1.1568e-03, 5.1069e-04, 1.6909e-03,
          1.2436e-03, 6.9189e-04, 1.3084e-03, 7.6675e-04, 6.6948e-03, 8.3685e-04,
          6.1750e-04, 4.6372e-04, 5.1022e-04, 3.7193e-04, 4.9553e-03, 1.2770e-03,
          3.7003e-03, 4.1656e-03, 4.4022e-03, 3.4466e-03, 2.3232e-03, 1.0538e-03,
          8.2159e-04, 1.1129e-03, 1.1034e-03, 6.6853e-04, 1.0529e-03, 8.1348e-04,
          8.7452e-04, 5.4779e-03, 1.3723e-03, 1.7729e-03, 5.1498e-04, 1.3504e-03,
          8.7452e-04, 3.8791e-04, 4.5705e-04, 3.8409e-04, 3.2067e-04, 5.1451e-04,
          7.6056e-04, 1.7662e-03, 1.2426e-03, 2.0561e-03, 6.5041e-04, 6.7091e-04,
          1.4458e-03, 6.7854e-04, 5.6887e-04, 6.0654e-04, 5.3215e-04, 5.3549e-04,
          6.3324e-04, 3.3627e-03, 1.0929e-03, 6.2132e-04, 3.6955e-04, 4.6492e-04,
          1.0099e-03, 8.5115e-04, 3.7217e-04, 7.4434e-04, 5.3787e-04, 7.1287e-04,
          7.4654e-03, 7.5912e-04, 2.6970e-03, 3.4180e-03, 6.0511e-04, 5.8270e-04,
          1.0214e-03, 6.6376e-04, 3.4094e-04, 1.0128e-03, 1.0805e-03, 3.1757e-03,
          1.9970e-03, 8.4114e-04, 1.0710e-03, 6.6948e-04, 6.3372e-04, 6.1989e-04,
          6.6090e-04, 7.7820e-04, 1.8520e-03, 1.5144e-03, 9.0551e-04, 3.0918e-03,
          4.5967e-03, 1.2856e-03, 4.0364e-04, 8.0776e-04, 1.1787e-03, 3.5024e-04,
          5.8413e-04, 4.9925e-04, 7.1049e-04, 1.3056e-03, 1.7586e-03, 2.2221e-03,
          1.1473e-03, 4.3392e-04, 3.0208e-04, 4.0030e-04, 5.8556e-04, 4.6897e-04,
          5.5542e-03, 3.8033e-03, 3.0270e-03, 3.4599e-03, 3.1281e-03, 4.3144e-03,
          5.9509e-04, 4.7064e-04, 5.5361e-04, 5.4312e-04, 6.0797e-04, 1.1454e-03,
          1.8940e-03, 1.0767e-03, 8.9693e-04, 4.6272e-03, 2.5482e-03, 1.4744e-03,
          9.2697e-04, 1.3008e-03, 4.0603e-04, 5.1117e-04, 1.7204e-03, 4.8790e-03,
          1.4706e-03, 1.8244e-03, 7.1945e-03, 2.0695e-03, 1.8091e-03, 1.1253e-03,
          9.1219e-04, 9.3222e-04, 7.7391e-04, 7.1478e-04, 3.2616e-03, 1.8587e-03,
          1.2894e-03, 1.1225e-03, 1.7824e-03, 1.7004e-03, 1.4114e-03, 1.7548e-03,
          7.6294e-04, 7.4196e-04, 1.2388e-03, 2.0313e-03, 2.3136e-03, 1.1320e-03,
          8.6498e-04, 6.0272e-03, 2.5425e-03, 1.0080e-03, 7.5960e-04, 1.1358e-03,
          1.7748e-03, 5.1212e-04, 1.1883e-03, 8.6441e-03, 9.2316e-04, 1.1330e-03,
          1.9112e-03, 3.9635e-03, 1.8482e-03, 6.1131e-04, 9.8133e-04, 8.4591e-04,
          5.5313e-04, 8.1587e-04, 6.5756e-04, 8.2731e-04, 1.3037e-03, 2.2945e-03,
          1.6088e-03, 3.1185e-03, 1.2054e-03, 5.9605e-04, 1.5039e-03, 6.6710e-04,
          1.8911e-03, 4.9782e-03, 1.4086e-03, 7.8773e-04, 2.6875e-03, 4.3831e-03,
          8.3160e-03, 1.1339e-03, 1.1797e-03, 8.8167e-04, 1.9684e-03, 1.1473e-03,
          5.6934e-04, 8.6927e-04, 5.6267e-04, 1.4772e-03, 1.5268e-03, 5.1422e-03,
          2.1420e-03, 1.0843e-03, 1.0586e-03, 9.9659e-04, 2.1439e-03, 3.6392e-03,
          2.3842e-03, 2.4529e-03, 1.8644e-03, 4.7150e-03, 7.6027e-03, 5.3368e-03,
          1.9760e-03, 9.7609e-04, 9.3079e-04, 1.1091e-03, 1.3094e-03, 1.1959e-03,
          6.2084e-04, 5.9557e-04, 1.1768e-03, 1.5364e-03, 2.3708e-03, 9.5825e-03,
          2.0256e-03, 1.1997e-03, 1.3294e-03, 1.2655e-03, 9.5129e-04, 2.6302e-03,
          1.9321e-03, 1.0958e-03, 2.0485e-03, 8.9340e-03, 7.6981e-03, 1.4334e-03,
          2.0523e-03, 9.1028e-04, 1.5993e-03, 1.1969e-03, 2.6569e-03, 1.7118e-03,
          1.6470e-03, 1.4172e-03, 3.7193e-03, 4.4441e-03, 2.1210e-03, 1.6375e-03,
          1.0128e-03, 1.1587e-03, 1.0433e-03, 1.4095e-03, 4.6806e-03, 3.3302e-03,
          2.2774e-03, 3.1624e-03, 4.0970e-03, 8.4000e-03, 2.1172e-03, 2.4033e-03,
          2.0981e-03, 1.0967e-03, 1.0166e-03, 2.0428e-03, 2.2068e-03, 1.6212e-03,
          1.2312e-03, 2.0008e-03, 3.0193e-03, 1.0803e-02, 1.7586e-03, 2.1839e-03,
          2.3174e-03, 4.5280e-03, 4.6463e-03, 4.3640e-03, 1.6670e-03, 1.2569e-03,
          4.6043e-03, 4.6120e-03, 1.2634e-02, 3.4580e-03, 7.6523e-03, 2.1267e-03,
          2.5101e-03, 2.6150e-03, 7.0267e-03, 4.4785e-03, 1.9817e-03, 2.2640e-03,
          6.2675e-03, 5.0354e-03, 9.6359e-03, 3.0899e-03, 4.3678e-03, 9.3918e-03,
          4.9362e-03, 3.8815e-03, 7.3395e-03, 7.6332e-03, 2.2659e-03, 6.2866e-02,
          4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 2.6703e-04, 4.1902e-05,
          4.1902e-05, 4.1902e-05, 2.6703e-04, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          4.1902e-05, 2.6703e-04, 4.1902e-05, 4.1902e-05, 2.6703e-04, 2.6703e-04,
          4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          2.6703e-04, 2.6703e-04, 4.1902e-05, 4.1902e-05, 2.6703e-04, 4.1902e-05,
          4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          2.6703e-04, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          2.6703e-04, 4.1902e-05, 2.6703e-04, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          2.6703e-04, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          4.1902e-05, 2.6703e-04, 4.1902e-05, 4.1902e-05, 4.1902e-05, 4.1902e-05,
          2.6703e-04, 2.6703e-04, 2.6703e-04, 2.6703e-04, 7.7105e-04, 2.6703e-04,
          4.1902e-05, 2.6703e-04, 2.6703e-04, 2.6703e-04, 4.1902e-05, 2.6703e-04,
          2.6703e-04, 2.6703e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.6068e-03, 5.1575e-02, 1.1864e-02, 8.7128e-03, 3.2532e-02,
          4.6478e-02, 6.2256e-03, 1.1368e-03, 6.7186e-04, 1.1644e-03, 2.4357e-03,
          1.0052e-03, 7.2622e-04, 3.5954e-04, 1.6785e-03, 1.0214e-03, 5.6410e-04,
          3.7551e-04, 4.4870e-04, 2.0564e-04, 2.1446e-04, 1.8895e-04, 5.7554e-04,
          2.8706e-03, 2.0390e-03, 5.2376e-03, 1.7805e-03, 3.0947e-04, 7.1764e-04,
          4.4250e-04, 2.2030e-03, 3.0065e-04, 1.6232e-03, 7.2527e-04, 1.2217e-03,
          1.8435e-03, 1.0557e-03, 6.7806e-04, 1.0433e-03, 2.4700e-04, 6.0606e-04,
          4.4203e-04, 3.4666e-04, 4.2963e-04, 2.5916e-04, 6.1321e-04, 1.3475e-03,
          2.1553e-04, 3.4714e-04, 2.1172e-03, 2.0409e-03, 1.5163e-03, 8.9788e-04,
          7.8297e-04, 1.1482e-03, 1.4458e-03, 9.0599e-04, 4.5586e-03, 2.1629e-03,
          3.0861e-03, 6.6710e-04, 1.9321e-03, 1.4458e-03, 6.3896e-04, 2.6169e-03,
          7.1287e-04, 4.5657e-04, 3.8457e-04, 2.4724e-04, 2.4283e-04, 4.2844e-04,
          4.7040e-04, 1.8663e-03, 1.1463e-03, 2.6588e-03, 9.1887e-04, 7.2813e-04,
          1.5545e-03, 8.5735e-04, 8.7166e-04, 2.0657e-03, 1.2188e-03, 2.4738e-03,
          1.6794e-03, 3.2310e-03, 1.5602e-03, 9.7322e-04, 3.7789e-04, 4.3797e-04,
          2.1152e-03, 9.8324e-04, 4.1795e-04, 1.3094e-03, 7.9966e-04, 7.4816e-04,
          4.5896e-04, 6.1464e-04, 1.4725e-03, 1.4324e-03, 9.3699e-04, 1.0061e-03,
          2.7847e-03, 6.5041e-04, 7.4911e-04, 3.8948e-03, 4.7951e-03, 7.1764e-04,
          3.5820e-03, 5.1498e-04, 1.3008e-03, 5.6839e-04, 6.5804e-04, 4.7398e-04,
          2.1381e-03, 7.4673e-04, 3.1614e-04, 3.0861e-03, 9.9277e-04, 8.4734e-04,
          1.0042e-03, 2.0905e-03, 5.1117e-04, 1.1129e-03, 3.6678e-03, 4.6563e-04,
          7.8058e-04, 8.6212e-04, 1.2341e-03, 4.5090e-03, 1.4524e-03, 4.2496e-03,
          1.1568e-03, 4.3511e-04, 6.3419e-04, 2.7084e-04, 9.8610e-04, 3.9840e-04,
          1.1530e-03, 4.4298e-04, 1.0338e-03, 1.8721e-03, 1.5287e-03, 1.0443e-03,
          7.9250e-04, 9.6798e-04, 1.1187e-03, 1.0033e-03, 9.4795e-04, 8.2550e-03,
          4.3144e-03, 4.8409e-03, 4.1428e-03, 2.2125e-03, 4.0169e-03, 1.0691e-03,
          1.0080e-03, 8.1587e-04, 5.8937e-04, 5.7125e-04, 2.5330e-03, 6.7472e-04,
          1.5383e-03, 1.1339e-03, 9.1696e-04, 1.6108e-03, 1.1606e-03, 1.0061e-03,
          1.8930e-03, 1.0252e-03, 2.9793e-03, 1.5936e-03, 7.9012e-04, 2.4529e-03,
          3.3539e-02, 3.4729e-02, 7.9956e-03, 9.5367e-04, 2.2564e-03, 1.3371e-03,
          1.0319e-03, 1.4133e-03, 2.1763e-03, 2.4643e-03, 1.0710e-03, 9.0408e-04,
          1.6613e-03, 7.9727e-04, 1.2646e-03, 1.0595e-03, 1.0490e-03, 2.2335e-03,
          2.2240e-03, 6.9714e-04, 1.7967e-03, 1.2608e-03, 1.6098e-03, 2.2640e-03,
          2.0935e-02, 1.3113e-03, 2.9316e-03, 7.8487e-04, 6.8903e-04, 9.2793e-04,
          7.3099e-04, 1.0900e-03, 6.8951e-04, 1.1311e-03, 2.3842e-03, 5.2977e-04,
          1.4896e-03, 1.4315e-03, 1.1187e-03, 1.2598e-03, 3.7479e-03, 1.2980e-03,
          2.3403e-03, 1.6022e-03, 1.5640e-03, 1.4648e-03, 2.8629e-03, 1.8740e-03,
          2.5826e-03, 1.1568e-03, 1.6747e-03, 7.5388e-04, 4.5128e-03, 1.7033e-03,
          6.2227e-04, 1.5039e-03, 1.1139e-03, 1.6571e-02, 3.5248e-03, 8.2064e-04,
          1.5335e-03, 1.2197e-03, 1.8625e-03, 5.3291e-03, 6.2523e-03, 1.5930e-02,
          8.6670e-03, 5.6725e-03, 3.4599e-03, 2.8019e-03, 3.1109e-03, 1.3733e-03,
          1.0738e-03, 1.7300e-03, 2.0046e-03, 3.4180e-03, 3.2291e-03, 3.3512e-03,
          6.5041e-04, 8.6308e-04, 2.0084e-03, 2.0542e-03, 1.3437e-03, 3.7174e-03,
          1.9121e-03, 1.5230e-03, 8.0261e-03, 5.8365e-03, 3.7060e-03, 9.5901e-03,
          5.8975e-03, 1.3304e-03, 1.7147e-03, 3.2883e-03, 2.4643e-03, 3.3455e-03,
          2.8706e-03, 8.5926e-04, 4.1847e-03, 1.8702e-03, 2.2850e-03, 6.0005e-03,
          8.2626e-03, 6.2981e-03, 5.3167e-04, 9.9373e-04, 1.2131e-03, 2.1248e-03,
          1.3666e-03, 1.7757e-03, 1.2131e-03, 3.0861e-03, 4.7874e-03, 3.7613e-03,
          1.1253e-02, 2.9602e-03, 1.6041e-03, 1.5965e-03, 2.4719e-03, 4.8027e-03,
          1.7605e-03, 1.1225e-03, 8.6689e-04, 2.2087e-03, 3.1261e-03, 2.9812e-03,
          1.7796e-03, 1.3618e-03, 2.9049e-03, 2.8477e-03, 1.7290e-03, 7.7248e-03,
          3.6583e-03, 1.8549e-03, 3.7975e-03, 2.0542e-03, 2.3403e-03, 2.3632e-03,
          1.8387e-03, 1.4771e-02, 2.9583e-03, 4.0665e-03, 2.8534e-03, 2.9716e-03,
          6.9199e-03, 4.1313e-03, 1.6613e-03, 1.4238e-03, 1.8559e-03, 1.5640e-03,
          1.3742e-03, 1.3113e-03, 2.0618e-03, 3.1471e-03, 8.6670e-03, 3.3035e-03,
          4.2000e-03, 2.2373e-03, 3.1071e-03, 5.4016e-03, 1.5202e-03, 5.1384e-03,
          5.3223e-02, 6.4194e-05, 1.7357e-04, 1.7357e-04, 1.7357e-04, 1.7357e-04,
          6.4194e-05, 6.4194e-05, 1.7357e-04, 1.7357e-04, 6.4194e-05, 6.4194e-05,
          1.7357e-04, 1.7357e-04, 6.4194e-05, 6.4194e-05, 1.7357e-04, 1.7357e-04,
          6.4194e-05, 6.4194e-05, 6.4194e-05, 1.7357e-04, 1.7357e-04, 6.4194e-05,
          1.7357e-04, 1.7357e-04, 1.7357e-04, 6.4194e-05, 1.7357e-04, 6.4194e-05,
          1.7357e-04, 1.7357e-04, 6.4194e-05, 6.4194e-05, 6.4194e-05, 6.4194e-05,
          1.7357e-04, 6.4194e-05, 6.4194e-05, 6.4194e-05, 1.7357e-04, 6.4194e-05,
          1.7357e-04, 1.7357e-04, 1.7357e-04, 6.4194e-05, 6.4194e-05, 1.7357e-04,
          1.7357e-04, 1.7357e-04, 1.7357e-04, 6.4194e-05, 1.7357e-04, 1.7357e-04,
          1.7357e-04, 1.7357e-04, 6.4194e-05, 6.4194e-05, 6.4194e-05, 6.4194e-05,
          1.1148e-03, 4.5657e-04, 4.5657e-04, 2.4700e-04, 1.7357e-04, 1.7357e-04,
          6.4194e-05, 1.7357e-04, 1.7357e-04, 1.7357e-04, 1.7357e-04, 6.8474e-04,
          1.7357e-04, 1.7357e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.7498e-03, 5.1666e-02, 1.2619e-02, 1.6327e-02, 2.9114e-02,
          3.1830e-02, 1.0551e-02, 4.2343e-03, 6.3610e-04, 1.9169e-03, 2.2984e-03,
          9.4652e-04, 7.6008e-04, 5.3215e-04, 1.2751e-03, 3.3932e-03, 1.5879e-03,
          8.1062e-04, 8.1158e-04, 2.7370e-04, 3.3879e-04, 2.9802e-04, 1.4982e-03,
          2.6188e-03, 1.1568e-03, 8.9951e-03, 4.4403e-03, 3.8362e-04, 1.4791e-03,
          6.7043e-04, 1.3069e-02, 5.6648e-04, 2.7237e-03, 2.6875e-03, 6.4354e-03,
          1.6966e-03, 5.0783e-04, 4.9210e-03, 4.0436e-03, 3.6979e-04, 1.4038e-03,
          4.7541e-04, 3.6168e-04, 3.6931e-04, 3.1424e-04, 1.3351e-03, 2.6093e-03,
          3.7646e-04, 4.4370e-04, 2.2316e-03, 9.5825e-03, 1.3748e-02, 3.4924e-03,
          1.1377e-03, 5.7259e-03, 1.5936e-03, 4.5991e-04, 1.2169e-03, 6.4373e-04,
          7.4387e-04, 1.0319e-03, 2.1305e-03, 5.8556e-03, 4.7588e-04, 2.5787e-03,
          6.4898e-04, 4.3774e-04, 9.1982e-04, 2.4295e-04, 2.2483e-04, 3.5238e-04,
          4.5753e-04, 3.9368e-03, 2.1343e-03, 1.7609e-02, 1.4229e-03, 1.5726e-03,
          5.2299e-03, 3.7909e-04, 4.2295e-04, 1.3590e-03, 5.1212e-04, 5.6410e-04,
          5.6553e-04, 2.5284e-02, 1.8482e-03, 1.2579e-03, 3.7313e-04, 3.6192e-04,
          2.5024e-03, 1.5678e-03, 2.0969e-04, 1.4505e-03, 3.7384e-04, 5.0592e-04,
          7.0524e-04, 8.5878e-04, 1.3145e-02, 2.2430e-03, 3.2449e-04, 4.7946e-04,
          2.0523e-03, 5.6791e-04, 5.0783e-04, 1.3514e-03, 2.2240e-03, 7.7677e-04,
          1.2108e-02, 4.7112e-04, 1.8635e-03, 3.3545e-04, 5.0402e-04, 6.4945e-04,
          9.1887e-04, 7.1287e-04, 4.7374e-04, 2.6360e-03, 6.7186e-04, 8.0729e-04,
          9.4986e-04, 1.5011e-03, 2.2078e-04, 2.2278e-03, 2.4738e-03, 2.0909e-04,
          2.6989e-04, 2.9969e-04, 4.1485e-04, 2.8858e-03, 6.3057e-03, 1.2009e-02,
          8.7500e-04, 4.9448e-04, 9.3985e-04, 3.9601e-04, 1.0109e-03, 2.8801e-04,
          4.0131e-03, 7.3671e-04, 2.2869e-03, 7.1449e-03, 4.4327e-03, 1.1644e-03,
          2.9826e-04, 7.7200e-04, 4.6897e-04, 4.3511e-04, 4.5419e-04, 2.0905e-03,
          2.9144e-03, 2.2564e-03, 7.4530e-04, 1.1627e-02, 5.2376e-03, 1.4229e-03,
          1.2760e-03, 1.0586e-03, 8.8596e-04, 3.4595e-04, 2.4147e-03, 1.6680e-03,
          2.8019e-03, 3.6373e-03, 2.3918e-03, 5.3940e-03, 7.3528e-04, 7.5531e-04,
          1.3571e-03, 1.0042e-03, 1.4477e-03, 1.2741e-03, 1.5287e-03, 2.8133e-03,
          1.8597e-03, 1.3542e-03, 1.5574e-03, 1.0767e-03, 2.4147e-03, 3.6736e-03,
          1.2140e-03, 2.0790e-03, 2.7580e-03, 1.1536e-02, 3.3531e-03, 2.6340e-03,
          2.1286e-03, 1.4563e-03, 3.1414e-03, 5.0974e-04, 4.8184e-04, 2.8191e-03,
          1.1383e-02, 2.5034e-04, 1.9855e-03, 2.6398e-03, 1.7357e-03, 2.1706e-03,
          1.5783e-03, 9.0647e-04, 3.9978e-03, 6.9714e-04, 1.3037e-03, 1.6184e-03,
          4.3797e-04, 1.0843e-03, 5.3406e-04, 2.5463e-03, 2.7981e-03, 1.4277e-03,
          6.0387e-03, 1.2264e-03, 5.2786e-04, 8.0061e-04, 2.6741e-03, 4.2367e-04,
          1.8139e-03, 2.2793e-03, 5.1537e-03, 1.3161e-03, 4.5891e-03, 1.5907e-03,
          6.2866e-03, 5.4121e-04, 1.1511e-03, 1.0185e-03, 4.5357e-03, 3.7842e-03,
          6.7902e-04, 5.6839e-04, 4.6444e-04, 1.5802e-03, 2.4147e-03, 1.1444e-03,
          9.1171e-04, 8.1348e-04, 4.9973e-04, 1.7796e-03, 1.5593e-03, 2.5921e-03,
          9.6798e-04, 1.0262e-03, 8.4448e-04, 2.5558e-03, 2.9297e-03, 1.2369e-03,
          9.9277e-04, 1.0929e-03, 4.4394e-04, 6.1941e-04, 7.5054e-04, 2.4433e-03,
          4.2319e-04, 4.5848e-04, 1.0195e-03, 1.2064e-03, 3.9825e-03, 2.2232e-02,
          9.1028e-04, 9.1887e-04, 1.3294e-03, 1.4286e-03, 4.6110e-04, 1.2627e-03,
          2.1133e-03, 3.9444e-03, 2.2449e-03, 4.4975e-03, 3.8853e-03, 2.1362e-03,
          1.0700e-03, 8.5831e-04, 3.0956e-03, 8.9455e-04, 3.1872e-03, 2.6035e-03,
          1.4629e-03, 1.9159e-03, 7.9155e-04, 1.7881e-03, 8.0204e-04, 7.9346e-04,
          8.8310e-04, 5.1355e-04, 3.9220e-04, 6.7282e-04, 2.2335e-03, 2.7390e-03,
          1.7862e-03, 7.4501e-03, 3.3207e-03, 3.2597e-03, 1.0452e-03, 1.3418e-03,
          1.1835e-03, 6.1893e-04, 5.3453e-04, 4.2763e-03, 2.7809e-03, 1.2302e-03,
          8.6880e-04, 3.1757e-03, 1.2596e-02, 3.7422e-03, 5.8413e-04, 3.0727e-03,
          3.4981e-03, 9.6035e-04, 3.5858e-03, 2.2774e-03, 4.4518e-03, 2.6875e-03,
          2.2678e-03, 2.6417e-03, 2.5043e-03, 6.4354e-03, 9.0561e-03, 5.4092e-03,
          1.8396e-03, 5.6877e-03, 3.6068e-03, 2.2488e-03, 1.0824e-03, 1.2026e-03,
          2.8610e-03, 1.7719e-03, 3.7251e-03, 8.7023e-04, 2.5024e-03, 3.2940e-03,
          6.0425e-03, 1.5287e-03, 2.2469e-03, 1.6510e-02, 4.8409e-03, 4.8180e-03,
          2.0199e-03, 4.8157e-02, 4.1604e-05, 4.1604e-05, 4.1604e-05, 4.4656e-04,
          4.1604e-05, 4.1604e-05, 1.9956e-04, 1.9956e-04, 4.1604e-05, 1.9956e-04,
          1.9956e-04, 1.9956e-04, 4.1604e-05, 4.1604e-05, 1.9956e-04, 1.9956e-04,
          4.1604e-05, 1.9956e-04, 4.1604e-05, 4.1604e-05, 4.4656e-04, 4.1604e-05,
          1.9956e-04, 1.9956e-04, 4.1604e-05, 4.1604e-05, 4.1604e-05, 4.1604e-05,
          4.1604e-05, 1.9956e-04, 4.1604e-05, 1.9956e-04, 4.1604e-05, 4.1604e-05,
          1.9956e-04, 4.1604e-05, 4.1604e-05, 4.1604e-05, 1.9956e-04, 4.1604e-05,
          1.9956e-04, 4.4656e-04, 1.9956e-04, 4.1604e-05, 4.1604e-05, 1.9956e-04,
          1.9956e-04, 4.1604e-05, 4.4656e-04, 4.1604e-05, 4.1604e-05, 4.1604e-05,
          1.7033e-03, 1.9956e-04, 4.1604e-05, 4.1604e-05, 4.1604e-05, 4.1604e-05,
          4.1604e-05, 4.1604e-05, 1.9956e-04, 4.1604e-05, 4.1604e-05, 4.1604e-05,
          4.1604e-05, 4.1604e-05, 4.1604e-05, 4.1604e-05, 4.1604e-05, 1.9956e-04,
          1.9956e-04, 4.1604e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.5629e-03, 5.3131e-02, 1.2817e-02, 1.3634e-02, 2.7924e-02,
          3.3203e-02, 8.7585e-03, 6.3972e-03, 1.5793e-03, 1.4000e-03, 2.1324e-03,
          7.9775e-04, 9.5415e-04, 5.5218e-04, 1.0405e-03, 6.7024e-03, 1.6365e-03,
          9.3460e-04, 1.6403e-03, 2.6107e-04, 2.7061e-04, 3.1877e-04, 1.5764e-03,
          2.8343e-03, 1.0691e-03, 7.2899e-03, 4.5242e-03, 4.0770e-04, 1.6298e-03,
          6.7854e-04, 1.5945e-02, 5.9986e-04, 1.6518e-03, 1.3704e-03, 6.2675e-03,
          9.8801e-04, 3.7956e-04, 1.7538e-03, 1.6508e-03, 3.5644e-04, 9.7847e-04,
          4.0293e-04, 2.9421e-04, 3.4666e-04, 2.9612e-04, 1.5173e-03, 1.7567e-03,
          3.8409e-04, 3.9577e-04, 2.0275e-03, 9.0714e-03, 1.6327e-02, 3.7727e-03,
          1.2960e-03, 6.9199e-03, 1.4143e-03, 3.9268e-04, 1.2283e-03, 5.3549e-04,
          5.9080e-04, 9.0647e-04, 3.0384e-03, 3.6240e-03, 4.3941e-04, 2.0676e-03,
          6.9427e-04, 4.3654e-04, 5.7364e-04, 2.1052e-04, 1.7858e-04, 3.2616e-04,
          3.9339e-04, 3.0899e-03, 2.3899e-03, 1.9501e-02, 1.4830e-03, 2.0027e-03,
          6.2447e-03, 3.1185e-04, 3.9577e-04, 8.3447e-04, 4.6539e-04, 4.3583e-04,
          3.9148e-04, 2.5101e-02, 2.0447e-03, 1.1587e-03, 3.7479e-04, 3.4761e-04,
          2.4815e-03, 1.4181e-03, 2.1720e-04, 1.3990e-03, 3.2902e-04, 3.9554e-04,
          7.0715e-04, 1.0662e-03, 1.3489e-02, 1.8797e-03, 2.9683e-04, 4.1938e-04,
          1.9131e-03, 4.9162e-04, 4.4370e-04, 7.5865e-04, 1.5116e-03, 6.2561e-04,
          1.5472e-02, 4.6086e-04, 1.7519e-03, 3.2020e-04, 5.0831e-04, 5.7316e-04,
          8.1635e-04, 5.7745e-04, 5.0402e-04, 2.3117e-03, 5.2166e-04, 7.8392e-04,
          8.6880e-04, 1.0443e-03, 1.9038e-04, 3.5610e-03, 1.6718e-03, 1.8656e-04,
          2.1040e-04, 2.8133e-04, 3.7646e-04, 1.9045e-03, 7.0229e-03, 1.7166e-02,
          9.0218e-04, 4.6897e-04, 7.1812e-04, 3.7766e-04, 9.5320e-04, 2.6393e-04,
          4.2534e-03, 8.4639e-04, 3.6716e-03, 5.8327e-03, 3.0785e-03, 1.1330e-03,
          2.4867e-04, 5.5790e-04, 3.4809e-04, 3.9148e-04, 4.2033e-04, 1.4343e-03,
          2.1420e-03, 1.3361e-03, 5.5599e-04, 1.0803e-02, 6.5842e-03, 1.0529e-03,
          1.1921e-03, 9.3412e-04, 9.8515e-04, 3.4857e-04, 1.9083e-03, 1.7118e-03,
          2.2049e-03, 2.2774e-03, 2.8305e-03, 2.4815e-03, 6.1131e-04, 6.4802e-04,
          1.2989e-03, 8.0585e-04, 9.2936e-04, 9.2268e-04, 1.7376e-03, 1.7576e-03,
          1.6346e-03, 1.3351e-03, 8.2302e-04, 9.0361e-04, 1.5612e-03, 3.8834e-03,
          8.8310e-04, 1.2331e-03, 1.9875e-03, 1.5289e-02, 4.5166e-03, 1.4858e-03,
          2.1820e-03, 1.6603e-03, 2.6207e-03, 4.0293e-04, 4.5085e-04, 1.6651e-03,
          1.5045e-02, 1.9300e-04, 1.3380e-03, 2.8725e-03, 1.5011e-03, 1.3123e-03,
          1.0996e-03, 8.4782e-04, 2.4128e-03, 6.1655e-04, 1.2407e-03, 1.1148e-03,
          4.3249e-04, 1.0042e-03, 4.5943e-04, 4.7035e-03, 2.7580e-03, 1.5526e-03,
          1.0986e-02, 9.0694e-04, 4.4584e-04, 7.7152e-04, 1.8578e-03, 3.7026e-04,
          1.2093e-03, 2.2278e-03, 2.9945e-03, 1.1530e-03, 2.9106e-03, 1.9341e-03,
          4.0932e-03, 5.4073e-04, 9.8228e-04, 9.3460e-04, 3.5000e-03, 2.1172e-03,
          7.1669e-04, 6.0081e-04, 4.3154e-04, 1.6947e-03, 1.6518e-03, 1.7710e-03,
          7.9060e-04, 6.8665e-04, 5.0497e-04, 1.4915e-03, 1.0567e-03, 2.7008e-03,
          1.1282e-03, 9.2697e-04, 6.7425e-04, 1.7080e-03, 3.5172e-03, 1.2951e-03,
          8.2493e-04, 1.3142e-03, 5.3120e-04, 7.6246e-04, 9.0837e-04, 1.7395e-03,
          4.2462e-04, 3.7670e-04, 8.7404e-04, 1.0366e-03, 3.5286e-03, 1.7731e-02,
          6.9952e-04, 8.4972e-04, 1.0023e-03, 1.0538e-03, 4.1485e-04, 1.2598e-03,
          1.4572e-03, 1.7643e-03, 2.5539e-03, 6.0806e-03, 4.0665e-03, 1.6289e-03,
          1.3876e-03, 8.6975e-04, 2.5940e-03, 8.4162e-04, 2.7084e-03, 2.1248e-03,
          1.6298e-03, 2.2545e-03, 8.9741e-04, 1.8826e-03, 6.4278e-04, 8.1730e-04,
          9.2506e-04, 4.4703e-04, 3.3283e-04, 5.3072e-04, 2.4719e-03, 2.1858e-03,
          2.0313e-03, 4.0054e-03, 3.8147e-03, 4.2496e-03, 1.0538e-03, 1.7900e-03,
          1.1234e-03, 6.6185e-04, 4.9305e-04, 3.2158e-03, 2.4509e-03, 1.6174e-03,
          8.4591e-04, 3.8586e-03, 2.1942e-02, 4.9744e-03, 5.2500e-04, 2.8515e-03,
          2.2507e-03, 9.5034e-04, 2.3041e-03, 4.6349e-03, 2.8133e-03, 2.5520e-03,
          3.1891e-03, 2.7962e-03, 2.2888e-03, 5.1956e-03, 5.9624e-03, 3.3131e-03,
          2.8248e-03, 6.1913e-03, 4.6310e-03, 3.2597e-03, 9.0075e-04, 8.8978e-04,
          4.6959e-03, 3.1433e-03, 3.7060e-03, 7.3576e-04, 1.9569e-03, 3.5172e-03,
          4.3182e-03, 3.0117e-03, 2.4662e-03, 6.8932e-03, 3.7136e-03, 6.9122e-03,
          2.9583e-03, 2.5375e-02, 4.6692e-02, 6.6400e-05, 1.2274e-03, 6.6400e-05,
          6.6400e-05, 3.4142e-04, 2.2566e-04, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          6.6400e-05, 2.2566e-04, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          6.6400e-05, 6.6400e-05, 3.4142e-04, 6.6400e-05, 2.2566e-04, 2.2566e-04,
          6.6400e-05, 2.2566e-04, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          2.2566e-04, 6.6400e-05, 6.6400e-05, 1.0672e-03, 6.6400e-05, 6.6400e-05,
          2.2566e-04, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          2.2566e-04, 3.4142e-04, 2.2566e-04, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          2.2566e-04, 6.6400e-05, 5.0068e-04, 6.6400e-05, 2.2566e-04, 2.2566e-04,
          6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05, 6.6400e-05,
          2.2566e-04, 2.2566e-04, 2.2566e-04, 2.2566e-04, 2.2566e-04, 6.6400e-05,
          6.6400e-05, 2.2566e-04, 2.2566e-04, 2.2566e-04, 6.6400e-05, 2.2566e-04,
          2.2566e-04, 2.2566e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.6147e-03, 4.6478e-02, 1.7929e-02, 1.8875e-02, 6.4819e-02,
          2.6825e-02, 1.6144e-02, 6.6328e-04, 3.8266e-04, 7.0429e-04, 1.3456e-03,
          4.7660e-04, 1.6136e-03, 2.1172e-03, 7.6151e-04, 6.2037e-04, 4.2000e-03,
          2.2602e-03, 3.1328e-04, 1.3628e-03, 1.1692e-03, 1.4973e-03, 1.5659e-03,
          1.4410e-03, 8.0013e-04, 1.1780e-02, 3.3493e-03, 2.0962e-03, 2.9335e-03,
          2.2411e-03, 1.1187e-03, 1.5888e-03, 1.1101e-03, 5.3215e-04, 6.1417e-04,
          8.6880e-04, 5.5599e-04, 1.1406e-03, 7.8964e-04, 2.2793e-03, 1.0223e-03,
          8.6212e-04, 6.2609e-04, 7.4720e-04, 4.5490e-04, 2.5673e-03, 1.5688e-03,
          1.2474e-03, 2.0142e-03, 8.2169e-03, 9.7322e-04, 7.5102e-04, 4.0126e-04,
          4.0293e-04, 5.5218e-04, 7.0381e-04, 5.7077e-04, 6.6233e-04, 6.9523e-04,
          8.5831e-04, 5.2757e-03, 9.4748e-04, 1.1129e-03, 6.0654e-04, 1.6241e-03,
          8.0585e-04, 2.8276e-04, 4.8041e-04, 5.4455e-04, 5.3835e-04, 5.6934e-04,
          7.8392e-04, 1.3008e-03, 7.3528e-04, 1.0166e-03, 3.4881e-04, 2.7871e-04,
          8.3876e-04, 5.9080e-04, 5.1069e-04, 4.4084e-04, 4.5395e-04, 5.1880e-04,
          7.2289e-04, 1.6823e-03, 7.1049e-04, 5.4264e-04, 3.1757e-04, 4.7469e-04,
          9.2411e-04, 5.6124e-04, 4.1628e-04, 6.1226e-04, 6.2418e-04, 1.0414e-03,
          1.7681e-03, 5.2929e-04, 7.4196e-04, 1.0443e-03, 6.0177e-04, 4.3464e-04,
          8.9550e-04, 3.2234e-04, 2.9635e-04, 7.0715e-04, 8.9836e-04, 6.0234e-03,
          1.2712e-03, 9.7418e-04, 8.5211e-04, 5.5504e-04, 5.5790e-04, 4.2391e-04,
          5.2977e-04, 1.1444e-03, 4.0665e-03, 1.8282e-03, 1.3466e-03, 3.9368e-03,
          5.1498e-03, 7.1907e-04, 3.4833e-04, 3.2640e-04, 1.1072e-03, 2.4509e-04,
          5.1403e-04, 3.6526e-04, 4.4179e-04, 1.1187e-03, 8.0776e-04, 1.3504e-03,
          8.2111e-04, 4.3130e-04, 2.6774e-04, 2.8610e-04, 4.9496e-04, 6.7806e-04,
          4.5433e-03, 5.4054e-03, 3.1471e-03, 2.0885e-03, 1.4229e-03, 4.6883e-03,
          4.7588e-04, 2.2125e-04, 4.2534e-04, 3.6573e-04, 5.4646e-04, 9.5367e-04,
          9.6512e-04, 9.3508e-04, 7.7009e-04, 1.7090e-03, 1.6642e-03, 1.8320e-03,
          5.5456e-04, 7.5197e-04, 3.1710e-04, 5.8842e-04, 2.1019e-03, 9.0408e-03,
          1.5411e-03, 9.7275e-04, 3.7861e-03, 1.1683e-03, 1.4400e-03, 3.9649e-04,
          6.8760e-04, 3.6955e-04, 5.7650e-04, 5.3596e-04, 2.9545e-03, 1.2989e-03,
          6.9952e-04, 6.8712e-04, 2.2583e-03, 2.6684e-03, 1.2894e-03, 1.1282e-03,
          5.6505e-04, 7.3910e-04, 1.2512e-03, 9.4795e-04, 6.8808e-04, 8.0729e-04,
          6.7282e-04, 6.0234e-03, 1.1463e-03, 9.1553e-04, 5.9509e-04, 9.3842e-04,
          8.6641e-04, 4.1866e-04, 8.5545e-04, 4.6692e-03, 5.3263e-04, 9.2173e-04,
          1.5736e-03, 6.8398e-03, 1.9875e-03, 7.1669e-04, 6.8855e-04, 7.6866e-04,
          7.5054e-04, 9.2411e-04, 1.4467e-03, 4.4942e-04, 1.0738e-03, 1.9665e-03,
          7.5769e-04, 2.3727e-03, 1.3638e-03, 4.4322e-04, 1.2245e-03, 5.4169e-04,
          8.1205e-04, 5.8517e-03, 8.9884e-04, 5.4026e-04, 1.8883e-03, 6.2866e-03,
          4.9553e-03, 1.9722e-03, 1.3123e-03, 6.0654e-04, 2.3327e-03, 9.6703e-04,
          7.1049e-04, 7.2002e-04, 9.4604e-04, 9.7418e-04, 1.3018e-03, 6.8703e-03,
          2.5826e-03, 5.1928e-04, 5.5504e-04, 5.1880e-04, 1.5020e-03, 2.3918e-03,
          1.4496e-03, 2.0161e-03, 1.1911e-03, 2.8400e-03, 1.3870e-02, 2.0905e-02,
          3.5210e-03, 1.0138e-03, 1.3943e-03, 9.1839e-04, 1.9684e-03, 1.5392e-03,
          8.0872e-04, 1.5478e-03, 2.7008e-03, 1.7653e-03, 2.2812e-03, 2.5539e-03,
          1.4601e-03, 5.7030e-04, 6.6519e-04, 7.2765e-04, 6.6566e-04, 1.5497e-03,
          1.1454e-03, 5.6553e-04, 3.0670e-03, 1.4351e-02, 8.4534e-03, 1.5230e-03,
          1.7929e-03, 1.0176e-03, 2.1477e-03, 2.0943e-03, 2.2774e-03, 2.0027e-03,
          1.7958e-03, 7.0667e-04, 2.0859e-02, 7.5340e-03, 2.2373e-03, 1.0252e-03,
          7.6771e-04, 1.1673e-03, 1.1673e-03, 1.3866e-03, 3.3550e-03, 1.1253e-03,
          1.1616e-03, 1.9493e-03, 3.8013e-03, 3.3607e-03, 3.5534e-03, 2.5806e-03,
          4.3411e-03, 2.4223e-03, 2.2202e-03, 2.1305e-03, 1.4591e-03, 3.3150e-03,
          3.2120e-03, 1.8911e-03, 1.2913e-03, 1.0429e-02, 2.5043e-03, 1.3323e-03,
          1.5574e-03, 8.8196e-03, 1.9855e-03, 1.6241e-03, 6.2037e-04, 5.8603e-04,
          3.6106e-03, 1.3494e-03, 3.8574e-02, 1.9073e-03, 4.8409e-03, 2.4586e-03,
          1.4648e-03, 1.9989e-03, 1.0468e-02, 6.6872e-03, 6.2408e-03, 6.8398e-03,
          4.1733e-03, 5.1270e-03, 5.7449e-03, 4.1275e-03, 2.9068e-03, 1.5388e-02,
          2.4815e-03, 3.7823e-03, 1.7578e-02, 4.3716e-03, 1.0843e-03, 5.4970e-03,
          7.8344e-04, 2.2411e-03, 1.7862e-03, 8.0505e-02, 5.1320e-05, 5.1320e-05,
          5.1320e-05, 1.9264e-04, 5.1320e-05, 5.1320e-05, 2.9683e-04, 5.1320e-05,
          5.1320e-05, 2.3818e-04, 5.1320e-05, 5.1320e-05, 1.9264e-04, 1.9264e-04,
          2.9683e-04, 5.1320e-05, 1.9264e-04, 5.1320e-05, 5.1320e-05, 5.1320e-05,
          1.9264e-04, 1.9264e-04, 5.1320e-05, 5.1320e-05, 5.1320e-05, 5.1320e-05,
          1.9264e-04, 5.1320e-05, 5.1320e-05, 4.8351e-04, 5.1320e-05, 5.1320e-05,
          5.1320e-05, 5.1320e-05, 5.1320e-05, 5.1320e-05, 1.9264e-04, 5.1320e-05,
          5.1320e-05, 1.9264e-04, 3.7956e-04, 5.1320e-05, 5.1320e-05, 5.1320e-05,
          3.7956e-04, 5.1320e-05, 1.9264e-04, 5.1320e-05, 5.1320e-05, 5.1320e-05,
          5.1320e-05, 8.0156e-04, 5.1320e-05, 5.1320e-05, 5.1320e-05, 5.1320e-05,
          2.3818e-04, 4.6206e-04, 2.3818e-04, 3.2473e-04, 4.6206e-04, 5.1320e-05,
          5.1320e-05, 6.0320e-04, 4.6206e-04, 6.1560e-04, 5.1320e-05, 5.1320e-05,
          5.1320e-05, 5.1320e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.1924e-03, 5.4047e-02, 1.1581e-02, 1.8387e-02, 3.1677e-02,
          3.2440e-02, 1.1002e-02, 4.9543e-04, 2.9087e-04, 2.7618e-03, 5.0964e-03,
          2.1000e-03, 6.2180e-04, 4.1199e-04, 8.4066e-04, 5.2691e-04, 1.0929e-03,
          7.0000e-04, 3.2949e-04, 2.2221e-04, 3.7146e-04, 3.0875e-04, 1.0843e-03,
          6.9847e-03, 1.5554e-03, 9.0561e-03, 3.8300e-03, 3.1066e-04, 1.1396e-03,
          6.8283e-04, 9.6893e-04, 3.9744e-04, 1.4753e-03, 1.5821e-03, 4.7898e-04,
          1.6699e-03, 6.4087e-04, 3.8261e-03, 1.3857e-03, 3.0708e-04, 4.6425e-03,
          1.5182e-03, 1.0614e-03, 1.0958e-03, 8.7738e-04, 1.2503e-03, 2.4529e-03,
          3.5048e-04, 3.7384e-04, 2.5768e-03, 8.2874e-04, 7.5340e-04, 5.3740e-04,
          2.2769e-04, 4.4823e-04, 2.7633e-04, 1.2674e-03, 9.6798e-04, 1.9178e-03,
          8.5163e-04, 6.7711e-04, 6.8426e-04, 1.0395e-03, 2.9011e-03, 4.5547e-03,
          2.7504e-03, 2.2850e-03, 2.1038e-03, 8.9455e-04, 7.9632e-04, 8.3876e-04,
          7.3719e-04, 7.2441e-03, 4.2725e-04, 9.1600e-04, 2.2125e-04, 2.7609e-04,
          4.7731e-04, 1.4057e-03, 2.6169e-03, 3.2616e-03, 3.0022e-03, 7.1144e-04,
          4.8661e-04, 1.8072e-03, 5.1785e-04, 4.2381e-03, 2.0218e-03, 1.6861e-03,
          1.0162e-02, 7.3509e-03, 5.5885e-04, 5.4932e-03, 6.3133e-04, 6.1703e-04,
          3.9196e-04, 2.4199e-04, 9.8419e-04, 1.8997e-03, 1.4629e-03, 2.2793e-03,
          7.2250e-03, 1.0166e-03, 2.4147e-03, 3.0708e-03, 1.8473e-03, 5.7411e-04,
          1.4124e-03, 2.8286e-03, 1.1024e-02, 7.5531e-04, 1.4448e-03, 3.1376e-03,
          1.1320e-03, 2.1534e-03, 3.6645e-04, 3.3836e-03, 8.4448e-04, 4.8780e-04,
          7.3767e-04, 4.2305e-03, 5.8556e-04, 2.9969e-04, 2.4490e-03, 9.9754e-04,
          9.0837e-04, 1.3933e-03, 1.6069e-03, 1.8606e-03, 8.7786e-04, 1.5478e-03,
          1.8749e-03, 2.3708e-03, 4.8089e-04, 1.2894e-03, 7.1411e-03, 1.0128e-03,
          2.9125e-03, 6.8235e-04, 6.9761e-04, 1.9913e-03, 2.0199e-03, 8.5115e-04,
          9.9087e-04, 4.8757e-04, 9.8515e-04, 2.2697e-03, 2.0027e-03, 2.0084e-03,
          8.8120e-03, 2.1362e-03, 1.2217e-03, 2.1381e-03, 9.9564e-04, 6.9885e-03,
          9.7580e-03, 5.7640e-03, 1.2779e-03, 1.7815e-03, 4.3869e-03, 1.0176e-03,
          2.9125e-03, 5.9166e-03, 1.8473e-03, 1.6642e-03, 1.9188e-03, 3.2253e-03,
          6.0043e-03, 3.7861e-03, 1.6499e-03, 6.4468e-04, 1.0986e-03, 2.9945e-03,
          9.5606e-04, 6.7854e-04, 1.2894e-03, 3.8624e-03, 6.8741e-03, 1.7195e-03,
          2.2240e-03, 1.3056e-03, 4.0817e-03, 1.1034e-03, 7.4053e-04, 6.1226e-04,
          4.0894e-03, 6.3515e-04, 1.0595e-03, 1.5440e-03, 1.4172e-03, 1.8520e-03,
          9.2697e-04, 8.4496e-04, 1.8349e-03, 2.1305e-03, 6.6261e-03, 2.1954e-03,
          1.1625e-03, 8.1444e-04, 4.7836e-03, 6.1455e-03, 5.5313e-03, 3.2501e-03,
          2.8019e-03, 1.0216e-02, 1.7319e-03, 6.7616e-04, 1.0529e-02, 9.7132e-04,
          7.1049e-04, 1.9817e-03, 1.5087e-03, 3.2253e-03, 2.3346e-03, 1.3151e-03,
          2.7561e-03, 1.2913e-03, 3.7823e-03, 4.9877e-04, 2.5864e-03, 8.7976e-04,
          4.8866e-03, 1.8473e-03, 8.5526e-03, 3.4714e-03, 5.9471e-03, 2.5806e-03,
          4.4823e-03, 1.0433e-03, 1.2760e-03, 1.0624e-03, 2.1000e-03, 6.0654e-04,
          2.0065e-03, 2.3746e-03, 8.0919e-04, 8.1587e-04, 2.6207e-03, 2.7390e-03,
          1.2474e-03, 1.0700e-03, 1.4029e-03, 1.6680e-03, 1.9121e-03, 1.5306e-03,
          5.4016e-03, 4.4022e-03, 8.9693e-04, 1.1253e-03, 1.2188e-03, 5.4321e-03,
          2.0847e-03, 9.3699e-04, 1.5764e-03, 2.4586e-03, 1.8339e-03, 4.8752e-03,
          2.5711e-03, 2.3785e-03, 1.1578e-03, 9.9564e-04, 5.3120e-04, 1.8845e-03,
          1.3037e-03, 1.2856e-03, 6.3562e-04, 2.2621e-03, 3.5095e-03, 3.2997e-03,
          2.8267e-03, 4.9896e-03, 5.2147e-03, 3.0956e-03, 7.6218e-03, 3.5057e-03,
          2.3613e-03, 1.1091e-03, 1.0290e-03, 1.7738e-03, 1.4687e-03, 2.4052e-03,
          2.6665e-03, 8.3780e-04, 1.0509e-03, 7.1669e-04, 1.9264e-03, 1.5097e-03,
          1.4057e-03, 5.8289e-03, 1.7700e-03, 1.9083e-03, 2.0027e-03, 3.0365e-03,
          4.6883e-03, 2.4872e-03, 2.0714e-03, 3.3379e-03, 1.1307e-02, 2.2850e-03,
          1.4887e-03, 1.2369e-03, 1.5268e-03, 2.0370e-03, 1.5821e-03, 4.2877e-03,
          1.9741e-03, 8.1348e-04, 3.4313e-03, 6.4325e-04, 1.0185e-03, 1.1969e-03,
          8.6737e-04, 1.0033e-03, 3.0651e-03, 2.2659e-03, 5.9509e-03, 6.0768e-03,
          1.3208e-03, 1.8829e-02, 2.8629e-03, 1.6937e-03, 2.0523e-03, 2.4948e-03,
          1.4009e-03, 1.2617e-03, 2.5234e-03, 2.3632e-03, 3.1033e-03, 3.1319e-03,
          1.5442e-02, 7.8297e-04, 2.5692e-03, 1.7441e-02, 1.6870e-03, 1.8177e-03,
          7.7248e-04, 4.1389e-03, 2.2964e-03, 4.6730e-03, 3.5858e-02, 1.4925e-04,
          4.4525e-05, 4.4525e-05, 1.4925e-04, 4.4525e-05, 2.4676e-04, 1.4925e-04,
          1.4925e-04, 4.4525e-05, 4.4525e-05, 4.4525e-05, 3.1066e-04, 2.0599e-04,
          2.4676e-04, 1.4925e-04, 4.4525e-05, 4.4525e-05, 1.4925e-04, 4.4525e-05,
          3.1066e-04, 3.1066e-04, 2.4676e-04, 4.4525e-05, 1.4925e-04, 4.4525e-05,
          4.4525e-05, 1.4925e-04, 4.4525e-05, 4.4525e-05, 4.4525e-05, 4.4525e-05,
          4.4525e-05, 4.4525e-05, 4.4525e-05, 4.4525e-05, 3.1066e-04, 4.4525e-05,
          1.4925e-04, 1.4925e-04, 4.4525e-05, 4.4525e-05, 4.4525e-05, 1.4925e-04,
          4.0817e-04, 4.4525e-05, 1.4925e-04, 4.4525e-05, 4.4525e-05, 1.4925e-04,
          1.4925e-04, 1.4925e-04, 4.4525e-05, 1.4925e-04, 4.4525e-05, 4.4525e-05,
          4.4525e-05, 4.4525e-05, 4.4525e-05, 4.4525e-05, 4.4525e-05, 4.4525e-05,
          4.4525e-05, 4.4525e-05, 4.4525e-05, 2.3019e-04, 4.4525e-05, 4.4525e-05,
          4.4525e-05, 4.4525e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.8409e-03, 4.5837e-02, 1.3885e-02, 2.3727e-02, 4.2084e-02,
          2.9984e-02, 1.5274e-02, 7.4387e-04, 3.6120e-04, 8.1110e-04, 1.5659e-03,
          5.6887e-04, 1.4076e-03, 4.1199e-03, 6.4468e-04, 6.8474e-04, 3.6659e-03,
          2.9392e-03, 3.5334e-04, 6.5517e-04, 4.4918e-04, 7.5483e-04, 1.9398e-03,
          1.7672e-03, 6.4659e-04, 1.2260e-02, 6.0577e-03, 9.2983e-04, 2.5730e-03,
          8.2016e-03, 1.4219e-03, 1.0471e-03, 8.2588e-04, 6.2943e-04, 5.7697e-04,
          5.7268e-04, 3.8767e-04, 8.7214e-04, 6.6280e-04, 1.2817e-03, 6.7377e-04,
          5.4264e-04, 3.7837e-04, 5.4598e-04, 2.5582e-04, 2.9888e-03, 1.0929e-03,
          1.9369e-03, 1.4029e-03, 3.9749e-03, 8.5115e-04, 9.6464e-04, 4.7088e-04,
          4.1890e-04, 6.4850e-04, 5.3167e-04, 3.2663e-04, 8.7261e-04, 5.2643e-04,
          6.7997e-04, 3.2310e-03, 8.1348e-04, 6.2561e-04, 5.0640e-04, 1.3208e-03,
          1.1415e-03, 4.1676e-04, 5.2452e-04, 2.5606e-04, 2.5082e-04, 4.5514e-04,
          5.2214e-04, 1.7509e-03, 6.3992e-04, 1.3170e-03, 4.9496e-04, 3.5667e-04,
          1.0147e-03, 3.1376e-04, 5.6505e-04, 5.1594e-04, 7.1478e-04, 3.6097e-04,
          4.3035e-04, 1.7557e-03, 8.4686e-04, 6.7186e-04, 5.6410e-04, 1.0233e-03,
          1.3666e-03, 6.1417e-04, 3.2163e-04, 8.8263e-04, 3.1662e-04, 4.9162e-04,
          1.3838e-03, 4.1556e-04, 9.8610e-04, 7.5102e-04, 3.2687e-04, 3.2854e-04,
          1.1683e-03, 3.6955e-04, 4.5347e-04, 5.0783e-04, 7.3051e-04, 3.0365e-03,
          1.3695e-03, 5.0211e-04, 9.3079e-04, 4.0197e-04, 7.6818e-04, 6.0081e-04,
          3.8433e-04, 4.7469e-04, 1.5802e-03, 1.4400e-03, 7.6866e-04, 1.8444e-03,
          9.9716e-03, 5.7840e-04, 2.4819e-04, 4.4656e-04, 9.7561e-04, 2.6035e-04,
          2.3282e-04, 5.2023e-04, 5.9795e-04, 9.4080e-04, 1.1063e-03, 1.6603e-03,
          9.5558e-04, 6.5517e-04, 3.0565e-04, 3.4881e-04, 6.5470e-04, 4.0364e-04,
          4.8409e-03, 6.5384e-03, 2.9945e-03, 2.7370e-03, 1.7929e-03, 1.0765e-02,
          3.6931e-04, 1.6308e-04, 2.9540e-04, 2.7108e-04, 9.9277e-04, 8.4496e-04,
          1.0815e-03, 8.3542e-04, 4.5371e-04, 1.2350e-03, 1.5364e-03, 8.8501e-04,
          6.4468e-04, 5.3215e-04, 4.0293e-04, 7.7820e-04, 1.3371e-03, 7.6561e-03,
          1.2379e-03, 8.8930e-04, 5.4245e-03, 1.0691e-03, 6.2943e-04, 3.5095e-04,
          8.5068e-04, 3.7146e-04, 5.5933e-04, 4.4560e-04, 4.4975e-03, 9.0122e-04,
          6.0177e-04, 6.8760e-04, 9.4604e-04, 1.1206e-03, 9.9754e-04, 1.8129e-03,
          6.2418e-04, 7.4005e-04, 1.2465e-03, 1.2865e-03, 7.7534e-04, 5.8603e-04,
          9.7084e-04, 7.6523e-03, 9.4748e-04, 5.3883e-04, 1.1597e-03, 9.6083e-04,
          1.1024e-03, 2.4414e-04, 8.0061e-04, 6.2294e-03, 7.5865e-04, 8.4972e-04,
          9.9564e-04, 3.3245e-03, 1.6050e-03, 9.6655e-04, 8.0633e-04, 6.6280e-04,
          1.1320e-03, 9.9945e-04, 6.9189e-04, 5.3167e-04, 1.5726e-03, 1.8902e-03,
          8.8787e-04, 1.1358e-03, 7.1907e-04, 6.5374e-04, 1.2102e-03, 6.5994e-04,
          6.0320e-04, 4.1313e-03, 1.4439e-03, 4.3726e-04, 1.4744e-03, 1.0353e-02,
          3.2139e-03, 7.8249e-04, 1.0061e-03, 8.5258e-04, 2.4300e-03, 1.2732e-03,
          1.1444e-03, 7.2622e-04, 5.3930e-04, 7.8917e-04, 9.0313e-04, 6.4125e-03,
          1.4362e-03, 4.9686e-04, 5.9175e-04, 4.4799e-04, 8.9788e-04, 1.8177e-03,
          1.2388e-03, 1.2064e-03, 4.9019e-04, 1.0834e-03, 1.3901e-02, 1.5884e-02,
          1.2560e-03, 1.5306e-03, 7.5340e-04, 7.0667e-04, 1.1673e-03, 1.3981e-03,
          8.3351e-04, 6.4468e-04, 1.2436e-03, 8.8596e-04, 2.5120e-03, 3.0937e-03,
          7.7581e-04, 7.7915e-04, 4.4370e-04, 5.3596e-04, 3.9124e-04, 1.3847e-03,
          7.9679e-04, 6.2609e-04, 3.5210e-03, 2.9968e-02, 2.8976e-02, 1.3628e-03,
          2.1782e-03, 2.5024e-03, 2.0218e-03, 1.2045e-03, 1.7681e-03, 1.8625e-03,
          1.3933e-03, 6.1178e-04, 6.1989e-03, 2.5574e-02, 1.0338e-03, 1.3542e-03,
          1.1902e-03, 6.6757e-04, 5.4789e-04, 6.9332e-04, 2.3212e-03, 7.9060e-04,
          1.0262e-03, 2.1839e-03, 3.5686e-03, 4.1618e-03, 1.4610e-03, 1.8959e-03,
          2.1114e-03, 1.2598e-03, 9.0313e-04, 2.0256e-03, 1.3151e-03, 2.1362e-03,
          1.5802e-03, 2.1820e-03, 1.5879e-03, 9.0790e-03, 1.1501e-03, 1.2341e-03,
          1.3447e-03, 3.1261e-03, 1.2293e-03, 2.7065e-03, 5.5885e-04, 6.6376e-04,
          3.7937e-03, 8.8930e-04, 1.2032e-02, 2.8419e-03, 5.8403e-03, 2.6035e-03,
          1.0233e-03, 3.0212e-03, 8.1482e-03, 6.3324e-03, 2.1992e-03, 2.2488e-03,
          4.4289e-03, 1.1086e-02, 1.0368e-02, 1.5631e-03, 1.6928e-03, 4.4495e-02,
          2.2945e-03, 3.0117e-03, 3.9856e-02, 4.8561e-03, 9.5081e-04, 4.3488e-03,
          7.6008e-04, 2.7695e-03, 1.7796e-03, 2.7863e-02, 2.3880e-03, 5.3619e-02,
          4.4227e-05, 1.4055e-04, 4.4227e-05, 1.4055e-04, 4.4227e-05, 4.4227e-05,
          4.4227e-05, 4.4227e-05, 4.4227e-05, 4.4227e-05, 2.3913e-04, 1.4055e-04,
          4.4227e-05, 4.4227e-05, 1.4055e-04, 1.4055e-04, 4.4227e-05, 1.4055e-04,
          1.0347e-03, 1.0347e-03, 4.4227e-05, 4.4227e-05, 1.4281e-04, 1.4055e-04,
          1.4055e-04, 4.4227e-05, 4.4227e-05, 4.4227e-05, 4.4227e-05, 4.4227e-05,
          4.4227e-05, 4.4227e-05, 4.4227e-05, 4.4227e-05, 1.4055e-04, 4.4227e-05,
          8.3542e-04, 4.4227e-05, 2.3913e-04, 4.4227e-05, 4.4227e-05, 4.4227e-05,
          9.3174e-04, 4.4227e-05, 4.4227e-05, 4.4227e-05, 1.4055e-04, 2.3913e-04,
          4.4227e-05, 2.9888e-03, 4.4227e-05, 4.4227e-05, 4.4227e-05, 4.4227e-05,
          2.4724e-04, 2.4724e-04, 2.4724e-04, 3.1209e-04, 2.4724e-04, 1.4281e-04,
          4.4227e-05, 2.3913e-04, 2.4724e-04, 3.5286e-04, 1.4281e-04, 1.4281e-04,
          2.4724e-04, 2.4724e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.4937e-03, 6.4758e-02, 1.1902e-02, 1.7639e-02, 3.3936e-02,
          4.6234e-02, 1.3107e-02, 1.2550e-03, 6.7854e-04, 5.2757e-03, 1.2802e-02,
          3.1242e-03, 1.2159e-03, 4.2272e-04, 1.4505e-03, 1.2341e-03, 1.2093e-03,
          6.4659e-04, 9.8419e-04, 1.3292e-04, 2.0313e-04, 2.0313e-04, 1.5564e-03,
          1.4839e-02, 2.0008e-03, 9.7656e-03, 4.1771e-03, 2.4843e-04, 1.1044e-03,
          6.1750e-04, 2.2144e-03, 2.9993e-04, 1.4582e-03, 3.2997e-03, 7.7820e-04,
          9.0790e-04, 3.9101e-04, 1.6794e-03, 1.3552e-03, 1.8263e-04, 1.4544e-03,
          6.1798e-04, 4.9591e-04, 4.9639e-04, 4.9353e-04, 1.5249e-03, 1.7347e-03,
          2.8038e-04, 3.1686e-04, 2.8553e-03, 7.6866e-04, 1.1969e-03, 1.0195e-03,
          2.5988e-04, 6.1369e-04, 2.4676e-04, 4.0627e-04, 9.2888e-04, 7.3433e-04,
          6.3658e-04, 3.9315e-04, 5.9032e-04, 6.1846e-04, 1.2007e-03, 2.4166e-03,
          1.4124e-03, 1.3561e-03, 1.2007e-03, 2.8515e-04, 2.7943e-04, 4.2892e-04,
          4.3273e-04, 1.1658e-02, 3.2711e-04, 1.5574e-03, 2.9850e-04, 2.5845e-04,
          4.9114e-04, 3.5858e-04, 8.2064e-04, 1.4830e-03, 1.1044e-03, 4.7159e-04,
          3.7169e-04, 2.2240e-03, 3.8552e-04, 6.5765e-03, 9.2077e-04, 7.4863e-04,
          1.0811e-02, 3.6278e-03, 3.4428e-04, 6.1150e-03, 3.0112e-04, 3.9649e-04,
          3.0565e-04, 2.0897e-04, 1.2169e-03, 1.1568e-03, 3.8338e-04, 7.2861e-04,
          9.0332e-03, 1.1587e-03, 1.0920e-03, 6.4993e-04, 1.5249e-03, 3.1185e-04,
          1.7805e-03, 4.8184e-04, 4.2496e-03, 4.8780e-04, 1.0462e-03, 1.5001e-03,
          1.7576e-03, 4.2176e-04, 2.9612e-04, 2.3804e-03, 5.3644e-04, 4.2439e-04,
          5.4312e-04, 1.5917e-03, 2.7466e-04, 8.9407e-04, 3.1815e-03, 3.2520e-04,
          2.5249e-04, 4.5824e-04, 6.9904e-04, 1.4687e-03, 2.4490e-03, 2.2106e-03,
          1.4315e-03, 1.3676e-03, 2.1801e-03, 6.9809e-04, 2.8954e-03, 3.3522e-04,
          3.9330e-03, 5.7602e-04, 3.8815e-03, 6.7673e-03, 4.2610e-03, 7.7009e-04,
          4.1461e-04, 2.1534e-03, 5.9986e-04, 7.0047e-04, 9.7418e-04, 1.4086e-03,
          3.6736e-03, 1.2016e-03, 3.8695e-04, 7.3004e-04, 8.6546e-04, 8.4686e-04,
          2.3880e-03, 9.5558e-04, 3.9635e-03, 5.2834e-04, 1.3895e-03, 9.0361e-04,
          2.9774e-03, 3.8433e-03, 2.9926e-03, 5.7602e-03, 6.4039e-04, 1.1301e-03,
          7.5340e-03, 1.5688e-03, 1.1024e-03, 1.6689e-03, 1.4610e-03, 1.6584e-03,
          9.1505e-04, 6.5660e-04, 6.2752e-04, 6.3658e-04, 1.6642e-03, 4.2381e-03,
          1.4620e-03, 1.3447e-03, 2.9354e-03, 3.6392e-03, 2.9526e-03, 2.8019e-03,
          9.4376e-03, 1.1930e-03, 1.1147e-02, 5.2357e-04, 8.9836e-04, 2.5043e-03,
          3.3417e-03, 2.6298e-04, 1.7948e-03, 3.2654e-03, 7.2060e-03, 1.5965e-03,
          7.3957e-04, 4.3440e-04, 2.0466e-03, 1.2760e-03, 4.0550e-03, 1.8940e-03,
          8.4734e-04, 2.1935e-03, 6.1035e-04, 3.1643e-03, 1.1154e-02, 1.3304e-03,
          2.7294e-03, 8.1968e-04, 5.3501e-04, 1.7662e-03, 1.9970e-03, 7.7295e-04,
          1.0529e-03, 1.1406e-03, 1.0254e-02, 2.5120e-03, 2.5749e-03, 9.9087e-04,
          2.5330e-03, 4.4179e-04, 1.4143e-03, 2.4605e-03, 5.2834e-03, 3.3875e-03,
          1.6174e-03, 8.6260e-04, 7.6056e-04, 1.8196e-03, 1.7633e-03, 9.5081e-04,
          9.1648e-04, 1.1311e-03, 7.3290e-04, 1.8511e-03, 1.0424e-03, 2.4681e-03,
          1.0080e-03, 7.6532e-04, 5.4646e-04, 8.0633e-04, 1.4534e-03, 7.4625e-04,
          5.7936e-04, 2.0351e-03, 4.7970e-04, 9.2697e-04, 7.9298e-04, 1.9341e-03,
          6.7043e-04, 3.9625e-04, 8.5211e-04, 9.8133e-04, 4.3030e-03, 6.9847e-03,
          6.8808e-04, 1.8482e-03, 8.2350e-04, 9.0122e-04, 3.5477e-04, 1.2436e-03,
          1.1187e-03, 4.8790e-03, 3.1052e-03, 2.9507e-03, 5.9280e-03, 1.3084e-03,
          1.6012e-03, 1.9369e-03, 3.1910e-03, 8.7166e-04, 2.4719e-03, 2.0809e-03,
          1.6708e-03, 3.4962e-03, 7.0810e-04, 1.8454e-03, 6.7949e-04, 1.2302e-03,
          1.9321e-03, 4.9067e-04, 3.7169e-04, 4.5800e-04, 1.2493e-03, 1.6251e-03,
          2.4261e-03, 6.6071e-03, 3.0785e-03, 5.0507e-03, 8.4448e-04, 1.8015e-03,
          1.2913e-03, 8.3017e-04, 6.0606e-04, 3.3855e-03, 2.8629e-03, 1.3485e-03,
          8.1587e-04, 4.6196e-03, 6.0120e-03, 2.2678e-03, 6.7568e-04, 3.7994e-03,
          2.2678e-03, 5.4789e-04, 1.7776e-03, 1.7395e-03, 7.5417e-03, 6.9351e-03,
          1.5717e-03, 1.6870e-03, 1.3552e-03, 4.3526e-03, 5.7106e-03, 3.0785e-03,
          3.6373e-03, 2.3575e-02, 2.2221e-03, 2.0275e-03, 9.0170e-04, 9.1457e-04,
          3.8624e-03, 1.9951e-03, 4.2763e-03, 8.3876e-04, 1.8463e-03, 3.3226e-03,
          9.7504e-03, 1.4973e-03, 1.9913e-03, 1.1765e-02, 2.8412e-02, 3.7327e-03,
          1.5526e-03, 7.8201e-03, 6.3019e-03, 1.9302e-03, 6.8207e-03, 2.6913e-03,
          5.3619e-02, 6.1989e-05, 2.2960e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05,
          6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05, 3.2520e-04, 3.2520e-04,
          6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05, 2.2960e-04, 2.2960e-04,
          4.9305e-04, 4.9305e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05,
          6.1989e-05, 2.2960e-04, 6.1989e-05, 2.2960e-04, 6.1989e-05, 6.1989e-05,
          6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05, 3.2520e-04, 6.1989e-05,
          2.2960e-04, 2.2960e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05,
          6.1989e-05, 6.1989e-05, 2.2960e-04, 6.1989e-05, 6.1989e-05, 2.2960e-04,
          2.2960e-04, 2.2960e-04, 6.1989e-05, 6.1989e-05, 6.1989e-05, 6.1989e-05,
          2.2960e-04, 2.2960e-04, 6.1989e-05, 6.1989e-05, 2.2960e-04, 6.1989e-05,
          6.1989e-05, 2.2960e-04, 2.2960e-04, 2.2960e-04, 2.2960e-04, 2.2960e-04,
          6.1989e-05, 6.1989e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9177e-03, 5.8258e-02, 1.2009e-02, 1.4282e-02, 3.0197e-02,
          4.5166e-02, 1.0574e-02, 1.3428e-03, 5.0640e-04, 2.6035e-03, 4.4518e-03,
          1.3657e-03, 1.6861e-03, 3.9530e-04, 3.9253e-03, 1.4019e-03, 1.6956e-03,
          9.8038e-04, 6.6185e-04, 1.3816e-04, 1.7691e-04, 1.9979e-04, 1.4343e-03,
          5.3825e-03, 2.4567e-03, 8.3694e-03, 3.4294e-03, 2.4235e-04, 1.7834e-03,
          5.7936e-04, 2.6417e-03, 3.9315e-04, 2.9850e-03, 3.9635e-03, 5.1165e-04,
          1.7462e-03, 3.1376e-04, 2.1954e-03, 2.9221e-03, 1.9133e-04, 1.1835e-03,
          4.1270e-04, 3.2759e-04, 3.6359e-04, 2.7204e-04, 3.1242e-03, 3.2177e-03,
          2.5392e-04, 2.7800e-04, 2.5902e-03, 6.4516e-04, 1.1702e-03, 7.0333e-04,
          2.3794e-04, 6.0558e-04, 2.4962e-04, 3.2020e-04, 6.2418e-04, 5.1880e-04,
          5.4121e-04, 4.8089e-04, 4.9114e-04, 6.9284e-04, 4.8280e-04, 9.1553e-03,
          6.3372e-04, 5.8317e-04, 1.2226e-03, 1.8382e-04, 1.8668e-04, 2.7132e-04,
          3.0565e-04, 6.1378e-03, 2.8825e-04, 1.6775e-03, 2.3735e-04, 2.2101e-04,
          5.4884e-04, 2.5940e-04, 3.5977e-04, 2.4071e-03, 4.7827e-04, 4.0293e-04,
          3.2806e-04, 2.3861e-03, 3.3855e-04, 2.6169e-03, 3.8576e-04, 3.3426e-04,
          4.7112e-03, 2.0771e-03, 2.6464e-04, 2.3613e-03, 3.0112e-04, 3.4022e-04,
          3.1304e-04, 1.9598e-04, 1.1368e-03, 8.3733e-04, 2.9063e-04, 4.5180e-04,
          3.6106e-03, 4.7374e-04, 5.0640e-04, 1.6737e-03, 8.3923e-03, 4.3035e-04,
          1.6890e-03, 3.4714e-04, 2.5749e-03, 2.7537e-04, 4.9877e-04, 6.4135e-04,
          2.3766e-03, 5.4598e-04, 3.1376e-04, 7.1297e-03, 5.2309e-04, 4.3964e-04,
          6.3467e-04, 1.8549e-03, 2.0194e-04, 1.0490e-03, 2.0401e-02, 1.6212e-04,
          2.0921e-04, 2.3842e-04, 3.5596e-04, 4.7417e-03, 2.2106e-03, 2.3155e-03,
          8.3542e-04, 6.0606e-04, 3.3989e-03, 4.1795e-04, 1.5965e-03, 2.8181e-04,
          4.7684e-03, 6.2180e-04, 1.8444e-03, 3.4313e-03, 3.4008e-03, 8.8215e-04,
          2.8992e-04, 1.4687e-03, 5.6171e-04, 3.4761e-04, 4.5586e-04, 3.6449e-03,
          4.0474e-03, 3.8071e-03, 4.5180e-04, 8.6403e-04, 7.7677e-04, 1.1225e-03,
          1.7214e-03, 7.6532e-04, 1.8187e-03, 3.4642e-04, 2.7771e-03, 9.8705e-04,
          4.4441e-03, 3.2158e-03, 5.2032e-03, 8.0872e-03, 5.5027e-04, 5.1117e-04,
          2.3518e-03, 7.2718e-04, 2.7142e-03, 4.6349e-03, 1.4515e-03, 2.9030e-03,
          1.0948e-03, 5.9509e-04, 6.7616e-04, 6.5470e-04, 3.4676e-03, 7.3776e-03,
          3.0670e-03, 4.1313e-03, 1.3901e-02, 3.9749e-03, 2.1896e-03, 3.4046e-03,
          5.1308e-03, 8.4305e-04, 3.9825e-03, 5.1117e-04, 4.4775e-04, 1.1810e-02,
          3.6926e-03, 2.2244e-04, 3.7880e-03, 4.7951e-03, 3.1719e-03, 3.5915e-03,
          6.9141e-04, 5.3024e-04, 5.2261e-03, 8.7547e-04, 1.7481e-03, 3.7136e-03,
          5.6982e-04, 1.7309e-03, 6.0558e-04, 2.2392e-03, 5.0354e-03, 1.4124e-03,
          1.6174e-03, 8.6403e-04, 4.6158e-04, 9.4652e-04, 4.7455e-03, 4.0555e-04,
          1.3752e-03, 1.3027e-03, 8.5449e-03, 4.5242e-03, 4.3907e-03, 1.0252e-03,
          2.7752e-03, 4.6229e-04, 1.6527e-03, 1.1654e-03, 3.0243e-02, 1.0719e-02,
          9.2745e-04, 4.7970e-04, 5.2309e-04, 1.4038e-03, 3.2806e-03, 7.7534e-04,
          7.9823e-04, 5.9748e-04, 4.1461e-04, 2.4281e-03, 1.5640e-03, 2.1877e-03,
          7.2145e-04, 6.4993e-04, 4.5443e-04, 1.0757e-03, 2.0199e-03, 9.4748e-04,
          7.4720e-04, 1.5583e-03, 4.4870e-04, 7.1621e-04, 8.9693e-04, 7.1869e-03,
          4.2582e-04, 4.6206e-04, 9.2793e-04, 1.1845e-03, 6.0310e-03, 4.8065e-03,
          7.8487e-04, 7.8535e-04, 1.2426e-03, 1.5116e-03, 3.7575e-04, 8.8310e-04,
          2.0580e-03, 4.9744e-03, 1.7824e-03, 3.7193e-03, 4.5967e-03, 3.5648e-03,
          1.0586e-03, 1.1377e-03, 1.3199e-02, 1.3037e-03, 3.3092e-03, 5.1270e-03,
          1.8215e-03, 2.4548e-03, 6.2847e-04, 1.7290e-03, 6.4468e-04, 7.2241e-04,
          9.8991e-04, 4.5347e-04, 3.3283e-04, 5.1260e-04, 1.5059e-03, 1.8454e-03,
          1.7767e-03, 1.0681e-02, 4.3068e-03, 4.7989e-03, 9.0361e-04, 1.4572e-03,
          1.3685e-03, 7.7295e-04, 6.5756e-04, 8.1863e-03, 3.1624e-03, 1.3828e-03,
          9.8228e-04, 3.2520e-03, 4.0474e-03, 2.3365e-03, 5.6362e-04, 4.1885e-03,
          4.3983e-03, 5.7650e-04, 2.1820e-03, 1.0710e-03, 7.4120e-03, 4.4365e-03,
          1.4582e-03, 1.5011e-03, 1.5469e-03, 4.9629e-03, 7.6637e-03, 9.1476e-03,
          2.3155e-03, 1.3321e-02, 2.5120e-03, 2.2621e-03, 1.1683e-03, 1.2779e-03,
          2.8076e-03, 1.5354e-03, 3.4180e-03, 8.8644e-04, 2.5864e-03, 3.4809e-03,
          6.1188e-03, 1.2035e-03, 2.0905e-03, 1.0017e-02, 5.8823e-03, 1.9026e-03,
          1.2016e-03, 6.6223e-03, 3.5076e-03, 2.0885e-03, 5.2223e-03, 2.8191e-03,
          5.3291e-03, 3.8055e-02, 3.2759e-04, 5.5850e-05, 4.3392e-04, 2.1625e-04,
          9.8896e-04, 2.6817e-03, 5.5850e-05, 5.5850e-05, 3.2759e-04, 3.2759e-04,
          5.4502e-04, 5.5850e-05, 5.5850e-05, 5.5850e-05, 3.2759e-04, 4.3392e-04,
          3.2759e-04, 3.2759e-04, 4.3392e-04, 5.5850e-05, 2.1625e-04, 5.5850e-05,
          5.5850e-05, 3.2759e-04, 5.5850e-05, 4.3392e-04, 5.5850e-05, 5.5850e-05,
          5.5850e-05, 5.5850e-05, 5.5850e-05, 5.5850e-05, 3.2759e-04, 5.5850e-05,
          3.2759e-04, 3.2759e-04, 2.1625e-04, 5.5850e-05, 5.5850e-05, 2.1625e-04,
          7.0572e-04, 2.1625e-04, 3.2759e-04, 5.5850e-05, 5.5850e-05, 5.5850e-05,
          2.1625e-04, 2.1625e-04, 5.5850e-05, 2.1625e-04, 5.5850e-05, 5.5850e-05,
          2.1625e-04, 5.5850e-05, 5.5850e-05, 1.2827e-04, 2.1625e-04, 5.5850e-05,
          5.5850e-05, 5.5850e-05, 5.5850e-05, 3.9053e-04, 5.5850e-05, 5.5850e-05,
          5.5850e-05, 5.5850e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3482e-03, 4.9591e-02, 1.6815e-02, 1.7410e-02, 3.6560e-02,
          2.9755e-02, 1.1841e-02, 8.6308e-04, 5.2786e-04, 9.3222e-04, 1.8768e-03,
          6.4230e-04, 3.7556e-03, 3.3054e-03, 8.6117e-04, 9.3269e-04, 1.8204e-02,
          6.8855e-03, 4.5061e-04, 1.3485e-03, 7.8201e-04, 2.3117e-03, 3.2177e-03,
          2.3479e-03, 8.2684e-04, 8.9951e-03, 3.7956e-03, 2.0390e-03, 1.3885e-02,
          4.3755e-03, 2.0008e-03, 6.1188e-03, 1.5888e-03, 9.1648e-04, 7.5865e-04,
          7.1192e-04, 3.3045e-04, 1.1997e-03, 1.0271e-03, 1.7843e-03, 9.1839e-04,
          4.9353e-04, 3.3593e-04, 4.7684e-04, 3.3569e-04, 1.1871e-02, 2.2736e-03,
          2.4261e-03, 1.7385e-03, 3.4904e-03, 1.0700e-03, 1.4639e-03, 6.0463e-04,
          2.7156e-04, 8.0824e-04, 4.1556e-04, 3.3188e-04, 6.0797e-04, 4.9305e-04,
          5.4073e-04, 3.8719e-03, 5.7983e-04, 7.9870e-04, 4.9353e-04, 2.2888e-03,
          6.2990e-04, 3.8457e-04, 5.6744e-04, 2.6560e-04, 2.9492e-04, 3.9816e-04,
          5.4502e-04, 2.3460e-03, 4.4298e-04, 2.3232e-03, 2.9635e-04, 2.6345e-04,
          8.5115e-04, 2.9731e-04, 3.4690e-04, 5.2071e-04, 3.8886e-04, 3.5310e-04,
          3.9768e-04, 2.5120e-03, 4.6825e-04, 9.5415e-04, 3.2353e-04, 3.2067e-04,
          2.1343e-03, 1.2312e-03, 2.6059e-04, 1.2283e-03, 3.9577e-04, 5.9509e-04,
          1.7605e-03, 2.5511e-04, 1.4963e-03, 1.5221e-03, 2.7704e-04, 3.5501e-04,
          1.4658e-03, 2.6631e-04, 3.7479e-04, 6.0844e-04, 1.1663e-03, 2.6760e-03,
          1.8950e-03, 4.3821e-04, 1.3351e-03, 2.4414e-04, 4.3082e-04, 4.8494e-04,
          4.9734e-04, 7.5912e-04, 1.5631e-03, 2.7027e-03, 8.5211e-04, 1.3494e-03,
          2.4452e-03, 8.3923e-04, 1.6761e-04, 5.2452e-04, 1.4572e-03, 1.4043e-04,
          2.2388e-04, 2.5487e-04, 3.3617e-04, 1.2474e-03, 1.6909e-03, 2.3270e-03,
          7.2718e-04, 4.5037e-04, 3.6049e-04, 2.5582e-04, 8.7166e-04, 3.6025e-04,
          1.4381e-02, 3.8052e-03, 2.8839e-03, 2.8839e-03, 2.1286e-03, 3.9101e-03,
          2.3437e-04, 2.0087e-04, 2.4581e-04, 2.9588e-04, 4.2439e-04, 9.4604e-04,
          1.5240e-03, 9.4986e-04, 4.4346e-04, 1.6880e-03, 1.2074e-03, 1.1368e-03,
          8.4782e-04, 6.3324e-04, 4.9448e-04, 3.8862e-04, 2.3880e-03, 7.5607e-03,
          2.5749e-03, 1.6069e-03, 1.8875e-02, 1.7977e-03, 6.2275e-04, 3.6526e-04,
          9.4938e-04, 4.2701e-04, 6.3562e-04, 6.0987e-04, 6.0921e-03, 1.5850e-03,
          6.2656e-04, 4.7684e-04, 9.1171e-04, 1.0891e-03, 1.6012e-03, 2.9945e-03,
          7.0286e-04, 1.0185e-03, 1.8578e-03, 1.7929e-03, 1.6336e-03, 1.0986e-03,
          1.4114e-03, 4.8676e-03, 8.8882e-04, 4.1795e-04, 3.9482e-04, 1.2550e-03,
          1.7595e-03, 2.0230e-04, 1.2569e-03, 2.0416e-02, 9.9087e-04, 1.1597e-03,
          8.4639e-04, 2.3842e-03, 2.4033e-03, 7.7629e-04, 8.0252e-04, 9.5606e-04,
          4.9925e-04, 1.3666e-03, 7.8917e-04, 6.7902e-04, 2.1515e-03, 9.2545e-03,
          1.1053e-03, 1.1206e-03, 5.3883e-04, 5.5742e-04, 1.5583e-03, 3.3545e-04,
          8.5783e-04, 7.8125e-03, 1.4744e-03, 6.1941e-04, 3.1395e-03, 5.0125e-03,
          5.9547e-03, 6.6566e-04, 1.1921e-03, 6.9714e-04, 3.6201e-03, 1.6222e-03,
          8.4162e-04, 4.7660e-04, 5.1451e-04, 9.3937e-04, 1.4782e-03, 4.4899e-03,
          1.0328e-03, 4.4632e-04, 3.8266e-04, 6.5899e-04, 1.0262e-03, 1.9875e-03,
          8.5115e-04, 8.3685e-04, 4.5800e-04, 1.9436e-03, 9.1934e-03, 4.6921e-03,
          1.4315e-03, 1.3351e-03, 5.1785e-04, 6.2037e-04, 9.2602e-04, 1.9894e-03,
          4.9877e-04, 5.9843e-04, 1.1492e-03, 1.2999e-03, 5.6992e-03, 3.7441e-03,
          7.8726e-04, 6.2323e-04, 5.1546e-04, 6.3753e-04, 3.6597e-04, 9.4318e-04,
          9.8038e-04, 6.8903e-04, 2.0237e-03, 1.5930e-02, 1.0468e-02, 1.5955e-03,
          1.2941e-03, 1.0586e-03, 2.8191e-03, 1.1730e-03, 3.2215e-03, 2.4147e-03,
          1.6804e-03, 9.8515e-04, 4.2229e-03, 7.0267e-03, 8.4352e-04, 7.4482e-04,
          8.6927e-04, 4.9734e-04, 4.2081e-04, 5.8699e-04, 3.4733e-03, 1.4343e-03,
          1.2293e-03, 3.1300e-03, 1.7822e-02, 1.0704e-02, 1.1168e-03, 1.7767e-03,
          1.7796e-03, 8.7690e-04, 7.4339e-04, 3.7231e-03, 2.1458e-03, 1.7080e-03,
          1.3485e-03, 2.9602e-03, 2.2659e-03, 1.2726e-02, 7.7868e-04, 1.7061e-03,
          1.9178e-03, 2.1305e-03, 1.6956e-03, 2.2163e-03, 8.0395e-04, 8.7214e-04,
          5.4016e-03, 9.9277e-04, 6.9199e-03, 2.8439e-03, 9.3307e-03, 3.4370e-03,
          1.1520e-03, 4.4441e-03, 1.4740e-02, 1.3725e-02, 1.6994e-03, 2.0599e-03,
          9.4910e-03, 9.6054e-03, 9.5978e-03, 1.2817e-03, 2.0943e-03, 1.3283e-02,
          2.9793e-03, 5.9357e-03, 7.7248e-03, 5.6648e-03, 1.1234e-03, 4.8485e-03,
          9.0170e-04, 3.6640e-03, 3.2463e-03, 1.5625e-02, 3.2978e-03, 1.0674e-02,
          1.3161e-03, 1.8568e-03, 3.7659e-02, 1.3030e-04, 2.0432e-04, 4.7565e-05,
          1.3030e-04, 1.3030e-04, 1.3030e-04, 4.7565e-05, 1.3030e-04, 1.3030e-04,
          7.2813e-04, 4.7565e-05, 2.6107e-04, 1.3030e-04, 1.3030e-04, 6.1131e-04,
          6.5422e-04, 7.8487e-04, 1.3030e-04, 1.3030e-04, 1.3030e-04, 4.7565e-05,
          2.6107e-04, 1.3030e-04, 4.7565e-05, 4.1771e-04, 4.7565e-05, 4.7565e-05,
          1.3030e-04, 1.3030e-04, 1.3030e-04, 4.7565e-05, 6.5422e-04, 4.7565e-05,
          6.5422e-04, 2.6107e-04, 4.5443e-04, 4.7565e-05, 4.7565e-05, 1.3030e-04,
          1.3030e-04, 4.7565e-05, 2.6107e-04, 1.3030e-04, 2.6107e-04, 2.8706e-04,
          1.3030e-04, 8.4782e-04, 4.7565e-05, 4.7565e-05, 4.7565e-05, 4.7565e-05,
          3.2377e-04, 3.2377e-04, 1.3030e-04, 1.6892e-04, 1.3030e-04, 4.7565e-05,
          4.7565e-05, 4.5443e-04, 3.2377e-04, 4.0698e-04, 1.3030e-04, 1.3030e-04,
          1.3030e-04, 1.3030e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.5128e-03, 6.1249e-02, 1.3504e-02, 1.6205e-02, 3.2501e-02,
          4.9561e-02, 1.2276e-02, 1.7242e-03, 8.6164e-04, 5.4665e-03, 1.2001e-02,
          4.2305e-03, 1.7662e-03, 5.6553e-04, 1.5020e-03, 1.6918e-03, 1.5450e-03,
          1.0405e-03, 1.1797e-03, 1.4091e-04, 2.0730e-04, 2.5010e-04, 2.3708e-03,
          1.3054e-02, 1.7366e-03, 9.1705e-03, 5.0125e-03, 2.7728e-04, 1.5230e-03,
          8.2493e-04, 2.7809e-03, 4.1938e-04, 1.4343e-03, 3.0537e-03, 8.1015e-04,
          8.2445e-04, 3.7074e-04, 1.2207e-03, 1.2798e-03, 2.0397e-04, 1.1063e-03,
          5.5981e-04, 4.3249e-04, 4.9114e-04, 5.2643e-04, 2.2984e-03, 1.4973e-03,
          3.4976e-04, 3.4308e-04, 3.0384e-03, 8.6212e-04, 1.6708e-03, 1.0290e-03,
          2.8157e-04, 7.4387e-04, 2.7657e-04, 5.6791e-04, 1.4668e-03, 9.4604e-04,
          6.7568e-04, 4.8375e-04, 5.6219e-04, 5.4598e-04, 1.1883e-03, 2.1000e-03,
          1.0252e-03, 1.4191e-03, 1.0624e-03, 3.2854e-04, 2.8443e-04, 4.5681e-04,
          5.0831e-04, 1.0643e-02, 4.2868e-04, 2.0542e-03, 3.2806e-04, 2.8539e-04,
          6.7282e-04, 4.2391e-04, 8.2302e-04, 1.5688e-03, 1.0719e-03, 4.4394e-04,
          3.4404e-04, 2.4357e-03, 3.9411e-04, 6.1569e-03, 7.5769e-04, 5.6553e-04,
          1.0078e-02, 3.1509e-03, 3.2163e-04, 5.6572e-03, 2.6822e-04, 3.9697e-04,
          3.7599e-04, 2.3437e-04, 1.4629e-03, 1.0843e-03, 3.6883e-04, 6.9618e-04,
          9.0179e-03, 1.0185e-03, 1.3189e-03, 6.2990e-04, 1.5173e-03, 3.5501e-04,
          1.9913e-03, 3.9482e-04, 3.4218e-03, 3.3617e-04, 7.5436e-04, 1.1377e-03,
          1.3237e-03, 3.9124e-04, 3.0184e-04, 2.3308e-03, 5.3024e-04, 5.0116e-04,
          7.1430e-04, 1.3733e-03, 2.9230e-04, 1.0700e-03, 2.8954e-03, 2.9016e-04,
          2.6298e-04, 3.9196e-04, 6.0558e-04, 1.6031e-03, 2.7046e-03, 2.7561e-03,
          1.2217e-03, 1.1311e-03, 1.6651e-03, 5.7316e-04, 2.5101e-03, 3.3665e-04,
          4.0512e-03, 6.6900e-04, 3.8338e-03, 6.0043e-03, 4.1771e-03, 1.0767e-03,
          3.8600e-04, 1.5993e-03, 4.9162e-04, 6.4039e-04, 7.4482e-04, 1.3371e-03,
          3.7460e-03, 1.3475e-03, 4.3869e-04, 7.7295e-04, 9.7227e-04, 6.5660e-04,
          2.0275e-03, 7.5483e-04, 3.8357e-03, 4.4894e-04, 1.2102e-03, 1.0662e-03,
          2.5444e-03, 2.6703e-03, 4.2305e-03, 4.6272e-03, 5.7554e-04, 7.4577e-04,
          7.2289e-03, 1.1387e-03, 9.8419e-04, 1.6422e-03, 1.8454e-03, 1.3866e-03,
          1.0500e-03, 6.7520e-04, 5.5933e-04, 5.4932e-04, 1.3170e-03, 5.5428e-03,
          1.1415e-03, 1.2894e-03, 2.4490e-03, 4.8409e-03, 2.9621e-03, 2.5406e-03,
          1.0033e-02, 1.3342e-03, 9.5749e-03, 4.8876e-04, 7.8344e-04, 2.3212e-03,
          4.2915e-03, 2.4915e-04, 1.7843e-03, 4.0092e-03, 7.3547e-03, 1.6403e-03,
          7.1096e-04, 4.5776e-04, 1.5841e-03, 1.2846e-03, 3.3207e-03, 1.3466e-03,
          7.2813e-04, 1.7862e-03, 5.4455e-04, 3.0804e-03, 1.0361e-02, 1.3819e-03,
          2.7409e-03, 8.0061e-04, 5.1451e-04, 2.5635e-03, 2.0752e-03, 6.7377e-04,
          8.8453e-04, 1.2960e-03, 1.0620e-02, 2.3155e-03, 2.4796e-03, 1.4524e-03,
          1.9312e-03, 3.9339e-04, 1.1158e-03, 1.9159e-03, 4.4060e-03, 2.9583e-03,
          1.4658e-03, 6.7759e-04, 6.1703e-04, 1.5240e-03, 1.4057e-03, 1.1044e-03,
          9.6607e-04, 8.9931e-04, 6.6280e-04, 1.4009e-03, 9.5749e-04, 2.6302e-03,
          1.0548e-03, 7.4911e-04, 4.8923e-04, 7.4625e-04, 1.8120e-03, 8.3256e-04,
          5.0068e-04, 1.9684e-03, 4.5371e-04, 8.4114e-04, 6.4468e-04, 1.5574e-03,
          5.1928e-04, 3.5930e-04, 6.8235e-04, 6.9857e-04, 4.8294e-03, 5.7564e-03,
          6.5041e-04, 1.5478e-03, 7.1621e-04, 8.1539e-04, 3.6168e-04, 1.4181e-03,
          1.0853e-03, 3.8280e-03, 3.2063e-03, 4.2000e-03, 7.6141e-03, 1.0900e-03,
          1.3781e-03, 1.7996e-03, 2.5921e-03, 7.0381e-04, 1.6413e-03, 1.8082e-03,
          1.4181e-03, 2.5578e-03, 6.8665e-04, 2.0618e-03, 6.7568e-04, 1.1368e-03,
          2.0161e-03, 5.1594e-04, 3.6430e-04, 4.4346e-04, 1.3266e-03, 1.3762e-03,
          2.6302e-03, 5.7068e-03, 3.7537e-03, 5.6114e-03, 7.1812e-04, 1.6470e-03,
          1.0080e-03, 7.1096e-04, 5.0211e-04, 3.0136e-03, 2.1305e-03, 1.4458e-03,
          7.3385e-04, 4.8180e-03, 6.8130e-03, 2.7332e-03, 6.5470e-04, 3.5286e-03,
          2.1725e-03, 5.6314e-04, 1.4286e-03, 1.8120e-03, 4.5204e-03, 6.2294e-03,
          1.8263e-03, 1.4229e-03, 1.2703e-03, 4.3564e-03, 5.4131e-03, 2.4376e-03,
          2.9411e-03, 2.6169e-02, 2.2907e-03, 2.4147e-03, 7.1955e-04, 6.9761e-04,
          3.9940e-03, 2.4109e-03, 5.7259e-03, 8.3542e-04, 1.7471e-03, 4.1580e-03,
          6.9771e-03, 1.6747e-03, 2.3918e-03, 9.4910e-03, 2.6627e-02, 3.0231e-03,
          1.6279e-03, 6.1150e-03, 6.1569e-03, 1.7529e-03, 5.2719e-03, 2.9926e-03,
          2.6947e-02, 5.7487e-03, 2.8534e-03, 4.4037e-02, 2.7514e-04, 7.3791e-05,
          7.3791e-05, 4.2439e-04, 7.3791e-05, 7.3791e-05, 2.7514e-04, 2.7514e-04,
          2.7514e-04, 7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05, 2.2316e-04,
          4.2439e-04, 4.2439e-04, 7.3791e-05, 7.3791e-05, 2.2316e-04, 2.2316e-04,
          7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05,
          7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05, 2.7514e-04, 7.3791e-05,
          2.2316e-04, 7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05, 7.3791e-05,
          2.7514e-04, 2.2316e-04, 7.3791e-05, 7.3791e-05, 7.3791e-05, 2.2316e-04,
          2.2316e-04, 2.2316e-04, 7.3791e-05, 2.2316e-04, 7.3791e-05, 7.3791e-05,
          2.2316e-04, 2.2316e-04, 7.3791e-05, 7.3791e-05, 2.2316e-04, 7.3791e-05,
          7.3791e-05, 2.2316e-04, 7.3791e-05, 5.3167e-04, 2.2316e-04, 2.2316e-04,
          7.3791e-05, 7.3791e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.0384e-03, 4.7119e-02, 1.2627e-02, 1.1795e-02, 2.4933e-02,
          3.2440e-02, 7.4539e-03, 5.1994e-03, 7.1096e-04, 1.1845e-03, 1.7929e-03,
          7.4291e-04, 9.0218e-04, 6.4898e-04, 1.3285e-03, 4.6425e-03, 1.8530e-03,
          9.8515e-04, 1.0033e-03, 3.6120e-04, 3.7503e-04, 5.0831e-04, 1.4839e-03,
          2.2278e-03, 1.0490e-03, 6.3896e-03, 3.5515e-03, 4.0507e-04, 1.5297e-03,
          7.2289e-04, 1.3306e-02, 5.4741e-04, 1.7881e-03, 1.0595e-03, 7.4463e-03,
          1.2274e-03, 4.8709e-04, 1.7815e-03, 1.8826e-03, 4.8208e-04, 1.1892e-03,
          5.6267e-04, 4.5514e-04, 5.7268e-04, 3.8624e-04, 1.5974e-03, 2.1801e-03,
          5.6839e-04, 6.0320e-04, 2.0447e-03, 1.4046e-02, 1.3794e-02, 3.1109e-03,
          1.4315e-03, 5.7793e-03, 2.7351e-03, 4.4847e-04, 2.8362e-03, 8.3447e-04,
          1.1759e-03, 1.0767e-03, 5.3978e-03, 6.4430e-03, 6.0034e-04, 2.5921e-03,
          7.2193e-04, 4.8590e-04, 6.9332e-04, 3.5810e-04, 3.2091e-04, 6.8712e-04,
          7.3242e-04, 2.7504e-03, 4.4022e-03, 1.7593e-02, 1.7719e-03, 2.0542e-03,
          7.5455e-03, 3.4451e-04, 3.9530e-04, 9.9564e-04, 5.0545e-04, 7.1287e-04,
          5.9032e-04, 2.3026e-02, 3.5152e-03, 1.0033e-03, 3.7432e-04, 3.8743e-04,
          2.1439e-03, 1.5450e-03, 3.2711e-04, 1.5535e-03, 6.1560e-04, 1.0023e-03,
          9.1410e-04, 1.8482e-03, 1.3031e-02, 1.7805e-03, 3.2091e-04, 4.3368e-04,
          1.4801e-03, 4.3130e-04, 4.1962e-04, 1.3189e-03, 2.1648e-03, 7.5817e-04,
          1.5778e-02, 5.7840e-04, 1.7738e-03, 4.6706e-04, 5.1737e-04, 6.4898e-04,
          9.5558e-04, 9.3079e-04, 6.4993e-04, 3.2043e-03, 1.6317e-03, 6.2132e-04,
          8.7166e-04, 1.0538e-03, 2.0444e-04, 2.7962e-03, 1.7633e-03, 1.9419e-04,
          2.5249e-04, 3.4142e-04, 4.5776e-04, 2.8839e-03, 4.6463e-03, 1.7670e-02,
          9.4891e-04, 4.6444e-04, 5.2166e-04, 3.7980e-04, 1.0500e-03, 4.3392e-04,
          4.2152e-03, 9.5463e-04, 2.7523e-03, 5.0240e-03, 2.7981e-03, 1.0977e-03,
          2.7847e-04, 4.6802e-04, 3.5977e-04, 4.1366e-04, 4.1914e-04, 1.8215e-03,
          2.4014e-03, 2.4128e-03, 9.2506e-04, 1.3268e-02, 1.5945e-02, 1.4582e-03,
          1.3256e-03, 1.1549e-03, 7.1669e-04, 4.6539e-04, 2.8915e-03, 1.9321e-03,
          2.7370e-03, 2.7866e-03, 2.6855e-03, 2.0809e-03, 6.0558e-04, 5.6410e-04,
          9.8705e-04, 7.3814e-04, 9.5415e-04, 7.9012e-04, 1.3771e-03, 2.0962e-03,
          3.4485e-03, 2.3613e-03, 1.6050e-03, 1.2674e-03, 1.9503e-03, 2.2392e-03,
          1.0195e-03, 1.5421e-03, 2.1706e-03, 1.3260e-02, 3.0861e-03, 1.7805e-03,
          1.7691e-03, 1.6613e-03, 1.4820e-03, 4.3297e-04, 4.1389e-04, 1.6737e-03,
          1.0048e-02, 2.1911e-04, 1.4305e-03, 2.3975e-03, 1.2455e-03, 1.8721e-03,
          2.4338e-03, 1.0128e-03, 3.0994e-03, 6.9046e-04, 1.1597e-03, 1.2493e-03,
          4.8351e-04, 1.1215e-03, 7.5865e-04, 2.2869e-03, 2.6169e-03, 1.2627e-03,
          6.2943e-03, 9.6655e-04, 4.4584e-04, 5.6267e-04, 1.8291e-03, 3.6764e-04,
          1.2522e-03, 1.7872e-03, 2.4338e-03, 1.0777e-03, 2.9602e-03, 2.0809e-03,
          4.1199e-03, 6.5708e-04, 1.1892e-03, 9.3317e-04, 3.8090e-03, 2.2373e-03,
          6.9094e-04, 7.8917e-04, 7.2908e-04, 2.1515e-03, 2.0409e-03, 1.8253e-03,
          7.3147e-04, 6.2609e-04, 4.3774e-04, 1.3580e-03, 1.2693e-03, 2.2717e-03,
          9.8896e-04, 8.5163e-04, 7.3004e-04, 2.1381e-03, 3.0670e-03, 1.1702e-03,
          1.1539e-03, 1.1377e-03, 6.5708e-04, 1.0300e-03, 1.0109e-03, 2.0580e-03,
          5.5075e-04, 8.7547e-04, 2.0237e-03, 1.7033e-03, 3.2883e-03, 1.4687e-02,
          7.3195e-04, 6.9380e-04, 1.0567e-03, 1.1024e-03, 4.8041e-04, 1.3304e-03,
          1.4782e-03, 1.6146e-03, 1.9798e-03, 4.5586e-03, 3.3150e-03, 1.9665e-03,
          1.3180e-03, 7.3385e-04, 2.8915e-03, 9.4271e-04, 2.9888e-03, 2.9278e-03,
          2.3823e-03, 2.8267e-03, 1.3247e-03, 2.0256e-03, 6.2752e-04, 6.5947e-04,
          6.2656e-04, 4.2701e-04, 3.4046e-04, 5.5981e-04, 2.0123e-03, 1.8911e-03,
          1.8349e-03, 3.6526e-03, 2.4014e-03, 2.9125e-03, 1.1520e-03, 1.9875e-03,
          1.3256e-03, 7.3051e-04, 6.4039e-04, 3.5648e-03, 3.5191e-03, 2.7657e-03,
          1.3838e-03, 3.2310e-03, 1.1879e-02, 3.0708e-03, 5.0354e-04, 2.5768e-03,
          2.0390e-03, 8.0013e-04, 2.4147e-03, 1.8072e-03, 2.1286e-03, 2.0618e-03,
          1.9331e-03, 2.6875e-03, 2.1057e-03, 4.6883e-03, 5.1842e-03, 3.6526e-03,
          2.5902e-03, 4.5891e-03, 3.6125e-03, 2.5063e-03, 1.7300e-03, 1.4477e-03,
          3.6278e-03, 2.6016e-03, 2.7924e-03, 6.7377e-04, 2.0847e-03, 2.4662e-03,
          4.0016e-03, 1.4248e-03, 1.7805e-03, 5.6038e-03, 2.4071e-03, 3.9253e-03,
          3.0174e-03, 2.0767e-02, 1.7242e-02, 2.3766e-03, 3.5076e-03, 2.9736e-03,
          2.7103e-03, 3.0193e-03, 3.1738e-03, 2.6321e-03, 4.9316e-02, 4.9829e-05,
          1.7655e-04, 1.7655e-04, 4.9829e-05, 4.9829e-05, 1.7655e-04, 1.7655e-04,
          4.9829e-05, 4.9829e-05, 4.9829e-05, 4.9829e-05, 1.7655e-04, 4.9829e-05,
          1.7655e-04, 1.7655e-04, 4.9829e-05, 4.9829e-05, 1.7655e-04, 4.9829e-05,
          4.9829e-05, 1.7655e-04, 4.9829e-05, 4.9829e-05, 4.9829e-05, 4.9829e-05,
          1.7655e-04, 4.9829e-05, 4.9829e-05, 4.9829e-05, 1.7655e-04, 4.9829e-05,
          1.7655e-04, 1.7655e-04, 1.7655e-04, 4.9829e-05, 4.9829e-05, 1.7655e-04,
          1.7655e-04, 4.9829e-05, 1.7655e-04, 4.9829e-05, 4.9829e-05, 4.9829e-05,
          1.7655e-04, 1.7655e-04, 4.9829e-05, 4.9829e-05, 4.9829e-05, 4.9829e-05,
          7.2145e-04, 1.7655e-04, 1.7655e-04, 1.1778e-04, 1.7655e-04, 4.9829e-05,
          4.9829e-05, 1.7655e-04, 1.7655e-04, 1.0214e-03, 8.2684e-04, 4.2391e-04,
          1.7655e-04, 4.9829e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.3596e-03, 4.9042e-02, 1.7776e-02, 1.8616e-02, 3.5828e-02,
          2.6520e-02, 1.2657e-02, 5.3740e-04, 3.0804e-04, 7.9679e-04, 1.2655e-03,
          4.2558e-04, 2.0924e-03, 1.9121e-03, 8.3303e-04, 6.2227e-04, 8.0032e-03,
          3.4370e-03, 3.5405e-04, 1.1330e-03, 1.1206e-03, 1.6403e-03, 3.7098e-03,
          1.5860e-03, 8.5592e-04, 1.0040e-02, 5.2834e-03, 1.7157e-03, 6.1150e-03,
          3.8910e-03, 1.1177e-03, 2.0924e-03, 1.5898e-03, 6.5231e-04, 9.1076e-04,
          8.1205e-04, 3.6764e-04, 1.7319e-03, 1.0147e-03, 1.7509e-03, 1.3399e-03,
          6.5470e-04, 4.7755e-04, 5.5933e-04, 4.7708e-04, 7.6141e-03, 2.8782e-03,
          1.7691e-03, 2.2316e-03, 4.3602e-03, 1.3876e-03, 8.5688e-04, 6.9475e-04,
          2.8110e-04, 4.6682e-04, 4.6134e-04, 3.7622e-04, 4.6325e-04, 4.4918e-04,
          4.9877e-04, 4.4098e-03, 7.1383e-04, 1.2236e-03, 5.3978e-04, 2.5215e-03,
          5.9938e-04, 3.5191e-04, 6.2656e-04, 3.8457e-04, 3.9172e-04, 4.9639e-04,
          7.7820e-04, 2.5101e-03, 5.6982e-04, 1.0195e-03, 2.8849e-04, 2.6488e-04,
          6.6090e-04, 3.9673e-04, 3.0255e-04, 4.5466e-04, 3.1543e-04, 2.9707e-04,
          3.5143e-04, 1.8415e-03, 5.9843e-04, 6.6710e-04, 2.9278e-04, 3.3736e-04,
          1.2617e-03, 1.1263e-03, 2.8777e-04, 9.3412e-04, 4.9162e-04, 8.7976e-04,
          2.2144e-03, 3.6669e-04, 1.8644e-03, 2.5196e-03, 3.4261e-04, 3.7479e-04,
          8.8596e-04, 2.9111e-04, 2.4581e-04, 6.1941e-04, 1.1902e-03, 4.6654e-03,
          1.3800e-03, 6.2513e-04, 1.4353e-03, 3.4142e-04, 4.2105e-04, 5.2834e-04,
          5.5122e-04, 1.0328e-03, 3.1166e-03, 2.9430e-03, 1.3323e-03, 2.2831e-03,
          4.9133e-03, 9.2363e-04, 1.8704e-04, 3.3450e-04, 1.3781e-03, 1.3947e-04,
          2.8300e-04, 2.3639e-04, 3.0780e-04, 1.2646e-03, 1.4572e-03, 1.5221e-03,
          7.4387e-04, 4.1580e-04, 2.8467e-04, 2.7966e-04, 7.6342e-04, 4.3201e-04,
          1.6235e-02, 5.0316e-03, 3.3512e-03, 3.8643e-03, 3.3321e-03, 5.3940e-03,
          2.4772e-04, 1.8442e-04, 2.5725e-04, 3.0184e-04, 3.3736e-04, 8.8596e-04,
          1.1444e-03, 8.7261e-04, 4.7040e-04, 2.3136e-03, 1.3161e-03, 1.7548e-03,
          1.0176e-03, 1.0748e-03, 3.1948e-04, 3.9101e-04, 2.6455e-03, 8.9111e-03,
          3.8891e-03, 2.8038e-03, 1.2817e-02, 2.2640e-03, 1.0843e-03, 4.5919e-04,
          6.9857e-04, 5.5647e-04, 5.7983e-04, 5.5408e-04, 4.7989e-03, 2.1877e-03,
          5.4264e-04, 4.2939e-04, 9.6989e-04, 1.8587e-03, 1.7519e-03, 1.9312e-03,
          9.8419e-04, 8.8644e-04, 2.2297e-03, 1.2474e-03, 1.3800e-03, 1.3962e-03,
          9.0504e-04, 6.0654e-03, 1.5688e-03, 4.8566e-04, 3.2496e-04, 1.1063e-03,
          9.0599e-04, 2.3043e-04, 1.5116e-03, 1.4282e-02, 7.8487e-04, 1.2865e-03,
          8.7023e-04, 3.0022e-03, 2.0657e-03, 7.3051e-04, 9.3079e-04, 1.2302e-03,
          4.9877e-04, 1.4191e-03, 9.8133e-04, 5.0211e-04, 1.8444e-03, 5.1651e-03,
          9.1505e-04, 2.0809e-03, 6.3324e-04, 3.8481e-04, 1.2894e-03, 2.7990e-04,
          1.0519e-03, 7.3242e-03, 1.3523e-03, 5.1451e-04, 3.6812e-03, 4.3259e-03,
          7.0839e-03, 7.5960e-04, 1.3371e-03, 7.6389e-04, 3.5286e-03, 1.5059e-03,
          7.2241e-04, 4.6086e-04, 5.6314e-04, 8.4925e-04, 1.5612e-03, 5.8441e-03,
          1.2541e-03, 4.7183e-04, 3.0041e-04, 4.9496e-04, 9.6416e-04, 1.3523e-03,
          7.3576e-04, 8.4400e-04, 6.4707e-04, 2.9678e-03, 9.0485e-03, 5.8556e-03,
          2.2602e-03, 9.3889e-04, 4.9782e-04, 6.2037e-04, 8.5115e-04, 1.9112e-03,
          5.3215e-04, 7.6914e-04, 1.6451e-03, 1.9140e-03, 5.8365e-03, 5.8899e-03,
          9.5940e-04, 6.0225e-04, 4.3726e-04, 5.2786e-04, 3.1257e-04, 8.3923e-04,
          8.2064e-04, 7.6294e-04, 2.1954e-03, 1.5083e-02, 1.1864e-02, 1.6451e-03,
          1.1005e-03, 7.9679e-04, 2.5902e-03, 1.2465e-03, 4.4403e-03, 2.3193e-03,
          1.6413e-03, 8.2970e-04, 8.2855e-03, 8.5754e-03, 1.2932e-03, 5.7507e-04,
          5.7077e-04, 4.6182e-04, 4.4656e-04, 5.5361e-04, 3.8471e-03, 1.6260e-03,
          7.7677e-04, 4.1313e-03, 8.2779e-03, 8.1787e-03, 1.1587e-03, 1.4763e-03,
          2.0790e-03, 8.9645e-04, 8.2445e-04, 4.1237e-03, 2.7771e-03, 2.1839e-03,
          1.5574e-03, 3.3169e-03, 1.8244e-03, 1.3390e-02, 7.2384e-04, 1.4133e-03,
          1.7395e-03, 2.4548e-03, 2.2411e-03, 3.1013e-03, 6.6614e-04, 6.0749e-04,
          5.6763e-03, 1.1368e-03, 9.1782e-03, 2.2964e-03, 1.2398e-02, 3.2043e-03,
          1.0376e-03, 3.3951e-03, 1.7593e-02, 1.1116e-02, 2.4147e-03, 3.1242e-03,
          8.0414e-03, 8.5678e-03, 9.4070e-03, 1.3599e-03, 1.9369e-03, 1.3451e-02,
          3.7632e-03, 2.9526e-03, 6.8703e-03, 5.7068e-03, 1.1787e-03, 7.5073e-03,
          6.7186e-04, 4.8523e-03, 3.2806e-03, 1.7944e-02, 3.6526e-03, 1.0193e-02,
          1.3847e-03, 1.7548e-03, 1.9211e-02, 1.1711e-03, 4.7607e-03, 3.5797e-02,
          4.5419e-05, 2.5749e-04, 4.5419e-05, 4.5419e-05, 4.3750e-04, 2.2531e-04,
          4.5419e-05, 4.5419e-05, 2.2531e-04, 4.5419e-05, 4.5419e-05, 2.5749e-04,
          4.3750e-04, 4.3750e-04, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05,
          2.2531e-04, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05,
          4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05, 2.2531e-04, 4.5419e-05,
          2.5749e-04, 4.5419e-05, 4.3750e-04, 4.5419e-05, 4.5419e-05, 4.5419e-05,
          4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05,
          4.5419e-05, 2.5749e-04, 4.5419e-05, 4.5419e-05, 4.5419e-05, 4.5419e-05,
          2.5749e-04, 2.5749e-04, 2.5749e-04, 1.1814e-04, 2.5749e-04, 4.5419e-05,
          4.5419e-05, 2.5749e-04, 2.5749e-04, 8.5306e-04, 4.5419e-05, 4.5419e-05,
          2.5749e-04, 4.5419e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.2338e-03, 4.6143e-02, 1.8143e-02, 2.0569e-02, 5.4535e-02,
          2.7267e-02, 1.7822e-02, 5.8794e-04, 2.6608e-04, 4.9686e-04, 1.0481e-03,
          3.5191e-04, 1.0567e-03, 3.7003e-03, 4.8876e-04, 5.4264e-04, 3.0212e-03,
          1.9245e-03, 3.0613e-04, 1.3371e-03, 9.1887e-04, 1.7462e-03, 1.2999e-03,
          1.2751e-03, 6.7472e-04, 1.3474e-02, 2.9144e-03, 1.9741e-03, 1.6546e-03,
          6.3133e-03, 9.2554e-04, 9.7227e-04, 6.2704e-04, 3.7766e-04, 4.6229e-04,
          5.4264e-04, 4.1628e-04, 7.6723e-04, 5.0879e-04, 2.4338e-03, 7.1812e-04,
          7.4387e-04, 5.4741e-04, 7.2956e-04, 3.7742e-04, 1.8864e-03, 1.0929e-03,
          2.4242e-03, 3.0899e-03, 7.0267e-03, 7.7820e-04, 5.8937e-04, 3.5644e-04,
          4.6802e-04, 4.5133e-04, 6.7854e-04, 3.1209e-04, 7.0000e-04, 4.6730e-04,
          6.5136e-04, 3.4885e-03, 1.0490e-03, 7.6389e-04, 5.2786e-04, 1.2312e-03,
          1.0920e-03, 3.3593e-04, 4.7851e-04, 5.1689e-04, 5.2452e-04, 7.5483e-04,
          1.0452e-03, 1.0796e-03, 8.0872e-04, 8.3065e-04, 4.6301e-04, 3.1590e-04,
          8.9216e-04, 3.2854e-04, 4.5276e-04, 3.1877e-04, 4.5729e-04, 3.6120e-04,
          4.8423e-04, 1.4019e-03, 1.3943e-03, 4.7112e-04, 4.2105e-04, 9.2936e-04,
          8.6069e-04, 5.1069e-04, 4.7135e-04, 6.8808e-04, 6.0034e-04, 1.1826e-03,
          2.1992e-03, 5.5552e-04, 6.6805e-04, 7.8440e-04, 3.5477e-04, 2.8038e-04,
          6.8903e-04, 2.9564e-04, 2.8348e-04, 4.5681e-04, 6.3658e-04, 3.4695e-03,
          1.1806e-03, 6.9761e-04, 7.2432e-04, 4.9686e-04, 7.2670e-04, 5.5408e-04,
          4.5013e-04, 8.0204e-04, 3.6831e-03, 1.4143e-03, 1.8044e-03, 3.1109e-03,
          8.8425e-03, 4.5013e-04, 2.5392e-04, 3.3069e-04, 7.8869e-04, 2.2340e-04,
          2.7299e-04, 3.9625e-04, 4.5443e-04, 6.7425e-04, 7.1955e-04, 1.2836e-03,
          8.5449e-04, 5.7316e-04, 2.6274e-04, 3.2687e-04, 5.0354e-04, 6.4087e-04,
          3.0518e-03, 9.1324e-03, 3.0231e-03, 2.5120e-03, 1.2560e-03, 7.4425e-03,
          3.5167e-04, 1.9145e-04, 2.6226e-04, 2.3901e-04, 6.5279e-04, 6.5947e-04,
          7.0238e-04, 6.1560e-04, 4.8280e-04, 1.3905e-03, 1.7214e-03, 1.1187e-03,
          5.2261e-04, 6.8617e-04, 3.1996e-04, 7.9393e-04, 1.5087e-03, 7.8506e-03,
          1.2093e-03, 9.7656e-04, 3.2768e-03, 9.3651e-04, 7.2241e-04, 3.1114e-04,
          5.5695e-04, 3.0565e-04, 4.1533e-04, 3.7932e-04, 2.2793e-03, 8.4639e-04,
          4.8685e-04, 6.3848e-04, 1.0872e-03, 1.5545e-03, 8.6164e-04, 1.1292e-03,
          5.7697e-04, 5.9271e-04, 1.0443e-03, 9.4366e-04, 6.7902e-04, 7.4959e-04,
          7.5293e-04, 9.9411e-03, 1.1740e-03, 5.7030e-04, 7.6962e-04, 6.9189e-04,
          7.0715e-04, 2.8706e-04, 6.0272e-04, 3.7346e-03, 4.7922e-04, 7.1526e-04,
          1.1082e-03, 5.5199e-03, 1.3485e-03, 7.1764e-04, 6.4421e-04, 6.5899e-04,
          9.9659e-04, 8.2731e-04, 1.1787e-03, 4.6706e-04, 1.2283e-03, 1.6594e-03,
          8.7357e-04, 1.3847e-03, 9.6083e-04, 4.2558e-04, 7.9966e-04, 5.1880e-04,
          4.8971e-04, 3.3264e-03, 1.0462e-03, 4.3273e-04, 1.0757e-03, 8.4076e-03,
          3.0918e-03, 1.0405e-03, 9.2411e-04, 6.8712e-04, 2.0046e-03, 9.0075e-04,
          8.8644e-04, 8.9455e-04, 9.1696e-04, 1.0014e-03, 9.8610e-04, 8.8806e-03,
          1.9484e-03, 4.0817e-04, 5.3120e-04, 4.1580e-04, 8.1968e-04, 1.4915e-03,
          1.2007e-03, 1.4744e-03, 6.1512e-04, 1.1816e-03, 1.1391e-02, 1.6754e-02,
          1.7776e-03, 1.0252e-03, 9.1028e-04, 8.1348e-04, 1.5030e-03, 1.1797e-03,
          1.0204e-03, 1.3475e-03, 2.5520e-03, 1.3838e-03, 1.9855e-03, 2.2945e-03,
          8.4543e-04, 5.5933e-04, 4.1890e-04, 4.4680e-04, 4.2248e-04, 1.2989e-03,
          6.0225e-04, 5.1022e-04, 3.2482e-03, 1.7014e-02, 1.5747e-02, 1.1702e-03,
          1.7729e-03, 1.4668e-03, 1.7509e-03, 1.3933e-03, 1.7338e-03, 1.8568e-03,
          1.8387e-03, 6.7616e-04, 2.1332e-02, 1.8509e-02, 1.4715e-03, 1.0452e-03,
          7.9775e-04, 8.0204e-04, 7.4291e-04, 8.9693e-04, 1.9741e-03, 6.5136e-04,
          8.5497e-04, 1.4706e-03, 2.2182e-03, 3.1681e-03, 1.7385e-03, 1.8816e-03,
          2.4681e-03, 1.4257e-03, 1.1902e-03, 1.5984e-03, 1.2674e-03, 3.5515e-03,
          2.9793e-03, 2.2869e-03, 1.4534e-03, 7.4005e-03, 1.5745e-03, 9.0265e-04,
          9.9564e-04, 5.3253e-03, 9.9373e-04, 1.9379e-03, 5.3883e-04, 5.4932e-04,
          2.7733e-03, 7.8344e-04, 1.6190e-02, 1.7548e-03, 4.6768e-03, 1.9970e-03,
          9.4080e-04, 2.1305e-03, 8.0872e-03, 5.8937e-03, 4.7226e-03, 4.7035e-03,
          4.4174e-03, 9.2697e-03, 7.1411e-03, 2.1877e-03, 1.5078e-03, 2.3834e-02,
          1.6813e-03, 2.4033e-03, 2.4155e-02, 3.4351e-03, 9.7227e-04, 4.6234e-03,
          6.4182e-04, 2.3975e-03, 1.4381e-03, 3.2898e-02, 1.9331e-03, 3.1860e-02,
          1.0281e-03, 1.2398e-03, 1.5381e-02, 9.3842e-04, 2.2106e-03, 1.5373e-02,
          4.4250e-02, 2.1601e-04, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 5.1379e-05, 2.1803e-04, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          2.1803e-04, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 5.1379e-05, 2.1803e-04, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          5.1379e-05, 2.9945e-03, 5.1379e-05, 5.1379e-05, 5.1379e-05, 5.1379e-05,
          4.0054e-04, 4.0054e-04, 4.0054e-04, 4.7326e-04, 4.7326e-04, 2.1601e-04,
          5.1379e-05, 4.0054e-04, 4.0054e-04, 5.4502e-04, 2.1601e-04, 2.1601e-04,
          2.1601e-04, 2.1601e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.5320e-02, 7.0679e-02, 3.2959e-02, 5.0537e-02, 1.2561e-01,
          5.0751e-02, 5.7709e-02, 6.5231e-04, 4.1652e-04, 8.7214e-04, 1.9760e-03,
          5.9080e-04, 8.9884e-04, 1.1721e-03, 4.5156e-04, 4.9877e-04, 2.1896e-03,
          1.1625e-03, 2.8610e-04, 1.0395e-03, 1.2426e-03, 1.0643e-03, 1.2693e-03,
          2.0695e-03, 6.7854e-04, 3.5645e-02, 5.8174e-03, 2.1629e-03, 1.3647e-03,
          1.2856e-03, 7.8630e-04, 8.8978e-04, 6.0081e-04, 4.0436e-04, 4.6301e-04,
          6.2037e-04, 7.6294e-04, 8.8739e-04, 4.2725e-04, 8.0109e-04, 1.0014e-03,
          1.0128e-03, 5.8508e-04, 7.7724e-04, 4.5514e-04, 1.0662e-03, 7.7868e-04,
          7.5054e-04, 2.0370e-03, 2.7908e-02, 1.0719e-03, 6.8521e-04, 3.6025e-04,
          6.8808e-04, 5.7125e-04, 9.5367e-04, 6.2418e-04, 6.2323e-04, 6.6948e-04,
          7.6199e-04, 1.0843e-03, 9.7847e-04, 5.3120e-04, 4.1962e-04, 7.9393e-04,
          7.5626e-04, 3.0637e-04, 3.7789e-04, 6.7425e-04, 6.5279e-04, 7.5388e-04,
          1.9722e-03, 1.4524e-03, 1.7242e-03, 7.9870e-04, 4.5824e-04, 3.3712e-04,
          6.5374e-04, 5.6553e-04, 5.4455e-04, 3.8242e-04, 5.4121e-04, 4.5323e-04,
          6.4659e-04, 1.0252e-03, 5.9986e-04, 4.8923e-04, 3.2067e-04, 4.2677e-04,
          1.0242e-03, 6.0177e-04, 2.7609e-04, 6.8283e-04, 4.7445e-04, 2.1019e-03,
          1.7786e-03, 9.2554e-04, 5.1641e-04, 9.2316e-04, 7.9012e-04, 4.7159e-04,
          9.2363e-04, 3.3712e-04, 3.4499e-04, 4.6206e-04, 4.7255e-04, 7.0238e-04,
          6.9618e-04, 8.1921e-04, 9.0218e-04, 3.3021e-04, 4.6992e-04, 4.0436e-04,
          3.3522e-04, 6.0129e-04, 1.0099e-03, 7.6485e-04, 2.1324e-03, 2.5272e-03,
          1.6994e-03, 5.9414e-04, 5.0497e-04, 3.1447e-04, 7.3528e-04, 2.3830e-04,
          3.7718e-04, 3.6550e-04, 3.8862e-04, 5.0783e-04, 4.5943e-04, 7.4768e-04,
          5.2834e-04, 3.8576e-04, 2.8038e-04, 2.3961e-04, 5.1498e-04, 4.2343e-04,
          1.6413e-03, 9.5034e-04, 8.6308e-04, 2.2335e-03, 1.1168e-03, 1.2426e-03,
          5.2595e-04, 2.3210e-04, 2.6202e-04, 3.2496e-04, 5.7220e-04, 5.5790e-04,
          7.3862e-04, 4.6921e-04, 3.7003e-04, 7.8630e-04, 1.0767e-03, 9.2363e-04,
          5.0163e-04, 4.7731e-04, 3.0422e-04, 4.0698e-04, 6.6471e-04, 1.0681e-03,
          7.4244e-04, 7.0429e-04, 1.6193e-03, 8.8215e-04, 1.7576e-03, 4.6444e-04,
          8.1158e-04, 4.4060e-04, 3.7980e-04, 3.3116e-04, 1.2579e-03, 6.5422e-04,
          3.5000e-04, 4.1890e-04, 1.0700e-03, 1.6479e-03, 5.7268e-04, 8.3447e-04,
          4.1103e-04, 3.7456e-04, 7.0333e-04, 5.6267e-04, 5.7983e-04, 3.6502e-04,
          6.2275e-04, 1.1530e-03, 1.1463e-03, 1.1492e-03, 5.4073e-04, 6.4230e-04,
          6.2466e-04, 3.7289e-04, 4.7851e-04, 1.7853e-03, 5.3596e-04, 5.5456e-04,
          8.3876e-04, 1.3657e-03, 6.5947e-04, 4.7779e-04, 5.3596e-04, 3.9101e-04,
          5.6458e-04, 5.7459e-04, 6.2084e-04, 4.7517e-04, 1.0271e-03, 6.6853e-04,
          6.2227e-04, 2.1553e-03, 2.4776e-03, 5.5313e-04, 7.3099e-04, 4.9257e-04,
          5.4073e-04, 8.9788e-04, 9.5749e-04, 2.9063e-04, 7.5960e-04, 8.3685e-04,
          1.6136e-03, 9.2268e-04, 5.2881e-04, 4.6349e-04, 1.0443e-03, 6.5470e-04,
          6.0225e-04, 4.5133e-04, 3.9625e-04, 5.1928e-04, 5.9175e-04, 8.0204e-04,
          5.8327e-03, 5.6028e-04, 5.0783e-04, 3.6478e-04, 6.7186e-04, 9.7942e-04,
          1.0338e-03, 1.5821e-03, 5.2786e-04, 7.7581e-04, 1.1892e-03, 1.7309e-03,
          1.2808e-03, 5.4550e-04, 6.1464e-04, 4.1032e-04, 8.5497e-04, 5.7507e-04,
          3.9053e-04, 5.0783e-04, 1.3018e-03, 6.9952e-04, 7.3576e-04, 4.3259e-03,
          1.2894e-03, 6.2990e-04, 4.1699e-04, 3.9148e-04, 4.3964e-04, 7.8154e-04,
          5.2547e-04, 4.6229e-04, 5.0259e-04, 1.5364e-03, 2.6932e-03, 4.6968e-04,
          8.1110e-04, 8.2397e-04, 7.8392e-04, 6.7139e-04, 1.0071e-03, 6.4898e-04,
          5.9366e-04, 3.5572e-04, 2.7924e-03, 1.5049e-03, 2.9068e-03, 9.3603e-04,
          7.5388e-04, 8.6737e-04, 7.9012e-04, 9.7466e-04, 8.1539e-04, 4.5657e-04,
          5.3549e-04, 1.2884e-03, 1.0672e-03, 1.3857e-03, 1.3294e-03, 9.4509e-04,
          1.1892e-03, 7.2813e-04, 6.6805e-04, 6.6280e-04, 7.5483e-04, 1.1301e-03,
          1.3227e-03, 6.2799e-04, 9.3842e-04, 1.9436e-03, 2.0790e-03, 7.5150e-04,
          7.1430e-04, 1.6603e-03, 8.4257e-04, 5.0688e-04, 4.6062e-04, 5.3263e-04,
          5.3740e-04, 5.5742e-04, 3.4294e-03, 1.0738e-03, 1.9569e-03, 7.5769e-04,
          4.7803e-04, 1.4429e-03, 1.0967e-03, 8.2636e-04, 1.8206e-03, 2.6150e-03,
          1.2102e-03, 8.8692e-04, 3.3894e-03, 3.0594e-03, 1.2245e-03, 3.2406e-03,
          2.0370e-03, 7.1192e-04, 3.0212e-03, 2.8419e-03, 1.1730e-03, 2.0370e-03,
          4.4799e-04, 1.4553e-03, 8.5735e-04, 3.9711e-03, 1.2980e-03, 2.4357e-03,
          1.0595e-03, 7.3099e-04, 2.0924e-03, 7.3433e-04, 1.0881e-03, 2.0046e-03,
          2.6188e-03, 1.6846e-01, 7.1287e-05, 7.1287e-05, 1.3437e-03, 1.1911e-03,
          2.0676e-03, 7.1287e-05, 1.0605e-03, 5.8031e-04, 3.8934e-04, 1.8444e-03,
          1.7214e-03, 1.6327e-03, 2.2202e-03, 2.0385e-04, 1.6193e-03, 5.8031e-04,
          9.5034e-04, 5.6267e-04, 2.0385e-04, 2.1324e-03, 7.1287e-05, 7.1287e-05,
          7.1287e-05, 7.1287e-05, 7.1287e-05, 2.0385e-04, 1.1911e-03, 7.1287e-05,
          1.7214e-03, 3.8934e-04, 3.5954e-03, 3.7169e-04, 2.6202e-04, 7.1287e-05,
          2.6798e-03, 2.6202e-04, 8.6880e-04, 2.0385e-04, 5.8031e-04, 1.9112e-03,
          4.4775e-04, 1.6193e-03, 7.1287e-05, 7.1287e-05, 7.1287e-05, 7.1287e-05,
          1.8620e-04, 1.8620e-04, 5.0402e-04, 3.2377e-04, 1.8620e-04, 3.1877e-04,
          7.1287e-05, 4.8876e-04, 1.8620e-04, 6.4993e-04, 1.8620e-04, 1.8620e-04,
          1.8620e-04, 1.8620e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.7722e-03, 4.8096e-02, 1.8921e-02, 1.8631e-02, 6.6345e-02,
          3.3966e-02, 1.8158e-02, 6.5136e-04, 4.5586e-04, 1.3342e-03, 4.8599e-03,
          1.0786e-03, 6.8903e-04, 5.7125e-04, 1.0872e-03, 4.4632e-04, 8.8978e-04,
          4.2415e-04, 2.8658e-04, 4.2248e-04, 4.2725e-04, 3.3188e-04, 5.5981e-04,
          4.3831e-03, 1.8044e-03, 1.2451e-02, 2.0428e-03, 1.1406e-03, 8.4543e-04,
          6.8712e-04, 9.8515e-04, 3.9411e-04, 1.0843e-03, 5.1832e-04, 4.6992e-04,
          1.4315e-03, 1.4315e-03, 7.7677e-04, 6.6042e-04, 4.7231e-04, 1.0662e-03,
          1.0281e-03, 7.6580e-04, 8.4305e-04, 5.8794e-04, 7.1859e-04, 1.2989e-03,
          3.0565e-04, 5.4979e-04, 5.8212e-03, 1.3418e-03, 8.0013e-04, 5.4646e-04,
          5.8842e-04, 5.1641e-04, 9.7132e-04, 1.8253e-03, 1.9331e-03, 2.8267e-03,
          3.1204e-03, 6.0892e-04, 8.7309e-04, 6.0368e-04, 1.0939e-03, 1.9531e-03,
          1.2331e-03, 5.6744e-04, 5.2452e-04, 5.4836e-04, 5.7077e-04, 8.7261e-04,
          9.0122e-04, 2.1610e-03, 1.4668e-03, 1.0738e-03, 4.8089e-04, 3.2759e-04,
          8.4019e-04, 1.8663e-03, 1.4849e-03, 1.2197e-03, 1.5106e-03, 1.8797e-03,
          1.5860e-03, 1.0004e-03, 6.4278e-04, 1.2140e-03, 5.6839e-04, 8.0585e-04,
          2.3956e-03, 1.0986e-03, 5.7888e-04, 1.4009e-03, 1.3981e-03, 1.3580e-03,
          8.9169e-04, 8.2493e-04, 5.5027e-04, 1.8358e-03, 3.3665e-03, 1.9407e-03,
          3.4924e-03, 7.7057e-04, 7.6866e-04, 1.8654e-03, 1.7776e-03, 5.7220e-04,
          7.8773e-04, 9.1362e-04, 1.5011e-03, 7.0858e-04, 1.1168e-03, 6.3992e-04,
          1.6098e-03, 9.1505e-04, 3.4714e-04, 2.0256e-03, 1.6375e-03, 1.7748e-03,
          1.5297e-03, 2.6188e-03, 1.9875e-03, 3.9697e-04, 1.9264e-03, 6.7711e-04,
          1.6565e-03, 1.0471e-03, 1.2350e-03, 1.5469e-03, 5.0974e-04, 8.6260e-04,
          1.1568e-03, 7.2527e-04, 3.9053e-04, 3.0398e-04, 1.0195e-03, 6.1083e-04,
          9.7322e-04, 4.3583e-04, 6.5136e-04, 1.3971e-03, 1.0471e-03, 1.1196e-03,
          2.9888e-03, 1.2226e-03, 1.3733e-03, 1.5211e-03, 1.8787e-03, 2.8095e-03,
          2.6779e-03, 1.5316e-03, 2.1725e-03, 7.4911e-04, 1.1797e-03, 1.2703e-03,
          8.6927e-04, 7.0524e-04, 4.3011e-04, 6.7186e-04, 1.6603e-03, 5.0402e-04,
          1.2846e-03, 8.2588e-04, 8.7500e-04, 1.2989e-03, 4.7493e-03, 1.4963e-03,
          2.8419e-03, 1.1511e-03, 1.5354e-03, 9.7322e-04, 7.3624e-04, 1.6003e-03,
          2.9373e-03, 2.4624e-03, 3.7346e-03, 1.4467e-03, 1.3733e-03, 9.0599e-04,
          7.9346e-04, 6.3086e-04, 1.1673e-03, 6.4087e-04, 5.8317e-04, 7.9775e-04,
          1.1644e-03, 5.7888e-04, 1.0128e-03, 6.5193e-03, 2.1057e-03, 1.2903e-03,
          8.2874e-04, 2.2507e-03, 9.7275e-04, 1.3628e-03, 1.3876e-03, 1.2159e-03,
          4.0207e-03, 1.1711e-03, 1.2884e-03, 9.3746e-04, 6.8569e-04, 7.9918e-04,
          1.0185e-03, 1.1644e-03, 1.1711e-03, 5.7840e-04, 2.2678e-03, 4.4942e-04,
          6.8045e-04, 5.0049e-03, 1.0605e-02, 1.9379e-03, 2.8057e-03, 2.6302e-03,
          1.6403e-03, 1.0834e-03, 1.0910e-03, 7.0238e-04, 1.4534e-03, 9.0027e-04,
          2.1267e-03, 1.6060e-03, 1.4219e-03, 6.9904e-04, 2.1114e-03, 9.8038e-04,
          9.7179e-04, 1.1272e-03, 1.2932e-03, 4.3831e-03, 1.3494e-03, 5.1594e-04,
          1.4168e-02, 2.5024e-03, 3.7003e-03, 2.0123e-03, 6.5079e-03, 1.1711e-02,
          6.2523e-03, 6.0387e-03, 2.8992e-03, 1.8520e-03, 1.9455e-03, 1.6689e-03,
          1.7662e-03, 1.3103e-03, 1.4153e-03, 1.7576e-03, 1.6956e-03, 1.3504e-03,
          7.3147e-04, 1.4229e-03, 2.3460e-03, 1.3313e-03, 8.4782e-04, 2.6913e-03,
          7.9498e-03, 1.9703e-03, 3.5248e-03, 2.5673e-03, 4.6616e-03, 6.0997e-03,
          2.0752e-03, 6.4802e-04, 7.4291e-04, 1.6298e-03, 1.8911e-03, 1.1301e-03,
          2.2411e-03, 1.5965e-03, 1.9112e-03, 1.4372e-03, 1.7948e-03, 2.4681e-03,
          4.8141e-03, 2.1534e-03, 9.9659e-04, 8.6117e-04, 7.9193e-03, 5.6229e-03,
          2.8820e-03, 7.1297e-03, 6.2370e-03, 7.2136e-03, 2.2411e-03, 1.5812e-03,
          5.7373e-03, 2.1267e-03, 9.6560e-04, 1.1511e-03, 2.8057e-03, 3.3665e-03,
          2.3880e-03, 1.6003e-03, 1.2302e-03, 1.3304e-03, 1.6146e-03, 4.3526e-03,
          2.6932e-03, 6.6996e-04, 1.0548e-03, 2.9507e-03, 1.9531e-02, 4.1466e-03,
          2.0657e-03, 4.4365e-03, 2.4567e-03, 6.5327e-04, 1.0405e-03, 8.5878e-04,
          6.9141e-04, 2.5005e-03, 5.5542e-03, 1.3723e-03, 2.2469e-03, 1.3752e-03,
          1.8797e-03, 3.3512e-03, 1.1463e-03, 8.4686e-04, 3.7651e-03, 3.1319e-03,
          9.3937e-04, 6.9714e-04, 1.9722e-03, 4.9225e-02, 1.4511e-02, 4.1313e-03,
          4.5586e-03, 1.3084e-03, 4.4365e-03, 4.3755e-03, 1.2064e-03, 2.7351e-03,
          2.4433e-03, 2.0924e-03, 1.2321e-03, 7.0076e-03, 2.2869e-03, 1.9569e-03,
          1.4143e-03, 1.1673e-03, 1.3008e-03, 7.9298e-04, 1.4124e-03, 2.3270e-03,
          1.9083e-03, 1.1345e-02, 5.3467e-02, 6.5565e-05, 4.5753e-04, 2.5439e-04,
          6.5565e-05, 6.5565e-05, 6.5565e-05, 9.6083e-04, 6.5565e-05, 6.5565e-05,
          4.5753e-04, 4.5753e-04, 6.5565e-05, 6.5565e-05, 2.6870e-04, 6.5565e-05,
          9.6083e-04, 2.6870e-04, 6.5565e-05, 6.5565e-05, 6.5565e-05, 6.5565e-05,
          6.5565e-05, 6.5565e-05, 6.5565e-05, 6.5565e-05, 4.5753e-04, 6.5565e-05,
          4.5753e-04, 2.5439e-04, 2.5439e-04, 6.5565e-05, 6.5565e-05, 9.6083e-04,
          4.5753e-04, 6.5565e-05, 2.5439e-04, 6.5565e-05, 6.5565e-05, 9.6083e-04,
          6.5565e-05, 3.1033e-03, 6.5565e-05, 6.5565e-05, 6.5565e-05, 6.5565e-05,
          2.6870e-04, 2.6870e-04, 6.5565e-05, 1.4198e-04, 6.5565e-05, 6.5565e-05,
          6.5565e-05, 6.5565e-05, 6.5565e-05, 2.6870e-04, 6.5565e-05, 6.5565e-05,
          6.5565e-05, 6.5565e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.0430e-03, 5.0018e-02, 1.8661e-02, 1.7380e-02, 6.3477e-02,
          3.4424e-02, 1.6754e-02, 6.4898e-04, 4.6754e-04, 1.4887e-03, 5.2071e-03,
          1.1797e-03, 7.2479e-04, 5.0974e-04, 1.0509e-03, 4.5395e-04, 9.3126e-04,
          4.4084e-04, 2.7370e-04, 4.0102e-04, 3.9673e-04, 3.2210e-04, 4.9162e-04,
          4.6806e-03, 1.5411e-03, 1.1169e-02, 1.9207e-03, 1.1024e-03, 9.1124e-04,
          6.4468e-04, 1.0004e-03, 4.0174e-04, 1.1024e-03, 5.6076e-04, 4.7231e-04,
          1.3056e-03, 1.2398e-03, 7.6914e-04, 6.6233e-04, 4.2295e-04, 9.1791e-04,
          8.5020e-04, 6.0654e-04, 7.3767e-04, 5.3072e-04, 8.2588e-04, 1.2856e-03,
          3.0136e-04, 4.7445e-04, 5.4054e-03, 1.1568e-03, 8.2588e-04, 5.0545e-04,
          5.3501e-04, 5.7840e-04, 1.0538e-03, 1.5383e-03, 1.4219e-03, 2.2564e-03,
          2.9125e-03, 6.1846e-04, 8.4019e-04, 5.4741e-04, 8.3303e-04, 1.7996e-03,
          8.9312e-04, 4.4274e-04, 4.6110e-04, 4.6182e-04, 4.6515e-04, 6.4468e-04,
          7.8011e-04, 2.3746e-03, 1.3514e-03, 1.3418e-03, 5.0068e-04, 3.1519e-04,
          9.1028e-04, 1.6870e-03, 1.2150e-03, 1.1454e-03, 1.2798e-03, 1.4954e-03,
          1.4534e-03, 1.0605e-03, 6.0749e-04, 1.2093e-03, 3.9649e-04, 5.4598e-04,
          2.3365e-03, 1.0824e-03, 4.4942e-04, 1.3199e-03, 1.6155e-03, 1.4362e-03,
          9.0885e-04, 7.6580e-04, 4.9734e-04, 1.7004e-03, 3.0956e-03, 1.8063e-03,
          3.6774e-03, 5.6124e-04, 6.6996e-04, 1.4353e-03, 1.6108e-03, 6.3419e-04,
          8.3828e-04, 6.6328e-04, 1.4763e-03, 4.7207e-04, 7.9584e-04, 5.2595e-04,
          1.2741e-03, 9.3937e-04, 3.4952e-04, 2.0313e-03, 1.7385e-03, 1.6375e-03,
          1.4248e-03, 2.5692e-03, 1.7252e-03, 3.7265e-04, 1.9274e-03, 5.5408e-04,
          1.5602e-03, 7.9966e-04, 1.0538e-03, 1.4915e-03, 4.7898e-04, 9.0361e-04,
          8.6117e-04, 5.7364e-04, 2.9588e-04, 2.4652e-04, 8.9645e-04, 5.4216e-04,
          1.0643e-03, 4.2272e-04, 6.1226e-04, 1.2245e-03, 9.9373e-04, 1.1063e-03,
          2.6493e-03, 9.0837e-04, 1.1864e-03, 1.4534e-03, 1.4839e-03, 2.5139e-03,
          2.6302e-03, 1.4639e-03, 1.9217e-03, 7.1287e-04, 1.1215e-03, 1.0624e-03,
          7.4863e-04, 5.3549e-04, 4.0889e-04, 5.2023e-04, 1.4658e-03, 4.8065e-04,
          1.3561e-03, 7.3004e-04, 9.7275e-04, 1.0605e-03, 4.2610e-03, 1.2369e-03,
          2.7390e-03, 1.0118e-03, 1.4992e-03, 8.8692e-04, 7.8869e-04, 1.7824e-03,
          2.2182e-03, 1.6747e-03, 3.4332e-03, 1.2836e-03, 1.1721e-03, 8.3637e-04,
          7.1001e-04, 6.0558e-04, 1.1969e-03, 6.3229e-04, 4.9257e-04, 6.5327e-04,
          1.2188e-03, 5.3883e-04, 8.4066e-04, 5.8937e-03, 1.8330e-03, 1.3599e-03,
          8.1635e-04, 2.1381e-03, 1.0281e-03, 1.5059e-03, 1.5469e-03, 1.2951e-03,
          3.0460e-03, 1.1501e-03, 1.2875e-03, 7.6437e-04, 6.3467e-04, 7.4005e-04,
          8.5068e-04, 9.7132e-04, 1.0977e-03, 4.9829e-04, 2.2221e-03, 4.6253e-04,
          6.6853e-04, 4.3297e-03, 1.1002e-02, 1.6479e-03, 2.8934e-03, 2.2717e-03,
          1.5507e-03, 1.0986e-03, 1.0834e-03, 6.2895e-04, 1.5335e-03, 8.8835e-04,
          2.2411e-03, 1.3762e-03, 1.1883e-03, 6.2561e-04, 2.2144e-03, 1.0624e-03,
          7.4291e-04, 7.2908e-04, 1.1768e-03, 3.2082e-03, 1.1873e-03, 4.9114e-04,
          1.2993e-02, 2.0409e-03, 2.8419e-03, 1.4277e-03, 6.3629e-03, 1.0109e-02,
          5.6496e-03, 5.2490e-03, 2.1515e-03, 1.8339e-03, 1.7624e-03, 1.6184e-03,
          1.5030e-03, 1.0719e-03, 1.2217e-03, 1.3647e-03, 1.4477e-03, 1.1797e-03,
          6.1417e-04, 1.3990e-03, 2.0905e-03, 1.2045e-03, 9.1219e-04, 2.2774e-03,
          7.6027e-03, 1.7223e-03, 2.8839e-03, 2.3499e-03, 4.6310e-03, 5.2414e-03,
          1.9989e-03, 5.8460e-04, 6.7282e-04, 1.5507e-03, 1.8406e-03, 9.2030e-04,
          1.7662e-03, 1.3952e-03, 1.8635e-03, 1.1950e-03, 1.7548e-03, 2.2240e-03,
          5.8937e-03, 2.0485e-03, 9.8324e-04, 8.2016e-04, 7.5798e-03, 5.1842e-03,
          2.4681e-03, 7.4921e-03, 6.6185e-03, 7.9193e-03, 2.0428e-03, 1.7128e-03,
          4.3449e-03, 2.1381e-03, 1.0395e-03, 1.1845e-03, 2.6169e-03, 2.8629e-03,
          2.1496e-03, 1.4286e-03, 1.1358e-03, 1.5345e-03, 1.3905e-03, 4.4670e-03,
          2.5921e-03, 6.3944e-04, 1.0519e-03, 2.8076e-03, 2.0264e-02, 4.1504e-03,
          2.2316e-03, 4.5891e-03, 2.4414e-03, 6.3944e-04, 7.8583e-04, 8.0824e-04,
          6.8712e-04, 2.0199e-03, 5.9776e-03, 1.3151e-03, 2.4261e-03, 1.3266e-03,
          1.3218e-03, 3.6736e-03, 1.1139e-03, 8.6212e-04, 3.9406e-03, 3.0766e-03,
          9.9373e-04, 6.6471e-04, 1.9045e-03, 4.9561e-02, 1.4458e-02, 4.7379e-03,
          4.2572e-03, 1.2197e-03, 4.8065e-03, 4.5547e-03, 1.1501e-03, 2.5997e-03,
          2.1000e-03, 1.8158e-03, 1.1559e-03, 6.7940e-03, 1.8835e-03, 1.9608e-03,
          1.5163e-03, 1.1463e-03, 1.2655e-03, 8.0681e-04, 1.3313e-03, 2.2736e-03,
          1.7490e-03, 1.0902e-02, 4.1718e-02, 5.5725e-02, 3.5048e-04, 3.5048e-04,
          7.3254e-05, 3.2449e-04, 7.3254e-05, 3.2449e-04, 2.1267e-04, 7.3254e-05,
          3.5048e-04, 3.5048e-04, 3.2449e-04, 7.3254e-05, 3.5048e-04, 7.3254e-05,
          4.6372e-04, 2.1124e-04, 7.3254e-05, 7.3254e-05, 7.3254e-05, 7.3254e-05,
          7.3254e-05, 7.3254e-05, 7.3254e-05, 2.1267e-04, 3.5048e-04, 7.3254e-05,
          3.5048e-04, 2.1267e-04, 2.1267e-04, 3.2449e-04, 3.2449e-04, 3.2449e-04,
          6.0177e-04, 3.2449e-04, 2.1267e-04, 3.2449e-04, 3.2449e-04, 3.2449e-04,
          3.2449e-04, 6.6757e-04, 3.2449e-04, 3.2449e-04, 7.3254e-05, 7.3254e-05,
          4.5991e-04, 4.5991e-04, 2.1124e-04, 7.3254e-05, 2.1124e-04, 7.3254e-05,
          7.3254e-05, 2.1124e-04, 2.1124e-04, 2.1124e-04, 7.3254e-05, 7.3254e-05,
          7.3254e-05, 2.1124e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.3474e-03, 5.0781e-02, 1.2390e-02, 1.5572e-02, 2.7695e-02,
          3.7750e-02, 7.9727e-03, 7.1793e-03, 1.8206e-03, 1.6556e-03, 2.4338e-03,
          1.0405e-03, 1.2741e-03, 7.6008e-04, 1.1473e-03, 6.7863e-03, 2.1458e-03,
          1.4162e-03, 1.8969e-03, 2.5678e-04, 2.6083e-04, 2.6226e-04, 1.6870e-03,
          2.8400e-03, 9.0551e-04, 7.1564e-03, 5.0735e-03, 5.4264e-04, 2.5291e-03,
          1.0223e-03, 1.8814e-02, 8.7118e-04, 2.0180e-03, 1.6870e-03, 7.5951e-03,
          1.4591e-03, 5.1212e-04, 1.5469e-03, 1.7719e-03, 3.7098e-04, 8.1158e-04,
          3.9959e-04, 3.1018e-04, 3.3760e-04, 2.3448e-04, 1.9407e-03, 1.4744e-03,
          3.9554e-04, 3.3617e-04, 2.1439e-03, 1.2100e-02, 1.6663e-02, 4.1771e-03,
          1.7471e-03, 7.8201e-03, 2.2564e-03, 5.6124e-04, 2.0027e-03, 7.7200e-04,
          9.1839e-04, 9.4986e-04, 3.6316e-03, 3.5915e-03, 4.6253e-04, 1.6737e-03,
          7.1716e-04, 4.2748e-04, 5.8174e-04, 1.8167e-04, 1.5306e-04, 2.8253e-04,
          3.3712e-04, 3.5706e-03, 3.6373e-03, 2.2079e-02, 2.0790e-03, 2.5692e-03,
          7.7171e-03, 5.2071e-04, 6.1607e-04, 1.3037e-03, 7.2193e-04, 5.0926e-04,
          4.3297e-04, 2.4887e-02, 2.3823e-03, 1.1177e-03, 3.7861e-04, 3.5000e-04,
          2.2278e-03, 1.4009e-03, 1.8036e-04, 1.3618e-03, 2.5058e-04, 3.3164e-04,
          1.1406e-03, 1.4334e-03, 1.5404e-02, 1.7929e-03, 5.3024e-04, 6.7997e-04,
          2.4986e-03, 4.9639e-04, 6.8092e-04, 9.8515e-04, 1.7023e-03, 6.2323e-04,
          1.6388e-02, 4.2534e-04, 1.5965e-03, 2.9063e-04, 4.7946e-04, 5.2643e-04,
          5.7030e-04, 4.0841e-04, 3.0184e-04, 1.9703e-03, 4.4060e-04, 1.0252e-03,
          1.3628e-03, 1.4286e-03, 3.3498e-04, 3.6335e-03, 1.9779e-03, 2.6584e-04,
          3.0613e-04, 4.0555e-04, 5.8365e-04, 2.3022e-03, 5.7907e-03, 1.7273e-02,
          8.4019e-04, 4.6587e-04, 5.3263e-04, 2.8229e-04, 8.9216e-04, 2.2709e-04,
          3.5381e-03, 6.0940e-04, 1.9255e-03, 4.7874e-03, 2.9793e-03, 1.6584e-03,
          4.8757e-04, 4.4274e-04, 5.1117e-04, 5.4026e-04, 6.1131e-04, 1.7118e-03,
          2.7351e-03, 1.6251e-03, 7.0620e-04, 7.8278e-03, 7.4577e-03, 8.5354e-04,
          1.0309e-03, 7.4720e-04, 7.8726e-04, 3.3569e-04, 1.3857e-03, 1.2922e-03,
          1.7529e-03, 1.7185e-03, 3.2501e-03, 1.9522e-03, 8.1348e-04, 6.7854e-04,
          1.8473e-03, 8.5831e-04, 1.1368e-03, 9.1505e-04, 2.0657e-03, 1.9484e-03,
          1.6022e-03, 1.2684e-03, 9.5177e-04, 8.5688e-04, 1.2293e-03, 3.4599e-03,
          7.1526e-04, 9.5510e-04, 1.4772e-03, 1.3298e-02, 2.5005e-03, 7.8869e-04,
          1.8635e-03, 1.3046e-03, 1.4114e-03, 6.9618e-04, 6.5327e-04, 1.7910e-03,
          1.6861e-02, 2.8563e-04, 1.4181e-03, 3.4237e-03, 1.7176e-03, 1.5402e-03,
          1.1654e-03, 7.6342e-04, 1.4191e-03, 5.4789e-04, 9.3555e-04, 7.8154e-04,
          3.9482e-04, 7.7009e-04, 3.0446e-04, 2.4548e-03, 2.2945e-03, 1.1835e-03,
          6.6719e-03, 1.3046e-03, 7.4673e-04, 1.2083e-03, 2.2640e-03, 5.4359e-04,
          1.3876e-03, 2.2583e-03, 2.7332e-03, 9.9850e-04, 2.9850e-03, 2.7409e-03,
          3.6869e-03, 4.5085e-04, 6.9571e-04, 6.5279e-04, 2.5253e-03, 1.6518e-03,
          5.7697e-04, 4.6253e-04, 3.0780e-04, 1.0223e-03, 1.0233e-03, 1.0605e-03,
          1.4000e-03, 8.4591e-04, 7.1001e-04, 1.1635e-03, 1.5650e-03, 2.6150e-03,
          1.2217e-03, 9.4318e-04, 7.3957e-04, 2.1286e-03, 3.2177e-03, 1.0843e-03,
          6.7902e-04, 8.2111e-04, 4.2295e-04, 5.6791e-04, 6.6614e-04, 1.0366e-03,
          3.2687e-04, 2.4378e-04, 4.7326e-04, 6.0081e-04, 2.2888e-03, 1.2329e-02,
          1.0653e-03, 9.2220e-04, 1.1196e-03, 1.1101e-03, 5.3549e-04, 1.3332e-03,
          1.5545e-03, 1.3275e-03, 2.4090e-03, 5.4741e-03, 4.0817e-03, 8.5258e-04,
          1.0557e-03, 8.0729e-04, 1.4400e-03, 4.6825e-04, 1.7014e-03, 1.2255e-03,
          8.9979e-04, 1.0796e-03, 5.0068e-04, 1.7042e-03, 9.6083e-04, 1.2417e-03,
          1.2608e-03, 6.6090e-04, 4.4465e-04, 7.1430e-04, 2.2297e-03, 1.4601e-03,
          2.0313e-03, 3.1567e-03, 3.0270e-03, 3.4084e-03, 7.6294e-04, 1.1044e-03,
          7.3147e-04, 4.5061e-04, 3.3975e-04, 2.0580e-03, 1.5850e-03, 6.1989e-04,
          4.5061e-04, 1.7071e-03, 1.2383e-02, 4.4365e-03, 9.2125e-04, 2.7618e-03,
          2.1706e-03, 1.2970e-03, 2.2011e-03, 2.2755e-03, 1.4257e-03, 1.9588e-03,
          1.8682e-03, 2.1057e-03, 1.6384e-03, 3.4199e-03, 3.7880e-03, 1.6155e-03,
          1.1568e-03, 4.5242e-03, 2.1114e-03, 1.6937e-03, 5.4789e-04, 6.1083e-04,
          1.7309e-03, 1.7080e-03, 3.2730e-03, 1.3857e-03, 2.5539e-03, 4.5242e-03,
          3.9597e-03, 1.7176e-03, 2.9678e-03, 6.2065e-03, 1.8444e-03, 3.6125e-03,
          1.8539e-03, 1.3397e-02, 1.2367e-02, 1.5736e-03, 2.2774e-03, 2.4796e-03,
          1.6460e-03, 1.6661e-03, 2.3212e-03, 1.8425e-03, 1.5686e-02, 1.6575e-03,
          8.5878e-04, 4.8866e-03, 2.0981e-03, 1.5898e-03, 6.1676e-02, 6.6638e-05,
          6.6638e-05, 6.6638e-05, 3.4332e-04, 6.6638e-05, 1.6618e-04, 1.6618e-04,
          6.6638e-05, 6.6638e-05, 6.6638e-05, 6.6638e-05, 6.6638e-05, 6.6638e-05,
          1.6618e-04, 9.7215e-05, 6.6638e-05, 9.1887e-04, 6.6638e-05, 6.6638e-05,
          4.0841e-04, 6.6638e-05, 6.6638e-05, 6.6638e-05, 6.6638e-05, 3.1471e-04,
          6.6638e-05, 4.4298e-04, 1.6618e-04, 6.6638e-05, 6.6638e-05, 1.6618e-04,
          6.6638e-05, 8.9073e-04, 4.4298e-04, 6.6638e-05, 6.6638e-05, 1.6618e-04,
          4.1437e-04, 6.6638e-05, 6.6638e-05, 1.6618e-04, 6.6638e-05, 6.6638e-05,
          1.6618e-04, 6.6638e-05, 1.6618e-04, 1.6618e-04, 1.6618e-04, 6.6638e-05,
          6.6638e-05, 1.6618e-04, 1.6618e-04, 3.8433e-04, 1.6618e-04, 1.6618e-04,
          1.6618e-04, 1.6618e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.8629e-03, 5.0262e-02, 1.2383e-02, 1.1688e-02, 2.4612e-02,
          3.7659e-02, 6.4201e-03, 6.7329e-03, 1.5936e-03, 1.6499e-03, 2.3041e-03,
          1.0662e-03, 1.2398e-03, 8.6069e-04, 1.1959e-03, 6.2675e-03, 1.9989e-03,
          1.7700e-03, 1.5688e-03, 2.4486e-04, 2.1613e-04, 2.7251e-04, 1.8997e-03,
          2.6741e-03, 8.4496e-04, 5.7449e-03, 4.8332e-03, 5.6505e-04, 2.6207e-03,
          1.1578e-03, 1.9180e-02, 1.0118e-03, 1.9875e-03, 1.9875e-03, 6.8893e-03,
          1.2321e-03, 4.2629e-04, 1.2178e-03, 1.8549e-03, 3.8719e-04, 5.8079e-04,
          3.7503e-04, 2.8276e-04, 3.5167e-04, 2.0993e-04, 1.9760e-03, 1.3180e-03,
          4.7994e-04, 3.1543e-04, 1.8749e-03, 1.0193e-02, 1.7975e-02, 3.6583e-03,
          1.5478e-03, 8.5602e-03, 1.7824e-03, 4.7755e-04, 2.2182e-03, 7.2479e-04,
          8.5306e-04, 1.0672e-03, 3.4428e-03, 2.2774e-03, 3.9864e-04, 1.5717e-03,
          6.9094e-04, 4.1294e-04, 5.4026e-04, 1.5771e-04, 1.4091e-04, 2.8038e-04,
          3.2234e-04, 3.2425e-03, 2.9373e-03, 2.4353e-02, 1.8787e-03, 2.3098e-03,
          7.5111e-03, 4.1819e-04, 5.4598e-04, 1.2980e-03, 7.7581e-04, 5.0306e-04,
          4.3869e-04, 2.6413e-02, 2.3441e-03, 1.0843e-03, 4.1509e-04, 3.5501e-04,
          2.2831e-03, 1.1625e-03, 1.7571e-04, 1.3189e-03, 2.4438e-04, 3.2330e-04,
          1.5354e-03, 1.2188e-03, 1.3687e-02, 1.3790e-03, 3.6907e-04, 5.0497e-04,
          2.3708e-03, 4.9829e-04, 8.2445e-04, 1.0309e-03, 1.9588e-03, 6.0606e-04,
          1.6159e-02, 3.5715e-04, 1.2512e-03, 2.6083e-04, 4.7445e-04, 4.7159e-04,
          5.8651e-04, 3.4904e-04, 2.9755e-04, 2.0828e-03, 4.3583e-04, 8.4400e-04,
          1.2178e-03, 1.1091e-03, 2.4867e-04, 4.9667e-03, 2.0466e-03, 2.4152e-04,
          2.5511e-04, 3.8576e-04, 5.3549e-04, 2.8229e-03, 5.4283e-03, 1.8814e-02,
          7.7963e-04, 4.7588e-04, 6.1941e-04, 2.8157e-04, 7.8011e-04, 2.1148e-04,
          2.9316e-03, 5.6410e-04, 1.8692e-03, 4.6959e-03, 2.8133e-03, 1.5974e-03,
          3.3712e-04, 4.3893e-04, 4.5228e-04, 4.5681e-04, 5.8889e-04, 1.9989e-03,
          2.9850e-03, 1.9407e-03, 7.0381e-04, 7.0953e-03, 8.4381e-03, 6.3419e-04,
          8.4162e-04, 6.2418e-04, 8.8930e-04, 3.1376e-04, 1.2980e-03, 1.1120e-03,
          1.4734e-03, 1.4200e-03, 3.1948e-03, 1.9627e-03, 5.5933e-04, 5.6171e-04,
          1.5392e-03, 7.4863e-04, 1.1215e-03, 9.9087e-04, 1.7614e-03, 1.8044e-03,
          1.7567e-03, 1.3714e-03, 9.7466e-04, 7.0858e-04, 1.0090e-03, 4.1161e-03,
          6.5851e-04, 9.6035e-04, 1.3905e-03, 1.5991e-02, 2.7332e-03, 9.1076e-04,
          1.9798e-03, 1.2712e-03, 1.3895e-03, 4.6921e-04, 5.9509e-04, 1.8435e-03,
          2.0111e-02, 2.3079e-04, 1.2865e-03, 3.1395e-03, 1.7214e-03, 1.5755e-03,
          1.2121e-03, 7.0047e-04, 1.2999e-03, 4.9829e-04, 8.6737e-04, 6.9809e-04,
          3.9649e-04, 6.6280e-04, 2.7823e-04, 2.9964e-03, 2.1439e-03, 1.0958e-03,
          8.1177e-03, 8.6784e-04, 5.2118e-04, 1.0519e-03, 2.2049e-03, 4.8256e-04,
          1.1854e-03, 1.9894e-03, 3.0575e-03, 1.1206e-03, 2.7695e-03, 2.9335e-03,
          2.9755e-03, 3.5524e-04, 5.6314e-04, 6.3181e-04, 2.4147e-03, 1.5755e-03,
          6.1226e-04, 4.5991e-04, 2.9492e-04, 1.0614e-03, 1.0366e-03, 1.1768e-03,
          9.8515e-04, 6.8760e-04, 5.8174e-04, 1.1187e-03, 1.2960e-03, 2.2526e-03,
          1.0061e-03, 7.8583e-04, 5.5933e-04, 1.7614e-03, 3.2158e-03, 9.4557e-04,
          5.0926e-04, 7.6437e-04, 3.5691e-04, 4.7374e-04, 5.7220e-04, 9.8038e-04,
          3.1352e-04, 2.3437e-04, 4.4489e-04, 5.7316e-04, 2.5997e-03, 9.4452e-03,
          7.6342e-04, 8.1205e-04, 1.0033e-03, 1.1225e-03, 4.2486e-04, 1.1921e-03,
          1.3742e-03, 1.4105e-03, 2.3918e-03, 5.3596e-03, 4.3678e-03, 8.3685e-04,
          9.2459e-04, 8.0538e-04, 1.4105e-03, 4.0555e-04, 1.3924e-03, 1.2569e-03,
          8.9264e-04, 1.0853e-03, 4.6086e-04, 1.8663e-03, 6.4182e-04, 9.9659e-04,
          1.2341e-03, 5.1355e-04, 3.5620e-04, 5.7173e-04, 2.0561e-03, 1.2856e-03,
          1.8969e-03, 2.5330e-03, 2.9697e-03, 3.3131e-03, 6.4039e-04, 9.1696e-04,
          6.1703e-04, 3.9101e-04, 2.7537e-04, 1.9264e-03, 1.3685e-03, 6.0368e-04,
          4.0293e-04, 1.9627e-03, 1.6327e-02, 3.4275e-03, 6.2466e-04, 2.5806e-03,
          2.0466e-03, 9.5654e-04, 1.7233e-03, 1.8988e-03, 1.3609e-03, 2.2011e-03,
          1.6527e-03, 1.9321e-03, 1.2445e-03, 3.3722e-03, 3.4657e-03, 1.7242e-03,
          1.0796e-03, 4.6272e-03, 1.8024e-03, 1.8930e-03, 4.7827e-04, 4.7803e-04,
          1.9760e-03, 1.7357e-03, 3.6488e-03, 8.7404e-04, 1.8969e-03, 4.3640e-03,
          3.1090e-03, 1.4296e-03, 2.8572e-03, 4.9591e-03, 1.8663e-03, 3.2272e-03,
          1.7509e-03, 1.4206e-02, 1.2100e-02, 1.2360e-03, 1.9817e-03, 2.6264e-03,
          1.5764e-03, 1.6642e-03, 2.2984e-03, 1.9045e-03, 1.6159e-02, 1.5211e-03,
          8.0156e-04, 4.1275e-03, 1.4000e-03, 1.1101e-03, 4.6539e-02, 3.8330e-02,
          6.6698e-05, 1.7095e-04, 2.6274e-04, 6.6698e-05, 3.6716e-04, 1.7095e-04,
          6.6698e-05, 6.6698e-05, 6.6698e-05, 2.6274e-04, 6.6698e-05, 6.6698e-05,
          1.7095e-04, 1.7095e-04, 6.6698e-05, 7.8154e-04, 6.6698e-05, 6.6698e-05,
          4.6825e-04, 6.6698e-05, 1.7095e-04, 6.6698e-05, 6.6698e-05, 1.1215e-03,
          6.6698e-05, 7.8773e-04, 3.6716e-04, 6.6698e-05, 6.6698e-05, 5.9175e-04,
          6.6698e-05, 6.2704e-04, 7.8773e-04, 6.6698e-05, 6.6698e-05, 5.8508e-04,
          1.1520e-03, 6.6698e-05, 6.6698e-05, 1.7095e-04, 1.7095e-04, 6.6698e-05,
          1.7095e-04, 1.7095e-04, 1.7095e-04, 1.7095e-04, 2.8801e-04, 6.6698e-05,
          6.6698e-05, 3.6716e-04, 1.7095e-04, 2.6155e-04, 2.6155e-04, 3.6573e-04,
          6.6698e-05, 6.6698e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.1185e-03, 5.8075e-02, 9.8572e-03, 1.1627e-02, 2.5604e-02,
          4.5258e-02, 6.5918e-03, 3.0708e-03, 9.3937e-04, 1.4648e-03, 1.7967e-03,
          8.3160e-04, 1.0014e-03, 6.5804e-04, 1.4238e-03, 2.9430e-03, 2.0199e-03,
          9.7847e-04, 8.2207e-04, 3.6740e-04, 3.9530e-04, 2.8253e-04, 1.3466e-03,
          1.9522e-03, 1.0157e-03, 5.8441e-03, 3.8280e-03, 5.9605e-04, 2.0809e-03,
          7.8011e-04, 1.0834e-02, 8.7214e-04, 2.8095e-03, 1.7214e-03, 7.1564e-03,
          2.2011e-03, 6.5708e-04, 2.1820e-03, 2.4147e-03, 5.5504e-04, 1.1072e-03,
          4.4537e-04, 3.7360e-04, 3.9291e-04, 3.3236e-04, 1.4439e-03, 2.5272e-03,
          4.2892e-04, 4.1485e-04, 1.7471e-03, 1.7059e-02, 1.0406e-02, 3.3836e-03,
          1.5278e-03, 5.4054e-03, 3.3703e-03, 8.6880e-04, 3.1357e-03, 1.3561e-03,
          1.9665e-03, 1.4944e-03, 3.7937e-03, 8.1711e-03, 4.6086e-04, 2.9125e-03,
          4.7040e-04, 2.9445e-04, 4.7255e-04, 2.3127e-04, 1.9443e-04, 3.4857e-04,
          3.8862e-04, 3.0098e-03, 4.7722e-03, 1.3542e-02, 1.3037e-03, 1.6870e-03,
          9.5978e-03, 7.5197e-04, 5.3024e-04, 1.5583e-03, 6.9284e-04, 9.3651e-04,
          7.5579e-04, 1.5244e-02, 2.1057e-03, 7.5388e-04, 2.5678e-04, 2.8324e-04,
          1.5411e-03, 1.3828e-03, 2.3818e-04, 1.0986e-03, 4.4012e-04, 4.7612e-04,
          1.9455e-03, 1.9178e-03, 1.1482e-02, 3.1853e-03, 5.4693e-04, 7.7772e-04,
          1.9445e-03, 4.1938e-04, 5.9986e-04, 2.2793e-03, 2.6627e-03, 1.0710e-03,
          1.1002e-02, 5.6839e-04, 1.7214e-03, 2.7800e-04, 3.7241e-04, 4.6921e-04,
          7.7343e-04, 8.2874e-04, 5.1689e-04, 3.4542e-03, 6.1893e-04, 9.2745e-04,
          1.2922e-03, 2.0294e-03, 3.1447e-04, 1.7977e-03, 2.3346e-03, 2.7728e-04,
          4.1199e-04, 4.7827e-04, 6.5136e-04, 3.7670e-03, 4.7951e-03, 1.1826e-02,
          6.4516e-04, 3.1734e-04, 4.8614e-04, 2.1911e-04, 8.6308e-04, 2.8682e-04,
          4.5128e-03, 6.6996e-04, 1.8177e-03, 4.3983e-03, 3.7537e-03, 1.4267e-03,
          4.3464e-04, 6.7329e-04, 5.5742e-04, 6.6853e-04, 5.2691e-04, 2.6875e-03,
          3.7079e-03, 2.9907e-03, 1.4763e-03, 1.4275e-02, 7.8583e-03, 1.4391e-03,
          1.2865e-03, 1.0834e-03, 4.3559e-04, 3.2234e-04, 2.4471e-03, 1.7309e-03,
          2.5940e-03, 2.2106e-03, 2.4586e-03, 2.2488e-03, 1.1053e-03, 8.0347e-04,
          1.3895e-03, 1.0471e-03, 1.3704e-03, 1.2474e-03, 1.6632e-03, 3.3836e-03,
          3.7422e-03, 2.1858e-03, 2.4185e-03, 1.1740e-03, 1.7700e-03, 1.8435e-03,
          8.5306e-04, 1.1578e-03, 2.0199e-03, 5.4054e-03, 1.8301e-03, 1.2808e-03,
          1.2732e-03, 9.9945e-04, 1.3151e-03, 6.5851e-04, 5.4550e-04, 2.2926e-03,
          7.3204e-03, 3.4022e-04, 1.9283e-03, 2.5139e-03, 1.4906e-03, 2.5234e-03,
          3.0212e-03, 1.0548e-03, 2.3880e-03, 4.9782e-04, 8.1110e-04, 9.8419e-04,
          3.3307e-04, 9.4795e-04, 4.1842e-04, 1.6279e-03, 1.9627e-03, 1.0700e-03,
          2.9354e-03, 1.5802e-03, 6.1750e-04, 8.4972e-04, 2.7637e-03, 5.6887e-04,
          2.2678e-03, 2.8038e-03, 2.5063e-03, 1.1940e-03, 3.9787e-03, 2.1820e-03,
          5.3596e-03, 4.6849e-04, 8.6308e-04, 5.7411e-04, 3.1796e-03, 1.8244e-03,
          4.1580e-04, 3.8886e-04, 3.7050e-04, 1.3895e-03, 1.4944e-03, 1.0128e-03,
          1.0786e-03, 7.8201e-04, 5.0783e-04, 1.7977e-03, 2.1935e-03, 3.2501e-03,
          1.6251e-03, 1.2245e-03, 9.1219e-04, 2.8992e-03, 3.4542e-03, 1.1158e-03,
          8.6594e-04, 5.6934e-04, 3.9911e-04, 5.1641e-04, 6.3086e-04, 1.4515e-03,
          2.9182e-04, 3.4642e-04, 7.0477e-04, 1.0386e-03, 2.4452e-03, 1.5900e-02,
          1.1806e-03, 8.1253e-04, 1.7195e-03, 1.6518e-03, 8.1015e-04, 2.6188e-03,
          1.9064e-03, 1.8263e-03, 2.0676e-03, 4.0436e-03, 2.5234e-03, 1.1797e-03,
          7.6485e-04, 5.6124e-04, 2.0313e-03, 6.0368e-04, 2.9030e-03, 1.9569e-03,
          1.2064e-03, 1.4544e-03, 5.4073e-04, 1.2121e-03, 8.3494e-04, 8.6260e-04,
          7.5102e-04, 7.7438e-04, 5.2786e-04, 1.1101e-03, 2.9430e-03, 2.7370e-03,
          2.3956e-03, 3.9330e-03, 2.4834e-03, 2.6283e-03, 7.8201e-04, 1.1158e-03,
          7.8821e-04, 4.4942e-04, 3.5071e-04, 2.8038e-03, 2.5578e-03, 7.1907e-04,
          5.2643e-04, 1.5688e-03, 6.0196e-03, 4.3602e-03, 6.8998e-04, 2.8801e-03,
          2.8648e-03, 1.3294e-03, 3.3951e-03, 2.3670e-03, 2.1744e-03, 2.0370e-03,
          1.9855e-03, 2.7390e-03, 1.8787e-03, 3.2425e-03, 4.7913e-03, 2.0142e-03,
          1.1234e-03, 3.4904e-03, 2.6741e-03, 1.4811e-03, 6.3848e-04, 7.4291e-04,
          1.4887e-03, 1.3199e-03, 2.3479e-03, 1.0386e-03, 2.7275e-03, 2.8458e-03,
          4.6539e-03, 1.3533e-03, 2.0409e-03, 5.7106e-03, 1.8654e-03, 3.7270e-03,
          2.7885e-03, 1.5594e-02, 1.1208e-02, 1.5354e-03, 2.6321e-03, 1.8272e-03,
          1.4753e-03, 1.8778e-03, 2.5387e-03, 1.2732e-03, 1.9516e-02, 2.1133e-03,
          6.9284e-04, 3.5419e-03, 1.5764e-03, 1.5011e-03, 2.3422e-02, 1.2413e-02,
          4.1504e-02, 2.0409e-04, 9.9123e-05, 9.9123e-05, 2.0409e-04, 2.0409e-04,
          2.0409e-04, 2.0409e-04, 9.9123e-05, 9.9123e-05, 9.9123e-05, 9.9123e-05,
          2.0409e-04, 2.0409e-04, 9.9123e-05, 2.0409e-04, 9.9123e-05, 9.9123e-05,
          6.0368e-04, 9.9123e-05, 9.9123e-05, 9.9123e-05, 9.9123e-05, 9.9123e-05,
          2.0409e-04, 2.0409e-04, 2.0409e-04, 9.9123e-05, 9.9123e-05, 5.1975e-04,
          2.0409e-04, 9.9123e-05, 2.0409e-04, 9.9123e-05, 2.0409e-04, 2.0409e-04,
          2.0409e-04, 2.0409e-04, 9.9123e-05, 2.0409e-04, 2.0409e-04, 9.9123e-05,
          1.6451e-04, 1.6451e-04, 2.0409e-04, 1.6451e-04, 2.0409e-04, 5.9664e-05,
          5.9664e-05, 1.6451e-04, 1.6451e-04, 3.2330e-04, 3.6263e-04, 1.0319e-03,
          2.0409e-04, 1.6451e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.6087e-03, 5.3558e-02, 1.2962e-02, 1.2817e-02, 3.1128e-02,
          4.8828e-02, 9.1171e-03, 4.5547e-03, 1.5993e-03, 2.7122e-03, 6.8398e-03,
          1.5039e-03, 1.6441e-03, 9.7132e-04, 2.2678e-03, 4.7150e-03, 1.3504e-03,
          1.3943e-03, 1.8511e-03, 1.2374e-04, 1.3232e-04, 1.8668e-04, 2.2831e-03,
          5.9471e-03, 1.1129e-03, 7.3929e-03, 4.0054e-03, 2.9230e-04, 1.7109e-03,
          2.1820e-03, 8.4152e-03, 5.2404e-04, 1.3399e-03, 3.5076e-03, 5.4836e-04,
          5.3978e-04, 2.4271e-04, 8.7929e-04, 1.5192e-03, 2.2888e-04, 3.7909e-04,
          2.3043e-04, 1.6165e-04, 2.4748e-04, 1.4389e-04, 2.2240e-03, 1.0576e-03,
          4.9257e-04, 3.0208e-04, 2.2202e-03, 7.9298e-04, 4.3182e-03, 1.1654e-03,
          3.6407e-04, 1.8654e-03, 2.8348e-04, 2.0123e-04, 9.1219e-04, 3.4571e-04,
          4.3106e-04, 6.9666e-04, 4.5943e-04, 3.6407e-04, 2.3818e-04, 2.1725e-03,
          7.4148e-04, 4.6682e-04, 6.9046e-04, 1.0860e-04, 1.1307e-04, 2.0421e-04,
          2.3556e-04, 4.2114e-03, 4.1032e-04, 6.3820e-03, 4.5085e-04, 4.4966e-04,
          1.1625e-03, 1.7524e-04, 3.5644e-04, 9.6893e-04, 6.4707e-04, 2.6560e-04,
          2.2781e-04, 5.7716e-03, 4.0150e-04, 3.3588e-03, 4.2629e-04, 3.8075e-04,
          5.2071e-03, 1.1845e-03, 1.8311e-04, 1.8988e-03, 1.9324e-04, 2.0349e-04,
          4.0197e-04, 2.3663e-04, 1.5907e-03, 7.7438e-04, 2.0075e-04, 2.7394e-04,
          4.5700e-03, 6.0129e-04, 8.4162e-04, 5.4407e-04, 2.3003e-03, 4.3678e-04,
          3.1757e-03, 1.9383e-04, 1.1425e-03, 1.9383e-04, 4.5300e-04, 3.9816e-04,
          1.0233e-03, 3.4714e-04, 2.5678e-04, 2.2049e-03, 2.9469e-04, 5.4455e-04,
          1.3151e-03, 7.0238e-04, 1.4853e-04, 7.3662e-03, 4.5128e-03, 1.5652e-04,
          1.5175e-04, 2.5940e-04, 3.3450e-04, 1.7185e-03, 3.7289e-03, 6.1264e-03,
          7.0381e-04, 6.2656e-04, 6.6185e-03, 3.1543e-04, 8.5163e-04, 1.7452e-04,
          2.1458e-03, 8.5640e-04, 3.0327e-03, 6.1264e-03, 2.8439e-03, 2.6951e-03,
          2.0003e-04, 1.4362e-03, 2.9087e-04, 2.2066e-04, 5.5218e-04, 1.5030e-03,
          2.1820e-03, 1.2350e-03, 2.7680e-04, 6.8331e-04, 7.9727e-04, 3.5214e-04,
          8.3494e-04, 3.4904e-04, 2.8152e-03, 2.9373e-04, 8.7452e-04, 6.8092e-04,
          1.3275e-03, 1.3199e-03, 3.1471e-03, 5.8670e-03, 3.2640e-04, 3.2496e-04,
          2.1286e-03, 4.5800e-04, 9.2316e-04, 2.3785e-03, 1.6479e-03, 9.8038e-04,
          9.6655e-04, 7.3195e-04, 4.1842e-04, 3.3903e-04, 1.0347e-03, 2.3712e-02,
          8.7404e-04, 1.4296e-03, 2.5425e-03, 9.8114e-03, 1.2360e-02, 2.4815e-03,
          6.0196e-03, 1.0786e-03, 5.0812e-03, 3.1471e-04, 6.1798e-04, 2.9545e-03,
          1.3504e-02, 1.5986e-04, 1.2894e-03, 2.8858e-03, 2.9678e-03, 1.2445e-03,
          5.3692e-04, 4.5276e-04, 1.3456e-03, 5.1498e-04, 1.2655e-03, 7.3051e-04,
          4.1413e-04, 6.3992e-04, 2.3377e-04, 2.1530e-02, 2.9964e-03, 9.9468e-04,
          5.8708e-03, 5.5647e-04, 3.4738e-04, 1.3599e-03, 1.8520e-03, 4.1986e-04,
          6.6328e-04, 1.5841e-03, 7.6637e-03, 3.0136e-03, 2.1496e-03, 1.4305e-03,
          1.5650e-03, 2.6917e-04, 5.7316e-04, 7.0429e-04, 6.0310e-03, 3.1910e-03,
          8.3828e-04, 4.6325e-04, 2.7609e-04, 1.1015e-03, 9.9850e-04, 7.5006e-04,
          6.6423e-04, 4.7231e-04, 4.5347e-04, 1.8911e-03, 8.2731e-04, 2.3899e-03,
          8.6308e-04, 6.3419e-04, 3.3379e-04, 6.5088e-04, 1.5450e-03, 1.3704e-03,
          3.1900e-04, 9.7752e-04, 3.2520e-04, 4.3583e-04, 5.9414e-04, 1.3933e-03,
          3.0327e-04, 2.1338e-04, 4.1127e-04, 4.4703e-04, 4.3373e-03, 4.3297e-03,
          4.3488e-04, 6.9475e-04, 8.0490e-04, 8.8882e-04, 3.1805e-04, 1.4019e-03,
          9.4891e-04, 3.9597e-03, 3.7823e-03, 3.2444e-03, 1.3123e-02, 1.0586e-03,
          8.3828e-04, 1.2321e-03, 3.0136e-03, 4.7278e-04, 1.0471e-03, 1.5707e-03,
          9.4986e-04, 1.2684e-03, 4.3201e-04, 4.3068e-03, 5.0354e-04, 8.0919e-04,
          1.5802e-03, 3.9673e-04, 2.9659e-04, 4.3106e-04, 1.2465e-03, 1.2350e-03,
          1.6909e-03, 3.8395e-03, 3.4313e-03, 4.4327e-03, 4.9639e-04, 7.5388e-04,
          5.3978e-04, 3.5858e-04, 2.5749e-04, 2.1000e-03, 1.0014e-03, 6.2323e-04,
          4.4751e-04, 3.4981e-03, 1.0231e-02, 2.3098e-03, 4.5300e-04, 2.0866e-03,
          1.8387e-03, 6.5374e-04, 1.2321e-03, 1.7786e-03, 7.2365e-03, 6.6681e-03,
          1.9045e-03, 1.3514e-03, 1.1463e-03, 3.6488e-03, 3.4809e-03, 2.5902e-03,
          1.5287e-03, 8.8730e-03, 1.7986e-03, 1.7920e-03, 4.6921e-04, 4.6325e-04,
          2.8954e-03, 1.3809e-03, 4.1924e-03, 6.2084e-04, 1.4639e-03, 1.0223e-02,
          2.6035e-03, 1.4782e-03, 8.1787e-03, 5.1231e-03, 1.2253e-02, 3.9787e-03,
          1.2741e-03, 5.8479e-03, 6.3248e-03, 1.3752e-03, 1.9484e-03, 7.2708e-03,
          4.7874e-03, 7.0267e-03, 1.8692e-03, 5.2071e-03, 3.7785e-03, 1.6556e-03,
          1.4372e-03, 5.4550e-03, 1.1072e-03, 9.3031e-04, 9.4757e-03, 8.0795e-03,
          4.0970e-03, 1.0199e-01, 6.5267e-05, 6.5267e-05, 6.5267e-05, 6.2847e-04,
          2.0301e-04, 2.0301e-04, 2.0301e-04, 6.5267e-05, 2.0301e-04, 2.0301e-04,
          6.5267e-05, 6.5267e-05, 6.5267e-05, 1.0347e-03, 6.5267e-05, 6.5267e-05,
          2.0301e-04, 6.5267e-05, 6.5267e-05, 6.5267e-05, 2.0301e-04, 6.5267e-05,
          2.0301e-04, 6.5267e-05, 2.0301e-04, 6.5267e-05, 6.5267e-05, 6.5267e-05,
          2.0301e-04, 6.5267e-05, 6.5267e-05, 6.5267e-05, 6.5267e-05, 7.6628e-04,
          6.5267e-05, 2.0301e-04, 6.5267e-05, 6.5267e-05, 6.5267e-05, 6.5267e-05,
          1.2321e-03, 2.0301e-04, 2.0301e-04, 2.0301e-04, 2.0301e-04, 2.0301e-04,
          6.5267e-05, 2.0301e-04, 2.0301e-04, 2.0301e-04, 2.0301e-04, 2.0301e-04,
          2.0301e-04, 2.0301e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.9520e-03, 5.7983e-02, 1.1314e-02, 1.1620e-02, 2.7466e-02,
          4.1962e-02, 8.0109e-03, 7.0000e-04, 5.0640e-04, 4.7035e-03, 5.7945e-03,
          2.7027e-03, 7.6008e-04, 3.0375e-04, 2.8362e-03, 6.0987e-04, 6.8092e-04,
          4.3535e-04, 4.2915e-04, 1.1098e-04, 1.4484e-04, 1.2255e-04, 1.1444e-03,
          5.9128e-03, 2.3003e-03, 6.6109e-03, 3.0136e-03, 2.2149e-04, 8.8453e-04,
          4.1556e-04, 1.5764e-03, 3.0208e-04, 2.5005e-03, 4.7569e-03, 4.8518e-04,
          2.1019e-03, 5.2834e-04, 1.5860e-03, 2.1038e-03, 1.6463e-04, 1.3762e-03,
          3.5453e-04, 3.2902e-04, 3.3283e-04, 3.2139e-04, 9.1362e-04, 2.4242e-03,
          1.7738e-04, 1.9050e-04, 1.6565e-03, 7.2956e-04, 1.0529e-03, 7.5102e-04,
          2.2840e-04, 6.3372e-04, 3.2234e-04, 7.1144e-04, 1.3027e-03, 1.2321e-03,
          1.0910e-03, 4.0317e-04, 5.2309e-04, 5.2071e-04, 6.8855e-04, 6.3515e-03,
          7.1239e-04, 8.5878e-04, 1.1683e-03, 2.4557e-04, 2.0075e-04, 2.6703e-04,
          2.7633e-04, 7.2136e-03, 3.3760e-04, 1.3762e-03, 2.2030e-04, 2.4271e-04,
          6.6710e-04, 6.3229e-04, 7.8201e-04, 5.2834e-03, 1.3552e-03, 7.1287e-04,
          5.4979e-04, 1.4095e-03, 3.1662e-04, 2.9087e-03, 5.6171e-04, 4.0436e-04,
          5.3520e-03, 3.4637e-03, 2.4772e-04, 3.0117e-03, 2.6155e-04, 2.6703e-04,
          2.9850e-04, 2.2030e-04, 9.4986e-04, 1.3075e-03, 5.4502e-04, 1.0586e-03,
          7.6256e-03, 1.0786e-03, 1.6336e-03, 3.3379e-03, 6.8207e-03, 3.4618e-04,
          1.2941e-03, 3.8147e-04, 3.4103e-03, 2.9421e-04, 5.2691e-04, 8.6451e-04,
          2.1877e-03, 5.6887e-04, 1.8477e-04, 5.0545e-03, 3.8838e-04, 4.6086e-04,
          5.3167e-04, 3.8433e-03, 3.6788e-04, 6.3896e-04, 1.7914e-02, 3.8290e-04,
          4.3607e-04, 4.7827e-04, 7.1192e-04, 4.9782e-03, 1.6832e-03, 1.7462e-03,
          8.5354e-04, 7.4339e-04, 2.0504e-03, 5.2404e-04, 2.2430e-03, 2.8014e-04,
          2.2793e-03, 3.0470e-04, 1.3008e-03, 3.0956e-03, 3.6469e-03, 7.2813e-04,
          5.3167e-04, 2.6360e-03, 1.0166e-03, 9.8133e-04, 7.4482e-04, 5.2757e-03,
          1.5068e-02, 5.0125e-03, 1.1530e-03, 8.4066e-04, 8.6594e-04, 1.1330e-03,
          3.1796e-03, 1.5306e-03, 1.6928e-03, 3.6311e-04, 2.0771e-03, 4.8923e-04,
          3.6106e-03, 4.2038e-03, 1.6861e-03, 7.5569e-03, 7.9012e-04, 1.2274e-03,
          6.0196e-03, 2.0370e-03, 3.1395e-03, 3.2272e-03, 1.0014e-03, 3.8872e-03,
          1.6832e-03, 8.9550e-04, 1.6336e-03, 6.6423e-04, 3.9635e-03, 3.5839e-03,
          2.9163e-03, 2.0027e-03, 9.8648e-03, 1.7700e-03, 1.4582e-03, 2.5196e-03,
          5.1079e-03, 4.8923e-04, 4.7340e-03, 7.2002e-04, 7.7629e-04, 7.7858e-03,
          2.5024e-03, 4.1676e-04, 4.4937e-03, 2.1706e-03, 7.4425e-03, 4.7073e-03,
          1.4791e-03, 4.8661e-04, 3.2959e-03, 1.1272e-03, 2.8763e-03, 2.9488e-03,
          5.0449e-04, 2.2812e-03, 4.2510e-04, 1.7548e-03, 5.4970e-03, 7.7391e-04,
          1.2951e-03, 1.0147e-03, 5.7697e-04, 2.3327e-03, 4.7493e-03, 6.9571e-04,
          2.3842e-03, 1.1330e-03, 9.9487e-03, 3.8853e-03, 4.6539e-03, 8.6737e-04,
          2.3613e-03, 4.4680e-04, 1.8063e-03, 1.4467e-03, 2.2095e-02, 5.7220e-03,
          9.1171e-04, 4.1199e-04, 4.1652e-04, 1.0204e-03, 2.3155e-03, 4.3821e-04,
          8.5688e-04, 1.2722e-03, 6.7282e-04, 3.8261e-03, 2.0351e-03, 3.2158e-03,
          1.3533e-03, 1.0738e-03, 8.0252e-04, 1.4477e-03, 1.4105e-03, 6.5517e-04,
          6.7663e-04, 1.3399e-03, 3.6693e-04, 5.6314e-04, 6.5088e-04, 4.7150e-03,
          3.3402e-04, 3.0613e-04, 5.0640e-04, 9.3794e-04, 2.9163e-03, 5.0392e-03,
          1.0872e-03, 1.3752e-03, 1.9293e-03, 2.1172e-03, 6.0987e-04, 1.8959e-03,
          2.4776e-03, 6.9580e-03, 2.1305e-03, 2.6169e-03, 2.8458e-03, 2.1591e-03,
          9.4128e-04, 1.2712e-03, 1.1497e-02, 9.2459e-04, 3.0785e-03, 2.7046e-03,
          1.0338e-03, 1.7214e-03, 3.0470e-04, 8.0633e-04, 7.4911e-04, 1.1177e-03,
          1.6384e-03, 6.9046e-04, 5.3930e-04, 7.8106e-04, 2.1076e-03, 2.8591e-03,
          2.9411e-03, 1.1330e-02, 2.5158e-03, 3.1643e-03, 8.0585e-04, 1.0986e-03,
          9.8801e-04, 6.9666e-04, 5.3835e-04, 4.8180e-03, 4.4785e-03, 7.5388e-04,
          5.7125e-04, 2.0504e-03, 2.9888e-03, 2.1744e-03, 7.1096e-04, 6.6795e-03,
          4.3488e-03, 6.6423e-04, 3.3131e-03, 1.5421e-03, 8.0719e-03, 6.6299e-03,
          1.3895e-03, 2.3193e-03, 1.3599e-03, 3.7708e-03, 4.6082e-03, 4.2114e-03,
          2.0084e-03, 1.3466e-02, 1.4200e-03, 1.1797e-03, 6.1798e-04, 7.5054e-04,
          1.4839e-03, 8.1205e-04, 2.6970e-03, 9.9468e-04, 2.8763e-03, 2.2488e-03,
          9.4910e-03, 1.3409e-03, 1.6079e-03, 1.1185e-02, 8.0872e-03, 2.8324e-03,
          2.2182e-03, 6.4888e-03, 3.8013e-03, 1.2884e-03, 6.6032e-03, 1.4515e-03,
          6.1417e-03, 1.0719e-02, 1.2636e-03, 5.0278e-03, 3.8414e-03, 1.2445e-03,
          5.1165e-04, 4.1924e-03, 1.5345e-03, 1.2980e-03, 5.5923e-03, 3.4180e-03,
          4.9324e-03, 4.9553e-03, 2.5620e-02, 7.1049e-05, 1.7738e-04, 7.1049e-05,
          3.7575e-04, 3.7575e-04, 1.7738e-04, 1.7738e-04, 1.7738e-04, 7.1049e-05,
          1.7738e-04, 1.7738e-04, 7.1049e-05, 1.7738e-04, 7.1049e-05, 7.1049e-05,
          3.7575e-04, 7.1049e-05, 1.7738e-04, 7.1049e-05, 3.7575e-04, 1.7738e-04,
          3.7575e-04, 3.7575e-04, 7.1049e-05, 1.7738e-04, 1.7738e-04, 1.7738e-04,
          3.7575e-04, 1.7738e-04, 1.7738e-04, 7.1049e-05, 7.1049e-05, 1.7738e-04,
          1.7738e-04, 1.7738e-04, 7.1049e-05, 1.7738e-04, 1.7738e-04, 7.1049e-05,
          1.7738e-04, 7.1049e-05, 7.1049e-05, 1.7738e-04, 1.7738e-04, 1.7738e-04,
          7.1049e-05, 1.7738e-04, 1.7738e-04, 1.7738e-04, 1.7738e-04, 5.1928e-04,
          1.7738e-04, 7.1049e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.4861e-03, 6.0059e-02, 1.1772e-02, 1.1513e-02, 3.3173e-02,
          4.4189e-02, 9.4452e-03, 1.1177e-03, 5.1498e-04, 2.6398e-03, 6.2103e-03,
          1.8854e-03, 1.2636e-03, 7.3099e-04, 2.3918e-03, 9.4938e-04, 2.3289e-03,
          7.9107e-04, 6.9427e-04, 2.0468e-04, 2.2340e-04, 2.6727e-04, 1.3447e-03,
          5.6839e-03, 2.6455e-03, 7.0496e-03, 2.7809e-03, 3.7193e-04, 1.5354e-03,
          1.0490e-03, 2.3994e-03, 4.5800e-04, 1.6499e-03, 2.6436e-03, 6.1798e-04,
          9.5463e-04, 3.9005e-04, 1.1740e-03, 1.5059e-03, 2.7966e-04, 6.4278e-04,
          3.6335e-04, 2.9898e-04, 3.5524e-04, 2.5153e-04, 1.7767e-03, 1.4219e-03,
          3.1233e-04, 3.3689e-04, 2.4643e-03, 7.2622e-04, 1.2245e-03, 8.6451e-04,
          3.2997e-04, 6.7997e-04, 3.3331e-04, 3.5810e-04, 1.0939e-03, 6.9046e-04,
          7.6294e-04, 4.9591e-04, 4.4775e-04, 4.6802e-04, 6.1941e-04, 2.7142e-03,
          1.3723e-03, 7.6628e-04, 6.0701e-04, 2.1553e-04, 1.9825e-04, 3.4499e-04,
          3.3331e-04, 4.8447e-03, 3.7670e-04, 1.9379e-03, 3.6192e-04, 2.7776e-04,
          8.3542e-04, 3.6478e-04, 6.8426e-04, 1.6661e-03, 7.3099e-04, 5.4884e-04,
          4.5490e-04, 1.9112e-03, 3.1400e-04, 2.6207e-03, 7.2861e-04, 5.6601e-04,
          3.8795e-03, 1.4963e-03, 2.7776e-04, 2.0847e-03, 2.8539e-04, 3.2377e-04,
          3.7980e-04, 2.5082e-04, 1.0815e-03, 1.3247e-03, 3.9220e-04, 6.2132e-04,
          4.7874e-03, 1.8902e-03, 9.1553e-04, 1.1244e-03, 3.1414e-03, 3.9196e-04,
          1.6203e-03, 3.3045e-04, 1.7796e-03, 4.4870e-04, 1.3142e-03, 9.8801e-04,
          3.0212e-03, 4.3488e-04, 2.5988e-04, 2.7676e-03, 4.2701e-04, 5.2500e-04,
          6.8188e-04, 1.7309e-03, 3.0136e-04, 9.5367e-04, 4.4289e-03, 3.0112e-04,
          2.8133e-04, 3.8624e-04, 6.3181e-04, 3.3073e-03, 2.5482e-03, 2.6817e-03,
          2.5578e-03, 1.0357e-03, 3.2501e-03, 4.4918e-04, 1.1539e-03, 2.9540e-04,
          2.9354e-03, 6.2275e-04, 4.6349e-03, 5.6458e-03, 3.5248e-03, 1.1139e-03,
          4.5013e-04, 5.9929e-03, 6.3038e-04, 5.2834e-04, 8.5640e-04, 3.3875e-03,
          3.4676e-03, 2.7447e-03, 6.0558e-04, 6.5613e-04, 9.0122e-04, 6.2609e-04,
          1.1091e-03, 5.3740e-04, 2.3918e-03, 3.3593e-04, 1.5326e-03, 8.9359e-04,
          2.3441e-03, 2.3212e-03, 2.9621e-03, 8.8959e-03, 5.7936e-04, 8.0204e-04,
          4.0817e-03, 1.1253e-03, 2.1763e-03, 5.1193e-03, 1.2903e-03, 1.9007e-03,
          1.7109e-03, 1.0204e-03, 8.8596e-04, 4.4465e-04, 1.4935e-03, 3.6106e-03,
          1.4820e-03, 1.8110e-03, 3.0937e-03, 3.0308e-03, 2.5959e-03, 7.7362e-03,
          5.7793e-03, 1.6136e-03, 9.9640e-03, 5.7077e-04, 8.6689e-04, 3.7460e-03,
          3.1490e-03, 2.8181e-04, 2.9678e-03, 3.7270e-03, 4.0627e-03, 2.2678e-03,
          1.1530e-03, 5.5504e-04, 2.4700e-03, 7.0381e-04, 2.0504e-03, 1.3762e-03,
          4.2653e-04, 8.7976e-04, 4.0960e-04, 2.5711e-03, 4.1771e-03, 8.5258e-04,
          2.1267e-03, 8.0156e-04, 5.0402e-04, 1.7872e-03, 3.5152e-03, 5.9843e-04,
          1.4477e-03, 1.4038e-03, 1.0490e-02, 8.5831e-03, 3.4256e-03, 1.6460e-03,
          7.8659e-03, 3.6573e-04, 8.6117e-04, 1.8225e-03, 6.8245e-03, 3.9787e-03,
          1.2274e-03, 6.9809e-04, 4.0936e-04, 1.6346e-03, 1.8158e-03, 9.3651e-04,
          7.7868e-04, 9.0456e-04, 1.0462e-03, 6.0654e-03, 1.5068e-03, 3.9291e-03,
          1.2178e-03, 1.0853e-03, 6.1607e-04, 1.1272e-03, 2.2125e-03, 8.5211e-04,
          4.3249e-04, 1.4200e-03, 4.6682e-04, 8.8549e-04, 8.5783e-04, 2.1572e-03,
          4.3583e-04, 2.9063e-04, 4.9686e-04, 6.3848e-04, 5.0163e-03, 5.0850e-03,
          6.9475e-04, 1.5516e-03, 1.4324e-03, 2.1172e-03, 5.2500e-04, 1.9722e-03,
          2.1763e-03, 7.8506e-03, 3.9558e-03, 4.2610e-03, 8.6899e-03, 1.5192e-03,
          1.8015e-03, 1.4038e-03, 3.5744e-03, 6.8569e-04, 2.0370e-03, 2.8992e-03,
          1.3342e-03, 2.8191e-03, 4.5967e-04, 1.9236e-03, 6.3133e-04, 1.2808e-03,
          2.2526e-03, 6.5374e-04, 4.4298e-04, 6.7043e-04, 2.0466e-03, 2.6741e-03,
          3.2501e-03, 5.9776e-03, 2.5463e-03, 4.9400e-03, 6.5756e-04, 1.0986e-03,
          8.3303e-04, 5.2977e-04, 4.2796e-04, 3.9864e-03, 2.0676e-03, 9.0933e-04,
          5.7983e-04, 4.1351e-03, 4.8103e-03, 2.6455e-03, 7.1955e-04, 3.8967e-03,
          4.2191e-03, 8.3113e-04, 2.2888e-03, 2.0714e-03, 1.4160e-02, 8.3694e-03,
          1.9217e-03, 2.4281e-03, 1.4057e-03, 5.1765e-03, 4.7150e-03, 3.6087e-03,
          3.3627e-03, 1.1574e-02, 1.7633e-03, 1.4896e-03, 6.4135e-04, 6.6423e-04,
          3.1185e-03, 1.6098e-03, 4.6768e-03, 8.6641e-04, 2.3880e-03, 3.9101e-03,
          4.9706e-03, 1.6184e-03, 2.2831e-03, 7.8735e-03, 1.3954e-02, 4.2038e-03,
          2.5101e-03, 9.0408e-03, 5.0087e-03, 1.4544e-03, 2.9545e-03, 2.4071e-03,
          1.0292e-02, 8.3847e-03, 2.5978e-03, 7.6218e-03, 4.8904e-03, 1.6365e-03,
          7.5674e-04, 6.7749e-03, 1.5974e-03, 1.2960e-03, 6.2828e-03, 4.1618e-03,
          4.9210e-03, 7.7858e-03, 7.5188e-03, 5.1453e-02, 1.7691e-04, 1.7691e-04,
          1.7691e-04, 1.7691e-04, 1.7691e-04, 5.2989e-05, 1.7691e-04, 5.2989e-05,
          5.2989e-05, 1.7691e-04, 5.2989e-05, 1.7691e-04, 5.2989e-05, 5.2989e-05,
          1.7691e-04, 5.2989e-05, 1.7691e-04, 5.2989e-05, 1.7691e-04, 5.2989e-05,
          1.7691e-04, 1.7691e-04, 5.2989e-05, 1.7691e-04, 5.2989e-05, 1.7691e-04,
          1.7691e-04, 1.7691e-04, 1.7691e-04, 5.2989e-05, 5.2989e-05, 1.7691e-04,
          1.7691e-04, 1.7691e-04, 5.2989e-05, 1.7691e-04, 1.7691e-04, 5.2989e-05,
          1.7691e-04, 5.2989e-05, 5.2989e-05, 1.7691e-04, 1.7691e-04, 5.2989e-05,
          5.2989e-05, 1.7691e-04, 1.7691e-04, 1.7691e-04, 1.7691e-04, 1.7691e-04,
          5.2989e-05, 5.2989e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1575e-03, 4.2023e-02, 1.3016e-02, 2.2675e-02, 4.0497e-02,
          2.7084e-02, 1.3336e-02, 8.1158e-04, 3.3808e-04, 7.7200e-04, 1.4915e-03,
          5.5742e-04, 1.4639e-03, 4.2953e-03, 6.0511e-04, 6.9094e-04, 3.7689e-03,
          2.8057e-03, 3.2067e-04, 5.2738e-04, 3.4642e-04, 5.2452e-04, 1.6747e-03,
          1.5783e-03, 5.3835e-04, 1.0895e-02, 5.5885e-03, 9.9087e-04, 2.6512e-03,
          8.8425e-03, 1.5678e-03, 1.1244e-03, 8.2302e-04, 6.0511e-04, 6.1607e-04,
          6.3181e-04, 4.1223e-04, 7.4387e-04, 6.4754e-04, 1.1053e-03, 4.4107e-04,
          3.8767e-04, 2.7490e-04, 4.0293e-04, 1.8573e-04, 3.1204e-03, 8.4639e-04,
          1.7452e-03, 1.0233e-03, 3.5191e-03, 9.8038e-04, 1.1501e-03, 5.0497e-04,
          4.9257e-04, 7.4244e-04, 6.1178e-04, 3.4738e-04, 8.9550e-04, 5.4502e-04,
          7.0953e-04, 2.7618e-03, 7.9441e-04, 5.3549e-04, 3.6931e-04, 1.0395e-03,
          9.2411e-04, 3.2544e-04, 4.3082e-04, 1.7130e-04, 1.7512e-04, 3.4070e-04,
          3.9458e-04, 1.7452e-03, 7.7581e-04, 1.6203e-03, 5.5218e-04, 3.8362e-04,
          1.1768e-03, 3.3903e-04, 5.7125e-04, 5.4932e-04, 7.3576e-04, 3.8528e-04,
          4.4560e-04, 1.7576e-03, 7.4959e-04, 5.8985e-04, 4.7421e-04, 7.9155e-04,
          1.1492e-03, 5.0020e-04, 2.3055e-04, 7.4339e-04, 2.3985e-04, 3.7813e-04,
          1.5707e-03, 4.7636e-04, 1.2674e-03, 7.2479e-04, 3.5620e-04, 3.5810e-04,
          1.2417e-03, 3.9268e-04, 4.8280e-04, 5.6410e-04, 7.6485e-04, 2.6684e-03,
          1.3018e-03, 3.5000e-04, 7.5102e-04, 2.9254e-04, 6.4898e-04, 4.9114e-04,
          3.3092e-04, 3.1829e-04, 9.7275e-04, 1.1864e-03, 5.8460e-04, 1.8654e-03,
          9.2087e-03, 6.3086e-04, 2.5773e-04, 5.0688e-04, 1.0328e-03, 2.8539e-04,
          2.4700e-04, 5.1165e-04, 6.1750e-04, 1.0414e-03, 1.1444e-03, 1.5812e-03,
          8.3590e-04, 5.3358e-04, 2.7156e-04, 2.7204e-04, 5.2309e-04, 2.9874e-04,
          3.7537e-03, 4.5280e-03, 2.7370e-03, 2.5578e-03, 1.7900e-03, 9.8495e-03,
          3.9458e-04, 1.7905e-04, 3.2401e-04, 3.0065e-04, 1.0328e-03, 1.0099e-03,
          1.2236e-03, 9.5987e-04, 4.7398e-04, 1.0929e-03, 1.4887e-03, 5.8889e-04,
          5.2738e-04, 3.9291e-04, 3.6931e-04, 5.6267e-04, 9.6512e-04, 4.5700e-03,
          1.0109e-03, 7.2241e-04, 5.1651e-03, 1.0262e-03, 6.8760e-04, 4.1437e-04,
          9.3079e-04, 4.3702e-04, 6.4373e-04, 4.8304e-04, 4.3602e-03, 1.0128e-03,
          6.8521e-04, 7.0095e-04, 1.0204e-03, 8.2874e-04, 7.6342e-04, 1.6460e-03,
          5.5408e-04, 6.2084e-04, 1.0424e-03, 1.4153e-03, 7.4434e-04, 5.5361e-04,
          9.0885e-04, 6.8092e-03, 9.3365e-04, 6.1989e-04, 1.3189e-03, 1.0624e-03,
          1.3208e-03, 2.7609e-04, 8.6927e-04, 5.7144e-03, 8.2207e-04, 9.4986e-04,
          1.0166e-03, 2.7409e-03, 1.2379e-03, 6.8521e-04, 7.3338e-04, 5.1355e-04,
          7.6866e-04, 6.9714e-04, 3.8695e-04, 5.0449e-04, 1.3428e-03, 1.3609e-03,
          8.8835e-04, 1.2760e-03, 8.2302e-04, 7.8154e-04, 1.5001e-03, 7.6437e-04,
          7.3576e-04, 4.3259e-03, 1.4725e-03, 4.6492e-04, 1.6470e-03, 1.1215e-02,
          3.0155e-03, 5.3835e-04, 6.9809e-04, 7.2908e-04, 1.9007e-03, 1.1349e-03,
          7.8773e-04, 5.1737e-04, 3.2425e-04, 6.2418e-04, 7.1335e-04, 5.1231e-03,
          1.6460e-03, 6.3181e-04, 7.3862e-04, 5.3024e-04, 1.1606e-03, 2.2907e-03,
          1.5469e-03, 1.4534e-03, 6.1035e-04, 1.3685e-03, 1.4542e-02, 1.1475e-02,
          8.2541e-04, 9.8228e-04, 4.9591e-04, 4.6706e-04, 7.6151e-04, 9.5606e-04,
          5.4836e-04, 3.7074e-04, 7.4196e-04, 5.5742e-04, 2.2259e-03, 3.2349e-03,
          1.0090e-03, 1.0300e-03, 6.0129e-04, 7.1716e-04, 4.9639e-04, 1.7138e-03,
          1.0748e-03, 6.6233e-04, 3.5114e-03, 3.1525e-02, 3.1464e-02, 8.9693e-04,
          1.5793e-03, 2.0161e-03, 1.4400e-03, 7.0667e-04, 1.1272e-03, 1.2302e-03,
          7.8535e-04, 4.9925e-04, 2.6779e-03, 2.4673e-02, 1.2188e-03, 1.7614e-03,
          1.4772e-03, 8.3399e-04, 6.8569e-04, 8.5640e-04, 3.3512e-03, 1.0090e-03,
          1.3046e-03, 2.2640e-03, 3.6869e-03, 4.7722e-03, 1.2007e-03, 1.3504e-03,
          1.3533e-03, 8.3447e-04, 5.7268e-04, 1.4963e-03, 8.9884e-04, 1.0948e-03,
          8.6641e-04, 1.7157e-03, 1.6861e-03, 1.0551e-02, 1.4858e-03, 1.5554e-03,
          1.6727e-03, 3.6240e-03, 1.7271e-03, 3.0193e-03, 5.6410e-04, 7.4196e-04,
          3.7670e-03, 1.2274e-03, 8.8654e-03, 2.8191e-03, 4.4518e-03, 1.7471e-03,
          8.2493e-04, 2.6493e-03, 4.2496e-03, 4.0436e-03, 1.1730e-03, 1.2903e-03,
          3.8967e-03, 7.9346e-03, 1.0902e-02, 2.3003e-03, 2.6398e-03, 4.6173e-02,
          2.7752e-03, 3.6545e-03, 4.2908e-02, 5.7411e-03, 1.0395e-03, 4.9400e-03,
          9.6416e-04, 3.1891e-03, 1.9512e-03, 1.3496e-02, 1.7862e-03, 3.5431e-02,
          1.0147e-03, 1.2074e-03, 9.8419e-03, 9.6083e-04, 2.4433e-03, 4.6806e-03,
          7.1411e-03, 9.9030e-03, 3.3092e-03, 2.5520e-03, 5.7526e-03, 2.3022e-03,
          2.5902e-03, 1.6861e-03, 5.5265e-04, 6.6137e-04, 2.6733e-02, 1.7905e-04,
          1.0023e-03, 2.9325e-04, 8.8406e-04, 6.1154e-05, 1.7524e-04, 1.7905e-04,
          1.7905e-04, 6.1154e-05, 6.1154e-05, 6.1154e-05, 6.1154e-05, 6.1154e-05,
          6.1154e-05, 6.1154e-05, 6.1154e-05, 6.1154e-05, 1.7905e-04, 6.1154e-05,
          8.8406e-04, 6.1154e-05, 1.7905e-04, 6.1154e-05, 6.1154e-05, 6.1154e-05,
          1.1168e-03, 6.1154e-05, 6.1154e-05, 6.1154e-05, 1.7905e-04, 4.3464e-04,
          6.1154e-05, 9.9850e-04, 6.1154e-05, 6.1154e-05, 6.1154e-05, 6.1154e-05,
          7.0524e-04, 5.6410e-04, 2.9421e-04, 3.5667e-04, 2.9421e-04, 1.7524e-04,
          1.7524e-04, 2.9325e-04, 1.7524e-04, 2.8133e-04, 1.7524e-04, 6.1154e-05,
          2.9421e-04, 1.7524e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.1733e-03, 5.2521e-02, 1.0979e-02, 1.3184e-02, 2.8061e-02,
          4.4159e-02, 8.4534e-03, 1.0872e-03, 5.1212e-04, 1.8721e-03, 2.6073e-03,
          1.3752e-03, 8.1921e-04, 3.9864e-04, 2.0332e-03, 8.2827e-04, 8.3113e-04,
          4.3654e-04, 4.2796e-04, 2.0754e-04, 2.2447e-04, 1.6963e-04, 7.5722e-04,
          3.4428e-03, 2.2736e-03, 6.8817e-03, 3.0880e-03, 2.8872e-04, 8.0729e-04,
          4.7469e-04, 2.5654e-03, 3.2520e-04, 2.4109e-03, 1.5316e-03, 1.0748e-03,
          5.0316e-03, 1.4153e-03, 1.1435e-03, 1.6718e-03, 2.8348e-04, 1.0433e-03,
          4.6110e-04, 3.8171e-04, 4.4489e-04, 3.5954e-04, 7.9155e-04, 2.3060e-03,
          2.3150e-04, 2.9445e-04, 1.7881e-03, 2.2068e-03, 2.1877e-03, 8.8882e-04,
          5.5313e-04, 1.4763e-03, 9.8228e-04, 1.3313e-03, 5.5923e-03, 4.2610e-03,
          4.5929e-03, 6.0034e-04, 1.1826e-03, 1.4124e-03, 6.6566e-04, 5.6190e-03,
          7.5865e-04, 4.8494e-04, 5.4073e-04, 2.9945e-04, 2.4915e-04, 4.1175e-04,
          4.6897e-04, 3.0384e-03, 8.5592e-04, 3.0136e-03, 5.6314e-04, 5.6362e-04,
          2.1610e-03, 8.4352e-04, 7.2241e-04, 4.0016e-03, 1.1854e-03, 6.7177e-03,
          2.8439e-03, 4.9896e-03, 8.5306e-04, 1.0471e-03, 4.0746e-04, 4.1628e-04,
          2.5520e-03, 1.4839e-03, 3.2234e-04, 1.7614e-03, 6.9952e-04, 6.7806e-04,
          4.0936e-04, 4.5538e-04, 2.3060e-03, 2.7447e-03, 8.3160e-04, 1.1272e-03,
          3.6335e-03, 8.2016e-04, 9.2649e-04, 2.5528e-02, 9.5291e-03, 5.1785e-04,
          2.9583e-03, 5.6934e-04, 1.9932e-03, 4.8590e-04, 6.1893e-04, 6.3658e-04,
          2.2984e-03, 1.0262e-03, 2.4939e-04, 5.3368e-03, 8.7690e-04, 5.7554e-04,
          6.2084e-04, 3.3054e-03, 4.3082e-04, 9.4175e-04, 6.3324e-03, 3.9005e-04,
          6.2418e-04, 6.5374e-04, 8.6498e-04, 7.1678e-03, 1.4181e-03, 3.2291e-03,
          1.0214e-03, 4.7636e-04, 7.4863e-04, 3.2091e-04, 1.1778e-03, 3.6931e-04,
          1.7767e-03, 3.6144e-04, 1.2379e-03, 2.6932e-03, 2.3651e-03, 7.1383e-04,
          6.1560e-04, 1.9684e-03, 8.4972e-04, 1.1539e-03, 7.0238e-04, 9.3155e-03,
          6.3057e-03, 8.5449e-03, 3.0537e-03, 2.5253e-03, 2.1610e-03, 1.5450e-03,
          1.4296e-03, 1.2646e-03, 7.0763e-04, 4.6611e-04, 3.7899e-03, 6.3896e-04,
          3.0899e-03, 2.1400e-03, 1.1759e-03, 3.2463e-03, 1.2550e-03, 1.4353e-03,
          2.5749e-03, 1.5535e-03, 3.2616e-03, 2.4700e-03, 7.3099e-04, 4.2000e-03,
          7.7705e-03, 3.3531e-03, 4.6768e-03, 9.2983e-04, 3.4924e-03, 1.5202e-03,
          1.7548e-03, 1.8501e-03, 3.6678e-03, 1.8644e-03, 1.3580e-03, 1.6918e-03,
          2.2202e-03, 6.5184e-04, 1.9684e-03, 8.7452e-04, 7.0906e-04, 3.5858e-03,
          2.2697e-03, 4.5657e-04, 2.9125e-03, 1.3771e-03, 2.3575e-03, 4.2076e-03,
          5.3864e-03, 7.6389e-04, 4.3755e-03, 6.7902e-04, 8.6832e-04, 1.8597e-03,
          5.6934e-04, 1.4553e-03, 6.4707e-04, 1.3771e-03, 3.0270e-03, 5.7125e-04,
          1.5354e-03, 1.3914e-03, 8.1921e-04, 1.1806e-03, 4.5929e-03, 7.6580e-04,
          3.0231e-03, 1.2512e-03, 2.6665e-03, 2.7637e-03, 3.6278e-03, 9.5892e-04,
          2.8248e-03, 7.8344e-04, 2.2030e-03, 9.0456e-04, 7.4577e-03, 2.7084e-03,
          6.0940e-04, 7.5531e-04, 7.1955e-04, 3.8242e-03, 3.9864e-03, 6.0892e-04,
          1.1282e-03, 1.1244e-03, 9.1648e-04, 5.0049e-03, 3.4065e-03, 5.3215e-03,
          2.2335e-03, 1.7195e-03, 1.3714e-03, 1.9836e-03, 1.7433e-03, 7.4100e-04,
          9.9564e-04, 1.0519e-03, 9.6416e-04, 2.0676e-03, 1.4153e-03, 5.4817e-03,
          4.4632e-04, 5.8126e-04, 8.5211e-04, 1.4791e-03, 1.9588e-03, 6.2752e-03,
          1.5125e-03, 1.2732e-03, 4.6539e-03, 3.8528e-03, 1.0481e-03, 2.9716e-03,
          3.9139e-03, 2.3956e-03, 1.5297e-03, 2.0866e-03, 2.0313e-03, 4.2419e-03,
          1.4925e-03, 7.7534e-04, 7.4692e-03, 1.6022e-03, 3.3646e-03, 4.6196e-03,
          3.0727e-03, 4.4441e-03, 4.1294e-04, 7.9870e-04, 9.0885e-04, 1.2112e-03,
          9.8610e-04, 8.5735e-04, 6.6566e-04, 1.0920e-03, 2.6550e-03, 4.1504e-03,
          5.2376e-03, 4.8256e-03, 1.5545e-03, 2.1019e-03, 1.2770e-03, 2.5654e-03,
          1.4296e-03, 8.0633e-04, 6.7234e-04, 3.2940e-03, 4.0588e-03, 1.3199e-03,
          9.1934e-04, 1.6193e-03, 3.4103e-03, 2.0332e-03, 8.7500e-04, 4.6349e-03,
          4.5280e-03, 8.6021e-04, 3.3703e-03, 1.4496e-03, 4.2686e-03, 3.5324e-03,
          1.3342e-03, 6.8626e-03, 1.6308e-03, 3.4027e-03, 3.8662e-03, 4.0665e-03,
          3.8490e-03, 5.0583e-03, 1.5907e-03, 1.0910e-03, 1.1120e-03, 1.1768e-03,
          1.3218e-03, 9.9087e-04, 1.8320e-03, 1.3552e-03, 4.4289e-03, 1.9102e-03,
          5.4893e-03, 1.1435e-03, 1.5192e-03, 4.7874e-03, 2.7637e-03, 3.6469e-03,
          7.3700e-03, 1.0078e-02, 5.6992e-03, 1.4191e-03, 3.4161e-03, 1.4143e-03,
          3.0346e-03, 4.1161e-03, 1.2665e-03, 2.1286e-03, 8.3084e-03, 1.4219e-03,
          5.8699e-04, 3.4847e-03, 2.2202e-03, 1.7166e-03, 6.8932e-03, 3.9940e-03,
          1.0117e-02, 2.9392e-03, 4.8027e-03, 4.2076e-03, 1.0271e-03, 3.8116e-02,
          3.7479e-04, 3.7479e-04, 1.6129e-04, 5.7817e-05, 3.7479e-04, 5.7817e-05,
          5.7817e-05, 1.6129e-04, 1.6129e-04, 5.7817e-05, 5.7817e-05, 5.7817e-05,
          3.7479e-04, 5.7817e-05, 5.7817e-05, 5.7817e-05, 1.6129e-04, 3.7479e-04,
          3.7479e-04, 1.6129e-04, 1.6129e-04, 1.6129e-04, 5.7817e-05, 3.7479e-04,
          3.7479e-04, 3.7479e-04, 1.6129e-04, 5.7817e-05, 5.7817e-05, 1.6129e-04,
          3.7479e-04, 3.7479e-04, 5.7817e-05, 3.7479e-04, 1.6129e-04, 5.7817e-05,
          6.4230e-04, 1.6129e-04, 4.2820e-04, 1.6129e-04, 6.4230e-04, 1.6129e-04,
          1.6129e-04, 1.6129e-04, 1.6129e-04, 8.4162e-04, 6.0368e-04, 8.1778e-04,
          6.4230e-04, 1.6129e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.2139e-03, 5.0232e-02, 1.1589e-02, 1.6129e-02, 2.9068e-02,
          3.6224e-02, 8.0566e-03, 5.5733e-03, 1.0166e-03, 1.0405e-03, 1.8148e-03,
          6.7711e-04, 9.8133e-04, 7.1144e-04, 8.6308e-04, 5.8670e-03, 1.7128e-03,
          1.1396e-03, 9.4509e-04, 3.1400e-04, 3.1924e-04, 3.3522e-04, 1.3103e-03,
          2.1667e-03, 7.5960e-04, 7.3738e-03, 3.8891e-03, 5.5075e-04, 1.8749e-03,
          8.2445e-04, 1.5671e-02, 8.0490e-04, 1.3018e-03, 8.8215e-04, 5.0774e-03,
          1.1177e-03, 5.2357e-04, 1.2407e-03, 1.1282e-03, 4.6468e-04, 7.8297e-04,
          4.5228e-04, 3.2640e-04, 3.8910e-04, 2.4939e-04, 1.5135e-03, 1.2550e-03,
          4.2725e-04, 4.1103e-04, 2.3670e-03, 8.5068e-03, 1.5068e-02, 2.2373e-03,
          1.8969e-03, 7.8430e-03, 2.8057e-03, 5.4836e-04, 2.0618e-03, 8.1205e-04,
          8.8024e-04, 1.2083e-03, 5.7030e-03, 3.3894e-03, 4.3106e-04, 1.5221e-03,
          6.4373e-04, 3.3903e-04, 4.5061e-04, 2.3162e-04, 1.9813e-04, 3.5405e-04,
          4.5204e-04, 2.1648e-03, 3.9368e-03, 2.0462e-02, 2.3441e-03, 2.8553e-03,
          7.3738e-03, 4.2534e-04, 4.5180e-04, 8.4925e-04, 6.2275e-04, 6.0272e-04,
          5.6505e-04, 2.9297e-02, 3.6697e-03, 8.1253e-04, 3.1877e-04, 3.2926e-04,
          1.6537e-03, 1.0958e-03, 2.0778e-04, 1.0519e-03, 3.2234e-04, 4.4394e-04,
          1.1921e-03, 1.8158e-03, 1.0910e-02, 1.6413e-03, 4.0889e-04, 4.8232e-04,
          1.5697e-03, 3.8671e-04, 5.0163e-04, 1.0281e-03, 1.5316e-03, 7.2575e-04,
          2.0050e-02, 5.0688e-04, 1.2579e-03, 3.3760e-04, 4.3678e-04, 4.4799e-04,
          5.4455e-04, 5.2547e-04, 3.8147e-04, 1.9188e-03, 6.2037e-04, 8.8120e-04,
          1.0767e-03, 9.8324e-04, 2.3901e-04, 2.7580e-03, 1.5306e-03, 2.2161e-04,
          2.9731e-04, 4.0531e-04, 5.0354e-04, 1.8053e-03, 4.6349e-03, 2.0905e-02,
          8.3876e-04, 3.8671e-04, 3.8862e-04, 2.6488e-04, 7.4816e-04, 3.0565e-04,
          2.7447e-03, 6.7949e-04, 1.5135e-03, 3.3875e-03, 2.0332e-03, 1.2894e-03,
          3.1996e-04, 2.8920e-04, 3.7074e-04, 4.7231e-04, 4.7874e-04, 1.3990e-03,
          2.2030e-03, 1.4467e-03, 8.2827e-04, 1.0643e-02, 1.2970e-02, 1.0433e-03,
          8.9264e-04, 8.8882e-04, 5.7888e-04, 3.7003e-04, 1.4830e-03, 1.2589e-03,
          1.6069e-03, 1.5011e-03, 2.4509e-03, 1.3943e-03, 6.8617e-04, 5.9271e-04,
          1.0834e-03, 7.0906e-04, 8.0967e-04, 6.1131e-04, 1.4868e-03, 1.6747e-03,
          1.5488e-03, 1.4458e-03, 1.2255e-03, 1.0834e-03, 1.2093e-03, 2.6207e-03,
          6.4898e-04, 8.1682e-04, 1.2980e-03, 9.8190e-03, 1.9493e-03, 7.0095e-04,
          1.3113e-03, 1.1683e-03, 1.2083e-03, 4.9496e-04, 4.9067e-04, 1.2512e-03,
          1.0910e-02, 2.5105e-04, 1.0118e-03, 2.5673e-03, 1.0986e-03, 1.2283e-03,
          1.5287e-03, 1.0557e-03, 1.4801e-03, 5.5170e-04, 7.7915e-04, 7.2479e-04,
          4.3273e-04, 7.6008e-04, 4.0865e-04, 1.8940e-03, 1.8034e-03, 1.1110e-03,
          6.0692e-03, 1.0395e-03, 5.8317e-04, 7.0810e-04, 1.5411e-03, 4.4656e-04,
          1.2016e-03, 1.8101e-03, 1.7490e-03, 7.1955e-04, 2.1629e-03, 2.5406e-03,
          3.7270e-03, 6.2895e-04, 8.1062e-04, 6.2513e-04, 2.3937e-03, 1.2312e-03,
          5.5265e-04, 5.3692e-04, 3.6931e-04, 1.1225e-03, 1.0595e-03, 1.1988e-03,
          1.1015e-03, 6.6185e-04, 5.1785e-04, 8.9073e-04, 1.2159e-03, 2.1324e-03,
          1.1024e-03, 9.9087e-04, 7.2289e-04, 2.1324e-03, 3.0460e-03, 1.3142e-03,
          9.7418e-04, 8.7786e-04, 6.2513e-04, 7.0572e-04, 8.4400e-04, 1.0557e-03,
          3.7074e-04, 3.3975e-04, 6.2752e-04, 7.6580e-04, 1.8845e-03, 1.0292e-02,
          8.8215e-04, 6.6948e-04, 8.5211e-04, 8.7738e-04, 4.6206e-04, 1.2150e-03,
          1.1950e-03, 8.9359e-04, 1.5621e-03, 4.5128e-03, 3.2349e-03, 1.1005e-03,
          1.2913e-03, 7.5436e-04, 1.5011e-03, 6.4087e-04, 1.7881e-03, 1.3094e-03,
          1.1234e-03, 1.1034e-03, 7.2527e-04, 1.5898e-03, 8.4114e-04, 8.9741e-04,
          8.6021e-04, 5.6362e-04, 4.3797e-04, 6.2370e-04, 2.0332e-03, 1.2579e-03,
          1.6747e-03, 2.2697e-03, 2.4815e-03, 2.7294e-03, 1.1063e-03, 1.5736e-03,
          1.0900e-03, 6.2799e-04, 4.8494e-04, 1.9121e-03, 1.7824e-03, 9.6178e-04,
          6.2418e-04, 1.5879e-03, 1.0956e-02, 3.4580e-03, 6.8617e-04, 2.0714e-03,
          1.6432e-03, 1.2245e-03, 1.8711e-03, 1.8063e-03, 1.0290e-03, 1.3723e-03,
          1.7490e-03, 2.3918e-03, 2.2030e-03, 3.4828e-03, 3.4657e-03, 1.6375e-03,
          1.5059e-03, 3.5343e-03, 2.3270e-03, 2.0313e-03, 8.2970e-04, 8.2445e-04,
          1.9150e-03, 1.8873e-03, 3.0804e-03, 9.3412e-04, 1.7872e-03, 3.5419e-03,
          3.4332e-03, 1.7281e-03, 2.7885e-03, 4.8141e-03, 1.5059e-03, 3.7880e-03,
          2.1744e-03, 1.2367e-02, 1.2741e-02, 2.2831e-03, 2.3251e-03, 2.6493e-03,
          1.4668e-03, 1.4324e-03, 2.6169e-03, 1.6680e-03, 1.6281e-02, 2.1687e-03,
          1.1148e-03, 4.9057e-03, 1.4877e-03, 1.1225e-03, 3.5797e-02, 1.6830e-02,
          1.4328e-02, 2.4261e-03, 8.8739e-04, 9.6893e-04, 1.6956e-03, 1.9512e-03,
          4.6997e-02, 8.0824e-05, 8.0824e-05, 8.0824e-05, 8.0824e-05, 8.0824e-05,
          2.1291e-04, 8.0824e-05, 8.0824e-05, 2.1291e-04, 8.0824e-05, 8.0824e-05,
          2.1291e-04, 8.0824e-05, 8.0824e-05, 8.0824e-05, 8.0824e-05, 2.7275e-03,
          8.0824e-05, 3.7122e-04, 3.9721e-04, 2.1291e-04, 8.0824e-05, 2.7275e-03,
          8.0824e-05, 8.0824e-05, 3.7122e-04, 8.0824e-05, 2.6512e-04, 2.1291e-04,
          4.0221e-04, 8.0824e-05, 8.0824e-05, 8.0824e-05, 8.0824e-05, 8.0824e-05,
          2.1291e-04, 2.1291e-04, 2.1291e-04, 2.1291e-04, 2.9254e-04, 2.1291e-04,
          8.0824e-05, 2.1291e-04, 2.1291e-04, 4.2057e-04, 2.8849e-04, 2.1291e-04,
          2.1291e-04, 2.1291e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.1528e-03, 4.9561e-02, 1.2268e-02, 1.5503e-02, 2.8564e-02,
          3.5400e-02, 7.7972e-03, 5.2338e-03, 1.1206e-03, 1.1206e-03, 1.8473e-03,
          6.9904e-04, 1.0777e-03, 5.9366e-04, 9.5844e-04, 6.1760e-03, 1.8110e-03,
          1.3523e-03, 1.1358e-03, 2.3305e-04, 2.6941e-04, 3.0375e-04, 1.7176e-03,
          2.2583e-03, 7.7963e-04, 6.9237e-03, 4.6806e-03, 4.4322e-04, 2.1286e-03,
          8.2445e-04, 1.4763e-02, 7.3290e-04, 1.4668e-03, 1.1282e-03, 4.5280e-03,
          8.1730e-04, 3.5739e-04, 1.2245e-03, 1.3781e-03, 3.1281e-04, 5.6648e-04,
          3.6478e-04, 2.6822e-04, 3.3355e-04, 1.9205e-04, 1.7843e-03, 1.2379e-03,
          3.7265e-04, 3.2735e-04, 2.2850e-03, 5.6229e-03, 1.3359e-02, 2.3499e-03,
          1.4200e-03, 6.7596e-03, 1.4515e-03, 3.1614e-04, 1.3685e-03, 5.1928e-04,
          6.2418e-04, 9.7370e-04, 4.0703e-03, 2.7275e-03, 3.7026e-04, 1.4734e-03,
          6.4087e-04, 3.5119e-04, 4.9067e-04, 1.6582e-04, 1.5295e-04, 2.7442e-04,
          3.3712e-04, 2.5387e-03, 2.0847e-03, 1.7197e-02, 1.5011e-03, 2.1229e-03,
          5.0812e-03, 2.6488e-04, 3.5954e-04, 7.4291e-04, 4.9973e-04, 4.2224e-04,
          4.0245e-04, 2.5558e-02, 2.5787e-03, 9.2745e-04, 3.2234e-04, 3.0041e-04,
          1.8435e-03, 1.0567e-03, 1.8489e-04, 1.0662e-03, 2.6536e-04, 3.1757e-04,
          7.9823e-04, 8.7643e-04, 1.0277e-02, 1.3494e-03, 2.7156e-04, 3.4595e-04,
          1.6184e-03, 3.6287e-04, 4.5109e-04, 6.3515e-04, 1.4000e-03, 5.9938e-04,
          1.7868e-02, 3.7932e-04, 1.1549e-03, 2.6989e-04, 4.3273e-04, 3.9434e-04,
          5.3835e-04, 3.7527e-04, 3.3712e-04, 1.9407e-03, 4.7326e-04, 7.5245e-04,
          9.2554e-04, 8.2159e-04, 1.8561e-04, 3.1548e-03, 1.4639e-03, 1.6737e-04,
          1.8942e-04, 2.8539e-04, 3.8242e-04, 1.7605e-03, 5.6839e-03, 1.8967e-02,
          8.1110e-04, 4.0150e-04, 4.5371e-04, 2.6536e-04, 7.2718e-04, 2.3651e-04,
          3.2024e-03, 6.4278e-04, 1.8911e-03, 4.1428e-03, 2.3689e-03, 1.2178e-03,
          2.5511e-04, 3.0351e-04, 3.0446e-04, 3.2902e-04, 4.3249e-04, 1.2865e-03,
          1.9283e-03, 1.3084e-03, 5.3883e-04, 7.5798e-03, 8.3160e-03, 7.5531e-04,
          7.5865e-04, 6.1989e-04, 7.6103e-04, 3.3021e-04, 1.3046e-03, 1.2751e-03,
          1.4353e-03, 1.4105e-03, 3.3550e-03, 1.5841e-03, 4.8208e-04, 4.3631e-04,
          1.0748e-03, 5.3358e-04, 7.5340e-04, 6.9332e-04, 1.8778e-03, 1.4124e-03,
          1.2798e-03, 1.1454e-03, 8.0347e-04, 8.7500e-04, 1.1225e-03, 3.7060e-03,
          6.0797e-04, 9.9564e-04, 1.3046e-03, 1.3069e-02, 2.4319e-03, 7.6580e-04,
          1.5278e-03, 1.2989e-03, 1.2703e-03, 3.8624e-04, 4.5395e-04, 1.3437e-03,
          1.3062e-02, 1.7726e-04, 1.0309e-03, 3.3417e-03, 1.1635e-03, 1.1921e-03,
          1.0080e-03, 7.9775e-04, 1.6766e-03, 5.4836e-04, 8.0633e-04, 7.0286e-04,
          4.0960e-04, 7.4291e-04, 3.7169e-04, 2.1019e-03, 1.8778e-03, 1.2751e-03,
          6.5842e-03, 7.8964e-04, 4.6301e-04, 7.1335e-04, 1.5621e-03, 3.6716e-04,
          9.2459e-04, 1.8044e-03, 2.1439e-03, 8.0347e-04, 2.4033e-03, 2.5291e-03,
          3.3054e-03, 5.4026e-04, 7.6485e-04, 6.3848e-04, 2.5730e-03, 1.5497e-03,
          5.9986e-04, 5.3740e-04, 3.5906e-04, 1.1148e-03, 1.1377e-03, 1.2589e-03,
          8.8501e-04, 5.6314e-04, 4.9210e-04, 9.3937e-04, 1.0300e-03, 2.1191e-03,
          9.6035e-04, 8.3494e-04, 5.2261e-04, 1.5488e-03, 3.0766e-03, 1.2026e-03,
          7.8440e-04, 1.0386e-03, 5.5790e-04, 6.9809e-04, 8.4639e-04, 1.2197e-03,
          3.5572e-04, 3.1137e-04, 5.9223e-04, 7.0000e-04, 2.5406e-03, 9.7580e-03,
          6.4898e-04, 6.3324e-04, 7.4053e-04, 8.5592e-04, 3.7766e-04, 9.8419e-04,
          1.2808e-03, 1.0519e-03, 1.9178e-03, 5.4893e-03, 4.5242e-03, 1.2369e-03,
          1.3523e-03, 8.5449e-04, 1.7538e-03, 6.2275e-04, 1.5993e-03, 1.4219e-03,
          1.1969e-03, 1.1101e-03, 6.8331e-04, 1.9989e-03, 6.5947e-04, 8.5354e-04,
          9.8515e-04, 4.6492e-04, 3.2759e-04, 5.0306e-04, 1.9531e-03, 1.2503e-03,
          1.6365e-03, 2.5005e-03, 3.2310e-03, 3.1528e-03, 1.0357e-03, 1.4696e-03,
          1.0138e-03, 5.9366e-04, 4.4799e-04, 2.0943e-03, 1.6670e-03, 9.7609e-04,
          6.1417e-04, 1.8396e-03, 1.1635e-02, 3.4904e-03, 5.8556e-04, 2.1877e-03,
          1.9331e-03, 9.2030e-04, 1.6365e-03, 1.7900e-03, 1.0662e-03, 1.5144e-03,
          1.8320e-03, 1.9741e-03, 1.9798e-03, 4.0741e-03, 4.1695e-03, 2.2068e-03,
          1.5011e-03, 4.2267e-03, 2.5215e-03, 2.4891e-03, 8.3256e-04, 8.0299e-04,
          2.3613e-03, 2.1763e-03, 4.2686e-03, 8.3113e-04, 1.6432e-03, 4.4060e-03,
          3.0785e-03, 1.7605e-03, 3.0766e-03, 5.4321e-03, 1.6317e-03, 3.6087e-03,
          1.8835e-03, 1.2222e-02, 1.1871e-02, 2.3499e-03, 2.4014e-03, 3.3607e-03,
          1.6403e-03, 1.6384e-03, 3.4428e-03, 1.9817e-03, 1.5320e-02, 2.3098e-03,
          1.2169e-03, 5.1575e-03, 1.4448e-03, 1.0719e-03, 3.5828e-02, 1.6266e-02,
          1.3145e-02, 2.7714e-03, 9.5272e-04, 9.9087e-04, 2.0237e-03, 1.6098e-03,
          3.6133e-02, 4.1565e-02, 8.9645e-05, 8.9645e-05, 8.9645e-05, 8.9645e-05,
          4.0340e-04, 2.0933e-04, 8.9645e-05, 6.5660e-04, 8.9645e-05, 8.9645e-05,
          2.0933e-04, 8.9645e-05, 8.9645e-05, 8.9645e-05, 8.9645e-05, 8.9645e-05,
          8.9645e-05, 6.3133e-04, 6.3133e-04, 2.0933e-04, 8.9645e-05, 4.3726e-04,
          8.9645e-05, 2.5678e-04, 7.9823e-04, 8.9645e-05, 3.1757e-04, 2.0933e-04,
          6.0463e-04, 8.9645e-05, 8.9645e-05, 8.9645e-05, 2.0933e-04, 8.9645e-05,
          2.0933e-04, 2.0933e-04, 2.0933e-04, 2.0933e-04, 2.0933e-04, 2.0933e-04,
          8.9645e-05, 4.0340e-04, 2.0933e-04, 3.6001e-04, 2.4033e-04, 3.6001e-04,
          2.0933e-04, 2.0933e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.8915e-03, 6.5674e-02, 1.1055e-02, 1.0719e-02, 2.8397e-02,
          4.7455e-02, 7.1449e-03, 1.2722e-03, 7.1955e-04, 9.1314e-04, 1.5116e-03,
          5.9080e-04, 6.9618e-04, 4.9448e-04, 1.0300e-03, 1.9817e-03, 1.4372e-03,
          6.6710e-04, 7.7677e-04, 3.3236e-04, 4.4155e-04, 2.8920e-04, 8.6403e-04,
          1.8835e-03, 1.0319e-03, 6.1378e-03, 2.8839e-03, 4.0984e-04, 1.2455e-03,
          5.6410e-04, 3.8013e-03, 4.8423e-04, 1.5383e-03, 8.4639e-04, 2.8591e-03,
          1.6165e-03, 5.5122e-04, 3.9062e-03, 1.5984e-03, 4.2200e-04, 2.0599e-03,
          5.3072e-04, 4.6253e-04, 4.3392e-04, 3.9983e-04, 1.1187e-03, 1.8711e-03,
          3.2091e-04, 3.6025e-04, 1.8129e-03, 7.4272e-03, 4.1428e-03, 1.3227e-03,
          7.4482e-04, 2.3899e-03, 1.4124e-03, 7.0906e-04, 2.0123e-03, 1.0281e-03,
          1.1063e-03, 1.2827e-03, 5.1842e-03, 7.6256e-03, 5.8651e-04, 2.1973e-03,
          6.2037e-04, 3.4189e-04, 4.6611e-04, 3.2997e-04, 2.4223e-04, 3.7360e-04,
          4.1151e-04, 1.9217e-03, 2.3117e-03, 6.4468e-03, 6.9427e-04, 1.0796e-03,
          3.9978e-03, 6.4373e-04, 4.8923e-04, 1.1625e-03, 5.4455e-04, 7.6962e-04,
          6.7139e-04, 1.1856e-02, 3.6087e-03, 7.0286e-04, 3.2330e-04, 3.5262e-04,
          1.6270e-03, 1.3533e-03, 2.2388e-04, 1.0853e-03, 4.5013e-04, 5.0831e-04,
          7.1716e-04, 1.1196e-03, 3.9253e-03, 4.3449e-03, 8.0299e-04, 1.0014e-03,
          1.3599e-03, 3.5810e-04, 4.3154e-04, 2.0275e-03, 1.5755e-03, 8.5878e-04,
          1.2932e-02, 8.6260e-04, 1.9951e-03, 3.7813e-04, 4.5776e-04, 6.2084e-04,
          9.4175e-04, 1.4648e-03, 4.3344e-04, 2.4891e-03, 7.1621e-04, 6.9141e-04,
          9.1553e-04, 2.6264e-03, 2.6417e-04, 1.0424e-03, 1.5640e-03, 2.4462e-04,
          4.6945e-04, 4.4346e-04, 5.4026e-04, 2.2469e-03, 4.9820e-03, 1.5137e-02,
          8.6689e-04, 3.7432e-04, 3.7241e-04, 2.8062e-04, 1.0042e-03, 4.0126e-04,
          2.9583e-03, 6.6185e-04, 1.4868e-03, 2.5883e-03, 2.4376e-03, 1.0300e-03,
          4.3678e-04, 4.7302e-04, 4.2343e-04, 9.7847e-04, 4.6945e-04, 2.3746e-03,
          3.3340e-03, 2.4586e-03, 1.3247e-03, 2.0203e-02, 1.2978e-02, 3.2425e-03,
          1.6804e-03, 2.0046e-03, 4.3011e-04, 4.2629e-04, 3.3073e-03, 1.3494e-03,
          1.9283e-03, 2.8572e-03, 1.7757e-03, 1.7061e-03, 1.7624e-03, 1.3628e-03,
          1.0233e-03, 1.2569e-03, 1.2493e-03, 7.2050e-04, 1.0796e-03, 3.4332e-03,
          2.9888e-03, 2.1858e-03, 1.8902e-03, 1.8549e-03, 2.9297e-03, 1.7405e-03,
          9.0170e-04, 1.1787e-03, 1.6441e-03, 3.8013e-03, 1.4582e-03, 1.0595e-03,
          1.2789e-03, 9.9564e-04, 1.1263e-03, 7.8726e-04, 4.8113e-04, 1.4238e-03,
          3.0003e-03, 3.8624e-04, 1.4315e-03, 1.8635e-03, 1.1253e-03, 1.8034e-03,
          2.5749e-03, 1.2302e-03, 3.6011e-03, 7.7200e-04, 1.0910e-03, 1.1702e-03,
          5.1928e-04, 1.5059e-03, 7.2241e-04, 1.3962e-03, 2.1687e-03, 9.1362e-04,
          2.4185e-03, 1.7233e-03, 7.0190e-04, 6.3419e-04, 2.2030e-03, 5.5027e-04,
          3.4485e-03, 2.2526e-03, 2.0046e-03, 7.8773e-04, 3.1319e-03, 1.5202e-03,
          1.6800e-02, 9.1267e-04, 1.8473e-03, 8.2970e-04, 3.1319e-03, 1.6880e-03,
          6.5851e-04, 6.8426e-04, 5.5790e-04, 1.7509e-03, 1.7834e-03, 9.2840e-04,
          1.0090e-03, 8.7166e-04, 5.0640e-04, 2.3136e-03, 2.3956e-03, 3.3245e-03,
          1.4095e-03, 1.1616e-03, 1.5230e-03, 3.8948e-03, 3.2444e-03, 1.4000e-03,
          2.2602e-03, 1.0862e-03, 7.6246e-04, 9.8419e-04, 9.1934e-04, 2.0313e-03,
          5.5122e-04, 5.9414e-04, 1.1244e-03, 1.7309e-03, 1.9531e-03, 8.3771e-03,
          1.9493e-03, 9.2220e-04, 1.5783e-03, 1.4849e-03, 6.4421e-04, 2.1935e-03,
          1.7090e-03, 1.4496e-03, 1.5726e-03, 3.2005e-03, 2.5845e-03, 2.3022e-03,
          1.4210e-03, 8.1873e-04, 2.5177e-03, 1.2569e-03, 6.6605e-03, 2.7905e-03,
          1.8740e-03, 1.8415e-03, 7.5006e-04, 1.3332e-03, 9.6846e-04, 8.9598e-04,
          6.8665e-04, 6.0225e-04, 6.2227e-04, 8.0729e-04, 3.1776e-03, 2.9564e-03,
          2.0485e-03, 3.2825e-03, 1.9474e-03, 2.3613e-03, 1.6251e-03, 2.5063e-03,
          1.9140e-03, 9.0742e-04, 7.5197e-04, 2.5711e-03, 6.6109e-03, 2.1057e-03,
          1.0309e-03, 1.8063e-03, 4.4708e-03, 3.5172e-03, 7.1907e-04, 2.4128e-03,
          2.5120e-03, 1.3018e-03, 5.0354e-03, 1.8167e-03, 1.7481e-03, 1.6680e-03,
          1.7643e-03, 2.4872e-03, 3.0231e-03, 3.8700e-03, 5.1270e-03, 3.2921e-03,
          2.0981e-03, 4.0550e-03, 3.3932e-03, 1.9760e-03, 1.3914e-03, 1.3094e-03,
          2.0351e-03, 1.7471e-03, 2.2182e-03, 1.2789e-03, 3.0327e-03, 2.7046e-03,
          5.8708e-03, 1.3123e-03, 2.2411e-03, 8.1482e-03, 1.7242e-03, 4.9973e-03,
          2.5082e-03, 9.3994e-03, 8.7509e-03, 2.8210e-03, 5.6190e-03, 2.5139e-03,
          2.0008e-03, 2.0981e-03, 2.8515e-03, 1.6747e-03, 1.5915e-02, 2.9984e-03,
          1.1969e-03, 3.8414e-03, 1.5984e-03, 1.2903e-03, 1.0750e-02, 5.9204e-03,
          1.8570e-02, 2.1572e-03, 1.3819e-03, 1.5783e-03, 1.3390e-03, 2.8629e-03,
          1.3969e-02, 9.9487e-03, 4.1321e-02, 7.9632e-05, 7.9632e-05, 7.9632e-05,
          7.9632e-05, 2.3055e-04, 7.9632e-05, 2.3055e-04, 7.9632e-05, 7.9632e-05,
          2.3055e-04, 7.9632e-05, 7.9632e-05, 7.9632e-05, 7.9632e-05, 7.9632e-05,
          7.9632e-05, 2.3055e-04, 2.3055e-04, 2.3055e-04, 7.9632e-05, 2.3055e-04,
          7.9632e-05, 7.9632e-05, 2.3055e-04, 7.9632e-05, 7.9632e-05, 2.3055e-04,
          2.3055e-04, 7.9632e-05, 7.9632e-05, 2.3055e-04, 2.3055e-04, 7.9632e-05,
          7.9632e-05, 2.3055e-04, 2.3055e-04, 1.3936e-04, 2.3055e-04, 2.3055e-04,
          7.9632e-05, 2.3055e-04, 2.3055e-04, 5.0831e-04, 2.9778e-04, 4.4870e-04,
          2.3055e-04, 7.9632e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.8357e-03, 4.6143e-02, 1.1551e-02, 2.2675e-02, 2.6657e-02,
          3.1799e-02, 1.1169e-02, 1.0061e-03, 4.2224e-04, 2.3727e-03, 3.8471e-03,
          1.4381e-03, 1.9569e-03, 3.4885e-03, 8.5831e-04, 8.8882e-04, 7.1030e-03,
          3.0918e-03, 5.6744e-04, 2.8586e-04, 2.3985e-04, 4.0054e-04, 3.3379e-03,
          4.3526e-03, 9.7847e-04, 9.5596e-03, 6.7024e-03, 4.5276e-04, 4.1046e-03,
          8.2245e-03, 2.2774e-03, 1.2388e-03, 1.1635e-03, 2.8896e-03, 5.3930e-04,
          6.4468e-04, 3.2473e-04, 1.5230e-03, 1.3466e-03, 4.9782e-04, 7.7724e-04,
          4.1604e-04, 2.8300e-04, 3.9244e-04, 2.2864e-04, 6.2561e-03, 1.0786e-03,
          1.2474e-03, 6.5136e-04, 2.0695e-03, 9.0361e-04, 1.6594e-03, 7.7200e-04,
          3.1996e-04, 8.5068e-04, 2.9254e-04, 3.0255e-04, 1.5278e-03, 5.4312e-04,
          5.2071e-04, 1.0080e-03, 6.1464e-04, 5.1975e-04, 5.1832e-04, 1.4753e-03,
          1.2465e-03, 9.7179e-04, 9.9659e-04, 1.9252e-04, 1.9503e-04, 3.4070e-04,
          3.8195e-04, 4.9019e-03, 4.2033e-04, 2.3327e-03, 4.2772e-04, 3.8099e-04,
          8.4686e-04, 2.5725e-04, 7.8011e-04, 1.2465e-03, 1.2302e-03, 2.9302e-04,
          2.6083e-04, 2.5940e-03, 6.1846e-04, 2.1076e-03, 1.2426e-03, 9.0599e-04,
          4.6387e-03, 1.6689e-03, 3.5954e-04, 2.3766e-03, 2.0516e-04, 2.9969e-04,
          6.3133e-04, 2.4748e-04, 1.5736e-03, 9.9373e-04, 2.7728e-04, 3.6240e-04,
          3.2310e-03, 9.7942e-04, 1.4057e-03, 5.3787e-04, 9.7418e-04, 6.2943e-04,
          1.8120e-03, 3.3450e-04, 1.8053e-03, 3.7146e-04, 8.6308e-04, 1.1311e-03,
          7.3385e-04, 3.2282e-04, 4.4203e-04, 1.4820e-03, 4.2868e-04, 4.9543e-04,
          2.2602e-03, 8.8644e-04, 2.1160e-04, 7.3004e-04, 1.6184e-03, 2.9564e-04,
          2.1470e-04, 5.3215e-04, 7.2956e-04, 1.2636e-03, 2.1725e-03, 2.6760e-03,
          1.3380e-03, 1.0777e-03, 1.2693e-03, 6.7759e-04, 1.5163e-03, 3.0255e-04,
          4.8256e-03, 2.5749e-03, 3.3970e-03, 5.7945e-03, 4.8409e-03, 4.4479e-03,
          3.1209e-04, 6.2799e-04, 4.1509e-04, 3.0398e-04, 1.1787e-03, 9.4795e-04,
          2.5063e-03, 1.1187e-03, 3.4642e-04, 8.6641e-04, 9.4986e-04, 6.1464e-04,
          1.4458e-03, 7.5912e-04, 1.4448e-03, 5.9605e-04, 1.0319e-03, 2.2583e-03,
          1.4429e-03, 2.3079e-03, 7.9422e-03, 3.4885e-03, 4.8232e-04, 6.8092e-04,
          2.0618e-03, 8.1396e-04, 8.5974e-04, 8.8978e-04, 3.2101e-03, 1.1272e-03,
          6.9332e-04, 6.7425e-04, 4.7755e-04, 5.3215e-04, 1.1005e-03, 4.2343e-03,
          1.0967e-03, 1.0004e-03, 1.8702e-03, 2.4471e-03, 2.0809e-03, 1.6069e-03,
          3.7041e-03, 3.9597e-03, 4.1656e-03, 4.0317e-04, 1.1826e-03, 1.6232e-03,
          2.2697e-03, 2.0158e-04, 1.2531e-03, 7.2098e-03, 2.4204e-03, 1.2913e-03,
          6.0225e-04, 5.7077e-04, 1.5640e-03, 1.1959e-03, 2.2697e-03, 1.1234e-03,
          8.2874e-04, 1.2016e-03, 3.7527e-04, 1.7176e-03, 4.1542e-03, 1.3943e-03,
          1.6670e-03, 7.0858e-04, 4.4107e-04, 1.7033e-03, 1.5287e-03, 6.1512e-04,
          7.7200e-04, 2.1591e-03, 6.7482e-03, 1.1911e-03, 1.9989e-03, 3.2406e-03,
          2.8343e-03, 3.3879e-04, 8.4114e-04, 2.1038e-03, 3.1452e-03, 2.6531e-03,
          1.6212e-03, 6.5041e-04, 3.7813e-04, 7.7772e-04, 9.5749e-04, 1.7929e-03,
          7.2336e-04, 8.9455e-04, 7.2002e-04, 9.4748e-04, 7.7486e-04, 1.8988e-03,
          8.7404e-04, 6.5231e-04, 4.1723e-04, 7.0477e-04, 3.1605e-03, 2.4204e-03,
          4.8542e-04, 1.7929e-03, 3.1638e-04, 5.0163e-04, 4.6730e-04, 1.4124e-03,
          5.3740e-04, 2.6202e-04, 5.4359e-04, 5.4502e-04, 4.1161e-03, 6.7978e-03,
          5.2929e-04, 2.0390e-03, 6.1131e-04, 6.5470e-04, 2.7704e-04, 1.2941e-03,
          8.3351e-04, 2.9640e-03, 3.6316e-03, 8.2474e-03, 5.7617e-02, 9.6607e-04,
          1.2827e-03, 2.6779e-03, 2.1343e-03, 5.1546e-04, 1.5192e-03, 1.4191e-03,
          7.9441e-04, 1.0166e-03, 9.7418e-04, 1.5068e-02, 5.7888e-04, 1.3571e-03,
          2.5063e-03, 4.0627e-04, 3.1686e-04, 3.6860e-04, 1.3180e-03, 1.3685e-03,
          1.2789e-03, 5.1422e-03, 3.8548e-03, 7.2594e-03, 5.8842e-04, 9.5606e-04,
          7.9155e-04, 5.5981e-04, 3.8838e-04, 2.1572e-03, 1.5182e-03, 8.2970e-04,
          5.6744e-04, 3.1471e-03, 3.8109e-03, 4.0359e-03, 5.2214e-04, 1.8578e-03,
          1.6403e-03, 5.7983e-04, 1.3418e-03, 2.6970e-03, 2.9278e-03, 2.6493e-03,
          2.4910e-03, 9.9945e-04, 1.6489e-03, 3.4714e-03, 6.4621e-03, 2.4223e-03,
          1.6413e-03, 9.4299e-03, 2.9411e-03, 2.7409e-03, 6.2990e-04, 6.9046e-04,
          5.2948e-03, 3.8700e-03, 1.3893e-02, 6.4135e-04, 1.2274e-03, 2.4139e-02,
          4.6501e-03, 1.2627e-03, 1.1475e-02, 8.5449e-03, 9.0256e-03, 4.3182e-03,
          9.2936e-04, 7.2098e-03, 4.0512e-03, 2.1992e-03, 3.9597e-03, 1.6144e-02,
          5.6610e-03, 3.4943e-03, 6.6605e-03, 5.2795e-03, 4.8599e-03, 3.5877e-03,
          3.1929e-03, 4.8523e-03, 9.8991e-04, 7.6008e-04, 5.8365e-03, 3.5019e-03,
          3.8338e-03, 7.9041e-03, 2.3422e-03, 3.3016e-03, 9.8190e-03, 1.8072e-03,
          3.3264e-03, 3.4084e-03, 3.2482e-03, 4.0527e-02, 2.1052e-04, 5.5730e-05,
          5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05,
          5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05, 3.0112e-04, 5.5730e-05,
          2.1052e-04, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05,
          4.5586e-04, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05,
          5.5730e-05, 1.7166e-03, 5.5730e-05, 5.5730e-05, 5.5730e-05, 5.5730e-05,
          2.1052e-04, 2.1052e-04, 2.1052e-04, 2.7633e-04, 2.1052e-04, 5.5730e-05,
          5.5730e-05, 2.1052e-04, 2.1052e-04, 4.4036e-04, 2.1052e-04, 2.1052e-04,
          5.5730e-05, 5.5730e-05], dtype=torch.float16),
  tensor([0.0000e+00, 2.9964e-03, 4.8828e-02, 1.0529e-02, 1.3428e-02, 3.0746e-02,
          4.7485e-02, 1.0269e-02, 3.8891e-03, 1.1072e-03, 1.8950e-03, 4.2953e-03,
          1.0872e-03, 9.7466e-04, 4.3917e-04, 1.8473e-03, 4.8828e-03, 1.0214e-03,
          9.8419e-04, 1.6632e-03, 1.3053e-04, 1.5652e-04, 2.4438e-04, 1.6527e-03,
          4.2610e-03, 1.1234e-03, 8.6136e-03, 3.1681e-03, 2.8706e-04, 1.1988e-03,
          5.6362e-04, 7.8735e-03, 4.0436e-04, 1.1101e-03, 2.0828e-03, 9.0599e-04,
          5.5456e-04, 2.7061e-04, 8.6212e-04, 1.3418e-03, 1.8847e-04, 4.5490e-04,
          3.1710e-04, 2.4986e-04, 3.3331e-04, 1.7822e-04, 1.5316e-03, 1.1730e-03,
          3.1686e-04, 2.9635e-04, 2.1286e-03, 1.3533e-03, 4.8485e-03, 1.2178e-03,
          4.7326e-04, 2.4014e-03, 3.8457e-04, 2.0981e-04, 1.0414e-03, 3.7479e-04,
          4.8828e-04, 5.2643e-04, 8.5497e-04, 8.9407e-04, 3.0279e-04, 2.3022e-03,
          6.2418e-04, 4.6706e-04, 6.5279e-04, 1.4663e-04, 1.5247e-04, 3.0327e-04,
          3.2353e-04, 2.9831e-03, 5.2691e-04, 6.5842e-03, 5.8603e-04, 6.4182e-04,
          1.8349e-03, 1.6642e-04, 2.9850e-04, 8.5688e-04, 5.2214e-04, 3.2830e-04,
          3.0541e-04, 8.5983e-03, 5.4073e-04, 2.5864e-03, 3.4738e-04, 3.1281e-04,
          3.9253e-03, 1.0843e-03, 1.9503e-04, 1.7519e-03, 2.6464e-04, 3.1543e-04,
          5.0926e-04, 3.4428e-04, 2.0008e-03, 5.8889e-04, 1.8990e-04, 2.4366e-04,
          2.5997e-03, 4.7469e-04, 6.2895e-04, 5.7888e-04, 2.0561e-03, 3.2949e-04,
          8.4076e-03, 2.2948e-04, 1.1721e-03, 2.1982e-04, 4.1389e-04, 3.8433e-04,
          1.0433e-03, 4.0197e-04, 2.7657e-04, 2.4948e-03, 4.6873e-04, 5.2738e-04,
          6.0892e-04, 6.4707e-04, 1.4532e-04, 7.5493e-03, 3.3264e-03, 1.2267e-04,
          1.3614e-04, 2.4283e-04, 3.0708e-04, 1.5478e-03, 4.1656e-03, 1.3054e-02,
          7.2956e-04, 5.3120e-04, 4.7607e-03, 3.5048e-04, 9.1743e-04, 2.1386e-04,
          1.6108e-03, 4.7493e-04, 2.3727e-03, 5.3444e-03, 2.3212e-03, 7.9679e-04,
          1.8799e-04, 1.1625e-03, 2.6226e-04, 1.9550e-04, 3.8433e-04, 1.4057e-03,
          2.0237e-03, 1.2646e-03, 3.3188e-04, 1.5526e-03, 2.1095e-03, 4.4537e-04,
          7.6342e-04, 3.4428e-04, 2.4605e-03, 3.0231e-04, 1.1759e-03, 6.6280e-04,
          1.3504e-03, 1.4582e-03, 2.2545e-03, 4.3793e-03, 2.6846e-04, 2.6226e-04,
          1.3695e-03, 3.9530e-04, 8.1968e-04, 1.7872e-03, 8.7738e-04, 9.7275e-04,
          1.2283e-03, 9.4795e-04, 5.2786e-04, 4.2963e-04, 1.2074e-03, 1.4954e-02,
          9.4271e-04, 1.5688e-03, 2.6150e-03, 1.1353e-02, 9.5673e-03, 2.3212e-03,
          5.2528e-03, 7.9393e-04, 3.9482e-03, 2.7657e-04, 3.7360e-04, 2.2144e-03,
          1.4679e-02, 1.4389e-04, 1.3151e-03, 1.9817e-03, 2.1992e-03, 1.2455e-03,
          7.3099e-04, 4.4966e-04, 1.6527e-03, 5.6124e-04, 1.4105e-03, 8.2207e-04,
          3.9768e-04, 7.6103e-04, 3.3879e-04, 3.0121e-02, 3.0212e-03, 8.1968e-04,
          1.0727e-02, 4.6873e-04, 2.9540e-04, 8.8978e-04, 1.5182e-03, 3.0231e-04,
          5.9462e-04, 8.3780e-04, 6.0959e-03, 2.4776e-03, 1.9484e-03, 1.0414e-03,
          1.5526e-03, 3.2210e-04, 7.2718e-04, 7.4387e-04, 6.5651e-03, 3.0079e-03,
          8.5831e-04, 5.2357e-04, 3.7265e-04, 1.3742e-03, 1.2980e-03, 8.6641e-04,
          5.5075e-04, 3.9244e-04, 3.4046e-04, 1.7910e-03, 8.2445e-04, 1.6947e-03,
          6.4325e-04, 5.4216e-04, 3.0756e-04, 5.9271e-04, 1.2751e-03, 6.2275e-04,
          4.1246e-04, 1.1778e-03, 3.9530e-04, 5.6887e-04, 7.3624e-04, 1.8702e-03,
          3.1734e-04, 3.3498e-04, 6.9571e-04, 6.2561e-04, 4.1351e-03, 3.9406e-03,
          4.0054e-04, 5.5313e-04, 8.0729e-04, 9.4128e-04, 2.8872e-04, 8.1825e-04,
          9.7942e-04, 3.2349e-03, 2.9373e-03, 2.5349e-03, 4.5929e-03, 1.4429e-03,
          9.7942e-04, 1.1473e-03, 3.7441e-03, 6.1798e-04, 1.2770e-03, 2.0580e-03,
          1.4048e-03, 1.7538e-03, 4.9782e-04, 1.4172e-03, 4.1771e-04, 5.8270e-04,
          1.0204e-03, 3.1567e-04, 2.3043e-04, 4.0102e-04, 1.0462e-03, 1.1692e-03,
          1.5430e-03, 3.4142e-03, 2.4166e-03, 3.7479e-03, 6.8378e-04, 1.0433e-03,
          7.1764e-04, 4.2725e-04, 3.3402e-04, 2.5673e-03, 1.3304e-03, 1.0815e-03,
          6.6137e-04, 3.6488e-03, 1.8661e-02, 1.7653e-03, 3.6955e-04, 2.0962e-03,
          1.9989e-03, 4.7398e-04, 1.0824e-03, 1.7729e-03, 6.3019e-03, 5.4283e-03,
          1.5059e-03, 1.5726e-03, 1.0777e-03, 3.8776e-03, 3.7136e-03, 3.9062e-03,
          2.0943e-03, 8.4915e-03, 1.8234e-03, 1.9274e-03, 7.4387e-04, 6.5756e-04,
          2.9526e-03, 1.7700e-03, 3.4389e-03, 4.8184e-04, 1.3800e-03, 2.6093e-03,
          2.4452e-03, 1.4343e-03, 1.7767e-03, 4.6768e-03, 6.9504e-03, 3.2768e-03,
          1.6403e-03, 6.4888e-03, 1.0277e-02, 1.3409e-03, 2.2888e-03, 2.3708e-03,
          4.3945e-03, 7.7209e-03, 2.1954e-03, 4.7417e-03, 6.3248e-03, 1.9474e-03,
          1.0090e-03, 4.9324e-03, 9.1600e-04, 7.1001e-04, 1.1650e-02, 1.1147e-02,
          5.8136e-03, 6.0608e-02, 2.6608e-03, 3.1185e-03, 1.3218e-03, 2.6016e-03,
          1.2489e-02, 1.0796e-02, 5.5046e-03, 4.6425e-03, 3.9734e-02, 5.8711e-05,
          2.2018e-04, 2.2018e-04, 5.8711e-05, 9.1791e-04, 5.8711e-05, 5.8711e-05,
          2.2018e-04, 5.8711e-05, 5.8711e-05, 5.8711e-05, 5.8711e-05, 5.8711e-05,
          5.8711e-05, 2.2018e-04, 2.2018e-04, 5.8711e-05, 5.8711e-05, 2.2018e-04,
          5.8711e-05, 5.8711e-05, 2.2018e-04, 5.8711e-05, 5.8711e-05, 9.1791e-04,
          2.2018e-04, 5.8711e-05, 5.8711e-05, 5.8711e-05, 5.8711e-05, 5.8711e-05,
          5.8711e-05, 5.8711e-05, 5.8711e-05, 1.1623e-04, 2.2018e-04, 2.2018e-04,
          5.8711e-05, 2.2018e-04, 5.8711e-05, 6.0844e-04, 2.1601e-04, 2.1601e-04,
          5.8711e-05, 5.8711e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9940e-03, 5.1788e-02, 1.1581e-02, 2.2049e-02, 2.7863e-02,
          3.8483e-02, 1.1162e-02, 1.0042e-03, 4.6325e-04, 4.8103e-03, 6.1188e-03,
          1.9083e-03, 9.4891e-04, 4.5371e-04, 1.9836e-03, 9.5034e-04, 1.2989e-03,
          9.5367e-04, 5.0068e-04, 1.7130e-04, 2.5773e-04, 2.3782e-04, 1.8787e-03,
          6.4163e-03, 1.5955e-03, 9.3307e-03, 7.6981e-03, 3.2449e-04, 1.6127e-03,
          7.1144e-04, 2.0523e-03, 4.7922e-04, 3.8414e-03, 5.7526e-03, 7.5388e-04,
          2.3460e-03, 4.6086e-04, 7.6294e-03, 4.6806e-03, 2.4188e-04, 2.5253e-03,
          5.3692e-04, 4.6539e-04, 4.4155e-04, 4.2582e-04, 1.6966e-03, 4.3030e-03,
          3.1590e-04, 3.6716e-04, 2.7866e-03, 1.0786e-03, 1.3037e-03, 9.9564e-04,
          2.8753e-04, 6.3086e-04, 3.1519e-04, 5.5885e-04, 8.4734e-04, 6.9666e-04,
          6.2799e-04, 6.4754e-04, 6.5994e-04, 1.2484e-03, 6.9237e-04, 6.8893e-03,
          1.1349e-03, 9.1076e-04, 2.3022e-03, 2.3532e-04, 2.2483e-04, 3.3879e-04,
          4.0483e-04, 1.1391e-02, 4.1747e-04, 1.5888e-03, 2.7418e-04, 2.8658e-04,
          6.2275e-04, 4.2844e-04, 8.0109e-04, 2.7313e-03, 9.6464e-04, 4.0436e-04,
          3.6263e-04, 2.5139e-03, 4.1890e-04, 3.4714e-03, 8.4448e-04, 6.1131e-04,
          7.4158e-03, 4.5738e-03, 2.6393e-04, 3.7079e-03, 2.7704e-04, 3.9887e-04,
          4.5204e-04, 2.5749e-04, 1.6470e-03, 1.1768e-03, 4.1199e-04, 7.4244e-04,
          5.4245e-03, 7.1192e-04, 8.7547e-04, 1.6832e-03, 3.8528e-03, 5.1498e-04,
          1.8787e-03, 4.7874e-04, 4.6959e-03, 3.2115e-04, 7.9870e-04, 1.7757e-03,
          1.2999e-03, 6.7854e-04, 3.1996e-04, 4.2191e-03, 5.7888e-04, 5.6028e-04,
          8.3447e-04, 2.7924e-03, 2.5797e-04, 6.1226e-04, 7.6904e-03, 3.0279e-04,
          3.0589e-04, 4.1389e-04, 5.4502e-04, 3.0899e-03, 2.0676e-03, 2.4033e-03,
          1.3304e-03, 1.1587e-03, 1.1568e-03, 6.0892e-04, 2.6932e-03, 2.9135e-04,
          3.8910e-03, 5.6362e-04, 1.0796e-03, 3.7155e-03, 4.9934e-03, 1.0920e-03,
          3.7098e-04, 6.1607e-04, 7.4720e-04, 5.7316e-04, 8.7595e-04, 2.7542e-03,
          6.0921e-03, 2.2926e-03, 5.3406e-04, 1.5173e-03, 9.8610e-04, 1.7681e-03,
          3.8223e-03, 2.0561e-03, 1.0223e-03, 4.1199e-04, 2.6913e-03, 9.5224e-04,
          4.5128e-03, 7.4463e-03, 2.9583e-03, 6.9847e-03, 8.6403e-04, 1.3084e-03,
          3.4599e-03, 1.6966e-03, 2.1687e-03, 1.3847e-03, 1.5297e-03, 3.9024e-03,
          8.2779e-04, 5.3120e-04, 8.5258e-04, 9.7513e-04, 4.3068e-03, 4.3373e-03,
          3.0499e-03, 2.3880e-03, 7.8888e-03, 2.2850e-03, 1.2465e-03, 1.5697e-03,
          3.2101e-03, 6.4468e-04, 3.8090e-03, 6.2704e-04, 6.9189e-04, 4.8637e-03,
          2.4338e-03, 2.7752e-04, 3.5534e-03, 2.9602e-03, 5.5237e-03, 3.4561e-03,
          7.3338e-04, 5.9938e-04, 4.5433e-03, 1.4610e-03, 3.9024e-03, 3.3455e-03,
          6.2513e-04, 3.0975e-03, 5.7507e-04, 1.0262e-03, 6.0387e-03, 1.1711e-03,
          9.6321e-04, 1.3533e-03, 6.3181e-04, 1.2550e-03, 3.5152e-03, 4.9496e-04,
          1.6994e-03, 1.3037e-03, 1.0712e-02, 1.2541e-03, 4.4289e-03, 1.0805e-03,
          3.8662e-03, 5.2452e-04, 1.8625e-03, 2.1534e-03, 1.2505e-02, 6.9313e-03,
          1.4572e-03, 4.0674e-04, 4.2081e-04, 6.7806e-04, 2.0981e-03, 6.0034e-04,
          1.0624e-03, 1.0214e-03, 4.6825e-04, 1.2608e-03, 1.4420e-03, 1.7624e-03,
          6.9666e-04, 6.8140e-04, 5.4741e-04, 1.4467e-03, 1.6432e-03, 9.8991e-04,
          1.0519e-03, 1.4839e-03, 3.3593e-04, 4.9019e-04, 5.7983e-04, 3.9520e-03,
          4.4012e-04, 3.7694e-04, 8.0919e-04, 1.1320e-03, 3.3703e-03, 6.6757e-03,
          9.9277e-04, 1.4992e-03, 8.8310e-04, 1.0357e-03, 3.3545e-04, 8.9025e-04,
          1.5059e-03, 5.1689e-03, 1.0433e-03, 2.9507e-03, 5.1460e-03, 2.3193e-03,
          9.9754e-04, 1.6251e-03, 6.8359e-03, 9.7656e-04, 3.6964e-03, 2.8286e-03,
          1.0548e-03, 8.9550e-04, 6.3848e-04, 1.6317e-03, 8.2302e-04, 9.9373e-04,
          1.5230e-03, 4.3988e-04, 3.5310e-04, 5.2834e-04, 1.4544e-03, 1.3485e-03,
          1.1244e-03, 1.6968e-02, 2.8248e-03, 2.2659e-03, 9.2030e-04, 1.2331e-03,
          1.2875e-03, 6.6519e-04, 5.8985e-04, 4.9973e-03, 4.7302e-03, 9.4271e-04,
          7.3910e-04, 1.8435e-03, 2.4529e-03, 2.2926e-03, 6.5279e-04, 3.2082e-03,
          2.8267e-03, 6.0225e-04, 2.6588e-03, 8.4782e-04, 2.4281e-03, 1.8816e-03,
          1.0214e-03, 8.6355e-04, 1.6718e-03, 3.6316e-03, 8.3542e-03, 5.6686e-03,
          1.0033e-03, 1.3077e-02, 1.9836e-03, 1.5659e-03, 9.5177e-04, 1.2312e-03,
          1.4315e-03, 9.8705e-04, 3.5610e-03, 9.9945e-04, 2.1057e-03, 3.2997e-03,
          1.0818e-02, 8.0395e-04, 2.1763e-03, 2.3666e-02, 4.4289e-03, 1.3685e-03,
          6.8521e-04, 6.5575e-03, 2.3937e-03, 1.9140e-03, 7.5035e-03, 2.5196e-03,
          3.7384e-03, 6.1722e-03, 2.0885e-03, 2.9163e-03, 3.6182e-03, 2.0485e-03,
          9.2649e-04, 5.3596e-03, 1.1320e-03, 1.0281e-03, 5.0087e-03, 2.4796e-03,
          3.0231e-03, 2.9469e-03, 2.6169e-03, 1.4000e-03, 1.4067e-03, 1.2102e-03,
          2.4509e-03, 2.3098e-03, 2.6646e-03, 2.8152e-03, 1.5984e-03, 3.3203e-02,
          7.0572e-05, 5.0640e-04, 7.0572e-05, 4.1342e-04, 7.0572e-05, 7.0572e-05,
          7.0572e-05, 7.0572e-05, 7.0572e-05, 7.0572e-05, 8.4925e-04, 7.0572e-05,
          8.4925e-04, 5.0640e-04, 2.6417e-04, 7.0572e-05, 7.0572e-05, 2.6417e-04,
          8.4925e-04, 7.0572e-05, 5.0640e-04, 7.0572e-05, 7.0572e-05, 6.0701e-04,
          9.4557e-04, 2.9449e-03, 7.0572e-05, 2.6417e-04, 2.6417e-04, 7.0572e-05,
          7.0572e-05, 7.0572e-05, 7.0572e-05, 7.0572e-05, 2.6417e-04, 7.0572e-05,
          7.0572e-05, 7.0572e-05, 7.0572e-05, 3.5763e-04, 7.0572e-05, 2.6417e-04,
          7.0572e-05, 7.0572e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.9177e-03, 5.9509e-02, 1.2177e-02, 1.1452e-02, 2.8061e-02,
          4.8401e-02, 8.1787e-03, 1.0195e-03, 4.3201e-04, 2.1172e-03, 3.8605e-03,
          1.3151e-03, 1.1730e-03, 3.7646e-04, 5.3520e-03, 9.7847e-04, 9.7561e-04,
          7.4387e-04, 6.0606e-04, 1.2201e-04, 1.4734e-04, 1.8835e-04, 1.3485e-03,
          4.4823e-03, 2.0809e-03, 6.3286e-03, 3.3569e-03, 2.1124e-04, 1.1892e-03,
          5.1403e-04, 2.6875e-03, 3.5095e-04, 2.1286e-03, 3.5477e-03, 4.1699e-04,
          1.2388e-03, 2.6989e-04, 1.2131e-03, 2.3956e-03, 1.8942e-04, 6.1703e-04,
          2.9397e-04, 2.4045e-04, 3.1376e-04, 1.9181e-04, 1.8091e-03, 2.2221e-03,
          3.2020e-04, 2.8300e-04, 2.1381e-03, 5.9175e-04, 1.4353e-03, 6.3705e-04,
          2.3270e-04, 8.1396e-04, 2.3758e-04, 2.3961e-04, 9.3460e-04, 4.4870e-04,
          5.5218e-04, 4.7874e-04, 4.7493e-04, 5.6124e-04, 3.3164e-04, 6.2866e-03,
          5.0449e-04, 4.4179e-04, 7.4053e-04, 1.4436e-04, 1.4317e-04, 2.7370e-04,
          3.0255e-04, 4.3526e-03, 2.7776e-04, 2.2621e-03, 2.8133e-04, 2.5201e-04,
          7.7486e-04, 2.0182e-04, 2.9969e-04, 1.6546e-03, 4.8018e-04, 3.5453e-04,
          3.0231e-04, 3.1414e-03, 3.7026e-04, 1.8940e-03, 3.1090e-04, 2.7466e-04,
          4.0512e-03, 1.2798e-03, 2.0158e-04, 1.9894e-03, 2.5606e-04, 3.1304e-04,
          2.9087e-04, 1.9383e-04, 8.2207e-04, 6.3658e-04, 2.3115e-04, 3.4308e-04,
          3.0746e-03, 4.3297e-04, 6.7377e-04, 1.2512e-03, 7.2861e-03, 3.7813e-04,
          3.1662e-03, 2.4629e-04, 1.5373e-03, 2.3997e-04, 4.0150e-04, 4.1676e-04,
          2.2678e-03, 3.6454e-04, 3.0375e-04, 5.7831e-03, 5.3263e-04, 3.7646e-04,
          5.7936e-04, 9.4509e-04, 1.5032e-04, 9.3174e-04, 2.2186e-02, 1.3995e-04,
          1.6391e-04, 2.3699e-04, 3.5048e-04, 4.3297e-03, 2.2488e-03, 5.9547e-03,
          6.8998e-04, 4.9210e-04, 4.0779e-03, 3.3545e-04, 9.9754e-04, 2.2376e-04,
          3.3283e-03, 5.4979e-04, 1.9817e-03, 3.5038e-03, 3.3092e-03, 8.1873e-04,
          2.0981e-04, 1.4963e-03, 4.6945e-04, 2.6298e-04, 4.0936e-04, 3.3913e-03,
          2.7828e-03, 2.9984e-03, 4.4084e-04, 7.5674e-04, 9.8991e-04, 6.4373e-04,
          9.7561e-04, 4.5586e-04, 1.6899e-03, 2.5964e-04, 1.7128e-03, 7.9107e-04,
          3.4809e-03, 2.3270e-03, 3.0575e-03, 9.8572e-03, 3.5357e-04, 3.3426e-04,
          1.6975e-03, 5.0592e-04, 2.0161e-03, 5.8174e-03, 1.1654e-03, 2.1210e-03,
          1.3056e-03, 9.4604e-04, 6.2180e-04, 4.4179e-04, 1.6804e-03, 6.3324e-03,
          2.1267e-03, 3.1338e-03, 8.5678e-03, 3.3417e-03, 2.2850e-03, 4.5242e-03,
          4.3640e-03, 7.8344e-04, 5.2261e-03, 3.2735e-04, 4.0841e-04, 9.2087e-03,
          3.2806e-03, 1.7047e-04, 3.1185e-03, 2.8820e-03, 2.4185e-03, 2.5024e-03,
          8.4257e-04, 5.0783e-04, 3.2692e-03, 5.3930e-04, 1.2665e-03, 2.0962e-03,
          3.8457e-04, 8.8072e-04, 3.7694e-04, 2.6913e-03, 3.4676e-03, 1.1339e-03,
          1.7452e-03, 5.5361e-04, 3.4094e-04, 9.1267e-04, 3.7575e-03, 3.4475e-04,
          1.0004e-03, 1.1063e-03, 8.5449e-03, 6.7673e-03, 3.5229e-03, 1.0529e-03,
          1.7748e-03, 3.0422e-04, 8.1587e-04, 7.9679e-04, 2.2964e-02, 8.1406e-03,
          6.7806e-04, 4.1986e-04, 3.6287e-04, 1.0481e-03, 1.9350e-03, 7.6818e-04,
          6.0749e-04, 4.3559e-04, 3.8052e-04, 2.3251e-03, 1.1024e-03, 1.8435e-03,
          6.8092e-04, 5.6791e-04, 3.4165e-04, 9.6464e-04, 1.5974e-03, 7.3290e-04,
          4.3344e-04, 1.1072e-03, 3.3259e-04, 5.1451e-04, 6.5660e-04, 3.6430e-03,
          3.0971e-04, 3.4738e-04, 6.8331e-04, 7.1573e-04, 6.0997e-03, 3.7746e-03,
          5.0449e-04, 6.3610e-04, 9.3508e-04, 1.2178e-03, 3.1996e-04, 9.8705e-04,
          1.4982e-03, 5.3024e-03, 2.3994e-03, 2.9926e-03, 4.0474e-03, 2.1019e-03,
          8.8024e-04, 8.3494e-04, 8.9874e-03, 7.2527e-04, 2.1896e-03, 3.1929e-03,
          1.1873e-03, 2.0771e-03, 5.4073e-04, 1.5059e-03, 4.8780e-04, 6.3658e-04,
          8.6069e-04, 3.7265e-04, 2.8610e-04, 4.0674e-04, 1.3142e-03, 1.7166e-03,
          1.5078e-03, 7.5607e-03, 3.4485e-03, 4.0512e-03, 6.2943e-04, 9.4461e-04,
          8.1539e-04, 4.9114e-04, 3.9530e-04, 6.4430e-03, 1.8244e-03, 1.0786e-03,
          7.1669e-04, 3.9005e-03, 4.4594e-03, 1.8883e-03, 4.2844e-04, 3.0365e-03,
          3.4637e-03, 5.7125e-04, 1.5802e-03, 1.2226e-03, 9.6970e-03, 5.4245e-03,
          1.5020e-03, 1.4544e-03, 1.1425e-03, 4.5891e-03, 5.3062e-03, 6.3400e-03,
          1.8902e-03, 9.2392e-03, 2.1362e-03, 1.8311e-03, 8.5449e-04, 8.3828e-04,
          2.6665e-03, 1.4153e-03, 2.9182e-03, 6.2990e-04, 1.5945e-03, 2.8229e-03,
          3.5744e-03, 1.1415e-03, 1.8816e-03, 6.0349e-03, 8.4381e-03, 2.0466e-03,
          1.2712e-03, 5.7831e-03, 3.7346e-03, 1.4210e-03, 2.7809e-03, 2.4910e-03,
          5.5618e-03, 2.9602e-02, 2.0714e-03, 5.2795e-03, 4.0970e-03, 1.9436e-03,
          9.2888e-04, 4.2191e-03, 1.0233e-03, 7.9393e-04, 5.2643e-03, 4.0321e-03,
          3.3607e-03, 1.1528e-02, 4.7874e-03, 5.2185e-03, 1.3599e-03, 3.4122e-03,
          4.3907e-03, 3.9215e-03, 2.6627e-03, 4.6043e-03, 4.5395e-03, 9.4147e-03,
          4.7424e-02, 2.2137e-04, 9.7573e-05, 1.0347e-03, 9.7573e-05, 9.7573e-05,
          2.2137e-04, 9.7573e-05, 9.7573e-05, 9.7573e-05, 4.6802e-04, 2.2137e-04,
          1.2589e-03, 4.6802e-04, 2.2137e-04, 2.2137e-04, 2.2137e-04, 2.2137e-04,
          2.2137e-04, 2.2137e-04, 4.6802e-04, 9.7573e-05, 9.7573e-05, 2.2137e-04,
          2.2137e-04, 2.9430e-03, 9.7573e-05, 2.2137e-04, 2.2137e-04, 9.7573e-05,
          2.2137e-04, 2.2137e-04, 2.2137e-04, 2.9516e-04, 2.2137e-04, 2.2137e-04,
          2.2137e-04, 2.2137e-04, 2.2137e-04, 9.7322e-04, 8.9931e-04, 8.9931e-04,
          2.2137e-04, 9.7573e-05], dtype=torch.float16),
  tensor([0.0000e+00, 5.1422e-03, 4.1870e-02, 1.5045e-02, 2.0035e-02, 5.1880e-02,
          2.3560e-02, 1.5533e-02, 5.4264e-04, 2.7800e-04, 5.7316e-04, 1.1673e-03,
          3.9887e-04, 1.5736e-03, 5.8899e-03, 5.2452e-04, 5.0449e-04, 6.3972e-03,
          3.2043e-03, 3.0351e-04, 1.5812e-03, 7.6294e-04, 1.3552e-03, 1.4668e-03,
          1.3781e-03, 5.9986e-04, 1.1971e-02, 3.2711e-03, 1.7576e-03, 3.1757e-03,
          1.1108e-02, 9.3746e-04, 1.7767e-03, 7.0238e-04, 4.1723e-04, 4.3511e-04,
          5.3501e-04, 3.8505e-04, 7.7486e-04, 5.3644e-04, 2.4166e-03, 6.8951e-04,
          6.0940e-04, 4.2486e-04, 5.6076e-04, 3.0017e-04, 3.9215e-03, 1.1072e-03,
          2.9449e-03, 2.2449e-03, 5.8594e-03, 6.9141e-04, 6.1274e-04, 3.2878e-04,
          3.5071e-04, 4.4656e-04, 5.6744e-04, 3.0184e-04, 6.2656e-04, 4.6563e-04,
          6.3515e-04, 4.0283e-03, 7.3528e-04, 5.8556e-04, 4.8518e-04, 1.2589e-03,
          8.8549e-04, 3.0780e-04, 4.9686e-04, 3.5810e-04, 3.4070e-04, 5.1498e-04,
          6.6519e-04, 1.2941e-03, 6.3324e-04, 8.6641e-04, 3.7766e-04, 2.6560e-04,
          8.2302e-04, 3.1018e-04, 4.5013e-04, 3.6144e-04, 5.2404e-04, 3.6430e-04,
          4.6611e-04, 1.2054e-03, 7.4387e-04, 4.9067e-04, 3.9673e-04, 7.5006e-04,
          9.4891e-04, 4.9448e-04, 3.2496e-04, 6.8855e-04, 3.8624e-04, 7.1955e-04,
          1.6527e-03, 4.4656e-04, 6.2227e-04, 6.5088e-04, 3.1948e-04, 2.6798e-04,
          8.0061e-04, 2.6178e-04, 3.0994e-04, 4.5133e-04, 6.4230e-04, 3.2883e-03,
          9.7513e-04, 5.7650e-04, 7.2527e-04, 3.7336e-04, 5.9175e-04, 4.6182e-04,
          3.6407e-04, 6.2418e-04, 2.1095e-03, 1.3895e-03, 1.0796e-03, 1.6460e-03,
          5.9967e-03, 4.4703e-04, 2.3413e-04, 3.3545e-04, 8.2350e-04, 2.0874e-04,
          2.4283e-04, 3.9291e-04, 4.4775e-04, 7.4911e-04, 7.0763e-04, 1.1015e-03,
          7.0333e-04, 4.9210e-04, 2.3317e-04, 2.7180e-04, 4.8876e-04, 5.0163e-04,
          3.9368e-03, 7.3509e-03, 2.3708e-03, 2.0752e-03, 1.1921e-03, 7.7782e-03,
          3.2544e-04, 1.4353e-04, 2.4676e-04, 2.2149e-04, 6.8998e-04, 6.9618e-04,
          7.6675e-04, 6.6376e-04, 4.5967e-04, 1.0109e-03, 1.3514e-03, 1.0147e-03,
          4.8757e-04, 5.1737e-04, 2.9635e-04, 6.5374e-04, 1.4734e-03, 6.8474e-03,
          1.1787e-03, 8.1635e-04, 5.2872e-03, 8.4925e-04, 5.8794e-04, 2.7251e-04,
          5.9223e-04, 2.8157e-04, 4.3440e-04, 3.6216e-04, 3.1052e-03, 8.0061e-04,
          4.5681e-04, 5.5599e-04, 1.0672e-03, 1.3771e-03, 8.7976e-04, 1.2541e-03,
          5.4169e-04, 6.0892e-04, 1.0757e-03, 8.4305e-04, 6.7186e-04, 5.6887e-04,
          7.0333e-04, 7.2403e-03, 8.0347e-04, 5.2643e-04, 8.0442e-04, 7.5483e-04,
          7.1001e-04, 2.5344e-04, 6.4087e-04, 6.5193e-03, 5.2786e-04, 7.0286e-04,
          1.0281e-03, 3.6030e-03, 1.3990e-03, 6.9952e-04, 6.5994e-04, 5.9652e-04,
          8.3828e-04, 8.2350e-04, 8.9455e-04, 4.1747e-04, 1.1930e-03, 1.7052e-03,
          7.2527e-04, 1.0939e-03, 8.2350e-04, 4.3631e-04, 8.9359e-04, 5.1546e-04,
          4.7493e-04, 3.5954e-03, 1.0281e-03, 3.6287e-04, 1.1797e-03, 6.4812e-03,
          2.7637e-03, 9.3126e-04, 8.7118e-04, 5.9652e-04, 2.0905e-03, 9.4271e-04,
          8.2636e-04, 6.3992e-04, 6.3276e-04, 6.9571e-04, 8.8501e-04, 5.4588e-03,
          1.6737e-03, 3.7956e-04, 4.6110e-04, 3.6073e-04, 7.6914e-04, 1.3313e-03,
          1.0958e-03, 1.3046e-03, 4.7755e-04, 1.0462e-03, 9.0027e-03, 1.4923e-02,
          1.5650e-03, 9.8801e-04, 7.5340e-04, 5.9319e-04, 1.3514e-03, 1.2350e-03,
          7.9393e-04, 9.3031e-04, 1.7357e-03, 1.0386e-03, 2.0638e-03, 2.0046e-03,
          7.4768e-04, 5.4121e-04, 3.6955e-04, 4.3106e-04, 3.6693e-04, 1.1749e-03,
          6.1703e-04, 4.4370e-04, 2.4757e-03, 1.5762e-02, 1.6724e-02, 1.1530e-03,
          1.5125e-03, 1.4391e-03, 1.8044e-03, 1.2655e-03, 1.6804e-03, 1.7300e-03,
          1.3857e-03, 5.4169e-04, 9.4528e-03, 1.7334e-02, 1.1988e-03, 9.9945e-04,
          8.4543e-04, 7.8249e-04, 6.7043e-04, 8.1921e-04, 1.7614e-03, 5.9319e-04,
          7.5483e-04, 1.5850e-03, 2.8286e-03, 6.3171e-03, 1.6689e-03, 1.7080e-03,
          2.3785e-03, 1.3838e-03, 1.1206e-03, 1.7872e-03, 1.1168e-03, 2.5158e-03,
          2.1687e-03, 1.9045e-03, 1.1673e-03, 7.4844e-03, 1.6680e-03, 9.5844e-04,
          1.0672e-03, 3.6488e-03, 9.6369e-04, 2.2240e-03, 4.7445e-04, 5.1641e-04,
          2.9030e-03, 7.0524e-04, 1.4282e-02, 1.7376e-03, 4.6692e-03, 2.2049e-03,
          8.1873e-04, 2.2125e-03, 7.4501e-03, 5.6839e-03, 3.5725e-03, 3.8071e-03,
          5.2681e-03, 8.9340e-03, 8.2474e-03, 2.0428e-03, 1.4124e-03, 2.5192e-02,
          1.7529e-03, 2.2888e-03, 2.6474e-02, 3.4618e-03, 8.0061e-04, 4.0703e-03,
          5.4741e-04, 1.9550e-03, 1.2074e-03, 3.2318e-02, 1.8215e-03, 3.1494e-02,
          8.9407e-04, 1.2121e-03, 1.8021e-02, 7.6771e-04, 1.7099e-03, 9.2697e-03,
          1.8509e-02, 1.1368e-02, 2.1648e-03, 1.8663e-03, 2.3117e-03, 1.0290e-03,
          1.3332e-03, 1.3056e-03, 3.5763e-04, 5.8222e-04, 1.2657e-02, 4.7493e-04,
          1.9054e-03, 1.7271e-03, 1.4524e-03, 2.9411e-03, 9.4795e-04, 2.0084e-03,
          8.9979e-04, 4.0527e-02, 7.0810e-05, 1.9658e-04, 7.0810e-05, 7.0810e-05,
          1.9658e-04, 7.0810e-05, 7.0810e-05, 7.0810e-05, 2.0754e-04, 7.0810e-05,
          7.0810e-05, 7.0810e-05, 3.3331e-04, 7.0810e-05, 7.0810e-05, 7.0810e-05,
          1.9658e-04, 7.0810e-05, 7.0810e-05, 7.0810e-05, 2.0754e-04, 1.9658e-04,
          7.0810e-05, 1.9658e-04, 7.0810e-05, 7.0810e-05, 7.0810e-05, 7.0810e-05,
          3.3951e-04, 3.3951e-04, 3.3951e-04, 4.0483e-04, 3.3951e-04, 1.9658e-04,
          7.0810e-05, 4.7636e-04, 3.3951e-04, 5.5504e-04, 1.9658e-04, 1.9658e-04,
          3.3951e-04, 1.9658e-04], dtype=torch.float16),
  tensor([0.0000e+00, 3.4084e-03, 4.4647e-02, 1.3191e-02, 2.1072e-02, 2.8870e-02,
          4.1504e-02, 1.7792e-02, 3.1357e-03, 1.0405e-03, 2.4567e-03, 5.2528e-03,
          1.3742e-03, 1.2674e-03, 5.7554e-04, 1.9255e-03, 3.1853e-03, 1.3056e-03,
          1.2732e-03, 1.4601e-03, 1.5891e-04, 1.9562e-04, 3.4881e-04, 2.4986e-03,
          4.9820e-03, 1.1930e-03, 1.5038e-02, 3.8013e-03, 3.4571e-04, 1.5402e-03,
          7.7677e-04, 6.3324e-03, 4.5943e-04, 1.2865e-03, 3.1052e-03, 7.1430e-04,
          5.5695e-04, 3.0994e-04, 9.4891e-04, 1.6270e-03, 2.2423e-04, 5.0259e-04,
          3.4094e-04, 2.5678e-04, 3.5286e-04, 1.9360e-04, 2.0161e-03, 1.3304e-03,
          3.9697e-04, 3.9268e-04, 2.5082e-03, 1.2913e-03, 4.2801e-03, 1.2083e-03,
          4.6682e-04, 1.9722e-03, 3.9101e-04, 2.3186e-04, 1.0605e-03, 4.0913e-04,
          5.2118e-04, 5.7125e-04, 7.9632e-04, 7.9584e-04, 3.0231e-04, 2.2793e-03,
          6.4611e-04, 4.9257e-04, 7.1049e-04, 1.6105e-04, 1.7893e-04, 3.3855e-04,
          3.9220e-04, 3.5381e-03, 5.2071e-04, 5.0659e-03, 6.0129e-04, 5.8126e-04,
          1.3466e-03, 1.7858e-04, 3.0589e-04, 8.3256e-04, 5.1928e-04, 3.2949e-04,
          3.2234e-04, 5.6458e-03, 5.0926e-04, 2.6932e-03, 3.5858e-04, 3.1972e-04,
          4.3106e-03, 1.2283e-03, 2.0242e-04, 1.8768e-03, 2.9469e-04, 4.0340e-04,
          4.9496e-04, 3.3402e-04, 1.7080e-03, 6.7472e-04, 1.9789e-04, 2.4557e-04,
          2.9182e-03, 5.2929e-04, 7.7963e-04, 5.3835e-04, 2.0199e-03, 3.6502e-04,
          5.2528e-03, 2.5225e-04, 1.2989e-03, 2.2817e-04, 4.2248e-04, 4.1914e-04,
          9.6178e-04, 3.8981e-04, 3.7432e-04, 2.3975e-03, 6.3324e-04, 5.6553e-04,
          6.6376e-04, 6.6853e-04, 1.5557e-04, 5.8136e-03, 3.9825e-03, 1.3399e-04,
          1.4746e-04, 2.4331e-04, 3.1114e-04, 1.4582e-03, 3.4103e-03, 8.1024e-03,
          6.6280e-04, 5.7316e-04, 3.8605e-03, 3.7527e-04, 9.8419e-04, 2.2078e-04,
          2.2182e-03, 5.9986e-04, 2.8381e-03, 8.8348e-03, 3.0861e-03, 9.1314e-04,
          1.9217e-04, 1.0500e-03, 2.5249e-04, 1.9872e-04, 3.9172e-04, 1.2007e-03,
          1.9293e-03, 1.1616e-03, 3.1304e-04, 1.3733e-03, 1.8358e-03, 4.6110e-04,
          9.1600e-04, 3.8981e-04, 2.2259e-03, 2.9635e-04, 1.0719e-03, 7.6485e-04,
          1.5564e-03, 1.7958e-03, 3.0994e-03, 5.5237e-03, 2.9445e-04, 2.9564e-04,
          1.5211e-03, 4.5514e-04, 7.5960e-04, 1.4505e-03, 1.1578e-03, 1.0099e-03,
          9.2459e-04, 8.7690e-04, 5.8317e-04, 5.0497e-04, 1.2178e-03, 2.0660e-02,
          9.6273e-04, 1.4677e-03, 3.1109e-03, 1.0048e-02, 1.1078e-02, 2.3174e-03,
          5.4665e-03, 9.6989e-04, 4.5319e-03, 2.8563e-04, 3.9363e-04, 2.6588e-03,
          1.0521e-02, 1.4651e-04, 1.5879e-03, 2.5520e-03, 2.3212e-03, 1.2817e-03,
          7.2575e-04, 4.9114e-04, 1.5230e-03, 5.7507e-04, 1.5354e-03, 8.7023e-04,
          3.9792e-04, 7.6437e-04, 3.6311e-04, 1.5823e-02, 3.1834e-03, 9.3794e-04,
          6.7749e-03, 4.9019e-04, 3.4046e-04, 9.4128e-04, 1.3809e-03, 3.1424e-04,
          6.0797e-04, 9.2602e-04, 6.9962e-03, 2.0924e-03, 2.1629e-03, 1.0500e-03,
          1.6241e-03, 3.7456e-04, 6.5470e-04, 7.7391e-04, 7.4692e-03, 3.3836e-03,
          8.7500e-04, 4.7517e-04, 3.7456e-04, 1.1225e-03, 1.2312e-03, 8.4496e-04,
          6.3896e-04, 4.0579e-04, 3.4213e-04, 1.4448e-03, 7.2050e-04, 1.5898e-03,
          6.7282e-04, 5.7125e-04, 2.9588e-04, 5.7173e-04, 1.2321e-03, 6.7949e-04,
          4.5681e-04, 1.1187e-03, 3.6049e-04, 5.3596e-04, 6.8188e-04, 1.7033e-03,
          3.2687e-04, 3.6335e-04, 8.8120e-04, 6.4516e-04, 3.9482e-03, 4.8294e-03,
          4.1008e-04, 5.9128e-04, 7.1335e-04, 7.6437e-04, 2.9540e-04, 8.1587e-04,
          8.6021e-04, 3.2101e-03, 2.7466e-03, 2.6131e-03, 6.3934e-03, 1.2550e-03,
          8.2874e-04, 1.0500e-03, 3.6812e-03, 5.6171e-04, 1.3971e-03, 1.8167e-03,
          1.1158e-03, 1.3580e-03, 6.4516e-04, 1.8187e-03, 4.5180e-04, 5.6648e-04,
          1.0300e-03, 3.2425e-04, 2.3353e-04, 4.2129e-04, 9.9468e-04, 1.0710e-03,
          1.2388e-03, 4.1847e-03, 2.6855e-03, 3.5362e-03, 7.8249e-04, 1.1158e-03,
          7.2002e-04, 4.2510e-04, 3.3784e-04, 2.5921e-03, 1.3599e-03, 1.4133e-03,
          8.3542e-04, 3.1719e-03, 1.1185e-02, 1.7347e-03, 3.8004e-04, 1.8187e-03,
          1.6947e-03, 4.6206e-04, 1.0519e-03, 1.3247e-03, 5.9929e-03, 4.4289e-03,
          1.2770e-03, 1.1463e-03, 1.1225e-03, 3.4828e-03, 4.0436e-03, 3.1796e-03,
          1.7586e-03, 8.5602e-03, 1.7719e-03, 1.7014e-03, 8.3208e-04, 7.3385e-04,
          2.6665e-03, 1.6909e-03, 4.0131e-03, 4.7064e-04, 1.2245e-03, 3.2177e-03,
          2.6455e-03, 1.1148e-03, 2.0218e-03, 4.9400e-03, 8.1787e-03, 2.7275e-03,
          1.2779e-03, 6.5231e-03, 7.2517e-03, 1.3533e-03, 2.3422e-03, 2.7981e-03,
          4.8752e-03, 8.0032e-03, 2.1706e-03, 4.9782e-03, 5.9471e-03, 2.1191e-03,
          1.2016e-03, 5.6152e-03, 8.1348e-04, 6.7472e-04, 9.2773e-03, 7.8964e-03,
          5.2567e-03, 5.2246e-02, 2.6150e-03, 3.3417e-03, 1.5211e-03, 2.2488e-03,
          7.9117e-03, 7.2556e-03, 4.2229e-03, 5.0545e-03, 1.7410e-02, 6.3591e-03,
          7.0839e-03, 1.7433e-03, 4.3243e-02, 5.8270e-04, 6.9618e-05, 6.9618e-05,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 5.8270e-04, 9.6846e-04,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05, 6.9618e-05,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 1.1873e-04, 4.5586e-04, 6.9618e-05,
          6.9618e-05, 6.9618e-05, 6.9618e-05, 6.2418e-04, 2.9540e-04, 2.9540e-04,
          6.9618e-05, 6.9618e-05], dtype=torch.float16),
  tensor([0.0000e+00, 3.7766e-03, 5.7373e-02, 1.2878e-02, 1.6174e-02, 2.9022e-02,
          3.6438e-02, 1.0132e-02, 6.6996e-04, 3.3617e-04, 2.5158e-03, 5.3062e-03,
          1.7700e-03, 6.2370e-04, 4.1223e-04, 6.2561e-04, 6.3848e-04, 9.9277e-04,
          5.8222e-04, 3.9363e-04, 1.8454e-04, 2.5249e-04, 2.3997e-04, 9.3794e-04,
          7.2327e-03, 1.2350e-03, 8.3847e-03, 3.7556e-03, 2.5272e-04, 8.6927e-04,
          5.9795e-04, 1.2617e-03, 3.0017e-04, 1.0386e-03, 1.1683e-03, 4.3511e-04,
          1.5793e-03, 7.7343e-04, 1.8988e-03, 9.3746e-04, 2.4343e-04, 2.3575e-03,
          9.8324e-04, 7.0953e-04, 8.5068e-04, 7.3767e-04, 1.0328e-03, 1.3847e-03,
          3.0518e-04, 3.3879e-04, 2.2240e-03, 8.7309e-04, 9.2745e-04, 5.8603e-04,
          2.5392e-04, 5.2547e-04, 3.3689e-04, 1.2293e-03, 2.5787e-03, 3.3035e-03,
          1.9932e-03, 4.4346e-04, 1.0157e-03, 1.0529e-03, 1.6851e-03, 2.2087e-03,
          1.8759e-03, 1.7157e-03, 1.4029e-03, 5.2357e-04, 4.5466e-04, 8.2970e-04,
          6.9714e-04, 5.5618e-03, 4.8113e-04, 1.2445e-03, 2.9635e-04, 2.8944e-04,
          6.3276e-04, 8.7738e-04, 2.1305e-03, 2.3537e-03, 3.0670e-03, 1.2026e-03,
          7.4339e-04, 2.0714e-03, 6.7234e-04, 3.2177e-03, 1.5936e-03, 1.0214e-03,
          7.9269e-03, 5.3596e-03, 5.9509e-04, 5.0392e-03, 6.9904e-04, 8.0395e-04,
          3.3808e-04, 2.3723e-04, 9.7847e-04, 1.7624e-03, 1.1072e-03, 2.1114e-03,
          6.2523e-03, 9.8801e-04, 2.0638e-03, 2.6970e-03, 1.3704e-03, 3.6526e-04,
          1.6947e-03, 1.7405e-03, 6.4011e-03, 5.5933e-04, 1.1635e-03, 2.5406e-03,
          1.3924e-03, 1.2598e-03, 2.8896e-04, 2.1210e-03, 1.2245e-03, 3.9697e-04,
          5.2547e-04, 3.2082e-03, 5.0545e-04, 4.4250e-04, 1.6642e-03, 7.6246e-04,
          6.8903e-04, 1.4563e-03, 1.4801e-03, 1.5259e-03, 9.5463e-04, 2.0294e-03,
          1.6909e-03, 1.7281e-03, 4.2534e-04, 9.2793e-04, 4.7646e-03, 6.2609e-04,
          2.0199e-03, 5.3835e-04, 1.1854e-03, 2.3022e-03, 1.8787e-03, 6.5994e-04,
          8.5735e-04, 7.9918e-04, 8.8739e-04, 2.3003e-03, 1.5316e-03, 2.0199e-03,
          9.7275e-03, 1.9836e-03, 1.6422e-03, 1.8225e-03, 1.6060e-03, 3.7670e-03,
          5.0049e-03, 3.2082e-03, 1.1549e-03, 9.1314e-04, 2.5616e-03, 7.5817e-04,
          1.9417e-03, 4.4250e-03, 1.5926e-03, 1.5507e-03, 1.8110e-03, 3.5343e-03,
          4.8752e-03, 3.8395e-03, 1.3628e-03, 7.1621e-04, 7.9679e-04, 2.1667e-03,
          2.8038e-03, 1.5841e-03, 1.9569e-03, 1.9503e-03, 2.9812e-03, 1.4582e-03,
          1.4000e-03, 9.5892e-04, 2.0599e-03, 1.5631e-03, 1.1263e-03, 9.1982e-04,
          4.2267e-03, 7.5388e-04, 1.9875e-03, 1.1778e-03, 1.2808e-03, 1.3819e-03,
          1.2579e-03, 6.2656e-04, 1.2827e-03, 1.8311e-03, 6.6261e-03, 1.8015e-03,
          2.2030e-03, 5.4169e-04, 2.2488e-03, 3.1166e-03, 5.1498e-03, 1.9331e-03,
          1.3857e-03, 6.4735e-03, 1.0729e-03, 1.0281e-03, 8.4915e-03, 9.1171e-04,
          1.2236e-03, 1.3342e-03, 1.3247e-03, 2.4681e-03, 2.1629e-03, 1.2093e-03,
          2.2907e-03, 9.3126e-04, 3.1567e-03, 7.1335e-04, 1.8663e-03, 8.5402e-04,
          3.0918e-03, 1.0633e-03, 3.3588e-03, 3.3054e-03, 3.3798e-03, 1.8578e-03,
          2.8896e-03, 1.2293e-03, 1.2426e-03, 3.5286e-03, 2.0199e-03, 7.6103e-04,
          1.8501e-03, 3.0117e-03, 9.8991e-04, 1.0862e-03, 2.2793e-03, 4.7112e-03,
          2.4033e-03, 1.3237e-03, 1.4086e-03, 1.3485e-03, 1.5163e-03, 8.1730e-04,
          1.9741e-03, 3.2940e-03, 1.0300e-03, 2.0885e-03, 1.1091e-03, 2.6436e-03,
          1.1873e-03, 8.9741e-04, 1.7719e-03, 1.9875e-03, 1.8778e-03, 5.6038e-03,
          2.2182e-03, 2.5902e-03, 1.5793e-03, 1.1311e-03, 8.5878e-04, 3.2444e-03,
          1.3390e-03, 1.4915e-03, 1.0328e-03, 2.2430e-03, 3.3283e-03, 1.7471e-03,
          2.9163e-03, 3.1166e-03, 2.9659e-03, 1.5459e-03, 4.0474e-03, 2.8534e-03,
          3.8452e-03, 4.2267e-03, 7.0286e-04, 1.5593e-03, 1.1244e-03, 2.4204e-03,
          2.6627e-03, 8.9741e-04, 9.1219e-04, 8.0252e-04, 2.0485e-03, 2.2278e-03,
          3.6926e-03, 4.0436e-03, 1.6384e-03, 2.2240e-03, 1.5364e-03, 3.8242e-03,
          2.0542e-03, 1.2980e-03, 9.6989e-04, 2.3651e-03, 7.9346e-03, 3.5439e-03,
          1.7853e-03, 1.8921e-03, 2.5826e-03, 1.7557e-03, 1.4753e-03, 5.1460e-03,
          1.7891e-03, 5.1451e-04, 3.0136e-03, 1.0052e-03, 2.0046e-03, 2.2411e-03,
          1.0366e-03, 2.1973e-03, 1.4391e-03, 2.6474e-03, 4.2915e-03, 3.1090e-03,
          3.7441e-03, 1.6190e-02, 2.1076e-03, 1.6012e-03, 1.6766e-03, 1.5182e-03,
          2.1362e-03, 1.8082e-03, 2.5311e-03, 2.1267e-03, 3.6545e-03, 2.5158e-03,
          1.3130e-02, 9.8991e-04, 1.8864e-03, 1.1993e-02, 3.6621e-03, 2.7084e-03,
          2.9507e-03, 5.4321e-03, 3.7937e-03, 1.8187e-03, 1.4023e-02, 2.8114e-03,
          6.9122e-03, 3.0575e-03, 1.9817e-03, 4.8065e-03, 6.0005e-03, 2.6646e-03,
          1.3323e-03, 5.2032e-03, 2.5978e-03, 2.2373e-03, 3.9330e-03, 2.1839e-03,
          4.3030e-03, 1.6031e-03, 2.5101e-03, 1.6918e-03, 1.3189e-03, 3.7365e-03,
          3.4180e-03, 2.7733e-03, 5.4893e-03, 2.7294e-03, 2.0084e-03, 6.8398e-03,
          2.0294e-03, 1.8759e-03, 4.1161e-03, 3.2349e-02, 7.7724e-05, 7.7724e-05,
          2.3639e-04, 7.7724e-05, 7.7724e-05, 7.7724e-05, 5.4359e-04, 7.7724e-05,
          6.0892e-04, 7.7724e-05, 7.7724e-05, 7.7724e-05, 7.7724e-05, 7.7724e-05,
          2.3639e-04, 6.0892e-04, 7.7724e-05, 7.7724e-05, 7.7724e-05, 2.3639e-04,
          7.7724e-05, 6.0892e-04, 7.7724e-05, 7.7724e-05, 7.7724e-05, 7.7724e-05,
          1.1158e-03, 1.1158e-03, 7.7724e-05, 2.3639e-04, 8.5402e-04, 2.3639e-04,
          7.7724e-05, 2.3639e-04, 2.3639e-04, 1.0138e-03, 2.3639e-04, 1.0548e-03,
          4.8184e-04, 4.8184e-04], dtype=torch.float16),
  tensor([0.0000e+00, 6.3820e-03, 5.9296e-02, 1.6083e-02, 4.6051e-02, 3.6469e-02,
          4.2847e-02, 2.7481e-02, 8.5735e-04, 4.4465e-04, 3.2692e-03, 4.4823e-03,
          1.4448e-03, 7.8201e-04, 6.1321e-04, 1.1559e-03, 7.3910e-04, 1.3266e-03,
          8.7452e-04, 3.9887e-04, 2.9993e-04, 4.3154e-04, 3.7313e-04, 1.9855e-03,
          4.5280e-03, 1.2960e-03, 2.2385e-02, 1.1650e-02, 5.0259e-04, 1.3132e-03,
          8.3303e-04, 1.4658e-03, 5.1785e-04, 2.2793e-03, 2.3766e-03, 1.0052e-03,
          2.0027e-03, 5.7268e-04, 4.3678e-03, 2.3155e-03, 3.7432e-04, 1.9016e-03,
          5.8270e-04, 4.9019e-04, 4.8828e-04, 5.1832e-04, 1.1511e-03, 2.6875e-03,
          4.3821e-04, 6.5851e-04, 4.1084e-03, 1.9760e-03, 1.2474e-03, 9.7084e-04,
          3.9339e-04, 6.5184e-04, 4.2295e-04, 6.8712e-04, 1.1702e-03, 9.0122e-04,
          6.2084e-04, 8.4639e-04, 8.7786e-04, 1.5383e-03, 6.4564e-04, 2.8038e-03,
          1.1873e-03, 8.0776e-04, 1.4305e-03, 3.3784e-04, 2.9731e-04, 4.6992e-04,
          6.2895e-04, 7.6370e-03, 7.0000e-04, 1.4162e-03, 3.9554e-04, 3.7932e-04,
          7.3004e-04, 5.8508e-04, 9.0885e-04, 1.7662e-03, 1.0338e-03, 4.3869e-04,
          4.3726e-04, 1.7958e-03, 5.6505e-04, 1.9665e-03, 8.4257e-04, 6.2180e-04,
          4.3259e-03, 3.2101e-03, 2.4748e-04, 2.7428e-03, 3.4046e-04, 6.0797e-04,
          6.9141e-04, 3.7742e-04, 1.6155e-03, 1.7815e-03, 4.8661e-04, 8.6880e-04,
          3.7251e-03, 6.5231e-04, 7.2527e-04, 1.6794e-03, 1.4648e-03, 5.6887e-04,
          1.4629e-03, 5.6648e-04, 3.9177e-03, 3.0923e-04, 7.2718e-04, 1.6260e-03,
          8.0538e-04, 6.9952e-04, 5.1260e-04, 2.2812e-03, 7.2861e-04, 7.0238e-04,
          9.4414e-04, 2.0657e-03, 2.5415e-04, 4.4942e-04, 2.3670e-03, 3.2163e-04,
          3.6883e-04, 5.2547e-04, 5.2214e-04, 1.6632e-03, 1.5564e-03, 1.7929e-03,
          1.0929e-03, 9.3365e-04, 6.0368e-04, 5.1451e-04, 2.1076e-03, 3.2496e-04,
          3.5553e-03, 7.2622e-04, 1.3428e-03, 7.2136e-03, 6.1150e-03, 1.1091e-03,
          3.4857e-04, 3.7026e-04, 5.6982e-04, 6.8760e-04, 9.1696e-04, 1.5087e-03,
          5.1956e-03, 1.3456e-03, 6.0034e-04, 2.2221e-03, 1.6909e-03, 1.5802e-03,
          2.6875e-03, 1.9131e-03, 6.1941e-04, 4.3511e-04, 1.9875e-03, 1.1511e-03,
          5.5351e-03, 5.9280e-03, 2.2240e-03, 4.4975e-03, 1.3008e-03, 1.9007e-03,
          3.1567e-03, 2.0599e-03, 1.0786e-03, 6.4993e-04, 1.5831e-03, 3.2368e-03,
          6.2990e-04, 5.6505e-04, 1.0490e-03, 1.3447e-03, 2.3212e-03, 2.5845e-03,
          1.7271e-03, 9.4461e-04, 3.1490e-03, 1.4019e-03, 1.1549e-03, 1.0347e-03,
          1.7786e-03, 1.0757e-03, 3.6297e-03, 5.8603e-04, 6.7949e-04, 1.7176e-03,
          1.3952e-03, 2.9302e-04, 2.0294e-03, 2.4490e-03, 3.5686e-03, 2.0351e-03,
          8.3447e-04, 6.3801e-04, 2.2297e-03, 1.2140e-03, 3.3817e-03, 2.1057e-03,
          5.4407e-04, 2.1591e-03, 5.3310e-04, 9.9182e-04, 5.0278e-03, 1.2102e-03,
          1.2064e-03, 1.9102e-03, 7.5674e-04, 9.6083e-04, 1.8215e-03, 5.1165e-04,
          1.7557e-03, 1.4496e-03, 4.6349e-03, 6.6090e-04, 3.2883e-03, 1.0729e-03,
          3.3913e-03, 5.0926e-04, 1.3628e-03, 2.4319e-03, 4.4746e-03, 2.2545e-03,
          1.3351e-03, 4.7398e-04, 4.0984e-04, 6.3848e-04, 1.3657e-03, 7.4434e-04,
          1.4038e-03, 1.2112e-03, 4.1795e-04, 7.0381e-04, 1.2760e-03, 1.3657e-03,
          7.3719e-04, 8.0776e-04, 6.6185e-04, 1.6327e-03, 1.5621e-03, 1.1377e-03,
          1.1349e-03, 1.1034e-03, 3.3545e-04, 4.5395e-04, 5.7745e-04, 2.0790e-03,
          4.3178e-04, 3.9029e-04, 8.5831e-04, 1.1597e-03, 2.1400e-03, 1.1627e-02,
          1.1148e-03, 1.7881e-03, 6.8140e-04, 7.0572e-04, 3.1519e-04, 9.7179e-04,
          1.0729e-03, 2.2869e-03, 1.1091e-03, 2.4586e-03, 4.4823e-03, 1.4057e-03,
          8.9359e-04, 1.5402e-03, 2.7809e-03, 8.0538e-04, 3.9749e-03, 1.6632e-03,
          8.3399e-04, 7.7057e-04, 1.1406e-03, 1.9350e-03, 1.1415e-03, 9.1743e-04,
          1.4229e-03, 4.1890e-04, 3.8314e-04, 4.8327e-04, 1.3161e-03, 1.4305e-03,
          8.2350e-04, 8.3084e-03, 2.2259e-03, 2.1133e-03, 7.3051e-04, 9.6655e-04,
          1.1463e-03, 6.2180e-04, 5.1260e-04, 3.7270e-03, 4.0970e-03, 1.0948e-03,
          7.5245e-04, 1.7672e-03, 2.5291e-03, 2.5139e-03, 7.5817e-04, 2.3975e-03,
          1.7309e-03, 7.0333e-04, 2.4948e-03, 1.1282e-03, 1.7586e-03, 1.1530e-03,
          1.2827e-03, 8.3780e-04, 1.8969e-03, 2.6245e-03, 7.8354e-03, 2.7924e-03,
          9.2649e-04, 8.6975e-03, 2.5959e-03, 1.6127e-03, 9.3651e-04, 1.2522e-03,
          1.8110e-03, 1.3084e-03, 5.1918e-03, 1.1034e-03, 1.7462e-03, 3.3703e-03,
          1.1391e-02, 8.5878e-04, 2.6188e-03, 1.6327e-02, 4.0436e-03, 3.1738e-03,
          6.8235e-04, 5.3291e-03, 2.9182e-03, 2.2202e-03, 6.9504e-03, 2.8419e-03,
          4.1542e-03, 2.4071e-03, 2.2621e-03, 3.0937e-03, 3.7155e-03, 3.5629e-03,
          1.7347e-03, 8.7662e-03, 1.1845e-03, 9.9373e-04, 3.9101e-03, 2.0199e-03,
          3.3760e-03, 3.3531e-03, 1.6232e-03, 1.6060e-03, 1.6928e-03, 1.3208e-03,
          2.6569e-03, 2.2488e-03, 3.2730e-03, 3.9902e-03, 2.4357e-03, 1.1703e-02,
          1.8721e-03, 2.1782e-03, 6.2599e-03, 4.0817e-03, 5.9967e-02, 8.4460e-05,
          8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05,
          4.9734e-04, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05,
          8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05,
          8.4460e-05, 5.1956e-03, 8.4460e-05, 8.4460e-05, 8.4460e-05, 8.4460e-05,
          8.4460e-05, 8.4460e-05, 8.4460e-05, 1.5235e-04, 4.9734e-04, 8.4460e-05,
          8.4460e-05, 4.9734e-04, 8.4460e-05, 8.6594e-04, 8.4460e-05, 8.4460e-05,
          8.4460e-05, 8.4460e-05], dtype=torch.float16),
  tensor([0.0000e+00, 4.0436e-03, 5.8319e-02, 1.3405e-02, 1.9836e-02, 3.0731e-02,
          4.9835e-02, 1.3229e-02, 1.0080e-03, 4.5705e-04, 3.2234e-03, 4.1656e-03,
          1.3437e-03, 9.3126e-04, 4.7302e-04, 4.0207e-03, 1.0519e-03, 1.0138e-03,
          8.7452e-04, 5.6982e-04, 1.6308e-04, 2.0528e-04, 2.4414e-04, 1.9836e-03,
          4.1733e-03, 2.1591e-03, 1.0315e-02, 6.8893e-03, 2.9325e-04, 1.3237e-03,
          7.0620e-04, 1.9722e-03, 3.6740e-04, 3.4199e-03, 4.6539e-03, 6.6376e-04,
          1.1892e-03, 2.9421e-04, 2.4357e-03, 3.1109e-03, 2.2197e-04, 7.9536e-04,
          3.1829e-04, 2.7156e-04, 3.3712e-04, 2.2340e-04, 1.5507e-03, 2.8858e-03,
          3.5501e-04, 4.1461e-04, 2.6588e-03, 9.5797e-04, 1.2522e-03, 8.0776e-04,
          2.7776e-04, 6.8426e-04, 3.2854e-04, 2.4164e-04, 7.5769e-04, 4.1199e-04,
          4.3297e-04, 6.3133e-04, 6.8665e-04, 9.4557e-04, 3.3140e-04, 6.4240e-03,
          5.1737e-04, 4.2963e-04, 1.0452e-03, 1.6904e-04, 1.7166e-04, 2.9373e-04,
          3.7217e-04, 6.6414e-03, 4.1652e-04, 1.8721e-03, 3.8004e-04, 3.2473e-04,
          6.7616e-04, 2.1851e-04, 3.1447e-04, 1.5268e-03, 5.0592e-04, 2.9230e-04,
          2.8706e-04, 2.6379e-03, 4.4537e-04, 1.8911e-03, 3.5572e-04, 3.0041e-04,
          3.1776e-03, 1.5736e-03, 1.9419e-04, 1.7710e-03, 3.4142e-04, 3.8338e-04,
          4.1413e-04, 2.7180e-04, 1.2465e-03, 9.1553e-04, 2.2233e-04, 3.3498e-04,
          2.6970e-03, 3.7789e-04, 4.7970e-04, 1.0920e-03, 5.1880e-03, 4.2987e-04,
          2.6169e-03, 2.7823e-04, 1.6928e-03, 2.3007e-04, 3.6287e-04, 4.9496e-04,
          1.3914e-03, 4.0650e-04, 4.5085e-04, 4.9057e-03, 5.9938e-04, 4.9448e-04,
          6.7139e-04, 1.2417e-03, 1.4699e-04, 7.0858e-04, 1.2657e-02, 1.4317e-04,
          1.7154e-04, 2.3293e-04, 2.9016e-04, 3.1891e-03, 2.2964e-03, 3.5954e-03,
          6.5231e-04, 4.4394e-04, 2.1305e-03, 3.1495e-04, 1.0872e-03, 2.1601e-04,
          4.6005e-03, 6.8855e-04, 1.8330e-03, 5.0354e-03, 5.7259e-03, 9.9468e-04,
          1.9813e-04, 6.8521e-04, 4.3011e-04, 2.7418e-04, 4.1842e-04, 2.8419e-03,
          3.4866e-03, 1.8148e-03, 3.3283e-04, 1.2045e-03, 1.4400e-03, 6.9857e-04,
          1.2512e-03, 6.3753e-04, 9.1457e-04, 2.6202e-04, 1.7900e-03, 9.5797e-04,
          4.0550e-03, 3.3512e-03, 3.5629e-03, 1.3290e-02, 4.6992e-04, 4.4775e-04,
          1.4944e-03, 6.4564e-04, 1.6747e-03, 2.9984e-03, 1.6260e-03, 2.2812e-03,
          8.1968e-04, 6.2704e-04, 6.4468e-04, 5.4502e-04, 2.2163e-03, 5.6076e-03,
          2.3251e-03, 1.8930e-03, 8.3847e-03, 2.3575e-03, 1.8110e-03, 2.5120e-03,
          2.6073e-03, 1.0624e-03, 6.4163e-03, 3.1853e-04, 4.0793e-04, 5.3787e-03,
          2.3232e-03, 1.6522e-04, 3.9444e-03, 3.2387e-03, 2.1896e-03, 2.2564e-03,
          7.4148e-04, 5.1403e-04, 3.1109e-03, 6.3658e-04, 1.4782e-03, 1.8520e-03,
          3.5024e-04, 1.0338e-03, 3.9029e-04, 1.6241e-03, 3.1509e-03, 1.2178e-03,
          1.8530e-03, 7.6485e-04, 3.6025e-04, 7.0238e-04, 2.9430e-03, 3.1257e-04,
          1.0509e-03, 1.3418e-03, 7.8430e-03, 2.4605e-03, 4.1199e-03, 9.5177e-04,
          2.1973e-03, 3.0756e-04, 8.9025e-04, 8.3208e-04, 1.6739e-02, 6.3667e-03,
          6.8760e-04, 3.9053e-04, 3.3116e-04, 8.0967e-04, 1.7872e-03, 9.7942e-04,
          6.6519e-04, 4.5872e-04, 3.1829e-04, 1.3485e-03, 1.1845e-03, 1.7672e-03,
          6.2084e-04, 5.9223e-04, 3.5381e-04, 1.0004e-03, 1.4467e-03, 8.5449e-04,
          5.2071e-04, 9.8419e-04, 2.8229e-04, 4.3535e-04, 5.9032e-04, 3.2177e-03,
          3.1400e-04, 3.3712e-04, 8.6021e-04, 9.0981e-04, 4.1046e-03, 5.8441e-03,
          5.6696e-04, 6.8426e-04, 6.7949e-04, 9.0885e-04, 2.8419e-04, 8.9788e-04,
          1.4038e-03, 5.6038e-03, 1.5564e-03, 2.8782e-03, 4.7150e-03, 1.9760e-03,
          7.8726e-04, 9.0361e-04, 7.2975e-03, 6.5279e-04, 2.5673e-03, 2.6569e-03,
          9.6941e-04, 1.1892e-03, 9.7609e-04, 1.8568e-03, 5.6219e-04, 5.5408e-04,
          7.5817e-04, 3.3545e-04, 2.7370e-04, 4.3082e-04, 1.3018e-03, 1.4877e-03,
          1.0729e-03, 1.0452e-02, 3.5400e-03, 3.0365e-03, 6.1274e-04, 9.4414e-04,
          8.0585e-04, 4.5252e-04, 3.6979e-04, 4.6997e-03, 2.0981e-03, 1.1997e-03,
          7.7295e-04, 3.1395e-03, 3.8910e-03, 2.3327e-03, 4.1223e-04, 2.7504e-03,
          2.5005e-03, 5.5552e-04, 1.6956e-03, 1.3475e-03, 5.2719e-03, 2.4567e-03,
          1.4753e-03, 1.1024e-03, 1.3485e-03, 3.5629e-03, 6.8283e-03, 4.7493e-03,
          1.5240e-03, 7.7133e-03, 2.5635e-03, 1.9159e-03, 9.2793e-04, 1.0157e-03,
          2.9278e-03, 1.6832e-03, 4.0703e-03, 5.7936e-04, 1.5011e-03, 3.4409e-03,
          4.5090e-03, 1.2293e-03, 2.2354e-03, 7.3433e-03, 7.9498e-03, 3.5248e-03,
          9.0361e-04, 6.1111e-03, 4.0359e-03, 1.7986e-03, 3.0708e-03, 2.8267e-03,
          4.9171e-03, 1.0399e-02, 2.4281e-03, 4.3030e-03, 3.8452e-03, 3.3264e-03,
          1.5354e-03, 5.1880e-03, 9.0694e-04, 7.4768e-04, 4.7073e-03, 3.0994e-03,
          3.2253e-03, 9.2773e-03, 3.0918e-03, 3.5286e-03, 1.5211e-03, 1.8396e-03,
          4.7035e-03, 3.7766e-03, 3.1376e-03, 4.9286e-03, 4.9744e-03, 1.3412e-02,
          9.3689e-03, 2.0332e-03, 9.1858e-03, 2.1076e-03, 2.4734e-02, 3.4363e-02,
          2.4486e-04, 8.3685e-05, 8.3685e-05, 8.3685e-05, 2.4486e-04, 8.3685e-05,
          2.4486e-04, 2.4486e-04, 2.4486e-04, 2.4486e-04, 8.3685e-05, 8.3685e-05,
          2.4486e-04, 8.3685e-05, 2.4486e-04, 8.3685e-05, 8.3685e-05, 2.4486e-04,
          2.4486e-04, 5.6114e-03, 8.3685e-05, 8.3685e-05, 8.3685e-05, 8.3685e-05,
          8.3685e-05, 2.4486e-04, 2.4486e-04, 3.0851e-04, 2.4486e-04, 2.4486e-04,
          8.3685e-05, 2.4486e-04, 2.4486e-04, 8.1825e-04, 8.3685e-05, 2.4486e-04,
          2.4486e-04, 8.3685e-05], dtype=torch.float16),
  tensor([0.0000e+00, 1.6998e-02, 6.6895e-02, 3.3600e-02, 4.2145e-02, 1.2341e-01,
          4.0100e-02, 5.0934e-02, 5.9557e-04, 4.0817e-04, 1.1044e-03, 2.7103e-03,
          6.9809e-04, 5.7840e-04, 7.3624e-04, 4.8065e-04, 4.3178e-04, 1.0529e-03,
          6.5374e-04, 2.6345e-04, 6.9141e-04, 8.1444e-04, 8.6069e-04, 8.4448e-04,
          2.3174e-03, 5.8556e-04, 3.1067e-02, 4.7989e-03, 1.5612e-03, 8.1730e-04,
          8.0681e-04, 7.1573e-04, 5.4121e-04, 5.6362e-04, 4.1771e-04, 3.9148e-04,
          6.0177e-04, 7.2908e-04, 7.1239e-04, 3.8242e-04, 6.3992e-04, 6.7854e-04,
          7.6294e-04, 5.5170e-04, 6.1750e-04, 3.2282e-04, 6.3086e-04, 6.3753e-04,
          4.5586e-04, 1.1854e-03, 2.6566e-02, 9.7847e-04, 6.1321e-04, 2.9421e-04,
          7.5769e-04, 5.6791e-04, 1.0357e-03, 6.7043e-04, 7.0715e-04, 7.4148e-04,
          8.3876e-04, 8.6641e-04, 1.0014e-03, 4.4060e-04, 3.4261e-04, 6.9809e-04,
          7.2002e-04, 2.4772e-04, 3.4451e-04, 4.5609e-04, 4.4727e-04, 4.6897e-04,
          8.6880e-04, 1.3342e-03, 1.8806e-03, 6.9666e-04, 4.6992e-04, 3.4237e-04,
          6.7997e-04, 6.6376e-04, 5.7507e-04, 4.4537e-04, 5.7602e-04, 4.3893e-04,
          6.2799e-04, 1.1129e-03, 6.0320e-04, 5.8270e-04, 3.0828e-04, 4.1199e-04,
          1.0862e-03, 4.8804e-04, 2.1803e-04, 6.2847e-04, 3.4213e-04, 1.0118e-03,
          1.4458e-03, 1.0719e-03, 4.5371e-04, 6.2370e-04, 8.1444e-04, 4.8399e-04,
          1.2703e-03, 3.9268e-04, 3.4690e-04, 5.0068e-04, 5.3835e-04, 5.9891e-04,
          7.4196e-04, 6.1464e-04, 7.7105e-04, 3.0637e-04, 4.8876e-04, 3.2043e-04,
          3.3998e-04, 4.3249e-04, 7.1573e-04, 6.5851e-04, 1.2159e-03, 2.1400e-03,
          1.3857e-03, 5.3215e-04, 5.1975e-04, 3.1185e-04, 8.3828e-04, 2.3496e-04,
          4.1556e-04, 3.5691e-04, 3.9291e-04, 4.9877e-04, 3.7885e-04, 8.0919e-04,
          6.1512e-04, 3.7241e-04, 2.9588e-04, 1.8954e-04, 4.4799e-04, 3.4571e-04,
          9.1219e-04, 6.0320e-04, 5.8126e-04, 1.8167e-03, 8.4257e-04, 9.8133e-04,
          4.6992e-04, 1.8978e-04, 2.7728e-04, 3.1900e-04, 6.5851e-04, 5.8079e-04,
          7.8106e-04, 4.4370e-04, 3.5977e-04, 6.5804e-04, 1.2779e-03, 7.1573e-04,
          4.3130e-04, 3.9983e-04, 2.4068e-04, 3.3712e-04, 5.5504e-04, 6.8951e-04,
          6.3896e-04, 5.6458e-04, 8.4972e-04, 8.8358e-04, 1.3304e-03, 3.9625e-04,
          8.8167e-04, 3.5715e-04, 3.5119e-04, 3.3379e-04, 7.9107e-04, 5.5790e-04,
          3.0541e-04, 4.3726e-04, 1.0061e-03, 1.3847e-03, 5.4312e-04, 8.2493e-04,
          4.8566e-04, 3.3593e-04, 6.6376e-04, 4.7302e-04, 5.5170e-04, 3.7122e-04,
          6.1941e-04, 7.3862e-04, 1.1625e-03, 1.0557e-03, 5.7220e-04, 5.9080e-04,
          5.7888e-04, 3.8218e-04, 4.0460e-04, 1.0157e-03, 5.2404e-04, 4.8304e-04,
          8.6880e-04, 1.2913e-03, 5.4502e-04, 4.4632e-04, 4.8184e-04, 4.2748e-04,
          4.7398e-04, 4.8065e-04, 4.8876e-04, 4.6158e-04, 9.3508e-04, 4.3106e-04,
          5.2214e-04, 1.8320e-03, 2.1687e-03, 5.3835e-04, 6.5088e-04, 4.8733e-04,
          4.4584e-04, 7.1955e-04, 8.6260e-04, 3.0589e-04, 6.6280e-04, 6.6900e-04,
          1.0519e-03, 8.4114e-04, 4.7064e-04, 4.7946e-04, 1.0710e-03, 6.2752e-04,
          5.3787e-04, 3.9887e-04, 2.8205e-04, 3.9864e-04, 4.7803e-04, 5.9080e-04,
          5.1842e-03, 4.7088e-04, 6.0749e-04, 3.3998e-04, 6.5994e-04, 1.0338e-03,
          1.8654e-03, 1.8492e-03, 5.3215e-04, 6.8140e-04, 1.0014e-03, 1.5802e-03,
          9.9182e-04, 4.7636e-04, 5.7125e-04, 3.5977e-04, 7.9250e-04, 5.1403e-04,
          3.3474e-04, 4.1962e-04, 1.0290e-03, 5.3120e-04, 6.0368e-04, 3.1204e-03,
          1.0357e-03, 6.7520e-04, 3.9530e-04, 3.6788e-04, 4.8685e-04, 9.6655e-04,
          4.7207e-04, 4.6349e-04, 3.7074e-04, 1.1082e-03, 2.0905e-03, 4.5037e-04,
          7.9060e-04, 8.1205e-04, 7.4911e-04, 5.8174e-04, 7.7057e-04, 5.6076e-04,
          4.6158e-04, 2.9135e-04, 1.8845e-03, 1.0519e-03, 2.5597e-03, 9.2554e-04,
          8.1348e-04, 9.6369e-04, 8.8501e-04, 1.0796e-03, 7.6103e-04, 3.8886e-04,
          5.6124e-04, 1.2894e-03, 7.8058e-04, 8.7881e-04, 1.4000e-03, 8.8978e-04,
          1.1892e-03, 6.7472e-04, 6.0320e-04, 5.9319e-04, 6.0272e-04, 9.9373e-04,
          1.0767e-03, 5.9652e-04, 7.6008e-04, 1.5745e-03, 2.2640e-03, 7.2622e-04,
          6.1321e-04, 1.7891e-03, 7.0715e-04, 4.0317e-04, 4.1580e-04, 4.4394e-04,
          4.3821e-04, 5.3167e-04, 3.4771e-03, 9.9945e-04, 1.4868e-03, 6.9714e-04,
          4.4370e-04, 1.3590e-03, 8.9598e-04, 6.8283e-04, 1.9102e-03, 2.3327e-03,
          8.2541e-04, 6.4564e-04, 2.2755e-03, 3.1910e-03, 1.1606e-03, 2.6512e-03,
          1.7624e-03, 7.2575e-04, 2.7542e-03, 2.3632e-03, 1.2617e-03, 1.5965e-03,
          4.0126e-04, 1.2245e-03, 7.2145e-04, 4.2191e-03, 1.0328e-03, 2.0142e-03,
          9.5701e-04, 7.4863e-04, 1.1511e-03, 6.6900e-04, 8.8453e-04, 1.7385e-03,
          2.0638e-03, 8.5205e-02, 2.7046e-03, 2.6741e-03, 1.2560e-03, 6.4802e-04,
          8.0824e-04, 9.4748e-04, 3.1590e-04, 4.7493e-04, 1.3647e-03, 4.1962e-04,
          1.0204e-03, 9.0647e-04, 7.6914e-04, 8.7595e-04, 6.6662e-04, 1.5278e-03,
          5.9795e-04, 2.1076e-03, 1.8139e-03, 6.3467e-04, 4.6501e-03, 1.2627e-03,
          1.4587e-01, 1.0103e-04, 1.0103e-04, 1.0103e-04, 2.7351e-03, 1.0103e-04,
          2.5768e-03, 7.5436e-04, 2.3537e-03, 4.7970e-04, 1.0103e-04, 4.7970e-04,
          2.5997e-03, 1.0103e-04, 7.5436e-04, 3.7551e-04, 2.3537e-03, 3.1204e-03,
          4.7970e-04, 4.1084e-03, 1.0103e-04, 1.0103e-04, 1.0103e-04, 1.0103e-04,
          3.2544e-04, 3.2544e-04, 7.0429e-04, 4.6873e-04, 3.2544e-04, 3.2544e-04,
          1.0103e-04, 3.2544e-04, 1.0103e-04, 1.0103e-04, 1.0103e-04, 1.0103e-04,
          1.0103e-04, 1.0103e-04], dtype=torch.float16),
  tensor([0.0000, 0.0042, 0.0428, 0.0181, 0.0204, 0.0743, 0.0304, 0.0203, 0.0006,
          0.0003, 0.0012, 0.0043, 0.0009, 0.0006, 0.0006, 0.0009, 0.0004, 0.0008,
          0.0004, 0.0002, 0.0005, 0.0004, 0.0004, 0.0005, 0.0035, 0.0013, 0.0138,
          0.0022, 0.0017, 0.0007, 0.0007, 0.0008, 0.0004, 0.0009, 0.0004, 0.0004,
          0.0012, 0.0014, 0.0006, 0.0005, 0.0005, 0.0007, 0.0009, 0.0006, 0.0009,
          0.0005, 0.0007, 0.0010, 0.0003, 0.0006, 0.0059, 0.0011, 0.0007, 0.0004,
          0.0006, 0.0005, 0.0014, 0.0018, 0.0015, 0.0025, 0.0032, 0.0006, 0.0009,
          0.0005, 0.0007, 0.0014, 0.0008, 0.0004, 0.0004, 0.0005, 0.0005, 0.0008,
          0.0010, 0.0019, 0.0017, 0.0009, 0.0005, 0.0003, 0.0008, 0.0016, 0.0009,
          0.0010, 0.0010, 0.0018, 0.0018, 0.0009, 0.0007, 0.0008, 0.0003, 0.0005,
          0.0016, 0.0009, 0.0004, 0.0011, 0.0012, 0.0016, 0.0012, 0.0011, 0.0004,
          0.0013, 0.0023, 0.0013, 0.0027, 0.0004, 0.0005, 0.0015, 0.0014, 0.0007,
          0.0008, 0.0007, 0.0012, 0.0005, 0.0007, 0.0004, 0.0009, 0.0008, 0.0004,
          0.0015, 0.0021, 0.0017, 0.0013, 0.0020, 0.0012, 0.0003, 0.0013, 0.0004,
          0.0012, 0.0007, 0.0008, 0.0012, 0.0004, 0.0008, 0.0007, 0.0004, 0.0002,
          0.0002, 0.0007, 0.0005, 0.0009, 0.0005, 0.0005, 0.0011, 0.0008, 0.0010,
          0.0018, 0.0007, 0.0009, 0.0012, 0.0010, 0.0016, 0.0023, 0.0013, 0.0016,
          0.0006, 0.0012, 0.0010, 0.0006, 0.0005, 0.0003, 0.0005, 0.0013, 0.0005,
          0.0010, 0.0006, 0.0008, 0.0009, 0.0042, 0.0012, 0.0019, 0.0011, 0.0009,
          0.0006, 0.0006, 0.0012, 0.0017, 0.0016, 0.0039, 0.0013, 0.0009, 0.0007,
          0.0006, 0.0005, 0.0009, 0.0005, 0.0004, 0.0005, 0.0008, 0.0005, 0.0008,
          0.0043, 0.0012, 0.0009, 0.0005, 0.0015, 0.0008, 0.0012, 0.0013, 0.0010,
          0.0037, 0.0013, 0.0010, 0.0006, 0.0006, 0.0005, 0.0007, 0.0007, 0.0009,
          0.0004, 0.0017, 0.0004, 0.0005, 0.0041, 0.0080, 0.0010, 0.0018, 0.0014,
          0.0012, 0.0009, 0.0008, 0.0005, 0.0011, 0.0008, 0.0015, 0.0013, 0.0010,
          0.0005, 0.0015, 0.0008, 0.0006, 0.0007, 0.0009, 0.0023, 0.0010, 0.0005,
          0.0125, 0.0016, 0.0019, 0.0012, 0.0041, 0.0057, 0.0050, 0.0056, 0.0023,
          0.0014, 0.0015, 0.0016, 0.0014, 0.0008, 0.0011, 0.0010, 0.0013, 0.0010,
          0.0006, 0.0013, 0.0024, 0.0010, 0.0007, 0.0018, 0.0068, 0.0012, 0.0020,
          0.0020, 0.0028, 0.0047, 0.0018, 0.0005, 0.0006, 0.0013, 0.0016, 0.0009,
          0.0014, 0.0010, 0.0014, 0.0010, 0.0015, 0.0016, 0.0029, 0.0013, 0.0012,
          0.0009, 0.0078, 0.0031, 0.0015, 0.0055, 0.0055, 0.0068, 0.0018, 0.0014,
          0.0028, 0.0017, 0.0008, 0.0010, 0.0025, 0.0024, 0.0020, 0.0012, 0.0009,
          0.0011, 0.0013, 0.0046, 0.0028, 0.0006, 0.0008, 0.0023, 0.0148, 0.0033,
          0.0017, 0.0043, 0.0020, 0.0005, 0.0006, 0.0006, 0.0006, 0.0015, 0.0061,
          0.0010, 0.0019, 0.0012, 0.0014, 0.0027, 0.0010, 0.0007, 0.0041, 0.0033,
          0.0009, 0.0006, 0.0019, 0.0395, 0.0089, 0.0034, 0.0040, 0.0012, 0.0039,
          0.0032, 0.0013, 0.0025, 0.0016, 0.0014, 0.0009, 0.0074, 0.0019, 0.0019,
          0.0014, 0.0009, 0.0012, 0.0007, 0.0011, 0.0025, 0.0021, 0.0117, 0.0286,
          0.0346, 0.0017, 0.0008, 0.0014, 0.0010, 0.0006, 0.0006, 0.0016, 0.0010,
          0.0011, 0.0010, 0.0013, 0.0008, 0.0008, 0.0013, 0.0007, 0.0020, 0.0017,
          0.0013, 0.0034, 0.0010, 0.0263, 0.0850, 0.0001, 0.0001, 0.0005, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0001, 0.0014, 0.0001, 0.0001, 0.0001,
          0.0006, 0.0001, 0.0014, 0.0001, 0.0004, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0004, 0.0004, 0.0002, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0047, 0.0441, 0.0161, 0.0193, 0.0629, 0.0257, 0.0166, 0.0006,
          0.0003, 0.0016, 0.0047, 0.0011, 0.0007, 0.0005, 0.0010, 0.0004, 0.0009,
          0.0005, 0.0002, 0.0003, 0.0004, 0.0003, 0.0006, 0.0042, 0.0013, 0.0118,
          0.0025, 0.0009, 0.0008, 0.0007, 0.0009, 0.0004, 0.0012, 0.0005, 0.0004,
          0.0013, 0.0012, 0.0010, 0.0007, 0.0004, 0.0012, 0.0009, 0.0006, 0.0007,
          0.0005, 0.0007, 0.0014, 0.0003, 0.0005, 0.0052, 0.0010, 0.0007, 0.0004,
          0.0005, 0.0005, 0.0009, 0.0017, 0.0012, 0.0020, 0.0025, 0.0006, 0.0008,
          0.0006, 0.0008, 0.0018, 0.0010, 0.0005, 0.0006, 0.0005, 0.0004, 0.0007,
          0.0009, 0.0030, 0.0011, 0.0010, 0.0004, 0.0003, 0.0007, 0.0017, 0.0012,
          0.0012, 0.0012, 0.0012, 0.0014, 0.0010, 0.0006, 0.0011, 0.0005, 0.0006,
          0.0023, 0.0015, 0.0004, 0.0015, 0.0009, 0.0012, 0.0008, 0.0007, 0.0005,
          0.0018, 0.0024, 0.0020, 0.0039, 0.0007, 0.0007, 0.0015, 0.0018, 0.0006,
          0.0008, 0.0007, 0.0022, 0.0004, 0.0009, 0.0007, 0.0009, 0.0009, 0.0004,
          0.0020, 0.0015, 0.0015, 0.0013, 0.0030, 0.0012, 0.0003, 0.0018, 0.0006,
          0.0014, 0.0009, 0.0010, 0.0015, 0.0005, 0.0009, 0.0010, 0.0006, 0.0002,
          0.0003, 0.0011, 0.0005, 0.0011, 0.0005, 0.0006, 0.0013, 0.0011, 0.0010,
          0.0018, 0.0006, 0.0010, 0.0016, 0.0013, 0.0020, 0.0038, 0.0015, 0.0017,
          0.0008, 0.0011, 0.0013, 0.0012, 0.0010, 0.0003, 0.0005, 0.0016, 0.0005,
          0.0016, 0.0012, 0.0010, 0.0012, 0.0058, 0.0021, 0.0033, 0.0019, 0.0013,
          0.0007, 0.0009, 0.0020, 0.0016, 0.0017, 0.0037, 0.0014, 0.0014, 0.0008,
          0.0008, 0.0006, 0.0013, 0.0007, 0.0005, 0.0006, 0.0011, 0.0005, 0.0009,
          0.0042, 0.0016, 0.0013, 0.0008, 0.0018, 0.0011, 0.0015, 0.0019, 0.0014,
          0.0039, 0.0011, 0.0014, 0.0008, 0.0009, 0.0009, 0.0008, 0.0011, 0.0010,
          0.0005, 0.0026, 0.0005, 0.0007, 0.0049, 0.0069, 0.0015, 0.0024, 0.0017,
          0.0019, 0.0011, 0.0012, 0.0005, 0.0017, 0.0009, 0.0023, 0.0013, 0.0012,
          0.0008, 0.0021, 0.0010, 0.0007, 0.0007, 0.0009, 0.0021, 0.0013, 0.0005,
          0.0101, 0.0024, 0.0020, 0.0013, 0.0042, 0.0059, 0.0047, 0.0052, 0.0030,
          0.0021, 0.0020, 0.0018, 0.0017, 0.0010, 0.0009, 0.0010, 0.0013, 0.0013,
          0.0006, 0.0011, 0.0021, 0.0015, 0.0009, 0.0027, 0.0078, 0.0018, 0.0024,
          0.0018, 0.0026, 0.0047, 0.0020, 0.0006, 0.0006, 0.0017, 0.0023, 0.0011,
          0.0016, 0.0013, 0.0018, 0.0012, 0.0027, 0.0019, 0.0026, 0.0013, 0.0010,
          0.0010, 0.0073, 0.0040, 0.0024, 0.0051, 0.0056, 0.0054, 0.0023, 0.0022,
          0.0027, 0.0028, 0.0011, 0.0012, 0.0024, 0.0026, 0.0022, 0.0014, 0.0011,
          0.0017, 0.0027, 0.0035, 0.0023, 0.0007, 0.0010, 0.0030, 0.0114, 0.0042,
          0.0022, 0.0036, 0.0032, 0.0007, 0.0008, 0.0008, 0.0008, 0.0018, 0.0054,
          0.0015, 0.0029, 0.0014, 0.0014, 0.0039, 0.0013, 0.0009, 0.0033, 0.0032,
          0.0010, 0.0007, 0.0023, 0.0248, 0.0095, 0.0047, 0.0073, 0.0012, 0.0050,
          0.0063, 0.0011, 0.0028, 0.0017, 0.0019, 0.0012, 0.0065, 0.0049, 0.0024,
          0.0015, 0.0012, 0.0015, 0.0009, 0.0014, 0.0025, 0.0018, 0.0106, 0.0206,
          0.0227, 0.0024, 0.0011, 0.0019, 0.0011, 0.0008, 0.0007, 0.0020, 0.0013,
          0.0015, 0.0014, 0.0020, 0.0010, 0.0008, 0.0021, 0.0008, 0.0021, 0.0018,
          0.0022, 0.0050, 0.0012, 0.0226, 0.0414, 0.0296, 0.0001, 0.0006, 0.0001,
          0.0003, 0.0001, 0.0003, 0.0001, 0.0001, 0.0042, 0.0006, 0.0001, 0.0004,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0027, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0003, 0.0003, 0.0003, 0.0002, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0033, 0.0573, 0.0128, 0.0104, 0.0310, 0.0458, 0.0076, 0.0027,
          0.0015, 0.0022, 0.0048, 0.0012, 0.0019, 0.0005, 0.0026, 0.0031, 0.0018,
          0.0014, 0.0017, 0.0001, 0.0001, 0.0003, 0.0024, 0.0043, 0.0012, 0.0063,
          0.0030, 0.0003, 0.0021, 0.0007, 0.0057, 0.0006, 0.0018, 0.0031, 0.0006,
          0.0007, 0.0002, 0.0011, 0.0020, 0.0002, 0.0005, 0.0002, 0.0002, 0.0003,
          0.0002, 0.0029, 0.0013, 0.0003, 0.0003, 0.0020, 0.0012, 0.0035, 0.0011,
          0.0003, 0.0017, 0.0003, 0.0002, 0.0008, 0.0003, 0.0004, 0.0007, 0.0006,
          0.0007, 0.0003, 0.0025, 0.0005, 0.0004, 0.0006, 0.0001, 0.0001, 0.0002,
          0.0002, 0.0041, 0.0004, 0.0051, 0.0004, 0.0004, 0.0016, 0.0002, 0.0003,
          0.0010, 0.0005, 0.0002, 0.0002, 0.0063, 0.0004, 0.0026, 0.0003, 0.0003,
          0.0038, 0.0012, 0.0002, 0.0015, 0.0002, 0.0002, 0.0006, 0.0003, 0.0014,
          0.0007, 0.0002, 0.0003, 0.0033, 0.0005, 0.0006, 0.0006, 0.0026, 0.0004,
          0.0045, 0.0002, 0.0011, 0.0002, 0.0003, 0.0004, 0.0010, 0.0003, 0.0003,
          0.0027, 0.0003, 0.0006, 0.0008, 0.0008, 0.0001, 0.0044, 0.0047, 0.0001,
          0.0002, 0.0002, 0.0003, 0.0025, 0.0034, 0.0070, 0.0006, 0.0004, 0.0041,
          0.0002, 0.0008, 0.0002, 0.0030, 0.0005, 0.0025, 0.0042, 0.0027, 0.0013,
          0.0002, 0.0011, 0.0003, 0.0002, 0.0004, 0.0019, 0.0022, 0.0018, 0.0003,
          0.0010, 0.0010, 0.0004, 0.0008, 0.0003, 0.0020, 0.0002, 0.0011, 0.0007,
          0.0015, 0.0014, 0.0045, 0.0062, 0.0003, 0.0003, 0.0017, 0.0004, 0.0012,
          0.0025, 0.0016, 0.0013, 0.0012, 0.0007, 0.0005, 0.0004, 0.0011, 0.0263,
          0.0009, 0.0017, 0.0028, 0.0080, 0.0087, 0.0024, 0.0045, 0.0008, 0.0032,
          0.0003, 0.0004, 0.0033, 0.0113, 0.0002, 0.0017, 0.0039, 0.0025, 0.0016,
          0.0006, 0.0004, 0.0020, 0.0005, 0.0014, 0.0008, 0.0003, 0.0007, 0.0003,
          0.0073, 0.0026, 0.0013, 0.0037, 0.0006, 0.0003, 0.0010, 0.0026, 0.0004,
          0.0007, 0.0015, 0.0067, 0.0028, 0.0028, 0.0013, 0.0020, 0.0003, 0.0006,
          0.0006, 0.0065, 0.0035, 0.0006, 0.0003, 0.0002, 0.0007, 0.0011, 0.0007,
          0.0006, 0.0004, 0.0004, 0.0017, 0.0010, 0.0028, 0.0008, 0.0006, 0.0003,
          0.0009, 0.0020, 0.0007, 0.0003, 0.0009, 0.0003, 0.0004, 0.0005, 0.0016,
          0.0002, 0.0002, 0.0004, 0.0005, 0.0060, 0.0032, 0.0005, 0.0006, 0.0008,
          0.0009, 0.0003, 0.0013, 0.0012, 0.0034, 0.0033, 0.0033, 0.0055, 0.0012,
          0.0007, 0.0009, 0.0033, 0.0005, 0.0012, 0.0016, 0.0009, 0.0013, 0.0004,
          0.0013, 0.0005, 0.0007, 0.0011, 0.0004, 0.0003, 0.0005, 0.0016, 0.0014,
          0.0016, 0.0045, 0.0047, 0.0054, 0.0005, 0.0008, 0.0005, 0.0003, 0.0002,
          0.0029, 0.0011, 0.0006, 0.0004, 0.0042, 0.0083, 0.0034, 0.0004, 0.0023,
          0.0025, 0.0006, 0.0015, 0.0021, 0.0051, 0.0052, 0.0021, 0.0013, 0.0011,
          0.0038, 0.0041, 0.0031, 0.0012, 0.0080, 0.0021, 0.0022, 0.0005, 0.0005,
          0.0032, 0.0014, 0.0040, 0.0006, 0.0017, 0.0037, 0.0028, 0.0018, 0.0023,
          0.0063, 0.0067, 0.0029, 0.0011, 0.0048, 0.0061, 0.0013, 0.0020, 0.0022,
          0.0034, 0.0065, 0.0023, 0.0040, 0.0034, 0.0017, 0.0007, 0.0043, 0.0011,
          0.0009, 0.0085, 0.0072, 0.0038, 0.0484, 0.0022, 0.0029, 0.0015, 0.0018,
          0.0056, 0.0056, 0.0025, 0.0035, 0.0094, 0.0056, 0.0060, 0.0012, 0.0148,
          0.0017, 0.0089, 0.0064, 0.0067, 0.0015, 0.0014, 0.0490, 0.0001, 0.0001,
          0.0008, 0.0001, 0.0003, 0.0001, 0.0001, 0.0001, 0.0008, 0.0001, 0.0003,
          0.0001, 0.0005, 0.0005, 0.0001, 0.0038, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0003, 0.0003, 0.0003, 0.0003, 0.0003, 0.0001, 0.0001, 0.0003, 0.0003,
          0.0003, 0.0001, 0.0003, 0.0003, 0.0001], dtype=torch.float16),
  tensor([0.0000e+00, 3.0422e-03, 5.2643e-02, 1.1711e-02, 1.2268e-02, 2.5558e-02,
          3.7537e-02, 6.4049e-03, 2.9793e-03, 1.1454e-03, 9.5224e-04, 1.5411e-03,
          6.7759e-04, 8.0442e-04, 5.2643e-04, 8.5926e-04, 3.2673e-03, 1.4877e-03,
          9.3555e-04, 1.1988e-03, 2.7132e-04, 2.8181e-04, 2.4068e-04, 1.3590e-03,
          1.8873e-03, 7.0715e-04, 5.6763e-03, 3.5877e-03, 5.2404e-04, 1.8272e-03,
          7.4768e-04, 1.0307e-02, 7.6389e-04, 1.6737e-03, 1.1053e-03, 4.3831e-03,
          1.4944e-03, 4.0483e-04, 1.9007e-03, 1.5345e-03, 3.9482e-04, 6.9046e-04,
          3.1924e-04, 2.5725e-04, 2.7990e-04, 2.2256e-04, 1.3208e-03, 1.4887e-03,
          3.5000e-04, 3.3355e-04, 1.5774e-03, 9.7733e-03, 9.5444e-03, 2.5845e-03,
          1.0815e-03, 5.0201e-03, 1.8415e-03, 5.6696e-04, 1.7309e-03, 7.5674e-04,
          8.4448e-04, 9.4366e-04, 2.3670e-03, 3.9864e-03, 3.1686e-04, 1.6823e-03,
          4.2081e-04, 2.5129e-04, 3.8052e-04, 1.7524e-04, 1.4150e-04, 2.6345e-04,
          3.1948e-04, 2.3327e-03, 2.8095e-03, 1.4297e-02, 1.4763e-03, 1.6232e-03,
          6.6185e-03, 5.8365e-04, 4.3583e-04, 1.0576e-03, 5.9891e-04, 5.1451e-04,
          4.2558e-04, 1.7242e-02, 1.6203e-03, 6.2847e-04, 2.3592e-04, 2.3329e-04,
          1.5173e-03, 1.0090e-03, 1.3530e-04, 1.0004e-03, 2.5415e-04, 3.9482e-04,
          1.3561e-03, 1.3161e-03, 1.3206e-02, 2.1210e-03, 4.6563e-04, 5.9128e-04,
          1.7347e-03, 3.2282e-04, 5.4789e-04, 1.4038e-03, 1.4668e-03, 6.2895e-04,
          1.2825e-02, 3.8171e-04, 1.2522e-03, 1.9419e-04, 3.2377e-04, 3.7909e-04,
          4.8709e-04, 5.5265e-04, 3.9601e-04, 2.0103e-03, 4.5156e-04, 9.2125e-04,
          1.1921e-03, 1.3580e-03, 2.5010e-04, 1.7443e-03, 1.6670e-03, 2.1636e-04,
          3.8338e-04, 3.5620e-04, 4.6134e-04, 2.0390e-03, 4.1733e-03, 1.5610e-02,
          5.5265e-04, 2.9969e-04, 3.7289e-04, 2.2948e-04, 6.7472e-04, 2.0719e-04,
          3.4943e-03, 6.5279e-04, 1.3227e-03, 4.3602e-03, 3.1586e-03, 1.3924e-03,
          3.4714e-04, 3.4285e-04, 4.2796e-04, 5.1785e-04, 4.4632e-04, 1.5650e-03,
          2.6894e-03, 1.6270e-03, 8.3828e-04, 1.2413e-02, 6.5651e-03, 1.0223e-03,
          9.7656e-04, 8.9550e-04, 4.4107e-04, 2.5153e-04, 1.4362e-03, 1.4887e-03,
          1.7958e-03, 2.0199e-03, 2.6703e-03, 1.6928e-03, 9.4795e-04, 6.9952e-04,
          1.2598e-03, 8.9931e-04, 9.4509e-04, 7.4530e-04, 1.6804e-03, 2.4414e-03,
          1.7118e-03, 1.2350e-03, 1.1568e-03, 7.7677e-04, 1.2903e-03, 2.1954e-03,
          7.3576e-04, 7.9489e-04, 1.5888e-03, 5.9280e-03, 1.3857e-03, 7.3242e-04,
          1.2350e-03, 1.0052e-03, 1.0519e-03, 5.5265e-04, 4.7898e-04, 1.5402e-03,
          8.8348e-03, 2.8515e-04, 1.3876e-03, 2.6798e-03, 1.2636e-03, 1.5621e-03,
          1.4715e-03, 8.1253e-04, 1.5078e-03, 4.1461e-04, 7.7391e-04, 7.6056e-04,
          2.9445e-04, 7.3338e-04, 2.9612e-04, 1.4439e-03, 1.7433e-03, 1.1091e-03,
          3.5801e-03, 1.6022e-03, 6.1321e-04, 8.3160e-04, 1.8711e-03, 4.6015e-04,
          1.8387e-03, 2.3479e-03, 2.0313e-03, 8.4543e-04, 2.9755e-03, 2.0676e-03,
          5.1956e-03, 3.5453e-04, 6.6471e-04, 5.4264e-04, 2.3861e-03, 1.3247e-03,
          4.1533e-04, 3.2759e-04, 2.7227e-04, 9.5558e-04, 9.6941e-04, 9.7942e-04,
          1.2226e-03, 7.2765e-04, 4.7851e-04, 1.1110e-03, 1.7405e-03, 2.2793e-03,
          1.1501e-03, 9.5892e-04, 8.0967e-04, 2.7084e-03, 3.3092e-03, 1.0643e-03,
          7.3338e-04, 5.7888e-04, 3.4165e-04, 4.2772e-04, 5.4598e-04, 1.1015e-03,
          2.6989e-04, 2.6941e-04, 4.9400e-04, 6.8903e-04, 2.0142e-03, 1.6144e-02,
          1.0900e-03, 7.9107e-04, 1.2188e-03, 1.1711e-03, 5.2643e-04, 1.6861e-03,
          1.5354e-03, 1.0881e-03, 1.7548e-03, 4.5662e-03, 3.0727e-03, 9.0456e-04,
          7.7534e-04, 6.0034e-04, 1.5011e-03, 4.9114e-04, 2.5063e-03, 1.3275e-03,
          8.5592e-04, 1.0071e-03, 6.4516e-04, 1.4992e-03, 9.6178e-04, 9.0122e-04,
          8.5306e-04, 6.5994e-04, 4.8566e-04, 8.3065e-04, 2.6360e-03, 1.7633e-03,
          1.7681e-03, 3.3493e-03, 2.6741e-03, 2.7466e-03, 6.2847e-04, 9.6941e-04,
          7.0381e-04, 3.9339e-04, 3.0613e-04, 2.1534e-03, 2.3384e-03, 6.1560e-04,
          4.4036e-04, 1.3952e-03, 8.2474e-03, 4.6463e-03, 7.9679e-04, 2.3575e-03,
          2.1000e-03, 1.5774e-03, 2.8439e-03, 2.2144e-03, 1.3161e-03, 1.5812e-03,
          1.8930e-03, 2.0313e-03, 1.7500e-03, 2.9182e-03, 4.6730e-03, 1.6203e-03,
          9.9754e-04, 3.6526e-03, 2.6741e-03, 1.5440e-03, 5.4979e-04, 6.5947e-04,
          1.5268e-03, 1.7242e-03, 2.8763e-03, 1.4172e-03, 2.3670e-03, 3.7498e-03,
          4.3564e-03, 1.5688e-03, 2.5578e-03, 6.0158e-03, 1.4725e-03, 3.8643e-03,
          1.8616e-03, 1.6800e-02, 1.2306e-02, 1.5411e-03, 2.4891e-03, 2.1229e-03,
          1.3456e-03, 1.4734e-03, 2.8172e-03, 1.3819e-03, 1.5793e-02, 2.2964e-03,
          8.2350e-04, 3.9673e-03, 1.7242e-03, 1.4162e-03, 3.2654e-02, 1.7227e-02,
          2.3315e-02, 2.3422e-03, 9.7942e-04, 1.1692e-03, 1.4925e-03, 2.2392e-03,
          2.2049e-02, 1.7914e-02, 8.1253e-03, 1.8301e-03, 2.7313e-03, 2.6760e-03,
          1.4744e-03, 1.1082e-03, 3.6869e-03, 1.6899e-03, 7.2403e-03, 1.9569e-03,
          6.3629e-03, 2.3232e-03, 2.0828e-03, 3.3112e-03, 2.0584e-02, 9.7215e-05,
          9.7215e-05, 5.5552e-04, 2.3103e-04, 9.7215e-05, 9.7215e-05, 2.3103e-04,
          9.7215e-05, 9.7215e-05, 2.3103e-04, 9.7215e-05, 9.7215e-05, 2.3103e-04,
          2.3103e-04, 9.7215e-05, 9.7215e-05, 2.3103e-04, 2.3103e-04, 9.7215e-05,
          2.3103e-04, 2.3103e-04, 9.7215e-05, 2.3103e-04, 3.6979e-04, 9.7215e-05,
          9.7215e-05, 2.3103e-04, 2.3103e-04, 4.1580e-04, 9.7215e-05, 4.1580e-04,
          2.3103e-04, 2.3103e-04], dtype=torch.float16),
  tensor([0.0000, 0.0073, 0.0407, 0.0183, 0.0233, 0.0845, 0.0231, 0.0244, 0.0005,
          0.0004, 0.0007, 0.0013, 0.0005, 0.0016, 0.0021, 0.0007, 0.0004, 0.0044,
          0.0018, 0.0002, 0.0016, 0.0012, 0.0013, 0.0009, 0.0012, 0.0006, 0.0159,
          0.0035, 0.0033, 0.0024, 0.0022, 0.0008, 0.0015, 0.0009, 0.0004, 0.0006,
          0.0008, 0.0006, 0.0011, 0.0006, 0.0021, 0.0008, 0.0008, 0.0005, 0.0006,
          0.0004, 0.0019, 0.0011, 0.0011, 0.0020, 0.0109, 0.0011, 0.0006, 0.0004,
          0.0005, 0.0005, 0.0009, 0.0006, 0.0006, 0.0006, 0.0008, 0.0037, 0.0008,
          0.0007, 0.0004, 0.0011, 0.0006, 0.0002, 0.0004, 0.0004, 0.0004, 0.0005,
          0.0007, 0.0012, 0.0011, 0.0008, 0.0004, 0.0003, 0.0008, 0.0007, 0.0005,
          0.0004, 0.0004, 0.0005, 0.0007, 0.0011, 0.0006, 0.0004, 0.0002, 0.0004,
          0.0007, 0.0004, 0.0003, 0.0005, 0.0004, 0.0008, 0.0021, 0.0008, 0.0006,
          0.0011, 0.0009, 0.0005, 0.0008, 0.0003, 0.0003, 0.0006, 0.0008, 0.0034,
          0.0008, 0.0007, 0.0006, 0.0004, 0.0004, 0.0003, 0.0004, 0.0007, 0.0027,
          0.0012, 0.0010, 0.0046, 0.0045, 0.0007, 0.0006, 0.0003, 0.0010, 0.0003,
          0.0006, 0.0003, 0.0004, 0.0008, 0.0005, 0.0008, 0.0006, 0.0003, 0.0002,
          0.0002, 0.0004, 0.0005, 0.0026, 0.0028, 0.0019, 0.0016, 0.0014, 0.0036,
          0.0007, 0.0003, 0.0004, 0.0004, 0.0005, 0.0008, 0.0008, 0.0007, 0.0006,
          0.0011, 0.0012, 0.0011, 0.0004, 0.0005, 0.0002, 0.0004, 0.0011, 0.0030,
          0.0010, 0.0008, 0.0024, 0.0013, 0.0025, 0.0005, 0.0007, 0.0004, 0.0005,
          0.0005, 0.0020, 0.0011, 0.0005, 0.0006, 0.0018, 0.0018, 0.0007, 0.0008,
          0.0004, 0.0005, 0.0009, 0.0006, 0.0005, 0.0007, 0.0005, 0.0029, 0.0012,
          0.0015, 0.0007, 0.0009, 0.0007, 0.0006, 0.0007, 0.0033, 0.0005, 0.0008,
          0.0013, 0.0046, 0.0010, 0.0004, 0.0006, 0.0005, 0.0005, 0.0005, 0.0007,
          0.0004, 0.0009, 0.0010, 0.0006, 0.0036, 0.0024, 0.0005, 0.0010, 0.0005,
          0.0008, 0.0036, 0.0009, 0.0006, 0.0015, 0.0028, 0.0034, 0.0012, 0.0006,
          0.0004, 0.0014, 0.0008, 0.0004, 0.0004, 0.0004, 0.0007, 0.0008, 0.0026,
          0.0047, 0.0006, 0.0006, 0.0005, 0.0014, 0.0019, 0.0015, 0.0022, 0.0010,
          0.0022, 0.0064, 0.0115, 0.0020, 0.0005, 0.0007, 0.0004, 0.0009, 0.0007,
          0.0004, 0.0007, 0.0014, 0.0009, 0.0013, 0.0030, 0.0020, 0.0006, 0.0007,
          0.0006, 0.0007, 0.0014, 0.0010, 0.0006, 0.0017, 0.0057, 0.0051, 0.0007,
          0.0010, 0.0007, 0.0011, 0.0009, 0.0013, 0.0009, 0.0008, 0.0005, 0.0087,
          0.0035, 0.0039, 0.0012, 0.0008, 0.0015, 0.0016, 0.0017, 0.0022, 0.0009,
          0.0011, 0.0018, 0.0022, 0.0029, 0.0021, 0.0013, 0.0020, 0.0011, 0.0010,
          0.0012, 0.0007, 0.0014, 0.0015, 0.0010, 0.0009, 0.0069, 0.0036, 0.0012,
          0.0012, 0.0090, 0.0017, 0.0013, 0.0006, 0.0006, 0.0019, 0.0011, 0.0227,
          0.0013, 0.0032, 0.0011, 0.0008, 0.0015, 0.0037, 0.0023, 0.0033, 0.0037,
          0.0029, 0.0019, 0.0042, 0.0070, 0.0028, 0.0105, 0.0022, 0.0021, 0.0121,
          0.0042, 0.0011, 0.0046, 0.0008, 0.0021, 0.0012, 0.0273, 0.0012, 0.0073,
          0.0009, 0.0010, 0.0064, 0.0006, 0.0013, 0.0051, 0.0069, 0.0199, 0.0048,
          0.0045, 0.0021, 0.0010, 0.0015, 0.0011, 0.0004, 0.0007, 0.0045, 0.0006,
          0.0015, 0.0014, 0.0013, 0.0015, 0.0007, 0.0015, 0.0008, 0.0080, 0.0018,
          0.0006, 0.0037, 0.0011, 0.0367, 0.0116, 0.0061, 0.0012, 0.0009, 0.0737,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0066, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0004, 0.0004, 0.0001, 0.0001, 0.0004, 0.0001, 0.0001, 0.0004, 0.0004,
          0.0004, 0.0001, 0.0001, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0029, 0.0460, 0.0114, 0.0107, 0.0266, 0.0327, 0.0069, 0.0021,
          0.0008, 0.0007, 0.0011, 0.0005, 0.0006, 0.0006, 0.0006, 0.0030, 0.0014,
          0.0007, 0.0008, 0.0005, 0.0004, 0.0003, 0.0008, 0.0015, 0.0007, 0.0060,
          0.0023, 0.0007, 0.0014, 0.0007, 0.0094, 0.0007, 0.0010, 0.0006, 0.0032,
          0.0013, 0.0007, 0.0012, 0.0008, 0.0008, 0.0008, 0.0005, 0.0003, 0.0004,
          0.0003, 0.0011, 0.0009, 0.0004, 0.0004, 0.0019, 0.0118, 0.0095, 0.0014,
          0.0022, 0.0058, 0.0039, 0.0008, 0.0023, 0.0011, 0.0014, 0.0017, 0.0114,
          0.0057, 0.0004, 0.0012, 0.0005, 0.0003, 0.0004, 0.0003, 0.0002, 0.0003,
          0.0004, 0.0013, 0.0071, 0.0153, 0.0030, 0.0043, 0.0088, 0.0006, 0.0005,
          0.0008, 0.0006, 0.0007, 0.0007, 0.0309, 0.0074, 0.0005, 0.0003, 0.0003,
          0.0010, 0.0007, 0.0002, 0.0007, 0.0003, 0.0004, 0.0014, 0.0029, 0.0093,
          0.0017, 0.0007, 0.0006, 0.0012, 0.0003, 0.0004, 0.0015, 0.0012, 0.0011,
          0.0188, 0.0007, 0.0010, 0.0003, 0.0004, 0.0004, 0.0005, 0.0007, 0.0004,
          0.0015, 0.0006, 0.0011, 0.0014, 0.0011, 0.0003, 0.0013, 0.0012, 0.0002,
          0.0005, 0.0005, 0.0005, 0.0015, 0.0030, 0.0185, 0.0008, 0.0004, 0.0002,
          0.0002, 0.0006, 0.0004, 0.0017, 0.0006, 0.0012, 0.0023, 0.0015, 0.0013,
          0.0005, 0.0003, 0.0004, 0.0006, 0.0005, 0.0013, 0.0021, 0.0015, 0.0012,
          0.0189, 0.0205, 0.0016, 0.0008, 0.0012, 0.0003, 0.0004, 0.0015, 0.0010,
          0.0011, 0.0013, 0.0015, 0.0010, 0.0012, 0.0009, 0.0009, 0.0009, 0.0007,
          0.0005, 0.0009, 0.0018, 0.0020, 0.0017, 0.0017, 0.0014, 0.0014, 0.0011,
          0.0006, 0.0007, 0.0011, 0.0049, 0.0013, 0.0006, 0.0009, 0.0011, 0.0009,
          0.0007, 0.0005, 0.0011, 0.0048, 0.0004, 0.0009, 0.0018, 0.0008, 0.0012,
          0.0023, 0.0016, 0.0015, 0.0005, 0.0007, 0.0006, 0.0004, 0.0007, 0.0004,
          0.0009, 0.0014, 0.0007, 0.0034, 0.0017, 0.0007, 0.0007, 0.0014, 0.0005,
          0.0016, 0.0015, 0.0014, 0.0005, 0.0017, 0.0020, 0.0043, 0.0007, 0.0009,
          0.0006, 0.0019, 0.0010, 0.0005, 0.0004, 0.0003, 0.0012, 0.0010, 0.0009,
          0.0012, 0.0008, 0.0005, 0.0010, 0.0016, 0.0029, 0.0014, 0.0012, 0.0011,
          0.0027, 0.0028, 0.0015, 0.0013, 0.0007, 0.0006, 0.0007, 0.0007, 0.0009,
          0.0004, 0.0003, 0.0006, 0.0009, 0.0013, 0.0089, 0.0014, 0.0008, 0.0011,
          0.0009, 0.0006, 0.0019, 0.0013, 0.0008, 0.0011, 0.0033, 0.0025, 0.0010,
          0.0011, 0.0006, 0.0012, 0.0007, 0.0019, 0.0012, 0.0011, 0.0011, 0.0006,
          0.0015, 0.0011, 0.0010, 0.0008, 0.0007, 0.0006, 0.0009, 0.0021, 0.0016,
          0.0020, 0.0018, 0.0017, 0.0022, 0.0012, 0.0019, 0.0011, 0.0006, 0.0005,
          0.0013, 0.0020, 0.0009, 0.0006, 0.0011, 0.0060, 0.0032, 0.0008, 0.0017,
          0.0017, 0.0016, 0.0026, 0.0018, 0.0010, 0.0011, 0.0014, 0.0021, 0.0026,
          0.0033, 0.0028, 0.0014, 0.0014, 0.0025, 0.0017, 0.0014, 0.0008, 0.0008,
          0.0015, 0.0014, 0.0021, 0.0013, 0.0023, 0.0033, 0.0038, 0.0016, 0.0026,
          0.0048, 0.0011, 0.0058, 0.0026, 0.0152, 0.0112, 0.0020, 0.0025, 0.0022,
          0.0011, 0.0011, 0.0021, 0.0010, 0.0163, 0.0019, 0.0009, 0.0045, 0.0019,
          0.0014, 0.0228, 0.0090, 0.0175, 0.0014, 0.0007, 0.0009, 0.0015, 0.0022,
          0.0242, 0.0169, 0.0092, 0.0015, 0.0020, 0.0018, 0.0011, 0.0012, 0.0032,
          0.0022, 0.0050, 0.0016, 0.0072, 0.0026, 0.0022, 0.0021, 0.0104, 0.0030,
          0.0298, 0.0002, 0.0005, 0.0002, 0.0001, 0.0002, 0.0001, 0.0001, 0.0002,
          0.0001, 0.0002, 0.0004, 0.0002, 0.0001, 0.0001, 0.0002, 0.0002, 0.0001,
          0.0005, 0.0002, 0.0003, 0.0002, 0.0002, 0.0002, 0.0001, 0.0002, 0.0003,
          0.0004, 0.0002, 0.0004, 0.0003, 0.0002], dtype=torch.float16),
  tensor([0.0000, 0.0041, 0.0415, 0.0131, 0.0248, 0.0361, 0.0301, 0.0148, 0.0009,
          0.0005, 0.0014, 0.0029, 0.0009, 0.0026, 0.0034, 0.0007, 0.0008, 0.0064,
          0.0042, 0.0005, 0.0003, 0.0003, 0.0004, 0.0033, 0.0029, 0.0006, 0.0116,
          0.0062, 0.0007, 0.0044, 0.0084, 0.0017, 0.0015, 0.0010, 0.0012, 0.0005,
          0.0006, 0.0003, 0.0009, 0.0008, 0.0007, 0.0005, 0.0003, 0.0002, 0.0003,
          0.0002, 0.0056, 0.0009, 0.0014, 0.0009, 0.0031, 0.0008, 0.0013, 0.0006,
          0.0004, 0.0007, 0.0004, 0.0003, 0.0008, 0.0004, 0.0005, 0.0016, 0.0005,
          0.0004, 0.0004, 0.0012, 0.0010, 0.0006, 0.0006, 0.0001, 0.0001, 0.0003,
          0.0003, 0.0031, 0.0006, 0.0016, 0.0004, 0.0004, 0.0009, 0.0003, 0.0006,
          0.0007, 0.0008, 0.0003, 0.0003, 0.0016, 0.0005, 0.0012, 0.0007, 0.0008,
          0.0023, 0.0008, 0.0003, 0.0013, 0.0002, 0.0003, 0.0012, 0.0003, 0.0011,
          0.0008, 0.0003, 0.0003, 0.0021, 0.0005, 0.0006, 0.0004, 0.0008, 0.0014,
          0.0011, 0.0003, 0.0010, 0.0003, 0.0007, 0.0007, 0.0004, 0.0003, 0.0006,
          0.0012, 0.0004, 0.0011, 0.0060, 0.0007, 0.0002, 0.0006, 0.0012, 0.0002,
          0.0002, 0.0005, 0.0006, 0.0009, 0.0016, 0.0015, 0.0010, 0.0007, 0.0005,
          0.0004, 0.0008, 0.0003, 0.0049, 0.0034, 0.0025, 0.0032, 0.0026, 0.0078,
          0.0003, 0.0002, 0.0003, 0.0003, 0.0010, 0.0008, 0.0015, 0.0008, 0.0003,
          0.0008, 0.0009, 0.0005, 0.0008, 0.0005, 0.0007, 0.0005, 0.0008, 0.0035,
          0.0010, 0.0011, 0.0084, 0.0018, 0.0006, 0.0005, 0.0014, 0.0005, 0.0006,
          0.0005, 0.0050, 0.0009, 0.0005, 0.0005, 0.0006, 0.0006, 0.0008, 0.0030,
          0.0006, 0.0007, 0.0013, 0.0018, 0.0012, 0.0007, 0.0018, 0.0045, 0.0015,
          0.0005, 0.0012, 0.0012, 0.0018, 0.0002, 0.0009, 0.0091, 0.0015, 0.0009,
          0.0006, 0.0012, 0.0011, 0.0008, 0.0013, 0.0006, 0.0007, 0.0008, 0.0003,
          0.0007, 0.0021, 0.0017, 0.0009, 0.0009, 0.0006, 0.0010, 0.0014, 0.0006,
          0.0007, 0.0033, 0.0027, 0.0006, 0.0016, 0.0059, 0.0027, 0.0003, 0.0006,
          0.0010, 0.0022, 0.0015, 0.0011, 0.0005, 0.0003, 0.0005, 0.0006, 0.0028,
          0.0013, 0.0006, 0.0007, 0.0005, 0.0009, 0.0016, 0.0010, 0.0009, 0.0003,
          0.0009, 0.0060, 0.0056, 0.0005, 0.0011, 0.0003, 0.0004, 0.0005, 0.0010,
          0.0005, 0.0003, 0.0005, 0.0004, 0.0030, 0.0032, 0.0007, 0.0012, 0.0004,
          0.0005, 0.0003, 0.0011, 0.0008, 0.0010, 0.0023, 0.0113, 0.0515, 0.0007,
          0.0012, 0.0023, 0.0014, 0.0005, 0.0012, 0.0011, 0.0006, 0.0005, 0.0015,
          0.0167, 0.0008, 0.0015, 0.0019, 0.0005, 0.0004, 0.0005, 0.0019, 0.0008,
          0.0009, 0.0029, 0.0047, 0.0065, 0.0007, 0.0009, 0.0008, 0.0005, 0.0003,
          0.0014, 0.0010, 0.0007, 0.0006, 0.0017, 0.0019, 0.0075, 0.0009, 0.0014,
          0.0013, 0.0014, 0.0013, 0.0021, 0.0007, 0.0010, 0.0025, 0.0007, 0.0034,
          0.0026, 0.0043, 0.0016, 0.0006, 0.0044, 0.0033, 0.0033, 0.0007, 0.0008,
          0.0046, 0.0046, 0.0134, 0.0012, 0.0016, 0.0334, 0.0028, 0.0019, 0.0238,
          0.0061, 0.0024, 0.0037, 0.0006, 0.0029, 0.0017, 0.0049, 0.0020, 0.0215,
          0.0018, 0.0015, 0.0080, 0.0017, 0.0021, 0.0037, 0.0042, 0.0088, 0.0018,
          0.0014, 0.0043, 0.0023, 0.0020, 0.0025, 0.0006, 0.0007, 0.0153, 0.0006,
          0.0024, 0.0023, 0.0017, 0.0068, 0.0011, 0.0035, 0.0011, 0.0086, 0.0029,
          0.0010, 0.0079, 0.0022, 0.0155, 0.0028, 0.0021, 0.0022, 0.0013, 0.0072,
          0.0015, 0.0427, 0.0003, 0.0001, 0.0001, 0.0001, 0.0016, 0.0001, 0.0001,
          0.0001, 0.0003, 0.0004, 0.0001, 0.0034, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0006, 0.0003, 0.0002, 0.0004, 0.0003, 0.0002, 0.0001, 0.0004, 0.0002,
          0.0004, 0.0002, 0.0001, 0.0003, 0.0002], dtype=torch.float16),
  tensor([0.0000, 0.0049, 0.0472, 0.0110, 0.0191, 0.0298, 0.0324, 0.0106, 0.0010,
          0.0005, 0.0037, 0.0085, 0.0033, 0.0010, 0.0004, 0.0014, 0.0010, 0.0015,
          0.0010, 0.0005, 0.0001, 0.0002, 0.0002, 0.0016, 0.0101, 0.0013, 0.0087,
          0.0043, 0.0003, 0.0019, 0.0007, 0.0024, 0.0006, 0.0023, 0.0037, 0.0007,
          0.0014, 0.0004, 0.0019, 0.0024, 0.0002, 0.0017, 0.0004, 0.0003, 0.0004,
          0.0005, 0.0019, 0.0022, 0.0002, 0.0002, 0.0021, 0.0009, 0.0017, 0.0007,
          0.0002, 0.0008, 0.0002, 0.0007, 0.0011, 0.0011, 0.0007, 0.0005, 0.0004,
          0.0006, 0.0009, 0.0040, 0.0013, 0.0012, 0.0011, 0.0003, 0.0002, 0.0004,
          0.0004, 0.0087, 0.0004, 0.0022, 0.0003, 0.0003, 0.0006, 0.0006, 0.0013,
          0.0025, 0.0014, 0.0005, 0.0003, 0.0021, 0.0003, 0.0052, 0.0009, 0.0005,
          0.0094, 0.0039, 0.0002, 0.0052, 0.0002, 0.0003, 0.0004, 0.0002, 0.0015,
          0.0012, 0.0005, 0.0010, 0.0120, 0.0009, 0.0017, 0.0014, 0.0024, 0.0004,
          0.0015, 0.0005, 0.0056, 0.0003, 0.0008, 0.0015, 0.0009, 0.0006, 0.0002,
          0.0033, 0.0004, 0.0004, 0.0007, 0.0023, 0.0003, 0.0007, 0.0039, 0.0004,
          0.0003, 0.0005, 0.0007, 0.0029, 0.0022, 0.0021, 0.0014, 0.0014, 0.0010,
          0.0005, 0.0031, 0.0003, 0.0045, 0.0004, 0.0011, 0.0031, 0.0030, 0.0011,
          0.0005, 0.0008, 0.0007, 0.0008, 0.0011, 0.0025, 0.0056, 0.0023, 0.0007,
          0.0011, 0.0007, 0.0015, 0.0031, 0.0012, 0.0031, 0.0004, 0.0020, 0.0008,
          0.0036, 0.0029, 0.0042, 0.0033, 0.0008, 0.0009, 0.0101, 0.0014, 0.0018,
          0.0013, 0.0018, 0.0034, 0.0012, 0.0007, 0.0007, 0.0009, 0.0025, 0.0037,
          0.0016, 0.0014, 0.0038, 0.0034, 0.0011, 0.0010, 0.0093, 0.0005, 0.0024,
          0.0007, 0.0009, 0.0030, 0.0037, 0.0003, 0.0027, 0.0046, 0.0095, 0.0030,
          0.0007, 0.0004, 0.0024, 0.0017, 0.0031, 0.0017, 0.0007, 0.0027, 0.0004,
          0.0012, 0.0114, 0.0012, 0.0012, 0.0010, 0.0007, 0.0038, 0.0035, 0.0008,
          0.0015, 0.0015, 0.0067, 0.0014, 0.0039, 0.0016, 0.0028, 0.0004, 0.0017,
          0.0019, 0.0061, 0.0037, 0.0019, 0.0004, 0.0004, 0.0006, 0.0011, 0.0005,
          0.0012, 0.0010, 0.0007, 0.0013, 0.0017, 0.0031, 0.0013, 0.0008, 0.0005,
          0.0010, 0.0024, 0.0008, 0.0008, 0.0019, 0.0003, 0.0005, 0.0007, 0.0024,
          0.0004, 0.0003, 0.0004, 0.0006, 0.0028, 0.0052, 0.0010, 0.0018, 0.0009,
          0.0011, 0.0004, 0.0018, 0.0016, 0.0029, 0.0015, 0.0044, 0.0049, 0.0012,
          0.0011, 0.0022, 0.0035, 0.0007, 0.0030, 0.0019, 0.0008, 0.0008, 0.0003,
          0.0013, 0.0007, 0.0015, 0.0030, 0.0006, 0.0004, 0.0005, 0.0018, 0.0013,
          0.0019, 0.0075, 0.0032, 0.0034, 0.0006, 0.0011, 0.0009, 0.0006, 0.0004,
          0.0037, 0.0040, 0.0006, 0.0004, 0.0015, 0.0032, 0.0026, 0.0009, 0.0049,
          0.0030, 0.0006, 0.0021, 0.0009, 0.0019, 0.0038, 0.0011, 0.0009, 0.0011,
          0.0031, 0.0052, 0.0029, 0.0009, 0.0301, 0.0017, 0.0015, 0.0005, 0.0006,
          0.0014, 0.0009, 0.0037, 0.0013, 0.0023, 0.0046, 0.0083, 0.0010, 0.0028,
          0.0131, 0.0053, 0.0015, 0.0009, 0.0046, 0.0031, 0.0013, 0.0061, 0.0024,
          0.0045, 0.0033, 0.0028, 0.0048, 0.0040, 0.0014, 0.0005, 0.0050, 0.0016,
          0.0015, 0.0087, 0.0039, 0.0041, 0.0029, 0.0028, 0.0016, 0.0016, 0.0013,
          0.0036, 0.0035, 0.0027, 0.0025, 0.0017, 0.0083, 0.0023, 0.0011, 0.0031,
          0.0030, 0.0112, 0.0037, 0.0080, 0.0019, 0.0021, 0.0039, 0.0021, 0.0025,
          0.0016, 0.0043, 0.0309, 0.0002, 0.0002, 0.0002, 0.0009, 0.0002, 0.0003,
          0.0002, 0.0002, 0.0008, 0.0002, 0.0017, 0.0002, 0.0003, 0.0002, 0.0002,
          0.0003, 0.0002, 0.0002, 0.0002, 0.0003, 0.0003, 0.0002, 0.0005, 0.0003,
          0.0005, 0.0003, 0.0005, 0.0002, 0.0002], dtype=torch.float16),
  tensor([0.0000, 0.0042, 0.0486, 0.0140, 0.0135, 0.0312, 0.0412, 0.0097, 0.0020,
          0.0009, 0.0029, 0.0078, 0.0016, 0.0056, 0.0009, 0.0021, 0.0021, 0.0061,
          0.0037, 0.0011, 0.0002, 0.0003, 0.0006, 0.0064, 0.0072, 0.0010, 0.0074,
          0.0042, 0.0004, 0.0066, 0.0014, 0.0038, 0.0018, 0.0024, 0.0036, 0.0006,
          0.0007, 0.0002, 0.0016, 0.0022, 0.0003, 0.0007, 0.0003, 0.0002, 0.0003,
          0.0002, 0.0079, 0.0020, 0.0006, 0.0005, 0.0028, 0.0010, 0.0020, 0.0007,
          0.0002, 0.0008, 0.0002, 0.0002, 0.0007, 0.0004, 0.0004, 0.0011, 0.0004,
          0.0007, 0.0003, 0.0025, 0.0007, 0.0006, 0.0008, 0.0001, 0.0001, 0.0002,
          0.0003, 0.0057, 0.0003, 0.0029, 0.0002, 0.0002, 0.0008, 0.0002, 0.0004,
          0.0010, 0.0006, 0.0002, 0.0002, 0.0022, 0.0003, 0.0032, 0.0004, 0.0003,
          0.0054, 0.0017, 0.0002, 0.0025, 0.0002, 0.0002, 0.0008, 0.0002, 0.0012,
          0.0012, 0.0002, 0.0003, 0.0043, 0.0005, 0.0007, 0.0005, 0.0020, 0.0006,
          0.0015, 0.0002, 0.0015, 0.0002, 0.0004, 0.0005, 0.0008, 0.0003, 0.0004,
          0.0029, 0.0003, 0.0007, 0.0010, 0.0009, 0.0001, 0.0012, 0.0035, 0.0001,
          0.0001, 0.0002, 0.0003, 0.0019, 0.0023, 0.0024, 0.0007, 0.0007, 0.0020,
          0.0003, 0.0010, 0.0002, 0.0107, 0.0007, 0.0056, 0.0048, 0.0041, 0.0016,
          0.0002, 0.0010, 0.0003, 0.0002, 0.0005, 0.0012, 0.0024, 0.0012, 0.0003,
          0.0012, 0.0008, 0.0004, 0.0009, 0.0004, 0.0018, 0.0003, 0.0012, 0.0015,
          0.0024, 0.0021, 0.0115, 0.0064, 0.0004, 0.0003, 0.0021, 0.0005, 0.0009,
          0.0018, 0.0030, 0.0016, 0.0007, 0.0004, 0.0005, 0.0004, 0.0010, 0.0103,
          0.0009, 0.0013, 0.0025, 0.0041, 0.0053, 0.0037, 0.0047, 0.0014, 0.0045,
          0.0003, 0.0004, 0.0026, 0.0039, 0.0002, 0.0019, 0.0104, 0.0031, 0.0016,
          0.0005, 0.0005, 0.0017, 0.0005, 0.0016, 0.0009, 0.0003, 0.0008, 0.0003,
          0.0021, 0.0036, 0.0039, 0.0018, 0.0007, 0.0004, 0.0009, 0.0021, 0.0003,
          0.0007, 0.0023, 0.0076, 0.0026, 0.0038, 0.0018, 0.0026, 0.0002, 0.0005,
          0.0008, 0.0044, 0.0031, 0.0009, 0.0003, 0.0002, 0.0006, 0.0012, 0.0010,
          0.0007, 0.0005, 0.0003, 0.0010, 0.0008, 0.0016, 0.0006, 0.0005, 0.0003,
          0.0011, 0.0023, 0.0009, 0.0004, 0.0009, 0.0002, 0.0004, 0.0004, 0.0012,
          0.0002, 0.0002, 0.0004, 0.0006, 0.0116, 0.0037, 0.0005, 0.0007, 0.0005,
          0.0006, 0.0002, 0.0007, 0.0010, 0.0033, 0.0029, 0.0049, 0.0069, 0.0008,
          0.0007, 0.0012, 0.0023, 0.0004, 0.0013, 0.0013, 0.0006, 0.0011, 0.0007,
          0.0018, 0.0005, 0.0006, 0.0013, 0.0003, 0.0003, 0.0004, 0.0014, 0.0013,
          0.0011, 0.0054, 0.0123, 0.0111, 0.0004, 0.0007, 0.0005, 0.0003, 0.0002,
          0.0033, 0.0011, 0.0005, 0.0004, 0.0056, 0.0034, 0.0048, 0.0004, 0.0020,
          0.0023, 0.0006, 0.0015, 0.0017, 0.0031, 0.0037, 0.0029, 0.0009, 0.0014,
          0.0031, 0.0069, 0.0022, 0.0009, 0.0088, 0.0036, 0.0035, 0.0004, 0.0005,
          0.0073, 0.0018, 0.0079, 0.0006, 0.0015, 0.0042, 0.0033, 0.0020, 0.0024,
          0.0066, 0.0114, 0.0028, 0.0007, 0.0048, 0.0033, 0.0019, 0.0022, 0.0023,
          0.0048, 0.0044, 0.0043, 0.0050, 0.0031, 0.0029, 0.0009, 0.0066, 0.0009,
          0.0008, 0.0052, 0.0036, 0.0027, 0.0081, 0.0018, 0.0021, 0.0015, 0.0013,
          0.0034, 0.0039, 0.0020, 0.0041, 0.0030, 0.0061, 0.0031, 0.0015, 0.0061,
          0.0019, 0.0101, 0.0054, 0.0092, 0.0013, 0.0012, 0.0079, 0.0021, 0.0025,
          0.0019, 0.0052, 0.0064, 0.0471, 0.0001, 0.0001, 0.0009, 0.0001, 0.0003,
          0.0001, 0.0001, 0.0007, 0.0001, 0.0049, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0003, 0.0003, 0.0001, 0.0001, 0.0003, 0.0001, 0.0001, 0.0003, 0.0003,
          0.0003, 0.0001, 0.0003, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0052, 0.0460, 0.0126, 0.0146, 0.0350, 0.0276, 0.0097, 0.0005,
          0.0003, 0.0006, 0.0012, 0.0004, 0.0013, 0.0015, 0.0007, 0.0005, 0.0038,
          0.0017, 0.0003, 0.0005, 0.0004, 0.0006, 0.0017, 0.0012, 0.0006, 0.0076,
          0.0038, 0.0010, 0.0027, 0.0031, 0.0010, 0.0009, 0.0011, 0.0005, 0.0007,
          0.0007, 0.0004, 0.0009, 0.0007, 0.0009, 0.0005, 0.0004, 0.0003, 0.0004,
          0.0002, 0.0030, 0.0013, 0.0009, 0.0012, 0.0034, 0.0010, 0.0007, 0.0006,
          0.0003, 0.0005, 0.0005, 0.0004, 0.0006, 0.0005, 0.0006, 0.0026, 0.0006,
          0.0007, 0.0003, 0.0014, 0.0006, 0.0003, 0.0004, 0.0003, 0.0002, 0.0003,
          0.0004, 0.0017, 0.0006, 0.0011, 0.0004, 0.0003, 0.0007, 0.0004, 0.0004,
          0.0004, 0.0003, 0.0003, 0.0004, 0.0015, 0.0005, 0.0005, 0.0003, 0.0003,
          0.0009, 0.0006, 0.0003, 0.0006, 0.0003, 0.0004, 0.0012, 0.0004, 0.0012,
          0.0014, 0.0004, 0.0004, 0.0009, 0.0004, 0.0002, 0.0006, 0.0010, 0.0030,
          0.0012, 0.0005, 0.0008, 0.0004, 0.0004, 0.0004, 0.0005, 0.0005, 0.0014,
          0.0016, 0.0006, 0.0018, 0.0040, 0.0008, 0.0002, 0.0004, 0.0012, 0.0002,
          0.0003, 0.0003, 0.0004, 0.0012, 0.0011, 0.0013, 0.0007, 0.0003, 0.0002,
          0.0002, 0.0004, 0.0004, 0.0068, 0.0041, 0.0035, 0.0025, 0.0024, 0.0052,
          0.0004, 0.0002, 0.0004, 0.0003, 0.0005, 0.0010, 0.0011, 0.0009, 0.0005,
          0.0012, 0.0013, 0.0009, 0.0006, 0.0006, 0.0003, 0.0004, 0.0014, 0.0062,
          0.0017, 0.0013, 0.0063, 0.0016, 0.0009, 0.0005, 0.0007, 0.0005, 0.0007,
          0.0006, 0.0038, 0.0015, 0.0007, 0.0006, 0.0011, 0.0010, 0.0010, 0.0013,
          0.0006, 0.0006, 0.0013, 0.0011, 0.0009, 0.0009, 0.0006, 0.0062, 0.0012,
          0.0006, 0.0006, 0.0011, 0.0010, 0.0003, 0.0012, 0.0083, 0.0007, 0.0011,
          0.0011, 0.0026, 0.0013, 0.0005, 0.0008, 0.0007, 0.0004, 0.0006, 0.0004,
          0.0004, 0.0011, 0.0020, 0.0007, 0.0018, 0.0007, 0.0004, 0.0015, 0.0005,
          0.0011, 0.0062, 0.0011, 0.0005, 0.0028, 0.0072, 0.0047, 0.0006, 0.0007,
          0.0006, 0.0022, 0.0011, 0.0005, 0.0004, 0.0003, 0.0006, 0.0010, 0.0064,
          0.0013, 0.0006, 0.0006, 0.0006, 0.0016, 0.0023, 0.0013, 0.0014, 0.0009,
          0.0029, 0.0131, 0.0069, 0.0011, 0.0007, 0.0005, 0.0006, 0.0008, 0.0011,
          0.0004, 0.0004, 0.0008, 0.0009, 0.0033, 0.0041, 0.0012, 0.0008, 0.0006,
          0.0008, 0.0006, 0.0015, 0.0014, 0.0007, 0.0026, 0.0179, 0.0184, 0.0010,
          0.0012, 0.0008, 0.0015, 0.0007, 0.0019, 0.0014, 0.0010, 0.0007, 0.0037,
          0.0109, 0.0012, 0.0010, 0.0008, 0.0007, 0.0007, 0.0009, 0.0058, 0.0019,
          0.0013, 0.0030, 0.0051, 0.0061, 0.0012, 0.0016, 0.0013, 0.0007, 0.0006,
          0.0024, 0.0014, 0.0011, 0.0009, 0.0020, 0.0014, 0.0145, 0.0012, 0.0019,
          0.0020, 0.0037, 0.0028, 0.0041, 0.0007, 0.0007, 0.0059, 0.0021, 0.0103,
          0.0022, 0.0070, 0.0017, 0.0012, 0.0023, 0.0089, 0.0062, 0.0014, 0.0017,
          0.0055, 0.0075, 0.0084, 0.0023, 0.0034, 0.0262, 0.0037, 0.0054, 0.0138,
          0.0058, 0.0012, 0.0098, 0.0011, 0.0039, 0.0027, 0.0165, 0.0020, 0.0154,
          0.0012, 0.0014, 0.0160, 0.0010, 0.0031, 0.0100, 0.0046, 0.0069, 0.0027,
          0.0022, 0.0047, 0.0018, 0.0031, 0.0015, 0.0006, 0.0008, 0.0068, 0.0009,
          0.0031, 0.0030, 0.0023, 0.0041, 0.0010, 0.0026, 0.0010, 0.0077, 0.0026,
          0.0011, 0.0067, 0.0020, 0.0130, 0.0044, 0.0037, 0.0019, 0.0018, 0.0136,
          0.0028, 0.0083, 0.0022, 0.0034, 0.0310, 0.0001, 0.0003, 0.0001, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0003, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0006, 0.0006, 0.0003, 0.0003, 0.0006, 0.0003, 0.0001, 0.0003, 0.0006,
          0.0003, 0.0003, 0.0003, 0.0003, 0.0003], dtype=torch.float16),
  tensor([0.0000e+00, 4.1695e-03, 6.1096e-02, 1.0826e-02, 1.6785e-02, 2.5131e-02,
          5.0262e-02, 1.2299e-02, 4.4274e-04, 2.2566e-04, 1.6451e-03, 2.2144e-03,
          7.6437e-04, 6.5422e-04, 7.1764e-04, 1.9875e-03, 4.2701e-04, 1.0738e-03,
          5.9509e-04, 2.7418e-04, 1.4329e-04, 1.6308e-04, 1.6844e-04, 8.4114e-04,
          2.5253e-03, 2.3098e-03, 9.2010e-03, 2.4414e-03, 2.2602e-04, 1.0424e-03,
          1.5602e-03, 1.2026e-03, 2.8729e-04, 2.0332e-03, 1.9646e-03, 3.6931e-04,
          9.1267e-04, 2.4939e-04, 2.0294e-03, 1.7853e-03, 2.1803e-04, 5.0402e-04,
          2.6298e-04, 1.9681e-04, 2.4652e-04, 1.6081e-04, 1.1168e-03, 2.4662e-03,
          3.6383e-04, 2.9182e-04, 1.6441e-03, 4.7231e-04, 6.1417e-04, 4.8566e-04,
          1.4496e-04, 3.5501e-04, 2.0325e-04, 2.3508e-04, 6.8712e-04, 4.7421e-04,
          4.6897e-04, 6.6233e-04, 3.0756e-04, 3.9363e-04, 3.5572e-04, 4.1351e-03,
          9.7847e-04, 3.4475e-04, 5.5933e-04, 1.5545e-04, 1.3435e-04, 2.4152e-04,
          2.6441e-04, 3.8090e-03, 2.2376e-04, 1.0242e-03, 1.7953e-04, 1.3876e-04,
          3.5930e-04, 2.3758e-04, 4.5514e-04, 1.2121e-03, 5.0735e-04, 4.0555e-04,
          3.5596e-04, 1.2360e-03, 2.9230e-04, 1.1101e-03, 5.4407e-04, 5.6982e-04,
          2.1019e-03, 1.1778e-03, 2.2304e-04, 1.1435e-03, 2.6751e-04, 2.8944e-04,
          2.7728e-04, 1.6201e-04, 7.0286e-04, 1.1873e-03, 2.4223e-04, 3.6478e-04,
          2.0447e-03, 6.1560e-04, 4.3726e-04, 1.6308e-03, 3.3073e-03, 5.2309e-04,
          9.9754e-04, 2.9802e-04, 1.5202e-03, 3.3474e-04, 8.3160e-04, 6.7806e-04,
          3.4790e-03, 5.0735e-04, 3.1161e-04, 4.0321e-03, 4.1199e-04, 4.2439e-04,
          9.7656e-04, 1.9798e-03, 1.6809e-04, 3.0637e-04, 4.3030e-03, 1.8561e-04,
          2.1636e-04, 3.4356e-04, 4.2844e-04, 3.1548e-03, 1.5430e-03, 1.3371e-03,
          1.2283e-03, 5.1069e-04, 1.3294e-03, 2.9635e-04, 9.3317e-04, 2.2221e-04,
          3.3875e-03, 1.0080e-03, 2.0294e-03, 4.1275e-03, 3.8776e-03, 2.0599e-03,
          2.5678e-04, 1.6146e-03, 5.4169e-04, 3.1948e-04, 7.5531e-04, 4.8637e-03,
          2.7828e-03, 2.2678e-03, 4.4775e-04, 6.1321e-04, 6.0129e-04, 7.4339e-04,
          1.1826e-03, 6.9046e-04, 6.8855e-04, 3.3092e-04, 2.4033e-03, 9.4795e-04,
          3.4523e-03, 3.2597e-03, 2.0752e-03, 8.4915e-03, 5.3501e-04, 7.2193e-04,
          1.1320e-03, 7.7248e-04, 2.6493e-03, 2.7637e-03, 1.3008e-03, 2.6970e-03,
          1.0948e-03, 5.8317e-04, 6.6376e-04, 4.2844e-04, 1.8511e-03, 2.0809e-03,
          1.7357e-03, 1.4334e-03, 3.9482e-03, 1.1644e-03, 1.1578e-03, 3.7994e-03,
          1.8797e-03, 1.2360e-03, 6.3896e-03, 3.6192e-04, 7.1383e-04, 2.8439e-03,
          1.1663e-03, 1.7488e-04, 3.9368e-03, 2.3575e-03, 1.7300e-03, 2.1400e-03,
          6.8331e-04, 5.4264e-04, 3.1223e-03, 5.3930e-04, 1.2789e-03, 1.6050e-03,
          4.5204e-04, 7.7248e-04, 3.4809e-04, 1.0233e-03, 2.3270e-03, 8.6880e-04,
          1.0376e-03, 7.2813e-04, 3.7837e-04, 6.5756e-04, 3.9749e-03, 4.9019e-04,
          1.6060e-03, 1.8129e-03, 6.5308e-03, 3.4275e-03, 4.4823e-03, 1.0157e-03,
          7.9575e-03, 2.9516e-04, 7.7009e-04, 1.0586e-03, 7.8506e-03, 3.4180e-03,
          6.2466e-04, 5.2977e-04, 3.2258e-04, 1.2188e-03, 1.7176e-03, 7.9918e-04,
          6.6137e-04, 5.9128e-04, 5.6934e-04, 9.3155e-03, 1.7271e-03, 2.4681e-03,
          7.6437e-04, 6.8092e-04, 5.9032e-04, 9.9277e-04, 1.9331e-03, 1.7653e-03,
          4.5252e-04, 9.2411e-04, 2.9802e-04, 5.0020e-04, 5.5981e-04, 2.7905e-03,
          4.1676e-04, 2.5320e-04, 4.3869e-04, 8.0872e-04, 3.4084e-03, 5.2986e-03,
          6.0654e-04, 9.5749e-04, 1.1082e-03, 2.7008e-03, 2.6488e-04, 1.1339e-03,
          2.0370e-03, 4.3602e-03, 2.0046e-03, 3.0098e-03, 9.3002e-03, 1.6851e-03,
          1.0815e-03, 9.5367e-04, 4.6577e-03, 6.5088e-04, 3.8986e-03, 4.3640e-03,
          1.2617e-03, 1.6212e-03, 6.3229e-04, 4.5853e-03, 5.2309e-04, 8.5688e-04,
          9.7275e-04, 4.0054e-04, 3.1185e-04, 3.7360e-04, 2.5616e-03, 2.7313e-03,
          1.6842e-03, 7.9193e-03, 2.1591e-03, 2.3041e-03, 5.1785e-04, 8.7738e-04,
          7.3290e-04, 4.4584e-04, 3.4475e-04, 4.9171e-03, 3.2005e-03, 7.6056e-04,
          5.5504e-04, 2.4929e-03, 2.4509e-03, 2.7943e-03, 5.3072e-04, 3.0441e-03,
          4.9896e-03, 7.4291e-04, 2.5063e-03, 1.2589e-03, 5.6534e-03, 2.0866e-03,
          1.9550e-03, 1.7481e-03, 1.8244e-03, 3.3989e-03, 6.2447e-03, 3.7842e-03,
          1.6270e-03, 5.7220e-03, 2.5654e-03, 1.6508e-03, 7.4005e-04, 8.6498e-04,
          2.1706e-03, 1.4048e-03, 3.0727e-03, 7.1573e-04, 2.4052e-03, 9.6512e-03,
          5.1041e-03, 1.2589e-03, 7.6904e-03, 7.8430e-03, 6.5193e-03, 3.5076e-03,
          1.4009e-03, 6.7215e-03, 3.3054e-03, 2.1420e-03, 3.1185e-03, 8.5831e-03,
          5.1689e-03, 5.7182e-03, 2.4853e-03, 3.6793e-03, 3.8700e-03, 2.5330e-03,
          1.6546e-03, 4.9706e-03, 1.1301e-03, 9.0313e-04, 3.1834e-03, 2.0809e-03,
          3.8376e-03, 8.3771e-03, 4.1771e-03, 1.0948e-02, 5.9586e-03, 5.5695e-03,
          2.1706e-03, 1.9569e-03, 5.4741e-03, 7.9498e-03, 2.5558e-03, 8.4915e-03,
          4.7989e-03, 3.7003e-03, 5.2452e-03, 2.4929e-03, 1.6846e-02, 1.1055e-02,
          7.9041e-03, 1.5125e-03, 1.5688e-03, 4.9973e-03, 1.8520e-03, 3.3207e-03,
          1.8349e-03, 1.3443e-02, 5.5733e-03, 5.9853e-03, 3.8471e-03, 4.2145e-02,
          2.5535e-04, 9.7334e-05, 9.7334e-05, 9.7334e-05, 9.7334e-05, 9.7334e-05,
          9.7334e-05, 3.4523e-03, 9.7334e-05, 9.7334e-05, 9.7334e-05, 9.7334e-05,
          2.5535e-04, 2.5535e-04, 2.5535e-04, 2.5535e-04, 2.5535e-04, 9.7334e-05,
          9.7334e-05, 2.5535e-04, 2.5535e-04, 4.0174e-04, 2.5535e-04, 2.5535e-04,
          2.5535e-04, 2.5535e-04], dtype=torch.float16),
  tensor([0.0000, 0.0028, 0.0505, 0.0092, 0.0106, 0.0257, 0.0307, 0.0077, 0.0048,
          0.0008, 0.0009, 0.0018, 0.0006, 0.0006, 0.0005, 0.0006, 0.0042, 0.0010,
          0.0006, 0.0012, 0.0002, 0.0002, 0.0002, 0.0009, 0.0023, 0.0007, 0.0066,
          0.0021, 0.0003, 0.0010, 0.0009, 0.0138, 0.0004, 0.0008, 0.0007, 0.0017,
          0.0006, 0.0003, 0.0008, 0.0008, 0.0002, 0.0004, 0.0003, 0.0003, 0.0003,
          0.0002, 0.0010, 0.0008, 0.0003, 0.0003, 0.0017, 0.0038, 0.0103, 0.0014,
          0.0031, 0.0047, 0.0014, 0.0003, 0.0018, 0.0005, 0.0007, 0.0005, 0.0031,
          0.0018, 0.0004, 0.0012, 0.0013, 0.0005, 0.0004, 0.0002, 0.0001, 0.0003,
          0.0003, 0.0017, 0.0027, 0.0206, 0.0061, 0.0029, 0.0079, 0.0002, 0.0004,
          0.0006, 0.0005, 0.0004, 0.0004, 0.0206, 0.0037, 0.0009, 0.0005, 0.0005,
          0.0018, 0.0007, 0.0002, 0.0011, 0.0002, 0.0003, 0.0004, 0.0011, 0.0042,
          0.0006, 0.0002, 0.0003, 0.0017, 0.0007, 0.0004, 0.0007, 0.0010, 0.0004,
          0.0135, 0.0003, 0.0011, 0.0004, 0.0009, 0.0005, 0.0006, 0.0003, 0.0002,
          0.0015, 0.0004, 0.0005, 0.0006, 0.0008, 0.0002, 0.0038, 0.0012, 0.0002,
          0.0002, 0.0003, 0.0004, 0.0014, 0.0036, 0.0160, 0.0011, 0.0005, 0.0005,
          0.0003, 0.0007, 0.0003, 0.0013, 0.0005, 0.0019, 0.0029, 0.0013, 0.0008,
          0.0003, 0.0006, 0.0003, 0.0003, 0.0006, 0.0012, 0.0016, 0.0012, 0.0006,
          0.0038, 0.0155, 0.0005, 0.0006, 0.0004, 0.0011, 0.0004, 0.0011, 0.0007,
          0.0010, 0.0010, 0.0015, 0.0015, 0.0003, 0.0004, 0.0013, 0.0005, 0.0007,
          0.0008, 0.0007, 0.0008, 0.0017, 0.0017, 0.0009, 0.0006, 0.0010, 0.0020,
          0.0005, 0.0009, 0.0011, 0.0113, 0.0034, 0.0011, 0.0020, 0.0016, 0.0015,
          0.0004, 0.0005, 0.0014, 0.0090, 0.0002, 0.0009, 0.0018, 0.0013, 0.0011,
          0.0015, 0.0006, 0.0014, 0.0005, 0.0011, 0.0006, 0.0004, 0.0007, 0.0003,
          0.0025, 0.0023, 0.0007, 0.0118, 0.0005, 0.0003, 0.0008, 0.0015, 0.0004,
          0.0006, 0.0008, 0.0021, 0.0009, 0.0016, 0.0016, 0.0016, 0.0004, 0.0007,
          0.0009, 0.0024, 0.0015, 0.0008, 0.0008, 0.0004, 0.0013, 0.0012, 0.0010,
          0.0006, 0.0005, 0.0007, 0.0013, 0.0010, 0.0026, 0.0011, 0.0010, 0.0005,
          0.0008, 0.0019, 0.0007, 0.0005, 0.0012, 0.0006, 0.0009, 0.0012, 0.0012,
          0.0004, 0.0003, 0.0007, 0.0006, 0.0023, 0.0047, 0.0005, 0.0008, 0.0007,
          0.0009, 0.0004, 0.0016, 0.0013, 0.0014, 0.0021, 0.0040, 0.0049, 0.0011,
          0.0028, 0.0012, 0.0018, 0.0007, 0.0011, 0.0017, 0.0015, 0.0020, 0.0004,
          0.0019, 0.0004, 0.0010, 0.0013, 0.0004, 0.0003, 0.0006, 0.0014, 0.0015,
          0.0022, 0.0022, 0.0020, 0.0035, 0.0008, 0.0015, 0.0009, 0.0005, 0.0004,
          0.0019, 0.0014, 0.0015, 0.0007, 0.0025, 0.0106, 0.0023, 0.0005, 0.0019,
          0.0020, 0.0006, 0.0013, 0.0023, 0.0020, 0.0027, 0.0018, 0.0020, 0.0013,
          0.0043, 0.0033, 0.0024, 0.0022, 0.0056, 0.0017, 0.0018, 0.0008, 0.0008,
          0.0029, 0.0022, 0.0031, 0.0006, 0.0016, 0.0031, 0.0030, 0.0019, 0.0019,
          0.0043, 0.0027, 0.0031, 0.0025, 0.0108, 0.0115, 0.0016, 0.0026, 0.0025,
          0.0029, 0.0023, 0.0021, 0.0026, 0.0145, 0.0016, 0.0008, 0.0050, 0.0013,
          0.0011, 0.0224, 0.0110, 0.0117, 0.0029, 0.0016, 0.0022, 0.0017, 0.0029,
          0.0221, 0.0289, 0.0086, 0.0029, 0.0039, 0.0033, 0.0026, 0.0013, 0.0050,
          0.0027, 0.0075, 0.0032, 0.0071, 0.0017, 0.0015, 0.0041, 0.0077, 0.0020,
          0.0122, 0.0032, 0.0050, 0.0041, 0.0028, 0.0023, 0.0319, 0.0001, 0.0002,
          0.0001, 0.0001, 0.0002, 0.0002, 0.0001, 0.0001, 0.0002, 0.0001, 0.0001,
          0.0002, 0.0002, 0.0001, 0.0002, 0.0002, 0.0001, 0.0001, 0.0002, 0.0002,
          0.0005, 0.0004, 0.0004, 0.0002, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0063, 0.0423, 0.0161, 0.0189, 0.0717, 0.0261, 0.0193, 0.0005,
          0.0002, 0.0005, 0.0010, 0.0003, 0.0013, 0.0020, 0.0005, 0.0004, 0.0031,
          0.0014, 0.0002, 0.0014, 0.0012, 0.0013, 0.0010, 0.0011, 0.0005, 0.0136,
          0.0029, 0.0022, 0.0016, 0.0020, 0.0007, 0.0010, 0.0007, 0.0003, 0.0004,
          0.0006, 0.0005, 0.0008, 0.0005, 0.0018, 0.0007, 0.0008, 0.0006, 0.0007,
          0.0003, 0.0017, 0.0009, 0.0010, 0.0020, 0.0110, 0.0008, 0.0005, 0.0003,
          0.0004, 0.0004, 0.0007, 0.0004, 0.0006, 0.0005, 0.0007, 0.0031, 0.0009,
          0.0008, 0.0004, 0.0010, 0.0006, 0.0002, 0.0004, 0.0005, 0.0005, 0.0005,
          0.0008, 0.0009, 0.0008, 0.0006, 0.0004, 0.0003, 0.0007, 0.0004, 0.0004,
          0.0003, 0.0003, 0.0004, 0.0006, 0.0011, 0.0007, 0.0004, 0.0002, 0.0004,
          0.0006, 0.0004, 0.0003, 0.0005, 0.0005, 0.0009, 0.0017, 0.0005, 0.0004,
          0.0006, 0.0004, 0.0003, 0.0006, 0.0002, 0.0002, 0.0006, 0.0006, 0.0030,
          0.0009, 0.0007, 0.0005, 0.0005, 0.0005, 0.0003, 0.0005, 0.0008, 0.0025,
          0.0011, 0.0012, 0.0029, 0.0030, 0.0004, 0.0003, 0.0003, 0.0007, 0.0002,
          0.0004, 0.0003, 0.0003, 0.0007, 0.0005, 0.0009, 0.0006, 0.0003, 0.0002,
          0.0002, 0.0003, 0.0006, 0.0019, 0.0028, 0.0018, 0.0015, 0.0011, 0.0023,
          0.0004, 0.0002, 0.0003, 0.0002, 0.0004, 0.0006, 0.0006, 0.0006, 0.0005,
          0.0011, 0.0014, 0.0011, 0.0003, 0.0005, 0.0002, 0.0005, 0.0011, 0.0035,
          0.0009, 0.0006, 0.0021, 0.0009, 0.0009, 0.0003, 0.0004, 0.0002, 0.0004,
          0.0003, 0.0013, 0.0008, 0.0005, 0.0005, 0.0016, 0.0018, 0.0007, 0.0008,
          0.0004, 0.0005, 0.0008, 0.0006, 0.0005, 0.0005, 0.0005, 0.0039, 0.0009,
          0.0007, 0.0004, 0.0006, 0.0005, 0.0003, 0.0005, 0.0025, 0.0004, 0.0006,
          0.0011, 0.0036, 0.0011, 0.0004, 0.0005, 0.0005, 0.0005, 0.0005, 0.0009,
          0.0003, 0.0008, 0.0009, 0.0005, 0.0017, 0.0011, 0.0003, 0.0007, 0.0004,
          0.0005, 0.0021, 0.0007, 0.0003, 0.0010, 0.0031, 0.0026, 0.0013, 0.0007,
          0.0004, 0.0013, 0.0006, 0.0005, 0.0005, 0.0006, 0.0007, 0.0009, 0.0037,
          0.0023, 0.0003, 0.0004, 0.0004, 0.0009, 0.0014, 0.0011, 0.0016, 0.0008,
          0.0016, 0.0067, 0.0101, 0.0021, 0.0007, 0.0008, 0.0006, 0.0014, 0.0009,
          0.0006, 0.0010, 0.0019, 0.0011, 0.0011, 0.0020, 0.0010, 0.0004, 0.0004,
          0.0005, 0.0005, 0.0010, 0.0008, 0.0004, 0.0015, 0.0067, 0.0052, 0.0009,
          0.0012, 0.0007, 0.0012, 0.0012, 0.0013, 0.0012, 0.0011, 0.0005, 0.0120,
          0.0043, 0.0018, 0.0007, 0.0005, 0.0009, 0.0008, 0.0011, 0.0019, 0.0007,
          0.0008, 0.0012, 0.0016, 0.0030, 0.0023, 0.0015, 0.0024, 0.0013, 0.0013,
          0.0011, 0.0009, 0.0021, 0.0021, 0.0010, 0.0009, 0.0049, 0.0020, 0.0008,
          0.0009, 0.0052, 0.0013, 0.0010, 0.0005, 0.0004, 0.0015, 0.0010, 0.0241,
          0.0013, 0.0030, 0.0014, 0.0010, 0.0014, 0.0044, 0.0028, 0.0044, 0.0047,
          0.0032, 0.0027, 0.0043, 0.0034, 0.0018, 0.0080, 0.0016, 0.0018, 0.0100,
          0.0030, 0.0009, 0.0046, 0.0006, 0.0016, 0.0010, 0.0467, 0.0012, 0.0104,
          0.0008, 0.0008, 0.0083, 0.0006, 0.0012, 0.0064, 0.0111, 0.0185, 0.0028,
          0.0024, 0.0016, 0.0007, 0.0011, 0.0009, 0.0003, 0.0006, 0.0043, 0.0005,
          0.0014, 0.0013, 0.0011, 0.0016, 0.0008, 0.0013, 0.0007, 0.0151, 0.0019,
          0.0006, 0.0037, 0.0013, 0.0352, 0.0065, 0.0038, 0.0009, 0.0006, 0.0512,
          0.0009, 0.0035, 0.0011, 0.0017, 0.0105, 0.0013, 0.0006, 0.0681, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0001, 0.0035, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0005, 0.0005, 0.0002, 0.0005, 0.0001, 0.0001, 0.0005, 0.0001,
          0.0006, 0.0001, 0.0001, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0037, 0.0490, 0.0122, 0.0211, 0.0310, 0.0314, 0.0135, 0.0006,
          0.0002, 0.0007, 0.0014, 0.0005, 0.0019, 0.0027, 0.0005, 0.0006, 0.0056,
          0.0028, 0.0003, 0.0004, 0.0003, 0.0004, 0.0017, 0.0016, 0.0005, 0.0106,
          0.0046, 0.0006, 0.0037, 0.0069, 0.0014, 0.0010, 0.0006, 0.0005, 0.0004,
          0.0004, 0.0003, 0.0007, 0.0005, 0.0007, 0.0004, 0.0003, 0.0002, 0.0003,
          0.0002, 0.0062, 0.0010, 0.0014, 0.0009, 0.0025, 0.0006, 0.0009, 0.0004,
          0.0003, 0.0005, 0.0003, 0.0002, 0.0006, 0.0004, 0.0005, 0.0020, 0.0005,
          0.0004, 0.0003, 0.0011, 0.0008, 0.0004, 0.0004, 0.0001, 0.0002, 0.0003,
          0.0004, 0.0019, 0.0005, 0.0014, 0.0004, 0.0003, 0.0008, 0.0002, 0.0004,
          0.0004, 0.0005, 0.0002, 0.0003, 0.0015, 0.0005, 0.0006, 0.0004, 0.0007,
          0.0014, 0.0006, 0.0002, 0.0009, 0.0002, 0.0003, 0.0009, 0.0003, 0.0012,
          0.0008, 0.0002, 0.0002, 0.0011, 0.0003, 0.0004, 0.0004, 0.0006, 0.0016,
          0.0011, 0.0003, 0.0008, 0.0002, 0.0006, 0.0005, 0.0003, 0.0003, 0.0007,
          0.0013, 0.0005, 0.0009, 0.0052, 0.0004, 0.0002, 0.0004, 0.0009, 0.0002,
          0.0001, 0.0004, 0.0004, 0.0008, 0.0011, 0.0014, 0.0007, 0.0005, 0.0003,
          0.0003, 0.0006, 0.0003, 0.0074, 0.0042, 0.0030, 0.0028, 0.0021, 0.0075,
          0.0002, 0.0001, 0.0002, 0.0002, 0.0007, 0.0006, 0.0010, 0.0006, 0.0003,
          0.0009, 0.0010, 0.0006, 0.0007, 0.0004, 0.0004, 0.0005, 0.0010, 0.0054,
          0.0013, 0.0012, 0.0097, 0.0012, 0.0004, 0.0003, 0.0008, 0.0004, 0.0004,
          0.0004, 0.0043, 0.0008, 0.0004, 0.0005, 0.0006, 0.0006, 0.0008, 0.0018,
          0.0006, 0.0006, 0.0011, 0.0015, 0.0008, 0.0008, 0.0010, 0.0057, 0.0010,
          0.0003, 0.0009, 0.0008, 0.0012, 0.0002, 0.0007, 0.0107, 0.0008, 0.0008,
          0.0006, 0.0014, 0.0011, 0.0008, 0.0010, 0.0006, 0.0008, 0.0008, 0.0004,
          0.0004, 0.0017, 0.0019, 0.0008, 0.0006, 0.0004, 0.0006, 0.0010, 0.0005,
          0.0005, 0.0033, 0.0015, 0.0004, 0.0015, 0.0072, 0.0030, 0.0004, 0.0007,
          0.0008, 0.0021, 0.0011, 0.0009, 0.0005, 0.0003, 0.0005, 0.0006, 0.0039,
          0.0008, 0.0004, 0.0004, 0.0003, 0.0006, 0.0011, 0.0008, 0.0007, 0.0003,
          0.0007, 0.0067, 0.0063, 0.0006, 0.0012, 0.0003, 0.0004, 0.0005, 0.0011,
          0.0006, 0.0003, 0.0007, 0.0006, 0.0037, 0.0034, 0.0005, 0.0007, 0.0003,
          0.0004, 0.0002, 0.0009, 0.0005, 0.0006, 0.0026, 0.0184, 0.0320, 0.0009,
          0.0012, 0.0020, 0.0015, 0.0006, 0.0015, 0.0013, 0.0008, 0.0005, 0.0020,
          0.0194, 0.0006, 0.0010, 0.0011, 0.0004, 0.0003, 0.0004, 0.0018, 0.0007,
          0.0007, 0.0026, 0.0044, 0.0083, 0.0008, 0.0011, 0.0011, 0.0006, 0.0004,
          0.0023, 0.0012, 0.0012, 0.0008, 0.0021, 0.0016, 0.0080, 0.0006, 0.0011,
          0.0011, 0.0011, 0.0009, 0.0028, 0.0005, 0.0006, 0.0035, 0.0006, 0.0040,
          0.0023, 0.0062, 0.0021, 0.0007, 0.0034, 0.0059, 0.0051, 0.0011, 0.0011,
          0.0056, 0.0090, 0.0098, 0.0008, 0.0012, 0.0329, 0.0024, 0.0021, 0.0246,
          0.0048, 0.0012, 0.0041, 0.0006, 0.0032, 0.0019, 0.0076, 0.0025, 0.0279,
          0.0013, 0.0013, 0.0125, 0.0012, 0.0028, 0.0074, 0.0059, 0.0065, 0.0012,
          0.0009, 0.0041, 0.0019, 0.0020, 0.0018, 0.0005, 0.0006, 0.0156, 0.0006,
          0.0024, 0.0025, 0.0018, 0.0051, 0.0010, 0.0034, 0.0010, 0.0125, 0.0028,
          0.0010, 0.0087, 0.0021, 0.0116, 0.0019, 0.0016, 0.0016, 0.0011, 0.0056,
          0.0014, 0.0252, 0.0022, 0.0036, 0.0083, 0.0021, 0.0009, 0.0078, 0.0254,
          0.0001, 0.0003, 0.0004, 0.0001, 0.0025, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0003, 0.0003, 0.0003, 0.0004, 0.0004, 0.0002, 0.0001, 0.0002, 0.0002,
          0.0004, 0.0002, 0.0001, 0.0003, 0.0002], dtype=torch.float16),
  tensor([0.0000, 0.0030, 0.0557, 0.0105, 0.0129, 0.0288, 0.0504, 0.0089, 0.0026,
          0.0009, 0.0024, 0.0051, 0.0013, 0.0017, 0.0006, 0.0020, 0.0028, 0.0016,
          0.0015, 0.0012, 0.0001, 0.0001, 0.0002, 0.0030, 0.0051, 0.0012, 0.0072,
          0.0030, 0.0003, 0.0018, 0.0010, 0.0051, 0.0005, 0.0015, 0.0046, 0.0004,
          0.0006, 0.0002, 0.0010, 0.0019, 0.0002, 0.0004, 0.0002, 0.0002, 0.0002,
          0.0001, 0.0026, 0.0012, 0.0004, 0.0003, 0.0023, 0.0007, 0.0024, 0.0007,
          0.0003, 0.0010, 0.0003, 0.0002, 0.0009, 0.0003, 0.0004, 0.0006, 0.0005,
          0.0005, 0.0002, 0.0025, 0.0005, 0.0004, 0.0007, 0.0001, 0.0001, 0.0002,
          0.0003, 0.0040, 0.0004, 0.0030, 0.0003, 0.0003, 0.0009, 0.0001, 0.0003,
          0.0009, 0.0005, 0.0003, 0.0002, 0.0034, 0.0004, 0.0029, 0.0003, 0.0002,
          0.0050, 0.0013, 0.0002, 0.0021, 0.0002, 0.0002, 0.0004, 0.0002, 0.0010,
          0.0008, 0.0002, 0.0002, 0.0033, 0.0004, 0.0006, 0.0005, 0.0022, 0.0003,
          0.0027, 0.0002, 0.0012, 0.0002, 0.0003, 0.0003, 0.0009, 0.0003, 0.0003,
          0.0024, 0.0003, 0.0004, 0.0007, 0.0006, 0.0001, 0.0031, 0.0045, 0.0001,
          0.0001, 0.0002, 0.0002, 0.0016, 0.0024, 0.0047, 0.0005, 0.0005, 0.0043,
          0.0003, 0.0009, 0.0002, 0.0026, 0.0005, 0.0026, 0.0047, 0.0027, 0.0012,
          0.0002, 0.0009, 0.0002, 0.0002, 0.0004, 0.0013, 0.0020, 0.0011, 0.0002,
          0.0007, 0.0009, 0.0004, 0.0007, 0.0003, 0.0023, 0.0002, 0.0009, 0.0007,
          0.0014, 0.0014, 0.0039, 0.0060, 0.0003, 0.0003, 0.0016, 0.0004, 0.0008,
          0.0017, 0.0013, 0.0009, 0.0007, 0.0006, 0.0004, 0.0003, 0.0009, 0.0209,
          0.0009, 0.0012, 0.0029, 0.0060, 0.0065, 0.0023, 0.0053, 0.0008, 0.0042,
          0.0002, 0.0003, 0.0026, 0.0065, 0.0001, 0.0014, 0.0031, 0.0025, 0.0012,
          0.0005, 0.0004, 0.0016, 0.0005, 0.0013, 0.0008, 0.0003, 0.0007, 0.0002,
          0.0048, 0.0032, 0.0011, 0.0026, 0.0004, 0.0003, 0.0009, 0.0016, 0.0003,
          0.0005, 0.0011, 0.0085, 0.0025, 0.0019, 0.0010, 0.0014, 0.0002, 0.0005,
          0.0007, 0.0063, 0.0034, 0.0007, 0.0003, 0.0002, 0.0007, 0.0010, 0.0007,
          0.0005, 0.0003, 0.0003, 0.0011, 0.0006, 0.0016, 0.0006, 0.0005, 0.0002,
          0.0005, 0.0013, 0.0007, 0.0003, 0.0009, 0.0002, 0.0004, 0.0005, 0.0014,
          0.0002, 0.0002, 0.0004, 0.0004, 0.0051, 0.0030, 0.0003, 0.0005, 0.0006,
          0.0006, 0.0002, 0.0008, 0.0007, 0.0036, 0.0024, 0.0028, 0.0070, 0.0010,
          0.0007, 0.0010, 0.0031, 0.0004, 0.0011, 0.0015, 0.0007, 0.0011, 0.0005,
          0.0018, 0.0004, 0.0005, 0.0009, 0.0003, 0.0002, 0.0003, 0.0009, 0.0010,
          0.0010, 0.0039, 0.0036, 0.0048, 0.0005, 0.0007, 0.0005, 0.0003, 0.0002,
          0.0023, 0.0010, 0.0006, 0.0004, 0.0036, 0.0058, 0.0017, 0.0003, 0.0017,
          0.0015, 0.0004, 0.0009, 0.0012, 0.0063, 0.0055, 0.0013, 0.0010, 0.0009,
          0.0031, 0.0038, 0.0027, 0.0011, 0.0093, 0.0017, 0.0017, 0.0005, 0.0005,
          0.0032, 0.0013, 0.0046, 0.0004, 0.0011, 0.0039, 0.0023, 0.0010, 0.0025,
          0.0046, 0.0124, 0.0025, 0.0008, 0.0048, 0.0042, 0.0012, 0.0020, 0.0029,
          0.0047, 0.0081, 0.0020, 0.0056, 0.0033, 0.0017, 0.0009, 0.0046, 0.0007,
          0.0006, 0.0055, 0.0044, 0.0027, 0.0386, 0.0026, 0.0035, 0.0017, 0.0017,
          0.0046, 0.0041, 0.0025, 0.0056, 0.0087, 0.0059, 0.0077, 0.0016, 0.0155,
          0.0019, 0.0118, 0.0087, 0.0069, 0.0011, 0.0009, 0.0281, 0.0027, 0.0019,
          0.0026, 0.0065, 0.0080, 0.0170, 0.0019, 0.0067, 0.0031, 0.0017, 0.0026,
          0.0370, 0.0007, 0.0009, 0.0002, 0.0026, 0.0002, 0.0004, 0.0002, 0.0002,
          0.0004, 0.0004, 0.0004, 0.0004, 0.0004, 0.0002, 0.0002, 0.0004, 0.0004,
          0.0004, 0.0004, 0.0009, 0.0004, 0.0002], dtype=torch.float16),
  tensor([0.0000, 0.0035, 0.0459, 0.0097, 0.0120, 0.0259, 0.0341, 0.0098, 0.0007,
          0.0003, 0.0021, 0.0060, 0.0019, 0.0006, 0.0003, 0.0020, 0.0007, 0.0007,
          0.0004, 0.0004, 0.0001, 0.0001, 0.0002, 0.0008, 0.0075, 0.0038, 0.0082,
          0.0020, 0.0002, 0.0007, 0.0005, 0.0014, 0.0003, 0.0016, 0.0015, 0.0005,
          0.0010, 0.0004, 0.0010, 0.0013, 0.0002, 0.0009, 0.0005, 0.0004, 0.0004,
          0.0003, 0.0010, 0.0017, 0.0002, 0.0002, 0.0018, 0.0005, 0.0007, 0.0005,
          0.0002, 0.0004, 0.0002, 0.0004, 0.0009, 0.0009, 0.0007, 0.0004, 0.0004,
          0.0005, 0.0009, 0.0040, 0.0036, 0.0011, 0.0008, 0.0002, 0.0002, 0.0003,
          0.0003, 0.0041, 0.0002, 0.0010, 0.0002, 0.0002, 0.0004, 0.0004, 0.0011,
          0.0020, 0.0010, 0.0007, 0.0004, 0.0015, 0.0003, 0.0031, 0.0010, 0.0009,
          0.0055, 0.0019, 0.0003, 0.0032, 0.0003, 0.0003, 0.0002, 0.0002, 0.0009,
          0.0006, 0.0004, 0.0007, 0.0058, 0.0021, 0.0010, 0.0019, 0.0036, 0.0003,
          0.0011, 0.0004, 0.0033, 0.0007, 0.0046, 0.0009, 0.0046, 0.0005, 0.0002,
          0.0046, 0.0005, 0.0003, 0.0005, 0.0014, 0.0003, 0.0006, 0.0052, 0.0003,
          0.0002, 0.0005, 0.0008, 0.0032, 0.0015, 0.0015, 0.0073, 0.0015, 0.0020,
          0.0005, 0.0020, 0.0003, 0.0024, 0.0005, 0.0020, 0.0031, 0.0021, 0.0007,
          0.0005, 0.0045, 0.0007, 0.0005, 0.0015, 0.0048, 0.0033, 0.0025, 0.0008,
          0.0007, 0.0007, 0.0008, 0.0015, 0.0005, 0.0026, 0.0005, 0.0026, 0.0008,
          0.0028, 0.0016, 0.0018, 0.0042, 0.0004, 0.0005, 0.0049, 0.0009, 0.0029,
          0.0038, 0.0008, 0.0015, 0.0020, 0.0012, 0.0008, 0.0006, 0.0023, 0.0024,
          0.0016, 0.0020, 0.0045, 0.0027, 0.0012, 0.0034, 0.0068, 0.0010, 0.0041,
          0.0006, 0.0012, 0.0035, 0.0021, 0.0002, 0.0022, 0.0021, 0.0041, 0.0019,
          0.0010, 0.0004, 0.0028, 0.0013, 0.0016, 0.0015, 0.0007, 0.0020, 0.0006,
          0.0014, 0.0080, 0.0009, 0.0013, 0.0006, 0.0006, 0.0026, 0.0053, 0.0007,
          0.0009, 0.0012, 0.0044, 0.0041, 0.0033, 0.0012, 0.0021, 0.0005, 0.0020,
          0.0020, 0.0099, 0.0036, 0.0021, 0.0015, 0.0008, 0.0029, 0.0027, 0.0009,
          0.0008, 0.0008, 0.0026, 0.0043, 0.0031, 0.0066, 0.0016, 0.0009, 0.0006,
          0.0007, 0.0022, 0.0007, 0.0007, 0.0038, 0.0005, 0.0009, 0.0012, 0.0037,
          0.0009, 0.0004, 0.0008, 0.0008, 0.0038, 0.0035, 0.0006, 0.0017, 0.0014,
          0.0035, 0.0004, 0.0028, 0.0026, 0.0036, 0.0023, 0.0039, 0.0039, 0.0022,
          0.0043, 0.0018, 0.0061, 0.0011, 0.0018, 0.0094, 0.0028, 0.0032, 0.0005,
          0.0015, 0.0005, 0.0035, 0.0038, 0.0007, 0.0003, 0.0005, 0.0033, 0.0018,
          0.0037, 0.0043, 0.0020, 0.0028, 0.0008, 0.0016, 0.0013, 0.0008, 0.0006,
          0.0056, 0.0022, 0.0013, 0.0008, 0.0026, 0.0032, 0.0020, 0.0011, 0.0071,
          0.0054, 0.0004, 0.0013, 0.0010, 0.0067, 0.0052, 0.0013, 0.0018, 0.0010,
          0.0033, 0.0046, 0.0058, 0.0038, 0.0187, 0.0019, 0.0015, 0.0009, 0.0009,
          0.0020, 0.0015, 0.0023, 0.0010, 0.0040, 0.0031, 0.0043, 0.0010, 0.0017,
          0.0052, 0.0050, 0.0017, 0.0019, 0.0056, 0.0031, 0.0015, 0.0045, 0.0024,
          0.0060, 0.0079, 0.0023, 0.0043, 0.0046, 0.0016, 0.0008, 0.0040, 0.0020,
          0.0016, 0.0048, 0.0025, 0.0035, 0.0034, 0.0039, 0.0065, 0.0012, 0.0036,
          0.0027, 0.0027, 0.0028, 0.0031, 0.0021, 0.0063, 0.0056, 0.0013, 0.0039,
          0.0028, 0.0095, 0.0064, 0.0058, 0.0018, 0.0018, 0.0037, 0.0015, 0.0017,
          0.0014, 0.0029, 0.0135, 0.0046, 0.0019, 0.0061, 0.0024, 0.0019, 0.0017,
          0.0026, 0.0281, 0.0001, 0.0001, 0.0018, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0001, 0.0001, 0.0001, 0.0002, 0.0003, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0004, 0.0001, 0.0001, 0.0001, 0.0001], dtype=torch.float16),
  tensor([0.0000, 0.0033, 0.0516, 0.0127, 0.0117, 0.0260, 0.0410, 0.0088, 0.0006,
          0.0002, 0.0014, 0.0021, 0.0007, 0.0009, 0.0003, 0.0027, 0.0006, 0.0008,
          0.0005, 0.0004, 0.0002, 0.0002, 0.0003, 0.0009, 0.0026, 0.0031, 0.0069,
          0.0022, 0.0002, 0.0009, 0.0004, 0.0012, 0.0003, 0.0028, 0.0014, 0.0005,
          0.0024, 0.0005, 0.0017, 0.0023, 0.0002, 0.0011, 0.0005, 0.0004, 0.0005,
          0.0003, 0.0011, 0.0034, 0.0003, 0.0004, 0.0018, 0.0008, 0.0006, 0.0005,
          0.0002, 0.0004, 0.0003, 0.0006, 0.0010, 0.0010, 0.0010, 0.0005, 0.0008,
          0.0013, 0.0006, 0.0073, 0.0008, 0.0004, 0.0008, 0.0003, 0.0002, 0.0005,
          0.0005, 0.0027, 0.0003, 0.0010, 0.0002, 0.0002, 0.0006, 0.0005, 0.0005,
          0.0022, 0.0006, 0.0008, 0.0006, 0.0017, 0.0006, 0.0010, 0.0004, 0.0004,
          0.0022, 0.0015, 0.0003, 0.0018, 0.0005, 0.0006, 0.0004, 0.0002, 0.0008,
          0.0018, 0.0005, 0.0006, 0.0019, 0.0006, 0.0005, 0.0052, 0.0073, 0.0005,
          0.0018, 0.0004, 0.0019, 0.0004, 0.0008, 0.0006, 0.0070, 0.0010, 0.0004,
          0.0081, 0.0010, 0.0005, 0.0005, 0.0027, 0.0002, 0.0005, 0.0069, 0.0002,
          0.0005, 0.0004, 0.0005, 0.0070, 0.0013, 0.0022, 0.0014, 0.0005, 0.0009,
          0.0003, 0.0011, 0.0003, 0.0026, 0.0005, 0.0020, 0.0027, 0.0025, 0.0007,
          0.0003, 0.0014, 0.0008, 0.0006, 0.0005, 0.0167, 0.0043, 0.0069, 0.0018,
          0.0013, 0.0015, 0.0014, 0.0013, 0.0009, 0.0006, 0.0004, 0.0052, 0.0009,
          0.0043, 0.0028, 0.0017, 0.0052, 0.0009, 0.0007, 0.0011, 0.0007, 0.0064,
          0.0044, 0.0007, 0.0034, 0.0031, 0.0010, 0.0017, 0.0008, 0.0032, 0.0019,
          0.0021, 0.0023, 0.0046, 0.0014, 0.0018, 0.0042, 0.0019, 0.0009, 0.0027,
          0.0006, 0.0005, 0.0038, 0.0011, 0.0003, 0.0030, 0.0017, 0.0014, 0.0031,
          0.0016, 0.0007, 0.0053, 0.0006, 0.0011, 0.0019, 0.0004, 0.0012, 0.0007,
          0.0012, 0.0027, 0.0011, 0.0012, 0.0012, 0.0005, 0.0006, 0.0097, 0.0005,
          0.0024, 0.0012, 0.0032, 0.0032, 0.0045, 0.0009, 0.0039, 0.0005, 0.0015,
          0.0010, 0.0095, 0.0034, 0.0007, 0.0006, 0.0006, 0.0023, 0.0055, 0.0009,
          0.0008, 0.0006, 0.0006, 0.0081, 0.0037, 0.0032, 0.0008, 0.0008, 0.0009,
          0.0020, 0.0021, 0.0008, 0.0009, 0.0010, 0.0005, 0.0008, 0.0009, 0.0045,
          0.0004, 0.0006, 0.0011, 0.0016, 0.0043, 0.0037, 0.0011, 0.0008, 0.0029,
          0.0051, 0.0004, 0.0012, 0.0030, 0.0029, 0.0017, 0.0024, 0.0022, 0.0035,
          0.0012, 0.0007, 0.0062, 0.0012, 0.0041, 0.0089, 0.0025, 0.0037, 0.0008,
          0.0011, 0.0007, 0.0008, 0.0009, 0.0005, 0.0004, 0.0005, 0.0034, 0.0037,
          0.0022, 0.0055, 0.0025, 0.0027, 0.0009, 0.0014, 0.0011, 0.0006, 0.0005,
          0.0060, 0.0040, 0.0015, 0.0010, 0.0038, 0.0025, 0.0022, 0.0006, 0.0043,
          0.0063, 0.0007, 0.0032, 0.0011, 0.0041, 0.0022, 0.0015, 0.0030, 0.0015,
          0.0032, 0.0053, 0.0059, 0.0019, 0.0049, 0.0026, 0.0019, 0.0013, 0.0012,
          0.0032, 0.0015, 0.0019, 0.0009, 0.0044, 0.0017, 0.0041, 0.0011, 0.0012,
          0.0058, 0.0037, 0.0024, 0.0021, 0.0067, 0.0035, 0.0017, 0.0035, 0.0017,
          0.0033, 0.0059, 0.0018, 0.0026, 0.0054, 0.0026, 0.0010, 0.0033, 0.0012,
          0.0010, 0.0031, 0.0021, 0.0043, 0.0033, 0.0035, 0.0042, 0.0009, 0.0067,
          0.0030, 0.0024, 0.0042, 0.0026, 0.0024, 0.0059, 0.0050, 0.0014, 0.0042,
          0.0029, 0.0100, 0.0064, 0.0047, 0.0013, 0.0014, 0.0038, 0.0017, 0.0018,
          0.0019, 0.0018, 0.0039, 0.0055, 0.0022, 0.0064, 0.0024, 0.0020, 0.0015,
          0.0030, 0.0058, 0.0379, 0.0003, 0.0003, 0.0002, 0.0002, 0.0002, 0.0002,
          0.0003, 0.0003, 0.0003, 0.0004, 0.0003, 0.0002, 0.0002, 0.0003, 0.0003,
          0.0010, 0.0002, 0.0005, 0.0003, 0.0003], dtype=torch.float16),
  tensor([0.0000, 0.0046, 0.0480, 0.0124, 0.0235, 0.0332, 0.0275, 0.0137, 0.0005,
          0.0002, 0.0009, 0.0014, 0.0005, 0.0018, 0.0024, 0.0009, 0.0005, 0.0084,
          0.0030, 0.0003, 0.0008, 0.0008, 0.0009, 0.0034, 0.0016, 0.0007, 0.0109,
          0.0058, 0.0011, 0.0065, 0.0057, 0.0011, 0.0019, 0.0018, 0.0007, 0.0008,
          0.0007, 0.0003, 0.0017, 0.0011, 0.0016, 0.0009, 0.0004, 0.0003, 0.0003,
          0.0003, 0.0084, 0.0031, 0.0014, 0.0013, 0.0032, 0.0011, 0.0007, 0.0006,
          0.0002, 0.0004, 0.0003, 0.0003, 0.0005, 0.0004, 0.0004, 0.0029, 0.0005,
          0.0008, 0.0004, 0.0031, 0.0006, 0.0003, 0.0005, 0.0002, 0.0002, 0.0003,
          0.0004, 0.0030, 0.0004, 0.0008, 0.0002, 0.0002, 0.0006, 0.0003, 0.0003,
          0.0005, 0.0004, 0.0002, 0.0003, 0.0014, 0.0005, 0.0007, 0.0003, 0.0005,
          0.0014, 0.0010, 0.0002, 0.0009, 0.0003, 0.0004, 0.0013, 0.0003, 0.0015,
          0.0022, 0.0003, 0.0003, 0.0011, 0.0002, 0.0003, 0.0006, 0.0014, 0.0022,
          0.0010, 0.0004, 0.0012, 0.0002, 0.0004, 0.0004, 0.0004, 0.0006, 0.0011,
          0.0033, 0.0006, 0.0011, 0.0032, 0.0009, 0.0002, 0.0003, 0.0018, 0.0001,
          0.0002, 0.0003, 0.0003, 0.0014, 0.0012, 0.0011, 0.0005, 0.0004, 0.0003,
          0.0002, 0.0007, 0.0003, 0.0159, 0.0033, 0.0017, 0.0032, 0.0039, 0.0052,
          0.0002, 0.0001, 0.0002, 0.0002, 0.0005, 0.0009, 0.0013, 0.0008, 0.0003,
          0.0018, 0.0009, 0.0012, 0.0009, 0.0007, 0.0003, 0.0003, 0.0019, 0.0041,
          0.0041, 0.0025, 0.0119, 0.0024, 0.0007, 0.0004, 0.0008, 0.0005, 0.0006,
          0.0005, 0.0047, 0.0025, 0.0004, 0.0004, 0.0007, 0.0012, 0.0015, 0.0020,
          0.0009, 0.0006, 0.0025, 0.0011, 0.0010, 0.0009, 0.0008, 0.0025, 0.0012,
          0.0004, 0.0005, 0.0013, 0.0009, 0.0002, 0.0019, 0.0126, 0.0009, 0.0015,
          0.0007, 0.0016, 0.0017, 0.0007, 0.0009, 0.0010, 0.0005, 0.0010, 0.0005,
          0.0004, 0.0018, 0.0033, 0.0006, 0.0013, 0.0005, 0.0004, 0.0015, 0.0003,
          0.0009, 0.0049, 0.0015, 0.0005, 0.0043, 0.0025, 0.0057, 0.0005, 0.0010,
          0.0007, 0.0039, 0.0015, 0.0007, 0.0004, 0.0003, 0.0005, 0.0009, 0.0017,
          0.0010, 0.0004, 0.0003, 0.0004, 0.0009, 0.0011, 0.0006, 0.0007, 0.0004,
          0.0018, 0.0043, 0.0052, 0.0014, 0.0008, 0.0003, 0.0004, 0.0005, 0.0015,
          0.0004, 0.0004, 0.0008, 0.0008, 0.0044, 0.0059, 0.0007, 0.0006, 0.0004,
          0.0005, 0.0002, 0.0008, 0.0007, 0.0007, 0.0014, 0.0082, 0.0135, 0.0012,
          0.0009, 0.0010, 0.0024, 0.0007, 0.0032, 0.0017, 0.0009, 0.0004, 0.0023,
          0.0087, 0.0009, 0.0007, 0.0006, 0.0004, 0.0003, 0.0004, 0.0024, 0.0013,
          0.0007, 0.0053, 0.0063, 0.0068, 0.0008, 0.0010, 0.0014, 0.0007, 0.0005,
          0.0044, 0.0019, 0.0011, 0.0008, 0.0017, 0.0014, 0.0075, 0.0006, 0.0014,
          0.0018, 0.0014, 0.0018, 0.0020, 0.0005, 0.0005, 0.0030, 0.0006, 0.0052,
          0.0020, 0.0106, 0.0027, 0.0006, 0.0035, 0.0079, 0.0041, 0.0012, 0.0016,
          0.0037, 0.0036, 0.0083, 0.0010, 0.0015, 0.0171, 0.0039, 0.0014, 0.0132,
          0.0064, 0.0011, 0.0040, 0.0005, 0.0038, 0.0022, 0.0073, 0.0030, 0.0144,
          0.0012, 0.0016, 0.0100, 0.0010, 0.0035, 0.0123, 0.0056, 0.0062, 0.0011,
          0.0009, 0.0044, 0.0015, 0.0032, 0.0020, 0.0005, 0.0005, 0.0081, 0.0006,
          0.0024, 0.0019, 0.0027, 0.0036, 0.0009, 0.0044, 0.0011, 0.0089, 0.0026,
          0.0010, 0.0120, 0.0032, 0.0108, 0.0017, 0.0017, 0.0019, 0.0013, 0.0059,
          0.0012, 0.0127, 0.0027, 0.0037, 0.0069, 0.0031, 0.0007, 0.0074, 0.0107,
          0.0012, 0.0011, 0.0016, 0.0329, 0.0034, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0004, 0.0004, 0.0003, 0.0005, 0.0004, 0.0003, 0.0001, 0.0003, 0.0003,
          0.0005, 0.0003, 0.0001, 0.0003, 0.0003], dtype=torch.float16),
  tensor([0.0000, 0.0026, 0.0602, 0.0111, 0.0105, 0.0257, 0.0490, 0.0067, 0.0015,
          0.0006, 0.0007, 0.0011, 0.0004, 0.0008, 0.0004, 0.0008, 0.0027, 0.0016,
          0.0009, 0.0007, 0.0003, 0.0005, 0.0004, 0.0012, 0.0015, 0.0009, 0.0056,
          0.0031, 0.0003, 0.0017, 0.0005, 0.0065, 0.0006, 0.0016, 0.0010, 0.0028,
          0.0009, 0.0003, 0.0033, 0.0017, 0.0004, 0.0011, 0.0004, 0.0003, 0.0004,
          0.0003, 0.0017, 0.0023, 0.0004, 0.0004, 0.0018, 0.0054, 0.0051, 0.0016,
          0.0005, 0.0027, 0.0008, 0.0003, 0.0010, 0.0005, 0.0006, 0.0010, 0.0026,
          0.0054, 0.0004, 0.0024, 0.0004, 0.0003, 0.0005, 0.0003, 0.0002, 0.0004,
          0.0004, 0.0020, 0.0017, 0.0071, 0.0006, 0.0007, 0.0033, 0.0003, 0.0003,
          0.0008, 0.0003, 0.0004, 0.0004, 0.0117, 0.0015, 0.0006, 0.0002, 0.0002,
          0.0014, 0.0011, 0.0002, 0.0010, 0.0004, 0.0005, 0.0008, 0.0009, 0.0056,
          0.0034, 0.0003, 0.0004, 0.0010, 0.0002, 0.0003, 0.0012, 0.0015, 0.0007,
          0.0106, 0.0005, 0.0014, 0.0002, 0.0003, 0.0004, 0.0008, 0.0009, 0.0005,
          0.0029, 0.0008, 0.0006, 0.0008, 0.0017, 0.0001, 0.0009, 0.0016, 0.0001,
          0.0002, 0.0002, 0.0003, 0.0017, 0.0045, 0.0134, 0.0006, 0.0003, 0.0004,
          0.0002, 0.0007, 0.0003, 0.0054, 0.0007, 0.0016, 0.0038, 0.0034, 0.0010,
          0.0002, 0.0003, 0.0003, 0.0004, 0.0003, 0.0017, 0.0022, 0.0013, 0.0007,
          0.0144, 0.0070, 0.0021, 0.0013, 0.0013, 0.0004, 0.0003, 0.0027, 0.0017,
          0.0022, 0.0033, 0.0033, 0.0021, 0.0009, 0.0007, 0.0007, 0.0008, 0.0009,
          0.0006, 0.0015, 0.0027, 0.0014, 0.0009, 0.0011, 0.0013, 0.0024, 0.0026,
          0.0008, 0.0010, 0.0021, 0.0035, 0.0014, 0.0011, 0.0011, 0.0010, 0.0011,
          0.0004, 0.0003, 0.0012, 0.0045, 0.0002, 0.0015, 0.0030, 0.0009, 0.0015,
          0.0012, 0.0009, 0.0027, 0.0005, 0.0010, 0.0010, 0.0004, 0.0010, 0.0005,
          0.0013, 0.0016, 0.0014, 0.0026, 0.0012, 0.0004, 0.0004, 0.0016, 0.0003,
          0.0021, 0.0023, 0.0019, 0.0007, 0.0034, 0.0014, 0.0129, 0.0005, 0.0012,
          0.0006, 0.0035, 0.0014, 0.0005, 0.0004, 0.0004, 0.0012, 0.0017, 0.0011,
          0.0007, 0.0005, 0.0003, 0.0014, 0.0015, 0.0017, 0.0006, 0.0007, 0.0007,
          0.0024, 0.0030, 0.0011, 0.0018, 0.0007, 0.0004, 0.0006, 0.0006, 0.0020,
          0.0004, 0.0005, 0.0009, 0.0013, 0.0030, 0.0111, 0.0009, 0.0005, 0.0011,
          0.0011, 0.0003, 0.0008, 0.0012, 0.0010, 0.0017, 0.0031, 0.0029, 0.0019,
          0.0009, 0.0006, 0.0025, 0.0009, 0.0048, 0.0025, 0.0014, 0.0013, 0.0009,
          0.0016, 0.0006, 0.0005, 0.0005, 0.0003, 0.0003, 0.0004, 0.0030, 0.0023,
          0.0014, 0.0036, 0.0027, 0.0028, 0.0010, 0.0013, 0.0013, 0.0006, 0.0005,
          0.0030, 0.0047, 0.0013, 0.0008, 0.0018, 0.0061, 0.0035, 0.0004, 0.0022,
          0.0022, 0.0010, 0.0031, 0.0016, 0.0013, 0.0011, 0.0018, 0.0016, 0.0022,
          0.0027, 0.0062, 0.0030, 0.0013, 0.0034, 0.0039, 0.0020, 0.0012, 0.0011,
          0.0022, 0.0021, 0.0026, 0.0008, 0.0019, 0.0026, 0.0043, 0.0011, 0.0019,
          0.0063, 0.0016, 0.0041, 0.0014, 0.0095, 0.0090, 0.0020, 0.0039, 0.0026,
          0.0017, 0.0020, 0.0040, 0.0017, 0.0132, 0.0034, 0.0013, 0.0038, 0.0009,
          0.0008, 0.0136, 0.0079, 0.0159, 0.0025, 0.0010, 0.0011, 0.0011, 0.0022,
          0.0167, 0.0122, 0.0221, 0.0023, 0.0039, 0.0036, 0.0019, 0.0018, 0.0044,
          0.0022, 0.0087, 0.0031, 0.0058, 0.0012, 0.0012, 0.0040, 0.0090, 0.0020,
          0.0125, 0.0023, 0.0030, 0.0030, 0.0026, 0.0025, 0.0034, 0.0019, 0.0023,
          0.0020, 0.0019, 0.0037, 0.0046, 0.0208, 0.0001, 0.0001, 0.0001, 0.0001,
          0.0003, 0.0003, 0.0003, 0.0003, 0.0006, 0.0003, 0.0001, 0.0003, 0.0003,
          0.0005, 0.0004, 0.0004, 0.0003, 0.0003], dtype=torch.float16),
  tensor([0.0000, 0.0053, 0.0671, 0.0146, 0.0274, 0.0327, 0.0548, 0.0177, 0.0007,
          0.0003, 0.0025, 0.0034, 0.0010, 0.0007, 0.0004, 0.0018, 0.0006, 0.0008,
          0.0007, 0.0005, 0.0002, 0.0002, 0.0002, 0.0020, 0.0036, 0.0015, 0.0132,
          0.0074, 0.0003, 0.0009, 0.0006, 0.0013, 0.0004, 0.0021, 0.0035, 0.0004,
          0.0008, 0.0002, 0.0016, 0.0019, 0.0002, 0.0006, 0.0003, 0.0002, 0.0003,
          0.0002, 0.0010, 0.0017, 0.0003, 0.0004, 0.0031, 0.0006, 0.0007, 0.0006,
          0.0002, 0.0004, 0.0002, 0.0002, 0.0005, 0.0003, 0.0003, 0.0005, 0.0004,
          0.0005, 0.0003, 0.0025, 0.0005, 0.0004, 0.0008, 0.0001, 0.0002, 0.0002,
          0.0003, 0.0057, 0.0003, 0.0010, 0.0002, 0.0002, 0.0004, 0.0002, 0.0003,
          0.0010, 0.0004, 0.0002, 0.0002, 0.0013, 0.0003, 0.0016, 0.0004, 0.0003,
          0.0024, 0.0010, 0.0002, 0.0013, 0.0002, 0.0003, 0.0003, 0.0002, 0.0008,
          0.0007, 0.0002, 0.0002, 0.0019, 0.0004, 0.0004, 0.0006, 0.0016, 0.0004,
          0.0010, 0.0002, 0.0012, 0.0002, 0.0004, 0.0005, 0.0009, 0.0003, 0.0004,
          0.0021, 0.0004, 0.0004, 0.0006, 0.0008, 0.0001, 0.0004, 0.0046, 0.0001,
          0.0001, 0.0002, 0.0002, 0.0015, 0.0014, 0.0014, 0.0007, 0.0004, 0.0017,
          0.0003, 0.0007, 0.0002, 0.0025, 0.0006, 0.0021, 0.0056, 0.0057, 0.0010,
          0.0002, 0.0007, 0.0003, 0.0002, 0.0004, 0.0014, 0.0023, 0.0010, 0.0002,
          0.0006, 0.0006, 0.0005, 0.0008, 0.0004, 0.0010, 0.0002, 0.0011, 0.0007,
          0.0028, 0.0026, 0.0021, 0.0129, 0.0004, 0.0004, 0.0012, 0.0005, 0.0010,
          0.0019, 0.0012, 0.0016, 0.0004, 0.0003, 0.0004, 0.0004, 0.0012, 0.0044,
          0.0019, 0.0012, 0.0044, 0.0016, 0.0020, 0.0029, 0.0027, 0.0010, 0.0127,
          0.0003, 0.0004, 0.0029, 0.0016, 0.0001, 0.0028, 0.0021, 0.0017, 0.0015,
          0.0004, 0.0005, 0.0018, 0.0005, 0.0015, 0.0014, 0.0003, 0.0007, 0.0003,
          0.0013, 0.0027, 0.0010, 0.0012, 0.0007, 0.0003, 0.0006, 0.0018, 0.0003,
          0.0007, 0.0011, 0.0086, 0.0018, 0.0027, 0.0009, 0.0015, 0.0002, 0.0006,
          0.0010, 0.0076, 0.0040, 0.0006, 0.0003, 0.0003, 0.0006, 0.0011, 0.0007,
          0.0007, 0.0004, 0.0003, 0.0010, 0.0007, 0.0012, 0.0005, 0.0005, 0.0003,
          0.0007, 0.0011, 0.0009, 0.0004, 0.0009, 0.0002, 0.0003, 0.0004, 0.0017,
          0.0003, 0.0002, 0.0006, 0.0006, 0.0033, 0.0061, 0.0004, 0.0008, 0.0004,
          0.0006, 0.0002, 0.0006, 0.0009, 0.0057, 0.0017, 0.0023, 0.0048, 0.0011,
          0.0007, 0.0009, 0.0032, 0.0005, 0.0015, 0.0015, 0.0007, 0.0010, 0.0008,
          0.0017, 0.0005, 0.0005, 0.0007, 0.0003, 0.0002, 0.0003, 0.0010, 0.0011,
          0.0008, 0.0083, 0.0024, 0.0027, 0.0005, 0.0006, 0.0006, 0.0004, 0.0003,
          0.0030, 0.0012, 0.0008, 0.0006, 0.0035, 0.0027, 0.0020, 0.0003, 0.0018,
          0.0019, 0.0005, 0.0013, 0.0013, 0.0044, 0.0024, 0.0013, 0.0008, 0.0013,
          0.0033, 0.0064, 0.0032, 0.0010, 0.0078, 0.0020, 0.0016, 0.0007, 0.0008,
          0.0027, 0.0013, 0.0047, 0.0005, 0.0010, 0.0034, 0.0041, 0.0010, 0.0026,
          0.0073, 0.0095, 0.0031, 0.0006, 0.0054, 0.0029, 0.0018, 0.0025, 0.0030,
          0.0057, 0.0062, 0.0018, 0.0049, 0.0028, 0.0027, 0.0012, 0.0069, 0.0008,
          0.0006, 0.0033, 0.0021, 0.0024, 0.0075, 0.0028, 0.0042, 0.0020, 0.0016,
          0.0020, 0.0018, 0.0020, 0.0056, 0.0029, 0.0133, 0.0060, 0.0020, 0.0063,
          0.0019, 0.0267, 0.0265, 0.0112, 0.0011, 0.0012, 0.0072, 0.0016, 0.0029,
          0.0013, 0.0053, 0.0069, 0.0098, 0.0026, 0.0108, 0.0016, 0.0028, 0.0028,
          0.0064, 0.0042, 0.0048, 0.0056, 0.0025, 0.0458, 0.0002, 0.0002, 0.0002,
          0.0005, 0.0005, 0.0002, 0.0006, 0.0005, 0.0005, 0.0002, 0.0005, 0.0005,
          0.0006, 0.0005, 0.0005, 0.0005, 0.0005], dtype=torch.float16),
  tensor([0.0000, 0.0044, 0.0568, 0.0156, 0.0192, 0.0359, 0.0349, 0.0124, 0.0004,
          0.0002, 0.0012, 0.0023, 0.0006, 0.0006, 0.0004, 0.0006, 0.0004, 0.0017,
          0.0008, 0.0002, 0.0004, 0.0005, 0.0005, 0.0009, 0.0025, 0.0009, 0.0100,
          0.0040, 0.0005, 0.0017, 0.0006, 0.0007, 0.0006, 0.0014, 0.0006, 0.0006,
          0.0020, 0.0006, 0.0035, 0.0012, 0.0005, 0.0041, 0.0013, 0.0008, 0.0007,
          0.0009, 0.0013, 0.0024, 0.0004, 0.0006, 0.0029, 0.0014, 0.0007, 0.0005,
          0.0002, 0.0004, 0.0004, 0.0013, 0.0007, 0.0011, 0.0007, 0.0013, 0.0010,
          0.0018, 0.0014, 0.0032, 0.0010, 0.0006, 0.0011, 0.0008, 0.0005, 0.0006,
          0.0009, 0.0037, 0.0006, 0.0008, 0.0002, 0.0003, 0.0005, 0.0013, 0.0013,
          0.0013, 0.0008, 0.0005, 0.0005, 0.0020, 0.0007, 0.0012, 0.0007, 0.0005,
          0.0032, 0.0035, 0.0003, 0.0020, 0.0006, 0.0008, 0.0007, 0.0003, 0.0010,
          0.0026, 0.0014, 0.0019, 0.0025, 0.0004, 0.0005, 0.0023, 0.0013, 0.0012,
          0.0016, 0.0024, 0.0050, 0.0004, 0.0005, 0.0013, 0.0007, 0.0029, 0.0008,
          0.0028, 0.0011, 0.0008, 0.0010, 0.0034, 0.0004, 0.0002, 0.0016, 0.0003,
          0.0010, 0.0005, 0.0005, 0.0016, 0.0007, 0.0018, 0.0009, 0.0007, 0.0002,
          0.0004, 0.0021, 0.0007, 0.0039, 0.0008, 0.0010, 0.0020, 0.0018, 0.0012,
          0.0005, 0.0002, 0.0006, 0.0016, 0.0007, 0.0015, 0.0051, 0.0016, 0.0013,
          0.0038, 0.0018, 0.0079, 0.0041, 0.0044, 0.0003, 0.0007, 0.0040, 0.0016,
          0.0027, 0.0066, 0.0023, 0.0016, 0.0033, 0.0023, 0.0020, 0.0022, 0.0009,
          0.0004, 0.0011, 0.0037, 0.0008, 0.0006, 0.0019, 0.0045, 0.0046, 0.0010,
          0.0013, 0.0009, 0.0024, 0.0008, 0.0008, 0.0008, 0.0008, 0.0011, 0.0012,
          0.0012, 0.0005, 0.0012, 0.0006, 0.0006, 0.0015, 0.0025, 0.0025, 0.0017,
          0.0014, 0.0014, 0.0036, 0.0016, 0.0020, 0.0020, 0.0006, 0.0035, 0.0012,
          0.0004, 0.0030, 0.0015, 0.0007, 0.0032, 0.0013, 0.0006, 0.0017, 0.0005,
          0.0031, 0.0019, 0.0017, 0.0003, 0.0028, 0.0012, 0.0061, 0.0012, 0.0036,
          0.0015, 0.0036, 0.0014, 0.0011, 0.0005, 0.0006, 0.0008, 0.0016, 0.0014,
          0.0019, 0.0012, 0.0004, 0.0006, 0.0026, 0.0020, 0.0010, 0.0012, 0.0016,
          0.0032, 0.0034, 0.0018, 0.0062, 0.0010, 0.0005, 0.0006, 0.0008, 0.0029,
          0.0007, 0.0008, 0.0015, 0.0024, 0.0019, 0.0054, 0.0029, 0.0014, 0.0010,
          0.0008, 0.0005, 0.0016, 0.0012, 0.0009, 0.0006, 0.0030, 0.0023, 0.0025,
          0.0011, 0.0012, 0.0030, 0.0019, 0.0086, 0.0027, 0.0016, 0.0010, 0.0019,
          0.0016, 0.0016, 0.0008, 0.0008, 0.0006, 0.0008, 0.0007, 0.0027, 0.0018,
          0.0009, 0.0049, 0.0030, 0.0017, 0.0014, 0.0018, 0.0029, 0.0011, 0.0011,
          0.0029, 0.0101, 0.0019, 0.0013, 0.0016, 0.0014, 0.0038, 0.0011, 0.0023,
          0.0018, 0.0014, 0.0047, 0.0010, 0.0008, 0.0006, 0.0018, 0.0012, 0.0043,
          0.0021, 0.0062, 0.0039, 0.0009, 0.0049, 0.0046, 0.0030, 0.0019, 0.0025,
          0.0032, 0.0019, 0.0031, 0.0026, 0.0031, 0.0029, 0.0104, 0.0017, 0.0025,
          0.0135, 0.0014, 0.0035, 0.0007, 0.0039, 0.0025, 0.0056, 0.0106, 0.0029,
          0.0017, 0.0015, 0.0044, 0.0010, 0.0032, 0.0065, 0.0023, 0.0062, 0.0018,
          0.0016, 0.0026, 0.0012, 0.0030, 0.0012, 0.0007, 0.0007, 0.0014, 0.0011,
          0.0024, 0.0019, 0.0038, 0.0013, 0.0013, 0.0051, 0.0010, 0.0028, 0.0031,
          0.0043, 0.0164, 0.0032, 0.0099, 0.0023, 0.0027, 0.0017, 0.0013, 0.0047,
          0.0019, 0.0025, 0.0034, 0.0034, 0.0043, 0.0026, 0.0010, 0.0060, 0.0021,
          0.0013, 0.0013, 0.0030, 0.0067, 0.0032, 0.0044, 0.0443, 0.0002, 0.0002,
          0.0005, 0.0005, 0.0005, 0.0005, 0.0005, 0.0002, 0.0002, 0.0005, 0.0005,
          0.0008, 0.0002, 0.0005, 0.0005, 0.0005], dtype=torch.float16),
  tensor([0.0000, 0.0044, 0.0560, 0.0141, 0.0259, 0.0349, 0.0351, 0.0175, 0.0005,
          0.0002, 0.0015, 0.0036, 0.0008, 0.0005, 0.0005, 0.0005, 0.0005, 0.0009,
          0.0006, 0.0002, 0.0003, 0.0004, 0.0004, 0.0010, 0.0039, 0.0008, 0.0140,
          0.0047, 0.0005, 0.0008, 0.0011, 0.0008, 0.0003, 0.0009, 0.0006, 0.0005,
          0.0011, 0.0006, 0.0022, 0.0008, 0.0004, 0.0024, 0.0009, 0.0007, 0.0007,
          0.0007, 0.0008, 0.0014, 0.0005, 0.0009, 0.0036, 0.0012, 0.0007, 0.0005,
          0.0003, 0.0004, 0.0004, 0.0010, 0.0010, 0.0013, 0.0006, 0.0009, 0.0011,
          0.0012, 0.0014, 0.0022, 0.0028, 0.0011, 0.0014, 0.0006, 0.0005, 0.0009,
          0.0009, 0.0043, 0.0006, 0.0008, 0.0003, 0.0003, 0.0006, 0.0010, 0.0031,
          0.0014, 0.0020, 0.0004, 0.0005, 0.0016, 0.0008, 0.0018, 0.0018, 0.0013,
          0.0052, 0.0040, 0.0004, 0.0030, 0.0005, 0.0008, 0.0006, 0.0003, 0.0009,
          0.0014, 0.0011, 0.0016, 0.0035, 0.0008, 0.0010, 0.0018, 0.0009, 0.0008,
          0.0012, 0.0015, 0.0052, 0.0005, 0.0014, 0.0029, 0.0007, 0.0013, 0.0007,
          0.0017, 0.0011, 0.0007, 0.0010, 0.0021, 0.0005, 0.0003, 0.0013, 0.0008,
          0.0007, 0.0014, 0.0013, 0.0009, 0.0007, 0.0014, 0.0019, 0.0022, 0.0003,
          0.0007, 0.0033, 0.0006, 0.0019, 0.0009, 0.0008, 0.0030, 0.0022, 0.0014,
          0.0007, 0.0002, 0.0007, 0.0015, 0.0016, 0.0013, 0.0049, 0.0010, 0.0009,
          0.0023, 0.0018, 0.0038, 0.0041, 0.0033, 0.0004, 0.0009, 0.0022, 0.0010,
          0.0017, 0.0046, 0.0014, 0.0016, 0.0016, 0.0025, 0.0029, 0.0025, 0.0008,
          0.0004, 0.0009, 0.0022, 0.0006, 0.0007, 0.0013, 0.0023, 0.0025, 0.0013,
          0.0012, 0.0007, 0.0020, 0.0009, 0.0007, 0.0006, 0.0011, 0.0012, 0.0019,
          0.0011, 0.0010, 0.0009, 0.0008, 0.0006, 0.0010, 0.0016, 0.0037, 0.0012,
          0.0012, 0.0010, 0.0020, 0.0029, 0.0039, 0.0016, 0.0011, 0.0043, 0.0008,
          0.0005, 0.0040, 0.0008, 0.0008, 0.0019, 0.0014, 0.0012, 0.0013, 0.0011,
          0.0017, 0.0012, 0.0025, 0.0003, 0.0017, 0.0009, 0.0032, 0.0009, 0.0026,
          0.0029, 0.0027, 0.0012, 0.0032, 0.0008, 0.0006, 0.0007, 0.0011, 0.0008,
          0.0020, 0.0022, 0.0007, 0.0005, 0.0018, 0.0017, 0.0012, 0.0011, 0.0011,
          0.0015, 0.0015, 0.0018, 0.0023, 0.0015, 0.0005, 0.0007, 0.0007, 0.0020,
          0.0009, 0.0006, 0.0013, 0.0015, 0.0012, 0.0069, 0.0019, 0.0023, 0.0008,
          0.0006, 0.0004, 0.0019, 0.0008, 0.0010, 0.0007, 0.0018, 0.0035, 0.0015,
          0.0016, 0.0031, 0.0023, 0.0013, 0.0042, 0.0019, 0.0012, 0.0007, 0.0018,
          0.0023, 0.0014, 0.0021, 0.0029, 0.0007, 0.0008, 0.0007, 0.0015, 0.0013,
          0.0008, 0.0042, 0.0014, 0.0013, 0.0011, 0.0016, 0.0019, 0.0010, 0.0009,
          0.0018, 0.0058, 0.0018, 0.0012, 0.0010, 0.0016, 0.0021, 0.0016, 0.0022,
          0.0012, 0.0009, 0.0027, 0.0008, 0.0009, 0.0006, 0.0009, 0.0010, 0.0031,
          0.0020, 0.0049, 0.0028, 0.0008, 0.0062, 0.0024, 0.0015, 0.0015, 0.0018,
          0.0015, 0.0013, 0.0036, 0.0024, 0.0026, 0.0039, 0.0113, 0.0008, 0.0037,
          0.0137, 0.0021, 0.0035, 0.0007, 0.0044, 0.0021, 0.0037, 0.0107, 0.0042,
          0.0024, 0.0015, 0.0021, 0.0016, 0.0034, 0.0044, 0.0031, 0.0083, 0.0021,
          0.0018, 0.0029, 0.0013, 0.0031, 0.0017, 0.0010, 0.0007, 0.0023, 0.0012,
          0.0023, 0.0016, 0.0035, 0.0023, 0.0013, 0.0065, 0.0010, 0.0035, 0.0035,
          0.0069, 0.0231, 0.0038, 0.0126, 0.0023, 0.0027, 0.0017, 0.0013, 0.0043,
          0.0017, 0.0040, 0.0045, 0.0026, 0.0034, 0.0036, 0.0008, 0.0055, 0.0032,
          0.0015, 0.0013, 0.0024, 0.0079, 0.0027, 0.0053, 0.0259, 0.0389, 0.0002,
          0.0005, 0.0005, 0.0005, 0.0005, 0.0005, 0.0005, 0.0002, 0.0005, 0.0005,
          0.0007, 0.0003, 0.0002, 0.0002, 0.0002], dtype=torch.float16),
  tensor([0.0000e+00, 3.1143e-02, 8.1665e-02, 4.1168e-02, 5.3558e-02, 5.5756e-02,
          4.6692e-02, 4.8035e-02, 4.9496e-04, 4.4537e-04, 1.2274e-03, 1.8559e-03,
          7.8726e-04, 6.0654e-04, 6.8474e-04, 6.4373e-04, 4.2415e-04, 1.2579e-03,
          6.8235e-04, 2.7966e-04, 6.7997e-04, 1.0490e-03, 6.8331e-04, 1.0681e-03,
          2.1381e-03, 9.5654e-04, 2.9617e-02, 7.7667e-03, 1.2445e-03, 9.2220e-04,
          6.1417e-04, 7.0143e-04, 6.2895e-04, 1.0815e-03, 6.9618e-04, 6.5517e-04,
          1.2331e-03, 1.0405e-03, 1.2026e-03, 7.3147e-04, 6.0511e-04, 1.6508e-03,
          1.4830e-03, 7.4625e-04, 9.1839e-04, 6.6662e-04, 6.8283e-04, 1.0309e-03,
          4.4250e-04, 1.1187e-03, 8.8348e-03, 1.2341e-03, 5.8842e-04, 5.4646e-04,
          4.6897e-04, 4.2152e-04, 7.7248e-04, 1.0424e-03, 5.8889e-04, 8.5258e-04,
          8.3971e-04, 9.2888e-04, 8.6927e-04, 8.4019e-04, 5.4932e-04, 8.6832e-04,
          5.3692e-04, 2.7752e-04, 3.9911e-04, 5.4979e-04, 5.0783e-04, 5.6362e-04,
          1.0099e-03, 2.1687e-03, 1.0729e-03, 6.8188e-04, 2.9182e-04, 2.9731e-04,
          4.9925e-04, 8.7214e-04, 4.6110e-04, 6.3467e-04, 4.8709e-04, 4.6229e-04,
          7.3624e-04, 1.0252e-03, 6.0701e-04, 5.6124e-04, 2.5010e-04, 3.1114e-04,
          1.1530e-03, 7.7057e-04, 2.2233e-04, 8.2064e-04, 4.3082e-04, 9.7466e-04,
          9.6607e-04, 6.3038e-04, 5.9080e-04, 1.0967e-03, 7.8344e-04, 6.6614e-04,
          1.1663e-03, 2.8491e-04, 3.3092e-04, 7.5102e-04, 6.2704e-04, 7.3385e-04,
          6.8331e-04, 1.0939e-03, 1.1473e-03, 3.6573e-04, 3.2377e-04, 2.7204e-04,
          3.4857e-04, 6.9571e-04, 7.4959e-04, 8.2541e-04, 1.0710e-03, 1.5497e-03,
          1.0967e-03, 7.9250e-04, 3.8123e-04, 2.4068e-04, 8.8692e-04, 2.0278e-04,
          5.1165e-04, 2.9349e-04, 3.5572e-04, 7.5960e-04, 5.4741e-04, 7.6914e-04,
          4.3821e-04, 2.7585e-04, 2.6321e-04, 1.5152e-04, 4.9591e-04, 3.2139e-04,
          1.2684e-03, 5.3883e-04, 6.4945e-04, 2.0885e-03, 1.3885e-03, 7.4816e-04,
          3.7169e-04, 2.0111e-04, 2.8253e-04, 3.6430e-04, 3.9458e-04, 7.0953e-04,
          1.0271e-03, 6.3801e-04, 4.9973e-04, 8.9884e-04, 8.8930e-04, 1.2321e-03,
          5.3692e-04, 5.9843e-04, 3.1281e-04, 2.7680e-04, 7.6246e-04, 7.3957e-04,
          1.0920e-03, 9.4318e-04, 8.7452e-04, 1.1358e-03, 1.8625e-03, 4.5347e-04,
          7.8487e-04, 4.4084e-04, 3.9911e-04, 3.9792e-04, 6.8951e-04, 9.9087e-04,
          3.8743e-04, 4.1437e-04, 1.1187e-03, 1.8492e-03, 7.8487e-04, 6.8903e-04,
          4.0603e-04, 4.6635e-04, 7.6771e-04, 4.4847e-04, 3.6740e-04, 4.3917e-04,
          7.4959e-04, 6.4325e-04, 1.0729e-03, 7.4339e-04, 3.4189e-04, 6.5899e-04,
          5.5695e-04, 3.2091e-04, 5.3120e-04, 1.0557e-03, 5.6648e-04, 6.6662e-04,
          7.5293e-04, 8.6641e-04, 7.7343e-04, 3.5810e-04, 3.4785e-04, 4.8971e-04,
          3.2640e-04, 5.5885e-04, 4.9067e-04, 3.9291e-04, 1.1520e-03, 4.4036e-04,
          4.0507e-04, 2.1782e-03, 1.3142e-03, 4.3392e-04, 6.7663e-04, 2.8110e-04,
          7.0810e-04, 7.3195e-04, 1.0147e-03, 3.0637e-04, 9.0551e-04, 5.9843e-04,
          1.2035e-03, 7.2384e-04, 5.7888e-04, 2.9707e-04, 1.0614e-03, 7.3004e-04,
          3.4213e-04, 2.5868e-04, 2.6131e-04, 3.4952e-04, 6.2084e-04, 4.6110e-04,
          2.8305e-03, 3.4928e-04, 3.2330e-04, 3.1829e-04, 5.7650e-04, 8.3351e-04,
          7.8058e-04, 1.1692e-03, 5.4836e-04, 9.0551e-04, 8.7309e-04, 9.3126e-04,
          1.2646e-03, 3.8099e-04, 3.2616e-04, 3.1567e-04, 5.9128e-04, 5.8508e-04,
          2.3758e-04, 3.2878e-04, 7.7438e-04, 6.5708e-04, 6.9523e-04, 2.7771e-03,
          9.8801e-04, 3.2115e-04, 2.9612e-04, 3.1114e-04, 3.5095e-04, 6.4182e-04,
          5.2595e-04, 5.7030e-04, 3.9482e-04, 9.3079e-04, 1.0891e-03, 5.1785e-04,
          5.1737e-04, 3.7742e-04, 7.1526e-04, 5.8174e-04, 8.1491e-04, 5.3215e-04,
          4.2439e-04, 3.1161e-04, 1.0223e-03, 6.5708e-04, 1.9207e-03, 4.3535e-04,
          4.0030e-04, 4.3535e-04, 4.3440e-04, 6.3610e-04, 5.8317e-04, 4.0793e-04,
          4.1652e-04, 1.5421e-03, 5.9509e-04, 7.2098e-04, 8.6117e-04, 6.7997e-04,
          8.8453e-04, 4.2844e-04, 4.3964e-04, 5.6362e-04, 5.8270e-04, 5.3358e-04,
          6.2752e-04, 5.0545e-04, 5.8174e-04, 1.1435e-03, 8.4734e-04, 4.8184e-04,
          5.7364e-04, 8.3113e-04, 7.8726e-04, 3.5858e-04, 3.9029e-04, 4.1151e-04,
          3.3331e-04, 4.0126e-04, 1.5144e-03, 7.8201e-04, 1.7557e-03, 7.5960e-04,
          3.0684e-04, 1.3123e-03, 6.4707e-04, 4.5204e-04, 8.3256e-04, 1.2989e-03,
          4.6635e-04, 3.8433e-04, 1.3828e-03, 1.2608e-03, 6.7186e-04, 1.1263e-03,
          1.9178e-03, 2.8634e-04, 1.0977e-03, 1.9150e-03, 7.9441e-04, 1.2274e-03,
          2.8896e-04, 1.0376e-03, 5.8603e-04, 1.4391e-03, 8.9216e-04, 8.6737e-04,
          7.1621e-04, 5.2691e-04, 7.5436e-04, 5.2404e-04, 8.6546e-04, 1.2102e-03,
          7.0524e-04, 9.5444e-03, 8.6355e-04, 8.2350e-04, 1.0176e-03, 5.3978e-04,
          8.3494e-04, 4.8232e-04, 2.5368e-04, 3.5429e-04, 6.0320e-04, 3.3283e-04,
          6.4993e-04, 5.0735e-04, 4.8590e-04, 7.1716e-04, 3.7718e-04, 1.5306e-03,
          3.9554e-04, 7.2289e-04, 9.1791e-04, 4.5204e-04, 5.4588e-03, 1.2579e-03,
          1.4099e-02, 1.4915e-03, 1.3437e-03, 5.3740e-04, 4.1485e-04, 2.8915e-03,
          4.2510e-04, 9.3937e-04, 7.5197e-04, 7.9823e-04, 9.1457e-04, 3.9053e-04,
          2.3580e-04, 1.7195e-03, 4.4179e-04, 4.3988e-04, 3.7336e-04, 4.4847e-04,
          1.4687e-03, 3.9601e-04, 1.1187e-03, 2.0504e-03, 2.0027e-03, 2.7734e-01,
          3.2258e-04, 3.2258e-04, 3.2258e-04, 3.2258e-04, 1.5898e-03, 3.2258e-04,
          3.2258e-04, 3.2258e-04, 9.9087e-04, 3.2258e-04, 3.2258e-04, 3.2258e-04,
          3.2258e-04, 3.2258e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.5381e-02, 7.0618e-02, 9.2850e-03, 6.7017e-02, 3.0685e-02,
          4.7150e-02, 4.9438e-02, 5.3167e-04, 3.3140e-04, 1.8206e-03, 2.4300e-03,
          8.3303e-04, 5.3072e-04, 4.0865e-04, 8.4496e-04, 3.8290e-04, 8.7881e-04,
          4.5013e-04, 2.1672e-04, 2.8825e-04, 5.0211e-04, 2.6178e-04, 1.0433e-03,
          2.4357e-03, 1.1244e-03, 3.2043e-02, 1.0147e-02, 5.9319e-04, 6.7711e-04,
          4.2319e-04, 7.4387e-04, 3.3259e-04, 1.1625e-03, 9.2030e-04, 5.4169e-04,
          8.1062e-04, 5.0974e-04, 1.4353e-03, 8.6927e-04, 2.8539e-04, 7.5340e-04,
          4.8327e-04, 3.2306e-04, 3.3903e-04, 2.4533e-04, 4.9543e-04, 1.1282e-03,
          2.0301e-04, 5.0306e-04, 5.3711e-03, 7.5531e-04, 4.8304e-04, 4.2653e-04,
          2.9325e-04, 2.6965e-04, 4.2915e-04, 3.9577e-04, 4.4513e-04, 4.3869e-04,
          4.5943e-04, 4.6659e-04, 4.6945e-04, 6.0987e-04, 2.6345e-04, 1.0529e-03,
          4.2486e-04, 2.4414e-04, 3.6144e-04, 1.9956e-04, 1.7750e-04, 2.1517e-04,
          3.3331e-04, 3.0060e-03, 5.5599e-04, 5.3596e-04, 1.8084e-04, 1.5986e-04,
          2.9635e-04, 3.3808e-04, 2.9230e-04, 4.5228e-04, 3.1281e-04, 2.5654e-04,
          3.4904e-04, 9.3555e-04, 2.9492e-04, 5.8603e-04, 2.0897e-04, 1.9646e-04,
          1.1454e-03, 6.8283e-04, 9.2447e-05, 7.3910e-04, 1.4520e-04, 3.2568e-04,
          4.5753e-04, 3.3903e-04, 4.9353e-04, 7.7248e-04, 3.0375e-04, 2.9230e-04,
          1.0624e-03, 2.4021e-04, 2.1803e-04, 4.3726e-04, 6.5041e-04, 3.2806e-04,
          4.6873e-04, 3.3355e-04, 8.9216e-04, 1.5771e-04, 2.1374e-04, 2.0742e-04,
          3.2210e-04, 2.4891e-04, 2.4962e-04, 9.2030e-04, 3.7575e-04, 6.2799e-04,
          4.5347e-04, 4.8518e-04, 1.5616e-04, 1.9825e-04, 9.7752e-04, 9.2626e-05,
          1.7500e-04, 1.5175e-04, 1.8275e-04, 5.7411e-04, 4.3797e-04, 5.1451e-04,
          3.3665e-04, 1.9586e-04, 2.4652e-04, 8.4043e-05, 3.7003e-04, 1.1128e-04,
          1.2007e-03, 2.2030e-04, 6.3801e-04, 2.5826e-03, 2.0161e-03, 3.4809e-04,
          1.5402e-04, 1.7405e-04, 1.6129e-04, 1.5974e-04, 2.6083e-04, 5.3072e-04,
          8.8882e-04, 3.6168e-04, 1.9300e-04, 4.8518e-04, 5.4455e-04, 4.3416e-04,
          3.8028e-04, 2.9135e-04, 1.8978e-04, 1.2219e-04, 4.9877e-04, 3.3450e-04,
          1.1044e-03, 9.2030e-04, 6.4850e-04, 1.8177e-03, 6.6614e-04, 2.2960e-04,
          6.2227e-04, 2.9373e-04, 2.5439e-04, 3.3450e-04, 4.6587e-04, 7.2813e-04,
          1.8346e-04, 1.8370e-04, 5.0879e-04, 6.3133e-04, 5.2452e-04, 6.1131e-04,
          3.7646e-04, 2.5034e-04, 7.8535e-04, 4.3058e-04, 2.7394e-04, 3.9935e-04,
          5.5695e-04, 2.7037e-04, 1.6136e-03, 2.7156e-04, 1.8990e-04, 5.3072e-04,
          4.4537e-04, 1.1981e-04, 4.7970e-04, 6.5041e-04, 5.6601e-04, 5.3263e-04,
          3.2330e-04, 2.6464e-04, 4.1246e-04, 1.8203e-04, 4.5609e-04, 4.3106e-04,
          1.3781e-04, 2.8634e-04, 1.6487e-04, 3.1304e-04, 8.8787e-04, 2.2185e-04,
          2.9922e-04, 8.9931e-04, 4.5586e-04, 2.6011e-04, 4.7708e-04, 1.2851e-04,
          3.7408e-04, 3.9148e-04, 1.1072e-03, 2.8729e-04, 7.4339e-04, 2.6870e-04,
          8.3542e-04, 1.9383e-04, 2.5296e-04, 2.4724e-04, 1.1673e-03, 6.0272e-04,
          1.9169e-04, 1.0478e-04, 1.0157e-04, 1.9717e-04, 3.9172e-04, 1.9205e-04,
          1.1234e-03, 1.7583e-04, 1.3328e-04, 2.5964e-04, 2.7084e-04, 3.6740e-04,
          2.4390e-04, 4.0746e-04, 1.7989e-04, 3.9387e-04, 3.6645e-04, 3.2330e-04,
          3.4642e-04, 1.6487e-04, 1.0651e-04, 1.2290e-04, 2.2888e-04, 3.7193e-04,
          8.9526e-05, 9.4354e-05, 2.2829e-04, 2.6965e-04, 5.7030e-04, 2.9984e-03,
          3.3569e-04, 2.1064e-04, 1.6630e-04, 1.7941e-04, 1.1104e-04, 2.0385e-04,
          2.7370e-04, 5.7793e-04, 2.5868e-04, 5.1260e-04, 7.9298e-04, 2.3556e-04,
          1.9479e-04, 2.1589e-04, 4.8733e-04, 1.8048e-04, 6.2275e-04, 3.2401e-04,
          1.6212e-04, 1.5020e-04, 3.1137e-04, 3.1161e-04, 5.8031e-04, 1.9002e-04,
          2.4843e-04, 1.4377e-04, 1.3196e-04, 2.2256e-04, 2.4033e-04, 2.2876e-04,
          1.9956e-04, 1.9264e-03, 3.7885e-04, 4.2224e-04, 2.6298e-04, 2.4128e-04,
          2.6059e-04, 1.3554e-04, 1.3232e-04, 4.5919e-04, 3.8934e-04, 1.7321e-04,
          1.9324e-04, 2.8849e-04, 4.1151e-04, 5.5742e-04, 2.5201e-04, 4.6158e-04,
          3.9482e-04, 2.4986e-04, 4.0817e-04, 1.6809e-04, 3.1734e-04, 2.8515e-04,
          1.6427e-04, 2.0683e-04, 4.9543e-04, 4.8137e-04, 1.5240e-03, 4.3130e-04,
          1.9825e-04, 1.0929e-03, 3.0780e-04, 1.7369e-04, 2.8110e-04, 4.7922e-04,
          2.3651e-04, 1.4091e-04, 1.0042e-03, 3.4618e-04, 2.7227e-04, 4.8971e-04,
          1.5583e-03, 8.7559e-05, 3.8934e-04, 1.7090e-03, 9.5606e-04, 4.1437e-04,
          1.2434e-04, 9.5320e-04, 3.9339e-04, 3.8910e-04, 6.4325e-04, 3.3855e-04,
          8.5783e-04, 4.9114e-04, 3.1829e-04, 4.8876e-04, 6.4945e-04, 4.9639e-04,
          2.1517e-04, 5.1155e-03, 2.8110e-04, 2.8658e-04, 6.4754e-04, 3.1209e-04,
          6.2037e-04, 4.0817e-04, 2.1732e-04, 3.3355e-04, 2.3723e-04, 2.0993e-04,
          3.8433e-04, 3.0041e-04, 3.7384e-04, 6.6996e-04, 2.6894e-04, 2.0161e-03,
          3.3712e-04, 2.3413e-04, 7.0858e-04, 3.8218e-04, 6.6376e-03, 1.8206e-03,
          6.3438e-03, 4.9257e-04, 4.9496e-04, 3.7575e-04, 2.1648e-04, 8.2588e-04,
          1.7536e-04, 4.9257e-04, 6.4898e-04, 7.1192e-04, 3.4761e-04, 3.8266e-04,
          1.3149e-04, 5.2309e-04, 2.0373e-04, 2.9635e-04, 2.7442e-04, 3.8481e-04,
          9.0456e-04, 2.6584e-04, 1.2341e-03, 1.2312e-03, 1.2083e-03, 1.3574e-01,
          3.1494e-01, 5.0259e-04, 5.0259e-04, 1.3695e-03, 1.3695e-03, 1.3695e-03,
          5.0259e-04, 1.3695e-03, 5.0259e-04, 8.0824e-04, 5.0259e-04, 5.0259e-04,
          5.0259e-04, 5.0259e-04], dtype=torch.float16),
  tensor([0.0000e+00, 4.2572e-03, 4.0619e-02, 3.8662e-03, 2.9770e-02, 2.3102e-02,
          4.6814e-02, 1.7349e-02, 1.1234e-03, 4.7421e-04, 2.5997e-03, 3.5877e-03,
          1.1730e-03, 1.0538e-03, 7.9679e-04, 1.0815e-03, 8.7023e-04, 1.8997e-03,
          9.9945e-04, 3.9077e-04, 3.8433e-04, 4.5776e-04, 4.0412e-04, 1.5850e-03,
          3.5496e-03, 1.0357e-03, 1.3580e-02, 8.8501e-03, 6.8426e-04, 1.4353e-03,
          9.6941e-04, 1.7633e-03, 5.4550e-04, 1.8721e-03, 1.6518e-03, 1.4963e-03,
          1.3733e-03, 6.2227e-04, 2.5387e-03, 1.5802e-03, 5.2404e-04, 1.4248e-03,
          5.4646e-04, 4.6015e-04, 4.2105e-04, 4.6539e-04, 1.2054e-03, 1.7605e-03,
          3.8767e-04, 6.4754e-04, 3.5000e-03, 2.6741e-03, 1.3781e-03, 8.9169e-04,
          5.1641e-04, 6.8331e-04, 5.7220e-04, 6.6900e-04, 1.0538e-03, 7.3004e-04,
          5.5790e-04, 8.5783e-04, 6.8378e-04, 1.2264e-03, 4.5276e-04, 1.9627e-03,
          6.9189e-04, 5.2023e-04, 6.7949e-04, 2.5439e-04, 1.8609e-04, 2.9445e-04,
          4.2605e-04, 4.3526e-03, 8.8549e-04, 1.5478e-03, 3.2425e-04, 3.5334e-04,
          8.0013e-04, 5.2166e-04, 5.3692e-04, 9.8038e-04, 6.4135e-04, 4.1270e-04,
          3.7026e-04, 2.0714e-03, 4.1080e-04, 1.1206e-03, 4.2820e-04, 3.4595e-04,
          2.1458e-03, 1.3885e-03, 1.6856e-04, 1.3065e-03, 2.1541e-04, 4.1938e-04,
          7.4291e-04, 4.5252e-04, 1.7948e-03, 1.5421e-03, 4.1819e-04, 5.9605e-04,
          2.1038e-03, 4.4274e-04, 4.2367e-04, 9.6703e-04, 1.3781e-03, 5.4693e-04,
          1.1015e-03, 4.2105e-04, 1.6098e-03, 2.2805e-04, 3.6812e-04, 5.2214e-04,
          5.4741e-04, 4.1842e-04, 3.0637e-04, 1.7328e-03, 4.9210e-04, 6.2561e-04,
          8.4114e-04, 1.0910e-03, 1.9538e-04, 5.1355e-04, 2.1343e-03, 1.9705e-04,
          2.5797e-04, 2.7347e-04, 3.1066e-04, 1.2999e-03, 9.8133e-04, 1.1969e-03,
          5.7602e-04, 4.1914e-04, 4.4227e-04, 1.9014e-04, 8.0013e-04, 1.8394e-04,
          2.9125e-03, 4.5514e-04, 1.1311e-03, 2.9316e-03, 3.4504e-03, 8.3923e-04,
          2.3532e-04, 2.7108e-04, 3.4189e-04, 4.3464e-04, 4.9877e-04, 1.0061e-03,
          2.1648e-03, 7.6675e-04, 3.8052e-04, 1.4334e-03, 8.9216e-04, 8.9169e-04,
          8.9169e-04, 7.5865e-04, 3.7408e-04, 2.5916e-04, 9.5272e-04, 8.2684e-04,
          1.7128e-03, 1.7128e-03, 1.8511e-03, 2.2144e-03, 9.4652e-04, 7.6151e-04,
          1.3018e-03, 8.7976e-04, 5.5218e-04, 5.4741e-04, 1.1473e-03, 1.2388e-03,
          4.1342e-04, 4.8137e-04, 6.9952e-04, 7.4100e-04, 1.0481e-03, 1.5869e-03,
          6.8521e-04, 3.9768e-04, 1.7929e-03, 1.2026e-03, 6.8998e-04, 5.7364e-04,
          9.9373e-04, 6.9141e-04, 1.4753e-03, 3.8028e-04, 3.3927e-04, 1.0757e-03,
          1.1787e-03, 1.6642e-04, 8.6117e-04, 1.6775e-03, 1.0700e-03, 9.0170e-04,
          5.3978e-04, 3.6812e-04, 7.6437e-04, 4.2152e-04, 7.9203e-04, 6.9284e-04,
          2.3532e-04, 6.9427e-04, 2.1553e-04, 6.0225e-04, 1.7538e-03, 5.1403e-04,
          6.1226e-04, 1.1797e-03, 5.5313e-04, 4.6897e-04, 9.3842e-04, 2.6178e-04,
          7.9823e-04, 1.0500e-03, 1.9817e-03, 4.5085e-04, 1.5144e-03, 8.3876e-04,
          1.6699e-03, 2.5725e-04, 5.2404e-04, 6.6900e-04, 2.3899e-03, 9.7942e-04,
          4.2844e-04, 1.9312e-04, 1.6057e-04, 4.2987e-04, 7.0572e-04, 4.2844e-04,
          1.0929e-03, 4.6182e-04, 2.2018e-04, 3.7837e-04, 5.3453e-04, 6.7043e-04,
          3.5620e-04, 4.6849e-04, 4.0412e-04, 8.2684e-04, 1.1787e-03, 6.4421e-04,
          5.4026e-04, 3.3641e-04, 1.9324e-04, 2.4486e-04, 3.9315e-04, 9.3746e-04,
          1.3459e-04, 1.2755e-04, 2.5868e-04, 4.8018e-04, 1.1473e-03, 4.2801e-03,
          6.1369e-04, 6.1226e-04, 3.2616e-04, 3.2759e-04, 1.5640e-04, 3.9530e-04,
          4.9496e-04, 8.7261e-04, 6.4707e-04, 1.6899e-03, 2.6627e-03, 4.9210e-04,
          3.8600e-04, 5.4264e-04, 1.1625e-03, 3.4523e-04, 1.2121e-03, 7.2479e-04,
          3.2711e-04, 2.7442e-04, 3.6955e-04, 8.1253e-04, 6.8188e-04, 3.2663e-04,
          5.5027e-04, 1.8394e-04, 1.8156e-04, 2.3723e-04, 5.9366e-04, 4.9114e-04,
          3.9411e-04, 2.7332e-03, 1.0357e-03, 1.2808e-03, 2.9016e-04, 3.6693e-04,
          3.8433e-04, 2.2316e-04, 1.9169e-04, 8.9693e-04, 1.1234e-03, 2.7752e-04,
          2.3961e-04, 4.9210e-04, 1.0929e-03, 1.3971e-03, 3.3879e-04, 9.5797e-04,
          6.4850e-04, 3.3402e-04, 8.1015e-04, 4.5967e-04, 5.6458e-04, 4.5323e-04,
          4.4751e-04, 3.6001e-04, 7.9298e-04, 7.7724e-04, 2.8496e-03, 7.7581e-04,
          3.0398e-04, 2.1667e-03, 8.2922e-04, 3.8338e-04, 3.3212e-04, 5.4169e-04,
          6.7186e-04, 4.1771e-04, 2.4147e-03, 5.3120e-04, 5.6219e-04, 1.2827e-03,
          2.7599e-03, 1.5306e-04, 8.0919e-04, 2.7618e-03, 1.6251e-03, 1.1320e-03,
          2.4033e-04, 2.0294e-03, 1.2512e-03, 5.7936e-04, 1.4229e-03, 7.6580e-04,
          1.3199e-03, 9.0170e-04, 9.5129e-04, 9.2602e-04, 2.0351e-03, 9.3794e-04,
          3.0589e-04, 3.1052e-03, 5.1355e-04, 5.2261e-04, 2.5501e-03, 9.2173e-04,
          1.7357e-03, 9.9277e-04, 3.6979e-04, 4.4942e-04, 4.3845e-04, 4.8709e-04,
          1.1539e-03, 8.3828e-04, 7.3433e-04, 1.3447e-03, 4.5180e-04, 3.1376e-03,
          5.6696e-04, 3.9411e-04, 9.7752e-04, 8.5115e-04, 8.3618e-03, 2.2354e-03,
          3.4809e-03, 5.8460e-04, 7.2002e-04, 8.4877e-04, 6.3658e-04, 9.0551e-04,
          3.9172e-04, 1.0395e-03, 1.2760e-03, 1.6451e-03, 7.1335e-04, 4.6802e-04,
          2.5010e-04, 5.3549e-04, 4.6945e-04, 6.2466e-04, 4.2582e-04, 6.6185e-04,
          1.5945e-03, 6.1655e-04, 1.4019e-03, 2.2392e-03, 2.2430e-03, 4.6783e-02,
          2.1436e-01, 1.8372e-01, 3.1567e-04, 3.1567e-04, 3.1567e-04, 3.1567e-04,
          3.1567e-04, 3.1567e-04, 3.1567e-04, 1.7090e-03, 1.3618e-03, 1.3618e-03,
          8.4019e-04, 3.1567e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.1191e-03, 2.9556e-02, 2.5024e-03, 1.5114e-02, 9.0332e-03,
          2.8717e-02, 7.5264e-03, 7.4911e-04, 3.7909e-04, 4.4518e-03, 7.5264e-03,
          3.0441e-03, 8.0109e-04, 4.2725e-04, 1.1797e-03, 6.4993e-04, 1.2321e-03,
          7.7820e-04, 3.1328e-04, 1.5414e-04, 1.9193e-04, 1.8692e-04, 1.3323e-03,
          9.7504e-03, 1.2531e-03, 5.4245e-03, 2.9678e-03, 2.4116e-04, 1.2751e-03,
          5.7793e-04, 1.6689e-03, 4.9353e-04, 2.1267e-03, 2.9392e-03, 7.9823e-04,
          1.5945e-03, 5.4455e-04, 2.4605e-03, 1.9150e-03, 2.0790e-04, 2.0428e-03,
          6.0320e-04, 4.4227e-04, 4.3964e-04, 4.4656e-04, 1.1520e-03, 2.1248e-03,
          2.6155e-04, 2.4354e-04, 1.5097e-03, 1.1501e-03, 1.2102e-03, 6.5088e-04,
          2.0969e-04, 5.7077e-04, 2.1243e-04, 1.0023e-03, 1.3380e-03, 1.6289e-03,
          7.5388e-04, 5.1117e-04, 3.3379e-04, 6.7759e-04, 1.2064e-03, 3.0689e-03,
          1.0834e-03, 1.1702e-03, 1.0748e-03, 2.7657e-04, 2.4962e-04, 3.3736e-04,
          3.4118e-04, 6.7902e-03, 2.7704e-04, 1.3018e-03, 1.7035e-04, 2.0278e-04,
          4.4727e-04, 8.1635e-04, 1.5116e-03, 2.5673e-03, 1.5841e-03, 6.5708e-04,
          3.1877e-04, 1.7309e-03, 2.6655e-04, 3.0804e-03, 8.0872e-04, 5.0163e-04,
          6.4087e-03, 3.4351e-03, 2.2817e-04, 4.0627e-03, 2.1064e-04, 2.4235e-04,
          2.9039e-04, 1.3542e-04, 1.1902e-03, 1.7500e-03, 6.6042e-04, 1.2960e-03,
          6.9389e-03, 5.7840e-04, 1.3056e-03, 1.6136e-03, 1.9093e-03, 3.3784e-04,
          1.0681e-03, 5.8651e-04, 4.8904e-03, 2.5439e-04, 5.5027e-04, 9.3269e-04,
          7.7820e-04, 5.3549e-04, 1.3268e-04, 2.5215e-03, 2.8229e-04, 2.5892e-04,
          4.9543e-04, 2.0370e-03, 3.2806e-04, 4.2558e-04, 2.4815e-03, 3.4189e-04,
          3.5501e-04, 5.5408e-04, 6.8426e-04, 1.9741e-03, 1.1005e-03, 1.1845e-03,
          8.2397e-04, 8.2159e-04, 4.8470e-04, 3.5596e-04, 2.5539e-03, 2.4414e-04,
          2.1820e-03, 2.7037e-04, 6.7568e-04, 1.7567e-03, 1.8368e-03, 6.4039e-04,
          3.9506e-04, 3.4761e-04, 5.1355e-04, 9.6464e-04, 8.1968e-04, 2.1610e-03,
          5.2261e-03, 1.5535e-03, 6.3086e-04, 9.9850e-04, 4.2200e-04, 1.2369e-03,
          2.1763e-03, 1.1625e-03, 1.0662e-03, 3.6669e-04, 1.7853e-03, 5.6410e-04,
          1.8673e-03, 2.0142e-03, 1.3475e-03, 1.7290e-03, 8.7452e-04, 8.3923e-04,
          4.0894e-03, 1.2207e-03, 1.0223e-03, 6.6805e-04, 7.4720e-04, 1.5955e-03,
          6.7234e-04, 5.0974e-04, 5.9271e-04, 8.7166e-04, 2.0790e-03, 1.4381e-03,
          8.1444e-04, 7.7009e-04, 1.8072e-03, 1.2074e-03, 4.4966e-04, 5.4121e-04,
          3.3550e-03, 3.2496e-04, 1.2503e-03, 5.2834e-04, 5.4789e-04, 1.2531e-03,
          1.1415e-03, 2.6965e-04, 1.0958e-03, 1.3361e-03, 2.7351e-03, 1.4887e-03,
          5.1641e-04, 2.0540e-04, 1.5602e-03, 1.2674e-03, 1.6308e-03, 1.0242e-03,
          3.9768e-04, 2.1420e-03, 3.1662e-04, 5.4026e-04, 6.1646e-03, 4.3201e-04,
          4.7278e-04, 8.4639e-04, 6.4659e-04, 1.8988e-03, 1.6909e-03, 4.4060e-04,
          1.1683e-03, 7.8487e-04, 2.9011e-03, 5.4312e-04, 1.7385e-03, 7.1239e-04,
          1.6031e-03, 3.3832e-04, 1.2636e-03, 9.0933e-04, 2.8419e-03, 1.4181e-03,
          9.1839e-04, 2.2507e-04, 2.2316e-04, 3.6979e-04, 8.0967e-04, 2.9373e-04,
          9.2888e-04, 7.7963e-04, 2.9135e-04, 6.2656e-04, 9.6989e-04, 1.2836e-03,
          6.2513e-04, 4.7302e-04, 4.1008e-04, 6.7902e-04, 1.0786e-03, 3.5572e-04,
          6.3276e-04, 1.1902e-03, 2.0766e-04, 3.1805e-04, 4.2701e-04, 1.4524e-03,
          2.1279e-04, 1.5032e-04, 2.4605e-04, 4.2987e-04, 1.2560e-03, 2.6741e-03,
          7.4339e-04, 8.0013e-04, 4.3702e-04, 5.1498e-04, 2.0909e-04, 7.5436e-04,
          6.8665e-04, 9.2459e-04, 5.3024e-04, 1.3275e-03, 1.5783e-03, 7.5769e-04,
          4.9496e-04, 9.1362e-04, 1.4668e-03, 4.8637e-04, 1.6203e-03, 9.4557e-04,
          4.1771e-04, 2.9635e-04, 1.5259e-04, 5.5218e-04, 5.2500e-04, 4.8614e-04,
          9.8515e-04, 2.6035e-04, 2.4772e-04, 2.3794e-04, 7.4482e-04, 6.3086e-04,
          6.0892e-04, 2.8820e-03, 7.3004e-04, 9.4414e-04, 3.2115e-04, 5.8699e-04,
          4.8232e-04, 3.4618e-04, 2.6608e-04, 1.2293e-03, 1.7776e-03, 2.8300e-04,
          2.0540e-04, 4.6778e-04, 7.6628e-04, 8.6069e-04, 4.4966e-04, 1.6909e-03,
          1.0748e-03, 1.8561e-04, 9.7656e-04, 2.7418e-04, 4.5848e-04, 6.9666e-04,
          3.0446e-04, 3.9363e-04, 3.6430e-04, 9.9277e-04, 2.1629e-03, 1.3056e-03,
          3.5238e-04, 5.4436e-03, 5.8842e-04, 3.6883e-04, 2.3079e-04, 3.6478e-04,
          3.4451e-04, 2.5511e-04, 1.1120e-03, 5.9366e-04, 7.2479e-04, 9.2697e-04,
          3.4733e-03, 1.5295e-04, 6.1989e-04, 3.5114e-03, 1.4305e-03, 4.0793e-04,
          2.5010e-04, 1.6212e-03, 8.4496e-04, 3.1471e-04, 2.9583e-03, 5.7077e-04,
          1.3618e-03, 9.2983e-04, 5.3549e-04, 1.0624e-03, 1.5068e-03, 4.7421e-04,
          1.5295e-04, 1.2321e-03, 4.9639e-04, 4.3035e-04, 1.5535e-03, 6.6805e-04,
          1.5259e-03, 7.1526e-04, 5.6744e-04, 3.7122e-04, 3.3808e-04, 4.4298e-04,
          6.5374e-04, 5.6982e-04, 9.0122e-04, 1.0004e-03, 3.5810e-04, 2.7332e-03,
          5.4932e-04, 2.4199e-04, 6.7139e-04, 1.0195e-03, 3.8280e-03, 1.2817e-03,
          1.2522e-03, 5.1546e-04, 7.3719e-04, 7.1287e-04, 4.2748e-04, 4.4727e-04,
          2.4188e-04, 8.3160e-04, 2.3689e-03, 1.1921e-03, 3.3450e-04, 4.0054e-04,
          2.0981e-04, 2.6393e-04, 3.2187e-04, 4.7708e-04, 7.0620e-04, 9.6750e-04,
          7.6818e-04, 4.4322e-04, 8.0824e-04, 2.1763e-03, 1.6851e-03, 3.4424e-02,
          1.4490e-01, 1.2201e-01, 1.7322e-01, 3.0112e-04, 3.0112e-04, 3.0112e-04,
          3.0112e-04, 3.0112e-04, 3.0112e-04, 6.8855e-04, 6.8855e-04, 9.0599e-04,
          3.0112e-04, 3.0112e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.9369e-03, 3.1769e-02, 1.7529e-03, 1.7517e-02, 1.0399e-02,
          2.8168e-02, 7.3662e-03, 4.4465e-04, 2.6464e-04, 1.5335e-03, 1.9102e-03,
          7.1526e-04, 8.3828e-04, 6.9427e-04, 8.4352e-04, 3.5715e-04, 1.9531e-03,
          9.2793e-04, 2.1839e-04, 2.4188e-04, 3.6120e-04, 3.5095e-04, 1.5726e-03,
          2.0275e-03, 9.7084e-04, 6.0654e-03, 3.1662e-03, 3.2902e-04, 1.5469e-03,
          8.5402e-04, 8.5878e-04, 4.9734e-04, 1.9531e-03, 1.1196e-03, 6.2704e-04,
          1.4772e-03, 4.5276e-04, 2.3708e-03, 1.5717e-03, 2.6774e-04, 2.0504e-03,
          6.6519e-04, 5.1403e-04, 3.6192e-04, 3.9697e-04, 1.3332e-03, 1.8826e-03,
          2.8968e-04, 2.8563e-04, 1.5821e-03, 1.1826e-03, 5.8174e-04, 4.8494e-04,
          2.0349e-04, 3.0899e-04, 2.9039e-04, 8.2731e-04, 9.3031e-04, 7.9393e-04,
          5.1403e-04, 7.6294e-04, 3.8362e-04, 1.5450e-03, 8.3876e-04, 2.0256e-03,
          8.2779e-04, 4.8590e-04, 6.7568e-04, 3.0875e-04, 2.0802e-04, 2.2173e-04,
          2.3878e-04, 2.4395e-03, 3.7193e-04, 5.9748e-04, 1.4269e-04, 1.7798e-04,
          3.6478e-04, 7.9060e-04, 7.9155e-04, 1.1091e-03, 7.5912e-04, 3.3760e-04,
          2.7394e-04, 1.0252e-03, 3.1900e-04, 5.5599e-04, 4.7946e-04, 3.8695e-04,
          1.1673e-03, 1.4839e-03, 1.7941e-04, 8.5211e-04, 1.9944e-04, 2.2566e-04,
          3.8671e-04, 1.7893e-04, 6.9237e-04, 4.6387e-03, 7.5483e-04, 9.6464e-04,
          1.1902e-03, 4.1914e-04, 4.5776e-04, 9.6893e-04, 1.2350e-03, 5.0068e-04,
          7.1192e-04, 8.6069e-04, 1.8148e-03, 3.1400e-04, 3.6025e-04, 5.1498e-04,
          5.2786e-04, 7.1907e-04, 1.9276e-04, 1.6766e-03, 2.5153e-04, 3.1805e-04,
          5.7411e-04, 1.9112e-03, 2.3711e-04, 3.4666e-04, 1.2770e-03, 2.6131e-04,
          3.5548e-04, 5.2595e-04, 5.3787e-04, 1.2026e-03, 8.5354e-04, 7.3814e-04,
          5.1785e-04, 3.3736e-04, 2.3961e-04, 2.3937e-04, 8.2493e-04, 2.2292e-04,
          2.8305e-03, 3.1161e-04, 7.0333e-04, 1.5860e-03, 1.6012e-03, 6.4707e-04,
          4.0078e-04, 1.5616e-04, 4.7278e-04, 6.3705e-04, 5.4455e-04, 1.1349e-03,
          2.4281e-03, 9.4748e-04, 4.2629e-04, 2.0256e-03, 6.3705e-04, 2.0027e-03,
          1.0748e-03, 1.3561e-03, 1.9741e-04, 3.6597e-04, 1.7920e-03, 6.1607e-04,
          1.3676e-03, 1.9932e-03, 1.5335e-03, 1.3895e-03, 1.4677e-03, 8.9407e-04,
          6.7663e-04, 8.2731e-04, 5.6982e-04, 4.1032e-04, 6.7520e-04, 1.3609e-03,
          4.5586e-04, 3.7122e-04, 5.9223e-04, 1.2312e-03, 1.6432e-03, 9.7322e-04,
          5.7316e-04, 4.8375e-04, 9.6035e-04, 5.2261e-04, 6.0844e-04, 4.7827e-04,
          5.7316e-04, 2.7776e-04, 6.8808e-04, 5.3644e-04, 3.5620e-04, 6.2943e-04,
          4.6563e-04, 1.9741e-04, 7.3576e-04, 1.2856e-03, 6.0225e-04, 8.2016e-04,
          5.1975e-04, 3.0065e-04, 1.1072e-03, 6.1798e-04, 6.7282e-04, 7.5674e-04,
          2.6727e-04, 1.0643e-03, 2.5177e-04, 3.8409e-04, 9.5701e-04, 4.2725e-04,
          3.7456e-04, 1.3475e-03, 7.1716e-04, 4.5967e-04, 8.9741e-04, 2.9635e-04,
          1.5402e-03, 8.6689e-04, 1.1234e-03, 3.0184e-04, 1.7328e-03, 4.2844e-04,
          2.7981e-03, 3.5501e-04, 7.1859e-04, 5.4264e-04, 1.3971e-03, 6.5041e-04,
          3.2902e-04, 1.7405e-04, 1.5259e-04, 3.2210e-04, 8.4066e-04, 2.7299e-04,
          1.0824e-03, 8.6260e-04, 2.2292e-04, 4.0245e-04, 9.0408e-04, 8.5449e-04,
          3.7527e-04, 3.9840e-04, 6.2227e-04, 1.1768e-03, 8.2254e-04, 4.1771e-04,
          9.5701e-04, 2.8944e-04, 1.9109e-04, 2.2268e-04, 3.4356e-04, 8.1301e-04,
          1.4806e-04, 1.2362e-04, 2.1458e-04, 5.5695e-04, 1.1320e-03, 2.8534e-03,
          8.5974e-04, 4.9353e-04, 4.1318e-04, 3.6836e-04, 1.5855e-04, 4.7135e-04,
          5.0545e-04, 4.8590e-04, 5.1022e-04, 8.4686e-04, 1.3742e-03, 5.4455e-04,
          3.3116e-04, 3.6907e-04, 6.7425e-04, 4.0555e-04, 1.6680e-03, 6.2227e-04,
          3.0112e-04, 2.3603e-04, 1.8561e-04, 3.8719e-04, 6.8665e-04, 3.6144e-04,
          4.3678e-04, 2.2399e-04, 2.4891e-04, 2.0337e-04, 8.2779e-04, 1.0281e-03,
          3.5453e-04, 1.7853e-03, 8.2350e-04, 5.8985e-04, 2.8968e-04, 3.9148e-04,
          4.5800e-04, 2.5129e-04, 2.2018e-04, 7.8917e-04, 1.3580e-03, 2.0015e-04,
          1.6773e-04, 3.2830e-04, 4.3130e-04, 1.0185e-03, 4.6515e-04, 7.9536e-04,
          8.1348e-04, 2.6751e-04, 1.1606e-03, 2.7561e-04, 2.9898e-04, 2.3639e-04,
          3.3116e-04, 3.4189e-04, 5.7364e-04, 6.3324e-04, 2.4662e-03, 7.8726e-04,
          2.0564e-04, 1.0605e-03, 7.1764e-04, 3.3951e-04, 2.2078e-04, 4.0221e-04,
          3.8552e-04, 1.6749e-04, 1.1406e-03, 6.6280e-04, 5.6124e-04, 7.8583e-04,
          2.1896e-03, 1.0312e-04, 4.6563e-04, 2.3975e-03, 5.4026e-04, 7.0620e-04,
          2.0349e-04, 1.4200e-03, 5.5552e-04, 3.7766e-04, 1.6432e-03, 3.8052e-04,
          5.5027e-04, 5.0974e-04, 8.7261e-04, 3.4928e-04, 9.2697e-04, 6.7043e-04,
          1.1557e-04, 1.5249e-03, 4.3130e-04, 3.4952e-04, 6.5041e-04, 2.8706e-04,
          1.2140e-03, 4.7231e-04, 3.0208e-04, 2.4295e-04, 1.9681e-04, 4.3154e-04,
          3.7026e-04, 2.9826e-04, 1.5736e-03, 1.0386e-03, 2.8253e-04, 1.5907e-03,
          3.3021e-04, 2.1172e-04, 5.3644e-04, 6.4898e-04, 3.3550e-03, 8.3876e-04,
          1.3008e-03, 4.4370e-04, 5.4312e-04, 4.7183e-04, 2.9063e-04, 5.0068e-04,
          1.7989e-04, 4.5705e-04, 5.9700e-04, 9.0837e-04, 3.8815e-04, 3.0947e-04,
          2.0432e-04, 2.5129e-04, 2.6703e-04, 3.5334e-04, 2.3878e-04, 6.3133e-04,
          8.9312e-04, 4.4870e-04, 4.6635e-04, 2.1801e-03, 1.1463e-03, 2.5024e-02,
          1.5027e-01, 1.0315e-01, 1.6821e-01, 1.5295e-01, 2.9039e-04, 2.9039e-04,
          2.9039e-04, 1.3533e-03, 2.9039e-04, 8.8835e-04, 1.3533e-03, 2.0027e-03,
          5.7936e-04, 2.9039e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.0618e-03, 2.9984e-02, 1.7023e-03, 1.6525e-02, 1.0170e-02,
          2.6398e-02, 7.8735e-03, 4.9400e-04, 2.7919e-04, 1.2522e-03, 1.5173e-03,
          5.0354e-04, 1.3170e-03, 1.2188e-03, 6.7568e-04, 3.6955e-04, 3.2005e-03,
          1.4734e-03, 2.2590e-04, 3.9029e-04, 4.9591e-04, 4.7660e-04, 1.9360e-03,
          1.4486e-03, 6.1417e-04, 6.4964e-03, 3.3512e-03, 4.9686e-04, 2.4586e-03,
          1.6575e-03, 8.7404e-04, 8.6546e-04, 1.8139e-03, 1.0986e-03, 4.6921e-04,
          1.2531e-03, 3.6311e-04, 2.9221e-03, 1.5154e-03, 4.7231e-04, 1.8415e-03,
          4.5466e-04, 3.3116e-04, 2.4819e-04, 2.8610e-04, 2.1763e-03, 1.6623e-03,
          5.5075e-04, 4.1318e-04, 1.5268e-03, 1.1673e-03, 5.8889e-04, 4.5252e-04,
          1.9443e-04, 3.4404e-04, 2.9039e-04, 6.5517e-04, 6.3324e-04, 5.9938e-04,
          4.2367e-04, 1.2760e-03, 3.7289e-04, 1.4820e-03, 4.7517e-04, 1.5459e-03,
          6.1369e-04, 2.1946e-04, 4.8232e-04, 2.0790e-04, 1.4520e-04, 1.5509e-04,
          2.0623e-04, 2.2221e-03, 3.8099e-04, 6.0797e-04, 1.4114e-04, 2.0611e-04,
          4.4060e-04, 6.7854e-04, 4.5872e-04, 9.4318e-04, 4.2915e-04, 2.6083e-04,
          2.2423e-04, 1.0138e-03, 3.2330e-04, 4.1938e-04, 2.9206e-04, 3.1686e-04,
          8.1873e-04, 1.0786e-03, 1.3006e-04, 5.3501e-04, 1.4114e-04, 1.9205e-04,
          5.5790e-04, 2.0063e-04, 7.8106e-04, 6.1684e-03, 5.7268e-04, 8.2874e-04,
          8.8024e-04, 2.9254e-04, 2.5606e-04, 1.1196e-03, 1.0357e-03, 8.7357e-04,
          7.7724e-04, 6.4564e-04, 1.4000e-03, 1.9383e-04, 2.7370e-04, 4.1795e-04,
          4.0507e-04, 7.1621e-04, 2.9087e-04, 1.2121e-03, 2.4068e-04, 4.4203e-04,
          8.5402e-04, 2.1667e-03, 1.8525e-04, 3.8004e-04, 9.9850e-04, 1.6797e-04,
          3.6097e-04, 3.5477e-04, 3.5334e-04, 9.2983e-04, 8.9264e-04, 7.8535e-04,
          3.9172e-04, 2.2650e-04, 2.0480e-04, 1.3244e-04, 5.7125e-04, 2.0063e-04,
          3.6068e-03, 5.4836e-04, 8.3208e-04, 1.5802e-03, 1.5507e-03, 1.0233e-03,
          2.7490e-04, 1.2255e-04, 2.4021e-04, 6.0844e-04, 4.0030e-04, 9.5654e-04,
          2.2049e-03, 7.8583e-04, 3.8242e-04, 2.4281e-03, 6.1417e-04, 1.9970e-03,
          9.7179e-04, 1.5936e-03, 1.5819e-04, 4.4584e-04, 1.6489e-03, 9.9945e-04,
          9.0981e-04, 1.8415e-03, 2.3136e-03, 1.3170e-03, 1.5688e-03, 1.0443e-03,
          4.9877e-04, 8.8120e-04, 4.9973e-04, 3.4499e-04, 1.0328e-03, 1.2999e-03,
          3.3259e-04, 2.8229e-04, 5.4169e-04, 9.8801e-04, 1.7157e-03, 1.2217e-03,
          4.1604e-04, 3.4070e-04, 7.4053e-04, 4.2224e-04, 7.8583e-04, 4.5395e-04,
          4.7827e-04, 4.6945e-04, 5.9271e-04, 4.7159e-04, 2.7037e-04, 4.8780e-04,
          5.0020e-04, 2.2733e-04, 6.3992e-04, 2.0103e-03, 5.0306e-04, 6.7949e-04,
          4.8566e-04, 4.9257e-04, 1.0777e-03, 5.2547e-04, 4.8804e-04, 5.5647e-04,
          2.6417e-04, 9.0027e-04, 2.1732e-04, 3.4523e-04, 6.5899e-04, 5.9319e-04,
          3.2687e-04, 1.6718e-03, 5.3072e-04, 2.7633e-04, 7.6056e-04, 3.0923e-04,
          2.1534e-03, 1.4400e-03, 1.0567e-03, 2.4772e-04, 1.8463e-03, 7.1430e-04,
          4.3907e-03, 3.4523e-04, 8.0872e-04, 3.7265e-04, 1.0509e-03, 5.0306e-04,
          2.4843e-04, 1.6034e-04, 1.2338e-04, 2.6035e-04, 7.5626e-04, 3.0804e-04,
          8.9359e-04, 5.0259e-04, 1.7142e-04, 3.3092e-04, 7.1335e-04, 6.6042e-04,
          3.0136e-04, 3.7551e-04, 7.7677e-04, 1.9817e-03, 1.4935e-03, 7.9441e-04,
          9.4175e-04, 2.0778e-04, 1.6534e-04, 1.6487e-04, 1.9538e-04, 7.0953e-04,
          1.4448e-04, 1.0109e-04, 1.7715e-04, 6.0844e-04, 1.2159e-03, 2.3918e-03,
          9.3126e-04, 3.4928e-04, 3.9506e-04, 3.0708e-04, 1.1295e-04, 3.7146e-04,
          4.2653e-04, 4.5085e-04, 6.2895e-04, 1.3046e-03, 2.3174e-03, 5.5170e-04,
          2.6870e-04, 3.2425e-04, 5.7077e-04, 4.1509e-04, 2.3022e-03, 6.1321e-04,
          2.4354e-04, 2.0111e-04, 2.8419e-04, 6.8283e-04, 6.6376e-04, 3.0446e-04,
          3.4189e-04, 1.9979e-04, 2.8968e-04, 1.8799e-04, 1.3828e-03, 1.2655e-03,
          3.3879e-04, 1.5516e-03, 1.1320e-03, 1.1091e-03, 2.8300e-04, 3.1447e-04,
          4.7946e-04, 2.2638e-04, 1.9097e-04, 5.6887e-04, 1.8005e-03, 1.5843e-04,
          1.4591e-04, 4.3821e-04, 4.1676e-04, 1.4544e-03, 3.4881e-04, 5.1308e-04,
          6.3515e-04, 4.3464e-04, 1.6708e-03, 3.5787e-04, 2.6202e-04, 2.0230e-04,
          4.2319e-04, 2.9969e-04, 9.3985e-04, 6.7663e-04, 2.4776e-03, 7.0667e-04,
          1.5843e-04, 7.9632e-04, 9.7656e-04, 5.0974e-04, 2.1076e-04, 3.7718e-04,
          5.6696e-04, 2.6107e-04, 1.5936e-03, 5.4407e-04, 5.3883e-04, 1.2913e-03,
          1.7729e-03, 1.2672e-04, 7.9346e-04, 2.6188e-03, 4.7421e-04, 1.2474e-03,
          1.8907e-04, 1.2417e-03, 5.7983e-04, 5.6124e-04, 1.5879e-03, 7.1192e-04,
          3.8767e-04, 4.3511e-04, 1.0586e-03, 2.9063e-04, 8.8072e-04, 7.0524e-04,
          1.9526e-04, 1.4000e-03, 3.1567e-04, 2.9707e-04, 7.2813e-04, 3.0708e-04,
          1.0843e-03, 4.3249e-04, 2.3663e-04, 2.0647e-04, 3.7050e-04, 5.8174e-04,
          3.9339e-04, 2.8658e-04, 1.9588e-03, 1.2865e-03, 2.5177e-04, 1.3838e-03,
          2.9850e-04, 3.6597e-04, 5.1403e-04, 5.8556e-04, 2.9469e-03, 6.8378e-04,
          1.2712e-03, 3.7122e-04, 4.8780e-04, 4.9400e-04, 3.1543e-04, 6.2561e-04,
          2.0957e-04, 7.8297e-04, 4.9496e-04, 1.1015e-03, 4.8661e-04, 2.3103e-04,
          1.8239e-04, 2.9707e-04, 3.6335e-04, 4.1032e-04, 1.9181e-04, 5.1498e-04,
          8.6689e-04, 4.4250e-04, 4.4489e-04, 2.3365e-03, 1.1110e-03, 1.8814e-02,
          9.1614e-02, 8.0750e-02, 1.1749e-01, 1.6504e-01, 1.2488e-01, 2.7561e-04,
          2.7561e-04, 2.7561e-04, 2.7561e-04, 1.2531e-03, 1.0281e-03, 1.2531e-03,
          6.2561e-04, 2.7561e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.6169e-03, 2.9526e-02, 2.7599e-03, 1.8127e-02, 1.1986e-02,
          2.3254e-02, 8.9264e-03, 3.6049e-04, 2.1458e-04, 1.3924e-03, 1.5583e-03,
          5.6076e-04, 4.3559e-04, 3.4523e-04, 6.5517e-04, 2.3854e-04, 7.6342e-04,
          4.3917e-04, 1.2052e-04, 3.0184e-04, 5.5313e-04, 3.1853e-04, 7.8535e-04,
          1.4553e-03, 6.9952e-04, 6.1684e-03, 3.7975e-03, 3.9768e-04, 7.8249e-04,
          4.4489e-04, 7.0000e-04, 4.0817e-04, 1.5821e-03, 1.2083e-03, 3.8218e-04,
          1.5278e-03, 5.9366e-04, 6.1493e-03, 1.5659e-03, 2.9945e-04, 2.3937e-03,
          5.0116e-04, 3.9864e-04, 2.7442e-04, 2.5845e-04, 5.3692e-04, 1.6594e-03,
          2.2686e-04, 3.2902e-04, 1.9503e-03, 9.3174e-04, 5.6028e-04, 3.5906e-04,
          2.1672e-04, 3.1614e-04, 3.3140e-04, 1.2589e-03, 8.2922e-04, 1.1415e-03,
          7.2050e-04, 6.8855e-04, 4.0793e-04, 1.1435e-03, 5.9700e-04, 1.4133e-03,
          4.6802e-04, 2.1946e-04, 6.4802e-04, 1.9729e-04, 1.5104e-04, 1.6308e-04,
          2.0707e-04, 2.4395e-03, 3.8815e-04, 6.0987e-04, 1.2541e-04, 1.7726e-04,
          3.8171e-04, 1.6203e-03, 7.1621e-04, 1.1806e-03, 6.9761e-04, 4.5109e-04,
          3.6812e-04, 8.7643e-04, 2.8920e-04, 4.5729e-04, 2.7657e-04, 2.8920e-04,
          8.5735e-04, 1.3304e-03, 1.1742e-04, 5.9080e-04, 1.4365e-04, 1.9228e-04,
          4.4727e-04, 2.1374e-04, 5.2166e-04, 2.5349e-03, 1.6317e-03, 1.5326e-03,
          1.1625e-03, 3.2020e-04, 3.6645e-04, 1.5945e-03, 1.5020e-03, 4.7016e-04,
          5.8699e-04, 7.2575e-04, 1.5879e-03, 2.2042e-04, 3.0017e-04, 3.3736e-04,
          4.1223e-04, 6.5279e-04, 1.8609e-04, 1.1911e-03, 2.3365e-04, 3.9053e-04,
          5.1689e-04, 2.1477e-03, 3.3760e-04, 1.9491e-04, 1.3418e-03, 3.2735e-04,
          1.1215e-03, 5.2929e-04, 5.6505e-04, 1.2331e-03, 4.1127e-04, 5.5742e-04,
          4.8208e-04, 2.6560e-04, 1.3649e-04, 1.2028e-04, 6.5756e-04, 1.9372e-04,
          9.6941e-04, 2.0778e-04, 2.8920e-04, 9.7847e-04, 1.2398e-03, 4.5037e-04,
          5.6744e-04, 1.0604e-04, 3.9554e-04, 8.7023e-04, 5.0116e-04, 1.4381e-03,
          2.3441e-03, 1.1129e-03, 7.3147e-04, 1.0605e-03, 4.7016e-04, 2.3804e-03,
          1.1625e-03, 1.4191e-03, 1.1235e-04, 3.2997e-04, 1.2236e-03, 3.1185e-04,
          1.0109e-03, 2.4281e-03, 5.4407e-04, 1.0357e-03, 2.2545e-03, 1.1797e-03,
          6.7902e-04, 9.1028e-04, 6.2704e-04, 2.9016e-04, 3.7003e-04, 1.4496e-03,
          4.5753e-04, 4.9496e-04, 9.5510e-04, 1.2503e-03, 2.0676e-03, 5.2214e-04,
          5.3835e-04, 3.1257e-04, 7.2861e-04, 2.5988e-04, 2.2268e-04, 2.7442e-04,
          3.8791e-04, 1.8322e-04, 4.7517e-04, 1.0185e-03, 3.5501e-04, 5.6887e-04,
          3.4523e-04, 5.3024e-04, 6.5899e-04, 5.6839e-04, 5.3406e-04, 8.3065e-04,
          8.4066e-04, 4.2486e-04, 1.1759e-03, 6.2704e-04, 5.1641e-04, 6.5851e-04,
          2.7633e-04, 1.2140e-03, 2.8253e-04, 2.0504e-04, 6.8045e-04, 2.0576e-04,
          2.0683e-04, 2.1095e-03, 1.1940e-03, 4.3988e-04, 9.2602e-04, 4.5609e-04,
          1.5793e-03, 5.9366e-04, 9.4414e-04, 1.9598e-04, 1.2608e-03, 3.2425e-04,
          3.6507e-03, 4.6659e-04, 1.2169e-03, 3.4356e-04, 1.0967e-03, 5.0306e-04,
          2.6488e-04, 1.3030e-04, 1.3208e-04, 2.5582e-04, 5.6171e-04, 1.2827e-04,
          1.6069e-03, 7.0620e-04, 2.5511e-04, 3.2783e-04, 8.5020e-04, 7.6675e-04,
          6.0129e-04, 7.0429e-04, 1.2331e-03, 1.4734e-03, 6.2990e-04, 4.5943e-04,
          1.2016e-03, 2.2948e-04, 1.6963e-04, 1.5950e-04, 2.0146e-04, 8.3351e-04,
          1.4985e-04, 1.2267e-04, 2.0444e-04, 5.1498e-04, 4.9877e-04, 1.9779e-03,
          1.3666e-03, 3.8910e-04, 6.0177e-04, 3.5691e-04, 2.5606e-04, 6.3181e-04,
          4.3750e-04, 3.5238e-04, 2.2531e-04, 4.9734e-04, 6.5041e-04, 6.4182e-04,
          2.8729e-04, 2.9969e-04, 5.6744e-04, 4.6134e-04, 1.5812e-03, 5.2261e-04,
          2.4211e-04, 1.6725e-04, 1.9979e-04, 2.7752e-04, 1.0719e-03, 4.6992e-04,
          4.5276e-04, 3.6502e-04, 5.6887e-04, 3.3951e-04, 7.5817e-04, 6.9571e-04,
          3.0231e-04, 1.7738e-03, 3.2282e-04, 3.4308e-04, 3.1424e-04, 3.6764e-04,
          5.3215e-04, 2.4772e-04, 2.3210e-04, 6.0225e-04, 1.5125e-03, 1.6928e-04,
          1.6797e-04, 2.0051e-04, 2.6131e-04, 5.9128e-04, 6.3944e-04, 7.4863e-04,
          5.9366e-04, 3.4976e-04, 1.3914e-03, 1.4389e-04, 1.9169e-04, 1.5318e-04,
          1.4615e-04, 3.8385e-04, 7.4577e-04, 4.6349e-04, 1.8167e-03, 7.0286e-04,
          1.3471e-04, 7.2527e-04, 3.5787e-04, 1.7035e-04, 2.4235e-04, 4.0293e-04,
          1.6713e-04, 1.0097e-04, 6.0415e-04, 1.0672e-03, 6.6280e-04, 4.8590e-04,
          1.9875e-03, 7.1526e-05, 4.1080e-04, 3.9482e-03, 3.5024e-04, 3.2687e-04,
          2.1350e-04, 9.6560e-04, 3.4022e-04, 3.5763e-04, 1.6842e-03, 2.9039e-04,
          3.7050e-04, 3.9768e-04, 2.6417e-04, 2.0516e-04, 5.4359e-04, 4.1103e-04,
          1.3232e-04, 1.2283e-03, 4.9210e-04, 5.2452e-04, 4.5514e-04, 2.1529e-04,
          6.9380e-04, 2.3639e-04, 2.2268e-04, 1.4210e-04, 1.9145e-04, 4.7684e-04,
          2.7895e-04, 1.8847e-04, 7.2145e-04, 4.0483e-04, 1.6224e-04, 1.6851e-03,
          2.3413e-04, 2.0218e-04, 3.5048e-04, 5.6458e-04, 3.0346e-03, 6.5327e-04,
          1.5154e-03, 5.8413e-04, 7.4530e-04, 2.5702e-04, 1.9038e-04, 5.4169e-04,
          1.7023e-04, 3.3164e-04, 4.5180e-04, 4.4608e-04, 2.4986e-04, 2.0766e-04,
          9.8050e-05, 2.7966e-04, 1.4222e-04, 1.8823e-04, 1.8024e-04, 6.5994e-04,
          5.7840e-04, 2.6441e-04, 4.0150e-04, 2.0924e-03, 1.1253e-03, 2.3926e-02,
          9.6802e-02, 6.2378e-02, 9.2529e-02, 1.1493e-01, 6.6833e-02, 1.7175e-01,
          3.7122e-04, 3.7122e-04, 3.7122e-04, 3.7122e-04, 7.1764e-04, 1.2474e-03,
          7.1764e-04, 3.7122e-04], dtype=torch.float16),
  tensor([0.0000e+00, 5.3673e-03, 4.5502e-02, 7.6218e-03, 2.0248e-02, 1.4465e-02,
          3.3997e-02, 1.7120e-02, 2.5344e-04, 1.0645e-04, 7.0810e-04, 1.3809e-03,
          3.7336e-04, 2.1768e-04, 2.1803e-04, 3.2282e-04, 1.5879e-04, 3.2830e-04,
          1.8167e-04, 9.2685e-05, 1.5140e-04, 1.8787e-04, 1.3006e-04, 3.6860e-04,
          1.3552e-03, 5.3787e-04, 1.1040e-02, 2.6970e-03, 2.2984e-04, 2.5368e-04,
          2.2876e-04, 3.8218e-04, 1.3566e-04, 5.2786e-04, 3.3522e-04, 2.7299e-04,
          8.1015e-04, 5.4264e-04, 7.5150e-04, 4.1056e-04, 1.7190e-04, 7.8344e-04,
          4.1580e-04, 2.7227e-04, 2.3568e-04, 1.7142e-04, 1.9920e-04, 5.1975e-04,
          1.0574e-04, 2.1446e-04, 1.9951e-03, 5.2643e-04, 2.9874e-04, 2.1946e-04,
          1.7738e-04, 1.7917e-04, 2.5749e-04, 9.0551e-04, 1.1501e-03, 9.4509e-04,
          5.7459e-04, 2.6226e-04, 3.1567e-04, 3.5095e-04, 4.1676e-04, 5.5933e-04,
          9.1839e-04, 5.0402e-04, 3.1805e-04, 1.6975e-04, 1.2958e-04, 1.4770e-04,
          1.8108e-04, 1.2550e-03, 3.5858e-04, 3.2377e-04, 1.1688e-04, 1.2004e-04,
          2.0349e-04, 1.0414e-03, 1.0986e-03, 6.8808e-04, 1.2503e-03, 3.2830e-04,
          3.5596e-04, 3.8910e-04, 1.9622e-04, 3.4142e-04, 5.6696e-04, 5.4646e-04,
          6.6805e-04, 6.3705e-04, 1.4210e-04, 4.9877e-04, 9.3758e-05, 1.7571e-04,
          2.0826e-04, 2.1601e-04, 3.1495e-04, 8.9550e-04, 1.4133e-03, 1.0796e-03,
          9.6083e-04, 6.0272e-04, 7.2098e-04, 9.6750e-04, 5.3644e-04, 1.7631e-04,
          2.7299e-04, 4.5800e-04, 8.2874e-04, 3.5191e-04, 5.5313e-04, 4.0531e-04,
          3.5834e-04, 2.5773e-04, 9.7156e-05, 4.2748e-04, 2.0039e-04, 3.1710e-04,
          2.5272e-04, 7.4816e-04, 4.5252e-04, 1.1069e-04, 6.7568e-04, 5.1594e-04,
          6.3944e-04, 9.3985e-04, 8.4782e-04, 5.5122e-04, 1.9550e-04, 2.7895e-04,
          6.1417e-04, 4.1652e-04, 1.0812e-04, 2.1887e-04, 4.7231e-04, 1.4234e-04,
          3.8004e-04, 1.0842e-04, 2.1887e-04, 8.1873e-04, 7.0810e-04, 2.1708e-04,
          7.5531e-04, 1.2118e-04, 3.8838e-04, 7.4196e-04, 9.1648e-04, 8.9025e-04,
          1.4000e-03, 5.8126e-04, 3.7384e-04, 3.0637e-04, 3.6407e-04, 8.2922e-04,
          5.9700e-04, 6.8521e-04, 9.9421e-05, 3.2115e-04, 4.9782e-04, 1.7202e-04,
          4.9591e-04, 8.4686e-04, 2.2340e-04, 5.8365e-04, 1.5793e-03, 9.6464e-04,
          5.8746e-04, 6.5708e-04, 3.7813e-04, 1.7488e-04, 2.0945e-04, 6.5231e-04,
          2.4652e-04, 3.3617e-04, 5.8460e-04, 6.2180e-04, 5.9271e-04, 2.5392e-04,
          3.4690e-04, 1.2517e-04, 3.7980e-04, 1.4985e-04, 1.3661e-04, 1.5330e-04,
          2.9659e-04, 1.3626e-04, 5.6458e-04, 9.8705e-04, 6.9857e-04, 2.6846e-04,
          1.8895e-04, 2.8443e-04, 2.6989e-04, 2.4116e-04, 3.4213e-04, 3.3474e-04,
          4.2415e-04, 1.7166e-04, 3.4213e-04, 4.2391e-04, 4.0770e-04, 3.6693e-04,
          2.9993e-04, 5.0831e-04, 1.1986e-04, 1.3626e-04, 5.0592e-04, 1.0043e-04,
          1.3947e-04, 9.1410e-04, 1.9703e-03, 6.8998e-04, 4.4656e-04, 6.0940e-04,
          6.0177e-04, 3.0041e-04, 5.0735e-04, 1.2970e-04, 4.6158e-04, 2.0003e-04,
          5.1212e-04, 2.3448e-04, 4.0054e-04, 6.0415e-04, 5.5218e-04, 2.1768e-04,
          3.7074e-04, 2.1708e-04, 8.7321e-05, 1.9777e-04, 3.2020e-04, 8.3327e-05,
          2.7199e-03, 1.1253e-03, 5.0926e-04, 1.8322e-04, 4.5180e-04, 4.8351e-04,
          5.5981e-04, 6.8617e-04, 7.7152e-04, 5.4550e-04, 2.8992e-04, 2.2030e-04,
          3.3140e-04, 2.9469e-04, 1.7202e-04, 1.6499e-04, 2.1040e-04, 4.3917e-04,
          1.4257e-04, 6.9916e-05, 1.4687e-04, 2.6107e-04, 2.0111e-04, 1.0777e-03,
          9.6226e-04, 7.6008e-04, 3.4738e-04, 1.9324e-04, 1.5938e-04, 4.7946e-04,
          2.2233e-04, 2.3592e-04, 1.5152e-04, 2.5082e-04, 4.0388e-04, 2.4796e-04,
          4.3321e-04, 5.2738e-04, 2.9993e-04, 2.6178e-04, 5.6553e-04, 3.1281e-04,
          1.4520e-04, 1.1414e-04, 1.2219e-04, 1.5640e-04, 9.9277e-04, 7.9012e-04,
          7.0906e-04, 3.6025e-04, 4.1485e-04, 2.2519e-04, 3.0875e-04, 2.4891e-04,
          2.3615e-04, 7.6818e-04, 1.5438e-04, 1.7321e-04, 2.3341e-04, 3.0351e-04,
          2.6679e-04, 1.6963e-04, 1.5986e-04, 1.9586e-04, 5.8317e-04, 1.1712e-04,
          1.1224e-04, 1.2189e-04, 1.6022e-04, 3.2187e-04, 9.9087e-04, 4.5490e-04,
          2.1648e-04, 1.8024e-04, 4.9162e-04, 8.1539e-05, 1.7536e-04, 1.5128e-04,
          8.9347e-05, 2.8253e-04, 2.9874e-04, 2.5773e-04, 6.4754e-04, 3.1519e-04,
          1.5795e-04, 5.0831e-04, 1.9562e-04, 8.0705e-05, 1.6427e-04, 2.4652e-04,
          9.5308e-05, 7.0989e-05, 4.3011e-04, 1.0891e-03, 3.6502e-04, 2.4104e-04,
          1.1368e-03, 4.7624e-05, 2.2876e-04, 1.0872e-03, 4.1914e-04, 2.2626e-04,
          1.3256e-04, 3.4261e-04, 1.5891e-04, 1.9145e-04, 9.0265e-04, 1.7726e-04,
          3.5715e-04, 2.1482e-04, 1.2553e-04, 2.2089e-04, 2.9564e-04, 2.1625e-04,
          9.0599e-05, 2.4395e-03, 5.0354e-04, 4.5490e-04, 2.7633e-04, 1.4031e-04,
          2.9540e-04, 1.6665e-04, 1.7822e-04, 1.4174e-04, 1.2600e-04, 3.0112e-04,
          1.6785e-04, 1.2803e-04, 2.3603e-04, 3.0851e-04, 1.1820e-04, 8.3017e-04,
          1.3876e-04, 1.1569e-04, 3.0398e-04, 3.8457e-04, 2.5024e-03, 5.5599e-04,
          4.6082e-03, 5.6028e-04, 5.5933e-04, 1.4031e-04, 1.0443e-04, 3.3903e-04,
          1.0532e-04, 2.4533e-04, 2.9707e-04, 3.0971e-04, 1.6367e-04, 1.5879e-04,
          7.3791e-05, 2.0325e-04, 1.0633e-04, 1.1712e-04, 1.5986e-04, 3.2425e-04,
          3.4380e-04, 1.4293e-04, 3.7742e-04, 1.0147e-03, 8.1253e-04, 4.2877e-02,
          1.2976e-01, 6.4819e-02, 5.4871e-02, 5.0201e-02, 2.5452e-02, 1.1499e-01,
          1.8994e-01, 3.9244e-04, 3.9244e-04, 3.9244e-04, 3.9244e-04, 3.9244e-04,
          3.9244e-04, 3.9244e-04], dtype=torch.float16),
  tensor([0.0000e+00, 7.3128e-03, 4.5685e-02, 6.3248e-03, 3.2196e-02, 1.6998e-02,
          3.1494e-02, 2.6855e-02, 2.5773e-04, 1.2481e-04, 4.8184e-04, 1.2112e-03,
          2.6393e-04, 1.5342e-04, 1.2481e-04, 2.1112e-04, 1.5056e-04, 2.1887e-04,
          1.2326e-04, 9.3460e-05, 8.6248e-05, 1.5748e-04, 8.2076e-05, 3.6478e-04,
          1.0424e-03, 3.5715e-04, 1.6327e-02, 3.4733e-03, 2.3484e-04, 1.8883e-04,
          1.4091e-04, 3.0327e-04, 1.0198e-04, 2.1148e-04, 2.0587e-04, 1.1504e-04,
          2.1338e-04, 2.3305e-04, 3.1233e-04, 2.0564e-04, 8.5950e-05, 1.2338e-04,
          1.2004e-04, 8.1718e-05, 1.1706e-04, 5.3525e-05, 1.7285e-04, 1.9991e-04,
          6.8963e-05, 1.7059e-04, 2.8381e-03, 2.6679e-04, 1.9622e-04, 1.2732e-04,
          1.4794e-04, 1.2302e-04, 2.5487e-04, 1.3268e-04, 2.6774e-04, 1.8978e-04,
          2.8920e-04, 1.2970e-04, 2.4843e-04, 1.4865e-04, 5.8830e-05, 1.9586e-04,
          1.6999e-04, 5.8174e-05, 8.2135e-05, 4.8935e-05, 5.1916e-05, 8.0228e-05,
          1.3530e-04, 7.4911e-04, 3.3641e-04, 2.3639e-04, 1.0622e-04, 7.6056e-05,
          1.4222e-04, 1.0759e-04, 9.0301e-05, 1.1927e-04, 1.1909e-04, 1.3256e-04,
          2.2376e-04, 2.5392e-04, 1.4150e-04, 1.9920e-04, 6.4909e-05, 6.4969e-05,
          3.4380e-04, 1.1957e-04, 2.7716e-05, 2.0850e-04, 4.5776e-05, 1.1194e-04,
          1.8239e-04, 1.8120e-04, 1.1915e-04, 2.4867e-04, 9.8467e-05, 7.7665e-05,
          4.0007e-04, 1.1331e-04, 8.8274e-05, 1.4138e-04, 2.1088e-04, 8.4281e-05,
          1.6379e-04, 7.2896e-05, 1.7846e-04, 4.8101e-05, 9.7513e-05, 6.4611e-05,
          1.1736e-04, 5.2333e-05, 6.8188e-05, 1.8775e-04, 1.4269e-04, 2.6107e-04,
          1.3983e-04, 1.3649e-04, 6.2883e-05, 9.6679e-05, 3.4499e-04, 3.0875e-05,
          6.0141e-05, 6.7949e-05, 8.1420e-05, 1.4722e-04, 1.2732e-04, 1.8334e-04,
          2.0039e-04, 9.1612e-05, 1.2410e-04, 2.3901e-05, 8.3745e-05, 3.9041e-05,
          1.9407e-04, 5.2989e-05, 1.7846e-04, 8.3494e-04, 5.4932e-04, 9.5189e-05,
          5.7399e-05, 1.1820e-04, 4.4405e-05, 4.7624e-05, 1.0574e-04, 1.6963e-04,
          2.4414e-04, 1.2708e-04, 8.4758e-05, 1.1432e-04, 2.2578e-04, 8.4341e-05,
          7.0453e-05, 6.1631e-05, 8.6606e-05, 4.3273e-05, 1.0121e-04, 6.8903e-05,
          1.7965e-04, 1.7869e-04, 1.6916e-04, 5.8651e-04, 2.2328e-04, 6.5386e-05,
          2.3115e-04, 7.0512e-05, 7.3195e-05, 9.7394e-05, 9.1970e-05, 1.4091e-04,
          9.6798e-05, 1.5402e-04, 2.5511e-04, 1.6963e-04, 1.1283e-04, 2.5964e-04,
          9.3699e-05, 9.3281e-05, 1.7321e-04, 1.0484e-04, 1.5438e-04, 1.4079e-04,
          2.6202e-04, 7.8082e-05, 1.1911e-03, 9.5129e-05, 7.5877e-05, 1.5843e-04,
          1.3018e-04, 4.2558e-05, 1.0413e-04, 1.7476e-04, 1.6487e-04, 1.4412e-04,
          1.7846e-04, 9.9659e-05, 1.1426e-04, 4.9591e-05, 1.4722e-04, 7.9930e-05,
          5.5492e-05, 5.8949e-05, 4.9949e-05, 1.4865e-04, 2.8133e-04, 5.9783e-05,
          1.3101e-04, 2.6321e-04, 2.2185e-04, 1.2010e-04, 1.2159e-04, 5.5909e-05,
          1.0103e-04, 8.2970e-05, 4.1699e-04, 9.8646e-05, 1.4460e-04, 8.1122e-05,
          2.1696e-04, 7.7307e-05, 6.9618e-05, 7.5400e-05, 3.1734e-04, 1.9479e-04,
          7.0930e-05, 4.7088e-05, 3.3021e-05, 7.5161e-05, 1.1331e-04, 6.0380e-05,
          7.8154e-04, 6.0201e-05, 7.6473e-05, 1.2624e-04, 9.3937e-05, 1.8299e-04,
          1.7834e-04, 2.8563e-04, 8.2970e-05, 1.1504e-04, 9.6917e-05, 8.0466e-05,
          9.0063e-05, 6.2883e-05, 6.1274e-05, 7.0155e-05, 1.1367e-04, 9.8050e-05,
          3.2604e-05, 3.6776e-05, 1.0216e-04, 8.2016e-05, 1.3864e-04, 6.2943e-04,
          1.0270e-04, 7.4089e-05, 6.9737e-05, 6.4492e-05, 6.2704e-05, 1.2720e-04,
          9.0897e-05, 2.4164e-04, 8.2135e-05, 1.2147e-04, 2.1923e-04, 8.0526e-05,
          1.0222e-04, 7.8261e-05, 1.4114e-04, 5.6505e-05, 1.1092e-04, 9.6381e-05,
          5.7876e-05, 5.9843e-05, 8.2254e-05, 9.8348e-05, 3.0446e-04, 8.2195e-05,
          1.3852e-04, 7.9215e-05, 5.9009e-05, 1.1438e-04, 8.6665e-05, 7.6711e-05,
          9.5963e-05, 3.5095e-04, 9.6262e-05, 1.2803e-04, 1.3745e-04, 1.3399e-04,
          8.4579e-05, 4.6909e-05, 4.3571e-05, 9.7156e-05, 7.7188e-05, 8.2493e-05,
          8.4698e-05, 1.0055e-04, 1.3566e-04, 1.2302e-04, 1.2177e-04, 1.1593e-04,
          1.0347e-04, 1.0937e-04, 1.1671e-04, 5.6386e-05, 2.0862e-04, 1.5390e-04,
          3.9756e-05, 8.9169e-05, 1.5140e-04, 2.1005e-04, 3.2616e-04, 1.4353e-04,
          8.6904e-05, 3.5882e-04, 6.0260e-05, 5.0485e-05, 1.0902e-04, 1.3745e-04,
          7.6234e-05, 4.7266e-05, 3.6287e-04, 1.5402e-04, 1.0145e-04, 1.3387e-04,
          3.5214e-04, 3.4690e-05, 1.2851e-04, 5.0688e-04, 5.4550e-04, 1.4567e-04,
          6.8367e-05, 2.2233e-04, 8.6725e-05, 1.1164e-04, 1.4031e-04, 1.1629e-04,
          3.5667e-04, 1.7834e-04, 7.5817e-05, 2.3568e-04, 1.4114e-04, 1.1742e-04,
          6.7353e-05, 2.9488e-03, 1.3626e-04, 1.4329e-04, 1.6057e-04, 1.0598e-04,
          1.2970e-04, 2.2435e-04, 8.6844e-05, 2.2614e-04, 9.1851e-05, 8.6188e-05,
          1.1325e-04, 1.0490e-04, 1.1039e-04, 2.3460e-04, 1.3781e-04, 4.2582e-04,
          1.4222e-04, 7.5221e-05, 3.5071e-04, 1.2875e-04, 1.9741e-03, 5.5552e-04,
          3.0499e-03, 3.1114e-04, 2.2757e-04, 1.7452e-04, 5.6267e-05, 3.1447e-04,
          6.7294e-05, 1.5712e-04, 1.9193e-04, 2.3401e-04, 1.1456e-04, 1.5700e-04,
          5.6505e-05, 1.9395e-04, 6.5446e-05, 1.4544e-04, 1.0246e-04, 1.6654e-04,
          2.1064e-04, 8.0824e-05, 5.9938e-04, 3.0971e-04, 3.9053e-04, 3.9154e-02,
          1.0681e-01, 4.7485e-02, 1.8021e-02, 1.7639e-02, 1.4397e-02, 3.2990e-02,
          1.9324e-01, 2.6514e-01, 8.9931e-04, 1.7710e-03, 8.9931e-04, 8.9931e-04,
          1.4706e-03, 8.9931e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.3580e-02, 5.5786e-02, 1.1185e-02, 2.2980e-02, 1.2444e-02,
          2.9633e-02, 1.8448e-02, 1.2612e-04, 6.8188e-05, 2.8253e-04, 9.2125e-04,
          1.6105e-04, 9.9123e-05, 9.7275e-05, 1.1438e-04, 7.3493e-05, 2.2662e-04,
          8.6129e-05, 5.6684e-05, 7.2718e-05, 1.2946e-04, 6.3300e-05, 2.5630e-04,
          7.4196e-04, 1.8752e-04, 1.1574e-02, 2.1400e-03, 1.3757e-04, 1.4901e-04,
          1.0675e-04, 2.2793e-04, 6.5506e-05, 1.1313e-04, 1.1081e-04, 6.6996e-05,
          1.1355e-04, 1.3268e-04, 1.9622e-04, 1.0139e-04, 6.1035e-05, 9.1612e-05,
          9.5844e-05, 6.1333e-05, 9.1076e-05, 3.9399e-05, 1.2290e-04, 1.2445e-04,
          4.9949e-05, 1.3709e-04, 1.7967e-03, 1.3375e-04, 1.1498e-04, 7.3195e-05,
          7.6354e-05, 7.8499e-05, 1.5020e-04, 9.1493e-05, 1.3793e-04, 1.1092e-04,
          1.5140e-04, 8.0645e-05, 1.3423e-04, 8.3208e-05, 4.1187e-05, 1.1653e-04,
          1.3638e-04, 4.4823e-05, 6.5446e-05, 3.6180e-05, 3.8445e-05, 6.5088e-05,
          9.6083e-05, 3.6669e-04, 1.8263e-04, 1.3602e-04, 5.5492e-05, 4.5240e-05,
          8.6844e-05, 7.8321e-05, 7.0274e-05, 6.9857e-05, 8.1897e-05, 6.3717e-05,
          1.1843e-04, 1.4603e-04, 7.6652e-05, 1.2863e-04, 4.4465e-05, 5.4061e-05,
          2.5439e-04, 1.0657e-04, 2.3305e-05, 1.4794e-04, 3.2961e-05, 8.4221e-05,
          1.1075e-04, 1.3125e-04, 7.3314e-05, 1.4746e-04, 7.8559e-05, 5.9724e-05,
          2.8539e-04, 9.3699e-05, 6.0260e-05, 9.1314e-05, 1.0955e-04, 5.1737e-05,
          8.5175e-05, 5.0247e-05, 1.4973e-04, 3.4869e-05, 7.7546e-05, 4.3631e-05,
          6.6221e-05, 3.9697e-05, 4.3392e-05, 1.1861e-04, 1.1265e-04, 1.4389e-04,
          8.7082e-05, 8.0109e-05, 4.6134e-05, 5.6922e-05, 1.6797e-04, 2.4438e-05,
          4.2379e-05, 5.1498e-05, 5.4479e-05, 8.6069e-05, 6.2823e-05, 9.6262e-05,
          1.5199e-04, 5.9962e-05, 6.0558e-05, 2.2590e-05, 6.7770e-05, 3.0994e-05,
          1.1814e-04, 3.7909e-05, 1.2970e-04, 5.9652e-04, 3.3236e-04, 6.3479e-05,
          4.2796e-05, 6.5088e-05, 3.2485e-05, 3.7432e-05, 7.8261e-05, 9.2447e-05,
          1.4579e-04, 8.3029e-05, 5.2512e-05, 6.5207e-05, 1.3578e-04, 6.4194e-05,
          6.2168e-05, 5.2333e-05, 3.7551e-05, 3.9577e-05, 7.0274e-05, 4.4346e-05,
          1.2195e-04, 1.7643e-04, 1.1992e-04, 2.9683e-04, 1.5867e-04, 5.0724e-05,
          1.8334e-04, 5.8472e-05, 4.4942e-05, 5.2810e-05, 6.7532e-05, 1.1855e-04,
          4.6313e-05, 5.8293e-05, 1.5187e-04, 1.2279e-04, 6.8545e-05, 1.3185e-04,
          6.9082e-05, 4.3988e-05, 9.4593e-05, 6.8605e-05, 8.9824e-05, 9.9361e-05,
          1.1414e-04, 4.4525e-05, 4.8876e-04, 6.9737e-05, 5.2333e-05, 9.0003e-05,
          8.6129e-05, 3.1650e-05, 7.7069e-05, 1.3494e-04, 1.1641e-04, 1.0532e-04,
          1.1200e-04, 7.0155e-05, 6.3241e-05, 3.8326e-05, 1.5473e-04, 6.7592e-05,
          3.9279e-05, 4.8816e-05, 3.2723e-05, 8.1122e-05, 2.0528e-04, 3.2723e-05,
          7.2420e-05, 1.8167e-04, 1.5450e-04, 6.7294e-05, 7.6115e-05, 4.6849e-05,
          6.1750e-05, 5.6982e-05, 2.4259e-04, 5.6386e-05, 1.0109e-04, 4.5240e-05,
          1.2082e-04, 5.1618e-05, 4.6313e-05, 7.2002e-05, 1.6248e-04, 1.0276e-04,
          5.7936e-05, 3.0875e-05, 2.2709e-05, 3.5167e-05, 5.9724e-05, 3.5644e-05,
          4.4656e-04, 5.5730e-05, 4.8220e-05, 5.4955e-05, 5.7518e-05, 1.0407e-04,
          1.0031e-04, 1.5521e-04, 4.6492e-05, 6.5207e-05, 5.7697e-05, 5.3763e-05,
          6.3896e-05, 3.5882e-05, 3.4928e-05, 3.3617e-05, 5.6744e-05, 5.9545e-05,
          2.3365e-05, 2.7895e-05, 7.7188e-05, 6.8307e-05, 8.6904e-05, 4.9686e-04,
          7.3254e-05, 6.5327e-05, 5.3704e-05, 3.8743e-05, 3.5524e-05, 7.5519e-05,
          5.1796e-05, 1.1474e-04, 7.0393e-05, 7.6354e-05, 1.4591e-04, 4.8518e-05,
          6.1274e-05, 6.0499e-05, 7.3433e-05, 3.7611e-05, 9.7215e-05, 5.6684e-05,
          3.3021e-05, 2.9504e-05, 5.3346e-05, 6.9797e-05, 1.9908e-04, 6.2525e-05,
          1.1456e-04, 5.5730e-05, 4.2856e-05, 7.6056e-05, 4.5359e-05, 5.6267e-05,
          5.5134e-05, 2.3532e-04, 5.7817e-05, 9.1732e-05, 8.2552e-05, 7.0333e-05,
          5.9247e-05, 3.0875e-05, 3.1471e-05, 6.3956e-05, 6.0618e-05, 5.6326e-05,
          6.0380e-05, 6.4135e-05, 7.6473e-05, 7.6950e-05, 8.3029e-05, 8.8036e-05,
          6.5327e-05, 5.9307e-05, 7.8559e-05, 3.7372e-05, 9.8348e-05, 8.0824e-05,
          2.6405e-05, 4.8637e-05, 8.9467e-05, 8.3208e-05, 2.4343e-04, 7.6890e-05,
          5.4896e-05, 2.1756e-04, 3.6597e-05, 2.9564e-05, 8.2612e-05, 9.4771e-05,
          6.2883e-05, 2.5094e-05, 2.9469e-04, 1.0669e-04, 6.6578e-05, 9.2685e-05,
          3.0756e-04, 1.8954e-05, 9.1553e-05, 4.2605e-04, 2.8038e-04, 1.0622e-04,
          3.1829e-05, 1.7393e-04, 5.1618e-05, 6.3837e-05, 1.4067e-04, 8.7798e-05,
          2.3830e-04, 9.4950e-05, 5.3108e-05, 9.6917e-05, 8.7678e-05, 8.8513e-05,
          4.7803e-05, 2.1191e-03, 7.5459e-05, 8.8692e-05, 8.8036e-05, 6.1333e-05,
          9.7215e-05, 1.1140e-04, 4.8399e-05, 1.4281e-04, 6.2227e-05, 5.7101e-05,
          6.2943e-05, 6.1274e-05, 1.0294e-04, 1.7524e-04, 8.1658e-05, 3.3331e-04,
          8.2195e-05, 5.6922e-05, 2.1458e-04, 1.1367e-04, 1.6108e-03, 3.0494e-04,
          2.5902e-03, 1.6272e-04, 1.2660e-04, 7.9811e-05, 3.8385e-05, 1.7965e-04,
          3.9101e-05, 1.0735e-04, 1.2779e-04, 1.4889e-04, 7.1585e-05, 1.1629e-04,
          3.0875e-05, 1.2094e-04, 5.1141e-05, 8.6784e-05, 5.9783e-05, 8.5354e-05,
          1.7190e-04, 6.5446e-05, 3.0661e-04, 2.8205e-04, 3.8743e-04, 4.4189e-02,
          7.5378e-02, 3.7811e-02, 1.5869e-02, 1.6891e-02, 1.0307e-02, 2.5665e-02,
          1.3708e-01, 2.2522e-01, 1.8298e-01, 6.7329e-04, 6.7329e-04, 1.0147e-03,
          6.7329e-04, 6.7329e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.4526e-02, 5.4932e-02, 2.3911e-02, 3.2196e-02, 3.0975e-02,
          3.5339e-02, 3.2196e-02, 1.7250e-04, 1.1134e-04, 2.9302e-04, 6.5804e-04,
          2.0969e-04, 1.4174e-04, 1.8668e-04, 1.5414e-04, 1.2350e-04, 2.3472e-04,
          1.3387e-04, 7.6771e-05, 1.9169e-04, 2.6679e-04, 1.7536e-04, 2.6321e-04,
          6.6519e-04, 2.7204e-04, 1.9852e-02, 1.9703e-03, 3.9411e-04, 1.8227e-04,
          1.7715e-04, 2.5105e-04, 1.2106e-04, 2.0885e-04, 1.3995e-04, 1.2577e-04,
          3.3641e-04, 5.9748e-04, 2.7156e-04, 1.5414e-04, 1.6248e-04, 6.2656e-04,
          9.8515e-04, 4.2224e-04, 5.8460e-04, 2.3496e-04, 1.5855e-04, 1.9169e-04,
          1.0818e-04, 3.1519e-04, 4.7989e-03, 3.9816e-04, 2.1565e-04, 1.3697e-04,
          3.3164e-04, 1.6701e-04, 4.3631e-04, 6.2323e-04, 3.0184e-04, 5.9175e-04,
          4.8828e-04, 2.2566e-04, 5.5027e-04, 2.1553e-04, 2.3317e-04, 1.7512e-04,
          2.3639e-04, 7.9691e-05, 9.8526e-05, 3.0851e-04, 2.6989e-04, 2.6655e-04,
          4.1389e-04, 4.6158e-04, 7.3528e-04, 2.8777e-04, 1.3685e-04, 1.2600e-04,
          2.0957e-04, 5.6887e-04, 2.7275e-04, 1.7822e-04, 2.4128e-04, 2.1577e-04,
          3.2663e-04, 3.0994e-04, 3.9387e-04, 1.3399e-04, 9.3579e-05, 1.5092e-04,
          2.8157e-04, 2.1994e-04, 8.5473e-05, 2.0969e-04, 2.2721e-04, 4.3011e-04,
          2.9683e-04, 3.4881e-04, 1.5104e-04, 3.1519e-04, 7.6580e-04, 3.1400e-04,
          3.3689e-04, 1.1706e-04, 1.1098e-04, 2.6631e-04, 1.7071e-04, 1.6224e-04,
          2.1684e-04, 7.2575e-04, 4.0936e-04, 1.7142e-04, 1.2863e-04, 8.9049e-05,
          1.0359e-04, 2.2852e-04, 1.7202e-04, 1.8620e-04, 4.5228e-04, 3.7408e-04,
          2.5272e-04, 1.9813e-04, 2.5415e-04, 7.7665e-05, 2.1684e-04, 9.0480e-05,
          2.7919e-04, 1.5688e-04, 1.5926e-04, 1.9503e-04, 1.1718e-04, 2.4033e-04,
          1.5163e-04, 8.1420e-05, 5.9187e-05, 4.1902e-05, 1.4448e-04, 1.3244e-04,
          1.6212e-04, 1.0526e-04, 1.4019e-04, 5.9032e-04, 2.9492e-04, 1.4734e-04,
          2.4867e-04, 6.4969e-05, 8.1480e-05, 1.4985e-04, 2.0230e-04, 2.1791e-04,
          2.7895e-04, 2.0230e-04, 2.5415e-04, 2.6178e-04, 3.7861e-04, 4.9686e-04,
          1.5378e-04, 2.1267e-04, 6.4552e-05, 1.5032e-04, 1.7250e-04, 1.1623e-04,
          1.6010e-04, 1.8466e-04, 1.3685e-04, 2.2209e-04, 1.0242e-03, 1.7858e-04,
          2.4021e-04, 1.3208e-04, 1.0455e-04, 1.0151e-04, 8.8930e-05, 1.6880e-04,
          1.5819e-04, 2.0278e-04, 5.6362e-04, 1.0262e-03, 1.5712e-04, 1.3518e-04,
          8.1539e-05, 9.5487e-05, 1.3602e-04, 9.5844e-05, 9.0957e-05, 9.7156e-05,
          1.8585e-04, 1.2863e-04, 2.7108e-04, 5.0926e-04, 1.4758e-04, 1.5092e-04,
          1.3351e-04, 1.4818e-04, 1.0097e-04, 1.5700e-04, 1.3185e-04, 1.6296e-04,
          3.3474e-04, 2.0409e-04, 1.3769e-04, 1.0496e-04, 8.3745e-05, 8.1539e-05,
          1.3506e-04, 1.5044e-04, 1.7405e-04, 1.1444e-04, 2.8753e-04, 6.2108e-05,
          1.1623e-04, 7.1573e-04, 1.1377e-03, 1.4353e-04, 1.5604e-04, 1.3435e-04,
          1.7452e-04, 1.1581e-04, 2.3925e-04, 7.4327e-05, 1.4961e-04, 1.3673e-04,
          2.4438e-04, 3.1281e-04, 1.5485e-04, 8.2970e-05, 2.0885e-04, 1.4842e-04,
          9.1553e-05, 9.2983e-05, 1.0604e-04, 1.0633e-04, 1.5366e-04, 1.0610e-04,
          1.9970e-03, 1.6844e-04, 1.3030e-04, 9.2089e-05, 1.7738e-04, 2.5964e-04,
          3.9053e-04, 5.9605e-04, 2.0278e-04, 2.2626e-04, 1.6391e-04, 2.1017e-04,
          4.8089e-04, 9.5189e-05, 1.6952e-04, 1.0782e-04, 1.9622e-04, 1.2183e-04,
          7.4744e-05, 1.1587e-04, 3.0255e-04, 2.0444e-04, 1.2672e-04, 5.2309e-04,
          4.4322e-04, 1.2767e-04, 9.9659e-05, 8.1658e-05, 1.6928e-04, 2.5654e-04,
          1.2720e-04, 1.2290e-04, 1.0133e-04, 1.7810e-04, 2.2256e-04, 1.1319e-04,
          1.4889e-04, 1.1629e-04, 1.5497e-04, 1.4675e-04, 1.7869e-04, 1.3125e-04,
          1.3101e-04, 8.0466e-05, 2.6035e-04, 1.3459e-04, 9.0265e-04, 1.5283e-04,
          1.2219e-04, 2.1303e-04, 2.1732e-04, 2.4939e-04, 1.1992e-04, 1.1176e-04,
          1.1957e-04, 2.4319e-04, 8.8036e-05, 1.2529e-04, 3.2043e-04, 2.4354e-04,
          2.3663e-04, 1.2100e-04, 1.2147e-04, 9.2030e-05, 1.4818e-04, 2.1613e-04,
          2.5082e-04, 1.0496e-04, 1.3196e-04, 1.7214e-04, 4.1962e-04, 1.2064e-04,
          1.2219e-04, 1.8799e-04, 1.8704e-04, 7.3075e-05, 9.3997e-05, 1.0985e-04,
          5.6863e-05, 1.3256e-04, 4.3678e-04, 1.5604e-04, 2.7156e-04, 1.5366e-04,
          8.9467e-05, 2.8205e-04, 8.0109e-05, 6.7711e-05, 2.7013e-04, 3.4642e-04,
          7.3135e-05, 8.4281e-05, 2.7394e-04, 7.3910e-04, 1.9109e-04, 1.9205e-04,
          4.4870e-04, 6.3419e-05, 2.1768e-04, 3.9411e-04, 1.9133e-04, 2.6155e-04,
          9.0897e-05, 1.8787e-04, 9.4831e-05, 3.2115e-04, 2.3568e-04, 1.8525e-04,
          1.5247e-04, 1.1146e-04, 9.8944e-05, 9.9659e-05, 1.6272e-04, 1.7095e-04,
          1.4198e-04, 6.0959e-03, 4.0007e-04, 4.1366e-04, 1.9026e-04, 1.3196e-04,
          1.9407e-04, 1.0842e-04, 6.5625e-05, 9.9063e-05, 1.3220e-04, 1.1325e-04,
          1.4281e-04, 1.3220e-04, 1.4102e-04, 1.8048e-04, 1.1081e-04, 2.7895e-04,
          9.7871e-05, 1.4889e-04, 2.7823e-04, 1.5986e-04, 1.1349e-03, 2.2399e-04,
          7.4158e-03, 7.2050e-04, 5.1451e-04, 1.0288e-04, 8.7202e-05, 8.4543e-04,
          1.0699e-04, 1.9932e-04, 1.6463e-04, 1.6153e-04, 2.0444e-04, 9.1076e-05,
          6.9737e-05, 4.2057e-04, 9.4771e-05, 9.0241e-05, 9.6381e-05, 1.2326e-04,
          2.1744e-04, 1.0484e-04, 2.0254e-04, 4.7708e-04, 4.6539e-04, 4.8889e-02,
          4.4769e-02, 3.0930e-02, 1.6937e-02, 1.3176e-02, 8.9951e-03, 2.2949e-02,
          8.0261e-02, 7.1472e-02, 1.2256e-01, 1.8323e-01, 8.1921e-04, 9.3794e-04,
          8.1921e-04, 8.1921e-04], dtype=torch.float16),
  tensor([0.0000e+00, 1.4763e-02, 5.9448e-02, 2.9602e-02, 2.6627e-02, 1.8951e-02,
          3.4790e-02, 2.7100e-02, 8.7500e-05, 6.5386e-05, 1.9932e-04, 3.7003e-04,
          1.2445e-04, 7.1943e-05, 1.0967e-04, 1.0794e-04, 6.7055e-05, 1.2720e-04,
          8.5652e-05, 5.8472e-05, 1.0914e-04, 1.9562e-04, 1.0341e-04, 2.3854e-04,
          3.4976e-04, 1.8620e-04, 1.6556e-02, 2.5387e-03, 2.1899e-04, 1.1593e-04,
          1.1891e-04, 1.2052e-04, 8.7976e-05, 1.4305e-04, 1.2994e-04, 6.8963e-05,
          2.0897e-04, 2.1100e-04, 2.2745e-04, 1.2624e-04, 9.6202e-05, 1.6677e-04,
          1.6677e-04, 1.0651e-04, 1.4675e-04, 7.2837e-05, 8.7023e-05, 1.3781e-04,
          7.0155e-05, 2.1017e-04, 2.9564e-03, 1.6320e-04, 1.0067e-04, 8.4400e-05,
          9.0122e-05, 8.0585e-05, 1.7631e-04, 1.5819e-04, 1.4985e-04, 1.6427e-04,
          1.9038e-04, 1.3185e-04, 1.6415e-04, 1.2636e-04, 6.5863e-05, 1.0031e-04,
          8.8871e-05, 3.9935e-05, 7.3195e-05, 6.6936e-05, 6.5088e-05, 8.9765e-05,
          1.5771e-04, 3.9983e-04, 2.2590e-04, 1.3041e-04, 7.0333e-05, 5.7042e-05,
          1.0258e-04, 1.4126e-04, 8.0168e-05, 1.0782e-04, 9.6560e-05, 8.5175e-05,
          1.5128e-04, 1.4043e-04, 1.0294e-04, 7.5102e-05, 4.4882e-05, 5.7399e-05,
          1.4782e-04, 9.5963e-05, 3.5226e-05, 1.1617e-04, 5.2512e-05, 1.3876e-04,
          1.7762e-04, 1.5473e-04, 8.3685e-05, 1.6189e-04, 1.3578e-04, 1.0031e-04,
          1.7142e-04, 6.0678e-05, 6.9857e-05, 1.4985e-04, 9.0063e-05, 1.0002e-04,
          1.1164e-04, 1.0377e-04, 1.5497e-04, 4.9055e-05, 5.8174e-05, 4.6611e-05,
          5.4002e-05, 7.3254e-05, 8.1539e-05, 1.1098e-04, 1.7643e-04, 2.2185e-04,
          1.5044e-04, 1.1736e-04, 7.3075e-05, 5.0843e-05, 1.3089e-04, 3.9399e-05,
          8.4579e-05, 6.4313e-05, 7.1108e-05, 1.1516e-04, 8.0168e-05, 1.2589e-04,
          9.2268e-05, 5.2512e-05, 5.2571e-05, 2.8729e-05, 6.8784e-05, 4.7028e-05,
          8.9943e-05, 6.6102e-05, 9.5487e-05, 7.6818e-04, 3.3069e-04, 1.0705e-04,
          7.0214e-05, 5.4300e-05, 4.6551e-05, 5.2631e-05, 8.0705e-05, 1.2362e-04,
          1.5497e-04, 1.2016e-04, 8.6844e-05, 9.2030e-05, 1.6904e-04, 1.3638e-04,
          7.8261e-05, 8.4460e-05, 4.5598e-05, 4.4465e-05, 1.1206e-04, 6.9976e-05,
          1.0890e-04, 1.5903e-04, 8.3447e-05, 2.3913e-04, 2.8586e-04, 8.0764e-05,
          1.2130e-04, 8.1062e-05, 6.9201e-05, 6.6698e-05, 7.1883e-05, 1.3053e-04,
          6.6102e-05, 6.7651e-05, 2.3198e-04, 2.2447e-04, 1.1301e-04, 9.8348e-05,
          7.2300e-05, 7.2956e-05, 9.9897e-05, 5.4419e-05, 6.7055e-05, 7.8678e-05,
          1.1325e-04, 7.4267e-05, 3.0327e-04, 1.2189e-04, 6.8545e-05, 1.1384e-04,
          6.7413e-05, 5.6624e-05, 8.1003e-05, 9.2506e-05, 8.9705e-05, 1.2231e-04,
          1.4842e-04, 1.3840e-04, 1.0788e-04, 5.6922e-05, 5.9903e-05, 6.9141e-05,
          5.6982e-05, 7.0870e-05, 5.5969e-05, 7.8619e-05, 1.5414e-04, 4.5061e-05,
          7.8797e-05, 3.2449e-04, 2.4676e-04, 8.5771e-05, 1.1253e-04, 5.8949e-05,
          1.0854e-04, 7.7546e-05, 2.3961e-04, 5.7876e-05, 1.1861e-04, 7.1645e-05,
          1.3661e-04, 8.2910e-05, 7.7486e-05, 5.1498e-05, 1.3161e-04, 1.2016e-04,
          5.0545e-05, 4.1306e-05, 3.1650e-05, 4.3392e-05, 8.4162e-05, 5.8770e-05,
          6.6233e-04, 7.4029e-05, 5.8293e-05, 5.9247e-05, 9.1136e-05, 1.0967e-04,
          1.2791e-04, 2.1577e-04, 7.7903e-05, 1.1516e-04, 7.9036e-05, 1.1009e-04,
          1.3411e-04, 4.8697e-05, 4.3988e-05, 3.9637e-05, 6.6221e-05, 8.4460e-05,
          3.5882e-05, 4.5955e-05, 1.2004e-04, 8.8394e-05, 9.8169e-05, 4.8065e-04,
          1.2732e-04, 5.8234e-05, 5.8234e-05, 5.5790e-05, 5.5075e-05, 9.0480e-05,
          8.5056e-05, 1.2529e-04, 7.7784e-05, 8.4221e-05, 1.5152e-04, 7.4804e-05,
          7.4148e-05, 5.9605e-05, 9.2328e-05, 6.0618e-05, 9.8526e-05, 7.1645e-05,
          4.9949e-05, 4.1544e-05, 9.7811e-05, 1.0097e-04, 3.2163e-04, 8.2552e-05,
          6.8188e-05, 7.4148e-05, 6.9439e-05, 1.1319e-04, 7.3254e-05, 6.7949e-05,
          6.2823e-05, 2.4164e-04, 6.7830e-05, 7.0930e-05, 1.2255e-04, 9.1255e-05,
          1.0890e-04, 5.2094e-05, 5.6028e-05, 6.8367e-05, 7.1466e-05, 7.8499e-05,
          9.1791e-05, 7.7367e-05, 8.3327e-05, 1.0449e-04, 1.2457e-04, 7.2002e-05,
          9.3579e-05, 1.0508e-04, 1.1402e-04, 4.5002e-05, 8.8334e-05, 8.5652e-05,
          3.8028e-05, 6.9499e-05, 1.7786e-04, 1.1897e-04, 2.5225e-04, 1.2171e-04,
          5.4836e-05, 1.7345e-04, 5.5015e-05, 4.9412e-05, 1.2624e-04, 1.5819e-04,
          5.0843e-05, 4.1604e-05, 2.1946e-04, 1.6582e-04, 9.5487e-05, 1.4699e-04,
          2.9469e-04, 3.6776e-05, 1.4877e-04, 3.2234e-04, 1.5664e-04, 1.4591e-04,
          4.7266e-05, 1.3137e-04, 5.1856e-05, 1.3483e-04, 1.3840e-04, 1.3709e-04,
          1.1563e-04, 8.5592e-05, 6.3121e-05, 7.2539e-05, 9.0063e-05, 1.2338e-04,
          8.3208e-05, 3.2597e-03, 1.1790e-04, 1.1414e-04, 9.7573e-05, 8.4877e-05,
          1.0270e-04, 8.6367e-05, 5.6744e-05, 9.6858e-05, 1.0186e-04, 7.6115e-05,
          7.4208e-05, 7.9811e-05, 7.4029e-05, 1.5259e-04, 8.4519e-05, 3.5858e-04,
          7.7367e-05, 1.0139e-04, 2.5678e-04, 8.5890e-05, 1.6527e-03, 3.0518e-04,
          3.9902e-03, 2.2936e-04, 1.9729e-04, 7.9095e-05, 5.4061e-05, 3.1757e-04,
          6.7830e-05, 1.5795e-04, 1.1963e-04, 1.4210e-04, 1.1551e-04, 9.5248e-05,
          4.3154e-05, 2.1434e-04, 7.9036e-05, 7.1466e-05, 7.4863e-05, 8.3625e-05,
          2.1017e-04, 6.9737e-05, 2.8801e-04, 3.5715e-04, 4.1318e-04, 3.9124e-02,
          6.5186e-02, 3.4119e-02, 1.3985e-02, 1.4465e-02, 8.1253e-03, 2.2110e-02,
          8.4778e-02, 7.8552e-02, 1.0358e-01, 1.1627e-01, 1.3062e-01, 8.8406e-04,
          8.8406e-04, 8.8406e-04], dtype=torch.float16),
  tensor([0.0000e+00, 2.0809e-03, 3.9886e-02, 3.2444e-03, 7.3624e-03, 7.8735e-03,
          3.1982e-02, 6.2943e-03, 3.4571e-04, 1.1998e-04, 1.0815e-03, 2.5883e-03,
          5.9652e-04, 3.1781e-04, 2.5892e-04, 5.4550e-04, 2.1529e-04, 4.6825e-04,
          2.3377e-04, 1.1170e-04, 1.0127e-04, 1.0759e-04, 1.0937e-04, 3.7599e-04,
          2.2411e-03, 9.1743e-04, 4.3755e-03, 1.3590e-03, 2.0373e-04, 3.6120e-04,
          3.7622e-04, 6.4468e-04, 1.3888e-04, 8.4257e-04, 4.6778e-04, 3.1018e-04,
          1.7176e-03, 7.8344e-04, 6.6662e-04, 5.9462e-04, 1.6296e-04, 8.5545e-04,
          4.0913e-04, 3.4404e-04, 3.1805e-04, 2.1791e-04, 3.1400e-04, 8.1778e-04,
          1.2827e-04, 1.5187e-04, 8.9359e-04, 6.4468e-04, 5.1165e-04, 3.1018e-04,
          2.1887e-04, 3.0518e-04, 2.1672e-04, 1.4229e-03, 3.2997e-03, 3.1204e-03,
          1.4315e-03, 2.6870e-04, 3.1638e-04, 4.1509e-04, 6.1989e-04, 1.9388e-03,
          2.1896e-03, 8.0252e-04, 4.8780e-04, 1.9288e-04, 1.5962e-04, 3.2711e-04,
          2.5678e-04, 1.6222e-03, 2.9135e-04, 6.8283e-04, 1.7118e-04, 1.8919e-04,
          4.1318e-04, 1.4029e-03, 1.9512e-03, 2.7065e-03, 2.9240e-03, 9.5415e-04,
          4.7088e-04, 7.6866e-04, 2.3401e-04, 7.3290e-04, 1.1673e-03, 1.0490e-03,
          1.4391e-03, 1.1644e-03, 3.2210e-04, 9.4795e-04, 1.9622e-04, 2.3651e-04,
          1.9193e-04, 1.5509e-04, 5.3787e-04, 1.8740e-03, 1.5907e-03, 1.6251e-03,
          2.0905e-03, 1.0519e-03, 1.3762e-03, 4.9019e-03, 2.7409e-03, 1.7762e-04,
          5.5265e-04, 5.0163e-04, 1.2856e-03, 5.4598e-04, 1.2712e-03, 7.3814e-04,
          1.1368e-03, 3.7932e-04, 8.2076e-05, 1.4982e-03, 2.5773e-04, 1.8167e-04,
          2.9612e-04, 2.4128e-03, 4.7660e-04, 2.1815e-04, 2.1725e-03, 9.1600e-04,
          7.7486e-04, 2.3460e-03, 2.1172e-03, 1.6508e-03, 3.1638e-04, 6.6423e-04,
          1.4133e-03, 7.4148e-04, 1.8716e-04, 4.0221e-04, 9.4700e-04, 2.7061e-04,
          5.4598e-04, 1.4520e-04, 2.6989e-04, 6.9857e-04, 7.0620e-04, 2.9778e-04,
          8.6212e-04, 2.4557e-04, 9.3317e-04, 1.6031e-03, 2.1038e-03, 4.6158e-03,
          7.0801e-03, 2.0466e-03, 1.0853e-03, 5.7840e-04, 5.1069e-04, 1.0509e-03,
          1.0176e-03, 1.1129e-03, 1.7464e-04, 8.9264e-04, 1.1578e-03, 2.1398e-04,
          7.4434e-04, 9.5558e-04, 3.2783e-04, 7.6008e-04, 1.1978e-03, 1.7862e-03,
          1.0309e-03, 1.3123e-03, 1.2569e-03, 4.1652e-04, 2.6345e-04, 1.1435e-03,
          1.0357e-03, 9.8610e-04, 9.8991e-04, 6.5327e-04, 1.1721e-03, 4.2558e-04,
          6.2084e-04, 2.7657e-04, 1.0681e-03, 3.2043e-04, 2.2984e-04, 2.8634e-04,
          5.3501e-04, 1.9526e-04, 5.6696e-04, 1.0166e-03, 1.4725e-03, 6.3372e-04,
          3.3879e-04, 4.5896e-04, 5.5790e-04, 3.5620e-04, 9.4223e-04, 6.4898e-04,
          1.0843e-03, 1.5259e-04, 7.4005e-04, 8.0776e-04, 8.7500e-04, 6.6853e-04,
          5.7268e-04, 1.3771e-03, 2.2173e-04, 2.1791e-04, 9.0027e-04, 1.1635e-04,
          2.6608e-04, 7.8106e-04, 1.3037e-03, 1.3962e-03, 1.3561e-03, 1.4486e-03,
          1.3924e-03, 4.2105e-04, 7.7057e-04, 2.6917e-04, 8.3876e-04, 3.0994e-04,
          9.5892e-04, 4.3488e-04, 1.1158e-03, 1.0738e-03, 1.6451e-03, 4.2701e-04,
          7.4816e-04, 5.0688e-04, 2.0003e-04, 7.1621e-04, 1.0386e-03, 1.2231e-04,
          1.4906e-03, 1.7509e-03, 1.5001e-03, 5.6553e-04, 1.6613e-03, 1.8520e-03,
          1.3485e-03, 1.0767e-03, 1.3447e-03, 8.2588e-04, 5.4359e-04, 2.7895e-04,
          5.2643e-04, 6.0225e-04, 2.9016e-04, 4.2653e-04, 3.4404e-04, 1.2550e-03,
          2.8443e-04, 1.2326e-04, 2.4295e-04, 4.6825e-04, 4.0650e-04, 1.2455e-03,
          1.1597e-03, 1.1835e-03, 1.2197e-03, 7.9918e-04, 4.3583e-04, 1.6623e-03,
          6.8951e-04, 3.6716e-04, 2.6703e-04, 5.3310e-04, 9.6798e-04, 6.1846e-04,
          1.1215e-03, 1.0519e-03, 1.1387e-03, 5.8222e-04, 1.0242e-03, 1.3285e-03,
          4.8161e-04, 3.7503e-04, 1.1104e-04, 2.6059e-04, 5.8985e-04, 1.9121e-03,
          1.7242e-03, 5.7697e-04, 6.0272e-04, 5.0640e-04, 8.2254e-04, 7.5769e-04,
          1.0176e-03, 1.1005e-03, 2.4128e-04, 3.0303e-04, 3.6716e-04, 6.9237e-04,
          4.7517e-04, 3.1972e-04, 2.8634e-04, 5.1641e-04, 1.2760e-03, 2.6393e-04,
          1.9920e-04, 1.8167e-04, 3.4237e-04, 4.6277e-04, 9.9564e-04, 1.3905e-03,
          6.2132e-04, 1.7321e-04, 9.9945e-04, 1.3614e-04, 3.6550e-04, 2.5582e-04,
          1.5020e-04, 5.7364e-04, 3.3283e-04, 4.6062e-04, 7.9775e-04, 7.2145e-04,
          2.8563e-04, 1.1644e-03, 2.8610e-04, 1.3161e-04, 2.2745e-04, 2.9492e-04,
          1.8299e-04, 1.1766e-04, 5.5695e-04, 1.1663e-03, 1.2913e-03, 4.5371e-04,
          1.7061e-03, 7.1645e-05, 3.4761e-04, 1.3590e-03, 6.1274e-04, 3.8457e-04,
          4.3201e-04, 7.6962e-04, 3.6693e-04, 2.3043e-04, 1.4133e-03, 2.9206e-04,
          6.0749e-04, 5.3644e-04, 1.8644e-04, 3.1757e-04, 6.3562e-04, 2.2948e-04,
          1.1230e-04, 9.5367e-04, 9.9468e-04, 8.9455e-04, 5.4455e-04, 2.8706e-04,
          6.4373e-04, 3.3712e-04, 4.2224e-04, 2.7466e-04, 1.7750e-04, 1.3027e-03,
          2.8515e-04, 2.4724e-04, 7.4625e-04, 4.6778e-04, 2.3007e-04, 1.0147e-03,
          3.0613e-04, 1.4842e-04, 3.9458e-04, 9.3079e-04, 2.1515e-03, 6.4659e-04,
          1.0471e-03, 8.9931e-04, 9.7466e-04, 3.2806e-04, 1.8692e-04, 3.0565e-04,
          1.8871e-04, 4.5514e-04, 7.1955e-04, 4.5180e-04, 2.7871e-04, 3.1233e-04,
          1.6606e-04, 2.2089e-04, 1.5974e-04, 2.5058e-04, 3.5048e-04, 1.3275e-03,
          3.5381e-04, 2.6798e-04, 4.7350e-04, 1.5001e-03, 1.2484e-03, 1.5022e-02,
          3.3997e-02, 3.5339e-02, 5.6396e-02, 3.7048e-02, 1.2215e-02, 6.3843e-02,
          6.6406e-02, 2.0035e-02, 3.8727e-02, 4.3915e-02, 6.4758e-02, 1.0669e-01,
          1.4906e-03, 1.4906e-03], dtype=torch.float16),
  tensor([0.0000e+00, 5.0240e-03, 3.9337e-02, 3.0975e-03, 2.2476e-02, 1.1993e-02,
          2.8824e-02, 2.1622e-02, 1.3351e-04, 8.7857e-05, 3.8505e-04, 8.1444e-04,
          2.2876e-04, 1.0073e-04, 1.1027e-04, 1.6773e-04, 9.9182e-05, 2.0385e-04,
          1.5664e-04, 5.9664e-05, 7.7128e-05, 1.3769e-04, 7.9572e-05, 3.6263e-04,
          7.3576e-04, 2.3484e-04, 1.3046e-02, 2.8419e-03, 1.9693e-04, 2.2495e-04,
          1.4389e-04, 2.1327e-04, 1.4520e-04, 2.6870e-04, 2.5439e-04, 1.0270e-04,
          2.1100e-04, 1.7262e-04, 3.9315e-04, 2.3401e-04, 8.0645e-05, 1.6606e-04,
          1.2231e-04, 7.8559e-05, 1.0014e-04, 5.3823e-05, 1.6510e-04, 2.4629e-04,
          7.5638e-05, 1.5748e-04, 1.4963e-03, 2.0969e-04, 1.8537e-04, 1.1307e-04,
          6.8784e-05, 1.1593e-04, 1.3256e-04, 1.1659e-04, 1.7333e-04, 1.4889e-04,
          1.6785e-04, 1.2922e-04, 1.4281e-04, 1.2684e-04, 7.4446e-05, 2.1088e-04,
          1.2815e-04, 6.8963e-05, 1.0747e-04, 4.7147e-05, 4.7445e-05, 6.6400e-05,
          9.8825e-05, 7.4148e-04, 1.9217e-04, 2.3389e-04, 5.4955e-05, 5.0843e-05,
          1.1462e-04, 1.0413e-04, 9.7036e-05, 1.4448e-04, 1.2326e-04, 8.7380e-05,
          1.2815e-04, 2.2614e-04, 9.0480e-05, 1.6379e-04, 5.8770e-05, 6.1154e-05,
          3.5143e-04, 1.9097e-04, 2.9087e-05, 2.4188e-04, 3.9399e-05, 9.5010e-05,
          1.7357e-04, 1.2267e-04, 1.1390e-04, 2.6441e-04, 1.1450e-04, 1.1259e-04,
          4.4322e-04, 8.6963e-05, 1.2171e-04, 1.5306e-04, 1.6677e-04, 8.2314e-05,
          1.5378e-04, 8.7261e-05, 2.9421e-04, 4.0710e-05, 6.9618e-05, 6.3181e-05,
          8.4460e-05, 5.8651e-05, 5.9545e-05, 2.1672e-04, 1.1432e-04, 2.0766e-04,
          1.3793e-04, 1.9276e-04, 6.8367e-05, 6.7890e-05, 2.5868e-04, 3.6418e-05,
          6.1989e-05, 6.8128e-05, 7.2539e-05, 1.6189e-04, 9.9421e-05, 1.8144e-04,
          1.1092e-04, 8.4043e-05, 6.8963e-05, 3.4988e-05, 1.3518e-04, 3.7134e-05,
          2.1100e-04, 5.2333e-05, 1.1259e-04, 7.8678e-04, 5.3978e-04, 1.1045e-04,
          6.6578e-05, 5.3704e-05, 5.2989e-05, 6.8367e-05, 9.3877e-05, 1.5354e-04,
          2.8515e-04, 1.3530e-04, 7.8619e-05, 1.3316e-04, 1.4997e-04, 1.1986e-04,
          1.1295e-04, 8.2672e-05, 4.9233e-05, 4.5896e-05, 1.3006e-04, 7.0751e-05,
          2.4557e-04, 2.4843e-04, 2.1029e-04, 3.4952e-04, 2.7251e-04, 1.0926e-04,
          2.8324e-04, 1.3971e-04, 9.0063e-05, 8.3268e-05, 1.0765e-04, 2.1911e-04,
          7.1108e-05, 6.3360e-05, 2.0397e-04, 2.1279e-04, 1.4639e-04, 1.8024e-04,
          1.0127e-04, 9.0837e-05, 1.7750e-04, 9.0420e-05, 8.1360e-05, 9.7871e-05,
          1.5664e-04, 6.4433e-05, 3.1734e-04, 1.2445e-04, 7.2539e-05, 1.7190e-04,
          1.0425e-04, 4.6492e-05, 1.3840e-04, 2.0492e-04, 2.0146e-04, 1.8501e-04,
          1.1557e-04, 7.9095e-05, 1.1659e-04, 7.2122e-05, 2.1660e-04, 1.4257e-04,
          5.5015e-05, 9.8109e-05, 5.0128e-05, 1.0973e-04, 3.1328e-04, 8.8274e-05,
          1.1420e-04, 3.1424e-04, 2.1636e-04, 1.2577e-04, 1.6105e-04, 5.7220e-05,
          1.3280e-04, 1.0037e-04, 3.8576e-04, 7.3314e-05, 1.9872e-04, 6.2704e-05,
          2.4557e-04, 6.5744e-05, 8.6784e-05, 7.8976e-05, 2.3782e-04, 2.0134e-04,
          7.4148e-05, 3.5346e-05, 3.4094e-05, 5.1141e-05, 1.0365e-04, 4.5002e-05,
          6.0701e-04, 7.6413e-05, 5.0187e-05, 6.6757e-05, 9.7930e-05, 1.3590e-04,
          1.0556e-04, 1.8430e-04, 7.0691e-05, 1.1462e-04, 8.0705e-05, 8.0049e-05,
          9.7454e-05, 5.8234e-05, 3.2127e-05, 3.2544e-05, 5.5790e-05, 1.0675e-04,
          3.0577e-05, 3.1114e-05, 7.3671e-05, 8.1778e-05, 1.5080e-04, 1.0195e-03,
          1.4627e-04, 6.8069e-05, 7.2479e-05, 6.4015e-05, 4.4763e-05, 8.7023e-05,
          9.7811e-05, 1.2994e-04, 6.6400e-05, 1.0407e-04, 1.9431e-04, 6.9737e-05,
          6.9261e-05, 7.4685e-05, 1.2338e-04, 4.7803e-05, 2.3413e-04, 8.9884e-05,
          4.8578e-05, 4.2558e-05, 7.5817e-05, 1.0014e-04, 2.8491e-04, 7.7188e-05,
          1.0163e-04, 6.6042e-05, 5.3108e-05, 8.9765e-05, 7.5996e-05, 1.1986e-04,
          7.4506e-05, 3.9411e-04, 1.1408e-04, 1.0419e-04, 9.6500e-05, 8.7142e-05,
          7.8499e-05, 3.9935e-05, 4.1664e-05, 1.1998e-04, 1.0991e-04, 5.3942e-05,
          6.1333e-05, 7.0572e-05, 1.1319e-04, 1.4353e-04, 1.1212e-04, 1.5581e-04,
          1.3483e-04, 8.2672e-05, 1.3852e-04, 5.9485e-05, 9.8288e-05, 9.1136e-05,
          4.4048e-05, 6.1870e-05, 1.2863e-04, 1.2302e-04, 4.1294e-04, 1.3399e-04,
          5.0068e-05, 3.3855e-04, 7.0572e-05, 4.9531e-05, 7.8321e-05, 1.3256e-04,
          7.6115e-05, 3.8743e-05, 3.7026e-04, 1.7202e-04, 1.1688e-04, 1.5068e-04,
          5.9748e-04, 3.2306e-05, 1.3614e-04, 6.1417e-04, 2.3472e-04, 1.0115e-04,
          4.4405e-05, 3.0184e-04, 1.0502e-04, 1.0514e-04, 2.6965e-04, 1.1379e-04,
          1.9789e-04, 1.3316e-04, 1.0562e-04, 9.5546e-05, 1.7047e-04, 1.5616e-04,
          5.7757e-05, 1.7939e-03, 1.2195e-04, 1.4031e-04, 1.5080e-04, 1.0681e-04,
          1.8144e-04, 1.0616e-04, 7.5936e-05, 7.5936e-05, 8.5950e-05, 7.8917e-05,
          8.5592e-05, 8.9526e-05, 1.8752e-04, 1.7416e-04, 1.0210e-04, 4.1842e-04,
          1.0115e-04, 7.0095e-05, 2.5988e-04, 2.4724e-04, 2.4662e-03, 3.4642e-04,
          2.2049e-03, 2.4414e-04, 2.1338e-04, 9.8288e-05, 6.5029e-05, 2.7704e-04,
          5.8889e-05, 1.5247e-04, 2.1660e-04, 1.6367e-04, 9.6619e-05, 1.0312e-04,
          4.4107e-05, 1.6201e-04, 6.4850e-05, 1.0651e-04, 7.8201e-05, 1.0347e-04,
          2.4343e-04, 9.3818e-05, 2.4509e-04, 5.7459e-04, 6.8140e-04, 1.5160e-02,
          6.7017e-02, 3.7964e-02, 2.0813e-02, 2.5085e-02, 9.9487e-03, 3.2806e-02,
          9.7351e-02, 6.4453e-02, 9.2834e-02, 2.9190e-02, 6.8542e-02, 3.1281e-02,
          1.9250e-01, 2.8801e-03], dtype=torch.float16),
  tensor([0.0000e+00, 2.0995e-03, 2.7405e-02, 1.8757e-03, 1.6518e-02, 9.7674e-03,
          2.0832e-02, 1.4429e-02, 2.1065e-04, 7.7037e-05, 5.8581e-04, 1.4250e-03,
          2.9177e-04, 1.5615e-04, 1.3919e-04, 2.5103e-04, 1.2524e-04, 2.4706e-04,
          1.2310e-04, 6.6666e-05, 7.8322e-05, 1.1446e-04, 6.5871e-05, 4.1962e-04,
          1.1645e-03, 4.3152e-04, 9.8872e-03, 2.4708e-03, 1.4412e-04, 2.0080e-04,
          1.9734e-04, 3.4057e-04, 8.2279e-05, 3.3434e-04, 2.6524e-04, 1.2731e-04,
          2.9733e-04, 1.8690e-04, 4.2251e-04, 2.4262e-04, 8.3429e-05, 1.8682e-04,
          1.4841e-04, 9.4306e-05, 1.1355e-04, 5.8052e-05, 1.6379e-04, 3.6455e-04,
          6.7753e-05, 1.3450e-04, 1.1614e-03, 2.2301e-04, 2.3734e-04, 1.2042e-04,
          8.7030e-05, 1.1398e-04, 1.2669e-04, 1.7507e-04, 2.9081e-04, 2.1285e-04,
          2.0025e-04, 1.2710e-04, 1.4154e-04, 1.3130e-04, 9.9964e-05, 5.5416e-04,
          5.4445e-04, 1.5767e-04, 1.6429e-04, 5.0369e-05, 4.8870e-05, 7.7485e-05,
          1.0013e-04, 8.9446e-04, 1.7793e-04, 2.5433e-04, 7.0215e-05, 5.6076e-05,
          1.2590e-04, 1.6106e-04, 4.3351e-04, 2.5718e-04, 4.7299e-04, 1.2339e-04,
          1.4607e-04, 2.3589e-04, 8.7904e-05, 2.9971e-04, 2.5707e-04, 2.0849e-04,
          5.4416e-04, 2.8961e-04, 4.8098e-05, 3.3206e-04, 4.5613e-05, 9.0711e-05,
          1.2050e-04, 1.0690e-04, 1.5388e-04, 4.3277e-04, 1.7042e-04, 1.7599e-04,
          8.3460e-04, 3.5083e-04, 2.1440e-04, 2.5117e-04, 5.9451e-04, 8.2867e-05,
          1.3558e-04, 9.2849e-05, 3.9647e-04, 7.3888e-05, 3.6669e-04, 1.9151e-04,
          3.0562e-04, 8.1717e-05, 5.8042e-05, 4.5635e-04, 1.1870e-04, 1.6051e-04,
          1.3402e-04, 4.1933e-04, 1.1864e-04, 1.2055e-04, 6.3635e-04, 7.4235e-05,
          7.9467e-05, 1.8167e-04, 2.1731e-04, 2.9772e-04, 1.0390e-04, 1.4710e-04,
          4.6704e-04, 3.2933e-04, 9.6793e-05, 6.0079e-05, 1.7493e-04, 5.4936e-05,
          2.3987e-04, 6.5725e-05, 1.7009e-04, 7.9013e-04, 5.8277e-04, 1.1043e-04,
          1.6312e-04, 8.4647e-05, 1.2604e-04, 1.3627e-04, 4.3728e-04, 3.9532e-04,
          6.1274e-04, 2.0146e-04, 9.5331e-05, 1.3595e-04, 1.5405e-04, 1.3751e-04,
          1.7501e-04, 1.4274e-04, 7.0240e-05, 9.5953e-05, 2.0510e-04, 7.6978e-05,
          4.0205e-04, 3.9932e-04, 1.7846e-04, 5.4029e-04, 3.7338e-04, 3.5644e-04,
          4.9995e-04, 3.3334e-04, 1.7340e-04, 1.3439e-04, 1.1004e-04, 3.7257e-04,
          8.1710e-05, 1.0376e-04, 2.4555e-04, 2.1710e-04, 1.9823e-04, 2.4076e-04,
          2.6112e-04, 8.2458e-05, 3.4772e-04, 1.5195e-04, 1.4768e-04, 1.6743e-04,
          1.9006e-04, 7.7818e-05, 4.8230e-04, 1.8233e-04, 2.2664e-04, 2.4231e-04,
          1.5073e-04, 5.5789e-05, 2.2686e-04, 1.8810e-04, 3.0044e-04, 2.7613e-04,
          1.5855e-04, 8.2310e-05, 1.4121e-04, 1.1345e-04, 4.8753e-04, 3.0198e-04,
          1.0925e-04, 1.1696e-04, 4.9969e-05, 1.3454e-04, 4.7630e-04, 5.6139e-05,
          1.4386e-04, 3.7507e-04, 3.5993e-04, 2.8365e-04, 3.7485e-04, 1.4938e-04,
          2.2019e-04, 1.1113e-04, 4.8371e-04, 1.2875e-04, 2.7149e-04, 7.7015e-05,
          3.5546e-04, 8.1440e-05, 1.0748e-04, 3.1628e-04, 5.7318e-04, 2.2540e-04,
          1.8270e-04, 5.4165e-05, 3.3378e-05, 8.8320e-05, 1.2913e-04, 5.0935e-05,
          1.0670e-03, 1.9158e-04, 1.2707e-04, 1.3573e-04, 1.9572e-04, 2.5263e-04,
          1.6311e-04, 2.4661e-04, 1.4102e-04, 1.6365e-04, 1.0914e-04, 1.0444e-04,
          1.0842e-04, 9.3341e-05, 4.7737e-05, 4.5232e-05, 7.5839e-05, 1.4865e-04,
          3.6155e-05, 3.2668e-05, 7.9449e-05, 1.1816e-04, 1.7056e-04, 9.0694e-04,
          1.8932e-04, 2.3635e-04, 1.8588e-04, 1.2854e-04, 4.9513e-05, 1.5092e-04,
          1.3086e-04, 1.7316e-04, 1.0433e-04, 1.5422e-04, 3.5250e-04, 1.0327e-04,
          1.3559e-04, 2.0887e-04, 2.4345e-04, 6.6416e-05, 3.8511e-04, 1.6539e-04,
          6.1245e-05, 6.1404e-05, 7.7612e-05, 1.1043e-04, 3.2745e-04, 2.3557e-04,
          6.3927e-04, 1.0744e-04, 8.0083e-05, 1.0547e-04, 1.0222e-04, 2.2153e-04,
          1.0574e-04, 6.4695e-04, 1.0208e-04, 1.1976e-04, 1.0704e-04, 9.5863e-05,
          8.7803e-05, 5.1131e-05, 5.0665e-05, 2.5373e-04, 1.6662e-04, 6.6702e-05,
          6.4874e-05, 8.0515e-05, 1.4666e-04, 1.6256e-04, 1.9190e-04, 3.4264e-04,
          2.7973e-04, 7.9241e-05, 2.3414e-04, 7.1643e-05, 1.5972e-04, 9.3743e-05,
          5.0138e-05, 1.1735e-04, 1.3944e-04, 1.3560e-04, 5.6217e-04, 1.4262e-04,
          9.1778e-05, 5.0265e-04, 8.5356e-05, 5.0947e-05, 9.0430e-05, 1.3529e-04,
          9.6027e-05, 3.8390e-05, 4.2231e-04, 2.4103e-04, 2.0410e-04, 1.7589e-04,
          8.2773e-04, 3.0008e-05, 1.4848e-04, 1.0857e-03, 3.9088e-04, 1.2605e-04,
          5.4416e-05, 3.9584e-04, 1.3449e-04, 1.0714e-04, 3.8068e-04, 1.2090e-04,
          3.5730e-04, 2.9159e-04, 8.2976e-05, 1.6814e-04, 2.2068e-04, 1.4559e-04,
          6.0188e-05, 1.2359e-03, 1.6058e-04, 1.8238e-04, 1.8631e-04, 9.5567e-05,
          2.1929e-04, 1.7106e-04, 1.2618e-04, 1.3433e-04, 8.1627e-05, 1.3603e-04,
          1.0694e-04, 8.9566e-05, 2.7342e-04, 2.7877e-04, 1.3195e-04, 6.0129e-04,
          1.9970e-04, 7.0297e-05, 2.6676e-04, 3.1597e-04, 2.6740e-03, 5.2099e-04,
          1.4511e-03, 2.8406e-04, 2.3990e-04, 1.6956e-04, 5.7075e-05, 2.2302e-04,
          4.8397e-05, 1.6891e-04, 3.1918e-04, 2.3269e-04, 1.1079e-04, 1.9822e-04,
          5.6211e-05, 1.4769e-04, 6.7083e-05, 1.4303e-04, 1.0037e-04, 3.1506e-04,
          2.5585e-04, 1.3075e-04, 3.3099e-04, 6.9556e-04, 7.5481e-04, 1.0352e-02,
          5.5527e-02, 3.4235e-02, 2.9242e-02, 3.1150e-02, 1.1207e-02, 4.2191e-02,
          1.0118e-01, 4.8331e-02, 6.4940e-02, 1.8898e-02, 4.7815e-02, 2.4592e-02,
          1.2309e-01, 1.6283e-01, 0.0000e+00]),
  tensor([0.0000e+00, 1.1428e-03, 2.3274e-02, 1.1907e-03, 8.9420e-03, 6.5407e-03,
          2.0408e-02, 4.3832e-03, 3.1063e-04, 1.0249e-04, 8.4341e-04, 1.9057e-03,
          4.0820e-04, 2.6501e-04, 1.9689e-04, 2.7995e-04, 2.0532e-04, 4.4914e-04,
          1.7689e-04, 9.0413e-05, 9.0322e-05, 1.2604e-04, 8.4000e-05, 4.7947e-04,
          1.6504e-03, 3.8463e-04, 3.5429e-03, 2.2946e-03, 1.7889e-04, 3.4931e-04,
          2.8291e-04, 5.2981e-04, 1.1950e-04, 4.2238e-04, 3.7901e-04, 1.9096e-04,
          4.2443e-04, 2.3981e-04, 6.9420e-04, 3.9606e-04, 1.2204e-04, 3.2763e-04,
          1.5902e-04, 1.1275e-04, 1.1827e-04, 7.6164e-05, 3.2146e-04, 3.7329e-04,
          9.3480e-05, 1.1990e-04, 9.1333e-04, 4.9621e-04, 3.1537e-04, 1.7254e-04,
          1.1908e-04, 1.4859e-04, 1.5738e-04, 2.6356e-04, 3.3557e-04, 2.9307e-04,
          2.7339e-04, 1.7810e-04, 2.0775e-04, 2.7207e-04, 1.4190e-04, 5.0055e-04,
          6.9121e-04, 2.6799e-04, 1.9556e-04, 5.1097e-05, 4.6475e-05, 7.5740e-05,
          9.1364e-05, 1.3119e-03, 2.3765e-04, 3.3759e-04, 7.8370e-05, 6.9448e-05,
          1.7582e-04, 2.1065e-04, 3.7060e-04, 3.3075e-04, 3.9961e-04, 1.3346e-04,
          1.4844e-04, 3.2793e-04, 1.1029e-04, 3.9100e-04, 3.4012e-04, 2.4977e-04,
          8.2243e-04, 3.8835e-04, 4.9788e-05, 4.7690e-04, 4.6502e-05, 8.0739e-05,
          1.8790e-04, 1.3386e-04, 1.9380e-04, 5.6071e-04, 2.1159e-04, 2.7116e-04,
          1.0572e-03, 3.9170e-04, 2.8231e-04, 3.0235e-04, 4.6741e-04, 1.0664e-04,
          1.9242e-04, 1.0609e-04, 5.9688e-04, 8.7854e-05, 3.5421e-04, 2.3595e-04,
          2.1416e-04, 9.0704e-05, 5.3948e-05, 4.0097e-04, 1.0132e-04, 1.6391e-04,
          1.7380e-04, 5.1520e-04, 1.0464e-04, 1.5229e-04, 5.2985e-04, 8.3928e-05,
          1.0074e-04, 1.9376e-04, 2.0115e-04, 3.1249e-04, 1.6797e-04, 2.1608e-04,
          4.1185e-04, 2.7122e-04, 1.1074e-04, 8.6168e-05, 2.6278e-04, 5.7617e-05,
          3.5066e-04, 8.3157e-05, 2.7624e-04, 7.5434e-04, 7.3077e-04, 1.4125e-04,
          1.4431e-04, 1.1018e-04, 1.1807e-04, 2.2023e-04, 4.1189e-04, 3.4933e-04,
          9.4902e-04, 2.3166e-04, 1.2925e-04, 2.1159e-04, 1.9866e-04, 1.9730e-04,
          2.6248e-04, 2.4181e-04, 9.4549e-05, 1.1723e-04, 2.1668e-04, 1.0529e-04,
          3.4569e-04, 4.0233e-04, 3.1571e-04, 5.7224e-04, 4.2273e-04, 3.9981e-04,
          6.4458e-04, 3.9109e-04, 1.6246e-04, 1.5736e-04, 1.3749e-04, 2.9217e-04,
          1.1631e-04, 1.0899e-04, 2.5793e-04, 2.7019e-04, 2.7818e-04, 3.0057e-04,
          1.9653e-04, 1.0447e-04, 3.4721e-04, 2.1752e-04, 2.0327e-04, 1.6898e-04,
          2.5641e-04, 1.0682e-04, 5.7060e-04, 1.9673e-04, 2.0262e-04, 2.5019e-04,
          1.9895e-04, 6.9616e-05, 1.8872e-04, 3.3736e-04, 4.4242e-04, 2.6674e-04,
          2.0270e-04, 8.9850e-05, 2.0539e-04, 1.4989e-04, 4.5766e-04, 2.1001e-04,
          8.8902e-05, 2.1756e-04, 5.5733e-05, 1.5160e-04, 6.0036e-04, 7.3456e-05,
          1.6657e-04, 4.2415e-04, 3.2946e-04, 3.4496e-04, 2.8357e-04, 1.3462e-04,
          2.4563e-04, 1.5374e-04, 7.1286e-04, 1.2245e-04, 3.0415e-04, 1.2173e-04,
          4.1646e-04, 9.2782e-05, 1.9499e-04, 4.0677e-04, 5.2127e-04, 2.5313e-04,
          1.6872e-04, 5.7713e-05, 4.2190e-05, 1.0021e-04, 1.8036e-04, 6.8958e-05,
          7.9758e-04, 2.5371e-04, 9.6299e-05, 1.3192e-04, 2.0231e-04, 2.9497e-04,
          2.1062e-04, 2.8255e-04, 2.0349e-04, 2.0787e-04, 1.9647e-04, 1.2826e-04,
          1.4647e-04, 1.1400e-04, 4.9530e-05, 6.0109e-05, 7.6078e-05, 1.8782e-04,
          3.9765e-05, 3.1769e-05, 7.4599e-05, 1.1686e-04, 2.3437e-04, 9.4765e-04,
          2.2985e-04, 3.2091e-04, 1.3955e-04, 1.1344e-04, 5.5189e-05, 2.1087e-04,
          1.4978e-04, 2.2272e-04, 1.6644e-04, 2.3552e-04, 4.8570e-04, 1.2146e-04,
          1.3137e-04, 2.1900e-04, 2.3866e-04, 8.5194e-05, 3.0123e-04, 1.8123e-04,
          7.5671e-05, 7.5437e-05, 5.6762e-05, 1.4442e-04, 2.8319e-04, 1.8096e-04,
          5.6301e-04, 1.0835e-04, 8.8881e-05, 1.0820e-04, 1.4147e-04, 1.9324e-04,
          1.3809e-04, 5.6590e-04, 1.4434e-04, 2.1957e-04, 1.1988e-04, 1.2004e-04,
          1.1065e-04, 5.8625e-05, 5.2683e-05, 1.8368e-04, 2.5346e-04, 6.3669e-05,
          5.6535e-05, 1.0549e-04, 1.7528e-04, 2.3184e-04, 1.8204e-04, 3.4046e-04,
          2.0013e-04, 9.1819e-05, 2.4560e-04, 1.1775e-04, 1.6442e-04, 1.3183e-04,
          7.6755e-05, 1.1127e-04, 1.4755e-04, 2.1237e-04, 5.8895e-04, 1.9800e-04,
          1.0498e-04, 6.6457e-04, 1.1452e-04, 6.8602e-05, 8.3460e-05, 1.2257e-04,
          1.4771e-04, 5.2702e-05, 4.5888e-04, 2.4035e-04, 1.9363e-04, 2.2607e-04,
          9.3730e-04, 3.0023e-05, 1.7685e-04, 1.1332e-03, 5.6494e-04, 1.9593e-04,
          6.3419e-05, 4.4880e-04, 1.6253e-04, 1.0711e-04, 4.7416e-04, 1.4422e-04,
          4.8662e-04, 2.8576e-04, 1.0311e-04, 2.3134e-04, 2.6105e-04, 1.9422e-04,
          5.7233e-05, 8.3108e-04, 1.6177e-04, 1.6653e-04, 2.4872e-04, 1.2288e-04,
          2.4279e-04, 1.6256e-04, 1.2991e-04, 1.2907e-04, 9.6348e-05, 1.4485e-04,
          1.2896e-04, 1.1279e-04, 2.1468e-04, 4.7341e-04, 1.0918e-04, 7.2936e-04,
          1.8292e-04, 7.5761e-05, 2.3603e-04, 4.6083e-04, 2.5521e-03, 4.9356e-04,
          7.7311e-04, 2.4729e-04, 2.5329e-04, 1.4164e-04, 7.0850e-05, 1.9658e-04,
          6.6195e-05, 2.1890e-04, 3.9471e-04, 3.5494e-04, 1.3964e-04, 1.3762e-04,
          7.5412e-05, 1.2458e-04, 9.0459e-05, 1.5883e-04, 1.3026e-04, 2.3306e-04,
          3.0765e-04, 9.8926e-05, 3.8125e-04, 6.9503e-04, 7.9650e-04, 6.6652e-03,
          4.0395e-02, 4.1115e-02, 5.7639e-02, 4.3746e-02, 1.5590e-02, 5.7824e-02,
          9.0219e-02, 2.7566e-02, 3.4977e-02, 9.6270e-03, 2.5740e-02, 1.6211e-02,
          6.8687e-02, 1.3654e-01, 1.5380e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.7286e-04, 2.2771e-02, 7.8036e-04, 5.0904e-03, 4.0628e-03,
          2.2771e-02, 2.0210e-03, 3.4597e-04, 1.9295e-04, 2.0095e-03, 3.5531e-03,
          1.4617e-03, 4.3770e-04, 2.4688e-04, 6.6279e-04, 2.8111e-04, 6.6326e-04,
          4.3322e-04, 1.3499e-04, 7.9949e-05, 8.4731e-05, 1.0391e-04, 7.1528e-04,
          4.7657e-03, 7.0813e-04, 1.7315e-03, 1.2592e-03, 1.4907e-04, 7.1563e-04,
          2.9813e-04, 9.0370e-04, 2.9700e-04, 1.3158e-03, 1.4413e-03, 3.3546e-04,
          1.0328e-03, 3.3509e-04, 1.0053e-03, 1.1073e-03, 1.3359e-04, 8.8873e-04,
          2.1641e-04, 2.0189e-04, 1.9193e-04, 2.2559e-04, 5.8184e-04, 1.1853e-03,
          1.4145e-04, 1.0639e-04, 5.4079e-04, 6.1079e-04, 6.6427e-04, 3.0749e-04,
          1.1385e-04, 3.5702e-04, 1.0318e-04, 7.0429e-04, 9.0967e-04, 1.1044e-03,
          5.2027e-04, 3.6607e-04, 1.5530e-04, 3.7167e-04, 5.4024e-04, 1.5100e-03,
          4.9952e-04, 5.2123e-04, 4.8523e-04, 1.2589e-04, 1.0769e-04, 1.7123e-04,
          1.5657e-04, 2.9112e-03, 1.4274e-04, 9.2593e-04, 1.1881e-04, 1.1784e-04,
          3.3345e-04, 5.8137e-04, 7.6516e-04, 1.2876e-03, 1.0177e-03, 4.6127e-04,
          1.9935e-04, 7.6349e-04, 1.2809e-04, 1.3534e-03, 3.7585e-04, 2.6480e-04,
          3.1385e-03, 1.6537e-03, 1.1008e-04, 2.2643e-03, 1.0168e-04, 1.1584e-04,
          1.7667e-04, 6.9013e-05, 4.2950e-04, 1.3012e-03, 4.5368e-04, 7.1628e-04,
          3.5682e-03, 3.2466e-04, 7.5486e-04, 1.2307e-03, 1.3831e-03, 1.9176e-04,
          5.7249e-04, 1.5523e-04, 2.1078e-03, 1.0426e-04, 2.9957e-04, 3.3148e-04,
          4.7481e-04, 2.8876e-04, 5.3722e-05, 1.5839e-03, 1.1489e-04, 1.1523e-04,
          2.7448e-04, 1.0228e-03, 1.8513e-04, 1.8792e-04, 1.2832e-03, 2.0447e-04,
          2.6721e-04, 4.8275e-04, 4.7587e-04, 1.5166e-03, 5.3572e-04, 7.4356e-04,
          4.1895e-04, 3.4964e-04, 2.3567e-04, 1.3682e-04, 1.1626e-03, 9.8989e-05,
          8.4973e-04, 1.2917e-04, 3.0071e-04, 7.3699e-04, 1.0170e-03, 3.3464e-04,
          2.6129e-04, 2.3809e-04, 2.8780e-04, 5.8627e-04, 4.8401e-04, 1.5403e-03,
          2.3263e-03, 1.2760e-03, 4.4957e-04, 4.1345e-04, 2.5254e-04, 4.1435e-04,
          8.4981e-04, 3.7469e-04, 4.5488e-04, 1.2687e-04, 1.0005e-03, 3.1337e-04,
          9.7149e-04, 7.8572e-04, 5.7810e-04, 7.8672e-04, 4.9994e-04, 3.9274e-04,
          2.1916e-03, 6.2847e-04, 5.9373e-04, 3.7892e-04, 3.5926e-04, 9.9718e-04,
          5.9089e-04, 2.9871e-04, 3.4261e-04, 2.6734e-04, 8.0426e-04, 7.7091e-04,
          3.6582e-04, 4.0403e-04, 7.5744e-04, 4.0997e-04, 2.1212e-04, 2.8835e-04,
          1.6691e-03, 1.1856e-04, 5.1556e-04, 3.0496e-04, 3.3542e-04, 5.8674e-04,
          5.1542e-04, 1.7525e-04, 6.0836e-04, 6.1852e-04, 1.5029e-03, 9.2553e-04,
          3.3863e-04, 9.0054e-05, 9.5206e-04, 5.1345e-04, 6.1509e-04, 4.2728e-04,
          1.7250e-04, 7.9851e-04, 1.3385e-04, 2.6999e-04, 3.1086e-03, 2.3381e-04,
          1.8874e-04, 4.4699e-04, 4.1366e-04, 9.0383e-04, 1.0457e-03, 2.8331e-04,
          6.4531e-04, 4.3844e-04, 1.2674e-03, 3.0840e-04, 1.0514e-03, 3.7315e-04,
          7.3908e-04, 1.1800e-04, 5.1913e-04, 3.4878e-04, 1.1709e-03, 6.7962e-04,
          3.6661e-04, 1.1643e-04, 1.1054e-04, 2.0499e-04, 4.7668e-04, 1.7427e-04,
          6.4090e-04, 4.5731e-04, 1.7207e-04, 3.7586e-04, 7.2387e-04, 9.3499e-04,
          3.5878e-04, 2.8177e-04, 3.0024e-04, 4.7741e-04, 5.3967e-04, 1.5226e-04,
          2.1269e-04, 6.6179e-04, 9.5943e-05, 1.8566e-04, 2.1449e-04, 7.4755e-04,
          8.0497e-05, 6.7491e-05, 1.0145e-04, 1.7014e-04, 6.5325e-04, 1.0062e-03,
          4.3598e-04, 3.5394e-04, 3.6851e-04, 3.9281e-04, 1.3553e-04, 5.0908e-04,
          6.0978e-04, 3.9194e-04, 2.9274e-04, 5.6815e-04, 6.0545e-04, 3.8791e-04,
          2.7629e-04, 4.2858e-04, 7.4567e-04, 1.8318e-04, 6.2996e-04, 6.4295e-04,
          2.6214e-04, 1.8812e-04, 4.9992e-05, 2.2472e-04, 3.3139e-04, 2.7765e-04,
          5.2500e-04, 1.7058e-04, 1.9808e-04, 1.6276e-04, 4.6012e-04, 5.5175e-04,
          4.0947e-04, 1.1434e-03, 3.7058e-04, 3.8472e-04, 1.3055e-04, 3.2940e-04,
          1.8623e-04, 1.4078e-04, 1.0142e-04, 7.4190e-04, 5.5490e-04, 1.2860e-04,
          7.9703e-05, 2.1560e-04, 2.7285e-04, 4.8222e-04, 3.0027e-04, 1.0278e-03,
          9.7898e-04, 9.7094e-05, 5.3036e-04, 1.4270e-04, 2.6548e-04, 3.6470e-04,
          1.8014e-04, 2.6951e-04, 1.5222e-04, 4.4396e-04, 1.2483e-03, 7.5976e-04,
          2.0897e-04, 2.9657e-03, 2.8246e-04, 2.1327e-04, 9.0057e-05, 1.4178e-04,
          1.5418e-04, 1.1074e-04, 5.9121e-04, 4.0356e-04, 5.1514e-04, 3.9053e-04,
          1.5060e-03, 8.8354e-05, 2.7425e-04, 1.5179e-03, 6.8717e-04, 1.9061e-04,
          1.5174e-04, 4.5486e-04, 2.3127e-04, 1.3264e-04, 9.9868e-04, 2.3923e-04,
          6.9992e-04, 4.3032e-04, 2.2215e-04, 5.1844e-04, 4.1179e-04, 1.9535e-04,
          5.4195e-05, 4.0967e-04, 2.9737e-04, 3.0641e-04, 5.3301e-04, 3.3999e-04,
          6.9112e-04, 3.0314e-04, 3.1266e-04, 1.8599e-04, 1.4755e-04, 3.8114e-04,
          2.5808e-04, 2.7141e-04, 4.2897e-04, 3.9054e-04, 1.8030e-04, 9.9781e-04,
          2.6165e-04, 9.1624e-05, 2.7464e-04, 5.3114e-04, 1.4732e-03, 4.7182e-04,
          4.1303e-04, 3.4679e-04, 4.7495e-04, 3.2505e-04, 1.7402e-04, 1.7786e-04,
          1.0989e-04, 3.1978e-04, 1.2162e-03, 6.1983e-04, 1.5997e-04, 1.9523e-04,
          1.0782e-04, 1.0111e-04, 1.2507e-04, 2.0885e-04, 3.4807e-04, 7.4781e-04,
          3.3165e-04, 1.8589e-04, 2.9035e-04, 7.3176e-04, 6.4205e-04, 5.8082e-03,
          1.2972e-02, 2.3586e-02, 7.1743e-02, 5.4822e-02, 1.8954e-02, 8.0957e-02,
          4.8488e-02, 1.0542e-02, 1.4404e-02, 6.6976e-03, 1.3116e-02, 1.4983e-02,
          3.0827e-02, 7.1765e-02, 1.1551e-01, 1.3362e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.0560e-04, 2.4907e-02, 5.9794e-04, 5.7207e-03, 3.9906e-03,
          2.3623e-02, 1.9364e-03, 1.6044e-04, 1.0710e-04, 7.4690e-04, 8.6671e-04,
          3.0832e-04, 3.7043e-04, 1.7013e-04, 4.1428e-04, 1.2319e-04, 7.0174e-04,
          3.1063e-04, 8.7601e-05, 9.3652e-05, 1.3985e-04, 1.4444e-04, 8.1586e-04,
          8.8254e-04, 3.4756e-04, 1.7341e-03, 1.9200e-03, 1.6553e-04, 6.1113e-04,
          2.3463e-04, 3.3853e-04, 2.1676e-04, 1.3183e-03, 6.3246e-04, 1.9232e-04,
          7.7389e-04, 1.5505e-04, 1.7569e-03, 1.0698e-03, 1.1771e-04, 9.3174e-04,
          1.6524e-04, 1.6390e-04, 1.1159e-04, 1.6503e-04, 5.3374e-04, 1.3343e-03,
          9.1649e-05, 1.1583e-04, 5.6533e-04, 3.5947e-04, 2.0863e-04, 1.8373e-04,
          6.1552e-05, 1.1930e-04, 1.0209e-04, 3.5084e-04, 2.3737e-04, 2.8266e-04,
          2.4148e-04, 2.9818e-04, 1.0630e-04, 5.7750e-04, 2.2468e-04, 8.9098e-04,
          1.4880e-04, 9.3051e-05, 3.4723e-04, 7.3378e-05, 5.6421e-05, 7.4537e-05,
          8.8305e-05, 1.4003e-03, 1.0339e-04, 2.4873e-04, 4.8633e-05, 5.8433e-05,
          1.5521e-04, 3.4960e-04, 1.6988e-04, 4.8279e-04, 1.8946e-04, 1.2474e-04,
          1.0644e-04, 2.7726e-04, 9.0514e-05, 2.1190e-04, 7.9803e-05, 6.8589e-05,
          4.5893e-04, 6.9676e-04, 4.2232e-05, 3.4051e-04, 6.0073e-05, 7.8445e-05,
          1.6542e-04, 6.1519e-05, 2.0696e-04, 2.0426e-03, 2.8766e-04, 3.7185e-04,
          5.2092e-04, 9.7843e-05, 1.2730e-04, 3.7216e-04, 6.2907e-04, 1.7693e-04,
          2.0401e-04, 1.3536e-04, 6.7779e-04, 4.6137e-05, 7.7977e-05, 1.2638e-04,
          1.9802e-04, 2.5627e-04, 6.0400e-05, 8.1680e-04, 8.4465e-05, 1.1403e-04,
          1.9693e-04, 1.0758e-03, 7.4589e-05, 1.2809e-04, 5.5207e-04, 5.8950e-05,
          1.4802e-04, 1.4528e-04, 1.2315e-04, 6.6652e-04, 3.0166e-04, 2.2394e-04,
          1.4713e-04, 8.6702e-05, 1.0466e-04, 3.6694e-05, 2.4116e-04, 4.6804e-05,
          1.3056e-03, 9.0508e-05, 3.1939e-04, 7.3786e-04, 7.9237e-04, 2.1344e-04,
          1.1388e-04, 8.7805e-05, 1.4214e-04, 2.3875e-04, 1.1133e-04, 4.9381e-04,
          9.3284e-04, 5.1671e-04, 1.7765e-04, 3.8369e-04, 1.7615e-04, 6.0754e-04,
          4.3931e-04, 4.2072e-04, 7.1281e-05, 5.3558e-05, 7.7907e-04, 2.2559e-04,
          6.4287e-04, 1.0322e-03, 6.0066e-04, 7.0885e-04, 6.5587e-04, 3.0060e-04,
          2.9256e-04, 4.3491e-04, 2.6596e-04, 1.9127e-04, 2.8223e-04, 8.5382e-04,
          1.7655e-04, 9.5883e-05, 2.6232e-04, 2.8586e-04, 6.5996e-04, 4.2432e-04,
          2.3206e-04, 2.6273e-04, 4.0223e-04, 1.5376e-04, 2.5311e-04, 2.6551e-04,
          2.4048e-04, 8.4578e-05, 3.4703e-04, 1.7431e-04, 7.9560e-05, 2.9025e-04,
          1.7535e-04, 7.4190e-05, 4.8380e-04, 5.1374e-04, 2.7525e-04, 5.1383e-04,
          2.0320e-04, 1.0459e-04, 6.4590e-04, 1.8505e-04, 2.3482e-04, 3.2049e-04,
          5.5179e-05, 3.5015e-04, 8.6455e-05, 1.6435e-04, 4.1835e-04, 1.8874e-04,
          1.2186e-04, 7.3104e-04, 1.9934e-04, 1.2212e-04, 4.9441e-04, 9.3493e-05,
          6.9139e-04, 3.9989e-04, 5.7169e-04, 1.5529e-04, 1.1267e-03, 1.3518e-04,
          1.3415e-03, 7.4966e-05, 2.1131e-04, 1.2384e-04, 5.1240e-04, 3.8072e-04,
          7.6797e-05, 4.2901e-05, 4.3434e-05, 1.0013e-04, 3.8891e-04, 8.9513e-05,
          4.0144e-04, 1.8561e-04, 5.5144e-05, 1.7890e-04, 4.0266e-04, 3.2046e-04,
          1.0659e-04, 1.4071e-04, 1.7440e-04, 7.3211e-04, 3.0608e-04, 1.3003e-04,
          2.4371e-04, 7.5745e-05, 3.5547e-05, 6.5328e-05, 5.3354e-05, 2.7164e-04,
          2.9711e-05, 4.0123e-05, 6.8339e-05, 1.5684e-04, 6.0292e-04, 1.0540e-03,
          3.2664e-04, 1.2090e-04, 1.8934e-04, 1.9113e-04, 6.1808e-05, 1.3313e-04,
          2.9849e-04, 2.6359e-04, 2.5873e-04, 2.7975e-04, 4.3305e-04, 2.0824e-04,
          8.1823e-05, 9.4749e-05, 2.6496e-04, 7.8413e-05, 7.4421e-04, 2.8434e-04,
          1.0209e-04, 1.1152e-04, 5.8379e-05, 1.1257e-04, 2.6644e-04, 8.2569e-05,
          1.0535e-04, 7.4938e-05, 8.8261e-05, 1.0307e-04, 3.9501e-04, 4.9528e-04,
          1.4626e-04, 1.1519e-03, 3.6512e-04, 2.4853e-04, 8.2291e-05, 1.1803e-04,
          1.1270e-04, 4.9954e-05, 4.7462e-05, 4.8065e-04, 4.8209e-04, 5.7760e-05,
          5.3046e-05, 1.6207e-04, 1.3905e-04, 4.8406e-04, 1.3140e-04, 3.4219e-04,
          5.9760e-04, 1.0328e-04, 6.3653e-04, 1.1117e-04, 1.5986e-04, 1.1243e-04,
          1.4890e-04, 1.6010e-04, 2.1941e-04, 2.3705e-04, 1.5912e-03, 3.9739e-04,
          8.0440e-05, 4.8411e-04, 3.6257e-04, 1.6729e-04, 7.0955e-05, 1.2455e-04,
          1.6998e-04, 5.4474e-05, 5.0895e-04, 2.4246e-04, 3.2065e-04, 2.3654e-04,
          1.4836e-03, 4.8037e-05, 1.4867e-04, 1.6537e-03, 3.0352e-04, 2.5264e-04,
          7.3929e-05, 4.9423e-04, 1.7635e-04, 1.3637e-04, 6.6043e-04, 1.1426e-04,
          3.1581e-04, 2.5359e-04, 2.6444e-04, 1.6958e-04, 2.6751e-04, 2.8831e-04,
          3.8473e-05, 5.1326e-04, 1.3637e-04, 1.4975e-04, 2.0700e-04, 1.1449e-04,
          4.3965e-04, 1.9117e-04, 1.4123e-04, 1.1446e-04, 6.9182e-05, 2.1697e-04,
          1.0949e-04, 1.0318e-04, 5.3154e-04, 4.1174e-04, 1.2318e-04, 1.0324e-03,
          1.7001e-04, 6.2784e-05, 2.4483e-04, 3.3958e-04, 1.9040e-03, 3.7068e-04,
          4.0664e-04, 2.0174e-04, 2.4119e-04, 2.1556e-04, 1.0449e-04, 1.6731e-04,
          7.0051e-05, 1.5815e-04, 2.8986e-04, 5.1298e-04, 1.5077e-04, 1.3680e-04,
          8.0977e-05, 8.8358e-05, 7.6454e-05, 1.5895e-04, 1.0968e-04, 3.4210e-04,
          4.1466e-04, 1.6475e-04, 2.5017e-04, 1.1022e-03, 6.5474e-04, 4.0235e-03,
          1.3452e-02, 2.4685e-02, 2.4127e-02, 5.9290e-02, 3.5230e-02, 1.0587e-01,
          5.2090e-02, 1.0535e-02, 1.5337e-02, 5.8788e-03, 1.3643e-02, 1.2935e-02,
          3.3666e-02, 6.4544e-02, 1.0011e-01, 1.2372e-01, 1.1555e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.6587e-04, 2.1728e-02, 6.4390e-04, 5.3344e-03, 5.4775e-03,
          2.0594e-02, 1.9041e-03, 2.4622e-04, 1.2700e-04, 6.5077e-04, 8.7766e-04,
          2.7479e-04, 5.6153e-04, 4.1424e-04, 3.5786e-04, 1.6605e-04, 1.0071e-03,
          4.9308e-04, 1.0711e-04, 2.0170e-04, 2.8380e-04, 2.4784e-04, 9.8361e-04,
          8.1432e-04, 3.3000e-04, 1.7475e-03, 2.2247e-03, 2.7640e-04, 8.8908e-04,
          5.9152e-04, 4.5025e-04, 3.3113e-04, 9.9922e-04, 5.8304e-04, 2.0086e-04,
          7.7473e-04, 2.2817e-04, 2.4296e-03, 9.9786e-04, 2.4603e-04, 1.1048e-03,
          1.9549e-04, 1.9012e-04, 1.3130e-04, 1.6557e-04, 8.4809e-04, 1.0356e-03,
          2.0907e-04, 2.0224e-04, 8.4439e-04, 4.9405e-04, 2.7472e-04, 1.8501e-04,
          9.6171e-05, 1.6940e-04, 1.3569e-04, 4.4078e-04, 3.0053e-04, 4.1276e-04,
          2.9627e-04, 6.3410e-04, 1.3635e-04, 6.1891e-04, 2.1252e-04, 6.9927e-04,
          2.2333e-04, 1.0761e-04, 2.6187e-04, 8.4520e-05, 6.6649e-05, 8.4827e-05,
          1.0496e-04, 1.0960e-03, 1.5168e-04, 3.1859e-04, 7.1641e-05, 9.1000e-05,
          2.4104e-04, 4.3892e-04, 2.4185e-04, 4.7790e-04, 2.7277e-04, 1.5732e-04,
          1.2464e-04, 3.1475e-04, 1.1374e-04, 1.9919e-04, 1.1859e-04, 1.2150e-04,
          4.2331e-04, 6.1282e-04, 4.8111e-05, 2.9624e-04, 7.1726e-05, 9.8735e-05,
          2.9744e-04, 8.6664e-05, 2.6206e-04, 2.2843e-03, 3.5984e-04, 4.5547e-04,
          4.8896e-04, 1.4766e-04, 1.6910e-04, 5.8769e-04, 5.9476e-04, 3.8623e-04,
          2.2545e-04, 1.5948e-04, 6.6835e-04, 6.4234e-05, 1.2157e-04, 1.7009e-04,
          2.4222e-04, 3.1016e-04, 1.0619e-04, 6.0031e-04, 1.1119e-04, 1.8521e-04,
          3.2523e-04, 1.1773e-03, 9.7629e-05, 2.1939e-04, 5.3017e-04, 8.7205e-05,
          2.2334e-04, 2.1724e-04, 1.9499e-04, 5.4406e-04, 3.6878e-04, 2.5517e-04,
          1.9966e-04, 1.1607e-04, 1.2379e-04, 4.9170e-05, 2.5723e-04, 6.0543e-05,
          1.3468e-03, 1.7689e-04, 4.3313e-04, 7.6452e-04, 7.4241e-04, 3.5395e-04,
          1.5620e-04, 1.0666e-04, 1.3953e-04, 3.0363e-04, 1.8548e-04, 5.7995e-04,
          1.0432e-03, 4.8750e-04, 2.0524e-04, 5.0646e-04, 2.2309e-04, 7.7283e-04,
          4.7349e-04, 6.5462e-04, 8.1106e-05, 9.0265e-05, 6.8594e-04, 2.9431e-04,
          4.1012e-04, 1.0230e-03, 9.2447e-04, 7.0214e-04, 1.0623e-03, 4.6516e-04,
          2.6592e-04, 4.2171e-04, 2.6117e-04, 2.0140e-04, 3.8571e-04, 6.8391e-04,
          2.0890e-04, 1.0493e-04, 2.8365e-04, 3.3232e-04, 8.0910e-04, 5.5881e-04,
          2.0743e-04, 1.6737e-04, 3.2628e-04, 1.7086e-04, 4.3619e-04, 2.8912e-04,
          2.2860e-04, 1.5576e-04, 3.7983e-04, 2.7056e-04, 1.4220e-04, 2.3764e-04,
          2.0924e-04, 1.2657e-04, 3.5853e-04, 7.6382e-04, 2.4748e-04, 3.9013e-04,
          2.3199e-04, 2.2573e-04, 5.8524e-04, 2.4246e-04, 2.1586e-04, 2.6119e-04,
          1.0111e-04, 4.4994e-04, 9.1588e-05, 1.9342e-04, 3.4309e-04, 2.1724e-04,
          1.6826e-04, 1.1293e-03, 3.2779e-04, 1.6531e-04, 4.2406e-04, 1.6534e-04,
          9.9418e-04, 5.9898e-04, 6.0460e-04, 1.6294e-04, 8.8663e-04, 2.4508e-04,
          2.6165e-03, 9.6251e-05, 3.5032e-04, 1.7943e-04, 4.4331e-04, 2.7317e-04,
          1.1727e-04, 7.1055e-05, 5.2733e-05, 1.2715e-04, 3.5418e-04, 1.2659e-04,
          6.3827e-04, 2.6763e-04, 9.2715e-05, 2.0151e-04, 4.6411e-04, 4.0667e-04,
          1.5050e-04, 2.0945e-04, 4.2391e-04, 1.1801e-03, 6.1381e-04, 2.6092e-04,
          3.2495e-04, 1.1779e-04, 4.9670e-05, 8.5088e-05, 6.6986e-05, 2.9470e-04,
          4.7206e-05, 4.4794e-05, 8.1875e-05, 2.2147e-04, 6.0944e-04, 9.9779e-04,
          5.5961e-04, 1.8293e-04, 2.3659e-04, 1.9826e-04, 6.8821e-05, 1.9575e-04,
          2.7925e-04, 2.6616e-04, 3.5473e-04, 4.3095e-04, 9.3177e-04, 2.2859e-04,
          1.4401e-04, 1.6999e-04, 2.5330e-04, 1.2693e-04, 1.1100e-03, 3.1386e-04,
          1.2211e-04, 1.4504e-04, 9.4892e-05, 2.2153e-04, 4.5841e-04, 1.6269e-04,
          1.9385e-04, 1.1468e-04, 2.0212e-04, 1.2334e-04, 5.6796e-04, 5.3492e-04,
          1.9614e-04, 8.7095e-04, 4.4796e-04, 4.7899e-04, 9.4437e-05, 1.3899e-04,
          1.6191e-04, 7.2428e-05, 6.8920e-05, 2.7837e-04, 8.2154e-04, 6.6855e-05,
          6.0044e-05, 2.0067e-04, 1.8293e-04, 6.4519e-04, 3.0010e-04, 3.0557e-04,
          3.7227e-04, 2.2444e-04, 8.7540e-04, 1.6664e-04, 1.7920e-04, 1.2072e-04,
          1.8490e-04, 2.0434e-04, 4.1192e-04, 2.7412e-04, 1.4256e-03, 3.7010e-04,
          1.0384e-04, 4.0003e-04, 3.8594e-04, 1.9334e-04, 9.1531e-05, 1.6476e-04,
          2.6514e-04, 8.0804e-05, 8.5703e-04, 5.4871e-04, 3.5303e-04, 4.4951e-04,
          1.1536e-03, 5.1881e-05, 2.9081e-04, 1.7934e-03, 3.5977e-04, 5.4299e-04,
          9.5877e-05, 4.8924e-04, 2.1003e-04, 2.0400e-04, 9.0676e-04, 1.9863e-04,
          2.8336e-04, 2.5281e-04, 3.6081e-04, 1.8291e-04, 3.1815e-04, 3.8958e-04,
          6.8811e-05, 6.1120e-04, 2.1186e-04, 2.1937e-04, 2.5127e-04, 1.4577e-04,
          3.8199e-04, 2.3699e-04, 1.5524e-04, 1.4927e-04, 1.2502e-04, 3.9728e-04,
          1.3382e-04, 1.1477e-04, 7.0574e-04, 6.8766e-04, 1.3475e-04, 8.3511e-04,
          1.9183e-04, 1.2302e-04, 2.6622e-04, 3.1808e-04, 1.7854e-03, 3.9139e-04,
          5.0411e-04, 2.8310e-04, 3.4985e-04, 2.5098e-04, 1.0711e-04, 2.9261e-04,
          8.3960e-05, 3.2284e-04, 2.4991e-04, 7.3996e-04, 2.2538e-04, 1.3990e-04,
          1.0301e-04, 1.2580e-04, 1.3637e-04, 2.0873e-04, 1.2807e-04, 4.3750e-04,
          4.4813e-04, 1.6818e-04, 2.9523e-04, 1.6918e-03, 7.0121e-04, 4.0012e-03,
          9.4421e-03, 2.3547e-02, 1.8002e-02, 1.5052e-02, 3.3740e-02, 1.2836e-01,
          3.3576e-02, 8.4985e-03, 1.3387e-02, 5.4478e-03, 1.3300e-02, 1.2014e-02,
          2.6997e-02, 5.3274e-02, 7.9752e-02, 7.9804e-02, 1.3109e-01, 1.0764e-01,
          0.0000e+00]),
  tensor([0.0000e+00, 6.3286e-04, 2.8344e-02, 7.4669e-04, 4.9134e-03, 4.7843e-03,
          2.6535e-02, 1.9334e-03, 1.8244e-04, 9.2724e-05, 7.0789e-04, 1.2057e-03,
          3.3101e-04, 2.6285e-04, 1.5653e-04, 5.6781e-04, 1.1955e-04, 3.8828e-04,
          2.0950e-04, 6.9085e-05, 1.1896e-04, 1.9087e-04, 1.1603e-04, 4.4188e-04,
          1.0924e-03, 7.6542e-04, 1.6830e-03, 1.8126e-03, 1.7481e-04, 3.6879e-04,
          1.9885e-04, 3.2965e-04, 2.1268e-04, 8.9330e-04, 5.5135e-04, 2.3684e-04,
          9.4243e-04, 2.7256e-04, 3.2533e-03, 7.9451e-04, 1.5562e-04, 1.0822e-03,
          2.2303e-04, 1.8382e-04, 1.2666e-04, 1.4170e-04, 2.6427e-04, 1.2553e-03,
          1.0828e-04, 1.4891e-04, 8.3338e-04, 4.0488e-04, 2.4312e-04, 1.9322e-04,
          7.9144e-05, 1.4054e-04, 1.0988e-04, 6.6440e-04, 6.9790e-04, 6.2720e-04,
          4.0283e-04, 2.8995e-04, 1.3689e-04, 4.1698e-04, 2.8359e-04, 1.3595e-03,
          8.5284e-04, 2.4847e-04, 3.2513e-04, 9.2235e-05, 6.7095e-05, 8.6156e-05,
          9.6391e-05, 1.3202e-03, 1.2071e-04, 2.8115e-04, 5.8285e-05, 6.7980e-05,
          1.6563e-04, 8.7252e-04, 5.4744e-04, 7.6433e-04, 5.2643e-04, 2.5936e-04,
          1.7386e-04, 3.3449e-04, 9.9881e-05, 3.2552e-04, 3.6449e-04, 2.5871e-04,
          6.3532e-04, 8.0350e-04, 7.1280e-05, 4.4557e-04, 7.4039e-05, 1.0229e-04,
          1.8254e-04, 7.2930e-05, 2.8467e-04, 1.1377e-03, 8.4251e-04, 8.3980e-04,
          8.2625e-04, 8.6104e-04, 3.5923e-04, 1.0415e-03, 1.2717e-03, 1.7222e-04,
          2.4256e-04, 2.7272e-04, 9.0554e-04, 1.7313e-04, 9.5809e-04, 2.7046e-04,
          6.8124e-04, 3.1346e-04, 7.4373e-05, 1.3666e-03, 1.1819e-04, 1.1925e-04,
          1.7263e-04, 9.7686e-04, 1.5566e-04, 9.2376e-05, 1.0453e-03, 1.8048e-04,
          7.8395e-04, 3.7661e-04, 3.8920e-04, 9.0771e-04, 2.1728e-04, 2.4501e-04,
          8.9501e-04, 2.8641e-04, 1.1064e-04, 8.9264e-05, 4.2349e-04, 8.6318e-05,
          5.9470e-04, 8.9815e-05, 2.1756e-04, 5.7217e-04, 8.6432e-04, 1.7940e-04,
          2.7702e-04, 1.6701e-04, 2.8885e-04, 4.2887e-04, 5.4545e-04, 1.2938e-03,
          1.3899e-03, 7.6891e-04, 2.8955e-04, 3.7188e-04, 1.8005e-04, 1.0182e-03,
          7.2527e-04, 7.6923e-04, 8.2132e-05, 1.6295e-04, 9.3479e-04, 1.7122e-04,
          9.4348e-04, 1.1085e-03, 3.0806e-04, 7.3285e-04, 8.7663e-04, 6.0036e-04,
          4.5730e-04, 5.7535e-04, 5.0316e-04, 2.7086e-04, 2.2273e-04, 9.9034e-04,
          2.6237e-04, 1.9463e-04, 4.0748e-04, 4.5529e-04, 1.0515e-03, 3.1951e-04,
          4.5861e-04, 1.8048e-04, 5.6917e-04, 1.3622e-04, 1.3977e-04, 2.8330e-04,
          2.5503e-04, 1.1002e-04, 3.9179e-04, 3.7779e-04, 3.5387e-04, 3.6615e-04,
          1.7562e-04, 1.7665e-04, 5.0252e-04, 3.0097e-04, 4.0413e-04, 6.2347e-04,
          3.4528e-04, 1.3307e-04, 6.0570e-04, 3.4914e-04, 3.9676e-04, 4.8174e-04,
          1.2912e-04, 5.6122e-04, 1.1931e-04, 1.3258e-04, 4.7179e-04, 1.3637e-04,
          1.0008e-04, 7.7976e-04, 5.3888e-04, 4.2901e-04, 9.1829e-04, 2.5887e-04,
          7.5225e-04, 3.2573e-04, 6.1592e-04, 2.1353e-04, 8.1801e-04, 1.6942e-04,
          1.6241e-03, 1.6208e-04, 4.8965e-04, 3.8499e-04, 8.7303e-04, 3.3915e-04,
          1.8622e-04, 1.1603e-04, 6.1861e-05, 1.8245e-04, 3.5812e-04, 7.5868e-05,
          7.5945e-04, 3.6356e-04, 4.4100e-04, 3.5319e-04, 4.5444e-04, 5.7086e-04,
          2.1990e-04, 2.8375e-04, 4.3559e-04, 6.6483e-04, 3.3047e-04, 1.6176e-04,
          4.5697e-04, 1.8994e-04, 7.4990e-05, 9.5033e-05, 9.4127e-05, 5.5241e-04,
          6.3940e-05, 5.1166e-05, 8.5430e-05, 2.2791e-04, 3.6097e-04, 1.0275e-03,
          5.0502e-04, 3.3720e-04, 4.5696e-04, 3.6177e-04, 7.8180e-05, 2.7127e-04,
          3.5114e-04, 2.5893e-04, 1.7022e-04, 2.9986e-04, 3.4589e-04, 3.2005e-04,
          3.1638e-04, 2.3300e-04, 5.4394e-04, 1.8524e-04, 9.1177e-04, 5.2430e-04,
          1.3551e-04, 1.2634e-04, 7.6206e-05, 1.2928e-04, 4.2528e-04, 3.8348e-04,
          7.0151e-04, 1.9917e-04, 2.0588e-04, 1.3909e-04, 4.8881e-04, 4.8068e-04,
          2.4165e-04, 1.1843e-03, 2.0883e-04, 2.0863e-04, 1.2491e-04, 1.7951e-04,
          2.1743e-04, 1.1378e-04, 9.6332e-05, 5.8714e-04, 8.9435e-04, 7.8429e-05,
          7.1948e-05, 1.4046e-04, 1.3270e-04, 3.7904e-04, 3.3974e-04, 6.3742e-04,
          7.6417e-04, 1.1866e-04, 6.8809e-04, 8.6310e-05, 1.7217e-04, 1.2606e-04,
          1.1050e-04, 2.9065e-04, 2.7286e-04, 2.7143e-04, 1.2306e-03, 4.4224e-04,
          1.3827e-04, 5.4372e-04, 2.6249e-04, 1.0087e-04, 1.0355e-04, 1.7259e-04,
          1.1584e-04, 5.8742e-05, 4.2463e-04, 4.4790e-04, 4.6862e-04, 2.1014e-04,
          1.4108e-03, 3.4461e-05, 1.6348e-04, 2.0685e-03, 3.5081e-04, 1.8719e-04,
          1.3023e-04, 4.7983e-04, 1.7782e-04, 1.3652e-04, 1.1159e-03, 1.2815e-04,
          3.2279e-04, 3.0963e-04, 1.5199e-04, 1.6591e-04, 3.2981e-04, 2.6399e-04,
          5.2005e-05, 4.5939e-04, 2.4483e-04, 3.0518e-04, 2.3487e-04, 1.1954e-04,
          4.8615e-04, 1.8268e-04, 2.0209e-04, 1.8560e-04, 8.3022e-05, 3.9276e-04,
          1.2296e-04, 9.0415e-05, 4.5763e-04, 2.9408e-04, 1.1905e-04, 1.0830e-03,
          1.8653e-04, 7.8334e-05, 2.0160e-04, 3.7564e-04, 1.9189e-03, 4.4163e-04,
          5.4364e-04, 2.9713e-04, 3.6541e-04, 1.7233e-04, 1.0446e-04, 1.8553e-04,
          7.0259e-05, 1.7320e-04, 3.4709e-04, 3.7814e-04, 1.4184e-04, 2.6858e-04,
          5.8869e-05, 1.0857e-04, 8.4274e-05, 1.1228e-04, 3.3906e-04, 1.2267e-03,
          3.5784e-04, 1.5962e-04, 2.5540e-04, 1.4652e-03, 7.5597e-04, 4.8135e-03,
          1.2215e-02, 2.3541e-02, 1.9255e-02, 1.6958e-02, 1.0263e-02, 7.8072e-02,
          5.6227e-02, 1.0312e-02, 1.8119e-02, 6.7988e-03, 1.4151e-02, 1.4261e-02,
          3.1512e-02, 6.6336e-02, 6.9060e-02, 7.2894e-02, 7.9846e-02, 4.4743e-02,
          1.2309e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.9693e-04, 2.2647e-02, 5.3573e-04, 4.7362e-03, 4.0698e-03,
          2.2828e-02, 2.2536e-03, 2.4892e-04, 8.7895e-05, 9.6114e-04, 2.1132e-03,
          4.6408e-04, 2.8292e-04, 2.2627e-04, 5.8962e-04, 1.6090e-04, 3.5472e-04,
          1.8693e-04, 8.3841e-05, 6.8733e-05, 7.7609e-05, 6.0508e-05, 4.7188e-04,
          1.8148e-03, 9.0462e-04, 1.7508e-03, 1.4314e-03, 1.0324e-04, 2.5306e-04,
          2.9809e-04, 3.9192e-04, 8.1624e-05, 7.1688e-04, 4.0593e-04, 2.1005e-04,
          1.0875e-03, 3.2666e-04, 8.2403e-04, 5.2926e-04, 9.7445e-05, 4.7183e-04,
          2.4163e-04, 2.2228e-04, 2.0092e-04, 1.2255e-04, 2.2440e-04, 8.4359e-04,
          8.4777e-05, 9.3319e-05, 5.8336e-04, 3.1637e-04, 2.5827e-04, 1.8683e-04,
          1.0209e-04, 1.3196e-04, 1.3057e-04, 5.6205e-04, 1.1497e-03, 8.9678e-04,
          6.3399e-04, 1.4820e-04, 1.4251e-04, 2.9416e-04, 2.9944e-04, 1.6795e-03,
          2.7199e-03, 8.4012e-04, 3.5257e-04, 9.5824e-05, 7.7179e-05, 1.1863e-04,
          9.4477e-05, 1.4762e-03, 1.4063e-04, 2.9812e-04, 7.5034e-05, 7.4351e-05,
          1.7820e-04, 6.2898e-04, 2.5887e-03, 1.4395e-03, 2.3611e-03, 3.3300e-04,
          2.1972e-04, 2.9456e-04, 9.7722e-05, 5.5426e-04, 1.6079e-03, 1.2326e-03,
          1.0875e-03, 5.6768e-04, 1.9241e-04, 7.0924e-04, 7.6351e-05, 1.0270e-04,
          1.0813e-04, 8.5119e-05, 2.6797e-04, 9.2541e-04, 9.0058e-04, 6.9941e-04,
          1.5473e-03, 1.8087e-03, 1.1983e-03, 1.4604e-03, 1.8596e-03, 9.5773e-05,
          2.1004e-04, 1.9293e-04, 6.2530e-04, 3.2650e-04, 2.5056e-03, 6.0385e-04,
          1.0301e-03, 2.5551e-04, 3.9690e-05, 1.2918e-03, 1.2349e-04, 8.6277e-05,
          1.5201e-04, 1.2344e-03, 4.0676e-04, 1.2883e-04, 1.4850e-03, 5.7103e-04,
          3.4470e-04, 1.2019e-03, 1.1954e-03, 1.0671e-03, 1.9301e-04, 2.3542e-04,
          1.7858e-03, 1.1136e-03, 1.7777e-04, 2.2770e-04, 4.2162e-04, 1.4646e-04,
          4.4177e-04, 7.8982e-05, 2.3997e-04, 5.8945e-04, 7.9191e-04, 1.7358e-04,
          7.7278e-04, 2.3953e-04, 5.8914e-04, 5.7579e-04, 2.9937e-03, 2.4683e-03,
          2.4912e-03, 8.3097e-04, 3.2267e-04, 2.1354e-04, 1.9896e-04, 4.6693e-04,
          3.8031e-04, 3.3994e-04, 1.5821e-04, 3.4198e-04, 7.9033e-04, 1.1288e-04,
          7.1609e-04, 5.4319e-04, 2.5619e-04, 8.6957e-04, 6.5859e-04, 6.7385e-04,
          8.3713e-04, 5.2838e-04, 7.4930e-04, 3.2252e-04, 2.2587e-04, 6.1478e-04,
          2.4786e-04, 2.9528e-04, 4.8826e-04, 3.1443e-04, 6.9325e-04, 3.5684e-04,
          6.3120e-04, 1.6369e-04, 7.4813e-04, 2.0098e-04, 1.6759e-04, 3.3036e-04,
          4.0659e-04, 1.1844e-04, 5.7079e-04, 5.6002e-04, 1.7532e-03, 4.0978e-04,
          1.8978e-04, 1.7205e-04, 3.8841e-04, 2.4394e-04, 5.7824e-04, 4.4823e-04,
          3.6152e-04, 6.6332e-05, 3.7222e-04, 3.5766e-04, 6.3474e-04, 3.5720e-04,
          3.0614e-04, 3.0599e-04, 8.7614e-05, 1.2947e-04, 7.3954e-04, 7.3086e-05,
          1.3060e-04, 5.9312e-04, 8.5819e-04, 1.4565e-03, 1.1351e-03, 9.4590e-04,
          5.5653e-04, 2.2153e-04, 7.3437e-04, 2.4718e-04, 5.3931e-04, 1.6805e-04,
          5.5430e-04, 1.3016e-04, 2.3567e-04, 9.8150e-04, 1.0950e-03, 3.2431e-04,
          6.2775e-04, 2.1499e-04, 6.4049e-05, 1.7833e-04, 3.6137e-04, 6.0325e-05,
          1.1985e-03, 7.0613e-04, 1.2387e-03, 4.4416e-04, 8.1461e-04, 5.6005e-04,
          5.0254e-04, 4.5273e-04, 3.2807e-04, 3.0969e-04, 2.5954e-04, 1.2445e-04,
          1.9823e-04, 4.1197e-04, 9.8808e-05, 1.0846e-04, 1.1185e-04, 4.4922e-04,
          1.2085e-04, 4.4708e-05, 8.9021e-05, 1.6043e-04, 3.1102e-04, 8.6136e-04,
          5.5902e-04, 9.1583e-04, 6.4306e-04, 4.3155e-04, 1.0427e-04, 5.2125e-04,
          3.6981e-04, 2.9079e-04, 1.9074e-04, 3.0887e-04, 7.4862e-04, 2.2419e-04,
          7.2756e-04, 8.3405e-04, 6.6319e-04, 1.4663e-04, 4.2517e-04, 6.9969e-04,
          1.1926e-04, 1.1737e-04, 4.5641e-05, 1.5798e-04, 3.0281e-04, 1.4011e-03,
          2.8397e-03, 2.9315e-04, 2.8723e-04, 1.9283e-04, 4.7903e-04, 4.6140e-04,
          2.5968e-04, 8.4224e-04, 1.5314e-04, 1.8054e-04, 1.2559e-04, 1.8939e-04,
          1.6454e-04, 1.0637e-04, 9.6794e-05, 4.3623e-04, 3.3993e-04, 9.5644e-05,
          8.5074e-05, 1.1790e-04, 1.4837e-04, 2.7220e-04, 6.2243e-04, 6.8429e-04,
          5.2932e-04, 7.4414e-05, 4.5611e-04, 8.7306e-05, 2.2474e-04, 1.8087e-04,
          9.0210e-05, 3.5770e-04, 1.3186e-04, 2.2647e-04, 7.4589e-04, 3.1692e-04,
          1.5677e-04, 8.8654e-04, 1.4297e-04, 6.3705e-05, 1.1216e-04, 1.6954e-04,
          1.0582e-04, 5.4418e-05, 5.0904e-04, 5.1116e-04, 5.8544e-04, 2.7680e-04,
          9.3428e-04, 2.7535e-05, 1.7455e-04, 9.2755e-04, 5.6681e-04, 1.3318e-04,
          1.2676e-04, 4.5225e-04, 1.8640e-04, 8.5729e-05, 4.9409e-04, 1.4463e-04,
          5.3883e-04, 3.8729e-04, 9.5506e-05, 3.1937e-04, 3.2117e-04, 1.4456e-04,
          4.1269e-05, 5.1284e-04, 3.1135e-04, 4.0776e-04, 2.4739e-04, 1.0263e-04,
          3.7540e-04, 2.3643e-04, 2.2988e-04, 2.1033e-04, 9.0627e-05, 4.9704e-04,
          1.0379e-04, 8.9055e-05, 4.8246e-04, 4.7033e-04, 1.3136e-04, 8.3246e-04,
          2.2956e-04, 5.9252e-05, 1.8225e-04, 3.6606e-04, 1.5056e-03, 5.8092e-04,
          4.6636e-04, 3.8039e-04, 5.2132e-04, 2.0028e-04, 7.0941e-05, 1.3507e-04,
          4.4625e-05, 2.7729e-04, 4.4681e-04, 3.8243e-04, 1.2953e-04, 3.4309e-04,
          8.5492e-05, 8.3276e-05, 9.4735e-05, 1.3363e-04, 2.7379e-04, 1.2327e-03,
          2.5730e-04, 1.2274e-04, 2.6663e-04, 5.5855e-04, 5.9781e-04, 3.3433e-03,
          2.0108e-02, 2.4625e-02, 2.3767e-02, 2.1388e-02, 6.9276e-03, 2.6863e-02,
          5.4393e-02, 1.2100e-02, 2.0605e-02, 5.0308e-03, 1.5178e-02, 1.4513e-02,
          3.2578e-02, 6.7738e-02, 6.2118e-02, 6.1771e-02, 4.2208e-02, 2.0408e-02,
          9.3147e-02, 1.2580e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.8029e-04, 2.3704e-02, 5.7186e-04, 5.0922e-03, 3.7940e-03,
          2.5015e-02, 2.2238e-03, 3.5003e-04, 1.4214e-04, 1.5627e-03, 3.5431e-03,
          8.6810e-04, 3.0572e-04, 2.8047e-04, 4.0468e-04, 2.2382e-04, 3.9701e-04,
          2.6650e-04, 1.2135e-04, 6.1949e-05, 6.0772e-05, 6.7615e-05, 5.0533e-04,
          3.3458e-03, 5.2904e-04, 1.8180e-03, 1.2689e-03, 1.0508e-04, 3.1718e-04,
          3.6248e-04, 5.9249e-04, 1.1323e-04, 6.3235e-04, 6.3793e-04, 1.8003e-04,
          6.0163e-04, 3.6083e-04, 7.3409e-04, 5.4402e-04, 1.0008e-04, 5.5406e-04,
          3.2597e-04, 2.7662e-04, 2.6253e-04, 1.4566e-04, 2.8199e-04, 5.1637e-04,
          1.2097e-04, 9.2988e-05, 6.2147e-04, 3.7196e-04, 4.2189e-04, 2.0560e-04,
          1.3569e-04, 2.1320e-04, 1.2735e-04, 4.9804e-04, 1.6435e-03, 1.1111e-03,
          6.9290e-04, 1.7144e-04, 1.7992e-04, 3.1724e-04, 5.8279e-04, 8.8916e-04,
          4.0119e-03, 1.6085e-03, 5.5564e-04, 1.1049e-04, 9.0839e-05, 1.6846e-04,
          1.0469e-04, 2.0185e-03, 1.6686e-04, 5.1221e-04, 1.2181e-04, 1.1706e-04,
          2.5807e-04, 5.2536e-04, 3.5486e-03, 1.4546e-03, 5.1732e-03, 3.2052e-04,
          1.8889e-04, 4.2722e-04, 1.3208e-04, 1.0876e-03, 2.3855e-03, 1.5747e-03,
          2.4506e-03, 8.8245e-04, 2.3896e-04, 1.4231e-03, 6.9685e-05, 9.0421e-05,
          1.1391e-04, 8.1134e-05, 2.7928e-04, 4.8032e-04, 7.8325e-04, 7.4082e-04,
          2.9943e-03, 2.5036e-03, 3.0869e-03, 9.5058e-04, 9.7454e-04, 1.0301e-04,
          3.1739e-04, 2.2123e-04, 1.0676e-03, 4.1830e-04, 2.8014e-03, 9.6429e-04,
          7.1934e-04, 1.4728e-04, 3.4125e-05, 6.3995e-04, 1.1252e-04, 9.5665e-05,
          1.8703e-04, 1.0416e-03, 3.7951e-04, 2.0878e-04, 9.8918e-04, 7.6240e-04,
          3.1037e-04, 1.7575e-03, 1.9203e-03, 6.0975e-04, 2.7211e-04, 3.8895e-04,
          2.6088e-03, 2.3905e-03, 1.8715e-04, 4.6478e-04, 9.2714e-04, 1.5862e-04,
          3.3641e-04, 8.4282e-05, 2.5894e-04, 6.1624e-04, 6.4987e-04, 2.0508e-04,
          7.2227e-04, 1.9606e-04, 6.2840e-04, 6.6039e-04, 3.9175e-03, 1.2833e-03,
          3.3800e-03, 5.2009e-04, 2.8087e-04, 2.2137e-04, 2.6443e-04, 3.3216e-04,
          6.2264e-04, 4.0777e-04, 2.9624e-04, 4.6377e-04, 4.4237e-04, 1.1182e-04,
          3.5131e-04, 4.9147e-04, 2.9217e-04, 6.4730e-04, 4.0664e-04, 7.6334e-04,
          1.5246e-03, 6.6334e-04, 4.9710e-04, 2.4438e-04, 2.0954e-04, 3.7114e-04,
          2.4273e-04, 3.6406e-04, 3.7228e-04, 2.7726e-04, 4.5240e-04, 5.4565e-04,
          3.7803e-04, 1.7722e-04, 5.4165e-04, 2.7953e-04, 2.5511e-04, 2.3110e-04,
          8.3725e-04, 1.2492e-04, 6.2998e-04, 4.4515e-04, 2.5187e-03, 3.3833e-04,
          3.2170e-04, 1.8375e-04, 2.6676e-04, 2.8464e-04, 1.0461e-03, 3.3772e-04,
          3.3487e-04, 6.5122e-05, 3.3615e-04, 7.5416e-04, 1.2051e-03, 2.6776e-04,
          5.2692e-04, 6.2036e-04, 8.6595e-05, 1.9463e-04, 1.4665e-03, 7.7044e-05,
          2.0161e-04, 3.2791e-04, 6.1717e-04, 3.1338e-03, 6.2164e-04, 1.0763e-03,
          3.7689e-04, 1.9152e-04, 1.1623e-03, 1.9281e-04, 3.9260e-04, 2.5595e-04,
          4.4666e-04, 1.2346e-04, 2.9269e-04, 1.5800e-03, 7.5963e-04, 3.2974e-04,
          1.6600e-03, 3.4044e-04, 8.3801e-05, 1.7364e-04, 3.0067e-04, 7.5269e-05,
          8.3457e-04, 9.0119e-04, 1.5623e-03, 3.5418e-04, 6.0872e-04, 6.0032e-04,
          6.4152e-04, 4.5665e-04, 3.1265e-04, 2.3279e-04, 2.6441e-04, 1.2684e-04,
          1.3473e-04, 7.5920e-04, 1.0407e-04, 1.4595e-04, 1.2149e-04, 3.5993e-04,
          1.6648e-04, 4.1439e-05, 1.0382e-04, 1.4025e-04, 2.9884e-04, 6.2513e-04,
          3.6457e-04, 1.2780e-03, 4.2139e-04, 2.7428e-04, 1.0665e-04, 6.9383e-04,
          2.3917e-04, 3.4543e-04, 2.0566e-04, 3.2528e-04, 9.2002e-04, 1.8538e-04,
          1.0862e-03, 2.0429e-03, 4.4215e-04, 1.4517e-04, 2.8877e-04, 4.9045e-04,
          1.2167e-04, 1.0990e-04, 3.9678e-05, 2.0137e-04, 2.0245e-04, 1.8789e-03,
          5.2510e-03, 2.3836e-04, 2.2712e-04, 1.6457e-04, 2.7201e-04, 2.1761e-04,
          2.8171e-04, 6.4293e-04, 1.5490e-04, 2.2158e-04, 1.1756e-04, 2.5144e-04,
          1.5535e-04, 1.3682e-04, 9.9378e-05, 2.3252e-04, 3.5873e-04, 1.0505e-04,
          8.0935e-05, 1.4116e-04, 2.1357e-04, 2.0153e-04, 5.0479e-04, 5.9041e-04,
          2.7580e-04, 6.2444e-05, 3.4663e-04, 8.7285e-05, 2.3017e-04, 2.6979e-04,
          7.2395e-05, 1.8278e-04, 1.0571e-04, 3.4101e-04, 5.1783e-04, 2.8073e-04,
          1.4727e-04, 1.5605e-03, 1.0457e-04, 8.0415e-05, 8.5001e-05, 1.1629e-04,
          1.1389e-04, 6.6582e-05, 5.7877e-04, 3.1750e-04, 4.5262e-04, 3.2546e-04,
          1.0546e-03, 3.8150e-05, 2.2898e-04, 1.2328e-03, 8.3327e-04, 1.2530e-04,
          1.0670e-04, 3.9112e-04, 1.5971e-04, 7.9294e-05, 7.9230e-04, 2.0218e-04,
          6.7912e-04, 3.4703e-04, 1.0338e-04, 4.5805e-04, 2.7381e-04, 1.1500e-04,
          4.7447e-05, 5.9206e-04, 2.6242e-04, 3.0111e-04, 3.4504e-04, 1.7701e-04,
          2.6078e-04, 3.0580e-04, 2.4106e-04, 1.7937e-04, 1.2744e-04, 2.3285e-04,
          1.3748e-04, 1.4276e-04, 2.5709e-04, 4.9023e-04, 1.5757e-04, 8.9405e-04,
          2.3431e-04, 7.6940e-05, 2.5261e-04, 5.6950e-04, 1.3634e-03, 4.6902e-04,
          6.2157e-04, 2.7901e-04, 4.2769e-04, 2.3858e-04, 7.7094e-05, 1.1824e-04,
          6.3633e-05, 4.1168e-04, 6.5679e-04, 4.8840e-04, 1.1856e-04, 1.8560e-04,
          1.1759e-04, 8.1303e-05, 9.3594e-05, 1.9174e-04, 2.1427e-04, 4.9653e-04,
          1.7976e-04, 9.0349e-05, 2.7227e-04, 5.1195e-04, 7.2869e-04, 3.0969e-03,
          1.2214e-02, 2.1946e-02, 3.4249e-02, 1.4290e-02, 5.8906e-03, 2.1785e-02,
          3.2364e-02, 9.2108e-03, 1.3527e-02, 3.9889e-03, 9.9190e-03, 1.3239e-02,
          2.2894e-02, 4.6142e-02, 4.7495e-02, 6.3423e-02, 2.8093e-02, 1.4253e-02,
          5.9657e-02, 1.1686e-01, 1.2694e-01, 0.0000e+00]),
  tensor([0.0000e+00, 7.2978e-04, 3.2048e-02, 8.9254e-04, 5.3547e-03, 4.9462e-03,
          3.5940e-02, 1.8575e-03, 4.8233e-04, 3.2828e-04, 1.8475e-03, 5.8225e-03,
          1.5733e-03, 5.4251e-04, 3.6338e-04, 4.5019e-04, 3.4835e-04, 5.5556e-04,
          5.2081e-04, 2.2522e-04, 5.5971e-05, 4.5263e-05, 6.0443e-05, 7.2481e-04,
          5.7321e-03, 6.0221e-04, 1.7327e-03, 1.0919e-03, 1.1272e-04, 6.3966e-04,
          5.0810e-04, 1.0421e-03, 2.4813e-04, 5.6569e-04, 1.0525e-03, 1.9118e-04,
          4.9261e-04, 3.5904e-04, 3.8159e-04, 5.2128e-04, 1.0407e-04, 3.5173e-04,
          2.9223e-04, 2.3334e-04, 4.2234e-04, 1.1967e-04, 6.2437e-04, 3.9172e-04,
          1.7902e-04, 8.5294e-05, 6.4121e-04, 3.6424e-04, 8.5139e-04, 2.8661e-04,
          2.3064e-04, 5.0970e-04, 1.3502e-04, 4.2411e-04, 1.7815e-03, 1.3681e-03,
          8.4009e-04, 2.1886e-04, 2.4337e-04, 2.4244e-04, 6.5627e-04, 6.7893e-04,
          7.1380e-03, 2.0541e-03, 3.7035e-04, 1.0259e-04, 1.0249e-04, 2.8666e-04,
          1.3098e-04, 1.8667e-03, 1.9078e-04, 1.1455e-03, 3.2638e-04, 2.1778e-04,
          4.1928e-04, 3.8581e-04, 3.5217e-03, 1.0555e-03, 5.7421e-03, 4.1170e-04,
          1.6571e-04, 8.6000e-04, 2.3838e-04, 1.5453e-03, 3.3279e-03, 2.9476e-03,
          4.2608e-03, 8.2396e-04, 2.4424e-04, 2.0881e-03, 8.9599e-05, 1.0015e-04,
          1.1916e-04, 8.1412e-05, 2.9064e-04, 2.8600e-04, 5.6576e-04, 5.7930e-04,
          4.4145e-03, 3.8447e-03, 4.0939e-03, 7.5404e-04, 7.0048e-04, 1.1920e-04,
          6.8446e-04, 2.3950e-04, 1.1049e-03, 7.0015e-04, 6.0652e-03, 1.1606e-03,
          5.3136e-04, 1.1707e-04, 3.7942e-05, 6.2373e-04, 1.2856e-04, 1.1911e-04,
          2.4562e-04, 4.7989e-04, 4.3675e-04, 3.0718e-04, 7.6364e-04, 7.2344e-04,
          1.9368e-04, 1.8017e-03, 2.8302e-03, 6.8690e-04, 4.7354e-04, 9.3076e-04,
          5.7165e-03, 4.2797e-03, 3.4838e-04, 6.1982e-04, 1.2071e-03, 1.6187e-04,
          3.6491e-04, 1.0883e-04, 3.4439e-04, 6.1216e-04, 5.5634e-04, 3.0747e-04,
          6.7692e-04, 3.2898e-04, 3.5448e-04, 4.9104e-04, 5.5980e-03, 8.4780e-04,
          1.7649e-03, 6.6944e-04, 3.7602e-04, 2.0885e-04, 3.1135e-04, 2.5876e-04,
          6.4730e-04, 2.8429e-04, 6.8534e-04, 7.3370e-04, 3.2868e-04, 1.3573e-04,
          2.5354e-04, 2.6958e-04, 5.4167e-04, 3.8731e-04, 1.9442e-04, 3.1289e-04,
          1.9778e-03, 3.7377e-04, 3.9669e-04, 3.2970e-04, 2.7308e-04, 2.7129e-04,
          5.9479e-04, 5.8082e-04, 2.9518e-04, 2.1042e-04, 3.3682e-04, 9.2516e-04,
          1.8121e-04, 2.8446e-04, 3.7937e-04, 4.2624e-04, 2.8975e-04, 2.0427e-04,
          1.5547e-03, 1.5511e-04, 4.6689e-04, 3.4690e-04, 3.9545e-03, 3.8174e-04,
          6.3805e-04, 2.0168e-04, 2.3254e-04, 4.9513e-04, 1.1869e-03, 3.9207e-04,
          3.5649e-04, 7.6367e-05, 3.5545e-04, 1.9019e-03, 1.1715e-03, 1.7344e-04,
          1.6341e-03, 7.5392e-04, 1.0760e-04, 4.2226e-04, 1.8053e-03, 1.4591e-04,
          3.0049e-04, 1.6444e-04, 5.3070e-04, 5.9401e-03, 5.3796e-04, 1.2870e-03,
          2.2265e-04, 2.3548e-04, 1.2886e-03, 2.6018e-04, 3.1590e-04, 4.1083e-04,
          2.6333e-04, 1.6958e-04, 4.3184e-04, 1.5110e-03, 6.8272e-04, 4.7987e-04,
          3.4902e-03, 1.0888e-03, 1.9230e-04, 2.7947e-04, 2.3638e-04, 1.2759e-04,
          4.3281e-04, 6.6140e-04, 3.8085e-03, 3.0140e-04, 3.9779e-04, 1.2252e-03,
          9.3506e-04, 3.8142e-04, 1.8075e-04, 1.4913e-04, 3.4521e-04, 1.4279e-04,
          1.0931e-04, 4.9133e-03, 1.6999e-04, 2.6751e-04, 2.1553e-04, 3.9639e-04,
          4.7490e-04, 6.0877e-05, 1.6013e-04, 1.1655e-04, 4.5233e-04, 3.6777e-04,
          2.1374e-04, 1.9445e-03, 2.5891e-04, 2.4193e-04, 1.7770e-04, 9.8809e-04,
          2.5076e-04, 2.9746e-04, 3.1875e-04, 5.2244e-04, 1.3762e-03, 1.9045e-04,
          3.1363e-03, 3.8046e-03, 4.3697e-04, 1.8040e-04, 1.8343e-04, 3.7345e-04,
          2.2100e-04, 1.7834e-04, 3.8341e-05, 3.6835e-04, 1.4084e-04, 3.2123e-03,
          1.3053e-02, 2.4449e-04, 2.1151e-04, 2.0551e-04, 2.0689e-04, 1.5436e-04,
          5.5526e-04, 3.9608e-04, 2.2782e-04, 4.1353e-04, 1.4068e-04, 4.6921e-04,
          1.7849e-04, 2.6403e-04, 1.4730e-04, 1.9438e-04, 2.3982e-04, 2.0319e-04,
          1.1343e-04, 2.1443e-04, 3.3495e-04, 1.9036e-04, 5.4452e-04, 6.7544e-04,
          2.6043e-04, 5.4926e-05, 1.7554e-04, 1.0817e-04, 2.1727e-04, 5.0157e-04,
          9.1394e-05, 1.8084e-04, 9.3647e-05, 3.1626e-04, 3.6331e-04, 2.9527e-04,
          2.4974e-04, 1.9447e-03, 9.8596e-05, 1.2402e-04, 8.1181e-05, 7.9469e-05,
          1.5882e-04, 1.2456e-04, 6.2964e-04, 2.7133e-04, 3.1773e-04, 5.6084e-04,
          6.1367e-04, 7.2954e-05, 4.2575e-04, 6.8146e-04, 8.0474e-04, 1.1226e-04,
          1.5863e-04, 3.0533e-04, 1.8534e-04, 8.3612e-05, 5.0799e-04, 4.5051e-04,
          4.3929e-04, 2.6917e-04, 1.5993e-04, 3.3312e-04, 2.4369e-04, 7.9917e-05,
          6.1298e-05, 4.8541e-04, 3.2164e-04, 3.7566e-04, 4.9948e-04, 3.5073e-04,
          2.4774e-04, 3.4533e-04, 1.7673e-04, 2.0080e-04, 2.7490e-04, 1.9108e-04,
          2.0519e-04, 2.5944e-04, 1.6803e-04, 5.8035e-04, 1.9426e-04, 5.4312e-04,
          2.1386e-04, 1.1955e-04, 2.9697e-04, 3.5882e-04, 1.0298e-03, 2.8023e-04,
          4.4389e-04, 3.1080e-04, 3.6277e-04, 2.9259e-04, 1.0707e-04, 1.1113e-04,
          1.0036e-04, 6.7906e-04, 5.8888e-04, 5.4969e-04, 1.2506e-04, 1.0360e-04,
          1.2601e-04, 7.7282e-05, 1.6626e-04, 1.9163e-04, 2.5319e-04, 2.7737e-04,
          1.4584e-04, 9.0506e-05, 1.9169e-04, 2.9509e-04, 4.2229e-04, 3.7445e-03,
          6.9048e-03, 2.2529e-02, 3.6315e-02, 6.7420e-03, 3.9564e-03, 1.4272e-02,
          1.4846e-02, 5.2658e-03, 8.2982e-03, 3.4920e-03, 7.2903e-03, 8.6175e-03,
          1.3708e-02, 2.6260e-02, 2.5849e-02, 4.3048e-02, 1.2934e-02, 7.1001e-03,
          2.5195e-02, 4.6229e-02, 1.0532e-01, 1.8813e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.8700e-04, 2.5324e-02, 5.4487e-04, 3.6975e-03, 3.8973e-03,
          2.5402e-02, 1.5034e-03, 3.2061e-04, 1.5535e-04, 1.1639e-03, 4.4474e-03,
          7.9467e-04, 4.5067e-04, 2.7771e-04, 3.9817e-04, 2.2373e-04, 5.1167e-04,
          3.2362e-04, 1.3275e-04, 5.1165e-05, 4.3091e-05, 5.2871e-05, 5.5168e-04,
          4.2311e-03, 4.5041e-04, 1.4315e-03, 7.6311e-04, 8.5427e-05, 4.3179e-04,
          3.5560e-04, 5.7920e-04, 1.3612e-04, 3.6003e-04, 4.3948e-04, 1.6675e-04,
          3.2642e-04, 2.1688e-04, 2.3705e-04, 2.8517e-04, 8.6768e-05, 2.8994e-04,
          2.4141e-04, 2.4002e-04, 2.8591e-04, 1.1757e-04, 4.8704e-04, 3.4665e-04,
          1.2418e-04, 7.3638e-05, 4.4577e-04, 2.2886e-04, 3.9037e-04, 2.0545e-04,
          1.5835e-04, 2.2633e-04, 1.5406e-04, 3.5785e-04, 1.1364e-03, 8.6917e-04,
          5.3028e-04, 1.7317e-04, 2.1120e-04, 1.8980e-04, 5.2123e-04, 7.2058e-04,
          5.4013e-03, 1.3546e-03, 3.2281e-04, 9.8902e-05, 8.3817e-05, 2.6671e-04,
          1.1301e-04, 1.2805e-03, 1.2297e-04, 5.3079e-04, 1.8824e-04, 1.3586e-04,
          2.6471e-04, 2.9308e-04, 2.3082e-03, 7.8099e-04, 3.1760e-03, 2.6052e-04,
          1.4655e-04, 4.6281e-04, 2.0307e-04, 1.0632e-03, 2.0449e-03, 1.9888e-03,
          3.0205e-03, 8.4136e-04, 3.1082e-04, 1.5143e-03, 9.8294e-05, 8.2151e-05,
          1.0224e-04, 6.2566e-05, 1.8886e-04, 3.6061e-04, 5.0540e-04, 5.9994e-04,
          3.1605e-03, 2.5957e-03, 2.2262e-03, 5.0605e-04, 7.4121e-04, 1.1378e-04,
          3.6002e-04, 2.0493e-04, 1.0228e-03, 8.4359e-04, 4.5074e-03, 9.6245e-04,
          5.3256e-04, 1.1619e-04, 3.7762e-05, 6.6708e-04, 9.3910e-05, 8.5134e-05,
          1.9545e-04, 5.4771e-04, 3.5919e-04, 1.9997e-04, 7.1748e-04, 5.9692e-04,
          1.7430e-04, 1.7695e-03, 3.0807e-03, 4.9774e-04, 2.6671e-04, 4.6756e-04,
          6.8516e-03, 2.3563e-03, 2.0997e-04, 6.2822e-04, 1.2792e-03, 2.0555e-04,
          3.7181e-04, 9.8805e-05, 3.3327e-04, 4.8443e-04, 3.7725e-04, 2.0516e-04,
          5.7678e-04, 2.0651e-04, 3.5381e-04, 6.3604e-04, 3.6278e-03, 6.9662e-04,
          1.7890e-03, 4.0704e-04, 2.8185e-04, 1.7145e-04, 2.5414e-04, 2.3777e-04,
          7.3069e-04, 3.7814e-04, 3.6990e-04, 7.3000e-04, 3.1910e-04, 1.2025e-04,
          2.4441e-04, 2.6551e-04, 4.4321e-04, 3.8308e-04, 1.6562e-04, 3.8243e-04,
          1.3991e-03, 3.5286e-04, 3.0332e-04, 2.4504e-04, 2.0325e-04, 1.9853e-04,
          3.8686e-04, 4.8428e-04, 2.1284e-04, 1.5316e-04, 2.9330e-04, 5.7369e-04,
          2.1470e-04, 1.6205e-04, 3.6459e-04, 2.6532e-04, 2.8743e-04, 2.1854e-04,
          8.9823e-04, 1.3023e-04, 4.3643e-04, 2.7456e-04, 2.7602e-03, 2.6729e-04,
          3.2900e-04, 1.5525e-04, 1.9425e-04, 4.1444e-04, 9.0142e-04, 2.2779e-04,
          3.1023e-04, 8.6922e-05, 2.1818e-04, 1.0013e-03, 8.6580e-04, 1.4973e-04,
          1.0193e-03, 7.6801e-04, 8.9015e-05, 2.5379e-04, 1.1830e-03, 9.3232e-05,
          2.3152e-04, 1.3761e-04, 4.2042e-04, 3.4913e-03, 3.8439e-04, 1.1416e-03,
          2.0013e-04, 1.7694e-04, 9.0871e-04, 1.9580e-04, 2.3146e-04, 3.4083e-04,
          2.6241e-04, 1.2826e-04, 3.0678e-04, 1.5082e-03, 5.7634e-04, 2.9200e-04,
          2.0506e-03, 1.3195e-03, 1.4753e-04, 2.5094e-04, 2.1401e-04, 1.1270e-04,
          3.9769e-04, 7.1251e-04, 4.2304e-03, 2.2635e-04, 3.9558e-04, 8.0519e-04,
          7.3015e-04, 4.3963e-04, 2.0956e-04, 1.4436e-04, 3.5083e-04, 1.4892e-04,
          7.9091e-05, 3.4323e-03, 1.6085e-04, 2.1834e-04, 2.3135e-04, 3.2769e-04,
          9.1762e-04, 5.0814e-05, 1.3598e-04, 1.5207e-04, 3.5421e-04, 3.4345e-04,
          1.9469e-04, 2.0434e-03, 3.2800e-04, 1.9249e-04, 1.4478e-04, 7.2088e-04,
          1.7806e-04, 2.2135e-04, 2.5775e-04, 4.0951e-04, 9.6435e-04, 1.6275e-04,
          5.2445e-03, 2.6786e-03, 3.4503e-04, 1.7906e-04, 2.1813e-04, 3.9107e-04,
          2.4114e-04, 1.6136e-04, 3.9489e-05, 2.3949e-04, 1.1147e-04, 4.1225e-03,
          7.0544e-03, 2.3299e-04, 2.5815e-04, 1.6322e-04, 2.3717e-04, 1.6239e-04,
          3.6815e-04, 3.2384e-04, 1.8505e-04, 3.9883e-04, 1.0281e-04, 2.9075e-04,
          1.2968e-04, 2.0095e-04, 1.4218e-04, 1.8748e-04, 2.7056e-04, 1.5134e-04,
          8.1434e-05, 1.7616e-04, 2.1193e-04, 1.6925e-04, 5.7646e-04, 5.1106e-04,
          1.9635e-04, 5.0596e-05, 1.6885e-04, 8.5044e-05, 1.7488e-04, 2.9131e-04,
          7.6794e-05, 2.3540e-04, 9.6468e-05, 2.2612e-04, 2.8577e-04, 1.9193e-04,
          1.8676e-04, 1.2103e-03, 8.7405e-05, 9.1073e-05, 7.2815e-05, 7.1142e-05,
          1.9839e-04, 9.5557e-05, 4.4340e-04, 2.5489e-04, 2.9035e-04, 3.3175e-04,
          5.5275e-04, 4.9836e-05, 3.2355e-04, 5.3539e-04, 8.2202e-04, 1.2297e-04,
          1.2729e-04, 2.6575e-04, 1.4388e-04, 9.3536e-05, 5.3339e-04, 3.0624e-04,
          5.0186e-04, 2.5777e-04, 1.2917e-04, 3.3091e-04, 1.9706e-04, 7.6847e-05,
          4.9676e-05, 3.3166e-04, 2.4856e-04, 2.8363e-04, 2.8585e-04, 1.7720e-04,
          1.9109e-04, 2.5642e-04, 1.4346e-04, 1.3652e-04, 1.8813e-04, 1.7389e-04,
          1.2564e-04, 1.4350e-04, 1.5161e-04, 4.5642e-04, 1.4046e-04, 3.9809e-04,
          2.0166e-04, 1.0429e-04, 2.4152e-04, 4.4321e-04, 8.9175e-04, 2.4384e-04,
          3.3205e-04, 2.3338e-04, 2.9393e-04, 2.1523e-04, 6.6598e-05, 9.2948e-05,
          5.8719e-05, 4.2741e-04, 4.2938e-04, 5.9072e-04, 1.2067e-04, 9.6547e-05,
          8.9108e-05, 7.7645e-05, 1.1400e-04, 1.7083e-04, 1.4661e-04, 2.5489e-04,
          1.2451e-04, 7.4220e-05, 1.7395e-04, 3.6074e-04, 4.7535e-04, 2.6067e-03,
          5.9086e-03, 1.8470e-02, 2.6302e-02, 6.2804e-03, 3.7625e-03, 1.2251e-02,
          1.6212e-02, 4.6255e-03, 7.2224e-03, 2.8809e-03, 6.0669e-03, 8.6736e-03,
          1.1559e-02, 2.4213e-02, 2.3510e-02, 3.2653e-02, 9.0895e-03, 5.9759e-03,
          2.0512e-02, 3.8580e-02, 8.2316e-02, 1.8781e-01, 1.6319e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.7308e-04, 2.3106e-02, 4.6992e-04, 2.8572e-03, 2.9065e-03,
          2.3495e-02, 1.2624e-03, 2.8598e-04, 1.4751e-04, 1.4550e-03, 2.6541e-03,
          1.0865e-03, 4.1310e-04, 1.8012e-04, 1.1019e-03, 2.0888e-04, 4.4280e-04,
          2.9072e-04, 9.8998e-05, 4.6782e-05, 4.8517e-05, 4.4172e-05, 5.6738e-04,
          3.1423e-03, 8.2146e-04, 1.0530e-03, 8.8496e-04, 7.0214e-05, 4.4790e-04,
          2.5537e-04, 6.5254e-04, 1.4817e-04, 1.1129e-03, 8.2921e-04, 3.2530e-04,
          1.3704e-03, 2.4010e-04, 4.7625e-04, 6.2229e-04, 7.8616e-05, 4.8971e-04,
          1.7405e-04, 1.5479e-04, 1.7772e-04, 1.5961e-04, 4.1691e-04, 1.5231e-03,
          8.7247e-05, 6.2248e-05, 3.7809e-04, 3.8601e-04, 5.1313e-04, 2.4363e-04,
          1.1081e-04, 2.7328e-04, 8.4113e-05, 5.6823e-04, 1.6953e-03, 1.1610e-03,
          5.7197e-04, 1.7880e-04, 9.4751e-05, 1.9964e-04, 5.3663e-04, 2.4084e-03,
          7.6450e-04, 7.6898e-04, 3.5658e-04, 9.0569e-05, 8.8003e-05, 1.7075e-04,
          1.1720e-04, 2.0548e-03, 9.6955e-05, 6.4211e-04, 9.7108e-05, 1.0729e-04,
          2.1712e-04, 5.7525e-04, 1.2067e-03, 1.3778e-03, 1.6109e-03, 4.5151e-04,
          1.5462e-04, 4.8442e-04, 8.1274e-05, 1.0848e-03, 5.3536e-04, 3.8827e-04,
          2.2643e-03, 1.3459e-03, 1.1266e-04, 1.6738e-03, 8.7638e-05, 8.3257e-05,
          8.4977e-05, 4.8000e-05, 4.2324e-04, 1.2060e-03, 4.7332e-04, 6.4307e-04,
          2.6234e-03, 5.0280e-04, 1.1058e-03, 2.0810e-03, 3.8171e-03, 1.0583e-04,
          3.7841e-04, 1.5507e-04, 1.5406e-03, 1.6447e-04, 4.3146e-04, 4.4626e-04,
          8.5060e-04, 2.0071e-04, 2.7992e-05, 3.2821e-03, 8.2246e-05, 6.2864e-05,
          1.6815e-04, 1.1789e-03, 2.0644e-04, 1.5772e-04, 2.4872e-03, 4.4134e-04,
          3.2414e-04, 8.0875e-04, 8.9019e-04, 2.2718e-03, 3.8822e-04, 4.7233e-04,
          6.2441e-04, 5.5346e-04, 2.0690e-04, 2.0970e-04, 1.1728e-03, 8.7939e-05,
          7.5492e-04, 7.4918e-05, 2.4941e-04, 4.9785e-04, 8.9125e-04, 2.2197e-04,
          3.1471e-04, 2.1449e-04, 4.0319e-04, 6.4160e-04, 8.8962e-04, 3.3946e-03,
          2.9987e-03, 1.8953e-03, 4.6085e-04, 2.3078e-04, 1.5176e-04, 2.8349e-04,
          8.6899e-04, 3.7693e-04, 3.9720e-04, 2.0176e-04, 9.7587e-04, 1.5863e-04,
          1.1377e-03, 5.8494e-04, 4.8550e-04, 7.4442e-04, 3.9395e-04, 4.8206e-04,
          1.3859e-03, 6.0578e-04, 8.3268e-04, 4.4801e-04, 3.4225e-04, 9.9119e-04,
          6.0041e-04, 4.0804e-04, 3.8467e-04, 1.9047e-04, 6.3097e-04, 6.6185e-04,
          4.7146e-04, 2.2463e-04, 7.9628e-04, 2.8594e-04, 1.6151e-04, 2.8993e-04,
          1.3551e-03, 1.1223e-04, 3.5923e-04, 2.9117e-04, 7.0674e-04, 6.6318e-04,
          3.7108e-04, 1.8737e-04, 6.5147e-04, 4.3049e-04, 1.2400e-03, 8.1791e-04,
          3.4810e-04, 5.0930e-05, 4.7940e-04, 6.3320e-04, 7.2651e-04, 3.3460e-04,
          2.7548e-04, 7.9220e-04, 9.0909e-05, 1.9209e-04, 2.0439e-03, 1.4890e-04,
          1.4131e-04, 3.3187e-04, 3.8738e-04, 1.1436e-03, 1.9459e-03, 4.9180e-04,
          7.1477e-04, 3.9968e-04, 7.4993e-04, 3.3619e-04, 9.4681e-04, 2.6759e-04,
          5.4265e-04, 8.6065e-05, 4.0453e-04, 5.7000e-04, 1.4008e-03, 4.4107e-04,
          5.4674e-04, 1.5349e-04, 1.0126e-04, 2.9283e-04, 4.0324e-04, 9.3244e-05,
          5.3578e-04, 6.2757e-04, 4.6361e-04, 5.5996e-04, 8.1900e-04, 9.6496e-04,
          3.9529e-04, 3.1781e-04, 2.6668e-04, 3.7335e-04, 4.3528e-04, 1.0278e-04,
          1.1530e-04, 7.7013e-04, 7.5263e-05, 1.5855e-04, 1.6183e-04, 7.0770e-04,
          1.0211e-04, 4.8082e-05, 7.9044e-05, 1.2242e-04, 4.8442e-04, 8.4170e-04,
          5.5521e-04, 5.8919e-04, 9.7329e-04, 7.4278e-04, 1.5599e-04, 5.3542e-04,
          7.1475e-04, 2.6200e-04, 2.6157e-04, 4.9837e-04, 6.4170e-04, 2.7242e-04,
          3.8476e-04, 6.4888e-04, 8.7224e-04, 1.5064e-04, 4.8384e-04, 6.7052e-04,
          1.9353e-04, 1.8057e-04, 2.8805e-05, 1.7360e-04, 2.0310e-04, 5.1643e-04,
          9.1578e-04, 2.0376e-04, 2.5088e-04, 2.5331e-04, 7.8008e-04, 7.7073e-04,
          4.7815e-04, 1.0001e-03, 2.9515e-04, 2.8745e-04, 8.9188e-05, 2.5290e-04,
          1.2274e-04, 1.1970e-04, 8.5439e-05, 8.5052e-04, 3.9975e-04, 9.7796e-05,
          6.5997e-05, 1.3289e-04, 1.9675e-04, 4.0723e-04, 3.5462e-04, 1.0397e-03,
          1.1549e-03, 6.6289e-05, 4.7628e-04, 1.0111e-04, 2.2290e-04, 2.9327e-04,
          1.3711e-04, 5.1053e-04, 9.4947e-05, 2.6973e-04, 9.9037e-04, 3.8767e-04,
          1.9265e-04, 2.0281e-03, 2.1085e-04, 1.1656e-04, 6.1608e-05, 8.4912e-05,
          1.0932e-04, 8.4714e-05, 4.6472e-04, 4.5443e-04, 6.6821e-04, 3.2778e-04,
          9.2302e-04, 4.7068e-05, 2.1541e-04, 7.9696e-04, 5.2209e-04, 1.4619e-04,
          2.1228e-04, 4.3561e-04, 2.3485e-04, 7.1473e-05, 7.5037e-04, 1.6956e-04,
          4.3798e-04, 4.9390e-04, 1.6011e-04, 3.5915e-04, 4.0806e-04, 1.4511e-04,
          3.2986e-05, 2.6545e-04, 3.5805e-04, 4.4296e-04, 4.1254e-04, 2.0316e-04,
          6.6048e-04, 3.2325e-04, 2.8025e-04, 1.3801e-04, 1.0807e-04, 7.6810e-04,
          1.6739e-04, 1.5237e-04, 3.8361e-04, 3.4590e-04, 1.6500e-04, 5.9248e-04,
          3.2498e-04, 5.6993e-05, 2.2589e-04, 3.6292e-04, 9.7805e-04, 4.3533e-04,
          2.8909e-04, 4.1932e-04, 5.0445e-04, 3.0395e-04, 1.1389e-04, 1.0119e-04,
          6.7646e-05, 2.6343e-04, 8.7966e-04, 4.9434e-04, 1.2409e-04, 1.8972e-04,
          6.3846e-05, 5.9662e-05, 1.0544e-04, 1.7264e-04, 2.5168e-04, 1.5337e-03,
          2.5518e-04, 1.3506e-04, 1.9757e-04, 4.5517e-04, 4.5387e-04, 3.0232e-03,
          6.1448e-03, 1.7371e-02, 2.8280e-02, 1.0536e-02, 4.7549e-03, 2.0307e-02,
          2.1830e-02, 5.1086e-03, 7.8630e-03, 3.6716e-03, 6.5033e-03, 8.9638e-03,
          1.5841e-02, 3.2612e-02, 2.3957e-02, 4.0878e-02, 1.5505e-02, 7.6802e-03,
          3.4136e-02, 5.5514e-02, 5.7958e-02, 7.4644e-02, 1.2484e-01, 1.1215e-01,
          0.0000e+00]),
  tensor([0.0000e+00, 5.1638e-04, 2.9503e-02, 5.8240e-04, 3.8813e-03, 5.0845e-03,
          3.4423e-02, 1.5459e-03, 1.2557e-03, 2.6576e-04, 1.3130e-03, 2.6107e-03,
          6.6788e-04, 4.3990e-04, 2.7585e-04, 7.2658e-04, 6.0051e-04, 4.7810e-04,
          2.9425e-04, 1.9653e-04, 6.7922e-05, 6.2385e-05, 6.4206e-05, 5.8088e-04,
          2.3212e-03, 7.0434e-04, 1.3903e-03, 1.1675e-03, 9.8309e-05, 3.9460e-04,
          3.3701e-04, 1.7001e-03, 1.2497e-04, 9.9842e-04, 6.6896e-04, 6.9654e-04,
          1.0926e-03, 6.4087e-04, 5.3664e-04, 8.2522e-04, 1.0811e-04, 4.1953e-04,
          1.9996e-04, 1.9397e-04, 1.9051e-04, 1.5353e-04, 3.6498e-04, 7.2296e-04,
          1.0664e-04, 1.0216e-04, 5.8706e-04, 1.2981e-03, 1.4553e-03, 4.4985e-04,
          5.4120e-04, 8.5992e-04, 4.2940e-04, 8.9035e-04, 3.0562e-02, 2.4323e-03,
          2.8850e-03, 2.3021e-04, 4.1832e-04, 7.1763e-04, 3.7831e-04, 1.0803e-03,
          6.6010e-04, 3.7038e-04, 3.7497e-04, 1.1387e-04, 9.4857e-05, 1.8061e-04,
          1.3983e-04, 1.5986e-03, 5.2431e-04, 1.7441e-03, 4.2346e-04, 4.1856e-04,
          9.9372e-04, 7.4473e-04, 9.1241e-04, 1.4495e-03, 1.6461e-03, 1.7605e-03,
          4.7570e-04, 1.4541e-03, 3.5617e-04, 8.3472e-04, 4.3500e-04, 3.4837e-04,
          1.2917e-03, 6.1705e-04, 2.8915e-04, 7.8281e-04, 2.0328e-04, 1.5959e-04,
          1.4887e-04, 2.3534e-04, 8.8561e-04, 5.9728e-04, 4.8879e-04, 4.5367e-04,
          1.9392e-03, 6.2338e-04, 7.5240e-04, 2.1178e-03, 1.7425e-03, 1.6215e-04,
          1.1571e-03, 1.3575e-04, 5.6989e-04, 2.5392e-04, 4.6028e-04, 2.2950e-04,
          6.0411e-04, 1.9048e-04, 4.2532e-05, 1.0416e-03, 1.6216e-04, 1.1179e-04,
          3.0733e-04, 6.4232e-04, 2.2688e-04, 3.8711e-04, 1.1386e-03, 3.8756e-04,
          2.7773e-04, 6.2210e-04, 6.7038e-04, 1.6220e-03, 5.4123e-04, 1.3166e-03,
          6.0649e-04, 3.2223e-04, 2.8417e-04, 1.5756e-04, 3.9202e-04, 1.0684e-04,
          4.6372e-04, 9.7685e-05, 4.3843e-04, 7.9739e-04, 8.0355e-04, 3.6871e-04,
          3.4291e-04, 4.8811e-04, 4.5442e-04, 3.8745e-04, 1.1069e-03, 2.3646e-03,
          2.1701e-03, 1.6449e-03, 7.9537e-04, 6.1879e-04, 8.9270e-04, 3.1727e-04,
          3.1682e-04, 2.4867e-04, 2.8889e-04, 1.9539e-04, 7.2416e-04, 2.0994e-04,
          5.0128e-04, 4.4847e-04, 3.4334e-04, 6.1144e-04, 3.0276e-04, 2.7313e-04,
          6.6428e-04, 3.0075e-04, 6.3892e-04, 4.6897e-04, 2.5971e-04, 5.5040e-04,
          1.6075e-03, 2.7567e-03, 8.0828e-04, 2.6235e-04, 5.3966e-04, 6.5000e-04,
          2.4176e-04, 4.1493e-04, 5.3368e-04, 7.7685e-04, 3.3176e-04, 3.3116e-04,
          7.8318e-04, 2.3557e-04, 5.7168e-04, 2.5495e-04, 9.2811e-04, 4.2728e-04,
          7.5621e-04, 1.4607e-04, 3.9857e-04, 2.9777e-04, 7.2635e-04, 4.7855e-04,
          1.0942e-03, 9.3050e-05, 6.8358e-04, 2.5128e-04, 3.6507e-04, 2.2425e-04,
          1.8341e-04, 2.5268e-04, 1.2318e-04, 3.1368e-04, 6.9833e-04, 8.9636e-05,
          3.3937e-04, 2.8944e-04, 4.4219e-04, 1.2359e-03, 8.1693e-04, 9.3307e-04,
          4.6087e-04, 3.0155e-04, 6.8739e-04, 3.9297e-04, 6.4112e-04, 6.9322e-04,
          3.9807e-04, 1.7426e-04, 2.8901e-04, 2.6572e-04, 8.1452e-04, 3.8805e-04,
          2.6997e-04, 2.8709e-04, 1.3038e-04, 5.2580e-04, 5.0986e-04, 1.9245e-04,
          7.0143e-04, 3.7204e-04, 7.4964e-04, 6.2739e-04, 7.4992e-04, 1.7279e-03,
          1.1693e-03, 6.3753e-04, 4.2062e-04, 3.9403e-04, 7.4455e-04, 1.5303e-04,
          1.6821e-04, 3.7576e-04, 2.8444e-04, 4.7912e-04, 3.7388e-04, 6.5029e-04,
          8.9485e-05, 9.4641e-05, 1.7297e-04, 1.6765e-04, 5.2684e-04, 8.6280e-04,
          2.8234e-04, 2.9285e-04, 8.3370e-04, 5.2320e-04, 6.1494e-04, 1.1521e-03,
          6.3043e-04, 3.0784e-04, 4.7542e-04, 8.8198e-04, 1.0501e-03, 4.1482e-04,
          5.7471e-04, 3.5366e-04, 5.6045e-04, 1.9594e-04, 3.4469e-04, 6.6546e-04,
          3.9249e-04, 2.5994e-04, 4.8587e-05, 1.8540e-04, 2.0121e-04, 4.8332e-04,
          5.7309e-04, 2.7246e-04, 2.0015e-04, 4.3148e-04, 5.2378e-04, 3.2665e-04,
          7.0027e-04, 5.5362e-04, 1.8033e-04, 2.6833e-04, 1.7868e-04, 4.5971e-04,
          1.5848e-04, 1.0867e-04, 1.0855e-04, 3.5013e-04, 3.3651e-04, 2.2618e-04,
          1.3734e-04, 2.1633e-04, 4.5687e-04, 3.2358e-04, 3.8259e-04, 7.2004e-04,
          5.2827e-04, 9.1191e-05, 3.3154e-04, 1.3158e-04, 3.0794e-04, 3.4629e-04,
          1.2234e-04, 5.0942e-04, 1.3245e-04, 4.6428e-04, 5.5322e-04, 5.5311e-04,
          4.1899e-04, 9.9409e-04, 1.5219e-04, 1.3750e-04, 1.3902e-04, 1.4893e-04,
          1.3108e-04, 1.3935e-04, 5.6582e-04, 3.7343e-04, 5.1294e-04, 3.1925e-04,
          5.7184e-04, 1.1166e-04, 2.2172e-04, 5.7977e-04, 8.1813e-04, 1.5548e-04,
          3.7613e-04, 7.4352e-04, 3.3889e-04, 1.2403e-04, 3.7102e-04, 1.8184e-04,
          4.3309e-04, 3.0773e-04, 1.3842e-04, 4.4137e-04, 6.6017e-04, 1.0997e-04,
          4.5323e-05, 4.0169e-04, 5.2619e-04, 5.5767e-04, 8.4468e-04, 4.2172e-04,
          5.6827e-04, 3.9300e-04, 1.9881e-04, 2.6191e-04, 1.2594e-04, 6.8315e-04,
          3.9365e-04, 4.4103e-04, 3.4558e-04, 4.2210e-04, 2.2990e-04, 5.3716e-04,
          2.0387e-04, 6.7458e-05, 2.9301e-04, 2.4722e-04, 9.1737e-04, 3.4625e-04,
          4.0763e-04, 5.3288e-04, 4.4019e-04, 2.6751e-04, 1.4613e-04, 1.1231e-04,
          1.2632e-04, 4.9234e-04, 3.7545e-04, 6.4246e-04, 1.5561e-04, 1.5984e-04,
          2.4404e-04, 9.2896e-05, 7.5857e-05, 1.7412e-04, 2.9306e-04, 5.4384e-04,
          1.4128e-04, 1.3293e-04, 1.9935e-04, 3.6057e-04, 3.2455e-04, 2.7233e-03,
          6.2994e-03, 2.1752e-02, 1.5559e-02, 7.4162e-03, 3.9447e-03, 1.2210e-02,
          1.8378e-02, 4.9444e-03, 1.0161e-02, 2.5957e-03, 7.7328e-03, 1.0015e-02,
          1.5255e-02, 3.3748e-02, 2.5860e-02, 2.1207e-02, 1.1967e-02, 7.1857e-03,
          1.8799e-02, 4.2884e-02, 4.6989e-02, 3.9140e-02, 5.3972e-02, 8.3470e-02,
          1.6604e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.4916e-04, 2.5991e-02, 5.7746e-04, 4.0501e-03, 4.8323e-03,
          2.9392e-02, 1.6959e-03, 1.5046e-03, 3.2473e-04, 1.3546e-03, 2.8714e-03,
          8.2072e-04, 4.7359e-04, 3.5620e-04, 5.5874e-04, 7.5748e-04, 5.3164e-04,
          3.9531e-04, 2.3827e-04, 7.3746e-05, 6.2526e-05, 8.7949e-05, 6.4583e-04,
          2.6308e-03, 6.0940e-04, 1.5266e-03, 1.4176e-03, 1.1731e-04, 4.8714e-04,
          4.2478e-04, 2.0829e-03, 1.6692e-04, 8.0376e-04, 6.8137e-04, 6.9273e-04,
          8.7651e-04, 6.7475e-04, 4.4654e-04, 7.1220e-04, 1.2089e-04, 4.1153e-04,
          2.2174e-04, 1.8056e-04, 2.0767e-04, 1.3395e-04, 4.4654e-04, 5.1245e-04,
          1.5173e-04, 1.0426e-04, 6.0468e-04, 1.7213e-03, 1.8534e-03, 4.8594e-04,
          6.6708e-04, 1.1724e-03, 6.0734e-04, 6.4025e-04, 4.6464e-02, 1.5795e-03,
          1.8761e-03, 2.7647e-04, 8.9078e-04, 7.8568e-04, 3.9329e-04, 8.1367e-04,
          9.5838e-04, 5.0625e-04, 3.3099e-04, 9.0547e-05, 7.8455e-05, 1.6662e-04,
          1.2714e-04, 1.4570e-03, 7.3146e-04, 2.0453e-03, 5.7682e-04, 5.4452e-04,
          1.2025e-03, 4.8588e-04, 9.5715e-04, 1.1446e-03, 1.8390e-03, 9.9038e-04,
          3.8242e-04, 2.5446e-03, 6.5159e-04, 9.5950e-04, 5.4718e-04, 4.4709e-04,
          1.7132e-03, 5.9122e-04, 2.6557e-04, 9.0601e-04, 1.4527e-04, 1.2625e-04,
          1.7007e-04, 2.8931e-04, 9.6693e-04, 4.5150e-04, 3.4761e-04, 3.1939e-04,
          2.0904e-03, 6.6336e-04, 9.2066e-04, 1.4743e-03, 1.2733e-03, 1.9797e-04,
          2.0363e-03, 1.5995e-04, 5.5876e-04, 2.6991e-04, 5.8170e-04, 2.5467e-04,
          4.6802e-04, 1.3687e-04, 5.0154e-05, 7.6288e-04, 1.4464e-04, 1.4905e-04,
          4.1405e-04, 4.7101e-04, 1.9251e-04, 5.5585e-04, 9.0021e-04, 3.2275e-04,
          1.7400e-04, 6.3431e-04, 6.7892e-04, 1.1194e-03, 6.8973e-04, 2.2883e-03,
          6.8325e-04, 4.1625e-04, 2.6024e-04, 2.0582e-04, 4.3822e-04, 1.0462e-04,
          4.3030e-04, 1.2382e-04, 3.9513e-04, 7.3869e-04, 7.7328e-04, 4.4321e-04,
          2.6786e-04, 3.7474e-04, 3.0525e-04, 2.9032e-04, 1.0581e-03, 1.6462e-03,
          1.9591e-03, 1.1515e-03, 4.6585e-04, 7.9193e-04, 1.5517e-03, 2.4553e-04,
          3.3377e-04, 2.4929e-04, 3.6857e-04, 2.3244e-04, 4.5416e-04, 2.2510e-04,
          3.2825e-04, 3.4584e-04, 3.8967e-04, 5.0334e-04, 2.1656e-04, 2.4087e-04,
          7.1913e-04, 2.3636e-04, 4.7830e-04, 3.6581e-04, 2.9126e-04, 3.6094e-04,
          1.3108e-03, 1.5802e-03, 4.5606e-04, 2.5337e-04, 3.8517e-04, 7.8416e-04,
          1.8390e-04, 3.6606e-04, 4.1248e-04, 9.6568e-04, 3.7742e-04, 2.4160e-04,
          9.4422e-04, 2.2718e-04, 5.6854e-04, 2.0319e-04, 8.6569e-04, 3.5409e-04,
          8.6475e-04, 1.1370e-04, 2.7646e-04, 3.4282e-04, 7.1984e-04, 3.5907e-04,
          5.3674e-04, 1.0964e-04, 6.1886e-04, 2.8630e-04, 3.8713e-04, 1.7632e-04,
          2.5628e-04, 2.5964e-04, 9.5457e-05, 3.4698e-04, 7.6387e-04, 1.1320e-04,
          4.6423e-04, 2.1829e-04, 3.0457e-04, 1.1407e-03, 5.7109e-04, 5.7147e-04,
          2.7408e-04, 2.8740e-04, 8.3476e-04, 2.9868e-04, 4.5103e-04, 8.0316e-04,
          3.2799e-04, 1.9309e-04, 2.6278e-04, 3.0451e-04, 6.8318e-04, 3.5874e-04,
          3.8543e-04, 3.0790e-04, 1.1606e-04, 4.4091e-04, 3.5146e-04, 1.8702e-04,
          5.2851e-04, 2.8144e-04, 6.3276e-04, 4.9852e-04, 4.8616e-04, 1.6474e-03,
          9.8360e-04, 5.0323e-04, 3.6095e-04, 2.7534e-04, 6.8445e-04, 1.8136e-04,
          1.2376e-04, 4.8927e-04, 3.2016e-04, 4.8111e-04, 3.4946e-04, 5.0686e-04,
          1.1122e-04, 6.7375e-05, 1.7920e-04, 1.5233e-04, 4.9808e-04, 6.6845e-04,
          1.8376e-04, 3.0804e-04, 4.1740e-04, 3.3401e-04, 4.0418e-04, 1.0472e-03,
          4.1542e-04, 3.1743e-04, 4.5219e-04, 8.4364e-04, 1.2745e-03, 3.6756e-04,
          6.4783e-04, 4.7054e-04, 5.0168e-04, 1.8990e-04, 2.3538e-04, 4.9238e-04,
          3.0154e-04, 2.2478e-04, 5.1625e-05, 2.6932e-04, 1.4311e-04, 4.7079e-04,
          7.6833e-04, 1.7525e-04, 1.2261e-04, 2.5195e-04, 4.2988e-04, 2.3931e-04,
          6.3745e-04, 4.4474e-04, 1.9900e-04, 2.9545e-04, 1.8693e-04, 5.1536e-04,
          1.4091e-04, 1.0747e-04, 9.4246e-05, 2.6047e-04, 2.5261e-04, 1.9478e-04,
          1.2396e-04, 2.3938e-04, 5.2473e-04, 3.1555e-04, 2.5937e-04, 5.2386e-04,
          3.4822e-04, 9.2672e-05, 2.3556e-04, 1.4835e-04, 2.3738e-04, 3.7144e-04,
          1.2619e-04, 3.1362e-04, 1.2335e-04, 6.0586e-04, 4.4700e-04, 4.0516e-04,
          3.2688e-04, 1.1642e-03, 1.4279e-04, 1.5012e-04, 1.2648e-04, 1.2061e-04,
          1.4183e-04, 1.5685e-04, 6.6454e-04, 2.2316e-04, 3.6643e-04, 4.5010e-04,
          4.2193e-04, 1.1375e-04, 3.1988e-04, 5.5134e-04, 8.4993e-04, 1.6674e-04,
          2.8829e-04, 5.7609e-04, 3.1357e-04, 1.2116e-04, 3.2089e-04, 2.8876e-04,
          3.8116e-04, 2.9499e-04, 1.4698e-04, 4.8080e-04, 6.0711e-04, 1.0715e-04,
          5.7581e-05, 4.6450e-04, 3.2754e-04, 3.1714e-04, 1.0129e-03, 5.3158e-04,
          5.1855e-04, 4.0112e-04, 1.9134e-04, 2.2034e-04, 1.9522e-04, 4.5873e-04,
          5.9866e-04, 6.8720e-04, 3.0099e-04, 4.8271e-04, 2.1609e-04, 5.0390e-04,
          2.2491e-04, 1.0007e-04, 3.1840e-04, 2.1620e-04, 8.2180e-04, 3.3643e-04,
          4.5489e-04, 3.0859e-04, 3.2193e-04, 2.7326e-04, 1.8720e-04, 1.0954e-04,
          2.0320e-04, 6.5975e-04, 4.1446e-04, 7.1906e-04, 1.4045e-04, 1.2046e-04,
          3.0315e-04, 9.8250e-05, 1.0036e-04, 1.9042e-04, 2.2842e-04, 3.7793e-04,
          1.3675e-04, 1.4516e-04, 1.8689e-04, 2.7742e-04, 2.8345e-04, 2.6995e-03,
          5.7405e-03, 1.9910e-02, 1.5135e-02, 4.7848e-03, 3.1830e-03, 1.0147e-02,
          1.2430e-02, 4.9041e-03, 7.0413e-03, 2.3820e-03, 5.8078e-03, 7.1574e-03,
          1.2925e-02, 2.4234e-02, 1.9909e-02, 1.7331e-02, 6.8311e-03, 5.3175e-03,
          1.3303e-02, 2.0347e-02, 2.7967e-02, 3.7541e-02, 3.6424e-02, 4.7379e-02,
          1.8261e-01, 1.3718e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.6420e-04, 2.7601e-02, 6.7861e-04, 4.4264e-03, 4.8542e-03,
          3.4108e-02, 1.9332e-03, 1.0991e-03, 2.8238e-04, 1.1456e-03, 2.7098e-03,
          8.1606e-04, 4.2034e-04, 6.6273e-04, 3.6799e-04, 5.2389e-04, 5.0148e-04,
          5.0632e-04, 1.8907e-04, 8.3171e-05, 5.4791e-05, 1.0090e-04, 4.9933e-04,
          2.5142e-03, 4.4306e-04, 1.7951e-03, 9.7764e-04, 1.7229e-04, 4.3238e-04,
          7.9487e-04, 1.4786e-03, 1.8268e-04, 4.3937e-04, 4.7167e-04, 3.4104e-04,
          5.3005e-04, 6.8233e-04, 2.2960e-04, 3.8096e-04, 1.8287e-04, 2.2065e-04,
          2.1426e-04, 1.5882e-04, 2.4856e-04, 9.2762e-05, 4.4293e-04, 2.8602e-04,
          3.3862e-04, 1.3874e-04, 5.8149e-04, 9.7878e-04, 1.3882e-03, 2.8936e-04,
          1.2904e-03, 9.2276e-04, 6.3246e-04, 5.0235e-04, 7.6530e-02, 1.7102e-03,
          1.9195e-03, 4.5175e-04, 1.3213e-03, 3.6873e-04, 3.0032e-04, 4.7750e-04,
          1.0885e-03, 5.1212e-04, 2.0000e-04, 8.0598e-05, 8.0685e-05, 2.2750e-04,
          1.4229e-04, 1.1020e-03, 7.5919e-04, 1.9670e-03, 1.8181e-03, 6.6693e-04,
          1.2148e-03, 3.0961e-04, 8.6510e-04, 7.0919e-04, 2.3186e-03, 1.3977e-03,
          4.4833e-04, 1.8734e-03, 1.2982e-03, 7.7180e-04, 4.8411e-04, 5.7922e-04,
          1.6250e-03, 3.8606e-04, 2.3533e-04, 9.0714e-04, 1.2429e-04, 1.2635e-04,
          2.1594e-04, 2.8283e-04, 4.5498e-04, 1.9866e-04, 2.3075e-04, 2.0021e-04,
          1.7474e-03, 5.2845e-04, 8.6690e-04, 1.0824e-03, 6.8731e-04, 1.9214e-04,
          1.6815e-03, 1.2592e-04, 3.9179e-04, 3.2111e-04, 5.5804e-04, 1.9892e-04,
          3.1513e-04, 8.6952e-05, 5.6316e-05, 4.6481e-04, 1.5106e-04, 1.3186e-04,
          5.3627e-04, 1.9611e-04, 1.6378e-04, 4.3573e-04, 5.2078e-04, 2.7279e-04,
          1.2556e-04, 7.1311e-04, 7.8542e-04, 6.3018e-04, 4.7270e-04, 2.0098e-03,
          5.9624e-04, 4.1297e-04, 2.2593e-04, 2.0427e-04, 4.0820e-04, 1.1896e-04,
          2.5747e-04, 2.1657e-04, 2.9619e-04, 5.4197e-04, 4.1396e-04, 5.5310e-04,
          2.1110e-04, 4.4327e-04, 1.8411e-04, 1.8971e-04, 9.7763e-04, 7.7600e-04,
          1.0236e-03, 6.3202e-04, 3.8719e-04, 5.1449e-04, 2.1916e-03, 1.4286e-04,
          2.2932e-04, 1.3798e-04, 3.0629e-04, 3.8122e-04, 2.5438e-04, 2.1659e-04,
          1.6825e-04, 1.8025e-04, 3.9041e-04, 2.5455e-04, 8.5594e-05, 1.1138e-04,
          5.2049e-04, 1.2476e-04, 2.4617e-04, 1.9666e-04, 2.1398e-04, 1.7996e-04,
          8.9694e-04, 2.0829e-03, 3.5937e-04, 1.6963e-04, 1.9873e-04, 6.2171e-04,
          1.0760e-04, 1.9679e-04, 2.4723e-04, 7.3084e-04, 2.5563e-04, 1.4669e-04,
          7.2573e-04, 1.8806e-04, 4.0373e-04, 1.2285e-04, 7.9114e-04, 2.4046e-04,
          6.4733e-04, 9.1297e-05, 1.5575e-04, 2.8807e-04, 5.5708e-04, 2.1985e-04,
          5.6576e-04, 1.3082e-04, 2.5859e-04, 3.0986e-04, 2.6905e-04, 1.0224e-04,
          4.8514e-04, 2.3945e-04, 7.7240e-05, 3.0679e-04, 6.4169e-04, 9.4048e-05,
          4.5097e-04, 8.6132e-05, 1.7084e-04, 9.4554e-04, 3.2345e-04, 4.7110e-04,
          1.1993e-04, 2.2664e-04, 6.3902e-04, 1.8362e-04, 2.1186e-04, 8.8596e-04,
          1.8160e-04, 1.3844e-04, 1.9630e-04, 2.1402e-04, 4.0249e-04, 2.5360e-04,
          4.5752e-04, 5.1859e-04, 1.1836e-04, 3.7935e-04, 2.0142e-04, 1.8605e-04,
          2.3204e-04, 1.5082e-04, 6.9826e-04, 3.1911e-04, 2.3672e-04, 1.1823e-03,
          1.2917e-03, 4.8054e-04, 1.8720e-04, 1.4607e-04, 4.2629e-04, 2.7186e-04,
          8.5837e-05, 6.7149e-04, 3.6678e-04, 4.7841e-04, 3.6696e-04, 2.8718e-04,
          1.8400e-04, 6.4948e-05, 2.0328e-04, 9.2081e-05, 3.3035e-04, 3.5100e-04,
          9.0357e-05, 2.5872e-04, 1.8961e-04, 1.7663e-04, 2.4297e-04, 1.5083e-03,
          2.1023e-04, 1.8134e-04, 3.0286e-04, 5.5214e-04, 1.3563e-03, 1.8003e-04,
          8.8648e-04, 7.5009e-04, 3.0005e-04, 1.4837e-04, 1.2450e-04, 2.8870e-04,
          2.2531e-04, 1.5583e-04, 5.3833e-05, 5.9645e-04, 7.5094e-05, 4.6331e-04,
          7.3766e-04, 1.2364e-04, 9.9984e-05, 2.0635e-04, 1.9101e-04, 1.1174e-04,
          4.6297e-04, 2.0021e-04, 1.4144e-04, 2.3178e-04, 1.7856e-04, 4.8578e-04,
          1.3129e-04, 1.5302e-04, 1.0885e-04, 1.3452e-04, 1.3881e-04, 2.4584e-04,
          1.3708e-04, 1.5105e-04, 4.0122e-04, 1.6124e-04, 1.6017e-04, 3.8146e-04,
          1.7628e-04, 7.1402e-05, 1.1367e-04, 9.1657e-05, 1.5487e-04, 2.6692e-04,
          7.8707e-05, 2.4587e-04, 1.2197e-04, 3.5680e-04, 2.4349e-04, 2.2368e-04,
          2.8927e-04, 8.9963e-04, 9.3496e-05, 1.1137e-04, 1.2021e-04, 9.1417e-05,
          1.0305e-04, 1.4788e-04, 4.9846e-04, 1.2955e-04, 1.8876e-04, 7.8314e-04,
          2.2880e-04, 9.5030e-05, 8.6343e-04, 3.1754e-04, 7.1560e-04, 1.2459e-04,
          2.3186e-04, 3.0081e-04, 1.9981e-04, 1.3784e-04, 1.7819e-04, 7.8261e-04,
          2.2444e-04, 1.8845e-04, 1.2673e-04, 2.4462e-04, 3.4631e-04, 7.3989e-05,
          1.3409e-04, 4.1644e-04, 2.4256e-04, 2.3471e-04, 6.3597e-04, 4.4563e-04,
          3.1351e-04, 3.4918e-04, 1.0107e-04, 1.6515e-04, 5.1822e-04, 2.6503e-04,
          3.8353e-04, 4.5636e-04, 1.9188e-04, 4.5837e-04, 1.9417e-04, 2.6168e-04,
          1.7634e-04, 3.2838e-04, 2.7208e-04, 1.4003e-04, 6.1498e-04, 1.9117e-04,
          4.9632e-04, 3.0725e-04, 2.9450e-04, 2.0835e-04, 1.3339e-04, 1.2854e-04,
          1.8102e-04, 7.6404e-04, 2.9273e-04, 8.4897e-04, 1.5512e-04, 1.2940e-04,
          2.3836e-04, 1.0633e-04, 2.9788e-04, 1.5195e-04, 1.3501e-04, 1.9973e-04,
          1.8963e-04, 1.1767e-04, 1.3323e-04, 1.7664e-04, 2.4062e-04, 2.3692e-03,
          4.7437e-03, 2.0254e-02, 1.1708e-02, 3.1725e-03, 2.6520e-03, 7.1124e-03,
          9.2844e-03, 4.2011e-03, 6.2596e-03, 2.0844e-03, 5.8452e-03, 4.3659e-03,
          1.0988e-02, 2.0563e-02, 1.5458e-02, 1.2060e-02, 4.1935e-03, 3.6134e-03,
          8.8506e-03, 1.1856e-02, 1.4490e-02, 3.7781e-02, 3.4241e-02, 3.1387e-02,
          1.0167e-01, 1.2474e-01, 1.6752e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.1872e-04, 2.7311e-02, 6.6148e-04, 3.6564e-03, 3.9930e-03,
          3.0026e-02, 1.5699e-03, 7.6481e-04, 2.7331e-04, 7.5465e-04, 1.7052e-03,
          6.0440e-04, 4.1017e-04, 2.0455e-04, 6.2101e-04, 4.6063e-04, 5.1971e-04,
          3.2570e-04, 1.8387e-04, 6.6620e-05, 7.0409e-05, 8.4303e-05, 5.0251e-04,
          1.8389e-03, 5.5615e-04, 1.3923e-03, 8.8404e-04, 1.0803e-04, 5.4611e-04,
          2.3728e-04, 1.2994e-03, 2.2145e-04, 6.3838e-04, 4.4052e-04, 5.8877e-04,
          7.2243e-04, 5.1519e-04, 2.4581e-04, 4.2794e-04, 1.0054e-04, 2.5361e-04,
          1.4611e-04, 1.3479e-04, 1.4004e-04, 1.0175e-04, 4.0539e-04, 6.3967e-04,
          1.2441e-04, 9.1250e-05, 4.8952e-04, 1.1786e-03, 1.1568e-03, 3.5580e-04,
          4.7497e-04, 7.2608e-04, 3.8667e-04, 5.2761e-04, 3.1787e-02, 1.2579e-03,
          1.1158e-03, 2.5689e-04, 6.7007e-04, 5.3910e-04, 2.6650e-04, 7.8594e-04,
          5.2348e-04, 3.7735e-04, 1.8398e-04, 7.1574e-05, 5.5992e-05, 1.4054e-04,
          1.1809e-04, 8.3107e-04, 5.8720e-04, 1.4495e-03, 4.4013e-04, 3.0563e-04,
          5.4687e-04, 3.3900e-04, 4.4882e-04, 5.4557e-04, 1.0112e-03, 9.0017e-04,
          3.5928e-04, 1.4623e-03, 4.2327e-04, 5.6064e-04, 2.2923e-04, 1.9900e-04,
          1.2671e-03, 3.9853e-04, 1.3137e-04, 7.1921e-04, 1.2236e-04, 1.0446e-04,
          1.6518e-04, 1.8640e-04, 5.2636e-04, 4.3518e-04, 2.4278e-04, 2.4176e-04,
          1.4412e-03, 3.4170e-04, 6.0996e-04, 1.0232e-03, 1.4230e-03, 1.4969e-04,
          1.3467e-03, 8.9727e-05, 3.8963e-04, 1.5723e-04, 3.2458e-04, 1.4076e-04,
          3.5191e-04, 9.3488e-05, 4.4589e-05, 1.1397e-03, 1.0841e-04, 1.7652e-04,
          2.2025e-04, 2.5582e-04, 1.1597e-04, 4.9869e-04, 8.0593e-04, 1.2211e-04,
          1.1977e-04, 3.6012e-04, 4.2097e-04, 1.1340e-03, 5.3265e-04, 1.5257e-03,
          3.4596e-04, 2.9575e-04, 1.5105e-04, 9.9453e-05, 3.0982e-04, 6.1453e-05,
          4.0665e-04, 7.7732e-05, 3.1466e-04, 6.0120e-04, 6.4332e-04, 2.1130e-04,
          1.4536e-04, 3.2564e-04, 1.5468e-04, 2.3025e-04, 3.3353e-04, 1.0144e-03,
          8.1310e-04, 9.7441e-04, 3.7274e-04, 7.2914e-04, 1.9723e-03, 1.2524e-04,
          2.0452e-04, 1.1784e-04, 2.2179e-04, 1.0571e-04, 3.3473e-04, 1.6903e-04,
          3.8503e-04, 2.0886e-04, 3.4296e-04, 3.5739e-04, 1.4876e-04, 8.8457e-05,
          6.3778e-04, 1.0151e-04, 2.1786e-04, 2.5398e-04, 1.9279e-04, 3.7409e-04,
          1.0203e-03, 1.2110e-03, 3.9519e-04, 1.3255e-04, 2.1650e-04, 4.8168e-04,
          1.0826e-04, 2.1257e-04, 2.4979e-04, 5.6022e-04, 2.3030e-04, 1.9840e-04,
          5.5670e-04, 1.4435e-04, 2.9543e-04, 1.1914e-04, 2.2057e-04, 2.7846e-04,
          5.3374e-04, 6.6724e-05, 2.3646e-04, 3.3534e-04, 4.0633e-04, 3.2928e-04,
          4.8572e-04, 8.2785e-05, 2.7100e-04, 2.0618e-04, 2.0592e-04, 9.6710e-05,
          2.3139e-04, 1.8594e-04, 6.3634e-05, 2.5555e-04, 6.4525e-04, 1.3401e-04,
          3.4720e-04, 1.2140e-04, 1.8711e-04, 5.3693e-04, 4.3789e-04, 2.0659e-04,
          1.9106e-04, 2.2739e-04, 4.3538e-04, 2.4524e-04, 3.9040e-04, 4.3060e-04,
          2.2970e-04, 1.0506e-04, 1.7243e-04, 1.6198e-04, 5.3102e-04, 2.6829e-04,
          2.7059e-04, 2.2953e-04, 9.2195e-05, 4.3036e-04, 2.4879e-04, 2.0376e-04,
          2.2466e-04, 1.2659e-04, 2.7900e-04, 3.6194e-04, 3.0051e-04, 1.5216e-03,
          4.3963e-04, 3.0029e-04, 1.8220e-04, 2.2205e-04, 3.9161e-04, 9.2182e-05,
          7.0600e-05, 5.6333e-04, 2.1893e-04, 4.4254e-04, 2.1531e-04, 2.9549e-04,
          1.0766e-04, 4.6745e-05, 1.2035e-04, 9.4923e-05, 4.4304e-04, 6.1964e-04,
          1.3135e-04, 1.4491e-04, 3.6388e-04, 3.0059e-04, 1.9144e-04, 4.7632e-04,
          3.6127e-04, 1.6544e-04, 3.0598e-04, 4.8260e-04, 4.4474e-04, 1.8280e-04,
          4.7564e-04, 3.6586e-04, 3.4039e-04, 1.0607e-04, 1.5467e-04, 3.3950e-04,
          2.4923e-04, 1.7547e-04, 3.8504e-05, 1.6825e-04, 9.2286e-05, 2.1738e-04,
          4.3021e-04, 1.0490e-04, 7.4580e-05, 1.7144e-04, 3.3028e-04, 2.7868e-04,
          5.9960e-04, 3.3881e-04, 2.0382e-04, 2.7534e-04, 1.1131e-04, 4.9632e-04,
          8.7462e-05, 7.7012e-05, 5.9198e-05, 2.9846e-04, 1.5035e-04, 1.5910e-04,
          9.1291e-05, 1.6653e-04, 3.8410e-04, 2.3702e-04, 1.2860e-04, 4.3775e-04,
          4.1204e-04, 6.2171e-05, 1.6844e-04, 1.1361e-04, 1.5734e-04, 2.3830e-04,
          1.0376e-04, 4.4959e-04, 8.3132e-05, 3.5184e-04, 4.1621e-04, 2.3069e-04,
          2.8978e-04, 8.1339e-04, 1.1857e-04, 1.5569e-04, 8.0872e-05, 7.2629e-05,
          1.1229e-04, 1.2675e-04, 4.1097e-04, 1.6754e-04, 3.0025e-04, 2.2663e-04,
          2.3772e-04, 7.9622e-05, 1.7870e-04, 2.9052e-04, 6.0163e-04, 1.5068e-04,
          3.3029e-04, 3.3870e-04, 2.4537e-04, 8.5291e-05, 1.7462e-04, 1.6528e-04,
          2.7394e-04, 1.8676e-04, 1.7152e-04, 3.5137e-04, 4.5190e-04, 9.4251e-05,
          4.2479e-05, 2.9277e-04, 2.6796e-04, 2.6736e-04, 6.5391e-04, 4.5909e-04,
          5.8929e-04, 2.1576e-04, 1.2428e-04, 1.3740e-04, 1.1378e-04, 4.0683e-04,
          3.9947e-04, 3.8460e-04, 2.4801e-04, 2.5119e-04, 1.5826e-04, 2.4780e-04,
          1.5607e-04, 7.0514e-05, 2.3380e-04, 1.5345e-04, 6.4810e-04, 2.1939e-04,
          3.2903e-04, 2.4317e-04, 2.2598e-04, 1.9484e-04, 1.6412e-04, 8.2904e-05,
          1.7266e-04, 2.1812e-04, 2.5008e-04, 3.9437e-04, 1.1193e-04, 8.3787e-05,
          2.0890e-04, 7.6414e-05, 6.2875e-05, 1.2730e-04, 1.3821e-04, 4.1864e-04,
          1.1216e-04, 1.4261e-04, 1.2503e-04, 1.8850e-04, 1.8943e-04, 2.7888e-03,
          4.7136e-03, 1.8813e-02, 1.2850e-02, 4.5034e-03, 2.8708e-03, 7.2887e-03,
          1.4010e-02, 4.7907e-03, 6.7473e-03, 3.3234e-03, 5.6170e-03, 5.5292e-03,
          1.2904e-02, 2.6567e-02, 1.7534e-02, 1.6086e-02, 5.2396e-03, 3.6076e-03,
          1.2447e-02, 1.8522e-02, 1.5857e-02, 2.4431e-02, 2.4823e-02, 5.5346e-02,
          9.6783e-02, 5.7615e-02, 1.1917e-01, 1.6314e-01, 0.0000e+00]),
  tensor([0.0000e+00, 1.0949e-03, 2.2268e-02, 1.1909e-03, 6.1347e-03, 5.5105e-03,
          2.0418e-02, 3.7353e-03, 1.7860e-04, 1.0229e-04, 4.2585e-04, 8.0200e-04,
          2.2014e-04, 2.3369e-04, 1.1447e-04, 6.0185e-04, 1.1010e-04, 2.4263e-04,
          1.1053e-04, 7.1013e-05, 5.9521e-05, 7.2603e-05, 5.3813e-05, 2.7666e-04,
          7.2489e-04, 5.2404e-04, 2.7959e-03, 1.2947e-03, 1.0351e-04, 1.8226e-04,
          1.6065e-04, 2.9492e-04, 6.3887e-05, 4.5323e-04, 2.3096e-04, 3.4796e-04,
          4.4736e-04, 2.5728e-04, 3.4096e-04, 2.6677e-04, 8.5702e-05, 2.5801e-04,
          1.2527e-04, 1.1406e-04, 1.1387e-04, 7.1841e-05, 1.6698e-04, 6.3209e-04,
          6.8096e-05, 1.0639e-04, 6.8063e-04, 2.9763e-04, 2.1413e-04, 2.9773e-04,
          1.0751e-04, 1.2064e-04, 2.1027e-04, 2.9638e-04, 4.1738e-03, 4.2352e-04,
          3.3513e-04, 1.3001e-04, 1.4728e-04, 2.4933e-04, 1.4664e-04, 8.6384e-04,
          1.8721e-04, 9.0722e-05, 1.2541e-04, 6.1203e-05, 4.6490e-05, 9.9221e-05,
          1.1112e-04, 7.2872e-04, 1.1611e-04, 2.6052e-04, 6.6158e-05, 6.6312e-05,
          1.3783e-04, 2.7386e-04, 2.3723e-04, 4.4923e-04, 3.5560e-04, 3.8352e-04,
          2.5392e-04, 2.8530e-04, 9.2668e-05, 2.0202e-04, 1.0826e-04, 9.9107e-05,
          3.2974e-04, 3.2466e-04, 4.6943e-05, 2.5105e-04, 5.8856e-05, 9.7441e-05,
          1.0354e-04, 7.6333e-05, 3.3386e-04, 6.5099e-04, 1.5275e-04, 1.9416e-04,
          4.4679e-04, 1.8347e-04, 2.1011e-04, 8.6664e-04, 1.4200e-03, 9.3851e-05,
          1.7073e-04, 7.3264e-05, 4.0448e-04, 7.4957e-05, 1.5846e-04, 1.2160e-04,
          4.4864e-04, 7.0287e-05, 3.9632e-05, 1.1218e-03, 1.0826e-04, 8.1369e-05,
          9.9515e-05, 3.9228e-04, 6.4086e-05, 9.5641e-05, 1.2261e-03, 6.4610e-05,
          1.2837e-04, 2.1838e-04, 2.8487e-04, 6.0688e-04, 1.8279e-04, 1.7392e-04,
          3.1756e-04, 1.0667e-04, 1.2743e-04, 3.9163e-05, 1.7824e-04, 4.2241e-05,
          4.6391e-04, 5.0295e-05, 1.7875e-04, 5.6323e-04, 6.2423e-04, 9.4622e-05,
          7.7773e-05, 1.3758e-04, 1.1967e-04, 1.7378e-04, 2.0248e-04, 1.0258e-03,
          8.5239e-04, 5.1087e-04, 2.4355e-04, 1.9823e-04, 1.8374e-04, 1.5085e-04,
          2.3796e-04, 1.8776e-04, 5.7465e-05, 6.3790e-05, 3.4165e-04, 9.5112e-05,
          5.6243e-04, 4.7405e-04, 2.2266e-04, 5.7562e-04, 2.8875e-04, 1.7785e-04,
          2.9791e-04, 2.0216e-04, 2.6998e-04, 2.3697e-04, 1.6637e-04, 4.7745e-04,
          3.3174e-04, 3.0976e-04, 5.2784e-04, 1.6161e-04, 2.6251e-04, 2.7448e-04,
          2.3382e-04, 8.4209e-05, 4.6084e-04, 1.4839e-04, 1.1032e-04, 2.1559e-04,
          1.7987e-04, 8.6988e-05, 4.0492e-04, 1.1255e-04, 1.3571e-04, 3.2855e-04,
          1.5784e-04, 6.5382e-05, 3.0732e-04, 1.8185e-04, 2.7701e-04, 3.1875e-04,
          4.4437e-04, 7.8454e-05, 1.7837e-04, 9.9173e-05, 1.8530e-04, 2.1987e-04,
          7.0112e-05, 1.3694e-04, 5.2456e-05, 1.0818e-04, 3.1824e-04, 6.8720e-05,
          9.0436e-05, 2.7449e-04, 1.8330e-04, 1.4412e-04, 6.6453e-04, 1.1288e-04,
          2.9804e-04, 2.3326e-04, 2.8965e-04, 1.9440e-04, 4.1709e-04, 1.0536e-04,
          3.7711e-04, 6.1980e-05, 1.1848e-04, 1.4644e-04, 7.9953e-04, 1.7353e-04,
          8.2381e-05, 7.7113e-05, 5.5912e-05, 3.5328e-04, 2.5313e-04, 5.3672e-05,
          3.4913e-04, 1.4515e-04, 1.6307e-04, 3.7487e-04, 2.8077e-04, 4.8290e-04,
          2.3139e-04, 3.2798e-04, 2.4286e-04, 2.6545e-04, 2.3980e-04, 9.7660e-05,
          7.7318e-05, 7.5507e-05, 5.2555e-05, 7.9114e-05, 1.2696e-04, 2.6206e-04,
          5.4338e-05, 3.8443e-05, 9.2988e-05, 1.1244e-04, 2.0776e-04, 1.0503e-03,
          2.2755e-04, 1.4249e-04, 5.7393e-04, 3.6983e-04, 7.9156e-05, 2.3932e-04,
          2.8100e-04, 1.5972e-04, 1.4647e-04, 2.4448e-04, 2.6540e-04, 1.4567e-04,
          1.5511e-04, 9.7427e-05, 3.5339e-04, 7.1898e-05, 3.5075e-04, 3.3920e-04,
          1.0057e-04, 1.8501e-04, 4.9861e-05, 8.5567e-05, 1.8675e-04, 1.4912e-04,
          1.9388e-04, 1.1504e-04, 1.0337e-04, 1.5736e-04, 3.7247e-04, 3.1613e-04,
          2.6666e-04, 5.8089e-04, 1.5098e-04, 1.2832e-04, 8.6187e-05, 1.0091e-04,
          7.1094e-05, 4.7238e-05, 4.9482e-05, 3.7051e-04, 2.1852e-04, 7.0067e-05,
          6.0004e-05, 6.6218e-05, 1.5442e-04, 2.0778e-04, 1.3617e-04, 5.8878e-04,
          4.3551e-04, 7.4310e-05, 2.6892e-04, 5.4615e-05, 2.0807e-04, 9.4929e-05,
          7.9522e-05, 4.2362e-04, 1.3691e-04, 1.2595e-04, 5.2284e-04, 1.5313e-04,
          1.5114e-04, 3.8704e-04, 1.3326e-04, 4.0430e-05, 7.0828e-05, 9.6144e-05,
          6.5870e-05, 3.8407e-05, 2.4950e-04, 2.2867e-04, 4.1355e-04, 1.3742e-04,
          5.4959e-04, 2.7902e-05, 1.1873e-04, 5.2440e-04, 2.9745e-04, 1.3770e-04,
          1.8021e-04, 4.8587e-04, 2.1502e-04, 9.1963e-05, 2.8663e-04, 9.0168e-05,
          2.7836e-04, 3.0480e-04, 9.1988e-05, 1.3644e-04, 3.5886e-04, 1.3022e-04,
          3.8889e-05, 4.9729e-04, 1.9043e-04, 2.4287e-04, 2.6940e-04, 1.0348e-04,
          4.5335e-04, 2.1408e-04, 1.5841e-04, 1.4191e-04, 6.0559e-05, 4.7868e-04,
          1.4095e-04, 8.9189e-05, 2.8438e-04, 2.4094e-04, 1.0519e-04, 4.1664e-04,
          1.9094e-04, 5.3068e-05, 2.0951e-04, 2.0569e-04, 1.3674e-03, 4.2856e-04,
          5.9888e-04, 2.7372e-04, 3.0171e-04, 1.5462e-04, 9.1959e-05, 1.4482e-04,
          5.4701e-05, 1.1784e-04, 2.3292e-04, 2.3724e-04, 1.0747e-04, 1.9464e-04,
          3.6030e-05, 8.8649e-05, 5.8610e-05, 1.1001e-04, 1.3293e-04, 1.0297e-03,
          2.3060e-04, 1.3170e-04, 2.1036e-04, 5.1488e-04, 4.3239e-04, 9.9392e-03,
          1.4367e-02, 2.0108e-02, 9.8894e-03, 7.9582e-03, 4.2576e-03, 1.5099e-02,
          4.1690e-02, 1.2864e-02, 2.5238e-02, 2.0747e-02, 2.0148e-02, 8.0638e-03,
          3.2781e-02, 6.8054e-02, 3.7513e-02, 2.1162e-02, 1.1307e-02, 6.4015e-03,
          2.8696e-02, 4.2048e-02, 2.8832e-02, 1.8949e-02, 2.0812e-02, 3.3878e-02,
          4.9674e-02, 1.6057e-02, 2.1273e-02, 4.0634e-02, 1.5071e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.2060e-04, 2.7249e-02, 4.1054e-04, 2.4433e-03, 3.2430e-03,
          2.8838e-02, 1.3305e-03, 1.7891e-04, 8.6006e-05, 7.1169e-04, 9.5287e-04,
          3.2337e-04, 3.0699e-04, 8.9637e-05, 1.3074e-03, 1.2483e-04, 3.0692e-04,
          1.3847e-04, 6.3775e-05, 4.0433e-05, 5.2081e-05, 4.3867e-05, 2.7600e-04,
          9.4803e-04, 9.2797e-04, 1.0391e-03, 7.4376e-04, 5.6053e-05, 2.6068e-04,
          1.1224e-04, 3.1956e-04, 6.7880e-05, 1.1460e-03, 4.5306e-04, 2.8462e-04,
          1.5703e-03, 3.2979e-04, 6.0307e-04, 6.1995e-04, 5.8713e-05, 4.3819e-04,
          1.1506e-04, 1.1443e-04, 8.8435e-05, 1.0214e-04, 2.4447e-04, 1.7050e-03,
          4.3945e-05, 5.6966e-05, 3.5541e-04, 3.4597e-04, 2.1469e-04, 1.6546e-04,
          5.5874e-05, 1.1529e-04, 6.7767e-05, 6.7624e-04, 9.0435e-04, 1.0851e-03,
          4.5336e-04, 1.3454e-04, 1.0257e-04, 2.9813e-04, 2.1343e-04, 2.8424e-03,
          1.6695e-04, 1.2503e-04, 2.5446e-04, 6.4761e-05, 5.2022e-05, 7.8469e-05,
          7.3382e-05, 1.0934e-03, 7.9515e-05, 2.3040e-04, 4.8085e-05, 4.9091e-05,
          1.3258e-04, 5.2467e-04, 2.7016e-04, 1.4381e-03, 4.3614e-04, 5.5467e-04,
          2.5107e-04, 2.8280e-04, 6.7227e-05, 2.8695e-04, 1.1109e-04, 9.0896e-05,
          5.1782e-04, 5.1085e-04, 5.5754e-05, 3.7964e-04, 9.2865e-05, 9.1824e-05,
          6.9973e-05, 4.4561e-05, 2.6164e-04, 1.0494e-03, 3.9324e-04, 5.4143e-04,
          7.2856e-04, 1.3723e-04, 2.1383e-04, 3.0702e-03, 3.9873e-03, 8.1585e-05,
          2.2001e-04, 1.0584e-04, 5.5653e-04, 6.7158e-05, 1.1776e-04, 1.0752e-04,
          9.0718e-04, 2.0732e-04, 2.6484e-05, 3.8660e-03, 9.6285e-05, 4.5805e-05,
          9.9480e-05, 1.0701e-03, 8.2516e-05, 9.4671e-05, 3.2825e-03, 9.4048e-05,
          2.6736e-04, 2.0692e-04, 1.8463e-04, 2.3471e-03, 2.0421e-04, 2.4151e-04,
          1.8280e-04, 9.0846e-05, 1.8263e-04, 4.9192e-05, 2.9081e-04, 5.6009e-05,
          7.9600e-04, 4.5031e-05, 1.8591e-04, 4.4400e-04, 6.7099e-04, 1.1663e-04,
          1.2950e-04, 2.0803e-04, 2.7308e-04, 4.3303e-04, 1.9867e-04, 3.5440e-03,
          2.0245e-03, 1.7924e-03, 4.6189e-04, 2.9261e-04, 1.7267e-04, 3.7946e-04,
          3.8280e-04, 3.1310e-04, 9.0837e-05, 6.8468e-05, 1.1200e-03, 1.2148e-04,
          1.3718e-03, 6.6210e-04, 3.1876e-04, 8.2756e-04, 4.3849e-04, 3.0718e-04,
          3.2229e-04, 3.8937e-04, 9.6512e-04, 6.1787e-04, 2.2840e-04, 1.1359e-03,
          6.9575e-04, 3.9178e-04, 5.6546e-04, 1.9649e-04, 1.0171e-03, 3.4643e-04,
          4.9023e-04, 2.6331e-04, 1.2093e-03, 1.4640e-04, 1.1825e-04, 3.2637e-04,
          2.7620e-04, 7.8934e-05, 3.6657e-04, 2.0678e-04, 1.4076e-04, 8.9573e-04,
          1.7473e-04, 1.3585e-04, 6.9298e-04, 2.6390e-04, 3.7604e-04, 7.8342e-04,
          4.6195e-04, 5.2355e-05, 6.7439e-04, 1.7265e-04, 2.0450e-04, 4.0719e-04,
          6.7488e-05, 2.7259e-04, 8.4556e-05, 1.2821e-04, 4.0493e-04, 1.0289e-04,
          8.3628e-05, 3.8056e-04, 2.4758e-04, 1.9417e-04, 1.8703e-03, 1.6276e-04,
          6.7864e-04, 4.1596e-04, 4.0298e-04, 4.2389e-04, 1.0116e-03, 1.4418e-04,
          5.9414e-04, 8.4650e-05, 3.9561e-04, 1.3550e-04, 2.1900e-03, 3.8816e-04,
          8.6872e-05, 6.2416e-05, 6.6581e-05, 3.3356e-04, 7.3368e-04, 6.0507e-05,
          3.5399e-04, 2.9948e-04, 1.6001e-04, 6.5147e-04, 8.0276e-04, 7.1972e-04,
          2.1711e-04, 2.5430e-04, 3.5202e-04, 5.0438e-04, 4.0344e-04, 9.3231e-05,
          1.6351e-04, 1.3067e-04, 6.3649e-05, 1.0891e-04, 1.1672e-04, 9.0019e-04,
          3.1776e-05, 4.7534e-05, 7.9796e-05, 1.7639e-04, 3.7727e-04, 9.2799e-04,
          5.7627e-04, 1.2339e-04, 1.0351e-03, 8.4108e-04, 1.3402e-04, 2.9678e-04,
          6.9140e-04, 2.1272e-04, 1.7226e-04, 3.5096e-04, 2.7039e-04, 4.7621e-04,
          1.2966e-04, 1.0222e-04, 1.2587e-03, 1.5697e-04, 8.0507e-04, 8.4869e-04,
          2.0772e-04, 2.5682e-04, 2.8726e-05, 8.6992e-05, 2.1589e-04, 1.1797e-04,
          1.4146e-04, 1.3871e-04, 1.5580e-04, 2.2477e-04, 8.3410e-04, 8.9376e-04,
          3.3604e-04, 1.0185e-03, 2.3174e-04, 1.9771e-04, 1.1133e-04, 1.7582e-04,
          1.3754e-04, 7.7418e-05, 6.5000e-05, 1.0147e-03, 6.3889e-04, 9.3972e-05,
          6.7491e-05, 9.1608e-05, 1.2099e-04, 3.4740e-04, 1.8469e-04, 1.0468e-03,
          1.1155e-03, 8.8343e-05, 5.0511e-04, 6.5139e-05, 2.9942e-04, 1.3623e-04,
          1.2759e-04, 5.6835e-04, 1.3032e-04, 1.7648e-04, 9.3604e-04, 6.1635e-04,
          1.6859e-04, 5.4113e-04, 2.2101e-04, 6.3554e-05, 7.8487e-05, 1.1405e-04,
          8.0118e-05, 4.7174e-05, 2.4367e-04, 3.7766e-04, 7.2807e-04, 1.6799e-04,
          8.2139e-04, 2.4086e-05, 1.3238e-04, 6.2056e-04, 2.8808e-04, 1.5390e-04,
          2.7103e-04, 4.5504e-04, 2.0439e-04, 7.3735e-05, 5.1905e-04, 8.6357e-05,
          2.7765e-04, 6.3623e-04, 1.2673e-04, 1.7151e-04, 3.6295e-04, 1.6857e-04,
          2.4597e-05, 3.4627e-04, 3.1375e-04, 3.8185e-04, 2.3383e-04, 1.0169e-04,
          7.1858e-04, 2.6019e-04, 2.7431e-04, 1.5166e-04, 5.2080e-05, 9.1655e-04,
          1.1029e-04, 7.8739e-05, 5.0568e-04, 2.0025e-04, 1.3027e-04, 5.6676e-04,
          3.3016e-04, 3.6457e-05, 1.7855e-04, 2.1921e-04, 1.0581e-03, 4.9534e-04,
          2.8172e-04, 4.2503e-04, 4.8705e-04, 2.0166e-04, 9.7248e-05, 1.1247e-04,
          4.7022e-05, 1.0141e-04, 3.4137e-04, 3.0115e-04, 1.1300e-04, 2.7711e-04,
          4.0372e-05, 6.0829e-05, 6.2403e-05, 1.1513e-04, 2.2440e-04, 1.9583e-03,
          2.6638e-04, 1.7474e-04, 1.8563e-04, 6.3889e-04, 3.3094e-04, 2.4940e-03,
          7.1113e-03, 2.0011e-02, 1.5842e-02, 9.6649e-03, 4.2133e-03, 1.7576e-02,
          2.2077e-02, 5.1815e-03, 8.0842e-03, 3.0675e-03, 6.5999e-03, 7.9617e-03,
          1.7432e-02, 3.2461e-02, 2.6683e-02, 2.9138e-02, 1.2891e-02, 5.7870e-03,
          2.5858e-02, 4.7216e-02, 2.9208e-02, 1.8949e-02, 1.9840e-02, 3.5891e-02,
          4.8322e-02, 1.2835e-02, 1.4040e-02, 2.5309e-02, 1.2818e-01, 1.1693e-01,
          0.0000e+00]),
  tensor([0.0000e+00, 4.7537e-04, 2.3211e-02, 3.9616e-04, 2.7091e-03, 3.3435e-03,
          2.5505e-02, 1.4001e-03, 2.4320e-04, 9.4811e-05, 7.5131e-04, 9.6056e-04,
          3.1400e-04, 4.1263e-04, 1.5058e-04, 1.5138e-03, 1.6434e-04, 4.6231e-04,
          2.0629e-04, 6.4556e-05, 6.1658e-05, 7.4263e-05, 5.8559e-05, 4.1573e-04,
          9.0738e-04, 1.1613e-03, 1.1230e-03, 9.5610e-04, 9.7429e-05, 3.8447e-04,
          2.0151e-04, 4.0358e-04, 9.0803e-05, 1.4066e-03, 4.1729e-04, 3.2648e-04,
          1.5673e-03, 2.8384e-04, 6.9729e-04, 7.0972e-04, 9.7203e-05, 4.1857e-04,
          1.5523e-04, 1.5257e-04, 1.1306e-04, 1.1394e-04, 3.7202e-04, 2.0021e-03,
          6.4125e-05, 8.1827e-05, 5.0950e-04, 4.1389e-04, 2.4346e-04, 1.9370e-04,
          7.9742e-05, 1.2600e-04, 1.0128e-04, 5.3487e-04, 6.7352e-04, 7.3845e-04,
          4.7073e-04, 2.0744e-04, 1.2690e-04, 3.6231e-04, 1.6489e-04, 3.9018e-03,
          2.7686e-04, 1.2845e-04, 2.4086e-04, 8.4403e-05, 5.5353e-05, 8.1959e-05,
          8.1164e-05, 1.1134e-03, 1.1836e-04, 2.5803e-04, 6.0170e-05, 6.0962e-05,
          1.6247e-04, 4.7042e-04, 3.4879e-04, 1.4163e-03, 4.5568e-04, 4.1383e-04,
          2.3060e-04, 2.9810e-04, 8.8209e-05, 2.7196e-04, 1.4387e-04, 1.3545e-04,
          4.9400e-04, 4.2429e-04, 8.2870e-05, 3.4185e-04, 1.0974e-04, 1.1212e-04,
          1.1184e-04, 7.0451e-05, 3.0059e-04, 1.3061e-03, 3.5326e-04, 4.0253e-04,
          7.3576e-04, 1.8888e-04, 1.9790e-04, 2.4525e-03, 4.6567e-03, 1.2100e-04,
          2.2755e-04, 1.0780e-04, 4.1249e-04, 8.8007e-05, 1.9086e-04, 1.1623e-04,
          1.2661e-03, 2.6053e-04, 3.9427e-05, 4.4755e-03, 1.3507e-04, 7.0983e-05,
          1.5466e-04, 1.3165e-03, 9.1459e-05, 1.1182e-04, 3.5817e-03, 8.8099e-05,
          2.2984e-04, 1.9921e-04, 2.0928e-04, 2.7731e-03, 2.1309e-04, 2.4120e-04,
          2.3023e-04, 1.2714e-04, 1.7906e-04, 4.8060e-05, 2.1045e-04, 7.7277e-05,
          1.1147e-03, 6.5627e-05, 2.4656e-04, 4.5039e-04, 7.9299e-04, 1.8548e-04,
          1.4557e-04, 1.6662e-04, 3.8000e-04, 3.3197e-04, 3.1396e-04, 4.8172e-03,
          2.0560e-03, 1.9450e-03, 4.6342e-04, 3.1284e-04, 2.0388e-04, 4.2189e-04,
          2.4036e-04, 2.3739e-04, 7.9962e-05, 9.4468e-05, 1.3876e-03, 1.6873e-04,
          1.5506e-03, 6.0007e-04, 4.9099e-04, 9.8811e-04, 5.4197e-04, 2.3187e-04,
          3.1338e-04, 2.3905e-04, 1.2721e-03, 7.5995e-04, 3.8351e-04, 1.1277e-03,
          5.1636e-04, 3.8928e-04, 5.9161e-04, 2.1003e-04, 9.2930e-04, 3.9564e-04,
          5.7635e-04, 2.4652e-04, 1.3963e-03, 1.8213e-04, 1.4009e-04, 4.2384e-04,
          2.2053e-04, 1.0750e-04, 3.2427e-04, 2.5800e-04, 1.6666e-04, 1.0389e-03,
          1.8966e-04, 1.1695e-04, 7.5651e-04, 4.1208e-04, 3.4893e-04, 7.4061e-04,
          4.4831e-04, 7.7389e-05, 6.1946e-04, 1.1033e-04, 1.7701e-04, 3.5250e-04,
          6.6523e-05, 1.7196e-04, 9.2805e-05, 1.1162e-04, 3.2304e-04, 1.4504e-04,
          1.0141e-04, 5.4672e-04, 3.1438e-04, 1.8932e-04, 2.8079e-03, 2.1999e-04,
          7.8711e-04, 6.1051e-04, 3.8290e-04, 4.1671e-04, 1.1327e-03, 2.1093e-04,
          8.3494e-04, 1.0836e-04, 2.5141e-04, 1.3798e-04, 2.4392e-03, 4.1596e-04,
          9.8080e-05, 6.6859e-05, 6.7805e-05, 2.6492e-04, 9.2841e-04, 7.9827e-05,
          5.1954e-04, 1.7095e-04, 1.7118e-04, 6.5589e-04, 1.1475e-03, 5.9685e-04,
          2.2887e-04, 2.9108e-04, 2.9070e-04, 5.6130e-04, 6.3453e-04, 1.4673e-04,
          1.7750e-04, 9.4298e-05, 7.6372e-05, 9.0986e-05, 1.2449e-04, 7.7684e-04,
          3.5427e-05, 5.6923e-05, 9.7485e-05, 2.1252e-04, 4.8196e-04, 9.3855e-04,
          7.0638e-04, 1.3971e-04, 1.0610e-03, 8.0212e-04, 1.3518e-04, 2.8429e-04,
          6.6146e-04, 1.9997e-04, 2.2495e-04, 5.2502e-04, 4.2805e-04, 4.0482e-04,
          1.4276e-04, 1.1720e-04, 1.4290e-03, 1.6252e-04, 7.1765e-04, 1.3084e-03,
          1.7492e-04, 2.2014e-04, 4.2336e-05, 1.3100e-04, 2.6450e-04, 1.7292e-04,
          2.1728e-04, 1.5711e-04, 1.8668e-04, 2.3756e-04, 1.2451e-03, 8.1770e-04,
          2.6262e-04, 1.0282e-03, 3.4176e-04, 2.7407e-04, 1.4077e-04, 1.7366e-04,
          1.5844e-04, 8.4095e-05, 6.7269e-05, 1.2261e-03, 4.5479e-04, 1.3747e-04,
          1.1660e-04, 1.1801e-04, 1.2810e-04, 5.6447e-04, 2.5133e-04, 1.0020e-03,
          1.0960e-03, 1.2839e-04, 6.0211e-04, 9.6926e-05, 2.7034e-04, 1.1333e-04,
          1.9018e-04, 5.3064e-04, 1.9785e-04, 1.9335e-04, 1.0615e-03, 4.8443e-04,
          1.2472e-04, 4.9696e-04, 3.0004e-04, 8.3568e-05, 1.3104e-04, 1.6101e-04,
          1.0329e-04, 6.6356e-05, 3.3753e-04, 5.1069e-04, 8.8272e-04, 2.8210e-04,
          6.8076e-04, 3.0748e-05, 1.9294e-04, 6.2359e-04, 2.5342e-04, 1.9458e-04,
          1.9431e-04, 4.6151e-04, 2.3024e-04, 1.0800e-04, 3.2155e-04, 1.2668e-04,
          2.6433e-04, 8.1907e-04, 1.9457e-04, 1.7954e-04, 3.9635e-04, 2.3225e-04,
          3.0219e-05, 3.5964e-04, 3.7324e-04, 5.0657e-04, 2.8715e-04, 1.0521e-04,
          5.9467e-04, 3.0817e-04, 2.2735e-04, 1.4964e-04, 7.1298e-05, 8.5715e-04,
          1.3040e-04, 8.8502e-05, 5.5611e-04, 2.5231e-04, 1.4381e-04, 5.2584e-04,
          4.8891e-04, 4.9786e-05, 1.8976e-04, 1.8979e-04, 9.7093e-04, 5.4557e-04,
          3.4213e-04, 5.3045e-04, 5.7059e-04, 2.5879e-04, 9.3965e-05, 1.6978e-04,
          5.0281e-05, 1.3145e-04, 3.5745e-04, 3.4194e-04, 1.6057e-04, 3.5374e-04,
          5.0741e-05, 8.4216e-05, 1.0567e-04, 1.4495e-04, 2.3393e-04, 2.0654e-03,
          3.3613e-04, 1.5676e-04, 1.9216e-04, 5.7461e-04, 3.1416e-04, 2.4143e-03,
          8.3381e-03, 1.9419e-02, 1.2676e-02, 9.6917e-03, 4.1632e-03, 1.6357e-02,
          2.2358e-02, 5.1326e-03, 8.6788e-03, 2.5680e-03, 6.5913e-03, 6.7565e-03,
          1.5793e-02, 3.0095e-02, 2.3197e-02, 2.1699e-02, 1.1134e-02, 5.6284e-03,
          2.3073e-02, 4.7138e-02, 2.6831e-02, 1.4500e-02, 1.7102e-02, 2.7290e-02,
          3.6285e-02, 1.1003e-02, 1.0829e-02, 1.6817e-02, 9.7902e-02, 9.3981e-02,
          1.1158e-01, 0.0000e+00]),
  tensor([0.0000e+00, 4.1023e-04, 2.6558e-02, 5.2357e-04, 3.1328e-03, 3.8198e-03,
          2.8066e-02, 1.2411e-03, 2.5367e-04, 1.7442e-04, 8.2506e-04, 1.0780e-03,
          4.4361e-04, 9.3459e-04, 2.5862e-04, 1.8330e-03, 1.9825e-04, 1.1806e-03,
          5.2943e-04, 1.1831e-04, 8.3614e-05, 8.1378e-05, 6.5538e-05, 7.1469e-04,
          1.2825e-03, 1.1487e-03, 1.0709e-03, 8.8922e-04, 1.4195e-04, 1.0697e-03,
          3.6392e-04, 5.6337e-04, 2.4796e-04, 3.3592e-03, 8.2860e-04, 4.4267e-04,
          2.6193e-03, 2.6704e-04, 1.0348e-03, 1.6733e-03, 1.2324e-04, 4.8070e-04,
          1.1610e-04, 1.2545e-04, 9.2864e-05, 1.2102e-04, 1.0734e-03, 3.1680e-03,
          1.0299e-04, 9.6662e-05, 5.5295e-04, 5.8520e-04, 3.1526e-04, 3.1546e-04,
          6.8098e-05, 1.7162e-04, 1.3314e-04, 5.2917e-04, 4.1702e-04, 5.5826e-04,
          4.6587e-04, 4.1505e-04, 1.2576e-04, 5.3043e-04, 1.6428e-04, 5.4852e-03,
          1.8117e-04, 1.1375e-04, 3.0612e-04, 9.1851e-05, 4.8010e-05, 6.9130e-05,
          7.5977e-05, 1.2440e-03, 1.0597e-04, 3.6206e-04, 5.8452e-05, 6.0083e-05,
          1.9480e-04, 4.1230e-04, 2.1925e-04, 1.3241e-03, 2.3780e-04, 2.9650e-04,
          1.7059e-04, 4.6889e-04, 1.0684e-04, 3.3534e-04, 8.5721e-05, 8.3469e-05,
          7.8952e-04, 5.5355e-04, 7.2447e-05, 4.8094e-04, 1.7050e-04, 1.2768e-04,
          1.4694e-04, 5.8543e-05, 3.8534e-04, 1.9139e-03, 3.4181e-04, 4.6283e-04,
          1.0072e-03, 1.0372e-04, 1.3530e-04, 2.0416e-03, 4.3746e-03, 3.1104e-04,
          3.7339e-04, 1.0349e-04, 5.6537e-04, 5.8897e-05, 1.1245e-04, 8.5935e-05,
          1.4030e-03, 4.7453e-04, 8.4339e-05, 6.4769e-03, 1.4807e-04, 1.7262e-04,
          4.1559e-04, 1.8651e-03, 7.8925e-05, 1.4742e-04, 3.3138e-03, 6.6371e-05,
          2.5041e-04, 1.3573e-04, 1.5374e-04, 5.8856e-03, 4.8682e-04, 4.2764e-04,
          1.7521e-04, 8.9178e-05, 2.1308e-04, 3.7362e-05, 2.6316e-04, 6.6276e-05,
          2.6721e-03, 1.4006e-04, 4.3890e-04, 5.8330e-04, 1.0714e-03, 4.5381e-04,
          1.0970e-04, 2.0364e-04, 2.9930e-04, 3.7717e-04, 1.7047e-04, 5.2378e-03,
          1.9940e-03, 3.7923e-03, 7.0829e-04, 5.5359e-04, 2.4166e-04, 5.7558e-04,
          2.9311e-04, 2.5182e-04, 1.3878e-04, 7.0825e-05, 2.9827e-03, 5.1492e-04,
          2.9917e-03, 8.0812e-04, 1.1228e-03, 1.2668e-03, 6.4095e-04, 2.2063e-04,
          3.6503e-04, 2.2886e-04, 1.6049e-03, 1.3590e-03, 5.7641e-04, 2.1487e-03,
          7.2547e-04, 3.6092e-04, 6.0120e-04, 1.8739e-04, 1.1249e-03, 6.7967e-04,
          4.4262e-04, 7.2630e-04, 1.6781e-03, 2.5882e-04, 2.3473e-04, 6.8454e-04,
          3.8802e-04, 1.9346e-04, 3.0368e-04, 2.4101e-04, 1.0729e-04, 1.3567e-03,
          3.1660e-04, 1.1865e-04, 1.5587e-03, 9.8870e-04, 3.9063e-04, 1.5378e-03,
          5.1698e-04, 1.8293e-04, 1.6502e-03, 9.2984e-05, 1.2523e-04, 4.9950e-04,
          5.1643e-05, 2.0715e-04, 1.1227e-04, 1.6636e-04, 5.3169e-04, 3.0719e-04,
          1.3117e-04, 7.1095e-04, 2.1649e-04, 1.4805e-04, 3.7067e-03, 1.6240e-04,
          1.2460e-03, 1.3877e-03, 5.9621e-04, 5.9777e-04, 2.9048e-03, 4.3819e-04,
          1.9231e-03, 8.8665e-05, 2.4072e-04, 8.7117e-05, 2.6943e-03, 8.2753e-04,
          7.7080e-05, 5.3668e-05, 7.3884e-05, 3.5513e-04, 1.4622e-03, 1.7088e-04,
          3.6715e-04, 1.3315e-04, 9.2575e-05, 1.0509e-03, 1.7843e-03, 8.4304e-04,
          2.2582e-04, 2.5549e-04, 2.8500e-04, 1.5569e-03, 2.0650e-03, 3.2312e-04,
          1.9285e-04, 1.1453e-04, 6.1484e-05, 9.5875e-05, 1.3459e-04, 1.2110e-03,
          3.1457e-05, 8.0780e-05, 1.2596e-04, 3.3376e-04, 1.3478e-03, 1.1570e-03,
          6.7824e-04, 9.4116e-05, 1.0801e-03, 1.0898e-03, 1.3956e-04, 3.1984e-04,
          1.0389e-03, 3.3070e-04, 5.1141e-04, 1.3665e-03, 6.7899e-04, 6.8978e-04,
          1.1446e-04, 9.0436e-05, 1.4004e-03, 1.5454e-04, 1.5317e-03, 2.3414e-03,
          3.1671e-04, 3.1810e-04, 6.5882e-05, 3.2460e-04, 2.4943e-04, 1.3033e-04,
          1.3272e-04, 1.3922e-04, 1.6508e-04, 2.4418e-04, 2.9172e-03, 1.0683e-03,
          4.2079e-04, 1.4216e-03, 7.1419e-04, 6.8535e-04, 1.3751e-04, 2.0863e-04,
          1.5738e-04, 6.9977e-05, 5.9529e-05, 1.8920e-03, 7.1294e-04, 1.4796e-04,
          1.3041e-04, 3.2959e-04, 1.9759e-04, 1.1747e-03, 1.6905e-04, 1.0899e-03,
          1.7641e-03, 3.5038e-04, 1.1065e-03, 2.2375e-04, 3.0736e-04, 1.9999e-04,
          4.2383e-04, 4.7999e-04, 3.4956e-04, 4.5811e-04, 1.7354e-03, 1.4954e-03,
          1.8510e-04, 7.0948e-04, 7.5143e-04, 2.9702e-04, 1.4247e-04, 2.2035e-04,
          2.3948e-04, 1.4767e-04, 4.8845e-04, 3.4665e-04, 1.1177e-03, 7.8492e-04,
          7.8270e-04, 7.9637e-05, 5.7615e-04, 8.1523e-04, 2.4836e-04, 3.1072e-04,
          2.2041e-04, 6.6232e-04, 3.4220e-04, 2.5615e-04, 3.7871e-04, 2.8322e-04,
          2.4299e-04, 7.1584e-04, 5.8687e-04, 1.8075e-04, 5.5251e-04, 4.5818e-04,
          4.3124e-05, 3.0386e-04, 2.6972e-04, 3.3971e-04, 4.2237e-04, 1.7660e-04,
          8.2643e-04, 3.3631e-04, 2.1249e-04, 1.6315e-04, 1.3292e-04, 6.8085e-04,
          1.6889e-04, 1.3154e-04, 8.1449e-04, 2.9455e-04, 1.3064e-04, 6.6002e-04,
          4.7170e-04, 9.3362e-05, 2.4353e-04, 1.7564e-04, 9.9970e-04, 5.5488e-04,
          2.8446e-04, 3.9922e-04, 4.9377e-04, 3.6857e-04, 1.4351e-04, 3.0424e-04,
          7.6145e-05, 1.5652e-04, 3.5473e-04, 5.4779e-04, 3.3218e-04, 3.0211e-04,
          7.3397e-05, 1.0218e-04, 1.9741e-04, 1.9042e-04, 2.9118e-04, 1.6886e-03,
          4.7342e-04, 2.2418e-04, 2.1301e-04, 7.6505e-04, 2.6668e-04, 2.8254e-03,
          5.7248e-03, 1.8683e-02, 1.5817e-02, 7.0450e-03, 4.7154e-03, 1.7829e-02,
          1.3807e-02, 3.4434e-03, 5.2408e-03, 1.6935e-03, 5.2450e-03, 5.3243e-03,
          1.1547e-02, 2.0314e-02, 1.6576e-02, 2.3031e-02, 7.8227e-03, 5.3065e-03,
          2.1014e-02, 1.7095e-02, 1.2505e-02, 1.4595e-02, 1.2080e-02, 2.3408e-02,
          2.2562e-02, 9.3217e-03, 8.5321e-03, 1.3974e-02, 5.5444e-02, 5.5628e-02,
          7.9437e-02, 1.5291e-01, 0.0000e+00]),
  tensor([0.0000e+00, 3.7527e-04, 2.4209e-02, 5.1165e-04, 3.0273e-03, 3.5504e-03,
          2.5006e-02, 1.1472e-03, 2.5603e-04, 1.9944e-04, 7.1503e-04, 9.2345e-04,
          3.8045e-04, 9.0290e-04, 2.3770e-04, 1.6494e-03, 2.0125e-04, 1.0046e-03,
          4.6564e-04, 1.0641e-04, 8.0486e-05, 7.5734e-05, 7.2251e-05, 6.4995e-04,
          1.0731e-03, 1.0192e-03, 9.9633e-04, 7.7063e-04, 1.4491e-04, 9.7124e-04,
          3.1882e-04, 6.0490e-04, 2.4949e-04, 2.5298e-03, 7.6902e-04, 4.4151e-04,
          2.1893e-03, 2.4148e-04, 8.0253e-04, 1.2984e-03, 1.2308e-04, 4.1167e-04,
          1.0404e-04, 1.1836e-04, 8.6788e-05, 1.1437e-04, 9.0635e-04, 2.6253e-03,
          9.5166e-05, 9.0361e-05, 5.1893e-04, 5.8391e-04, 3.6964e-04, 3.7541e-04,
          1.1868e-04, 2.1837e-04, 1.5633e-04, 4.5735e-04, 3.5071e-04, 4.8452e-04,
          3.8972e-04, 3.7500e-04, 1.3175e-04, 4.2382e-04, 1.4688e-04, 4.2541e-03,
          1.2211e-04, 1.0060e-04, 2.3876e-04, 7.9750e-05, 4.3144e-05, 7.0366e-05,
          7.8075e-05, 1.0050e-03, 9.8792e-05, 5.1078e-04, 7.1544e-05, 7.1420e-05,
          2.2981e-04, 3.6446e-04, 1.6555e-04, 1.0677e-03, 1.8881e-04, 2.9045e-04,
          1.6101e-04, 5.2993e-04, 1.0249e-04, 2.7795e-04, 6.1587e-05, 6.4802e-05,
          6.3160e-04, 4.9293e-04, 5.6933e-05, 4.0785e-04, 1.9307e-04, 1.2847e-04,
          1.6575e-04, 5.6759e-05, 4.6591e-04, 1.5419e-03, 2.6877e-04, 3.5281e-04,
          8.1498e-04, 8.5017e-05, 1.1198e-04, 1.7777e-03, 3.9791e-03, 2.6939e-04,
          4.1262e-04, 8.4331e-05, 4.5109e-04, 5.6750e-05, 8.7777e-05, 6.8834e-05,
          1.1661e-03, 3.7372e-04, 8.8364e-05, 5.2448e-03, 1.4234e-04, 1.9669e-04,
          3.8911e-04, 1.3621e-03, 6.9545e-05, 1.4971e-04, 3.0259e-03, 5.3685e-05,
          2.3050e-04, 1.1250e-04, 1.4293e-04, 3.7542e-03, 4.9643e-04, 4.8548e-04,
          1.7088e-04, 6.6156e-05, 2.0813e-04, 2.5902e-05, 2.0650e-04, 5.9097e-05,
          2.7199e-03, 1.3259e-04, 4.9225e-04, 6.3547e-04, 1.0103e-03, 4.4151e-04,
          8.4881e-05, 1.9613e-04, 2.6079e-04, 2.9123e-04, 1.2946e-04, 3.7397e-03,
          1.5665e-03, 2.5166e-03, 6.2825e-04, 4.3050e-04, 2.3572e-04, 4.3754e-04,
          2.5971e-04, 2.2584e-04, 1.0998e-04, 5.8023e-05, 2.3458e-03, 4.7330e-04,
          2.1404e-03, 7.7155e-04, 1.0774e-03, 1.1461e-03, 4.8787e-04, 1.6936e-04,
          3.3003e-04, 1.9101e-04, 1.3928e-03, 1.2441e-03, 5.5680e-04, 1.6892e-03,
          7.6208e-04, 3.6399e-04, 6.0176e-04, 1.4857e-04, 8.7735e-04, 6.6970e-04,
          3.6637e-04, 5.5491e-04, 1.2795e-03, 2.5400e-04, 2.2441e-04, 7.1092e-04,
          3.4440e-04, 2.4999e-04, 2.9965e-04, 1.7470e-04, 8.7149e-05, 1.0946e-03,
          3.2034e-04, 1.0773e-04, 1.1192e-03, 9.2782e-04, 3.2664e-04, 1.1236e-03,
          5.5237e-04, 1.8262e-04, 1.1805e-03, 7.8023e-05, 9.8014e-05, 3.6418e-04,
          4.5365e-05, 1.5890e-04, 1.0501e-04, 1.5864e-04, 4.3963e-04, 2.9711e-04,
          1.1857e-04, 5.6256e-04, 1.6926e-04, 1.2072e-04, 2.6202e-03, 1.3534e-04,
          1.0277e-03, 1.2934e-03, 5.4015e-04, 5.9550e-04, 2.2230e-03, 4.6109e-04,
          1.4062e-03, 6.8042e-05, 1.9278e-04, 7.3828e-05, 2.1232e-03, 6.2548e-04,
          6.3179e-05, 4.8099e-05, 6.8123e-05, 4.0996e-04, 1.3153e-03, 1.8938e-04,
          2.5884e-04, 1.1130e-04, 1.1368e-04, 9.8072e-04, 1.3468e-03, 9.0222e-04,
          2.2533e-04, 2.7207e-04, 2.8836e-04, 1.4193e-03, 2.1463e-03, 3.1447e-04,
          1.4726e-04, 1.0032e-04, 5.5164e-05, 1.0315e-04, 1.2429e-04, 8.8718e-04,
          3.0066e-05, 7.3808e-05, 1.2546e-04, 2.5880e-04, 1.3401e-03, 1.2035e-03,
          5.0160e-04, 8.8902e-05, 9.8793e-04, 9.5747e-04, 1.4193e-04, 3.0159e-04,
          9.4564e-04, 2.9823e-04, 5.3440e-04, 1.6080e-03, 7.1295e-04, 5.5620e-04,
          1.1345e-04, 7.7348e-05, 1.1189e-03, 1.2966e-04, 1.0232e-03, 1.6987e-03,
          3.2163e-04, 3.4240e-04, 6.2325e-05, 3.5986e-04, 1.9843e-04, 1.1652e-04,
          1.1408e-04, 1.2543e-04, 1.4517e-04, 2.1074e-04, 2.2967e-03, 9.1596e-04,
          4.5366e-04, 1.1387e-03, 7.6343e-04, 6.2907e-04, 1.0968e-04, 2.0712e-04,
          1.2284e-04, 5.8052e-05, 4.7473e-05, 1.3672e-03, 5.1857e-04, 1.4392e-04,
          1.1859e-04, 2.9894e-04, 2.0434e-04, 1.2070e-03, 1.3819e-04, 1.0212e-03,
          1.3608e-03, 3.2752e-04, 1.0328e-03, 2.5823e-04, 3.2021e-04, 1.8210e-04,
          4.4925e-04, 5.8280e-04, 3.2790e-04, 3.5436e-04, 1.7055e-03, 9.0044e-04,
          1.6526e-04, 5.8447e-04, 7.2780e-04, 2.4172e-04, 1.2404e-04, 1.6451e-04,
          2.1554e-04, 1.8300e-04, 4.4454e-04, 3.1063e-04, 1.0828e-03, 9.0842e-04,
          6.6755e-04, 7.8261e-05, 6.9944e-04, 6.6808e-04, 2.4395e-04, 2.8201e-04,
          2.7105e-04, 6.6973e-04, 3.4161e-04, 2.1703e-04, 3.2073e-04, 3.1784e-04,
          2.5546e-04, 6.9493e-04, 5.5594e-04, 1.7716e-04, 5.5889e-04, 4.3985e-04,
          4.0694e-05, 2.8725e-04, 2.4686e-04, 3.1270e-04, 4.8695e-04, 2.0807e-04,
          9.0556e-04, 4.1130e-04, 2.1399e-04, 1.6982e-04, 1.4916e-04, 6.4548e-04,
          1.8126e-04, 1.4033e-04, 6.6153e-04, 2.9769e-04, 1.4084e-04, 5.2591e-04,
          5.1674e-04, 9.3715e-05, 2.6210e-04, 1.5229e-04, 8.1169e-04, 4.6605e-04,
          3.1195e-04, 3.5210e-04, 4.4225e-04, 4.2908e-04, 1.5262e-04, 2.5398e-04,
          8.4870e-05, 1.4592e-04, 3.2431e-04, 5.4743e-04, 3.4397e-04, 3.1975e-04,
          6.2429e-05, 1.0179e-04, 2.0036e-04, 1.8307e-04, 2.7642e-04, 1.3787e-03,
          4.5245e-04, 2.8214e-04, 2.1180e-04, 6.1089e-04, 2.2577e-04, 2.5793e-03,
          4.8214e-03, 1.6668e-02, 1.3033e-02, 5.5357e-03, 3.9850e-03, 1.5119e-02,
          1.2765e-02, 3.3684e-03, 4.8305e-03, 2.2698e-03, 4.3634e-03, 4.8306e-03,
          1.0727e-02, 1.7805e-02, 1.3445e-02, 1.7157e-02, 5.7789e-03, 4.4375e-03,
          1.6813e-02, 1.2969e-02, 9.4222e-03, 1.3461e-02, 1.1035e-02, 2.0561e-02,
          1.8760e-02, 8.0980e-03, 8.6777e-03, 1.3407e-02, 4.8067e-02, 4.1227e-02,
          5.6328e-02, 1.6958e-01, 1.1816e-01, 0.0000e+00]),
  tensor([0.0000e+00, 3.5356e-04, 2.3934e-02, 4.5305e-04, 2.6057e-03, 3.8905e-03,
          2.3935e-02, 1.3042e-03, 1.9565e-04, 1.3496e-04, 5.3785e-04, 7.0706e-04,
          2.5751e-04, 8.8575e-04, 2.0284e-04, 1.8618e-03, 1.6413e-04, 1.1193e-03,
          4.6304e-04, 1.0135e-04, 8.1352e-05, 7.7034e-05, 6.3532e-05, 6.4271e-04,
          7.2903e-04, 9.3474e-04, 1.0296e-03, 8.7770e-04, 1.1781e-04, 9.7114e-04,
          2.7702e-04, 3.8916e-04, 2.2811e-04, 1.7521e-03, 5.5286e-04, 4.3815e-04,
          8.3154e-04, 1.0395e-04, 7.4844e-04, 9.6567e-04, 1.1176e-04, 2.4636e-04,
          7.3835e-05, 6.9878e-05, 7.0230e-05, 6.7559e-05, 9.7083e-04, 1.7517e-03,
          9.3573e-05, 1.2330e-04, 4.7417e-04, 3.8204e-04, 2.0823e-04, 2.7315e-04,
          6.8446e-05, 1.2598e-04, 9.0344e-05, 1.8824e-04, 2.2870e-04, 2.2767e-04,
          2.0006e-04, 2.7773e-04, 9.2466e-05, 2.4286e-04, 9.1235e-05, 2.1822e-03,
          1.0407e-04, 5.9102e-05, 1.6565e-04, 5.1642e-05, 3.5105e-05, 5.6076e-05,
          6.7792e-05, 8.8515e-04, 8.0827e-05, 2.3966e-04, 6.5934e-05, 6.3155e-05,
          1.4805e-04, 1.8729e-04, 1.0621e-04, 4.8420e-04, 1.1115e-04, 1.8626e-04,
          1.0587e-04, 3.1722e-04, 7.0298e-05, 2.0404e-04, 5.3380e-05, 5.8005e-05,
          3.6889e-04, 3.0208e-04, 5.5350e-05, 2.4850e-04, 1.1275e-04, 9.2201e-05,
          1.2968e-04, 5.1133e-05, 4.0642e-04, 1.3050e-03, 1.3552e-04, 1.8412e-04,
          4.5952e-04, 7.0244e-05, 7.0292e-05, 7.6299e-04, 2.6198e-03, 2.3047e-04,
          2.5475e-04, 5.5183e-05, 2.9827e-04, 5.1090e-05, 7.1939e-05, 5.7909e-05,
          9.7725e-04, 1.6709e-04, 7.7559e-05, 2.8562e-03, 1.1197e-04, 1.2558e-04,
          2.7360e-04, 7.8122e-04, 4.5222e-05, 1.0945e-04, 1.9326e-03, 3.6603e-05,
          1.1057e-04, 6.8678e-05, 9.0279e-05, 2.1141e-03, 4.4557e-04, 2.9305e-04,
          1.3281e-04, 5.2176e-05, 2.7910e-04, 2.8336e-05, 1.3505e-04, 5.6162e-05,
          2.8531e-03, 1.1465e-04, 7.2651e-04, 8.3935e-04, 1.3169e-03, 3.5245e-04,
          5.6253e-05, 2.4356e-04, 1.7673e-04, 1.7024e-04, 1.0006e-04, 1.8170e-03,
          7.7494e-04, 1.0684e-03, 2.9637e-04, 3.0215e-04, 1.6391e-04, 1.9567e-04,
          1.8779e-04, 1.6090e-04, 9.3451e-05, 5.1765e-05, 8.6864e-04, 4.5059e-04,
          1.5127e-03, 6.8239e-04, 1.2040e-03, 1.4246e-03, 2.9908e-04, 1.2653e-04,
          1.9947e-04, 1.3665e-04, 6.0570e-04, 1.1363e-03, 6.1966e-04, 1.0906e-03,
          4.0101e-04, 2.5039e-04, 3.5819e-04, 9.9235e-05, 4.5975e-04, 5.8825e-04,
          3.1742e-04, 3.7459e-04, 9.2856e-04, 2.0120e-04, 2.2062e-04, 9.4532e-04,
          2.6529e-04, 2.1484e-04, 4.5001e-04, 1.0081e-04, 7.2334e-05, 7.6859e-04,
          2.4050e-04, 5.9217e-05, 8.4729e-04, 9.8447e-04, 2.2190e-04, 6.0850e-04,
          3.0658e-04, 1.3704e-04, 7.1424e-04, 5.0884e-05, 8.4985e-05, 2.7890e-04,
          3.7576e-05, 1.0118e-04, 6.8625e-05, 1.5288e-04, 2.6899e-04, 2.6969e-04,
          9.9337e-05, 3.2304e-04, 1.0797e-04, 7.5105e-05, 1.3626e-03, 9.3829e-05,
          6.6086e-04, 1.2278e-03, 5.2862e-04, 8.9339e-04, 1.6412e-03, 3.7369e-04,
          1.3215e-03, 4.7278e-05, 1.1233e-04, 6.0263e-05, 1.4333e-03, 5.5508e-04,
          4.8075e-05, 4.5110e-05, 5.2869e-05, 2.5238e-04, 4.7992e-04, 1.9188e-04,
          1.9574e-04, 7.7134e-05, 7.3789e-05, 7.2113e-04, 5.4621e-04, 4.6406e-04,
          1.3064e-04, 1.8940e-04, 1.8498e-04, 7.1489e-04, 1.4165e-03, 2.5652e-04,
          8.3216e-05, 6.0033e-05, 4.6953e-05, 6.4815e-05, 9.7524e-05, 5.3718e-04,
          2.5538e-05, 5.2239e-05, 8.6809e-05, 1.7618e-04, 1.4824e-03, 1.5803e-03,
          2.9865e-04, 6.7862e-05, 5.4487e-04, 6.1560e-04, 9.1611e-05, 1.6120e-04,
          5.8475e-04, 3.3901e-04, 6.5950e-04, 1.3510e-03, 5.8175e-04, 3.4858e-04,
          8.7857e-05, 6.3963e-05, 6.9683e-04, 9.7620e-05, 8.4327e-04, 8.2672e-04,
          1.8395e-04, 2.6019e-04, 6.1011e-05, 2.8128e-04, 1.2687e-04, 7.3328e-05,
          7.7958e-05, 8.2137e-05, 8.6360e-05, 1.4847e-04, 1.8174e-03, 9.5555e-04,
          2.2519e-04, 9.8392e-04, 7.5143e-04, 7.2096e-04, 7.6063e-05, 1.2439e-04,
          7.9595e-05, 4.6039e-05, 3.9262e-05, 1.2749e-03, 3.9801e-04, 9.9741e-05,
          8.4360e-05, 3.1644e-04, 1.9456e-04, 1.2090e-03, 9.8225e-05, 5.3830e-04,
          1.1792e-03, 2.7346e-04, 7.4921e-04, 2.7493e-04, 4.5372e-04, 1.7145e-04,
          4.7359e-04, 4.0565e-04, 2.8002e-04, 2.8628e-04, 1.8927e-03, 5.6385e-04,
          1.3934e-04, 4.1603e-04, 7.1170e-04, 2.3226e-04, 1.0028e-04, 1.3007e-04,
          3.1819e-04, 2.0014e-04, 4.7586e-04, 1.8807e-04, 5.2566e-04, 6.3008e-04,
          5.1502e-04, 7.4357e-05, 4.9621e-04, 6.6336e-04, 3.1127e-04, 2.8899e-04,
          1.4986e-04, 7.8587e-04, 3.7834e-04, 1.7403e-04, 2.4624e-04, 2.5100e-04,
          3.2933e-04, 5.8154e-04, 5.7877e-04, 2.4636e-04, 6.1306e-04, 4.8140e-04,
          4.3604e-05, 2.9502e-04, 1.5204e-04, 1.9189e-04, 4.1097e-04, 1.4533e-04,
          1.1208e-03, 3.8237e-04, 2.1436e-04, 3.0559e-04, 1.3432e-04, 4.6817e-04,
          1.4918e-04, 1.1555e-04, 5.7802e-04, 3.4518e-04, 1.3625e-04, 6.4041e-04,
          4.2412e-04, 9.6570e-05, 2.7096e-04, 1.3554e-04, 1.1469e-03, 7.3478e-04,
          3.2560e-04, 2.5336e-04, 3.0759e-04, 3.7304e-04, 1.5790e-04, 2.4560e-04,
          6.3024e-05, 1.6473e-04, 2.9619e-04, 6.6345e-04, 4.1516e-04, 4.2912e-04,
          5.2700e-05, 1.0198e-04, 1.9078e-04, 1.9069e-04, 3.0067e-04, 1.1373e-03,
          5.4451e-04, 2.7978e-04, 2.4727e-04, 6.3134e-04, 2.5991e-04, 2.3170e-03,
          5.8787e-03, 1.7086e-02, 1.0231e-02, 5.4567e-03, 3.7480e-03, 1.1131e-02,
          1.3771e-02, 3.6557e-03, 5.4663e-03, 2.3100e-03, 4.8146e-03, 3.7729e-03,
          1.3048e-02, 1.9473e-02, 1.4309e-02, 1.1408e-02, 4.8989e-03, 3.7916e-03,
          1.2902e-02, 1.1335e-02, 8.4334e-03, 8.4268e-03, 8.1069e-03, 1.3494e-02,
          1.4272e-02, 6.4474e-03, 6.6840e-03, 9.1580e-03, 4.8811e-02, 3.1529e-02,
          3.3429e-02, 1.1179e-01, 1.2451e-01, 1.7635e-01, 0.0000e+00]),
  tensor([0.0000e+00, 3.1462e-04, 2.3063e-02, 3.4372e-04, 2.1409e-03, 3.3242e-03,
          2.6713e-02, 1.1746e-03, 3.2784e-04, 8.0899e-05, 4.4598e-04, 6.6600e-04,
          2.0255e-04, 1.8520e-04, 7.1833e-05, 1.2139e-03, 2.1152e-04, 1.8591e-04,
          9.3199e-05, 6.7467e-05, 3.8118e-05, 5.1045e-05, 3.2431e-05, 2.4999e-04,
          6.0495e-04, 9.9768e-04, 9.5575e-04, 7.3578e-04, 5.4082e-05, 1.5849e-04,
          8.2288e-05, 5.1814e-04, 4.8967e-05, 1.1554e-03, 3.4091e-04, 6.7411e-04,
          1.2010e-03, 1.5478e-04, 6.0387e-04, 7.9377e-04, 4.9130e-05, 2.5067e-04,
          6.5502e-05, 5.9637e-05, 5.2168e-05, 5.3229e-05, 1.4702e-04, 1.1323e-03,
          3.8392e-05, 5.9885e-05, 3.8287e-04, 4.9846e-04, 2.9716e-04, 2.7050e-04,
          6.6568e-05, 1.5837e-04, 9.9114e-05, 2.5852e-04, 5.9406e-04, 3.4236e-04,
          2.5339e-04, 1.0462e-04, 1.0190e-04, 3.4822e-04, 6.9002e-05, 2.0924e-03,
          8.4353e-05, 4.5177e-05, 1.3613e-04, 3.9446e-05, 2.8063e-05, 4.7317e-05,
          5.7540e-05, 7.8131e-04, 9.0888e-05, 3.1013e-04, 6.4156e-05, 6.5883e-05,
          1.5498e-04, 2.0110e-04, 9.7304e-05, 8.5017e-04, 1.3243e-04, 4.7142e-04,
          2.0871e-04, 4.4503e-04, 7.3756e-05, 2.0846e-04, 4.4162e-05, 4.8859e-05,
          2.9605e-04, 2.8178e-04, 3.8560e-05, 1.9780e-04, 1.0851e-04, 9.0871e-05,
          6.3843e-05, 5.6953e-05, 6.8859e-04, 7.0499e-04, 1.0993e-04, 1.6677e-04,
          4.1833e-04, 5.9694e-05, 5.9796e-05, 2.9678e-03, 3.4772e-03, 8.0919e-05,
          3.1365e-04, 5.8889e-05, 2.8076e-04, 4.9697e-05, 6.2094e-05, 5.6872e-05,
          8.8793e-04, 1.5770e-04, 3.6565e-05, 1.9122e-03, 1.0669e-04, 5.3966e-05,
          9.1603e-05, 6.2108e-04, 3.0272e-05, 1.2888e-04, 3.3770e-03, 3.4928e-05,
          1.0698e-04, 6.7627e-05, 6.9095e-05, 1.8504e-03, 2.2171e-04, 3.0797e-04,
          1.4380e-04, 4.4307e-05, 3.0236e-04, 2.1133e-05, 1.2597e-04, 4.0968e-05,
          5.1207e-04, 3.9705e-05, 1.9967e-04, 5.8000e-04, 7.0865e-04, 9.8957e-05,
          4.0979e-05, 2.3098e-04, 1.8511e-04, 1.3767e-04, 9.2772e-05, 3.0177e-03,
          1.1250e-03, 1.4531e-03, 3.4148e-04, 2.8393e-04, 1.6549e-04, 2.4633e-04,
          1.8947e-04, 1.7446e-04, 5.9921e-05, 4.4618e-05, 7.7376e-04, 1.1169e-04,
          1.0969e-03, 5.4412e-04, 1.8506e-04, 1.2083e-03, 2.7631e-04, 1.2300e-04,
          1.9083e-04, 1.5301e-04, 6.9933e-04, 6.4276e-04, 1.4302e-04, 1.0031e-03,
          1.5254e-03, 6.2454e-04, 5.7139e-04, 1.1463e-04, 7.8871e-04, 3.9291e-04,
          4.6395e-04, 4.0399e-04, 1.4826e-03, 2.7388e-04, 1.4404e-04, 4.2015e-04,
          1.8649e-04, 8.0986e-05, 4.4617e-04, 7.7669e-05, 6.1646e-05, 8.8507e-04,
          2.9702e-04, 4.9503e-05, 6.0176e-04, 1.5897e-04, 2.2447e-04, 5.7176e-04,
          4.7017e-04, 6.7963e-05, 7.9230e-04, 4.7897e-05, 1.0793e-04, 3.5025e-04,
          3.1044e-05, 1.0288e-04, 6.1381e-05, 2.0210e-04, 2.1016e-04, 5.6723e-05,
          9.7300e-05, 3.0508e-04, 8.9080e-05, 6.7885e-05, 1.1205e-03, 7.3709e-05,
          5.7651e-04, 3.0132e-04, 3.6350e-04, 6.4375e-04, 8.3702e-04, 1.4878e-04,
          5.3573e-04, 5.7498e-05, 1.7288e-04, 6.3365e-05, 2.8174e-03, 4.6094e-04,
          3.8209e-05, 5.5048e-05, 4.5487e-05, 6.9274e-04, 7.2634e-04, 6.3612e-05,
          1.5967e-04, 8.2536e-05, 1.1573e-04, 1.7723e-03, 6.4293e-04, 7.5843e-04,
          1.5129e-04, 1.9725e-04, 2.6269e-04, 5.3463e-04, 5.0601e-04, 8.5051e-05,
          1.1112e-04, 4.5767e-05, 6.0221e-05, 8.0786e-05, 2.1428e-04, 1.1224e-03,
          2.1343e-05, 5.1404e-05, 8.0791e-05, 1.7151e-04, 3.4935e-04, 1.4872e-03,
          2.8966e-04, 6.3894e-05, 1.2076e-03, 1.0209e-03, 1.1142e-04, 2.0067e-04,
          7.2949e-04, 2.6050e-04, 1.9136e-04, 4.7327e-04, 2.5201e-04, 6.6716e-04,
          1.1606e-04, 4.7534e-05, 1.5787e-03, 1.4687e-04, 5.7580e-04, 8.2699e-04,
          2.1735e-04, 3.1698e-04, 3.3374e-05, 7.7645e-05, 1.3363e-04, 5.7541e-05,
          5.9388e-05, 6.3632e-05, 6.0729e-05, 1.4364e-04, 7.0106e-04, 5.7946e-04,
          4.3866e-04, 1.0312e-03, 1.3888e-04, 1.2962e-04, 1.0616e-04, 1.2091e-04,
          1.0111e-04, 4.7831e-05, 4.3312e-05, 7.1557e-04, 3.7451e-04, 1.0579e-04,
          7.7282e-05, 9.6151e-05, 2.5993e-04, 2.7649e-04, 7.9817e-05, 7.1630e-04,
          7.0151e-04, 9.2033e-05, 4.6931e-04, 5.7122e-05, 5.4736e-04, 1.1568e-04,
          8.9840e-05, 1.2864e-03, 1.4516e-04, 2.5504e-04, 7.3582e-04, 7.7087e-04,
          2.4613e-04, 3.6502e-04, 1.7329e-04, 5.4463e-05, 1.0020e-04, 1.3100e-04,
          5.4367e-05, 4.8410e-05, 1.8423e-04, 1.6824e-04, 5.3404e-04, 1.4664e-04,
          5.0552e-04, 2.5202e-05, 1.1077e-04, 5.5272e-04, 2.4866e-04, 1.3802e-04,
          6.5316e-04, 1.1431e-03, 5.3384e-04, 8.3449e-05, 2.6395e-04, 8.6306e-05,
          2.3732e-04, 8.7165e-04, 8.1712e-05, 1.6056e-04, 9.9283e-04, 1.1157e-04,
          2.5771e-05, 2.7851e-04, 1.8365e-04, 2.1113e-04, 6.8514e-04, 2.1241e-04,
          1.2943e-03, 7.2583e-04, 3.0665e-04, 1.8858e-04, 5.4513e-05, 1.2419e-03,
          2.8631e-04, 1.4244e-04, 5.0609e-04, 1.8600e-04, 3.3631e-04, 5.6815e-04,
          5.7457e-04, 4.0788e-05, 3.0074e-04, 1.1856e-04, 1.0121e-03, 7.6753e-04,
          2.9546e-04, 2.7216e-04, 3.1175e-04, 3.5395e-04, 1.6831e-04, 1.1098e-04,
          6.3255e-05, 8.6261e-05, 2.9174e-04, 2.5801e-04, 9.7859e-05, 3.5164e-04,
          4.5418e-05, 7.1593e-05, 4.3543e-05, 1.9684e-04, 2.6204e-04, 1.5244e-03,
          1.6563e-04, 2.4074e-04, 2.7597e-04, 5.3155e-04, 2.3994e-04, 2.1077e-03,
          5.8422e-03, 1.7942e-02, 7.9672e-03, 5.4461e-03, 2.8996e-03, 1.0466e-02,
          1.4685e-02, 4.0663e-03, 6.7067e-03, 1.8483e-03, 5.4508e-03, 4.3280e-03,
          1.0971e-02, 2.1039e-02, 1.5429e-02, 9.3761e-03, 4.8738e-03, 3.0077e-03,
          1.1688e-02, 1.4442e-02, 8.6455e-03, 5.9256e-03, 6.6411e-03, 1.0645e-02,
          1.4464e-02, 7.1747e-03, 5.9044e-03, 9.1314e-03, 5.7108e-02, 4.0878e-02,
          4.6968e-02, 7.9703e-02, 6.9106e-02, 1.3204e-01, 1.2416e-01, 0.0000e+00]),
  tensor([0.0000e+00, 5.2436e-04, 2.6352e-02, 4.2937e-04, 2.5947e-03, 3.2190e-03,
          3.0048e-02, 1.2111e-03, 1.4766e-04, 9.2391e-05, 7.0780e-04, 1.1488e-03,
          3.3327e-04, 1.6578e-04, 1.0011e-04, 9.5498e-04, 1.1266e-04, 1.6937e-04,
          9.6848e-05, 7.3029e-05, 3.6889e-05, 4.1142e-05, 4.0004e-05, 3.5763e-04,
          1.0413e-03, 1.0893e-03, 9.6056e-04, 8.9097e-04, 5.9098e-05, 1.5021e-04,
          1.1637e-04, 2.4957e-04, 5.5211e-05, 7.6723e-04, 5.4825e-04, 1.7339e-04,
          6.2745e-04, 1.3250e-04, 4.6939e-04, 6.0673e-04, 4.9285e-05, 1.9571e-04,
          7.1742e-05, 7.2066e-05, 6.7653e-05, 5.5400e-05, 1.3557e-04, 6.9107e-04,
          5.5435e-05, 6.5011e-05, 5.0100e-04, 2.0800e-04, 1.4539e-04, 1.5183e-04,
          5.5194e-05, 8.2534e-05, 8.5020e-05, 1.3380e-04, 3.7652e-04, 2.8246e-04,
          2.4320e-04, 1.1389e-04, 1.0235e-04, 2.2905e-04, 1.0150e-04, 1.6680e-03,
          2.7541e-04, 1.1932e-04, 2.5208e-04, 3.9376e-05, 3.0850e-05, 5.6457e-05,
          5.8976e-05, 1.1501e-03, 6.7522e-05, 1.8698e-04, 5.3279e-05, 4.9804e-05,
          1.1129e-04, 1.0770e-04, 2.1198e-04, 8.2340e-04, 3.2127e-04, 2.8123e-04,
          1.6126e-04, 2.2952e-04, 7.8618e-05, 4.3383e-04, 1.7721e-04, 1.4742e-04,
          5.3845e-04, 3.5133e-04, 1.3287e-04, 3.2744e-04, 8.8029e-05, 8.1268e-05,
          7.5178e-05, 5.2513e-05, 2.0576e-04, 3.1683e-04, 1.2005e-04, 1.8015e-04,
          6.3553e-04, 2.0522e-04, 1.4505e-04, 1.4651e-03, 2.1614e-03, 6.7768e-05,
          1.8740e-04, 6.1497e-05, 3.4422e-04, 9.4790e-05, 2.1387e-04, 1.4599e-04,
          1.2416e-03, 1.1841e-04, 2.9178e-05, 1.1318e-03, 1.0998e-04, 4.3866e-05,
          8.5580e-05, 4.8660e-04, 4.0175e-05, 7.3691e-05, 3.5501e-03, 7.1583e-05,
          7.4599e-05, 1.4295e-04, 1.4168e-04, 9.7055e-04, 2.0219e-04, 2.2670e-04,
          3.1493e-04, 1.3338e-04, 4.6503e-04, 8.5389e-05, 2.0727e-04, 7.8551e-05,
          3.6305e-04, 5.2394e-05, 2.1511e-04, 6.2757e-04, 6.8627e-04, 9.8357e-05,
          8.1883e-05, 3.1067e-04, 4.7043e-04, 1.7055e-04, 3.2805e-04, 1.3897e-03,
          1.2660e-03, 7.0784e-04, 1.8299e-04, 1.5893e-04, 1.8393e-04, 1.8771e-04,
          2.4499e-04, 2.0832e-04, 1.1887e-04, 1.0503e-04, 4.6113e-04, 9.0018e-05,
          7.0141e-04, 5.0453e-04, 1.5758e-04, 1.8875e-03, 1.8440e-04, 1.9558e-04,
          2.8183e-04, 1.9668e-04, 4.8445e-04, 5.9107e-04, 1.3047e-04, 4.8477e-04,
          3.0228e-04, 2.4388e-04, 3.0954e-04, 1.1141e-04, 5.8709e-04, 4.7899e-04,
          6.8091e-04, 2.8935e-04, 1.9095e-03, 1.3484e-04, 1.5823e-04, 7.1787e-04,
          3.6370e-04, 6.7742e-05, 1.5296e-03, 9.4012e-05, 1.9561e-04, 1.0660e-03,
          1.5007e-04, 4.5934e-05, 4.6440e-04, 1.4350e-04, 3.3691e-04, 4.1358e-04,
          2.5922e-04, 5.5829e-05, 4.8293e-04, 9.1869e-05, 2.5689e-04, 3.7562e-04,
          6.1148e-05, 1.8609e-04, 5.5361e-05, 1.4191e-04, 3.5855e-04, 5.0877e-05,
          7.8766e-05, 1.9926e-04, 1.0519e-04, 1.4569e-04, 6.1411e-04, 1.5958e-04,
          2.8029e-04, 1.4531e-04, 1.1538e-03, 8.1576e-04, 5.0650e-04, 1.1807e-04,
          3.3811e-04, 6.6869e-05, 1.7488e-04, 2.0039e-04, 3.4254e-03, 6.4460e-04,
          1.0575e-04, 8.8841e-05, 5.3106e-05, 2.0282e-04, 4.5374e-04, 5.7979e-05,
          1.6213e-04, 1.3943e-04, 1.8651e-04, 9.4700e-04, 4.5385e-04, 3.3043e-04,
          1.9965e-04, 1.8578e-04, 1.7810e-04, 2.3256e-04, 2.0130e-04, 8.2341e-05,
          9.9685e-05, 1.0753e-04, 1.0224e-04, 1.0246e-04, 1.5975e-04, 9.5909e-04,
          3.9611e-05, 4.0430e-05, 8.5407e-05, 1.4766e-04, 3.6494e-04, 6.9241e-04,
          1.9918e-04, 1.5097e-04, 4.4818e-04, 3.9409e-04, 7.0996e-05, 2.0183e-04,
          3.2028e-04, 8.1596e-04, 1.6059e-04, 2.2635e-04, 3.6235e-04, 4.9496e-04,
          2.1567e-04, 1.3540e-04, 1.5561e-03, 2.0220e-04, 3.7831e-04, 7.0836e-04,
          1.4578e-04, 1.6953e-04, 3.3307e-05, 9.5064e-05, 9.7360e-05, 1.3140e-04,
          2.2673e-04, 6.5144e-05, 6.8806e-05, 9.6554e-05, 2.4824e-04, 2.4526e-04,
          1.7117e-04, 1.0524e-03, 1.2259e-04, 1.1780e-04, 1.0273e-04, 1.9338e-04,
          1.2001e-04, 8.0560e-05, 6.3638e-05, 4.5826e-04, 3.1386e-04, 1.1631e-04,
          9.4972e-05, 1.7405e-04, 1.2654e-04, 1.7823e-04, 9.4348e-05, 6.1211e-04,
          3.6338e-04, 5.7089e-05, 3.2112e-04, 6.1977e-05, 9.3286e-04, 2.1806e-04,
          6.6857e-05, 2.7339e-04, 1.1035e-04, 3.0496e-04, 5.5832e-04, 4.9435e-04,
          1.9905e-04, 6.2645e-04, 1.1498e-04, 5.8767e-05, 1.1378e-04, 1.2686e-04,
          6.8430e-05, 4.3899e-05, 2.8904e-04, 1.2640e-04, 3.5455e-04, 1.4079e-04,
          6.1581e-04, 2.6207e-05, 1.0532e-04, 6.1951e-04, 6.5280e-04, 1.1208e-04,
          1.2432e-04, 3.8638e-04, 1.7314e-04, 7.6771e-05, 3.3074e-04, 9.8678e-05,
          5.8198e-04, 1.3961e-03, 7.0960e-05, 3.6571e-04, 2.9394e-04, 9.7259e-05,
          2.9516e-05, 3.3798e-04, 1.3325e-04, 1.6120e-04, 2.0558e-04, 9.8929e-05,
          2.9511e-04, 4.3954e-04, 7.5212e-04, 3.8307e-04, 6.0420e-05, 4.5405e-04,
          8.9520e-05, 7.3539e-05, 2.2536e-04, 3.1912e-04, 1.7959e-04, 9.1799e-04,
          9.3404e-04, 4.5449e-05, 3.4315e-04, 1.9394e-04, 1.2688e-03, 1.5264e-03,
          3.3048e-04, 2.2185e-04, 2.7295e-04, 3.8546e-04, 6.8410e-05, 1.0609e-04,
          4.0820e-05, 1.4515e-04, 4.6401e-04, 3.8272e-04, 9.8602e-05, 7.0144e-04,
          4.9607e-05, 8.3283e-05, 4.9242e-05, 2.9826e-04, 4.5015e-04, 1.0617e-03,
          1.5307e-04, 1.2508e-04, 6.2961e-04, 4.3990e-04, 3.7641e-04, 2.4485e-03,
          7.8410e-03, 2.0246e-02, 1.1122e-02, 6.9360e-03, 3.2262e-03, 1.0176e-02,
          1.6593e-02, 5.4259e-03, 8.5053e-03, 1.7944e-03, 5.9443e-03, 5.3022e-03,
          1.4788e-02, 2.3527e-02, 1.6080e-02, 1.2085e-02, 5.0882e-03, 3.0752e-03,
          1.2115e-02, 1.7382e-02, 1.0647e-02, 9.2995e-03, 8.9997e-03, 1.3744e-02,
          1.4746e-02, 6.7996e-03, 5.7728e-03, 8.5828e-03, 5.3005e-02, 3.6921e-02,
          3.7671e-02, 4.6928e-02, 2.9379e-02, 6.9080e-02, 1.0937e-01, 1.3744e-01,
          0.0000e+00]),
  tensor([0.0000e+00, 4.8505e-04, 2.6784e-02, 4.3034e-04, 2.4884e-03, 3.0039e-03,
          3.2281e-02, 1.3020e-03, 1.7672e-04, 8.2000e-05, 5.4561e-04, 8.4774e-04,
          2.3815e-04, 1.3410e-04, 6.0541e-05, 1.2128e-03, 1.2698e-04, 1.3157e-04,
          7.6094e-05, 7.1812e-05, 2.6074e-05, 3.5401e-05, 2.6295e-05, 2.8776e-04,
          7.5048e-04, 9.4395e-04, 1.0298e-03, 8.1965e-04, 4.7261e-05, 1.1269e-04,
          7.6173e-05, 2.3864e-04, 3.6908e-05, 9.8655e-04, 5.2717e-04, 3.8552e-04,
          6.9242e-04, 9.6201e-05, 4.4091e-04, 6.4708e-04, 3.4532e-05, 1.5420e-04,
          5.7273e-05, 5.2726e-05, 5.0899e-05, 4.1116e-05, 1.0883e-04, 8.8479e-04,
          3.4838e-05, 5.3006e-05, 4.1362e-04, 2.5618e-04, 1.4484e-04, 1.8951e-04,
          5.1233e-05, 8.3815e-05, 6.3301e-05, 9.9556e-05, 3.9999e-04, 1.7097e-04,
          1.6212e-04, 7.4268e-05, 9.5618e-05, 2.3881e-04, 6.0051e-05, 1.7671e-03,
          8.2441e-05, 5.7629e-05, 1.5593e-04, 3.0404e-05, 2.2496e-05, 4.1112e-05,
          5.1559e-05, 9.8168e-04, 6.1710e-05, 1.6116e-04, 4.8573e-05, 4.4738e-05,
          8.7378e-05, 7.6179e-05, 7.1403e-05, 7.4552e-04, 1.2720e-04, 2.6893e-04,
          1.3800e-04, 2.0874e-04, 6.8885e-05, 3.0887e-04, 5.5495e-05, 5.1307e-05,
          3.5823e-04, 2.6582e-04, 4.2494e-05, 2.3505e-04, 7.1661e-05, 7.6992e-05,
          5.0950e-05, 4.0197e-05, 2.8845e-04, 4.1168e-04, 5.5702e-05, 9.1913e-05,
          4.3913e-04, 6.4138e-05, 7.7795e-05, 1.5842e-03, 2.5558e-03, 4.9833e-05,
          1.8939e-04, 4.4282e-05, 2.6049e-04, 4.5854e-05, 5.4802e-05, 7.2038e-05,
          7.4233e-04, 8.9143e-05, 2.4404e-05, 1.4540e-03, 8.7745e-05, 4.3283e-05,
          6.3523e-05, 4.3037e-04, 2.2440e-05, 9.6425e-05, 4.4321e-03, 2.7091e-05,
          4.0137e-05, 4.8600e-05, 5.4802e-05, 1.1062e-03, 2.1041e-04, 2.0568e-04,
          1.3703e-04, 5.3970e-05, 4.6231e-04, 2.5112e-05, 1.4639e-04, 3.3903e-05,
          4.3796e-04, 3.2995e-05, 1.9014e-04, 7.3888e-04, 7.9527e-04, 7.2505e-05,
          3.2523e-05, 2.8806e-04, 1.5771e-04, 1.0151e-04, 9.1207e-05, 1.7254e-03,
          1.0954e-03, 7.3187e-04, 1.3610e-04, 1.6078e-04, 1.4707e-04, 1.4147e-04,
          1.8838e-04, 1.3113e-04, 8.2772e-05, 4.0496e-05, 4.5085e-04, 7.9866e-05,
          7.8851e-04, 4.8068e-04, 1.5880e-04, 1.9202e-03, 1.2281e-04, 1.0835e-04,
          2.0393e-04, 1.4117e-04, 4.1660e-04, 5.5538e-04, 1.1724e-04, 6.1402e-04,
          4.9877e-04, 2.8721e-04, 2.6490e-04, 9.4526e-05, 7.0645e-04, 4.9772e-04,
          4.9289e-04, 3.8350e-04, 1.9587e-03, 1.5511e-04, 1.6185e-04, 5.0164e-04,
          2.8279e-04, 5.7060e-05, 1.0133e-03, 4.8283e-05, 6.4863e-05, 1.1572e-03,
          1.6122e-04, 2.6784e-05, 5.1463e-04, 1.3109e-04, 2.7606e-04, 4.6118e-04,
          2.1490e-04, 4.0327e-05, 4.9526e-04, 5.4189e-05, 1.6856e-04, 3.0243e-04,
          3.1812e-05, 1.1258e-04, 4.1072e-05, 2.0294e-04, 2.5455e-04, 5.2338e-05,
          7.0183e-05, 1.4370e-04, 5.8357e-05, 7.8254e-05, 6.3259e-04, 5.5052e-05,
          2.9012e-04, 1.5098e-04, 7.3281e-04, 7.7646e-04, 6.2031e-04, 1.0477e-04,
          3.3312e-04, 5.0046e-05, 1.4842e-04, 1.0398e-04, 4.0732e-03, 6.4824e-04,
          4.5474e-05, 4.8256e-05, 3.5630e-05, 2.6582e-04, 5.0038e-04, 5.2837e-05,
          1.2293e-04, 6.6168e-05, 7.5244e-05, 1.3451e-03, 3.6022e-04, 3.2680e-04,
          9.9734e-05, 1.4073e-04, 1.3788e-04, 2.3465e-04, 2.3490e-04, 5.5844e-05,
          8.2839e-05, 5.6306e-05, 5.8692e-05, 7.2334e-05, 1.6452e-04, 9.1668e-04,
          2.0352e-05, 3.3095e-05, 6.7984e-05, 1.1143e-04, 3.3000e-04, 9.0461e-04,
          1.5498e-04, 7.6714e-05, 4.5559e-04, 6.4200e-04, 5.9164e-05, 1.1782e-04,
          4.4544e-04, 4.9192e-04, 1.5108e-04, 2.4711e-04, 2.6318e-04, 5.4471e-04,
          9.9773e-05, 6.6543e-05, 1.8733e-03, 1.4371e-04, 3.5500e-04, 4.8601e-04,
          1.4157e-04, 1.9046e-04, 2.8102e-05, 8.0881e-05, 7.5198e-05, 4.5710e-05,
          6.6713e-05, 4.4125e-05, 3.6560e-05, 8.1940e-05, 3.9584e-04, 3.5638e-04,
          1.7454e-04, 1.2374e-03, 1.3988e-04, 9.8584e-05, 1.0145e-04, 1.3727e-04,
          9.3210e-05, 5.3057e-05, 3.7966e-05, 5.7939e-04, 2.7496e-04, 7.8808e-05,
          6.0176e-05, 1.1244e-04, 1.3512e-04, 1.7186e-04, 5.4004e-05, 6.3598e-04,
          4.7826e-04, 4.9247e-05, 2.9317e-04, 4.6223e-05, 9.6471e-04, 1.6799e-04,
          6.3786e-05, 4.0106e-04, 8.7806e-05, 2.7145e-04, 5.8309e-04, 5.9474e-04,
          2.1219e-04, 4.8060e-04, 1.2080e-04, 3.9531e-05, 8.0339e-05, 9.9010e-05,
          5.0388e-05, 4.3831e-05, 1.9262e-04, 9.3650e-05, 2.9678e-04, 1.2189e-04,
          4.8331e-04, 2.0492e-05, 8.9732e-05, 5.4753e-04, 4.5566e-04, 1.0012e-04,
          2.1468e-04, 5.3519e-04, 2.1262e-04, 5.8924e-05, 2.4140e-04, 8.4665e-05,
          3.9427e-04, 1.2749e-03, 6.5045e-05, 2.3372e-04, 4.2609e-04, 9.2666e-05,
          2.3719e-05, 2.7570e-04, 1.1408e-04, 1.2468e-04, 2.3527e-04, 9.4942e-05,
          5.1996e-04, 8.6503e-04, 6.6139e-04, 2.8015e-04, 5.4259e-05, 6.0769e-04,
          9.5059e-05, 6.4900e-05, 2.5121e-04, 2.2499e-04, 3.4708e-04, 7.8344e-04,
          9.0112e-04, 3.6968e-05, 4.7051e-04, 1.6792e-04, 1.1944e-03, 1.5068e-03,
          2.8578e-04, 1.9784e-04, 2.2997e-04, 5.4295e-04, 8.0406e-05, 8.9600e-05,
          3.9657e-05, 1.1195e-04, 3.8538e-04, 3.1133e-04, 7.9252e-05, 4.1491e-04,
          3.0835e-05, 7.2330e-05, 4.2626e-05, 2.6375e-04, 3.0198e-04, 9.7161e-04,
          1.5122e-04, 1.5778e-04, 3.9502e-04, 3.8056e-04, 2.5798e-04, 2.3575e-03,
          8.4120e-03, 2.0249e-02, 8.5394e-03, 5.9434e-03, 3.2774e-03, 8.5796e-03,
          1.5796e-02, 5.9333e-03, 9.0227e-03, 1.6045e-03, 6.0876e-03, 4.2318e-03,
          1.3750e-02, 2.3758e-02, 1.6979e-02, 1.0035e-02, 5.1917e-03, 3.1003e-03,
          9.6635e-03, 1.6088e-02, 9.3492e-03, 6.3528e-03, 5.9933e-03, 9.9987e-03,
          1.4572e-02, 6.2159e-03, 4.6109e-03, 7.2350e-03, 5.0056e-02, 3.0027e-02,
          2.8218e-02, 4.1248e-02, 2.2344e-02, 5.1578e-02, 8.9183e-02, 1.3908e-01,
          9.8159e-02, 0.0000e+00]),
  tensor([0.0000e+00, 2.7907e-04, 2.2806e-02, 2.9069e-04, 1.9984e-03, 2.4762e-03,
          2.7317e-02, 9.8982e-04, 3.0037e-04, 1.1199e-04, 8.3013e-04, 1.2329e-03,
          3.7242e-04, 2.5298e-04, 1.0229e-04, 1.3734e-03, 2.1594e-04, 2.7322e-04,
          1.4530e-04, 9.3236e-05, 2.6223e-05, 2.7287e-05, 2.7152e-05, 3.9316e-04,
          1.1512e-03, 9.7308e-04, 7.9934e-04, 6.8869e-04, 3.9033e-05, 2.3582e-04,
          1.2488e-04, 4.9967e-04, 5.3662e-05, 1.2659e-03, 7.0119e-04, 3.1519e-04,
          1.1500e-03, 1.2002e-04, 6.0977e-04, 8.8702e-04, 3.6934e-05, 2.7831e-04,
          6.9191e-05, 6.9089e-05, 6.0355e-05, 6.2970e-05, 2.3626e-04, 1.0241e-03,
          4.2737e-05, 4.3168e-05, 3.2076e-04, 3.2312e-04, 2.7942e-04, 2.1165e-04,
          5.7097e-05, 1.3421e-04, 5.5992e-05, 1.7815e-04, 5.0491e-04, 3.5112e-04,
          2.3525e-04, 9.2712e-05, 7.6794e-05, 2.4808e-04, 1.0562e-04, 2.9423e-03,
          2.2996e-04, 1.1530e-04, 2.9700e-04, 3.7329e-05, 2.5810e-05, 4.6450e-05,
          4.5241e-05, 1.1540e-03, 6.0729e-05, 3.1936e-04, 5.6816e-05, 5.6530e-05,
          1.3414e-04, 1.4606e-04, 1.9106e-04, 1.4879e-03, 3.3621e-04, 3.5140e-04,
          1.1552e-04, 3.7928e-04, 6.3988e-05, 5.0201e-04, 1.5496e-04, 1.1968e-04,
          7.3171e-04, 5.0052e-04, 8.1068e-05, 4.0313e-04, 7.9738e-05, 5.9479e-05,
          4.7472e-05, 3.4026e-05, 3.9204e-04, 5.2850e-04, 1.1599e-04, 1.7785e-04,
          8.5910e-04, 1.6454e-04, 1.6460e-04, 3.0276e-03, 4.1131e-03, 6.3752e-05,
          2.9716e-04, 5.1683e-05, 4.4915e-04, 6.4747e-05, 1.7066e-04, 1.4025e-04,
          1.1786e-03, 1.4922e-04, 2.3303e-05, 2.0448e-03, 7.8104e-05, 3.6351e-05,
          8.7332e-05, 7.6357e-04, 3.4288e-05, 1.9280e-04, 6.2850e-03, 5.6964e-05,
          7.9307e-05, 1.1191e-04, 1.3016e-04, 2.2534e-03, 3.0651e-04, 3.3830e-04,
          2.7021e-04, 1.4190e-04, 5.8114e-04, 5.1593e-05, 2.4441e-04, 5.2788e-05,
          6.6710e-04, 4.7700e-05, 2.3840e-04, 6.1326e-04, 6.8341e-04, 1.3075e-04,
          6.1982e-05, 3.2331e-04, 3.4623e-04, 1.6528e-04, 2.6557e-04, 2.9975e-03,
          1.9786e-03, 1.5660e-03, 2.5248e-04, 2.0834e-04, 1.5042e-04, 2.2952e-04,
          3.3279e-04, 2.2883e-04, 1.6490e-04, 7.5215e-05, 7.6724e-04, 1.2091e-04,
          7.8840e-04, 6.4185e-04, 3.0356e-04, 1.7119e-03, 1.6932e-04, 1.7202e-04,
          3.2473e-04, 1.9534e-04, 8.2288e-04, 8.6082e-04, 1.7307e-04, 7.8681e-04,
          5.4518e-04, 3.5408e-04, 2.5098e-04, 9.1903e-05, 9.5032e-04, 8.2803e-04,
          6.3803e-04, 6.6983e-04, 2.4915e-03, 3.4005e-04, 2.6410e-04, 6.4604e-04,
          4.8192e-04, 7.1796e-05, 9.5114e-04, 8.2604e-05, 1.4930e-04, 1.7157e-03,
          3.8365e-04, 4.4922e-05, 6.0588e-04, 2.7158e-04, 4.5920e-04, 5.7396e-04,
          2.2895e-04, 3.8191e-05, 8.9924e-04, 9.8207e-05, 2.7084e-04, 4.2696e-04,
          4.8401e-05, 1.9414e-04, 5.3199e-05, 2.3528e-04, 4.1446e-04, 7.4673e-05,
          1.2296e-04, 1.7206e-04, 8.4366e-05, 1.6485e-04, 1.1398e-03, 1.3966e-04,
          4.2263e-04, 2.3109e-04, 1.0137e-03, 1.0146e-03, 7.8499e-04, 1.8310e-04,
          4.7097e-04, 5.4612e-05, 2.0984e-04, 1.8174e-04, 5.7142e-03, 8.6571e-04,
          1.1390e-04, 6.2016e-05, 4.2507e-05, 2.5686e-04, 5.4867e-04, 6.3512e-05,
          1.3311e-04, 1.1993e-04, 1.5640e-04, 1.4259e-03, 6.1199e-04, 4.0979e-04,
          1.5943e-04, 1.3796e-04, 1.4124e-04, 2.8209e-04, 4.3140e-04, 7.2922e-05,
          8.4685e-05, 1.0930e-04, 6.3490e-05, 8.0485e-05, 1.2850e-04, 1.3685e-03,
          2.8988e-05, 3.4891e-05, 6.4648e-05, 1.3206e-04, 5.6551e-04, 8.0371e-04,
          2.0488e-04, 1.3349e-04, 6.7454e-04, 6.1528e-04, 7.9577e-05, 1.8738e-04,
          4.4954e-04, 6.1636e-04, 2.2508e-04, 4.7833e-04, 5.0543e-04, 7.1058e-04,
          1.5805e-04, 1.3052e-04, 2.6754e-03, 1.7464e-04, 5.0473e-04, 9.1872e-04,
          1.5949e-04, 1.8422e-04, 2.3399e-05, 1.1050e-04, 7.4115e-05, 1.1767e-04,
          2.3001e-04, 5.5480e-05, 5.8283e-05, 8.9182e-05, 5.4033e-04, 3.4487e-04,
          2.1596e-04, 1.0949e-03, 2.3077e-04, 1.9798e-04, 7.8018e-05, 1.3164e-04,
          1.1113e-04, 6.1769e-05, 4.7484e-05, 7.6417e-04, 4.4282e-04, 9.8928e-05,
          7.4328e-05, 1.8089e-04, 2.5068e-04, 2.4709e-04, 8.0517e-05, 7.0390e-04,
          5.6736e-04, 5.5989e-05, 4.3643e-04, 6.6912e-05, 9.7964e-04, 2.7754e-04,
          8.6682e-05, 4.0213e-04, 8.3095e-05, 3.3919e-04, 5.8835e-04, 8.3871e-04,
          1.5898e-04, 7.5736e-04, 1.5574e-04, 8.4654e-05, 8.3575e-05, 9.2814e-05,
          8.2403e-05, 6.1210e-05, 2.6672e-04, 1.1996e-04, 4.3040e-04, 2.1587e-04,
          6.0570e-04, 3.6427e-05, 1.2345e-04, 7.1238e-04, 6.3861e-04, 1.1498e-04,
          1.6583e-04, 5.2879e-04, 2.7301e-04, 6.0235e-05, 4.2787e-04, 1.0720e-04,
          4.8616e-04, 1.9399e-03, 1.1061e-04, 3.4319e-04, 4.4838e-04, 9.4708e-05,
          2.1893e-05, 2.4042e-04, 1.2305e-04, 1.5393e-04, 4.2992e-04, 1.9414e-04,
          4.8322e-04, 1.0107e-03, 6.4440e-04, 2.8501e-04, 5.8922e-05, 6.1821e-04,
          1.5992e-04, 1.2786e-04, 3.3531e-04, 3.0722e-04, 3.3795e-04, 8.4948e-04,
          1.2823e-03, 3.7390e-05, 4.9078e-04, 2.0153e-04, 8.9134e-04, 1.1067e-03,
          2.6610e-04, 2.0890e-04, 2.5280e-04, 7.8103e-04, 9.9810e-05, 7.7854e-05,
          4.7197e-05, 1.6404e-04, 5.5534e-04, 5.3987e-04, 7.8982e-05, 4.4930e-04,
          4.5062e-05, 5.1489e-05, 5.6223e-05, 4.3345e-04, 4.0586e-04, 1.2835e-03,
          1.4012e-04, 1.6599e-04, 4.2722e-04, 4.7842e-04, 2.9062e-04, 1.7518e-03,
          5.4216e-03, 1.7213e-02, 1.2124e-02, 5.3908e-03, 2.6104e-03, 1.0760e-02,
          1.0319e-02, 3.2901e-03, 4.9650e-03, 1.2194e-03, 3.8553e-03, 3.6733e-03,
          9.7541e-03, 1.5282e-02, 1.1365e-02, 1.2987e-02, 5.5191e-03, 2.6936e-03,
          1.1150e-02, 1.3728e-02, 9.4349e-03, 1.0078e-02, 8.0820e-03, 1.4428e-02,
          1.6162e-02, 6.7896e-03, 5.6440e-03, 9.0928e-03, 3.6310e-02, 2.2929e-02,
          1.9731e-02, 4.6501e-02, 3.2900e-02, 5.6678e-02, 6.4055e-02, 9.0791e-02,
          6.9034e-02, 9.5461e-02, 0.0000e+00]),
  tensor([0.0000e+00, 3.2101e-04, 2.5191e-02, 4.0695e-04, 2.2520e-03, 2.7140e-03,
          2.5353e-02, 1.1655e-03, 2.1107e-04, 1.2612e-04, 7.2448e-04, 1.1326e-03,
          3.9890e-04, 3.3623e-04, 7.8725e-05, 1.6334e-03, 1.4327e-04, 3.2227e-04,
          1.8398e-04, 1.0181e-04, 3.0753e-05, 2.9862e-05, 3.5360e-05, 4.7076e-04,
          1.2210e-03, 8.6667e-04, 8.1141e-04, 6.6305e-04, 5.6244e-05, 3.7205e-04,
          1.1224e-04, 4.0234e-04, 9.7900e-05, 1.1678e-03, 7.9617e-04, 1.8253e-04,
          5.5523e-04, 6.5135e-05, 4.9416e-04, 8.0953e-04, 4.2948e-05, 1.8536e-04,
          4.8788e-05, 5.0828e-05, 5.2772e-05, 4.9372e-05, 3.7204e-04, 1.1476e-03,
          5.8487e-05, 5.7497e-05, 3.4155e-04, 2.1691e-04, 2.4932e-04, 2.4523e-04,
          4.9644e-05, 1.5012e-04, 7.3486e-05, 8.4081e-05, 1.9597e-04, 1.6042e-04,
          1.4131e-04, 1.0102e-04, 8.3826e-05, 1.5966e-04, 8.5097e-05, 2.4713e-03,
          9.0678e-05, 6.7745e-05, 2.0208e-04, 3.0988e-05, 2.2540e-05, 4.2291e-05,
          4.2577e-05, 1.1186e-03, 5.0818e-05, 3.5501e-04, 6.2539e-05, 5.9198e-05,
          1.5013e-04, 6.6239e-05, 6.7949e-05, 6.3084e-04, 1.1657e-04, 1.7843e-04,
          8.2992e-05, 3.7722e-04, 6.6510e-05, 3.5135e-04, 5.2976e-05, 4.6135e-05,
          6.2967e-04, 3.9304e-04, 4.2316e-05, 4.0534e-04, 5.8726e-05, 5.3150e-05,
          7.5370e-05, 3.2337e-05, 2.5143e-04, 4.2865e-04, 5.7396e-05, 9.9522e-05,
          7.3594e-04, 6.3287e-05, 8.8848e-05, 8.5454e-04, 3.1023e-03, 6.1889e-05,
          2.9451e-04, 3.7967e-05, 3.5627e-04, 4.1275e-05, 5.8583e-05, 5.3884e-05,
          6.2460e-04, 9.2550e-05, 2.7970e-05, 2.4365e-03, 7.2779e-05, 5.1742e-05,
          8.6947e-05, 4.4531e-04, 2.4484e-05, 1.2677e-04, 4.8503e-03, 2.4914e-05,
          4.1044e-05, 5.1906e-05, 7.4160e-05, 1.9089e-03, 2.8128e-04, 4.0267e-04,
          1.4517e-04, 6.5788e-05, 6.8094e-04, 2.4536e-05, 1.7301e-04, 3.5141e-05,
          1.2848e-03, 4.7422e-05, 2.7813e-04, 5.9538e-04, 8.5978e-04, 1.1601e-04,
          3.6113e-05, 3.2943e-04, 1.3022e-04, 8.4267e-05, 7.8659e-05, 1.5251e-03,
          8.3733e-04, 1.1597e-03, 1.6864e-04, 1.5206e-04, 1.8250e-04, 1.3185e-04,
          1.9686e-04, 1.1545e-04, 1.5723e-04, 3.6287e-05, 5.4979e-04, 1.3723e-04,
          9.9557e-04, 7.5276e-04, 5.0374e-04, 1.7270e-03, 1.0492e-04, 7.0403e-05,
          2.7921e-04, 1.0115e-04, 6.1681e-04, 9.8258e-04, 2.3360e-04, 7.4180e-04,
          3.3007e-04, 1.4792e-04, 1.9468e-04, 6.0384e-05, 5.4787e-04, 7.8582e-04,
          5.6822e-04, 6.1961e-04, 2.5700e-03, 2.0947e-04, 2.3287e-04, 6.2390e-04,
          4.6754e-04, 5.8898e-05, 1.0877e-03, 5.0775e-05, 5.4563e-05, 1.7087e-03,
          2.7067e-04, 2.8589e-05, 6.9244e-04, 4.2560e-04, 3.3924e-04, 6.4739e-04,
          1.7487e-04, 4.6614e-05, 7.3134e-04, 5.8608e-05, 1.2758e-04, 4.1830e-04,
          3.3903e-05, 1.1170e-04, 4.3294e-05, 2.2008e-04, 4.3094e-04, 1.3335e-04,
          1.0137e-04, 1.0838e-04, 5.2561e-05, 9.5361e-05, 9.3741e-04, 5.4654e-05,
          3.5322e-04, 2.9387e-04, 8.6062e-04, 1.1070e-03, 9.6199e-04, 1.7231e-04,
          3.8680e-04, 3.0981e-05, 1.2949e-04, 6.9441e-05, 4.5544e-03, 1.0025e-03,
          5.6307e-05, 3.5403e-05, 3.4995e-05, 2.0206e-04, 3.5965e-04, 8.0379e-05,
          1.0379e-04, 5.7482e-05, 4.8987e-05, 6.2242e-04, 3.0048e-04, 3.0635e-04,
          8.1912e-05, 1.1066e-04, 7.8606e-05, 3.0855e-04, 4.0916e-04, 6.5114e-05,
          4.8496e-05, 8.1993e-05, 3.8786e-05, 6.3174e-05, 9.0642e-05, 1.0944e-03,
          2.1866e-05, 3.0964e-05, 6.5671e-05, 1.3555e-04, 7.9451e-04, 8.1904e-04,
          1.1182e-04, 6.2221e-05, 2.7875e-04, 3.2462e-04, 5.4621e-05, 9.6943e-05,
          3.8639e-04, 5.1196e-04, 2.2193e-04, 3.9245e-04, 3.2268e-04, 4.7957e-04,
          8.2959e-05, 7.0098e-05, 1.7563e-03, 1.0763e-04, 5.6292e-04, 6.7306e-04,
          1.3253e-04, 1.9286e-04, 2.4440e-05, 1.1768e-04, 6.1315e-05, 5.1438e-05,
          8.9076e-05, 3.6675e-05, 3.8677e-05, 7.5472e-05, 4.6464e-04, 3.2879e-04,
          1.7992e-04, 1.1734e-03, 4.0890e-04, 2.7779e-04, 5.9140e-05, 1.1043e-04,
          6.1282e-05, 3.8623e-05, 3.4299e-05, 9.6236e-04, 2.7950e-04, 7.5277e-05,
          6.1137e-05, 1.7755e-04, 1.8742e-04, 3.2782e-04, 4.5473e-05, 5.9831e-04,
          6.2219e-04, 5.1061e-05, 4.4703e-04, 9.0593e-05, 1.1788e-03, 2.8441e-04,
          1.2247e-04, 2.4846e-04, 7.7078e-05, 2.8087e-04, 9.1030e-04, 8.3098e-04,
          1.6036e-04, 6.7272e-04, 2.2936e-04, 9.9505e-05, 7.0163e-05, 8.7413e-05,
          1.0534e-04, 6.6348e-05, 2.4978e-04, 6.5335e-05, 2.5237e-04, 2.0376e-04,
          5.2854e-04, 3.9728e-05, 1.2765e-04, 5.8771e-04, 6.0906e-04, 9.9798e-05,
          1.1159e-04, 3.9865e-04, 2.0123e-04, 6.1997e-05, 2.4542e-04, 1.0973e-04,
          5.1415e-04, 1.8826e-03, 1.9325e-04, 2.9028e-04, 3.1617e-04, 1.5192e-04,
          2.1141e-05, 2.1055e-04, 7.5617e-05, 9.4240e-05, 2.9770e-04, 1.5031e-04,
          4.4909e-04, 7.5093e-04, 3.9186e-04, 2.8500e-04, 6.4006e-05, 3.8898e-04,
          1.1811e-04, 1.1279e-04, 2.1168e-04, 2.6416e-04, 2.0850e-04, 8.0787e-04,
          1.1309e-03, 4.5285e-05, 4.1155e-04, 1.3618e-04, 1.0155e-03, 1.0437e-03,
          2.8389e-04, 1.4391e-04, 1.8904e-04, 6.1597e-04, 1.0010e-04, 7.4645e-05,
          4.9793e-05, 1.2474e-04, 4.3593e-04, 6.3777e-04, 1.1342e-04, 3.4495e-04,
          3.9674e-05, 5.4075e-05, 6.5631e-05, 3.1652e-04, 3.5985e-04, 7.8126e-04,
          2.2154e-04, 1.7253e-04, 4.0867e-04, 4.0635e-04, 2.2322e-04, 2.7436e-03,
          4.6982e-03, 1.5537e-02, 1.1092e-02, 4.5344e-03, 2.5910e-03, 8.7382e-03,
          1.0445e-02, 3.4852e-03, 4.9227e-03, 2.7688e-03, 3.6983e-03, 3.0182e-03,
          1.2712e-02, 1.5770e-02, 1.0114e-02, 1.2635e-02, 4.4479e-03, 2.7562e-03,
          1.0195e-02, 8.5324e-03, 5.6431e-03, 9.8508e-03, 8.5451e-03, 1.4384e-02,
          1.1258e-02, 4.8391e-03, 5.6991e-03, 9.8843e-03, 3.9780e-02, 2.0549e-02,
          1.2243e-02, 4.0614e-02, 3.4624e-02, 6.6238e-02, 5.1093e-02, 5.5604e-02,
          3.9809e-02, 7.3357e-02, 1.4501e-01, 0.0000e+00]),
  tensor([0.0000e+00, 9.6642e-04, 1.9551e-02, 1.2728e-03, 4.6891e-03, 4.2057e-03,
          1.7593e-02, 3.2999e-03, 4.0839e-04, 6.2827e-05, 2.5955e-04, 5.6297e-04,
          1.3495e-04, 1.3762e-04, 8.6044e-05, 2.5906e-04, 1.9726e-04, 2.0049e-04,
          7.7407e-05, 7.0586e-05, 4.8392e-05, 5.8991e-05, 4.1046e-05, 2.2208e-04,
          5.0511e-04, 3.1377e-04, 2.3222e-03, 9.0143e-04, 9.9716e-05, 1.2606e-04,
          1.2481e-04, 6.3087e-04, 4.7978e-05, 2.7232e-04, 1.4855e-04, 5.2557e-04,
          2.2753e-04, 1.0473e-04, 3.1216e-04, 2.0582e-04, 5.6420e-05, 1.4494e-04,
          6.5229e-05, 4.8589e-05, 5.6261e-05, 3.7981e-05, 1.1333e-04, 2.9332e-04,
          4.7091e-05, 8.2939e-05, 6.0141e-04, 3.0636e-04, 4.2318e-04, 2.7779e-04,
          7.8897e-05, 1.8291e-04, 9.7421e-05, 1.0402e-04, 2.8503e-04, 1.3940e-04,
          1.5010e-04, 7.8869e-05, 9.0174e-05, 1.3217e-04, 4.5894e-05, 3.5679e-04,
          1.0683e-04, 3.8150e-05, 7.9229e-05, 3.0120e-05, 2.5242e-05, 4.9116e-05,
          6.4062e-05, 4.8390e-04, 1.1823e-04, 5.4468e-04, 9.3845e-05, 7.7930e-05,
          1.4250e-04, 9.6392e-05, 7.1413e-05, 1.7397e-04, 9.2501e-05, 1.6148e-04,
          1.1745e-04, 4.7645e-04, 6.5585e-05, 1.4843e-04, 4.8791e-05, 5.1858e-05,
          2.3029e-04, 1.7533e-04, 2.4066e-05, 1.5393e-04, 4.8786e-05, 7.2647e-05,
          8.9217e-05, 7.4143e-05, 8.1435e-04, 3.7377e-04, 6.4957e-05, 8.8043e-05,
          2.7857e-04, 1.2024e-04, 5.5643e-05, 3.5403e-04, 4.7557e-04, 5.7228e-05,
          1.6820e-04, 4.7912e-05, 2.1330e-04, 4.3079e-05, 1.0509e-04, 6.5919e-05,
          2.2734e-04, 5.3082e-05, 3.7658e-05, 3.5120e-04, 8.7275e-05, 7.5514e-05,
          8.3436e-05, 1.6670e-04, 2.9999e-05, 1.3045e-04, 4.6959e-04, 2.4396e-05,
          4.7300e-05, 5.5506e-05, 7.4629e-05, 2.6310e-04, 1.5521e-04, 1.8881e-04,
          2.8130e-04, 6.1790e-05, 9.9557e-05, 1.7692e-05, 8.6149e-05, 2.7706e-05,
          2.4239e-04, 3.6053e-05, 1.8550e-04, 5.9787e-04, 4.4804e-04, 6.6599e-05,
          3.5145e-05, 9.0719e-05, 5.7821e-05, 7.3538e-05, 8.2061e-05, 2.9764e-04,
          3.5729e-04, 1.7765e-04, 8.9923e-05, 1.5357e-04, 1.3210e-04, 1.0388e-04,
          1.2285e-04, 1.1785e-04, 4.4532e-05, 3.6612e-05, 1.6248e-04, 8.0999e-05,
          3.5105e-04, 3.3051e-04, 1.4225e-04, 4.6155e-04, 1.6805e-04, 1.1041e-04,
          1.8835e-04, 1.1685e-04, 1.0480e-04, 1.1516e-04, 8.9841e-05, 2.8161e-04,
          2.7435e-04, 1.9418e-04, 2.3463e-04, 9.3169e-05, 1.7253e-04, 1.9937e-04,
          2.4718e-04, 7.2822e-05, 2.5942e-04, 3.6662e-04, 1.0679e-04, 1.5850e-04,
          1.3920e-04, 9.6411e-05, 4.0693e-04, 5.8109e-05, 5.7175e-05, 1.5622e-04,
          3.3435e-04, 2.4860e-05, 1.4183e-04, 1.2690e-04, 1.5086e-04, 1.6718e-04,
          1.9393e-04, 5.3731e-05, 1.3642e-04, 4.1118e-05, 1.2869e-04, 1.7357e-04,
          2.9542e-05, 6.7479e-05, 3.2530e-05, 1.0599e-04, 2.1870e-04, 3.9106e-05,
          1.9826e-04, 1.5878e-04, 9.4459e-05, 6.4960e-05, 1.7017e-04, 5.1794e-05,
          1.8765e-04, 1.0701e-04, 2.4715e-04, 1.2252e-04, 2.1518e-04, 7.6116e-05,
          3.0314e-04, 3.7837e-05, 6.3560e-05, 9.2746e-05, 4.2831e-04, 1.5741e-04,
          4.5548e-05, 5.0424e-05, 3.2529e-05, 4.8069e-04, 1.5543e-04, 5.5861e-05,
          1.8655e-04, 6.9191e-05, 1.2343e-04, 2.6603e-04, 1.1105e-04, 6.7837e-04,
          1.5420e-04, 2.3784e-04, 1.3670e-04, 1.7984e-04, 1.9779e-04, 6.5438e-05,
          5.7028e-05, 4.0631e-05, 4.1625e-05, 6.4575e-05, 1.3801e-04, 2.1719e-04,
          1.7014e-05, 2.3666e-05, 5.6463e-05, 8.4276e-05, 1.4548e-04, 1.6726e-03,
          9.4690e-05, 8.4326e-05, 2.0885e-04, 1.7540e-04, 4.3430e-05, 1.6032e-04,
          1.3682e-04, 1.4164e-04, 1.1728e-04, 2.7709e-04, 2.2280e-04, 1.1319e-04,
          1.0095e-04, 4.7810e-05, 1.8703e-04, 5.7101e-05, 2.2493e-04, 1.3903e-04,
          9.4094e-05, 1.6905e-04, 4.0063e-05, 7.3195e-05, 9.9862e-05, 6.8352e-05,
          1.1721e-04, 6.3742e-05, 4.3403e-05, 6.8848e-05, 1.6314e-04, 1.7617e-04,
          4.0650e-04, 4.3590e-04, 7.7122e-05, 9.9826e-05, 6.2472e-05, 7.9772e-05,
          5.7760e-05, 3.0326e-05, 3.0989e-05, 1.6730e-04, 1.4327e-04, 5.8322e-05,
          4.3479e-05, 5.4034e-05, 4.2414e-04, 1.4636e-04, 7.3848e-05, 2.1705e-04,
          1.6509e-04, 6.3587e-05, 1.8549e-04, 4.5878e-05, 1.7137e-04, 6.6326e-05,
          5.2509e-05, 2.9039e-04, 1.0604e-04, 1.1339e-04, 3.8060e-04, 1.3556e-04,
          1.4768e-04, 2.6316e-04, 7.7549e-05, 3.0634e-05, 6.1587e-05, 8.9103e-05,
          5.9965e-05, 3.1192e-05, 2.2446e-04, 1.1933e-04, 1.5628e-04, 1.1577e-04,
          4.3063e-04, 2.1713e-05, 1.0717e-04, 4.5538e-04, 2.6471e-04, 1.9605e-04,
          1.7962e-04, 1.2953e-03, 5.2067e-04, 7.7913e-05, 1.9124e-04, 7.7284e-05,
          2.3355e-04, 1.9980e-04, 7.3586e-05, 1.0943e-04, 8.5414e-04, 9.8183e-05,
          3.0349e-05, 4.9737e-04, 1.2485e-04, 1.4285e-04, 7.5839e-04, 2.7952e-04,
          9.3132e-04, 1.6845e-04, 9.2098e-05, 1.5544e-04, 5.4011e-05, 2.3522e-04,
          3.6745e-04, 2.4570e-04, 2.3920e-04, 2.3261e-04, 1.0245e-04, 3.8077e-04,
          1.3351e-04, 4.1528e-05, 1.9489e-04, 1.2751e-04, 1.1942e-03, 4.4721e-04,
          5.9431e-04, 1.6009e-04, 1.4681e-04, 1.1361e-04, 1.9481e-04, 1.2656e-04,
          4.7657e-05, 8.3481e-05, 1.7527e-04, 1.7430e-04, 9.4941e-05, 1.6367e-04,
          4.2367e-05, 8.1887e-05, 3.9663e-05, 9.0630e-05, 9.6395e-05, 3.0233e-04,
          1.7113e-04, 1.1476e-04, 2.1482e-04, 4.2981e-04, 3.3727e-04, 9.9520e-03,
          1.2407e-02, 1.6981e-02, 7.0578e-03, 5.3986e-03, 3.2480e-03, 1.2336e-02,
          3.4263e-02, 1.1808e-02, 2.1991e-02, 1.8441e-02, 1.8058e-02, 4.8741e-03,
          3.1279e-02, 5.1422e-02, 2.4050e-02, 1.0665e-02, 6.6106e-03, 3.5835e-03,
          1.6808e-02, 1.7281e-02, 1.0317e-02, 8.7570e-03, 8.1516e-03, 1.2133e-02,
          1.2115e-02, 6.0916e-03, 6.1798e-03, 9.1614e-03, 7.8321e-02, 3.2691e-02,
          1.8770e-02, 2.2554e-02, 1.4028e-02, 2.3082e-02, 3.3641e-02, 2.8087e-02,
          1.6274e-02, 1.5575e-02, 4.8695e-02, 1.3433e-01, 0.0000e+00])])
torch.Size([410])

Now I’ll turn this into a PyTorch tensor.

torch.Size([437, 437])

Now it’s time to visualize! A higher gamma factor highlights lower attention values more.

Object `np.power` not found.

The sharp spike of attention at the beginning of the graph is most likely the system prompt.

What I now want to do is visualize how the total attention on the image itself varied as each token was generated.

410
['<|im_start|>',
 'user',
 'Ċ',
 '<|vision_start|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|image_pad|>',
 '<|vision_end|>',
 'Describe',
 'Ġthe',
 'Ġcat',
 'Ġon',
 'Ġthe',
 'Ġleft',
 '.<',
 'think',
 '>',
 '<|im_end|>',
 'Ċ',
 '<|im_start|>',
 'assistant',
 'Ċ']

I’ll save the indicies of the first and last vision tokens.

(3, 395)

I’ll now sum the attentions for each generated token, over all image tokens.

torch.Size([27, 437])
torch.Size([437])
torch.Size([392])
[0.1649223268032074,
 0.14924326539039612,
 0.24785716831684113,
 0.15989810228347778,
 0.18078750371932983,
 0.19855275750160217,
 0.2221490442752838,
 0.25423598289489746,
 0.33117902278900146,
 0.2534129023551941,
 0.23635193705558777,
 0.27973249554634094,
 0.2829405665397644,
 0.2884763479232788,
 0.20834580063819885,
 0.13748674094676971,
 0.19547797739505768,
 0.20929457247257233,
 0.2825366258621216,
 0.24532952904701233,
 0.19346649944782257,
 0.1775532066822052,
 0.16503523290157318,
 0.15991057455539703,
 0.19774067401885986,
 0.1700247824192047,
 0.10200894623994827]

Now I can visualize the sum of all attentions on all image tokens, as each new token was generated.

Next up, I want to visualize the attention weights of the vision encoder.

Qwen2_5_VLVisionAttention(
  (qkv): Linear(in_features=1280, out_features=3840, bias=True)
  (proj): Linear(in_features=1280, out_features=1280, bias=True)
)

The vision encoder uses SPDA attention. This implementation uses C++/CUDA under the hood, so I’ll have to use a manual implementation instead to be able to access the weights.

'eager'
<bound method Module.named_modules of Qwen2_5_VisionTransformerPretrainedModel(
  (patch_embed): Qwen2_5_VisionPatchEmbed(
    (proj): Conv3d(3, 1280, kernel_size=(2, 14, 14), stride=(2, 14, 14), bias=False)
  )
  (rotary_pos_emb): Qwen2_5_VisionRotaryEmbedding()
  (blocks): ModuleList(
    (0-31): 32 x Qwen2_5_VLVisionBlock(
      (norm1): Qwen2RMSNorm((1280,), eps=1e-06)
      (norm2): Qwen2RMSNorm((1280,), eps=1e-06)
      (attn): Qwen2_5_VLVisionAttention(
        (qkv): Linear(in_features=1280, out_features=3840, bias=True)
        (proj): Linear(in_features=1280, out_features=1280, bias=True)
      )
      (mlp): Qwen2_5_VLMLP(
        (gate_proj): Linear(in_features=1280, out_features=3420, bias=True)
        (up_proj): Linear(in_features=1280, out_features=3420, bias=True)
        (down_proj): Linear(in_features=3420, out_features=1280, bias=True)
        (act_fn): SiLU()
      )
    )
  )
  (merger): Qwen2_5_VLPatchMerger(
    (ln_q): Qwen2RMSNorm((1280,), eps=1e-06)
    (mlp): Sequential(
      (0): Linear(in_features=5120, out_features=5120, bias=True)
      (1): GELU(approximate='none')
      (2): Linear(in_features=5120, out_features=2048, bias=True)
    )
  )
)>
Traceback (most recent call last):
  File "/tmp/ipykernel_265136/256214625.py", line 5, in suplog
    try: yield
  File "/tmp/ipykernel_265136/862200727.py", line 2, in <module>
    with t.no_grad(): x = vlm.visual(inps.pixel_values, inps.image_grid_thw, output_attentions=True)
  File "/home/data/Salman/miniforge3/envs/robobrain2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/home/data/Salman/miniforge3/envs/robobrain2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
TypeError: Qwen2_5_VisionTransformerPretrainedModel.forward() got an unexpected keyword argument 'output_attentions'

The output_attentions parameter doesn’t exist.

<bound method Module.named_modules of Qwen2_5_VisionTransformerPretrainedModel(
  (patch_embed): Qwen2_5_VisionPatchEmbed(
    (proj): Conv3d(3, 1280, kernel_size=(2, 14, 14), stride=(2, 14, 14), bias=False)
  )
  (rotary_pos_emb): Qwen2_5_VisionRotaryEmbedding()
  (blocks): ModuleList(
    (0-31): 32 x Qwen2_5_VLVisionBlock(
      (norm1): Qwen2RMSNorm((1280,), eps=1e-06)
      (norm2): Qwen2RMSNorm((1280,), eps=1e-06)
      (attn): Qwen2_5_VLVisionAttention(
        (qkv): Linear(in_features=1280, out_features=3840, bias=True)
        (proj): Linear(in_features=1280, out_features=1280, bias=True)
      )
      (mlp): Qwen2_5_VLMLP(
        (gate_proj): Linear(in_features=1280, out_features=3420, bias=True)
        (up_proj): Linear(in_features=1280, out_features=3420, bias=True)
        (down_proj): Linear(in_features=3420, out_features=1280, bias=True)
        (act_fn): SiLU()
      )
    )
  )
  (merger): Qwen2_5_VLPatchMerger(
    (ln_q): Qwen2RMSNorm((1280,), eps=1e-06)
    (mlp): Sequential(
      (0): Linear(in_features=5120, out_features=5120, bias=True)
      (1): GELU(approximate='none')
      (2): Linear(in_features=5120, out_features=2048, bias=True)
    )
  )
)>
Signature:       vlm.visual.blocks[0].attn(*args, **kwargs)

Type:            Qwen2_5_VLVisionAttention

String form:    

Qwen2_5_VLVisionAttention(

  (qkv): Linear(in_features=1280, out_features=3840, bias=True)

  (proj): Linear(in_features=1280, out_features=1280, bias=True)

)

File:            ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Source:         

class Qwen2_5_VLVisionAttention(nn.Module):

    def __init__(self, dim: int, num_heads: int = 16) -> None:

        super().__init__()

        self.num_heads = num_heads

        self.head_dim = dim // num_heads

        self.qkv = nn.Linear(dim, dim * 3, bias=True)

        self.proj = nn.Linear(dim, dim)



    def forward(

        self,

        hidden_states: torch.Tensor,

        cu_seqlens: torch.Tensor,

        rotary_pos_emb: Optional[torch.Tensor] = None,

        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

    ) -> torch.Tensor:

        seq_length = hidden_states.shape[0]

        q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

        if position_embeddings is None:

            logger.warning_once(

                "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

                "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

                "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

                "removed and `position_embeddings` will be mandatory."

            )

            emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

            cos = emb.cos()

            sin = emb.sin()

        else:

            cos, sin = position_embeddings

        q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



        attention_mask = torch.full(

            [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

        )

        for i in range(1, len(cu_seqlens)):

            attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



        q = q.transpose(0, 1)

        k = k.transpose(0, 1)

        v = v.transpose(0, 1)

        attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

        attn_weights = attn_weights + attention_mask

        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

        attn_output = torch.matmul(attn_weights, v)

        attn_output = attn_output.transpose(0, 1)

        attn_output = attn_output.reshape(seq_length, -1)

        attn_output = self.proj(attn_output)

        return attn_output

Class docstring:

Base class for all neural network modules.



Your models should also subclass this class.



Modules can also contain other Modules, allowing to nest them in

a tree structure. You can assign the submodules as regular attributes::



    import torch.nn as nn

    import torch.nn.functional as F



    class Model(nn.Module):

        def __init__(self) -> None:

            super().__init__()

            self.conv1 = nn.Conv2d(1, 20, 5)

            self.conv2 = nn.Conv2d(20, 20, 5)



        def forward(self, x):

            x = F.relu(self.conv1(x))

            return F.relu(self.conv2(x))



Submodules assigned in this way will be registered, and will have their

parameters converted too when you call :meth:`to`, etc.



.. note::

    As per the example above, an ``__init__()`` call to the parent class

    must be made before assignment on the child.



:ivar training: Boolean represents whether this module is in training or

                evaluation mode.

:vartype training: bool

Init docstring:  Initialize internal Module state, shared by both nn.Module and ScriptModule.
`loss_type=None` was set in the config but it is unrecognised.Using the default loss: `ForCausalLMLoss`.

Attaching a forward hook to this module will probably be the most straightforward method.

attaching hook to blocks.0.attn
attaching hook to blocks.1.attn
attaching hook to blocks.2.attn
attaching hook to blocks.3.attn
attaching hook to blocks.4.attn
attaching hook to blocks.5.attn
attaching hook to blocks.6.attn
attaching hook to blocks.7.attn
attaching hook to blocks.8.attn
attaching hook to blocks.9.attn
attaching hook to blocks.10.attn
attaching hook to blocks.11.attn
attaching hook to blocks.12.attn
attaching hook to blocks.13.attn
attaching hook to blocks.14.attn
attaching hook to blocks.15.attn
attaching hook to blocks.16.attn
attaching hook to blocks.17.attn
attaching hook to blocks.18.attn
attaching hook to blocks.19.attn
attaching hook to blocks.20.attn
attaching hook to blocks.21.attn
attaching hook to blocks.22.attn
attaching hook to blocks.23.attn
attaching hook to blocks.24.attn
attaching hook to blocks.25.attn
attaching hook to blocks.26.attn
attaching hook to blocks.27.attn
attaching hook to blocks.28.attn
attaching hook to blocks.29.attn
attaching hook to blocks.30.attn
attaching hook to blocks.31.attn
Forward hooks found on module: blocks.0.attn -> OrderedDict([(0, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.1.attn -> OrderedDict([(1, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.2.attn -> OrderedDict([(2, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.3.attn -> OrderedDict([(3, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.4.attn -> OrderedDict([(4, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.5.attn -> OrderedDict([(5, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.6.attn -> OrderedDict([(6, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.7.attn -> OrderedDict([(7, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.8.attn -> OrderedDict([(8, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.9.attn -> OrderedDict([(9, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.10.attn -> OrderedDict([(10, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.11.attn -> OrderedDict([(11, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.12.attn -> OrderedDict([(12, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.13.attn -> OrderedDict([(13, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.14.attn -> OrderedDict([(14, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.15.attn -> OrderedDict([(15, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.16.attn -> OrderedDict([(16, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.17.attn -> OrderedDict([(17, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.18.attn -> OrderedDict([(18, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.19.attn -> OrderedDict([(19, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.20.attn -> OrderedDict([(20, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.21.attn -> OrderedDict([(21, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.22.attn -> OrderedDict([(22, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.23.attn -> OrderedDict([(23, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.24.attn -> OrderedDict([(24, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.25.attn -> OrderedDict([(25, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.26.attn -> OrderedDict([(26, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.27.attn -> OrderedDict([(27, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.28.attn -> OrderedDict([(28, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.29.attn -> OrderedDict([(29, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.30.attn -> OrderedDict([(30, <function get_attn_hook at 0x7dae5c114ca0>)])
Forward hooks found on module: blocks.31.attn -> OrderedDict([(31, <function get_attn_hook at 0x7dae5c114ca0>)])
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================
============================================================
MODULE: Qwen2_5_VLVisionAttention
Module attrs: ['T_destination', 'add_module', 'apply', 'bfloat16', 'buffers', 'call_super_init', 'children', 'compile', 'cpu', 'cuda', 'double', 'dump_patches', 'eval', 'extra_repr', 'float', 'forward', 'get_buffer', 'get_extra_state', 'get_parameter', 'get_submodule', 'half', 'head_dim', 'ipu', 'load_state_dict', 'modules', 'mtia', 'named_buffers', 'named_children', 'named_modules', 'named_parameters', 'num_heads', 'parameters', 'proj', 'qkv', 'register_backward_hook', 'register_buffer', 'register_forward_hook', 'register_forward_pre_hook', 'register_full_backward_hook', 'register_full_backward_pre_hook', 'register_load_state_dict_post_hook', 'register_load_state_dict_pre_hook', 'register_module', 'register_parameter', 'register_state_dict_post_hook', 'register_state_dict_pre_hook', 'requires_grad_', 'set_extra_state', 'set_submodule', 'share_memory', 'state_dict', 'to', 'to_empty', 'train', 'training', 'type', 'xpu', 'zero_grad']
------------------------------
INPUT: <class 'tuple'> | Shape: torch.Size([1564, 1280])
Input[0] attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
------------------------------
OUTPUT: <class 'torch.Tensor'> | Shape: torch.Size([1564, 1280])
Output attrs: ['H', 'T', 'abs', 'abs_', 'absolute', 'absolute_', 'acos', 'acos_', 'acosh', 'acosh_', 'add', 'add_', 'addbmm', 'addbmm_', 'addcdiv', 'addcdiv_', 'addcmul', 'addcmul_', 'addmm', 'addmm_', 'addmv', 'addmv_', 'addr', 'addr_', 'adjoint', 'align_as', 'align_to', 'all', 'allclose', 'amax', 'amin', 'aminmax', 'angle', 'any', 'apply_', 'arccos', 'arccos_', 'arccosh', 'arccosh_', 'arcsin', 'arcsin_', 'arcsinh', 'arcsinh_', 'arctan', 'arctan2', 'arctan2_', 'arctan_', 'arctanh', 'arctanh_', 'argmax', 'argmin', 'argsort', 'argwhere', 'as_strided', 'as_strided_', 'as_strided_scatter', 'as_subclass', 'asin', 'asin_', 'asinh', 'asinh_', 'atan', 'atan2', 'atan2_', 'atan_', 'atanh', 'atanh_', 'backward', 'baddbmm', 'baddbmm_', 'bernoulli', 'bernoulli_', 'bfloat16', 'bincount', 'bitwise_and', 'bitwise_and_', 'bitwise_left_shift', 'bitwise_left_shift_', 'bitwise_not', 'bitwise_not_', 'bitwise_or', 'bitwise_or_', 'bitwise_right_shift', 'bitwise_right_shift_', 'bitwise_xor', 'bitwise_xor_', 'bmm', 'bool', 'broadcast_to', 'byte', 'cauchy_', 'ccol_indices', 'cdouble', 'ceil', 'ceil_', 'cfloat', 'chalf', 'char', 'cholesky', 'cholesky_inverse', 'cholesky_solve', 'chunk', 'clamp', 'clamp_', 'clamp_max', 'clamp_max_', 'clamp_min', 'clamp_min_', 'clip', 'clip_', 'clone', 'coalesce', 'col_indices', 'conj', 'conj_physical', 'conj_physical_', 'contiguous', 'copy_', 'copysign', 'copysign_', 'corrcoef', 'cos', 'cos_', 'cosh', 'cosh_', 'count_nonzero', 'cov', 'cpu', 'cross', 'crow_indices', 'cuda', 'cummax', 'cummin', 'cumprod', 'cumprod_', 'cumsum', 'cumsum_', 'data', 'data_ptr', 'deg2rad', 'deg2rad_', 'dense_dim', 'dequantize', 'det', 'detach', 'detach_', 'device', 'diag', 'diag_embed', 'diagflat', 'diagonal', 'diagonal_scatter', 'diff', 'digamma', 'digamma_', 'dim', 'dim_order', 'dist', 'div', 'div_', 'divide', 'divide_', 'dot', 'double', 'dsplit', 'dtype', 'eig', 'element_size', 'eq', 'eq_', 'equal', 'erf', 'erf_', 'erfc', 'erfc_', 'erfinv', 'erfinv_', 'exp', 'exp2', 'exp2_', 'exp_', 'expand', 'expand_as', 'expm1', 'expm1_', 'exponential_', 'fill_', 'fill_diagonal_', 'fix', 'fix_', 'flatten', 'flip', 'fliplr', 'flipud', 'float', 'float_power', 'float_power_', 'floor', 'floor_', 'floor_divide', 'floor_divide_', 'fmax', 'fmin', 'fmod', 'fmod_', 'frac', 'frac_', 'frexp', 'gather', 'gcd', 'gcd_', 'ge', 'ge_', 'geometric_', 'geqrf', 'ger', 'get_device', 'grad', 'grad_fn', 'greater', 'greater_', 'greater_equal', 'greater_equal_', 'gt', 'gt_', 'half', 'hardshrink', 'has_names', 'heaviside', 'heaviside_', 'histc', 'histogram', 'hsplit', 'hypot', 'hypot_', 'i0', 'i0_', 'igamma', 'igamma_', 'igammac', 'igammac_', 'imag', 'index_add', 'index_add_', 'index_copy', 'index_copy_', 'index_fill', 'index_fill_', 'index_put', 'index_put_', 'index_reduce', 'index_reduce_', 'index_select', 'indices', 'inner', 'int', 'int_repr', 'inverse', 'ipu', 'is_coalesced', 'is_complex', 'is_conj', 'is_contiguous', 'is_cpu', 'is_cuda', 'is_distributed', 'is_floating_point', 'is_inference', 'is_ipu', 'is_leaf', 'is_maia', 'is_meta', 'is_mkldnn', 'is_mps', 'is_mtia', 'is_neg', 'is_nested', 'is_nonzero', 'is_pinned', 'is_quantized', 'is_same_size', 'is_set_to', 'is_shared', 'is_signed', 'is_sparse', 'is_sparse_csr', 'is_vulkan', 'is_xla', 'is_xpu', 'isclose', 'isfinite', 'isinf', 'isnan', 'isneginf', 'isposinf', 'isreal', 'istft', 'item', 'itemsize', 'kron', 'kthvalue', 'layout', 'lcm', 'lcm_', 'ldexp', 'ldexp_', 'le', 'le_', 'lerp', 'lerp_', 'less', 'less_', 'less_equal', 'less_equal_', 'lgamma', 'lgamma_', 'log', 'log10', 'log10_', 'log1p', 'log1p_', 'log2', 'log2_', 'log_', 'log_normal_', 'log_softmax', 'logaddexp', 'logaddexp2', 'logcumsumexp', 'logdet', 'logical_and', 'logical_and_', 'logical_not', 'logical_not_', 'logical_or', 'logical_or_', 'logical_xor', 'logical_xor_', 'logit', 'logit_', 'logsumexp', 'long', 'lstsq', 'lt', 'lt_', 'lu', 'lu_solve', 'mH', 'mT', 'map2_', 'map_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'matrix_exp', 'matrix_power', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mode', 'module_load', 'moveaxis', 'movedim', 'msort', 'mtia', 'mul', 'mul_', 'multinomial', 'multiply', 'multiply_', 'mv', 'mvlgamma', 'mvlgamma_', 'name', 'names', 'nan_to_num', 'nan_to_num_', 'nanmean', 'nanmedian', 'nanquantile', 'nansum', 'narrow', 'narrow_copy', 'nbytes', 'ndim', 'ndimension', 'ne', 'ne_', 'neg', 'neg_', 'negative', 'negative_', 'nelement', 'new', 'new_empty', 'new_empty_strided', 'new_full', 'new_ones', 'new_tensor', 'new_zeros', 'nextafter', 'nextafter_', 'nonzero', 'nonzero_static', 'norm', 'normal_', 'not_equal', 'not_equal_', 'numel', 'numpy', 'orgqr', 'ormqr', 'outer', 'output_nr', 'permute', 'pin_memory', 'pinverse', 'polygamma', 'polygamma_', 'positive', 'pow', 'pow_', 'prelu', 'prod', 'put', 'put_', 'q_per_channel_axis', 'q_per_channel_scales', 'q_per_channel_zero_points', 'q_scale', 'q_zero_point', 'qr', 'qscheme', 'quantile', 'rad2deg', 'rad2deg_', 'random_', 'ravel', 'real', 'reciprocal', 'reciprocal_', 'record_stream', 'refine_names', 'register_hook', 'register_post_accumulate_grad_hook', 'reinforce', 'relu', 'relu_', 'remainder', 'remainder_', 'rename', 'rename_', 'renorm', 'renorm_', 'repeat', 'repeat_interleave', 'requires_grad', 'requires_grad_', 'reshape', 'reshape_as', 'resize', 'resize_', 'resize_as', 'resize_as_', 'resize_as_sparse_', 'resolve_conj', 'resolve_neg', 'retain_grad', 'retains_grad', 'roll', 'rot90', 'round', 'round_', 'row_indices', 'rsqrt', 'rsqrt_', 'scatter', 'scatter_', 'scatter_add', 'scatter_add_', 'scatter_reduce', 'scatter_reduce_', 'select', 'select_scatter', 'set_', 'sgn', 'sgn_', 'shape', 'share_memory_', 'short', 'sigmoid', 'sigmoid_', 'sign', 'sign_', 'signbit', 'sin', 'sin_', 'sinc', 'sinc_', 'sinh', 'sinh_', 'size', 'slice_inverse', 'slice_scatter', 'slogdet', 'smm', 'softmax', 'solve', 'sort', 'sparse_dim', 'sparse_mask', 'sparse_resize_', 'sparse_resize_and_clear_', 'split', 'split_with_sizes', 'sqrt', 'sqrt_', 'square', 'square_', 'squeeze', 'squeeze_', 'sspaddmm', 'std', 'stft', 'storage', 'storage_offset', 'storage_type', 'stride', 'sub', 'sub_', 'subtract', 'subtract_', 'sum', 'sum_to_size', 'svd', 'swapaxes', 'swapaxes_', 'swapdims', 'swapdims_', 'symeig', 't', 't_', 'take', 'take_along_dim', 'tan', 'tan_', 'tanh', 'tanh_', 'tensor_split', 'tile', 'to', 'to_dense', 'to_mkldnn', 'to_padded_tensor', 'to_sparse', 'to_sparse_bsc', 'to_sparse_bsr', 'to_sparse_coo', 'to_sparse_csc', 'to_sparse_csr', 'tolist', 'topk', 'trace', 'transpose', 'transpose_', 'triangular_solve', 'tril', 'tril_', 'triu', 'triu_', 'true_divide', 'true_divide_', 'trunc', 'trunc_', 'type', 'type_as', 'unbind', 'unflatten', 'unfold', 'uniform_', 'unique', 'unique_consecutive', 'unsafe_chunk', 'unsafe_split', 'unsafe_split_with_sizes', 'unsqueeze', 'unsqueeze_', 'untyped_storage', 'values', 'var', 'vdot', 'view', 'view_as', 'vsplit', 'where', 'xlogy', 'xlogy_', 'xpu', 'zero_']
****************************** CHILDREN ******************************
  └── qkv: Linear
  └── proj: Linear
============================================================

The output is a 1564x1280 tensor. There are (644/14)·(476/14)=46·34=1564 patches. The reason why this doesn’t match the number of image tokens produced by the tokenizer is that these 1564 patches haven’t been passed through the projection layer yet.

The Qwen 2.5 VL projection layer combines patches into 2x2 blocks. This will result in, in this case, (46/2)·(34/2)=392. This matches the number of image tokens.

(32, torch.Size([1564, 1280]))

Adding a hook doesn’t work as I only get the output feature vectors back, and not the intermediate attention weights. I need to monkey patch the forward method so it captures the weights.

Signature:       vlm.visual(*args, **kwargs)

Type:            Qwen2_5_VisionTransformerPretrainedModel

String form:    

Qwen2_5_VisionTransformerPretrainedModel(

  (patch_embed): Qwen2_5_VisionPatchEmbed(

    (proj): Conv3d(3, 1280, kernel_size=(2, 14, 14), stride=(2, 14, 14), bias=False)

  )

  (rotary_pos_emb): Qwen2_5_VisionRotaryEmbedding()

  (blocks): ModuleList(

    (0-31): 32 x Qwen2_5_VLVisionBlock(

      (norm1): Qwen2RMSNorm((1280,), eps=1e-06)

      (norm2): Qwen2RMSNorm((1280,), eps=1e-06)

      (attn): Qwen2_5_VLVisionAttention(

        (qkv): Linear(in_features=1280, out_features=3840, bias=True)

        (proj): Linear(in_features=1280, out_features=1280, bias=True)

      )

      (mlp): Qwen2_5_VLMLP(

        (gate_proj): Linear(in_features=1280, out_features=3420, bias=True)

        (up_proj): Linear(in_features=1280, out_features=3420, bias=True)

        (down_proj): Linear(in_features=3420, out_features=1280, bias=True)

        (act_fn): SiLU()

      )

    )

  )

  (merger): Qwen2_5_VLPatchMerger(

    (ln_q): Qwen2RMSNorm((1280,), eps=1e-06)

    (mlp): Sequential(

      (0): Linear(in_features=5120, out_features=5120, bias=True)

      (1): GELU(approximate='none')

      (2): Linear(in_features=5120, out_features=2048, bias=True)

    )

  )

)

File:            ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Source:         

class Qwen2_5_VisionTransformerPretrainedModel(Qwen2_5_VLPreTrainedModel):

    config_class = Qwen2_5_VLVisionConfig

    _no_split_modules = ["Qwen2_5_VLVisionBlock"]



    def __init__(self, config, *inputs, **kwargs) -> None:

        super().__init__(config, *inputs, **kwargs)

        self.spatial_merge_size = config.spatial_merge_size

        self.patch_size = config.patch_size

        self.fullatt_block_indexes = config.fullatt_block_indexes

        self.window_size = config.window_size

        self.spatial_merge_unit = self.spatial_merge_size * self.spatial_merge_size



        self.patch_embed = Qwen2_5_VisionPatchEmbed(

            patch_size=config.patch_size,

            temporal_patch_size=config.temporal_patch_size,

            in_channels=config.in_channels,

            embed_dim=config.hidden_size,

        )



        head_dim = config.hidden_size // config.num_heads

        self.rotary_pos_emb = Qwen2_5_VisionRotaryEmbedding(head_dim // 2)



        self.blocks = nn.ModuleList(

            [Qwen2_5_VLVisionBlock(config, config._attn_implementation) for _ in range(config.depth)]

        )

        self.merger = Qwen2_5_VLPatchMerger(

            dim=config.out_hidden_size,

            context_dim=config.hidden_size,

            spatial_merge_size=config.spatial_merge_size,

        )

        self.gradient_checkpointing = False



    def rot_pos_emb(self, grid_thw):

        pos_ids = []

        for t, h, w in grid_thw:

            hpos_ids = torch.arange(h).unsqueeze(1).expand(-1, w)

            hpos_ids = hpos_ids.reshape(

                h // self.spatial_merge_size,

                self.spatial_merge_size,

                w // self.spatial_merge_size,

                self.spatial_merge_size,

            )

            hpos_ids = hpos_ids.permute(0, 2, 1, 3)

            hpos_ids = hpos_ids.flatten()



            wpos_ids = torch.arange(w).unsqueeze(0).expand(h, -1)

            wpos_ids = wpos_ids.reshape(

                h // self.spatial_merge_size,

                self.spatial_merge_size,

                w // self.spatial_merge_size,

                self.spatial_merge_size,

            )

            wpos_ids = wpos_ids.permute(0, 2, 1, 3)

            wpos_ids = wpos_ids.flatten()

            pos_ids.append(torch.stack([hpos_ids, wpos_ids], dim=-1).repeat(t, 1))

        pos_ids = torch.cat(pos_ids, dim=0)

        max_grid_size = grid_thw[:, 1:].max()

        rotary_pos_emb_full = self.rotary_pos_emb(max_grid_size)

        rotary_pos_emb = rotary_pos_emb_full[pos_ids].flatten(1)

        return rotary_pos_emb



    def get_window_index(self, grid_thw):

        window_index: list = []

        cu_window_seqlens: list = [0]

        window_index_id = 0

        vit_merger_window_size = self.window_size // self.spatial_merge_size // self.patch_size



        for grid_t, grid_h, grid_w in grid_thw:

            llm_grid_h, llm_grid_w = (

                grid_h // self.spatial_merge_size,

                grid_w // self.spatial_merge_size,

            )

            index = torch.arange(grid_t * llm_grid_h * llm_grid_w).reshape(grid_t, llm_grid_h, llm_grid_w)

            pad_h = vit_merger_window_size - llm_grid_h % vit_merger_window_size

            pad_w = vit_merger_window_size - llm_grid_w % vit_merger_window_size

            num_windows_h = (llm_grid_h + pad_h) // vit_merger_window_size

            num_windows_w = (llm_grid_w + pad_w) // vit_merger_window_size

            index_padded = F.pad(index, (0, pad_w, 0, pad_h), "constant", -100)

            index_padded = index_padded.reshape(

                grid_t,

                num_windows_h,

                vit_merger_window_size,

                num_windows_w,

                vit_merger_window_size,

            )

            index_padded = index_padded.permute(0, 1, 3, 2, 4).reshape(

                grid_t,

                num_windows_h * num_windows_w,

                vit_merger_window_size,

                vit_merger_window_size,

            )

            seqlens = (index_padded != -100).sum([2, 3]).reshape(-1)

            index_padded = index_padded.reshape(-1)

            index_new = index_padded[index_padded != -100]

            window_index.append(index_new + window_index_id)

            cu_seqlens_tmp = seqlens.cumsum(0) * self.spatial_merge_unit + cu_window_seqlens[-1]

            cu_window_seqlens.extend(cu_seqlens_tmp.tolist())

            window_index_id += (grid_t * llm_grid_h * llm_grid_w).item()

        window_index = torch.cat(window_index, dim=0)



        return window_index, cu_window_seqlens



    def forward(self, hidden_states: torch.Tensor, grid_thw: torch.Tensor) -> torch.Tensor:

        """

        Args:

            hidden_states (`torch.Tensor` of shape `(seq_len, hidden_size)`):

                The final hidden states of the model.

            grid_thw (`torch.Tensor` of shape `(num_images_or_videos, 3)`):

                The temporal, height and width of feature shape of each image in LLM.



        Returns:

            `torch.Tensor`: hidden_states.

        """

        hidden_states = self.patch_embed(hidden_states)

        rotary_pos_emb = self.rot_pos_emb(grid_thw)

        window_index, cu_window_seqlens = self.get_window_index(grid_thw)

        cu_window_seqlens = torch.tensor(

            cu_window_seqlens,

            device=hidden_states.device,

            dtype=grid_thw.dtype if torch.jit.is_tracing() else torch.int32,

        )

        cu_window_seqlens = torch.unique_consecutive(cu_window_seqlens)



        seq_len, _ = hidden_states.size()

        hidden_states = hidden_states.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)

        hidden_states = hidden_states[window_index, :, :]

        hidden_states = hidden_states.reshape(seq_len, -1)

        rotary_pos_emb = rotary_pos_emb.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)

        rotary_pos_emb = rotary_pos_emb[window_index, :, :]

        rotary_pos_emb = rotary_pos_emb.reshape(seq_len, -1)

        emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

        position_embeddings = (emb.cos(), emb.sin())



        cu_seqlens = torch.repeat_interleave(grid_thw[:, 1] * grid_thw[:, 2], grid_thw[:, 0]).cumsum(

            dim=0,

            # Select dtype based on the following factors:

            #  - FA2 requires that cu_seqlens_q must have dtype int32

            #  - torch.onnx.export requires that cu_seqlens_q must have same dtype as grid_thw

            # See https://github.com/huggingface/transformers/pull/34852 for more information

            dtype=grid_thw.dtype if torch.jit.is_tracing() else torch.int32,

        )

        cu_seqlens = F.pad(cu_seqlens, (1, 0), value=0)



        for layer_num, blk in enumerate(self.blocks):

            if layer_num in self.fullatt_block_indexes:

                cu_seqlens_now = cu_seqlens

            else:

                cu_seqlens_now = cu_window_seqlens

            if self.gradient_checkpointing and self.training:

                hidden_states = self._gradient_checkpointing_func(

                    blk.__call__, hidden_states, cu_seqlens_now, None, position_embeddings

                )

            else:

                hidden_states = blk(hidden_states, cu_seqlens=cu_seqlens_now, position_embeddings=position_embeddings)



        hidden_states = self.merger(hidden_states)

        reverse_indices = torch.argsort(window_index)

        hidden_states = hidden_states[reverse_indices, :]



        return hidden_states

Class docstring:

The bare Qwen2_5_VL Model outputting raw hidden-states without any specific head on top.

This model inherits from [`PreTrainedModel`]. Check the superclass documentation for the generic methods the

library implements for all its model (such as downloading or saving, resizing the input embeddings, pruning heads

etc.)



This model is also a PyTorch [torch.nn.Module](https://pytorch.org/docs/stable/nn.html#torch.nn.Module) subclass.

Use it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage

and behavior.



Parameters:

    config ([`Qwen2_5_VLConfig`]):

        Model configuration class with all the parameters of the model. Initializing with a config file does not

        load the weights associated with the model, only the configuration. Check out the

        [`~PreTrainedModel.from_pretrained`] method to load the model weights.

Init docstring:  Initialize internal Module state, shared by both nn.Module and ScriptModule.

It seems I’ll need to monkey patch something within the blocks themselves.

Signature:       vlm.visual.blocks[0].attn(*args, **kwargs)

Type:            Qwen2_5_VLVisionAttention

String form:    

Qwen2_5_VLVisionAttention(

  (qkv): Linear(in_features=1280, out_features=3840, bias=True)

  (proj): Linear(in_features=1280, out_features=1280, bias=True)

)

File:            ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Source:         

class Qwen2_5_VLVisionAttention(nn.Module):

    def __init__(self, dim: int, num_heads: int = 16) -> None:

        super().__init__()

        self.num_heads = num_heads

        self.head_dim = dim // num_heads

        self.qkv = nn.Linear(dim, dim * 3, bias=True)

        self.proj = nn.Linear(dim, dim)



    def forward(

        self,

        hidden_states: torch.Tensor,

        cu_seqlens: torch.Tensor,

        rotary_pos_emb: Optional[torch.Tensor] = None,

        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

    ) -> torch.Tensor:

        seq_length = hidden_states.shape[0]

        q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

        if position_embeddings is None:

            logger.warning_once(

                "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

                "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

                "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

                "removed and `position_embeddings` will be mandatory."

            )

            emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

            cos = emb.cos()

            sin = emb.sin()

        else:

            cos, sin = position_embeddings

        q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



        attention_mask = torch.full(

            [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

        )

        for i in range(1, len(cu_seqlens)):

            attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



        q = q.transpose(0, 1)

        k = k.transpose(0, 1)

        v = v.transpose(0, 1)

        attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

        attn_weights = attn_weights + attention_mask

        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

        attn_output = torch.matmul(attn_weights, v)

        attn_output = attn_output.transpose(0, 1)

        attn_output = attn_output.reshape(seq_length, -1)

        attn_output = self.proj(attn_output)

        return attn_output

Class docstring:

Base class for all neural network modules.



Your models should also subclass this class.



Modules can also contain other Modules, allowing to nest them in

a tree structure. You can assign the submodules as regular attributes::



    import torch.nn as nn

    import torch.nn.functional as F



    class Model(nn.Module):

        def __init__(self) -> None:

            super().__init__()

            self.conv1 = nn.Conv2d(1, 20, 5)

            self.conv2 = nn.Conv2d(20, 20, 5)



        def forward(self, x):

            x = F.relu(self.conv1(x))

            return F.relu(self.conv2(x))



Submodules assigned in this way will be registered, and will have their

parameters converted too when you call :meth:`to`, etc.



.. note::

    As per the example above, an ``__init__()`` call to the parent class

    must be made before assignment on the child.



:ivar training: Boolean represents whether this module is in training or

                evaluation mode.

:vartype training: bool

Init docstring:  Initialize internal Module state, shared by both nn.Module and ScriptModule.

I’ll need to monkey patch the forward method of the attention class.

Signature:

vlm.visual.blocks[0].attn.forward(

    hidden_states: torch.Tensor,

    cu_seqlens: torch.Tensor,

    rotary_pos_emb: Optional[torch.Tensor] = None,

    position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

) -> torch.Tensor

Docstring:

Define the computation performed at every call.



Should be overridden by all subclasses.



.. note::

    Although the recipe for forward pass needs to be defined within

    this function, one should call the :class:`Module` instance afterwards

    instead of this since the former takes care of running the

    registered hooks while the latter silently ignores them.

Source:   

    def forward(

        self,

        hidden_states: torch.Tensor,

        cu_seqlens: torch.Tensor,

        rotary_pos_emb: Optional[torch.Tensor] = None,

        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

    ) -> torch.Tensor:

        seq_length = hidden_states.shape[0]

        q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

        if position_embeddings is None:

            logger.warning_once(

                "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

                "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

                "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

                "removed and `position_embeddings` will be mandatory."

            )

            emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

            cos = emb.cos()

            sin = emb.sin()

        else:

            cos, sin = position_embeddings

        q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



        attention_mask = torch.full(

            [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

        )

        for i in range(1, len(cu_seqlens)):

            attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



        q = q.transpose(0, 1)

        k = k.transpose(0, 1)

        v = v.transpose(0, 1)

        attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

        attn_weights = attn_weights + attention_mask

        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

        attn_output = torch.matmul(attn_weights, v)

        attn_output = attn_output.transpose(0, 1)

        attn_output = attn_output.reshape(seq_length, -1)

        attn_output = self.proj(attn_output)

        return attn_output

File:      ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Type:      method
Init signature: Qwen2_5_VLVisionAttention(dim: int, num_heads: int = 16) -> None

Docstring:     

Base class for all neural network modules.



Your models should also subclass this class.



Modules can also contain other Modules, allowing to nest them in

a tree structure. You can assign the submodules as regular attributes::



    import torch.nn as nn

    import torch.nn.functional as F



    class Model(nn.Module):

        def __init__(self) -> None:

            super().__init__()

            self.conv1 = nn.Conv2d(1, 20, 5)

            self.conv2 = nn.Conv2d(20, 20, 5)



        def forward(self, x):

            x = F.relu(self.conv1(x))

            return F.relu(self.conv2(x))



Submodules assigned in this way will be registered, and will have their

parameters converted too when you call :meth:`to`, etc.



.. note::

    As per the example above, an ``__init__()`` call to the parent class

    must be made before assignment on the child.



:ivar training: Boolean represents whether this module is in training or

                evaluation mode.

:vartype training: bool

Source:        

class Qwen2_5_VLVisionAttention(nn.Module):

    def __init__(self, dim: int, num_heads: int = 16) -> None:

        super().__init__()

        self.num_heads = num_heads

        self.head_dim = dim // num_heads

        self.qkv = nn.Linear(dim, dim * 3, bias=True)

        self.proj = nn.Linear(dim, dim)



    def forward(

        self,

        hidden_states: torch.Tensor,

        cu_seqlens: torch.Tensor,

        rotary_pos_emb: Optional[torch.Tensor] = None,

        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

    ) -> torch.Tensor:

        seq_length = hidden_states.shape[0]

        q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

        if position_embeddings is None:

            logger.warning_once(

                "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

                "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

                "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

                "removed and `position_embeddings` will be mandatory."

            )

            emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

            cos = emb.cos()

            sin = emb.sin()

        else:

            cos, sin = position_embeddings

        q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



        attention_mask = torch.full(

            [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

        )

        for i in range(1, len(cu_seqlens)):

            attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



        q = q.transpose(0, 1)

        k = k.transpose(0, 1)

        v = v.transpose(0, 1)

        attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

        attn_weights = attn_weights + attention_mask

        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

        attn_output = torch.matmul(attn_weights, v)

        attn_output = attn_output.transpose(0, 1)

        attn_output = attn_output.reshape(seq_length, -1)

        attn_output = self.proj(attn_output)

        return attn_output

File:           ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Type:           type

Subclasses:     

Confirming patch has be introduced.

Signature:       vlm.visual.blocks[0].attn(*args, **kwargs)

Type:            Qwen2_5_VLVisionAttention

String form:    

Qwen2_5_VLVisionAttention(

  (qkv): Linear(in_features=1280, out_features=3840, bias=True)

  (proj): Linear(in_features=1280, out_features=1280, bias=True)

)

File:            ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py

Source:         

class Qwen2_5_VLVisionAttention(nn.Module):

    def __init__(self, dim: int, num_heads: int = 16) -> None:

        super().__init__()

        self.num_heads = num_heads

        self.head_dim = dim // num_heads

        self.qkv = nn.Linear(dim, dim * 3, bias=True)

        self.proj = nn.Linear(dim, dim)



    def forward(

        self,

        hidden_states: torch.Tensor,

        cu_seqlens: torch.Tensor,

        rotary_pos_emb: Optional[torch.Tensor] = None,

        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,

    ) -> torch.Tensor:

        seq_length = hidden_states.shape[0]

        q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

        if position_embeddings is None:

            logger.warning_once(

                "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

                "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

                "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

                "removed and `position_embeddings` will be mandatory."

            )

            emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

            cos = emb.cos()

            sin = emb.sin()

        else:

            cos, sin = position_embeddings

        q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



        attention_mask = torch.full(

            [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

        )

        for i in range(1, len(cu_seqlens)):

            attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



        q = q.transpose(0, 1)

        k = k.transpose(0, 1)

        v = v.transpose(0, 1)

        attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

        attn_weights = attn_weights + attention_mask

        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

        attn_output = torch.matmul(attn_weights, v)

        attn_output = attn_output.transpose(0, 1)

        attn_output = attn_output.reshape(seq_length, -1)

        attn_output = self.proj(attn_output)

        return attn_output

Class docstring:

Base class for all neural network modules.



Your models should also subclass this class.



Modules can also contain other Modules, allowing to nest them in

a tree structure. You can assign the submodules as regular attributes::



    import torch.nn as nn

    import torch.nn.functional as F



    class Model(nn.Module):

        def __init__(self) -> None:

            super().__init__()

            self.conv1 = nn.Conv2d(1, 20, 5)

            self.conv2 = nn.Conv2d(20, 20, 5)



        def forward(self, x):

            x = F.relu(self.conv1(x))

            return F.relu(self.conv2(x))



Submodules assigned in this way will be registered, and will have their

parameters converted too when you call :meth:`to`, etc.



.. note::

    As per the example above, an ``__init__()`` call to the parent class

    must be made before assignment on the child.



:ivar training: Boolean represents whether this module is in training or

                evaluation mode.

:vartype training: bool

Init docstring:  Initialize internal Module state, shared by both nn.Module and ScriptModule.
Signature:

vlm.visual.blocks[0].attn.forward(

    hidden_states,

    cu_seqlens,

    rotary_pos_emb,

    position_embeddings,

)

Docstring:

Define the computation performed at every call.



Should be overridden by all subclasses.



.. note::

    Although the recipe for forward pass needs to be defined within

    this function, one should call the :class:`Module` instance afterwards

    instead of this since the former takes care of running the

    registered hooks while the latter silently ignores them.

Source:   

@patch

def forward(self: Qwen2_5_VLVisionAttention, hidden_states, cu_seqlens, rotary_pos_emb, position_embeddings):

  seq_length = hidden_states.shape[0]

  q, k, v = self.qkv(hidden_states).reshape(seq_length, 3, self.num_heads, -1).permute(1, 0, 2, 3).unbind(0)

  if position_embeddings is None:

    logger.warning_once(

      "The attention layers in this model are transitioning from computing the RoPE embeddings internally "

      "through `rotary_pos_emb` (2D tensor of RoPE theta values), to using externally computed "

      "`position_embeddings` (Tuple of tensors, containing cos and sin). In v4.54 `rotary_pos_emb` will be "

      "removed and `position_embeddings` will be mandatory."

    )

    emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)

    cos = emb.cos()

    sin = emb.sin()

  else:

    cos, sin = position_embeddings

  q, k = apply_rotary_pos_emb_vision(q, k, cos, sin)



  attention_mask = torch.full(

    [1, seq_length, seq_length], torch.finfo(q.dtype).min, device=q.device, dtype=q.dtype

  )

  for i in range(1, len(cu_seqlens)):

    attention_mask[..., cu_seqlens[i - 1] : cu_seqlens[i], cu_seqlens[i - 1] : cu_seqlens[i]] = 0



  q = q.transpose(0, 1)

  k = k.transpose(0, 1)

  v = v.transpose(0, 1)

  attn_weights = torch.matmul(q, k.transpose(1, 2)) / math.sqrt(self.head_dim)

  attn_weights = attn_weights + attention_mask

  attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(q.dtype)

  attn_ws.append(attn_weights) #<< Addition

  attn_output = torch.matmul(attn_weights, v)

  attn_output = attn_output.transpose(0, 1)

  attn_output = attn_output.reshape(seq_length, -1)

  attn_output = self.proj(attn_output)

  return attn_output

File:      /tmp/ipykernel_265136/3723538712.py

Type:      method

Let’s give it a run. But first, remove the existing hooks.

(32, torch.Size([16, 1564, 1564]))

It’s worked! The reason why the attention weights is a different shape is that the patches/feature vectors to which the attention weights correspond to haven’t pass through what’s known as the projection/pooling layer.

The RoboBrain model is based on Qwen 2.5 VL. The vision encoder in this model uses a kernel size of 14.

14

The image at the beginning of the notebook has dimensions, after padding, 644x476.

(644, 476)

Therefore, the number of patches produced is (644/14)·(476/14)=46·34=1564. After the vision encoder produces the feature vector for each of these 1564 patches, the vectors are passed through the projection/pooling layer. In the Qwen 2.5 VL model, each 2x2 patch is combined, which results in (46/2)·(34/2)=392 patches. This matches the number of image tokens produced by the tokenizer.

I will now aggregate all the attention weights.

torch.Size([16, 410, 410])

In the reference notebook, a normalization step occurs after taking the mean across all heads. This is because in the LLaVa implementation, the first token is a special token which doens’t need to be visualized. Here, I do not need to perform such processes, since far as I’m aware, there is no special patch token.

If there were, I would have to perform the following.

vec = attns_per_head[1:,1:].cpu()
vec/vec.sum(-1,keepdim=True)
torch.Size([1564, 1564])

Time to visualize. I’ll begin by visualizing the attention of only the first token.

torch.Size([1564, 1564])
437
(#27) [410,411,412,413,414,415,416,417,418,419,420,421,422,423,424,425,426,427,428,429...]
27

The target token is currently the first generated token.

410
torch.Size([437, 437])
torch.Size([437])
((3, 395), torch.Size([392]))

I’ll obtain the attention weights over the vision tokens, and then normalize those weights.

(tensor(3.0008e-05), tensor(0.0208))
(tensor(0.0002), tensor(0.1263))
(torch.Size([392]), torch.Size([1564, 1564]))
(#392) [(tensor(0.0114), tensor([7.2815e-02, 2.7817e-02, 2.7298e-02,  ..., 3.5703e-05, 9.5844e-05,
        1.3089e-04], dtype=torch.float16)),(tensor(0.1002), tensor([5.5511e-02, 3.5126e-02, 2.1759e-02,  ..., 6.6400e-05, 6.7353e-05,
        8.6725e-05], dtype=torch.float16)),(tensor(0.0592), tensor([5.8105e-02, 1.9745e-02, 4.0253e-02,  ..., 4.3392e-05, 5.0604e-05,
        7.5161e-05], dtype=torch.float16)),(tensor(0.1263), tensor([4.4434e-02, 2.7466e-02, 2.9800e-02,  ..., 4.5955e-05, 4.0114e-05,
        5.9068e-05], dtype=torch.float16)),(tensor(0.0875), tensor([4.9561e-02, 2.5208e-02, 1.5625e-02,  ..., 5.1498e-05, 6.5625e-05,
        1.1742e-04], dtype=torch.float16)),(tensor(0.0013), tensor([4.0619e-02, 2.0660e-02, 1.8677e-02,  ..., 1.0329e-04, 8.6665e-05,
        1.9145e-04], dtype=torch.float16)),(tensor(0.0005), tensor([3.3173e-02, 1.9882e-02, 1.4587e-02,  ..., 5.0008e-05, 7.7903e-05,
        9.0420e-05], dtype=torch.float16)),(tensor(0.0036), tensor([1.8967e-02, 1.4717e-02, 1.7731e-02,  ..., 7.9393e-05, 4.9829e-05,
        4.2319e-05], dtype=torch.float16)),(tensor(0.0086), tensor([0.0320, 0.0208, 0.0108,  ..., 0.0001, 0.0001, 0.0002],
       dtype=torch.float16)),(tensor(0.0018), tensor([2.7313e-02, 1.5854e-02, 1.1726e-02,  ..., 1.1587e-04, 1.2165e-04,
        9.3877e-05], dtype=torch.float16)),(tensor(0.0009), tensor([1.7624e-02, 1.1581e-02, 1.1177e-02,  ..., 6.8367e-05, 9.2864e-05,
        1.7357e-04], dtype=torch.float16)),(tensor(0.0008), tensor([1.8936e-02, 1.0399e-02, 1.0033e-02,  ..., 3.6418e-05, 5.7578e-05,
        4.4882e-05], dtype=torch.float16)),(tensor(0.0015), tensor([2.3804e-02, 1.3443e-02, 1.0323e-02,  ..., 3.4571e-05, 4.4703e-05,
        1.3959e-04], dtype=torch.float16)),(tensor(0.0008), tensor([3.4149e-02, 1.6403e-02, 1.2215e-02,  ..., 2.6405e-05, 4.3452e-05,
        5.7697e-05], dtype=torch.float16)),(tensor(0.0015), tensor([2.0523e-02, 1.3504e-02, 1.1162e-02,  ..., 3.9041e-05, 4.8935e-05,
        7.4387e-05], dtype=torch.float16)),(tensor(0.0007), tensor([1.9135e-02, 1.0620e-02, 1.3245e-02,  ..., 2.6762e-05, 5.5313e-05,
        3.9697e-05], dtype=torch.float16)),(tensor(0.0004), tensor([3.3081e-02, 1.1414e-02, 3.4821e-02,  ..., 2.5630e-05, 5.0366e-05,
        7.6592e-05], dtype=torch.float16)),(tensor(0.0005), tensor([2.1408e-02, 1.3779e-02, 2.8259e-02,  ..., 1.3947e-05, 3.1412e-05,
        3.7611e-05], dtype=torch.float16)),(tensor(0.0007), tensor([2.5253e-02, 1.3268e-02, 1.7975e-02,  ..., 8.7857e-05, 1.4389e-04,
        2.6059e-04], dtype=torch.float16)),(tensor(0.0004), tensor([1.5945e-02, 7.9346e-03, 1.0849e-02,  ..., 1.1802e-05, 4.8161e-05,
        3.2246e-05], dtype=torch.float16))...]
((tensor(0.0114),
  tensor([7.2815e-02, 2.7817e-02, 2.7298e-02,  ..., 3.5703e-05, 9.5844e-05,
          1.3089e-04], dtype=torch.float16)),
 torch.Size([1564]))

So what I have here, y, is the LLM attention weights over the image tokens, and the ViT attention weights of the image patches.

(46, 34)

I need to resize the ViT/vision encoder tensors so it matches the number of image tokens. In addition, I’ll also take the product between the resulting reshaped attention tensor and the LLM attention weights to get the final attention over the image.

torch.Size([46, 34])
(392, torch.Size([46, 34]))
torch.Size([46, 34])
Signature:

F.interpolate(

    input: torch.Tensor,

    size: Optional[int] = None,

    scale_factor: Optional[List[float]] = None,

    mode: str = 'nearest',

    align_corners: Optional[bool] = None,

    recompute_scale_factor: Optional[bool] = None,

    antialias: bool = False,

) -> torch.Tensor

Docstring:

Down/up samples the input.



Tensor interpolated to either the given :attr:`size` or the given

:attr:`scale_factor`



The algorithm used for interpolation is determined by :attr:`mode`.



Currently temporal, spatial and volumetric sampling are supported, i.e.

expected inputs are 3-D, 4-D or 5-D in shape.



The input dimensions are interpreted in the form:

`mini-batch x channels x [optional depth] x [optional height] x width`.



The modes available for resizing are: `nearest`, `linear` (3D-only),

`bilinear`, `bicubic` (4D-only), `trilinear` (5D-only), `area`, `nearest-exact`



Args:

    input (Tensor): the input tensor

    size (int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int]):

        output spatial size.

    scale_factor (float or Tuple[float]): multiplier for spatial size. If `scale_factor` is a tuple,

        its length has to match the number of spatial dimensions; `input.dim() - 2`.

    mode (str): algorithm used for upsampling:

        ``'nearest'`` | ``'linear'`` | ``'bilinear'`` | ``'bicubic'`` |

        ``'trilinear'`` | ``'area'`` | ``'nearest-exact'``. Default: ``'nearest'``

    align_corners (bool, optional): Geometrically, we consider the pixels of the

        input and output as squares rather than points.

        If set to ``True``, the input and output tensors are aligned by the

        center points of their corner pixels, preserving the values at the corner pixels.

        If set to ``False``, the input and output tensors are aligned by the corner

        points of their corner pixels, and the interpolation uses edge value padding

        for out-of-boundary values, making this operation *independent* of input size

        when :attr:`scale_factor` is kept the same. This only has an effect when :attr:`mode`

        is ``'linear'``, ``'bilinear'``, ``'bicubic'`` or ``'trilinear'``.

        Default: ``False``

    recompute_scale_factor (bool, optional): recompute the scale_factor for use in the

        interpolation calculation. If `recompute_scale_factor` is ``True``, then

        `scale_factor` must be passed in and `scale_factor` is used to compute the

        output `size`. The computed output `size` will be used to infer new scales for

        the interpolation. Note that when `scale_factor` is floating-point, it may differ

        from the recomputed `scale_factor` due to rounding and precision issues.

        If `recompute_scale_factor` is ``False``, then `size` or `scale_factor` will

        be used directly for interpolation. Default: ``None``.

    antialias (bool, optional): flag to apply anti-aliasing. Default: ``False``. Using anti-alias

        option together with ``align_corners=False``, interpolation result would match Pillow

        result for downsampling operation. Supported modes: ``'bilinear'``, ``'bicubic'``.



.. note::

    With ``mode='bicubic'``, it's possible to cause overshoot, in other words it can produce

    negative values or values greater than 255 for images.

    Explicitly call ``result.clamp(min=0, max=255)`` if you want to reduce the overshoot

    when displaying the image.



.. note::

    Mode ``mode='nearest-exact'`` matches Scikit-Image and PIL nearest neighbours interpolation

    algorithms and fixes known issues with ``mode='nearest'``. This mode is introduced to keep

    backward compatibility.

    Mode ``mode='nearest'`` matches buggy OpenCV's ``INTER_NEAREST`` interpolation algorithm.



.. note::

    The gradients for the dtype ``float16`` on CUDA may be inaccurate in the upsample operation

    when using modes ``['linear', 'bilinear', 'bicubic', 'trilinear', 'area']``.

    For more details, please refer to the discussion in

    `issue#104157 <https://github.com/pytorch/pytorch/issues/104157>`_.



Note:

    This operation may produce nondeterministic gradients when given tensors on a CUDA device. See :doc:`/notes/randomness` for more information.

Source:   

def interpolate(  # noqa: F811

    input: Tensor,

    size: Optional[int] = None,

    scale_factor: Optional[List[float]] = None,

    mode: str = "nearest",

    align_corners: Optional[bool] = None,

    recompute_scale_factor: Optional[bool] = None,

    antialias: bool = False,

) -> Tensor:  # noqa: B950

    r"""Down/up samples the input.



    Tensor interpolated to either the given :attr:`size` or the given

    :attr:`scale_factor`



    The algorithm used for interpolation is determined by :attr:`mode`.



    Currently temporal, spatial and volumetric sampling are supported, i.e.

    expected inputs are 3-D, 4-D or 5-D in shape.



    The input dimensions are interpreted in the form:

    `mini-batch x channels x [optional depth] x [optional height] x width`.



    The modes available for resizing are: `nearest`, `linear` (3D-only),

    `bilinear`, `bicubic` (4D-only), `trilinear` (5D-only), `area`, `nearest-exact`



    Args:

        input (Tensor): the input tensor

        size (int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int]):

            output spatial size.

        scale_factor (float or Tuple[float]): multiplier for spatial size. If `scale_factor` is a tuple,

            its length has to match the number of spatial dimensions; `input.dim() - 2`.

        mode (str): algorithm used for upsampling:

            ``'nearest'`` | ``'linear'`` | ``'bilinear'`` | ``'bicubic'`` |

            ``'trilinear'`` | ``'area'`` | ``'nearest-exact'``. Default: ``'nearest'``

        align_corners (bool, optional): Geometrically, we consider the pixels of the

            input and output as squares rather than points.

            If set to ``True``, the input and output tensors are aligned by the

            center points of their corner pixels, preserving the values at the corner pixels.

            If set to ``False``, the input and output tensors are aligned by the corner

            points of their corner pixels, and the interpolation uses edge value padding

            for out-of-boundary values, making this operation *independent* of input size

            when :attr:`scale_factor` is kept the same. This only has an effect when :attr:`mode`

            is ``'linear'``, ``'bilinear'``, ``'bicubic'`` or ``'trilinear'``.

            Default: ``False``

        recompute_scale_factor (bool, optional): recompute the scale_factor for use in the

            interpolation calculation. If `recompute_scale_factor` is ``True``, then

            `scale_factor` must be passed in and `scale_factor` is used to compute the

            output `size`. The computed output `size` will be used to infer new scales for

            the interpolation. Note that when `scale_factor` is floating-point, it may differ

            from the recomputed `scale_factor` due to rounding and precision issues.

            If `recompute_scale_factor` is ``False``, then `size` or `scale_factor` will

            be used directly for interpolation. Default: ``None``.

        antialias (bool, optional): flag to apply anti-aliasing. Default: ``False``. Using anti-alias

            option together with ``align_corners=False``, interpolation result would match Pillow

            result for downsampling operation. Supported modes: ``'bilinear'``, ``'bicubic'``.



    .. note::

        With ``mode='bicubic'``, it's possible to cause overshoot, in other words it can produce

        negative values or values greater than 255 for images.

        Explicitly call ``result.clamp(min=0, max=255)`` if you want to reduce the overshoot

        when displaying the image.



    .. note::

        Mode ``mode='nearest-exact'`` matches Scikit-Image and PIL nearest neighbours interpolation

        algorithms and fixes known issues with ``mode='nearest'``. This mode is introduced to keep

        backward compatibility.

        Mode ``mode='nearest'`` matches buggy OpenCV's ``INTER_NEAREST`` interpolation algorithm.



    .. note::

        The gradients for the dtype ``float16`` on CUDA may be inaccurate in the upsample operation

        when using modes ``['linear', 'bilinear', 'bicubic', 'trilinear', 'area']``.

        For more details, please refer to the discussion in

        `issue#104157 <https://github.com/pytorch/pytorch/issues/104157>`_.



    Note:

        {backward_reproducibility_note}

    """

    if has_torch_function_unary(input):

        return handle_torch_function(

            interpolate,

            (input,),

            input,

            size=size,

            scale_factor=scale_factor,

            mode=mode,

            align_corners=align_corners,

            recompute_scale_factor=recompute_scale_factor,

            antialias=antialias,

        )



    if mode in ("nearest", "area", "nearest-exact"):

        if align_corners is not None:

            raise ValueError(

                "align_corners option can only be set with the "

                "interpolating modes: linear | bilinear | bicubic | trilinear"

            )

    else:

        if align_corners is None:

            align_corners = False



    dim = input.dim() - 2  # Number of spatial dimensions.



    # Process size and scale_factor.  Validate that exactly one is set.

    # Validate its length if it is a list, or expand it if it is a scalar.

    # After this block, exactly one of output_size and scale_factors will

    # be non-None, and it will be a list (or tuple).

    if size is not None and scale_factor is not None:

        raise ValueError("only one of size or scale_factor should be defined")

    elif size is not None:

        assert scale_factor is None

        scale_factors = None

        if isinstance(size, (list, tuple)):

            if len(size) != dim:

                raise ValueError(

                    "Input and output must have the same number of spatial dimensions, but got "

                    f"input with spatial dimensions of {list(input.shape[2:])} and output size of {size}. "

                    "Please provide input tensor in (N, C, d1, d2, ...,dK) format and "

                    "output size in (o1, o2, ...,oK) format."

                )

            if not torch.jit.is_scripting():

                if not all(_is_integer(x) for x in size):

                    raise TypeError(

                        "expected size to be one of int or Tuple[int] or Tuple[int, int] or "

                        f"Tuple[int, int, int], but got size with types {[type(x) for x in size]}"

                    )

            output_size = size

        else:

            output_size = [size for _ in range(dim)]

    elif scale_factor is not None:

        assert size is None

        output_size = None

        if isinstance(scale_factor, (list, tuple)):

            if len(scale_factor) != dim:

                raise ValueError(

                    "Input and scale_factor must have the same number of spatial dimensions, but "

                    f"got input with spatial dimensions of {list(input.shape[2:])} and "

                    f"scale_factor of shape {scale_factor}. "

                    "Please provide input tensor in (N, C, d1, d2, ...,dK) format and "

                    "scale_factor in (s1, s2, ...,sK) format."

                )

            scale_factors = scale_factor

        else:

            scale_factors = [scale_factor for _ in range(dim)]

    else:

        raise ValueError("either size or scale_factor should be defined")



    if (

        recompute_scale_factor is not None

        and recompute_scale_factor

        and size is not None

    ):

        raise ValueError(

            "recompute_scale_factor is not meaningful with an explicit size."

        )



    # "area" mode always requires an explicit size rather than scale factor.

    # Re-use the recompute_scale_factor code path.

    if mode == "area" and output_size is None:

        recompute_scale_factor = True



    if recompute_scale_factor is not None and recompute_scale_factor:

        # We compute output_size here, then un-set scale_factors.

        # The C++ code will recompute it based on the (integer) output size.

        assert scale_factors is not None

        if not torch.jit.is_scripting() and torch._C._get_tracing_state():

            # make scale_factor a tensor in tracing so constant doesn't get baked in

            output_size = [

                (

                    torch.floor(

                        (

                            input.size(i + 2).float()

                            * torch.tensor(scale_factors[i], dtype=torch.float32)

                        ).float()

                    )

                )

                for i in range(dim)

            ]

        elif torch.jit.is_scripting():

            output_size = [

                int(math.floor(float(input.size(i + 2)) * scale_factors[i]))

                for i in range(dim)

            ]

        else:

            output_size = [

                _sym_int(input.size(i + 2) * scale_factors[i]) for i in range(dim)

            ]

        scale_factors = None



    if antialias and not (mode in ("bilinear", "bicubic") and input.ndim == 4):

        raise ValueError(

            "Anti-alias option is restricted to bilinear and bicubic modes and requires a 4-D tensor as input"

        )



    if input.dim() == 3 and mode == "nearest":

        return torch._C._nn.upsample_nearest1d(input, output_size, scale_factors)

    if input.dim() == 4 and mode == "nearest":

        return torch._C._nn.upsample_nearest2d(input, output_size, scale_factors)

    if input.dim() == 5 and mode == "nearest":

        return torch._C._nn.upsample_nearest3d(input, output_size, scale_factors)



    if input.dim() == 3 and mode == "nearest-exact":

        return torch._C._nn._upsample_nearest_exact1d(input, output_size, scale_factors)

    if input.dim() == 4 and mode == "nearest-exact":

        return torch._C._nn._upsample_nearest_exact2d(input, output_size, scale_factors)

    if input.dim() == 5 and mode == "nearest-exact":

        return torch._C._nn._upsample_nearest_exact3d(input, output_size, scale_factors)



    if input.dim() == 3 and mode == "area":

        assert output_size is not None

        return adaptive_avg_pool1d(input, output_size)

    if input.dim() == 4 and mode == "area":

        assert output_size is not None

        return adaptive_avg_pool2d(input, output_size)

    if input.dim() == 5 and mode == "area":

        assert output_size is not None

        return adaptive_avg_pool3d(input, output_size)



    if input.dim() == 3 and mode == "linear":

        assert align_corners is not None

        return torch._C._nn.upsample_linear1d(

            input, output_size, align_corners, scale_factors

        )

    if input.dim() == 4 and mode == "bilinear":

        assert align_corners is not None

        if antialias:

            return torch._C._nn._upsample_bilinear2d_aa(

                input, output_size, align_corners, scale_factors

            )

        # Two levels are necessary to prevent TorchScript from touching

        # are_deterministic_algorithms_enabled.

        if not torch.jit.is_scripting():

            if torch.are_deterministic_algorithms_enabled() and (

                input.is_cuda or input.is_xpu

            ):

                # Use slow decomp whose backward will be in terms of index_put

                # importlib is required because the import cannot be top level

                # (cycle) and cannot be nested (TS doesn't support)

                return importlib.import_module(

                    "torch._decomp.decompositions"

                )._upsample_linear_vec(input, output_size, align_corners, scale_factors)

        return torch._C._nn.upsample_bilinear2d(

            input, output_size, align_corners, scale_factors

        )

    if input.dim() == 5 and mode == "trilinear":

        assert align_corners is not None

        return torch._C._nn.upsample_trilinear3d(

            input, output_size, align_corners, scale_factors

        )

    if input.dim() == 4 and mode == "bicubic":

        assert align_corners is not None

        if antialias:

            return torch._C._nn._upsample_bicubic2d_aa(

                input, output_size, align_corners, scale_factors

            )

        return torch._C._nn.upsample_bicubic2d(

            input, output_size, align_corners, scale_factors

        )



    if input.dim() == 3 and mode == "bilinear":

        raise NotImplementedError("Got 3D input, but bilinear mode needs 4D input")

    if input.dim() == 3 and mode == "trilinear":

        raise NotImplementedError("Got 3D input, but trilinear mode needs 5D input")

    if input.dim() == 4 and mode == "linear":

        raise NotImplementedError("Got 4D input, but linear mode needs 3D input")

    if input.dim() == 4 and mode == "trilinear":

        raise NotImplementedError("Got 4D input, but trilinear mode needs 5D input")

    if input.dim() == 5 and mode == "linear":

        raise NotImplementedError("Got 5D input, but linear mode needs 3D input")

    if input.dim() == 5 and mode == "bilinear":

        raise NotImplementedError("Got 5D input, but bilinear mode needs 4D input")



    raise NotImplementedError(

        "Input Error: Only 3D, 4D and 5D input Tensors supported"

        f" (got {input.dim()}D) for the modes: nearest | linear | bilinear | bicubic | trilinear | area | nearest-exact"

        f" (got {mode})"

    )

File:      ~/miniforge3/envs/robobrain2/lib/python3.10/site-packages/torch/nn/functional.py

Type:      function

remove period

The reason why I’m interpolating here is to align the resolution of the current attention heatmap. to the original input image. In essence, I’m upscaling the attention heatmap.

torch.Size([1, 1, 46, 34])
torch.Size([1, 1, 644, 476])

mode='nearest' is an operation that could be thought of as a constrast slider when editing photos. more='nearest' prevents the resulting image from looking too flat or smooth.

(644, 476, 3)
(644, 476, 3)

I’ll now overlay the attention mask on the image.

Traceback (most recent call last):
  File "/tmp/ipykernel_265136/256214625.py", line 5, in suplog
    try: yield
  File "/tmp/ipykernel_265136/3629630572.py", line 2, in <module>
    with suplog(): heatmap = cv2.applyColorMap(np.uint8(255*attn_over_img.numpy()), cv2.COLOMAP_HSV)
AttributeError: module 'cv2' has no attribute 'COLOMAP_HSV'
array([[[  0, 247, 250],
        [  0, 247, 250],
        [  0, 247, 250],
        ...,
        [  0,  48, 255],
        [  0,  48, 255],
        [  0,  48, 255]],

       [[  0, 247, 250],
        [  0, 247, 250],
        [  0, 247, 250],
        ...,
        [  0,  48, 255],
        [  0,  48, 255],
        [  0,  48, 255]],

       [[  0, 247, 250],
        [  0, 247, 250],
        [  0, 247, 250],
        ...,
        [  0,  48, 255],
        [  0,  48, 255],
        [  0,  48, 255]],

       ...,

       [[  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        ...,
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255]],

       [[  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        ...,
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255]],

       [[  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        ...,
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255]]], dtype=uint8)
((644, 476, 3), numpy.ndarray, dtype('uint64'))
Traceback (most recent call last):
  File "/home/data/Salman/miniforge3/envs/robobrain2/lib/python3.10/site-packages/PIL/Image.py", line 3311, in fromarray
    mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 3), '<u8')

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/tmp/ipykernel_265136/256214625.py", line 5, in suplog
    try: yield
  File "/tmp/ipykernel_265136/844388640.py", line 2, in <module>
    with suplog(): PILImage.fromarray(img_with_attn)
  File "/home/data/Salman/miniforge3/envs/robobrain2/lib/python3.10/site-packages/PIL/Image.py", line 3315, in fromarray
    raise TypeError(msg) from e
TypeError: Cannot handle this data type: (1, 1, 3), <u8

Now I’ll visualize the map for all tokens.

1.3529411764705883
27
4

PIL.Image.Image
/tmp/ipykernel_265136/951595502.py:26: RuntimeWarning: invalid value encountered in cast
  hm = cv2.applyColorMap(np.uint8(255*attn_over_img.cpu().numpy()), cv2.COLORMAP_HSV)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)
(476, 644, 3)

Back to top